Remove from hashtable as we return from HasOnlyBlittableFields to mak…

…e sure we're handling our recursion check correctly.
dotnet · May 14, 2021 · 070f1e7 · 070f1e7
1 parent 7b935da
commit 070f1e7
Showing 1 changed file with 20 additions and 16 deletions.
diff --git a/DllImportGenerator/DllImportGenerator/TypeSymbolExtensions.cs b/DllImportGenerator/DllImportGenerator/TypeSymbolExtensions.cs
@@ -29,26 +29,30 @@ private static bool HasOnlyBlittableFields(this ITypeSymbol type, HashSet<ITypeS
             seenTypes.Add(type);
             foreach (var field in type.GetMembers().OfType<IFieldSymbol>())
             {
-                bool? fieldBlittable = field switch
+                if (!field.IsStatic)
                 {
-                    { IsStatic: true } => null,
-                    { Type: { IsReferenceType: true } } => false,
-                    { Type: IPointerTypeSymbol ptr } => IsConsideredBlittable(ptr.PointedAtType),
-                    { Type: IFunctionPointerTypeSymbol } => true,
-                    not { Type: { SpecialType: SpecialType.None } } => IsSpecialTypeBlittable(field.Type.SpecialType),
-                    // Assume that type parameters that can be blittable are blittable.
-                    // We'll re-evaluate blittability for generic fields of generic types at instantation time.
-                    { Type: ITypeParameterSymbol } => true,
-                    { Type: { IsValueType: false } } => false,
-                    _ => IsConsideredBlittable(field.Type, seenTypes)
-                };
-
-                if (fieldBlittable is false)
-                {
-                    return false;
+                    bool fieldBlittable = field switch
+                    {
+                        { Type: { IsReferenceType: true } } => false,
+                        { Type: IPointerTypeSymbol ptr } => IsConsideredBlittable(ptr.PointedAtType),
+                        { Type: IFunctionPointerTypeSymbol } => true,
+                        not { Type: { SpecialType: SpecialType.None } } => IsSpecialTypeBlittable(field.Type.SpecialType),
+                        // Assume that type parameters that can be blittable are blittable.
+                        // We'll re-evaluate blittability for generic fields of generic types at instantation time.
+                        { Type: ITypeParameterSymbol } => true,
+                        { Type: { IsValueType: false } } => false,
+                        _ => IsConsideredBlittable(field.Type, seenTypes)
+                    };
+
+                    if (!fieldBlittable)
+                    {
+                        seenTypes.Remove(type);
+                        return false;
+                    }
                 }
             }
 
+            seenTypes.Remove(type);
             return true;
         }