Refactor all state into the Session struct
This commit is contained in:
parent
c568a471d3
commit
9471a65ca3
327
protocol7/connection.go
Normal file
327
protocol7/connection.go
Normal file
|
|
@ -0,0 +1,327 @@
|
|||
package protocol7
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"slices"
|
||||
|
||||
"github.com/teeworlds-go/huffman"
|
||||
"github.com/teeworlds-go/teeworlds/chunk"
|
||||
message "github.com/teeworlds-go/teeworlds/messages"
|
||||
"github.com/teeworlds-go/teeworlds/packer"
|
||||
"github.com/teeworlds-go/teeworlds/packet"
|
||||
)
|
||||
|
||||
const (
|
||||
MaxClients = 64
|
||||
|
||||
msgCtrlKeepAlive = 0x00
|
||||
msgCtrlConnect = 0x01
|
||||
msgCtrlAccept = 0x02
|
||||
msgCtrlToken = 0x05
|
||||
msgCtrlClose = 0x04
|
||||
|
||||
msgSysMapChange = 2
|
||||
msgSysConReady = 5
|
||||
msgSysSnapSingle = 8
|
||||
|
||||
msgGameSvMotd = 1
|
||||
msgGameSvChat = 3
|
||||
msgGameReadyToEnter = 8
|
||||
msgGameSvClientInfo = 18
|
||||
msgGameClStartInfo = 27
|
||||
)
|
||||
|
||||
type Player struct {
|
||||
Info message.SvClientInfo
|
||||
}
|
||||
|
||||
type Connection struct {
|
||||
ClientToken [4]byte
|
||||
ServerToken [4]byte
|
||||
Conn net.Conn
|
||||
|
||||
// The amount of vital chunks received
|
||||
Ack int
|
||||
|
||||
// The amount of vital chunks sent
|
||||
Sequence int
|
||||
|
||||
// The amount of vital chunks acknowledged by the peer
|
||||
PeerAck int
|
||||
|
||||
Players []Player
|
||||
}
|
||||
|
||||
func (c *Connection) SendCtrlToken(myToken []byte) {
|
||||
header := []byte{0x04, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff}
|
||||
ctrlToken := append([]byte{0x05}, myToken...)
|
||||
zeros := []byte{512: 0}
|
||||
data := slices.Concat(header, ctrlToken, zeros)
|
||||
c.Conn.Write(data)
|
||||
}
|
||||
|
||||
func (c *Connection) SendCtrlMsg(data []byte) {
|
||||
header := packet.PacketHeader{
|
||||
Flags: packet.PacketFlags{
|
||||
Connless: false,
|
||||
Compression: false,
|
||||
Resend: false,
|
||||
Control: true,
|
||||
},
|
||||
Ack: c.Ack,
|
||||
NumChunks: 0,
|
||||
Token: c.ServerToken,
|
||||
}
|
||||
|
||||
packet := slices.Concat(header.Pack(), data)
|
||||
c.Conn.Write(packet)
|
||||
}
|
||||
|
||||
func (c *Connection) SendKeepAlive() {
|
||||
c.SendCtrlMsg([]byte{msgCtrlKeepAlive})
|
||||
}
|
||||
|
||||
func (c *Connection) SendReady() {
|
||||
ready := []byte{0x40, 0x01, 0x02, 0x25}
|
||||
|
||||
c.Sequence++
|
||||
c.SendPacket(ready, 1)
|
||||
}
|
||||
|
||||
type ChunkArgs struct {
|
||||
MsgId int
|
||||
System bool
|
||||
Flags chunk.ChunkFlags
|
||||
Payload []byte
|
||||
}
|
||||
|
||||
func (client *Connection) PackChunk(c ChunkArgs) []byte {
|
||||
c.MsgId <<= 1
|
||||
if c.System {
|
||||
c.MsgId |= 1
|
||||
}
|
||||
|
||||
client.Sequence++
|
||||
msgAndSys := packer.PackInt(c.MsgId)
|
||||
|
||||
chunkHeader := chunk.ChunkHeader{
|
||||
Flags: c.Flags,
|
||||
Size: len(msgAndSys) + len(c.Payload),
|
||||
Seq: client.Sequence,
|
||||
}
|
||||
|
||||
data := slices.Concat(
|
||||
chunkHeader.Pack(),
|
||||
msgAndSys,
|
||||
c.Payload,
|
||||
)
|
||||
|
||||
fmt.Printf("packed chunk: %x\n", data)
|
||||
|
||||
return data
|
||||
}
|
||||
|
||||
func (client *Connection) SendPacket(payload []byte, numChunks int) {
|
||||
header := packet.PacketHeader{
|
||||
Flags: packet.PacketFlags{
|
||||
Connless: false,
|
||||
Compression: false,
|
||||
Resend: false,
|
||||
Control: false,
|
||||
},
|
||||
Ack: client.Ack,
|
||||
NumChunks: numChunks,
|
||||
Token: client.ServerToken,
|
||||
}
|
||||
|
||||
packet := slices.Concat(header.Pack(), payload)
|
||||
client.Conn.Write(packet)
|
||||
}
|
||||
|
||||
func (client *Connection) SendInfo() {
|
||||
info := []byte{0x40, 0x28, 0x01, 0x03, 0x30, 0x2E, 0x37, 0x20, 0x38, 0x30, 0x32, 0x66,
|
||||
0x31, 0x62, 0x65, 0x36, 0x30, 0x61, 0x30, 0x35, 0x36, 0x36, 0x35, 0x66,
|
||||
0x00, 0x6D, 0x79, 0x5F, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6F, 0x72, 0x64,
|
||||
0x5F, 0x31, 0x32, 0x33, 0x00, 0x85, 0x1C, 0x00}
|
||||
|
||||
client.Sequence++
|
||||
client.SendPacket(info, 1)
|
||||
}
|
||||
|
||||
func (client *Connection) SendStartInfo() {
|
||||
info := message.ClStartInfo{
|
||||
Name: "gopher",
|
||||
Clan: "",
|
||||
Country: 0,
|
||||
Body: "greensward",
|
||||
Marking: "duodonny",
|
||||
Decoration: "",
|
||||
Hands: "standard",
|
||||
Feet: "standard",
|
||||
Eyes: "standard",
|
||||
CustomColorBody: false,
|
||||
CustomColorMarking: false,
|
||||
CustomColorDecoration: false,
|
||||
CustomColorHands: false,
|
||||
CustomColorFeet: false,
|
||||
CustomColorEyes: false,
|
||||
ColorBody: 0,
|
||||
ColorMarking: 0,
|
||||
ColorDecoration: 0,
|
||||
ColorHands: 0,
|
||||
ColorFeet: 0,
|
||||
ColorEyes: 0,
|
||||
}
|
||||
|
||||
payload := client.PackChunk(ChunkArgs{
|
||||
MsgId: msgGameClStartInfo,
|
||||
Flags: chunk.ChunkFlags{
|
||||
Vital: true,
|
||||
},
|
||||
Payload: info.Pack(),
|
||||
})
|
||||
|
||||
client.SendPacket(payload, 1)
|
||||
}
|
||||
|
||||
func (client *Connection) SendEnterGame() {
|
||||
enter := []byte{
|
||||
0x40, 0x01, 0x04, 0x27,
|
||||
}
|
||||
|
||||
client.SendPacket(enter, 1)
|
||||
}
|
||||
|
||||
func byteSliceToString(s []byte) string {
|
||||
n := bytes.IndexByte(s, 0)
|
||||
if n >= 0 {
|
||||
s = s[:n]
|
||||
}
|
||||
return string(s)
|
||||
}
|
||||
|
||||
func (client *Connection) OnSystemMsg(msg int, chunk chunk.Chunk, u *packer.Unpacker) {
|
||||
if msg == msgSysMapChange {
|
||||
fmt.Println("got map change")
|
||||
client.SendReady()
|
||||
} else if msg == msgSysConReady {
|
||||
fmt.Println("got ready")
|
||||
client.SendStartInfo()
|
||||
} else if msg == msgSysSnapSingle {
|
||||
// tick := u.GetInt()
|
||||
// fmt.Printf("got snap single tick=%d\n", tick)
|
||||
client.SendKeepAlive()
|
||||
} else {
|
||||
fmt.Printf("unknown system message id=%d data=%x\n", msg, chunk.Data)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *Connection) OnChatMessage(mode int, clientId int, targetId int, message string) {
|
||||
name := client.Players[clientId].Info.Name
|
||||
fmt.Printf("[chat] <%s> %s\n", name, message)
|
||||
}
|
||||
|
||||
func (client *Connection) OnMotd(motd string) {
|
||||
fmt.Printf("[motd] %s\n", motd)
|
||||
}
|
||||
|
||||
func (client *Connection) OnGameMsg(msg int, chunk chunk.Chunk, u *packer.Unpacker) {
|
||||
if msg == msgGameReadyToEnter {
|
||||
fmt.Println("got ready to enter")
|
||||
client.SendEnterGame()
|
||||
} else if msg == msgGameSvMotd {
|
||||
motd := u.GetString()
|
||||
if motd != "" {
|
||||
client.OnMotd(motd)
|
||||
}
|
||||
} else if msg == msgGameSvChat {
|
||||
mode := u.GetInt()
|
||||
clientId := u.GetInt()
|
||||
targetId := u.GetInt()
|
||||
message := u.GetString()
|
||||
client.OnChatMessage(mode, clientId, targetId, message)
|
||||
} else if msg == msgGameSvClientInfo {
|
||||
clientId := packer.UnpackInt(chunk.Data[1:])
|
||||
client.Players[clientId].Info.Unpack(u)
|
||||
|
||||
fmt.Printf("got client info id=%d name=%s\n", clientId, client.Players[clientId].Info.Name)
|
||||
} else {
|
||||
fmt.Printf("unknown game message id=%d data=%x\n", msg, chunk.Data)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *Connection) OnMessage(chunk chunk.Chunk) {
|
||||
// fmt.Printf("got chunk size=%d data=%v\n", chunk.Header.Size, chunk.Data)
|
||||
|
||||
if chunk.Header.Flags.Vital {
|
||||
client.Ack++
|
||||
}
|
||||
|
||||
u := packer.Unpacker{}
|
||||
u.Reset(chunk.Data)
|
||||
|
||||
msg := u.GetInt()
|
||||
|
||||
sys := msg&1 != 0
|
||||
msg >>= 1
|
||||
|
||||
if sys {
|
||||
client.OnSystemMsg(msg, chunk, &u)
|
||||
} else {
|
||||
client.OnGameMsg(msg, chunk, &u)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *Connection) OnPacketPayload(header []byte, data []byte) {
|
||||
chunks := chunk.UnpackChunks(data)
|
||||
|
||||
for _, c := range chunks {
|
||||
client.OnMessage(c)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *Connection) OnPacket(data []byte) {
|
||||
header := packet.PacketHeader{}
|
||||
headerRaw := data[:7]
|
||||
payload := data[7:]
|
||||
header.Unpack(headerRaw)
|
||||
|
||||
if header.Flags.Control {
|
||||
ctrlMsg := payload[0]
|
||||
fmt.Printf("got ctrl msg %d\n", ctrlMsg)
|
||||
if ctrlMsg == msgCtrlToken {
|
||||
copy(client.ServerToken[:], payload[1:5])
|
||||
fmt.Printf("got server token %x\n", client.ServerToken)
|
||||
client.SendCtrlMsg(slices.Concat([]byte{msgCtrlConnect}, client.ClientToken[:]))
|
||||
} else if ctrlMsg == msgCtrlAccept {
|
||||
fmt.Println("got accept")
|
||||
client.SendInfo()
|
||||
} else if ctrlMsg == msgCtrlClose {
|
||||
// TODO: get length from packet header to determine if a reason is set or not
|
||||
// len(data) -> is 1400 (maxPacketLen)
|
||||
|
||||
reason := byteSliceToString(payload)
|
||||
fmt.Printf("disconnected (%s)\n", reason)
|
||||
|
||||
os.Exit(0)
|
||||
} else {
|
||||
fmt.Printf("unknown control message: %x\n", data)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
if header.Flags.Compression {
|
||||
huff := huffman.Huffman{}
|
||||
var err error
|
||||
payload, err = huff.Decompress(payload)
|
||||
if err != nil {
|
||||
fmt.Printf("huffman error: %v\n", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
client.OnPacketPayload(headerRaw, payload)
|
||||
}
|
||||
332
teeworlds.go
332
teeworlds.go
|
|
@ -2,50 +2,20 @@ package main
|
|||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"slices"
|
||||
"time"
|
||||
|
||||
"github.com/teeworlds-go/huffman"
|
||||
"github.com/teeworlds-go/teeworlds/chunk"
|
||||
message "github.com/teeworlds-go/teeworlds/messages"
|
||||
"github.com/teeworlds-go/teeworlds/packer"
|
||||
"github.com/teeworlds-go/teeworlds/packet"
|
||||
"github.com/teeworlds-go/teeworlds/protocol7"
|
||||
)
|
||||
|
||||
const (
|
||||
maxPacksize = 1400
|
||||
|
||||
MaxClients = 64
|
||||
|
||||
msgCtrlKeepAlive = 0x00
|
||||
msgCtrlConnect = 0x01
|
||||
msgCtrlAccept = 0x02
|
||||
msgCtrlToken = 0x05
|
||||
msgCtrlClose = 0x04
|
||||
|
||||
msgSysMapChange = 2
|
||||
msgSysConReady = 5
|
||||
msgSysSnapSingle = 8
|
||||
|
||||
msgGameSvMotd = 1
|
||||
msgGameSvChat = 3
|
||||
msgGameReadyToEnter = 8
|
||||
msgGameSvClientInfo = 18
|
||||
msgGameClStartInfo = 27
|
||||
)
|
||||
|
||||
func ctrlToken(myToken []byte) []byte {
|
||||
header := []byte{0x04, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff}
|
||||
ctrlToken := append([]byte{0x05}, myToken...)
|
||||
zeros := []byte{512: 0}
|
||||
data := slices.Concat(header, ctrlToken, zeros)
|
||||
return data
|
||||
}
|
||||
|
||||
func getConnection() (net.Conn, error) {
|
||||
conn, err := net.Dial("udp", "127.0.0.1:8303")
|
||||
if err != nil {
|
||||
|
|
@ -72,290 +42,6 @@ func readNetwork(ch chan<- []byte, conn net.Conn) {
|
|||
conn.Close()
|
||||
}
|
||||
|
||||
type Player struct {
|
||||
info message.SvClientInfo
|
||||
}
|
||||
|
||||
type TeeworldsClient struct {
|
||||
clientToken [4]byte
|
||||
serverToken [4]byte
|
||||
conn net.Conn
|
||||
|
||||
// The amount of vital chunks received
|
||||
Ack int
|
||||
|
||||
// The amount of vital chunks sent
|
||||
Sequence int
|
||||
|
||||
// The amount of vital chunks acknowledged by the peer
|
||||
PeerAck int
|
||||
|
||||
Players []Player
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendCtrlMsg(data []byte) {
|
||||
header := packet.PacketHeader{
|
||||
Flags: packet.PacketFlags{
|
||||
Connless: false,
|
||||
Compression: false,
|
||||
Resend: false,
|
||||
Control: true,
|
||||
},
|
||||
Ack: client.Ack,
|
||||
NumChunks: 0,
|
||||
Token: client.serverToken,
|
||||
}
|
||||
|
||||
packet := slices.Concat(header.Pack(), data)
|
||||
client.conn.Write(packet)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendKeepAlive() {
|
||||
client.sendCtrlMsg([]byte{msgCtrlKeepAlive})
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendReady() {
|
||||
ready := []byte{0x40, 0x01, 0x02, 0x25}
|
||||
|
||||
client.Sequence++
|
||||
client.sendPacket(ready, 1)
|
||||
}
|
||||
|
||||
type ChunkArgs struct {
|
||||
MsgId int
|
||||
System bool
|
||||
Flags chunk.ChunkFlags
|
||||
Payload []byte
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) packChunk(c ChunkArgs) []byte {
|
||||
c.MsgId <<= 1
|
||||
if c.System {
|
||||
c.MsgId |= 1
|
||||
}
|
||||
|
||||
client.Sequence++
|
||||
msgAndSys := packer.PackInt(c.MsgId)
|
||||
|
||||
chunkHeader := chunk.ChunkHeader{
|
||||
Flags: c.Flags,
|
||||
Size: len(msgAndSys) + len(c.Payload),
|
||||
Seq: client.Sequence,
|
||||
}
|
||||
|
||||
data := slices.Concat(
|
||||
chunkHeader.Pack(),
|
||||
msgAndSys,
|
||||
c.Payload,
|
||||
)
|
||||
|
||||
fmt.Printf("packed chunk: %x\n", data)
|
||||
|
||||
return data
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendPacket(payload []byte, numChunks int) {
|
||||
header := packet.PacketHeader{
|
||||
Flags: packet.PacketFlags{
|
||||
Connless: false,
|
||||
Compression: false,
|
||||
Resend: false,
|
||||
Control: false,
|
||||
},
|
||||
Ack: client.Ack,
|
||||
NumChunks: numChunks,
|
||||
Token: client.serverToken,
|
||||
}
|
||||
|
||||
packet := slices.Concat(header.Pack(), payload)
|
||||
client.conn.Write(packet)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendInfo() {
|
||||
info := []byte{0x40, 0x28, 0x01, 0x03, 0x30, 0x2E, 0x37, 0x20, 0x38, 0x30, 0x32, 0x66,
|
||||
0x31, 0x62, 0x65, 0x36, 0x30, 0x61, 0x30, 0x35, 0x36, 0x36, 0x35, 0x66,
|
||||
0x00, 0x6D, 0x79, 0x5F, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6F, 0x72, 0x64,
|
||||
0x5F, 0x31, 0x32, 0x33, 0x00, 0x85, 0x1C, 0x00}
|
||||
|
||||
client.Sequence++
|
||||
client.sendPacket(info, 1)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendStartInfo() {
|
||||
info := message.ClStartInfo{
|
||||
Name: "gopher",
|
||||
Clan: "",
|
||||
Country: 0,
|
||||
Body: "greensward",
|
||||
Marking: "duodonny",
|
||||
Decoration: "",
|
||||
Hands: "standard",
|
||||
Feet: "standard",
|
||||
Eyes: "standard",
|
||||
CustomColorBody: false,
|
||||
CustomColorMarking: false,
|
||||
CustomColorDecoration: false,
|
||||
CustomColorHands: false,
|
||||
CustomColorFeet: false,
|
||||
CustomColorEyes: false,
|
||||
ColorBody: 0,
|
||||
ColorMarking: 0,
|
||||
ColorDecoration: 0,
|
||||
ColorHands: 0,
|
||||
ColorFeet: 0,
|
||||
ColorEyes: 0,
|
||||
}
|
||||
|
||||
payload := client.packChunk(ChunkArgs{
|
||||
MsgId: msgGameClStartInfo,
|
||||
Flags: chunk.ChunkFlags{
|
||||
Vital: true,
|
||||
},
|
||||
Payload: info.Pack(),
|
||||
})
|
||||
|
||||
client.sendPacket(payload, 1)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) sendEnterGame() {
|
||||
enter := []byte{
|
||||
0x40, 0x01, 0x04, 0x27,
|
||||
}
|
||||
|
||||
client.sendPacket(enter, 1)
|
||||
}
|
||||
|
||||
func byteSliceToString(s []byte) string {
|
||||
n := bytes.IndexByte(s, 0)
|
||||
if n >= 0 {
|
||||
s = s[:n]
|
||||
}
|
||||
return string(s)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) onSystemMsg(msg int, chunk chunk.Chunk, u *packer.Unpacker) {
|
||||
if msg == msgSysMapChange {
|
||||
fmt.Println("got map change")
|
||||
client.sendReady()
|
||||
} else if msg == msgSysConReady {
|
||||
fmt.Println("got ready")
|
||||
client.sendStartInfo()
|
||||
} else if msg == msgSysSnapSingle {
|
||||
// tick := u.GetInt()
|
||||
// fmt.Printf("got snap single tick=%d\n", tick)
|
||||
client.sendKeepAlive()
|
||||
} else {
|
||||
fmt.Printf("unknown system message id=%d data=%x\n", msg, chunk.Data)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) OnChatMessage(mode int, clientId int, targetId int, message string) {
|
||||
name := client.Players[clientId].info.Name
|
||||
fmt.Printf("[chat] <%s> %s\n", name, message)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) OnMotd(motd string) {
|
||||
fmt.Printf("[motd] %s\n", motd)
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) onGameMsg(msg int, chunk chunk.Chunk, u *packer.Unpacker) {
|
||||
if msg == msgGameReadyToEnter {
|
||||
fmt.Println("got ready to enter")
|
||||
client.sendEnterGame()
|
||||
} else if msg == msgGameSvMotd {
|
||||
motd := u.GetString()
|
||||
if motd != "" {
|
||||
client.OnMotd(motd)
|
||||
}
|
||||
} else if msg == msgGameSvChat {
|
||||
mode := u.GetInt()
|
||||
clientId := u.GetInt()
|
||||
targetId := u.GetInt()
|
||||
message := u.GetString()
|
||||
client.OnChatMessage(mode, clientId, targetId, message)
|
||||
} else if msg == msgGameSvClientInfo {
|
||||
clientId := packer.UnpackInt(chunk.Data[1:])
|
||||
client.Players[clientId].info.Unpack(u)
|
||||
|
||||
fmt.Printf("got client info id=%d name=%s\n", clientId, client.Players[clientId].info.Name)
|
||||
} else {
|
||||
fmt.Printf("unknown game message id=%d data=%x\n", msg, chunk.Data)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) onMessage(chunk chunk.Chunk) {
|
||||
// fmt.Printf("got chunk size=%d data=%v\n", chunk.Header.Size, chunk.Data)
|
||||
|
||||
if chunk.Header.Flags.Vital {
|
||||
client.Ack++
|
||||
}
|
||||
|
||||
u := packer.Unpacker{}
|
||||
u.Reset(chunk.Data)
|
||||
|
||||
msg := u.GetInt()
|
||||
|
||||
sys := msg&1 != 0
|
||||
msg >>= 1
|
||||
|
||||
if sys {
|
||||
client.onSystemMsg(msg, chunk, &u)
|
||||
} else {
|
||||
client.onGameMsg(msg, chunk, &u)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) onPacketPayload(header []byte, data []byte) {
|
||||
chunks := chunk.UnpackChunks(data)
|
||||
|
||||
for _, c := range chunks {
|
||||
client.onMessage(c)
|
||||
}
|
||||
}
|
||||
|
||||
func (client *TeeworldsClient) onPacket(data []byte) {
|
||||
header := packet.PacketHeader{}
|
||||
headerRaw := data[:7]
|
||||
payload := data[7:]
|
||||
header.Unpack(headerRaw)
|
||||
|
||||
if header.Flags.Control {
|
||||
ctrlMsg := payload[0]
|
||||
fmt.Printf("got ctrl msg %d\n", ctrlMsg)
|
||||
if ctrlMsg == msgCtrlToken {
|
||||
copy(client.serverToken[:], payload[1:5])
|
||||
fmt.Printf("got server token %x\n", client.serverToken)
|
||||
client.sendCtrlMsg(slices.Concat([]byte{msgCtrlConnect}, client.clientToken[:]))
|
||||
} else if ctrlMsg == msgCtrlAccept {
|
||||
fmt.Println("got accept")
|
||||
client.sendInfo()
|
||||
} else if ctrlMsg == msgCtrlClose {
|
||||
// TODO: get length from packet header to determine if a reason is set or not
|
||||
// len(data) -> is 1400 (maxPacketLen)
|
||||
|
||||
reason := byteSliceToString(payload)
|
||||
fmt.Printf("disconnected (%s)\n", reason)
|
||||
|
||||
os.Exit(0)
|
||||
} else {
|
||||
fmt.Printf("unknown control message: %x\n", data)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
if header.Flags.Compression {
|
||||
huff := huffman.Huffman{}
|
||||
var err error
|
||||
payload, err = huff.Decompress(payload)
|
||||
if err != nil {
|
||||
fmt.Printf("huffman error: %v\n", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
client.onPacketPayload(headerRaw, payload)
|
||||
}
|
||||
|
||||
func main() {
|
||||
ch := make(chan []byte, maxPacksize)
|
||||
|
||||
|
|
@ -365,23 +51,23 @@ func main() {
|
|||
os.Exit(1)
|
||||
}
|
||||
|
||||
client := &TeeworldsClient{
|
||||
clientToken: [4]byte{0x01, 0x02, 0x03, 0x04},
|
||||
serverToken: [4]byte{0xff, 0xff, 0xff, 0xff},
|
||||
conn: conn,
|
||||
client := &protocol7.Connection{
|
||||
ClientToken: [4]byte{0x01, 0x02, 0x03, 0x04},
|
||||
ServerToken: [4]byte{0xff, 0xff, 0xff, 0xff},
|
||||
Conn: conn,
|
||||
Ack: 0,
|
||||
Players: make([]Player, MaxClients),
|
||||
Players: make([]protocol7.Player, MaxClients),
|
||||
}
|
||||
|
||||
go readNetwork(ch, client.conn)
|
||||
go readNetwork(ch, client.Conn)
|
||||
|
||||
conn.Write(ctrlToken(client.clientToken[:]))
|
||||
client.SendCtrlToken([]byte{0xff, 0xff, 0xff, 0xff})
|
||||
|
||||
for {
|
||||
time.Sleep(10_000_000)
|
||||
select {
|
||||
case msg := <-ch:
|
||||
client.onPacket(msg)
|
||||
client.OnPacket(msg)
|
||||
default:
|
||||
// do nothing
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue