Fix index out of range error.

skycoin · Jun 1, 2019 · c3b94f4 · c3b94f4
1 parent 504c92a
commit c3b94f4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/dms/transport.go b/pkg/dms/transport.go
@@ -60,7 +60,10 @@ func (c *Transport) close() (closed bool) {
 
 func (c *Transport) awaitResponse(ctx context.Context) error {
 	select {
-	case f := <-c.readCh:
+	case f, ok := <-c.readCh:
+		if !ok {
+			return io.ErrClosedPipe
+		}
 		if f.Type() == AcceptType {
 			return nil
 		}