diff options
-rw-r--r-- | lib/Rewrite/RewriteModernObjC.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/lib/Rewrite/RewriteModernObjC.cpp b/lib/Rewrite/RewriteModernObjC.cpp index 37e780aba3..256984d881 100644 --- a/lib/Rewrite/RewriteModernObjC.cpp +++ b/lib/Rewrite/RewriteModernObjC.cpp @@ -5959,7 +5959,11 @@ void RewriteModernObjC::Initialize(ASTContext &context) { Preamble += "#define __block\n"; Preamble += "#define __weak\n"; } - Preamble += "\nextern \"C\" void * memset(void *b, int c, unsigned long len);\n"; + Preamble += "\n#if defined(_MSC_VER)\n"; + Preamble += "#include <string.h>\n"; + Preamble += "#else\n"; + Preamble += "extern \"C\" void * memset(void *b, int c, unsigned long len);\n"; + Preamble += "#endif\n"; // Declarations required for modern objective-c array and dictionary literals. Preamble += "\n#include <stdarg.h>\n"; |