Merge pull request #14 from PowerShell/cbnilrem-refactor

Apply mitigation to CimInstance disposal race condition

Author: cbnilrem

src/Microsoft.Management.Infrastructure/CimInstance.cs

@@ -384,7 +384,10 @@ private void Dispose(bool disposing)
             {
                 if (this.InstanceHandle != null && !this.InstanceHandle.IsNull)
                 {
-                    this.InstanceHandle.Delete();
+                    using (this.InstanceHandle)
+                    {
+                        this.InstanceHandle.Delete();
+                    }
                 }
             }
         }

src/Microsoft.Management.Infrastructure/Native/MI_Instance.cs

@@ -3,7 +3,7 @@
 
 namespace Microsoft.Management.Infrastructure.Native
 {
-    internal class MI_Instance : MI_NativeObjectWithFT<MI_Instance.MI_InstanceFT>
+    internal class MI_Instance : MI_NativeObjectWithFT<MI_Instance.MI_InstanceFT>, IDisposable
     {
         internal MI_Result GetElement(
             string name,
@@ -124,21 +124,6 @@ internal MI_Result Delete()
             return localResult;
         }
 
-        private void ZeroPtr()
-        {
-            if (this.isDirect)
-            {
-                this.allocatedData = IntPtr.Zero;
-            }
-            else if (this.allocatedData != IntPtr.Zero)
-            {
-                unsafe
-                {
-                    *(IntPtr*)this.allocatedData = IntPtr.Zero;
-                }
-            }
-        }
-
         internal MI_Result IsA(
             MI_ClassDecl classDecl,
             out bool flag

src/Microsoft.Management.Infrastructure/Native/MI_NativeObject.cs

@@ -3,7 +3,7 @@
 
 namespace Microsoft.Management.Infrastructure.Native
 {
-    internal abstract class MI_NativeObject
+    internal abstract class MI_NativeObject : IDisposable
     {
         [StructLayout(LayoutKind.Sequential, CharSet = MI_PlatformSpecific.AppropriateCharSet)]
         protected struct MI_NativeObjectNormalMembersLayout
@@ -60,11 +60,13 @@ internal struct ArrayPtr
         protected IntPtr allocatedData;
         protected bool isDirect;
 
-        ~MI_NativeObject()
+        public virtual void Dispose()
         {
             if (this.allocatedData != IntPtr.Zero)
             {
-                Marshal.FreeHGlobal(this.allocatedData);
+                IntPtr tmp = this.allocatedData;
+                this.ZeroPtr();
+                Marshal.FreeHGlobal(tmp);
             }
         }
 
@@ -126,5 +128,20 @@ internal IntPtr Ptr
                 return structurePtr;
             }
         }
+
+        protected void ZeroPtr()
+        {
+            if (this.isDirect)
+            {
+                this.allocatedData = IntPtr.Zero;
+            }
+            else if (this.allocatedData != IntPtr.Zero)
+            {
+                unsafe
+                {
+                    *(IntPtr*)this.allocatedData = IntPtr.Zero;
+                }
+            }
+        }
     }
 }

src/Microsoft.Management.Infrastructure/Native/Structures/MI_Value.cs

@@ -77,7 +77,7 @@ internal MI_Value() : base(true)
             this.Dispose(false);
         }
 
-        internal void Dispose()
+        public override void Dispose()
         {
             this.Dispose(true);
             GC.SuppressFinalize(this);
@@ -105,6 +105,8 @@ private void Dispose(bool disposing)
                     this.allocatedData = IntPtr.Zero;
                 }
             }
+
+            base.Dispose();
         }
 
         internal string String