1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
//===--- AvoidPragmaOnceCheck.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 "AvoidPragmaOnceCheck.h"
#include "clang/Basic/SourceManager.h"
#include "clang/Lex/PPCallbacks.h"
#include "clang/Lex/Preprocessor.h"
#include "llvm/ADT/StringRef.h"
namespace clang::tidy::portability {
class PragmaOnceCallbacks : public PPCallbacks {
public:
PragmaOnceCallbacks(AvoidPragmaOnceCheck *Check, const SourceManager &SM)
: Check(Check), SM(SM) {}
void PragmaDirective(SourceLocation Loc,
PragmaIntroducerKind Introducer) override {
auto Str = llvm::StringRef(SM.getCharacterData(Loc));
if (!Str.consume_front("#"))
return;
Str = Str.trim();
if (!Str.consume_front("pragma"))
return;
Str = Str.trim();
if (Str.starts_with("once"))
Check->diag(Loc,
"avoid 'pragma once' directive; use include guards instead");
}
private:
AvoidPragmaOnceCheck *Check;
const SourceManager &SM;
};
void AvoidPragmaOnceCheck::registerPPCallbacks(const SourceManager &SM,
Preprocessor *PP,
Preprocessor *ModuleExpanderPP) {
PP->addPPCallbacks(std::make_unique<PragmaOnceCallbacks>(this, SM));
}
} // namespace clang::tidy::portability
|
