1 //===-- Value.cpp - Implement the Value class -----------------------------===// 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 file implements the Value, ValueHandle, and User classes. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "llvm/IR/Value.h" 15 #include "LLVMContextImpl.h" 16 #include "llvm/ADT/DenseMap.h" 17 #include "llvm/ADT/SmallString.h" 18 #include "llvm/IR/CallSite.h" 19 #include "llvm/IR/Constant.h" 20 #include "llvm/IR/Constants.h" 21 #include "llvm/IR/DataLayout.h" 22 #include "llvm/IR/DerivedTypes.h" 23 #include "llvm/IR/GetElementPtrTypeIterator.h" 24 #include "llvm/IR/InstrTypes.h" 25 #include "llvm/IR/Instructions.h" 26 #include "llvm/IR/Module.h" 27 #include "llvm/IR/Operator.h" 28 #include "llvm/IR/ValueHandle.h" 29 #include "llvm/IR/ValueSymbolTable.h" 30 #include "llvm/Support/Debug.h" 31 #include "llvm/Support/ErrorHandling.h" 32 #include "llvm/Support/ManagedStatic.h" 33 #include <algorithm> 34 using namespace llvm; 35 36 //===----------------------------------------------------------------------===// 37 // Value Class 38 //===----------------------------------------------------------------------===// 39 40 static inline Type *checkType(Type *Ty) { 41 assert(Ty && "Value defined with a null type: Error!"); 42 return Ty; 43 } 44 45 Value::Value(Type *ty, unsigned scid) 46 : VTy(checkType(ty)), UseList(nullptr), SubclassID(scid), HasValueHandle(0), 47 SubclassOptionalData(0), SubclassData(0), NumOperands(0) { 48 // FIXME: Why isn't this in the subclass gunk?? 49 // Note, we cannot call isa<CallInst> before the CallInst has been 50 // constructed. 51 if (SubclassID == Instruction::Call || SubclassID == Instruction::Invoke) 52 assert((VTy->isFirstClassType() || VTy->isVoidTy() || VTy->isStructTy()) && 53 "invalid CallInst type!"); 54 else if (SubclassID != BasicBlockVal && 55 (SubclassID < ConstantFirstVal || SubclassID > ConstantLastVal)) 56 assert((VTy->isFirstClassType() || VTy->isVoidTy()) && 57 "Cannot create non-first-class values except for constants!"); 58 } 59 60 Value::~Value() { 61 // Notify all ValueHandles (if present) that this value is going away. 62 if (HasValueHandle) 63 ValueHandleBase::ValueIsDeleted(this); 64 if (isUsedByMetadata()) 65 ValueAsMetadata::handleDeletion(this); 66 67 #ifndef NDEBUG // Only in -g mode... 68 // Check to make sure that there are no uses of this value that are still 69 // around when the value is destroyed. If there are, then we have a dangling 70 // reference and something is wrong. This code is here to print out what is 71 // still being referenced. The value in question should be printed as 72 // a <badref> 73 // 74 if (!use_empty()) { 75 dbgs() << "While deleting: " << *VTy << " %" << getName() << "\n"; 76 for (use_iterator I = use_begin(), E = use_end(); I != E; ++I) 77 dbgs() << "Use still stuck around after Def is destroyed:" 78 << **I << "\n"; 79 } 80 #endif 81 assert(use_empty() && "Uses remain when a value is destroyed!"); 82 83 // If this value is named, destroy the name. This should not be in a symtab 84 // at this point. 85 destroyValueName(); 86 } 87 88 void Value::destroyValueName() { 89 ValueName *Name = getValueName(); 90 if (Name) 91 Name->Destroy(); 92 setValueName(nullptr); 93 } 94 95 bool Value::hasNUses(unsigned N) const { 96 const_use_iterator UI = use_begin(), E = use_end(); 97 98 for (; N; --N, ++UI) 99 if (UI == E) return false; // Too few. 100 return UI == E; 101 } 102 103 bool Value::hasNUsesOrMore(unsigned N) const { 104 const_use_iterator UI = use_begin(), E = use_end(); 105 106 for (; N; --N, ++UI) 107 if (UI == E) return false; // Too few. 108 109 return true; 110 } 111 112 bool Value::isUsedInBasicBlock(const BasicBlock *BB) const { 113 // This can be computed either by scanning the instructions in BB, or by 114 // scanning the use list of this Value. Both lists can be very long, but 115 // usually one is quite short. 116 // 117 // Scan both lists simultaneously until one is exhausted. This limits the 118 // search to the shorter list. 119 BasicBlock::const_iterator BI = BB->begin(), BE = BB->end(); 120 const_user_iterator UI = user_begin(), UE = user_end(); 121 for (; BI != BE && UI != UE; ++BI, ++UI) { 122 // Scan basic block: Check if this Value is used by the instruction at BI. 123 if (std::find(BI->op_begin(), BI->op_end(), this) != BI->op_end()) 124 return true; 125 // Scan use list: Check if the use at UI is in BB. 126 const Instruction *User = dyn_cast<Instruction>(*UI); 127 if (User && User->getParent() == BB) 128 return true; 129 } 130 return false; 131 } 132 133 unsigned Value::getNumUses() const { 134 return (unsigned)std::distance(use_begin(), use_end()); 135 } 136 137 static bool getSymTab(Value *V, ValueSymbolTable *&ST) { 138 ST = nullptr; 139 if (Instruction *I = dyn_cast<Instruction>(V)) { 140 if (BasicBlock *P = I->getParent()) 141 if (Function *PP = P->getParent()) 142 ST = &PP->getValueSymbolTable(); 143 } else if (BasicBlock *BB = dyn_cast<BasicBlock>(V)) { 144 if (Function *P = BB->getParent()) 145 ST = &P->getValueSymbolTable(); 146 } else if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) { 147 if (Module *P = GV->getParent()) 148 ST = &P->getValueSymbolTable(); 149 } else if (Argument *A = dyn_cast<Argument>(V)) { 150 if (Function *P = A->getParent()) 151 ST = &P->getValueSymbolTable(); 152 } else { 153 assert(isa<Constant>(V) && "Unknown value type!"); 154 return true; // no name is setable for this. 155 } 156 return false; 157 } 158 159 StringRef Value::getName() const { 160 // Make sure the empty string is still a C string. For historical reasons, 161 // some clients want to call .data() on the result and expect it to be null 162 // terminated. 163 if (!getValueName()) 164 return StringRef("", 0); 165 return getValueName()->getKey(); 166 } 167 168 void Value::setName(const Twine &NewName) { 169 // Fast path for common IRBuilder case of setName("") when there is no name. 170 if (NewName.isTriviallyEmpty() && !hasName()) 171 return; 172 173 SmallString<256> NameData; 174 StringRef NameRef = NewName.toStringRef(NameData); 175 assert(NameRef.find_first_of(0) == StringRef::npos && 176 "Null bytes are not allowed in names"); 177 178 // Name isn't changing? 179 if (getName() == NameRef) 180 return; 181 182 assert(!getType()->isVoidTy() && "Cannot assign a name to void values!"); 183 184 // Get the symbol table to update for this object. 185 ValueSymbolTable *ST; 186 if (getSymTab(this, ST)) 187 return; // Cannot set a name on this value (e.g. constant). 188 189 if (Function *F = dyn_cast<Function>(this)) 190 getContext().pImpl->IntrinsicIDCache.erase(F); 191 192 if (!ST) { // No symbol table to update? Just do the change. 193 if (NameRef.empty()) { 194 // Free the name for this value. 195 destroyValueName(); 196 return; 197 } 198 199 // NOTE: Could optimize for the case the name is shrinking to not deallocate 200 // then reallocated. 201 destroyValueName(); 202 203 // Create the new name. 204 setValueName(ValueName::Create(NameRef)); 205 getValueName()->setValue(this); 206 return; 207 } 208 209 // NOTE: Could optimize for the case the name is shrinking to not deallocate 210 // then reallocated. 211 if (hasName()) { 212 // Remove old name. 213 ST->removeValueName(getValueName()); 214 destroyValueName(); 215 216 if (NameRef.empty()) 217 return; 218 } 219 220 // Name is changing to something new. 221 setValueName(ST->createValueName(NameRef, this)); 222 } 223 224 void Value::takeName(Value *V) { 225 ValueSymbolTable *ST = nullptr; 226 // If this value has a name, drop it. 227 if (hasName()) { 228 // Get the symtab this is in. 229 if (getSymTab(this, ST)) { 230 // We can't set a name on this value, but we need to clear V's name if 231 // it has one. 232 if (V->hasName()) V->setName(""); 233 return; // Cannot set a name on this value (e.g. constant). 234 } 235 236 // Remove old name. 237 if (ST) 238 ST->removeValueName(getValueName()); 239 destroyValueName(); 240 } 241 242 // Now we know that this has no name. 243 244 // If V has no name either, we're done. 245 if (!V->hasName()) return; 246 247 // Get this's symtab if we didn't before. 248 if (!ST) { 249 if (getSymTab(this, ST)) { 250 // Clear V's name. 251 V->setName(""); 252 return; // Cannot set a name on this value (e.g. constant). 253 } 254 } 255 256 // Get V's ST, this should always succed, because V has a name. 257 ValueSymbolTable *VST; 258 bool Failure = getSymTab(V, VST); 259 assert(!Failure && "V has a name, so it should have a ST!"); (void)Failure; 260 261 // If these values are both in the same symtab, we can do this very fast. 262 // This works even if both values have no symtab yet. 263 if (ST == VST) { 264 // Take the name! 265 setValueName(V->getValueName()); 266 V->setValueName(nullptr); 267 getValueName()->setValue(this); 268 return; 269 } 270 271 // Otherwise, things are slightly more complex. Remove V's name from VST and 272 // then reinsert it into ST. 273 274 if (VST) 275 VST->removeValueName(V->getValueName()); 276 setValueName(V->getValueName()); 277 V->setValueName(nullptr); 278 getValueName()->setValue(this); 279 280 if (ST) 281 ST->reinsertValue(this); 282 } 283 284 #ifndef NDEBUG 285 static bool contains(SmallPtrSetImpl<ConstantExpr *> &Cache, ConstantExpr *Expr, 286 Constant *C) { 287 if (!Cache.insert(Expr).second) 288 return false; 289 290 for (auto &O : Expr->operands()) { 291 if (O == C) 292 return true; 293 auto *CE = dyn_cast<ConstantExpr>(O); 294 if (!CE) 295 continue; 296 if (contains(Cache, CE, C)) 297 return true; 298 } 299 return false; 300 } 301 302 static bool contains(Value *Expr, Value *V) { 303 if (Expr == V) 304 return true; 305 306 auto *C = dyn_cast<Constant>(V); 307 if (!C) 308 return false; 309 310 auto *CE = dyn_cast<ConstantExpr>(Expr); 311 if (!CE) 312 return false; 313 314 SmallPtrSet<ConstantExpr *, 4> Cache; 315 return contains(Cache, CE, C); 316 } 317 #endif 318 319 void Value::replaceAllUsesWith(Value *New) { 320 assert(New && "Value::replaceAllUsesWith(<null>) is invalid!"); 321 assert(!contains(New, this) && 322 "this->replaceAllUsesWith(expr(this)) is NOT valid!"); 323 assert(New->getType() == getType() && 324 "replaceAllUses of value with new value of different type!"); 325 326 // Notify all ValueHandles (if present) that this value is going away. 327 if (HasValueHandle) 328 ValueHandleBase::ValueIsRAUWd(this, New); 329 if (isUsedByMetadata()) 330 ValueAsMetadata::handleRAUW(this, New); 331 332 while (!use_empty()) { 333 Use &U = *UseList; 334 // Must handle Constants specially, we cannot call replaceUsesOfWith on a 335 // constant because they are uniqued. 336 if (auto *C = dyn_cast<Constant>(U.getUser())) { 337 if (!isa<GlobalValue>(C)) { 338 C->replaceUsesOfWithOnConstant(this, New, &U); 339 continue; 340 } 341 } 342 343 U.set(New); 344 } 345 346 if (BasicBlock *BB = dyn_cast<BasicBlock>(this)) 347 BB->replaceSuccessorsPhiUsesWith(cast<BasicBlock>(New)); 348 } 349 350 // Like replaceAllUsesWith except it does not handle constants or basic blocks. 351 // This routine leaves uses within BB. 352 void Value::replaceUsesOutsideBlock(Value *New, BasicBlock *BB) { 353 assert(New && "Value::replaceUsesOutsideBlock(<null>, BB) is invalid!"); 354 assert(!contains(New, this) && 355 "this->replaceUsesOutsideBlock(expr(this), BB) is NOT valid!"); 356 assert(New->getType() == getType() && 357 "replaceUses of value with new value of different type!"); 358 assert(BB && "Basic block that may contain a use of 'New' must be defined\n"); 359 360 use_iterator UI = use_begin(), E = use_end(); 361 for (; UI != E;) { 362 Use &U = *UI; 363 ++UI; 364 auto *Usr = dyn_cast<Instruction>(U.getUser()); 365 if (Usr && Usr->getParent() == BB) 366 continue; 367 U.set(New); 368 } 369 return; 370 } 371 372 namespace { 373 // Various metrics for how much to strip off of pointers. 374 enum PointerStripKind { 375 PSK_ZeroIndices, 376 PSK_ZeroIndicesAndAliases, 377 PSK_InBoundsConstantIndices, 378 PSK_InBounds 379 }; 380 381 template <PointerStripKind StripKind> 382 static Value *stripPointerCastsAndOffsets(Value *V) { 383 if (!V->getType()->isPointerTy()) 384 return V; 385 386 // Even though we don't look through PHI nodes, we could be called on an 387 // instruction in an unreachable block, which may be on a cycle. 388 SmallPtrSet<Value *, 4> Visited; 389 390 Visited.insert(V); 391 do { 392 if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) { 393 switch (StripKind) { 394 case PSK_ZeroIndicesAndAliases: 395 case PSK_ZeroIndices: 396 if (!GEP->hasAllZeroIndices()) 397 return V; 398 break; 399 case PSK_InBoundsConstantIndices: 400 if (!GEP->hasAllConstantIndices()) 401 return V; 402 // fallthrough 403 case PSK_InBounds: 404 if (!GEP->isInBounds()) 405 return V; 406 break; 407 } 408 V = GEP->getPointerOperand(); 409 } else if (Operator::getOpcode(V) == Instruction::BitCast || 410 Operator::getOpcode(V) == Instruction::AddrSpaceCast) { 411 V = cast<Operator>(V)->getOperand(0); 412 } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) { 413 if (StripKind == PSK_ZeroIndices || GA->mayBeOverridden()) 414 return V; 415 V = GA->getAliasee(); 416 } else { 417 return V; 418 } 419 assert(V->getType()->isPointerTy() && "Unexpected operand type!"); 420 } while (Visited.insert(V).second); 421 422 return V; 423 } 424 } // namespace 425 426 Value *Value::stripPointerCasts() { 427 return stripPointerCastsAndOffsets<PSK_ZeroIndicesAndAliases>(this); 428 } 429 430 Value *Value::stripPointerCastsNoFollowAliases() { 431 return stripPointerCastsAndOffsets<PSK_ZeroIndices>(this); 432 } 433 434 Value *Value::stripInBoundsConstantOffsets() { 435 return stripPointerCastsAndOffsets<PSK_InBoundsConstantIndices>(this); 436 } 437 438 Value *Value::stripAndAccumulateInBoundsConstantOffsets(const DataLayout &DL, 439 APInt &Offset) { 440 if (!getType()->isPointerTy()) 441 return this; 442 443 assert(Offset.getBitWidth() == DL.getPointerSizeInBits(cast<PointerType>( 444 getType())->getAddressSpace()) && 445 "The offset must have exactly as many bits as our pointer."); 446 447 // Even though we don't look through PHI nodes, we could be called on an 448 // instruction in an unreachable block, which may be on a cycle. 449 SmallPtrSet<Value *, 4> Visited; 450 Visited.insert(this); 451 Value *V = this; 452 do { 453 if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) { 454 if (!GEP->isInBounds()) 455 return V; 456 APInt GEPOffset(Offset); 457 if (!GEP->accumulateConstantOffset(DL, GEPOffset)) 458 return V; 459 Offset = GEPOffset; 460 V = GEP->getPointerOperand(); 461 } else if (Operator::getOpcode(V) == Instruction::BitCast || 462 Operator::getOpcode(V) == Instruction::AddrSpaceCast) { 463 V = cast<Operator>(V)->getOperand(0); 464 } else if (GlobalAlias *GA = dyn_cast<GlobalAlias>(V)) { 465 V = GA->getAliasee(); 466 } else { 467 return V; 468 } 469 assert(V->getType()->isPointerTy() && "Unexpected operand type!"); 470 } while (Visited.insert(V).second); 471 472 return V; 473 } 474 475 Value *Value::stripInBoundsOffsets() { 476 return stripPointerCastsAndOffsets<PSK_InBounds>(this); 477 } 478 479 /// \brief Check if Value is always a dereferenceable pointer. 480 /// 481 /// Test if V is always a pointer to allocated and suitably aligned memory for 482 /// a simple load or store. 483 static bool isDereferenceablePointer(const Value *V, const DataLayout *DL, 484 SmallPtrSetImpl<const Value *> &Visited) { 485 // Note that it is not safe to speculate into a malloc'd region because 486 // malloc may return null. 487 488 // These are obviously ok. 489 if (isa<AllocaInst>(V)) return true; 490 491 // It's not always safe to follow a bitcast, for example: 492 // bitcast i8* (alloca i8) to i32* 493 // would result in a 4-byte load from a 1-byte alloca. However, 494 // if we're casting from a pointer from a type of larger size 495 // to a type of smaller size (or the same size), and the alignment 496 // is at least as large as for the resulting pointer type, then 497 // we can look through the bitcast. 498 if (DL) 499 if (const BitCastInst* BC = dyn_cast<BitCastInst>(V)) { 500 Type *STy = BC->getSrcTy()->getPointerElementType(), 501 *DTy = BC->getDestTy()->getPointerElementType(); 502 if (STy->isSized() && DTy->isSized() && 503 (DL->getTypeStoreSize(STy) >= 504 DL->getTypeStoreSize(DTy)) && 505 (DL->getABITypeAlignment(STy) >= 506 DL->getABITypeAlignment(DTy))) 507 return isDereferenceablePointer(BC->getOperand(0), DL, Visited); 508 } 509 510 // Global variables which can't collapse to null are ok. 511 if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)) 512 return !GV->hasExternalWeakLinkage(); 513 514 // byval arguments are okay. Arguments specifically marked as 515 // dereferenceable are okay too. 516 if (const Argument *A = dyn_cast<Argument>(V)) { 517 if (A->hasByValAttr()) 518 return true; 519 else if (uint64_t Bytes = A->getDereferenceableBytes()) { 520 Type *Ty = V->getType()->getPointerElementType(); 521 if (Ty->isSized() && DL && DL->getTypeStoreSize(Ty) <= Bytes) 522 return true; 523 } 524 525 return false; 526 } 527 528 // Return values from call sites specifically marked as dereferenceable are 529 // also okay. 530 if (ImmutableCallSite CS = V) { 531 if (uint64_t Bytes = CS.getDereferenceableBytes(0)) { 532 Type *Ty = V->getType()->getPointerElementType(); 533 if (Ty->isSized() && DL && DL->getTypeStoreSize(Ty) <= Bytes) 534 return true; 535 } 536 } 537 538 // For GEPs, determine if the indexing lands within the allocated object. 539 if (const GEPOperator *GEP = dyn_cast<GEPOperator>(V)) { 540 // Conservatively require that the base pointer be fully dereferenceable. 541 if (!Visited.insert(GEP->getOperand(0)).second) 542 return false; 543 if (!isDereferenceablePointer(GEP->getOperand(0), DL, Visited)) 544 return false; 545 // Check the indices. 546 gep_type_iterator GTI = gep_type_begin(GEP); 547 for (User::const_op_iterator I = GEP->op_begin()+1, 548 E = GEP->op_end(); I != E; ++I) { 549 Value *Index = *I; 550 Type *Ty = *GTI++; 551 // Struct indices can't be out of bounds. 552 if (isa<StructType>(Ty)) 553 continue; 554 ConstantInt *CI = dyn_cast<ConstantInt>(Index); 555 if (!CI) 556 return false; 557 // Zero is always ok. 558 if (CI->isZero()) 559 continue; 560 // Check to see that it's within the bounds of an array. 561 ArrayType *ATy = dyn_cast<ArrayType>(Ty); 562 if (!ATy) 563 return false; 564 if (CI->getValue().getActiveBits() > 64) 565 return false; 566 if (CI->getZExtValue() >= ATy->getNumElements()) 567 return false; 568 } 569 // Indices check out; this is dereferenceable. 570 return true; 571 } 572 573 if (const AddrSpaceCastInst *ASC = dyn_cast<AddrSpaceCastInst>(V)) 574 return isDereferenceablePointer(ASC->getOperand(0), DL, Visited); 575 576 // If we don't know, assume the worst. 577 return false; 578 } 579 580 bool Value::isDereferenceablePointer(const DataLayout *DL) const { 581 // When dereferenceability information is provided by a dereferenceable 582 // attribute, we know exactly how many bytes are dereferenceable. If we can 583 // determine the exact offset to the attributed variable, we can use that 584 // information here. 585 Type *Ty = getType()->getPointerElementType(); 586 if (Ty->isSized() && DL) { 587 APInt Offset(DL->getTypeStoreSizeInBits(getType()), 0); 588 const Value *BV = stripAndAccumulateInBoundsConstantOffsets(*DL, Offset); 589 590 APInt DerefBytes(Offset.getBitWidth(), 0); 591 if (const Argument *A = dyn_cast<Argument>(BV)) 592 DerefBytes = A->getDereferenceableBytes(); 593 else if (ImmutableCallSite CS = BV) 594 DerefBytes = CS.getDereferenceableBytes(0); 595 596 if (DerefBytes.getBoolValue() && Offset.isNonNegative()) { 597 if (DerefBytes.uge(Offset + DL->getTypeStoreSize(Ty))) 598 return true; 599 } 600 } 601 602 SmallPtrSet<const Value *, 32> Visited; 603 return ::isDereferenceablePointer(this, DL, Visited); 604 } 605 606 Value *Value::DoPHITranslation(const BasicBlock *CurBB, 607 const BasicBlock *PredBB) { 608 PHINode *PN = dyn_cast<PHINode>(this); 609 if (PN && PN->getParent() == CurBB) 610 return PN->getIncomingValueForBlock(PredBB); 611 return this; 612 } 613 614 LLVMContext &Value::getContext() const { return VTy->getContext(); } 615 616 void Value::reverseUseList() { 617 if (!UseList || !UseList->Next) 618 // No need to reverse 0 or 1 uses. 619 return; 620 621 Use *Head = UseList; 622 Use *Current = UseList->Next; 623 Head->Next = nullptr; 624 while (Current) { 625 Use *Next = Current->Next; 626 Current->Next = Head; 627 Head->setPrev(&Current->Next); 628 Head = Current; 629 Current = Next; 630 } 631 UseList = Head; 632 Head->setPrev(&UseList); 633 } 634 635 //===----------------------------------------------------------------------===// 636 // ValueHandleBase Class 637 //===----------------------------------------------------------------------===// 638 639 void ValueHandleBase::AddToExistingUseList(ValueHandleBase **List) { 640 assert(List && "Handle list is null?"); 641 642 // Splice ourselves into the list. 643 Next = *List; 644 *List = this; 645 setPrevPtr(List); 646 if (Next) { 647 Next->setPrevPtr(&Next); 648 assert(V == Next->V && "Added to wrong list?"); 649 } 650 } 651 652 void ValueHandleBase::AddToExistingUseListAfter(ValueHandleBase *List) { 653 assert(List && "Must insert after existing node"); 654 655 Next = List->Next; 656 setPrevPtr(&List->Next); 657 List->Next = this; 658 if (Next) 659 Next->setPrevPtr(&Next); 660 } 661 662 void ValueHandleBase::AddToUseList() { 663 assert(V && "Null pointer doesn't have a use list!"); 664 665 LLVMContextImpl *pImpl = V->getContext().pImpl; 666 667 if (V->HasValueHandle) { 668 // If this value already has a ValueHandle, then it must be in the 669 // ValueHandles map already. 670 ValueHandleBase *&Entry = pImpl->ValueHandles[V]; 671 assert(Entry && "Value doesn't have any handles?"); 672 AddToExistingUseList(&Entry); 673 return; 674 } 675 676 // Ok, it doesn't have any handles yet, so we must insert it into the 677 // DenseMap. However, doing this insertion could cause the DenseMap to 678 // reallocate itself, which would invalidate all of the PrevP pointers that 679 // point into the old table. Handle this by checking for reallocation and 680 // updating the stale pointers only if needed. 681 DenseMap<Value*, ValueHandleBase*> &Handles = pImpl->ValueHandles; 682 const void *OldBucketPtr = Handles.getPointerIntoBucketsArray(); 683 684 ValueHandleBase *&Entry = Handles[V]; 685 assert(!Entry && "Value really did already have handles?"); 686 AddToExistingUseList(&Entry); 687 V->HasValueHandle = true; 688 689 // If reallocation didn't happen or if this was the first insertion, don't 690 // walk the table. 691 if (Handles.isPointerIntoBucketsArray(OldBucketPtr) || 692 Handles.size() == 1) { 693 return; 694 } 695 696 // Okay, reallocation did happen. Fix the Prev Pointers. 697 for (DenseMap<Value*, ValueHandleBase*>::iterator I = Handles.begin(), 698 E = Handles.end(); I != E; ++I) { 699 assert(I->second && I->first == I->second->V && 700 "List invariant broken!"); 701 I->second->setPrevPtr(&I->second); 702 } 703 } 704 705 void ValueHandleBase::RemoveFromUseList() { 706 assert(V && V->HasValueHandle && 707 "Pointer doesn't have a use list!"); 708 709 // Unlink this from its use list. 710 ValueHandleBase **PrevPtr = getPrevPtr(); 711 assert(*PrevPtr == this && "List invariant broken"); 712 713 *PrevPtr = Next; 714 if (Next) { 715 assert(Next->getPrevPtr() == &Next && "List invariant broken"); 716 Next->setPrevPtr(PrevPtr); 717 return; 718 } 719 720 // If the Next pointer was null, then it is possible that this was the last 721 // ValueHandle watching VP. If so, delete its entry from the ValueHandles 722 // map. 723 LLVMContextImpl *pImpl = V->getContext().pImpl; 724 DenseMap<Value*, ValueHandleBase*> &Handles = pImpl->ValueHandles; 725 if (Handles.isPointerIntoBucketsArray(PrevPtr)) { 726 Handles.erase(V); 727 V->HasValueHandle = false; 728 } 729 } 730 731 732 void ValueHandleBase::ValueIsDeleted(Value *V) { 733 assert(V->HasValueHandle && "Should only be called if ValueHandles present"); 734 735 // Get the linked list base, which is guaranteed to exist since the 736 // HasValueHandle flag is set. 737 LLVMContextImpl *pImpl = V->getContext().pImpl; 738 ValueHandleBase *Entry = pImpl->ValueHandles[V]; 739 assert(Entry && "Value bit set but no entries exist"); 740 741 // We use a local ValueHandleBase as an iterator so that ValueHandles can add 742 // and remove themselves from the list without breaking our iteration. This 743 // is not really an AssertingVH; we just have to give ValueHandleBase a kind. 744 // Note that we deliberately do not the support the case when dropping a value 745 // handle results in a new value handle being permanently added to the list 746 // (as might occur in theory for CallbackVH's): the new value handle will not 747 // be processed and the checking code will mete out righteous punishment if 748 // the handle is still present once we have finished processing all the other 749 // value handles (it is fine to momentarily add then remove a value handle). 750 for (ValueHandleBase Iterator(Assert, *Entry); Entry; Entry = Iterator.Next) { 751 Iterator.RemoveFromUseList(); 752 Iterator.AddToExistingUseListAfter(Entry); 753 assert(Entry->Next == &Iterator && "Loop invariant broken."); 754 755 switch (Entry->getKind()) { 756 case Assert: 757 break; 758 case Tracking: 759 // Mark that this value has been deleted by setting it to an invalid Value 760 // pointer. 761 Entry->operator=(DenseMapInfo<Value *>::getTombstoneKey()); 762 break; 763 case Weak: 764 // Weak just goes to null, which will unlink it from the list. 765 Entry->operator=(nullptr); 766 break; 767 case Callback: 768 // Forward to the subclass's implementation. 769 static_cast<CallbackVH*>(Entry)->deleted(); 770 break; 771 } 772 } 773 774 // All callbacks, weak references, and assertingVHs should be dropped by now. 775 if (V->HasValueHandle) { 776 #ifndef NDEBUG // Only in +Asserts mode... 777 dbgs() << "While deleting: " << *V->getType() << " %" << V->getName() 778 << "\n"; 779 if (pImpl->ValueHandles[V]->getKind() == Assert) 780 llvm_unreachable("An asserting value handle still pointed to this" 781 " value!"); 782 783 #endif 784 llvm_unreachable("All references to V were not removed?"); 785 } 786 } 787 788 789 void ValueHandleBase::ValueIsRAUWd(Value *Old, Value *New) { 790 assert(Old->HasValueHandle &&"Should only be called if ValueHandles present"); 791 assert(Old != New && "Changing value into itself!"); 792 assert(Old->getType() == New->getType() && 793 "replaceAllUses of value with new value of different type!"); 794 795 // Get the linked list base, which is guaranteed to exist since the 796 // HasValueHandle flag is set. 797 LLVMContextImpl *pImpl = Old->getContext().pImpl; 798 ValueHandleBase *Entry = pImpl->ValueHandles[Old]; 799 800 assert(Entry && "Value bit set but no entries exist"); 801 802 // We use a local ValueHandleBase as an iterator so that 803 // ValueHandles can add and remove themselves from the list without 804 // breaking our iteration. This is not really an AssertingVH; we 805 // just have to give ValueHandleBase some kind. 806 for (ValueHandleBase Iterator(Assert, *Entry); Entry; Entry = Iterator.Next) { 807 Iterator.RemoveFromUseList(); 808 Iterator.AddToExistingUseListAfter(Entry); 809 assert(Entry->Next == &Iterator && "Loop invariant broken."); 810 811 switch (Entry->getKind()) { 812 case Assert: 813 // Asserting handle does not follow RAUW implicitly. 814 break; 815 case Tracking: 816 // Tracking goes to new value like a WeakVH. Note that this may make it 817 // something incompatible with its templated type. We don't want to have a 818 // virtual (or inline) interface to handle this though, so instead we make 819 // the TrackingVH accessors guarantee that a client never sees this value. 820 821 // FALLTHROUGH 822 case Weak: 823 // Weak goes to the new value, which will unlink it from Old's list. 824 Entry->operator=(New); 825 break; 826 case Callback: 827 // Forward to the subclass's implementation. 828 static_cast<CallbackVH*>(Entry)->allUsesReplacedWith(New); 829 break; 830 } 831 } 832 833 #ifndef NDEBUG 834 // If any new tracking or weak value handles were added while processing the 835 // list, then complain about it now. 836 if (Old->HasValueHandle) 837 for (Entry = pImpl->ValueHandles[Old]; Entry; Entry = Entry->Next) 838 switch (Entry->getKind()) { 839 case Tracking: 840 case Weak: 841 dbgs() << "After RAUW from " << *Old->getType() << " %" 842 << Old->getName() << " to " << *New->getType() << " %" 843 << New->getName() << "\n"; 844 llvm_unreachable("A tracking or weak value handle still pointed to the" 845 " old value!\n"); 846 default: 847 break; 848 } 849 #endif 850 } 851 852 // Pin the vtable to this file. 853 void CallbackVH::anchor() {} 854