maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Introduce a SpecialCaseList ctor which takes a MemoryBuffer to make 

it more unit testable, and fix memory leak in the other ctor.

Differential Revision: http://llvm-reviews.chandlerc.com/D1090

llvm-svn: 185976
This commit is contained in:
Peter Collingbourne 2013-07-09 22:03:09 +00:00
parent 015370e23a
commit 2eb048d230
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
llvm/include/llvm/Transforms/Utils/SpecialCaseList.h
View File

@ -35,6 +35,7 @@
namespace llvm {
class Function;
class GlobalVariable;
class MemoryBuffer;
class Module;
class Regex;
class StringRef;
@ -42,6 +43,8 @@ class StringRef;
class SpecialCaseList {
public:
SpecialCaseList(const StringRef Path);
SpecialCaseList(const MemoryBuffer *MB);
// Returns whether either this function or it's source file are blacklisted.
bool isIn(const Function &F) const;
// Returns whether either this global or it's source file are blacklisted.
@ -53,6 +56,7 @@ class SpecialCaseList {
private:
StringMap<Regex*> Entries;
void init(const MemoryBuffer *MB);
bool inSection(const StringRef Section, const StringRef Query) const;
};

10
llvm/lib/Transforms/Utils/SpecialCaseList.cpp
View File

@ -39,9 +39,17 @@ SpecialCaseList::SpecialCaseList(const StringRef Path) {
EC.message());
}
init(File.get());
}
SpecialCaseList::SpecialCaseList(const MemoryBuffer *MB) {
init(MB);
}
void SpecialCaseList::init(const MemoryBuffer *MB) {
// Iterate through each line in the blacklist file.
SmallVector<StringRef, 16> Lines;
SplitString(File.take()->getBuffer(), Lines, "\n\r");
SplitString(MB->getBuffer(), Lines, "\n\r");
StringMap<std::string> Regexps;
for (SmallVectorImpl<StringRef>::iterator I = Lines.begin(), E = Lines.end();
I != E; ++I) {