neoneo-go/pkg/wire/util/util.go

50 lines
937 B
Go
Raw Normal View History

2019-02-25 22:44:14 +00:00
package util
import (
"bytes"
"crypto/sha256"
"io"
"io/ioutil"
)
// Convenience function
// BufferLength returns the length of a buffer as uint32
2019-02-25 22:44:14 +00:00
func BufferLength(buf *bytes.Buffer) uint32 {
return uint32(buf.Len())
}
// SumSHA256 returns the sha256 sum of the data
2019-02-25 22:44:14 +00:00
func SumSHA256(b []byte) []byte {
h := sha256.New()
h.Write(b)
return h.Sum(nil)
}
// CalculateHash takes a function with a binary writer and returns
// the double hash of the io.Writer
2019-02-25 22:44:14 +00:00
func CalculateHash(f func(bw *BinWriter)) (Uint256, error) {
buf := new(bytes.Buffer)
bw := &BinWriter{W: buf}
f(bw)
var hash Uint256
hash = sha256.Sum256(buf.Bytes())
hash = sha256.Sum256(hash.Bytes())
return hash, bw.Err
}
//ReaderToBuffer converts a io.Reader into a bytes.Buffer
func ReaderToBuffer(r io.Reader) (*bytes.Buffer, error) {
2019-02-25 22:44:14 +00:00
byt, err := ioutil.ReadAll(r)
if err != nil {
return nil, err
2019-02-25 22:44:14 +00:00
}
buf := bytes.NewBuffer(byt)
return buf, nil
2019-02-25 22:44:14 +00:00
}