1 files changed, 151 insertions, 0 deletions
diff --git a/netwerk/protocol/http/Http2StreamTunnel.h b/netwerk/protocol/http/Http2StreamTunnel.h
new file mode 100644
index 0000000000..78f0300f4c
--- /dev/null
+++ b/netwerk/protocol/http/Http2StreamTunnel.h
@@ -0,0 +1,151 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set sw=2 ts=8 et tw=80 : */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef mozilla_net_Http2StreamTunnel_h
+#define mozilla_net_Http2StreamTunnel_h
+
+#include "Http2StreamBase.h"
+#include "nsHttpConnection.h"
+#include "nsWeakReference.h"
+
+namespace mozilla::net {
+
+class OutputStreamTunnel;
+class InputStreamTunnel;
+
+// c881f764-a183-45cb-9dec-d9872b2f47b2
+#define NS_HTTP2STREAMTUNNEL_IID                     \
+  {                                                  \
+    0xc881f764, 0xa183, 0x45cb, {                    \
+      0x9d, 0xec, 0xd9, 0x87, 0x2b, 0x2f, 0x47, 0xb2 \
+    }                                                \
+  }
+
+class Http2StreamTunnel : public Http2StreamBase,
+                          public nsISocketTransport,
+                          public nsSupportsWeakReference {
+ public:
+  NS_DECLARE_STATIC_IID_ACCESSOR(NS_HTTP2STREAMTUNNEL_IID)
+  NS_DECL_THREADSAFE_ISUPPORTS
+  NS_DECL_NSITRANSPORT
+  NS_DECL_NSISOCKETTRANSPORT
+
+  Http2StreamTunnel(Http2Session* session, int32_t priority, uint64_t bcId,
+                    nsHttpConnectionInfo* aConnectionInfo);
+  bool IsTunnel() override { return true; }
+
+  already_AddRefed<nsHttpConnection> CreateHttpConnection(
+      nsAHttpTransaction* httpTransaction, nsIInterfaceRequestor* aCallbacks,
+      PRIntervalTime aRtt, bool aIsWebSocket);
+
+  nsHttpConnectionInfo* ConnectionInfo() override { return mConnectionInfo; }
+
+  void SetRequestDone() { mSendClosed = true; }
+  nsresult Condition() override { return mCondition; }
+  void CloseStream(nsresult reason) override;
+  void DisableSpdy() override {
+    if (mTransaction) {
+      mTransaction->DisableHttp2ForProxy();
+    }
+  }
+  void ReuseConnectionOnRestartOK(bool aReuse) override {
+    // Do nothing here. We'll never reuse the connection.
+  }
+  void MakeNonSticky() override {}
+
+ protected:
+  ~Http2StreamTunnel();
+  nsresult CallToReadData(uint32_t count, uint32_t* countRead) override;
+  nsresult CallToWriteData(uint32_t count, uint32_t* countWritten) override;
+  void HandleResponseHeaders(nsACString& aHeadersOut,
+                             int32_t httpResponseCode) override;
+  nsresult GenerateHeaders(nsCString& aCompressedData,
+                           uint8_t& firstFrameFlags) override;
+  bool CloseSendStreamWhenDone() override { return mSendClosed; }
+
+  RefPtr<nsAHttpTransaction> mTransaction;
+
+  void ClearTransactionsBlockedOnTunnel();
+  bool DispatchRelease();
+
+  RefPtr<OutputStreamTunnel> mOutput;
+  RefPtr<InputStreamTunnel> mInput;
+  RefPtr<nsHttpConnectionInfo> mConnectionInfo;
+  bool mSendClosed{false};
+  nsresult mCondition{NS_OK};
+};
+
+// f9d10060-f5d4-443e-ba59-f84ea975c5f0
+#define NS_OUTPUTSTREAMTUNNEL_IID                    \
+  {                                                  \
+    0xf9d10060, 0xf5d4, 0x443e, {                    \
+      0xba, 0x59, 0xf8, 0x4e, 0xa9, 0x75, 0xc5, 0xf0 \
+    }                                                \
+  }
+
+class OutputStreamTunnel : public nsIAsyncOutputStream {
+ public:
+  NS_DECLARE_STATIC_IID_ACCESSOR(NS_OUTPUTSTREAMTUNNEL_IID)
+  NS_DECL_THREADSAFE_ISUPPORTS
+  NS_DECL_NSIOUTPUTSTREAM
+  NS_DECL_NSIASYNCOUTPUTSTREAM
+
+  explicit OutputStreamTunnel(Http2StreamTunnel* aStream);
+
+  nsresult OnSocketReady(nsresult condition);
+  void MaybeSetRequestDone(nsIOutputStreamCallback* aCallback);
+
+ private:
+  virtual ~OutputStreamTunnel();
+
+  nsresult GetStream(Http2StreamTunnel** aStream);
+  nsresult GetSession(Http2Session** aSession);
+
+  nsWeakPtr mWeakStream;
+  nsCOMPtr<nsIOutputStreamCallback> mCallback;
+  nsresult mCondition{NS_OK};
+};
+
+class InputStreamTunnel : public nsIAsyncInputStream {
+ public:
+  NS_DECL_THREADSAFE_ISUPPORTS
+  NS_DECL_NSIINPUTSTREAM
+  NS_DECL_NSIASYNCINPUTSTREAM
+
+  explicit InputStreamTunnel(Http2StreamTunnel* aStream);
+
+  nsresult OnSocketReady(nsresult condition);
+  bool HasCallback() { return !!mCallback; }
+
+ private:
+  virtual ~InputStreamTunnel();
+
+  nsresult GetStream(Http2StreamTunnel** aStream);
+  nsresult GetSession(Http2Session** aSession);
+
+  nsWeakPtr mWeakStream;
+  nsCOMPtr<nsIInputStreamCallback> mCallback;
+  nsresult mCondition{NS_OK};
+};
+
+NS_DEFINE_STATIC_IID_ACCESSOR(Http2StreamTunnel, NS_HTTP2STREAMTUNNEL_IID)
+NS_DEFINE_STATIC_IID_ACCESSOR(OutputStreamTunnel, NS_OUTPUTSTREAMTUNNEL_IID)
+
+class Http2StreamWebSocket : public Http2StreamTunnel {
+ public:
+  Http2StreamWebSocket(Http2Session* session, int32_t priority, uint64_t bcId,
+                       nsHttpConnectionInfo* aConnectionInfo);
+  void CloseStream(nsresult reason) override;
+
+ protected:
+  virtual ~Http2StreamWebSocket();
+  nsresult GenerateHeaders(nsCString& aCompressedData,
+                           uint8_t& firstFrameFlags) override;
+};
+
+}  // namespace mozilla::net
+
+#endif  // mozilla_net_Http2StreamTunnel_h