77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"math/big"
|
|
)
|
|
|
|
// IntToHex converts an int64 to a byte array
|
|
func IntToHex(num int64) []byte {
|
|
buf := new(bytes.Buffer)
|
|
err := binary.Write(buf, binary.BigEndian, num)
|
|
if err != nil {
|
|
exitWithError(err)
|
|
}
|
|
return buf.Bytes()
|
|
}
|
|
|
|
// ReverseBytes reverses a byte array
|
|
func ReverseBytes(data []byte) {
|
|
for i, j := 0, len(data)-1; i < j; i, j = i+1, j-1 {
|
|
data[i], data[j] = data[j], data[i]
|
|
}
|
|
}
|
|
|
|
// Base58 Encode/Decode Functions
|
|
var b58Alphabet = []byte("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz")
|
|
|
|
// Base58Encode encodes a byte array to Base58
|
|
func Base58Encode(input []byte) []byte {
|
|
var result []byte
|
|
x := big.NewInt(0).SetBytes(input)
|
|
|
|
base := big.NewInt(int64(len(b58Alphabet)))
|
|
zero := big.NewInt(0)
|
|
mod := &big.Int{}
|
|
|
|
for x.Cmp(zero) != 0 {
|
|
x.DivMod(x, base, mod)
|
|
result = append(result, b58Alphabet[mod.Int64()])
|
|
}
|
|
|
|
ReverseBytes(result)
|
|
for b := range input {
|
|
if b == 0x00 {
|
|
result = append([]byte{b58Alphabet[0]}, result...)
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
// Base58Decode decodes Base58-encoded data
|
|
func Base58Decode(input []byte) []byte {
|
|
result := big.NewInt(0)
|
|
zeroBytes := 0
|
|
|
|
for b := range input {
|
|
if b == 0x00 {
|
|
zeroBytes++
|
|
}
|
|
}
|
|
|
|
payload := input[zeroBytes:]
|
|
for _, b := range payload {
|
|
charIndex := bytes.IndexByte(b58Alphabet, b)
|
|
result.Mul(result, big.NewInt(58))
|
|
result.Add(result, big.NewInt(int64(charIndex)))
|
|
}
|
|
|
|
decoded := result.Bytes()
|
|
decoded = append(bytes.Repeat([]byte{byte(0x00)}, zeroBytes), decoded...)
|
|
|
|
return decoded
|
|
}
|