55 lines
922 B
Go
55 lines
922 B
Go
|
package wsign
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"io/ioutil"
|
||
|
|
"service/bizcommon/util"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
Salt = "6%dkszcU*&lk@#$T#(lknkj.359"
|
||
|
|
)
|
||
|
|
|
||
|
|
func CalcSignature(s string) (signature string, err error) {
|
||
|
|
if len([]rune(s)) <= 0 {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if len([]rune(s)) <= 20 {
|
||
|
|
err = errors.New(fmt.Sprintf("to short s: %s", s))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
payload := []rune(s + Salt)
|
||
|
|
l := len(payload)
|
||
|
|
|
||
|
|
a1 := string(payload[:10])
|
||
|
|
a2 := string(payload[10 : l-10])
|
||
|
|
a3 := string(payload[l-10:])
|
||
|
|
|
||
|
|
signature = util.StringToMd5(a1 + a3 + a2)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func CheckSignature(ctx *gin.Context) bool {
|
||
|
|
signature := ctx.Query("signature")
|
||
|
|
if len(signature) <= 0 {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
if signature == "wishpalgogogo" {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
bs, _ := ioutil.ReadAll(ctx.Request.Body)
|
||
|
|
calcSignature, err := CalcSignature(string(bs))
|
||
|
|
if err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
if signature != calcSignature {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|