func countClientNums(buf []byte) (count int, err error) {
p := cubecode.NewPacket(buf)
for p.HasRemaining() {
_, err = p.ReadInt()
if err != nil {
return
}
count++
}
return
}
// queries the given server and returns the response and an error in case something went wrong. clientNum is optional, put 0 if not needed.
func (s *Server) queryServer(request []byte) (response *cubecode.Packet, err error) {
// connect to server at port+1 (port is the port you connect to in game, sauerbraten listens on the one higher port for BasicInfo queries
var conn *net.UDPConn
conn, err = net.DialUDP("udp", nil, s.addr)
if err != nil {
return
}
defer conn.Close()
// set up a buffered reader
bufconn := bufio.NewReader(conn)
// send the request to server
_, err = conn.Write(request)
if err != nil {
return
}
// receive response from server with 5 second timeout
rawResponse := make([]byte, MaxPacketLength)
var bytesRead int
conn.SetReadDeadline(time.Now().Add(s.timeOut))
bytesRead, err = bufconn.Read(rawResponse)
if err != nil {
return
}
// trim response to what's actually from the server
packet := cubecode.NewPacket(rawResponse[:bytesRead])
if bytesRead < 5 {
err = errors.New("extinfo: invalid response: too short")
return
}
// do some basic checks on the response
infoType, err := packet.ReadByte()
if err != nil {
return
}
command, err := packet.ReadByte() // only valid if infoType == EXTENDED_INFO
if err != nil {
return
}
if infoType == InfoTypeExtended {
var version, commandError byte
if command == ExtInfoTypeClientInfo {
if bytesRead < 6 {
err = errors.New("extinfo: invalid response: too short")
return
}
version = rawResponse[4]
commandError = rawResponse[5]
} else {
version = rawResponse[3]
commandError = rawResponse[4]
}
if infoType != request[0] || command != request[1] {
err = errors.New("extinfo: invalid response: response does not match request")
return
}
// this package only support extinfo protocol version 105
if version != ExtInfoVersion {
err = errors.New("extinfo: wrong version: expected " + strconv.Itoa(int(ExtInfoVersion)) + ", got " + strconv.Itoa(int(version)))
return
}
if commandError == ExtInfoError {
switch command {
case ExtInfoTypeClientInfo:
err = errors.New("extinfo: no client with cn " + strconv.Itoa(int(request[2])))
case ExtInfoTypeTeamScores:
err = errors.New("extinfo: server is not running a team mode")
}
return
}
}
// if not a response to EXTENDED_INFO_CLIENT_INFO, we are done
if infoType != InfoTypeExtended || command != ExtInfoTypeClientInfo {
offset := 0
if infoType == InfoTypeExtended {
switch command {
case ExtInfoTypeUptime:
offset = 4
case ExtInfoTypeTeamScores:
offset = 5
}
}
response = cubecode.NewPacket(rawResponse[offset:])
return
}
// handle response to EXTENDED_INFO_CLIENT_INFO
// some server mods silently fail to implement responses → fail gracefully
if len(rawResponse) < 7 || rawResponse[6] != ClientInfoResponseTypeCNs {
err = errors.New("extinfo: invalid response")
return
}
// get CNs out of the reponse, ignore 7 first bytes, which were:
// EXTENDED_INFO, EXTENDED_INFO_CLIENT_INFO, CN from request, EXTENDED_INFO_ACK, EXTENDED_INFO_VERSION, EXTENDED_INFO_NO_ERROR, EXTENDED_INFO_CLIENT_INFO_RESPONSE_CNS
clientNums := rawResponse[7:]
numberOfClients, err := countClientNums(clientNums)
if err != nil {
return
}
// for each client, receive a packet and append it to a new slice
clientInfos := make([]byte, 0, MaxPacketLength*numberOfClients)
for i := 0; i < numberOfClients; i++ {
// read from connection
clientInfo := make([]byte, MaxPacketLength)
conn.SetReadDeadline(time.Now().Add(s.timeOut))
_, err = bufconn.Read(clientInfo)
if err != nil {
return
}
// append bytes to slice
clientInfos = append(clientInfos, clientInfo...)
}
response = cubecode.NewPacket(clientInfos)
return
}
