reset ownership if handle creation fails (#213)

microsoft · Dec 4, 2019 · d96a164 · d96a164
1 parent 312a778
commit d96a164
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/cs/src/core/Device/FixedPool.cs b/cs/src/core/Device/FixedPool.cs
@@ -26,7 +26,7 @@ public FixedPool(int size, Func<T> creator)
         {
             while (true)
             {
-                for (int i=0; i<size; i++)
+                for (int i = 0; i < size; i++)
                 {
                     if (disposed)
                         throw new Exception("Disposed");
@@ -36,7 +36,16 @@ public FixedPool(int size, Func<T> creator)
                     {
                         if (Interlocked.CompareExchange(ref owners[i], 2, val) == val)
                         {
-                            items[i] = creator();
+                            try
+                            {
+                                items[i] = creator();
+                            }
+                            catch
+                            {
+                                Interlocked.Exchange(ref owners[i], val);
+                                throw;
+                            }
+
                             return (items[i], i);
                         }
                     }