LLVM Passe which fixes weak aliases overriden from bitcode
This commit is contained in:
parent
f4a2713ac8
commit
c03229e600
2 changed files with 112 additions and 0 deletions
26
external/bsd/llvm/passes/WeakAliasModuleOverride/Makefile
vendored
Normal file
26
external/bsd/llvm/passes/WeakAliasModuleOverride/Makefile
vendored
Normal file
|
@ -0,0 +1,26 @@
|
|||
QUIET=@
|
||||
ECHO=echo
|
||||
CP=cp
|
||||
|
||||
PASSLIBNAME:= WeakAliasModuleOverride.so
|
||||
|
||||
LLVM_VERSION = $($LLVMPREFIX/bin/llvm-config --version | sed "s/[^0-9]//g")
|
||||
CFLAGS += -Wno-long-long -Wall -W -Wno-unused-parameter -Wwrite-strings -DHAVE_EXCEPTIONS=0
|
||||
CFLAGS += $(shell $(LLVMPREFIX)/bin/llvm-config --cxxflags) -g -DLLVM_VERSION=$(LLVM_VERSION)
|
||||
LDFLAGS += $(shell $(LLVMPREFIX)/bin/llvm-config --ldflags)
|
||||
OBJS= WeakAliasModuleOverride.o
|
||||
|
||||
$(PASSLIBNAME): $(OBJS)
|
||||
$(QUIET) $(ECHO) " [LINK] $@"
|
||||
$(QUIET) $(CXX) $(CFLAGS) -shared -o $@ $(CPPS) $(OBJS) $(LDFLAGS) $(LIBS)
|
||||
|
||||
%.o: %.cpp $(HEADERS)
|
||||
$(QUIET) $(ECHO) " [C++] $<"
|
||||
$(QUIET) $(CXX) $(CFLAGS) $(INCLUDES) -c -o $@ $<
|
||||
|
||||
install: $(PASSLIBNAME)
|
||||
$(QUIET) -mkdir -p ../lib
|
||||
$(QUIET) $(CP) $(PASSLIBNAME) ../lib/$(PASSLIBNAME)
|
||||
|
||||
clean:
|
||||
-rm -f *.o *.so
|
86
external/bsd/llvm/passes/WeakAliasModuleOverride/WeakAliasModuleOverride.cpp
vendored
Normal file
86
external/bsd/llvm/passes/WeakAliasModuleOverride/WeakAliasModuleOverride.cpp
vendored
Normal file
|
@ -0,0 +1,86 @@
|
|||
|
||||
#include "llvm/IR/GlobalVariable.h"
|
||||
#include "llvm/IR/Module.h"
|
||||
#include "llvm/Pass.h"
|
||||
|
||||
#include "llvm/Support/raw_ostream.h"
|
||||
#include "llvm/Support/Regex.h"
|
||||
|
||||
using namespace llvm;
|
||||
|
||||
//#define passLog(M) (errs() << "WeakAliasModuleOverride: " << M << "\n")
|
||||
#define passLog(M) /* nothing */
|
||||
|
||||
namespace {
|
||||
class WeakAliasModuleOverride : public ModulePass {
|
||||
|
||||
public:
|
||||
static char ID;
|
||||
WeakAliasModuleOverride() : ModulePass(ID) {
|
||||
}
|
||||
|
||||
virtual bool runOnModule(Module &M) {
|
||||
const Module::FunctionListType &listFcts = M.getFunctionList();
|
||||
const Module::GlobalListType &listGlobalVars = M.getGlobalList();
|
||||
|
||||
std::string Asm = M.getModuleInlineAsm();
|
||||
|
||||
passLog("ASM START\n"
|
||||
<< Asm
|
||||
<< "ASM END\n");
|
||||
|
||||
// Filter out Function symbols
|
||||
for(Module::const_iterator it = listFcts.begin(), end=listFcts.end(); it!= end; ++it)
|
||||
{
|
||||
if (it->isDeclaration())
|
||||
continue;
|
||||
|
||||
// Filter out the assembly weak symbol as well as its default value
|
||||
std::string symbolName = it->getName();
|
||||
std::string matchWeak = "(^.weak.* " + symbolName + "\n)";
|
||||
std::string matchAssignement = "(^" + symbolName + " .*=.*\n)";
|
||||
|
||||
Regex filterWeak(matchWeak, Regex::Newline);
|
||||
Regex filterAssignement(matchAssignement, Regex::Newline);
|
||||
|
||||
while(filterWeak.match(Asm))
|
||||
Asm = filterWeak.sub("", Asm);
|
||||
|
||||
while(filterAssignement.match(Asm))
|
||||
Asm = filterAssignement.sub("", Asm);
|
||||
}
|
||||
|
||||
// Filter out GlobalVariable symbols
|
||||
for(Module::const_global_iterator it = listGlobalVars.begin(), end=listGlobalVars.end(); it!= end; ++it)
|
||||
{
|
||||
if (! it->hasInitializer())
|
||||
continue;
|
||||
|
||||
// Filter out the assembly weak symbol as well as its default value
|
||||
std::string symbolName = it->getName();
|
||||
std::string matchWeak = "(^.weak.* " + symbolName + "\n)";
|
||||
std::string matchAssignement = "(^" + symbolName + " .*=.*\n)";
|
||||
|
||||
Regex filterWeak(matchWeak, Regex::Newline);
|
||||
Regex filterAssignement(matchAssignement, Regex::Newline);
|
||||
|
||||
while(filterWeak.match(Asm))
|
||||
Asm = filterWeak.sub("", Asm);
|
||||
|
||||
while(filterAssignement.match(Asm))
|
||||
Asm = filterAssignement.sub("", Asm);
|
||||
}
|
||||
|
||||
M.setModuleInlineAsm(Asm);
|
||||
|
||||
passLog("ASM START - registered\n"
|
||||
<< M.getModuleInlineAsm()
|
||||
<< "ASM END\n");
|
||||
|
||||
return true;
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
char WeakAliasModuleOverride::ID = 0;
|
||||
RegisterPass<WeakAliasModuleOverride> WEAK_ALIAS_MODULE_OVERRIDE("weak-alias-module-override", "Fix Weak Alias overrides");
|
Loading…
Reference in a new issue