blob: 98a8aaaf86d4d9f07c6cd15673ba3377a3938d35 [file] [log] [blame]
Reid Spencer5f016e22007-07-11 17:01:13 +00001//===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Reid Spencer5f016e22007-07-11 17:01:13 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This coordinates the per-module state used while generating code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenModule.h"
15#include "CodeGenFunction.h"
16#include "clang/AST/ASTContext.h"
17#include "clang/AST/Decl.h"
Chris Lattner2c8569d2007-12-02 07:19:18 +000018#include "clang/Basic/Diagnostic.h"
Chris Lattner45e8cbd2007-11-28 05:34:05 +000019#include "clang/Basic/LangOptions.h"
Nate Begeman8bd4afe2008-04-19 04:17:09 +000020#include "clang/Basic/SourceManager.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000021#include "clang/Basic/TargetInfo.h"
Nate Begemanec9426c2008-03-09 03:09:36 +000022#include "llvm/CallingConv.h"
Chris Lattner8f32f712007-07-14 00:23:28 +000023#include "llvm/Constants.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000024#include "llvm/DerivedTypes.h"
Chris Lattnerbef20ac2007-08-31 04:31:45 +000025#include "llvm/Module.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000026#include "llvm/Intrinsics.h"
Christopher Lambce39faa2007-12-02 08:49:54 +000027#include <algorithm>
Reid Spencer5f016e22007-07-11 17:01:13 +000028using namespace clang;
29using namespace CodeGen;
30
31
Chris Lattner45e8cbd2007-11-28 05:34:05 +000032CodeGenModule::CodeGenModule(ASTContext &C, const LangOptions &LO,
Chris Lattnerfb97b032007-12-02 01:40:18 +000033 llvm::Module &M, const llvm::TargetData &TD,
34 Diagnostic &diags)
35 : Context(C), Features(LO), TheModule(M), TheTargetData(TD), Diags(diags),
Chris Lattner2b94fe32008-03-01 08:45:05 +000036 Types(C, M, TD), MemCpyFn(0), MemSetFn(0), CFConstantStringClassRef(0) {
37 //TODO: Make this selectable at runtime
Chris Lattner391d77a2008-03-30 23:03:07 +000038 Runtime = CreateObjCRuntime(M,
39 getTypes().ConvertType(getContext().IntTy),
40 getTypes().ConvertType(getContext().LongTy));
Chris Lattner2b94fe32008-03-01 08:45:05 +000041}
42
43CodeGenModule::~CodeGenModule() {
Chris Lattner391d77a2008-03-30 23:03:07 +000044 llvm::Function *ObjCInitFunction = Runtime->ModuleInitFunction();
Chris Lattnerce5605e2008-03-30 23:25:33 +000045 if (ObjCInitFunction)
Chris Lattner391d77a2008-03-30 23:03:07 +000046 AddGlobalCtor(ObjCInitFunction);
Nate Begeman4c13b7a2008-04-20 06:29:50 +000047 EmitStatics();
Chris Lattner6d397602008-03-14 17:18:18 +000048 EmitGlobalCtors();
Nate Begeman532485c2008-04-18 23:43:57 +000049 EmitAnnotations();
Chris Lattner2b94fe32008-03-01 08:45:05 +000050 delete Runtime;
51}
Reid Spencer5f016e22007-07-11 17:01:13 +000052
Chris Lattner2c8569d2007-12-02 07:19:18 +000053/// WarnUnsupported - Print out a warning that codegen doesn't support the
54/// specified stmt yet.
55void CodeGenModule::WarnUnsupported(const Stmt *S, const char *Type) {
56 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Warning,
57 "cannot codegen this %0 yet");
58 SourceRange Range = S->getSourceRange();
59 std::string Msg = Type;
Ted Kremenek9c728dc2007-12-12 22:39:36 +000060 getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID,
Ted Kremenek7a9d49f2007-12-11 21:27:55 +000061 &Msg, 1, &Range, 1);
Chris Lattner2c8569d2007-12-02 07:19:18 +000062}
Chris Lattner58c3f9e2007-12-02 06:27:33 +000063
Chris Lattnerc6fdc342008-01-12 07:05:38 +000064/// WarnUnsupported - Print out a warning that codegen doesn't support the
65/// specified decl yet.
66void CodeGenModule::WarnUnsupported(const Decl *D, const char *Type) {
67 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Warning,
68 "cannot codegen this %0 yet");
69 std::string Msg = Type;
70 getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID,
71 &Msg, 1);
72}
73
Chris Lattner6d397602008-03-14 17:18:18 +000074/// AddGlobalCtor - Add a function to the list that will be called before
75/// main() runs.
76void CodeGenModule::AddGlobalCtor(llvm::Function * Ctor) {
77 // TODO: Type coercion of void()* types.
78 GlobalCtors.push_back(Ctor);
79}
80
Chris Lattner391d77a2008-03-30 23:03:07 +000081/// EmitGlobalCtors - Generates the array of contsturctor functions to be
82/// called on module load, if any have been registered with AddGlobalCtor.
Chris Lattner6d397602008-03-14 17:18:18 +000083void CodeGenModule::EmitGlobalCtors() {
Chris Lattner391d77a2008-03-30 23:03:07 +000084 if (GlobalCtors.empty()) return;
Chris Lattnerce5605e2008-03-30 23:25:33 +000085
Chris Lattner6d397602008-03-14 17:18:18 +000086 // Get the type of @llvm.global_ctors
87 std::vector<const llvm::Type*> CtorFields;
88 CtorFields.push_back(llvm::IntegerType::get(32));
89 // Constructor function type
90 std::vector<const llvm::Type*> VoidArgs;
Chris Lattnerce5605e2008-03-30 23:25:33 +000091 llvm::FunctionType* CtorFuncTy =
92 llvm::FunctionType::get(llvm::Type::VoidTy, VoidArgs, false);
93
Chris Lattner6d397602008-03-14 17:18:18 +000094 // i32, function type pair
Chris Lattner572cf092008-03-19 05:24:56 +000095 const llvm::Type *FPType = llvm::PointerType::getUnqual(CtorFuncTy);
96 llvm::StructType* CtorStructTy =
97 llvm::StructType::get(llvm::Type::Int32Ty, FPType, NULL);
Chris Lattner6d397602008-03-14 17:18:18 +000098 // Array of fields
Chris Lattner572cf092008-03-19 05:24:56 +000099 llvm::ArrayType* GlobalCtorsTy =
100 llvm::ArrayType::get(CtorStructTy, GlobalCtors.size());
Chris Lattner6d397602008-03-14 17:18:18 +0000101
Chris Lattner6d397602008-03-14 17:18:18 +0000102 // Define the global variable
Chris Lattner572cf092008-03-19 05:24:56 +0000103 llvm::GlobalVariable *GlobalCtorsVal =
104 new llvm::GlobalVariable(GlobalCtorsTy, false,
105 llvm::GlobalValue::AppendingLinkage,
106 (llvm::Constant*)0, "llvm.global_ctors",
107 &TheModule);
Chris Lattner6d397602008-03-14 17:18:18 +0000108
109 // Populate the array
110 std::vector<llvm::Constant*> CtorValues;
Chris Lattner572cf092008-03-19 05:24:56 +0000111 llvm::Constant *MagicNumber =
112 llvm::ConstantInt::get(llvm::Type::Int32Ty, 65535, false);
113 std::vector<llvm::Constant*> StructValues;
Chris Lattner6d397602008-03-14 17:18:18 +0000114 for (std::vector<llvm::Constant*>::iterator I = GlobalCtors.begin(),
Chris Lattner572cf092008-03-19 05:24:56 +0000115 E = GlobalCtors.end(); I != E; ++I) {
116 StructValues.clear();
Chris Lattner6d397602008-03-14 17:18:18 +0000117 StructValues.push_back(MagicNumber);
118 StructValues.push_back(*I);
119
Chris Lattner572cf092008-03-19 05:24:56 +0000120 CtorValues.push_back(llvm::ConstantStruct::get(CtorStructTy, StructValues));
Chris Lattner6d397602008-03-14 17:18:18 +0000121 }
Chris Lattner572cf092008-03-19 05:24:56 +0000122
123 GlobalCtorsVal->setInitializer(llvm::ConstantArray::get(GlobalCtorsTy,
124 CtorValues));
Chris Lattner6d397602008-03-14 17:18:18 +0000125}
126
Chris Lattner391d77a2008-03-30 23:03:07 +0000127
128
Nate Begeman532485c2008-04-18 23:43:57 +0000129void CodeGenModule::EmitAnnotations() {
130 if (Annotations.empty())
131 return;
132
133 // Create a new global variable for the ConstantStruct in the Module.
134 llvm::Constant *Array =
135 llvm::ConstantArray::get(llvm::ArrayType::get(Annotations[0]->getType(),
136 Annotations.size()),
137 Annotations);
138 llvm::GlobalValue *gv =
139 new llvm::GlobalVariable(Array->getType(), false,
140 llvm::GlobalValue::AppendingLinkage, Array,
141 "llvm.global.annotations", &TheModule);
142 gv->setSection("llvm.metadata");
143}
144
Chris Lattner58c3f9e2007-12-02 06:27:33 +0000145/// ReplaceMapValuesWith - This is a really slow and bad function that
146/// searches for any entries in GlobalDeclMap that point to OldVal, changing
147/// them to point to NewVal. This is badbadbad, FIXME!
148void CodeGenModule::ReplaceMapValuesWith(llvm::Constant *OldVal,
149 llvm::Constant *NewVal) {
150 for (llvm::DenseMap<const Decl*, llvm::Constant*>::iterator
151 I = GlobalDeclMap.begin(), E = GlobalDeclMap.end(); I != E; ++I)
152 if (I->second == OldVal) I->second = NewVal;
153}
154
155
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000156llvm::Constant *CodeGenModule::GetAddrOfFunctionDecl(const FunctionDecl *D,
157 bool isDefinition) {
158 // See if it is already in the map. If so, just return it.
Reid Spencer5f016e22007-07-11 17:01:13 +0000159 llvm::Constant *&Entry = GlobalDeclMap[D];
160 if (Entry) return Entry;
161
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000162 const llvm::Type *Ty = getTypes().ConvertType(D->getType());
163
164 // Check to see if the function already exists.
165 llvm::Function *F = getModule().getFunction(D->getName());
166 const llvm::FunctionType *FTy = cast<llvm::FunctionType>(Ty);
167
168 // If it doesn't already exist, just create and return an entry.
169 if (F == 0) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000170 // FIXME: param attributes for sext/zext etc.
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000171 F = llvm::Function::Create(FTy, llvm::Function::ExternalLinkage,
172 D->getName(), &getModule());
Nate Begemanec9426c2008-03-09 03:09:36 +0000173
174 // Set the appropriate calling convention for the Function.
175 if (D->getAttr<FastCallAttr>())
176 F->setCallingConv(llvm::CallingConv::Fast);
177 return Entry = F;
Reid Spencer5f016e22007-07-11 17:01:13 +0000178 }
179
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000180 // If the pointer type matches, just return it.
Christopher Lambddc23f32007-12-17 01:11:20 +0000181 llvm::Type *PFTy = llvm::PointerType::getUnqual(Ty);
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000182 if (PFTy == F->getType()) return Entry = F;
Chris Lattnerfafad832007-12-02 06:30:46 +0000183
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000184 // If this isn't a definition, just return it casted to the right type.
185 if (!isDefinition)
186 return Entry = llvm::ConstantExpr::getBitCast(F, PFTy);
187
188 // Otherwise, we have a definition after a prototype with the wrong type.
189 // F is the Function* for the one with the wrong type, we must make a new
190 // Function* and update everything that used F (a declaration) with the new
191 // Function* (which will be a definition).
192 //
193 // This happens if there is a prototype for a function (e.g. "int f()") and
194 // then a definition of a different type (e.g. "int f(int x)"). Start by
195 // making a new function of the correct type, RAUW, then steal the name.
Gabor Greif984d0b42008-04-06 20:42:52 +0000196 llvm::Function *NewFn = llvm::Function::Create(FTy,
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000197 llvm::Function::ExternalLinkage,
198 "", &getModule());
199 NewFn->takeName(F);
200
201 // Replace uses of F with the Function we will endow with a body.
202 llvm::Constant *NewPtrForOldDecl =
203 llvm::ConstantExpr::getBitCast(NewFn, F->getType());
204 F->replaceAllUsesWith(NewPtrForOldDecl);
205
206 // FIXME: Update the globaldeclmap for the previous decl of this name. We
207 // really want a way to walk all of these, but we don't have it yet. This
208 // is incredibly slow!
209 ReplaceMapValuesWith(F, NewPtrForOldDecl);
210
211 // Ok, delete the old function now, which is dead.
212 assert(F->isDeclaration() && "Shouldn't replace non-declaration");
213 F->eraseFromParent();
214
215 // Return the new function which has the right type.
216 return Entry = NewFn;
217}
218
Chris Lattnerc4b23a52008-02-05 06:37:34 +0000219static bool IsZeroElementArray(const llvm::Type *Ty) {
220 if (const llvm::ArrayType *ATy = dyn_cast<llvm::ArrayType>(Ty))
221 return ATy->getNumElements() == 0;
222 return false;
223}
224
Chris Lattner2b9d2ca2007-12-18 08:16:44 +0000225llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D,
226 bool isDefinition) {
227 assert(D->hasGlobalStorage() && "Not a global variable");
228
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000229 // See if it is already in the map.
230 llvm::Constant *&Entry = GlobalDeclMap[D];
231 if (Entry) return Entry;
232
Christopher Lambebb97e92008-02-04 02:31:56 +0000233 QualType ASTTy = D->getType();
234 const llvm::Type *Ty = getTypes().ConvertTypeForMem(ASTTy);
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000235
236 // Check to see if the global already exists.
Chris Lattner49573782008-02-02 04:43:11 +0000237 llvm::GlobalVariable *GV = getModule().getGlobalVariable(D->getName(), true);
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000238
239 // If it doesn't already exist, just create and return an entry.
240 if (GV == 0) {
241 return Entry = new llvm::GlobalVariable(Ty, false,
242 llvm::GlobalValue::ExternalLinkage,
Christopher Lambebb97e92008-02-04 02:31:56 +0000243 0, D->getName(), &getModule(), 0,
244 ASTTy.getAddressSpace());
Chris Lattnerfafad832007-12-02 06:30:46 +0000245 }
246
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000247 // If the pointer type matches, just return it.
Christopher Lambddc23f32007-12-17 01:11:20 +0000248 llvm::Type *PTy = llvm::PointerType::getUnqual(Ty);
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000249 if (PTy == GV->getType()) return Entry = GV;
250
251 // If this isn't a definition, just return it casted to the right type.
252 if (!isDefinition)
253 return Entry = llvm::ConstantExpr::getBitCast(GV, PTy);
254
255
256 // Otherwise, we have a definition after a prototype with the wrong type.
257 // GV is the GlobalVariable* for the one with the wrong type, we must make a
258 /// new GlobalVariable* and update everything that used GV (a declaration)
259 // with the new GlobalVariable* (which will be a definition).
260 //
261 // This happens if there is a prototype for a global (e.g. "extern int x[];")
262 // and then a definition of a different type (e.g. "int x[10];"). Start by
263 // making a new global of the correct type, RAUW, then steal the name.
264 llvm::GlobalVariable *NewGV =
265 new llvm::GlobalVariable(Ty, false, llvm::GlobalValue::ExternalLinkage,
Christopher Lambebb97e92008-02-04 02:31:56 +0000266 0, D->getName(), &getModule(), 0,
267 ASTTy.getAddressSpace());
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000268 NewGV->takeName(GV);
269
270 // Replace uses of GV with the globalvalue we will endow with a body.
271 llvm::Constant *NewPtrForOldDecl =
272 llvm::ConstantExpr::getBitCast(NewGV, GV->getType());
273 GV->replaceAllUsesWith(NewPtrForOldDecl);
274
275 // FIXME: Update the globaldeclmap for the previous decl of this name. We
276 // really want a way to walk all of these, but we don't have it yet. This
277 // is incredibly slow!
278 ReplaceMapValuesWith(GV, NewPtrForOldDecl);
279
Chris Lattnerc4b23a52008-02-05 06:37:34 +0000280 // Verify that GV was a declaration or something like x[] which turns into
281 // [0 x type].
282 assert((GV->isDeclaration() ||
283 IsZeroElementArray(GV->getType()->getElementType())) &&
284 "Shouldn't replace non-declaration");
285
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000286 // Ok, delete the old global now, which is dead.
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000287 GV->eraseFromParent();
288
289 // Return the new global which has the right type.
290 return Entry = NewGV;
Reid Spencer5f016e22007-07-11 17:01:13 +0000291}
292
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000293
Chris Lattner391d77a2008-03-30 23:03:07 +0000294void CodeGenModule::EmitObjCMethod(const ObjCMethodDecl *OMD) {
295 // If this is not a prototype, emit the body.
296 if (OMD->getBody())
297 CodeGenFunction(*this).GenerateObjCMethod(OMD);
298}
299
Chris Lattner88a69ad2007-07-13 05:13:43 +0000300void CodeGenModule::EmitFunction(const FunctionDecl *FD) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000301 // If this is not a prototype, emit the body.
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000302 if (FD->getBody()) {
303 // If the function is a static, defer code generation until later so we can
304 // easily omit unused statics.
305 if (FD->getStorageClass() == FunctionDecl::Static) {
Nate Begeman1a1d92a2008-04-20 20:38:08 +0000306 // We need to check the Module here to see if GetAddrOfFunctionDecl() has
307 // already added this function to the Module because the address of the
308 // function's prototype was taken. If this is the case, call
309 // GetAddrOfFunctionDecl to insert the static FunctionDecl into the used
310 // GlobalDeclsMap, so that EmitStatics will generate code for it later.
311 //
312 // Example:
313 // static int foo();
314 // int bar() { return foo(); }
315 // static int foo() { return 5; }
316 if (getModule().getFunction(FD->getName()))
317 GetAddrOfFunctionDecl(FD, true);
318
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000319 StaticDecls.push_back(FD);
320 return;
321 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000322 CodeGenFunction(*this).GenerateCode(FD);
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000323 }
324}
325
326void CodeGenModule::EmitStatics() {
327 // Emit code for each used static decl encountered. Since a previously unused
328 // static decl may become used during the generation of code for a static
329 // function, iterate until no changes are made.
330 bool Changed;
331 do {
332 Changed = false;
333 for (unsigned i = 0, e = StaticDecls.size(); i != e; ++i) {
334 // Check the map of used decls for our static. If not found, continue.
335 const Decl *D = StaticDecls[i];
Nate Begeman1a1d92a2008-04-20 20:38:08 +0000336 if (!GlobalDeclMap.count(D))
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000337 continue;
338
339 // If this is a function decl, generate code for the static function if it
340 // has a body. Otherwise, we must have a var decl for a static global
341 // variable.
342 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
343 if (FD->getBody())
344 CodeGenFunction(*this).GenerateCode(FD);
345 } else {
Nate Begeman1a1d92a2008-04-20 20:38:08 +0000346 EmitGlobalVarInit(cast<VarDecl>(D));
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000347 }
348 // Erase the used decl from the list.
349 StaticDecls[i] = StaticDecls.back();
350 StaticDecls.pop_back();
351 --i;
352 --e;
353
354 // Remember that we made a change.
355 Changed = true;
356 }
357 } while (Changed);
358
359 // Warn about all statics that are still unused at end of code generation.
360 for (unsigned i = 0, e = StaticDecls.size(); i != e; ++i) {
Nate Begeman1a1d92a2008-04-20 20:38:08 +0000361 const NamedDecl *D = StaticDecls[i];
362 std::string Msg = D->getName();
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000363 getDiags().Report(Context.getFullLoc(D->getLocation()),
364 diag::warn_unused_static, &Msg, 1);
365 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000366}
367
Anders Carlsson3b1d57b2008-01-26 01:36:00 +0000368llvm::Constant *CodeGenModule::EmitGlobalInit(const Expr *Expr) {
369 return EmitConstantExpr(Expr);
Devang Patel9e32d4b2007-10-30 21:27:20 +0000370}
371
Nate Begeman8bd4afe2008-04-19 04:17:09 +0000372/// EmitAnnotateAttr - Generate the llvm::ConstantStruct which contains the
373/// annotation information for a given GlobalValue. The annotation struct is
374/// {i8 *, i8 *, i8 *, i32}. The first field is a constant expression, the
375/// GlobalValue being annotated. The second filed is thee constant string
376/// created from the AnnotateAttr's annotation. The third field is a constant
377/// string containing the name of the translation unit. The fourth field is
378/// the line number in the file of the annotated value declaration.
379///
380/// FIXME: this does not unique the annotation string constants, as llvm-gcc
381/// appears to.
382///
383llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
384 const AnnotateAttr *AA,
385 unsigned LineNo) {
386 llvm::Module *M = &getModule();
387
388 // get [N x i8] constants for the annotation string, and the filename string
389 // which are the 2nd and 3rd elements of the global annotation structure.
390 const llvm::Type *SBP = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
391 llvm::Constant *anno = llvm::ConstantArray::get(AA->getAnnotation(), true);
392 llvm::Constant *unit = llvm::ConstantArray::get(M->getModuleIdentifier(),
393 true);
394
395 // Get the two global values corresponding to the ConstantArrays we just
396 // created to hold the bytes of the strings.
397 llvm::GlobalValue *annoGV =
398 new llvm::GlobalVariable(anno->getType(), false,
399 llvm::GlobalValue::InternalLinkage, anno,
400 GV->getName() + ".str", M);
401 // translation unit name string, emitted into the llvm.metadata section.
402 llvm::GlobalValue *unitGV =
403 new llvm::GlobalVariable(unit->getType(), false,
404 llvm::GlobalValue::InternalLinkage, unit, ".str", M);
405
406 // Create the ConstantStruct that is the global annotion.
407 llvm::Constant *Fields[4] = {
408 llvm::ConstantExpr::getBitCast(GV, SBP),
409 llvm::ConstantExpr::getBitCast(annoGV, SBP),
410 llvm::ConstantExpr::getBitCast(unitGV, SBP),
411 llvm::ConstantInt::get(llvm::Type::Int32Ty, LineNo)
412 };
413 return llvm::ConstantStruct::get(Fields, 4, false);
414}
415
Steve Naroff248a7532008-04-15 22:42:06 +0000416void CodeGenModule::EmitGlobalVar(const VarDecl *D) {
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000417 // If the VarDecl is a static, defer code generation until later so we can
418 // easily omit unused statics.
419 if (D->getStorageClass() == VarDecl::Static) {
420 StaticDecls.push_back(D);
421 return;
422 }
423
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000424 // If this is just a forward declaration of the variable, don't emit it now,
425 // allow it to be emitted lazily on its first use.
Chris Lattner88a69ad2007-07-13 05:13:43 +0000426 if (D->getStorageClass() == VarDecl::Extern && D->getInit() == 0)
427 return;
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000428
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000429 EmitGlobalVarInit(D);
430}
431
432void CodeGenModule::EmitGlobalVarInit(const VarDecl *D) {
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000433 // Get the global, forcing it to be a direct reference.
434 llvm::GlobalVariable *GV =
Chris Lattner2b9d2ca2007-12-18 08:16:44 +0000435 cast<llvm::GlobalVariable>(GetAddrOfGlobalVar(D, true));
Chris Lattner9cd4fe42007-12-02 07:09:19 +0000436
437 // Convert the initializer, or use zero if appropriate.
Chris Lattner8f32f712007-07-14 00:23:28 +0000438 llvm::Constant *Init = 0;
439 if (D->getInit() == 0) {
Chris Lattner88a69ad2007-07-13 05:13:43 +0000440 Init = llvm::Constant::getNullValue(GV->getType()->getElementType());
Chris Lattner8f32f712007-07-14 00:23:28 +0000441 } else if (D->getType()->isIntegerType()) {
Hartmut Kaiser7b660002007-10-17 15:00:17 +0000442 llvm::APSInt Value(static_cast<uint32_t>(
Chris Lattner98be4942008-03-05 18:54:05 +0000443 getContext().getTypeSize(D->getInit()->getType())));
Chris Lattner590b6642007-07-15 23:26:56 +0000444 if (D->getInit()->isIntegerConstantExpr(Value, Context))
Chris Lattner8f32f712007-07-14 00:23:28 +0000445 Init = llvm::ConstantInt::get(Value);
446 }
Devang Patel8e53e722007-10-26 16:31:40 +0000447
Devang Patel9e32d4b2007-10-30 21:27:20 +0000448 if (!Init)
Oliver Hunt28247232007-12-02 00:11:25 +0000449 Init = EmitGlobalInit(D->getInit());
Devang Patel8e53e722007-10-26 16:31:40 +0000450
Nate Begeman8bd4afe2008-04-19 04:17:09 +0000451 if (const AnnotateAttr *AA = D->getAttr<AnnotateAttr>()) {
452 SourceManager &SM = Context.getSourceManager();
453 AddAnnotation(EmitAnnotateAttr(GV, AA,
454 SM.getLogicalLineNumber(D->getLocation())));
455 }
456
Chris Lattnerf89dfb22007-12-10 00:05:55 +0000457 assert(GV->getType()->getElementType() == Init->getType() &&
458 "Initializer codegen type mismatch!");
Chris Lattner88a69ad2007-07-13 05:13:43 +0000459 GV->setInitializer(Init);
Chris Lattnerddee4232008-03-03 03:28:21 +0000460
461 if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>())
462 GV->setVisibility(attr->getVisibility());
463 // FIXME: else handle -fvisibility
Chris Lattner88a69ad2007-07-13 05:13:43 +0000464
465 // Set the llvm linkage type as appropriate.
Chris Lattnerddee4232008-03-03 03:28:21 +0000466 if (D->getAttr<DLLImportAttr>())
467 GV->setLinkage(llvm::Function::DLLImportLinkage);
468 else if (D->getAttr<DLLExportAttr>())
469 GV->setLinkage(llvm::Function::DLLExportLinkage);
470 else if (D->getAttr<WeakAttr>()) {
471 GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
472
473 } else {
474 // FIXME: This isn't right. This should handle common linkage and other
475 // stuff.
476 switch (D->getStorageClass()) {
477 case VarDecl::Auto:
478 case VarDecl::Register:
479 assert(0 && "Can't have auto or register globals");
480 case VarDecl::None:
481 if (!D->getInit())
482 GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
483 break;
484 case VarDecl::Extern:
485 case VarDecl::PrivateExtern:
486 // todo: common
487 break;
488 case VarDecl::Static:
489 GV->setLinkage(llvm::GlobalVariable::InternalLinkage);
490 break;
491 }
Chris Lattner88a69ad2007-07-13 05:13:43 +0000492 }
493}
Reid Spencer5f016e22007-07-11 17:01:13 +0000494
Chris Lattner32b266c2007-07-14 00:16:50 +0000495/// EmitGlobalVarDeclarator - Emit all the global vars attached to the specified
496/// declarator chain.
Steve Naroff248a7532008-04-15 22:42:06 +0000497void CodeGenModule::EmitGlobalVarDeclarator(const VarDecl *D) {
498 for (; D; D = cast_or_null<VarDecl>(D->getNextDeclarator()))
499 if (D->isFileVarDecl())
500 EmitGlobalVar(D);
Chris Lattner32b266c2007-07-14 00:16:50 +0000501}
Reid Spencer5f016e22007-07-11 17:01:13 +0000502
Chris Lattnerc5b88062008-02-06 05:08:19 +0000503void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
504 // Make sure that this type is translated.
505 Types.UpdateCompletedType(TD);
Chris Lattnerd86e6bc2008-02-05 08:06:13 +0000506}
507
508
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000509/// getBuiltinLibFunction
510llvm::Function *CodeGenModule::getBuiltinLibFunction(unsigned BuiltinID) {
Chris Lattner1426fec2007-12-13 00:38:03 +0000511 if (BuiltinID > BuiltinFunctions.size())
512 BuiltinFunctions.resize(BuiltinID);
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000513
Chris Lattner1426fec2007-12-13 00:38:03 +0000514 // Cache looked up functions. Since builtin id #0 is invalid we don't reserve
515 // a slot for it.
516 assert(BuiltinID && "Invalid Builtin ID");
517 llvm::Function *&FunctionSlot = BuiltinFunctions[BuiltinID-1];
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000518 if (FunctionSlot)
519 return FunctionSlot;
520
521 assert(Context.BuiltinInfo.isLibFunction(BuiltinID) && "isn't a lib fn");
522
523 // Get the name, skip over the __builtin_ prefix.
524 const char *Name = Context.BuiltinInfo.GetName(BuiltinID)+10;
525
526 // Get the type for the builtin.
527 QualType Type = Context.BuiltinInfo.GetBuiltinType(BuiltinID, Context);
528 const llvm::FunctionType *Ty =
529 cast<llvm::FunctionType>(getTypes().ConvertType(Type));
530
531 // FIXME: This has a serious problem with code like this:
532 // void abs() {}
533 // ... __builtin_abs(x);
534 // The two versions of abs will collide. The fix is for the builtin to win,
535 // and for the existing one to be turned into a constantexpr cast of the
536 // builtin. In the case where the existing one is a static function, it
537 // should just be renamed.
Chris Lattnerc5e940f2007-08-31 04:44:06 +0000538 if (llvm::Function *Existing = getModule().getFunction(Name)) {
539 if (Existing->getFunctionType() == Ty && Existing->hasExternalLinkage())
540 return FunctionSlot = Existing;
541 assert(Existing == 0 && "FIXME: Name collision");
542 }
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000543
544 // FIXME: param attributes for sext/zext etc.
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000545 return FunctionSlot =
546 llvm::Function::Create(Ty, llvm::Function::ExternalLinkage, Name,
547 &getModule());
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000548}
549
Chris Lattner7acda7c2007-12-18 00:25:38 +0000550llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,const llvm::Type **Tys,
551 unsigned NumTys) {
552 return llvm::Intrinsic::getDeclaration(&getModule(),
553 (llvm::Intrinsic::ID)IID, Tys, NumTys);
554}
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000555
Reid Spencer5f016e22007-07-11 17:01:13 +0000556llvm::Function *CodeGenModule::getMemCpyFn() {
557 if (MemCpyFn) return MemCpyFn;
558 llvm::Intrinsic::ID IID;
Chris Lattnerf72a4432008-03-08 08:34:58 +0000559 switch (Context.Target.getPointerWidth(0)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000560 default: assert(0 && "Unknown ptr width");
561 case 32: IID = llvm::Intrinsic::memcpy_i32; break;
562 case 64: IID = llvm::Intrinsic::memcpy_i64; break;
563 }
Chris Lattner7acda7c2007-12-18 00:25:38 +0000564 return MemCpyFn = getIntrinsic(IID);
Reid Spencer5f016e22007-07-11 17:01:13 +0000565}
Anders Carlssonc9e20912007-08-21 00:21:21 +0000566
Lauro Ramos Venancio41ef30e2008-02-19 22:01:01 +0000567llvm::Function *CodeGenModule::getMemSetFn() {
568 if (MemSetFn) return MemSetFn;
569 llvm::Intrinsic::ID IID;
Chris Lattnerf72a4432008-03-08 08:34:58 +0000570 switch (Context.Target.getPointerWidth(0)) {
Lauro Ramos Venancio41ef30e2008-02-19 22:01:01 +0000571 default: assert(0 && "Unknown ptr width");
572 case 32: IID = llvm::Intrinsic::memset_i32; break;
573 case 64: IID = llvm::Intrinsic::memset_i64; break;
574 }
575 return MemSetFn = getIntrinsic(IID);
576}
Chris Lattner7acda7c2007-12-18 00:25:38 +0000577
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000578llvm::Constant *CodeGenModule::
579GetAddrOfConstantCFString(const std::string &str) {
Anders Carlssonc9e20912007-08-21 00:21:21 +0000580 llvm::StringMapEntry<llvm::Constant *> &Entry =
581 CFConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
582
583 if (Entry.getValue())
584 return Entry.getValue();
585
586 std::vector<llvm::Constant*> Fields;
587
588 if (!CFConstantStringClassRef) {
589 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
590 Ty = llvm::ArrayType::get(Ty, 0);
591
592 CFConstantStringClassRef =
593 new llvm::GlobalVariable(Ty, false,
594 llvm::GlobalVariable::ExternalLinkage, 0,
595 "__CFConstantStringClassReference",
596 &getModule());
597 }
598
599 // Class pointer.
600 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty);
601 llvm::Constant *Zeros[] = { Zero, Zero };
602 llvm::Constant *C =
603 llvm::ConstantExpr::getGetElementPtr(CFConstantStringClassRef, Zeros, 2);
604 Fields.push_back(C);
605
606 // Flags.
607 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
608 Fields.push_back(llvm::ConstantInt::get(Ty, 1992));
609
610 // String pointer.
611 C = llvm::ConstantArray::get(str);
612 C = new llvm::GlobalVariable(C->getType(), true,
613 llvm::GlobalValue::InternalLinkage,
614 C, ".str", &getModule());
615
616 C = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2);
617 Fields.push_back(C);
618
619 // String length.
620 Ty = getTypes().ConvertType(getContext().LongTy);
621 Fields.push_back(llvm::ConstantInt::get(Ty, str.length()));
622
623 // The struct.
624 Ty = getTypes().ConvertType(getContext().getCFConstantStringType());
625 C = llvm::ConstantStruct::get(cast<llvm::StructType>(Ty), Fields);
Anders Carlsson0c678292007-11-01 00:41:52 +0000626 llvm::GlobalVariable *GV =
627 new llvm::GlobalVariable(C->getType(), true,
628 llvm::GlobalVariable::InternalLinkage,
629 C, "", &getModule());
630 GV->setSection("__DATA,__cfstring");
631 Entry.setValue(GV);
632 return GV;
Anders Carlssonc9e20912007-08-21 00:21:21 +0000633}
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000634
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000635/// GenerateWritableString -- Creates storage for a string literal.
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000636static llvm::Constant *GenerateStringLiteral(const std::string &str,
637 bool constant,
Chris Lattner2c8569d2007-12-02 07:19:18 +0000638 CodeGenModule &CGM) {
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000639 // Create Constant for this string literal
640 llvm::Constant *C=llvm::ConstantArray::get(str);
641
642 // Create a global variable for this string
643 C = new llvm::GlobalVariable(C->getType(), constant,
644 llvm::GlobalValue::InternalLinkage,
Chris Lattner2c8569d2007-12-02 07:19:18 +0000645 C, ".str", &CGM.getModule());
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000646 return C;
647}
648
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000649/// CodeGenModule::GetAddrOfConstantString -- returns a pointer to the character
650/// array containing the literal. The result is pointer to array type.
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000651llvm::Constant *CodeGenModule::GetAddrOfConstantString(const std::string &str) {
652 // Don't share any string literals if writable-strings is turned on.
653 if (Features.WritableStrings)
654 return GenerateStringLiteral(str, false, *this);
655
656 llvm::StringMapEntry<llvm::Constant *> &Entry =
657 ConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
658
659 if (Entry.getValue())
660 return Entry.getValue();
661
662 // Create a global variable for this.
663 llvm::Constant *C = GenerateStringLiteral(str, true, *this);
664 Entry.setValue(C);
665 return C;
666}