public/gotron
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

check checksum in NewFromBase58

Browse Source
This commit is contained in:
anon 2024-12-28 17:25:50 -03:00
parent b3a47c5c7b
commit 54bdad99f1
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkg/address/address.go
View File

@ -1,6 +1,7 @@
package address package address
import ( import (
"bytes"
"crypto/ecdsa" "crypto/ecdsa"
"crypto/sha256" "crypto/sha256"
"encoding/hex" "encoding/hex"
@ -77,16 +78,22 @@ func NewFromHex(s string) (Address, error) {
} }
func NewFromBase58(s string) (Address, error) { func NewFromBase58(s string) (Address, error) {
bytes, err := base58.Decode(s) b, err := base58.Decode(s)
if err != nil { if err != nil {
return Address{}, err return Address{}, err
} }
if len(bytes) != 25 { if len(b) != 25 {
return Address{}, fmt.Errorf("tron address is not 25 bytes but %d", len(bytes)) return Address{}, fmt.Errorf("tron address is not 25 bytes but %d", len(b))
} }
return *(*[21]byte)(bytes[:21]), nil hash := SHA256(SHA256(b[:21]))
checksum := hash[:4]
if !bytes.Equal(checksum, b[21:]) {
return Address{}, fmt.Errorf("invalid checksum")
}
return *(*[21]byte)(b[:21]), nil
} }
func NewFromString(s string) (Address, error) { func NewFromString(s string) (Address, error) {