Ignore wrong responses

Author: sanych-sun

src/MongoDB.Driver/Core/Connections/BinaryConnection.cs

@@ -392,12 +392,19 @@ public ResponseMessage ReceiveMessage(
             try
             {
                 helper.ReceivingMessage();
-                using (var buffer = ReceiveBuffer(operationContext))
+                while (true)
                 {
-                    EnsureResponseToIsCorrect(buffer, responseTo);
-                    var message = helper.DecodeMessage(operationContext, buffer, encoderSelector);
-                    helper.ReceivedMessage(buffer, message);
-                    return message;
+                    using (var buffer = ReceiveBuffer(operationContext))
+                    {
+                        if (responseTo != GetResponseTo(buffer))
+                        {
+                            continue;
+                        }
+
+                        var message = helper.DecodeMessage(operationContext, buffer, encoderSelector);
+                        helper.ReceivedMessage(buffer, message);
+                        return message;
+                    }
                 }
             }
             catch (Exception ex)
@@ -421,12 +428,19 @@ public async Task<ResponseMessage> ReceiveMessageAsync(
             try
             {
                 helper.ReceivingMessage();
-                using (var buffer = await ReceiveBufferAsync(operationContext).ConfigureAwait(false))
+                while (true)
                 {
-                    EnsureResponseToIsCorrect(buffer, responseTo);
-                    var message = helper.DecodeMessage(operationContext, buffer, encoderSelector);
-                    helper.ReceivedMessage(buffer, message);
-                    return message;
+                    using (var buffer = await ReceiveBufferAsync(operationContext).ConfigureAwait(false))
+                    {
+                        if (responseTo != GetResponseTo(buffer))
+                        {
+                            continue;
+                        }
+
+                        var message = helper.DecodeMessage(operationContext, buffer, encoderSelector);
+                        helper.ReceivedMessage(buffer, message);
+                        return message;
+                    }
                 }
             }
             catch (Exception ex)
@@ -437,21 +451,10 @@ public async Task<ResponseMessage> ReceiveMessageAsync(
             }
         }
 
-        private IByteBuffer EnsureResponseToIsCorrect(IByteBuffer message, int responseTo)
+        private int GetResponseTo(IByteBuffer message)
         {
-            var receivedResponseTo = GetResponseTo(message);
-            if (receivedResponseTo != responseTo)
-            {
-                throw new InvalidOperationException($"Expected responseTo to be {responseTo} but was {receivedResponseTo}."); // should not be reached
-            }
-
-            return message;
-
-            int GetResponseTo(IByteBuffer message)
-            {
-                var backingBytes = message.AccessBackingBytes(8);
-                return BitConverter.ToInt32(backingBytes.Array, backingBytes.Offset);
-            }
+            var backingBytes = message.AccessBackingBytes(8);
+            return BitConverter.ToInt32(backingBytes.Array, backingBytes.Offset);
         }
 
         private void SendBuffer(OperationContext operationContext, IByteBuffer buffer)