fix reading in udp conn

whao · May 27, 2018 · 46dbbff · 46dbbff
1 parent 13f3c35
commit 46dbbff
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/app/proxyman/inbound/worker.go b/app/proxyman/inbound/worker.go
@@ -131,17 +131,20 @@ 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:
 	for {
 		select {
 		case in := <-c.input:
 			payload.Append(in)
-		case <-c.done.Wait():
-			break L
 		default:
 			break L
 		}