[asan] call __asan_init from .preinit_array. This simplifies __asan_init vs malloc...

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@146284 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/Transforms/Instrumentation/AddressSanitizer.cpp b/lib/Transforms/Instrumentation/AddressSanitizer.cpp
index 83be31eada626c88f568808822b5efd1bff0ad3f..fdea1b59466e925ea76cddf83fdd87907610a9e6 100644 (file)
--- a/lib/Transforms/Instrumentation/AddressSanitizer.cpp
+++ b/lib/Transforms/Instrumentation/AddressSanitizer.cpp
@@ -163,6 +163,8 @@ struct AddressSanitizer : public ModulePass {
 
  private:
 
+  void appendToPreinitArray(Module &M, Function *F);
+
   uint64_t getAllocaSizeInBytes(AllocaInst *AI) {
     Type *Ty = AI->getAllocatedType();
     uint64_t SizeInBytes = TD->getTypeStoreSizeInBits(Ty) / 8;
@@ -563,6 +565,18 @@ bool AddressSanitizer::insertGlobalRedzones(Module &M) {
   return true;
 }
 
+// .preinit_array is something that hapens before all other inits.
+// On systems where .preinit_array is honored, we will call __asan_init early.
+// On other systems this will make no effect.
+void AddressSanitizer::appendToPreinitArray(Module &M, Function *F) {
+  IRBuilder<> IRB(M.getContext());
+  GlobalVariable *Var =
+      new GlobalVariable(M, PointerType::getUnqual(F->getFunctionType()),
+                         false, GlobalValue::PrivateLinkage,
+                         F, "__asan_preinit_private");
+  Var->setSection(".preinit_array");
+}
+
 // virtual
 bool AddressSanitizer::runOnModule(Module &M) {
   // Initialize the private fields. No one has accessed them before.
@@ -633,6 +647,7 @@ bool AddressSanitizer::runOnModule(Module &M) {
   }
 
   appendToGlobalCtors(M, AsanCtorFunction, 1 /*high priority*/);
+  appendToPreinitArray(M, AsanInitFunction);
 
   return Res;
 }