sing-box/outbound/default.go

117 lines
3.2 KiB
Go
Raw Normal View History

package outbound
import (
"context"
"net"
2022-10-20 02:57:57 +00:00
"os"
"runtime"
"time"
2022-07-08 15:03:57 +00:00
"github.com/sagernet/sing-box/adapter"
C "github.com/sagernet/sing-box/constant"
"github.com/sagernet/sing-box/log"
"github.com/sagernet/sing/common"
"github.com/sagernet/sing/common/buf"
"github.com/sagernet/sing/common/bufio"
2023-03-23 07:14:21 +00:00
"github.com/sagernet/sing/common/canceler"
E "github.com/sagernet/sing/common/exceptions"
2022-07-07 15:36:32 +00:00
N "github.com/sagernet/sing/common/network"
)
type myOutboundAdapter struct {
protocol string
2022-07-23 01:15:47 +00:00
network []string
router adapter.Router
2022-07-12 07:17:29 +00:00
logger log.ContextLogger
tag string
}
func (a *myOutboundAdapter) Type() string {
return a.protocol
}
func (a *myOutboundAdapter) Tag() string {
return a.tag
}
func (a *myOutboundAdapter) Network() []string {
return a.network
}
2022-07-07 15:36:32 +00:00
func NewConnection(ctx context.Context, this N.Dialer, conn net.Conn, metadata adapter.InboundContext) error {
ctx = adapter.WithContext(ctx, &metadata)
var outConn net.Conn
var err error
if len(metadata.DestinationAddresses) > 0 {
2022-07-29 16:29:22 +00:00
outConn, err = N.DialSerial(ctx, this, N.NetworkTCP, metadata.Destination, metadata.DestinationAddresses)
2022-07-07 15:36:32 +00:00
} else {
2022-07-29 16:29:22 +00:00
outConn, err = this.DialContext(ctx, N.NetworkTCP, metadata.Destination)
2022-07-07 15:36:32 +00:00
}
if err != nil {
2022-07-29 16:29:22 +00:00
return N.HandshakeFailure(conn, err)
2022-07-07 15:36:32 +00:00
}
return CopyEarlyConn(ctx, conn, outConn)
}
func NewPacketConnection(ctx context.Context, this N.Dialer, conn N.PacketConn, metadata adapter.InboundContext) error {
ctx = adapter.WithContext(ctx, &metadata)
var outConn net.PacketConn
var err error
if len(metadata.DestinationAddresses) > 0 {
2022-07-11 10:44:59 +00:00
outConn, err = N.ListenSerial(ctx, this, metadata.Destination, metadata.DestinationAddresses)
2022-07-07 15:36:32 +00:00
} else {
outConn, err = this.ListenPacket(ctx, metadata.Destination)
}
if err != nil {
2022-07-29 16:29:22 +00:00
return N.HandshakeFailure(conn, err)
2022-07-07 15:36:32 +00:00
}
2022-08-12 04:13:57 +00: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-11-24 04:00:01 +00:00
return bufio.CopyPacketConn(ctx, conn, bufio.NewPacketConn(outConn))
}
func CopyEarlyConn(ctx context.Context, conn net.Conn, serverConn net.Conn) error {
2022-08-08 12:56:53 +00:00
if cachedReader, isCached := conn.(N.CachedReader); isCached {
2022-07-11 12:55:14 +00:00
payload := cachedReader.ReadCached()
if payload != nil && !payload.IsEmpty() {
_, err := serverConn.Write(payload.Bytes())
if err != nil {
return err
}
return bufio.CopyConn(ctx, conn, serverConn)
}
}
2023-02-28 12:16:15 +00:00
if earlyConn, isEarlyConn := common.Cast[N.EarlyConn](serverConn); isEarlyConn && earlyConn.NeedHandshake() {
2023-02-26 15:08:20 +00:00
_payload := buf.StackNew()
payload := common.Dup(_payload)
err := conn.SetReadDeadline(time.Now().Add(C.ReadPayloadTimeout))
if err != os.ErrInvalid {
if err != nil {
return err
}
_, err = payload.ReadOnceFrom(conn)
if err != nil && !E.IsTimeout(err) {
return E.Cause(err, "read payload")
}
err = conn.SetReadDeadline(time.Time{})
if err != nil {
payload.Release()
return err
}
2022-10-20 02:57:57 +00:00
}
2023-02-26 15:08:20 +00:00
_, err = serverConn.Write(payload.Bytes())
2022-10-20 02:57:57 +00:00
if err != nil {
2023-02-26 15:08:20 +00:00
return N.HandshakeFailure(conn, err)
2022-10-20 02:57:57 +00:00
}
2023-02-26 15:08:20 +00:00
runtime.KeepAlive(_payload)
payload.Release()
}
return bufio.CopyConn(ctx, conn, serverConn)
}