mirror of
https://github.com/Mrs4s/MiraiGo.git
synced 2025-05-04 11:07:40 +08:00
132 lines
2.4 KiB
Go
132 lines
2.4 KiB
Go
package binary
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"encoding/hex"
|
|
|
|
"github.com/Mrs4s/MiraiGo/utils"
|
|
)
|
|
|
|
// Writer 写入
|
|
type Writer bytes.Buffer
|
|
|
|
func NewWriterF(f func(writer *Writer)) []byte {
|
|
w := SelectWriter()
|
|
f(w)
|
|
b := append([]byte(nil), w.Bytes()...)
|
|
PutWriter(w)
|
|
return b
|
|
}
|
|
|
|
func (w *Writer) Write(b []byte) {
|
|
(*bytes.Buffer)(w).Write(b)
|
|
}
|
|
|
|
func (w *Writer) WriteHex(h string) {
|
|
b, _ := hex.DecodeString(h)
|
|
w.Write(b)
|
|
}
|
|
|
|
func (w *Writer) WriteByte(b byte) {
|
|
(*bytes.Buffer)(w).WriteByte(b)
|
|
}
|
|
|
|
func (w *Writer) WriteUInt16(v uint16) {
|
|
b := make([]byte, 2)
|
|
binary.BigEndian.PutUint16(b, v)
|
|
w.Write(b)
|
|
}
|
|
|
|
func (w *Writer) WriteUInt32(v uint32) {
|
|
b := make([]byte, 4)
|
|
binary.BigEndian.PutUint32(b, v)
|
|
w.Write(b)
|
|
}
|
|
|
|
func (w *Writer) WriteUInt64(v uint64) {
|
|
b := make([]byte, 8)
|
|
binary.BigEndian.PutUint64(b, v)
|
|
w.Write(b)
|
|
}
|
|
|
|
func (w *Writer) WriteString(v string) {
|
|
payload := utils.S2B(v)
|
|
w.WriteUInt32(uint32(len(payload) + 4))
|
|
w.Write(payload)
|
|
}
|
|
|
|
func (w *Writer) WriteStringShort(v string) {
|
|
w.WriteBytesShort(utils.S2B(v))
|
|
}
|
|
|
|
func (w *Writer) WriteBool(b bool) {
|
|
if b {
|
|
w.WriteByte(0x01)
|
|
} else {
|
|
w.WriteByte(0x00)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) EncryptAndWrite(key []byte, data []byte) {
|
|
tea := NewTeaCipher(key)
|
|
ed := tea.Encrypt(data)
|
|
w.Write(ed)
|
|
}
|
|
|
|
func (w *Writer) WriteIntLvPacket(offset int, f func(writer *Writer)) {
|
|
data := NewWriterF(f)
|
|
w.WriteUInt32(uint32(len(data) + offset))
|
|
w.Write(data)
|
|
}
|
|
|
|
func (w *Writer) WriteUniPacket(commandName string, sessionId, extraData, body []byte) {
|
|
w1 := SelectWriter()
|
|
{ // WriteIntLvPacket
|
|
w1.WriteString(commandName)
|
|
w1.WriteUInt32(8)
|
|
w1.Write(sessionId)
|
|
if len(extraData) == 0 {
|
|
w1.WriteUInt32(0x04)
|
|
} else {
|
|
w1.WriteUInt32(uint32(len(extraData) + 4))
|
|
w1.Write(extraData)
|
|
}
|
|
}
|
|
data := w1.Bytes()
|
|
w.WriteUInt32(uint32(len(data) + 4))
|
|
w.Write(data)
|
|
PutWriter(w1)
|
|
w.WriteUInt32(uint32(len(body) + 4)) // WriteIntLvPacket
|
|
w.Write(body)
|
|
}
|
|
|
|
func (w *Writer) WriteBytesShort(data []byte) {
|
|
w.WriteUInt16(uint16(len(data)))
|
|
w.Write(data)
|
|
}
|
|
|
|
func (w *Writer) WriteTlvLimitedSize(data []byte, limit int) {
|
|
if len(data) <= limit {
|
|
w.WriteBytesShort(data)
|
|
return
|
|
}
|
|
w.WriteBytesShort(data[:limit])
|
|
}
|
|
|
|
func (w *Writer) Bytes() []byte {
|
|
return (*bytes.Buffer)(w).Bytes()
|
|
}
|
|
|
|
func (w *Writer) Cap() int {
|
|
return (*bytes.Buffer)(w).Cap()
|
|
}
|
|
|
|
func (w *Writer) Reset() {
|
|
(*bytes.Buffer)(w).Reset()
|
|
}
|
|
|
|
func (w *Writer) Grow(n int) {
|
|
(*bytes.Buffer)(w).Grow(n)
|
|
}
|