class RewriteTest : public ASTConsumer {
Rewriter Rewrite;
Diagnostic &Diags;
+ const LangOptions &LangOpts;
unsigned RewriteFailedDiag;
ASTContext *Context;
// Top Level Driver code.
virtual void HandleTopLevelDecl(Decl *D);
void HandleDeclInMainFile(Decl *D);
- RewriteTest(bool isHeader, Diagnostic &D) : Diags(D) {
+ RewriteTest(bool isHeader, Diagnostic &D, const LangOptions &LOpts) :
+ Diags(D), LangOpts(LOpts) {
IsHeader = isHeader;
RewriteFailedDiag = Diags.getCustomDiagID(Diagnostic::Warning,
"rewriting sub-expression within a macro (may not be correct)");
}
ASTConsumer *clang::CreateCodeRewriterTest(const std::string& InFile,
- Diagnostic &Diags) {
- return new RewriteTest(IsHeaderFile(InFile), Diags);
+ Diagnostic &Diags,
+ const LangOptions &LOpts) {
+ return new RewriteTest(IsHeaderFile(InFile), Diags, LOpts);
}
void RewriteTest::Initialize(ASTContext &context) {
"unsigned long extra[5];\n};\n"
"#define __FASTENUMERATIONSTATE\n"
"#endif\n";
+ if (LangOpts.Microsoft) {
+ std::string S = s;
+ S += "#define __attribute__(X)\n";
+ s = S.c_str();
+ }
if (IsHeader) {
// insert the whole string when rewriting a header file
InsertText(SourceLocation::getFileLoc(MainFileID, 0), s, strlen(s));
Result += "\t" + utostr(OBJC_ABI_VERSION) +
", sizeof(struct _objc_module), \"\", &_OBJC_SYMBOLS\n";
Result += "};\n\n";
-
+
+ if (LangOpts.Microsoft) {
+ Result += "#pragma section(\".objc_module_info$B\",long,read,write)\n";
+ Result += "#pragma data_seq(push, \".objc_module_info$B\")\n";
+ Result += "static struct _objc_module *_POINTER_OBJC_MODULES = ";
+ Result += "&_OBJC_MODULES;\n";
+ Result += "#pragma data_seg(pop)\n\n";
+ }
}