blob: 188ea4d9b3cbf5e04b6703f82de6d0bea82f69d9 (
plain)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
//===-- FunctionBlackList.cpp - blacklist of functions --------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This is a utility class for instrumentation passes (like AddressSanitizer
// or ThreadSanitizer) to avoid instrumenting some functions based on
// user-supplied blacklist.
//
//===----------------------------------------------------------------------===//
#include "FunctionBlackList.h"
#include "llvm/ADT/OwningPtr.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/Function.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/Regex.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/Support/system_error.h"
namespace llvm {
FunctionBlackList::FunctionBlackList(const std::string &Path) {
Functions = NULL;
const char *kFunPrefix = "fun:";
if (!Path.size()) return;
std::string Fun;
OwningPtr<MemoryBuffer> File;
if (error_code EC = MemoryBuffer::getFile(Path.c_str(), File)) {
report_fatal_error("Can't open blacklist file " + Path + ": " +
EC.message());
}
MemoryBuffer *Buff = File.take();
const char *Data = Buff->getBufferStart();
size_t DataLen = Buff->getBufferSize();
SmallVector<StringRef, 16> Lines;
SplitString(StringRef(Data, DataLen), Lines, "\n\r");
for (size_t i = 0, numLines = Lines.size(); i < numLines; i++) {
if (Lines[i].startswith(kFunPrefix)) {
std::string ThisFunc = Lines[i].substr(strlen(kFunPrefix));
std::string ThisFuncRE;
// add ThisFunc replacing * with .*
for (size_t j = 0, n = ThisFunc.size(); j < n; j++) {
if (ThisFunc[j] == '*')
ThisFuncRE += '.';
ThisFuncRE += ThisFunc[j];
}
// Check that the regexp is valid.
Regex CheckRE(ThisFuncRE);
std::string Error;
if (!CheckRE.isValid(Error))
report_fatal_error("malformed blacklist regex: " + ThisFunc +
": " + Error);
// Append to the final regexp.
if (Fun.size())
Fun += "|";
Fun += ThisFuncRE;
}
}
if (Fun.size()) {
Functions = new Regex(Fun);
}
}
bool FunctionBlackList::isIn(const Function &F) {
if (Functions) {
bool Res = Functions->match(F.getName());
return Res;
}
return false;
}
} // namespace llvm
|