onyx_online/q3rcon-proxy
1
0
Fork 0
mirror of https://github.com/onyx-and-iris/q3rcon-proxy.git synced 2025-01-18 08:40:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

add isValidPacket

Browse Source 
ignore packets that don't match q3 rcon/query
This commit is contained in:
onyx-and-iris 2024-01-27 14:48:39 +00:00
parent 270f993cbc
commit ab91c219b9
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/udpproxy/udpproxy.go
View File

@ -36,6 +36,10 @@ func New(port, target string) (*Client, error) {
}, nil }, nil
} }
func (c *Client) isValidPacket(header []byte) bool {
return string(header[:8]) == "\xff\xff\xff\xffrcon" || string(header[:13]) == "\xff\xff\xff\xffgetstatus" || string(header[:11]) == "\xff\xff\xff\xffgetinfo"
}
func (c *Client) ListenAndServe() error { func (c *Client) ListenAndServe() error {
var err error var err error
c.proxyConn, err = net.ListenUDP("udp", c.laddr) c.proxyConn, err = net.ListenUDP("udp", c.laddr)
@ -52,6 +56,10 @@ func (c *Client) ListenAndServe() error {
log.Println(err) log.Println(err)
} }
if !c.isValidPacket(buf[:16]) {
continue
}
session, found := c.sessions[caddr.String()] session, found := c.sessions[caddr.String()]
if !found { if !found {
session, err = createSession(caddr, c.raddr, c.proxyConn) session, err = createSession(caddr, c.raddr, c.proxyConn)