mirror of
https://github.com/onyx-and-iris/q3rcon-proxy.git
synced 2025-04-20 04:23:48 +01:00
Compare commits
2 Commits
2c8ae43303
...
e8f213fded
Author | SHA1 | Date | |
---|---|---|---|
e8f213fded | |||
3a8758a4d7 |
@ -29,6 +29,7 @@ func newSession(caddr *net.UDPAddr, raddr *net.UDPAddr, proxyConn *net.UDPConn)
|
||||
proxyConn: proxyConn,
|
||||
caddr: caddr,
|
||||
updateTime: time.Now(),
|
||||
validator: newValidator(),
|
||||
}
|
||||
|
||||
go session.listen()
|
||||
|
@ -1,14 +1,36 @@
|
||||
package udpproxy
|
||||
|
||||
import "bytes"
|
||||
|
||||
type validator struct {
|
||||
rconRequestHeader []byte
|
||||
getstatusRequestHeader []byte
|
||||
getinfoRequestHeader []byte
|
||||
rconResponseHeader []byte
|
||||
getstatusResponseHeader []byte
|
||||
getinfoResponseHeader []byte
|
||||
badRconIdentifier []byte
|
||||
}
|
||||
|
||||
func newValidator() validator {
|
||||
v := 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"),
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
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 {
|
||||
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 {
|
||||
@ -16,11 +38,11 @@ func (v *validator) isValidRequestPacket(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 {
|
||||
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 {
|
||||
@ -28,5 +50,5 @@ func (v *validator) isValidResponsePacket(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)
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user