diff options
author | Steven Roose <stevenroose@gmail.com> | 2017-12-05 02:34:15 +0800 |
---|---|---|
committer | Felix Lange <fjl@users.noreply.github.com> | 2017-12-05 02:34:15 +0800 |
commit | afb8154eab2961996b321ac3fe5a21602e3b1aff (patch) | |
tree | 13011886b99de7a66e8d916882d731e12afbb2b1 /common/bytes.go | |
parent | 1d06e41f04d75c31334c455063e9ec7b4136bf23 (diff) | |
download | dexon-afb8154eab2961996b321ac3fe5a21602e3b1aff.tar.gz dexon-afb8154eab2961996b321ac3fe5a21602e3b1aff.tar.zst dexon-afb8154eab2961996b321ac3fe5a21602e3b1aff.zip |
common: improve IsHexAddress and add tests (#15551)
Also unexport isHex, hasHexPrefix because IsHexAddress is the only caller.
Fixes #15550
Diffstat (limited to 'common/bytes.go')
-rw-r--r-- | common/bytes.go | 26 |
1 files changed, 17 insertions, 9 deletions
diff --git a/common/bytes.go b/common/bytes.go index bb40ac1d7..ba00e8a4b 100644 --- a/common/bytes.go +++ b/common/bytes.go @@ -17,9 +17,7 @@ // Package common contains various helper functions. package common -import ( - "encoding/hex" -) +import "encoding/hex" func ToHex(b []byte) string { hex := Bytes2Hex(b) @@ -55,14 +53,24 @@ func CopyBytes(b []byte) (copiedBytes []byte) { return } -func HasHexPrefix(str string) bool { - l := len(str) - return l >= 2 && str[0:2] == "0x" +func hasHexPrefix(str string) bool { + return len(str) >= 2 && str[0] == '0' && (str[1] == 'x' || str[1] == 'X') } -func IsHex(str string) bool { - l := len(str) - return l >= 4 && l%2 == 0 && str[0:2] == "0x" +func isHexCharacter(c byte) bool { + return ('0' <= c && c <= '9') || ('a' <= c && c <= 'f') || ('A' <= c && c <= 'F') +} + +func isHex(str string) bool { + if len(str)%2 != 0 { + return false + } + for _, c := range []byte(str) { + if !isHexCharacter(c) { + return false + } + } + return true } func Bytes2Hex(d []byte) string { |