mirror of
https://github.com/c64scene-ar/llvm-6502.git
synced 2024-09-06 09:54:49 +00:00
d04a8d4b33
Sooooo many of these had incorrect or strange main module includes. I have manually inspected all of these, and fixed the main module include to be the nearest plausible thing I could find. If you own or care about any of these source files, I encourage you to take some time and check that these edits were sensible. I can't have broken anything (I strictly added headers, and reordered them, never removed), but they may not be the headers you'd really like to identify as containing the API being implemented. Many forward declarations and missing includes were added to a header files to allow them to parse cleanly when included first. The main module rule does in fact have its merits. =] git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@169131 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/DerivedTypes.h"
|
|
#include "llvm/Function.h"
|
|
#include "llvm/GlobalVariable.h"
|
|
#include "llvm/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
|