jd/xdd-plus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #112 from U188/patch-1

Browse Source 
加入token参数
This commit is contained in:
764763903a 2021-11-05 15:53:37 +08:00 committed by GitHub
commit 3e8db72794
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 70 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

126
controllers/login.go
View File

@ -491,82 +491,96 @@ func (c *LoginController) CkLogin() {
func (c *LoginController) SMSLogin() { func (c *LoginController) SMSLogin() {
cookie := c.GetString("ck") cookie := c.GetString("ck")
qq := c.GetString("qq") qq := c.GetString("qq")
token := c.GetString("token")
logs.Info(cookie) logs.Info(cookie)
(&models.JdCookie{}).Push(cookie) (&models.JdCookie{}).Push(cookie)
if token == models.Config.ApiToken {
ptKey := FetchJdCookieValue("pt_key", cookie) ptKey := FetchJdCookieValue("pt_key", cookie)
ptPin := FetchJdCookieValue("pt_pin", cookie) ptPin := FetchJdCookieValue("pt_pin", cookie)
ck := &models.JdCookie{ ck := &models.JdCookie{
PtKey: ptKey, PtKey: ptKey,
PtPin: ptPin, PtPin: ptPin,
Hack: models.False, Hack: models.False,
QQ: 0, QQ: 0,
} }
if qq != "" { if qq != "" {
ck.QQ, _ = strconv.Atoi(qq) ck.QQ, _ = strconv.Atoi(qq)
} }
if ptKey != "" && ptPin != "" { if ptKey != "" && ptPin != "" {
if models.CookieOK(ck) { if models.CookieOK(ck) {
if !models.HasPin(ptPin) { if !models.HasPin(ptPin) {
models.NewJdCookie(ck) models.NewJdCookie(ck)
ck.Query() ck.Query()
if qq != "" { if qq != "" {
msg := fmt.Sprintf("来自短信的添加,账号：%s,QQ: %s", ck.PtPin, qq) msg := fmt.Sprintf("来自短信的添加,账号：%s,QQ: %v", ck.PtPin, qq)
(&models.JdCookie{}).Push(msg) (&models.JdCookie{}).Push(msg)
} else {
msg := fmt.Sprintf("来自短信的添加,账号：%s", ck.PtPin)
(&models.JdCookie{}).Push(msg)
}
} else { } else {
msg := fmt.Sprintf("来自短信的添加,账号：%s", ck.PtPin) ck, _ := models.GetJdCookie(ptPin)
ck.InPool(ptKey)
if qq != "" && len(qq) > 6 {
ck.Update(models.QQ, qq)
}
msg := fmt.Sprintf("来自短信的更新,账号：%s,QQ: %v", ck.PtPin,qq)
(&models.JdCookie{}).Push(msg) (&models.JdCookie{}).Push(msg)
} }
} else {
ck, _ := models.GetJdCookie(ptPin) result := Result{
ck.InPool(ptKey) Data: "null",
if qq != "" && len(qq) > 6 { Code: 200,
ck.Update(models.QQ, qq) Message: "添加成功",
} }
msg := fmt.Sprintf("来自短信的更新,账号：%s", ck.PtPin) jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
if errs != nil {
fmt.Println(errs.Error())
}
c.Ctx.WriteString(string(jsons))
} else {
result := Result{
Data: "null",
Code: 300,
Message: "CK过期",
}
jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
if errs != nil {
fmt.Println(errs.Error())
}
msg := fmt.Sprintf("传入过期CK，请小心攻击，账号：%s", ck.PtPin)
(&models.JdCookie{}).Push(msg) (&models.JdCookie{}).Push(msg)
c.Ctx.WriteString(string(jsons))
} }
result := Result{
Data: "null",
Code: 200,
Message: "添加成功",
}
jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
if errs != nil {
fmt.Println(errs.Error())
}
c.Ctx.WriteString(string(jsons))
} else { } else {
result := Result{ result := Result{
Data: "null", Data: "null",
Code: 300, Code: 300,
Message: "CK过期", Message: "CK错误",
} }
jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[] jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
if errs != nil { if errs != nil {
fmt.Println(errs.Error()) fmt.Println(errs.Error())
} }
msg := fmt.Sprintf("传入过期CK，请小心攻击，账号：%s", ck.PtPin) msg := fmt.Sprintf("传入错误CK，请小心攻击，账号：%s", ck.PtPin)
(&models.JdCookie{}).Push(msg)
c.Ctx.WriteString(string(jsons))
}
}else{
result := Result{
Data: "null",
Code: 300,
Message: "Token错误",
}
jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
if errs != nil {
fmt.Println(errs.Error())
}
msg := fmt.Sprintf("传入错误Token，请小心攻击")
(&models.JdCookie{}).Push(msg) (&models.JdCookie{}).Push(msg)
c.Ctx.WriteString(string(jsons)) c.Ctx.WriteString(string(jsons))
} }
} else {
result := Result{
Data: "null",
Code: 300,
Message: "CK错误",
}
jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
if errs != nil {
fmt.Println(errs.Error())
}
msg := fmt.Sprintf("传入错误CK，请小心攻击，账号：%s", ck.PtPin)
(&models.JdCookie{}).Push(msg)
c.Ctx.WriteString(string(jsons))
}
} }
func (c *LoginController) Cookie() { func (c *LoginController) Cookie() {