mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2025-05-17 11:26:37 +00:00
e08464fb45
Most clients of SourceManager.h need to do things like turning source
locations into file & line number pairs, but this doesn't require
bringing in FileManager.h and LLVM's FS headers.
The main code change here is to sink SM::createFileID into the cpp file.
I reason that this is not performance critical because it doesn't happen
on the diagnostic path, it happens along the paths of macro expansion
(could be hot) and new includes (less hot).
Saves some includes:
309 - /usr/local/google/home/rnk/llvm-project/clang/include/clang/Basic/FileManager.h
272 - /usr/local/google/home/rnk/llvm-project/clang/include/clang/Basic/FileSystemOptions.h
271 - /usr/local/google/home/rnk/llvm-project/llvm/include/llvm/Support/VirtualFileSystem.h
267 - /usr/local/google/home/rnk/llvm-project/llvm/include/llvm/Support/FileSystem.h
266 - /usr/local/google/home/rnk/llvm-project/llvm/include/llvm/Support/Chrono.h
Differential Revision: https://reviews.llvm.org/D75406
115 lines
3.8 KiB
C++
115 lines
3.8 KiB
C++
//===- unittests/Frontend/ASTUnitTest.cpp - ASTUnit tests -----------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include <fstream>
|
|
|
|
#include "clang/Basic/FileManager.h"
|
|
#include "clang/Frontend/ASTUnit.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/CompilerInvocation.h"
|
|
#include "clang/Frontend/PCHContainerOperations.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace clang;
|
|
|
|
namespace {
|
|
|
|
class ASTUnitTest : public ::testing::Test {
|
|
protected:
|
|
int FD;
|
|
llvm::SmallString<256> InputFileName;
|
|
std::unique_ptr<ToolOutputFile> input_file;
|
|
IntrusiveRefCntPtr<DiagnosticsEngine> Diags;
|
|
std::shared_ptr<CompilerInvocation> CInvok;
|
|
std::shared_ptr<PCHContainerOperations> PCHContainerOps;
|
|
|
|
std::unique_ptr<ASTUnit> createASTUnit(bool isVolatile) {
|
|
EXPECT_FALSE(llvm::sys::fs::createTemporaryFile("ast-unit", "cpp", FD,
|
|
InputFileName));
|
|
input_file = std::make_unique<ToolOutputFile>(InputFileName, FD);
|
|
input_file->os() << "";
|
|
|
|
const char *Args[] = {"clang", "-xc++", InputFileName.c_str()};
|
|
|
|
Diags = CompilerInstance::createDiagnostics(new DiagnosticOptions());
|
|
|
|
CInvok = createInvocationFromCommandLine(Args, Diags);
|
|
|
|
if (!CInvok)
|
|
return nullptr;
|
|
|
|
FileManager *FileMgr =
|
|
new FileManager(FileSystemOptions(), vfs::getRealFileSystem());
|
|
PCHContainerOps = std::make_shared<PCHContainerOperations>();
|
|
|
|
return ASTUnit::LoadFromCompilerInvocation(
|
|
CInvok, PCHContainerOps, Diags, FileMgr, false, CaptureDiagsKind::None,
|
|
0, TU_Complete, false, false, isVolatile);
|
|
}
|
|
};
|
|
|
|
TEST_F(ASTUnitTest, SaveLoadPreservesLangOptionsInPrintingPolicy) {
|
|
// Check that the printing policy is restored with the correct language
|
|
// options when loading an ASTUnit from a file. To this end, an ASTUnit
|
|
// for a C++ translation unit is set up and written to a temporary file.
|
|
|
|
// By default `UseVoidForZeroParams` is true for non-C++ language options,
|
|
// thus we can check this field after loading the ASTUnit to deduce whether
|
|
// the correct (C++) language options were used when setting up the printing
|
|
// policy.
|
|
|
|
{
|
|
PrintingPolicy PolicyWithDefaultLangOpt(LangOptions{});
|
|
EXPECT_TRUE(PolicyWithDefaultLangOpt.UseVoidForZeroParams);
|
|
}
|
|
|
|
std::unique_ptr<ASTUnit> AST = createASTUnit(false);
|
|
|
|
if (!AST)
|
|
FAIL() << "failed to create ASTUnit";
|
|
|
|
EXPECT_FALSE(AST->getASTContext().getPrintingPolicy().UseVoidForZeroParams);
|
|
|
|
llvm::SmallString<256> ASTFileName;
|
|
ASSERT_FALSE(
|
|
llvm::sys::fs::createTemporaryFile("ast-unit", "ast", FD, ASTFileName));
|
|
ToolOutputFile ast_file(ASTFileName, FD);
|
|
AST->Save(ASTFileName.str());
|
|
|
|
EXPECT_TRUE(llvm::sys::fs::exists(ASTFileName));
|
|
|
|
std::unique_ptr<ASTUnit> AU = ASTUnit::LoadFromASTFile(
|
|
std::string(ASTFileName.str()), PCHContainerOps->getRawReader(),
|
|
ASTUnit::LoadEverything, Diags, FileSystemOptions(),
|
|
/*UseDebugInfo=*/false);
|
|
|
|
if (!AU)
|
|
FAIL() << "failed to load ASTUnit";
|
|
|
|
EXPECT_FALSE(AU->getASTContext().getPrintingPolicy().UseVoidForZeroParams);
|
|
}
|
|
|
|
TEST_F(ASTUnitTest, GetBufferForFileMemoryMapping) {
|
|
std::unique_ptr<ASTUnit> AST = createASTUnit(true);
|
|
|
|
if (!AST)
|
|
FAIL() << "failed to create ASTUnit";
|
|
|
|
std::unique_ptr<llvm::MemoryBuffer> memoryBuffer =
|
|
AST->getBufferForFile(InputFileName);
|
|
|
|
EXPECT_NE(memoryBuffer->getBufferKind(),
|
|
llvm::MemoryBuffer::MemoryBuffer_MMap);
|
|
}
|
|
|
|
} // anonymous namespace
|