正在显示
6 个修改的文件
包含
202 行增加
和
13 行删除
1 | -package common | |
1 | +package socialwork_sdk | |
2 | 2 | |
3 | 3 | import ( |
4 | 4 | "bytes" |
... | ... | @@ -7,7 +7,7 @@ import ( |
7 | 7 | "encoding/base64" |
8 | 8 | ) |
9 | 9 | |
10 | -func Aes_CBC_Decrypt(crypted []byte, key, iv []byte) ([]byte, error) { | |
10 | +func aes_CBC_Decrypt(crypted []byte, key, iv []byte) ([]byte, error) { | |
11 | 11 | block, err := aes.NewCipher(key) |
12 | 12 | if err != nil { |
13 | 13 | return nil, err |
... | ... | @@ -26,7 +26,7 @@ func unpadding(cipherText []byte) []byte { |
26 | 26 | return cipherText |
27 | 27 | } |
28 | 28 | |
29 | -func Aes_CBC_Encrypt(data []byte, key, iv []byte) (string, error) { | |
29 | +func aes_CBC_Encrypt(data []byte, key, iv []byte) (string, error) { | |
30 | 30 | block, err := aes.NewCipher(key) |
31 | 31 | if err != nil { |
32 | 32 | return "", err | ... | ... |
constant.go
0 → 100644
httpclient.go
0 → 100644
1 | +package socialwork_sdk | |
2 | + | |
3 | +import ( | |
4 | + "bytes" | |
5 | + "encoding/json" | |
6 | + "errors" | |
7 | + "fmt" | |
8 | + "io" | |
9 | + "io/ioutil" | |
10 | + "net/http" | |
11 | + "net/http/cookiejar" | |
12 | + "strings" | |
13 | +) | |
14 | + | |
15 | +const ( | |
16 | + post = "POST" | |
17 | + get = "GET" | |
18 | + put = "PUT" | |
19 | +) | |
20 | + | |
21 | +func makeHttpRequestV4(method, url string, headers map[string]string, entity map[string]interface{}, jar *cookiejar.Jar) (string, int, error) { | |
22 | + var body io.Reader | |
23 | + var err error | |
24 | + | |
25 | + if entity != nil { | |
26 | + switch method { | |
27 | + case post, put: | |
28 | + if len(entity) == 1 && entity["jsonbody"] != nil { | |
29 | + jsonBody, ok := entity["jsonbody"].(string) | |
30 | + if ok { | |
31 | + body = bytes.NewBuffer([]byte(jsonBody)) | |
32 | + } | |
33 | + } else { | |
34 | + b, err := json.Marshal(entity) | |
35 | + if err != nil { | |
36 | + return "", 0, err | |
37 | + } | |
38 | + | |
39 | + b = bytes.Replace(b, []byte("\\u003c"), []byte("<"), -1) | |
40 | + b = bytes.Replace(b, []byte("\\u003e"), []byte(">"), -1) | |
41 | + b = bytes.Replace(b, []byte("\\u0026"), []byte("&"), -1) | |
42 | + | |
43 | + body = bytes.NewBuffer(b) | |
44 | + } | |
45 | + | |
46 | + case get: | |
47 | + if len(entity) > 0 { | |
48 | + params := make([]string, len(entity)) | |
49 | + index := 0 | |
50 | + for k, v := range entity { | |
51 | + _v := fmt.Sprintf("%v", v) | |
52 | + params[index] = fmt.Sprintf("%s=%v", k, _v) | |
53 | + index++ | |
54 | + } | |
55 | + queryStr := strings.Join(params, "&") | |
56 | + url = fmt.Sprintf("%s?%s", url, queryStr) | |
57 | + } | |
58 | + } | |
59 | + } | |
60 | + | |
61 | + req, err := http.NewRequest(method, url, body) | |
62 | + if err != nil { | |
63 | + return "", 0, err | |
64 | + } | |
65 | + // set the http request header | |
66 | + for key, value := range headers { | |
67 | + req.Header.Set(key, value) | |
68 | + } | |
69 | + if entity != nil && (method == post || method == put) { | |
70 | + req.Header.Set("Content-Type", "application/json;charset=utf-8") | |
71 | + req.Header.Set("Accept", "application/json") | |
72 | + } | |
73 | + req.Header.Add("Connection", "close") | |
74 | + req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux i686; U;) Gecko/20070322 Kazehakase/0.4.5") | |
75 | + | |
76 | + //跳过证书验证 | |
77 | + // tr := &http.Transport{ | |
78 | + // TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, | |
79 | + // } | |
80 | + client := http.DefaultClient | |
81 | + if jar != nil { | |
82 | + client = &http.Client{ | |
83 | + Jar: jar, | |
84 | + } | |
85 | + } | |
86 | + // client.Transport = tr | |
87 | + | |
88 | + fmt.Printf("xuht================Request: %+v\n", *req) | |
89 | + res, err := client.Do(req) | |
90 | + if err != nil { | |
91 | + fmt.Println("faild to do the request with error ", err) | |
92 | + return "", 0, err | |
93 | + } | |
94 | + defer res.Body.Close() | |
95 | + | |
96 | + if res.StatusCode != http.StatusOK && res.StatusCode != http.StatusCreated && res.StatusCode != http.StatusNoContent { | |
97 | + resBody, _ := ioutil.ReadAll(res.Body) | |
98 | + fmt.Println("code is not 200 ", res.StatusCode, string(resBody)) | |
99 | + return "", 0, errors.New("http request failed to call") | |
100 | + } | |
101 | + resBody, err := ioutil.ReadAll(res.Body) | |
102 | + if err != nil { | |
103 | + fmt.Println("could not read the response body") | |
104 | + return "", 0, errors.New("the response could not be read") | |
105 | + } | |
106 | + | |
107 | + return string(resBody), res.StatusCode, nil | |
108 | +} | ... | ... |
model/order_client.go
0 → 100644
1 | 1 | package socialwork_sdk |
2 | 2 | |
3 | 3 | import ( |
4 | + "encoding/base64" | |
4 | 5 | "encoding/json" |
5 | - "gitlab.workai.com.cn/chenang/socialwork-sdk/common" | |
6 | + "gitlab.workai.com.cn/chenang/socialwork-sdk/model" | |
6 | 7 | "log" |
7 | 8 | ) |
8 | 9 | |
... | ... | @@ -11,26 +12,28 @@ type OrderClient struct { |
11 | 12 | IV string `json:"iv"` |
12 | 13 | PublicKey string `json:"public_key"` |
13 | 14 | CustomerPrivateKey string `json:"customer_private_key"` |
15 | + PtCode string `json:"pt_code"` | |
16 | + IP string `json:"ip"` | |
14 | 17 | } |
15 | 18 | |
16 | -func NewOrderClient(key, iv, customerPrivateKey, publicKey string) *OrderClient { | |
19 | +func NewOrderClient(key, iv, customerPrivateKey, publicKey, ip string) *OrderClient { | |
17 | 20 | return &OrderClient{ |
18 | 21 | Key: key, |
19 | 22 | IV: iv, |
20 | 23 | PublicKey: publicKey, |
21 | 24 | CustomerPrivateKey: customerPrivateKey, |
25 | + IP: ip, | |
22 | 26 | } |
23 | 27 | } |
24 | 28 | |
25 | -func (o *OrderClient) EncryptAndSign(res interface{}) (string, string, error) { | |
26 | - bs, _ := json.Marshal(res) | |
27 | - cipherText, err := common.Aes_CBC_Encrypt(bs, []byte(o.Key), []byte(o.IV)) | |
29 | +func (o *OrderClient) EncryptAndSign(bs []byte) (string, string, error) { | |
30 | + cipherText, err := aes_CBC_Encrypt(bs, []byte(o.Key), []byte(o.IV)) | |
28 | 31 | if err != nil { |
29 | 32 | log.Printf("Aes_CBC_Encrypt.failed.err=%v\n", err) |
30 | 33 | return "", "", err |
31 | 34 | } |
32 | 35 | |
33 | - sign, err := common.RsaSignWithMd5Hex(cipherText, o.CustomerPrivateKey) | |
36 | + sign, err := rsaSignWithMd5Hex(cipherText, o.CustomerPrivateKey) | |
34 | 37 | if err != nil { |
35 | 38 | log.Printf("RsaSignWithMd5Hex.failed.err=%v\n", err) |
36 | 39 | return "", "", err |
... | ... | @@ -38,6 +41,52 @@ func (o *OrderClient) EncryptAndSign(res interface{}) (string, string, error) { |
38 | 41 | return cipherText, sign, nil |
39 | 42 | } |
40 | 43 | |
41 | -func (o *OrderClient) PrintVersion() string { | |
42 | - return "1.0" | |
44 | +func (o *OrderClient) DecryptAndVerySign(cipherText, sign string) ([]byte, error) { | |
45 | + _cipherText, err := base64.StdEncoding.DecodeString(cipherText) | |
46 | + if err != nil { | |
47 | + return nil, err | |
48 | + } | |
49 | + err = rsaVerifySignWithMd5Base64(cipherText, sign, o.PublicKey) | |
50 | + if err != nil { | |
51 | + return nil, err | |
52 | + } | |
53 | + content, err := aes_CBC_Decrypt(_cipherText, []byte(o.Key), []byte(o.IV)) | |
54 | + if err != nil { | |
55 | + return nil, err | |
56 | + } | |
57 | + return content, nil | |
58 | +} | |
59 | + | |
60 | +func (o *OrderClient) SendRequest(businessCode, reqBody string) (string, error) { | |
61 | + //校验参数 reqbody | |
62 | + checkErr := o.checkParams(businessCode, reqBody) | |
63 | + if checkErr != nil { | |
64 | + return "", checkErr | |
65 | + } | |
66 | + cipherText, sign, err := o.EncryptAndSign([]byte(reqBody)) | |
67 | + if err != nil { | |
68 | + return "", err | |
69 | + } | |
70 | + | |
71 | + orderModel := model.OrderModel{ | |
72 | + Data: cipherText, | |
73 | + Sign: sign, | |
74 | + PtCode: o.PtCode, | |
75 | + BusinessCode: businessCode, | |
76 | + } | |
77 | + | |
78 | + _orderModel, err := json.Marshal(orderModel) | |
79 | + if err != nil { | |
80 | + return "", err | |
81 | + } | |
82 | + respBody, _, err := makeHttpRequestV4(post, o.IP+orderUrl, nil, map[string]interface{}{"jsonbody": string(_orderModel)}, nil) | |
83 | + if err != nil { | |
84 | + return "", err | |
85 | + } | |
86 | + | |
87 | + return respBody, nil | |
88 | +} | |
89 | + | |
90 | +func (o *OrderClient) checkParams(businessCode, reqBody string) error { | |
91 | + return nil | |
43 | 92 | } | ... | ... |
1 | -package common | |
1 | +package socialwork_sdk | |
2 | 2 | |
3 | 3 | import ( |
4 | 4 | "crypto" |
... | ... | @@ -10,7 +10,7 @@ import ( |
10 | 10 | "encoding/pem" |
11 | 11 | ) |
12 | 12 | |
13 | -func RsaSignWithMd5Hex(data string, prvKey string) (string, error) { | |
13 | +func rsaSignWithMd5Hex(data string, prvKey string) (string, error) { | |
14 | 14 | block, _ := pem.Decode([]byte(prvKey)) |
15 | 15 | privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes) |
16 | 16 | if err != nil { |
... | ... | @@ -29,3 +29,22 @@ func RsaSignWithMd5Hex(data string, prvKey string) (string, error) { |
29 | 29 | out := base64.StdEncoding.EncodeToString(signature) |
30 | 30 | return out, nil |
31 | 31 | } |
32 | + | |
33 | +func rsaVerifySignWithMd5Base64(originalData, signData, pubKey string) error { | |
34 | + sign, err := base64.StdEncoding.DecodeString(signData) | |
35 | + if err != nil { | |
36 | + return err | |
37 | + } | |
38 | + block, _ := pem.Decode([]byte(pubKey)) | |
39 | + pub, err := x509.ParsePKIXPublicKey(block.Bytes) | |
40 | + if err != nil { | |
41 | + return err | |
42 | + } | |
43 | + hash := md5.New() | |
44 | + hash.Write([]byte(originalData)) | |
45 | + err = rsa.VerifyPKCS1v15(pub.(*rsa.PublicKey), crypto.MD5, hash.Sum(nil), sign) | |
46 | + if err != nil { | |
47 | + return err | |
48 | + } | |
49 | + return nil | |
50 | +} | ... | ... |
请
注册
或
登录
后发表评论