diff options
author | Peter Collingbourne <peter@pcc.me.uk> | 2013-08-22 20:08:15 +0000 |
---|---|---|
committer | Peter Collingbourne <peter@pcc.me.uk> | 2013-08-22 20:08:15 +0000 |
commit | 054cec05b84e878a68e7ecc71342312e76850649 (patch) | |
tree | 7b4a1bded7e49ab644a0effef0d3b76013c537fa /lib/Transforms/Instrumentation | |
parent | 4f68e9ea8e6a9d98b60bbdde719dcb9d68991980 (diff) | |
download | llvm-054cec05b84e878a68e7ecc71342312e76850649.tar.gz llvm-054cec05b84e878a68e7ecc71342312e76850649.tar.bz2 llvm-054cec05b84e878a68e7ecc71342312e76850649.tar.xz |
DataFlowSanitizer: Replace non-instrumented aliases of instrumented functions, and vice versa, with wrappers.
Differential Revision: http://llvm-reviews.chandlerc.com/D1442
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@189054 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib/Transforms/Instrumentation')
-rw-r--r-- | lib/Transforms/Instrumentation/DataFlowSanitizer.cpp | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp b/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp index e92d88de1d..1bf6b99ea9 100644 --- a/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp +++ b/lib/Transforms/Instrumentation/DataFlowSanitizer.cpp @@ -470,7 +470,8 @@ bool DataFlowSanitizer::runOnModule(Module &M) { FnsToInstrument.push_back(&*i); } - // Give function aliases prefixes when necessary. + // Give function aliases prefixes when necessary, and build wrappers where the + // instrumentedness is inconsistent. for (Module::alias_iterator i = M.alias_begin(), e = M.alias_end(); i != e;) { GlobalAlias *GA = &*i; ++i; @@ -481,6 +482,16 @@ bool DataFlowSanitizer::runOnModule(Module &M) { bool GAInst = isInstrumented(GA), FInst = isInstrumented(F); if (GAInst && FInst) { addGlobalNamePrefix(GA); + } else if (GAInst != FInst) { + // Non-instrumented alias of an instrumented function, or vice versa. + // Replace the alias with a native-ABI wrapper of the aliasee. The pass + // below will take care of instrumenting it. + Function *NewF = + buildWrapperFunction(F, "", GA->getLinkage(), F->getFunctionType()); + GA->replaceAllUsesWith(NewF); + NewF->takeName(GA); + GA->eraseFromParent(); + FnsToInstrument.push_back(NewF); } } } |