cg: Generate FormatArgAttr.

trailofbits · Nov 4, 2024 · 955d6c6 · 955d6c6
1 parent 5e6b250
commit 955d6c6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/include/vast/CodeGen/DefaultAttrVisitor.hpp b/include/vast/CodeGen/DefaultAttrVisitor.hpp
@@ -29,6 +29,7 @@ namespace vast::cg {
         mlir_attr VisitAliasAttr(const clang::AliasAttr *attr);
         mlir_attr VisitErrorAttr(const clang::ErrorAttr *attr);
         mlir_attr VisitFormatAttr(const clang::FormatAttr *attr);
+        mlir_attr VisitFormatArgAttr(const clang::FormatArgAttr *attr);
         mlir_attr VisitAnnotateAttr(const clang::AnnotateAttr *attr);
         mlir_attr VisitAlignedAttr(const clang::AlignedAttr *attr);
         mlir_attr VisitAlwaysInlineAttr(const clang::AlwaysInlineAttr *attr);

diff --git a/include/vast/Dialect/HighLevel/HighLevelAttributes.td b/include/vast/Dialect/HighLevel/HighLevelAttributes.td
@@ -67,6 +67,11 @@ class HighLevel_FlatSymbolReferringAttr<string name, string attr_mnemonic>
 
 def HighLevel_AnnotationAttr       : HighLevel_NameAttr< "Annotation", "annotation" >;
 def HighLevel_FormatAttr           : HighLevel_NameAttr< "Format", "format" >;
+def HighLevel_FormatArgAttr        : HighLevel_NameAttr< "FormatArg", "format_arg" >
+{
+  let parameters = (ins "unsigned":$format_idx);
+  let assemblyFormat = "`<` $format_idx `>`";
+}
 def HighLevel_SectionAttr          : HighLevel_NameAttr< "Section", "section" >;
 def HighLevel_AliasAttr            : HighLevel_NameAttr< "Alias", "alias" >;
 def HighLevel_ErrorAttr            : HighLevel_NameAttr< "Error", "error" >;

diff --git a/lib/vast/CodeGen/DefaultAttrVisitor.cpp b/lib/vast/CodeGen/DefaultAttrVisitor.cpp
@@ -22,11 +22,14 @@ namespace vast::cg
         return make< hl::ErrorAttr >(attr->getUserDiagnostic());
     }
 
-
     mlir_attr default_attr_visitor::VisitFormatAttr(const clang::FormatAttr *attr) {
         return make< hl::FormatAttr >(attr->getType()->getName());
     }
 
+    mlir_attr default_attr_visitor::VisitFormatArgAttr(const clang::FormatArgAttr *attr) {
+        return make< hl::FormatArgAttr >(attr->getFormatIdx().serialize());
+    }
+
     mlir_attr default_attr_visitor::VisitAnnotateAttr(const clang::AnnotateAttr *attr) {
         return make< hl::AnnotationAttr >(attr->getAnnotation());
     }