Swift: extract ConsumeExpr

Author: AlexDenisov

swift/extractor/infra/SwiftTagTraits.h

@@ -202,7 +202,7 @@ MAP(swift::Expr, ExprTag)
   MAP(swift::TypeJoinExpr, void)  // TODO (introduced in 5.8)
   MAP(swift::MacroExpansionExpr, void)  // TODO (introduced in 5.8)
   MAP(swift::CopyExpr, CopyExprTag)
-  MAP(swift::ConsumeExpr, void)  // TODO (introduced in 5.9)
+  MAP(swift::ConsumeExpr, ConsumeExprTag)
   MAP(swift::MaterializePackExpr, void)  // TODO (introduced in 5.9)
   MAP(swift::SingleValueStmtExpr, SingleValueStmtExprTag)
 

swift/extractor/translators/ExprTranslator.cpp

@@ -663,4 +663,10 @@ codeql::CopyExpr ExprTranslator::translateCopyExpr(const swift::CopyExpr& expr)
   return entry;
 }
 
+codeql::ConsumeExpr ExprTranslator::translateConsumeExpr(const swift::ConsumeExpr& expr) {
+  auto entry = createExprEntry(expr);
+  entry.sub_expr = dispatcher.fetchLabel(expr.getSubExpr());
+  return entry;
+}
+
 }  // namespace codeql

swift/extractor/translators/ExprTranslator.h

@@ -123,6 +123,7 @@ class ExprTranslator : public AstTranslatorBase<ExprTranslator> {
   codeql::PackExpansionExpr translatePackExpansionExpr(const swift::PackExpansionExpr& expr);
   codeql::PackElementExpr translatePackElementExpr(const swift::PackElementExpr& expr);
   codeql::CopyExpr translateCopyExpr(const swift::CopyExpr& expr);
+  codeql::ConsumeExpr translateConsumeExpr(const swift::ConsumeExpr& expr);
 
  private:
   void fillClosureExpr(const swift::AbstractClosureExpr& expr, codeql::ClosureExpr& entry);