blob: 8f7ae2eb56a70dd588bbf159eb8d4f85c7188bb2 [file] [log] [blame]
Reid Spencer5f016e22007-07-11 17:01:13 +00001//===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Reid Spencer5f016e22007-07-11 17:01:13 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This coordinates the per-module state used while generating code.
11//
12//===----------------------------------------------------------------------===//
13
Sanjiv Guptae8b9f5b2008-05-08 08:54:20 +000014#include "CGDebugInfo.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000015#include "CodeGenModule.h"
16#include "CodeGenFunction.h"
Daniel Dunbar0dbe2272008-09-08 21:33:45 +000017#include "CGCall.h"
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +000018#include "CGObjCRuntime.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000019#include "clang/AST/ASTContext.h"
Daniel Dunbarc4a1dea2008-08-11 05:35:13 +000020#include "clang/AST/DeclObjC.h"
Chris Lattner2c8569d2007-12-02 07:19:18 +000021#include "clang/Basic/Diagnostic.h"
Nate Begeman8bd4afe2008-04-19 04:17:09 +000022#include "clang/Basic/SourceManager.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000023#include "clang/Basic/TargetInfo.h"
Nate Begemanec9426c2008-03-09 03:09:36 +000024#include "llvm/CallingConv.h"
Chris Lattnerbef20ac2007-08-31 04:31:45 +000025#include "llvm/Module.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000026#include "llvm/Intrinsics.h"
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000027#include "llvm/Target/TargetData.h"
Chris Lattnera1945fa2008-04-30 16:05:42 +000028#include "llvm/Analysis/Verifier.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000029using namespace clang;
30using namespace CodeGen;
31
32
Chris Lattner45e8cbd2007-11-28 05:34:05 +000033CodeGenModule::CodeGenModule(ASTContext &C, const LangOptions &LO,
Chris Lattnerfb97b032007-12-02 01:40:18 +000034 llvm::Module &M, const llvm::TargetData &TD,
Daniel Dunbarf77ac862008-08-11 21:35:06 +000035 Diagnostic &diags, bool GenerateDebugInfo)
Chris Lattnerfb97b032007-12-02 01:40:18 +000036 : Context(C), Features(LO), TheModule(M), TheTargetData(TD), Diags(diags),
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000037 Types(C, M, TD), Runtime(0), MemCpyFn(0), MemMoveFn(0), MemSetFn(0),
Eli Friedman0c995092008-05-26 12:59:39 +000038 CFConstantStringClassRef(0) {
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000039
40 if (Features.ObjC1) {
Daniel Dunbarf77ac862008-08-11 21:35:06 +000041 if (Features.NeXTRuntime) {
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000042 Runtime = CreateMacObjCRuntime(*this);
43 } else {
44 Runtime = CreateGNUObjCRuntime(*this);
45 }
Daniel Dunbarc17a4d32008-08-11 02:45:11 +000046 }
Sanjiv Guptae8b9f5b2008-05-08 08:54:20 +000047
48 // If debug info generation is enabled, create the CGDebugInfo object.
Ted Kremenek815c78f2008-08-05 18:50:11 +000049 DebugInfo = GenerateDebugInfo ? new CGDebugInfo(this) : 0;
Chris Lattner2b94fe32008-03-01 08:45:05 +000050}
51
52CodeGenModule::~CodeGenModule() {
Ted Kremenek815c78f2008-08-05 18:50:11 +000053 delete Runtime;
54 delete DebugInfo;
55}
56
57void CodeGenModule::Release() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000058 EmitStatics();
Daniel Dunbar219df662008-09-08 23:44:31 +000059 EmitAliases();
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000060 if (Runtime)
61 if (llvm::Function *ObjCInitFunction = Runtime->ModuleInitFunction())
62 AddGlobalCtor(ObjCInitFunction);
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +000063 EmitCtorList(GlobalCtors, "llvm.global_ctors");
64 EmitCtorList(GlobalDtors, "llvm.global_dtors");
Nate Begeman532485c2008-04-18 23:43:57 +000065 EmitAnnotations();
Chris Lattnera1945fa2008-04-30 16:05:42 +000066 // Run the verifier to check that the generated code is consistent.
Daniel Dunbar219df662008-09-08 23:44:31 +000067 if (verifyModule(TheModule, llvm::PrintMessageAction)) {
68 TheModule.dump();
69 assert(0 && "Module failed verification!");
70 }
Chris Lattner2b94fe32008-03-01 08:45:05 +000071}
Reid Spencer5f016e22007-07-11 17:01:13 +000072
Daniel Dunbar488e9932008-08-16 00:56:44 +000073/// ErrorUnsupported - Print out an error that codegen doesn't support the
Chris Lattner2c8569d2007-12-02 07:19:18 +000074/// specified stmt yet.
Daniel Dunbar90df4b62008-09-04 03:43:08 +000075void CodeGenModule::ErrorUnsupported(const Stmt *S, const char *Type,
76 bool OmitOnError) {
77 if (OmitOnError && getDiags().hasErrorOccurred())
78 return;
Daniel Dunbar488e9932008-08-16 00:56:44 +000079 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Error,
Chris Lattner2c8569d2007-12-02 07:19:18 +000080 "cannot codegen this %0 yet");
81 SourceRange Range = S->getSourceRange();
82 std::string Msg = Type;
Ted Kremenek9c728dc2007-12-12 22:39:36 +000083 getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID,
Ted Kremenek7a9d49f2007-12-11 21:27:55 +000084 &Msg, 1, &Range, 1);
Chris Lattner2c8569d2007-12-02 07:19:18 +000085}
Chris Lattner58c3f9e2007-12-02 06:27:33 +000086
Daniel Dunbar488e9932008-08-16 00:56:44 +000087/// ErrorUnsupported - Print out an error that codegen doesn't support the
Chris Lattnerc6fdc342008-01-12 07:05:38 +000088/// specified decl yet.
Daniel Dunbar90df4b62008-09-04 03:43:08 +000089void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type,
90 bool OmitOnError) {
91 if (OmitOnError && getDiags().hasErrorOccurred())
92 return;
Daniel Dunbar488e9932008-08-16 00:56:44 +000093 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Error,
Chris Lattnerc6fdc342008-01-12 07:05:38 +000094 "cannot codegen this %0 yet");
95 std::string Msg = Type;
96 getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID,
97 &Msg, 1);
98}
99
Daniel Dunbar41071de2008-08-15 23:26:23 +0000100/// setGlobalVisibility - Set the visibility for the given LLVM
101/// GlobalValue according to the given clang AST visibility value.
102static void setGlobalVisibility(llvm::GlobalValue *GV,
103 VisibilityAttr::VisibilityTypes Vis) {
Dan Gohman4f8d1232008-05-22 00:50:06 +0000104 switch (Vis) {
105 default: assert(0 && "Unknown visibility!");
106 case VisibilityAttr::DefaultVisibility:
107 GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
108 break;
109 case VisibilityAttr::HiddenVisibility:
110 GV->setVisibility(llvm::GlobalValue::HiddenVisibility);
111 break;
112 case VisibilityAttr::ProtectedVisibility:
113 GV->setVisibility(llvm::GlobalValue::ProtectedVisibility);
114 break;
115 }
116}
117
Chris Lattner6d397602008-03-14 17:18:18 +0000118/// AddGlobalCtor - Add a function to the list that will be called before
119/// main() runs.
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000120void CodeGenModule::AddGlobalCtor(llvm::Function * Ctor, int Priority) {
Chris Lattner6d397602008-03-14 17:18:18 +0000121 // TODO: Type coercion of void()* types.
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000122 GlobalCtors.push_back(std::make_pair(Ctor, Priority));
Chris Lattner6d397602008-03-14 17:18:18 +0000123}
124
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000125/// AddGlobalDtor - Add a function to the list that will be called
126/// when the module is unloaded.
127void CodeGenModule::AddGlobalDtor(llvm::Function * Dtor, int Priority) {
128 // TODO: Type coercion of void()* types.
129 GlobalDtors.push_back(std::make_pair(Dtor, Priority));
130}
131
132void CodeGenModule::EmitCtorList(const CtorList &Fns, const char *GlobalName) {
133 // Ctor function type is void()*.
134 llvm::FunctionType* CtorFTy =
135 llvm::FunctionType::get(llvm::Type::VoidTy,
136 std::vector<const llvm::Type*>(),
137 false);
138 llvm::Type *CtorPFTy = llvm::PointerType::getUnqual(CtorFTy);
139
140 // Get the type of a ctor entry, { i32, void ()* }.
Chris Lattner572cf092008-03-19 05:24:56 +0000141 llvm::StructType* CtorStructTy =
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000142 llvm::StructType::get(llvm::Type::Int32Ty,
143 llvm::PointerType::getUnqual(CtorFTy), NULL);
Chris Lattner6d397602008-03-14 17:18:18 +0000144
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000145 // Construct the constructor and destructor arrays.
146 std::vector<llvm::Constant*> Ctors;
147 for (CtorList::const_iterator I = Fns.begin(), E = Fns.end(); I != E; ++I) {
148 std::vector<llvm::Constant*> S;
149 S.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, I->second, false));
150 S.push_back(llvm::ConstantExpr::getBitCast(I->first, CtorPFTy));
151 Ctors.push_back(llvm::ConstantStruct::get(CtorStructTy, S));
Chris Lattner6d397602008-03-14 17:18:18 +0000152 }
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000153
154 if (!Ctors.empty()) {
155 llvm::ArrayType *AT = llvm::ArrayType::get(CtorStructTy, Ctors.size());
156 new llvm::GlobalVariable(AT, false,
157 llvm::GlobalValue::AppendingLinkage,
158 llvm::ConstantArray::get(AT, Ctors),
159 GlobalName,
160 &TheModule);
161 }
Chris Lattner6d397602008-03-14 17:18:18 +0000162}
163
Nate Begeman532485c2008-04-18 23:43:57 +0000164void CodeGenModule::EmitAnnotations() {
165 if (Annotations.empty())
166 return;
167
168 // Create a new global variable for the ConstantStruct in the Module.
169 llvm::Constant *Array =
170 llvm::ConstantArray::get(llvm::ArrayType::get(Annotations[0]->getType(),
171 Annotations.size()),
172 Annotations);
173 llvm::GlobalValue *gv =
174 new llvm::GlobalVariable(Array->getType(), false,
175 llvm::GlobalValue::AppendingLinkage, Array,
176 "llvm.global.annotations", &TheModule);
177 gv->setSection("llvm.metadata");
178}
179
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000180static void SetGlobalValueAttributes(const Decl *D,
181 bool IsInternal,
182 bool IsInline,
Daniel Dunbar219df662008-09-08 23:44:31 +0000183 llvm::GlobalValue *GV,
184 bool ForDefinition) {
Nuno Lopesd4cbda62008-06-08 15:45:52 +0000185 // TODO: Set up linkage and many other things. Note, this is a simple
186 // approximation of what we really want.
Daniel Dunbar219df662008-09-08 23:44:31 +0000187 if (!ForDefinition) {
188 // Only a few attributes are set on declarations.
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000189 if (D->getAttr<DLLImportAttr>())
190 GV->setLinkage(llvm::Function::DLLImportLinkage);
Daniel Dunbar219df662008-09-08 23:44:31 +0000191 } else {
192 if (IsInternal) {
193 GV->setLinkage(llvm::Function::InternalLinkage);
194 } else {
195 if (D->getAttr<DLLImportAttr>())
196 GV->setLinkage(llvm::Function::DLLImportLinkage);
197 else if (D->getAttr<DLLExportAttr>())
198 GV->setLinkage(llvm::Function::DLLExportLinkage);
199 else if (D->getAttr<WeakAttr>() || IsInline)
200 GV->setLinkage(llvm::Function::WeakLinkage);
201 }
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000202 }
Nuno Lopesd4cbda62008-06-08 15:45:52 +0000203
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000204 if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>())
Daniel Dunbar41071de2008-08-15 23:26:23 +0000205 setGlobalVisibility(GV, attr->getVisibility());
Nuno Lopesd4cbda62008-06-08 15:45:52 +0000206 // FIXME: else handle -fvisibility
Daniel Dunbara735ad82008-08-06 00:03:29 +0000207
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000208 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Daniel Dunbara735ad82008-08-06 00:03:29 +0000209 // Prefaced with special LLVM marker to indicate that the name
210 // should not be munged.
211 GV->setName("\01" + ALA->getLabel());
212 }
Nuno Lopesd4cbda62008-06-08 15:45:52 +0000213}
214
Daniel Dunbar219df662008-09-08 23:44:31 +0000215static void SetFunctionParamAttrs(const CGFunctionInfo &Info, llvm::Function *F) {
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000216 ParamAttrListType ParamAttrList;
217 Info.constructParamAttrList(ParamAttrList);
Eli Friedmanc134fcb2008-06-04 19:41:28 +0000218
Eli Friedmanff4a2d92008-06-01 15:54:49 +0000219 F->setParamAttrs(llvm::PAListPtr::get(ParamAttrList.begin(),
220 ParamAttrList.size()));
221
222 // Set the appropriate calling convention for the Function.
Daniel Dunbar0dbe2272008-09-08 21:33:45 +0000223 if (Info.getDecl()->getAttr<FastCallAttr>())
Eli Friedmanff4a2d92008-06-01 15:54:49 +0000224 F->setCallingConv(llvm::CallingConv::Fast);
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000225}
226
227/// SetFunctionAttributesForDefinition - Set function attributes
228/// specific to a function definition.
Daniel Dunbar219df662008-09-08 23:44:31 +0000229void CodeGenModule::SetFunctionAttributesForDefinition(const Decl *D,
230 llvm::Function *F) {
231 if (isa<ObjCMethodDecl>(D)) {
232 SetGlobalValueAttributes(D, true, false, F, true);
233 } else {
234 const FunctionDecl *FD = cast<FunctionDecl>(D);
235 SetGlobalValueAttributes(FD, FD->getStorageClass() == FunctionDecl::Static,
236 FD->isInline(), F, true);
237 }
238
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000239 if (!Features.Exceptions)
240 F->addParamAttr(0, llvm::ParamAttr::NoUnwind);
241}
242
243void CodeGenModule::SetMethodAttributes(const ObjCMethodDecl *MD,
244 llvm::Function *F) {
Daniel Dunbar219df662008-09-08 23:44:31 +0000245 SetFunctionParamAttrs(CGFunctionInfo(MD, Context), F);
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000246
Daniel Dunbar219df662008-09-08 23:44:31 +0000247 SetFunctionAttributesForDefinition(MD, F);
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000248}
249
250void CodeGenModule::SetFunctionAttributes(const FunctionDecl *FD,
251 llvm::Function *F) {
Daniel Dunbar219df662008-09-08 23:44:31 +0000252 SetFunctionParamAttrs(CGFunctionInfo(FD), F);
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000253
Daniel Dunbar219df662008-09-08 23:44:31 +0000254 SetGlobalValueAttributes(FD, FD->getStorageClass() == FunctionDecl::Static,
255 FD->isInline(), F, false);
256}
257
258void CodeGenModule::EmitAliases() {
259 for (unsigned i = 0, e = Aliases.size(); i != e; ++i) {
260 const FunctionDecl *D = Aliases[i];
261 const AliasAttr *AA = D->getAttr<AliasAttr>();
262
263 // This is something of a hack, if the FunctionDecl got overridden
264 // then its attributes will be moved to the new declaration. In
265 // this case the current decl has no alias attribute, but we will
266 // eventually see it.
267 if (!AA)
268 continue;
269
270 const std::string& aliaseeName = AA->getAliasee();
271 llvm::Function *aliasee = getModule().getFunction(aliaseeName);
272 if (!aliasee) {
273 // FIXME: This isn't unsupported, this is just an error, which
274 // sema should catch, but...
275 ErrorUnsupported(D, "alias referencing a missing function");
276 continue;
277 }
278
279 llvm::GlobalValue *GA =
280 new llvm::GlobalAlias(aliasee->getType(),
281 llvm::Function::ExternalLinkage,
282 D->getName(),
283 aliasee,
284 &getModule());
285
286 llvm::GlobalValue *&Entry = GlobalDeclMap[D->getIdentifier()];
287 if (Entry) {
288 // If we created a dummy function for this then replace it.
289 GA->takeName(Entry);
290
291 llvm::Value *Casted =
292 llvm::ConstantExpr::getBitCast(GA, Entry->getType());
293 Entry->replaceAllUsesWith(Casted);
294 Entry->eraseFromParent();
295
296 Entry = GA;
297 }
298
299 // Alias should never be internal or inline.
300 SetGlobalValueAttributes(D, false, false, GA, true);
301 }
Eli Friedmanff4a2d92008-06-01 15:54:49 +0000302}
303
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000304void CodeGenModule::EmitStatics() {
305 // Emit code for each used static decl encountered. Since a previously unused
306 // static decl may become used during the generation of code for a static
307 // function, iterate until no changes are made.
308 bool Changed;
309 do {
310 Changed = false;
311 for (unsigned i = 0, e = StaticDecls.size(); i != e; ++i) {
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000312 const ValueDecl *D = StaticDecls[i];
Eli Friedman6f7e2ee2008-05-27 04:58:01 +0000313
314 // Check if we have used a decl with the same name
315 // FIXME: The AST should have some sort of aggregate decls or
316 // global symbol map.
Daniel Dunbar219df662008-09-08 23:44:31 +0000317 // FIXME: This is missing some important cases. For example, we
318 // need to check for uses in an alias and in a constructor.
Daniel Dunbar90db8822008-08-25 06:18:57 +0000319 if (!GlobalDeclMap.count(D->getIdentifier()))
Daniel Dunbara735ad82008-08-06 00:03:29 +0000320 continue;
Eli Friedman6f7e2ee2008-05-27 04:58:01 +0000321
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000322 // Emit the definition.
323 EmitGlobalDefinition(D);
324
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000325 // Erase the used decl from the list.
326 StaticDecls[i] = StaticDecls.back();
327 StaticDecls.pop_back();
328 --i;
329 --e;
330
331 // Remember that we made a change.
332 Changed = true;
333 }
334 } while (Changed);
Reid Spencer5f016e22007-07-11 17:01:13 +0000335}
336
Nate Begeman8bd4afe2008-04-19 04:17:09 +0000337/// EmitAnnotateAttr - Generate the llvm::ConstantStruct which contains the
338/// annotation information for a given GlobalValue. The annotation struct is
339/// {i8 *, i8 *, i8 *, i32}. The first field is a constant expression, the
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000340/// GlobalValue being annotated. The second field is the constant string
Nate Begeman8bd4afe2008-04-19 04:17:09 +0000341/// created from the AnnotateAttr's annotation. The third field is a constant
342/// string containing the name of the translation unit. The fourth field is
343/// the line number in the file of the annotated value declaration.
344///
345/// FIXME: this does not unique the annotation string constants, as llvm-gcc
346/// appears to.
347///
348llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
349 const AnnotateAttr *AA,
350 unsigned LineNo) {
351 llvm::Module *M = &getModule();
352
353 // get [N x i8] constants for the annotation string, and the filename string
354 // which are the 2nd and 3rd elements of the global annotation structure.
355 const llvm::Type *SBP = llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
356 llvm::Constant *anno = llvm::ConstantArray::get(AA->getAnnotation(), true);
357 llvm::Constant *unit = llvm::ConstantArray::get(M->getModuleIdentifier(),
358 true);
359
360 // Get the two global values corresponding to the ConstantArrays we just
361 // created to hold the bytes of the strings.
362 llvm::GlobalValue *annoGV =
363 new llvm::GlobalVariable(anno->getType(), false,
364 llvm::GlobalValue::InternalLinkage, anno,
365 GV->getName() + ".str", M);
366 // translation unit name string, emitted into the llvm.metadata section.
367 llvm::GlobalValue *unitGV =
368 new llvm::GlobalVariable(unit->getType(), false,
369 llvm::GlobalValue::InternalLinkage, unit, ".str", M);
370
371 // Create the ConstantStruct that is the global annotion.
372 llvm::Constant *Fields[4] = {
373 llvm::ConstantExpr::getBitCast(GV, SBP),
374 llvm::ConstantExpr::getBitCast(annoGV, SBP),
375 llvm::ConstantExpr::getBitCast(unitGV, SBP),
376 llvm::ConstantInt::get(llvm::Type::Int32Ty, LineNo)
377 };
378 return llvm::ConstantStruct::get(Fields, 4, false);
379}
380
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000381void CodeGenModule::EmitGlobal(const ValueDecl *Global) {
382 bool isDef, isStatic;
383
384 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Global)) {
Daniel Dunbar219df662008-09-08 23:44:31 +0000385 // Aliases are deferred until code for everything else has been
386 // emitted.
387 if (FD->getAttr<AliasAttr>()) {
388 assert(!FD->isThisDeclarationADefinition() &&
389 "Function alias cannot have a definition!");
390 Aliases.push_back(FD);
391 return;
392 }
393
394 isDef = FD->isThisDeclarationADefinition();
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000395 isStatic = FD->getStorageClass() == FunctionDecl::Static;
396 } else if (const VarDecl *VD = cast<VarDecl>(Global)) {
397 assert(VD->isFileVarDecl() && "Cannot emit local var decl as global.");
398
399 isDef = !(VD->getStorageClass() == VarDecl::Extern && VD->getInit() == 0);
400 isStatic = VD->getStorageClass() == VarDecl::Static;
401 } else {
402 assert(0 && "Invalid argument to EmitGlobal");
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000403 return;
404 }
405
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000406 // Forward declarations are emitted lazily on first use.
407 if (!isDef)
Chris Lattner88a69ad2007-07-13 05:13:43 +0000408 return;
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000409
410 // If the global is a static, defer code generation until later so
411 // we can easily omit unused statics.
412 if (isStatic) {
413 StaticDecls.push_back(Global);
414 return;
415 }
416
417 // Otherwise emit the definition.
418 EmitGlobalDefinition(Global);
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000419}
420
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000421void CodeGenModule::EmitGlobalDefinition(const ValueDecl *D) {
422 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
423 EmitGlobalFunctionDefinition(FD);
424 } else if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
425 EmitGlobalVarDefinition(VD);
426 } else {
427 assert(0 && "Invalid argument to EmitGlobalDefinition()");
428 }
429}
430
Daniel Dunbar9986eab2008-07-30 16:32:24 +0000431 llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D) {
Eli Friedman77ba7082008-05-30 19:50:47 +0000432 assert(D->hasGlobalStorage() && "Not a global variable");
433
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000434 QualType ASTTy = D->getType();
435 const llvm::Type *Ty = getTypes().ConvertTypeForMem(ASTTy);
Daniel Dunbar9986eab2008-07-30 16:32:24 +0000436 const llvm::Type *PTy = llvm::PointerType::get(Ty, ASTTy.getAddressSpace());
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000437
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000438 // Lookup the entry, lazily creating it if necessary.
Daniel Dunbar90db8822008-08-25 06:18:57 +0000439 llvm::GlobalValue *&Entry = GlobalDeclMap[D->getIdentifier()];
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000440 if (!Entry)
441 Entry = new llvm::GlobalVariable(Ty, false,
442 llvm::GlobalValue::ExternalLinkage,
443 0, D->getName(), &getModule(), 0,
444 ASTTy.getAddressSpace());
445
Daniel Dunbar9986eab2008-07-30 16:32:24 +0000446 // Make sure the result is of the correct type.
447 return llvm::ConstantExpr::getBitCast(Entry, PTy);
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000448}
449
450void CodeGenModule::EmitGlobalVarDefinition(const VarDecl *D) {
Chris Lattner8f32f712007-07-14 00:23:28 +0000451 llvm::Constant *Init = 0;
Eli Friedman77ba7082008-05-30 19:50:47 +0000452 QualType ASTTy = D->getType();
453 const llvm::Type *VarTy = getTypes().ConvertTypeForMem(ASTTy);
Eli Friedman77ba7082008-05-30 19:50:47 +0000454
Chris Lattner8f32f712007-07-14 00:23:28 +0000455 if (D->getInit() == 0) {
Eli Friedmancd5f4aa2008-05-30 20:39:54 +0000456 // This is a tentative definition; tentative definitions are
457 // implicitly initialized with { 0 }
458 const llvm::Type* InitTy;
459 if (ASTTy->isIncompleteArrayType()) {
460 // An incomplete array is normally [ TYPE x 0 ], but we need
461 // to fix it to [ TYPE x 1 ].
462 const llvm::ArrayType* ATy = cast<llvm::ArrayType>(VarTy);
463 InitTy = llvm::ArrayType::get(ATy->getElementType(), 1);
464 } else {
465 InitTy = VarTy;
466 }
467 Init = llvm::Constant::getNullValue(InitTy);
Eli Friedman77ba7082008-05-30 19:50:47 +0000468 } else {
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000469 Init = EmitConstantExpr(D->getInit());
Eli Friedman77ba7082008-05-30 19:50:47 +0000470 }
471 const llvm::Type* InitType = Init->getType();
472
Daniel Dunbar90db8822008-08-25 06:18:57 +0000473 llvm::GlobalValue *&Entry = GlobalDeclMap[D->getIdentifier()];
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000474 llvm::GlobalVariable *GV = cast_or_null<llvm::GlobalVariable>(Entry);
475
Eli Friedman77ba7082008-05-30 19:50:47 +0000476 if (!GV) {
477 GV = new llvm::GlobalVariable(InitType, false,
478 llvm::GlobalValue::ExternalLinkage,
479 0, D->getName(), &getModule(), 0,
480 ASTTy.getAddressSpace());
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000481 } else if (GV->getType() !=
482 llvm::PointerType::get(InitType, ASTTy.getAddressSpace())) {
Eli Friedman77ba7082008-05-30 19:50:47 +0000483 // We have a definition after a prototype with the wrong type.
484 // We must make a new GlobalVariable* and update everything that used OldGV
485 // (a declaration or tentative definition) with the new GlobalVariable*
486 // (which will be a definition).
487 //
488 // This happens if there is a prototype for a global (e.g. "extern int x[];")
489 // and then a definition of a different type (e.g. "int x[10];"). This also
490 // happens when an initializer has a different type from the type of the
491 // global (this happens with unions).
Eli Friedmancd5f4aa2008-05-30 20:39:54 +0000492 //
493 // FIXME: This also ends up happening if there's a definition followed by
494 // a tentative definition! (Although Sema rejects that construct
495 // at the moment.)
Eli Friedman77ba7082008-05-30 19:50:47 +0000496
497 // Save the old global
498 llvm::GlobalVariable *OldGV = GV;
499
500 // Make a new global with the correct type
501 GV = new llvm::GlobalVariable(InitType, false,
502 llvm::GlobalValue::ExternalLinkage,
503 0, D->getName(), &getModule(), 0,
504 ASTTy.getAddressSpace());
505 // Steal the name of the old global
506 GV->takeName(OldGV);
507
508 // Replace all uses of the old global with the new global
509 llvm::Constant *NewPtrForOldDecl =
510 llvm::ConstantExpr::getBitCast(GV, OldGV->getType());
511 OldGV->replaceAllUsesWith(NewPtrForOldDecl);
Eli Friedman77ba7082008-05-30 19:50:47 +0000512
513 // Erase the old global, since it is no longer used.
514 OldGV->eraseFromParent();
Chris Lattner8f32f712007-07-14 00:23:28 +0000515 }
Devang Patel8e53e722007-10-26 16:31:40 +0000516
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000517 Entry = GV;
Devang Patel8e53e722007-10-26 16:31:40 +0000518
Nate Begeman8bd4afe2008-04-19 04:17:09 +0000519 if (const AnnotateAttr *AA = D->getAttr<AnnotateAttr>()) {
520 SourceManager &SM = Context.getSourceManager();
521 AddAnnotation(EmitAnnotateAttr(GV, AA,
522 SM.getLogicalLineNumber(D->getLocation())));
523 }
524
Chris Lattner88a69ad2007-07-13 05:13:43 +0000525 GV->setInitializer(Init);
Nuno Lopesb381aac2008-09-01 11:33:04 +0000526 GV->setConstant(D->getType().isConstant(Context));
Chris Lattnerddee4232008-03-03 03:28:21 +0000527
Eli Friedmancd5f4aa2008-05-30 20:39:54 +0000528 // FIXME: This is silly; getTypeAlign should just work for incomplete arrays
529 unsigned Align;
Chris Lattnerc63a1f22008-08-04 07:31:14 +0000530 if (const IncompleteArrayType* IAT =
531 Context.getAsIncompleteArrayType(D->getType()))
Eli Friedmancd5f4aa2008-05-30 20:39:54 +0000532 Align = Context.getTypeAlign(IAT->getElementType());
533 else
534 Align = Context.getTypeAlign(D->getType());
Eli Friedman08d78022008-05-29 11:10:27 +0000535 if (const AlignedAttr* AA = D->getAttr<AlignedAttr>()) {
536 Align = std::max(Align, AA->getAlignment());
537 }
538 GV->setAlignment(Align / 8);
539
Chris Lattnerddee4232008-03-03 03:28:21 +0000540 if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>())
Daniel Dunbar41071de2008-08-15 23:26:23 +0000541 setGlobalVisibility(GV, attr->getVisibility());
Chris Lattnerddee4232008-03-03 03:28:21 +0000542 // FIXME: else handle -fvisibility
Daniel Dunbara735ad82008-08-06 00:03:29 +0000543
544 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
545 // Prefaced with special LLVM marker to indicate that the name
546 // should not be munged.
547 GV->setName("\01" + ALA->getLabel());
548 }
Chris Lattner88a69ad2007-07-13 05:13:43 +0000549
550 // Set the llvm linkage type as appropriate.
Chris Lattner8fabd782008-05-04 01:44:26 +0000551 if (D->getStorageClass() == VarDecl::Static)
552 GV->setLinkage(llvm::Function::InternalLinkage);
553 else if (D->getAttr<DLLImportAttr>())
Chris Lattnerddee4232008-03-03 03:28:21 +0000554 GV->setLinkage(llvm::Function::DLLImportLinkage);
555 else if (D->getAttr<DLLExportAttr>())
556 GV->setLinkage(llvm::Function::DLLExportLinkage);
Chris Lattner8fabd782008-05-04 01:44:26 +0000557 else if (D->getAttr<WeakAttr>())
Chris Lattnerddee4232008-03-03 03:28:21 +0000558 GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
Chris Lattner8fabd782008-05-04 01:44:26 +0000559 else {
Chris Lattnerddee4232008-03-03 03:28:21 +0000560 // FIXME: This isn't right. This should handle common linkage and other
561 // stuff.
562 switch (D->getStorageClass()) {
Chris Lattner8fabd782008-05-04 01:44:26 +0000563 case VarDecl::Static: assert(0 && "This case handled above");
Chris Lattnerddee4232008-03-03 03:28:21 +0000564 case VarDecl::Auto:
565 case VarDecl::Register:
566 assert(0 && "Can't have auto or register globals");
567 case VarDecl::None:
568 if (!D->getInit())
Eli Friedmana07b7642008-05-29 11:03:17 +0000569 GV->setLinkage(llvm::GlobalVariable::CommonLinkage);
Chris Lattnerddee4232008-03-03 03:28:21 +0000570 break;
571 case VarDecl::Extern:
572 case VarDecl::PrivateExtern:
573 // todo: common
574 break;
Chris Lattnerddee4232008-03-03 03:28:21 +0000575 }
Chris Lattner88a69ad2007-07-13 05:13:43 +0000576 }
Sanjiv Gupta686226b2008-06-05 08:59:10 +0000577
578 // Emit global variable debug information.
579 CGDebugInfo *DI = getDebugInfo();
580 if(DI) {
581 if(D->getLocation().isValid())
582 DI->setLocation(D->getLocation());
583 DI->EmitGlobalVariable(GV, D);
584 }
Chris Lattner88a69ad2007-07-13 05:13:43 +0000585}
Reid Spencer5f016e22007-07-11 17:01:13 +0000586
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000587llvm::GlobalValue *
588CodeGenModule::EmitForwardFunctionDefinition(const FunctionDecl *D) {
Daniel Dunbar219df662008-09-08 23:44:31 +0000589 const llvm::Type *Ty = getTypes().ConvertType(D->getType());
590 llvm::Function *F = llvm::Function::Create(cast<llvm::FunctionType>(Ty),
591 llvm::Function::ExternalLinkage,
592 D->getName(), &getModule());
593 SetFunctionAttributes(D, F);
594 return F;
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000595}
596
597llvm::Constant *CodeGenModule::GetAddrOfFunction(const FunctionDecl *D) {
Daniel Dunbar9986eab2008-07-30 16:32:24 +0000598 QualType ASTTy = D->getType();
599 const llvm::Type *Ty = getTypes().ConvertTypeForMem(ASTTy);
600 const llvm::Type *PTy = llvm::PointerType::get(Ty, ASTTy.getAddressSpace());
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000601
602 // Lookup the entry, lazily creating it if necessary.
Daniel Dunbar90db8822008-08-25 06:18:57 +0000603 llvm::GlobalValue *&Entry = GlobalDeclMap[D->getIdentifier()];
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000604 if (!Entry)
605 Entry = EmitForwardFunctionDefinition(D);
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000606
Daniel Dunbar9986eab2008-07-30 16:32:24 +0000607 return llvm::ConstantExpr::getBitCast(Entry, PTy);
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000608}
609
610void CodeGenModule::EmitGlobalFunctionDefinition(const FunctionDecl *D) {
Daniel Dunbar90db8822008-08-25 06:18:57 +0000611 llvm::GlobalValue *&Entry = GlobalDeclMap[D->getIdentifier()];
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000612 if (!Entry) {
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000613 Entry = EmitForwardFunctionDefinition(D);
614 } else {
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000615 // If the types mismatch then we have to rewrite the definition.
616 const llvm::Type *Ty = getTypes().ConvertType(D->getType());
617 if (Entry->getType() != llvm::PointerType::getUnqual(Ty)) {
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000618 // Otherwise, we have a definition after a prototype with the wrong type.
619 // F is the Function* for the one with the wrong type, we must make a new
620 // Function* and update everything that used F (a declaration) with the new
621 // Function* (which will be a definition).
622 //
623 // This happens if there is a prototype for a function (e.g. "int f()") and
624 // then a definition of a different type (e.g. "int f(int x)"). Start by
625 // making a new function of the correct type, RAUW, then steal the name.
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000626 llvm::GlobalValue *NewFn = EmitForwardFunctionDefinition(D);
627 NewFn->takeName(Entry);
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000628
629 // Replace uses of F with the Function we will endow with a body.
630 llvm::Constant *NewPtrForOldDecl =
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000631 llvm::ConstantExpr::getBitCast(NewFn, Entry->getType());
632 Entry->replaceAllUsesWith(NewPtrForOldDecl);
633
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000634 // Ok, delete the old function now, which is dead.
Daniel Dunbar3c827a72008-08-05 23:31:02 +0000635 assert(Entry->isDeclaration() && "Shouldn't replace non-declaration");
Daniel Dunbar219df662008-09-08 23:44:31 +0000636 Entry->eraseFromParent();
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000637
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000638 Entry = NewFn;
639 }
640 }
641
Daniel Dunbar219df662008-09-08 23:44:31 +0000642 llvm::Function *Fn = cast<llvm::Function>(Entry);
643 CodeGenFunction(*this).GenerateCode(D, Fn);
Daniel Dunbar6bfed7e2008-08-01 00:01:51 +0000644
Daniel Dunbar219df662008-09-08 23:44:31 +0000645 SetFunctionAttributesForDefinition(D, Fn);
646
647 if (const ConstructorAttr *CA = D->getAttr<ConstructorAttr>()) {
648 AddGlobalCtor(Fn, CA->getPriority());
649 } else if (const DestructorAttr *DA = D->getAttr<DestructorAttr>()) {
650 AddGlobalDtor(Fn, DA->getPriority());
Daniel Dunbarbd012ff2008-07-29 23:18:29 +0000651 }
652}
653
Chris Lattnerc5b88062008-02-06 05:08:19 +0000654void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
655 // Make sure that this type is translated.
656 Types.UpdateCompletedType(TD);
Chris Lattnerd86e6bc2008-02-05 08:06:13 +0000657}
658
659
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000660/// getBuiltinLibFunction
661llvm::Function *CodeGenModule::getBuiltinLibFunction(unsigned BuiltinID) {
Chris Lattner1426fec2007-12-13 00:38:03 +0000662 if (BuiltinID > BuiltinFunctions.size())
663 BuiltinFunctions.resize(BuiltinID);
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000664
Chris Lattner1426fec2007-12-13 00:38:03 +0000665 // Cache looked up functions. Since builtin id #0 is invalid we don't reserve
666 // a slot for it.
667 assert(BuiltinID && "Invalid Builtin ID");
668 llvm::Function *&FunctionSlot = BuiltinFunctions[BuiltinID-1];
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000669 if (FunctionSlot)
670 return FunctionSlot;
671
672 assert(Context.BuiltinInfo.isLibFunction(BuiltinID) && "isn't a lib fn");
673
674 // Get the name, skip over the __builtin_ prefix.
675 const char *Name = Context.BuiltinInfo.GetName(BuiltinID)+10;
676
677 // Get the type for the builtin.
678 QualType Type = Context.BuiltinInfo.GetBuiltinType(BuiltinID, Context);
679 const llvm::FunctionType *Ty =
680 cast<llvm::FunctionType>(getTypes().ConvertType(Type));
681
682 // FIXME: This has a serious problem with code like this:
683 // void abs() {}
684 // ... __builtin_abs(x);
685 // The two versions of abs will collide. The fix is for the builtin to win,
686 // and for the existing one to be turned into a constantexpr cast of the
687 // builtin. In the case where the existing one is a static function, it
688 // should just be renamed.
Chris Lattnerc5e940f2007-08-31 04:44:06 +0000689 if (llvm::Function *Existing = getModule().getFunction(Name)) {
690 if (Existing->getFunctionType() == Ty && Existing->hasExternalLinkage())
691 return FunctionSlot = Existing;
692 assert(Existing == 0 && "FIXME: Name collision");
693 }
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000694
695 // FIXME: param attributes for sext/zext etc.
Nate Begeman4c13b7a2008-04-20 06:29:50 +0000696 return FunctionSlot =
697 llvm::Function::Create(Ty, llvm::Function::ExternalLinkage, Name,
698 &getModule());
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000699}
700
Chris Lattner7acda7c2007-12-18 00:25:38 +0000701llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,const llvm::Type **Tys,
702 unsigned NumTys) {
703 return llvm::Intrinsic::getDeclaration(&getModule(),
704 (llvm::Intrinsic::ID)IID, Tys, NumTys);
705}
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000706
Reid Spencer5f016e22007-07-11 17:01:13 +0000707llvm::Function *CodeGenModule::getMemCpyFn() {
708 if (MemCpyFn) return MemCpyFn;
709 llvm::Intrinsic::ID IID;
Chris Lattnerf72a4432008-03-08 08:34:58 +0000710 switch (Context.Target.getPointerWidth(0)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000711 default: assert(0 && "Unknown ptr width");
712 case 32: IID = llvm::Intrinsic::memcpy_i32; break;
713 case 64: IID = llvm::Intrinsic::memcpy_i64; break;
714 }
Chris Lattner7acda7c2007-12-18 00:25:38 +0000715 return MemCpyFn = getIntrinsic(IID);
Reid Spencer5f016e22007-07-11 17:01:13 +0000716}
Anders Carlssonc9e20912007-08-21 00:21:21 +0000717
Eli Friedman0c995092008-05-26 12:59:39 +0000718llvm::Function *CodeGenModule::getMemMoveFn() {
719 if (MemMoveFn) return MemMoveFn;
720 llvm::Intrinsic::ID IID;
721 switch (Context.Target.getPointerWidth(0)) {
722 default: assert(0 && "Unknown ptr width");
723 case 32: IID = llvm::Intrinsic::memmove_i32; break;
724 case 64: IID = llvm::Intrinsic::memmove_i64; break;
725 }
726 return MemMoveFn = getIntrinsic(IID);
727}
728
Lauro Ramos Venancio41ef30e2008-02-19 22:01:01 +0000729llvm::Function *CodeGenModule::getMemSetFn() {
730 if (MemSetFn) return MemSetFn;
731 llvm::Intrinsic::ID IID;
Chris Lattnerf72a4432008-03-08 08:34:58 +0000732 switch (Context.Target.getPointerWidth(0)) {
Lauro Ramos Venancio41ef30e2008-02-19 22:01:01 +0000733 default: assert(0 && "Unknown ptr width");
734 case 32: IID = llvm::Intrinsic::memset_i32; break;
735 case 64: IID = llvm::Intrinsic::memset_i64; break;
736 }
737 return MemSetFn = getIntrinsic(IID);
738}
Chris Lattner7acda7c2007-12-18 00:25:38 +0000739
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000740// We still need to work out the details of handling UTF-16.
741// See: <rdr://2996215>
Chris Lattnerbef20ac2007-08-31 04:31:45 +0000742llvm::Constant *CodeGenModule::
743GetAddrOfConstantCFString(const std::string &str) {
Anders Carlssonc9e20912007-08-21 00:21:21 +0000744 llvm::StringMapEntry<llvm::Constant *> &Entry =
745 CFConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
746
747 if (Entry.getValue())
748 return Entry.getValue();
749
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000750 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty);
751 llvm::Constant *Zeros[] = { Zero, Zero };
Anders Carlssonc9e20912007-08-21 00:21:21 +0000752
753 if (!CFConstantStringClassRef) {
754 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
755 Ty = llvm::ArrayType::get(Ty, 0);
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000756
757 // FIXME: This is fairly broken if
758 // __CFConstantStringClassReference is already defined, in that it
759 // will get renamed and the user will most likely see an opaque
760 // error message. This is a general issue with relying on
761 // particular names.
762 llvm::GlobalVariable *GV =
Anders Carlssonc9e20912007-08-21 00:21:21 +0000763 new llvm::GlobalVariable(Ty, false,
764 llvm::GlobalVariable::ExternalLinkage, 0,
765 "__CFConstantStringClassReference",
766 &getModule());
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000767
768 // Decay array -> ptr
769 CFConstantStringClassRef =
770 llvm::ConstantExpr::getGetElementPtr(GV, Zeros, 2);
Anders Carlssonc9e20912007-08-21 00:21:21 +0000771 }
772
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000773 std::vector<llvm::Constant*> Fields(4);
774
Anders Carlssonc9e20912007-08-21 00:21:21 +0000775 // Class pointer.
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000776 Fields[0] = CFConstantStringClassRef;
Anders Carlssonc9e20912007-08-21 00:21:21 +0000777
778 // Flags.
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000779 const llvm::Type *Ty = getTypes().ConvertType(getContext().UnsignedIntTy);
780 Fields[1] = llvm::ConstantInt::get(Ty, 0x07C8);
Anders Carlssonc9e20912007-08-21 00:21:21 +0000781
782 // String pointer.
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000783 llvm::Constant *C = llvm::ConstantArray::get(str);
Anders Carlssonc9e20912007-08-21 00:21:21 +0000784 C = new llvm::GlobalVariable(C->getType(), true,
785 llvm::GlobalValue::InternalLinkage,
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000786 C, ".str", &getModule());
787 Fields[2] = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2);
Anders Carlssonc9e20912007-08-21 00:21:21 +0000788
789 // String length.
790 Ty = getTypes().ConvertType(getContext().LongTy);
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000791 Fields[3] = llvm::ConstantInt::get(Ty, str.length());
Anders Carlssonc9e20912007-08-21 00:21:21 +0000792
793 // The struct.
794 Ty = getTypes().ConvertType(getContext().getCFConstantStringType());
795 C = llvm::ConstantStruct::get(cast<llvm::StructType>(Ty), Fields);
Anders Carlsson0c678292007-11-01 00:41:52 +0000796 llvm::GlobalVariable *GV =
797 new llvm::GlobalVariable(C->getType(), true,
798 llvm::GlobalVariable::InternalLinkage,
799 C, "", &getModule());
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000800
Anders Carlsson0c678292007-11-01 00:41:52 +0000801 GV->setSection("__DATA,__cfstring");
802 Entry.setValue(GV);
Daniel Dunbar3e9df992008-08-23 18:37:06 +0000803
Anders Carlsson0c678292007-11-01 00:41:52 +0000804 return GV;
Anders Carlssonc9e20912007-08-21 00:21:21 +0000805}
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000806
Daniel Dunbar61432932008-08-13 23:20:05 +0000807/// GetStringForStringLiteral - Return the appropriate bytes for a
Daniel Dunbar1e049762008-08-10 20:25:57 +0000808/// string literal, properly padded to match the literal type.
Daniel Dunbar61432932008-08-13 23:20:05 +0000809std::string CodeGenModule::GetStringForStringLiteral(const StringLiteral *E) {
Daniel Dunbar662174c82008-08-29 17:28:43 +0000810 if (E->isWide()) {
811 ErrorUnsupported(E, "wide string");
812 return "FIXME";
813 }
814
Daniel Dunbar1e049762008-08-10 20:25:57 +0000815 const char *StrData = E->getStrData();
816 unsigned Len = E->getByteLength();
817
818 const ConstantArrayType *CAT =
819 getContext().getAsConstantArrayType(E->getType());
820 assert(CAT && "String isn't pointer or array!");
821
822 // Resize the string to the right size
823 // FIXME: What about wchar_t strings?
824 std::string Str(StrData, StrData+Len);
825 uint64_t RealLen = CAT->getSize().getZExtValue();
826 Str.resize(RealLen, '\0');
827
828 return Str;
829}
830
Daniel Dunbar61432932008-08-13 23:20:05 +0000831/// GetAddrOfConstantStringFromLiteral - Return a pointer to a
832/// constant array for the given string literal.
833llvm::Constant *
834CodeGenModule::GetAddrOfConstantStringFromLiteral(const StringLiteral *S) {
835 // FIXME: This can be more efficient.
836 return GetAddrOfConstantString(GetStringForStringLiteral(S));
837}
838
Chris Lattnera7ad98f2008-02-11 00:02:17 +0000839/// GenerateWritableString -- Creates storage for a string literal.
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000840static llvm::Constant *GenerateStringLiteral(const std::string &str,
841 bool constant,
Chris Lattner2c8569d2007-12-02 07:19:18 +0000842 CodeGenModule &CGM) {
Daniel Dunbar61432932008-08-13 23:20:05 +0000843 // Create Constant for this string literal. Don't add a '\0'.
844 llvm::Constant *C = llvm::ConstantArray::get(str, false);
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000845
846 // Create a global variable for this string
847 C = new llvm::GlobalVariable(C->getType(), constant,
848 llvm::GlobalValue::InternalLinkage,
Chris Lattner2c8569d2007-12-02 07:19:18 +0000849 C, ".str", &CGM.getModule());
Daniel Dunbar61432932008-08-13 23:20:05 +0000850
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000851 return C;
852}
853
Daniel Dunbar61432932008-08-13 23:20:05 +0000854/// GetAddrOfConstantString - Returns a pointer to a character array
855/// containing the literal. This contents are exactly that of the
856/// given string, i.e. it will not be null terminated automatically;
857/// see GetAddrOfConstantCString. Note that whether the result is
858/// actually a pointer to an LLVM constant depends on
859/// Feature.WriteableStrings.
860///
861/// The result has pointer to array type.
Chris Lattner45e8cbd2007-11-28 05:34:05 +0000862llvm::Constant *CodeGenModule::GetAddrOfConstantString(const std::string &str) {
863 // Don't share any string literals if writable-strings is turned on.
864 if (Features.WritableStrings)
865 return GenerateStringLiteral(str, false, *this);
866
867 llvm::StringMapEntry<llvm::Constant *> &Entry =
868 ConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]);
869
870 if (Entry.getValue())
871 return Entry.getValue();
872
873 // Create a global variable for this.
874 llvm::Constant *C = GenerateStringLiteral(str, true, *this);
875 Entry.setValue(C);
876 return C;
877}
Daniel Dunbar61432932008-08-13 23:20:05 +0000878
879/// GetAddrOfConstantCString - Returns a pointer to a character
880/// array containing the literal and a terminating '\-'
881/// character. The result has pointer to array type.
882llvm::Constant *CodeGenModule::GetAddrOfConstantCString(const std::string &str) {
Daniel Dunbar9d9b09c2008-08-15 18:29:12 +0000883 return GetAddrOfConstantString(str + "\0");
Daniel Dunbar61432932008-08-13 23:20:05 +0000884}
Daniel Dunbar41071de2008-08-15 23:26:23 +0000885
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000886/// EmitObjCPropertyImplementations - Emit information for synthesized
887/// properties for an implementation.
888void CodeGenModule::EmitObjCPropertyImplementations(const
889 ObjCImplementationDecl *D) {
890 for (ObjCImplementationDecl::propimpl_iterator i = D->propimpl_begin(),
891 e = D->propimpl_end(); i != e; ++i) {
892 ObjCPropertyImplDecl *PID = *i;
893
894 // Dynamic is just for type-checking.
895 if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize) {
896 ObjCPropertyDecl *PD = PID->getPropertyDecl();
897
898 // Determine which methods need to be implemented, some may have
899 // been overridden. Note that ::isSynthesized is not the method
900 // we want, that just indicates if the decl came from a
901 // property. What we want to know is if the method is defined in
902 // this implementation.
903 if (!D->getInstanceMethod(PD->getGetterName()))
904 CodeGenFunction(*this).GenerateObjCGetter(PID);
905 if (!PD->isReadOnly() &&
906 !D->getInstanceMethod(PD->getSetterName()))
907 CodeGenFunction(*this).GenerateObjCSetter(PID);
908 }
909 }
910}
911
Daniel Dunbar41071de2008-08-15 23:26:23 +0000912/// EmitTopLevelDecl - Emit code for a single top level declaration.
913void CodeGenModule::EmitTopLevelDecl(Decl *D) {
914 // If an error has occurred, stop code generation, but continue
915 // parsing and semantic analysis (to ensure all warnings and errors
916 // are emitted).
917 if (Diags.hasErrorOccurred())
918 return;
919
920 switch (D->getKind()) {
921 case Decl::Function:
922 case Decl::Var:
923 EmitGlobal(cast<ValueDecl>(D));
924 break;
925
926 case Decl::Namespace:
Daniel Dunbar662174c82008-08-29 17:28:43 +0000927 ErrorUnsupported(D, "namespace");
Daniel Dunbar41071de2008-08-15 23:26:23 +0000928 break;
929
930 // Objective-C Decls
931
932 // Forward declarations, no (immediate) code generation.
933 case Decl::ObjCClass:
934 case Decl::ObjCCategory:
935 case Decl::ObjCForwardProtocol:
936 case Decl::ObjCInterface:
937 break;
938
939 case Decl::ObjCProtocol:
940 Runtime->GenerateProtocol(cast<ObjCProtocolDecl>(D));
941 break;
942
943 case Decl::ObjCCategoryImpl:
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000944 // Categories have properties but don't support synthesize so we
945 // can ignore them here.
946
Daniel Dunbar41071de2008-08-15 23:26:23 +0000947 Runtime->GenerateCategory(cast<ObjCCategoryImplDecl>(D));
948 break;
949
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000950 case Decl::ObjCImplementation: {
951 ObjCImplementationDecl *OMD = cast<ObjCImplementationDecl>(D);
952 EmitObjCPropertyImplementations(OMD);
953 Runtime->GenerateClass(OMD);
Daniel Dunbar41071de2008-08-15 23:26:23 +0000954 break;
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000955 }
Daniel Dunbar41071de2008-08-15 23:26:23 +0000956 case Decl::ObjCMethod: {
957 ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(D);
958 // If this is not a prototype, emit the body.
959 if (OMD->getBody())
960 CodeGenFunction(*this).GenerateObjCMethod(OMD);
961 break;
962 }
Daniel Dunbar41071de2008-08-15 23:26:23 +0000963 case Decl::ObjCCompatibleAlias:
Daniel Dunbar662174c82008-08-29 17:28:43 +0000964 ErrorUnsupported(D, "Objective-C compatible alias");
Daniel Dunbar41071de2008-08-15 23:26:23 +0000965 break;
966
967 case Decl::LinkageSpec: {
968 LinkageSpecDecl *LSD = cast<LinkageSpecDecl>(D);
969 if (LSD->getLanguage() == LinkageSpecDecl::lang_cxx)
Daniel Dunbar488e9932008-08-16 00:56:44 +0000970 ErrorUnsupported(LSD, "linkage spec");
Daniel Dunbar41071de2008-08-15 23:26:23 +0000971 // FIXME: implement C++ linkage, C linkage works mostly by C
972 // language reuse already.
973 break;
974 }
975
976 case Decl::FileScopeAsm: {
977 FileScopeAsmDecl *AD = cast<FileScopeAsmDecl>(D);
978 std::string AsmString(AD->getAsmString()->getStrData(),
979 AD->getAsmString()->getByteLength());
980
981 const std::string &S = getModule().getModuleInlineAsm();
982 if (S.empty())
983 getModule().setModuleInlineAsm(AsmString);
984 else
985 getModule().setModuleInlineAsm(S + '\n' + AsmString);
986 break;
987 }
988
989 default:
990 // Make sure we handled everything we should, every other kind is
991 // a non-top-level decl. FIXME: Would be nice to have an
992 // isTopLevelDeclKind function. Need to recode Decl::Kind to do
993 // that easily.
994 assert(isa<TypeDecl>(D) && "Unsupported decl kind");
995 }
996}
997