CSHARP-2768: Server selection times again out after launching mongocryptd process.

Author: DmitryLukyanov

src/MongoDB.Driver/Encryption/AutoEncryptionLibMongoController.cs

@@ -14,10 +14,13 @@
 */
 
 using System;
+using System.Diagnostics;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using MongoDB.Bson;
 using MongoDB.Driver.Core.Misc;
+using MongoDB.Driver.Core.Servers;
 using MongoDB.Driver.Core.WireProtocol;
 using MongoDB.Libmongocrypt;
 
@@ -181,6 +184,7 @@ private void ProcessNeedMongoMarkingsState(CryptContext context, string database
                 catch (TimeoutException) when (attempt == 1)
                 {
                     _mongocryptdFactory.SpawnMongocryptdProcessIfRequired();
+                    WaitForMongocryptdReady();
                 }
             }
 
@@ -205,6 +209,7 @@ private async Task ProcessNeedMongoMarkingsStateAsync(CryptContext context, stri
                 catch (TimeoutException) when (attempt == 1)
                 {
                     _mongocryptdFactory.SpawnMongocryptdProcessIfRequired();
+                    await WaitForMongocryptdReadyAsync().ConfigureAwait(false);
                 }
             }
 
@@ -223,5 +228,35 @@ private void RestoreDbNodeInResponse(BsonDocument request, BsonDocument response
                 }
             }
         }
+
+        private void WaitForMongocryptdReady()
+        {
+            var stopwatch = Stopwatch.StartNew();
+            while (stopwatch.Elapsed < TimeSpan.FromSeconds(5))
+            {
+                var clusterDescription = _mongocryptdClient.Cluster?.Description;
+                var mongocryptdServer = clusterDescription?.Servers?.FirstOrDefault();
+                if (mongocryptdServer != null && mongocryptdServer.Type != ServerType.Unknown)
+                {
+                    return;
+                }
+                Thread.Sleep(TimeSpan.FromMilliseconds(5));
+            }
+        }
+
+        private async Task WaitForMongocryptdReadyAsync()
+        {
+            var stopwatch = Stopwatch.StartNew();
+            while (stopwatch.Elapsed < TimeSpan.FromSeconds(5))
+            {
+                var clusterDescription = _mongocryptdClient.Cluster?.Description;
+                var mongocryptdServer = clusterDescription?.Servers?.FirstOrDefault();
+                if (mongocryptdServer != null && mongocryptdServer.Type != ServerType.Unknown)
+                {
+                    return;
+                }
+                await Task.Delay(TimeSpan.FromMilliseconds(5)).ConfigureAwait(false);
+            }
+        }
     }
 }