writer close

[bytom/vapor.git] / common / address.go

diff --git a/common/address.go b/common/address.go
index a5162e4..3fe4999 100644 (file)
--- a/common/address.go
+++ b/common/address.go
@@ -116,7 +116,7 @@ func DecodeAddress(addr string, param *consensus.Params) (Address, error) {
        oneIndex := strings.LastIndexByte(addr, '1')
        if oneIndex > 1 {
                prefix := addr[:oneIndex+1]
-               if consensus.IsBech32SegwitPrefix(prefix, param) {
+               if IsBech32SegwitPrefix(prefix, param) {
                        witnessVer, witnessProg, err := decodeSegWitAddress(addr)
                        if err != nil {
                                return nil, err
@@ -144,6 +144,13 @@ func DecodeAddress(addr string, param *consensus.Params) (Address, error) {
        return nil, ErrUnknownAddressType
 }
 
+// IsBech32SegwitPrefix returns whether the prefix is a known prefix for segwit
+// addresses on any default or registered network.  This is used when decoding
+// an address string into a specific address type.
+func IsBech32SegwitPrefix(prefix string, params *consensus.Params) bool {
+       return strings.ToLower(prefix) == params.Bech32HRPSegwit+"1"
+}
+
 // decodeSegWitAddress parses a bech32 encoded segwit address string and
 // returns the witness version and witness program byte representation.
 func decodeSegWitAddress(address string) (byte, []byte, error) {