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

Update udpant usages

Browse source
This commit is contained in:
世界 2024-11-26 11:37:44 +08:00
parent c177abb523
commit 59a6bdc1fa
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
5 changed files with 34 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

6
stack_gvisor.go
View file

@ -34,6 +34,7 @@ type GVisor struct {
logger logger.Logger
stack *stack.Stack
endpoint stack.LinkEndpoint
udpForwarder *UDPForwarder
}
type GVisorTun interface {
@ -71,9 +72,11 @@ func (t *GVisor) Start() error {
return err
}
ipStack.SetTransportProtocolHandler(tcp.ProtocolNumber, NewTCPForwarder(t.ctx, ipStack, t.handler).HandlePacket)
ipStack.SetTransportProtocolHandler(udp.ProtocolNumber, NewUDPForwarder(t.ctx, ipStack, t.handler, t.udpTimeout).HandlePacket)
udpForwarder := NewUDPForwarder(t.ctx, ipStack, t.handler, t.udpTimeout)
ipStack.SetTransportProtocolHandler(udp.ProtocolNumber, udpForwarder.HandlePacket)
t.stack = ipStack
t.endpoint = linkEndpoint
t.udpForwarder = udpForwarder
return nil
}
@ -86,6 +89,7 @@ func (t *GVisor) Close() error {
for _, endpoint := range t.stack.CleanupEndpoints() {
endpoint.Abort()
}
t.udpForwarder.Close()
return nil
}