1 | //===--- RestrictSystemLibcHeadersCheck.cpp - clang-tidy ------------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "RestrictSystemLibcHeadersCheck.h" |
10 | #include "clang/AST/ASTContext.h" |
11 | #include "clang/Lex/HeaderSearch.h" |
12 | #include "clang/Lex/HeaderSearchOptions.h" |
13 | #include "clang/Lex/Preprocessor.h" |
14 | |
15 | // FixItHint - Hint to check documentation script to mark this check as |
16 | // providing a FixIt. |
17 | |
18 | namespace clang::tidy::llvm_libc { |
19 | |
20 | namespace { |
21 | |
22 | class RestrictedIncludesPPCallbacks |
23 | : public portability::RestrictedIncludesPPCallbacks { |
24 | public: |
25 | explicit ( |
26 | RestrictSystemLibcHeadersCheck &Check, const SourceManager &SM, |
27 | const SmallString<128> CompilerIncudeDir) |
28 | : portability::RestrictedIncludesPPCallbacks(Check, SM), |
29 | CompilerIncudeDir(CompilerIncudeDir) {} |
30 | |
31 | void InclusionDirective(SourceLocation HashLoc, const Token &IncludeTok, |
32 | StringRef FileName, bool IsAngled, |
33 | CharSourceRange FilenameRange, |
34 | OptionalFileEntryRef File, StringRef SearchPath, |
35 | StringRef RelativePath, const Module *SuggestedModule, |
36 | bool ModuleImported, |
37 | SrcMgr::CharacteristicKind FileType) override; |
38 | |
39 | private: |
40 | const SmallString<128> CompilerIncudeDir; |
41 | }; |
42 | |
43 | } // namespace |
44 | |
45 | void RestrictedIncludesPPCallbacks::InclusionDirective( |
46 | SourceLocation HashLoc, const Token &IncludeTok, StringRef FileName, |
47 | bool IsAngled, CharSourceRange FilenameRange, OptionalFileEntryRef File, |
48 | StringRef SearchPath, StringRef RelativePath, const Module *SuggestedModule, |
49 | bool ModuleImported, SrcMgr::CharacteristicKind FileType) { |
50 | // Compiler provided headers are allowed (e.g stddef.h). |
51 | if (SrcMgr::isSystem(CK: FileType) && SearchPath == CompilerIncudeDir) |
52 | return; |
53 | portability::RestrictedIncludesPPCallbacks::InclusionDirective( |
54 | HashLoc, IncludeTok, FileName, IsAngled, FilenameRange, File, SearchPath, |
55 | RelativePath, SuggestedModule, ModuleImported, FileType); |
56 | } |
57 | |
58 | void RestrictSystemLibcHeadersCheck::( |
59 | const SourceManager &SM, Preprocessor *PP, Preprocessor *ModuleExpanderPP) { |
60 | SmallString<128> CompilerIncudeDir = |
61 | StringRef(PP->getHeaderSearchInfo().getHeaderSearchOpts().ResourceDir); |
62 | llvm::sys::path::append(path&: CompilerIncudeDir, a: "include" ); |
63 | PP->addPPCallbacks(C: std::make_unique<RestrictedIncludesPPCallbacks>( |
64 | args&: *this, args: SM, args&: CompilerIncudeDir)); |
65 | } |
66 | |
67 | } // namespace clang::tidy::llvm_libc |
68 | |