onyx_online/q3rcon-proxy
1
0
Fork 0
mirror of https://github.com/onyx-and-iris/q3rcon-proxy.git synced 2025-02-23 02:05:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove string casts

Browse Source 
compare byte slices
This commit is contained in:
onyx-and-iris 2024-04-11 19:24:43 +01:00
parent 2c8ae43303
commit 3a8758a4d7
2 changed files with 27 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/udpproxy/session.go
View File

@ -29,6 +29,7 @@ func newSession(caddr *net.UDPAddr, raddr *net.UDPAddr, proxyConn *net.UDPConn)
proxyConn: proxyConn, proxyConn: proxyConn,
caddr: caddr, caddr: caddr,
updateTime: time.Now(), updateTime: time.Now(),
validator: newValidator(),
} }
go session.listen() go session.listen()

31
pkg/udpproxy/validator.go
View File

@ -1,14 +1,35 @@
package udpproxy package udpproxy
import "bytes"
type validator struct { type validator struct {
rconRequestHeader []byte
getstatusRequestHeader []byte
getinfoRequestHeader []byte
rconResponseHeader []byte
getstatusResponseHeader []byte
getinfoResponseHeader []byte
badRconIdentifier []byte
}
func newValidator() validator {
return validator{
rconRequestHeader: []byte("\xff\xff\xff\xffrcon"),
getstatusRequestHeader: []byte("\xff\xff\xff\xffgetstatus"),
getinfoRequestHeader: []byte("\xff\xff\xff\xffgetinfo"),
rconResponseHeader: []byte("\xff\xff\xff\xffprint\n"),
getstatusResponseHeader: []byte("\xff\xff\xff\xffstatusResponse\n"),
getinfoResponseHeader: []byte("\xff\xff\xff\xffinfoResponse\n"),
badRconIdentifier: []byte("Bad rcon"),
}
} }
func (v *validator) isRconRequestPacket(buf []byte) bool { func (v *validator) isRconRequestPacket(buf []byte) bool {
return string(buf[:8]) == "\xff\xff\xff\xffrcon" return bytes.Equal(buf[:len(v.rconRequestHeader)], v.rconRequestHeader)
} }
func (v *validator) isQueryRequestPacket(buf []byte) bool { func (v *validator) isQueryRequestPacket(buf []byte) bool {
return string(buf[:13]) == "\xff\xff\xff\xffgetstatus" || string(buf[:11]) == "\xff\xff\xff\xffgetinfo" return bytes.Equal(buf[:len(v.getstatusRequestHeader)], v.getstatusRequestHeader) || bytes.Equal(buf[:len(v.getinfoRequestHeader)], v.getinfoRequestHeader)
} }
func (v *validator) isValidRequestPacket(buf []byte) bool { func (v *validator) isValidRequestPacket(buf []byte) bool {
@ -16,11 +37,11 @@ func (v *validator) isValidRequestPacket(buf []byte) bool {
} }
func (v *validator) isRconResponsePacket(buf []byte) bool { func (v *validator) isRconResponsePacket(buf []byte) bool {
return string(buf[:9]) == "\xff\xff\xff\xffprint" return bytes.Equal(buf[:len(v.rconResponseHeader)], v.rconResponseHeader)
} }
func (v *validator) isQueryResponsePacket(buf []byte) bool { func (v *validator) isQueryResponsePacket(buf []byte) bool {
return string(buf[:18]) == "\xff\xff\xff\xffstatusResponse" || string(buf[:16]) == "\xff\xff\xff\xffinfoResponse" return bytes.Equal(buf[:len(v.getstatusResponseHeader)], v.getstatusResponseHeader) || bytes.Equal(buf[:len(v.getinfoResponseHeader)], v.getinfoResponseHeader)
} }
func (v *validator) isValidResponsePacket(buf []byte) bool { func (v *validator) isValidResponsePacket(buf []byte) bool {
@ -28,5 +49,5 @@ func (v *validator) isValidResponsePacket(buf []byte) bool {
} }
func (v *validator) isBadRconResponse(buf []byte) bool { func (v *validator) isBadRconResponse(buf []byte) bool {
return string(buf[10:18]) == "Bad rcon" return bytes.Equal(buf[len(v.rconResponseHeader):len(v.rconResponseHeader)+len(v.badRconIdentifier)], v.badRconIdentifier)
} }