mirrors/SpoofDPI
1
0
Fork 0
mirror of https://github.com/xvzc/SpoofDPI.git synced 2024-12-22 14:26:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

perf: remove loop

Browse Source
This commit is contained in:
Andrey Semenov 2024-08-09 01:52:59 +05:00 committed by LiquidTheDangerous
parent 10d1ee4860
commit dbee4964f8
1 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
proxy/io.go
View File

@ -28,24 +28,17 @@ func ReadBytes(conn *net.TCPConn, dest []byte) ([]byte, error) {
return dest[:n], err
}
func readBytesInternal(in io.Reader, dest []byte) (int, error) {
totalRead := 0
for {
numRead, readErr := in.Read(dest[totalRead:])
totalRead += numRead
if readErr != nil {
switch readErr.(type) {
func readBytesInternal(conn *net.TCPConn, dest []byte) (int, error) {
totalRead, err := conn.Read(dest)
if err != nil {
switch err.(type) {
case *net.OpError:
return totalRead, errors.New("timed out")
default:
return totalRead, readErr
return totalRead, err
}
}
if totalRead == 0 {
return 0, io.EOF
}
return totalRead, nil
}
}
func Serve(from *net.TCPConn, to *net.TCPConn, proto string, fd string, td string, timeout int, bufferSize int) {