/
bizdata_crypt.go
116 lines (90 loc) · 2.6 KB
/
bizdata_crypt.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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package mp
import (
"encoding/base64"
"encoding/json"
"errors"
"fmt"
"github.com/arden/wechat/utils"
)
// UserData 用户数据
type UserData struct {
OpenID string `json:"openId"`
Language string `json:"language"`
City string `json:"city"`
Province string `json:"province"`
AvatarURL string `json:"avatarUrl"`
NickName string `json:"nickName"`
Gender int `json:"gender"`
Country string `json:"country"`
UnionID string `json:"unionId"`
WaterMark *WaterMark `json:"watermark"`
}
// PhoneData 用户手机号绑定数据
type PhoneData struct {
PhoneNumber string `json:"phoneNumber"`
PurePhoneNumber string `json:"purePhoneNumber"`
CountryCode string `json:"countryCode"`
WaterMark *WaterMark `json:"watermark"`
}
// WaterMark 水印
type WaterMark struct {
Timestamp int64 `json:"timestamp"`
AppID string `json:"appid"`
}
// BizDataCrypt 小程序数据解析
type BizDataCrypt struct {
mp *WXMP
encryptedData string
sessionKey string
iv string
data []byte
}
// Decrypt 数据解密
func (b *BizDataCrypt) Decrypt() error {
aesData, err := base64.StdEncoding.DecodeString(b.encryptedData)
if err != nil {
return err
}
aesKey, err := base64.StdEncoding.DecodeString(b.sessionKey)
if err != nil {
return err
}
aesIV, err := base64.StdEncoding.DecodeString(b.iv)
if err != nil {
return err
}
plainText, err := utils.AESCBCDecrypt(aesData, aesKey, aesIV...)
if err != nil {
return err
}
b.data = plainText
return nil
}
// GetUserData 获取用户数据(需先解密)
func (b *BizDataCrypt) GetUserData() (*UserData, error) {
if len(b.data) == 0 {
return nil, errors.New("gochat: empty data, check whether decrypted")
}
userData := new(UserData)
if err := json.Unmarshal(b.data, &userData); err != nil {
return nil, err
}
if userData.WaterMark.AppID != b.mp.AppID {
return nil, fmt.Errorf("gochat: wxmp user bizdata appid mismatch, want: %s, got: %s", b.mp.AppID, userData.WaterMark.AppID)
}
return userData, nil
}
// GetPhoneData 获取用户手机号绑定数据(需先解密)
func (b *BizDataCrypt) GetPhoneData() (*PhoneData, error) {
if len(b.data) == 0 {
return nil, errors.New("gochat: empty data, check whether decrypted")
}
phoneData := new(PhoneData)
if err := json.Unmarshal(b.data, &phoneData); err != nil {
return nil, err
}
if phoneData.WaterMark.AppID != b.mp.AppID {
return nil, fmt.Errorf("gochat: wxmp phone bizdata appid mismatch, want: %s, got: %s", b.mp.AppID, phoneData.WaterMark.AppID)
}
return phoneData, nil
}