Initial implementation of CronetBidirectionalStream.

components/cronet/android/cronet_bidirectional_stream.h

+// Copyright 2015 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.
+
+#ifndef COMPONENTS_CRONET_ANDROID_CRONET_BIDIRECTIONAL_STREAM_H_
+#define COMPONENTS_CRONET_ANDROID_CRONET_BIDIRECTIONAL_STREAM_H_
+
+#include <jni.h>
+#include <string>
+
+#include "base/android/jni_android.h"
+#include "base/android/jni_array.h"
+#include "base/android/jni_string.h"
+#include "base/android/scoped_java_ref.h"
+#include "base/callback.h"
+#include "base/location.h"
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "net/base/request_priority.h"
+#include "net/http/bidirectional_stream.h"
+#include "url/gurl.h"
+
+namespace base {
+class SingleThreadTaskRunner;

[xunjieli, 2015/12/08 17:59:55]

I don't see this being used.

[mef, 2015/12/11 21:28:25]

Done.

+}  // namespace base
+
+namespace net {
+enum Error;

[xunjieli, 2015/12/08 17:59:55]

I believe the style guide recommends against forward declaring enums.

[mef, 2015/12/11 21:28:25]

Done.

+class HttpRequestHeaders;
+class HttpResponseHeaders;
+class SSLInfo;
+class SpdyHeaderBlock;
+class UploadDataStream;

[xunjieli, 2015/12/08 17:59:55]

UploadDataStream is not used.

[mef, 2015/12/11 21:28:25]

Done.

+}  // namespace net
+
+namespace cronet {
+
+class CronetURLRequestContextAdapter;
+
+bool CronetBidirectionalStreamRegisterJni(JNIEnv* env);
+
+// An adapter from Java BidirectionalStream object to net::BidirectionalStream.
+// Created and configured from a Java thread. Start, ReadData, and Destroy are
+// posted to network thread and all callbacks into the Java BidirectionalStream
+// are done on the network thread. Java BidirectionalStream is expected to
+// initiate the next step like ReadData or Destroy. Public methods can be called
+// on any thread except PopulateResponseHeaders and Get* methods, which can only

[xunjieli, 2015/12/08 17:59:55]

nit: outdated comment. There isn't a PopulateResponseHeaders.

[mef, 2015/12/11 21:28:25]

Done.

+// be called on the network thread.
+class CronetBidirectionalStream : public net::BidirectionalStream::Delegate {
+ public:
+  CronetBidirectionalStream(CronetURLRequestContextAdapter* context,
+                            JNIEnv* env,
+                            jobject jbidi_stream);
+  ~CronetBidirectionalStream() override;
+
+  // Starts the request.
+  jint Start(JNIEnv* env,
+             jobject jcaller,
+             jstring jurl,
+             jstring jmethod,
+             jobjectArray jheaders);
+
+  // Reads more data.
+  jboolean ReadData(JNIEnv* env,
+                    jobject jcaller,
+                    jobject jbyte_buffer,
+                    jint jposition,
+                    jint jcapacity);
+
+  // Writes more data.
+  jboolean WriteData(JNIEnv* env,
+                     jobject jcaller,
+                     jobject jbyte_buffer,
+                     jint jposition,
+                     jint jcapacity,
+                     jboolean jend_of_stream);
+
+  // Releases all resources for the request and deletes the object itself.
+  // |jsend_on_canceled| indicates if Java onCanceled callback should be
+  // issued to indicate when no more callbacks will be issued.
+  void Destroy(JNIEnv* env, jobject jcaller, jboolean jsend_on_canceled);
+
+  // Gets NPN or ALPN Negotiated Protocol (if any) from HttpResponseInfo.
+  // Can only be called on the network thread.
+  base::android::ScopedJavaLocalRef<jstring> GetNegotiatedProtocol(
+      JNIEnv* env,
+      jobject jcaller) const;
+
+  // net::BidirectionalStream::Delegate implementations:
+
+  // Called when the request headers have been sent.
+  void OnRequestHeadersSent() override;
+
+  // Called when response headers are received.
+  void OnHeaders(const net::SpdyHeaderBlock& response_headers) override;
+
+  // Called when read is completed asynchronously. |bytes_read| specifies how
+  // much data is available.
+  void OnReadCompleted(int bytes_read) override;
+
+  // Called when the entire buffer passed through SendData is sent.
+  void OnDataSent() override;
+
+  // Called when trailers are received.
+  void OnTrailers(const net::SpdyHeaderBlock& trailers) override;
+
+  // Called when the stream is closed with error. No other delegate functions
+  // will be called after this.
+  void OnFailed(int error) override;
+
+ private:
+  class IOBufferWithByteBuffer;
+
+  void StartOnNetworkThread(
+      scoped_ptr<net::BidirectionalStream::RequestInfo> request_info);
+  // Gets headers as Java array.
+  base::android::ScopedJavaLocalRef<jobjectArray> GetHeadersArray(
+      JNIEnv* env,
+      const net::SpdyHeaderBlock& header_block);
+  void ReadDataOnNetworkThread(
+      scoped_refptr<IOBufferWithByteBuffer> read_buffer,
+      int buffer_size);
+  void WriteDataOnNetworkThread(
+      scoped_refptr<IOBufferWithByteBuffer> read_buffer,
+      int buffer_size,
+      bool end_of_stream);
+  void DestroyOnNetworkThread(bool send_on_canceled);
+
+  CronetURLRequestContextAdapter* context_;
+
+  // Java object that owns this CronetURLRequestContextAdapter.

[xunjieli, 2015/12/08 17:59:55]

nit: s/CronetURLRequestContextAdapter/CrnetBidirectionalStream

[mef, 2015/12/11 21:28:25]

Done.

+  base::android::ScopedJavaGlobalRef<jobject> owner_;
+
+  scoped_refptr<IOBufferWithByteBuffer> read_buffer_;
+  scoped_refptr<IOBufferWithByteBuffer> write_buffer_;
+  scoped_ptr<net::BidirectionalStream> bidi_stream_;
+
+  DISALLOW_COPY_AND_ASSIGN(CronetBidirectionalStream);
+};
+
+}  // namespace cronet
+
+#endif  // COMPONENTS_CRONET_ANDROID_CRONET_BIDIRECTIONAL_STREAM_H_