1 //===-- WebAssemblyLowerGlobalDtors.cpp - Lower @llvm.global_dtors --------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 ///
10 /// \file
11 /// Lower @llvm.global_dtors.
12 ///
13 /// WebAssembly doesn't have a builtin way to invoke static destructors.
14 /// Implement @llvm.global_dtors by creating wrapper functions that are
15 /// registered in @llvm.global_ctors and which contain a call to
16 /// `__cxa_atexit` to register their destructor functions.
17 ///
18 //===----------------------------------------------------------------------===//
19 
20 #include "WebAssembly.h"
21 #include "llvm/ADT/MapVector.h"
22 #include "llvm/IR/Constants.h"
23 #include "llvm/IR/Instructions.h"
24 #include "llvm/IR/Intrinsics.h"
25 #include "llvm/IR/Module.h"
26 #include "llvm/Pass.h"
27 #include "llvm/Support/Debug.h"
28 #include "llvm/Support/raw_ostream.h"
29 #include "llvm/Transforms/Utils/ModuleUtils.h"
30 using namespace llvm;
31 
32 #define DEBUG_TYPE "wasm-lower-global-dtors"
33 
34 namespace {
35 class LowerGlobalDtors final : public ModulePass {
getPassName() const36   StringRef getPassName() const override {
37     return "WebAssembly Lower @llvm.global_dtors";
38   }
39 
getAnalysisUsage(AnalysisUsage & AU) const40   void getAnalysisUsage(AnalysisUsage &AU) const override {
41     AU.setPreservesCFG();
42     ModulePass::getAnalysisUsage(AU);
43   }
44 
45   bool runOnModule(Module &M) override;
46 
47 public:
48   static char ID;
LowerGlobalDtors()49   LowerGlobalDtors() : ModulePass(ID) {}
50 };
51 } // End anonymous namespace
52 
53 char LowerGlobalDtors::ID = 0;
54 INITIALIZE_PASS(LowerGlobalDtors, DEBUG_TYPE,
55                 "Lower @llvm.global_dtors for WebAssembly", false, false)
56 
createWebAssemblyLowerGlobalDtors()57 ModulePass *llvm::createWebAssemblyLowerGlobalDtors() {
58   return new LowerGlobalDtors();
59 }
60 
runOnModule(Module & M)61 bool LowerGlobalDtors::runOnModule(Module &M) {
62   LLVM_DEBUG(dbgs() << "********** Lower Global Destructors **********\n");
63 
64   GlobalVariable *GV = M.getGlobalVariable("llvm.global_dtors");
65   if (!GV)
66     return false;
67 
68   const ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
69   if (!InitList)
70     return false;
71 
72   // Sanity-check @llvm.global_dtor's type.
73   StructType *ETy = dyn_cast<StructType>(InitList->getType()->getElementType());
74   if (!ETy || ETy->getNumElements() != 3 ||
75       !ETy->getTypeAtIndex(0U)->isIntegerTy() ||
76       !ETy->getTypeAtIndex(1U)->isPointerTy() ||
77       !ETy->getTypeAtIndex(2U)->isPointerTy())
78     return false; // Not (int, ptr, ptr).
79 
80   // Collect the contents of @llvm.global_dtors, collated by priority and
81   // associated symbol.
82   std::map<uint16_t, MapVector<Constant *, std::vector<Constant *>>> DtorFuncs;
83   for (Value *O : InitList->operands()) {
84     ConstantStruct *CS = dyn_cast<ConstantStruct>(O);
85     if (!CS)
86       continue; // Malformed.
87 
88     ConstantInt *Priority = dyn_cast<ConstantInt>(CS->getOperand(0));
89     if (!Priority)
90       continue; // Malformed.
91     uint16_t PriorityValue = Priority->getLimitedValue(UINT16_MAX);
92 
93     Constant *DtorFunc = CS->getOperand(1);
94     if (DtorFunc->isNullValue())
95       break; // Found a null terminator, skip the rest.
96 
97     Constant *Associated = CS->getOperand(2);
98     Associated = cast<Constant>(Associated->stripPointerCastsNoFollowAliases());
99 
100     DtorFuncs[PriorityValue][Associated].push_back(DtorFunc);
101   }
102   if (DtorFuncs.empty())
103     return false;
104 
105   // extern "C" int __cxa_atexit(void (*f)(void *), void *p, void *d);
106   LLVMContext &C = M.getContext();
107   PointerType *VoidStar = Type::getInt8PtrTy(C);
108   Type *AtExitFuncArgs[] = {VoidStar};
109   FunctionType *AtExitFuncTy =
110       FunctionType::get(Type::getVoidTy(C), AtExitFuncArgs,
111                         /*isVarArg=*/false);
112 
113   Type *AtExitArgs[] = {PointerType::get(AtExitFuncTy, 0), VoidStar, VoidStar};
114   FunctionType *AtExitTy = FunctionType::get(Type::getInt32Ty(C), AtExitArgs,
115                                              /*isVarArg=*/false);
116   Constant *AtExit = M.getOrInsertFunction("__cxa_atexit", AtExitTy);
117 
118   // Declare __dso_local.
119   Constant *DsoHandle = M.getNamedValue("__dso_handle");
120   if (!DsoHandle) {
121     Type *DsoHandleTy = Type::getInt8Ty(C);
122     GlobalVariable *Handle = new GlobalVariable(
123         M, DsoHandleTy, /*isConstant=*/true,
124         GlobalVariable::ExternalWeakLinkage, nullptr, "__dso_handle");
125     Handle->setVisibility(GlobalVariable::HiddenVisibility);
126     DsoHandle = Handle;
127   }
128 
129   // For each unique priority level and associated symbol, generate a function
130   // to call all the destructors at that level, and a function to register the
131   // first function with __cxa_atexit.
132   for (auto &PriorityAndMore : DtorFuncs) {
133     uint16_t Priority = PriorityAndMore.first;
134     for (auto &AssociatedAndMore : PriorityAndMore.second) {
135       Constant *Associated = AssociatedAndMore.first;
136 
137       Function *CallDtors = Function::Create(
138           AtExitFuncTy, Function::PrivateLinkage,
139           "call_dtors" +
140               (Priority != UINT16_MAX ? (Twine(".") + Twine(Priority))
141                                       : Twine()) +
142               (!Associated->isNullValue() ? (Twine(".") + Associated->getName())
143                                           : Twine()),
144           &M);
145       BasicBlock *BB = BasicBlock::Create(C, "body", CallDtors);
146 
147       for (auto Dtor : AssociatedAndMore.second)
148         CallInst::Create(Dtor, "", BB);
149       ReturnInst::Create(C, BB);
150 
151       FunctionType *VoidVoid = FunctionType::get(Type::getVoidTy(C),
152                                                  /*isVarArg=*/false);
153       Function *RegisterCallDtors = Function::Create(
154           VoidVoid, Function::PrivateLinkage,
155           "register_call_dtors" +
156               (Priority != UINT16_MAX ? (Twine(".") + Twine(Priority))
157                                       : Twine()) +
158               (!Associated->isNullValue() ? (Twine(".") + Associated->getName())
159                                           : Twine()),
160           &M);
161       BasicBlock *EntryBB = BasicBlock::Create(C, "entry", RegisterCallDtors);
162       BasicBlock *FailBB = BasicBlock::Create(C, "fail", RegisterCallDtors);
163       BasicBlock *RetBB = BasicBlock::Create(C, "return", RegisterCallDtors);
164 
165       Value *Null = ConstantPointerNull::get(VoidStar);
166       Value *Args[] = {CallDtors, Null, DsoHandle};
167       Value *Res = CallInst::Create(AtExit, Args, "call", EntryBB);
168       Value *Cmp = new ICmpInst(*EntryBB, ICmpInst::ICMP_NE, Res,
169                                 Constant::getNullValue(Res->getType()));
170       BranchInst::Create(FailBB, RetBB, Cmp, EntryBB);
171 
172       // If `__cxa_atexit` hits out-of-memory, trap, so that we don't misbehave.
173       // This should be very rare, because if the process is running out of
174       // memory before main has even started, something is wrong.
175       CallInst::Create(Intrinsic::getDeclaration(&M, Intrinsic::trap), "",
176                        FailBB);
177       new UnreachableInst(C, FailBB);
178 
179       ReturnInst::Create(C, RetBB);
180 
181       // Now register the registration function with @llvm.global_ctors.
182       appendToGlobalCtors(M, RegisterCallDtors, Priority, Associated);
183     }
184   }
185 
186   // Now that we've lowered everything, remove @llvm.global_dtors.
187   GV->eraseFromParent();
188 
189   return true;
190 }
191