diff --git a/netstack/forward_tcp.go b/netstack/forward_tcp.go index 2961397..e48d69b 100644 --- a/netstack/forward_tcp.go +++ b/netstack/forward_tcp.go @@ -34,8 +34,8 @@ func ForwardTCPConn(originConn *TcpConn, stun_quic_conn quic.Connection) { binary.BigEndian.PutUint16(portBytes, originConn.ID().LocalPort) new_quic_stream.Write(portBytes) - go proxy.ForwardQ2T(new_quic_stream, originConn, stun_quic_conn) - go proxy.ForwardT2Q(originConn, new_quic_stream, stun_quic_conn) + go proxy.ForwardQ2T(new_quic_stream, originConn) + go proxy.ForwardT2Q(originConn, new_quic_stream) } func NewTcpForwarder(s *stack.Stack, stun_quic_conn quic.Connection) *tcp.Forwarder { diff --git a/netstack/forward_udp.go b/netstack/forward_udp.go index 21aad10..a4f6485 100644 --- a/netstack/forward_udp.go +++ b/netstack/forward_udp.go @@ -33,8 +33,8 @@ func ForwardUdpConn(originConn *udpConn, stun_quic_conn quic.Connection) { binary.BigEndian.PutUint16(portBytes, originConn.ID().LocalPort) new_quic_stream.Write(portBytes) - go proxy.ForwardQ2T(new_quic_stream, originConn, stun_quic_conn) - go proxy.ForwardT2Q(originConn, new_quic_stream, stun_quic_conn) + go proxy.ForwardQ2T(new_quic_stream, originConn) + go proxy.ForwardT2Q(originConn, new_quic_stream) } func NewUdpForwarder(s *stack.Stack, stun_quic_conn quic.Connection) *udp.Forwarder {