99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
package BitSrun
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func LoginStatus(c *Conf) (online bool, ip string, e error) {
|
|
c.initApi()
|
|
|
|
res, e := c.api.GetUserInfo()
|
|
if e != nil {
|
|
return false, "", e
|
|
}
|
|
|
|
err, ok := res["error"]
|
|
if !ok {
|
|
return false, "", ErrResultCannotFound
|
|
}
|
|
|
|
ipInterface, ok := res["client_ip"]
|
|
if !ok {
|
|
ipInterface, ok = res["online_ip"]
|
|
if !ok {
|
|
return false, "", ErrResultCannotFound
|
|
}
|
|
}
|
|
|
|
return err.(string) == "ok", ipInterface.(string), nil
|
|
}
|
|
|
|
func DoLogin(clientIP string, c *Conf) error {
|
|
log.Debugln("正在获取 Token")
|
|
|
|
if c.LoginInfo.Form.UserType != "" {
|
|
c.LoginInfo.Form.UserName += "@" + c.LoginInfo.Form.UserType
|
|
}
|
|
|
|
res, e := c.api.GetChallenge(c.LoginInfo.Form.UserName, clientIP)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
token, ok := res["challenge"]
|
|
if !ok {
|
|
return ErrResultCannotFound
|
|
}
|
|
tokenStr := token.(string)
|
|
log.Debugln("token: ", tokenStr)
|
|
|
|
log.Debugln("发送登录请求")
|
|
|
|
info, e := json.Marshal(map[string]string{
|
|
"username": c.LoginInfo.Form.UserName,
|
|
"password": c.LoginInfo.Form.PassWord,
|
|
"ip": clientIP,
|
|
"acid": c.LoginInfo.Meta.Acid,
|
|
"enc_ver": c.LoginInfo.Meta.Enc,
|
|
})
|
|
if e != nil {
|
|
return e
|
|
}
|
|
EncryptedInfo := "{SRBX1}" + Base64(XEncode(string(info), tokenStr))
|
|
Md5Str := Md5(tokenStr)
|
|
EncryptedMd5 := "{MD5}" + Md5Str
|
|
EncryptedChkstr := Sha1(
|
|
tokenStr + c.LoginInfo.Form.UserName + tokenStr + Md5Str +
|
|
tokenStr + c.LoginInfo.Meta.Acid + tokenStr + clientIP +
|
|
tokenStr + c.LoginInfo.Meta.N + tokenStr + c.LoginInfo.Meta.Type +
|
|
tokenStr + EncryptedInfo,
|
|
)
|
|
|
|
res, e = c.api.Login(
|
|
c.LoginInfo.Form.UserName,
|
|
EncryptedMd5,
|
|
c.LoginInfo.Meta.Acid,
|
|
clientIP,
|
|
EncryptedInfo,
|
|
EncryptedChkstr,
|
|
c.LoginInfo.Meta.N,
|
|
c.LoginInfo.Meta.Type,
|
|
)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
var result interface{}
|
|
result, ok = res["error"]
|
|
if !ok {
|
|
return ErrResultCannotFound
|
|
}
|
|
LoginResult := result.(string)
|
|
|
|
if LoginResult != "ok" {
|
|
return errors.New(LoginResult)
|
|
}
|
|
|
|
return nil
|
|
}
|