Introduce MetadataBase, a base class for MDString and MDNode.
[oota-llvm.git] / lib / VMCore / Value.cpp
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/Constant.h"
15 #include "llvm/Constants.h"
16 #include "llvm/DerivedTypes.h"
17 #include "llvm/InstrTypes.h"
18 #include "llvm/Instructions.h"
19 #include "llvm/Operator.h"
20 #include "llvm/Module.h"
21 #include "llvm/MDNode.h"
22 #include "llvm/ValueSymbolTable.h"
23 #include "llvm/Support/Debug.h"
24 #include "llvm/Support/ErrorHandling.h"
25 #include "llvm/Support/LeakDetector.h"
26 #include "llvm/Support/ManagedStatic.h"
27 #include "llvm/Support/ValueHandle.h"
28 #include "llvm/Support/raw_ostream.h"
29 #include "llvm/System/RWMutex.h"
30 #include "llvm/System/Threading.h"
31 #include "llvm/ADT/DenseMap.h"
32 #include <algorithm>
33 using namespace llvm;
34
35 //===----------------------------------------------------------------------===//
36 //                                Value Class
37 //===----------------------------------------------------------------------===//
38
39 static inline const Type *checkType(const Type *Ty) {
40   assert(Ty && "Value defined with a null type: Error!");
41   return Ty;
42 }
43
44 Value::Value(const Type *ty, unsigned scid)
45   : SubclassID(scid), HasValueHandle(0), SubclassOptionalData(0),
46     SubclassData(0), VTy(checkType(ty)),
47     UseList(0), Name(0) {
48   if (isa<CallInst>(this) || isa<InvokeInst>(this))
49     assert((VTy->isFirstClassType() || VTy == Type::VoidTy ||
50             isa<OpaqueType>(ty) || VTy->getTypeID() == Type::StructTyID) &&
51            "invalid CallInst  type!");
52   else if (!isa<Constant>(this) && !isa<BasicBlock>(this))
53     assert((VTy->isFirstClassType() || VTy == Type::VoidTy ||
54            isa<OpaqueType>(ty)) &&
55            "Cannot create non-first-class values except for constants!");
56 }
57
58 Value::~Value() {
59   // Notify all ValueHandles (if present) that this value is going away.
60   if (HasValueHandle)
61     ValueHandleBase::ValueIsDeleted(this);
62   
63 #ifndef NDEBUG      // Only in -g mode...
64   // Check to make sure that there are no uses of this value that are still
65   // around when the value is destroyed.  If there are, then we have a dangling
66   // reference and something is wrong.  This code is here to print out what is
67   // still being referenced.  The value in question should be printed as
68   // a <badref>
69   //
70   if (!use_empty()) {
71     cerr << "While deleting: " << *VTy << " %" << getNameStr() << "\n";
72     for (use_iterator I = use_begin(), E = use_end(); I != E; ++I)
73       cerr << "Use still stuck around after Def is destroyed:"
74            << **I << "\n";
75   }
76 #endif
77   assert(use_empty() && "Uses remain when a value is destroyed!");
78
79   // If this value is named, destroy the name.  This should not be in a symtab
80   // at this point.
81   if (Name)
82     Name->Destroy();
83   
84   // There should be no uses of this object anymore, remove it.
85   LeakDetector::removeGarbageObject(this);
86 }
87
88 /// hasNUses - Return true if this Value has exactly N users.
89 ///
90 bool Value::hasNUses(unsigned N) const {
91   use_const_iterator UI = use_begin(), E = use_end();
92
93   for (; N; --N, ++UI)
94     if (UI == E) return false;  // Too few.
95   return UI == E;
96 }
97
98 /// hasNUsesOrMore - Return true if this value has N users or more.  This is
99 /// logically equivalent to getNumUses() >= N.
100 ///
101 bool Value::hasNUsesOrMore(unsigned N) const {
102   use_const_iterator UI = use_begin(), E = use_end();
103
104   for (; N; --N, ++UI)
105     if (UI == E) return false;  // Too few.
106
107   return true;
108 }
109
110 /// isUsedInBasicBlock - Return true if this value is used in the specified
111 /// basic block.
112 bool Value::isUsedInBasicBlock(const BasicBlock *BB) const {
113   for (use_const_iterator I = use_begin(), E = use_end(); I != E; ++I) {
114     const Instruction *User = dyn_cast<Instruction>(*I);
115     if (User && User->getParent() == BB)
116       return true;
117   }
118   return false;
119 }
120
121
122 /// getNumUses - This method computes the number of uses of this Value.  This
123 /// is a linear time operation.  Use hasOneUse or hasNUses to check for specific
124 /// values.
125 unsigned Value::getNumUses() const {
126   return (unsigned)std::distance(use_begin(), use_end());
127 }
128
129 static bool getSymTab(Value *V, ValueSymbolTable *&ST) {
130   ST = 0;
131   if (Instruction *I = dyn_cast<Instruction>(V)) {
132     if (BasicBlock *P = I->getParent())
133       if (Function *PP = P->getParent())
134         ST = &PP->getValueSymbolTable();
135   } else if (BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
136     if (Function *P = BB->getParent()) 
137       ST = &P->getValueSymbolTable();
138   } else if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
139     if (Module *P = GV->getParent()) 
140       ST = &P->getValueSymbolTable();
141   } else if (Argument *A = dyn_cast<Argument>(V)) {
142     if (Function *P = A->getParent()) 
143       ST = &P->getValueSymbolTable();
144   } else if (isa<MDString>(V))
145     return true;
146   else {
147     assert(isa<Constant>(V) && "Unknown value type!");
148     return true;  // no name is setable for this.
149   }
150   return false;
151 }
152
153 /// getNameStart - Return a pointer to a null terminated string for this name.
154 /// Note that names can have null characters within the string as well as at
155 /// their end.  This always returns a non-null pointer.
156 const char *Value::getNameStart() const {
157   if (Name == 0) return "";
158   return Name->getKeyData();
159 }
160
161 /// getNameLen - Return the length of the string, correctly handling nul
162 /// characters embedded into them.
163 unsigned Value::getNameLen() const {
164   return Name ? Name->getKeyLength() : 0;
165 }
166
167 /// isName - Return true if this value has the name specified by the provided
168 /// nul terminated string.
169 bool Value::isName(const char *N) const {
170   unsigned InLen = strlen(N);
171   return InLen == getNameLen() && memcmp(getNameStart(), N, InLen) == 0;
172 }
173
174
175 std::string Value::getNameStr() const {
176   if (Name == 0) return "";
177   return std::string(Name->getKeyData(),
178                      Name->getKeyData()+Name->getKeyLength());
179 }
180
181 void Value::setName(const std::string &name) {
182   setName(&name[0], name.size());
183 }
184
185 void Value::setName(const char *Name) {
186   setName(Name, Name ? strlen(Name) : 0);
187 }
188
189 void Value::setName(const char *NameStr, unsigned NameLen) {
190   if (NameLen == 0 && !hasName()) return;
191   assert(getType() != Type::VoidTy && "Cannot assign a name to void values!");
192   
193   // Get the symbol table to update for this object.
194   ValueSymbolTable *ST;
195   if (getSymTab(this, ST))
196     return;  // Cannot set a name on this value (e.g. constant).
197
198   if (!ST) { // No symbol table to update?  Just do the change.
199     if (NameLen == 0) {
200       // Free the name for this value.
201       Name->Destroy();
202       Name = 0;
203       return;
204     }
205     
206     if (Name) {
207       // Name isn't changing?
208       if (NameLen == Name->getKeyLength() &&
209           !memcmp(Name->getKeyData(), NameStr, NameLen))
210         return;
211       Name->Destroy();
212     }
213     
214     // NOTE: Could optimize for the case the name is shrinking to not deallocate
215     // then reallocated.
216       
217     // Create the new name.
218     Name = ValueName::Create(NameStr, NameStr+NameLen);
219     Name->setValue(this);
220     return;
221   }
222   
223   // NOTE: Could optimize for the case the name is shrinking to not deallocate
224   // then reallocated.
225   if (hasName()) {
226     // Name isn't changing?
227     if (NameLen == Name->getKeyLength() &&
228         !memcmp(Name->getKeyData(), NameStr, NameLen))
229       return;
230
231     // Remove old name.
232     ST->removeValueName(Name);
233     Name->Destroy();
234     Name = 0;
235
236     if (NameLen == 0)
237       return;
238   }
239
240   // Name is changing to something new.
241   Name = ST->createValueName(NameStr, NameLen, this);
242 }
243
244
245 /// takeName - transfer the name from V to this value, setting V's name to
246 /// empty.  It is an error to call V->takeName(V). 
247 void Value::takeName(Value *V) {
248   ValueSymbolTable *ST = 0;
249   // If this value has a name, drop it.
250   if (hasName()) {
251     // Get the symtab this is in.
252     if (getSymTab(this, ST)) {
253       // We can't set a name on this value, but we need to clear V's name if
254       // it has one.
255       if (V->hasName()) V->setName(0, 0);
256       return;  // Cannot set a name on this value (e.g. constant).
257     }
258     
259     // Remove old name.
260     if (ST)
261       ST->removeValueName(Name);
262     Name->Destroy();
263     Name = 0;
264   } 
265   
266   // Now we know that this has no name.
267   
268   // If V has no name either, we're done.
269   if (!V->hasName()) return;
270    
271   // Get this's symtab if we didn't before.
272   if (!ST) {
273     if (getSymTab(this, ST)) {
274       // Clear V's name.
275       V->setName(0, 0);
276       return;  // Cannot set a name on this value (e.g. constant).
277     }
278   }
279   
280   // Get V's ST, this should always succed, because V has a name.
281   ValueSymbolTable *VST;
282   bool Failure = getSymTab(V, VST);
283   assert(!Failure && "V has a name, so it should have a ST!"); Failure=Failure;
284   
285   // If these values are both in the same symtab, we can do this very fast.
286   // This works even if both values have no symtab yet.
287   if (ST == VST) {
288     // Take the name!
289     Name = V->Name;
290     V->Name = 0;
291     Name->setValue(this);
292     return;
293   }
294   
295   // Otherwise, things are slightly more complex.  Remove V's name from VST and
296   // then reinsert it into ST.
297   
298   if (VST)
299     VST->removeValueName(V->Name);
300   Name = V->Name;
301   V->Name = 0;
302   Name->setValue(this);
303   
304   if (ST)
305     ST->reinsertValue(this);
306 }
307
308
309 // uncheckedReplaceAllUsesWith - This is exactly the same as replaceAllUsesWith,
310 // except that it doesn't have all of the asserts.  The asserts fail because we
311 // are half-way done resolving types, which causes some types to exist as two
312 // different Type*'s at the same time.  This is a sledgehammer to work around
313 // this problem.
314 //
315 void Value::uncheckedReplaceAllUsesWith(Value *New) {
316   // Notify all ValueHandles (if present) that this value is going away.
317   if (HasValueHandle)
318     ValueHandleBase::ValueIsRAUWd(this, New);
319  
320   while (!use_empty()) {
321     Use &U = *UseList;
322     // Must handle Constants specially, we cannot call replaceUsesOfWith on a
323     // constant because they are uniqued.
324     if (Constant *C = dyn_cast<Constant>(U.getUser())) {
325       if (!isa<GlobalValue>(C)) {
326         C->replaceUsesOfWithOnConstant(this, New, &U);
327         continue;
328       }
329     }
330     
331     U.set(New);
332   }
333 }
334
335 void Value::replaceAllUsesWith(Value *New) {
336   assert(New && "Value::replaceAllUsesWith(<null>) is invalid!");
337   assert(New != this && "this->replaceAllUsesWith(this) is NOT valid!");
338   assert(New->getType() == getType() &&
339          "replaceAllUses of value with new value of different type!");
340
341   uncheckedReplaceAllUsesWith(New);
342 }
343
344 Value *Value::stripPointerCasts() {
345   if (!isa<PointerType>(getType()))
346     return this;
347   Value *V = this;
348   do {
349     if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) {
350       if (!GEP->hasAllZeroIndices())
351         return V;
352       V = GEP->getPointerOperand();
353     } else if (Operator::getOpcode(V) == Instruction::BitCast) {
354       V = cast<Operator>(V)->getOperand(0);
355     } else {
356       return V;
357     }
358     assert(isa<PointerType>(V->getType()) && "Unexpected operand type!");
359   } while (1);
360 }
361
362 Value *Value::getUnderlyingObject() {
363   if (!isa<PointerType>(getType()))
364     return this;
365   Value *V = this;
366   unsigned MaxLookup = 6;
367   do {
368     if (GEPOperator *GEP = dyn_cast<GEPOperator>(V)) {
369       V = GEP->getPointerOperand();
370     } else if (Operator::getOpcode(V) == Instruction::BitCast) {
371       V = cast<Operator>(V)->getOperand(0);
372     } else {
373       return V;
374     }
375     assert(isa<PointerType>(V->getType()) && "Unexpected operand type!");
376   } while (--MaxLookup);
377   return V;
378 }
379
380 /// DoPHITranslation - If this value is a PHI node with CurBB as its parent,
381 /// return the value in the PHI node corresponding to PredBB.  If not, return
382 /// ourself.  This is useful if you want to know the value something has in a
383 /// predecessor block.
384 Value *Value::DoPHITranslation(const BasicBlock *CurBB, 
385                                const BasicBlock *PredBB) {
386   PHINode *PN = dyn_cast<PHINode>(this);
387   if (PN && PN->getParent() == CurBB)
388     return PN->getIncomingValueForBlock(PredBB);
389   return this;
390 }
391
392 LLVMContext &Value::getContext() const { return VTy->getContext(); }
393
394 //===----------------------------------------------------------------------===//
395 //                             ValueHandleBase Class
396 //===----------------------------------------------------------------------===//
397
398 /// ValueHandles - This map keeps track of all of the value handles that are
399 /// watching a Value*.  The Value::HasValueHandle bit is used to know whether or
400 /// not a value has an entry in this map.
401 typedef DenseMap<Value*, ValueHandleBase*> ValueHandlesTy;
402 static ManagedStatic<ValueHandlesTy> ValueHandles;
403 static ManagedStatic<sys::SmartRWMutex<true> > ValueHandlesLock;
404
405 /// AddToExistingUseList - Add this ValueHandle to the use list for VP, where
406 /// List is known to point into the existing use list.
407 void ValueHandleBase::AddToExistingUseList(ValueHandleBase **List) {
408   assert(List && "Handle list is null?");
409   
410   // Splice ourselves into the list.
411   Next = *List;
412   *List = this;
413   setPrevPtr(List);
414   if (Next) {
415     Next->setPrevPtr(&Next);
416     assert(VP == Next->VP && "Added to wrong list?");
417   }
418 }
419
420 /// AddToUseList - Add this ValueHandle to the use list for VP.
421 void ValueHandleBase::AddToUseList() {
422   assert(VP && "Null pointer doesn't have a use list!");
423   if (VP->HasValueHandle) {
424     // If this value already has a ValueHandle, then it must be in the
425     // ValueHandles map already.
426     sys::SmartScopedReader<true> Reader(*ValueHandlesLock);
427     ValueHandleBase *&Entry = (*ValueHandles)[VP];
428     assert(Entry != 0 && "Value doesn't have any handles?");
429     AddToExistingUseList(&Entry);
430     return;
431   }
432   
433   // Ok, it doesn't have any handles yet, so we must insert it into the
434   // DenseMap.  However, doing this insertion could cause the DenseMap to
435   // reallocate itself, which would invalidate all of the PrevP pointers that
436   // point into the old table.  Handle this by checking for reallocation and
437   // updating the stale pointers only if needed.
438   sys::SmartScopedWriter<true> Writer(*ValueHandlesLock);
439   ValueHandlesTy &Handles = *ValueHandles;
440   const void *OldBucketPtr = Handles.getPointerIntoBucketsArray();
441   
442   ValueHandleBase *&Entry = Handles[VP];
443   assert(Entry == 0 && "Value really did already have handles?");
444   AddToExistingUseList(&Entry);
445   VP->HasValueHandle = true;
446   
447   // If reallocation didn't happen or if this was the first insertion, don't
448   // walk the table.
449   if (Handles.isPointerIntoBucketsArray(OldBucketPtr) || 
450       Handles.size() == 1) {
451     return;
452   }
453   
454   // Okay, reallocation did happen.  Fix the Prev Pointers.
455   for (ValueHandlesTy::iterator I = Handles.begin(), E = Handles.end();
456        I != E; ++I) {
457     assert(I->second && I->first == I->second->VP && "List invariant broken!");
458     I->second->setPrevPtr(&I->second);
459   }
460 }
461
462 /// RemoveFromUseList - Remove this ValueHandle from its current use list.
463 void ValueHandleBase::RemoveFromUseList() {
464   assert(VP && VP->HasValueHandle && "Pointer doesn't have a use list!");
465
466   // Unlink this from its use list.
467   ValueHandleBase **PrevPtr = getPrevPtr();
468   assert(*PrevPtr == this && "List invariant broken");
469   
470   *PrevPtr = Next;
471   if (Next) {
472     assert(Next->getPrevPtr() == &Next && "List invariant broken");
473     Next->setPrevPtr(PrevPtr);
474     return;
475   }
476   
477   // If the Next pointer was null, then it is possible that this was the last
478   // ValueHandle watching VP.  If so, delete its entry from the ValueHandles
479   // map.
480   sys::SmartScopedWriter<true> Writer(*ValueHandlesLock);
481   ValueHandlesTy &Handles = *ValueHandles;
482   if (Handles.isPointerIntoBucketsArray(PrevPtr)) {
483     Handles.erase(VP);
484     VP->HasValueHandle = false;
485   }
486 }
487
488
489 void ValueHandleBase::ValueIsDeleted(Value *V) {
490   assert(V->HasValueHandle && "Should only be called if ValueHandles present");
491
492   // Get the linked list base, which is guaranteed to exist since the
493   // HasValueHandle flag is set.
494   ValueHandlesLock->reader_acquire();
495   ValueHandleBase *Entry = (*ValueHandles)[V];
496   ValueHandlesLock->reader_release();
497   assert(Entry && "Value bit set but no entries exist");
498   
499   while (Entry) {
500     // Advance pointer to avoid invalidation.
501     ValueHandleBase *ThisNode = Entry;
502     Entry = Entry->Next;
503     
504     switch (ThisNode->getKind()) {
505     case Assert:
506 #ifndef NDEBUG      // Only in -g mode...
507       cerr << "While deleting: " << *V->getType() << " %" << V->getNameStr()
508            << "\n";
509 #endif
510       llvm_unreachable("An asserting value handle still pointed to this"
511                        " value!");
512     case Weak:
513       // Weak just goes to null, which will unlink it from the list.
514       ThisNode->operator=(0);
515       break;
516     case Callback:
517       // Forward to the subclass's implementation.
518       static_cast<CallbackVH*>(ThisNode)->deleted();
519       break;
520     }
521   }
522   
523   // All callbacks and weak references should be dropped by now.
524   assert(!V->HasValueHandle && "All references to V were not removed?");
525 }
526
527
528 void ValueHandleBase::ValueIsRAUWd(Value *Old, Value *New) {
529   assert(Old->HasValueHandle &&"Should only be called if ValueHandles present");
530   assert(Old != New && "Changing value into itself!");
531   
532   // Get the linked list base, which is guaranteed to exist since the
533   // HasValueHandle flag is set.
534   ValueHandlesLock->reader_acquire();
535   ValueHandleBase *Entry = (*ValueHandles)[Old];
536   ValueHandlesLock->reader_release();
537   assert(Entry && "Value bit set but no entries exist");
538   
539   while (Entry) {
540     // Advance pointer to avoid invalidation.
541     ValueHandleBase *ThisNode = Entry;
542     Entry = Entry->Next;
543     
544     switch (ThisNode->getKind()) {
545     case Assert:
546       // Asserting handle does not follow RAUW implicitly.
547       break;
548     case Weak:
549       // Weak goes to the new value, which will unlink it from Old's list.
550       ThisNode->operator=(New);
551       break;
552     case Callback:
553       // Forward to the subclass's implementation.
554       static_cast<CallbackVH*>(ThisNode)->allUsesReplacedWith(New);
555       break;
556     }
557   }
558 }
559
560 /// ~CallbackVH. Empty, but defined here to avoid emitting the vtable
561 /// more than once.
562 CallbackVH::~CallbackVH() {}
563
564
565 //===----------------------------------------------------------------------===//
566 //                                 User Class
567 //===----------------------------------------------------------------------===//
568
569 // replaceUsesOfWith - Replaces all references to the "From" definition with
570 // references to the "To" definition.
571 //
572 void User::replaceUsesOfWith(Value *From, Value *To) {
573   if (From == To) return;   // Duh what?
574
575   assert((!isa<Constant>(this) || isa<GlobalValue>(this)) &&
576          "Cannot call User::replaceUsesofWith on a constant!");
577
578   for (unsigned i = 0, E = getNumOperands(); i != E; ++i)
579     if (getOperand(i) == From) {  // Is This operand is pointing to oldval?
580       // The side effects of this setOperand call include linking to
581       // "To", adding "this" to the uses list of To, and
582       // most importantly, removing "this" from the use list of "From".
583       setOperand(i, To); // Fix it now...
584     }
585 }