github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-07-05 15:00:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

read full http proxy request, fix UnexpectedEOF error

Browse Source
This commit is contained in:
ffdfgdfg 2020-02-11 00:38:24 +08:00
parent 196c08cb32
commit e2d249a576
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lib/conn/conn.go
View File

@ -35,11 +35,30 @@ func NewConn(conn net.Conn) *Conn {
return &Conn{Conn: conn} return &Conn{Conn: conn}
} }
func (s *Conn) readRequest(buf []byte) (n int, err error) {
var rd int
for {
rd, err = s.Read(buf[n:])
if err != nil {
return
}
n += rd
if string(buf[n-4:n]) == "\r\n\r\n" {
return
}
// buf is full, can't contain the request
if n == cap(buf) {
err = io.ErrUnexpectedEOF
return
}
}
}
//get host 、connection type、method...from connection //get host 、connection type、method...from connection
func (s *Conn) GetHost() (method, address string, rb []byte, err error, r *http.Request) { func (s *Conn) GetHost() (method, address string, rb []byte, err error, r *http.Request) {
var b [32 * 1024]byte var b [32 * 1024]byte
var n int var n int
if n, err = s.Read(b[:]); err != nil { if n, err = s.readRequest(b[:]); err != nil {
return return
} }
rb = b[:n] rb = b[:n]