1 //===--- SanitizerSpecialCaseList.cpp - SCL for sanitizers ----------------===//
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 // An extension of SpecialCaseList to allowing querying sections by
10 // SanitizerMask.
11 //
12 //===----------------------------------------------------------------------===//
13 #include "clang/Basic/SanitizerSpecialCaseList.h"
14 
15 using namespace clang;
16 
17 std::unique_ptr<SanitizerSpecialCaseList>
create(const std::vector<std::string> & Paths,llvm::vfs::FileSystem & VFS,std::string & Error)18 SanitizerSpecialCaseList::create(const std::vector<std::string> &Paths,
19                                  llvm::vfs::FileSystem &VFS,
20                                  std::string &Error) {
21   std::unique_ptr<clang::SanitizerSpecialCaseList> SSCL(
22       new SanitizerSpecialCaseList());
23   if (SSCL->createInternal(Paths, VFS, Error)) {
24     SSCL->createSanitizerSections();
25     return SSCL;
26   }
27   return nullptr;
28 }
29 
30 std::unique_ptr<SanitizerSpecialCaseList>
createOrDie(const std::vector<std::string> & Paths,llvm::vfs::FileSystem & VFS)31 SanitizerSpecialCaseList::createOrDie(const std::vector<std::string> &Paths,
32                                       llvm::vfs::FileSystem &VFS) {
33   std::string Error;
34   if (auto SSCL = create(Paths, VFS, Error))
35     return SSCL;
36   llvm::report_fatal_error(StringRef(Error));
37 }
38 
createSanitizerSections()39 void SanitizerSpecialCaseList::createSanitizerSections() {
40   for (auto &It : Sections) {
41     auto &S = It.second;
42     SanitizerMask Mask;
43 
44 #define SANITIZER(NAME, ID)                                                    \
45   if (S.SectionMatcher->match(NAME))                                           \
46     Mask |= SanitizerKind::ID;
47 #define SANITIZER_GROUP(NAME, ID, ALIAS) SANITIZER(NAME, ID)
48 
49 #include "clang/Basic/Sanitizers.def"
50 #undef SANITIZER
51 #undef SANITIZER_GROUP
52 
53     SanitizerSections.emplace_back(Mask, S.Entries);
54   }
55 }
56 
inSection(SanitizerMask Mask,StringRef Prefix,StringRef Query,StringRef Category) const57 bool SanitizerSpecialCaseList::inSection(SanitizerMask Mask, StringRef Prefix,
58                                          StringRef Query,
59                                          StringRef Category) const {
60   for (auto &S : SanitizerSections)
61     if ((S.Mask & Mask) &&
62         SpecialCaseList::inSectionBlame(S.Entries, Prefix, Query, Category))
63       return true;
64 
65   return false;
66 }
67