C#: Extract alignment and format clauses of string interpolation expressions.

Author: michaelnebel

csharp/extractor/Semmle.Extraction.CSharp/Entities/Expressions/InterpolatedString.cs

@@ -1,5 +1,4 @@
 using System.IO;
-using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
 using Semmle.Extraction.Kinds;
@@ -21,15 +20,7 @@ protected override void PopulateExpression(TextWriter trapFile)
                 {
                     case SyntaxKind.Interpolation:
                         var interpolation = (InterpolationSyntax)c;
-                        var exp = interpolation.Expression;
-                        if (Context.GetTypeInfo(exp).Type is ITypeSymbol type && !type.ImplementsIFormattable())
-                        {
-                            ImplicitToString.Create(Context, exp, this, child++);
-                        }
-                        else
-                        {
-                            Create(Context, exp, this, child++);
-                        }
+                        new InterpolatedStringInsert(Context, interpolation, this, child++);
                         break;
                     case SyntaxKind.InterpolatedStringText:
                         // Create a string literal

csharp/extractor/Semmle.Extraction.CSharp/Entities/Expressions/InterpolatedStringInsert.cs

@@ -0,0 +1,38 @@
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+using Semmle.Extraction.Kinds;
+
+namespace Semmle.Extraction.CSharp.Entities.Expressions
+{
+    internal class InterpolatedStringInsert : Expression
+    {
+        public InterpolatedStringInsert(Context cx, InterpolationSyntax syntax, IExpressionParentEntity parent, int child) :
+            base(new ExpressionInfo(cx, null, cx.CreateLocation(syntax.GetLocation()), ExprKind.INTERPOLATED_STRING_INSERT, parent, child, isCompilerGenerated: false, null))
+        {
+            var exp = syntax.Expression;
+            if (Context.GetTypeInfo(exp).Type is ITypeSymbol type && !type.ImplementsIFormattable())
+            {
+                ImplicitToString.Create(Context, exp, this, 0);
+            }
+            {
+                Create(Context, exp, this, 0);
+            }
+
+            // Hardcode the child number of the optional alignment clause to 1 and format clause to 2.
+            // This simplifies the logic in QL.
+            if (syntax.AlignmentClause?.Value is ExpressionSyntax alignment)
+            {
+                Create(Context, alignment, this, 1);
+            }
+
+            if (syntax.FormatClause is InterpolationFormatClauseSyntax format)
+            {
+                var f = format.FormatStringToken.ValueText;
+                var t = AnnotatedTypeSymbol.CreateNotAnnotated(Context.Compilation.GetSpecialType(SpecialType.System_String));
+                new Expression(new ExpressionInfo(Context, t, Context.CreateLocation(format.GetLocation()), ExprKind.UTF16_STRING_LITERAL, this, 2, isCompilerGenerated: false, f));
+            }
+
+        }
+    }
+
+}

csharp/extractor/Semmle.Extraction.CSharp/Kinds/ExprKind.cs

@@ -132,6 +132,7 @@ public enum ExprKind
         UTF8_STRING_LITERAL = 135,
         COLLECTION = 136,
         SPREAD_ELEMENT = 137,
+        INTERPOLATED_STRING_INSERT = 138,
         DEFINE_SYMBOL = 999,
     }
 }

csharp/ql/lib/semmle/code/csharp/exprs/Expr.qll

@@ -34,8 +34,9 @@ private import semmle.code.csharp.TypeRef
  * `as` expression (`AsExpr`), a cast (`CastExpr`), a `typeof` expression
  * (`TypeofExpr`), a `default` expression (`DefaultValueExpr`), an `await`
  * expression (`AwaitExpr`), a `nameof` expression (`NameOfExpr`), an
- * interpolated string (`InterpolatedStringExpr`), a qualifiable expression
- * (`QualifiableExpr`), or a literal (`Literal`).
+ * interpolated string (`InterpolatedStringExpr`), an interpolated string
+ * insert (`InterpolatedStringInsertExpr`), a qualifiable expression (`QualifiableExpr`),
+ * or a literal (`Literal`).
  */
 class Expr extends ControlFlowElement, @expr {
   override Location getALocation() { expr_location(this, result) }
@@ -917,6 +918,40 @@ class NameOfExpr extends Expr, @nameof_expr {
   override string getAPrimaryQlClass() { result = "NameOfExpr" }
 }
 
+/**
+ * An interpolated string insert, for example `{pi, align:F3}` on line 3 in
+ *
+ * ```csharp
+ * void Pi() {
+ *     float pi = 3.14159f;
+ *     Console.WriteLine($"Hello Pi, {pi,6:F3}");
+ * }
+ * ```
+ */
+class InterpolatedStringInsertExpr extends Expr, @interpolated_string_insert_expr {
+  override string toString() { result = "{...}" }
+
+  override string getAPrimaryQlClass() { result = "InterpolatedStringInsertExpr" }
+
+  /**
+   * Gets the insert expression in this interpolated string insert. For
+   * example, the insert expression in `{pi, align:F3}` is `pi`.
+   */
+  Expr getInsert() { result = this.getChild(0) }
+
+  /**
+   * Gets the alignment expression in this interpolated string insert, if any.
+   * For example, the alignment expression in `{pi, align:F3}` is `align`.
+   */
+  Expr getAlignment() { result = this.getChild(1) }
+
+  /**
+   * Gets the format expression in this interpolated string insert, if any.
+   * For example, the format expression in `{pi, align:F3}` is `F3`.
+   */
+  StringLiteral getFormat() { result = this.getChild(2) }
+}
+
 /**
  * An interpolated string, for example `$"Hello, {name}!"` on line 2 in
  *
@@ -931,16 +966,20 @@ class InterpolatedStringExpr extends Expr, @interpolated_string_expr {
 
   override string getAPrimaryQlClass() { result = "InterpolatedStringExpr" }
 
+  /**
+   * Gets the interpolated string insert at index `i` in this interpolated string, if any.
+   * For example, the interpolated string insert at index `i = 1` is `{pi, align:F3}`
+   * in `$"Hello, {pi, align:F3}!"`.
+   */
+  InterpolatedStringInsertExpr getInterpolatedInsert(int i) { result = this.getChild(i) }
+
   /**
    * Gets the insert at index `i` in this interpolated string, if any. For
    * example, the insert at index `i = 1` is `name` in `$"Hello, {name}!"`.
    * Note that there is no insert at index `i = 0`, but instead a text
    * element (`getText(0)` gets the text).
    */
-  Expr getInsert(int i) {
-    result = this.getChild(i) and
-    not result instanceof StringLiteral
-  }
+  Expr getInsert(int i) { result = this.getInterpolatedInsert(i).getInsert() }
 
   /**
    * Gets the text element at index `i` in this interpolated string, if any.

csharp/ql/lib/semmlecode.csharp.dbscheme

@@ -1168,6 +1168,7 @@ case @expr.kind of
 /* C# 12.0 */
 | 136 = @collection_expr
 | 137 = @spread_element_expr
+| 138 = @interpolated_string_insert_expr
 /* Preprocessor */
 | 999 = @define_symbol_expr
 ;