mirror of
https://github.com/XTLS/Xray-core.git
synced 2026-05-14 18:09:05 +00:00
Fangliding
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package singbridge
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
N "github.com/sagernet/sing/common/network"
|
|
"github.com/xtls/xray-core/common/buf"
|
|
"github.com/xtls/xray-core/common/errors"
|
|
"github.com/xtls/xray-core/common/net"
|
|
"github.com/xtls/xray-core/features/routing"
|
|
"github.com/xtls/xray-core/transport"
|
|
)
|
|
|
|
var (
|
|
_ N.TCPConnectionHandler = (*Dispatcher)(nil)
|
|
_ N.UDPConnectionHandler = (*Dispatcher)(nil)
|
|
)
|
|
|
|
type Dispatcher struct {
|
|
upstream routing.Dispatcher
|
|
newErrorFunc func(values ...any) *errors.Error
|
|
}
|
|
|
|
func NewDispatcher(dispatcher routing.Dispatcher, newErrorFunc func(values ...any) *errors.Error) *Dispatcher {
|
|
return &Dispatcher{
|
|
upstream: dispatcher,
|
|
newErrorFunc: newErrorFunc,
|
|
}
|
|
}
|
|
|
|
func (d *Dispatcher) NewConnection(ctx context.Context, conn net.Conn, metadata M.Metadata) error {
|
|
dest, err := ToDestination(metadata.Destination, net.Network_TCP)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
xConn := NewConn(conn)
|
|
return d.upstream.DispatchLink(ctx, dest, &transport.Link{
|
|
Reader: xConn,
|
|
Writer: xConn,
|
|
})
|
|
}
|
|
|
|
func (d *Dispatcher) NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata M.Metadata) error {
|
|
dest, err := ToDestination(metadata.Destination, net.Network_UDP)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return d.upstream.DispatchLink(ctx, dest, &transport.Link{
|
|
Reader: buf.NewPacketReader(conn.(io.Reader)),
|
|
Writer: buf.NewWriter(conn.(io.Writer)),
|
|
})
|
|
}
|
|
|
|
func (d *Dispatcher) NewError(ctx context.Context, err error) {
|
|
errors.LogInfo(ctx, err.Error())
|
|
}
|