|
|
|
@ -41,6 +41,7 @@ func (s *Server) policy() core.Policy {
|
|
|
|
|
return p
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Network implements proxy.Inbound.
|
|
|
|
|
func (s *Server) Network() net.NetworkList {
|
|
|
|
|
list := net.NetworkList{
|
|
|
|
|
Network: []net.Network{net.Network_TCP},
|
|
|
|
@ -51,6 +52,7 @@ func (s *Server) Network() net.NetworkList {
|
|
|
|
|
return list
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Process implements proxy.Inbound.
|
|
|
|
|
func (s *Server) Process(ctx context.Context, network net.Network, conn internet.Connection, dispatcher core.Dispatcher) error {
|
|
|
|
|
switch network {
|
|
|
|
|
case net.Network_TCP:
|
|
|
|
@ -63,7 +65,10 @@ func (s *Server) Process(ctx context.Context, network net.Network, conn internet
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (s *Server) processTCP(ctx context.Context, conn internet.Connection, dispatcher core.Dispatcher) error {
|
|
|
|
|
conn.SetReadDeadline(time.Now().Add(s.policy().Timeouts.Handshake))
|
|
|
|
|
if err := conn.SetReadDeadline(time.Now().Add(s.policy().Timeouts.Handshake)); err != nil {
|
|
|
|
|
newError("failed to set deadline").Base(err).WithContext(ctx).WriteToLog()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
reader := buf.NewBufferedReader(buf.NewReader(conn))
|
|
|
|
|
|
|
|
|
|
inboundDest, ok := proxy.InboundEntryPointFromContext(ctx)
|
|
|
|
@ -87,7 +92,10 @@ func (s *Server) processTCP(ctx context.Context, conn internet.Connection, dispa
|
|
|
|
|
}
|
|
|
|
|
return newError("failed to read request").Base(err)
|
|
|
|
|
}
|
|
|
|
|
conn.SetReadDeadline(time.Time{})
|
|
|
|
|
|
|
|
|
|
if err := conn.SetReadDeadline(time.Time{}); err != nil {
|
|
|
|
|
newError("failed to clear deadline").Base(err).WithContext(ctx).WriteToLog()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if request.Command == protocol.RequestCommandTCP {
|
|
|
|
|
dest := request.Destination()
|
|
|
|
@ -111,16 +119,16 @@ func (s *Server) processTCP(ctx context.Context, conn internet.Connection, dispa
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (*Server) handleUDP(c net.Conn) error {
|
|
|
|
|
func (*Server) handleUDP(c io.Reader) error {
|
|
|
|
|
// The TCP connection closes after this method returns. We need to wait until
|
|
|
|
|
// the client closes it.
|
|
|
|
|
_, err := io.Copy(buf.DiscardBytes, c)
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (v *Server) transport(ctx context.Context, reader io.Reader, writer io.Writer, dest net.Destination, dispatcher core.Dispatcher) error {
|
|
|
|
|
func (s *Server) transport(ctx context.Context, reader io.Reader, writer io.Writer, dest net.Destination, dispatcher core.Dispatcher) error {
|
|
|
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
|
|
|
timer := signal.CancelAfterInactivity(ctx, cancel, v.policy().Timeouts.ConnectionIdle)
|
|
|
|
|
timer := signal.CancelAfterInactivity(ctx, cancel, s.policy().Timeouts.ConnectionIdle)
|
|
|
|
|
|
|
|
|
|
ray, err := dispatcher.Dispatch(ctx, dest)
|
|
|
|
|
if err != nil {
|
|
|
|
@ -131,8 +139,8 @@ func (v *Server) transport(ctx context.Context, reader io.Reader, writer io.Writ
|
|
|
|
|
output := ray.InboundOutput()
|
|
|
|
|
|
|
|
|
|
requestDone := signal.ExecuteAsync(func() error {
|
|
|
|
|
defer timer.SetTimeout(v.policy().Timeouts.DownlinkOnly)
|
|
|
|
|
defer input.Close()
|
|
|
|
|
defer timer.SetTimeout(s.policy().Timeouts.DownlinkOnly)
|
|
|
|
|
defer common.Must(input.Close())
|
|
|
|
|
|
|
|
|
|
v2reader := buf.NewReader(reader)
|
|
|
|
|
if err := buf.Copy(v2reader, input, buf.UpdateActivity(timer)); err != nil {
|
|
|
|
@ -143,7 +151,7 @@ func (v *Server) transport(ctx context.Context, reader io.Reader, writer io.Writ
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
responseDone := signal.ExecuteAsync(func() error {
|
|
|
|
|
defer timer.SetTimeout(v.policy().Timeouts.UplinkOnly)
|
|
|
|
|
defer timer.SetTimeout(s.policy().Timeouts.UplinkOnly)
|
|
|
|
|
|
|
|
|
|
v2writer := buf.NewWriter(writer)
|
|
|
|
|
if err := buf.Copy(output, v2writer, buf.UpdateActivity(timer)); err != nil {
|
|
|
|
@ -162,7 +170,7 @@ func (v *Server) transport(ctx context.Context, reader io.Reader, writer io.Writ
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (v *Server) handleUDPPayload(ctx context.Context, conn internet.Connection, dispatcher core.Dispatcher) error {
|
|
|
|
|
func (s *Server) handleUDPPayload(ctx context.Context, conn internet.Connection, dispatcher core.Dispatcher) error {
|
|
|
|
|
udpServer := udp.NewDispatcher(dispatcher)
|
|
|
|
|
|
|
|
|
|
if source, ok := proxy.SourceFromContext(ctx); ok {
|
|
|
|
|