|
|
package CheckHandler
|
|
|
|
|
|
import (
|
|
|
"github.com/gin-gonic/gin"
|
|
|
"regexp"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
//在注释标签中声明的标识字符串：X-EmailLimit
|
|
|
var emailLimit = "x-emaillimit"
|
|
|
|
|
|
//形成一个map,用于O(1)速度获取到一个接口是不是需要进行判断拦截，注意，一个接口可以存在post+get两种形式，所以map里面是一个数组，需要遍历
|
|
|
type emailLimitStruct struct {
|
|
|
parameterName string
|
|
|
httpType string
|
|
|
must bool
|
|
|
}
|
|
|
|
|
|
//长度限制的map,以string为key(key的命名是接口名+"_get" 或者 接口名+"_post"),里面是emailLimitStruct形式的数组
|
|
|
var emailLimitMap = make(map[string][]emailLimitStruct)
|
|
|
|
|
|
/**
|
|
|
功能：功能函数，添当前的检查字典里
|
|
|
*/
|
|
|
func emailAddToMap(d map[string]interface{}, t map[string]interface{}, k string, httpType string) {
|
|
|
var emailLimitArray []emailLimitStruct
|
|
|
d = t[httpType].(map[string]interface{})
|
|
|
if d != nil && d[emailLimit] != nil {
|
|
|
for _, value := range d[emailLimit].([]interface{}) {
|
|
|
var t emailLimitStruct
|
|
|
t.parameterName = value.(string)
|
|
|
t.httpType = httpType
|
|
|
//此参数是否为必填写项
|
|
|
for _, v2 := range d["parameters"].([]interface{}) {
|
|
|
c2 := v2.(map[string]interface{})
|
|
|
for k2, v2 := range c2 {
|
|
|
if k2 == "required" {
|
|
|
t.must = v2.(bool) //标识是否是必须项
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
emailLimitArray = append(emailLimitArray, t)
|
|
|
}
|
|
|
emailLimitMap[k+"_"+httpType] = emailLimitArray
|
|
|
}
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：初始化邮箱限制字典
|
|
|
*/
|
|
|
func emailLimitInit(inter interface{}) map[string][]emailLimitStruct {
|
|
|
//参数检查器
|
|
|
m := inter.(map[string]interface{})
|
|
|
n := m["paths"].(map[string]interface{})
|
|
|
for k, v := range n {
|
|
|
t := v.(map[string]interface{})
|
|
|
//参数长度检查器
|
|
|
var d map[string]interface{}
|
|
|
if t["get"] != nil {
|
|
|
emailAddToMap(d, t, k, "get")
|
|
|
}
|
|
|
if t["post"] != nil {
|
|
|
emailAddToMap(d, t, k, "post")
|
|
|
}
|
|
|
}
|
|
|
return emailLimitMap
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：检查字符串是否是邮箱
|
|
|
作者：黄海
|
|
|
时间：2020-03-17
|
|
|
*/
|
|
|
func emailLimitIsLegal(c *gin.Context, interName string) (bool, ResultStruct) {
|
|
|
var resultStruct ResultStruct
|
|
|
//接口传入的请求方式:get?post?
|
|
|
codeHttpType := strings.ToLower(c.Request.Method)
|
|
|
//1、如果此接口地址需要进行参数长度检查
|
|
|
if emailLimitMap[interName+"_"+codeHttpType] != nil {
|
|
|
arr := emailLimitMap[interName+"_"+codeHttpType]
|
|
|
resultStruct.InterfaceName = interName
|
|
|
//遍历arr
|
|
|
for i := 0; i < len(arr); i++ {
|
|
|
parameterName := arr[i].parameterName
|
|
|
httpType := arr[i].httpType
|
|
|
var p string
|
|
|
if httpType == "get" {
|
|
|
p = c.Query(parameterName)
|
|
|
} else {
|
|
|
p = c.PostForm(parameterName)
|
|
|
}
|
|
|
resultStruct.Parameter = parameterName
|
|
|
//为空，不需要检查，通过
|
|
|
if p == "" && !arr[i].must {
|
|
|
return true, resultStruct
|
|
|
}
|
|
|
//为空，需要检查，禁止
|
|
|
if p == "" && arr[i].must{
|
|
|
resultStruct.Message = "参数没有找到!"
|
|
|
return false, resultStruct
|
|
|
}
|
|
|
if !verifyEmailFormat(p) {
|
|
|
resultStruct.HttpType = httpType
|
|
|
resultStruct.Message = "不是合法邮箱！"
|
|
|
return false, resultStruct
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
return true, resultStruct
|
|
|
}
|
|
|
func verifyEmailFormat(email string) bool {
|
|
|
pattern := `^[0-9a-z][_.0-9a-z-]{0,31}@([0-9a-z][0-9a-z-]{0,30}[0-9a-z]\.){1,4}[a-z]{2,4}$`
|
|
|
reg := regexp.MustCompile(pattern)
|
|
|
return reg.MatchString(email)
|
|
|
}
|
