...

1 files changed, 88 insertions, 0 deletions

diff --git a/stream.go b/stream.go
new file mode 100644
index 0000000..2727565
--- /dev/null
+++ b/stream.go
@@ -0,0 +1,88 @@
+package salticidae
+
+// #include "salticidae/stream.h"
+import "C"
+
+type ByteArray = *C.struct_bytearray_t
+
+func NewByteArray() ByteArray { return C.bytearray_new() }
+func (self ByteArray) Free() { C.bytearray_free(self) }
+
+type DataStream = *C.struct_datastream_t
+
+func NewDataStream() DataStream { return C.datastream_new() }
+func NewDataStreamFromBytes(bytes []byte) DataStream {
+    base := uintptr(rawptr_t(&bytes[0]))
+    return C.datastream_new_from_bytes(
+        (*C.uint8_t)(rawptr_t(base)),
+        (*C.uint8_t)(rawptr_t(base + uintptr(len(bytes)))))
+}
+
+func (self DataStream) Free() { C.datastream_free(self) }
+
+func (self DataStream) AssignByCopy(src DataStream) {
+    C.datastream_assign_by_copy(self, src)
+}
+
+func (self DataStream) AssignByMove(src DataStream) {
+    C.datastream_assign_by_move(self, src)
+}
+
+// TODO: datastream_data
+
+func (self DataStream) Clear() { C.datastream_clear(self) }
+
+func (self DataStream) Size() int { return int(C.datastream_size(self)) }
+
+func (self DataStream) PutU8(v uint8) { C.datastream_put_u8(self, C.uint8_t(v)) }
+func (self DataStream) PutU16(v uint16) { C.datastream_put_u16(self, C.uint16_t(v)) }
+func (self DataStream) PutU32(v uint32) { C.datastream_put_u32(self, C.uint32_t(v)) }
+func (self DataStream) PutU64(v uint32) { C.datastream_put_u64(self, C.uint64_t(v)) }
+
+func (self DataStream) PutI8(v int8) { C.datastream_put_i8(self, C.int8_t(v)) }
+func (self DataStream) PutI16(v int16) { C.datastream_put_i16(self, C.int16_t(v)) }
+func (self DataStream) PutI32(v int32) { C.datastream_put_i32(self, C.int32_t(v)) }
+func (self DataStream) PutI64(v int32) { C.datastream_put_i64(self, C.int64_t(v)) }
+
+func (self DataStream) PutData(bytes []byte) {
+    base := uintptr(rawptr_t(&bytes[0]))
+    C.datastream_put_data(self,
+        (*C.uint8_t)(rawptr_t(base)),
+        (*C.uint8_t)(rawptr_t(base + uintptr(len(bytes)))))
+
+}
+
+func (self DataStream) GetU8() uint8 { return uint8(C.datastream_get_u8(self)) }
+func (self DataStream) GetU16() uint16 { return uint16(C.datastream_get_u16(self)) }
+func (self DataStream) GetU32() uint32 { return uint32(C.datastream_get_u32(self)) }
+func (self DataStream) GetU64() uint64 { return uint64(C.datastream_get_u64(self)) }
+
+func (self DataStream) GetI8() int8 { return int8(C.datastream_get_i8(self)) }
+func (self DataStream) GetI16() int16 { return int16(C.datastream_get_i16(self)) }
+func (self DataStream) GetI32() int32 { return int32(C.datastream_get_i32(self)) }
+func (self DataStream) GetI64() int64 { return int64(C.datastream_get_i64(self)) }
+
+
+func (self DataStream) GetDataInPlace(length int) *C.uint8_t {
+    return C.datastream_get_data_inplace(self, C.size_t(length))
+}
+
+func (self DataStream) GetData(length int) []byte {
+    return C.GoBytes(rawptr_t(self.GetDataInPlace(length)), C.int(length))
+}
+
+type UInt256 = *C.uint256_t
+
+func NewUInt256() UInt256 { return C.uint256_new() }
+func (self UInt256) UInt256IsNull() bool { return bool(C.uint256_is_null(self)) }
+func (self UInt256) UInt256IsEq(other UInt256) bool { return bool(C.uint256_is_eq(self, other)) }
+func (self UInt256) Serialize(s DataStream) { C.uint256_serialize(self, s) }
+func (self UInt256) Unserialize(s DataStream) { C.uint256_unserialize(self, s) }
+
+func (self DataStream) GetHash() UInt256 {
+    return C.datastream_get_hash(self)
+}
+
+func (_moved_self DataStream) ToByteArray() ByteArray {
+    return C.datastream_to_bytearray(_moved_self)
+}