mirrors/SpoofDPI
1
0
Fork 0
mirror of https://github.com/xvzc/SpoofDPI.git synced 2025-01-03 04:50:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
xvzc 2022-05-12 21:21:54 +09:00
parent 8ef36894b7
commit 8f03e79849
1 changed files with 12 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
net/conn.go
View File

@ -1,7 +1,7 @@
package net package net
import ( import (
"io" "errors"
"net" "net"
"time" "time"
@ -70,45 +70,27 @@ func (conn *Conn) WriteChunks(c [][]byte) (n int, err error) {
} }
func (conn *Conn) ReadBytes() ([]byte, error) { func (conn *Conn) ReadBytes() ([]byte, error) {
// ret := make([]byte, 0) ret := make([]byte, 0)
// buf := make([]byte, BUF_SIZE) buf := make([]byte, BUF_SIZE)
// for {
// n, err := conn.Read(buf)
// if err != nil {
// switch err.(type) {
// case *net.OpError:
// return nil, errors.New("timed out")
// default:
// return nil, err
// }
// }
// ret = append(ret, buf[:n]...)
// if n == 0 {
// return nil, io.EOF
// }
// if n < BUF_SIZE {
// break
// }
// }
buf := make([]byte, 0, 4096) // big buffer
tmp := make([]byte, 256) // using small tmo buffer for demonstrating
for { for {
n, err := conn.Read(tmp) n, err := conn.Read(buf)
if err != nil { if err != nil {
if err != io.EOF { switch err.(type) {
case *net.OpError:
return nil, errors.New("timed out")
default:
return nil, err return nil, err
} }
}
ret = append(ret, buf[:n]...)
if n < BUF_SIZE {
break break
} }
buf = append(buf, tmp[:n]...)
} }
return buf, nil return ret, nil
} }
func (lConn *Conn) HandleHttp(p *packet.HttpPacket) { func (lConn *Conn) HandleHttp(p *packet.HttpPacket) {