rcon/rcon_test.go

193 lines
4.1 KiB
Go
Raw Permalink Normal View History

package rcon
import (
"bytes"
"encoding/binary"
"net"
"testing"
)
const password = "password"
func startTestServer(fn func(net.Conn, *bytes.Buffer)) (string, error) {
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
return "", err
}
go func() {
conn, err := listener.Accept()
if err != nil {
return
}
defer conn.Close()
buf := make([]byte, maxPackageSize)
_, err = conn.Read(buf)
if err != nil {
return
}
var packetSize, requestID, cmdType int32
var str []byte
b := bytes.NewBuffer(buf)
binary.Read(b, binary.LittleEndian, &packetSize)
binary.Read(b, binary.LittleEndian, &requestID)
binary.Read(b, binary.LittleEndian, &cmdType)
str, err = b.ReadBytes(0x00)
if err != nil {
return
}
if string(str[:len(str)-1]) != password {
requestID = -1
}
2021-03-06 16:09:18 +01:00
conn.Write(buildPackage(requestID, typeAuthResponse, []byte{}).Bytes())
if fn != nil {
b.Reset()
fn(conn, b)
}
}()
return listener.Addr().String(), nil
}
2021-03-06 16:09:18 +01:00
func buildStartOfPackage(size int, requestID, responseType int32, body []byte) *bytes.Buffer {
b := bytes.NewBuffer([]byte{})
binary.Write(b, binary.LittleEndian, int32(minPackageSize+size))
binary.Write(b, binary.LittleEndian, int32(requestID))
binary.Write(b, binary.LittleEndian, int32(responseType))
binary.Write(b, binary.LittleEndian, body)
return b
}
func buildEndOfPackage(body []byte) *bytes.Buffer {
b := bytes.NewBuffer([]byte{})
binary.Write(b, binary.LittleEndian, body)
binary.Write(b, binary.LittleEndian, byte(0))
binary.Write(b, binary.LittleEndian, byte(0))
return b
}
func buildPackage(requestID, responseType int32, body []byte) *bytes.Buffer {
b := bytes.NewBuffer([]byte{})
b.Write(buildStartOfPackage(len(body), requestID, responseType, body).Bytes())
b.Write(buildEndOfPackage([]byte{}).Bytes())
return b
}
func buildPackageWithLength(requestID, responseType int32, length int) *bytes.Buffer {
return buildPackage(requestID, responseType, writeSpace(length).Bytes())
}
func writeSpace(length int) *bytes.Buffer {
b := bytes.NewBuffer([]byte{})
for i := 0; i < length; i++ {
binary.Write(b, binary.LittleEndian, byte(' '))
}
return b
}
func TestAuth(t *testing.T) {
addr, err := startTestServer(nil)
if err != nil {
t.Fatal(err)
}
rc, err := Dial(addr, password)
if err != nil {
t.Fatal(err)
}
err = rc.Close()
if err != nil {
t.Fatal(err)
}
}
func TestMultipacket(t *testing.T) {
2021-03-06 16:09:18 +01:00
t.Run("write one package at once", func(t *testing.T) {
addr, err := startTestServer(func(c net.Conn, b *bytes.Buffer) {
b.Write(buildPackageWithLength(123, typeResponseValue, 4000).Bytes())
c.Write(b.Bytes())
})
if err != nil {
t.Fatal(err)
}
rc, err := Dial(addr, password)
2021-03-06 16:09:18 +01:00
if err != nil {
t.Fatal(err)
}
str, _, err := rc.Read()
if err != nil {
t.Fatal(err)
}
if len(str) != 4000 {
t.Fatal("response length not correct")
}
})
2021-03-06 16:09:18 +01:00
t.Run("write one package body over multiple writes", func(t *testing.T) {
length := 4000
addr, err := startTestServer(func(c net.Conn, b *bytes.Buffer) {
c.Write(buildStartOfPackage(length, 123, typeResponseValue, writeSpace(length/2).Bytes()).Bytes())
2021-03-06 16:09:18 +01:00
c.Write(buildEndOfPackage(writeSpace(length / 2).Bytes()).Bytes())
})
if err != nil {
t.Fatal(err)
}
rc, err := Dial(addr, password)
2021-03-06 16:09:18 +01:00
if err != nil {
t.Fatal(err)
}
2021-03-06 16:09:18 +01:00
str, _, err := rc.Read()
if err != nil {
t.Fatal(err)
}
if len(str) != 4000 {
t.Fatal("response length not correct")
}
2021-03-06 16:09:18 +01:00
})
t.Run("write one package size over multiple writes", func(t *testing.T) {
length := 2000
addr, err := startTestServer(func(c net.Conn, b *bytes.Buffer) {
binary.Write(b, binary.LittleEndian, int32(length+minPackageSize))
c.Write(b.Bytes()[:len(b.Bytes())-3]) // send without the last three bytes
b.Reset()
c.Write(buildPackageWithLength(123, typeResponseValue, length).Bytes()[1:]) // skip first byte
})
if err != nil {
t.Fatal(err)
}
rc, err := Dial(addr, password)
2021-03-06 16:09:18 +01:00
if err != nil {
t.Fatal(err)
}
str, _, err := rc.Read()
if err != nil {
t.Fatal(err)
}
if len(str) != 2000 {
t.Fatal("response length not correct")
}
})
}