forked from golang/hotime
101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
|
package pay
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/fatih/structs"
|
|||
|
|
"github.com/silenceper/wechat/util"
|
|||
|
|
"github.com/spf13/cast"
|
|||
|
|
"reflect"
|
|||
|
|
"sort"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// doc: https://pay.weixin.qq.com/wiki/doc/api/jsapi.php?chapter=9_7&index=8
|
|||
|
|
|
|||
|
|
// NotifyResult 下单回调
|
|||
|
|
type NotifyResult struct {
|
|||
|
|
ReturnCode *string `xml:"return_code"`
|
|||
|
|
ReturnMsg *string `xml:"return_msg"`
|
|||
|
|
|
|||
|
|
AppID *string `xml:"appid" json:"appid"`
|
|||
|
|
MchID *string `xml:"mch_id"`
|
|||
|
|
DeviceInfo *string `xml:"device_info"`
|
|||
|
|
NonceStr *string `xml:"nonce_str"`
|
|||
|
|
Sign *string `xml:"sign"`
|
|||
|
|
SignType *string `xml:"sign_type"`
|
|||
|
|
ResultCode *string `xml:"result_code"`
|
|||
|
|
ErrCode *string `xml:"err_code"`
|
|||
|
|
ErrCodeDes *string `xml:"err_code_des"`
|
|||
|
|
OpenID *string `xml:"openid"`
|
|||
|
|
IsSubscribe *string `xml:"is_subscribe"`
|
|||
|
|
TradeType *string `xml:"trade_type"`
|
|||
|
|
BankType *string `xml:"bank_type"`
|
|||
|
|
TotalFee *int `xml:"total_fee"`
|
|||
|
|
SettlementTotalFee *int `xml:"settlement_total_fee"`
|
|||
|
|
FeeType *string `xml:"fee_type"`
|
|||
|
|
CashFee *string `xml:"cash_fee"`
|
|||
|
|
CashFeeType *string `xml:"cash_fee_type"`
|
|||
|
|
CouponFee *int `xml:"coupon_fee"`
|
|||
|
|
CouponCount *int `xml:"coupon_count"`
|
|||
|
|
|
|||
|
|
// coupon_type_$n 这里只声明 3 个，如果有更多的可以自己组合
|
|||
|
|
CouponType0 *string `xml:"coupon_type_0"`
|
|||
|
|
CouponType1 *string `xml:"coupon_type_1"`
|
|||
|
|
CouponType2 *string `xml:"coupon_type_2"`
|
|||
|
|
CouponID0 *string `xml:"coupon_id_0"`
|
|||
|
|
CouponID1 *string `xml:"coupon_id_1"`
|
|||
|
|
CouponID2 *string `xml:"coupon_id_2"`
|
|||
|
|
CouponFeed0 *string `xml:"coupon_fee_0"`
|
|||
|
|
CouponFeed1 *string `xml:"coupon_fee_1"`
|
|||
|
|
CouponFeed2 *string `xml:"coupon_fee_2"`
|
|||
|
|
|
|||
|
|
TransactionID *string `xml:"transaction_id"`
|
|||
|
|
OutTradeNo *string `xml:"out_trade_no"`
|
|||
|
|
Attach *string `xml:"attach"`
|
|||
|
|
TimeEnd *string `xml:"time_end"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NotifyResp 消息通知返回
|
|||
|
|
type NotifyResp struct {
|
|||
|
|
ReturnCode string `xml:"return_code"`
|
|||
|
|
ReturnMsg string `xml:"return_msg"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// VerifySign 验签
|
|||
|
|
func (pcf *Pay) VerifySign(notifyRes NotifyResult) bool {
|
|||
|
|
// STEP1, 转换 struct 为 map，并对 map keys 做排序
|
|||
|
|
resMap := structs.Map(notifyRes)
|
|||
|
|
|
|||
|
|
sortedKeys := make([]string, 0, len(resMap))
|
|||
|
|
for k := range resMap {
|
|||
|
|
sortedKeys = append(sortedKeys, k)
|
|||
|
|
}
|
|||
|
|
sort.Strings(sortedKeys)
|
|||
|
|
|
|||
|
|
// STEP2, 对key=value的键值对用&连接起来，略过空值 & sign
|
|||
|
|
var signStrings string
|
|||
|
|
for _, k := range sortedKeys {
|
|||
|
|
value := fmt.Sprintf("%v", cast.ToString(resMap[k]))
|
|||
|
|
if value != "" && strings.ToLower(k) != "sign" {
|
|||
|
|
signStrings = signStrings + getTagKeyName(k, ¬ifyRes) + "=" + value + "&"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// STEP3, 在键值对的最后加上key=API_KEY
|
|||
|
|
signStrings = signStrings + "key=" + pcf.PayKey
|
|||
|
|
|
|||
|
|
// STEP4, 进行MD5签名并且将所有字符转为大写.
|
|||
|
|
sign := util.MD5Sum(signStrings)
|
|||
|
|
if sign != *notifyRes.Sign {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func getTagKeyName(key string, notifyRes *NotifyResult) string {
|
|||
|
|
s := reflect.TypeOf(notifyRes).Elem()
|
|||
|
|
f, _ := s.FieldByName(key)
|
|||
|
|
name := f.Tag.Get("xml")
|
|||
|
|
return name
|
|||
|
|
}
|