-
Notifications
You must be signed in to change notification settings - Fork 0
/
address.go
89 lines (76 loc) · 2.14 KB
/
address.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package address
import (
"bytes"
"crypto/ecdsa"
"encoding/base64"
"math/big"
"github.com/ethereum/go-ethereum/crypto"
"github.com/jkevinp/gotron-sdk/pkg/common"
)
const (
// HashLength is the expected length of the hash
HashLength = 32
// AddressLength is the expected length of the address
AddressLength = 21
// AddressLengthBase58 is the expected length of the address in base58format
AddressLengthBase58 = 34
// TronBytePrefix is the hex prefix to address
TronBytePrefix = byte(0x41)
)
// Address represents the 21 byte address of an Tron account.
type Address []byte
// Bytes get bytes from address
func (a Address) Bytes() []byte {
return a[:]
}
// Hex get bytes from address in string
func (a Address) Hex() string {
return common.ToHex(a[:])
}
// BigToAddress returns Address with byte values of b.
// If b is larger than len(h), b will be cropped from the left.
func BigToAddress(b *big.Int) Address {
id := b.Bytes()
base := bytes.Repeat([]byte{0}, AddressLength-len(id))
return append(base, id...)
}
// HexToAddress returns Address with byte values of s.
// If s is larger than len(h), s will be cropped from the left.
func HexToAddress(s string) Address {
addr, err := common.FromHex(s)
if err != nil {
return nil
}
return addr
}
// Base58ToAddress returns Address with byte values of s.
func Base58ToAddress(s string) (Address, error) {
addr, err := common.DecodeCheck(s)
if err != nil {
return nil, err
}
return addr, nil
}
// Base64ToAddress returns Address with byte values of s.
func Base64ToAddress(s string) (Address, error) {
decoded, err := base64.StdEncoding.DecodeString(s)
if err != nil {
return nil, err
}
return Address(decoded), nil
}
// String implements fmt.Stringer.
func (a Address) String() string {
if a[0] == 0 {
return new(big.Int).SetBytes(a.Bytes()).String()
}
return common.EncodeCheck(a.Bytes())
}
// PubkeyToAddress returns address from ecdsa public key
func PubkeyToAddress(p ecdsa.PublicKey) Address {
address := crypto.PubkeyToAddress(p)
addressTron := make([]byte, 0)
addressTron = append(addressTron, TronBytePrefix)
addressTron = append(addressTron, address.Bytes()...)
return addressTron
}