apps/mochi-mqtt
1
0
Fork 0
mirror of https://github.com/mochi-mqtt/server.git synced 2025-11-03 02:23:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enforce server max packet (#121)

Browse Source 
* Enforce Server Maximum Packet Size on client read
* Fix tests
This commit is contained in:
JB
2022-12-16 00:30:23 +00:00
committed by GitHub
parent 88861c219d
commit d8f28cb843
4 changed files with 40 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
server_test.go
View File

@@ -2291,6 +2291,21 @@ func TestServerRecievePacketDisconnectClientZeroNonZero(t *testing.T) {
require.Equal(t, packets.TPacketData[packets.Disconnect].Get(packets.TDisconnectZeroNonZeroExpiry).RawBytes, buf)
}
func TestServerRecievePacketDisconnectClient(t *testing.T) {
s := newServer()
cl, r, w := newTestClient()
go func() {
err := s.DisconnectClient(cl, packets.CodeDisconnect)
require.NoError(t, err)
w.Close()
}()
buf, err := io.ReadAll(r)
require.NoError(t, err)
require.Equal(t, packets.TPacketData[packets.Disconnect].Get(packets.TDisconnect).RawBytes, buf)
}
func TestServerProcessPacketDisconnect(t *testing.T) {
s := newServer()
cl, _, _ := newTestClient()