pkg/udpproxy/session.go

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

pkg/udpproxy/validator.go

@@ -1,36 +1,14 @@
 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 bytes.Equal(buf[:len(v.rconRequestHeader)], v.rconRequestHeader)
+	return string(buf[:8]) == "\xff\xff\xff\xffrcon"
 }
 
 func (v *validator) isQueryRequestPacket(buf []byte) bool {
-	return bytes.Equal(buf[:len(v.getstatusRequestHeader)], v.getstatusRequestHeader) || bytes.Equal(buf[:len(v.getinfoRequestHeader)], v.getinfoRequestHeader)
+	return string(buf[:13]) == "\xff\xff\xff\xffgetstatus" || string(buf[:11]) == "\xff\xff\xff\xffgetinfo"
 }
 
 func (v *validator) isValidRequestPacket(buf []byte) bool {
@@ -38,11 +16,11 @@ func (v *validator) isValidRequestPacket(buf []byte) bool {
 }
 
 func (v *validator) isRconResponsePacket(buf []byte) bool {
-	return bytes.Equal(buf[:len(v.rconResponseHeader)], v.rconResponseHeader)
+	return string(buf[:9]) == "\xff\xff\xff\xffprint"
 }
 
 func (v *validator) isQueryResponsePacket(buf []byte) bool {
-	return bytes.Equal(buf[:len(v.getstatusResponseHeader)], v.getstatusResponseHeader) || bytes.Equal(buf[:len(v.getinfoResponseHeader)], v.getinfoResponseHeader)
+	return string(buf[:18]) == "\xff\xff\xff\xffstatusResponse" || string(buf[:16]) == "\xff\xff\xff\xffinfoResponse"
 }
 
 func (v *validator) isValidResponsePacket(buf []byte) bool {
@@ -50,5 +28,5 @@ func (v *validator) isValidResponsePacket(buf []byte) bool {
 }
 
 func (v *validator) isBadRconResponse(buf []byte) bool {
-	return bytes.Equal(buf[len(v.rconResponseHeader):len(v.rconResponseHeader)+len(v.badRconIdentifier)], v.badRconIdentifier)
+	return string(buf[10:18]) == "Bad rcon"
 }