bl/logic/service/player/pack.go

package player

import (
	"blazing/common/socket/errorcode"
	"blazing/cool"
	"blazing/logic/service/common"
	"encoding/binary"
	"encoding/hex"
	"sync"
	"sync/atomic"

	"context"

	"bytes"
	"fmt"
	"reflect"

	"github.com/bruceshao/lockfree"
	"github.com/gobwas/ws"
	"github.com/gobwas/ws/wsutil"
	"github.com/gogf/gf/v2/os/glog"
	"github.com/lunixbochs/struc"
	"github.com/panjf2000/gnet/v2"
)

const (
	minPacketLen = 17
	maxPacketLen = 10 * 1024
)

// getUnderlyingValue 递归解析reflect.Value，解包指针、interface{}到底层具体类型
func getUnderlyingValue(val reflect.Value) (reflect.Value, error) {
	for {
		switch val.Kind() {
		// 解包指针：获取指针指向的值
		case reflect.Ptr:
			if val.IsNil() {
				return reflect.Value{}, nil
			}
			val = val.Elem()

		// 解包interface{}：获取接口包裹的动态值
		case reflect.Interface:
			if val.IsNil() {
				return reflect.Value{}, nil
			}
			val = val.Elem()

		// 非指针/接口类型，终止递归
		default:
			return val, nil
		}
	}
}

func setFieldByIndex(root reflect.Value, index []int, value reflect.Value) bool {
	current := root
	for pos, idx := range index {
		if current.Kind() == reflect.Ptr {
			if current.IsNil() {
				current.Set(reflect.New(current.Type().Elem()))
			}
			current = current.Elem()
		}

		if current.Kind() != reflect.Struct || idx < 0 || idx >= current.NumField() {
			return false
		}

		field := current.Field(idx)
		if pos == len(index)-1 {
			if !field.CanSet() {
				return false
			}
			if value.Type().AssignableTo(field.Type()) {
				field.Set(value)
				return true
			}
			if field.Kind() == reflect.Ptr && value.Type().AssignableTo(field.Type().Elem()) {
				ptr := reflect.New(field.Type().Elem())
				ptr.Elem().Set(value)
				field.Set(ptr)
				return true
			}
			return false
		}

		current = field
	}

	return false
}

// XORDecryptU 原地执行异或解密，避免额外分配和拷贝。
func XORDecryptU(encryptedData []byte, key uint32) []byte {
	if len(encryptedData) == 0 {
		return []byte{}
	}

	var keyBytes [4]byte
	binary.BigEndian.PutUint32(keyBytes[:], key)
	keyLen := len(keyBytes)

	for i, b := range encryptedData {
		encryptedData[i] = b ^ keyBytes[i%keyLen]
	}

	return encryptedData
}

var packetDataPoolSizes = [...]int{64, 128, 256, 512, 1024, 2048, 4096}

var packetDataPools = func() []sync.Pool {
	pools := make([]sync.Pool, len(packetDataPoolSizes))
	for i, size := range packetDataPoolSizes {
		size := size
		pools[i].New = func() any {
			return make([]byte, size)
		}
	}
	return pools
}()

func getPacketData(size int) []byte {
	if size <= 0 {
		return nil
	}
	for i, poolSize := range packetDataPoolSizes {
		if size <= poolSize {
			return packetDataPools[i].Get().([]byte)[:size]
		}
	}
	return make([]byte, size)
}

func putPacketData(buf []byte) {
	if buf == nil {
		return
	}
	for i, poolSize := range packetDataPoolSizes {
		if cap(buf) == poolSize {
			packetDataPools[i].Put(buf[:poolSize])
			return
		}
	}
}

func (h *ClientData) PushEvent(v []byte, submit func(task func()) error) {
	if h == nil || h.IsClosed() {
		return
	}
	if len(v) < minPacketLen || len(v) > maxPacketLen {
		return
	}
	if binary.BigEndian.Uint32(v[0:4]) != uint32(len(v)) {
		return
	}

	var header common.TomeeHeader
	header.Len = binary.BigEndian.Uint32(v[0:4])
	header.CMD = binary.BigEndian.Uint32(v[5:9])
	header.UserID = binary.BigEndian.Uint32(v[9:13])
	if dataLen := len(v) - 17; dataLen > 0 {
		header.Data = getPacketData(dataLen)
		copy(header.Data, v[17:])
	}

	err := submit(func() {
		if h.IsClosed() || h.LF == nil || !h.LF.Running() {
			putPacketData(header.Data)
			return
		}
		if err := h.LF.Producer().Write(header); err != nil {
			putPacketData(header.Data)
		}
	})
	if err != nil {
		putPacketData(header.Data)
	}
}

