Added byte helpers

1 files changed, 27 insertions, 0 deletions

diff --git a/bytes.go b/bytes.go
new file mode 100644
index 000000000..6bf381343
--- /dev/null
+++ b/bytes.go
@@ -0,0 +1,27 @@
+package main
+
+import (
+  "bytes"
+  "encoding/binary"
+  "fmt"
+)
+
+func NumberToBytes(num uint64, bits int) []byte {
+  buf := new(bytes.Buffer)
+  err := binary.Write(buf, binary.BigEndian, num)
+  if err != nil {
+    fmt.Println("binary.Write failed:", err)
+  }
+
+  return buf.Bytes()[buf.Len()-(bits / 8):]
+}
+
+func BytesToNumber(b []byte) (number uint64) {
+  buf := bytes.NewReader(b)
+  err := binary.Read(buf, binary.LittleEndian, &number)
+  if err != nil {
+    fmt.Println("binary.Read failed:", err)
+  }
+
+  return
+}