OrgGo
/
frp
mirror of https://github.com/fatedier/frp.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

utils/vhost: update TcpConn with bufio.Reader

This commit is contained in:
fatedier 2016-07-20 16:33:42 +08:00
parent 4f49458af0
commit 926d0b74a9
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/frp/utils/conn/conn.go
View File

@ -117,6 +117,11 @@ func ConnectServer(host string, port int64) (c *Conn, err error) {
return c, nil 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) { func (c *Conn) GetRemoteAddr() (addr string) {
return c.TcpConn.RemoteAddr().String() return c.TcpConn.RemoteAddr().String()
} }

2
src/frp/utils/vhost/vhost.go
View File

@ -105,7 +105,7 @@ func (v *VhostMuxer) handle(c *conn.Conn) {
if err = sConn.SetDeadline(time.Time{}); err != nil { if err = sConn.SetDeadline(time.Time{}); err != nil {
return return
} }
c.TcpConn = sConn c.SetTcpConn(sConn)
l.accept <- c l.accept <- c
} }