提交 36f526f12608d3d5e89098cc1e0cfba7cf195438

作者 yuyang
1 个父辈 f129d0d1

update

... ... @@ -32,8 +32,6 @@ func aes_CBC_Encrypt(data []byte, key, iv []byte) (string, error) {
32 32 return "", err
33 33 }
34 34 origData := padding([]byte(data), block.BlockSize())
35   - println("block ", block.BlockSize())
36   - println("block2 ", len(iv))
37 35 blockMode := cipher.NewCBCEncrypter(block, iv)
38 36 crypted := make([]byte, len(origData))
39 37 blockMode.CryptBlocks(crypted, origData)
... ...
... ... @@ -35,6 +35,7 @@ KmUhzAS0v+cvRYEBdiD+ckC1i80KaGPsiSvKSgJIxEEo1urgsrGJBGP0uBMM50Bm
35 35 -----END PUBLIC KEY-----`
36 36 IP = "http://47.110.250.177:20000/"
37 37 //IP = "http://127.0.0.1:51393/"
  38 + PTCODE = "730088256896503808"
38 39 STR = `{"response_type":"code","client_id":"432437055486697481","redirect_uri":"https://dev3.workai.com.cn/wso-workai?state=socialwork_strait_certification_list","state":"client_credential","scope":"global_access:end_user","user_mobile":"13301169757","user_name":"sasl_dispose","callback":"","certification":"0","ptcode":"730088256896503808","path_type":"","path":""}`
39 40
40 41 V1DATA = "Wxy8h961GO12QmdOdwOYaUZdy6P+o67Olid6YbbjP+SiZNKYofmbQ33eyixsWXIP33sLIrrUrYZ/BNd1+iC8Bng9itBpIcJEfO7fJX9+8Sgp+5hORLESdPRoiyIjeQMX1JKpBk0Uiu7eSsxRRepi33oMi1+580LtFa91lc44PAwnrAEPoTZbEpPQQQmNPrGM0DzmZRUxEIT/SajlwoNg34DvKrGpsb2g8GZ7wUR8GuTOYUg2odF8WjTWNj2GWuzCUw8WPN8FGXdgTzRMn0glivn5SizGmJn0oOJZxo4JRG0Q6h0hV3xi8nPtvwjyOdy0Wspjcd4jQny1jaOKCaPg/5CRLWdmQKLE9d7UhSyQoQ6jzNxEkd/r7vAbtuvRfwvB6zN7NKODjVc/kJf7qovou8u6k2xBGQ7C1CT8wULjqRy3bAYfJ4A5YYWIwtMPJ/l1swlEYsalq5M4Y1g5RjnX1auMczdhnTCOzNeXm2tTJSQ="
... ... @@ -44,7 +45,7 @@ KmUhzAS0v+cvRYEBdiD+ckC1i80KaGPsiSvKSgJIxEEo1urgsrGJBGP0uBMM50Bm
44 45 var client *OrderClient
45 46
46 47 func init() {
47   - client = NewOrderClient(KEY, IV, PRIVATEKEY, PUBLICKEY, IP)
  48 + client = NewOrderClient(KEY, IV, PRIVATEKEY, PUBLICKEY, IP,PTCODE)
48 49 }
49 50
50 51 func TestCheckParams(t *testing.T) {
... ...
... ... @@ -83,9 +83,6 @@ func makeHttpRequestV4(method, url string, headers map[string]string, entity map
83 83 Jar: jar,
84 84 }
85 85 }
86   - // client.Transport = tr
87   -
88   - fmt.Printf("xuht================Request: %+v\n", *req)
89 86 res, err := client.Do(req)
90 87 if err != nil {
91 88 fmt.Println("faild to do the request with error ", err)
... ...
  1 +package main
  2 +
  3 +import (
  4 + "bufio"
  5 + "crypto/rand"
  6 + "crypto/rsa"
  7 + "crypto/x509"
  8 + "encoding/pem"
  9 + "flag"
  10 + socialwork_sdk "gitlab.workai.com.cn/chenang/socialwork-sdk"
  11 + "io/ioutil"
  12 + "os"
  13 +)
  14 +
  15 +
  16 +/*func init(){
  17 + var bits int
  18 + flag.IntVar(&bits, "b", 1024, "密钥长度,默认为1024位")
  19 + flag.Parse()
  20 +
  21 + if err := GenRsaKey(bits); err != nil {
  22 + log.Fatal("密钥文件生成失败!")
  23 + }
  24 + log.Println("密钥文件生成成功!")
  25 +}
  26 +*/
  27 +func main() {
  28 + var key string
  29 + var iv string
  30 + var privateKeyPath string
  31 + var publicKeyPath string
  32 + var ip string
  33 + var bussinessCode string
  34 + var ptCode string
  35 + var reqBody string
  36 + var errFilePath string
  37 + var resultFilePath string
  38 + flag.StringVar(&key, "key", "", "key")
  39 + flag.StringVar(&iv, "iv", "", "iv")
  40 + flag.StringVar(&privateKeyPath, "privateKeyPath", "", "privateKeyPath")
  41 + flag.StringVar(&publicKeyPath, "publicKeyPath", "", "publicKeyPath")
  42 + flag.StringVar(&ip, "ip", "", "ip")
  43 + flag.StringVar(&bussinessCode, "bussiness_code", "", "bussiness_code")
  44 + flag.StringVar(&ptCode, "ptCode", "", "ptCode")
  45 + flag.StringVar(&reqBody, "reqBody", "", "reqBody")
  46 + flag.StringVar(&errFilePath, "errFilePath", "", "errFilePath")
  47 + flag.StringVar(&resultFilePath, "resultFilePath", "", "resultFilePath")
  48 + flag.Parse()
  49 +
  50 +
  51 + fd, _ := os.OpenFile(errFilePath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
  52 + defer fd.Close()
  53 +
  54 + publicKey, err := ioutil.ReadFile(publicKeyPath)
  55 + if err != nil {
  56 + write := bufio.NewWriter(fd)
  57 + write.WriteString("命令行执行错误:" + err.Error() + "\r\n")
  58 + write.Flush()
  59 + return
  60 + }
  61 + privateKey, err := ioutil.ReadFile(privateKeyPath)
  62 + if err != nil {
  63 + write := bufio.NewWriter(fd)
  64 + write.WriteString("命令行执行错误:" + err.Error() + "\r\n")
  65 + write.Flush()
  66 + return
  67 + }
  68 +
  69 + client := socialwork_sdk.NewOrderClient(key, iv, string(privateKey), string(publicKey), ip, ptCode)
  70 + req, err := client.SendRequest(bussinessCode, reqBody)
  71 + if err != nil {
  72 + write := bufio.NewWriter(fd)
  73 + write.WriteString("命令行执行错误:" + err.Error() + "\r\n")
  74 + write.Flush()
  75 + return
  76 + } else {
  77 + file, err := os.Create(resultFilePath)
  78 + defer file.Close()
  79 +
  80 + if err != nil {
  81 + write := bufio.NewWriter(fd)
  82 + write.WriteString("命令行执行错误:" + err.Error() + "\r\n")
  83 + write.Flush()
  84 + return
  85 + }
  86 + file.WriteString(req)
  87 + return
  88 + }
  89 +}
  90 +
  91 +func GenRsaKey(bits int) error {
  92 + // 生成私钥文件
  93 + privateKey, err := rsa.GenerateKey(rand.Reader, bits)
  94 + if err != nil {
  95 + return err
  96 + }
  97 + derStream := x509.MarshalPKCS1PrivateKey(privateKey)
  98 + block := &pem.Block{
  99 + Type: "私钥",
  100 + Bytes: derStream,
  101 + }
  102 + file, err := os.Create("private.pem")
  103 + if err != nil {
  104 + return err
  105 + }
  106 + err = pem.Encode(file, block)
  107 + if err != nil {
  108 + return err
  109 + }
  110 + // 生成公钥文件
  111 + publicKey := &privateKey.PublicKey
  112 + derPkix, err := x509.MarshalPKIXPublicKey(publicKey)
  113 + if err != nil {
  114 + return err
  115 + }
  116 + block = &pem.Block{
  117 + Type: "公钥",
  118 + Bytes: derPkix,
  119 + }
  120 + file, err = os.Create("public.pem")
  121 + if err != nil {
  122 + return err
  123 + }
  124 + err = pem.Encode(file, block)
  125 + if err != nil {
  126 + return err
  127 + }
  128 + return nil
  129 +}
... ...
  1 +-----BEGIN PRIVATE KEY-----
  2 +MIICdwIBADANBgkqhkiG9w0BAQEFAASCAmEwggJdAgEAAoGBAL+HIjm9wYlMx5dR
  3 +hCGw9Uw5X1ZKPUfOvKXq/8mIuNVtH2ls6jz1fBlhoSHrYMjxY5t7EPSRZzOLeqGI
  4 +7FKbwlHPLAC4cN3njho8TE4Ao7EqipG7L81qHL6Wll/NrziP67+dDROG0ozscmNh
  5 +kq0JFxIkxQn3aKGbeEdJbLcLRRM7AgMBAAECgYEAn72Rah89v8nt3eL7FS6GKGDh
  6 +hxK8MZvX83oghHIn1roArOlOJDGKo5Ct16B9RJsvvInUBmcPcAow6XFNyUhntGnk
  7 +evnBZSPz8SJlwGMTTz/P4ebQURrcup5UejQzmBOSXrIXsrW4Ty0AdvnwM1OB/lmz
  8 +tQZyU3xMzK6QbXJkCgECQQDqWeR9/gBwWZLkZBPQASxWbb7NKwyCNN7clzMZv6MN
  9 +ea8sR7Zp6cNfPtfcVVTeHRX/sXRhB7inRyLi/ec3dSJBAkEA0TiGYiTv1l5prTCC
  10 +PK6lyShDK9SR8ZBcSxO5KsttFeOs3f9l2uU0ddd8fAuLJPm2xjd3RAbnH9dAfEC1
  11 +GXoeewJBAJ/aB0iFnIkGIHNXbEhCtJqTq0VJavPZ2nZ/UCTtGexUJ0+97shZ4qEy
  12 +PY2WSxhioKHTyn1xBdxcYoiRBklGxcECQAqDOZVi3744vjyajGC6MdBauo1QQAkQ
  13 +ROLs2cgBsrh/2SbKd3a5P5qQOkwNVG+GMJ3jGZYMdr6N1UqpAbCHB1MCQH1aSvf2
  14 ++CXW9fbkUz+qyjIWLN0pDyI+uh+zAHRRs9sL5UVOLfyU03W3120bUL7aekrbynP8
  15 +1+ihe8QolWZgvUk=
  16 +-----END PRIVATE KEY-----
... ...
  1 +-----BEGIN PUBLIC KEY-----
  2 +MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDHk0VRd3+CqwUHA2jC4dzrqe7/
  3 +pjX8BTbC7T7DnzAGO7iN09kf2abbzudLTM+xbOKr2wSTLkhffBhycZ4zlf3PuVu/
  4 +KmUhzAS0v+cvRYEBdiD+ckC1i80KaGPsiSvKSgJIxEEo1urgsrGJBGP0uBMM50Bm
  5 +3YZBKCNST3LB1JNSeQIDAQAB
  6 +-----END PUBLIC KEY-----
... ...
... ... @@ -21,13 +21,14 @@ func HelloWorld() {
21 21 fmt.Println("欢迎使用小爱,socialwork-sdk")
22 22 }
23 23
24   -func NewOrderClient(key, iv, customerPrivateKey, publicKey, ip string) *OrderClient {
  24 +func NewOrderClient(key, iv, customerPrivateKey, publicKey, ip, ptCode string) *OrderClient {
25 25 return &OrderClient{
26 26 Key: key,
27 27 IV: iv,
28 28 PublicKey: publicKey,
29 29 CustomerPrivateKey: customerPrivateKey,
30 30 IP: ip,
  31 + PtCode: ptCode,
31 32 }
32 33 }
33 34
... ... @@ -79,7 +80,6 @@ func (o *OrderClient) SendRequest(businessCode, reqBody string) (string, error)
79 80 PtCode: o.PtCode,
80 81 BusinessCode: businessCode,
81 82 }
82   -
83 83 _orderModel, err := json.Marshal(orderModel)
84 84 if err != nil {
85 85 return "", err
... ... @@ -102,98 +102,98 @@ func (o *OrderClient) checkParams(businessCode, reqBody string) error {
102 102 switch businessCode {
103 103 case "AUTHORIZE":
104 104 var authorize model.Authorize
105   - if err := json.Unmarshal([]byte(reqBody),&authorize); err != nil {
  105 + if err := json.Unmarshal([]byte(reqBody), &authorize); err != nil {
106 106 return err
107 107 }
108 108 return nil
109 109 case "CREATEORDER":
110 110 var createOrder model.CreateOrder
111   - if err := json.Unmarshal([]byte(reqBody),&createOrder); err != nil {
  111 + if err := json.Unmarshal([]byte(reqBody), &createOrder); err != nil {
112 112 return err
113 113 }
114 114 return nil
115 115 case "ADDFREEDOMEEMP":
116 116 var freedomEmps model.FreedomEmps
117   - if err := json.Unmarshal([]byte(reqBody),&freedomEmps); err != nil {
  117 + if err := json.Unmarshal([]byte(reqBody), &freedomEmps); err != nil {
118 118 return err
119 119 }
120 120 return nil
121 121 case "UPLOADATTACHMENT":
122 122 var uploadFreedomEmps model.UploadFreedomEmps
123   - if err := json.Unmarshal([]byte(reqBody),&uploadFreedomEmps); err != nil {
  123 + if err := json.Unmarshal([]byte(reqBody), &uploadFreedomEmps); err != nil {
124 124 return err
125 125 }
126 126 return nil
127 127 case "UPLOADATTACHMENTBYURL":
128 128 var uploadFreedomEmpsByUrl model.UploadFreedomEmpsByUrl
129   - if err := json.Unmarshal([]byte(reqBody),&uploadFreedomEmpsByUrl); err != nil {
  129 + if err := json.Unmarshal([]byte(reqBody), &uploadFreedomEmpsByUrl); err != nil {
130 130 return err
131 131 }
132 132 return nil
133 133 case "ORDERINFO":
134 134 var simpleOrderInfo model.SimpleOrderInfo
135   - if err := json.Unmarshal([]byte(reqBody),&simpleOrderInfo); err != nil {
  135 + if err := json.Unmarshal([]byte(reqBody), &simpleOrderInfo); err != nil {
136 136 return err
137 137 }
138 138 return nil
139 139 case "FREEDOMEMOSTATUS":
140 140 var freedomEmpStatus model.FreedomEmpStatus
141   - if err := json.Unmarshal([]byte(reqBody),&freedomEmpStatus); err != nil {
  141 + if err := json.Unmarshal([]byte(reqBody), &freedomEmpStatus); err != nil {
142 142 return err
143 143 }
144 144 return nil
145 145 case "SUBACCOUNTBALANCE":
146 146 var subAccountBalance model.SubAccountBalance
147   - if err := json.Unmarshal([]byte(reqBody),&subAccountBalance); err != nil {
  147 + if err := json.Unmarshal([]byte(reqBody), &subAccountBalance); err != nil {
148 148 return err
149 149 }
150 150 return nil
151 151 case "UPDATEORDERDETAIL":
152 152 var updateOrderDetail model.UpdateOrderDetail
153   - if err := json.Unmarshal([]byte(reqBody),&updateOrderDetail); err != nil {
  153 + if err := json.Unmarshal([]byte(reqBody), &updateOrderDetail); err != nil {
154 154 return err
155 155 }
156 156 return nil
157 157 case "EXISTRECEIPT":
158 158 var receipt model.Receipt
159   - if err := json.Unmarshal([]byte(reqBody),&receipt); err != nil {
  159 + if err := json.Unmarshal([]byte(reqBody), &receipt); err != nil {
160 160 return err
161 161 }
162 162 return nil
163 163 case "DOWNLOADRECEIPT":
164 164 var downloadReceipt model.DownloadReceipt
165   - if err := json.Unmarshal([]byte(reqBody),&downloadReceipt); err != nil {
  165 + if err := json.Unmarshal([]byte(reqBody), &downloadReceipt); err != nil {
166 166 return err
167 167 }
168 168 return nil
169 169 case "PAYORDER":
170 170 var payOrder model.PayOrder
171   - if err := json.Unmarshal([]byte(reqBody),&payOrder); err != nil {
  171 + if err := json.Unmarshal([]byte(reqBody), &payOrder); err != nil {
172 172 return err
173 173 }
174 174 return nil
175 175 case "USERINFO":
176 176 var getUserInfo model.GetUserInfo
177   - if err := json.Unmarshal([]byte(reqBody),&getUserInfo); err != nil {
  177 + if err := json.Unmarshal([]byte(reqBody), &getUserInfo); err != nil {
178 178 return err
179 179 }
180 180 return nil
181 181 case "PAYORDERV1":
182 182 var mergeCreateOrder model.MergeCreateOrder
183   - if err := json.Unmarshal([]byte(reqBody),&mergeCreateOrder); err != nil {
  183 + if err := json.Unmarshal([]byte(reqBody), &mergeCreateOrder); err != nil {
184 184 return err
185 185 }
186 186 return nil
187 187 case "UPDATEMOBILE":
188 188 var updateMobile model.UpdateMobile
189   - if err := json.Unmarshal([]byte(reqBody),&updateMobile); err != nil {
  189 + if err := json.Unmarshal([]byte(reqBody), &updateMobile); err != nil {
190 190 return err
191 191 }
192 192 return nil
193 193
194 194 case "CREATEEMPLOYMENTPOOL":
195 195 var createEmploymentPool model.CreateEmploymentPool
196   - if err := json.Unmarshal([]byte(reqBody),&createEmploymentPool); err != nil {
  196 + if err := json.Unmarshal([]byte(reqBody), &createEmploymentPool); err != nil {
197 197 return err
198 198 }
199 199 return nil
... ...
注册登录 后发表评论