|
@ -0,0 +1,106 @@ |
|
|
|
|
|
package weixinrpc |
|
|
|
|
|
|
|
|
|
|
|
import ( |
|
|
|
|
|
"encoding/json" |
|
|
|
|
|
"errors" |
|
|
|
|
|
"strconv" |
|
|
|
|
|
"time" |
|
|
|
|
|
|
|
|
|
|
|
"git.tetele.net/tgo/crypter" |
|
|
|
|
|
"github.com/golang/protobuf/proto" |
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
func SetResData(data interface{}, res *Response) { |
|
|
|
|
|
res_data_json, err := json.Marshal(data) |
|
|
|
|
|
if err == nil { |
|
|
|
|
|
|
|
|
|
|
|
encryData := crypter.DesEn(string(res_data_json), DES_KEY) |
|
|
|
|
|
|
|
|
|
|
|
now_str := strconv.FormatInt(time.Now().Unix(), 10) |
|
|
|
|
|
|
|
|
|
|
|
res_sign := Sign(encryData, now_str) |
|
|
|
|
|
|
|
|
|
|
|
res.Data = proto.String(encryData) |
|
|
|
|
|
res.Time = proto.String(now_str) |
|
|
|
|
|
res.Sign = proto.String(res_sign) |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func SetReqData(arg interface{}) (*Request, error) { |
|
|
|
|
|
data_json, err := json.Marshal(arg) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
return nil, err |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
now_int64 := time.Now().Unix() |
|
|
|
|
|
|
|
|
|
|
|
encryData := crypter.DesEn(string(data_json), DES_KEY) |
|
|
|
|
|
|
|
|
|
|
|
now := strconv.FormatInt(now_int64, 10) |
|
|
|
|
|
|
|
|
|
|
|
sign := Sign(encryData, now) |
|
|
|
|
|
|
|
|
|
|
|
return &Request{proto.String(encryData), proto.String(now), proto.String(sign), nil}, nil |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func GetReqData(req *Request) (string, error) { |
|
|
|
|
|
res_data := req.GetData() |
|
|
|
|
|
|
|
|
|
|
|
if res_data != "" { |
|
|
|
|
|
|
|
|
|
|
|
time_int64, err := strconv.ParseInt(req.GetTime(), 10, 64) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
return "", err |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
now_int64 := time.Now().Unix() |
|
|
|
|
|
|
|
|
|
|
|
if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 { |
|
|
|
|
|
//时间误差前后10秒,返回
|
|
|
|
|
|
return "", errors.New("返回时间错误") |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
check_sign := CheckSign(req.GetSign(), res_data, req.GetTime()) |
|
|
|
|
|
if !check_sign { |
|
|
|
|
|
return "", errors.New("返回数据签名错误") |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
//解密
|
|
|
|
|
|
return crypter.DesDe(res_data, DES_KEY), nil |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
return "", nil |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
func GetResData(res *Response) (string, error) { |
|
|
|
|
|
res_data := res.GetData() |
|
|
|
|
|
|
|
|
|
|
|
if res_data != "" { |
|
|
|
|
|
|
|
|
|
|
|
time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
return "", err |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
now_int64 := time.Now().Unix() |
|
|
|
|
|
|
|
|
|
|
|
if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 { |
|
|
|
|
|
//时间误差前后10秒,返回
|
|
|
|
|
|
return "", errors.New("返回时间错误") |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
check_sign := CheckSign(res.GetSign(), res_data, res.GetTime()) |
|
|
|
|
|
if !check_sign { |
|
|
|
|
|
return "", errors.New("返回数据签名错误") |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
//解密
|
|
|
|
|
|
return crypter.DesDe(res_data, DES_KEY), nil |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
return "", nil |
|
|
|
|
|
} |