fix reading in udp conn

7 years ago · 46dbbfff8c
parent 13f3c356ca
commit 46dbbfff8c
1 changed files with 7 additions and 4 deletions
--- a/app/proxyman/inbound/worker.go
+++ b/app/proxyman/inbound/worker.go
@ -131,8 +131,13 @@ func (c *udpConn) ReadMultiBuffer() (buf.MultiBuffer, error) {
 	select {
 	case in := <-c.input:
 		payload.Append(in)
-	case <-c.done.Wait():
-		return nil, io.EOF
+	default:
+		select {
+		case in := <-c.input:
+			payload.Append(in)
+		case <-c.done.Wait():
+			return nil, io.EOF
+		}
 	}

 L:
@ -140,8 +145,6 @@ L:
 		select {
 		case in := <-c.input:
 			payload.Append(in)
-		case <-c.done.Wait():
-			break L
 		default:
 			break L
 		}