完善机器人

2026-02-04 14:22:26 +08:00 · 2023-05-13 20:40:40 +08:00
parent 9f10142c47
commit 0bc54baf12
9 changed files with 163 additions and 62 deletions
--- 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 # 压力测试是否循环登录退出
--- 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
 		}
 	}
--- 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
 }
--- 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
 	}
--- 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()