69 lines
2 KiB
Go
69 lines
2 KiB
Go
package address
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/lbryio/lbryschema.go/address/base58"
|
|
)
|
|
|
|
const lbrycrdMainPubkeyPrefix = byte(85)
|
|
const lbrycrdMainScriptPrefix = byte(122)
|
|
const lbrycrdTestnetPubkeyPrefix = byte(111)
|
|
const lbrycrdTestnetScriptPrefix = byte(196)
|
|
const lbrycrdRegtestPubkeyPrefix = byte(111)
|
|
const lbrycrdRegtestScriptPrefix = byte(196)
|
|
|
|
const prefixLength = 1
|
|
const pubkeyLength = 20
|
|
const checksumLength = 4
|
|
const addressLength = prefixLength + pubkeyLength + checksumLength
|
|
const lbrycrdMain = "lbrycrd_main"
|
|
const lbrycrdTestnet = "lbrycrd_testnet"
|
|
const lbrycrdRegtest = "lbrycrd_regtest"
|
|
|
|
var addressPrefixes = map[string][2]byte{}
|
|
|
|
func SetPrefixes() {
|
|
addressPrefixes[lbrycrdMain] = [2]byte{lbrycrdMainPubkeyPrefix, lbrycrdMainScriptPrefix}
|
|
addressPrefixes[lbrycrdTestnet] = [2]byte{lbrycrdTestnetPubkeyPrefix, lbrycrdTestnetScriptPrefix}
|
|
addressPrefixes[lbrycrdRegtest] = [2]byte{lbrycrdRegtestPubkeyPrefix, lbrycrdRegtestScriptPrefix}
|
|
}
|
|
|
|
func PrefixIsValid(address [addressLength]byte, blockchainName string) bool {
|
|
SetPrefixes()
|
|
prefix := address[0]
|
|
for _, addrPrefix := range addressPrefixes[blockchainName] {
|
|
if addrPrefix == prefix {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func PubKeyIsValid(address [addressLength]byte) bool {
|
|
pubkey := address[prefixLength : pubkeyLength+prefixLength]
|
|
// TODO: validate this for real
|
|
if len(pubkey) != pubkeyLength {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func ChecksumIsValid(address [addressLength]byte) bool {
|
|
return base58.VerifyBase58Checksum(address[:])
|
|
}
|
|
|
|
func ValidateAddress(address [addressLength]byte, blockchainName string) ([addressLength]byte, error) {
|
|
if blockchainName != lbrycrdMain && blockchainName != lbrycrdTestnet && blockchainName != lbrycrdRegtest {
|
|
return address, errors.New("invalid blockchain name")
|
|
}
|
|
if !PrefixIsValid(address, blockchainName) {
|
|
return address, errors.New("invalid prefix")
|
|
}
|
|
if !PubKeyIsValid(address) {
|
|
return address, errors.New("invalid pubkey")
|
|
}
|
|
if !ChecksumIsValid(address) {
|
|
return address, errors.New("invalid address checksum")
|
|
}
|
|
return address, nil
|
|
}
|