Jelajahi Sumber

Improve the request for UDP Associate in Socks5

RPRX 4 tahun lalu
induk
melakukan
ceff4185dc
1 mengubah file dengan 6 tambahan dan 2 penghapusan
  1. 6 2
      proxy/socks/protocol.go

+ 6 - 2
proxy/socks/protocol.go

@@ -465,8 +465,12 @@ func ClientHandshake(request *protocol.RequestHeader, reader io.Reader, writer i
 		command = byte(cmdUDPAssociate)
 		command = byte(cmdUDPAssociate)
 	}
 	}
 	common.Must2(b.Write([]byte{socks5Version, command, 0x00 /* reserved */}))
 	common.Must2(b.Write([]byte{socks5Version, command, 0x00 /* reserved */}))
-	if err := addrParser.WriteAddressPort(b, request.Address, request.Port); err != nil {
-		return nil, err
+	if request.Command == protocol.RequestCommandUDP {
+		common.Must2(b.Write([]byte{1, 0, 0, 0, 0, 0, 0 /* RFC 1928 */}))
+	} else {
+		if err := addrParser.WriteAddressPort(b, request.Address, request.Port); err != nil {
+			return nil, err
+		}
 	}
 	}
 
 
 	if err := buf.WriteAllBytes(writer, b.Bytes()); err != nil {
 	if err := buf.WriteAllBytes(writer, b.Bytes()); err != nil {