package mob import ( "applet/app/db" "applet/app/e" "applet/app/lib/sms" "applet/app/utils" "applet/app/utils/logx" "bytes" "crypto/cipher" "crypto/des" "crypto/md5" "encoding/base64" "encoding/hex" "encoding/json" "errors" "fmt" "io/ioutil" "net/http" "sort" "time" "github.com/gin-gonic/gin" "github.com/tidwall/gjson" ) const base string = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" // Mob is mob sdk var Mob *SDK // MobMap is 每个站长都要有自己的mob 对象 var MobMap map[string]*SDK // Init 初始化 func Init() { // 后续可能要传请求的上下文来获取对应的配置 // mob 目前都是我们来管理每个站长的app 所以使用template 库 //fmt.Println("Mob SDK init ....") ch := make(chan struct{}) // 只是做信号标志的话 空struct 更省点资源 MobMap = make(map[string]*SDK) // 初始化 for k, e := range db.DBs { m := db.SysCfgGetWithDb(e, k, "third_app_push_set") if m == "" { fmt.Printf("masterid:%s 找不到推送配置", k) continue } key := gjson.Get(m, "mobAppKey").String() secret := gjson.Get(m, "mobAppSecret").String() if key == "" || secret == "" { fmt.Println(k + ":mob no config") continue } // fmt.Println(k, key, secret) mob := new(SDK) mob.AppKey = key mob.AppSecret = secret MobMap[k] = mob fmt.Println(k + ":mob config success") } go func() { ch <- struct{}{} }() // 定时任务 go func(MobMap map[string]*SDK, ch chan struct{}) { <-ch ticker := time.NewTicker(time.Duration(time.Second * 15)) //每 15s 一次更新一次mob 配置 for range ticker.C { for k, e := range db.DBs { if err := e.Ping(); err != nil { logx.Info(err) continue } m := db.SysCfgGetWithDb(e, k, "third_app_push_set") if m == "" { fmt.Printf("masterid:%s 找不到推送配置", k) continue } key := gjson.Get(m, "mobAppKey").String() secret := gjson.Get(m, "mobAppSecret").String() if key == "" || secret == "" { fmt.Println(k + ":mob no config") continue } // fmt.Println(k, key, secret) mob := new(SDK) mob.AppKey = key mob.AppSecret = secret MobMap[k] = mob // fmt.Println(k + ":mob config success") } } }(MobMap, ch) } // GetMobSDK is 获取mob 的sdk func GetMobSDK(mid string) (*SDK, error) { return &SDK{AppKey: "30dc33054b635", AppSecret: "396e98c293130c9976fb7428b6b434d6"}, nil } // SDK is mob_push 的sdk type SDK struct { AppKey string AppSecret string } // MobFreeLogin is 秒验 func (s *SDK) MobFreeLogin(args map[string]interface{}) (string, error) { var url string = "http://identify.verify.mob.com/auth/auth/sdkClientFreeLogin" // https://www.mob.com/wiki/detailed/?wiki=miaoyan_for_fuwuduan_mianmifuwuduanjieru&id=78 //加appkey args["appkey"] = s.AppKey //加签名 args["sign"] = generateSign(args, s.AppSecret) b, err := json.Marshal(args) if err != nil { return "", logx.Warn(err) } // 发送请求 respBody, err := httpPostBody(url, b) if err != nil { return "", logx.Warn(err) } // 反序列化 ret := struct { Status int `json:"status"` Error string `json:"error"` Res interface{} `json:"res"` }{} // 要拿 ret 里面 Res 再解密 if err := json.Unmarshal(respBody, &ret); err != nil { return "", logx.Warn(err) } //fmt.Println(ret) // ret里面的Res 反序列化为结构体 res := struct { IsValid int `json:"isValid"` Phone string `json:"phone"` }{} // 判断是否返回正确 状态码 if ret.Status == 200 { decode, _ := base64Decode([]byte(ret.Res.(string))) decr, _ := desDecrypt(decode, []byte(s.AppSecret)[0:8]) if err := json.Unmarshal(decr, &res); err != nil { return "", logx.Warn(err) } } // 有效则拿出res 里的电话号码 if res.IsValid == 1 { return res.Phone, nil } // Status 不等于200 则返回空 return "", fmt.Errorf("Mob error , status code %v ", ret.Status) } // MobSMS is mob 的短信验证 func (s *SDK) MobSMS(c *gin.Context, args map[string]interface{}) (bool, error) { // mob 的短信验证 // https://www.mob.com/wiki/detailed/?wiki=SMSSDK_for_yanzhengmafuwuduanxiaoyanjiekou&id=23 url := "https://webapi.sms.mob.com/sms/verify" //加appkey args["appkey"] = s.AppKey fmt.Println(args) //fmt.Println(args) // 发送请求 respBody, err := utils.CurlPost(url, args, nil) if err != nil { fmt.Println(err) return false, logx.Warn(err) } fmt.Println("=======================mob") fmt.Println("mob", string(respBody)) code := gjson.GetBytes(respBody, "status").Int() if code == 468 { return false, errors.New("验证码错误") } if code != 200 { utils.FilePutContents("sms", string(respBody)) return false, errors.New("验证码错误~") } return true, nil } func pkcs5UnPadding(origData []byte) []byte { length := len(origData) // 去掉最后一个字节 unpadding 次 unpadding := int(origData[length-1]) return origData[:(length - unpadding)] } func desDecrypt(crypted, key []byte) ([]byte, error) { block, err := des.NewCipher(key) if err != nil { return nil, err } blockMode := cipher.NewCBCDecrypter(block, []byte("00000000")) origData := make([]byte, len(crypted)) // origData := crypted blockMode.CryptBlocks(origData, crypted) origData = pkcs5UnPadding(origData) // origData = ZeroUnPadding(origData) return origData, nil } func base64Decode(src []byte) ([]byte, error) { var coder *base64.Encoding coder = base64.NewEncoding(base) return coder.DecodeString(string(src)) } func httpPostBody(url string, msg []byte) ([]byte, error) { resp, err := http.Post(url, "application/json;charset=utf-8", bytes.NewBuffer(msg)) if err != nil { return []byte(""), err } defer resp.Body.Close() body, err := ioutil.ReadAll(resp.Body) return body, err } func generateSign(request map[string]interface{}, secret string) string { ret := "" var keys []string for k := range request { keys = append(keys, k) } sort.Strings(keys) for _, k := range keys { ret = ret + fmt.Sprintf("%v=%v&", k, request[k]) } ret = ret[:len(ret)-1] + secret md5Ctx := md5.New() md5Ctx.Write([]byte(ret)) cipherStr := md5Ctx.Sum(nil) return hex.EncodeToString(cipherStr) } func Check(c *gin.Context, phone, zone, validCode string, ok1 bool) (bool, error) { smsPlatform := sms.GetSmsPlatform(c) if smsPlatform == "mob" { mob1, errr := GetMobSDK(c.GetString("master_id")) if errr != nil { return false, e.NewErr(400, "mob配置错误") } send := map[string]interface{}{ "phone": phone, "zone": zone, "code": validCode, } if zone == "" { send["zone"] = "86" } c.Set("not_deduction_doing", "1") ok, err := mob1.MobSMS(c, send) fmt.Println(ok) if err != nil { fmt.Println(err) return false, e.NewErr(400, "验证码校验错误") } return ok, nil } return ok1, nil }