blob: f2112ebd7212c3f6303e66ff800d6242580e7365 [file] [log] [blame]
Chris Lattner4b009652007-07-25 00:24:17 +00001//===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner959e5be2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattner4b009652007-07-25 00:24:17 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This coordinates the per-module state used while generating code.
11//
12//===----------------------------------------------------------------------===//
13
Sanjiv Gupta40e56a12008-05-08 08:54:20 +000014#include "CGDebugInfo.h"
Chris Lattner4b009652007-07-25 00:24:17 +000015#include "CodeGenModule.h"
16#include "CodeGenFunction.h"
17#include "clang/AST/ASTContext.h"
18#include "clang/AST/Decl.h"
Chris Lattnercf9c9d02007-12-02 07:19:18 +000019#include "clang/Basic/Diagnostic.h"
Chris Lattnerdb6be562007-11-28 05:34:05 +000020#include "clang/Basic/LangOptions.h"
Nate Begeman8a704172008-04-19 04:17:09 +000021#include "clang/Basic/SourceManager.h"
Chris Lattner4b009652007-07-25 00:24:17 +000022#include "clang/Basic/TargetInfo.h"
Nate Begemandc6262e2008-03-09 03:09:36 +000023#include "llvm/CallingConv.h"
Chris Lattner4b009652007-07-25 00:24:17 +000024#include "llvm/Constants.h"
25#include "llvm/DerivedTypes.h"
Chris Lattnerab862cc2007-08-31 04:31:45 +000026#include "llvm/Module.h"
Chris Lattner4b009652007-07-25 00:24:17 +000027#include "llvm/Intrinsics.h"
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000028#include "llvm/Target/TargetData.h"
Chris Lattnerb033c4a2008-04-30 16:05:42 +000029#include "llvm/Analysis/Verifier.h"
Christopher Lamb6db92f32007-12-02 08:49:54 +000030#include <algorithm>
Chris Lattner4b009652007-07-25 00:24:17 +000031using namespace clang;
32using namespace CodeGen;
33
34
Chris Lattnerdb6be562007-11-28 05:34:05 +000035CodeGenModule::CodeGenModule(ASTContext &C, const LangOptions &LO,
Chris Lattner22595b82007-12-02 01:40:18 +000036 llvm::Module &M, const llvm::TargetData &TD,
Sanjiv Gupta40e56a12008-05-08 08:54:20 +000037 Diagnostic &diags, bool GenerateDebugInfo)
Chris Lattner22595b82007-12-02 01:40:18 +000038 : Context(C), Features(LO), TheModule(M), TheTargetData(TD), Diags(diags),
Eli Friedman8f08a252008-05-26 12:59:39 +000039 Types(C, M, TD), MemCpyFn(0), MemMoveFn(0), MemSetFn(0),
40 CFConstantStringClassRef(0) {
Chris Lattnercbfb5512008-03-01 08:45:05 +000041 //TODO: Make this selectable at runtime
Chris Lattner547907c2008-06-26 04:19:03 +000042 Runtime = CreateObjCRuntime(*this);
Sanjiv Gupta40e56a12008-05-08 08:54:20 +000043
44 // If debug info generation is enabled, create the CGDebugInfo object.
45 if (GenerateDebugInfo)
46 DebugInfo = new CGDebugInfo(this);
47 else
48 DebugInfo = NULL;
Chris Lattnercbfb5512008-03-01 08:45:05 +000049}
50
51CodeGenModule::~CodeGenModule() {
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000052 EmitStatics();
Chris Lattnerb326b172008-03-30 23:03:07 +000053 llvm::Function *ObjCInitFunction = Runtime->ModuleInitFunction();
Chris Lattnerc61e9f82008-03-30 23:25:33 +000054 if (ObjCInitFunction)
Chris Lattnerb326b172008-03-30 23:03:07 +000055 AddGlobalCtor(ObjCInitFunction);
Chris Lattner753d2592008-03-14 17:18:18 +000056 EmitGlobalCtors();
Nate Begeman52da5c72008-04-18 23:43:57 +000057 EmitAnnotations();
Chris Lattnercbfb5512008-03-01 08:45:05 +000058 delete Runtime;
Sanjiv Gupta40e56a12008-05-08 08:54:20 +000059 delete DebugInfo;
Chris Lattnerb033c4a2008-04-30 16:05:42 +000060 // Run the verifier to check that the generated code is consistent.
61 assert(!verifyModule(TheModule));
Chris Lattnercbfb5512008-03-01 08:45:05 +000062}
Chris Lattner4b009652007-07-25 00:24:17 +000063
Chris Lattnercf9c9d02007-12-02 07:19:18 +000064/// WarnUnsupported - Print out a warning that codegen doesn't support the
65/// specified stmt yet.
66void CodeGenModule::WarnUnsupported(const Stmt *S, const char *Type) {
67 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Warning,
68 "cannot codegen this %0 yet");
69 SourceRange Range = S->getSourceRange();
70 std::string Msg = Type;
Ted Kremenekd7f64cd2007-12-12 22:39:36 +000071 getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID,
Ted Kremenekb3ee1932007-12-11 21:27:55 +000072 &Msg, 1, &Range, 1);
Chris Lattnercf9c9d02007-12-02 07:19:18 +000073}
Chris Lattner0e4755d2007-12-02 06:27:33 +000074
Chris Lattner806a5f52008-01-12 07:05:38 +000075/// WarnUnsupported - Print out a warning that codegen doesn't support the
76/// specified decl yet.
77void CodeGenModule::WarnUnsupported(const Decl *D, const char *Type) {
78 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Warning,
79 "cannot codegen this %0 yet");
80 std::string Msg = Type;
81 getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID,
82 &Msg, 1);
83}
84
Dan Gohman4751a3a2008-05-22 00:50:06 +000085/// setVisibility - Set the visibility for the given LLVM GlobalValue
86/// according to the given clang AST visibility value.
87void CodeGenModule::setVisibility(llvm::GlobalValue *GV,
88 VisibilityAttr::VisibilityTypes Vis) {
89 switch (Vis) {
90 default: assert(0 && "Unknown visibility!");
91 case VisibilityAttr::DefaultVisibility:
92 GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
93 break;
94 case VisibilityAttr::HiddenVisibility:
95 GV->setVisibility(llvm::GlobalValue::HiddenVisibility);
96 break;
97 case VisibilityAttr::ProtectedVisibility:
98 GV->setVisibility(llvm::GlobalValue::ProtectedVisibility);
99 break;
100 }
101}
102
Chris Lattner753d2592008-03-14 17:18:18 +0000103/// AddGlobalCtor - Add a function to the list that will be called before
104/// main() runs.
105void CodeGenModule::AddGlobalCtor(llvm::Function * Ctor) {
106 // TODO: Type coercion of void()* types.
107 GlobalCtors.push_back(Ctor);
108}
109
Chris Lattnerb326b172008-03-30 23:03:07 +0000110/// EmitGlobalCtors - Generates the array of contsturctor functions to be
111/// called on module load, if any have been registered with AddGlobalCtor.
Chris Lattner753d2592008-03-14 17:18:18 +0000112void CodeGenModule::EmitGlobalCtors() {
Chris Lattnerb326b172008-03-30 23:03:07 +0000113 if (GlobalCtors.empty()) return;
Chris Lattnerc61e9f82008-03-30 23:25:33 +0000114
Chris Lattner753d2592008-03-14 17:18:18 +0000115 // Get the type of @llvm.global_ctors
116 std::vector<const llvm::Type*> CtorFields;
117 CtorFields.push_back(llvm::IntegerType::get(32));
118 // Constructor function type
119 std::vector<const llvm::Type*> VoidArgs;
Chris Lattnerc61e9f82008-03-30 23:25:33 +0000120 llvm::FunctionType* CtorFuncTy =
121 llvm::FunctionType::get(llvm::Type::VoidTy, VoidArgs, false);
122
Chris Lattner753d2592008-03-14 17:18:18 +0000123 // i32, function type pair
Chris Lattnera18c12e2008-03-19 05:24:56 +0000124 const llvm::Type *FPType = llvm::PointerType::getUnqual(CtorFuncTy);
125 llvm::StructType* CtorStructTy =
126 llvm::StructType::get(llvm::Type::Int32Ty, FPType, NULL);
Chris Lattner753d2592008-03-14 17:18:18 +0000127 // Array of fields
Chris Lattnera18c12e2008-03-19 05:24:56 +0000128 llvm::ArrayType* GlobalCtorsTy =
129 llvm::ArrayType::get(CtorStructTy, GlobalCtors.size());
Chris Lattner753d2592008-03-14 17:18:18 +0000130
Chris Lattner753d2592008-03-14 17:18:18 +0000131 // Define the global variable
Chris Lattnera18c12e2008-03-19 05:24:56 +0000132 llvm::GlobalVariable *GlobalCtorsVal =
133 new llvm::GlobalVariable(GlobalCtorsTy, false,
134 llvm::GlobalValue::AppendingLinkage,
135 (llvm::Constant*)0, "llvm.global_ctors",
136 &TheModule);
Chris Lattner753d2592008-03-14 17:18:18 +0000137
138 // Populate the array
139 std::vector<llvm::Constant*> CtorValues;
Chris Lattnera18c12e2008-03-19 05:24:56 +0000140 llvm::Constant *MagicNumber =
141 llvm::ConstantInt::get(llvm::Type::Int32Ty, 65535, false);
142 std::vector<llvm::Constant*> StructValues;
Chris Lattner753d2592008-03-14 17:18:18 +0000143 for (std::vector<llvm::Constant*>::iterator I = GlobalCtors.begin(),
Chris Lattnera18c12e2008-03-19 05:24:56 +0000144 E = GlobalCtors.end(); I != E; ++I) {
145 StructValues.clear();
Chris Lattner753d2592008-03-14 17:18:18 +0000146 StructValues.push_back(MagicNumber);
147 StructValues.push_back(*I);
148
Chris Lattnera18c12e2008-03-19 05:24:56 +0000149 CtorValues.push_back(llvm::ConstantStruct::get(CtorStructTy, StructValues));
Chris Lattner753d2592008-03-14 17:18:18 +0000150 }
Chris Lattnera18c12e2008-03-19 05:24:56 +0000151
152 GlobalCtorsVal->setInitializer(llvm::ConstantArray::get(GlobalCtorsTy,
153 CtorValues));
Chris Lattner753d2592008-03-14 17:18:18 +0000154}
155
Chris Lattnerb326b172008-03-30 23:03:07 +0000156
157
Nate Begeman52da5c72008-04-18 23:43:57 +0000158void CodeGenModule::EmitAnnotations() {
159 if (Annotations.empty())
160 return;
161
162 // Create a new global variable for the ConstantStruct in the Module.
163 llvm::Constant *Array =
164 llvm::ConstantArray::get(llvm::ArrayType::get(Annotations[0]->getType(),
165 Annotations.size()),
166 Annotations);
167 llvm::GlobalValue *gv =
168 new llvm::GlobalVariable(Array->getType(), false,
169 llvm::GlobalValue::AppendingLinkage, Array,
170 "llvm.global.annotations", &TheModule);
171 gv->setSection("llvm.metadata");
172}
173
Chris Lattner0e4755d2007-12-02 06:27:33 +0000174/// ReplaceMapValuesWith - This is a really slow and bad function that
175/// searches for any entries in GlobalDeclMap that point to OldVal, changing
176/// them to point to NewVal. This is badbadbad, FIXME!
177void CodeGenModule::ReplaceMapValuesWith(llvm::Constant *OldVal,
178 llvm::Constant *NewVal) {
179 for (llvm::DenseMap<const Decl*, llvm::Constant*>::iterator
180 I = GlobalDeclMap.begin(), E = GlobalDeclMap.end(); I != E; ++I)
181 if (I->second == OldVal) I->second = NewVal;
182}
183
Eli Friedman9be42212008-06-01 15:54:49 +0000184bool hasAggregateLLVMType(QualType T) {
185 return !T->isRealType() && !T->isPointerLikeType() &&
186 !T->isVoidType() && !T->isVectorType() && !T->isFunctionType();
187}
188
Nuno Lopes78534382008-06-08 15:45:52 +0000189void CodeGenModule::SetGlobalValueAttributes(const FunctionDecl *FD,
190 llvm::GlobalValue *GV) {
191 // TODO: Set up linkage and many other things. Note, this is a simple
192 // approximation of what we really want.
193 if (FD->getStorageClass() == FunctionDecl::Static)
194 GV->setLinkage(llvm::Function::InternalLinkage);
195 else if (FD->getAttr<DLLImportAttr>())
196 GV->setLinkage(llvm::Function::DLLImportLinkage);
197 else if (FD->getAttr<DLLExportAttr>())
198 GV->setLinkage(llvm::Function::DLLExportLinkage);
199 else if (FD->getAttr<WeakAttr>() || FD->isInline())
200 GV->setLinkage(llvm::Function::WeakLinkage);
201
202 if (const VisibilityAttr *attr = FD->getAttr<VisibilityAttr>())
203 CodeGenModule::setVisibility(GV, attr->getVisibility());
204 // FIXME: else handle -fvisibility
205}
206
Eli Friedman9be42212008-06-01 15:54:49 +0000207void CodeGenModule::SetFunctionAttributes(const FunctionDecl *FD,
208 llvm::Function *F,
209 const llvm::FunctionType *FTy) {
210 unsigned FuncAttrs = 0;
211 if (FD->getAttr<NoThrowAttr>())
212 FuncAttrs |= llvm::ParamAttr::NoUnwind;
213 if (FD->getAttr<NoReturnAttr>())
214 FuncAttrs |= llvm::ParamAttr::NoReturn;
215
216 llvm::SmallVector<llvm::ParamAttrsWithIndex, 8> ParamAttrList;
217 if (FuncAttrs)
218 ParamAttrList.push_back(llvm::ParamAttrsWithIndex::get(0, FuncAttrs));
219 // Note that there is parallel code in CodeGenFunction::EmitCallExpr
220 bool AggregateReturn = hasAggregateLLVMType(FD->getResultType());
221 if (AggregateReturn)
222 ParamAttrList.push_back(
223 llvm::ParamAttrsWithIndex::get(1, llvm::ParamAttr::StructRet));
224 unsigned increment = AggregateReturn ? 2 : 1;
Eli Friedmanfa94dff2008-06-04 19:41:28 +0000225 const FunctionTypeProto* FTP = dyn_cast<FunctionTypeProto>(FD->getType());
226 if (FTP) {
227 for (unsigned i = 0; i < FTP->getNumArgs(); i++) {
228 QualType ParamType = FTP->getArgType(i);
229 unsigned ParamAttrs = 0;
230 if (ParamType->isRecordType())
231 ParamAttrs |= llvm::ParamAttr::ByVal;
Chris Lattner578279d2008-06-26 05:08:00 +0000232 if (ParamType->isSignedIntegerType() &&
233 ParamType->isPromotableIntegerType())
Eli Friedmanfa94dff2008-06-04 19:41:28 +0000234 ParamAttrs |= llvm::ParamAttr::SExt;
Chris Lattner578279d2008-06-26 05:08:00 +0000235 if (ParamType->isUnsignedIntegerType() &&
236 ParamType->isPromotableIntegerType())
Eli Friedmanfa94dff2008-06-04 19:41:28 +0000237 ParamAttrs |= llvm::ParamAttr::ZExt;
238 if (ParamAttrs)
239 ParamAttrList.push_back(llvm::ParamAttrsWithIndex::get(i + increment,
240 ParamAttrs));
241 }
Eli Friedman9be42212008-06-01 15:54:49 +0000242 }
Eli Friedmanfa94dff2008-06-04 19:41:28 +0000243
Eli Friedman9be42212008-06-01 15:54:49 +0000244 F->setParamAttrs(llvm::PAListPtr::get(ParamAttrList.begin(),
245 ParamAttrList.size()));
246
247 // Set the appropriate calling convention for the Function.
248 if (FD->getAttr<FastCallAttr>())
249 F->setCallingConv(llvm::CallingConv::Fast);
250
Nuno Lopes78534382008-06-08 15:45:52 +0000251 SetGlobalValueAttributes(FD, F);
Eli Friedman9be42212008-06-01 15:54:49 +0000252}
253
254
Chris Lattner0e4755d2007-12-02 06:27:33 +0000255
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000256llvm::Constant *CodeGenModule::GetAddrOfFunctionDecl(const FunctionDecl *D,
257 bool isDefinition) {
258 // See if it is already in the map. If so, just return it.
Chris Lattner4b009652007-07-25 00:24:17 +0000259 llvm::Constant *&Entry = GlobalDeclMap[D];
Eli Friedman5dcae522008-05-30 11:13:18 +0000260 if (!isDefinition && Entry) return Entry;
Eli Friedman725d2be2008-05-20 09:21:07 +0000261
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000262 const llvm::Type *Ty = getTypes().ConvertType(D->getType());
263
264 // Check to see if the function already exists.
265 llvm::Function *F = getModule().getFunction(D->getName());
266 const llvm::FunctionType *FTy = cast<llvm::FunctionType>(Ty);
267
268 // If it doesn't already exist, just create and return an entry.
269 if (F == 0) {
Chris Lattner4b009652007-07-25 00:24:17 +0000270 // FIXME: param attributes for sext/zext etc.
Nuno Lopes78534382008-06-08 15:45:52 +0000271 if (D->getBody() || !D->getAttr<AliasAttr>())
272 F = llvm::Function::Create(FTy, llvm::Function::ExternalLinkage,
273 D->getName(), &getModule());
274 else {
275 const std::string& aliaseeName = D->getAttr<AliasAttr>()->getAliasee();
276 llvm::Function *aliasee = getModule().getFunction(aliaseeName);
277 llvm::GlobalValue *alias = new llvm::GlobalAlias(aliasee->getType(),
Chris Lattner578279d2008-06-26 05:08:00 +0000278 llvm::Function::ExternalLinkage,
Nuno Lopes78534382008-06-08 15:45:52 +0000279 D->getName(),
280 aliasee,
281 &getModule());
282 SetGlobalValueAttributes(D, alias);
283 return Entry = alias;
284 }
Nate Begemandc6262e2008-03-09 03:09:36 +0000285
Eli Friedman9be42212008-06-01 15:54:49 +0000286 SetFunctionAttributes(D, F, FTy);
Nate Begemandc6262e2008-03-09 03:09:36 +0000287 return Entry = F;
Chris Lattner4b009652007-07-25 00:24:17 +0000288 }
289
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000290 // If the pointer type matches, just return it.
Christopher Lamb4fe5e702007-12-17 01:11:20 +0000291 llvm::Type *PFTy = llvm::PointerType::getUnqual(Ty);
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000292 if (PFTy == F->getType()) return Entry = F;
Chris Lattner77ce67c2007-12-02 06:30:46 +0000293
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000294 // If this isn't a definition, just return it casted to the right type.
295 if (!isDefinition)
296 return Entry = llvm::ConstantExpr::getBitCast(F, PFTy);
297
298 // Otherwise, we have a definition after a prototype with the wrong type.
299 // F is the Function* for the one with the wrong type, we must make a new
300 // Function* and update everything that used F (a declaration) with the new
301 // Function* (which will be a definition).
302 //
303 // This happens if there is a prototype for a function (e.g. "int f()") and
304 // then a definition of a different type (e.g. "int f(int x)"). Start by
305 // making a new function of the correct type, RAUW, then steal the name.
Gabor Greif815e2c12008-04-06 20:42:52 +0000306 llvm::Function *NewFn = llvm::Function::Create(FTy,
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000307 llvm::Function::ExternalLinkage,
308 "", &getModule());
309 NewFn->takeName(F);
310
311 // Replace uses of F with the Function we will endow with a body.
312 llvm::Constant *NewPtrForOldDecl =
313 llvm::ConstantExpr::getBitCast(NewFn, F->getType());
314 F->replaceAllUsesWith(NewPtrForOldDecl);
315
316 // FIXME: Update the globaldeclmap for the previous decl of this name. We
317 // really want a way to walk all of these, but we don't have it yet. This
318 // is incredibly slow!
319 ReplaceMapValuesWith(F, NewPtrForOldDecl);
320
321 // Ok, delete the old function now, which is dead.
322 assert(F->isDeclaration() && "Shouldn't replace non-declaration");
323 F->eraseFromParent();
324
Eli Friedman9be42212008-06-01 15:54:49 +0000325 SetFunctionAttributes(D, NewFn, FTy);
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000326 // Return the new function which has the right type.
327 return Entry = NewFn;
328}
329
Chris Lattnerd2df2b52007-12-18 08:16:44 +0000330llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D,
331 bool isDefinition) {
332 assert(D->hasGlobalStorage() && "Not a global variable");
Eli Friedman43a0ce82008-05-30 19:50:47 +0000333 assert(!isDefinition && "This shouldn't be called for definitions!");
334
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000335 // See if it is already in the map.
336 llvm::Constant *&Entry = GlobalDeclMap[D];
337 if (Entry) return Entry;
Eli Friedman43a0ce82008-05-30 19:50:47 +0000338
Christopher Lamb2a72bb32008-02-04 02:31:56 +0000339 QualType ASTTy = D->getType();
340 const llvm::Type *Ty = getTypes().ConvertTypeForMem(ASTTy);
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000341
342 // Check to see if the global already exists.
Chris Lattnered430e92008-02-02 04:43:11 +0000343 llvm::GlobalVariable *GV = getModule().getGlobalVariable(D->getName(), true);
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000344
345 // If it doesn't already exist, just create and return an entry.
346 if (GV == 0) {
347 return Entry = new llvm::GlobalVariable(Ty, false,
348 llvm::GlobalValue::ExternalLinkage,
Christopher Lamb2a72bb32008-02-04 02:31:56 +0000349 0, D->getName(), &getModule(), 0,
350 ASTTy.getAddressSpace());
Chris Lattner77ce67c2007-12-02 06:30:46 +0000351 }
Chris Lattner4b009652007-07-25 00:24:17 +0000352
Eli Friedman43a0ce82008-05-30 19:50:47 +0000353 // Otherwise, it already exists; return the existing version
354 llvm::PointerType *PTy = llvm::PointerType::get(Ty, ASTTy.getAddressSpace());
355 return Entry = llvm::ConstantExpr::getBitCast(GV, PTy);
356}
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000357
Chris Lattnerb326b172008-03-30 23:03:07 +0000358void CodeGenModule::EmitObjCMethod(const ObjCMethodDecl *OMD) {
359 // If this is not a prototype, emit the body.
360 if (OMD->getBody())
361 CodeGenFunction(*this).GenerateObjCMethod(OMD);
362}
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000363void CodeGenModule::EmitObjCProtocolImplementation(const ObjCProtocolDecl *PD){
364 llvm::SmallVector<std::string, 16> Protocols;
Chris Lattner0be08822008-07-21 21:32:27 +0000365 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
366 E = PD->protocol_end(); PI != E; ++PI)
367 Protocols.push_back((*PI)->getName());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000368 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
369 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
370 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
Chris Lattner0be08822008-07-21 21:32:27 +0000371 E = PD->instmeth_end(); iter != E; iter++) {
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000372 std::string TypeStr;
Chris Lattner0be08822008-07-21 21:32:27 +0000373 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000374 InstanceMethodNames.push_back(
375 GetAddrOfConstantString((*iter)->getSelector().getName()));
376 InstanceMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
377 }
378 // Collect information about class methods:
379 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
380 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
381 for (ObjCProtocolDecl::classmeth_iterator iter = PD->classmeth_begin(),
382 endIter = PD->classmeth_end() ; iter != endIter ; iter++) {
383 std::string TypeStr;
384 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
385 ClassMethodNames.push_back(
386 GetAddrOfConstantString((*iter)->getSelector().getName()));
387 ClassMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
388 }
389 Runtime->GenerateProtocol(PD->getName(), Protocols, InstanceMethodNames,
390 InstanceMethodTypes, ClassMethodNames, ClassMethodTypes);
391}
392
393void CodeGenModule::EmitObjCCategoryImpl(const ObjCCategoryImplDecl *OCD) {
394
395 // Collect information about instance methods
Chris Lattner578279d2008-06-26 05:08:00 +0000396 llvm::SmallVector<Selector, 16> InstanceMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000397 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
398 for (ObjCCategoryDecl::instmeth_iterator iter = OCD->instmeth_begin(),
399 endIter = OCD->instmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000400 InstanceMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000401 std::string TypeStr;
Chris Lattnerbcb3e862008-06-26 04:52:29 +0000402 Context.getObjCEncodingForMethodDecl(*iter,TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000403 InstanceMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
404 }
405
406 // Collect information about class methods
Chris Lattner578279d2008-06-26 05:08:00 +0000407 llvm::SmallVector<Selector, 16> ClassMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000408 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
409 for (ObjCCategoryDecl::classmeth_iterator iter = OCD->classmeth_begin(),
410 endIter = OCD->classmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000411 ClassMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000412 std::string TypeStr;
Chris Lattnerbcb3e862008-06-26 04:52:29 +0000413 Context.getObjCEncodingForMethodDecl(*iter,TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000414 ClassMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
415 }
416
417 // Collect the names of referenced protocols
418 llvm::SmallVector<std::string, 16> Protocols;
Chris Lattner8bcb5252008-07-21 18:19:38 +0000419 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
420 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
421 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
422 E = Protos.end(); I != E; ++I)
423 Protocols.push_back((*I)->getName());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000424
425 // Generate the category
426 Runtime->GenerateCategory(OCD->getClassInterface()->getName(),
Chris Lattner578279d2008-06-26 05:08:00 +0000427 OCD->getName(), InstanceMethodSels, InstanceMethodTypes,
428 ClassMethodSels, ClassMethodTypes, Protocols);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000429}
430
431void CodeGenModule::EmitObjCClassImplementation(
432 const ObjCImplementationDecl *OID) {
433 // Get the superclass name.
434 const ObjCInterfaceDecl * SCDecl = OID->getClassInterface()->getSuperClass();
435 const char * SCName = NULL;
436 if (SCDecl) {
437 SCName = SCDecl->getName();
438 }
439
440 // Get the class name
441 ObjCInterfaceDecl * ClassDecl = (ObjCInterfaceDecl*)OID->getClassInterface();
442 const char * ClassName = ClassDecl->getName();
443
444 // Get the size of instances. For runtimes that support late-bound instances
445 // this should probably be something different (size just of instance
446 // varaibles in this class, not superclasses?).
447 int instanceSize = 0;
448 const llvm::Type *ObjTy;
449 if (!Runtime->LateBoundIVars()) {
450 ObjTy = getTypes().ConvertType(Context.getObjCInterfaceType(ClassDecl));
451 instanceSize = TheTargetData.getABITypeSize(ObjTy);
452 }
453
454 // Collect information about instance variables.
455 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
456 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
457 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
458 const llvm::StructLayout *Layout =
459 TheTargetData.getStructLayout(cast<llvm::StructType>(ObjTy));
460 ObjTy = llvm::PointerType::getUnqual(ObjTy);
461 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
462 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
463 // Store the name
464 IvarNames.push_back(GetAddrOfConstantString((*iter)->getName()));
465 // Get the type encoding for this ivar
466 std::string TypeStr;
467 llvm::SmallVector<const RecordType *, 8> EncodingRecordTypes;
468 Context.getObjCEncodingForType((*iter)->getType(), TypeStr,
469 EncodingRecordTypes);
470 IvarTypes.push_back(GetAddrOfConstantString(TypeStr));
471 // Get the offset
472 int offset =
473 (int)Layout->getElementOffset(getTypes().getLLVMFieldNo(*iter));
474 IvarOffsets.push_back(
475 llvm::ConstantInt::get(llvm::Type::Int32Ty, offset));
476 }
477
478 // Collect information about instance methods
Chris Lattner578279d2008-06-26 05:08:00 +0000479 llvm::SmallVector<Selector, 16> InstanceMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000480 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
481 for (ObjCImplementationDecl::instmeth_iterator iter = OID->instmeth_begin(),
482 endIter = OID->instmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000483 InstanceMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000484 std::string TypeStr;
485 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000486 InstanceMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
487 }
488
489 // Collect information about class methods
Chris Lattner578279d2008-06-26 05:08:00 +0000490 llvm::SmallVector<Selector, 16> ClassMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000491 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
492 for (ObjCImplementationDecl::classmeth_iterator iter = OID->classmeth_begin(),
493 endIter = OID->classmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000494 ClassMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000495 std::string TypeStr;
496 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000497 ClassMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
498 }
499 // Collect the names of referenced protocols
500 llvm::SmallVector<std::string, 16> Protocols;
Chris Lattner8bcb5252008-07-21 18:19:38 +0000501 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
502 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
503 E = Protos.end(); I != E; ++I)
504 Protocols.push_back((*I)->getName());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000505
506 // Generate the category
507 Runtime->GenerateClass(ClassName, SCName, instanceSize, IvarNames, IvarTypes,
Chris Lattner578279d2008-06-26 05:08:00 +0000508 IvarOffsets, InstanceMethodSels, InstanceMethodTypes,
509 ClassMethodSels, ClassMethodTypes, Protocols);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000510}
511
Chris Lattnerb326b172008-03-30 23:03:07 +0000512
Chris Lattner4b009652007-07-25 00:24:17 +0000513void CodeGenModule::EmitFunction(const FunctionDecl *FD) {
514 // If this is not a prototype, emit the body.
Nuno Lopes78534382008-06-08 15:45:52 +0000515 if (!FD->isThisDeclarationADefinition()) {
516 if (FD->getAttr<AliasAttr>())
517 GetAddrOfFunctionDecl(FD, true);
Chris Lattner74bf5552008-05-04 02:29:49 +0000518 return;
Nuno Lopes78534382008-06-08 15:45:52 +0000519 }
Chris Lattner74bf5552008-05-04 02:29:49 +0000520
521 // If the function is a static, defer code generation until later so we can
522 // easily omit unused statics.
523 if (FD->getStorageClass() != FunctionDecl::Static) {
Chris Lattner4b009652007-07-25 00:24:17 +0000524 CodeGenFunction(*this).GenerateCode(FD);
Chris Lattner74bf5552008-05-04 02:29:49 +0000525 return;
Nate Begemanad320b62008-04-20 06:29:50 +0000526 }
Chris Lattner74bf5552008-05-04 02:29:49 +0000527
528 StaticDecls.push_back(FD);
Nate Begemanad320b62008-04-20 06:29:50 +0000529}
530
531void CodeGenModule::EmitStatics() {
532 // Emit code for each used static decl encountered. Since a previously unused
533 // static decl may become used during the generation of code for a static
534 // function, iterate until no changes are made.
535 bool Changed;
536 do {
537 Changed = false;
538 for (unsigned i = 0, e = StaticDecls.size(); i != e; ++i) {
Nate Begemanad320b62008-04-20 06:29:50 +0000539 const Decl *D = StaticDecls[i];
Eli Friedmana4d4e2f2008-05-27 04:58:01 +0000540
541 // Check if we have used a decl with the same name
542 // FIXME: The AST should have some sort of aggregate decls or
543 // global symbol map.
544 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
545 if (!getModule().getFunction(FD->getName()))
546 continue;
547 } else {
548 if (!getModule().getNamedGlobal(cast<VarDecl>(D)->getName()))
549 continue;
550 }
551
Nate Begemanad320b62008-04-20 06:29:50 +0000552 // If this is a function decl, generate code for the static function if it
553 // has a body. Otherwise, we must have a var decl for a static global
554 // variable.
555 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
556 if (FD->getBody())
557 CodeGenFunction(*this).GenerateCode(FD);
Nuno Lopes78534382008-06-08 15:45:52 +0000558 else if (FD->getAttr<AliasAttr>())
559 GetAddrOfFunctionDecl(FD, true);
Nate Begemanad320b62008-04-20 06:29:50 +0000560 } else {
Nate Begeman91f52012008-04-20 20:38:08 +0000561 EmitGlobalVarInit(cast<VarDecl>(D));
Nate Begemanad320b62008-04-20 06:29:50 +0000562 }
563 // Erase the used decl from the list.
564 StaticDecls[i] = StaticDecls.back();
565 StaticDecls.pop_back();
566 --i;
567 --e;
568
569 // Remember that we made a change.
570 Changed = true;
571 }
572 } while (Changed);
Chris Lattner4b009652007-07-25 00:24:17 +0000573}
574
Anders Carlssond76cead2008-01-26 01:36:00 +0000575llvm::Constant *CodeGenModule::EmitGlobalInit(const Expr *Expr) {
576 return EmitConstantExpr(Expr);
Devang Patel08a10cc2007-10-30 21:27:20 +0000577}
578
Nate Begeman8a704172008-04-19 04:17:09 +0000579/// EmitAnnotateAttr - Generate the llvm::ConstantStruct which contains the
580/// annotation information for a given GlobalValue. The annotation struct is
581/// {i8 *, i8 *, i8 *, i32}. The first field is a constant expression, the
582/// GlobalValue being annotated. The second filed is thee constant string
583/// created from the AnnotateAttr's annotation. The third field is a constant
584/// string containing the name of the translation unit. The fourth field is
585/// the line number in the file of the annotated value declaration.
586///
587/// FIXME: this does not unique the annotation string constants, as llvm-gcc
588/// appears to.
589///
590llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
591 const AnnotateAttr *AA,
592 unsigned LineNo) {
593 llvm::Module *M = &getModule();
594
595 // get [N x i8] constants for the annotation string, and the filename string
596 // which are the 2nd and 3rd elements of the global annotation structure.
597 const llvm::Type *SBP = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
598 llvm::Constant *anno = llvm::ConstantArray::get(AA->getAnnotation(), true);
599 llvm::Constant *unit = llvm::ConstantArray::get(M->getModuleIdentifier(),
600 true);
601
602 // Get the two global values corresponding to the ConstantArrays we just
603 // created to hold the bytes of the strings.
604 llvm::GlobalValue *annoGV =
605 new llvm::GlobalVariable(anno->getType(), false,
606 llvm::GlobalValue::InternalLinkage, anno,
607 GV->getName() + ".str", M);
608 // translation unit name string, emitted into the llvm.metadata section.
609 llvm::GlobalValue *unitGV =
610 new llvm::GlobalVariable(unit->getType(), false,
611 llvm::GlobalValue::InternalLinkage, unit, ".str", M);
612
613 // Create the ConstantStruct that is the global annotion.
614 llvm::Constant *Fields[4] = {
615 llvm::ConstantExpr::getBitCast(GV, SBP),
616 llvm::ConstantExpr::getBitCast(annoGV, SBP),
617 llvm::ConstantExpr::getBitCast(unitGV, SBP),
618 llvm::ConstantInt::get(llvm::Type::Int32Ty, LineNo)
619 };
620 return llvm::ConstantStruct::get(Fields, 4, false);
621}
622
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000623void CodeGenModule::EmitGlobalVar(const VarDecl *D) {
Nate Begemanad320b62008-04-20 06:29:50 +0000624 // If the VarDecl is a static, defer code generation until later so we can
625 // easily omit unused statics.
626 if (D->getStorageClass() == VarDecl::Static) {
627 StaticDecls.push_back(D);
628 return;
629 }
630
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000631 // If this is just a forward declaration of the variable, don't emit it now,
632 // allow it to be emitted lazily on its first use.
Chris Lattner4b009652007-07-25 00:24:17 +0000633 if (D->getStorageClass() == VarDecl::Extern && D->getInit() == 0)
634 return;
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000635
Nate Begemanad320b62008-04-20 06:29:50 +0000636 EmitGlobalVarInit(D);
637}
638
639void CodeGenModule::EmitGlobalVarInit(const VarDecl *D) {
Eli Friedman43a0ce82008-05-30 19:50:47 +0000640 assert(D->hasGlobalStorage() && "Not a global variable");
641
Chris Lattner4b009652007-07-25 00:24:17 +0000642 llvm::Constant *Init = 0;
Eli Friedman43a0ce82008-05-30 19:50:47 +0000643 QualType ASTTy = D->getType();
644 const llvm::Type *VarTy = getTypes().ConvertTypeForMem(ASTTy);
645 const llvm::Type *VarPtrTy =
646 llvm::PointerType::get(VarTy, ASTTy.getAddressSpace());
647
Chris Lattner4b009652007-07-25 00:24:17 +0000648 if (D->getInit() == 0) {
Eli Friedman7008e9a2008-05-30 20:39:54 +0000649 // This is a tentative definition; tentative definitions are
650 // implicitly initialized with { 0 }
651 const llvm::Type* InitTy;
652 if (ASTTy->isIncompleteArrayType()) {
653 // An incomplete array is normally [ TYPE x 0 ], but we need
654 // to fix it to [ TYPE x 1 ].
655 const llvm::ArrayType* ATy = cast<llvm::ArrayType>(VarTy);
656 InitTy = llvm::ArrayType::get(ATy->getElementType(), 1);
657 } else {
658 InitTy = VarTy;
659 }
660 Init = llvm::Constant::getNullValue(InitTy);
Eli Friedman43a0ce82008-05-30 19:50:47 +0000661 } else {
662 Init = EmitGlobalInit(D->getInit());
663 }
664 const llvm::Type* InitType = Init->getType();
665
666 llvm::GlobalVariable *GV = getModule().getGlobalVariable(D->getName(), true);
667
668 if (!GV) {
669 GV = new llvm::GlobalVariable(InitType, false,
670 llvm::GlobalValue::ExternalLinkage,
671 0, D->getName(), &getModule(), 0,
672 ASTTy.getAddressSpace());
673 } else if (GV->getType()->getElementType() != InitType ||
674 GV->getType()->getAddressSpace() != ASTTy.getAddressSpace()) {
675 // We have a definition after a prototype with the wrong type.
676 // We must make a new GlobalVariable* and update everything that used OldGV
677 // (a declaration or tentative definition) with the new GlobalVariable*
678 // (which will be a definition).
679 //
680 // This happens if there is a prototype for a global (e.g. "extern int x[];")
681 // and then a definition of a different type (e.g. "int x[10];"). This also
682 // happens when an initializer has a different type from the type of the
683 // global (this happens with unions).
Eli Friedman7008e9a2008-05-30 20:39:54 +0000684 //
685 // FIXME: This also ends up happening if there's a definition followed by
686 // a tentative definition! (Although Sema rejects that construct
687 // at the moment.)
Eli Friedman43a0ce82008-05-30 19:50:47 +0000688
689 // Save the old global
690 llvm::GlobalVariable *OldGV = GV;
691
692 // Make a new global with the correct type
693 GV = new llvm::GlobalVariable(InitType, false,
694 llvm::GlobalValue::ExternalLinkage,
695 0, D->getName(), &getModule(), 0,
696 ASTTy.getAddressSpace());
697 // Steal the name of the old global
698 GV->takeName(OldGV);
699
700 // Replace all uses of the old global with the new global
701 llvm::Constant *NewPtrForOldDecl =
702 llvm::ConstantExpr::getBitCast(GV, OldGV->getType());
703 OldGV->replaceAllUsesWith(NewPtrForOldDecl);
704 // Make sure we don't keep around any stale references to globals
705 // FIXME: This is really slow; we need a better way to walk all
706 // the decls with the same name
707 ReplaceMapValuesWith(OldGV, NewPtrForOldDecl);
708
709 // Erase the old global, since it is no longer used.
710 OldGV->eraseFromParent();
Chris Lattner4b009652007-07-25 00:24:17 +0000711 }
Devang Patel8b5f5302007-10-26 16:31:40 +0000712
Eli Friedman43a0ce82008-05-30 19:50:47 +0000713 GlobalDeclMap[D] = llvm::ConstantExpr::getBitCast(GV, VarPtrTy);
Devang Patel8b5f5302007-10-26 16:31:40 +0000714
Nate Begeman8a704172008-04-19 04:17:09 +0000715 if (const AnnotateAttr *AA = D->getAttr<AnnotateAttr>()) {
716 SourceManager &SM = Context.getSourceManager();
717 AddAnnotation(EmitAnnotateAttr(GV, AA,
718 SM.getLogicalLineNumber(D->getLocation())));
719 }
720
Chris Lattner4b009652007-07-25 00:24:17 +0000721 GV->setInitializer(Init);
Chris Lattner402b3372008-03-03 03:28:21 +0000722
Eli Friedman7008e9a2008-05-30 20:39:54 +0000723 // FIXME: This is silly; getTypeAlign should just work for incomplete arrays
724 unsigned Align;
725 if (const IncompleteArrayType* IAT = D->getType()->getAsIncompleteArrayType())
726 Align = Context.getTypeAlign(IAT->getElementType());
727 else
728 Align = Context.getTypeAlign(D->getType());
Eli Friedmanb232e992008-05-29 11:10:27 +0000729 if (const AlignedAttr* AA = D->getAttr<AlignedAttr>()) {
730 Align = std::max(Align, AA->getAlignment());
731 }
732 GV->setAlignment(Align / 8);
733
Chris Lattner402b3372008-03-03 03:28:21 +0000734 if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>())
Dan Gohman4751a3a2008-05-22 00:50:06 +0000735 setVisibility(GV, attr->getVisibility());
Chris Lattner402b3372008-03-03 03:28:21 +0000736 // FIXME: else handle -fvisibility
Chris Lattner4b009652007-07-25 00:24:17 +0000737
738 // Set the llvm linkage type as appropriate.
Chris Lattner25094a42008-05-04 01:44:26 +0000739 if (D->getStorageClass() == VarDecl::Static)
740 GV->setLinkage(llvm::Function::InternalLinkage);
741 else if (D->getAttr<DLLImportAttr>())
Chris Lattner402b3372008-03-03 03:28:21 +0000742 GV->setLinkage(llvm::Function::DLLImportLinkage);
743 else if (D->getAttr<DLLExportAttr>())
744 GV->setLinkage(llvm::Function::DLLExportLinkage);
Chris Lattner25094a42008-05-04 01:44:26 +0000745 else if (D->getAttr<WeakAttr>())
Chris Lattner402b3372008-03-03 03:28:21 +0000746 GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
Chris Lattner25094a42008-05-04 01:44:26 +0000747 else {
Chris Lattner402b3372008-03-03 03:28:21 +0000748 // FIXME: This isn't right. This should handle common linkage and other
749 // stuff.
750 switch (D->getStorageClass()) {
Chris Lattner25094a42008-05-04 01:44:26 +0000751 case VarDecl::Static: assert(0 && "This case handled above");
Chris Lattner402b3372008-03-03 03:28:21 +0000752 case VarDecl::Auto:
753 case VarDecl::Register:
754 assert(0 && "Can't have auto or register globals");
755 case VarDecl::None:
756 if (!D->getInit())
Eli Friedmana7f46332008-05-29 11:03:17 +0000757 GV->setLinkage(llvm::GlobalVariable::CommonLinkage);
Chris Lattner402b3372008-03-03 03:28:21 +0000758 break;
759 case VarDecl::Extern:
760 case VarDecl::PrivateExtern:
761 // todo: common
762 break;
Chris Lattner402b3372008-03-03 03:28:21 +0000763 }
Chris Lattner4b009652007-07-25 00:24:17 +0000764 }
Sanjiv Gupta54d97542008-06-05 08:59:10 +0000765
766 // Emit global variable debug information.
767 CGDebugInfo *DI = getDebugInfo();
768 if(DI) {
769 if(D->getLocation().isValid())
770 DI->setLocation(D->getLocation());
771 DI->EmitGlobalVariable(GV, D);
772 }
Chris Lattner4b009652007-07-25 00:24:17 +0000773}
774
775/// EmitGlobalVarDeclarator - Emit all the global vars attached to the specified
776/// declarator chain.
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000777void CodeGenModule::EmitGlobalVarDeclarator(const VarDecl *D) {
778 for (; D; D = cast_or_null<VarDecl>(D->getNextDeclarator()))
779 if (D->isFileVarDecl())
780 EmitGlobalVar(D);
Chris Lattner4b009652007-07-25 00:24:17 +0000781}
782
Chris Lattner9ec3ca22008-02-06 05:08:19 +0000783void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
784 // Make sure that this type is translated.
785 Types.UpdateCompletedType(TD);
Chris Lattner1b22f8b2008-02-05 08:06:13 +0000786}
787
788
Chris Lattnerab862cc2007-08-31 04:31:45 +0000789/// getBuiltinLibFunction
790llvm::Function *CodeGenModule::getBuiltinLibFunction(unsigned BuiltinID) {
Chris Lattner9f2d6892007-12-13 00:38:03 +0000791 if (BuiltinID > BuiltinFunctions.size())
792 BuiltinFunctions.resize(BuiltinID);
Chris Lattnerab862cc2007-08-31 04:31:45 +0000793
Chris Lattner9f2d6892007-12-13 00:38:03 +0000794 // Cache looked up functions. Since builtin id #0 is invalid we don't reserve
795 // a slot for it.
796 assert(BuiltinID && "Invalid Builtin ID");
797 llvm::Function *&FunctionSlot = BuiltinFunctions[BuiltinID-1];
Chris Lattnerab862cc2007-08-31 04:31:45 +0000798 if (FunctionSlot)
799 return FunctionSlot;
800
801 assert(Context.BuiltinInfo.isLibFunction(BuiltinID) && "isn't a lib fn");
802
803 // Get the name, skip over the __builtin_ prefix.
804 const char *Name = Context.BuiltinInfo.GetName(BuiltinID)+10;
805
806 // Get the type for the builtin.
807 QualType Type = Context.BuiltinInfo.GetBuiltinType(BuiltinID, Context);
808 const llvm::FunctionType *Ty =
809 cast<llvm::FunctionType>(getTypes().ConvertType(Type));
810
811 // FIXME: This has a serious problem with code like this:
812 // void abs() {}
813 // ... __builtin_abs(x);
814 // The two versions of abs will collide. The fix is for the builtin to win,
815 // and for the existing one to be turned into a constantexpr cast of the
816 // builtin. In the case where the existing one is a static function, it
817 // should just be renamed.
Chris Lattner02c60f52007-08-31 04:44:06 +0000818 if (llvm::Function *Existing = getModule().getFunction(Name)) {
819 if (Existing->getFunctionType() == Ty && Existing->hasExternalLinkage())
820 return FunctionSlot = Existing;
821 assert(Existing == 0 && "FIXME: Name collision");
822 }
Chris Lattnerab862cc2007-08-31 04:31:45 +0000823
824 // FIXME: param attributes for sext/zext etc.
Nate Begemanad320b62008-04-20 06:29:50 +0000825 return FunctionSlot =
826 llvm::Function::Create(Ty, llvm::Function::ExternalLinkage, Name,
827 &getModule());
Chris Lattnerab862cc2007-08-31 04:31:45 +0000828}
829
Chris Lattner4b23f942007-12-18 00:25:38 +0000830llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,const llvm::Type **Tys,
831 unsigned NumTys) {
832 return llvm::Intrinsic::getDeclaration(&getModule(),
833 (llvm::Intrinsic::ID)IID, Tys, NumTys);
834}
Chris Lattnerab862cc2007-08-31 04:31:45 +0000835
Chris Lattner4b009652007-07-25 00:24:17 +0000836llvm::Function *CodeGenModule::getMemCpyFn() {
837 if (MemCpyFn) return MemCpyFn;
838 llvm::Intrinsic::ID IID;
Chris Lattner461a6c52008-03-08 08:34:58 +0000839 switch (Context.Target.getPointerWidth(0)) {
Chris Lattner4b009652007-07-25 00:24:17 +0000840 default: assert(0 && "Unknown ptr width");
841 case 32: IID = llvm::Intrinsic::memcpy_i32; break;
842 case 64: IID = llvm::Intrinsic::memcpy_i64; break;
843 }
Chris Lattner4b23f942007-12-18 00:25:38 +0000844 return MemCpyFn = getIntrinsic(IID);
Chris Lattner4b009652007-07-25 00:24:17 +0000845}
Anders Carlsson36a04872007-08-21 00:21:21 +0000846
Eli Friedman8f08a252008-05-26 12:59:39 +0000847llvm::Function *CodeGenModule::getMemMoveFn() {
848 if (MemMoveFn) return MemMoveFn;
849 llvm::Intrinsic::ID IID;
850 switch (Context.Target.getPointerWidth(0)) {
851 default: assert(0 && "Unknown ptr width");
852 case 32: IID = llvm::Intrinsic::memmove_i32; break;
853 case 64: IID = llvm::Intrinsic::memmove_i64; break;
854 }
855 return MemMoveFn = getIntrinsic(IID);
856}
857
Lauro Ramos Venancioe5bef732008-02-19 22:01:01 +0000858llvm::Function *CodeGenModule::getMemSetFn() {
859 if (MemSetFn) return MemSetFn;
860 llvm::Intrinsic::ID IID;
Chris Lattner461a6c52008-03-08 08:34:58 +0000861 switch (Context.Target.getPointerWidth(0)) {
Lauro Ramos Venancioe5bef732008-02-19 22:01:01 +0000862 default: assert(0 && "Unknown ptr width");
863 case 32: IID = llvm::Intrinsic::memset_i32; break;
864 case 64: IID = llvm::Intrinsic::memset_i64; break;
865 }
866 return MemSetFn = getIntrinsic(IID);
867}
Chris Lattner4b23f942007-12-18 00:25:38 +0000868
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000869// FIXME: This needs moving into an Apple Objective-C runtime class
Chris Lattnerab862cc2007-08-31 04:31:45 +0000870llvm::Constant *CodeGenModule::
871GetAddrOfConstantCFString(const std::string &str) {
Anders Carlsson36a04872007-08-21 00:21:21 +0000872 llvm::StringMapEntry<llvm::Constant *> &Entry =
873 CFConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
874
875 if (Entry.getValue())
876 return Entry.getValue();
877
878 std::vector<llvm::Constant*> Fields;
879
880 if (!CFConstantStringClassRef) {
881 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
882 Ty = llvm::ArrayType::get(Ty, 0);
883
884 CFConstantStringClassRef =
885 new llvm::GlobalVariable(Ty, false,
886 llvm::GlobalVariable::ExternalLinkage, 0,
887 "__CFConstantStringClassReference",
888 &getModule());
889 }
890
891 // Class pointer.
892 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty);
893 llvm::Constant *Zeros[] = { Zero, Zero };
894 llvm::Constant *C =
895 llvm::ConstantExpr::getGetElementPtr(CFConstantStringClassRef, Zeros, 2);
896 Fields.push_back(C);
897
898 // Flags.
899 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
900 Fields.push_back(llvm::ConstantInt::get(Ty, 1992));
901
902 // String pointer.
903 C = llvm::ConstantArray::get(str);
904 C = new llvm::GlobalVariable(C->getType(), true,
905 llvm::GlobalValue::InternalLinkage,
906 C, ".str", &getModule());
907
908 C = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2);
909 Fields.push_back(C);
910
911 // String length.
912 Ty = getTypes().ConvertType(getContext().LongTy);
913 Fields.push_back(llvm::ConstantInt::get(Ty, str.length()));
914
915 // The struct.
916 Ty = getTypes().ConvertType(getContext().getCFConstantStringType());
917 C = llvm::ConstantStruct::get(cast<llvm::StructType>(Ty), Fields);
Anders Carlsson9be009e2007-11-01 00:41:52 +0000918 llvm::GlobalVariable *GV =
919 new llvm::GlobalVariable(C->getType(), true,
920 llvm::GlobalVariable::InternalLinkage,
921 C, "", &getModule());
922 GV->setSection("__DATA,__cfstring");
923 Entry.setValue(GV);
924 return GV;
Anders Carlsson36a04872007-08-21 00:21:21 +0000925}
Chris Lattnerdb6be562007-11-28 05:34:05 +0000926
Chris Lattnera6dcce32008-02-11 00:02:17 +0000927/// GenerateWritableString -- Creates storage for a string literal.
Chris Lattnerdb6be562007-11-28 05:34:05 +0000928static llvm::Constant *GenerateStringLiteral(const std::string &str,
929 bool constant,
Chris Lattnercf9c9d02007-12-02 07:19:18 +0000930 CodeGenModule &CGM) {
Chris Lattnerdb6be562007-11-28 05:34:05 +0000931 // Create Constant for this string literal
932 llvm::Constant *C=llvm::ConstantArray::get(str);
933
934 // Create a global variable for this string
935 C = new llvm::GlobalVariable(C->getType(), constant,
936 llvm::GlobalValue::InternalLinkage,
Chris Lattnercf9c9d02007-12-02 07:19:18 +0000937 C, ".str", &CGM.getModule());
Chris Lattnerdb6be562007-11-28 05:34:05 +0000938 return C;
939}
940
Chris Lattnera6dcce32008-02-11 00:02:17 +0000941/// CodeGenModule::GetAddrOfConstantString -- returns a pointer to the character
942/// array containing the literal. The result is pointer to array type.
Chris Lattnerdb6be562007-11-28 05:34:05 +0000943llvm::Constant *CodeGenModule::GetAddrOfConstantString(const std::string &str) {
944 // Don't share any string literals if writable-strings is turned on.
945 if (Features.WritableStrings)
946 return GenerateStringLiteral(str, false, *this);
947
948 llvm::StringMapEntry<llvm::Constant *> &Entry =
949 ConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
950
951 if (Entry.getValue())
952 return Entry.getValue();
953
954 // Create a global variable for this.
955 llvm::Constant *C = GenerateStringLiteral(str, true, *this);
956 Entry.setValue(C);
957 return C;
958}