blob: b9c15427c3fe0e9024e9991bffa4fc19ced0396f [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;
365 for (unsigned i = 0, e = PD->getNumReferencedProtocols() ; i < e ; i++)
366 Protocols.push_back(PD->getReferencedProtocols()[i]->getName());
367 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
368 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
369 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
370 endIter = PD->instmeth_end() ; iter != endIter ; iter++) {
371 std::string TypeStr;
372 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
373 InstanceMethodNames.push_back(
374 GetAddrOfConstantString((*iter)->getSelector().getName()));
375 InstanceMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
376 }
377 // Collect information about class methods:
378 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
379 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
380 for (ObjCProtocolDecl::classmeth_iterator iter = PD->classmeth_begin(),
381 endIter = PD->classmeth_end() ; iter != endIter ; iter++) {
382 std::string TypeStr;
383 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
384 ClassMethodNames.push_back(
385 GetAddrOfConstantString((*iter)->getSelector().getName()));
386 ClassMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
387 }
388 Runtime->GenerateProtocol(PD->getName(), Protocols, InstanceMethodNames,
389 InstanceMethodTypes, ClassMethodNames, ClassMethodTypes);
390}
391
392void CodeGenModule::EmitObjCCategoryImpl(const ObjCCategoryImplDecl *OCD) {
393
394 // Collect information about instance methods
Chris Lattner578279d2008-06-26 05:08:00 +0000395 llvm::SmallVector<Selector, 16> InstanceMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000396 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
397 for (ObjCCategoryDecl::instmeth_iterator iter = OCD->instmeth_begin(),
398 endIter = OCD->instmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000399 InstanceMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000400 std::string TypeStr;
Chris Lattnerbcb3e862008-06-26 04:52:29 +0000401 Context.getObjCEncodingForMethodDecl(*iter,TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000402 InstanceMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
403 }
404
405 // Collect information about class methods
Chris Lattner578279d2008-06-26 05:08:00 +0000406 llvm::SmallVector<Selector, 16> ClassMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000407 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
408 for (ObjCCategoryDecl::classmeth_iterator iter = OCD->classmeth_begin(),
409 endIter = OCD->classmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000410 ClassMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000411 std::string TypeStr;
Chris Lattnerbcb3e862008-06-26 04:52:29 +0000412 Context.getObjCEncodingForMethodDecl(*iter,TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000413 ClassMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
414 }
415
416 // Collect the names of referenced protocols
417 llvm::SmallVector<std::string, 16> Protocols;
Chris Lattner8bcb5252008-07-21 18:19:38 +0000418 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
419 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
420 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
421 E = Protos.end(); I != E; ++I)
422 Protocols.push_back((*I)->getName());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000423
424 // Generate the category
425 Runtime->GenerateCategory(OCD->getClassInterface()->getName(),
Chris Lattner578279d2008-06-26 05:08:00 +0000426 OCD->getName(), InstanceMethodSels, InstanceMethodTypes,
427 ClassMethodSels, ClassMethodTypes, Protocols);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000428}
429
430void CodeGenModule::EmitObjCClassImplementation(
431 const ObjCImplementationDecl *OID) {
432 // Get the superclass name.
433 const ObjCInterfaceDecl * SCDecl = OID->getClassInterface()->getSuperClass();
434 const char * SCName = NULL;
435 if (SCDecl) {
436 SCName = SCDecl->getName();
437 }
438
439 // Get the class name
440 ObjCInterfaceDecl * ClassDecl = (ObjCInterfaceDecl*)OID->getClassInterface();
441 const char * ClassName = ClassDecl->getName();
442
443 // Get the size of instances. For runtimes that support late-bound instances
444 // this should probably be something different (size just of instance
445 // varaibles in this class, not superclasses?).
446 int instanceSize = 0;
447 const llvm::Type *ObjTy;
448 if (!Runtime->LateBoundIVars()) {
449 ObjTy = getTypes().ConvertType(Context.getObjCInterfaceType(ClassDecl));
450 instanceSize = TheTargetData.getABITypeSize(ObjTy);
451 }
452
453 // Collect information about instance variables.
454 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
455 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
456 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
457 const llvm::StructLayout *Layout =
458 TheTargetData.getStructLayout(cast<llvm::StructType>(ObjTy));
459 ObjTy = llvm::PointerType::getUnqual(ObjTy);
460 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
461 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
462 // Store the name
463 IvarNames.push_back(GetAddrOfConstantString((*iter)->getName()));
464 // Get the type encoding for this ivar
465 std::string TypeStr;
466 llvm::SmallVector<const RecordType *, 8> EncodingRecordTypes;
467 Context.getObjCEncodingForType((*iter)->getType(), TypeStr,
468 EncodingRecordTypes);
469 IvarTypes.push_back(GetAddrOfConstantString(TypeStr));
470 // Get the offset
471 int offset =
472 (int)Layout->getElementOffset(getTypes().getLLVMFieldNo(*iter));
473 IvarOffsets.push_back(
474 llvm::ConstantInt::get(llvm::Type::Int32Ty, offset));
475 }
476
477 // Collect information about instance methods
Chris Lattner578279d2008-06-26 05:08:00 +0000478 llvm::SmallVector<Selector, 16> InstanceMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000479 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
480 for (ObjCImplementationDecl::instmeth_iterator iter = OID->instmeth_begin(),
481 endIter = OID->instmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000482 InstanceMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000483 std::string TypeStr;
484 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000485 InstanceMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
486 }
487
488 // Collect information about class methods
Chris Lattner578279d2008-06-26 05:08:00 +0000489 llvm::SmallVector<Selector, 16> ClassMethodSels;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000490 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
491 for (ObjCImplementationDecl::classmeth_iterator iter = OID->classmeth_begin(),
492 endIter = OID->classmeth_end() ; iter != endIter ; iter++) {
Chris Lattner578279d2008-06-26 05:08:00 +0000493 ClassMethodSels.push_back((*iter)->getSelector());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000494 std::string TypeStr;
495 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000496 ClassMethodTypes.push_back(GetAddrOfConstantString(TypeStr));
497 }
498 // Collect the names of referenced protocols
499 llvm::SmallVector<std::string, 16> Protocols;
Chris Lattner8bcb5252008-07-21 18:19:38 +0000500 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
501 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
502 E = Protos.end(); I != E; ++I)
503 Protocols.push_back((*I)->getName());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000504
505 // Generate the category
506 Runtime->GenerateClass(ClassName, SCName, instanceSize, IvarNames, IvarTypes,
Chris Lattner578279d2008-06-26 05:08:00 +0000507 IvarOffsets, InstanceMethodSels, InstanceMethodTypes,
508 ClassMethodSels, ClassMethodTypes, Protocols);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000509}
510
Chris Lattnerb326b172008-03-30 23:03:07 +0000511
Chris Lattner4b009652007-07-25 00:24:17 +0000512void CodeGenModule::EmitFunction(const FunctionDecl *FD) {
513 // If this is not a prototype, emit the body.
Nuno Lopes78534382008-06-08 15:45:52 +0000514 if (!FD->isThisDeclarationADefinition()) {
515 if (FD->getAttr<AliasAttr>())
516 GetAddrOfFunctionDecl(FD, true);
Chris Lattner74bf5552008-05-04 02:29:49 +0000517 return;
Nuno Lopes78534382008-06-08 15:45:52 +0000518 }
Chris Lattner74bf5552008-05-04 02:29:49 +0000519
520 // If the function is a static, defer code generation until later so we can
521 // easily omit unused statics.
522 if (FD->getStorageClass() != FunctionDecl::Static) {
Chris Lattner4b009652007-07-25 00:24:17 +0000523 CodeGenFunction(*this).GenerateCode(FD);
Chris Lattner74bf5552008-05-04 02:29:49 +0000524 return;
Nate Begemanad320b62008-04-20 06:29:50 +0000525 }
Chris Lattner74bf5552008-05-04 02:29:49 +0000526
527 StaticDecls.push_back(FD);
Nate Begemanad320b62008-04-20 06:29:50 +0000528}
529
530void CodeGenModule::EmitStatics() {
531 // Emit code for each used static decl encountered. Since a previously unused
532 // static decl may become used during the generation of code for a static
533 // function, iterate until no changes are made.
534 bool Changed;
535 do {
536 Changed = false;
537 for (unsigned i = 0, e = StaticDecls.size(); i != e; ++i) {
Nate Begemanad320b62008-04-20 06:29:50 +0000538 const Decl *D = StaticDecls[i];
Eli Friedmana4d4e2f2008-05-27 04:58:01 +0000539
540 // Check if we have used a decl with the same name
541 // FIXME: The AST should have some sort of aggregate decls or
542 // global symbol map.
543 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
544 if (!getModule().getFunction(FD->getName()))
545 continue;
546 } else {
547 if (!getModule().getNamedGlobal(cast<VarDecl>(D)->getName()))
548 continue;
549 }
550
Nate Begemanad320b62008-04-20 06:29:50 +0000551 // If this is a function decl, generate code for the static function if it
552 // has a body. Otherwise, we must have a var decl for a static global
553 // variable.
554 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
555 if (FD->getBody())
556 CodeGenFunction(*this).GenerateCode(FD);
Nuno Lopes78534382008-06-08 15:45:52 +0000557 else if (FD->getAttr<AliasAttr>())
558 GetAddrOfFunctionDecl(FD, true);
Nate Begemanad320b62008-04-20 06:29:50 +0000559 } else {
Nate Begeman91f52012008-04-20 20:38:08 +0000560 EmitGlobalVarInit(cast<VarDecl>(D));
Nate Begemanad320b62008-04-20 06:29:50 +0000561 }
562 // Erase the used decl from the list.
563 StaticDecls[i] = StaticDecls.back();
564 StaticDecls.pop_back();
565 --i;
566 --e;
567
568 // Remember that we made a change.
569 Changed = true;
570 }
571 } while (Changed);
Chris Lattner4b009652007-07-25 00:24:17 +0000572}
573
Anders Carlssond76cead2008-01-26 01:36:00 +0000574llvm::Constant *CodeGenModule::EmitGlobalInit(const Expr *Expr) {
575 return EmitConstantExpr(Expr);
Devang Patel08a10cc2007-10-30 21:27:20 +0000576}
577
Nate Begeman8a704172008-04-19 04:17:09 +0000578/// EmitAnnotateAttr - Generate the llvm::ConstantStruct which contains the
579/// annotation information for a given GlobalValue. The annotation struct is
580/// {i8 *, i8 *, i8 *, i32}. The first field is a constant expression, the
581/// GlobalValue being annotated. The second filed is thee constant string
582/// created from the AnnotateAttr's annotation. The third field is a constant
583/// string containing the name of the translation unit. The fourth field is
584/// the line number in the file of the annotated value declaration.
585///
586/// FIXME: this does not unique the annotation string constants, as llvm-gcc
587/// appears to.
588///
589llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
590 const AnnotateAttr *AA,
591 unsigned LineNo) {
592 llvm::Module *M = &getModule();
593
594 // get [N x i8] constants for the annotation string, and the filename string
595 // which are the 2nd and 3rd elements of the global annotation structure.
596 const llvm::Type *SBP = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
597 llvm::Constant *anno = llvm::ConstantArray::get(AA->getAnnotation(), true);
598 llvm::Constant *unit = llvm::ConstantArray::get(M->getModuleIdentifier(),
599 true);
600
601 // Get the two global values corresponding to the ConstantArrays we just
602 // created to hold the bytes of the strings.
603 llvm::GlobalValue *annoGV =
604 new llvm::GlobalVariable(anno->getType(), false,
605 llvm::GlobalValue::InternalLinkage, anno,
606 GV->getName() + ".str", M);
607 // translation unit name string, emitted into the llvm.metadata section.
608 llvm::GlobalValue *unitGV =
609 new llvm::GlobalVariable(unit->getType(), false,
610 llvm::GlobalValue::InternalLinkage, unit, ".str", M);
611
612 // Create the ConstantStruct that is the global annotion.
613 llvm::Constant *Fields[4] = {
614 llvm::ConstantExpr::getBitCast(GV, SBP),
615 llvm::ConstantExpr::getBitCast(annoGV, SBP),
616 llvm::ConstantExpr::getBitCast(unitGV, SBP),
617 llvm::ConstantInt::get(llvm::Type::Int32Ty, LineNo)
618 };
619 return llvm::ConstantStruct::get(Fields, 4, false);
620}
621
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000622void CodeGenModule::EmitGlobalVar(const VarDecl *D) {
Nate Begemanad320b62008-04-20 06:29:50 +0000623 // If the VarDecl is a static, defer code generation until later so we can
624 // easily omit unused statics.
625 if (D->getStorageClass() == VarDecl::Static) {
626 StaticDecls.push_back(D);
627 return;
628 }
629
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000630 // If this is just a forward declaration of the variable, don't emit it now,
631 // allow it to be emitted lazily on its first use.
Chris Lattner4b009652007-07-25 00:24:17 +0000632 if (D->getStorageClass() == VarDecl::Extern && D->getInit() == 0)
633 return;
Chris Lattner1a3c1e22007-12-02 07:09:19 +0000634
Nate Begemanad320b62008-04-20 06:29:50 +0000635 EmitGlobalVarInit(D);
636}
637
638void CodeGenModule::EmitGlobalVarInit(const VarDecl *D) {
Eli Friedman43a0ce82008-05-30 19:50:47 +0000639 assert(D->hasGlobalStorage() && "Not a global variable");
640
Chris Lattner4b009652007-07-25 00:24:17 +0000641 llvm::Constant *Init = 0;
Eli Friedman43a0ce82008-05-30 19:50:47 +0000642 QualType ASTTy = D->getType();
643 const llvm::Type *VarTy = getTypes().ConvertTypeForMem(ASTTy);
644 const llvm::Type *VarPtrTy =
645 llvm::PointerType::get(VarTy, ASTTy.getAddressSpace());
646
Chris Lattner4b009652007-07-25 00:24:17 +0000647 if (D->getInit() == 0) {
Eli Friedman7008e9a2008-05-30 20:39:54 +0000648 // This is a tentative definition; tentative definitions are
649 // implicitly initialized with { 0 }
650 const llvm::Type* InitTy;
651 if (ASTTy->isIncompleteArrayType()) {
652 // An incomplete array is normally [ TYPE x 0 ], but we need
653 // to fix it to [ TYPE x 1 ].
654 const llvm::ArrayType* ATy = cast<llvm::ArrayType>(VarTy);
655 InitTy = llvm::ArrayType::get(ATy->getElementType(), 1);
656 } else {
657 InitTy = VarTy;
658 }
659 Init = llvm::Constant::getNullValue(InitTy);
Eli Friedman43a0ce82008-05-30 19:50:47 +0000660 } else {
661 Init = EmitGlobalInit(D->getInit());
662 }
663 const llvm::Type* InitType = Init->getType();
664
665 llvm::GlobalVariable *GV = getModule().getGlobalVariable(D->getName(), true);
666
667 if (!GV) {
668 GV = new llvm::GlobalVariable(InitType, false,
669 llvm::GlobalValue::ExternalLinkage,
670 0, D->getName(), &getModule(), 0,
671 ASTTy.getAddressSpace());
672 } else if (GV->getType()->getElementType() != InitType ||
673 GV->getType()->getAddressSpace() != ASTTy.getAddressSpace()) {
674 // We have a definition after a prototype with the wrong type.
675 // We must make a new GlobalVariable* and update everything that used OldGV
676 // (a declaration or tentative definition) with the new GlobalVariable*
677 // (which will be a definition).
678 //
679 // This happens if there is a prototype for a global (e.g. "extern int x[];")
680 // and then a definition of a different type (e.g. "int x[10];"). This also
681 // happens when an initializer has a different type from the type of the
682 // global (this happens with unions).
Eli Friedman7008e9a2008-05-30 20:39:54 +0000683 //
684 // FIXME: This also ends up happening if there's a definition followed by
685 // a tentative definition! (Although Sema rejects that construct
686 // at the moment.)
Eli Friedman43a0ce82008-05-30 19:50:47 +0000687
688 // Save the old global
689 llvm::GlobalVariable *OldGV = GV;
690
691 // Make a new global with the correct type
692 GV = new llvm::GlobalVariable(InitType, false,
693 llvm::GlobalValue::ExternalLinkage,
694 0, D->getName(), &getModule(), 0,
695 ASTTy.getAddressSpace());
696 // Steal the name of the old global
697 GV->takeName(OldGV);
698
699 // Replace all uses of the old global with the new global
700 llvm::Constant *NewPtrForOldDecl =
701 llvm::ConstantExpr::getBitCast(GV, OldGV->getType());
702 OldGV->replaceAllUsesWith(NewPtrForOldDecl);
703 // Make sure we don't keep around any stale references to globals
704 // FIXME: This is really slow; we need a better way to walk all
705 // the decls with the same name
706 ReplaceMapValuesWith(OldGV, NewPtrForOldDecl);
707
708 // Erase the old global, since it is no longer used.
709 OldGV->eraseFromParent();
Chris Lattner4b009652007-07-25 00:24:17 +0000710 }
Devang Patel8b5f5302007-10-26 16:31:40 +0000711
Eli Friedman43a0ce82008-05-30 19:50:47 +0000712 GlobalDeclMap[D] = llvm::ConstantExpr::getBitCast(GV, VarPtrTy);
Devang Patel8b5f5302007-10-26 16:31:40 +0000713
Nate Begeman8a704172008-04-19 04:17:09 +0000714 if (const AnnotateAttr *AA = D->getAttr<AnnotateAttr>()) {
715 SourceManager &SM = Context.getSourceManager();
716 AddAnnotation(EmitAnnotateAttr(GV, AA,
717 SM.getLogicalLineNumber(D->getLocation())));
718 }
719
Chris Lattner4b009652007-07-25 00:24:17 +0000720 GV->setInitializer(Init);
Chris Lattner402b3372008-03-03 03:28:21 +0000721
Eli Friedman7008e9a2008-05-30 20:39:54 +0000722 // FIXME: This is silly; getTypeAlign should just work for incomplete arrays
723 unsigned Align;
724 if (const IncompleteArrayType* IAT = D->getType()->getAsIncompleteArrayType())
725 Align = Context.getTypeAlign(IAT->getElementType());
726 else
727 Align = Context.getTypeAlign(D->getType());
Eli Friedmanb232e992008-05-29 11:10:27 +0000728 if (const AlignedAttr* AA = D->getAttr<AlignedAttr>()) {
729 Align = std::max(Align, AA->getAlignment());
730 }
731 GV->setAlignment(Align / 8);
732
Chris Lattner402b3372008-03-03 03:28:21 +0000733 if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>())
Dan Gohman4751a3a2008-05-22 00:50:06 +0000734 setVisibility(GV, attr->getVisibility());
Chris Lattner402b3372008-03-03 03:28:21 +0000735 // FIXME: else handle -fvisibility
Chris Lattner4b009652007-07-25 00:24:17 +0000736
737 // Set the llvm linkage type as appropriate.
Chris Lattner25094a42008-05-04 01:44:26 +0000738 if (D->getStorageClass() == VarDecl::Static)
739 GV->setLinkage(llvm::Function::InternalLinkage);
740 else if (D->getAttr<DLLImportAttr>())
Chris Lattner402b3372008-03-03 03:28:21 +0000741 GV->setLinkage(llvm::Function::DLLImportLinkage);
742 else if (D->getAttr<DLLExportAttr>())
743 GV->setLinkage(llvm::Function::DLLExportLinkage);
Chris Lattner25094a42008-05-04 01:44:26 +0000744 else if (D->getAttr<WeakAttr>())
Chris Lattner402b3372008-03-03 03:28:21 +0000745 GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
Chris Lattner25094a42008-05-04 01:44:26 +0000746 else {
Chris Lattner402b3372008-03-03 03:28:21 +0000747 // FIXME: This isn't right. This should handle common linkage and other
748 // stuff.
749 switch (D->getStorageClass()) {
Chris Lattner25094a42008-05-04 01:44:26 +0000750 case VarDecl::Static: assert(0 && "This case handled above");
Chris Lattner402b3372008-03-03 03:28:21 +0000751 case VarDecl::Auto:
752 case VarDecl::Register:
753 assert(0 && "Can't have auto or register globals");
754 case VarDecl::None:
755 if (!D->getInit())
Eli Friedmana7f46332008-05-29 11:03:17 +0000756 GV->setLinkage(llvm::GlobalVariable::CommonLinkage);
Chris Lattner402b3372008-03-03 03:28:21 +0000757 break;
758 case VarDecl::Extern:
759 case VarDecl::PrivateExtern:
760 // todo: common
761 break;
Chris Lattner402b3372008-03-03 03:28:21 +0000762 }
Chris Lattner4b009652007-07-25 00:24:17 +0000763 }
Sanjiv Gupta54d97542008-06-05 08:59:10 +0000764
765 // Emit global variable debug information.
766 CGDebugInfo *DI = getDebugInfo();
767 if(DI) {
768 if(D->getLocation().isValid())
769 DI->setLocation(D->getLocation());
770 DI->EmitGlobalVariable(GV, D);
771 }
Chris Lattner4b009652007-07-25 00:24:17 +0000772}
773
774/// EmitGlobalVarDeclarator - Emit all the global vars attached to the specified
775/// declarator chain.
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000776void CodeGenModule::EmitGlobalVarDeclarator(const VarDecl *D) {
777 for (; D; D = cast_or_null<VarDecl>(D->getNextDeclarator()))
778 if (D->isFileVarDecl())
779 EmitGlobalVar(D);
Chris Lattner4b009652007-07-25 00:24:17 +0000780}
781
Chris Lattner9ec3ca22008-02-06 05:08:19 +0000782void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
783 // Make sure that this type is translated.
784 Types.UpdateCompletedType(TD);
Chris Lattner1b22f8b2008-02-05 08:06:13 +0000785}
786
787
Chris Lattnerab862cc2007-08-31 04:31:45 +0000788/// getBuiltinLibFunction
789llvm::Function *CodeGenModule::getBuiltinLibFunction(unsigned BuiltinID) {
Chris Lattner9f2d6892007-12-13 00:38:03 +0000790 if (BuiltinID > BuiltinFunctions.size())
791 BuiltinFunctions.resize(BuiltinID);
Chris Lattnerab862cc2007-08-31 04:31:45 +0000792
Chris Lattner9f2d6892007-12-13 00:38:03 +0000793 // Cache looked up functions. Since builtin id #0 is invalid we don't reserve
794 // a slot for it.
795 assert(BuiltinID && "Invalid Builtin ID");
796 llvm::Function *&FunctionSlot = BuiltinFunctions[BuiltinID-1];
Chris Lattnerab862cc2007-08-31 04:31:45 +0000797 if (FunctionSlot)
798 return FunctionSlot;
799
800 assert(Context.BuiltinInfo.isLibFunction(BuiltinID) && "isn't a lib fn");
801
802 // Get the name, skip over the __builtin_ prefix.
803 const char *Name = Context.BuiltinInfo.GetName(BuiltinID)+10;
804
805 // Get the type for the builtin.
806 QualType Type = Context.BuiltinInfo.GetBuiltinType(BuiltinID, Context);
807 const llvm::FunctionType *Ty =
808 cast<llvm::FunctionType>(getTypes().ConvertType(Type));
809
810 // FIXME: This has a serious problem with code like this:
811 // void abs() {}
812 // ... __builtin_abs(x);
813 // The two versions of abs will collide. The fix is for the builtin to win,
814 // and for the existing one to be turned into a constantexpr cast of the
815 // builtin. In the case where the existing one is a static function, it
816 // should just be renamed.
Chris Lattner02c60f52007-08-31 04:44:06 +0000817 if (llvm::Function *Existing = getModule().getFunction(Name)) {
818 if (Existing->getFunctionType() == Ty && Existing->hasExternalLinkage())
819 return FunctionSlot = Existing;
820 assert(Existing == 0 && "FIXME: Name collision");
821 }
Chris Lattnerab862cc2007-08-31 04:31:45 +0000822
823 // FIXME: param attributes for sext/zext etc.
Nate Begemanad320b62008-04-20 06:29:50 +0000824 return FunctionSlot =
825 llvm::Function::Create(Ty, llvm::Function::ExternalLinkage, Name,
826 &getModule());
Chris Lattnerab862cc2007-08-31 04:31:45 +0000827}
828
Chris Lattner4b23f942007-12-18 00:25:38 +0000829llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,const llvm::Type **Tys,
830 unsigned NumTys) {
831 return llvm::Intrinsic::getDeclaration(&getModule(),
832 (llvm::Intrinsic::ID)IID, Tys, NumTys);
833}
Chris Lattnerab862cc2007-08-31 04:31:45 +0000834
Chris Lattner4b009652007-07-25 00:24:17 +0000835llvm::Function *CodeGenModule::getMemCpyFn() {
836 if (MemCpyFn) return MemCpyFn;
837 llvm::Intrinsic::ID IID;
Chris Lattner461a6c52008-03-08 08:34:58 +0000838 switch (Context.Target.getPointerWidth(0)) {
Chris Lattner4b009652007-07-25 00:24:17 +0000839 default: assert(0 && "Unknown ptr width");
840 case 32: IID = llvm::Intrinsic::memcpy_i32; break;
841 case 64: IID = llvm::Intrinsic::memcpy_i64; break;
842 }
Chris Lattner4b23f942007-12-18 00:25:38 +0000843 return MemCpyFn = getIntrinsic(IID);
Chris Lattner4b009652007-07-25 00:24:17 +0000844}
Anders Carlsson36a04872007-08-21 00:21:21 +0000845
Eli Friedman8f08a252008-05-26 12:59:39 +0000846llvm::Function *CodeGenModule::getMemMoveFn() {
847 if (MemMoveFn) return MemMoveFn;
848 llvm::Intrinsic::ID IID;
849 switch (Context.Target.getPointerWidth(0)) {
850 default: assert(0 && "Unknown ptr width");
851 case 32: IID = llvm::Intrinsic::memmove_i32; break;
852 case 64: IID = llvm::Intrinsic::memmove_i64; break;
853 }
854 return MemMoveFn = getIntrinsic(IID);
855}
856
Lauro Ramos Venancioe5bef732008-02-19 22:01:01 +0000857llvm::Function *CodeGenModule::getMemSetFn() {
858 if (MemSetFn) return MemSetFn;
859 llvm::Intrinsic::ID IID;
Chris Lattner461a6c52008-03-08 08:34:58 +0000860 switch (Context.Target.getPointerWidth(0)) {
Lauro Ramos Venancioe5bef732008-02-19 22:01:01 +0000861 default: assert(0 && "Unknown ptr width");
862 case 32: IID = llvm::Intrinsic::memset_i32; break;
863 case 64: IID = llvm::Intrinsic::memset_i64; break;
864 }
865 return MemSetFn = getIntrinsic(IID);
866}
Chris Lattner4b23f942007-12-18 00:25:38 +0000867
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000868// FIXME: This needs moving into an Apple Objective-C runtime class
Chris Lattnerab862cc2007-08-31 04:31:45 +0000869llvm::Constant *CodeGenModule::
870GetAddrOfConstantCFString(const std::string &str) {
Anders Carlsson36a04872007-08-21 00:21:21 +0000871 llvm::StringMapEntry<llvm::Constant *> &Entry =
872 CFConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
873
874 if (Entry.getValue())
875 return Entry.getValue();
876
877 std::vector<llvm::Constant*> Fields;
878
879 if (!CFConstantStringClassRef) {
880 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
881 Ty = llvm::ArrayType::get(Ty, 0);
882
883 CFConstantStringClassRef =
884 new llvm::GlobalVariable(Ty, false,
885 llvm::GlobalVariable::ExternalLinkage, 0,
886 "__CFConstantStringClassReference",
887 &getModule());
888 }
889
890 // Class pointer.
891 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty);
892 llvm::Constant *Zeros[] = { Zero, Zero };
893 llvm::Constant *C =
894 llvm::ConstantExpr::getGetElementPtr(CFConstantStringClassRef, Zeros, 2);
895 Fields.push_back(C);
896
897 // Flags.
898 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
899 Fields.push_back(llvm::ConstantInt::get(Ty, 1992));
900
901 // String pointer.
902 C = llvm::ConstantArray::get(str);
903 C = new llvm::GlobalVariable(C->getType(), true,
904 llvm::GlobalValue::InternalLinkage,
905 C, ".str", &getModule());
906
907 C = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2);
908 Fields.push_back(C);
909
910 // String length.
911 Ty = getTypes().ConvertType(getContext().LongTy);
912 Fields.push_back(llvm::ConstantInt::get(Ty, str.length()));
913
914 // The struct.
915 Ty = getTypes().ConvertType(getContext().getCFConstantStringType());
916 C = llvm::ConstantStruct::get(cast<llvm::StructType>(Ty), Fields);
Anders Carlsson9be009e2007-11-01 00:41:52 +0000917 llvm::GlobalVariable *GV =
918 new llvm::GlobalVariable(C->getType(), true,
919 llvm::GlobalVariable::InternalLinkage,
920 C, "", &getModule());
921 GV->setSection("__DATA,__cfstring");
922 Entry.setValue(GV);
923 return GV;
Anders Carlsson36a04872007-08-21 00:21:21 +0000924}
Chris Lattnerdb6be562007-11-28 05:34:05 +0000925
Chris Lattnera6dcce32008-02-11 00:02:17 +0000926/// GenerateWritableString -- Creates storage for a string literal.
Chris Lattnerdb6be562007-11-28 05:34:05 +0000927static llvm::Constant *GenerateStringLiteral(const std::string &str,
928 bool constant,
Chris Lattnercf9c9d02007-12-02 07:19:18 +0000929 CodeGenModule &CGM) {
Chris Lattnerdb6be562007-11-28 05:34:05 +0000930 // Create Constant for this string literal
931 llvm::Constant *C=llvm::ConstantArray::get(str);
932
933 // Create a global variable for this string
934 C = new llvm::GlobalVariable(C->getType(), constant,
935 llvm::GlobalValue::InternalLinkage,
Chris Lattnercf9c9d02007-12-02 07:19:18 +0000936 C, ".str", &CGM.getModule());
Chris Lattnerdb6be562007-11-28 05:34:05 +0000937 return C;
938}
939
Chris Lattnera6dcce32008-02-11 00:02:17 +0000940/// CodeGenModule::GetAddrOfConstantString -- returns a pointer to the character
941/// array containing the literal. The result is pointer to array type.
Chris Lattnerdb6be562007-11-28 05:34:05 +0000942llvm::Constant *CodeGenModule::GetAddrOfConstantString(const std::string &str) {
943 // Don't share any string literals if writable-strings is turned on.
944 if (Features.WritableStrings)
945 return GenerateStringLiteral(str, false, *this);
946
947 llvm::StringMapEntry<llvm::Constant *> &Entry =
948 ConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
949
950 if (Entry.getValue())
951 return Entry.getValue();
952
953 // Create a global variable for this.
954 llvm::Constant *C = GenerateStringLiteral(str, true, *this);
955 Entry.setValue(C);
956 return C;
957}