1 files changed, 59 insertions, 0 deletions
diff --git a/netwerk/protocol/http/Http2Stream.h b/netwerk/protocol/http/Http2Stream.h
new file mode 100644
index 0000000000..aae1ca1b5d
--- /dev/null
+++ b/netwerk/protocol/http/Http2Stream.h
@@ -0,0 +1,59 @@
+/* -*- 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_Http2Stream_h
+#define mozilla_net_Http2Stream_h
+
+#include "Http2StreamBase.h"
+
+namespace mozilla::net {
+
+class Http2Stream : public Http2StreamBase {
+ public:
+  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(Http2Stream, override)
+
+  Http2Stream(nsAHttpTransaction* httpTransaction, Http2Session* session,
+              int32_t priority, uint64_t bcId);
+
+  void CloseStream(nsresult reason) override;
+  Http2Stream* GetHttp2Stream() override { return this; }
+  uint32_t GetWireStreamId() override;
+
+  nsresult OnWriteSegment(char* buf, uint32_t count,
+                          uint32_t* countWritten) override;
+
+  nsresult CheckPushCache();
+  Http2PushedStream* PushSource() { return mPushSource; }
+  bool IsReadingFromPushStream();
+  void ClearPushSource();
+
+  nsAHttpTransaction* Transaction() override { return mTransaction; }
+  nsIRequestContext* RequestContext() override {
+    return mTransaction ? mTransaction->RequestContext() : nullptr;
+  }
+
+ protected:
+  ~Http2Stream();
+  nsresult CallToReadData(uint32_t count, uint32_t* countRead) override;
+  nsresult CallToWriteData(uint32_t count, uint32_t* countWritten) override;
+  nsresult GenerateHeaders(nsCString& aCompressedData,
+                           uint8_t& firstFrameFlags) override;
+
+ private:
+  // For Http2Push
+  void AdjustPushedPriority();
+  Http2PushedStream* mPushSource{nullptr};
+
+  // The underlying HTTP transaction. This pointer is used as the key
+  // in the Http2Session mStreamTransactionHash so it is important to
+  // keep a reference to it as long as this stream is a member of that hash.
+  // (i.e. don't change it or release it after it is set in the ctor).
+  RefPtr<nsAHttpTransaction> mTransaction;
+};
+
+}  // namespace mozilla::net
+
+#endif  // mozilla_net_Http2Stream_h