SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-04 20:37:40 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add unidirectional NATPacketConn

Browse source
This commit is contained in:
世界 2023-10-28 20:59:48 +08:00
parent 8002db54c0
commit afcc9cb766
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 52 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

63
common/bufio/nat.go
View file

@ -9,21 +9,62 @@ import (
N "github.com/sagernet/sing/common/network" N "github.com/sagernet/sing/common/network"
) )
type NATPacketConn struct { type NATPacketConn interface {
N.NetPacketConn N.NetPacketConn
origin M.Socksaddr UpdateDestination(destinationAddress netip.Addr)
destination M.Socksaddr
} }
func NewNATPacketConn(conn N.NetPacketConn, origin M.Socksaddr, destination M.Socksaddr) *NATPacketConn { func NewUnidirectionalNATPacketConn(conn N.NetPacketConn, origin M.Socksaddr, destination M.Socksaddr) NATPacketConn {
return &NATPacketConn{ return &unidirectionalNATPacketConn{
NetPacketConn: conn, NetPacketConn: conn,
origin: origin, origin: origin,
destination: destination, destination: destination,
} }
} }
func (c *NATPacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) { func NewNATPacketConn(conn N.NetPacketConn, origin M.Socksaddr, destination M.Socksaddr) NATPacketConn {
return &bidirectionalNATPacketConn{
NetPacketConn: conn,
origin: origin,
destination: destination,
}
}
type unidirectionalNATPacketConn struct {
N.NetPacketConn
origin M.Socksaddr
destination M.Socksaddr
}
func (c *unidirectionalNATPacketConn) WriteTo(p []byte, addr net.Addr) (n int, err error) {
if M.SocksaddrFromNet(addr) == c.destination {
addr = c.origin.UDPAddr()
}
return c.NetPacketConn.WriteTo(p, addr)
}
func (c *unidirectionalNATPacketConn) WritePacket(buffer *buf.Buffer, destination M.Socksaddr) error {
if destination == c.destination {
destination = c.origin
}
return c.NetPacketConn.WritePacket(buffer, destination)
}
func (c *unidirectionalNATPacketConn) UpdateDestination(destinationAddress netip.Addr) {
c.destination = M.SocksaddrFrom(destinationAddress, c.destination.Port)
}
func (c *unidirectionalNATPacketConn) Upstream() any {
return c.NetPacketConn
}
type bidirectionalNATPacketConn struct {
N.NetPacketConn
origin M.Socksaddr
destination M.Socksaddr
}
func (c *bidirectionalNATPacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
n, addr, err = c.NetPacketConn.ReadFrom(p) n, addr, err = c.NetPacketConn.ReadFrom(p)
if err == nil && M.SocksaddrFromNet(addr) == c.origin { if err == nil && M.SocksaddrFromNet(addr) == c.origin {
addr = c.destination.UDPAddr() addr = c.destination.UDPAddr()
@ -31,14 +72,14 @@ func (c *NATPacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
return return
} }
func (c *NATPacketConn) WriteTo(p []byte, addr net.Addr) (n int, err error) { func (c *bidirectionalNATPacketConn) WriteTo(p []byte, addr net.Addr) (n int, err error) {
if M.SocksaddrFromNet(addr) == c.destination { if M.SocksaddrFromNet(addr) == c.destination {
addr = c.origin.UDPAddr() addr = c.origin.UDPAddr()
} }
return c.NetPacketConn.WriteTo(p, addr) return c.NetPacketConn.WriteTo(p, addr)
} }
func (c *NATPacketConn) ReadPacket(buffer *buf.Buffer) (destination M.Socksaddr, err error) { func (c *bidirectionalNATPacketConn) ReadPacket(buffer *buf.Buffer) (destination M.Socksaddr, err error) {
destination, err = c.NetPacketConn.ReadPacket(buffer) destination, err = c.NetPacketConn.ReadPacket(buffer)
if destination == c.origin { if destination == c.origin {
destination = c.destination destination = c.destination
@ -46,17 +87,17 @@ func (c *NATPacketConn) ReadPacket(buffer *buf.Buffer) (destination M.Socksaddr,
return return
} }
func (c *NATPacketConn) WritePacket(buffer *buf.Buffer, destination M.Socksaddr) error { func (c *bidirectionalNATPacketConn) WritePacket(buffer *buf.Buffer, destination M.Socksaddr) error {
if destination == c.destination { if destination == c.destination {
destination = c.origin destination = c.origin
} }
return c.NetPacketConn.WritePacket(buffer, destination) return c.NetPacketConn.WritePacket(buffer, destination)
} }
func (c *NATPacketConn) UpdateDestination(destinationAddress netip.Addr) { func (c *bidirectionalNATPacketConn) UpdateDestination(destinationAddress netip.Addr) {
c.destination = M.SocksaddrFrom(destinationAddress, c.destination.Port) c.destination = M.SocksaddrFrom(destinationAddress, c.destination.Port)
} }
func (c *NATPacketConn) Upstream() any { func (c *bidirectionalNATPacketConn) Upstream() any {
return c.NetPacketConn return c.NetPacketConn
} }