Merge pull request #50 from libp2p/fix/wait-on-handshake

fix: wait for handshake to complete before returning
libp2p · Jun 15, 2019 · 94b9134 · 94b9134
2 parents 2078160 + d81b453
commit 94b9134
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/protocol.go b/protocol.go
@@ -125,9 +125,12 @@ func (s *secureSession) runHandshake(ctx context.Context) error {
 	var err error
 	select {
 	case <-ctx.Done():
+		err = ctx.Err()
+
 		// State unknown. We *have* to close this.
 		s.insecure.Close()
-		err = ctx.Err()
+		// Wait for the handshake to return.
+		<-result
 	case err = <-result:
 	}
 	return err