ObjCAtCatchStmt's ParamStmt is always a DeclStmt.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@65759 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/include/clang/AST/Stmt.h b/include/clang/AST/Stmt.h
index 696aa00..c4c2291 100644
--- a/include/clang/AST/Stmt.h
+++ b/include/clang/AST/Stmt.h
@@ -1066,7 +1066,8 @@
public:
ObjCAtCatchStmt(SourceLocation atCatchLoc, SourceLocation rparenloc,
- Stmt *catchVarStmtDecl, Stmt *atCatchStmt, Stmt *atCatchList);
+ DeclStmt *catchVarStmtDecl,
+ Stmt *atCatchStmt, Stmt *atCatchList);
const Stmt *getCatchBody() const { return SubExprs[BODY]; }
Stmt *getCatchBody() { return SubExprs[BODY]; }
@@ -1078,8 +1079,11 @@
return static_cast<ObjCAtCatchStmt*>(SubExprs[NEXT_CATCH]);
}
- const Stmt *getCatchParamStmt() const { return SubExprs[SELECTOR]; }
- Stmt *getCatchParamStmt() { return SubExprs[SELECTOR]; }
+ const DeclStmt *getCatchParamStmt() const {
+ return static_cast<const DeclStmt*>(SubExprs[SELECTOR]); }
+ DeclStmt *getCatchParamStmt() {
+ return static_cast<DeclStmt*>(SubExprs[SELECTOR]);
+ }
SourceLocation getRParenLoc() const { return RParenLoc; }
diff --git a/lib/AST/Stmt.cpp b/lib/AST/Stmt.cpp
index b85a67e..e987d84 100644
--- a/lib/AST/Stmt.cpp
+++ b/lib/AST/Stmt.cpp
@@ -180,12 +180,13 @@
ObjCAtCatchStmt::ObjCAtCatchStmt(SourceLocation atCatchLoc,
SourceLocation rparenloc,
- Stmt *catchVarStmtDecl, Stmt *atCatchStmt,
+ DeclStmt *catchVarStmtDecl, Stmt *atCatchStmt,
Stmt *atCatchList)
: Stmt(ObjCAtCatchStmtClass) {
SubExprs[SELECTOR] = catchVarStmtDecl;
SubExprs[BODY] = atCatchStmt;
SubExprs[NEXT_CATCH] = NULL;
+ // FIXME: O(N^2) in number of catch blocks.
if (atCatchList) {
ObjCAtCatchStmt *AtCatchList = static_cast<ObjCAtCatchStmt*>(atCatchList);
diff --git a/lib/CodeGen/CGObjCMac.cpp b/lib/CodeGen/CGObjCMac.cpp
index 549a6b9..26a11c9 100644
--- a/lib/CodeGen/CGObjCMac.cpp
+++ b/lib/CodeGen/CGObjCMac.cpp
@@ -1999,8 +1999,7 @@
for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
llvm::BasicBlock *NextCatchBlock = CGF.createBasicBlock("catch");
- const DeclStmt *CatchParam =
- cast_or_null<DeclStmt>(CatchStmt->getCatchParamStmt());
+ const DeclStmt *CatchParam = CatchStmt->getCatchParamStmt();
const VarDecl *VD = 0;
const PointerType *PT = 0;
diff --git a/lib/Sema/SemaStmt.cpp b/lib/Sema/SemaStmt.cpp
index dfcb65a..39b211f 100644
--- a/lib/Sema/SemaStmt.cpp
+++ b/lib/Sema/SemaStmt.cpp
@@ -964,7 +964,7 @@
StmtArg Body, StmtArg catchList) {
Stmt *CatchList = static_cast<Stmt*>(catchList.release());
ObjCAtCatchStmt *CS = new (Context) ObjCAtCatchStmt(AtLoc, RParen,
- static_cast<Stmt*>(Parm.release()), static_cast<Stmt*>(Body.release()),
+ static_cast<DeclStmt*>(Parm.release()), static_cast<Stmt*>(Body.release()),
CatchList);
return Owned(CatchList ? CatchList : CS);
}