[android_webview] Don't block the IO thread when reading from an InputStream.

android_webview/native/input_stream.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "android_webview/native/input_stream.h"
+
+#include "base/android/jni_android.h"
+#include "net/base/io_buffer.h"
+// Disable "Warnings treated as errors" for input_stream_jni as it's a Java
+// system class and we have to generate C++ hooks for all methods in the class
+// even if they're unused.
+#pragma GCC diagnostic ignored "-Wunused-function"
+#include "jni/InputStream_jni.h"
+
+using base::android::ClearException;
+using base::android::JavaRef;
+using JNI_InputStream::Java_InputStream_available;
+using JNI_InputStream::Java_InputStream_skip;
+using JNI_InputStream::Java_InputStream_readI_AB_I_I;
+
+namespace android_webview {
+
+bool RegisterInputStream(JNIEnv* env) {
+  return JNI_InputStream::RegisterNativesImpl(env);
+}
+
+// Maximum number of bytes to be read in a single read.
+const int InputStream::kBufferSize = 4096;
+
+InputStream::InputStream() {
+}
+
+InputStream::InputStream(const JavaRef<jobject>& stream)
+    : jobject_(stream) {
+  DCHECK(!stream.is_null());
+}
+
+InputStream::~InputStream() {
+}
+
+bool InputStream::BytesAvailable(JNIEnv* env,
+                                 int* bytes_available) const {
+  // TODO: Use unsafe version once BUG 157880 is fixed.
+  int bytes = Java_InputStream_available(env, jobject_.obj());
+  if (ClearException(env))
+    return false;
+  *bytes_available = bytes;
+  return true;
+}
+
+bool InputStream::Skip(JNIEnv* env,
+                       int64_t n,
+                       int64_t* bytes_skipped) {
+  // TODO: Use unsafe version once BUG 157880 is fixed.
+  int bytes = Java_InputStream_skip(env, jobject_.obj(), n);
+  if (ClearException(env))
+    return false;
+  *bytes_skipped = bytes;
+  return true;
+}
+
+bool InputStream::Read(JNIEnv* env,
+                       net::IOBuffer* dest,
+                       int length,
+                       int* bytes_read) {
+  if (!buffer_.obj()) {
+    // Allocate transfer buffer.
+    buffer_.Reset(env, env->NewByteArray(kBufferSize));
+    if (ClearException(env))
+      return false;
+  }
+
+  jbyteArray buffer = buffer_.obj();
+  int read_size = std::min(length, kBufferSize);
+  *bytes_read = 0;
+
+  while (length > 0) {
+    int32_t byte_count =
+      Java_InputStream_readI_AB_I_I(
+          env, jobject_.obj(), buffer, 0, read_size);
+
+    if (ClearException(env))
+      return false;
+
+    if (byte_count <= 0)
+      break;
+
+#ifndef NDEBUG
+    int32_t buffer_length = env->GetArrayLength(buffer);
+    DCHECK_GE(read_size, byte_count);
+    DCHECK_GE(buffer_length, byte_count);
+#endif // NDEBUG
+
+    // Copy the data over to the provided C++ side buffer.
+    DCHECK_GE(length, byte_count);
+    env->GetByteArrayRegion(buffer, 0, byte_count,
+        reinterpret_cast<jbyte*>(dest->data() + *bytes_read));
+    if (ClearException(env))
+      return false;
+
+    *bytes_read += byte_count;
+    length -= byte_count;
+  }
+  return true;
+}
+
+} // namespace android_webview