forked from OSchip/llvm-project
117 lines
4.5 KiB
C++
117 lines
4.5 KiB
C++
//===--- RestrictSystemIncludesCheck.cpp - clang-tidy----------------------===//
|
|
//
|
|
// 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 "RestrictSystemIncludesCheck.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Lex/HeaderSearch.h"
|
|
#include "clang/Lex/PPCallbacks.h"
|
|
#include "clang/Lex/Preprocessor.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include <cstring>
|
|
|
|
namespace clang {
|
|
namespace tidy {
|
|
namespace fuchsia {
|
|
|
|
class RestrictedIncludesPPCallbacks : public PPCallbacks {
|
|
public:
|
|
explicit RestrictedIncludesPPCallbacks(RestrictSystemIncludesCheck &Check,
|
|
const SourceManager &SM)
|
|
: Check(Check), SM(SM) {}
|
|
|
|
void InclusionDirective(SourceLocation HashLoc, const Token &IncludeTok,
|
|
StringRef FileName, bool IsAngled,
|
|
CharSourceRange FilenameRange, const FileEntry *File,
|
|
StringRef SearchPath, StringRef RelativePath,
|
|
const Module *Imported,
|
|
SrcMgr::CharacteristicKind FileType) override;
|
|
void EndOfMainFile() override;
|
|
|
|
private:
|
|
struct IncludeDirective {
|
|
IncludeDirective() = default;
|
|
IncludeDirective(SourceLocation Loc, CharSourceRange Range,
|
|
StringRef Filename, StringRef FullPath, bool IsInMainFile)
|
|
: Loc(Loc), Range(Range), IncludeFile(Filename), IncludePath(FullPath),
|
|
IsInMainFile(IsInMainFile) {}
|
|
|
|
SourceLocation Loc; // '#' location in the include directive
|
|
CharSourceRange Range; // SourceRange for the file name
|
|
std::string IncludeFile; // Filename as a string
|
|
std::string IncludePath; // Full file path as a string
|
|
bool IsInMainFile; // Whether or not the include is in the main file
|
|
};
|
|
|
|
using FileIncludes = llvm::SmallVector<IncludeDirective, 8>;
|
|
llvm::SmallDenseMap<FileID, FileIncludes> IncludeDirectives;
|
|
|
|
RestrictSystemIncludesCheck &Check;
|
|
const SourceManager &SM;
|
|
};
|
|
|
|
void RestrictedIncludesPPCallbacks::InclusionDirective(
|
|
SourceLocation HashLoc, const Token &IncludeTok, StringRef FileName,
|
|
bool IsAngled, CharSourceRange FilenameRange, const FileEntry *File,
|
|
StringRef SearchPath, StringRef RelativePath, const Module *Imported,
|
|
SrcMgr::CharacteristicKind FileType) {
|
|
if (!Check.contains(FileName) && SrcMgr::isSystem(FileType)) {
|
|
SmallString<256> FullPath;
|
|
llvm::sys::path::append(FullPath, SearchPath);
|
|
llvm::sys::path::append(FullPath, RelativePath);
|
|
// Bucket the allowed include directives by the id of the file they were
|
|
// declared in.
|
|
IncludeDirectives[SM.getFileID(HashLoc)].emplace_back(
|
|
HashLoc, FilenameRange, FileName, FullPath.str(),
|
|
SM.isInMainFile(HashLoc));
|
|
}
|
|
}
|
|
|
|
void RestrictedIncludesPPCallbacks::EndOfMainFile() {
|
|
for (const auto &Bucket : IncludeDirectives) {
|
|
const FileIncludes &FileDirectives = Bucket.second;
|
|
|
|
// Emit fixits for all restricted includes.
|
|
for (const auto &Include : FileDirectives) {
|
|
// Fetch the length of the include statement from the start to just after
|
|
// the newline, for finding the end (including the newline).
|
|
unsigned ToLen = std::strcspn(SM.getCharacterData(Include.Loc), "\n") + 1;
|
|
CharSourceRange ToRange = CharSourceRange::getCharRange(
|
|
Include.Loc, Include.Loc.getLocWithOffset(ToLen));
|
|
|
|
if (!Include.IsInMainFile) {
|
|
auto D = Check.diag(
|
|
Include.Loc,
|
|
"system include %0 not allowed, transitively included from %1");
|
|
D << Include.IncludeFile << SM.getFilename(Include.Loc);
|
|
D << FixItHint::CreateRemoval(ToRange);
|
|
continue;
|
|
}
|
|
auto D = Check.diag(Include.Loc, "system include %0 not allowed");
|
|
D << Include.IncludeFile;
|
|
D << FixItHint::CreateRemoval(ToRange);
|
|
}
|
|
}
|
|
}
|
|
|
|
void RestrictSystemIncludesCheck::registerPPCallbacks(
|
|
const SourceManager &SM, Preprocessor *PP, Preprocessor *ModuleExpanderPP) {
|
|
PP->addPPCallbacks(
|
|
std::make_unique<RestrictedIncludesPPCallbacks>(*this, SM));
|
|
}
|
|
|
|
void RestrictSystemIncludesCheck::storeOptions(
|
|
ClangTidyOptions::OptionMap &Opts) {
|
|
Options.store(Opts, "Includes", AllowedIncludes);
|
|
}
|
|
|
|
} // namespace fuchsia
|
|
} // namespace tidy
|
|
} // namespace clang
|