Swift: extract ExtracFunctionIsolationExpr

Author: redsun82

swift/extractor/infra/SwiftTagTraits.h

@@ -202,7 +202,7 @@ MAP(swift::Expr, ExprTag)
   MAP(swift::ConsumeExpr, ConsumeExprTag)
   MAP(swift::MaterializePackExpr, MaterializePackExprTag)
   MAP(swift::SingleValueStmtExpr, SingleValueStmtExprTag)
-  MAP(swift::ExtractFunctionIsolationExpr, void)  // TODO swift 6.0
+  MAP(swift::ExtractFunctionIsolationExpr, ExtractFunctionIsolationExprTag)
   MAP(swift::CurrentContextIsolationExpr, void)  // TODO swift 6.0
 MAP(swift::Decl, DeclTag)
   MAP(swift::ValueDecl, ValueDeclTag)

swift/extractor/translators/ExprTranslator.cpp

@@ -671,4 +671,11 @@ codeql::MaterializePackExpr ExprTranslator::translateMaterializePackExpr(
   return entry;
 }
 
+codeql::ExtractFunctionIsolationExpr ExprTranslator::translateExtractFunctionIsolationExpr(
+    const swift::ExtractFunctionIsolationExpr& expr) {
+  auto entry = createExprEntry(expr);
+  entry.function_expr = dispatcher.fetchLabel(expr.getFunctionExpr());
+  return entry;
+}
+
 }  // namespace codeql

swift/extractor/translators/ExprTranslator.h

@@ -125,6 +125,8 @@ class ExprTranslator : public AstTranslatorBase<ExprTranslator> {
   codeql::CopyExpr translateCopyExpr(const swift::CopyExpr& expr);
   codeql::ConsumeExpr translateConsumeExpr(const swift::ConsumeExpr& expr);
   codeql::MaterializePackExpr translateMaterializePackExpr(const swift::MaterializePackExpr& expr);
+  codeql::ExtractFunctionIsolationExpr translateExtractFunctionIsolationExpr(
+      const swift::ExtractFunctionIsolationExpr& expr);
 
  private:
   void fillClosureExpr(const swift::AbstractClosureExpr& expr, codeql::ClosureExpr& entry);