package taskrpc 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 }