1 //===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This coordinates the per-module state used while generating code. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "CodeGenModule.h" 15 #include "CodeGenFunction.h" 16 #include "clang/AST/ASTContext.h" 17 #include "clang/AST/Decl.h" 18 #include "clang/Basic/Diagnostic.h" 19 #include "clang/Basic/LangOptions.h" 20 #include "clang/Basic/SourceManager.h" 21 #include "clang/Basic/TargetInfo.h" 22 #include "llvm/CallingConv.h" 23 #include "llvm/Constants.h" 24 #include "llvm/DerivedTypes.h" 25 #include "llvm/Module.h" 26 #include "llvm/Intrinsics.h" 27 #include "llvm/Analysis/Verifier.h" 28 #include <algorithm> 29 using namespace clang; 30 using namespace CodeGen; 31 32 33 CodeGenModule::CodeGenModule(ASTContext &C, const LangOptions &LO, 34 llvm::Module &M, const llvm::TargetData &TD, 35 Diagnostic &diags) 36 : Context(C), Features(LO), TheModule(M), TheTargetData(TD), Diags(diags), 37 Types(C, M, TD), MemCpyFn(0), MemSetFn(0), CFConstantStringClassRef(0) { 38 //TODO: Make this selectable at runtime 39 Runtime = CreateObjCRuntime(M, 40 getTypes().ConvertType(getContext().IntTy), 41 getTypes().ConvertType(getContext().LongTy)); 42 } 43 44 CodeGenModule::~CodeGenModule() { 45 llvm::Function *ObjCInitFunction = Runtime->ModuleInitFunction(); 46 if (ObjCInitFunction) 47 AddGlobalCtor(ObjCInitFunction); 48 EmitStatics(); 49 EmitGlobalCtors(); 50 EmitAnnotations(); 51 delete Runtime; 52 53 // Run the verifier to check that the generated code is consistent. 54 assert(!verifyModule(TheModule)); 55 } 56 57 /// WarnUnsupported - Print out a warning that codegen doesn't support the 58 /// specified stmt yet. 59 void CodeGenModule::WarnUnsupported(const Stmt *S, const char *Type) { 60 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Warning, 61 "cannot codegen this %0 yet"); 62 SourceRange Range = S->getSourceRange(); 63 std::string Msg = Type; 64 getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID, 65 &Msg, 1, &Range, 1); 66 } 67 68 /// WarnUnsupported - Print out a warning that codegen doesn't support the 69 /// specified decl yet. 70 void CodeGenModule::WarnUnsupported(const Decl *D, const char *Type) { 71 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Warning, 72 "cannot codegen this %0 yet"); 73 std::string Msg = Type; 74 getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID, 75 &Msg, 1); 76 } 77 78 /// AddGlobalCtor - Add a function to the list that will be called before 79 /// main() runs. 80 void CodeGenModule::AddGlobalCtor(llvm::Function * Ctor) { 81 // TODO: Type coercion of void()* types. 82 GlobalCtors.push_back(Ctor); 83 } 84 85 /// EmitGlobalCtors - Generates the array of contsturctor functions to be 86 /// called on module load, if any have been registered with AddGlobalCtor. 87 void CodeGenModule::EmitGlobalCtors() { 88 if (GlobalCtors.empty()) return; 89 90 // Get the type of @llvm.global_ctors 91 std::vector<const llvm::Type*> CtorFields; 92 CtorFields.push_back(llvm::IntegerType::get(32)); 93 // Constructor function type 94 std::vector<const llvm::Type*> VoidArgs; 95 llvm::FunctionType* CtorFuncTy = 96 llvm::FunctionType::get(llvm::Type::VoidTy, VoidArgs, false); 97 98 // i32, function type pair 99 const llvm::Type *FPType = llvm::PointerType::getUnqual(CtorFuncTy); 100 llvm::StructType* CtorStructTy = 101 llvm::StructType::get(llvm::Type::Int32Ty, FPType, NULL); 102 // Array of fields 103 llvm::ArrayType* GlobalCtorsTy = 104 llvm::ArrayType::get(CtorStructTy, GlobalCtors.size()); 105 106 // Define the global variable 107 llvm::GlobalVariable *GlobalCtorsVal = 108 new llvm::GlobalVariable(GlobalCtorsTy, false, 109 llvm::GlobalValue::AppendingLinkage, 110 (llvm::Constant*)0, "llvm.global_ctors", 111 &TheModule); 112 113 // Populate the array 114 std::vector<llvm::Constant*> CtorValues; 115 llvm::Constant *MagicNumber = 116 llvm::ConstantInt::get(llvm::Type::Int32Ty, 65535, false); 117 std::vector<llvm::Constant*> StructValues; 118 for (std::vector<llvm::Constant*>::iterator I = GlobalCtors.begin(), 119 E = GlobalCtors.end(); I != E; ++I) { 120 StructValues.clear(); 121 StructValues.push_back(MagicNumber); 122 StructValues.push_back(*I); 123 124 CtorValues.push_back(llvm::ConstantStruct::get(CtorStructTy, StructValues)); 125 } 126 127 GlobalCtorsVal->setInitializer(llvm::ConstantArray::get(GlobalCtorsTy, 128 CtorValues)); 129 } 130 131 132 133 void CodeGenModule::EmitAnnotations() { 134 if (Annotations.empty()) 135 return; 136 137 // Create a new global variable for the ConstantStruct in the Module. 138 llvm::Constant *Array = 139 llvm::ConstantArray::get(llvm::ArrayType::get(Annotations[0]->getType(), 140 Annotations.size()), 141 Annotations); 142 llvm::GlobalValue *gv = 143 new llvm::GlobalVariable(Array->getType(), false, 144 llvm::GlobalValue::AppendingLinkage, Array, 145 "llvm.global.annotations", &TheModule); 146 gv->setSection("llvm.metadata"); 147 } 148 149 /// ReplaceMapValuesWith - This is a really slow and bad function that 150 /// searches for any entries in GlobalDeclMap that point to OldVal, changing 151 /// them to point to NewVal. This is badbadbad, FIXME! 152 void CodeGenModule::ReplaceMapValuesWith(llvm::Constant *OldVal, 153 llvm::Constant *NewVal) { 154 for (llvm::DenseMap<const Decl*, llvm::Constant*>::iterator 155 I = GlobalDeclMap.begin(), E = GlobalDeclMap.end(); I != E; ++I) 156 if (I->second == OldVal) I->second = NewVal; 157 } 158 159 160 llvm::Constant *CodeGenModule::GetAddrOfFunctionDecl(const FunctionDecl *D, 161 bool isDefinition) { 162 // See if it is already in the map. If so, just return it. 163 llvm::Constant *&Entry = GlobalDeclMap[D]; 164 if (Entry) return Entry; 165 166 const llvm::Type *Ty = getTypes().ConvertType(D->getType()); 167 168 // Check to see if the function already exists. 169 llvm::Function *F = getModule().getFunction(D->getName()); 170 const llvm::FunctionType *FTy = cast<llvm::FunctionType>(Ty); 171 172 // If it doesn't already exist, just create and return an entry. 173 if (F == 0) { 174 // FIXME: param attributes for sext/zext etc. 175 F = llvm::Function::Create(FTy, llvm::Function::ExternalLinkage, 176 D->getName(), &getModule()); 177 178 // Set the appropriate calling convention for the Function. 179 if (D->getAttr<FastCallAttr>()) 180 F->setCallingConv(llvm::CallingConv::Fast); 181 return Entry = F; 182 } 183 184 // If the pointer type matches, just return it. 185 llvm::Type *PFTy = llvm::PointerType::getUnqual(Ty); 186 if (PFTy == F->getType()) return Entry = F; 187 188 // If this isn't a definition, just return it casted to the right type. 189 if (!isDefinition) 190 return Entry = llvm::ConstantExpr::getBitCast(F, PFTy); 191 192 // Otherwise, we have a definition after a prototype with the wrong type. 193 // F is the Function* for the one with the wrong type, we must make a new 194 // Function* and update everything that used F (a declaration) with the new 195 // Function* (which will be a definition). 196 // 197 // This happens if there is a prototype for a function (e.g. "int f()") and 198 // then a definition of a different type (e.g. "int f(int x)"). Start by 199 // making a new function of the correct type, RAUW, then steal the name. 200 llvm::Function *NewFn = llvm::Function::Create(FTy, 201 llvm::Function::ExternalLinkage, 202 "", &getModule()); 203 NewFn->takeName(F); 204 205 // Replace uses of F with the Function we will endow with a body. 206 llvm::Constant *NewPtrForOldDecl = 207 llvm::ConstantExpr::getBitCast(NewFn, F->getType()); 208 F->replaceAllUsesWith(NewPtrForOldDecl); 209 210 // FIXME: Update the globaldeclmap for the previous decl of this name. We 211 // really want a way to walk all of these, but we don't have it yet. This 212 // is incredibly slow! 213 ReplaceMapValuesWith(F, NewPtrForOldDecl); 214 215 // Ok, delete the old function now, which is dead. 216 assert(F->isDeclaration() && "Shouldn't replace non-declaration"); 217 F->eraseFromParent(); 218 219 // Return the new function which has the right type. 220 return Entry = NewFn; 221 } 222 223 static bool IsZeroElementArray(const llvm::Type *Ty) { 224 if (const llvm::ArrayType *ATy = dyn_cast<llvm::ArrayType>(Ty)) 225 return ATy->getNumElements() == 0; 226 return false; 227 } 228 229 llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D, 230 bool isDefinition) { 231 assert(D->hasGlobalStorage() && "Not a global variable"); 232 233 // See if it is already in the map. 234 llvm::Constant *&Entry = GlobalDeclMap[D]; 235 if (Entry) return Entry; 236 237 QualType ASTTy = D->getType(); 238 const llvm::Type *Ty = getTypes().ConvertTypeForMem(ASTTy); 239 240 // Check to see if the global already exists. 241 llvm::GlobalVariable *GV = getModule().getGlobalVariable(D->getName(), true); 242 243 // If it doesn't already exist, just create and return an entry. 244 if (GV == 0) { 245 return Entry = new llvm::GlobalVariable(Ty, false, 246 llvm::GlobalValue::ExternalLinkage, 247 0, D->getName(), &getModule(), 0, 248 ASTTy.getAddressSpace()); 249 } 250 251 // If the pointer type matches, just return it. 252 llvm::Type *PTy = llvm::PointerType::getUnqual(Ty); 253 if (PTy == GV->getType()) return Entry = GV; 254 255 // If this isn't a definition, just return it casted to the right type. 256 if (!isDefinition) 257 return Entry = llvm::ConstantExpr::getBitCast(GV, PTy); 258 259 260 // Otherwise, we have a definition after a prototype with the wrong type. 261 // GV is the GlobalVariable* for the one with the wrong type, we must make a 262 /// new GlobalVariable* and update everything that used GV (a declaration) 263 // with the new GlobalVariable* (which will be a definition). 264 // 265 // This happens if there is a prototype for a global (e.g. "extern int x[];") 266 // and then a definition of a different type (e.g. "int x[10];"). Start by 267 // making a new global of the correct type, RAUW, then steal the name. 268 llvm::GlobalVariable *NewGV = 269 new llvm::GlobalVariable(Ty, false, llvm::GlobalValue::ExternalLinkage, 270 0, D->getName(), &getModule(), 0, 271 ASTTy.getAddressSpace()); 272 NewGV->takeName(GV); 273 274 // Replace uses of GV with the globalvalue we will endow with a body. 275 llvm::Constant *NewPtrForOldDecl = 276 llvm::ConstantExpr::getBitCast(NewGV, GV->getType()); 277 GV->replaceAllUsesWith(NewPtrForOldDecl); 278 279 // FIXME: Update the globaldeclmap for the previous decl of this name. We 280 // really want a way to walk all of these, but we don't have it yet. This 281 // is incredibly slow! 282 ReplaceMapValuesWith(GV, NewPtrForOldDecl); 283 284 // Verify that GV was a declaration or something like x[] which turns into 285 // [0 x type]. 286 assert((GV->isDeclaration() || 287 IsZeroElementArray(GV->getType()->getElementType())) && 288 "Shouldn't replace non-declaration"); 289 290 // Ok, delete the old global now, which is dead. 291 GV->eraseFromParent(); 292 293 // Return the new global which has the right type. 294 return Entry = NewGV; 295 } 296 297 298 void CodeGenModule::EmitObjCMethod(const ObjCMethodDecl *OMD) { 299 // If this is not a prototype, emit the body. 300 if (OMD->getBody()) 301 CodeGenFunction(*this).GenerateObjCMethod(OMD); 302 } 303 304 void CodeGenModule::EmitFunction(const FunctionDecl *FD) { 305 // If this is not a prototype, emit the body. 306 if (!FD->isThisDeclarationADefinition()) 307 return; 308 309 // If the function is a static, defer code generation until later so we can 310 // easily omit unused statics. 311 if (FD->getStorageClass() != FunctionDecl::Static) { 312 CodeGenFunction(*this).GenerateCode(FD); 313 return; 314 } 315 316 // We need to check the Module here to see if GetAddrOfFunctionDecl() has 317 // already added this function to the Module because the address of the 318 // function's prototype was taken. If this is the case, call 319 // GetAddrOfFunctionDecl to insert the static FunctionDecl into the used 320 // GlobalDeclsMap, so that EmitStatics will generate code for it later. 321 // 322 // Example: 323 // static int foo(); 324 // int bar() { return foo(); } 325 // static int foo() { return 5; } 326 if (getModule().getFunction(FD->getName())) 327 GetAddrOfFunctionDecl(FD, true); 328 329 StaticDecls.push_back(FD); 330 } 331 332 void CodeGenModule::EmitStatics() { 333 // Emit code for each used static decl encountered. Since a previously unused 334 // static decl may become used during the generation of code for a static 335 // function, iterate until no changes are made. 336 bool Changed; 337 do { 338 Changed = false; 339 for (unsigned i = 0, e = StaticDecls.size(); i != e; ++i) { 340 // Check the map of used decls for our static. If not found, continue. 341 const Decl *D = StaticDecls[i]; 342 if (!GlobalDeclMap.count(D)) 343 continue; 344 345 // If this is a function decl, generate code for the static function if it 346 // has a body. Otherwise, we must have a var decl for a static global 347 // variable. 348 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { 349 if (FD->getBody()) 350 CodeGenFunction(*this).GenerateCode(FD); 351 } else { 352 EmitGlobalVarInit(cast<VarDecl>(D)); 353 } 354 // Erase the used decl from the list. 355 StaticDecls[i] = StaticDecls.back(); 356 StaticDecls.pop_back(); 357 --i; 358 --e; 359 360 // Remember that we made a change. 361 Changed = true; 362 } 363 } while (Changed); 364 } 365 366 llvm::Constant *CodeGenModule::EmitGlobalInit(const Expr *Expr) { 367 return EmitConstantExpr(Expr); 368 } 369 370 /// EmitAnnotateAttr - Generate the llvm::ConstantStruct which contains the 371 /// annotation information for a given GlobalValue. The annotation struct is 372 /// {i8 *, i8 *, i8 *, i32}. The first field is a constant expression, the 373 /// GlobalValue being annotated. The second filed is thee constant string 374 /// created from the AnnotateAttr's annotation. The third field is a constant 375 /// string containing the name of the translation unit. The fourth field is 376 /// the line number in the file of the annotated value declaration. 377 /// 378 /// FIXME: this does not unique the annotation string constants, as llvm-gcc 379 /// appears to. 380 /// 381 llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV, 382 const AnnotateAttr *AA, 383 unsigned LineNo) { 384 llvm::Module *M = &getModule(); 385 386 // get [N x i8] constants for the annotation string, and the filename string 387 // which are the 2nd and 3rd elements of the global annotation structure. 388 const llvm::Type *SBP = llvm::PointerType::getUnqual(llvm::Type::Int8Ty); 389 llvm::Constant *anno = llvm::ConstantArray::get(AA->getAnnotation(), true); 390 llvm::Constant *unit = llvm::ConstantArray::get(M->getModuleIdentifier(), 391 true); 392 393 // Get the two global values corresponding to the ConstantArrays we just 394 // created to hold the bytes of the strings. 395 llvm::GlobalValue *annoGV = 396 new llvm::GlobalVariable(anno->getType(), false, 397 llvm::GlobalValue::InternalLinkage, anno, 398 GV->getName() + ".str", M); 399 // translation unit name string, emitted into the llvm.metadata section. 400 llvm::GlobalValue *unitGV = 401 new llvm::GlobalVariable(unit->getType(), false, 402 llvm::GlobalValue::InternalLinkage, unit, ".str", M); 403 404 // Create the ConstantStruct that is the global annotion. 405 llvm::Constant *Fields[4] = { 406 llvm::ConstantExpr::getBitCast(GV, SBP), 407 llvm::ConstantExpr::getBitCast(annoGV, SBP), 408 llvm::ConstantExpr::getBitCast(unitGV, SBP), 409 llvm::ConstantInt::get(llvm::Type::Int32Ty, LineNo) 410 }; 411 return llvm::ConstantStruct::get(Fields, 4, false); 412 } 413 414 void CodeGenModule::EmitGlobalVar(const VarDecl *D) { 415 // If the VarDecl is a static, defer code generation until later so we can 416 // easily omit unused statics. 417 if (D->getStorageClass() == VarDecl::Static) { 418 StaticDecls.push_back(D); 419 return; 420 } 421 422 // If this is just a forward declaration of the variable, don't emit it now, 423 // allow it to be emitted lazily on its first use. 424 if (D->getStorageClass() == VarDecl::Extern && D->getInit() == 0) 425 return; 426 427 EmitGlobalVarInit(D); 428 } 429 430 void CodeGenModule::EmitGlobalVarInit(const VarDecl *D) { 431 // Get the global, forcing it to be a direct reference. 432 llvm::GlobalVariable *GV = 433 cast<llvm::GlobalVariable>(GetAddrOfGlobalVar(D, true)); 434 435 // Convert the initializer, or use zero if appropriate. 436 llvm::Constant *Init = 0; 437 if (D->getInit() == 0) { 438 Init = llvm::Constant::getNullValue(GV->getType()->getElementType()); 439 } else if (D->getType()->isIntegerType()) { 440 llvm::APSInt Value(static_cast<uint32_t>( 441 getContext().getTypeSize(D->getInit()->getType()))); 442 if (D->getInit()->isIntegerConstantExpr(Value, Context)) 443 Init = llvm::ConstantInt::get(Value); 444 } 445 446 if (!Init) 447 Init = EmitGlobalInit(D->getInit()); 448 449 if (const AnnotateAttr *AA = D->getAttr<AnnotateAttr>()) { 450 SourceManager &SM = Context.getSourceManager(); 451 AddAnnotation(EmitAnnotateAttr(GV, AA, 452 SM.getLogicalLineNumber(D->getLocation()))); 453 } 454 455 assert(GV->getType()->getElementType() == Init->getType() && 456 "Initializer codegen type mismatch!"); 457 GV->setInitializer(Init); 458 459 if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>()) 460 GV->setVisibility(attr->getVisibility()); 461 // FIXME: else handle -fvisibility 462 463 // Set the llvm linkage type as appropriate. 464 if (D->getStorageClass() == VarDecl::Static) 465 GV->setLinkage(llvm::Function::InternalLinkage); 466 else if (D->getAttr<DLLImportAttr>()) 467 GV->setLinkage(llvm::Function::DLLImportLinkage); 468 else if (D->getAttr<DLLExportAttr>()) 469 GV->setLinkage(llvm::Function::DLLExportLinkage); 470 else if (D->getAttr<WeakAttr>()) 471 GV->setLinkage(llvm::GlobalVariable::WeakLinkage); 472 else { 473 // FIXME: This isn't right. This should handle common linkage and other 474 // stuff. 475 switch (D->getStorageClass()) { 476 case VarDecl::Static: assert(0 && "This case handled above"); 477 case VarDecl::Auto: 478 case VarDecl::Register: 479 assert(0 && "Can't have auto or register globals"); 480 case VarDecl::None: 481 if (!D->getInit()) 482 GV->setLinkage(llvm::GlobalVariable::WeakLinkage); 483 break; 484 case VarDecl::Extern: 485 case VarDecl::PrivateExtern: 486 // todo: common 487 break; 488 } 489 } 490 } 491 492 /// EmitGlobalVarDeclarator - Emit all the global vars attached to the specified 493 /// declarator chain. 494 void CodeGenModule::EmitGlobalVarDeclarator(const VarDecl *D) { 495 for (; D; D = cast_or_null<VarDecl>(D->getNextDeclarator())) 496 if (D->isFileVarDecl()) 497 EmitGlobalVar(D); 498 } 499 500 void CodeGenModule::UpdateCompletedType(const TagDecl *TD) { 501 // Make sure that this type is translated. 502 Types.UpdateCompletedType(TD); 503 } 504 505 506 /// getBuiltinLibFunction 507 llvm::Function *CodeGenModule::getBuiltinLibFunction(unsigned BuiltinID) { 508 if (BuiltinID > BuiltinFunctions.size()) 509 BuiltinFunctions.resize(BuiltinID); 510 511 // Cache looked up functions. Since builtin id #0 is invalid we don't reserve 512 // a slot for it. 513 assert(BuiltinID && "Invalid Builtin ID"); 514 llvm::Function *&FunctionSlot = BuiltinFunctions[BuiltinID-1]; 515 if (FunctionSlot) 516 return FunctionSlot; 517 518 assert(Context.BuiltinInfo.isLibFunction(BuiltinID) && "isn't a lib fn"); 519 520 // Get the name, skip over the __builtin_ prefix. 521 const char *Name = Context.BuiltinInfo.GetName(BuiltinID)+10; 522 523 // Get the type for the builtin. 524 QualType Type = Context.BuiltinInfo.GetBuiltinType(BuiltinID, Context); 525 const llvm::FunctionType *Ty = 526 cast<llvm::FunctionType>(getTypes().ConvertType(Type)); 527 528 // FIXME: This has a serious problem with code like this: 529 // void abs() {} 530 // ... __builtin_abs(x); 531 // The two versions of abs will collide. The fix is for the builtin to win, 532 // and for the existing one to be turned into a constantexpr cast of the 533 // builtin. In the case where the existing one is a static function, it 534 // should just be renamed. 535 if (llvm::Function *Existing = getModule().getFunction(Name)) { 536 if (Existing->getFunctionType() == Ty && Existing->hasExternalLinkage()) 537 return FunctionSlot = Existing; 538 assert(Existing == 0 && "FIXME: Name collision"); 539 } 540 541 // FIXME: param attributes for sext/zext etc. 542 return FunctionSlot = 543 llvm::Function::Create(Ty, llvm::Function::ExternalLinkage, Name, 544 &getModule()); 545 } 546 547 llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,const llvm::Type **Tys, 548 unsigned NumTys) { 549 return llvm::Intrinsic::getDeclaration(&getModule(), 550 (llvm::Intrinsic::ID)IID, Tys, NumTys); 551 } 552 553 llvm::Function *CodeGenModule::getMemCpyFn() { 554 if (MemCpyFn) return MemCpyFn; 555 llvm::Intrinsic::ID IID; 556 switch (Context.Target.getPointerWidth(0)) { 557 default: assert(0 && "Unknown ptr width"); 558 case 32: IID = llvm::Intrinsic::memcpy_i32; break; 559 case 64: IID = llvm::Intrinsic::memcpy_i64; break; 560 } 561 return MemCpyFn = getIntrinsic(IID); 562 } 563 564 llvm::Function *CodeGenModule::getMemSetFn() { 565 if (MemSetFn) return MemSetFn; 566 llvm::Intrinsic::ID IID; 567 switch (Context.Target.getPointerWidth(0)) { 568 default: assert(0 && "Unknown ptr width"); 569 case 32: IID = llvm::Intrinsic::memset_i32; break; 570 case 64: IID = llvm::Intrinsic::memset_i64; break; 571 } 572 return MemSetFn = getIntrinsic(IID); 573 } 574 575 llvm::Constant *CodeGenModule:: 576 GetAddrOfConstantCFString(const std::string &str) { 577 llvm::StringMapEntry<llvm::Constant *> &Entry = 578 CFConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]); 579 580 if (Entry.getValue()) 581 return Entry.getValue(); 582 583 std::vector<llvm::Constant*> Fields; 584 585 if (!CFConstantStringClassRef) { 586 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy); 587 Ty = llvm::ArrayType::get(Ty, 0); 588 589 CFConstantStringClassRef = 590 new llvm::GlobalVariable(Ty, false, 591 llvm::GlobalVariable::ExternalLinkage, 0, 592 "__CFConstantStringClassReference", 593 &getModule()); 594 } 595 596 // Class pointer. 597 llvm::Constant *Zero = llvm::Constant::getNullValue(llvm::Type::Int32Ty); 598 llvm::Constant *Zeros[] = { Zero, Zero }; 599 llvm::Constant *C = 600 llvm::ConstantExpr::getGetElementPtr(CFConstantStringClassRef, Zeros, 2); 601 Fields.push_back(C); 602 603 // Flags. 604 const llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy); 605 Fields.push_back(llvm::ConstantInt::get(Ty, 1992)); 606 607 // String pointer. 608 C = llvm::ConstantArray::get(str); 609 C = new llvm::GlobalVariable(C->getType(), true, 610 llvm::GlobalValue::InternalLinkage, 611 C, ".str", &getModule()); 612 613 C = llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2); 614 Fields.push_back(C); 615 616 // String length. 617 Ty = getTypes().ConvertType(getContext().LongTy); 618 Fields.push_back(llvm::ConstantInt::get(Ty, str.length())); 619 620 // The struct. 621 Ty = getTypes().ConvertType(getContext().getCFConstantStringType()); 622 C = llvm::ConstantStruct::get(cast<llvm::StructType>(Ty), Fields); 623 llvm::GlobalVariable *GV = 624 new llvm::GlobalVariable(C->getType(), true, 625 llvm::GlobalVariable::InternalLinkage, 626 C, "", &getModule()); 627 GV->setSection("__DATA,__cfstring"); 628 Entry.setValue(GV); 629 return GV; 630 } 631 632 /// GenerateWritableString -- Creates storage for a string literal. 633 static llvm::Constant *GenerateStringLiteral(const std::string &str, 634 bool constant, 635 CodeGenModule &CGM) { 636 // Create Constant for this string literal 637 llvm::Constant *C=llvm::ConstantArray::get(str); 638 639 // Create a global variable for this string 640 C = new llvm::GlobalVariable(C->getType(), constant, 641 llvm::GlobalValue::InternalLinkage, 642 C, ".str", &CGM.getModule()); 643 return C; 644 } 645 646 /// CodeGenModule::GetAddrOfConstantString -- returns a pointer to the character 647 /// array containing the literal. The result is pointer to array type. 648 llvm::Constant *CodeGenModule::GetAddrOfConstantString(const std::string &str) { 649 // Don't share any string literals if writable-strings is turned on. 650 if (Features.WritableStrings) 651 return GenerateStringLiteral(str, false, *this); 652 653 llvm::StringMapEntry<llvm::Constant *> &Entry = 654 ConstantStringMap.GetOrCreateValue(&str[0], &str[str.length()]); 655 656 if (Entry.getValue()) 657 return Entry.getValue(); 658 659 // Create a global variable for this. 660 llvm::Constant *C = GenerateStringLiteral(str, true, *this); 661 Entry.setValue(C); 662 return C; 663 } 664