[asan] Initialize __asan_set_shadow_* callbacks

[vitalybuka]

Summary

Callbacks are not being used yet.

PR27453

[vitalybuka]

Test

[vitalybuka]

constant

[eugenis]

why do you want this change if the callbacks are not being used?
Dead code is bad.

test/Instrumentation/AddressSanitizer/stack-poisoning-experimental.ll

17 ↗

(On Diff #68461)

does that actually do anything without a ":" after CHECK?

[vitalybuka]

why do you want this change if the callbacks are not being used?

I am going to use in in future patches. This way we can have smaller reviews.

[vitalybuka]

fix in test

lib/Transforms/Instrumentation/AddressSanitizer.cpp

 #include "llvm/Transforms/Scalar.h"
 #include "llvm/Transforms/Utils/ASanStackFrameLayout.h"
 #include "llvm/Transforms/Utils/BasicBlockUtils.h"
 #include "llvm/Transforms/Utils/Cloning.h"
 #include "llvm/Transforms/Utils/Local.h"
 #include "llvm/Transforms/Utils/ModuleUtils.h"
 #include "llvm/Transforms/Utils/PromoteMemToReg.h"
 #include <algorithm>
+#include <cstdlib>
 #include <string>
 #include <system_error>
 
 using namespace llvm;
 
 #define DEBUG_TYPE "asan"
 
 static const uint64_t kDefaultShadowScale = 3;

 static const char *const kAsanPtrSub = "__sanitizer_ptr_sub";
 static const char *const kAsanHandleNoReturnName = "__asan_handle_no_return";
 static const int kMaxAsanStackMallocSizeClass = 10;
 static const char *const kAsanStackMallocNameTemplate = "__asan_stack_malloc_";
 static const char *const kAsanStackFreeNameTemplate = "__asan_stack_free_";
 static const char *const kAsanGenPrefix = "__asan_gen_";
 static const char *const kODRGenPrefix = "__odr_asan_gen_";
 static const char *const kSanCovGenPrefix = "__sancov_gen_";
+static const char *const kAsanSetShadow00Func = "__asan_set_shadow_00";
+static const char *const kAsanSetShadowf1Func = "__asan_set_shadow_f1";
+static const char *const kAsanSetShadowf2Func = "__asan_set_shadow_f2";
+static const char *const kAsanSetShadowf3Func = "__asan_set_shadow_f3";
+static const char *const kAsanSetShadowf5Func = "__asan_set_shadow_f5";
+static const char *const kAsanSetShadowf8Func = "__asan_set_shadow_f8";
 static const char *const kAsanPoisonStackMemoryName =
     "__asan_poison_stack_memory";
 static const char *const kAsanUnpoisonStackMemoryName =
     "__asan_unpoison_stack_memory";
 static const char *const kAsanGlobalsRegisteredFlagName =
     "__asan_globals_registered";
 
 static const char *const kAsanOptionDetectUseAfterReturn =
struct FunctionStackPoisoner : public InstVisitor<FunctionStackPoisoner> {
 
   SmallVector<AllocaInst *, 16> AllocaVec;
   SmallSetVector<AllocaInst *, 16> NonInstrumentedStaticAllocaVec;
   SmallVector<Instruction *, 8> RetVec;
   unsigned StackAlignment;
 
   Function *AsanStackMallocFunc[kMaxAsanStackMallocSizeClass + 1],
       *AsanStackFreeFunc[kMaxAsanStackMallocSizeClass + 1];
+  Function *AsanSetShadowFunc[1 << sizeof(uint8_t)] = {};
   Function *AsanPoisonStackMemoryFunc, *AsanUnpoisonStackMemoryFunc;
   Function *AsanAllocaPoisonFunc, *AsanAllocasUnpoisonFunc;
 
   // Stores a place and arguments of poisoning/unpoisoning call for alloca.
   struct AllocaPoisonCall {
     IntrinsicInst *InsBefore;
     AllocaInst *AI;
     uint64_t Size;
if (ASan.UseAfterScope) {
     AsanPoisonStackMemoryFunc = checkSanitizerInterfaceFunction(
         M.getOrInsertFunction(kAsanPoisonStackMemoryName, IRB.getVoidTy(),
                               IntptrTy, IntptrTy, nullptr));
     AsanUnpoisonStackMemoryFunc = checkSanitizerInterfaceFunction(
         M.getOrInsertFunction(kAsanUnpoisonStackMemoryName, IRB.getVoidTy(),
                               IntptrTy, IntptrTy, nullptr));
   }
 
+  for (const char *Name :
+       {kAsanSetShadow00Func, kAsanSetShadowf1Func, kAsanSetShadowf2Func,
+        kAsanSetShadowf3Func, kAsanSetShadowf5Func, kAsanSetShadowf8Func}) {
+    char *End = 0;
+    AsanSetShadowFunc[std::strtoul(Name + strlen(Name) - 2, &End, 16)] =
+        checkSanitizerInterfaceFunction(M.getOrInsertFunction(
+            Name, IRB.getVoidTy(), IntptrTy, IntptrTy, nullptr));
+  }
+
   AsanAllocaPoisonFunc = checkSanitizerInterfaceFunction(M.getOrInsertFunction(
       kAsanAllocaPoison, IRB.getVoidTy(), IntptrTy, IntptrTy, nullptr));
   AsanAllocasUnpoisonFunc =
       checkSanitizerInterfaceFunction(M.getOrInsertFunction(
           kAsanAllocasUnpoison, IRB.getVoidTy(), IntptrTy, IntptrTy, nullptr));
 }
 
 void FunctionStackPoisoner::poisonRedZones(ArrayRef<uint8_t> ShadowBytes,