1 //===- unittests/CodeGen/BufferSourceTest.cpp - MemoryBuffer source tests -===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "clang/AST/ASTConsumer.h" 11 #include "clang/AST/ASTContext.h" 12 #include "clang/AST/RecursiveASTVisitor.h" 13 #include "clang/Basic/TargetInfo.h" 14 #include "clang/CodeGen/ModuleBuilder.h" 15 #include "clang/Frontend/CompilerInstance.h" 16 #include "clang/Lex/Preprocessor.h" 17 #include "clang/Parse/ParseAST.h" 18 #include "clang/Sema/Sema.h" 19 #include "llvm/ADT/Triple.h" 20 #include "llvm/IR/LLVMContext.h" 21 #include "llvm/Support/Host.h" 22 #include "llvm/Support/MemoryBuffer.h" 23 #include "gtest/gtest.h" 24 25 using namespace llvm; 26 using namespace clang; 27 28 namespace { 29 30 // Emitting constructors for global objects involves looking 31 // at the source file name. This makes sure that we don't crash 32 // if the source file is a memory buffer. 33 const char TestProgram[] = 34 "class EmitCXXGlobalInitFunc " 35 "{ " 36 "public: " 37 " EmitCXXGlobalInitFunc() {} " 38 "}; " 39 "EmitCXXGlobalInitFunc test; "; 40 41 TEST(BufferSourceTest, EmitCXXGlobalInitFunc) { 42 LLVMContext Context; 43 CompilerInstance compiler; 44 45 compiler.createDiagnostics(); 46 compiler.getLangOpts().CPlusPlus = 1; 47 compiler.getLangOpts().CPlusPlus11 = 1; 48 49 compiler.getTargetOpts().Triple = llvm::Triple::normalize( 50 llvm::sys::getProcessTriple()); 51 compiler.setTarget(clang::TargetInfo::CreateTargetInfo( 52 compiler.getDiagnostics(), 53 std::make_shared<clang::TargetOptions>( 54 compiler.getTargetOpts()))); 55 56 compiler.createFileManager(); 57 compiler.createSourceManager(compiler.getFileManager()); 58 compiler.createPreprocessor(clang::TU_Prefix); 59 60 compiler.createASTContext(); 61 62 compiler.setASTConsumer(std::unique_ptr<ASTConsumer>( 63 CreateLLVMCodeGen( 64 compiler.getDiagnostics(), 65 "EmitCXXGlobalInitFuncTest", 66 compiler.getHeaderSearchOpts(), 67 compiler.getPreprocessorOpts(), 68 compiler.getCodeGenOpts(), 69 Context))); 70 71 compiler.createSema(clang::TU_Prefix, nullptr); 72 73 clang::SourceManager &sm = compiler.getSourceManager(); 74 sm.setMainFileID(sm.createFileID( 75 llvm::MemoryBuffer::getMemBuffer(TestProgram), clang::SrcMgr::C_User)); 76 77 clang::ParseAST(compiler.getSema(), false, false); 78 } 79 80 } // end anonymous namespace 81
