mirror of
https://github.com/RPCSX/llvm.git
synced 2024-11-26 05:00:26 +00:00
0b8c9a80f2
into their new header subdirectory: include/llvm/IR. This matches the directory structure of lib, and begins to correct a long standing point of file layout clutter in LLVM. There are still more header files to move here, but I wanted to handle them in separate commits to make tracking what files make sense at each layer easier. The only really questionable files here are the target intrinsic tablegen files. But that's a battle I'd rather not fight today. I've updated both CMake and Makefile build systems (I think, and my tests think, but I may have missed something). I've also re-sorted the includes throughout the project. I'll be committing updates to Clang, DragonEgg, and Polly momentarily. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@171366 91177308-0d34-0410-b5e6-96231b3b80d8
125 lines
4.0 KiB
C++
125 lines
4.0 KiB
C++
//===-- BlackList.cpp - blacklist for sanitizers --------------------------===//
|
|
//
|
|
// 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 or global
|
|
// variables based on a user-supplied blacklist.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "BlackList.h"
|
|
#include "llvm/ADT/OwningPtr.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/IR/DerivedTypes.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/Regex.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/system_error.h"
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
namespace llvm {
|
|
|
|
BlackList::BlackList(const StringRef Path) {
|
|
// Validate and open blacklist file.
|
|
if (Path.empty()) return;
|
|
OwningPtr<MemoryBuffer> File;
|
|
if (error_code EC = MemoryBuffer::getFile(Path, File)) {
|
|
report_fatal_error("Can't open blacklist file: " + Path + ": " +
|
|
EC.message());
|
|
}
|
|
|
|
// Iterate through each line in the blacklist file.
|
|
SmallVector<StringRef, 16> Lines;
|
|
SplitString(File.take()->getBuffer(), Lines, "\n\r");
|
|
StringMap<std::string> Regexps;
|
|
for (SmallVector<StringRef, 16>::iterator I = Lines.begin(), E = Lines.end();
|
|
I != E; ++I) {
|
|
// Ignore empty lines and lines starting with "#"
|
|
if (I->empty() || I->startswith("#"))
|
|
continue;
|
|
// Get our prefix and unparsed regexp.
|
|
std::pair<StringRef, StringRef> SplitLine = I->split(":");
|
|
StringRef Prefix = SplitLine.first;
|
|
std::string Regexp = SplitLine.second;
|
|
if (Regexp.empty()) {
|
|
// Missing ':' in the line.
|
|
report_fatal_error("malformed blacklist line: " + SplitLine.first);
|
|
}
|
|
|
|
// Replace * with .*
|
|
for (size_t pos = 0; (pos = Regexp.find("*", pos)) != std::string::npos;
|
|
pos += strlen(".*")) {
|
|
Regexp.replace(pos, strlen("*"), ".*");
|
|
}
|
|
|
|
// Check that the regexp is valid.
|
|
Regex CheckRE(Regexp);
|
|
std::string Error;
|
|
if (!CheckRE.isValid(Error)) {
|
|
report_fatal_error("malformed blacklist regex: " + SplitLine.second +
|
|
": " + Error);
|
|
}
|
|
|
|
// Add this regexp into the proper group by its prefix.
|
|
if (!Regexps[Prefix].empty())
|
|
Regexps[Prefix] += "|";
|
|
Regexps[Prefix] += Regexp;
|
|
}
|
|
|
|
// Iterate through each of the prefixes, and create Regexs for them.
|
|
for (StringMap<std::string>::iterator I = Regexps.begin(), E = Regexps.end();
|
|
I != E; ++I) {
|
|
Entries[I->getKey()] = new Regex(I->getValue());
|
|
}
|
|
}
|
|
|
|
bool BlackList::isIn(const Function &F) {
|
|
return isIn(*F.getParent()) || inSection("fun", F.getName());
|
|
}
|
|
|
|
bool BlackList::isIn(const GlobalVariable &G) {
|
|
return isIn(*G.getParent()) || inSection("global", G.getName());
|
|
}
|
|
|
|
bool BlackList::isIn(const Module &M) {
|
|
return inSection("src", M.getModuleIdentifier());
|
|
}
|
|
|
|
static StringRef GetGVTypeString(const GlobalVariable &G) {
|
|
// Types of GlobalVariables are always pointer types.
|
|
Type *GType = G.getType()->getElementType();
|
|
// For now we support blacklisting struct types only.
|
|
if (StructType *SGType = dyn_cast<StructType>(GType)) {
|
|
if (!SGType->isLiteral())
|
|
return SGType->getName();
|
|
}
|
|
return "<unknown type>";
|
|
}
|
|
|
|
bool BlackList::isInInit(const GlobalVariable &G) {
|
|
return (isIn(*G.getParent()) ||
|
|
inSection("global-init", G.getName()) ||
|
|
inSection("global-init-type", GetGVTypeString(G)));
|
|
}
|
|
|
|
bool BlackList::inSection(const StringRef Section, const StringRef Query) {
|
|
StringMap<Regex*>::iterator I = Entries.find(Section);
|
|
if (I == Entries.end()) return false;
|
|
|
|
Regex *FunctionRegex = I->getValue();
|
|
return FunctionRegex->match(Query);
|
|
}
|
|
|
|
} // namespace llvm
|