fix xnet.Pipe
This commit is contained in:
@@ -62,14 +62,14 @@ func (c *serverConn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *serverConn) CloseWrite() error {
|
func (c *serverConn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
type packetConn struct {
|
type packetConn struct {
|
||||||
|
|||||||
@@ -44,7 +44,7 @@ func (c *relayConnector) bindTCP(ctx context.Context, conn net.Conn, network, ad
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
log.Debugf("bind on %s/%s OK", laddr, laddr.Network())
|
log.Infof("bind on %s/%s OK", laddr, laddr.Network())
|
||||||
|
|
||||||
session, err := mux.ServerSession(conn, c.md.muxCfg)
|
session, err := mux.ServerSession(conn, c.md.muxCfg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -63,7 +63,7 @@ func (c *relayConnector) bindUDP(ctx context.Context, conn net.Conn, network, ad
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
log.Debugf("bind on %s/%s OK", laddr, laddr.Network())
|
log.Infof("bind on %s/%s OK", laddr, laddr.Network())
|
||||||
|
|
||||||
ln := udp.NewListener(
|
ln := udp.NewListener(
|
||||||
relay_util.UDPTunClientPacketConn(conn),
|
relay_util.UDPTunClientPacketConn(conn),
|
||||||
|
|||||||
@@ -462,13 +462,13 @@ func (h *httpHandler) handleProxy(ctx context.Context, conn net.Conn, req *http.
|
|||||||
log.Trace(string(dump))
|
log.Trace(string(dump))
|
||||||
}
|
}
|
||||||
|
|
||||||
if close, err := h.proxyRoundTrip(ctx, xio.NewReadWriter(br, conn), req, ro, &pStats, log); err != nil || close {
|
if close, err := h.proxyRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), req, ro, &pStats, log); err != nil || close {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *httpHandler) proxyRoundTrip(ctx context.Context, rw io.ReadWriter, req *http.Request, ro *xrecorder.HandlerRecorderObject, pStats stats.Stats, log logger.Logger) (close bool, err error) {
|
func (h *httpHandler) proxyRoundTrip(ctx context.Context, rw io.ReadWriteCloser, req *http.Request, ro *xrecorder.HandlerRecorderObject, pStats stats.Stats, log logger.Logger) (close bool, err error) {
|
||||||
close = true
|
close = true
|
||||||
|
|
||||||
ro2 := &xrecorder.HandlerRecorderObject{}
|
ro2 := &xrecorder.HandlerRecorderObject{}
|
||||||
@@ -682,7 +682,7 @@ func upgradeType(h http.Header) string {
|
|||||||
return h.Get("Upgrade")
|
return h.Get("Upgrade")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *httpHandler) handleUpgradeResponse(ctx context.Context, rw io.ReadWriter, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
func (h *httpHandler) handleUpgradeResponse(ctx context.Context, rw io.ReadWriteCloser, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
||||||
reqUpType := upgradeType(req.Header)
|
reqUpType := upgradeType(req.Header)
|
||||||
resUpType := upgradeType(res.Header)
|
resUpType := upgradeType(res.Header)
|
||||||
if !strings.EqualFold(reqUpType, resUpType) {
|
if !strings.EqualFold(reqUpType, resUpType) {
|
||||||
|
|||||||
@@ -316,7 +316,7 @@ func (h *http2Handler) roundTrip(ctx context.Context, w http.ResponseWriter, req
|
|||||||
start := time.Now()
|
start := time.Now()
|
||||||
log.Infof("%s <-> %s", req.RemoteAddr, host)
|
log.Infof("%s <-> %s", req.RemoteAddr, host)
|
||||||
// xnet.Transport(rw, cc)
|
// xnet.Transport(rw, cc)
|
||||||
xnet.Pipe(ctx, rw, cc)
|
xnet.Pipe(ctx, xio.NewReadWriteCloser(rw, rw, req.Body), cc)
|
||||||
log.WithFields(map[string]any{
|
log.WithFields(map[string]any{
|
||||||
"duration": time.Since(start),
|
"duration": time.Since(start),
|
||||||
}).Infof("%s >-< %s", req.RemoteAddr, host)
|
}).Infof("%s >-< %s", req.RemoteAddr, host)
|
||||||
|
|||||||
@@ -143,7 +143,7 @@ func (h *relayHandler) bindTCP(ctx context.Context, conn net.Conn, network, addr
|
|||||||
)
|
)
|
||||||
|
|
||||||
log = log.WithFields(map[string]any{})
|
log = log.WithFields(map[string]any{})
|
||||||
log.Debugf("bind on %s/%s OK", ln.Addr(), ln.Addr().Network())
|
log.Infof("bind on %s/%s OK", ln.Addr(), ln.Addr().Network())
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
defer srv.Close()
|
defer srv.Close()
|
||||||
@@ -201,7 +201,7 @@ func (h *relayHandler) bindUDP(ctx context.Context, conn net.Conn, network, addr
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debugf("bind on %s OK", pc.LocalAddr())
|
log.Infof("bind on %s OK", pc.LocalAddr())
|
||||||
|
|
||||||
r := udp.NewRelay(relay_util.UDPTunServerConn(conn), pc).
|
r := udp.NewRelay(relay_util.UDPTunServerConn(conn), pc).
|
||||||
WithBypass(h.options.Bypass).
|
WithBypass(h.options.Bypass).
|
||||||
|
|||||||
@@ -224,7 +224,7 @@ func (ep *entrypoint) handleHTTP(ctx context.Context, conn net.Conn, ro *xrecord
|
|||||||
|
|
||||||
ro.Time = time.Time{}
|
ro.Time = time.Time{}
|
||||||
|
|
||||||
if err := ep.httpRoundTrip(ctx, xio.NewReadWriter(br, conn), req, ro, &pStats, log); err != nil {
|
if err := ep.httpRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), req, ro, &pStats, log); err != nil {
|
||||||
log.Error(err)
|
log.Error(err)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -245,13 +245,13 @@ func (ep *entrypoint) handleHTTP(ctx context.Context, conn net.Conn, ro *xrecord
|
|||||||
log.Trace(string(dump))
|
log.Trace(string(dump))
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ep.httpRoundTrip(ctx, xio.NewReadWriter(br, conn), req, ro, &pStats, log); err != nil {
|
if err := ep.httpRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), req, ro, &pStats, log); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ep *entrypoint) httpRoundTrip(ctx context.Context, rw io.ReadWriter, req *http.Request, ro *xrecorder.HandlerRecorderObject, pStats stats.Stats, log logger.Logger) (err error) {
|
func (ep *entrypoint) httpRoundTrip(ctx context.Context, rw io.ReadWriteCloser, req *http.Request, ro *xrecorder.HandlerRecorderObject, pStats stats.Stats, log logger.Logger) (err error) {
|
||||||
ro2 := &xrecorder.HandlerRecorderObject{}
|
ro2 := &xrecorder.HandlerRecorderObject{}
|
||||||
*ro2 = *ro
|
*ro2 = *ro
|
||||||
ro = ro2
|
ro = ro2
|
||||||
@@ -413,7 +413,7 @@ func upgradeType(h http.Header) string {
|
|||||||
return h.Get("Upgrade")
|
return h.Get("Upgrade")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ep *entrypoint) handleUpgradeResponse(ctx context.Context, rw io.ReadWriter, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
func (ep *entrypoint) handleUpgradeResponse(ctx context.Context, rw io.ReadWriteCloser, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
||||||
reqUpType := upgradeType(req.Header)
|
reqUpType := upgradeType(req.Header)
|
||||||
resUpType := upgradeType(res.Header)
|
resUpType := upgradeType(res.Header)
|
||||||
if !strings.EqualFold(reqUpType, resUpType) {
|
if !strings.EqualFold(reqUpType, resUpType) {
|
||||||
|
|||||||
@@ -218,7 +218,7 @@ func (h *unixHandler) forwardUnix(ctx context.Context, conn net.Conn, target *ch
|
|||||||
}
|
}
|
||||||
defer cc.Close()
|
defer cc.Close()
|
||||||
|
|
||||||
var rw io.ReadWriter = conn
|
var rw io.ReadWriteCloser = conn
|
||||||
if h.md.sniffing {
|
if h.md.sniffing {
|
||||||
if h.md.sniffingTimeout > 0 {
|
if h.md.sniffingTimeout > 0 {
|
||||||
conn.SetReadDeadline(time.Now().Add(h.md.sniffingTimeout))
|
conn.SetReadDeadline(time.Now().Add(h.md.sniffingTimeout))
|
||||||
@@ -232,7 +232,7 @@ func (h *unixHandler) forwardUnix(ctx context.Context, conn net.Conn, target *ch
|
|||||||
conn.SetReadDeadline(time.Time{})
|
conn.SetReadDeadline(time.Time{})
|
||||||
}
|
}
|
||||||
|
|
||||||
rw = xio.NewReadWriter(br, conn)
|
rw = xio.NewReadWriteCloser(br, conn, conn)
|
||||||
switch proto {
|
switch proto {
|
||||||
case sniffing.ProtoHTTP:
|
case sniffing.ProtoHTTP:
|
||||||
ro2 := &xrecorder.HandlerRecorderObject{}
|
ro2 := &xrecorder.HandlerRecorderObject{}
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ const (
|
|||||||
defaultBodySize = 1024 * 1024 // 1MB
|
defaultBodySize = 1024 * 1024 // 1MB
|
||||||
)
|
)
|
||||||
|
|
||||||
func (h *unixHandler) handleHTTP(ctx context.Context, rw, cc io.ReadWriter, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
func (h *unixHandler) handleHTTP(ctx context.Context, rw, cc io.ReadWriteCloser, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
||||||
br := bufio.NewReader(rw)
|
br := bufio.NewReader(rw)
|
||||||
req, err := http.ReadRequest(br)
|
req, err := http.ReadRequest(br)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -81,7 +81,7 @@ func (h *unixHandler) handleHTTP(ctx context.Context, rw, cc io.ReadWriter, ro *
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *unixHandler) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriter, req *http.Request, ro *xrecorder.HandlerRecorderObject, log logger.Logger) (close bool, err error) {
|
func (h *unixHandler) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriteCloser, req *http.Request, ro *xrecorder.HandlerRecorderObject, log logger.Logger) (close bool, err error) {
|
||||||
close = true
|
close = true
|
||||||
|
|
||||||
if req == nil {
|
if req == nil {
|
||||||
@@ -206,7 +206,7 @@ func (h *unixHandler) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriter, r
|
|||||||
return resp.Close, nil
|
return resp.Close, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *unixHandler) handleTLS(ctx context.Context, rw, cc io.ReadWriter, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
func (h *unixHandler) handleTLS(ctx context.Context, rw, cc io.ReadWriteCloser, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
||||||
buf := new(bytes.Buffer)
|
buf := new(bytes.Buffer)
|
||||||
|
|
||||||
clientHello, err := dissector.ParseClientHello(io.TeeReader(rw, buf))
|
clientHello, err := dissector.ParseClientHello(io.TeeReader(rw, buf))
|
||||||
|
|||||||
+21
-2
@@ -1,10 +1,15 @@
|
|||||||
package io
|
package io
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrUnsupported = errors.New("unsupported")
|
||||||
|
)
|
||||||
|
|
||||||
type CloseRead interface {
|
type CloseRead interface {
|
||||||
CloseRead() error
|
CloseRead() error
|
||||||
}
|
}
|
||||||
@@ -29,14 +34,14 @@ func (rw *readWriter) CloseRead() error {
|
|||||||
if sc, ok := rw.Writer.(CloseRead); ok {
|
if sc, ok := rw.Writer.(CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rw *readWriter) CloseWrite() error {
|
func (rw *readWriter) CloseWrite() error {
|
||||||
if sc, ok := rw.Writer.(CloseWrite); ok {
|
if sc, ok := rw.Writer.(CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
type readWriteCloser struct {
|
type readWriteCloser struct {
|
||||||
@@ -53,6 +58,20 @@ func NewReadWriteCloser(r io.Reader, w io.Writer, c io.Closer) io.ReadWriteClose
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (rwc *readWriteCloser) CloseRead() error {
|
||||||
|
if sc, ok := rwc.Writer.(CloseRead); ok {
|
||||||
|
return sc.CloseRead()
|
||||||
|
}
|
||||||
|
return ErrUnsupported
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rwc *readWriteCloser) CloseWrite() error {
|
||||||
|
if sc, ok := rwc.Writer.(CloseWrite); ok {
|
||||||
|
return sc.CloseWrite()
|
||||||
|
}
|
||||||
|
return ErrUnsupported
|
||||||
|
}
|
||||||
|
|
||||||
type setReadDeadline interface {
|
type setReadDeadline interface {
|
||||||
SetReadDeadline(t time.Time) error
|
SetReadDeadline(t time.Time) error
|
||||||
}
|
}
|
||||||
|
|||||||
+2
-2
@@ -129,12 +129,12 @@ func (c *readWriteConn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *readWriteConn) CloseWrite() error {
|
func (c *readWriteConn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ const (
|
|||||||
tcpWaitTimeout = 10 * time.Second
|
tcpWaitTimeout = 10 * time.Second
|
||||||
)
|
)
|
||||||
|
|
||||||
func Pipe(ctx context.Context, rw1, rw2 io.ReadWriter) error {
|
func Pipe(ctx context.Context, rw1, rw2 io.ReadWriteCloser) error {
|
||||||
wg := sync.WaitGroup{}
|
wg := sync.WaitGroup{}
|
||||||
wg.Add(2)
|
wg.Add(2)
|
||||||
|
|
||||||
@@ -55,7 +55,7 @@ func Pipe(ctx context.Context, rw1, rw2 io.ReadWriter) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func pipeBuffer(dst io.ReadWriter, src io.ReadWriter, bufferSize int) error {
|
func pipeBuffer(dst io.ReadWriteCloser, src io.ReadWriteCloser, bufferSize int) error {
|
||||||
buf := bufpool.Get(bufferSize)
|
buf := bufpool.Get(bufferSize)
|
||||||
defer bufpool.Put(buf)
|
defer bufpool.Put(buf)
|
||||||
|
|
||||||
@@ -65,8 +65,13 @@ func pipeBuffer(dst io.ReadWriter, src io.ReadWriter, bufferSize int) error {
|
|||||||
if cr, ok := src.(xio.CloseRead); ok {
|
if cr, ok := src.(xio.CloseRead); ok {
|
||||||
cr.CloseRead()
|
cr.CloseRead()
|
||||||
}
|
}
|
||||||
|
|
||||||
if cw, ok := dst.(xio.CloseWrite); ok {
|
if cw, ok := dst.(xio.CloseWrite); ok {
|
||||||
cw.CloseWrite()
|
if e := cw.CloseWrite(); e == xio.ErrUnsupported {
|
||||||
|
dst.Close()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
dst.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set TCP half-close timeout.
|
// Set TCP half-close timeout.
|
||||||
|
|||||||
@@ -3,9 +3,38 @@ package proxyproto
|
|||||||
import (
|
import (
|
||||||
"net"
|
"net"
|
||||||
|
|
||||||
|
xio "github.com/go-gost/x/internal/io"
|
||||||
proxyproto "github.com/pires/go-proxyproto"
|
proxyproto "github.com/pires/go-proxyproto"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type serverConn struct {
|
||||||
|
net.Conn
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *serverConn) CloseRead() error {
|
||||||
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
|
return sc.CloseRead()
|
||||||
|
}
|
||||||
|
if conn, ok := c.Conn.(*proxyproto.Conn); ok {
|
||||||
|
if tcpConn, ok := conn.TCPConn(); ok {
|
||||||
|
return tcpConn.CloseRead()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return xio.ErrUnsupported
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *serverConn) CloseWrite() error {
|
||||||
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
|
return sc.CloseWrite()
|
||||||
|
}
|
||||||
|
if conn, ok := c.Conn.(*proxyproto.Conn); ok {
|
||||||
|
if tcpConn, ok := conn.TCPConn(); ok {
|
||||||
|
return tcpConn.CloseWrite()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return xio.ErrUnsupported
|
||||||
|
}
|
||||||
|
|
||||||
func WrapClientConn(ppv int, src, dst net.Addr, c net.Conn) net.Conn {
|
func WrapClientConn(ppv int, src, dst net.Addr, c net.Conn) net.Conn {
|
||||||
if ppv <= 0 {
|
if ppv <= 0 {
|
||||||
return c
|
return c
|
||||||
|
|||||||
@@ -7,13 +7,27 @@ import (
|
|||||||
proxyproto "github.com/pires/go-proxyproto"
|
proxyproto "github.com/pires/go-proxyproto"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type listener struct {
|
||||||
|
net.Listener
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ln *listener) Accept() (net.Conn, error) {
|
||||||
|
conn, err := ln.Listener.Accept()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &serverConn{Conn: conn}, nil
|
||||||
|
}
|
||||||
|
|
||||||
func WrapListener(ppv int, ln net.Listener, readHeaderTimeout time.Duration) net.Listener {
|
func WrapListener(ppv int, ln net.Listener, readHeaderTimeout time.Duration) net.Listener {
|
||||||
if ppv <= 0 {
|
if ppv <= 0 {
|
||||||
return ln
|
return ln
|
||||||
}
|
}
|
||||||
|
|
||||||
return &proxyproto.Listener{
|
return &listener{
|
||||||
Listener: ln,
|
Listener: &proxyproto.Listener{
|
||||||
ReadHeaderTimeout: readHeaderTimeout,
|
Listener: ln,
|
||||||
|
ReadHeaderTimeout: readHeaderTimeout,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -191,7 +191,7 @@ func (h *Sniffer) HandleHTTP(ctx context.Context, conn net.Conn, opts ...HandleO
|
|||||||
ro.Dst = cc.RemoteAddr().String()
|
ro.Dst = cc.RemoteAddr().String()
|
||||||
ro.Time = time.Time{}
|
ro.Time = time.Time{}
|
||||||
|
|
||||||
shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriter(br, conn), cc, node, req, &pStats, &ho)
|
shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), cc, node, req, &pStats, &ho)
|
||||||
if err != nil || shouldClose {
|
if err != nil || shouldClose {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -212,7 +212,7 @@ func (h *Sniffer) HandleHTTP(ctx context.Context, conn net.Conn, opts ...HandleO
|
|||||||
log.Trace(string(dump))
|
log.Trace(string(dump))
|
||||||
}
|
}
|
||||||
|
|
||||||
if shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriter(br, conn), cc, node, req, &pStats, &ho); err != nil || shouldClose {
|
if shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), cc, node, req, &pStats, &ho); err != nil || shouldClose {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -365,7 +365,7 @@ func (h *Sniffer) serveH2(ctx context.Context, conn net.Conn, ho *HandleOptions)
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Sniffer) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriter, node *chain.Node, req *http.Request, pStats stats.Stats, ho *HandleOptions) (close bool, err error) {
|
func (h *Sniffer) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriteCloser, node *chain.Node, req *http.Request, pStats stats.Stats, ho *HandleOptions) (close bool, err error) {
|
||||||
close = true
|
close = true
|
||||||
|
|
||||||
log := ho.Log
|
log := ho.Log
|
||||||
@@ -586,7 +586,7 @@ func upgradeType(h http.Header) string {
|
|||||||
return h.Get("Upgrade")
|
return h.Get("Upgrade")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Sniffer) handleUpgradeResponse(ctx context.Context, rw io.ReadWriter, cc io.ReadWriter, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
func (h *Sniffer) handleUpgradeResponse(ctx context.Context, rw, cc io.ReadWriteCloser, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
||||||
reqUpType := upgradeType(req.Header)
|
reqUpType := upgradeType(req.Header)
|
||||||
resUpType := upgradeType(res.Header)
|
resUpType := upgradeType(res.Header)
|
||||||
if !strings.EqualFold(reqUpType, resUpType) {
|
if !strings.EqualFold(reqUpType, resUpType) {
|
||||||
|
|||||||
@@ -198,7 +198,7 @@ func (h *Sniffer) HandleHTTP(ctx context.Context, conn net.Conn, opts ...HandleO
|
|||||||
ro.Dst = cc.RemoteAddr().String()
|
ro.Dst = cc.RemoteAddr().String()
|
||||||
ro.Time = time.Time{}
|
ro.Time = time.Time{}
|
||||||
|
|
||||||
shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriter(br, conn), cc, req, ro, &pStats, log)
|
shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), cc, req, ro, &pStats, log)
|
||||||
if err != nil || shouldClose {
|
if err != nil || shouldClose {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -219,7 +219,7 @@ func (h *Sniffer) HandleHTTP(ctx context.Context, conn net.Conn, opts ...HandleO
|
|||||||
log.Trace(string(dump))
|
log.Trace(string(dump))
|
||||||
}
|
}
|
||||||
|
|
||||||
if shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriter(br, conn), cc, req, ro, &pStats, log); err != nil || shouldClose {
|
if shouldClose, err := h.httpRoundTrip(ctx, xio.NewReadWriteCloser(br, conn, conn), cc, req, ro, &pStats, log); err != nil || shouldClose {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -277,7 +277,7 @@ func (h *Sniffer) serveH2(ctx context.Context, conn net.Conn, ho *HandleOptions)
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Sniffer) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriter, req *http.Request, ro *xrecorder.HandlerRecorderObject, pStats stats.Stats, log logger.Logger) (close bool, err error) {
|
func (h *Sniffer) httpRoundTrip(ctx context.Context, rw, cc io.ReadWriteCloser, req *http.Request, ro *xrecorder.HandlerRecorderObject, pStats stats.Stats, log logger.Logger) (close bool, err error) {
|
||||||
close = true
|
close = true
|
||||||
|
|
||||||
ro2 := &xrecorder.HandlerRecorderObject{}
|
ro2 := &xrecorder.HandlerRecorderObject{}
|
||||||
@@ -433,7 +433,7 @@ func upgradeType(h http.Header) string {
|
|||||||
return h.Get("Upgrade")
|
return h.Get("Upgrade")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Sniffer) handleUpgradeResponse(ctx context.Context, rw io.ReadWriter, cc io.ReadWriter, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
func (h *Sniffer) handleUpgradeResponse(ctx context.Context, rw, cc io.ReadWriteCloser, req *http.Request, res *http.Response, ro *xrecorder.HandlerRecorderObject, log logger.Logger) error {
|
||||||
reqUpType := upgradeType(req.Header)
|
reqUpType := upgradeType(req.Header)
|
||||||
resUpType := upgradeType(res.Header)
|
resUpType := upgradeType(res.Header)
|
||||||
if !strings.EqualFold(reqUpType, resUpType) {
|
if !strings.EqualFold(reqUpType, resUpType) {
|
||||||
|
|||||||
@@ -46,12 +46,12 @@ func (c *shadowConn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *shadowConn) CloseWrite() error {
|
func (c *shadowConn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -55,12 +55,12 @@ func (c *serverConn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *serverConn) CloseWrite() error {
|
func (c *serverConn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,9 +11,9 @@ import (
|
|||||||
"github.com/go-gost/core/limiter"
|
"github.com/go-gost/core/limiter"
|
||||||
"github.com/go-gost/core/limiter/traffic"
|
"github.com/go-gost/core/limiter/traffic"
|
||||||
"github.com/go-gost/core/metadata"
|
"github.com/go-gost/core/metadata"
|
||||||
|
xio "github.com/go-gost/x/internal/io"
|
||||||
xnet "github.com/go-gost/x/internal/net"
|
xnet "github.com/go-gost/x/internal/net"
|
||||||
"github.com/go-gost/x/internal/net/udp"
|
"github.com/go-gost/x/internal/net/udp"
|
||||||
xio "github.com/go-gost/x/internal/io"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -118,14 +118,14 @@ func (c *limitConn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *limitConn) CloseWrite() error {
|
func (c *limitConn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
type packetConn struct {
|
type packetConn struct {
|
||||||
|
|||||||
@@ -92,14 +92,14 @@ func (c *serverConn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *serverConn) CloseWrite() error {
|
func (c *serverConn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
type packetConn struct {
|
type packetConn struct {
|
||||||
|
|||||||
@@ -94,14 +94,14 @@ func (c *conn) CloseRead() error {
|
|||||||
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
if sc, ok := c.Conn.(xio.CloseRead); ok {
|
||||||
return sc.CloseRead()
|
return sc.CloseRead()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *conn) CloseWrite() error {
|
func (c *conn) CloseWrite() error {
|
||||||
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
if sc, ok := c.Conn.(xio.CloseWrite); ok {
|
||||||
return sc.CloseWrite()
|
return sc.CloseWrite()
|
||||||
}
|
}
|
||||||
return nil
|
return xio.ErrUnsupported
|
||||||
}
|
}
|
||||||
|
|
||||||
type packetConn struct {
|
type packetConn struct {
|
||||||
|
|||||||
+1
-1
@@ -78,7 +78,7 @@ func (r *httpRecorder) Record(ctx context.Context, b []byte, opts ...recorder.Re
|
|||||||
}
|
}
|
||||||
|
|
||||||
if r.header != nil {
|
if r.header != nil {
|
||||||
req.Header = r.header
|
req.Header = r.header.Clone()
|
||||||
}
|
}
|
||||||
|
|
||||||
if req.Header.Get("Content-Type") == "" {
|
if req.Header.Get("Content-Type") == "" {
|
||||||
|
|||||||
+6
-2
@@ -160,6 +160,8 @@ func (s *defaultService) Serve() error {
|
|||||||
defer v.Dec()
|
defer v.Dec()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log := s.options.logger
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
defer wg.Wait()
|
defer wg.Wait()
|
||||||
|
|
||||||
@@ -184,16 +186,18 @@ func (s *defaultService) Serve() error {
|
|||||||
|
|
||||||
s.setState(StateFailed)
|
s.setState(StateFailed)
|
||||||
|
|
||||||
s.options.logger.Warnf("accept: %v, retrying in %v", e, tempDelay)
|
log.Warnf("accept: %v, retrying in %v", e, tempDelay)
|
||||||
time.Sleep(tempDelay)
|
time.Sleep(tempDelay)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
s.setState(StateClosed)
|
s.setState(StateClosed)
|
||||||
|
|
||||||
if !errors.Is(e, net.ErrClosed) {
|
if !errors.Is(e, net.ErrClosed) {
|
||||||
s.options.logger.Errorf("accept: %v", e)
|
log.Errorf("accept: %v", e)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log.Debugf("service %s exited!", s.name)
|
||||||
|
|
||||||
return e
|
return e
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user