basichost: handle the SetProtocol error in NewStream (#2555)

libp2p · Sep 4, 2023 · dce9424 · dce9424
1 parent 9f14eb7
commit dce9424
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/p2p/host/basic/basic_host.go b/p2p/host/basic/basic_host.go
@@ -666,7 +666,9 @@ func (h *BasicHost) NewStream(ctx context.Context, p peer.ID, pids ...protocol.I
 	}
 
 	if pref != "" {
-		s.SetProtocol(pref)
+		if err := s.SetProtocol(pref); err != nil {
+			return nil, err
+		}
 		lzcon := msmux.NewMSSelect(s, pref)
 		return &streamWrapper{
 			Stream: s,