teeworlds-go/go-teeworlds-protocol
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Compress packet payload if compression flag is set

Browse source
This commit is contained in:
ChillerDragon 2024-06-21 11:00:40 +08:00
parent 4f37d5de48
commit 4d05818f15
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
protocol7/packet.go
View file

@ -3,6 +3,7 @@ package protocol7
import ( import (
"slices" "slices"
"github.com/teeworlds-go/huffman"
"github.com/teeworlds-go/teeworlds/chunk7" "github.com/teeworlds-go/teeworlds/chunk7"
"github.com/teeworlds-go/teeworlds/messages7" "github.com/teeworlds-go/teeworlds/messages7"
"github.com/teeworlds-go/teeworlds/network7" "github.com/teeworlds-go/teeworlds/network7"
@ -96,6 +97,16 @@ func (packet *Packet) Pack(connection *Connection) []byte {
packet.Header.Flags.Control = true packet.Header.Flags.Control = true
} }
if packet.Header.Flags.Compression {
// TODO: store huffman object in connection to avoid reallocating memory
huff := huffman.Huffman{}
var err error
payload, err = huff.Compress(payload)
if err != nil {
panic(err)
}
}
return slices.Concat( return slices.Concat(
packet.Header.Pack(), packet.Header.Pack(),
payload, payload,