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