partial bitcoin#8149: import OverrideStream from "BIP144: Serialization, hashes, relay (sender side)"

kwvg · kwvg · commit 5246a8664653 · 2021-05-25T11:08:41.000+05:30
diff --git a/src/streams.h b/src/streams.h
@@ -168,6 +168,39 @@ class VectorReader
         }
     }
 };
+template<typename Stream>
+class OverrideStream
+{
+    Stream* stream;
+public:
+    const int nType;
+    const int nVersion;
+
+    OverrideStream(Stream* stream_, int nType_, int nVersion_) : stream(stream_), nType(nType_), nVersion(nVersion_) {}
+
+    template<typename T>
+    OverrideStream<Stream>& operator<<(const T& obj)
+    {
+        // Serialize to this stream
+        ::Serialize(*this->stream, obj);
+        return (*this);
+    }
+
+    template<typename T>
+    OverrideStream<Stream>& operator>>(T& obj)
+    {
+        // Unserialize from this stream
+        ::Unserialize(*this->stream, obj);
+        return (*this);
+    }
+};
+
+template<typename S>
+OverrideStream<S> WithOrVersion(S* s, int nVersionFlag)
+{
+    return OverrideStream<S>(s, s->GetType(), s->GetVersion() | nVersionFlag);
+}
+
 
 /** Double ended buffer combining vector and stream-like interfaces.
  *
