Fix ASan init function detection after clang r208128.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@208141 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Nico Weber 2014-05-06 23:17:26 +00:00
parent 26668d093e
commit 9e9ec9b69c

View File

@ -846,8 +846,29 @@ void AddressSanitizer::instrumentAddress(Instruction *OrigIns,
void AddressSanitizerModule::createInitializerPoisonCalls(
Module &M, GlobalValue *ModuleName) {
// We do all of our poisoning and unpoisoning within _GLOBAL__I_a.
Function *GlobalInit = M.getFunction("_GLOBAL__I_a");
// We do all of our poisoning and unpoisoning within a global constructor.
// These are called _GLOBAL__(sub_)?I_.*.
// TODO: Consider looking through the functions in
// M.getGlobalVariable("llvm.global_ctors") instead of using this stringly
// typed approach.
Function *GlobalInit = nullptr;
for (auto &F : M.getFunctionList()) {
StringRef FName = F.getName();
const char kGlobalPrefix[] = "_GLOBAL__";
if (!FName.startswith(kGlobalPrefix))
continue;
FName = FName.substr(strlen(kGlobalPrefix));
const char kOptionalSub[] = "sub_";
if (FName.startswith(kOptionalSub))
FName = FName.substr(strlen(kOptionalSub));
if (FName.startswith("I_")) {
GlobalInit = &F;
break;
}
}
// If that function is not present, this TU contains no globals, or they have
// all been optimized away
if (!GlobalInit)
@ -862,7 +883,7 @@ void AddressSanitizerModule::createInitializerPoisonCalls(
// Add calls to unpoison all globals before each return instruction.
for (Function::iterator I = GlobalInit->begin(), E = GlobalInit->end();
I != E; ++I) {
I != E; ++I) {
if (ReturnInst *RI = dyn_cast<ReturnInst>(I->getTerminator())) {
CallInst::Create(AsanUnpoisonGlobals, "", RI);
}