2022-07-03 20:59:25 +08:00
|
|
|
package outbound
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"net"
|
2023-04-02 12:01:19 +08:00
|
|
|
"net/netip"
|
2022-10-20 10:57:57 +08:00
|
|
|
"os"
|
2022-07-03 20:59:25 +08:00
|
|
|
"time"
|
|
|
|
|
2022-07-08 23:03:57 +08:00
|
|
|
"github.com/sagernet/sing-box/adapter"
|
2024-11-12 19:37:10 +08:00
|
|
|
"github.com/sagernet/sing-box/common/dialer"
|
2022-07-08 23:03:57 +08:00
|
|
|
C "github.com/sagernet/sing-box/constant"
|
2022-07-03 20:59:25 +08:00
|
|
|
"github.com/sagernet/sing/common"
|
|
|
|
"github.com/sagernet/sing/common/buf"
|
|
|
|
"github.com/sagernet/sing/common/bufio"
|
2023-03-23 15:14:21 +08:00
|
|
|
"github.com/sagernet/sing/common/canceler"
|
2022-07-03 20:59:25 +08:00
|
|
|
E "github.com/sagernet/sing/common/exceptions"
|
2023-10-25 12:00:00 +08:00
|
|
|
M "github.com/sagernet/sing/common/metadata"
|
2022-07-07 23:36:32 +08:00
|
|
|
N "github.com/sagernet/sing/common/network"
|
2022-07-03 20:59:25 +08:00
|
|
|
)
|
|
|
|
|
2022-07-07 23:36:32 +08:00
|
|
|
func NewConnection(ctx context.Context, this N.Dialer, conn net.Conn, metadata adapter.InboundContext) error {
|
2024-11-07 13:44:00 +08:00
|
|
|
defer conn.Close()
|
2022-07-07 23:36:32 +08:00
|
|
|
ctx = adapter.WithContext(ctx, &metadata)
|
|
|
|
var outConn net.Conn
|
|
|
|
var err error
|
|
|
|
if len(metadata.DestinationAddresses) > 0 {
|
2024-11-12 19:37:10 +08:00
|
|
|
if parallelDialer, isParallelDialer := this.(dialer.ParallelInterfaceDialer); isParallelDialer {
|
2024-11-13 19:05:28 +08:00
|
|
|
outConn, err = dialer.DialSerialNetwork(ctx, parallelDialer, N.NetworkTCP, metadata.Destination, metadata.DestinationAddresses, metadata.NetworkStrategy, metadata.NetworkType, metadata.FallbackNetworkType, metadata.FallbackDelay)
|
2024-11-12 19:37:10 +08:00
|
|
|
} else {
|
|
|
|
outConn, err = N.DialSerial(ctx, this, N.NetworkTCP, metadata.Destination, metadata.DestinationAddresses)
|
2023-09-06 19:13:39 +08:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
outConn, err = this.DialContext(ctx, N.NetworkTCP, metadata.Destination)
|
|
|
|
}
|
|
|
|
if err != nil {
|
2023-08-31 20:07:32 +08:00
|
|
|
return N.ReportHandshakeFailure(conn, err)
|
|
|
|
}
|
2024-10-21 23:38:34 +08:00
|
|
|
err = N.ReportConnHandshakeSuccess(conn, outConn)
|
2023-08-31 20:07:32 +08:00
|
|
|
if err != nil {
|
2023-11-13 13:53:52 +08:00
|
|
|
outConn.Close()
|
2023-08-31 20:07:32 +08:00
|
|
|
return err
|
2023-09-06 19:13:39 +08:00
|
|
|
}
|
|
|
|
return CopyEarlyConn(ctx, conn, outConn)
|
|
|
|
}
|
|
|
|
|
2022-07-07 23:36:32 +08:00
|
|
|
func NewPacketConnection(ctx context.Context, this N.Dialer, conn N.PacketConn, metadata adapter.InboundContext) error {
|
2024-11-07 13:44:00 +08:00
|
|
|
defer conn.Close()
|
2022-07-07 23:36:32 +08:00
|
|
|
ctx = adapter.WithContext(ctx, &metadata)
|
2024-11-07 13:44:00 +08:00
|
|
|
var (
|
|
|
|
outPacketConn net.PacketConn
|
|
|
|
outConn net.Conn
|
|
|
|
destinationAddress netip.Addr
|
|
|
|
err error
|
|
|
|
)
|
|
|
|
if metadata.UDPConnect {
|
|
|
|
if len(metadata.DestinationAddresses) > 0 {
|
2024-11-12 19:37:10 +08:00
|
|
|
if parallelDialer, isParallelDialer := this.(dialer.ParallelInterfaceDialer); isParallelDialer {
|
2024-11-13 19:05:28 +08:00
|
|
|
outConn, err = dialer.DialSerialNetwork(ctx, parallelDialer, N.NetworkUDP, metadata.Destination, metadata.DestinationAddresses, metadata.NetworkStrategy, metadata.NetworkType, metadata.FallbackNetworkType, metadata.FallbackDelay)
|
2024-11-12 19:37:10 +08:00
|
|
|
} else {
|
|
|
|
outConn, err = N.DialSerial(ctx, this, N.NetworkUDP, metadata.Destination, metadata.DestinationAddresses)
|
|
|
|
}
|
2024-11-07 13:44:00 +08:00
|
|
|
} else {
|
|
|
|
outConn, err = this.DialContext(ctx, N.NetworkUDP, metadata.Destination)
|
|
|
|
}
|
|
|
|
if err != nil {
|
|
|
|
return N.ReportHandshakeFailure(conn, err)
|
|
|
|
}
|
|
|
|
outPacketConn = bufio.NewUnbindPacketConn(outConn)
|
|
|
|
connRemoteAddr := M.AddrFromNet(outConn.RemoteAddr())
|
|
|
|
if connRemoteAddr != metadata.Destination.Addr {
|
|
|
|
destinationAddress = connRemoteAddr
|
|
|
|
}
|
2022-07-07 23:36:32 +08:00
|
|
|
} else {
|
2024-11-07 13:44:00 +08:00
|
|
|
if len(metadata.DestinationAddresses) > 0 {
|
2024-11-12 19:37:10 +08:00
|
|
|
if parallelDialer, isParallelDialer := this.(dialer.ParallelInterfaceDialer); isParallelDialer {
|
2024-11-13 19:05:28 +08:00
|
|
|
outPacketConn, destinationAddress, err = dialer.ListenSerialNetworkPacket(ctx, parallelDialer, metadata.Destination, metadata.DestinationAddresses, metadata.NetworkStrategy, metadata.NetworkType, metadata.FallbackNetworkType, metadata.FallbackDelay)
|
2024-11-12 19:37:10 +08:00
|
|
|
} else {
|
|
|
|
outPacketConn, destinationAddress, err = N.ListenSerial(ctx, this, metadata.Destination, metadata.DestinationAddresses)
|
|
|
|
}
|
2024-11-07 13:44:00 +08:00
|
|
|
} else {
|
|
|
|
outPacketConn, err = this.ListenPacket(ctx, metadata.Destination)
|
|
|
|
}
|
|
|
|
if err != nil {
|
|
|
|
return N.ReportHandshakeFailure(conn, err)
|
|
|
|
}
|
2023-08-31 20:07:32 +08:00
|
|
|
}
|
2024-11-07 13:44:00 +08:00
|
|
|
err = N.ReportPacketConnHandshakeSuccess(conn, outPacketConn)
|
2023-08-31 20:07:32 +08:00
|
|
|
if err != nil {
|
2024-11-07 13:44:00 +08:00
|
|
|
outPacketConn.Close()
|
2023-08-31 20:07:32 +08:00
|
|
|
return err
|
2022-07-07 23:36:32 +08:00
|
|
|
}
|
2023-04-02 12:01:19 +08:00
|
|
|
if destinationAddress.IsValid() {
|
2023-10-25 12:00:00 +08:00
|
|
|
if metadata.Destination.IsFqdn() {
|
2024-10-21 23:38:34 +08:00
|
|
|
if metadata.UDPDisableDomainUnmapping {
|
2024-11-07 13:44:00 +08:00
|
|
|
outPacketConn = bufio.NewUnidirectionalNATPacketConn(bufio.NewPacketConn(outPacketConn), M.SocksaddrFrom(destinationAddress, metadata.Destination.Port), metadata.Destination)
|
2023-10-30 12:00:00 +08:00
|
|
|
} else {
|
2024-11-07 13:44:00 +08:00
|
|
|
outPacketConn = bufio.NewNATPacketConn(bufio.NewPacketConn(outPacketConn), M.SocksaddrFrom(destinationAddress, metadata.Destination.Port), metadata.Destination)
|
2023-10-30 12:00:00 +08:00
|
|
|
}
|
2023-10-25 12:00:00 +08:00
|
|
|
}
|
2023-10-30 12:00:00 +08:00
|
|
|
if natConn, loaded := common.Cast[bufio.NATPacketConn](conn); loaded {
|
2023-04-02 12:01:19 +08:00
|
|
|
natConn.UpdateDestination(destinationAddress)
|
|
|
|
}
|
|
|
|
}
|
2022-08-12 12:13:57 +08:00
|
|
|
switch metadata.Protocol {
|
|
|
|
case C.ProtocolSTUN:
|
|
|
|
ctx, conn = canceler.NewPacketConn(ctx, conn, C.STUNTimeout)
|
|
|
|
case C.ProtocolQUIC:
|
|
|
|
ctx, conn = canceler.NewPacketConn(ctx, conn, C.QUICTimeout)
|
|
|
|
case C.ProtocolDNS:
|
|
|
|
ctx, conn = canceler.NewPacketConn(ctx, conn, C.DNSTimeout)
|
2022-07-30 08:39:13 +08:00
|
|
|
}
|
2024-11-07 13:44:00 +08:00
|
|
|
return bufio.CopyPacketConn(ctx, conn, bufio.NewPacketConn(outPacketConn))
|
2022-07-30 08:39:13 +08:00
|
|
|
}
|
|
|
|
|
2022-07-03 20:59:25 +08:00
|
|
|
func CopyEarlyConn(ctx context.Context, conn net.Conn, serverConn net.Conn) error {
|
2022-08-08 20:56:53 +08:00
|
|
|
if cachedReader, isCached := conn.(N.CachedReader); isCached {
|
2022-07-11 20:55:14 +08:00
|
|
|
payload := cachedReader.ReadCached()
|
|
|
|
if payload != nil && !payload.IsEmpty() {
|
|
|
|
_, err := serverConn.Write(payload.Bytes())
|
2023-09-24 14:39:56 +08:00
|
|
|
payload.Release()
|
2022-07-11 20:55:14 +08:00
|
|
|
if err != nil {
|
2023-11-13 13:53:52 +08:00
|
|
|
serverConn.Close()
|
2022-07-11 20:55:14 +08:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
return bufio.CopyConn(ctx, conn, serverConn)
|
|
|
|
}
|
|
|
|
}
|
2023-02-28 20:16:15 +08:00
|
|
|
if earlyConn, isEarlyConn := common.Cast[N.EarlyConn](serverConn); isEarlyConn && earlyConn.NeedHandshake() {
|
2023-07-03 21:45:32 +08:00
|
|
|
payload := buf.NewPacket()
|
2023-02-26 23:08:20 +08:00
|
|
|
err := conn.SetReadDeadline(time.Now().Add(C.ReadPayloadTimeout))
|
|
|
|
if err != os.ErrInvalid {
|
|
|
|
if err != nil {
|
2023-09-24 14:39:56 +08:00
|
|
|
payload.Release()
|
2023-11-13 13:53:52 +08:00
|
|
|
serverConn.Close()
|
2023-02-26 23:08:20 +08:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
_, err = payload.ReadOnceFrom(conn)
|
|
|
|
if err != nil && !E.IsTimeout(err) {
|
2023-09-24 14:39:56 +08:00
|
|
|
payload.Release()
|
2023-11-13 13:53:52 +08:00
|
|
|
serverConn.Close()
|
2023-02-26 23:08:20 +08:00
|
|
|
return E.Cause(err, "read payload")
|
|
|
|
}
|
|
|
|
err = conn.SetReadDeadline(time.Time{})
|
|
|
|
if err != nil {
|
|
|
|
payload.Release()
|
2023-11-13 13:53:52 +08:00
|
|
|
serverConn.Close()
|
2023-02-26 23:08:20 +08:00
|
|
|
return err
|
|
|
|
}
|
2022-10-20 10:57:57 +08:00
|
|
|
}
|
2023-02-26 23:08:20 +08:00
|
|
|
_, err = serverConn.Write(payload.Bytes())
|
2023-09-24 14:39:56 +08:00
|
|
|
payload.Release()
|
2022-10-20 10:57:57 +08:00
|
|
|
if err != nil {
|
2023-11-13 13:53:52 +08:00
|
|
|
serverConn.Close()
|
2023-08-31 20:07:32 +08:00
|
|
|
return N.ReportHandshakeFailure(conn, err)
|
2022-10-20 10:57:57 +08:00
|
|
|
}
|
2022-07-03 20:59:25 +08:00
|
|
|
}
|
|
|
|
return bufio.CopyConn(ctx, conn, serverConn)
|
|
|
|
}
|