diff --git a/src/frp/utils/conn/conn.go b/src/frp/utils/conn/conn.go index a398110..a5a7a33 100644 --- a/src/frp/utils/conn/conn.go +++ b/src/frp/utils/conn/conn.go @@ -117,6 +117,11 @@ func ConnectServer(host string, port int64) (c *Conn, err error) { return c, nil } +func (c *Conn) SetTcpConn(tcpConn net.Conn) { + c.TcpConn = tcpConn + c.Reader = bufio.NewReader(c.TcpConn) +} + func (c *Conn) GetRemoteAddr() (addr string) { return c.TcpConn.RemoteAddr().String() } diff --git a/src/frp/utils/vhost/vhost.go b/src/frp/utils/vhost/vhost.go index ecf080d..ae67209 100644 --- a/src/frp/utils/vhost/vhost.go +++ b/src/frp/utils/vhost/vhost.go @@ -105,7 +105,7 @@ func (v *VhostMuxer) handle(c *conn.Conn) { if err = sConn.SetDeadline(time.Time{}); err != nil { return } - c.TcpConn = sConn + c.SetTcpConn(sConn) l.accept <- c }