aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/Rewrite/RewriteModernObjC.cpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/lib/Rewrite/RewriteModernObjC.cpp b/lib/Rewrite/RewriteModernObjC.cpp
index 3099499805..87d49fd08f 100644
--- a/lib/Rewrite/RewriteModernObjC.cpp
+++ b/lib/Rewrite/RewriteModernObjC.cpp
@@ -5146,6 +5146,10 @@ void RewriteModernObjC::HandleTranslationUnit(ASTContext &C) {
void RewriteModernObjC::Initialize(ASTContext &context) {
InitializeCommon(context);
+ Preamble += "#ifndef __OBJC2__\n";
+ Preamble += "#define __OBJC2__\n";
+ Preamble += "#endif\n";
+
// declaring objc_selector outside the parameter list removes a silly
// scope related warning...
if (IsHeader)