MiraiGo/client/tlv_decoders.go

package client

import (
	"crypto/md5"
	"fmt"
	"time"

	"github.com/Mrs4s/MiraiGo/client/internal/auth"
	"github.com/Mrs4s/MiraiGo/utils"

	"github.com/Mrs4s/MiraiGo/binary"
)

// --- tlv decoders for qq client ---

func (c *QQClient) decodeT161(data []byte) {
	/*
		reader := binary.NewReader(data)
		reader.ReadBytes(2)
		t := reader.ReadTlvMap(2)
		if t172, ok := t[0x172]; ok {
			c.rollbackSig = t172
		}
	*/
}

func (c *QQClient) decodeT119(data, ek []byte) {
	tea := binary.NewTeaCipher(ek)
	reader := binary.NewReader(tea.Decrypt(data))
	reader.ReadBytes(2)
	m := reader.ReadTlvMap(2)
	if t130, ok := m[0x130]; ok {
		c.decodeT130(t130)
	}
	if t113, ok := m[0x113]; ok {
		c.decodeT113(t113)
	}
	/*
		if t528, ok := m[0x528]; ok {
			c.t528 = t528
		}
		if t530, ok := m[0x530]; ok {
			c.t530 = t530
		}
	*/
	if t108, ok := m[0x108]; ok {
		c.ksid = t108
	}

	var (
		// openId   []byte
		// openKey  []byte
		// payToken []byte
		// pf       []byte
		// pfkey    []byte
		gender uint16 = 0
		age    uint16 = 0
		nick          = ""
		// a1       []byte
		// noPicSig []byte
		// ctime           = time.Now().Unix()
		// etime           = ctime + 2160000
		psKeyMap    map[string][]byte
		pt4TokenMap map[string][]byte
	)

	if _, ok := m[0x125]; ok {
		// openId, openKey = readT125(t125)
	}
	if t186, ok := m[0x186]; ok {
		c.decodeT186(t186)
	}
	if t11a, ok := m[0x11a]; ok {
		nick, age, gender = readT11A(t11a)
	}
	if _, ok := m[0x199]; ok {
		// openId, payToken = readT199(t199)
	}
	if _, ok := m[0x200]; ok {
		// pf, pfkey = readT200(t200)
	}
	if t512, ok := m[0x512]; ok {
		psKeyMap, pt4TokenMap = readT512(t512)
	}
	if _, ok := m[0x531]; ok {
		// a1, noPicSig = readT531(t531)
	}

	if _, ok := m[0x138]; ok {
		// readT138(t138) // chg time
	}

	// we don't use `c.sigInfo = &auth.SigInfo{...}` here,
	// because we need spread `SigInfo` to other places
	*c.sigInfo = auth.SigInfo{
		LoginBitmap:        0,
		SrmToken:           utils.Select(m[0x16a], c.sigInfo.SrmToken),
		T133:               utils.Select(m[0x133], c.sigInfo.T133),
		EncryptedA1:        utils.Select(m[0x106], c.sigInfo.EncryptedA1),
		TGT:                m[0x10a],
		TGTKey:             m[0x10d],
		UserStKey:          m[0x10e],
		UserStWebSig:       m[0x103],
		SKey:               m[0x120],
		SKeyExpiredTime:    time.Now().Unix() + 21600,
		D2:                 m[0x143],
		D2Key:              m[0x305],
		WtSessionTicketKey: utils.Select(m[0x134], c.sigInfo.WtSessionTicketKey),
		DeviceToken:        m[0x322],

		PsKeyMap:    psKeyMap,
		Pt4TokenMap: pt4TokenMap,
	}
	if len(c.PasswordMd5[:]) > 0 {
		key := md5.Sum(append(append(c.PasswordMd5[:], []byte{0x00, 0x00, 0x00, 0x00}...), binary.NewWriterF(func(w *binary.Writer) { w.WriteUInt32(uint32(c.Uin)) })...))
		decrypted := binary.NewTeaCipher(key[:]).Decrypt(c.sigInfo.EncryptedA1)
		if len(decrypted) > 51+16 {
			dr := binary.NewReader(decrypted)
			dr.ReadBytes(51)
			c.deviceInfo.TgtgtKey = dr.ReadBytes(16)
		}
	}
	c.Nickname = nick
	c.Age = age
	c.Gender = gender
}

// wtlogin.exchange_emp
func (c *QQClient) decodeT119R(data []byte) {
	tea := binary.NewTeaCipher(c.deviceInfo.TgtgtKey)
	reader := binary.NewReader(tea.Decrypt(data))
	reader.ReadBytes(2)
	m := reader.ReadTlvMap(2)
	if t120, ok := m[0x120]; ok {
		c.sigInfo.SKey = t120
		c.sigInfo.SKeyExpiredTime = time.Now().Unix() + 21600
		c.Debug("skey updated: %v", c.sigInfo.SKey)
	}
	if t11a, ok := m[0x11a]; ok {
		c.Nickname, c.Age, c.Gender = readT11A(t11a)
		c.Debug("account info updated: " + c.Nickname)
	}
}

func (c *QQClient) decodeT130(data []byte) {
	reader := binary.NewReader(data)
	reader.ReadBytes(2)
	// c.timeDiff = int64(reader.ReadInt32()) - time.Now().Unix()
	// c.t149 = reader.ReadBytes(4)
}

func (c *QQClient) decodeT113(data []byte) {
	reader := binary.NewReader(data)
	uin := reader.ReadInt32() // ?
	fmt.Println("got t113 uin:", uin)
}

func (c *QQClient) decodeT186(data []byte) {
	// c.pwdFlag = data[1] == 1
}

// --- tlv readers ---

func readT125(data []byte) (openID, openKey []byte) {
	reader := binary.NewReader(data)
	openID = reader.ReadBytesShort()
	openKey = reader.ReadBytesShort()
	return
}

func readT11A(data []byte) (nick string, age, gender uint16) {
	reader := binary.NewReader(data)
	reader.ReadUInt16()
	age = uint16(reader.ReadByte())
	gender = uint16(reader.ReadByte())
	nick = reader.ReadStringLimit(int(reader.ReadByte()) & 0xff)
	return
}

func readT199(data []byte) (openID, payToken []byte) {
	reader := binary.NewReader(data)
	openID = reader.ReadBytesShort()
	payToken = reader.ReadBytesShort()
	return
}

func readT200(data []byte) (pf, pfKey []byte) {
	reader := binary.NewReader(data)
	pf = reader.ReadBytesShort()
	pfKey = reader.ReadBytesShort()
	return
}

func readT512(data []byte) (psKeyMap map[string][]byte, pt4TokenMap map[string][]byte) {
	reader := binary.NewReader(data)
	length := int(reader.ReadUInt16())

	psKeyMap = make(map[string][]byte, length)
	pt4TokenMap = make(map[string][]byte, length)

	for i := 0; i < length; i++ {
		domain := reader.ReadStringShort()
		psKey := reader.ReadBytesShort()
		pt4Token := reader.ReadBytesShort()

		if len(psKey) > 0 {
			psKeyMap[domain] = psKey
		}

		if len(pt4Token) > 0 {
			pt4TokenMap[domain] = pt4Token
		}
	}

	return
}

func readT531(data []byte) (a1, noPicSig []byte) {
	reader := binary.NewReader(data)
	m := reader.ReadTlvMap(2)
	if m.Exists(0x103) && m.Exists(0x16a) && m.Exists(0x113) && m.Exists(0x10c) {
		a1 = append(m[0x106], m[0x10c]...)
		noPicSig = m[0x16a]
	}
	return
}