diff --git a/common/config/config.go b/common/config/config.go index 9ed1ea1d..b77060fd 100644 --- a/common/config/config.go +++ b/common/config/config.go @@ -10,12 +10,13 @@ var CONF *Config = nil // Config 配置 type Config struct { - HttpPort int32 `toml:"http_port"` - Logger Logger `toml:"logger"` - Database Database `toml:"database"` - Redis Redis `toml:"redis"` - Hk4e Hk4e `toml:"hk4e"` - MQ MQ `toml:"mq"` + HttpPort int32 `toml:"http_port"` + Logger Logger `toml:"logger"` + Database Database `toml:"database"` + Redis Redis `toml:"redis"` + Hk4e Hk4e `toml:"hk4e"` + Hk4eRobot Hk4eRobot `toml:"hk4e_robot"` + MQ MQ `toml:"mq"` } // Logger 日志 @@ -36,7 +37,7 @@ type Redis struct { Password string `toml:"password"` } -// Hk4e 原神相关 +// Hk4e 原神服务器 type Hk4e struct { KcpAddr string `toml:"kcp_addr"` // 该地址只用来注册到节点服务器 填网关的外网地址 网关本地监听为0.0.0.0 KcpPort int32 `toml:"kcp_port"` @@ -51,6 +52,22 @@ type Hk4e struct { DispatchUrl string `toml:"dispatch_url"` // 二级dispatch地址 将域名改为dispatch的外网地址 } +// Hk4eRobot 原神机器人 +type Hk4eRobot struct { + RegionListUrl string `toml:"region_list_url"` // 一级dispatch地址 + RegionListParam string `toml:"region_list_param"` // 一级dispatch的url参数 + CurRegionUrl string `toml:"cur_region_url"` // 二级dispatch地址 + CurRegionParam string `toml:"cur_region_param"` // 二级dispatch的url参数 + KeyId string `toml:"key_id"` // 客户端密钥编号 + LoginSdkUrl string `toml:"login_sdk_url"` // sdk登录服务器地址 + Account string `toml:"account"` // 帐号 + Password string `toml:"password"` // base64编码的rsa公钥加密后的密码 + ClientVersion string `toml:"client_version"` // 客户端版本号 + DosEnable bool `toml:"dos_enable"` // 是否开启压力测试 + DosNum int32 `toml:"dos_num"` // 压力测试并发数量 帐号自动添加后缀编号 + DosLoopLogin bool `toml:"dos_loop_login"` // 压力测试是否循环登录退出 +} + // MQ 消息队列 type MQ struct { NatsUrl string `toml:"nats_url"` diff --git a/gate/kcp/session.go b/gate/kcp/session.go index a567393f..24ce7d02 100644 --- a/gate/kcp/session.go +++ b/gate/kcp/session.go @@ -903,39 +903,33 @@ func DialWithOptions(raddr string) (*UDPSession, error) { network = "udp" } - conn, err := net.ListenUDP(network, nil) + conn, err := net.DialUDP(network, nil, udpaddr) if err != nil { return nil, errors.WithStack(err) } - - hsconn, err := net.DialUDP(network, nil, udpaddr) - if err != nil { - return nil, err - } enet := &Enet{ - Addr: raddr, + Addr: udpaddr.String(), ConvId: 0, ConnType: ConnEnetSyn, EnetType: EnetClientConnectKey, } data := BuildEnet(enet.ConnType, enet.EnetType, enet.ConvId) - _, err = hsconn.Write(data) + _, err = conn.Write(data) if err != nil { - return nil, err + return nil, errors.WithStack(err) } buf := make([]byte, mtuLimit) - n, addr, err := hsconn.ReadFrom(buf) + n, addr, err := conn.ReadFrom(buf) if err != nil { - return nil, err + return nil, errors.WithStack(err) } - if addr.String() != raddr { - // TODO 本质是为了安全考虑 但是用域名连接会出现这种情况看之后找个方法解决一下 - // return nil, errors.New("recv packet remote addr not match") + if addr.String() != udpaddr.String() { + return nil, errors.WithStack(errors.New("recv packet remote addr not match")) } udpPayload := buf[:n] connType, enetType, conv, err := ParseEnet(udpPayload) if err != nil || connType != ConnEnetEst || enetType != EnetClientConnectKey { - return nil, errors.New("recv packet format error") + return nil, errors.WithStack(errors.New("recv packet format error")) } return newUDPSession(conv, nil, conn, true, udpaddr), nil diff --git a/gate/net/kcp_connect_manager.go b/gate/net/kcp_connect_manager.go index 9f4431a8..13290878 100644 --- a/gate/net/kcp_connect_manager.go +++ b/gate/net/kcp_connect_manager.go @@ -150,6 +150,7 @@ func (k *KcpConnectManager) acceptHandle(listener *kcp.Listener) { } conn.SetACKNoDelay(true) conn.SetWriteDelay(false) + conn.SetWindowSize(255, 255) atomic.AddInt32(&CLIENT_CONN_NUM, 1) logger.Info("client connect, convId: %v", convId) kcpRawSendChan := make(chan *ProtoMsg, 1000) diff --git a/gate/net/kcp_endecode.go b/gate/net/kcp_endecode.go index 00a46ff7..e6e26c77 100644 --- a/gate/net/kcp_endecode.go +++ b/gate/net/kcp_endecode.go @@ -94,7 +94,13 @@ func EncodePayloadToBin(kcpMsg *KcpMsg, xorKey []byte) (bin []byte) { if kcpMsg.ProtoData == nil { kcpMsg.ProtoData = make([]byte, 0) } - bin = make([]byte, len(kcpMsg.HeadData)+len(kcpMsg.ProtoData)+12) + // 检查长度 + packetLen := len(kcpMsg.HeadData) + len(kcpMsg.ProtoData) + 12 + if packetLen > PacketMaxLen { + logger.Error("packet len too long") + return make([]byte, 0) + } + bin = make([]byte, packetLen) // 头部幻数 bin[0] = 0x45 bin[1] = 0x67 diff --git a/robot/cmd/application.toml b/robot/cmd/application.toml index 12346ebe..07bef0c5 100644 --- a/robot/cmd/application.toml +++ b/robot/cmd/application.toml @@ -3,3 +3,20 @@ level = "DEBUG" mode = "CONSOLE" track = true max_size = 10485760 + +[hk4e] +client_proto_proxy_enable = false + +[hk4e_robot] +region_list_url = "https://hk4e.flswld.com" # 一级dispatch地址 +region_list_param = "" # 一级dispatch的url参数 +cur_region_url = "https://hk4e.flswld.com/query_cur_region" # 二级dispatch地址 +cur_region_param = "?version=OSRELWin3.2.0&key_id=5" # 二级dispatch的url参数 +key_id = "5" # 客户端密钥编号 +login_sdk_url = "https://hk4e.flswld.com" # sdk登录服务器地址 +account = "flswld" # 帐号 +password = "" # base64编码的rsa公钥加密后的密码 +client_version = "CNRELWin3.2.0_R11611027_S11212885_D11793813" # 客户端版本号 +dos_enable = false # 是否开启压力测试 +dos_num = 1 # 压力测试并发数量 帐号自动添加后缀编号 +dos_loop_login = false # 压力测试是否循环登录退出 diff --git a/robot/cmd/main.go b/robot/cmd/main.go index 8a9a68e9..4cbc2872 100644 --- a/robot/cmd/main.go +++ b/robot/cmd/main.go @@ -2,12 +2,16 @@ package main import ( "encoding/base64" + "encoding/hex" "os" "os/signal" "strconv" "syscall" "time" + "hk4e/pkg/endec" + "hk4e/robot/net" + "hk4e/common/config" "hk4e/pkg/logger" "hk4e/protocol/cmd" @@ -22,8 +26,6 @@ func main() { logger.CloseLogger() }() - config.GetConfig().Hk4e.ClientProtoProxyEnable = false - // // DPDK模式需开启 // err := engine.InitEngine("00:0C:29:3E:3E:DF", "192.168.199.199", "255.255.255.0", "192.168.199.1") // if err != nil { @@ -32,8 +34,12 @@ func main() { // engine.RunEngine([]int{0, 1, 2, 3}, 4, 1, "0.0.0.0") // time.Sleep(time.Second * 30) - for i := 0; i < 1; i++ { - go runRobot("test_" + strconv.Itoa(i)) + if config.GetConfig().Hk4eRobot.DosEnable { + for i := 0; i < int(config.GetConfig().Hk4eRobot.DosNum); i++ { + go httpLogin(config.GetConfig().Hk4eRobot.Account + "_" + strconv.Itoa(i)) + } + } else { + httpLogin(config.GetConfig().Hk4eRobot.Account) } c := make(chan os.Signal, 1) @@ -52,31 +58,63 @@ func main() { } } -func runRobot(name string) { - logger.Info("robot start, name: %v", name) - dispatchInfo, err := login.GetDispatchInfo("https://hk4e.flswld.com", - "https://hk4e.flswld.com/query_cur_region", - "", - "?version=OSRELWin3.2.0&key_id=5", - "5") +func httpLogin(account string) { + logger.Info("robot start, account: %v", account) + dispatchInfo, err := login.GetDispatchInfo(config.GetConfig().Hk4eRobot.RegionListUrl, + config.GetConfig().Hk4eRobot.RegionListParam, + config.GetConfig().Hk4eRobot.CurRegionUrl, + config.GetConfig().Hk4eRobot.CurRegionParam, + config.GetConfig().Hk4eRobot.KeyId) if err != nil { logger.Error("get dispatch info error: %v", err) return } - accountInfo, err := login.AccountLogin("https://hk4e.flswld.com", name+"@@12345678", base64.StdEncoding.EncodeToString([]byte{0x00})) + accountInfo, err := login.AccountLogin(config.GetConfig().Hk4eRobot.LoginSdkUrl, account, config.GetConfig().Hk4eRobot.Password) if err != nil { logger.Error("account login error: %v", err) return } - session, err := login.GateLogin(dispatchInfo, accountInfo, "5") + for { + gateLogin(account, dispatchInfo, accountInfo) + if !config.GetConfig().Hk4eRobot.DosLoopLogin { + break + } + time.Sleep(time.Second) + continue + } +} + +func gateLogin(account string, dispatchInfo *login.DispatchInfo, accountInfo *login.AccountInfo) { + session, err := login.GateLogin(dispatchInfo, accountInfo, config.GetConfig().Hk4eRobot.KeyId) if err != nil { logger.Error("gate login error: %v", err) return } + clientVersionHashData, err := hex.DecodeString( + endec.Sha1Str(config.GetConfig().Hk4eRobot.ClientVersion + session.ClientVersionRandomKey + "mhy2020"), + ) + if err != nil { + logger.Error("gen clientVersionHashData error: %v", err) + return + } session.SendMsg(cmd.PlayerLoginReq, &proto.PlayerLoginReq{ - AccountUid: strconv.Itoa(int(accountInfo.AccountId)), - Token: accountInfo.ComboToken, + AccountType: 1, + SubChannelId: 1, + LanguageType: 2, + PlatformType: 3, + Checksum: "$008094416f86a051270e64eb0b405a38825", + ChecksumClientVersion: "CNRELWin3.2.0", + ClientDataVersion: 11793813, + ClientVerisonHash: base64.StdEncoding.EncodeToString(clientVersionHashData), + ClientVersion: config.GetConfig().Hk4eRobot.ClientVersion, + SecurityCmdReply: session.SecurityCmdBuffer, + SecurityLibraryMd5: "574a507ffee2eb6f997d11f71c8ae1fa", + Token: accountInfo.ComboToken, }) + clientLogic(account, session) +} + +func clientLogic(account string, session *net.Session) { ticker := time.NewTicker(time.Second) pingSeq := uint32(0) for { @@ -92,20 +130,32 @@ func runRobot(name string) { // 从这个管道接收服务器发来的消息 logger.Debug("recv protoMsg: %v", protoMsg) switch protoMsg.CmdId { + case cmd.PlayerLoginRsp: + rsp := protoMsg.PayloadMessage.(*proto.PlayerLoginRsp) + logger.Info("login ok, rsp: %v", rsp) case cmd.DoSetPlayerBornDataNotify: session.SendMsg(cmd.SetPlayerBornDataReq, &proto.SetPlayerBornDataReq{ AvatarId: 10000007, - NickName: name, + NickName: account, }) case cmd.PlayerEnterSceneNotify: ntf := protoMsg.PayloadMessage.(*proto.PlayerEnterSceneNotify) session.SendMsg(cmd.EnterSceneReadyReq, &proto.EnterSceneReadyReq{EnterSceneToken: ntf.EnterSceneToken}) + case cmd.EnterSceneReadyRsp: + ntf := protoMsg.PayloadMessage.(*proto.EnterSceneReadyRsp) session.SendMsg(cmd.SceneInitFinishReq, &proto.SceneInitFinishReq{EnterSceneToken: ntf.EnterSceneToken}) + case cmd.SceneInitFinishRsp: + ntf := protoMsg.PayloadMessage.(*proto.SceneInitFinishRsp) session.SendMsg(cmd.EnterSceneDoneReq, &proto.EnterSceneDoneReq{EnterSceneToken: ntf.EnterSceneToken}) + case cmd.EnterSceneDoneRsp: + ntf := protoMsg.PayloadMessage.(*proto.EnterSceneDoneRsp) session.SendMsg(cmd.PostEnterSceneReq, &proto.PostEnterSceneReq{EnterSceneToken: ntf.EnterSceneToken}) + if config.GetConfig().Hk4eRobot.DosLoopLogin { + return + } } case <-session.DeadEvent: - logger.Info("robot exit, name: %v", name) + logger.Info("robot exit, account: %v", account) return } } diff --git a/robot/login/gate_login.go b/robot/login/gate_login.go index 9710b66d..a241afb4 100644 --- a/robot/login/gate_login.go +++ b/robot/login/gate_login.go @@ -52,6 +52,10 @@ func GateLogin(dispatchInfo *DispatchInfo, accountInfo *AccountInfo, keyId strin AccountUid: strconv.Itoa(int(accountInfo.AccountId)), KeyId: uint32(keyIdInt), ClientRandKey: clientSeedBase64, + AccountType: 1, + ChannelId: 1, + SubChannelId: 1, + PlatformType: 3, }) protoMsg := <-session.RecvChan if protoMsg.CmdId != cmd.GetPlayerTokenRsp { @@ -92,5 +96,7 @@ func GateLogin(dispatchInfo *DispatchInfo, accountInfo *AccountInfo, keyId strin key := make([]byte, 4096) copy(key, xorKey[:]) session.XorKey = key + session.ClientVersionRandomKey = getPlayerTokenRsp.ClientVersionRandomKey + session.SecurityCmdBuffer = getPlayerTokenRsp.SecurityCmdBuffer return session, nil } diff --git a/robot/login/http_login.go b/robot/login/http_login.go index abf1a85d..9583a525 100644 --- a/robot/login/http_login.go +++ b/robot/login/http_login.go @@ -24,7 +24,7 @@ type DispatchInfo struct { DispatchKey []byte } -func GetDispatchInfo(regionListUrl string, curRegionUrl string, regionListParam string, curRegionParam string, keyId string) (*DispatchInfo, error) { +func GetDispatchInfo(regionListUrl string, regionListParam string, curRegionUrl string, curRegionParam string, keyId string) (*DispatchInfo, error) { logger.Info("http get url: %v", regionListUrl+"/query_region_list"+regionListParam) regionListBase64, err := httpclient.GetRaw(regionListUrl + "/query_region_list" + regionListParam) if err != nil { @@ -115,14 +115,14 @@ type AccountInfo struct { ComboToken string } -func AccountLogin(url string, account string, password string) (*AccountInfo, error) { +func AccountLogin(loginSdkUrl string, account string, password string) (*AccountInfo, error) { loginAccountRequestJson := &api.LoginAccountRequestJson{ Account: account, Password: password, IsCrypto: true, } - logger.Info("http post url: %v", url+"/hk4e_global/mdk/shield/api/login") - loginResult, err := httpclient.PostJson[api.LoginResult](url+"/hk4e_global/mdk/shield/api/login", loginAccountRequestJson) + logger.Info("http post url: %v", loginSdkUrl+"/hk4e_global/mdk/shield/api/login") + loginResult, err := httpclient.PostJson[api.LoginResult](loginSdkUrl+"/hk4e_global/mdk/shield/api/login", loginAccountRequestJson) if err != nil { return nil, err } @@ -147,8 +147,8 @@ func AccountLogin(url string, account string, password string) (*AccountInfo, er ChannelID: 1, Data: string(loginTokenDataJson), } - logger.Info("http post url: %v", url+"/hk4e_global/combo/granter/login/v2/login") - comboTokenRsp, err := httpclient.PostJson[api.ComboTokenRsp](url+"/hk4e_global/combo/granter/login/v2/login", comboTokenReq) + logger.Info("http post url: %v", loginSdkUrl+"/hk4e_global/combo/granter/login/v2/login") + comboTokenRsp, err := httpclient.PostJson[api.ComboTokenRsp](loginSdkUrl+"/hk4e_global/combo/granter/login/v2/login", comboTokenReq) if err != nil { return nil, err } diff --git a/robot/net/session.go b/robot/net/session.go index 4b8d4716..ec44cf09 100644 --- a/robot/net/session.go +++ b/robot/net/session.go @@ -4,6 +4,8 @@ import ( "sync/atomic" "time" + "hk4e/common/config" + "hk4e/gate/client_proto" "hk4e/gate/kcp" hk4egatenet "hk4e/gate/net" @@ -15,14 +17,16 @@ import ( ) type Session struct { - Conn *kcp.UDPSession - XorKey []byte - SendChan chan *hk4egatenet.ProtoMsg - RecvChan chan *hk4egatenet.ProtoMsg - ServerCmdProtoMap *cmd.CmdProtoMap - ClientCmdProtoMap *client_proto.ClientCmdProtoMap - ClientSeq uint32 - DeadEvent chan bool + Conn *kcp.UDPSession + XorKey []byte + SendChan chan *hk4egatenet.ProtoMsg + RecvChan chan *hk4egatenet.ProtoMsg + ServerCmdProtoMap *cmd.CmdProtoMap + ClientCmdProtoMap *client_proto.ClientCmdProtoMap + ClientSeq uint32 + DeadEvent chan bool + ClientVersionRandomKey string + SecurityCmdBuffer []byte } func NewSession(gateAddr string, dispatchKey []byte, localPort int) (*Session, error) { @@ -36,15 +40,21 @@ func NewSession(gateAddr string, dispatchKey []byte, localPort int) (*Session, e } conn.SetACKNoDelay(true) conn.SetWriteDelay(false) + conn.SetWindowSize(255, 255) r := &Session{ - Conn: conn, - XorKey: dispatchKey, - SendChan: make(chan *hk4egatenet.ProtoMsg, 1000), - RecvChan: make(chan *hk4egatenet.ProtoMsg, 1000), - ServerCmdProtoMap: cmd.NewCmdProtoMap(), - ClientCmdProtoMap: client_proto.NewClientCmdProtoMap(), - ClientSeq: 0, - DeadEvent: make(chan bool, 10), + Conn: conn, + XorKey: dispatchKey, + SendChan: make(chan *hk4egatenet.ProtoMsg, 1000), + RecvChan: make(chan *hk4egatenet.ProtoMsg, 1000), + ServerCmdProtoMap: cmd.NewCmdProtoMap(), + ClientCmdProtoMap: nil, + ClientSeq: 0, + DeadEvent: make(chan bool, 10), + ClientVersionRandomKey: "", + SecurityCmdBuffer: nil, + } + if config.GetConfig().Hk4e.ClientProtoProxyEnable { + r.ClientCmdProtoMap = client_proto.NewClientCmdProtoMap() } go r.recvHandle() go r.sendHandle()