// 重写
// 遍历结构体方法并执行RECV_cmd
func (h *ClientData) OnEvent(data common.TomeeHeader) {
	defer func() {
		if err := recover(); err != nil { // 恢复 panic，err 为 panic 错误值
			// 1. 打印错误信息
			if h.Player != nil {
				if h.Player.Info != nil {
					cool.Logger.Error(context.TODO(), "panic 错误：", cool.Config.ServerInfo.OnlineID, h.Player.Info.UserID, err)
				} else {
					cool.Logger.Error(context.TODO(), "panic 错误：", cool.Config.ServerInfo.OnlineID, err)

				}

			} else {
				cool.Logger.Error(context.TODO(), "panic 错误：", err)
			}

		}
	}()
	if data.CMD > 1001 {

		if h.Player == nil {
			fmt.Println(data.UserID, "账号未注册")
			return
		}
		if h.Player.Info == nil {
			fmt.Println(data.UserID, "未创建角色")
			return
		}
		if data.Data != nil {
			defer putPacketData(data.Data)
			data.Res = XORDecryptU(data.Data, h.Player.Hash)
		}

	} else if data.Data != nil {
		defer putPacketData(data.Data)
		data.Res = data.Data
	}
	if cool.Config.ServerInfo.IsDebug != 0 {
		fmt.Println("接收数据", data.UserID, data.CMD)
	}
	if data.UserID == 0 {
		return
	}
	cmdlister, ok := cool.CmdCache[data.CMD]
	if !ok {

		//	glog.Debug(context.Background(), data.UserID, data.CMD, "cmd未注册")
		return //TODO 待实现cmd未注册
	}

	var ptrValue reflect.Value
	if cmdlister.NewReqValue != nil {
		ptrValue = cmdlister.NewReqValue()
	} else {
		ptrValue = reflect.New(cmdlister.Req)
	}

	if data.Res != nil {
		err := struc.Unpack(bytes.NewBuffer(data.Res), ptrValue.Interface())

		if err != nil {
			cool.Logger.Error(context.Background(), data.UserID, data.CMD, "解包失败,", err, hex.EncodeToString(data.Res))
			data.Result = uint32(errorcode.ErrorCodes.ErrSystemProcessingError)
			h.SendPack(data.Pack(nil))
			return
		}
	}

	if !setFieldByIndex(ptrValue.Elem(), cmdlister.HeaderFieldIndex, reflect.ValueOf(data)) {
		cool.Logger.Warning(context.Background(), data.UserID, data.CMD, "设置请求头失败")
		return
	}

	var params [2]reflect.Value
	params[0] = ptrValue
	if cmdlister.UseConn {
		params[1] = reflect.ValueOf(h.Conn)
	} else {
		params[1] = reflect.ValueOf(h.Player)
	}

	ret := cmdlister.Func.Call(params[:])

	if len(ret) <= 0 { //如果判断没有参数,那就说明这个包没有返回参数
		return
	}

	aa, ok := ret[1].Interface().(errorcode.ErrorCode) //判断错误
	data.Result = uint32(aa)
	if aa == -1 {
		return

	}

	if ok && aa != 0 { //这里实现回复错误包

		glog.Info(context.Background(), data.UserID, data.CMD, aa.Code())
		h.SendPack(data.Pack(nil))

		return

	}

	t1 := data.Pack(ret[0].Interface())

	h.SendPack(t1)

}

type ClientData struct {
	Player             *Player //客户实体
	ERROR_CONNUT       int
	Wsmsg              *WsCodec
	Conn               gnet.Conn
	LF                 *lockfree.Lockfree[common.TomeeHeader]
	closed             int32
	crossDomainChecked uint32
}

func (p *ClientData) IsClosed() bool {
	return atomic.LoadInt32(&p.closed) == 1
}

func (p *ClientData) Close() {
	if !atomic.CompareAndSwapInt32(&p.closed, 0, 1) {
		return
	}
	if p.LF != nil && p.LF.Running() {
		_ = p.LF.Close()
	}
}

func (p *ClientData) IsCrossDomainChecked() bool {
	return atomic.LoadUint32(&p.crossDomainChecked) == 1
}

func (p *ClientData) MarkCrossDomainChecked() {
	atomic.StoreUint32(&p.crossDomainChecked, 1)
}

func (p *ClientData) GetPlayer(userid uint32) *Player { //TODO 这里待优化,可能存在内存泄漏问题
	if p.Player == nil {
		p.Player = NewPlayer(p.Conn)
	}
	_, ok := Mainplayer.LoadOrStore(userid, p)
	if !ok {
		p.Player = NewPlayer(p.Conn)
	}

	return p.Player
	//	return nil
}
func NewClientData(c gnet.Conn) *ClientData {
	cd := &ClientData{
		Conn:  c,
		Wsmsg: &WsCodec{},
	}
	cd.LF = lockfree.NewLockfree(
		8,
		cd,
		lockfree.NewConditionBlockStrategy(),
	)
	if err := cd.LF.Start(); err != nil {
		panic(err)
	}
	return cd

}

// XORDecrypt 异或解密函数（密钥改为uint32版本）
// 核心逻辑：将uint32密钥拆分为4字节数组（大端序，适配AS3二进制处理习惯），循环与加密数据异或
// 参数：
//
//	encryptedData - 待解密的字节数组
//	key - 32位无符号整数密钥（替代原字符串密钥）
//
// 返回值：解密后的字节数组

func XORDecrypt(encryptedData []byte, keyStr string) []byte {
	if len(encryptedData) == 0 || keyStr == "" {
		return []byte{}
	}

	// 1. 将密钥字符串转换为UTF-8字节数组（对应AS3的writeUTFBytes(_arg_2)）
	keyBytes := []byte(keyStr) // Go中string转[]byte默认是UTF-8编码，与AS3的writeUTFBytes一致
	keyLen := len(keyBytes)
	if keyLen == 0 {
		return encryptedData // 空密钥不加密，直接返回
	}

	// 2. 执行异或操作（与加密逻辑一致，异或两次还原数据）
	decrypted := make([]byte, len(encryptedData))
	for i, b := range encryptedData {
		// 循环复用密钥字节（索引取模）
		keyIndex := i % keyLen
		decrypted[i] = b ^ keyBytes[keyIndex]
	}

	return decrypted
}

func (p *ClientData) SendPack(b []byte) error {
	if p.Wsmsg == nil {
		return fmt.Errorf("ws空")
	}

	if p.Wsmsg.Upgraded {
		// This is the echo server
		wsutil.WriteServerMessage(p.Conn, ws.OpBinary, b)

	} else {

		p.Conn.AsyncWrite(b, nil)

	}

	return nil
}