go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix double copy in transport

Browse Source
This commit is contained in:
ginuerzh 2022-08-18 13:18:59 +08:00
parent bd60c2cf14
commit a1a2ea59d4
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/net/transport.go
View File

@ -9,7 +9,7 @@ import (
)
func Transport(rw1, rw2 io.ReadWriter) error {
errc := make(chan error, 1)
errc := make(chan error, 2)
go func() {
errc <- copyBuffer(rw1, rw2)
}()
@ -19,10 +19,16 @@ func Transport(rw1, rw2 io.ReadWriter) error {
}()
err := <-errc
if err != nil && err == io.EOF {
err = nil
err2 := <-errc
if err != nil && err != io.EOF {
return err
}
return err
if err2 != nil && err2 != io.EOF {
return err2
}
return nil
}
func copyBuffer(dst io.Writer, src io.Reader) error {