hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-04-26 17:25:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

C#: Include AttributeUsages in stub generator

Browse Source
This commit is contained in:
Tom Hvitved
2023-10-27 08:34:20 +02:00
parent a5bfeb68a8
commit abc16abf6f
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
csharp/extractor/Semmle.Extraction.CSharp.StubGenerator/StubVisitor.cs
View File

@@ -186,7 +186,8 @@ internal sealed class StubVisitor : SymbolVisitor
}
break;
case TypedConstantKind.Enum:
stubWriter.Write("throw null");
stubWriter.Write($"({c.Type!.GetQualifiedName()}) ");
stubWriter.Write(c.Value!.ToString());
break;
case TypedConstantKind.Array:
stubWriter.Write("new []{");
@@ -200,7 +201,8 @@ internal sealed class StubVisitor : SymbolVisitor
}
private static readonly HashSet<string> attributeAllowList = new() {
"System.FlagsAttribute"
"System.FlagsAttribute",
"System.AttributeUsageAttribute"
};
private void StubAttribute(AttributeData a, string prefix)
@@ -219,6 +221,14 @@ internal sealed class StubVisitor : SymbolVisitor
{
stubWriter.Write("(");
WriteCommaSep(a.ConstructorArguments, StubTypedConstant);
if (a.ConstructorArguments.Any() && a.NamedArguments.Any())
stubWriter.Write(",");
WriteCommaSep(a.NamedArguments, arg =>
{
stubWriter.Write(arg.Key);
stubWriter.Write(" = ");
StubTypedConstant(arg.Value);
});
stubWriter.Write(")");
}
stubWriter.WriteLine("]");