1 | //===--- KernelNameRestrictionCheck.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 "KernelNameRestrictionCheck.h" |
10 | #include "clang/Frontend/CompilerInstance.h" |
11 | #include "clang/Lex/PPCallbacks.h" |
12 | #include "clang/Lex/Preprocessor.h" |
13 | #include <vector> |
14 | |
15 | using namespace clang::ast_matchers; |
16 | |
17 | namespace clang::tidy::altera { |
18 | |
19 | namespace { |
20 | |
21 | class KernelNameRestrictionPPCallbacks : public PPCallbacks { |
22 | public: |
23 | explicit KernelNameRestrictionPPCallbacks(ClangTidyCheck &Check, |
24 | const SourceManager &SM) |
25 | : Check(Check), SM(SM) {} |
26 | |
27 | void InclusionDirective(SourceLocation HashLoc, const Token &IncludeTok, |
28 | StringRef FileName, bool IsAngled, |
29 | CharSourceRange FileNameRange, |
30 | OptionalFileEntryRef File, StringRef SearchPath, |
31 | StringRef RelativePath, const Module *SuggestedModule, |
32 | bool ModuleImported, |
33 | SrcMgr::CharacteristicKind FileType) override; |
34 | |
35 | void EndOfMainFile() override; |
36 | |
37 | private: |
38 | /// Returns true if the name of the file with path FileName is 'kernel.cl', |
39 | /// 'verilog.cl', or 'vhdl.cl'. The file name check is case insensitive. |
40 | bool fileNameIsRestricted(StringRef FileName); |
41 | |
42 | struct IncludeDirective { |
43 | SourceLocation Loc; // Location in the include directive. |
44 | StringRef FileName; // Filename as a string. |
45 | }; |
46 | |
47 | std::vector<IncludeDirective> IncludeDirectives; |
48 | ClangTidyCheck &Check; |
49 | const SourceManager &SM; |
50 | }; |
51 | |
52 | } // namespace |
53 | |
54 | void KernelNameRestrictionCheck::registerPPCallbacks(const SourceManager &SM, |
55 | Preprocessor *PP, |
56 | Preprocessor *) { |
57 | PP->addPPCallbacks( |
58 | C: std::make_unique<KernelNameRestrictionPPCallbacks>(args&: *this, args: SM)); |
59 | } |
60 | |
61 | void KernelNameRestrictionPPCallbacks::InclusionDirective( |
62 | SourceLocation HashLoc, const Token &, StringRef FileName, bool, |
63 | CharSourceRange, OptionalFileEntryRef, StringRef, StringRef, const Module *, |
64 | bool, SrcMgr::CharacteristicKind) { |
65 | IncludeDirective ID = {.Loc: HashLoc, .FileName: FileName}; |
66 | IncludeDirectives.push_back(x: std::move(ID)); |
67 | } |
68 | |
69 | bool KernelNameRestrictionPPCallbacks::fileNameIsRestricted( |
70 | StringRef FileName) { |
71 | return FileName.equals_insensitive(RHS: "kernel.cl" ) || |
72 | FileName.equals_insensitive(RHS: "verilog.cl" ) || |
73 | FileName.equals_insensitive(RHS: "vhdl.cl" ); |
74 | } |
75 | |
76 | void KernelNameRestrictionPPCallbacks::EndOfMainFile() { |
77 | |
78 | // Check main file for restricted names. |
79 | OptionalFileEntryRef Entry = SM.getFileEntryRefForID(FID: SM.getMainFileID()); |
80 | StringRef FileName = llvm::sys::path::filename(path: Entry->getName()); |
81 | if (fileNameIsRestricted(FileName)) |
82 | Check.diag(Loc: SM.getLocForStartOfFile(FID: SM.getMainFileID()), |
83 | Description: "compiling '%0' may cause additional compilation errors due " |
84 | "to the name of the kernel source file; consider renaming the " |
85 | "included kernel source file" ) |
86 | << FileName; |
87 | |
88 | if (IncludeDirectives.empty()) |
89 | return; |
90 | |
91 | // Check included files for restricted names. |
92 | for (const IncludeDirective &ID : IncludeDirectives) { |
93 | StringRef FileName = llvm::sys::path::filename(path: ID.FileName); |
94 | if (fileNameIsRestricted(FileName)) |
95 | Check.diag(Loc: ID.Loc, |
96 | Description: "including '%0' may cause additional compilation errors due " |
97 | "to the name of the kernel source file; consider renaming the " |
98 | "included kernel source file" ) |
99 | << FileName; |
100 | } |
101 | } |
102 | |
103 | } // namespace clang::tidy::altera |
104 | |