1 //===--- ConstantInitBuilder.cpp - Global initializer builder -------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines out-of-line routines for building initializers for
11 // global variables, in particular the kind of globals that are implicitly
12 // introduced by various language ABIs.
14 //===----------------------------------------------------------------------===//
16 #include "clang/CodeGen/ConstantInitBuilder.h"
17 #include "CodeGenModule.h"
19 using namespace clang;
20 using namespace CodeGen;
22 llvm::Type *ConstantInitFuture::getType() const {
23 assert(Data && "dereferencing null future");
24 if (Data.is<llvm::Constant*>()) {
25 return Data.get<llvm::Constant*>()->getType();
27 return Data.get<ConstantInitBuilderBase*>()->Buffer[0]->getType();
31 void ConstantInitFuture::abandon() {
32 assert(Data && "abandoning null future");
33 if (auto builder = Data.dyn_cast<ConstantInitBuilderBase*>()) {
39 void ConstantInitFuture::installInGlobal(llvm::GlobalVariable *GV) {
40 assert(Data && "installing null future");
41 if (Data.is<llvm::Constant*>()) {
42 GV->setInitializer(Data.get<llvm::Constant*>());
44 auto &builder = *Data.get<ConstantInitBuilderBase*>();
45 assert(builder.Buffer.size() == 1);
46 builder.setGlobalInitializer(GV, builder.Buffer[0]);
47 builder.Buffer.clear();
53 ConstantInitBuilderBase::createFuture(llvm::Constant *initializer) {
54 assert(Buffer.empty() && "buffer not current empty");
55 Buffer.push_back(initializer);
56 return ConstantInitFuture(this);
59 // Only used in this file.
60 inline ConstantInitFuture::ConstantInitFuture(ConstantInitBuilderBase *builder)
62 assert(!builder->Frozen);
63 assert(builder->Buffer.size() == 1);
64 assert(builder->Buffer[0] != nullptr);
67 llvm::GlobalVariable *
68 ConstantInitBuilderBase::createGlobal(llvm::Constant *initializer,
69 const llvm::Twine &name,
72 llvm::GlobalValue::LinkageTypes linkage,
73 unsigned addressSpace) {
74 auto GV = new llvm::GlobalVariable(CGM.getModule(),
75 initializer->getType(),
80 /*insert before*/ nullptr,
81 llvm::GlobalValue::NotThreadLocal,
83 GV->setAlignment(alignment.getQuantity());
84 resolveSelfReferences(GV);
88 void ConstantInitBuilderBase::setGlobalInitializer(llvm::GlobalVariable *GV,
89 llvm::Constant *initializer){
90 GV->setInitializer(initializer);
92 if (!SelfReferences.empty())
93 resolveSelfReferences(GV);
96 void ConstantInitBuilderBase::resolveSelfReferences(llvm::GlobalVariable *GV) {
97 for (auto &entry : SelfReferences) {
98 llvm::Constant *resolvedReference =
99 llvm::ConstantExpr::getInBoundsGetElementPtr(
100 GV->getValueType(), GV, entry.Indices);
101 auto dummy = entry.Dummy;
102 dummy->replaceAllUsesWith(resolvedReference);
103 dummy->eraseFromParent();
105 SelfReferences.clear();
108 void ConstantInitBuilderBase::abandon(size_t newEnd) {
109 // Remove all the entries we've added.
110 Buffer.erase(Buffer.begin() + newEnd, Buffer.end());
112 // If we're abandoning all the way to the beginning, destroy
113 // all the self-references, because we might not get another
116 for (auto &entry : SelfReferences) {
117 auto dummy = entry.Dummy;
118 dummy->replaceAllUsesWith(llvm::UndefValue::get(dummy->getType()));
119 dummy->eraseFromParent();
121 SelfReferences.clear();
125 void ConstantAggregateBuilderBase::addSize(CharUnits size) {
126 add(Builder.CGM.getSize(size));
130 ConstantAggregateBuilderBase::getRelativeOffset(llvm::IntegerType *offsetType,
131 llvm::Constant *target) {
132 // Compute the address of the relative-address slot.
133 auto base = getAddrOfCurrentPosition(offsetType);
136 base = llvm::ConstantExpr::getPtrToInt(base, Builder.CGM.IntPtrTy);
137 target = llvm::ConstantExpr::getPtrToInt(target, Builder.CGM.IntPtrTy);
138 llvm::Constant *offset = llvm::ConstantExpr::getSub(target, base);
140 // Truncate to the relative-address type if necessary.
141 if (Builder.CGM.IntPtrTy != offsetType) {
142 offset = llvm::ConstantExpr::getTrunc(offset, offsetType);
149 ConstantAggregateBuilderBase::getAddrOfCurrentPosition(llvm::Type *type) {
150 // Make a global variable. We will replace this with a GEP to this
151 // position after installing the initializer.
153 new llvm::GlobalVariable(Builder.CGM.getModule(), type, true,
154 llvm::GlobalVariable::PrivateLinkage,
156 Builder.SelfReferences.emplace_back(dummy);
157 auto &entry = Builder.SelfReferences.back();
158 (void) getGEPIndicesToCurrentPosition(entry.Indices);
162 void ConstantAggregateBuilderBase::getGEPIndicesTo(
163 llvm::SmallVectorImpl<llvm::Constant*> &indices,
164 size_t position) const {
165 // Recurse on the parent builder if present.
167 Parent->getGEPIndicesTo(indices, Begin);
169 // Otherwise, add an index to drill into the first level of pointer.
171 assert(indices.empty());
172 indices.push_back(llvm::ConstantInt::get(Builder.CGM.Int32Ty, 0));
175 assert(position >= Begin);
176 // We have to use i32 here because struct GEPs demand i32 indices.
177 // It's rather unlikely to matter in practice.
178 indices.push_back(llvm::ConstantInt::get(Builder.CGM.Int32Ty,
182 ConstantAggregateBuilderBase::PlaceholderPosition
183 ConstantAggregateBuilderBase::addPlaceholderWithSize(llvm::Type *type) {
184 // Bring the offset up to the last field.
185 CharUnits offset = getNextOffsetFromGlobal();
187 // Create the placeholder.
188 auto position = addPlaceholder();
190 // Advance the offset past that field.
191 auto &layout = Builder.CGM.getDataLayout();
193 offset = offset.alignTo(CharUnits::fromQuantity(
194 layout.getABITypeAlignment(type)));
195 offset += CharUnits::fromQuantity(layout.getTypeStoreSize(type));
197 CachedOffsetEnd = Builder.Buffer.size();
198 CachedOffsetFromGlobal = offset;
203 CharUnits ConstantAggregateBuilderBase::getOffsetFromGlobalTo(size_t end) const{
204 size_t cacheEnd = CachedOffsetEnd;
205 assert(cacheEnd <= end);
207 // Fast path: if the cache is valid, just use it.
208 if (cacheEnd == end) {
209 return CachedOffsetFromGlobal;
212 // If the cached range ends before the index at which the current
213 // aggregate starts, recurse for the parent.
215 if (cacheEnd < Begin) {
216 assert(cacheEnd == 0);
217 assert(Parent && "Begin != 0 for root builder");
219 offset = Parent->getOffsetFromGlobalTo(Begin);
221 offset = CachedOffsetFromGlobal;
224 // Perform simple layout on the elements in cacheEnd..<end.
225 if (cacheEnd != end) {
226 auto &layout = Builder.CGM.getDataLayout();
228 llvm::Constant *element = Builder.Buffer[cacheEnd];
229 assert(element != nullptr &&
230 "cannot compute offset when a placeholder is present");
231 llvm::Type *elementType = element->getType();
233 offset = offset.alignTo(CharUnits::fromQuantity(
234 layout.getABITypeAlignment(elementType)));
235 offset += CharUnits::fromQuantity(layout.getTypeStoreSize(elementType));
236 } while (++cacheEnd != end);
240 CachedOffsetEnd = cacheEnd;
241 CachedOffsetFromGlobal = offset;
245 llvm::Constant *ConstantAggregateBuilderBase::finishArray(llvm::Type *eltTy) {
248 auto &buffer = getBuffer();
249 assert((Begin < buffer.size() ||
250 (Begin == buffer.size() && eltTy))
251 && "didn't add any array elements without element type");
252 auto elts = llvm::makeArrayRef(buffer).slice(Begin);
253 if (!eltTy) eltTy = elts[0]->getType();
254 auto type = llvm::ArrayType::get(eltTy, elts.size());
255 auto constant = llvm::ConstantArray::get(type, elts);
256 buffer.erase(buffer.begin() + Begin, buffer.end());
261 ConstantAggregateBuilderBase::finishStruct(llvm::StructType *ty) {
264 auto &buffer = getBuffer();
265 auto elts = llvm::makeArrayRef(buffer).slice(Begin);
267 if (ty == nullptr && elts.empty())
268 ty = llvm::StructType::get(Builder.CGM.getLLVMContext(), {}, Packed);
270 llvm::Constant *constant;
272 assert(ty->isPacked() == Packed);
273 constant = llvm::ConstantStruct::get(ty, elts);
275 constant = llvm::ConstantStruct::getAnon(elts, Packed);
278 buffer.erase(buffer.begin() + Begin, buffer.end());