Fix gvisor icmp destination

Author: nekohasekai

ping/destination.go

@@ -1,6 +1,7 @@
 package ping
 
 import (
+	"context"
 	"errors"
 	"net/netip"
 	"os"
@@ -16,12 +17,13 @@ import (
 var _ tun.DirectRouteDestination = (*Destination)(nil)
 
 type Destination struct {
-	logger       logger.Logger
+	ctx          context.Context
+	logger       logger.ContextLogger
 	routeContext tun.DirectRouteContext
 	conn         *Conn
 }
 
-func ConnectDestination(logger logger.Logger, controlFunc control.Func, address netip.Addr, routeContext tun.DirectRouteContext) (tun.DirectRouteDestination, error) {
+func ConnectDestination(ctx context.Context, logger logger.ContextLogger, controlFunc control.Func, address netip.Addr, routeContext tun.DirectRouteContext) (tun.DirectRouteDestination, error) {
 	var (
 		conn *Conn
 		err  error
@@ -39,6 +41,7 @@ func ConnectDestination(logger logger.Logger, controlFunc control.Func, address
 		return nil, err
 	}
 	d := &Destination{
+		ctx:          ctx,
 		logger:       logger,
 		routeContext: routeContext,
 		conn:         conn,
@@ -54,13 +57,13 @@ func (d *Destination) loopRead() {
 		if err != nil {
 			buffer.Release()
 			if !E.IsClosed(err) {
-				d.logger.Error(E.Cause(err, "receive ICMP echo reply"))
+				d.logger.ErrorContext(d.ctx, E.Cause(err, "receive ICMP echo reply"))
 			}
 			return
 		}
 		err = d.routeContext.WritePacket(buffer.Bytes())
 		if err != nil {
-			d.logger.Error(E.Cause(err, "write ICMP echo reply"))
+			d.logger.Error(d.ctx, E.Cause(err, "write ICMP echo reply"))
 		}
 		buffer.Release()
 	}

ping/socket_unix.go

@@ -12,6 +12,7 @@ import (
 	"github.com/sagernet/sing/common/control"
 	E "github.com/sagernet/sing/common/exceptions"
 	M "github.com/sagernet/sing/common/metadata"
+
 	"golang.org/x/sys/unix"
 )
 

route_nat.go

@@ -6,8 +6,14 @@ import (
 
 	"github.com/sagernet/sing-tun/internal/gtcpip/checksum"
 	"github.com/sagernet/sing-tun/internal/gtcpip/header"
+	"github.com/sagernet/sing/common/buf"
 )
 
+type DirectRouteDestination interface {
+	WritePacket(packet *buf.Buffer) error
+	Close() error
+}
+
 type NatMapping struct {
 	access    sync.RWMutex
 	sessions  map[DirectRouteSession]DirectRouteContext

route_nat_gvisor.go

@@ -5,14 +5,14 @@ package tun
 import (
 	"github.com/sagernet/gvisor/pkg/tcpip"
 	"github.com/sagernet/gvisor/pkg/tcpip/header"
-	stack "github.com/sagernet/gvisor/pkg/tcpip/stack"
+	"github.com/sagernet/gvisor/pkg/tcpip/stack"
 	"github.com/sagernet/sing/common/buf"
 )
 
-type DirectRouteDestination interface {
-	WritePacket(packet *buf.Buffer) error
-	WritePacketBuffer(packetBuffer *stack.PacketBuffer) error
-	Close() error
+func icmpWritePacketBuffer(action DirectRouteDestination, packetBuffer *stack.PacketBuffer) error {
+	packetSlice := packetBuffer.TransportHeader().Slice()
+	packetSlice = append(packetSlice, packetBuffer.Data().AsRange().ToSlice()...)
+	return action.WritePacket(buf.As(packetSlice))
 }
 
 func (w *NatWriter) RewritePacketBuffer(packetBuffer *stack.PacketBuffer) {

route_nat_non_gvisor.go

@@ -62,8 +62,7 @@ func (f *ICMPForwarder) HandlePacket(id stack.TransportEndpointID, pkt *stack.Pa
 		}
 		if action != nil {
 			// TODO: handle error
-			pkt.IncRef()
-			_ = action.WritePacketBuffer(pkt)
+			_ = icmpWritePacketBuffer(action, pkt)
 			return true
 		}
 		icmpHdr.SetType(header.ICMPv4EchoReply)
@@ -120,7 +119,7 @@ func (f *ICMPForwarder) HandlePacket(id stack.TransportEndpointID, pkt *stack.Pa
 		if action != nil {
 			// TODO: handle error
 			pkt.IncRef()
-			_ = action.WritePacketBuffer(pkt)
+			_ = icmpWritePacketBuffer(action, pkt)
 			return true
 		}
 		icmpHdr.SetType(header.ICMPv6EchoReply)