1 //===-- Attributes.cpp - Implement AttributesList -------------------------===// 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 // \file 11 // \brief This file implements the Attribute, AttributeImpl, AttrBuilder, 12 // AttributeSetImpl, and AttributeSet classes. 13 // 14 //===----------------------------------------------------------------------===// 15 16 #include "llvm/IR/Attributes.h" 17 #include "AttributeImpl.h" 18 #include "LLVMContextImpl.h" 19 #include "llvm/ADT/STLExtras.h" 20 #include "llvm/ADT/StringExtras.h" 21 #include "llvm/IR/Type.h" 22 #include "llvm/Support/Atomic.h" 23 #include "llvm/Support/Debug.h" 24 #include "llvm/Support/ManagedStatic.h" 25 #include "llvm/Support/Mutex.h" 26 #include "llvm/Support/raw_ostream.h" 27 #include <algorithm> 28 using namespace llvm; 29 30 //===----------------------------------------------------------------------===// 31 // Attribute Construction Methods 32 //===----------------------------------------------------------------------===// 33 34 Attribute Attribute::get(LLVMContext &Context, Attribute::AttrKind Kind, 35 uint64_t Val) { 36 LLVMContextImpl *pImpl = Context.pImpl; 37 FoldingSetNodeID ID; 38 ID.AddInteger(Kind); 39 if (Val) ID.AddInteger(Val); 40 41 void *InsertPoint; 42 AttributeImpl *PA = pImpl->AttrsSet.FindNodeOrInsertPos(ID, InsertPoint); 43 44 if (!PA) { 45 // If we didn't find any existing attributes of the same shape then create a 46 // new one and insert it. 47 if (!Val) 48 PA = new EnumAttributeImpl(Kind); 49 else 50 PA = new IntAttributeImpl(Kind, Val); 51 pImpl->AttrsSet.InsertNode(PA, InsertPoint); 52 } 53 54 // Return the Attribute that we found or created. 55 return Attribute(PA); 56 } 57 58 Attribute Attribute::get(LLVMContext &Context, StringRef Kind, StringRef Val) { 59 LLVMContextImpl *pImpl = Context.pImpl; 60 FoldingSetNodeID ID; 61 ID.AddString(Kind); 62 if (!Val.empty()) ID.AddString(Val); 63 64 void *InsertPoint; 65 AttributeImpl *PA = pImpl->AttrsSet.FindNodeOrInsertPos(ID, InsertPoint); 66 67 if (!PA) { 68 // If we didn't find any existing attributes of the same shape then create a 69 // new one and insert it. 70 PA = new StringAttributeImpl(Kind, Val); 71 pImpl->AttrsSet.InsertNode(PA, InsertPoint); 72 } 73 74 // Return the Attribute that we found or created. 75 return Attribute(PA); 76 } 77 78 Attribute Attribute::getWithAlignment(LLVMContext &Context, uint64_t Align) { 79 assert(isPowerOf2_32(Align) && "Alignment must be a power of two."); 80 assert(Align <= 0x40000000 && "Alignment too large."); 81 return get(Context, Alignment, Align); 82 } 83 84 Attribute Attribute::getWithStackAlignment(LLVMContext &Context, 85 uint64_t Align) { 86 assert(isPowerOf2_32(Align) && "Alignment must be a power of two."); 87 assert(Align <= 0x100 && "Alignment too large."); 88 return get(Context, StackAlignment, Align); 89 } 90 91 Attribute Attribute::getWithDereferenceableBytes(LLVMContext &Context, 92 uint64_t Bytes) { 93 assert(Bytes && "Bytes must be non-zero."); 94 return get(Context, Dereferenceable, Bytes); 95 } 96 97 Attribute Attribute::getWithDereferenceableOrNullBytes(LLVMContext &Context, 98 uint64_t Bytes) { 99 assert(Bytes && "Bytes must be non-zero."); 100 return get(Context, DereferenceableOrNull, Bytes); 101 } 102 103 //===----------------------------------------------------------------------===// 104 // Attribute Accessor Methods 105 //===----------------------------------------------------------------------===// 106 107 bool Attribute::isEnumAttribute() const { 108 return pImpl && pImpl->isEnumAttribute(); 109 } 110 111 bool Attribute::isIntAttribute() const { 112 return pImpl && pImpl->isIntAttribute(); 113 } 114 115 bool Attribute::isStringAttribute() const { 116 return pImpl && pImpl->isStringAttribute(); 117 } 118 119 Attribute::AttrKind Attribute::getKindAsEnum() const { 120 if (!pImpl) return None; 121 assert((isEnumAttribute() || isIntAttribute()) && 122 "Invalid attribute type to get the kind as an enum!"); 123 return pImpl ? pImpl->getKindAsEnum() : None; 124 } 125 126 uint64_t Attribute::getValueAsInt() const { 127 if (!pImpl) return 0; 128 assert(isIntAttribute() && 129 "Expected the attribute to be an integer attribute!"); 130 return pImpl ? pImpl->getValueAsInt() : 0; 131 } 132 133 StringRef Attribute::getKindAsString() const { 134 if (!pImpl) return StringRef(); 135 assert(isStringAttribute() && 136 "Invalid attribute type to get the kind as a string!"); 137 return pImpl ? pImpl->getKindAsString() : StringRef(); 138 } 139 140 StringRef Attribute::getValueAsString() const { 141 if (!pImpl) return StringRef(); 142 assert(isStringAttribute() && 143 "Invalid attribute type to get the value as a string!"); 144 return pImpl ? pImpl->getValueAsString() : StringRef(); 145 } 146 147 bool Attribute::hasAttribute(AttrKind Kind) const { 148 return (pImpl && pImpl->hasAttribute(Kind)) || (!pImpl && Kind == None); 149 } 150 151 bool Attribute::hasAttribute(StringRef Kind) const { 152 if (!isStringAttribute()) return false; 153 return pImpl && pImpl->hasAttribute(Kind); 154 } 155 156 /// This returns the alignment field of an attribute as a byte alignment value. 157 unsigned Attribute::getAlignment() const { 158 assert(hasAttribute(Attribute::Alignment) && 159 "Trying to get alignment from non-alignment attribute!"); 160 return pImpl->getValueAsInt(); 161 } 162 163 /// This returns the stack alignment field of an attribute as a byte alignment 164 /// value. 165 unsigned Attribute::getStackAlignment() const { 166 assert(hasAttribute(Attribute::StackAlignment) && 167 "Trying to get alignment from non-alignment attribute!"); 168 return pImpl->getValueAsInt(); 169 } 170 171 /// This returns the number of dereferenceable bytes. 172 uint64_t Attribute::getDereferenceableBytes() const { 173 assert(hasAttribute(Attribute::Dereferenceable) && 174 "Trying to get dereferenceable bytes from " 175 "non-dereferenceable attribute!"); 176 return pImpl->getValueAsInt(); 177 } 178 179 uint64_t Attribute::getDereferenceableOrNullBytes() const { 180 assert(hasAttribute(Attribute::DereferenceableOrNull) && 181 "Trying to get dereferenceable bytes from " 182 "non-dereferenceable attribute!"); 183 return pImpl->getValueAsInt(); 184 } 185 186 std::string Attribute::getAsString(bool InAttrGrp) const { 187 if (!pImpl) return ""; 188 189 if (hasAttribute(Attribute::SanitizeAddress)) 190 return "sanitize_address"; 191 if (hasAttribute(Attribute::AlwaysInline)) 192 return "alwaysinline"; 193 if (hasAttribute(Attribute::ArgMemOnly)) 194 return "argmemonly"; 195 if (hasAttribute(Attribute::Builtin)) 196 return "builtin"; 197 if (hasAttribute(Attribute::ByVal)) 198 return "byval"; 199 if (hasAttribute(Attribute::Convergent)) 200 return "convergent"; 201 if (hasAttribute(Attribute::InAlloca)) 202 return "inalloca"; 203 if (hasAttribute(Attribute::InlineHint)) 204 return "inlinehint"; 205 if (hasAttribute(Attribute::InReg)) 206 return "inreg"; 207 if (hasAttribute(Attribute::JumpTable)) 208 return "jumptable"; 209 if (hasAttribute(Attribute::MinSize)) 210 return "minsize"; 211 if (hasAttribute(Attribute::Naked)) 212 return "naked"; 213 if (hasAttribute(Attribute::Nest)) 214 return "nest"; 215 if (hasAttribute(Attribute::NoAlias)) 216 return "noalias"; 217 if (hasAttribute(Attribute::NoBuiltin)) 218 return "nobuiltin"; 219 if (hasAttribute(Attribute::NoCapture)) 220 return "nocapture"; 221 if (hasAttribute(Attribute::NoDuplicate)) 222 return "noduplicate"; 223 if (hasAttribute(Attribute::NoImplicitFloat)) 224 return "noimplicitfloat"; 225 if (hasAttribute(Attribute::NoInline)) 226 return "noinline"; 227 if (hasAttribute(Attribute::NonLazyBind)) 228 return "nonlazybind"; 229 if (hasAttribute(Attribute::NonNull)) 230 return "nonnull"; 231 if (hasAttribute(Attribute::NoRedZone)) 232 return "noredzone"; 233 if (hasAttribute(Attribute::NoReturn)) 234 return "noreturn"; 235 if (hasAttribute(Attribute::NoRecurse)) 236 return "norecurse"; 237 if (hasAttribute(Attribute::NoUnwind)) 238 return "nounwind"; 239 if (hasAttribute(Attribute::OptimizeNone)) 240 return "optnone"; 241 if (hasAttribute(Attribute::OptimizeForSize)) 242 return "optsize"; 243 if (hasAttribute(Attribute::ReadNone)) 244 return "readnone"; 245 if (hasAttribute(Attribute::ReadOnly)) 246 return "readonly"; 247 if (hasAttribute(Attribute::Returned)) 248 return "returned"; 249 if (hasAttribute(Attribute::ReturnsTwice)) 250 return "returns_twice"; 251 if (hasAttribute(Attribute::SExt)) 252 return "signext"; 253 if (hasAttribute(Attribute::StackProtect)) 254 return "ssp"; 255 if (hasAttribute(Attribute::StackProtectReq)) 256 return "sspreq"; 257 if (hasAttribute(Attribute::StackProtectStrong)) 258 return "sspstrong"; 259 if (hasAttribute(Attribute::SafeStack)) 260 return "safestack"; 261 if (hasAttribute(Attribute::StructRet)) 262 return "sret"; 263 if (hasAttribute(Attribute::SanitizeThread)) 264 return "sanitize_thread"; 265 if (hasAttribute(Attribute::SanitizeMemory)) 266 return "sanitize_memory"; 267 if (hasAttribute(Attribute::UWTable)) 268 return "uwtable"; 269 if (hasAttribute(Attribute::ZExt)) 270 return "zeroext"; 271 if (hasAttribute(Attribute::Cold)) 272 return "cold"; 273 274 // FIXME: These should be output like this: 275 // 276 // align=4 277 // alignstack=8 278 // 279 if (hasAttribute(Attribute::Alignment)) { 280 std::string Result; 281 Result += "align"; 282 Result += (InAttrGrp) ? "=" : " "; 283 Result += utostr(getValueAsInt()); 284 return Result; 285 } 286 287 auto AttrWithBytesToString = [&](const char *Name) { 288 std::string Result; 289 Result += Name; 290 if (InAttrGrp) { 291 Result += "="; 292 Result += utostr(getValueAsInt()); 293 } else { 294 Result += "("; 295 Result += utostr(getValueAsInt()); 296 Result += ")"; 297 } 298 return Result; 299 }; 300 301 if (hasAttribute(Attribute::StackAlignment)) 302 return AttrWithBytesToString("alignstack"); 303 304 if (hasAttribute(Attribute::Dereferenceable)) 305 return AttrWithBytesToString("dereferenceable"); 306 307 if (hasAttribute(Attribute::DereferenceableOrNull)) 308 return AttrWithBytesToString("dereferenceable_or_null"); 309 310 // Convert target-dependent attributes to strings of the form: 311 // 312 // "kind" 313 // "kind" = "value" 314 // 315 if (isStringAttribute()) { 316 std::string Result; 317 Result += (Twine('"') + getKindAsString() + Twine('"')).str(); 318 319 StringRef Val = pImpl->getValueAsString(); 320 if (Val.empty()) return Result; 321 322 Result += ("=\"" + Val + Twine('"')).str(); 323 return Result; 324 } 325 326 llvm_unreachable("Unknown attribute"); 327 } 328 329 bool Attribute::operator<(Attribute A) const { 330 if (!pImpl && !A.pImpl) return false; 331 if (!pImpl) return true; 332 if (!A.pImpl) return false; 333 return *pImpl < *A.pImpl; 334 } 335 336 //===----------------------------------------------------------------------===// 337 // AttributeImpl Definition 338 //===----------------------------------------------------------------------===// 339 340 // Pin the vtables to this file. 341 AttributeImpl::~AttributeImpl() {} 342 void EnumAttributeImpl::anchor() {} 343 void IntAttributeImpl::anchor() {} 344 void StringAttributeImpl::anchor() {} 345 346 bool AttributeImpl::hasAttribute(Attribute::AttrKind A) const { 347 if (isStringAttribute()) return false; 348 return getKindAsEnum() == A; 349 } 350 351 bool AttributeImpl::hasAttribute(StringRef Kind) const { 352 if (!isStringAttribute()) return false; 353 return getKindAsString() == Kind; 354 } 355 356 Attribute::AttrKind AttributeImpl::getKindAsEnum() const { 357 assert(isEnumAttribute() || isIntAttribute()); 358 return static_cast<const EnumAttributeImpl *>(this)->getEnumKind(); 359 } 360 361 uint64_t AttributeImpl::getValueAsInt() const { 362 assert(isIntAttribute()); 363 return static_cast<const IntAttributeImpl *>(this)->getValue(); 364 } 365 366 StringRef AttributeImpl::getKindAsString() const { 367 assert(isStringAttribute()); 368 return static_cast<const StringAttributeImpl *>(this)->getStringKind(); 369 } 370 371 StringRef AttributeImpl::getValueAsString() const { 372 assert(isStringAttribute()); 373 return static_cast<const StringAttributeImpl *>(this)->getStringValue(); 374 } 375 376 bool AttributeImpl::operator<(const AttributeImpl &AI) const { 377 // This sorts the attributes with Attribute::AttrKinds coming first (sorted 378 // relative to their enum value) and then strings. 379 if (isEnumAttribute()) { 380 if (AI.isEnumAttribute()) return getKindAsEnum() < AI.getKindAsEnum(); 381 if (AI.isIntAttribute()) return true; 382 if (AI.isStringAttribute()) return true; 383 } 384 385 if (isIntAttribute()) { 386 if (AI.isEnumAttribute()) return false; 387 if (AI.isIntAttribute()) return getValueAsInt() < AI.getValueAsInt(); 388 if (AI.isStringAttribute()) return true; 389 } 390 391 if (AI.isEnumAttribute()) return false; 392 if (AI.isIntAttribute()) return false; 393 if (getKindAsString() == AI.getKindAsString()) 394 return getValueAsString() < AI.getValueAsString(); 395 return getKindAsString() < AI.getKindAsString(); 396 } 397 398 uint64_t AttributeImpl::getAttrMask(Attribute::AttrKind Val) { 399 // FIXME: Remove this. 400 switch (Val) { 401 case Attribute::EndAttrKinds: 402 llvm_unreachable("Synthetic enumerators which should never get here"); 403 404 case Attribute::None: return 0; 405 case Attribute::ZExt: return 1 << 0; 406 case Attribute::SExt: return 1 << 1; 407 case Attribute::NoReturn: return 1 << 2; 408 case Attribute::InReg: return 1 << 3; 409 case Attribute::StructRet: return 1 << 4; 410 case Attribute::NoUnwind: return 1 << 5; 411 case Attribute::NoAlias: return 1 << 6; 412 case Attribute::ByVal: return 1 << 7; 413 case Attribute::Nest: return 1 << 8; 414 case Attribute::ReadNone: return 1 << 9; 415 case Attribute::ReadOnly: return 1 << 10; 416 case Attribute::NoInline: return 1 << 11; 417 case Attribute::AlwaysInline: return 1 << 12; 418 case Attribute::OptimizeForSize: return 1 << 13; 419 case Attribute::StackProtect: return 1 << 14; 420 case Attribute::StackProtectReq: return 1 << 15; 421 case Attribute::Alignment: return 31 << 16; 422 case Attribute::NoCapture: return 1 << 21; 423 case Attribute::NoRedZone: return 1 << 22; 424 case Attribute::NoImplicitFloat: return 1 << 23; 425 case Attribute::Naked: return 1 << 24; 426 case Attribute::InlineHint: return 1 << 25; 427 case Attribute::StackAlignment: return 7 << 26; 428 case Attribute::ReturnsTwice: return 1 << 29; 429 case Attribute::UWTable: return 1 << 30; 430 case Attribute::NonLazyBind: return 1U << 31; 431 case Attribute::SanitizeAddress: return 1ULL << 32; 432 case Attribute::MinSize: return 1ULL << 33; 433 case Attribute::NoDuplicate: return 1ULL << 34; 434 case Attribute::StackProtectStrong: return 1ULL << 35; 435 case Attribute::SanitizeThread: return 1ULL << 36; 436 case Attribute::SanitizeMemory: return 1ULL << 37; 437 case Attribute::NoBuiltin: return 1ULL << 38; 438 case Attribute::Returned: return 1ULL << 39; 439 case Attribute::Cold: return 1ULL << 40; 440 case Attribute::Builtin: return 1ULL << 41; 441 case Attribute::OptimizeNone: return 1ULL << 42; 442 case Attribute::InAlloca: return 1ULL << 43; 443 case Attribute::NonNull: return 1ULL << 44; 444 case Attribute::JumpTable: return 1ULL << 45; 445 case Attribute::Convergent: return 1ULL << 46; 446 case Attribute::SafeStack: return 1ULL << 47; 447 case Attribute::NoRecurse: return 1ULL << 48; 448 case Attribute::Dereferenceable: 449 llvm_unreachable("dereferenceable attribute not supported in raw format"); 450 break; 451 case Attribute::DereferenceableOrNull: 452 llvm_unreachable("dereferenceable_or_null attribute not supported in raw " 453 "format"); 454 break; 455 case Attribute::ArgMemOnly: 456 llvm_unreachable("argmemonly attribute not supported in raw format"); 457 break; 458 } 459 llvm_unreachable("Unsupported attribute type"); 460 } 461 462 //===----------------------------------------------------------------------===// 463 // AttributeSetNode Definition 464 //===----------------------------------------------------------------------===// 465 466 AttributeSetNode *AttributeSetNode::get(LLVMContext &C, 467 ArrayRef<Attribute> Attrs) { 468 if (Attrs.empty()) 469 return nullptr; 470 471 // Otherwise, build a key to look up the existing attributes. 472 LLVMContextImpl *pImpl = C.pImpl; 473 FoldingSetNodeID ID; 474 475 SmallVector<Attribute, 8> SortedAttrs(Attrs.begin(), Attrs.end()); 476 array_pod_sort(SortedAttrs.begin(), SortedAttrs.end()); 477 478 for (SmallVectorImpl<Attribute>::iterator I = SortedAttrs.begin(), 479 E = SortedAttrs.end(); I != E; ++I) 480 I->Profile(ID); 481 482 void *InsertPoint; 483 AttributeSetNode *PA = 484 pImpl->AttrsSetNodes.FindNodeOrInsertPos(ID, InsertPoint); 485 486 // If we didn't find any existing attributes of the same shape then create a 487 // new one and insert it. 488 if (!PA) { 489 // Coallocate entries after the AttributeSetNode itself. 490 void *Mem = ::operator new(totalSizeToAlloc<Attribute>(SortedAttrs.size())); 491 PA = new (Mem) AttributeSetNode(SortedAttrs); 492 pImpl->AttrsSetNodes.InsertNode(PA, InsertPoint); 493 } 494 495 // Return the AttributesListNode that we found or created. 496 return PA; 497 } 498 499 bool AttributeSetNode::hasAttribute(Attribute::AttrKind Kind) const { 500 for (iterator I = begin(), E = end(); I != E; ++I) 501 if (I->hasAttribute(Kind)) 502 return true; 503 return false; 504 } 505 506 bool AttributeSetNode::hasAttribute(StringRef Kind) const { 507 for (iterator I = begin(), E = end(); I != E; ++I) 508 if (I->hasAttribute(Kind)) 509 return true; 510 return false; 511 } 512 513 Attribute AttributeSetNode::getAttribute(Attribute::AttrKind Kind) const { 514 for (iterator I = begin(), E = end(); I != E; ++I) 515 if (I->hasAttribute(Kind)) 516 return *I; 517 return Attribute(); 518 } 519 520 Attribute AttributeSetNode::getAttribute(StringRef Kind) const { 521 for (iterator I = begin(), E = end(); I != E; ++I) 522 if (I->hasAttribute(Kind)) 523 return *I; 524 return Attribute(); 525 } 526 527 unsigned AttributeSetNode::getAlignment() const { 528 for (iterator I = begin(), E = end(); I != E; ++I) 529 if (I->hasAttribute(Attribute::Alignment)) 530 return I->getAlignment(); 531 return 0; 532 } 533 534 unsigned AttributeSetNode::getStackAlignment() const { 535 for (iterator I = begin(), E = end(); I != E; ++I) 536 if (I->hasAttribute(Attribute::StackAlignment)) 537 return I->getStackAlignment(); 538 return 0; 539 } 540 541 uint64_t AttributeSetNode::getDereferenceableBytes() const { 542 for (iterator I = begin(), E = end(); I != E; ++I) 543 if (I->hasAttribute(Attribute::Dereferenceable)) 544 return I->getDereferenceableBytes(); 545 return 0; 546 } 547 548 uint64_t AttributeSetNode::getDereferenceableOrNullBytes() const { 549 for (iterator I = begin(), E = end(); I != E; ++I) 550 if (I->hasAttribute(Attribute::DereferenceableOrNull)) 551 return I->getDereferenceableOrNullBytes(); 552 return 0; 553 } 554 555 std::string AttributeSetNode::getAsString(bool InAttrGrp) const { 556 std::string Str; 557 for (iterator I = begin(), E = end(); I != E; ++I) { 558 if (I != begin()) 559 Str += ' '; 560 Str += I->getAsString(InAttrGrp); 561 } 562 return Str; 563 } 564 565 //===----------------------------------------------------------------------===// 566 // AttributeSetImpl Definition 567 //===----------------------------------------------------------------------===// 568 569 uint64_t AttributeSetImpl::Raw(unsigned Index) const { 570 for (unsigned I = 0, E = getNumAttributes(); I != E; ++I) { 571 if (getSlotIndex(I) != Index) continue; 572 const AttributeSetNode *ASN = getSlotNode(I); 573 uint64_t Mask = 0; 574 575 for (AttributeSetNode::iterator II = ASN->begin(), 576 IE = ASN->end(); II != IE; ++II) { 577 Attribute Attr = *II; 578 579 // This cannot handle string attributes. 580 if (Attr.isStringAttribute()) continue; 581 582 Attribute::AttrKind Kind = Attr.getKindAsEnum(); 583 584 if (Kind == Attribute::Alignment) 585 Mask |= (Log2_32(ASN->getAlignment()) + 1) << 16; 586 else if (Kind == Attribute::StackAlignment) 587 Mask |= (Log2_32(ASN->getStackAlignment()) + 1) << 26; 588 else if (Kind == Attribute::Dereferenceable) 589 llvm_unreachable("dereferenceable not supported in bit mask"); 590 else 591 Mask |= AttributeImpl::getAttrMask(Kind); 592 } 593 594 return Mask; 595 } 596 597 return 0; 598 } 599 600 void AttributeSetImpl::dump() const { 601 AttributeSet(const_cast<AttributeSetImpl *>(this)).dump(); 602 } 603 604 //===----------------------------------------------------------------------===// 605 // AttributeSet Construction and Mutation Methods 606 //===----------------------------------------------------------------------===// 607 608 AttributeSet 609 AttributeSet::getImpl(LLVMContext &C, 610 ArrayRef<std::pair<unsigned, AttributeSetNode*> > Attrs) { 611 LLVMContextImpl *pImpl = C.pImpl; 612 FoldingSetNodeID ID; 613 AttributeSetImpl::Profile(ID, Attrs); 614 615 void *InsertPoint; 616 AttributeSetImpl *PA = pImpl->AttrsLists.FindNodeOrInsertPos(ID, InsertPoint); 617 618 // If we didn't find any existing attributes of the same shape then 619 // create a new one and insert it. 620 if (!PA) { 621 // Coallocate entries after the AttributeSetImpl itself. 622 void *Mem = ::operator new( 623 AttributeSetImpl::totalSizeToAlloc<IndexAttrPair>(Attrs.size())); 624 PA = new (Mem) AttributeSetImpl(C, Attrs); 625 pImpl->AttrsLists.InsertNode(PA, InsertPoint); 626 } 627 628 // Return the AttributesList that we found or created. 629 return AttributeSet(PA); 630 } 631 632 AttributeSet AttributeSet::get(LLVMContext &C, 633 ArrayRef<std::pair<unsigned, Attribute> > Attrs){ 634 // If there are no attributes then return a null AttributesList pointer. 635 if (Attrs.empty()) 636 return AttributeSet(); 637 638 #ifndef NDEBUG 639 for (unsigned i = 0, e = Attrs.size(); i != e; ++i) { 640 assert((!i || Attrs[i-1].first <= Attrs[i].first) && 641 "Misordered Attributes list!"); 642 assert(!Attrs[i].second.hasAttribute(Attribute::None) && 643 "Pointless attribute!"); 644 } 645 #endif 646 647 // Create a vector if (unsigned, AttributeSetNode*) pairs from the attributes 648 // list. 649 SmallVector<std::pair<unsigned, AttributeSetNode*>, 8> AttrPairVec; 650 for (ArrayRef<std::pair<unsigned, Attribute> >::iterator I = Attrs.begin(), 651 E = Attrs.end(); I != E; ) { 652 unsigned Index = I->first; 653 SmallVector<Attribute, 4> AttrVec; 654 while (I != E && I->first == Index) { 655 AttrVec.push_back(I->second); 656 ++I; 657 } 658 659 AttrPairVec.push_back(std::make_pair(Index, 660 AttributeSetNode::get(C, AttrVec))); 661 } 662 663 return getImpl(C, AttrPairVec); 664 } 665 666 AttributeSet AttributeSet::get(LLVMContext &C, 667 ArrayRef<std::pair<unsigned, 668 AttributeSetNode*> > Attrs) { 669 // If there are no attributes then return a null AttributesList pointer. 670 if (Attrs.empty()) 671 return AttributeSet(); 672 673 return getImpl(C, Attrs); 674 } 675 676 AttributeSet AttributeSet::get(LLVMContext &C, unsigned Index, 677 const AttrBuilder &B) { 678 if (!B.hasAttributes()) 679 return AttributeSet(); 680 681 // Add target-independent attributes. 682 SmallVector<std::pair<unsigned, Attribute>, 8> Attrs; 683 for (Attribute::AttrKind Kind = Attribute::None; 684 Kind != Attribute::EndAttrKinds; Kind = Attribute::AttrKind(Kind + 1)) { 685 if (!B.contains(Kind)) 686 continue; 687 688 if (Kind == Attribute::Alignment) 689 Attrs.push_back(std::make_pair(Index, Attribute:: 690 getWithAlignment(C, B.getAlignment()))); 691 else if (Kind == Attribute::StackAlignment) 692 Attrs.push_back(std::make_pair(Index, Attribute:: 693 getWithStackAlignment(C, B.getStackAlignment()))); 694 else if (Kind == Attribute::Dereferenceable) 695 Attrs.push_back(std::make_pair(Index, 696 Attribute::getWithDereferenceableBytes(C, 697 B.getDereferenceableBytes()))); 698 else if (Kind == Attribute::DereferenceableOrNull) 699 Attrs.push_back( 700 std::make_pair(Index, Attribute::getWithDereferenceableOrNullBytes( 701 C, B.getDereferenceableOrNullBytes()))); 702 else 703 Attrs.push_back(std::make_pair(Index, Attribute::get(C, Kind))); 704 } 705 706 // Add target-dependent (string) attributes. 707 for (const AttrBuilder::td_type &TDA : B.td_attrs()) 708 Attrs.push_back( 709 std::make_pair(Index, Attribute::get(C, TDA.first, TDA.second))); 710 711 return get(C, Attrs); 712 } 713 714 AttributeSet AttributeSet::get(LLVMContext &C, unsigned Index, 715 ArrayRef<Attribute::AttrKind> Kind) { 716 SmallVector<std::pair<unsigned, Attribute>, 8> Attrs; 717 for (ArrayRef<Attribute::AttrKind>::iterator I = Kind.begin(), 718 E = Kind.end(); I != E; ++I) 719 Attrs.push_back(std::make_pair(Index, Attribute::get(C, *I))); 720 return get(C, Attrs); 721 } 722 723 AttributeSet AttributeSet::get(LLVMContext &C, ArrayRef<AttributeSet> Attrs) { 724 if (Attrs.empty()) return AttributeSet(); 725 if (Attrs.size() == 1) return Attrs[0]; 726 727 SmallVector<std::pair<unsigned, AttributeSetNode*>, 8> AttrNodeVec; 728 AttributeSetImpl *A0 = Attrs[0].pImpl; 729 if (A0) 730 AttrNodeVec.append(A0->getNode(0), A0->getNode(A0->getNumAttributes())); 731 // Copy all attributes from Attrs into AttrNodeVec while keeping AttrNodeVec 732 // ordered by index. Because we know that each list in Attrs is ordered by 733 // index we only need to merge each successive list in rather than doing a 734 // full sort. 735 for (unsigned I = 1, E = Attrs.size(); I != E; ++I) { 736 AttributeSetImpl *AS = Attrs[I].pImpl; 737 if (!AS) continue; 738 SmallVector<std::pair<unsigned, AttributeSetNode *>, 8>::iterator 739 ANVI = AttrNodeVec.begin(), ANVE; 740 for (const IndexAttrPair *AI = AS->getNode(0), 741 *AE = AS->getNode(AS->getNumAttributes()); 742 AI != AE; ++AI) { 743 ANVE = AttrNodeVec.end(); 744 while (ANVI != ANVE && ANVI->first <= AI->first) 745 ++ANVI; 746 ANVI = AttrNodeVec.insert(ANVI, *AI) + 1; 747 } 748 } 749 750 return getImpl(C, AttrNodeVec); 751 } 752 753 AttributeSet AttributeSet::addAttribute(LLVMContext &C, unsigned Index, 754 Attribute::AttrKind Attr) const { 755 if (hasAttribute(Index, Attr)) return *this; 756 return addAttributes(C, Index, AttributeSet::get(C, Index, Attr)); 757 } 758 759 AttributeSet AttributeSet::addAttribute(LLVMContext &C, unsigned Index, 760 StringRef Kind) const { 761 llvm::AttrBuilder B; 762 B.addAttribute(Kind); 763 return addAttributes(C, Index, AttributeSet::get(C, Index, B)); 764 } 765 766 AttributeSet AttributeSet::addAttribute(LLVMContext &C, unsigned Index, 767 StringRef Kind, StringRef Value) const { 768 llvm::AttrBuilder B; 769 B.addAttribute(Kind, Value); 770 return addAttributes(C, Index, AttributeSet::get(C, Index, B)); 771 } 772 773 AttributeSet AttributeSet::addAttribute(LLVMContext &C, 774 ArrayRef<unsigned> Indices, 775 Attribute A) const { 776 unsigned I = 0, E = pImpl ? pImpl->getNumAttributes() : 0; 777 auto IdxI = Indices.begin(), IdxE = Indices.end(); 778 SmallVector<AttributeSet, 4> AttrSet; 779 780 while (I != E && IdxI != IdxE) { 781 if (getSlotIndex(I) < *IdxI) 782 AttrSet.emplace_back(getSlotAttributes(I++)); 783 else if (getSlotIndex(I) > *IdxI) 784 AttrSet.emplace_back(AttributeSet::get(C, std::make_pair(*IdxI++, A))); 785 else { 786 AttrBuilder B(getSlotAttributes(I), *IdxI); 787 B.addAttribute(A); 788 AttrSet.emplace_back(AttributeSet::get(C, *IdxI, B)); 789 ++I; 790 ++IdxI; 791 } 792 } 793 794 while (I != E) 795 AttrSet.emplace_back(getSlotAttributes(I++)); 796 797 while (IdxI != IdxE) 798 AttrSet.emplace_back(AttributeSet::get(C, std::make_pair(*IdxI++, A))); 799 800 return get(C, AttrSet); 801 } 802 803 AttributeSet AttributeSet::addAttributes(LLVMContext &C, unsigned Index, 804 AttributeSet Attrs) const { 805 if (!pImpl) return Attrs; 806 if (!Attrs.pImpl) return *this; 807 808 #ifndef NDEBUG 809 // FIXME it is not obvious how this should work for alignment. For now, say 810 // we can't change a known alignment. 811 unsigned OldAlign = getParamAlignment(Index); 812 unsigned NewAlign = Attrs.getParamAlignment(Index); 813 assert((!OldAlign || !NewAlign || OldAlign == NewAlign) && 814 "Attempt to change alignment!"); 815 #endif 816 817 // Add the attribute slots before the one we're trying to add. 818 SmallVector<AttributeSet, 4> AttrSet; 819 uint64_t NumAttrs = pImpl->getNumAttributes(); 820 AttributeSet AS; 821 uint64_t LastIndex = 0; 822 for (unsigned I = 0, E = NumAttrs; I != E; ++I) { 823 if (getSlotIndex(I) >= Index) { 824 if (getSlotIndex(I) == Index) AS = getSlotAttributes(LastIndex++); 825 break; 826 } 827 LastIndex = I + 1; 828 AttrSet.push_back(getSlotAttributes(I)); 829 } 830 831 // Now add the attribute into the correct slot. There may already be an 832 // AttributeSet there. 833 AttrBuilder B(AS, Index); 834 835 for (unsigned I = 0, E = Attrs.pImpl->getNumAttributes(); I != E; ++I) 836 if (Attrs.getSlotIndex(I) == Index) { 837 for (AttributeSetImpl::iterator II = Attrs.pImpl->begin(I), 838 IE = Attrs.pImpl->end(I); II != IE; ++II) 839 B.addAttribute(*II); 840 break; 841 } 842 843 AttrSet.push_back(AttributeSet::get(C, Index, B)); 844 845 // Add the remaining attribute slots. 846 for (unsigned I = LastIndex, E = NumAttrs; I < E; ++I) 847 AttrSet.push_back(getSlotAttributes(I)); 848 849 return get(C, AttrSet); 850 } 851 852 AttributeSet AttributeSet::removeAttribute(LLVMContext &C, unsigned Index, 853 Attribute::AttrKind Attr) const { 854 if (!hasAttribute(Index, Attr)) return *this; 855 return removeAttributes(C, Index, AttributeSet::get(C, Index, Attr)); 856 } 857 858 AttributeSet AttributeSet::removeAttributes(LLVMContext &C, unsigned Index, 859 AttributeSet Attrs) const { 860 if (!pImpl) return AttributeSet(); 861 if (!Attrs.pImpl) return *this; 862 863 // FIXME it is not obvious how this should work for alignment. 864 // For now, say we can't pass in alignment, which no current use does. 865 assert(!Attrs.hasAttribute(Index, Attribute::Alignment) && 866 "Attempt to change alignment!"); 867 868 // Add the attribute slots before the one we're trying to add. 869 SmallVector<AttributeSet, 4> AttrSet; 870 uint64_t NumAttrs = pImpl->getNumAttributes(); 871 AttributeSet AS; 872 uint64_t LastIndex = 0; 873 for (unsigned I = 0, E = NumAttrs; I != E; ++I) { 874 if (getSlotIndex(I) >= Index) { 875 if (getSlotIndex(I) == Index) AS = getSlotAttributes(LastIndex++); 876 break; 877 } 878 LastIndex = I + 1; 879 AttrSet.push_back(getSlotAttributes(I)); 880 } 881 882 // Now remove the attribute from the correct slot. There may already be an 883 // AttributeSet there. 884 AttrBuilder B(AS, Index); 885 886 for (unsigned I = 0, E = Attrs.pImpl->getNumAttributes(); I != E; ++I) 887 if (Attrs.getSlotIndex(I) == Index) { 888 B.removeAttributes(Attrs.pImpl->getSlotAttributes(I), Index); 889 break; 890 } 891 892 AttrSet.push_back(AttributeSet::get(C, Index, B)); 893 894 // Add the remaining attribute slots. 895 for (unsigned I = LastIndex, E = NumAttrs; I < E; ++I) 896 AttrSet.push_back(getSlotAttributes(I)); 897 898 return get(C, AttrSet); 899 } 900 901 AttributeSet AttributeSet::removeAttributes(LLVMContext &C, unsigned Index, 902 const AttrBuilder &Attrs) const { 903 if (!pImpl) return AttributeSet(); 904 905 // FIXME it is not obvious how this should work for alignment. 906 // For now, say we can't pass in alignment, which no current use does. 907 assert(!Attrs.hasAlignmentAttr() && "Attempt to change alignment!"); 908 909 // Add the attribute slots before the one we're trying to add. 910 SmallVector<AttributeSet, 4> AttrSet; 911 uint64_t NumAttrs = pImpl->getNumAttributes(); 912 AttributeSet AS; 913 uint64_t LastIndex = 0; 914 for (unsigned I = 0, E = NumAttrs; I != E; ++I) { 915 if (getSlotIndex(I) >= Index) { 916 if (getSlotIndex(I) == Index) AS = getSlotAttributes(LastIndex++); 917 break; 918 } 919 LastIndex = I + 1; 920 AttrSet.push_back(getSlotAttributes(I)); 921 } 922 923 // Now remove the attribute from the correct slot. There may already be an 924 // AttributeSet there. 925 AttrBuilder B(AS, Index); 926 B.remove(Attrs); 927 928 AttrSet.push_back(AttributeSet::get(C, Index, B)); 929 930 // Add the remaining attribute slots. 931 for (unsigned I = LastIndex, E = NumAttrs; I < E; ++I) 932 AttrSet.push_back(getSlotAttributes(I)); 933 934 return get(C, AttrSet); 935 } 936 937 AttributeSet AttributeSet::addDereferenceableAttr(LLVMContext &C, unsigned Index, 938 uint64_t Bytes) const { 939 llvm::AttrBuilder B; 940 B.addDereferenceableAttr(Bytes); 941 return addAttributes(C, Index, AttributeSet::get(C, Index, B)); 942 } 943 944 AttributeSet AttributeSet::addDereferenceableOrNullAttr(LLVMContext &C, 945 unsigned Index, 946 uint64_t Bytes) const { 947 llvm::AttrBuilder B; 948 B.addDereferenceableOrNullAttr(Bytes); 949 return addAttributes(C, Index, AttributeSet::get(C, Index, B)); 950 } 951 952 //===----------------------------------------------------------------------===// 953 // AttributeSet Accessor Methods 954 //===----------------------------------------------------------------------===// 955 956 LLVMContext &AttributeSet::getContext() const { 957 return pImpl->getContext(); 958 } 959 960 AttributeSet AttributeSet::getParamAttributes(unsigned Index) const { 961 return pImpl && hasAttributes(Index) ? 962 AttributeSet::get(pImpl->getContext(), 963 ArrayRef<std::pair<unsigned, AttributeSetNode*> >( 964 std::make_pair(Index, getAttributes(Index)))) : 965 AttributeSet(); 966 } 967 968 AttributeSet AttributeSet::getRetAttributes() const { 969 return pImpl && hasAttributes(ReturnIndex) ? 970 AttributeSet::get(pImpl->getContext(), 971 ArrayRef<std::pair<unsigned, AttributeSetNode*> >( 972 std::make_pair(ReturnIndex, 973 getAttributes(ReturnIndex)))) : 974 AttributeSet(); 975 } 976 977 AttributeSet AttributeSet::getFnAttributes() const { 978 return pImpl && hasAttributes(FunctionIndex) ? 979 AttributeSet::get(pImpl->getContext(), 980 ArrayRef<std::pair<unsigned, AttributeSetNode*> >( 981 std::make_pair(FunctionIndex, 982 getAttributes(FunctionIndex)))) : 983 AttributeSet(); 984 } 985 986 bool AttributeSet::hasAttribute(unsigned Index, Attribute::AttrKind Kind) const{ 987 AttributeSetNode *ASN = getAttributes(Index); 988 return ASN ? ASN->hasAttribute(Kind) : false; 989 } 990 991 bool AttributeSet::hasAttribute(unsigned Index, StringRef Kind) const { 992 AttributeSetNode *ASN = getAttributes(Index); 993 return ASN ? ASN->hasAttribute(Kind) : false; 994 } 995 996 bool AttributeSet::hasAttributes(unsigned Index) const { 997 AttributeSetNode *ASN = getAttributes(Index); 998 return ASN ? ASN->hasAttributes() : false; 999 } 1000 1001 /// \brief Return true if the specified attribute is set for at least one 1002 /// parameter or for the return value. 1003 bool AttributeSet::hasAttrSomewhere(Attribute::AttrKind Attr) const { 1004 if (!pImpl) return false; 1005 1006 for (unsigned I = 0, E = pImpl->getNumAttributes(); I != E; ++I) 1007 for (AttributeSetImpl::iterator II = pImpl->begin(I), 1008 IE = pImpl->end(I); II != IE; ++II) 1009 if (II->hasAttribute(Attr)) 1010 return true; 1011 1012 return false; 1013 } 1014 1015 Attribute AttributeSet::getAttribute(unsigned Index, 1016 Attribute::AttrKind Kind) const { 1017 AttributeSetNode *ASN = getAttributes(Index); 1018 return ASN ? ASN->getAttribute(Kind) : Attribute(); 1019 } 1020 1021 Attribute AttributeSet::getAttribute(unsigned Index, 1022 StringRef Kind) const { 1023 AttributeSetNode *ASN = getAttributes(Index); 1024 return ASN ? ASN->getAttribute(Kind) : Attribute(); 1025 } 1026 1027 unsigned AttributeSet::getParamAlignment(unsigned Index) const { 1028 AttributeSetNode *ASN = getAttributes(Index); 1029 return ASN ? ASN->getAlignment() : 0; 1030 } 1031 1032 unsigned AttributeSet::getStackAlignment(unsigned Index) const { 1033 AttributeSetNode *ASN = getAttributes(Index); 1034 return ASN ? ASN->getStackAlignment() : 0; 1035 } 1036 1037 uint64_t AttributeSet::getDereferenceableBytes(unsigned Index) const { 1038 AttributeSetNode *ASN = getAttributes(Index); 1039 return ASN ? ASN->getDereferenceableBytes() : 0; 1040 } 1041 1042 uint64_t AttributeSet::getDereferenceableOrNullBytes(unsigned Index) const { 1043 AttributeSetNode *ASN = getAttributes(Index); 1044 return ASN ? ASN->getDereferenceableOrNullBytes() : 0; 1045 } 1046 1047 std::string AttributeSet::getAsString(unsigned Index, 1048 bool InAttrGrp) const { 1049 AttributeSetNode *ASN = getAttributes(Index); 1050 return ASN ? ASN->getAsString(InAttrGrp) : std::string(""); 1051 } 1052 1053 /// \brief The attributes for the specified index are returned. 1054 AttributeSetNode *AttributeSet::getAttributes(unsigned Index) const { 1055 if (!pImpl) return nullptr; 1056 1057 // Loop through to find the attribute node we want. 1058 for (unsigned I = 0, E = pImpl->getNumAttributes(); I != E; ++I) 1059 if (pImpl->getSlotIndex(I) == Index) 1060 return pImpl->getSlotNode(I); 1061 1062 return nullptr; 1063 } 1064 1065 AttributeSet::iterator AttributeSet::begin(unsigned Slot) const { 1066 if (!pImpl) 1067 return ArrayRef<Attribute>().begin(); 1068 return pImpl->begin(Slot); 1069 } 1070 1071 AttributeSet::iterator AttributeSet::end(unsigned Slot) const { 1072 if (!pImpl) 1073 return ArrayRef<Attribute>().end(); 1074 return pImpl->end(Slot); 1075 } 1076 1077 //===----------------------------------------------------------------------===// 1078 // AttributeSet Introspection Methods 1079 //===----------------------------------------------------------------------===// 1080 1081 /// \brief Return the number of slots used in this attribute list. This is the 1082 /// number of arguments that have an attribute set on them (including the 1083 /// function itself). 1084 unsigned AttributeSet::getNumSlots() const { 1085 return pImpl ? pImpl->getNumAttributes() : 0; 1086 } 1087 1088 unsigned AttributeSet::getSlotIndex(unsigned Slot) const { 1089 assert(pImpl && Slot < pImpl->getNumAttributes() && 1090 "Slot # out of range!"); 1091 return pImpl->getSlotIndex(Slot); 1092 } 1093 1094 AttributeSet AttributeSet::getSlotAttributes(unsigned Slot) const { 1095 assert(pImpl && Slot < pImpl->getNumAttributes() && 1096 "Slot # out of range!"); 1097 return pImpl->getSlotAttributes(Slot); 1098 } 1099 1100 uint64_t AttributeSet::Raw(unsigned Index) const { 1101 // FIXME: Remove this. 1102 return pImpl ? pImpl->Raw(Index) : 0; 1103 } 1104 1105 void AttributeSet::dump() const { 1106 dbgs() << "PAL[\n"; 1107 1108 for (unsigned i = 0, e = getNumSlots(); i < e; ++i) { 1109 uint64_t Index = getSlotIndex(i); 1110 dbgs() << " { "; 1111 if (Index == ~0U) 1112 dbgs() << "~0U"; 1113 else 1114 dbgs() << Index; 1115 dbgs() << " => " << getAsString(Index) << " }\n"; 1116 } 1117 1118 dbgs() << "]\n"; 1119 } 1120 1121 //===----------------------------------------------------------------------===// 1122 // AttrBuilder Method Implementations 1123 //===----------------------------------------------------------------------===// 1124 1125 AttrBuilder::AttrBuilder(AttributeSet AS, unsigned Index) 1126 : Attrs(0), Alignment(0), StackAlignment(0), DerefBytes(0), 1127 DerefOrNullBytes(0) { 1128 AttributeSetImpl *pImpl = AS.pImpl; 1129 if (!pImpl) return; 1130 1131 for (unsigned I = 0, E = pImpl->getNumAttributes(); I != E; ++I) { 1132 if (pImpl->getSlotIndex(I) != Index) continue; 1133 1134 for (AttributeSetImpl::iterator II = pImpl->begin(I), 1135 IE = pImpl->end(I); II != IE; ++II) 1136 addAttribute(*II); 1137 1138 break; 1139 } 1140 } 1141 1142 void AttrBuilder::clear() { 1143 Attrs.reset(); 1144 TargetDepAttrs.clear(); 1145 Alignment = StackAlignment = DerefBytes = DerefOrNullBytes = 0; 1146 } 1147 1148 AttrBuilder &AttrBuilder::addAttribute(Attribute::AttrKind Val) { 1149 assert((unsigned)Val < Attribute::EndAttrKinds && "Attribute out of range!"); 1150 assert(Val != Attribute::Alignment && Val != Attribute::StackAlignment && 1151 Val != Attribute::Dereferenceable && 1152 "Adding integer attribute without adding a value!"); 1153 Attrs[Val] = true; 1154 return *this; 1155 } 1156 1157 AttrBuilder &AttrBuilder::addAttribute(Attribute Attr) { 1158 if (Attr.isStringAttribute()) { 1159 addAttribute(Attr.getKindAsString(), Attr.getValueAsString()); 1160 return *this; 1161 } 1162 1163 Attribute::AttrKind Kind = Attr.getKindAsEnum(); 1164 Attrs[Kind] = true; 1165 1166 if (Kind == Attribute::Alignment) 1167 Alignment = Attr.getAlignment(); 1168 else if (Kind == Attribute::StackAlignment) 1169 StackAlignment = Attr.getStackAlignment(); 1170 else if (Kind == Attribute::Dereferenceable) 1171 DerefBytes = Attr.getDereferenceableBytes(); 1172 else if (Kind == Attribute::DereferenceableOrNull) 1173 DerefOrNullBytes = Attr.getDereferenceableOrNullBytes(); 1174 return *this; 1175 } 1176 1177 AttrBuilder &AttrBuilder::addAttribute(StringRef A, StringRef V) { 1178 TargetDepAttrs[A] = V; 1179 return *this; 1180 } 1181 1182 AttrBuilder &AttrBuilder::removeAttribute(Attribute::AttrKind Val) { 1183 assert((unsigned)Val < Attribute::EndAttrKinds && "Attribute out of range!"); 1184 Attrs[Val] = false; 1185 1186 if (Val == Attribute::Alignment) 1187 Alignment = 0; 1188 else if (Val == Attribute::StackAlignment) 1189 StackAlignment = 0; 1190 else if (Val == Attribute::Dereferenceable) 1191 DerefBytes = 0; 1192 else if (Val == Attribute::DereferenceableOrNull) 1193 DerefOrNullBytes = 0; 1194 1195 return *this; 1196 } 1197 1198 AttrBuilder &AttrBuilder::removeAttributes(AttributeSet A, uint64_t Index) { 1199 unsigned Slot = ~0U; 1200 for (unsigned I = 0, E = A.getNumSlots(); I != E; ++I) 1201 if (A.getSlotIndex(I) == Index) { 1202 Slot = I; 1203 break; 1204 } 1205 1206 assert(Slot != ~0U && "Couldn't find index in AttributeSet!"); 1207 1208 for (AttributeSet::iterator I = A.begin(Slot), E = A.end(Slot); I != E; ++I) { 1209 Attribute Attr = *I; 1210 if (Attr.isEnumAttribute() || Attr.isIntAttribute()) { 1211 Attribute::AttrKind Kind = I->getKindAsEnum(); 1212 Attrs[Kind] = false; 1213 1214 if (Kind == Attribute::Alignment) 1215 Alignment = 0; 1216 else if (Kind == Attribute::StackAlignment) 1217 StackAlignment = 0; 1218 else if (Kind == Attribute::Dereferenceable) 1219 DerefBytes = 0; 1220 else if (Kind == Attribute::DereferenceableOrNull) 1221 DerefOrNullBytes = 0; 1222 } else { 1223 assert(Attr.isStringAttribute() && "Invalid attribute type!"); 1224 std::map<std::string, std::string>::iterator 1225 Iter = TargetDepAttrs.find(Attr.getKindAsString()); 1226 if (Iter != TargetDepAttrs.end()) 1227 TargetDepAttrs.erase(Iter); 1228 } 1229 } 1230 1231 return *this; 1232 } 1233 1234 AttrBuilder &AttrBuilder::removeAttribute(StringRef A) { 1235 std::map<std::string, std::string>::iterator I = TargetDepAttrs.find(A); 1236 if (I != TargetDepAttrs.end()) 1237 TargetDepAttrs.erase(I); 1238 return *this; 1239 } 1240 1241 AttrBuilder &AttrBuilder::addAlignmentAttr(unsigned Align) { 1242 if (Align == 0) return *this; 1243 1244 assert(isPowerOf2_32(Align) && "Alignment must be a power of two."); 1245 assert(Align <= 0x40000000 && "Alignment too large."); 1246 1247 Attrs[Attribute::Alignment] = true; 1248 Alignment = Align; 1249 return *this; 1250 } 1251 1252 AttrBuilder &AttrBuilder::addStackAlignmentAttr(unsigned Align) { 1253 // Default alignment, allow the target to define how to align it. 1254 if (Align == 0) return *this; 1255 1256 assert(isPowerOf2_32(Align) && "Alignment must be a power of two."); 1257 assert(Align <= 0x100 && "Alignment too large."); 1258 1259 Attrs[Attribute::StackAlignment] = true; 1260 StackAlignment = Align; 1261 return *this; 1262 } 1263 1264 AttrBuilder &AttrBuilder::addDereferenceableAttr(uint64_t Bytes) { 1265 if (Bytes == 0) return *this; 1266 1267 Attrs[Attribute::Dereferenceable] = true; 1268 DerefBytes = Bytes; 1269 return *this; 1270 } 1271 1272 AttrBuilder &AttrBuilder::addDereferenceableOrNullAttr(uint64_t Bytes) { 1273 if (Bytes == 0) 1274 return *this; 1275 1276 Attrs[Attribute::DereferenceableOrNull] = true; 1277 DerefOrNullBytes = Bytes; 1278 return *this; 1279 } 1280 1281 AttrBuilder &AttrBuilder::merge(const AttrBuilder &B) { 1282 // FIXME: What if both have alignments, but they don't match?! 1283 if (!Alignment) 1284 Alignment = B.Alignment; 1285 1286 if (!StackAlignment) 1287 StackAlignment = B.StackAlignment; 1288 1289 if (!DerefBytes) 1290 DerefBytes = B.DerefBytes; 1291 1292 if (!DerefOrNullBytes) 1293 DerefOrNullBytes = B.DerefOrNullBytes; 1294 1295 Attrs |= B.Attrs; 1296 1297 for (auto I : B.td_attrs()) 1298 TargetDepAttrs[I.first] = I.second; 1299 1300 return *this; 1301 } 1302 1303 AttrBuilder &AttrBuilder::remove(const AttrBuilder &B) { 1304 // FIXME: What if both have alignments, but they don't match?! 1305 if (B.Alignment) 1306 Alignment = 0; 1307 1308 if (B.StackAlignment) 1309 StackAlignment = 0; 1310 1311 if (B.DerefBytes) 1312 DerefBytes = 0; 1313 1314 if (B.DerefOrNullBytes) 1315 DerefOrNullBytes = 0; 1316 1317 Attrs &= ~B.Attrs; 1318 1319 for (auto I : B.td_attrs()) 1320 TargetDepAttrs.erase(I.first); 1321 1322 return *this; 1323 } 1324 1325 bool AttrBuilder::overlaps(const AttrBuilder &B) const { 1326 // First check if any of the target independent attributes overlap. 1327 if ((Attrs & B.Attrs).any()) 1328 return true; 1329 1330 // Then check if any target dependent ones do. 1331 for (auto I : td_attrs()) 1332 if (B.contains(I.first)) 1333 return true; 1334 1335 return false; 1336 } 1337 1338 bool AttrBuilder::contains(StringRef A) const { 1339 return TargetDepAttrs.find(A) != TargetDepAttrs.end(); 1340 } 1341 1342 bool AttrBuilder::hasAttributes() const { 1343 return !Attrs.none() || !TargetDepAttrs.empty(); 1344 } 1345 1346 bool AttrBuilder::hasAttributes(AttributeSet A, uint64_t Index) const { 1347 unsigned Slot = ~0U; 1348 for (unsigned I = 0, E = A.getNumSlots(); I != E; ++I) 1349 if (A.getSlotIndex(I) == Index) { 1350 Slot = I; 1351 break; 1352 } 1353 1354 assert(Slot != ~0U && "Couldn't find the index!"); 1355 1356 for (AttributeSet::iterator I = A.begin(Slot), E = A.end(Slot); 1357 I != E; ++I) { 1358 Attribute Attr = *I; 1359 if (Attr.isEnumAttribute() || Attr.isIntAttribute()) { 1360 if (Attrs[I->getKindAsEnum()]) 1361 return true; 1362 } else { 1363 assert(Attr.isStringAttribute() && "Invalid attribute kind!"); 1364 return TargetDepAttrs.find(Attr.getKindAsString())!=TargetDepAttrs.end(); 1365 } 1366 } 1367 1368 return false; 1369 } 1370 1371 bool AttrBuilder::hasAlignmentAttr() const { 1372 return Alignment != 0; 1373 } 1374 1375 bool AttrBuilder::operator==(const AttrBuilder &B) { 1376 if (Attrs != B.Attrs) 1377 return false; 1378 1379 for (td_const_iterator I = TargetDepAttrs.begin(), 1380 E = TargetDepAttrs.end(); I != E; ++I) 1381 if (B.TargetDepAttrs.find(I->first) == B.TargetDepAttrs.end()) 1382 return false; 1383 1384 return Alignment == B.Alignment && StackAlignment == B.StackAlignment && 1385 DerefBytes == B.DerefBytes; 1386 } 1387 1388 AttrBuilder &AttrBuilder::addRawValue(uint64_t Val) { 1389 // FIXME: Remove this in 4.0. 1390 if (!Val) return *this; 1391 1392 for (Attribute::AttrKind I = Attribute::None; I != Attribute::EndAttrKinds; 1393 I = Attribute::AttrKind(I + 1)) { 1394 if (I == Attribute::Dereferenceable || 1395 I == Attribute::DereferenceableOrNull || 1396 I == Attribute::ArgMemOnly) 1397 continue; 1398 if (uint64_t A = (Val & AttributeImpl::getAttrMask(I))) { 1399 Attrs[I] = true; 1400 1401 if (I == Attribute::Alignment) 1402 Alignment = 1ULL << ((A >> 16) - 1); 1403 else if (I == Attribute::StackAlignment) 1404 StackAlignment = 1ULL << ((A >> 26)-1); 1405 } 1406 } 1407 1408 return *this; 1409 } 1410 1411 //===----------------------------------------------------------------------===// 1412 // AttributeFuncs Function Defintions 1413 //===----------------------------------------------------------------------===// 1414 1415 /// \brief Which attributes cannot be applied to a type. 1416 AttrBuilder AttributeFuncs::typeIncompatible(Type *Ty) { 1417 AttrBuilder Incompatible; 1418 1419 if (!Ty->isIntegerTy()) 1420 // Attribute that only apply to integers. 1421 Incompatible.addAttribute(Attribute::SExt) 1422 .addAttribute(Attribute::ZExt); 1423 1424 if (!Ty->isPointerTy()) 1425 // Attribute that only apply to pointers. 1426 Incompatible.addAttribute(Attribute::ByVal) 1427 .addAttribute(Attribute::Nest) 1428 .addAttribute(Attribute::NoAlias) 1429 .addAttribute(Attribute::NoCapture) 1430 .addAttribute(Attribute::NonNull) 1431 .addDereferenceableAttr(1) // the int here is ignored 1432 .addDereferenceableOrNullAttr(1) // the int here is ignored 1433 .addAttribute(Attribute::ReadNone) 1434 .addAttribute(Attribute::ReadOnly) 1435 .addAttribute(Attribute::StructRet) 1436 .addAttribute(Attribute::InAlloca); 1437 1438 return Incompatible; 1439 } 1440