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)