root/gost_software
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #870

Browse Source
This commit is contained in:
ginuerzh 2022-08-31 17:52:52 +08:00
parent 81854a61e0
commit 97dda762e0
1 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
server.go
View File

@ -103,7 +103,7 @@ type Listener interface {
} }
func transport(rw1, rw2 io.ReadWriter) error { func transport(rw1, rw2 io.ReadWriter) error {
errc := make(chan error, 2) errc := make(chan error, 1)
go func() { go func() {
errc <- copyBuffer(rw1, rw2) errc <- copyBuffer(rw1, rw2)
}() }()
@ -112,18 +112,11 @@ func transport(rw1, rw2 io.ReadWriter) error {
errc <- copyBuffer(rw2, rw1) errc <- copyBuffer(rw2, rw1)
}() }()
err := <-errc if err := <-errc; err != nil && err != io.EOF {
err2 := <-errc
if err != nil && err == io.EOF {
err = nil
}
if err != nil {
return err return err
} }
if err2 != nil && err2 == io.EOF {
err2 = nil return nil
}
return err2
} }
func copyBuffer(dst io.Writer, src io.Reader) error { func copyBuffer(dst io.Writer, src io.Reader) error {