cg: Add missing attribute visits.

trailofbits · Nov 4, 2024 · 57aca21 · 57aca21
1 parent 4bae47d
commit 57aca21
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lib/vast/CodeGen/DefaultAttrVisitor.cpp b/lib/vast/CodeGen/DefaultAttrVisitor.cpp
@@ -81,7 +81,7 @@ namespace vast::cg
 
     mlir_attr default_attr_visitor::VisitOverloadableAttr(const clang::OverloadableAttr *attr) {
         return make< hl::OverloadableAttr >();
-    }
+}
 
     mlir_attr default_attr_visitor::VisitNonNullAttr(const clang::NonNullAttr *) {
         return make< hl::NonNullAttr >();
@@ -160,6 +160,14 @@ namespace vast::cg
         return make< hl::TransparentUnionAttr >();
     }
 
+    mlir_attr default_attr_visitor::VisitUnavailableAttr(const clang::UnavailableAttr *attr) {
+        return make< hl::UnavailableAttr >(attr->getMessage());
+    }
+
+    mlir_attr default_attr_visitor::VisitDiagnoseIfAttr(const clang::DiagnoseIfAttr *attr) {
+        return make< hl::DiagnoseIfAttr >();
+    }
+
     mlir_attr default_attr_visitor::VisitReturnsTwiceAttr(const clang::ReturnsTwiceAttr *attr) {
         return make< hl::ReturnsTwiceAttr >();
     }