// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/cronet/android/io_buffer_with_byte_buffer.h"

#include "base/android/jni_bytebuffer.h"
#include "base/check_op.h"
#include "base/compiler_specific.h"
#include "base/numerics/safe_conversions.h"

namespace cronet {

IOBufferWithByteBuffer::IOBufferWithByteBuffer(
    JNIEnv* env,
    const base::android::JavaRef<jobject>& jbyte_buffer,
    int32_t position,
    int32_t limit)
    : net::WrappedIOBuffer(
          base::android::JavaByteBufferToMutableSpan(env, jbyte_buffer)
              .subspan(base::checked_cast<size_t>(position),
                       base::checked_cast<size_t>(limit - position))),
      byte_buffer_(env, jbyte_buffer),
      initial_position_(position),
      initial_limit_(limit) {}

IOBufferWithByteBuffer::~IOBufferWithByteBuffer() = default;

ByteBufferWithIOBuffer::ByteBufferWithIOBuffer(
    JNIEnv* env,
    scoped_refptr<net::IOBuffer> io_buffer,
    int io_buffer_len)
    : io_buffer_(std::move(io_buffer)), io_buffer_len_(io_buffer_len) {
  // An intermediate ScopedJavaLocalRef is needed here to release the local
  // reference created by env->NewDirectByteBuffer().
  auto java_buffer = jni_zero::AdoptRef(
      env, env->NewDirectByteBuffer(io_buffer_->data(), io_buffer_len_));
  byte_buffer_.Reset(env, java_buffer);
}

ByteBufferWithIOBuffer::~ByteBufferWithIOBuffer() = default;

}  // namespace cronet