102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package BitSrun
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
log "github.com/sirupsen/logrus"
|
||
"strings"
|
||
)
|
||
|
||
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
|
||
}
|
||
}
|
||
|
||
// 如果深澜分配的 ip 不是内网 ip，说明已经在线且拥有固定 ip
|
||
inet := strings.HasPrefix(ip, "192.168.") || strings.HasPrefix(ip, "10.") || strings.HasPrefix(ip, "172.")
|
||
return err.(string) == "ok" || !inet, 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
|
||
}
|