init

src/middleware/check_sign.go

@@ -0,0 +1,37 @@
+package middleware
+
+import (
+	"bytes"
+	"github.com/assimon/luuu/config"
+	"github.com/assimon/luuu/util/constant"
+	"github.com/assimon/luuu/util/json"
+	"github.com/assimon/luuu/util/sign"
+	"github.com/labstack/echo/v4"
+	"io/ioutil"
+)
+
+func CheckApiSign() echo.MiddlewareFunc {
+	return func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(ctx echo.Context) error {
+			params, err := ioutil.ReadAll(ctx.Request().Body)
+			if err != nil {
+				return constant.SignatureErr
+			}
+			m := make(map[string]interface{})
+			err = json.Cjson.Unmarshal(params, &m)
+			signature, ok := m["signature"]
+			if !ok {
+				return constant.SignatureErr
+			}
+			checkSignature, err := sign.Get(m, config.GetApiAuthToken())
+			if err != nil {
+				return constant.SignatureErr
+			}
+			if checkSignature != signature {
+				return constant.SignatureErr
+			}
+			ctx.Request().Body = ioutil.NopCloser(bytes.NewBuffer(params))
+			return next(ctx)
+		}
+	}
+}