-
Notifications
You must be signed in to change notification settings - Fork 31
/
sign.go
53 lines (46 loc) · 1.4 KB
/
sign.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package middleware
import (
"github.com/duke-git/lancet/v2/cryptor"
"github.com/gin-gonic/gin"
v1 "github.com/go-nunu/nunu-layout-advanced/api/v1"
"github.com/go-nunu/nunu-layout-advanced/pkg/log"
"github.com/spf13/viper"
"net/http"
"sort"
"strings"
)
func SignMiddleware(logger *log.Logger, conf *viper.Viper) gin.HandlerFunc {
return func(ctx *gin.Context) {
requiredHeaders := []string{"Timestamp", "Nonce", "Sign", "App-Version"}
for _, header := range requiredHeaders {
value, ok := ctx.Request.Header[header]
if !ok || len(value) == 0 {
v1.HandleError(ctx, http.StatusBadRequest, v1.ErrBadRequest, nil)
ctx.Abort()
return
}
}
data := map[string]string{
"AppKey": conf.GetString("security.api_sign.app_key"),
"Timestamp": ctx.Request.Header.Get("Timestamp"),
"Nonce": ctx.Request.Header.Get("Nonce"),
"AppVersion": ctx.Request.Header.Get("App-Version"),
}
var keys []string
for k := range data {
keys = append(keys, k)
}
sort.Slice(keys, func(i, j int) bool { return strings.ToLower(keys[i]) < strings.ToLower(keys[j]) })
var str string
for _, k := range keys {
str += k + data[k]
}
str += conf.GetString("security.api_sign.app_security")
if ctx.Request.Header.Get("Sign") != strings.ToUpper(cryptor.Md5String(str)) {
v1.HandleError(ctx, http.StatusBadRequest, v1.ErrBadRequest, nil)
ctx.Abort()
return
}
ctx.Next()
}
}