mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-13 05:45:02 +00:00
f8979fe7af
* golint and minor changes to make code readable
88 lines
1.8 KiB
Go
Executable file
88 lines
1.8 KiB
Go
Executable file
package hash
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"io"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/wire/util"
|
|
"golang.org/x/crypto/ripemd160"
|
|
)
|
|
|
|
// Sha256 hashes the incoming byte slice
|
|
// using the sha256 algorithm
|
|
func Sha256(data []byte) (util.Uint256, error) {
|
|
var hash util.Uint256
|
|
hasher := sha256.New()
|
|
hasher.Reset()
|
|
_, err := hasher.Write(data)
|
|
|
|
hash, err = util.Uint256DecodeBytes(hasher.Sum(nil))
|
|
if err != nil {
|
|
return hash, err
|
|
}
|
|
return hash, nil
|
|
}
|
|
|
|
// DoubleSha256 performs sha256 twice on the given data
|
|
func DoubleSha256(data []byte) (util.Uint256, error) {
|
|
var hash util.Uint256
|
|
|
|
h1, err := Sha256(data)
|
|
if err != nil {
|
|
return hash, err
|
|
}
|
|
|
|
hash, err = Sha256(h1.Bytes())
|
|
if err != nil {
|
|
return hash, err
|
|
}
|
|
return hash, nil
|
|
}
|
|
|
|
// RipeMD160 performs the RIPEMD160 hash algorithm
|
|
// on the given data
|
|
func RipeMD160(data []byte) (util.Uint160, error) {
|
|
var hash util.Uint160
|
|
hasher := ripemd160.New()
|
|
hasher.Reset()
|
|
_, err := io.WriteString(hasher, string(data))
|
|
|
|
hash, err = util.Uint160DecodeBytes(hasher.Sum(nil))
|
|
if err != nil {
|
|
return hash, err
|
|
}
|
|
return hash, nil
|
|
}
|
|
|
|
// Hash160 performs sha256 and then ripemd160
|
|
// on the given data
|
|
func Hash160(data []byte) (util.Uint160, error) {
|
|
var hash util.Uint160
|
|
h1, err := Sha256(data)
|
|
|
|
h2, err := RipeMD160(h1.Bytes())
|
|
|
|
hash, err = util.Uint160DecodeBytes(h2.Bytes())
|
|
|
|
if err != nil {
|
|
return hash, err
|
|
}
|
|
return hash, nil
|
|
}
|
|
|
|
// Checksum returns the checksum for a given piece of data
|
|
// using sha256 twice as the hash algorithm
|
|
func Checksum(data []byte) ([]byte, error) {
|
|
hash, err := Sum(data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return hash[:4], nil
|
|
}
|
|
|
|
// Sum performs sha256 twice on the given data
|
|
// XXX(issue): We should remove this and just do doublesha256
|
|
func Sum(b []byte) (util.Uint256, error) {
|
|
hash, err := DoubleSha256((b))
|
|
return hash, err
|
|
}
|