mirror of
https://github.com/SagerNet/sing-tun.git
synced 2025-04-03 20:07:40 +03:00
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
//go:build with_gvisor
|
|
|
|
package tun
|
|
|
|
import (
|
|
"context"
|
|
"math"
|
|
"net/netip"
|
|
|
|
"github.com/sagernet/gvisor/pkg/buffer"
|
|
"github.com/sagernet/gvisor/pkg/tcpip"
|
|
"github.com/sagernet/gvisor/pkg/tcpip/checksum"
|
|
"github.com/sagernet/gvisor/pkg/tcpip/header"
|
|
"github.com/sagernet/gvisor/pkg/tcpip/stack"
|
|
"github.com/sagernet/sing/common/buf"
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
N "github.com/sagernet/sing/common/network"
|
|
"github.com/sagernet/sing/common/udpnat"
|
|
)
|
|
|
|
type UDPForwarder struct {
|
|
ctx context.Context
|
|
stack *stack.Stack
|
|
udpNat *udpnat.Service[netip.AddrPort]
|
|
}
|
|
|
|
func NewUDPForwarder(ctx context.Context, stack *stack.Stack, handler Handler, udpTimeout int64) *UDPForwarder {
|
|
return &UDPForwarder{
|
|
ctx: ctx,
|
|
stack: stack,
|
|
udpNat: udpnat.New[netip.AddrPort](udpTimeout, handler),
|
|
}
|
|
}
|
|
|
|
func (f *UDPForwarder) HandlePacket(id stack.TransportEndpointID, pkt stack.PacketBufferPtr) bool {
|
|
var upstreamMetadata M.Metadata
|
|
upstreamMetadata.Source = M.SocksaddrFrom(AddrFromAddress(id.RemoteAddress), id.RemotePort)
|
|
upstreamMetadata.Destination = M.SocksaddrFrom(AddrFromAddress(id.LocalAddress), id.LocalPort)
|
|
var netProto tcpip.NetworkProtocolNumber
|
|
if upstreamMetadata.Source.IsIPv4() {
|
|
netProto = header.IPv4ProtocolNumber
|
|
} else {
|
|
netProto = header.IPv6ProtocolNumber
|
|
}
|
|
f.udpNat.NewPacket(
|
|
f.ctx,
|
|
upstreamMetadata.Source.AddrPort(),
|
|
buf.As(pkt.Data().AsRange().ToSlice()),
|
|
upstreamMetadata,
|
|
func(natConn N.PacketConn) N.PacketWriter {
|
|
return &UDPBackWriter{f.stack, id.RemoteAddress, id.RemotePort, netProto}
|
|
},
|
|
)
|
|
return true
|
|
}
|
|
|
|
type UDPBackWriter struct {
|
|
stack *stack.Stack
|
|
source tcpip.Address
|
|
sourcePort uint16
|
|
sourceNetwork tcpip.NetworkProtocolNumber
|
|
}
|
|
|
|
func (w *UDPBackWriter) WritePacket(packetBuffer *buf.Buffer, destination M.Socksaddr) error {
|
|
defer packetBuffer.Release()
|
|
|
|
route, err := w.stack.FindRoute(
|
|
defaultNIC,
|
|
AddressFromAddr(destination.Addr),
|
|
w.source,
|
|
w.sourceNetwork,
|
|
false,
|
|
)
|
|
if err != nil {
|
|
return wrapStackError(err)
|
|
}
|
|
defer route.Release()
|
|
|
|
packet := stack.NewPacketBuffer(stack.PacketBufferOptions{
|
|
ReserveHeaderBytes: header.UDPMinimumSize + int(route.MaxHeaderLength()),
|
|
Payload: buffer.MakeWithData(packetBuffer.Bytes()),
|
|
})
|
|
defer packet.DecRef()
|
|
|
|
packet.TransportProtocolNumber = header.UDPProtocolNumber
|
|
udpHdr := header.UDP(packet.TransportHeader().Push(header.UDPMinimumSize))
|
|
pLen := uint16(packet.Size())
|
|
udpHdr.Encode(&header.UDPFields{
|
|
SrcPort: destination.Port,
|
|
DstPort: w.sourcePort,
|
|
Length: pLen,
|
|
})
|
|
|
|
if route.RequiresTXTransportChecksum() && w.sourceNetwork == header.IPv6ProtocolNumber {
|
|
xsum := udpHdr.CalculateChecksum(checksum.Combine(
|
|
route.PseudoHeaderChecksum(header.UDPProtocolNumber, pLen),
|
|
packet.Data().Checksum(),
|
|
))
|
|
if xsum != math.MaxUint16 {
|
|
xsum = ^xsum
|
|
}
|
|
udpHdr.SetChecksum(xsum)
|
|
}
|
|
|
|
err = route.WritePacket(stack.NetworkHeaderParams{
|
|
Protocol: header.UDPProtocolNumber,
|
|
TTL: route.DefaultTTL(),
|
|
TOS: 0,
|
|
}, packet)
|
|
|
|
if err != nil {
|
|
route.Stats().UDP.PacketSendErrors.Increment()
|
|
return wrapStackError(err)
|
|
}
|
|
|
|
route.Stats().UDP.PacketsSent.Increment()
|
|
return nil
|
|
}
|