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