[clang] Added a new attribute 'DisableTryStmt'

Author: gmh5225

clang/lib/CodeGen/CGException.cpp

@@ -633,7 +633,8 @@ void CodeGenFunction::EmitCXXTryStmt(const CXXTryStmt &S) {
 
   // If try statements are disabled, just emit the try statement as a compound
   // stmt.
-  if (CGM.getCodeGenOpts().DisableTryStmt) {
+  if (CGM.getCodeGenOpts().DisableTryStmt ||
+      (CurCodeDecl && CurCodeDecl->hasAttr<DisableTryStmtAttr>())) {
     EmitStmt(S.getTryBlock());
     return;
   }
@@ -1752,7 +1753,8 @@ void CodeGenFunction::EmitSEHTryStmt(const SEHTryStmt &S) {
 
   // If try statements are disabled, just emit the try statement as a compound
   // stmt.
-  if (CGM.getCodeGenOpts().DisableTryStmt) {
+  if (CGM.getCodeGenOpts().DisableTryStmt ||
+      (CurCodeDecl && CurCodeDecl->hasAttr<DisableTryStmtAttr>())) {
     EmitStmt(S.getTryBlock());
     return;
   }

clang/lib/CodeGen/CodeGenFunction.cpp

@@ -408,6 +408,10 @@ void CodeGenFunction::FinishFunction(SourceLocation EndLoc) {
   if (CurCodeDecl && CurCodeDecl->hasAttr<VolatileAttr>())
     CurFn->setVolatileAndAppendToUsed();
 
+  // Check 'DisableTryStmt' attribute
+  if (CurCodeDecl && CurCodeDecl->hasAttr<DisableTryStmtAttr>())
+    CurFn->addFnAttr("no-try-stmt");
+  
   // Reset the debug location to that of the simple 'return' expression, if any
   // rather than that of the end of the function's scope '}'.
   ApplyDebugLocation AL(*this, Loc);

clang/lib/Sema/SemaDeclAttr.cpp

@@ -2199,6 +2199,10 @@ static void handleVolatileAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
   D->addAttr(::new (S.Context) UsedAttr(S.Context, AL));
 }
 
+static void handleDisableTryStmtAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
+  D->addAttr(::new (S.Context) DisableTryStmtAttr(S.Context, AL));
+}
+
 static void handleNoReturnAttr(Sema &S, Decl *D, const ParsedAttr &Attrs) {
   if (hasDeclarator(D)) return;
 
@@ -9159,6 +9163,9 @@ ProcessDeclAttribute(Sema &S, Scope *scope, Decl *D, const ParsedAttr &AL,
   case ParsedAttr::AT_Volatile:
     handleVolatileAttr(S, D, AL);
     break;
+  case ParsedAttr::AT_DisableTryStmt:
+    handleDisableTryStmtAttr(S, D, AL);
+    break;
   case ParsedAttr::AT_NoReturn:
     handleNoReturnAttr(S, D, AL);
     break;