codeql/csharp/extractor/Semmle.Extraction.CSharp/Populators/Ast.cs

using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Semmle.Extraction.CSharp.Entities;

namespace Semmle.Extraction.CSharp.Populators
{
    class Ast : CSharpSyntaxVisitor
    {
        readonly Context cx;
        readonly IExpressionParentEntity parent;
        readonly int child;

        public Ast(Context cx, IExpressionParentEntity parent, int child)
        {
            this.cx = cx;
            this.parent = parent;
            this.child = child;
        }

        public override void DefaultVisit(SyntaxNode node)
        {
            cx.ModelError(node, "Unhandled syntax node {0}", node.Kind());
        }

        public override void VisitPropertyDeclaration(PropertyDeclarationSyntax node)
        {
            ((Property)parent).VisitDeclaration(cx, node);
        }


        public override void VisitArgumentList(ArgumentListSyntax node)
        {
            int c = 0;
            foreach (var m in node.Arguments)
            {
                cx.Extract(m, parent, c++);
            }
        }

        public override void VisitArgument(ArgumentSyntax node)
        {
            Expression.Create(cx, node.Expression, parent, child);
        }
    }

    public static class AstExtensions
    {
        public static void Extract(this Context cx, CSharpSyntaxNode node, IExpressionParentEntity parent, int child)
        {
            using (cx.StackGuard)
            {
                try
                {
                    node.Accept(new Ast(cx, parent, child));
                }
                catch (System.Exception e)
                {
                    cx.ModelError(node, "Exception processing syntax node of type {0}: {1}", node.Kind(), e);
                }
            }
        }

        public static void Extract(this Context cx, SyntaxNode node, IEntity parent, int child)
        {
            cx.Extract(((CSharpSyntaxNode)node), parent, child);
        }
    }
}