Update: wait both direction and close

- origin code channel no closed
- io.Copy will never return EOF error

server.go

@@ -3,6 +3,7 @@ package gost
 import (
 	"io"
 	"net"
+	"sync"
 	"time"
 
 	"github.com/go-log/log"
@@ -99,20 +100,24 @@ func (ln tcpKeepAliveListener) Accept() (c net.Conn, err error) {
 }
 
 func transport(rw1, rw2 io.ReadWriter) error {
-	errc := make(chan error, 1)
+	var err1 error
-	go func() {
+	var err2 error
-		_, err := io.Copy(rw1, rw2)
+	wg := sync.WaitGroup{}
-		errc <- err
+	wg.Add(2)
-	}()
 
 	go func() {
-		_, err := io.Copy(rw2, rw1)
+		_, err1 = io.Copy(rw1, rw2)
-		errc <- err
+		wg.Done()
+	}()
+	go func() {
+		_, err2 = io.Copy(rw2, rw1)
+		wg.Done()
 	}()
 
-	err := <-errc
+	// wait for both direction complete
-	if err != nil && err == io.EOF {
+	wg.Wait()
-		err = nil
+	if err1 != nil {
+		return err1
 	}
-	return err
+	return err2
 }