src/rustllvm/RustWrapper.cpp

   1 // Copyright 2013 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 #include "rustllvm.h"
  12 #include "llvm/IR/DiagnosticInfo.h"
  13 #include "llvm/IR/DiagnosticPrinter.h"
  14 #include "llvm/IR/Instructions.h"
  15 #include "llvm/Object/Archive.h"
  16 #include "llvm/Object/ObjectFile.h"
  17
  18 #include "llvm/IR/CallSite.h"
  19
  20 //===----------------------------------------------------------------------===
  21 //
  22 // This file defines alternate interfaces to core functions that are more
  23 // readily callable by Rust's FFI.
  24 //
  25 //===----------------------------------------------------------------------===
  26
  27 using namespace llvm;
  28 using namespace llvm::sys;
  29 using namespace llvm::object;
  30
  31 // LLVMAtomicOrdering is already an enum - don't create another
  32 // one.
  33 static AtomicOrdering fromRust(LLVMAtomicOrdering Ordering) {
  34   switch (Ordering) {
  35   case LLVMAtomicOrderingNotAtomic:
  36     return AtomicOrdering::NotAtomic;
  37   case LLVMAtomicOrderingUnordered:
  38     return AtomicOrdering::Unordered;
  39   case LLVMAtomicOrderingMonotonic:
  40     return AtomicOrdering::Monotonic;
  41   case LLVMAtomicOrderingAcquire:
  42     return AtomicOrdering::Acquire;
  43   case LLVMAtomicOrderingRelease:
  44     return AtomicOrdering::Release;
  45   case LLVMAtomicOrderingAcquireRelease:
  46     return AtomicOrdering::AcquireRelease;
  47   case LLVMAtomicOrderingSequentiallyConsistent:
  48     return AtomicOrdering::SequentiallyConsistent;
  49   }
  50
  51   llvm_unreachable("Invalid LLVMAtomicOrdering value!");
  52 }
  53
  54 static char *LastError;
  55
  56 extern "C" LLVMMemoryBufferRef
  57 LLVMRustCreateMemoryBufferWithContentsOfFile(const char *Path) {
  58   ErrorOr<std::unique_ptr<MemoryBuffer>> BufOr =
  59       MemoryBuffer::getFile(Path, -1, false);
  60   if (!BufOr) {
  61     LLVMRustSetLastError(BufOr.getError().message().c_str());
  62     return nullptr;
  63   }
  64   return wrap(BufOr.get().release());
  65 }
  66
  67 extern "C" char *LLVMRustGetLastError(void) {
  68   char *Ret = LastError;
  69   LastError = nullptr;
  70   return Ret;
  71 }
  72
  73 void LLVMRustSetLastError(const char *Err) {
  74   free((void *)LastError);
  75   LastError = strdup(Err);
  76 }
  77
  78 extern "C" void LLVMRustSetNormalizedTarget(LLVMModuleRef M,
  79                                             const char *Triple) {
  80   unwrap(M)->setTargetTriple(Triple::normalize(Triple));
  81 }
  82
  83 extern "C" void LLVMRustPrintPassTimings() {
  84   raw_fd_ostream OS(2, false); // stderr.
  85   TimerGroup::printAll(OS);
  86 }
  87
  88 extern "C" LLVMValueRef LLVMRustGetNamedValue(LLVMModuleRef M,
  89                                               const char *Name) {
  90   return wrap(unwrap(M)->getNamedValue(Name));
  91 }
  92
  93 extern "C" LLVMValueRef LLVMRustGetOrInsertFunction(LLVMModuleRef M,
  94                                                     const char *Name,
  95                                                     LLVMTypeRef FunctionTy) {
  96   return wrap(
  97       unwrap(M)->getOrInsertFunction(Name, unwrap<FunctionType>(FunctionTy)));
  98 }
  99
 100 extern "C" LLVMValueRef
 101 LLVMRustGetOrInsertGlobal(LLVMModuleRef M, const char *Name, LLVMTypeRef Ty) {
 102   return wrap(unwrap(M)->getOrInsertGlobal(Name, unwrap(Ty)));
 103 }
 104
 105 extern "C" LLVMTypeRef LLVMRustMetadataTypeInContext(LLVMContextRef C) {
 106   return wrap(Type::getMetadataTy(*unwrap(C)));
 107 }
 108
 109 static Attribute::AttrKind fromRust(LLVMRustAttribute Kind) {
 110   switch (Kind) {
 111   case AlwaysInline:
 112     return Attribute::AlwaysInline;
 113   case ByVal:
 114     return Attribute::ByVal;
 115   case Cold:
 116     return Attribute::Cold;
 117   case InlineHint:
 118     return Attribute::InlineHint;
 119   case MinSize:
 120     return Attribute::MinSize;
 121   case Naked:
 122     return Attribute::Naked;
 123   case NoAlias:
 124     return Attribute::NoAlias;
 125   case NoCapture:
 126     return Attribute::NoCapture;
 127   case NoInline:
 128     return Attribute::NoInline;
 129   case NonNull:
 130     return Attribute::NonNull;
 131   case NoRedZone:
 132     return Attribute::NoRedZone;
 133   case NoReturn:
 134     return Attribute::NoReturn;
 135   case NoUnwind:
 136     return Attribute::NoUnwind;
 137   case OptimizeForSize:
 138     return Attribute::OptimizeForSize;
 139   case ReadOnly:
 140     return Attribute::ReadOnly;
 141   case SExt:
 142     return Attribute::SExt;
 143   case StructRet:
 144     return Attribute::StructRet;
 145   case UWTable:
 146     return Attribute::UWTable;
 147   case ZExt:
 148     return Attribute::ZExt;
 149   case InReg:
 150     return Attribute::InReg;
 151   case SanitizeThread:
 152     return Attribute::SanitizeThread;
 153   case SanitizeAddress:
 154     return Attribute::SanitizeAddress;
 155   case SanitizeMemory:
 156     return Attribute::SanitizeMemory;
 157   }
 158   llvm_unreachable("bad AttributeKind");
 159 }
 160
 161 extern "C" void LLVMRustAddCallSiteAttribute(LLVMValueRef Instr, unsigned Index,
 162                                              LLVMRustAttribute RustAttr) {
 163   CallSite Call = CallSite(unwrap<Instruction>(Instr));
 164   Attribute Attr = Attribute::get(Call->getContext(), fromRust(RustAttr));
 165   AttrBuilder B(Attr);
 166   Call.setAttributes(Call.getAttributes().addAttributes(
 167       Call->getContext(), Index,
 168       AttributeSet::get(Call->getContext(), Index, B)));
 169 }
 170
 171 extern "C" void LLVMRustAddDereferenceableCallSiteAttr(LLVMValueRef Instr,
 172                                                        unsigned Index,
 173                                                        uint64_t Bytes) {
 174   CallSite Call = CallSite(unwrap<Instruction>(Instr));
 175   AttrBuilder B;
 176   B.addDereferenceableAttr(Bytes);
 177   Call.setAttributes(Call.getAttributes().addAttributes(
 178       Call->getContext(), Index,
 179       AttributeSet::get(Call->getContext(), Index, B)));
 180 }
 181
 182 extern "C" void LLVMRustAddFunctionAttribute(LLVMValueRef Fn, unsigned Index,
 183                                              LLVMRustAttribute RustAttr) {
 184   Function *A = unwrap<Function>(Fn);
 185   Attribute Attr = Attribute::get(A->getContext(), fromRust(RustAttr));
 186   AttrBuilder B(Attr);
 187   A->addAttributes(Index, AttributeSet::get(A->getContext(), Index, B));
 188 }
 189
 190 extern "C" void LLVMRustAddDereferenceableAttr(LLVMValueRef Fn, unsigned Index,
 191                                                uint64_t Bytes) {
 192   Function *A = unwrap<Function>(Fn);
 193   AttrBuilder B;
 194   B.addDereferenceableAttr(Bytes);
 195   A->addAttributes(Index, AttributeSet::get(A->getContext(), Index, B));
 196 }
 197
 198 extern "C" void LLVMRustAddFunctionAttrStringValue(LLVMValueRef Fn,
 199                                                    unsigned Index,
 200                                                    const char *Name,
 201                                                    const char *Value) {
 202   Function *F = unwrap<Function>(Fn);
 203   AttrBuilder B;
 204   B.addAttribute(Name, Value);
 205   F->addAttributes(Index, AttributeSet::get(F->getContext(), Index, B));
 206 }
 207
 208 extern "C" void LLVMRustRemoveFunctionAttributes(LLVMValueRef Fn,
 209                                                  unsigned Index,
 210                                                  LLVMRustAttribute RustAttr) {
 211   Function *F = unwrap<Function>(Fn);
 212   const AttributeSet PAL = F->getAttributes();
 213   Attribute Attr = Attribute::get(F->getContext(), fromRust(RustAttr));
 214   AttrBuilder B(Attr);
 215   const AttributeSet PALNew = PAL.removeAttributes(
 216       F->getContext(), Index, AttributeSet::get(F->getContext(), Index, B));
 217   F->setAttributes(PALNew);
 218 }
 219
 220 // enable fpmath flag UnsafeAlgebra
 221 extern "C" void LLVMRustSetHasUnsafeAlgebra(LLVMValueRef V) {
 222   if (auto I = dyn_cast<Instruction>(unwrap<Value>(V))) {
 223     I->setHasUnsafeAlgebra(true);
 224   }
 225 }
 226
 227 extern "C" LLVMValueRef
 228 LLVMRustBuildAtomicLoad(LLVMBuilderRef B, LLVMValueRef Source, const char *Name,
 229                         LLVMAtomicOrdering Order, unsigned Alignment) {
 230   LoadInst *LI = new LoadInst(unwrap(Source), 0);
 231   LI->setAtomic(fromRust(Order));
 232   LI->setAlignment(Alignment);
 233   return wrap(unwrap(B)->Insert(LI, Name));
 234 }
 235
 236 extern "C" LLVMValueRef LLVMRustBuildAtomicStore(LLVMBuilderRef B,
 237                                                  LLVMValueRef V,
 238                                                  LLVMValueRef Target,
 239                                                  LLVMAtomicOrdering Order,
 240                                                  unsigned Alignment) {
 241   StoreInst *SI = new StoreInst(unwrap(V), unwrap(Target));
 242   SI->setAtomic(fromRust(Order));
 243   SI->setAlignment(Alignment);
 244   return wrap(unwrap(B)->Insert(SI));
 245 }
 246
 247 extern "C" LLVMValueRef
 248 LLVMRustBuildAtomicCmpXchg(LLVMBuilderRef B, LLVMValueRef Target,
 249                            LLVMValueRef Old, LLVMValueRef Source,
 250                            LLVMAtomicOrdering Order,
 251                            LLVMAtomicOrdering FailureOrder, LLVMBool Weak) {
 252   AtomicCmpXchgInst *ACXI = unwrap(B)->CreateAtomicCmpXchg(
 253       unwrap(Target), unwrap(Old), unwrap(Source), fromRust(Order),
 254       fromRust(FailureOrder));
 255   ACXI->setWeak(Weak);
 256   return wrap(ACXI);
 257 }
 258
 259 enum class LLVMRustSynchronizationScope {
 260   Other,
 261   SingleThread,
 262   CrossThread,
 263 };
 264
 265 static SynchronizationScope fromRust(LLVMRustSynchronizationScope Scope) {
 266   switch (Scope) {
 267   case LLVMRustSynchronizationScope::SingleThread:
 268     return SingleThread;
 269   case LLVMRustSynchronizationScope::CrossThread:
 270     return CrossThread;
 271   default:
 272     llvm_unreachable("bad SynchronizationScope.");
 273   }
 274 }
 275
 276 extern "C" LLVMValueRef
 277 LLVMRustBuildAtomicFence(LLVMBuilderRef B, LLVMAtomicOrdering Order,
 278                          LLVMRustSynchronizationScope Scope) {
 279   return wrap(unwrap(B)->CreateFence(fromRust(Order), fromRust(Scope)));
 280 }
 281
 282 extern "C" void LLVMRustSetDebug(int Enabled) {
 283 #ifndef NDEBUG
 284   DebugFlag = Enabled;
 285 #endif
 286 }
 287
 288 enum class LLVMRustAsmDialect {
 289   Other,
 290   Att,
 291   Intel,
 292 };
 293
 294 static InlineAsm::AsmDialect fromRust(LLVMRustAsmDialect Dialect) {
 295   switch (Dialect) {
 296   case LLVMRustAsmDialect::Att:
 297     return InlineAsm::AD_ATT;
 298   case LLVMRustAsmDialect::Intel:
 299     return InlineAsm::AD_Intel;
 300   default:
 301     llvm_unreachable("bad AsmDialect.");
 302   }
 303 }
 304
 305 extern "C" LLVMValueRef LLVMRustInlineAsm(LLVMTypeRef Ty, char *AsmString,
 306                                           char *Constraints,
 307                                           LLVMBool HasSideEffects,
 308                                           LLVMBool IsAlignStack,
 309                                           LLVMRustAsmDialect Dialect) {
 310   return wrap(InlineAsm::get(unwrap<FunctionType>(Ty), AsmString, Constraints,
 311                              HasSideEffects, IsAlignStack, fromRust(Dialect)));
 312 }
 313
 314 typedef DIBuilder *LLVMRustDIBuilderRef;
 315
 316 typedef struct LLVMOpaqueMetadata *LLVMRustMetadataRef;
 317
 318 namespace llvm {
 319 DEFINE_ISA_CONVERSION_FUNCTIONS(Metadata, LLVMRustMetadataRef)
 320
 321 inline Metadata **unwrap(LLVMRustMetadataRef *Vals) {
 322   return reinterpret_cast<Metadata **>(Vals);
 323 }
 324 }
 325
 326 template <typename DIT> DIT *unwrapDIPtr(LLVMRustMetadataRef Ref) {
 327   return (DIT *)(Ref ? unwrap<MDNode>(Ref) : nullptr);
 328 }
 329
 330 #define DIDescriptor DIScope
 331 #define DIArray DINodeArray
 332 #define unwrapDI unwrapDIPtr
 333
 334 // These values **must** match debuginfo::DIFlags! They also *happen*
 335 // to match LLVM, but that isn't required as we do giant sets of
 336 // matching below. The value shouldn't be directly passed to LLVM.
 337 enum class LLVMRustDIFlags : uint32_t {
 338   FlagZero = 0,
 339   FlagPrivate = 1,
 340   FlagProtected = 2,
 341   FlagPublic = 3,
 342   FlagFwdDecl = (1 << 2),
 343   FlagAppleBlock = (1 << 3),
 344   FlagBlockByrefStruct = (1 << 4),
 345   FlagVirtual = (1 << 5),
 346   FlagArtificial = (1 << 6),
 347   FlagExplicit = (1 << 7),
 348   FlagPrototyped = (1 << 8),
 349   FlagObjcClassComplete = (1 << 9),
 350   FlagObjectPointer = (1 << 10),
 351   FlagVector = (1 << 11),
 352   FlagStaticMember = (1 << 12),
 353   FlagLValueReference = (1 << 13),
 354   FlagRValueReference = (1 << 14),
 355   // Do not add values that are not supported by the minimum LLVM
 356   // version we support!
 357 };
 358
 359 inline LLVMRustDIFlags operator&(LLVMRustDIFlags A, LLVMRustDIFlags B) {
 360   return static_cast<LLVMRustDIFlags>(static_cast<uint32_t>(A) &
 361                                       static_cast<uint32_t>(B));
 362 }
 363
 364 inline LLVMRustDIFlags operator|(LLVMRustDIFlags A, LLVMRustDIFlags B) {
 365   return static_cast<LLVMRustDIFlags>(static_cast<uint32_t>(A) |
 366                                       static_cast<uint32_t>(B));
 367 }
 368
 369 inline LLVMRustDIFlags &operator|=(LLVMRustDIFlags &A, LLVMRustDIFlags B) {
 370   return A = A | B;
 371 }
 372
 373 inline bool isSet(LLVMRustDIFlags F) { return F != LLVMRustDIFlags::FlagZero; }
 374
 375 inline LLVMRustDIFlags visibility(LLVMRustDIFlags F) {
 376   return static_cast<LLVMRustDIFlags>(static_cast<uint32_t>(F) & 0x3);
 377 }
 378
 379 #if LLVM_VERSION_GE(4, 0)
 380 static DINode::DIFlags fromRust(LLVMRustDIFlags Flags) {
 381   DINode::DIFlags Result = DINode::DIFlags::FlagZero;
 382 #else
 383 static unsigned fromRust(LLVMRustDIFlags Flags) {
 384   unsigned Result = 0;
 385 #endif
 386
 387   switch (visibility(Flags)) {
 388   case LLVMRustDIFlags::FlagPrivate:
 389     Result |= DINode::DIFlags::FlagPrivate;
 390     break;
 391   case LLVMRustDIFlags::FlagProtected:
 392     Result |= DINode::DIFlags::FlagProtected;
 393     break;
 394   case LLVMRustDIFlags::FlagPublic:
 395     Result |= DINode::DIFlags::FlagPublic;
 396     break;
 397   default:
 398     // The rest are handled below
 399     break;
 400   }
 401
 402   if (isSet(Flags & LLVMRustDIFlags::FlagFwdDecl)) {
 403     Result |= DINode::DIFlags::FlagFwdDecl;
 404   }
 405   if (isSet(Flags & LLVMRustDIFlags::FlagAppleBlock)) {
 406     Result |= DINode::DIFlags::FlagAppleBlock;
 407   }
 408   if (isSet(Flags & LLVMRustDIFlags::FlagBlockByrefStruct)) {
 409     Result |= DINode::DIFlags::FlagBlockByrefStruct;
 410   }
 411   if (isSet(Flags & LLVMRustDIFlags::FlagVirtual)) {
 412     Result |= DINode::DIFlags::FlagVirtual;
 413   }
 414   if (isSet(Flags & LLVMRustDIFlags::FlagArtificial)) {
 415     Result |= DINode::DIFlags::FlagArtificial;
 416   }
 417   if (isSet(Flags & LLVMRustDIFlags::FlagExplicit)) {
 418     Result |= DINode::DIFlags::FlagExplicit;
 419   }
 420   if (isSet(Flags & LLVMRustDIFlags::FlagPrototyped)) {
 421     Result |= DINode::DIFlags::FlagPrototyped;
 422   }
 423   if (isSet(Flags & LLVMRustDIFlags::FlagObjcClassComplete)) {
 424     Result |= DINode::DIFlags::FlagObjcClassComplete;
 425   }
 426   if (isSet(Flags & LLVMRustDIFlags::FlagObjectPointer)) {
 427     Result |= DINode::DIFlags::FlagObjectPointer;
 428   }
 429   if (isSet(Flags & LLVMRustDIFlags::FlagVector)) {
 430     Result |= DINode::DIFlags::FlagVector;
 431   }
 432   if (isSet(Flags & LLVMRustDIFlags::FlagStaticMember)) {
 433     Result |= DINode::DIFlags::FlagStaticMember;
 434   }
 435   if (isSet(Flags & LLVMRustDIFlags::FlagLValueReference)) {
 436     Result |= DINode::DIFlags::FlagLValueReference;
 437   }
 438   if (isSet(Flags & LLVMRustDIFlags::FlagRValueReference)) {
 439     Result |= DINode::DIFlags::FlagRValueReference;
 440   }
 441
 442   return Result;
 443 }
 444
 445 extern "C" uint32_t LLVMRustDebugMetadataVersion() {
 446   return DEBUG_METADATA_VERSION;
 447 }
 448
 449 extern "C" uint32_t LLVMRustVersionMinor() { return LLVM_VERSION_MINOR; }
 450
 451 extern "C" uint32_t LLVMRustVersionMajor() { return LLVM_VERSION_MAJOR; }
 452
 453 extern "C" void LLVMRustAddModuleFlag(LLVMModuleRef M, const char *Name,
 454                                       uint32_t Value) {
 455   unwrap(M)->addModuleFlag(Module::Warning, Name, Value);
 456 }
 457
 458 extern "C" LLVMRustDIBuilderRef LLVMRustDIBuilderCreate(LLVMModuleRef M) {
 459   return new DIBuilder(*unwrap(M));
 460 }
 461
 462 extern "C" void LLVMRustDIBuilderDispose(LLVMRustDIBuilderRef Builder) {
 463   delete Builder;
 464 }
 465
 466 extern "C" void LLVMRustDIBuilderFinalize(LLVMRustDIBuilderRef Builder) {
 467   Builder->finalize();
 468 }
 469
 470 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateCompileUnit(
 471     LLVMRustDIBuilderRef Builder, unsigned Lang, const char *File,
 472     const char *Dir, const char *Producer, bool isOptimized, const char *Flags,
 473     unsigned RuntimeVer, const char *SplitName) {
 474   return wrap(Builder->createCompileUnit(Lang, File, Dir, Producer, isOptimized,
 475                                          Flags, RuntimeVer, SplitName));
 476 }
 477
 478 extern "C" LLVMRustMetadataRef
 479 LLVMRustDIBuilderCreateFile(LLVMRustDIBuilderRef Builder, const char *Filename,
 480                             const char *Directory) {
 481   return wrap(Builder->createFile(Filename, Directory));
 482 }
 483
 484 extern "C" LLVMRustMetadataRef
 485 LLVMRustDIBuilderCreateSubroutineType(LLVMRustDIBuilderRef Builder,
 486                                       LLVMRustMetadataRef File,
 487                                       LLVMRustMetadataRef ParameterTypes) {
 488   return wrap(Builder->createSubroutineType(
 489 #if LLVM_VERSION_EQ(3, 7)
 490       unwrapDI<DIFile>(File),
 491 #endif
 492       DITypeRefArray(unwrap<MDTuple>(ParameterTypes))));
 493 }
 494
 495 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateFunction(
 496     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope, const char *Name,
 497     const char *LinkageName, LLVMRustMetadataRef File, unsigned LineNo,
 498     LLVMRustMetadataRef Ty, bool IsLocalToUnit, bool IsDefinition,
 499     unsigned ScopeLine, LLVMRustDIFlags Flags, bool IsOptimized,
 500     LLVMValueRef Fn, LLVMRustMetadataRef TParam, LLVMRustMetadataRef Decl) {
 501 #if LLVM_VERSION_GE(3, 8)
 502   DITemplateParameterArray TParams =
 503       DITemplateParameterArray(unwrap<MDTuple>(TParam));
 504   DISubprogram *Sub = Builder->createFunction(
 505       unwrapDI<DIScope>(Scope), Name, LinkageName, unwrapDI<DIFile>(File),
 506       LineNo, unwrapDI<DISubroutineType>(Ty), IsLocalToUnit, IsDefinition,
 507       ScopeLine, fromRust(Flags), IsOptimized, TParams,
 508       unwrapDIPtr<DISubprogram>(Decl));
 509   unwrap<Function>(Fn)->setSubprogram(Sub);
 510   return wrap(Sub);
 511 #else
 512   return wrap(Builder->createFunction(
 513       unwrapDI<DIScope>(Scope), Name, LinkageName, unwrapDI<DIFile>(File),
 514       LineNo, unwrapDI<DISubroutineType>(Ty), IsLocalToUnit, IsDefinition,
 515       ScopeLine, fromRust(Flags), IsOptimized, unwrap<Function>(Fn),
 516       unwrapDIPtr<MDNode>(TParam), unwrapDIPtr<MDNode>(Decl)));
 517 #endif
 518 }
 519
 520 extern "C" LLVMRustMetadataRef
 521 LLVMRustDIBuilderCreateBasicType(LLVMRustDIBuilderRef Builder, const char *Name,
 522                                  uint64_t SizeInBits, uint32_t AlignInBits,
 523                                  unsigned Encoding) {
 524   return wrap(Builder->createBasicType(Name, SizeInBits,
 525 #if LLVM_VERSION_LE(3, 9)
 526                                        AlignInBits,
 527 #endif
 528                                        Encoding));
 529 }
 530
 531 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreatePointerType(
 532     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef PointeeTy,
 533     uint64_t SizeInBits, uint32_t AlignInBits, const char *Name) {
 534   return wrap(Builder->createPointerType(unwrapDI<DIType>(PointeeTy),
 535                                          SizeInBits, AlignInBits, Name));
 536 }
 537
 538 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateStructType(
 539     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope, const char *Name,
 540     LLVMRustMetadataRef File, unsigned LineNumber, uint64_t SizeInBits,
 541     uint32_t AlignInBits, LLVMRustDIFlags Flags,
 542     LLVMRustMetadataRef DerivedFrom, LLVMRustMetadataRef Elements,
 543     unsigned RunTimeLang, LLVMRustMetadataRef VTableHolder,
 544     const char *UniqueId) {
 545   return wrap(Builder->createStructType(
 546       unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIFile>(File), LineNumber,
 547       SizeInBits, AlignInBits, fromRust(Flags), unwrapDI<DIType>(DerivedFrom),
 548       DINodeArray(unwrapDI<MDTuple>(Elements)), RunTimeLang,
 549       unwrapDI<DIType>(VTableHolder), UniqueId));
 550 }
 551
 552 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateMemberType(
 553     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope, const char *Name,
 554     LLVMRustMetadataRef File, unsigned LineNo, uint64_t SizeInBits,
 555     uint32_t AlignInBits, uint64_t OffsetInBits, LLVMRustDIFlags Flags,
 556     LLVMRustMetadataRef Ty) {
 557   return wrap(Builder->createMemberType(unwrapDI<DIDescriptor>(Scope), Name,
 558                                         unwrapDI<DIFile>(File), LineNo,
 559                                         SizeInBits, AlignInBits, OffsetInBits,
 560                                         fromRust(Flags), unwrapDI<DIType>(Ty)));
 561 }
 562
 563 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateLexicalBlock(
 564     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope,
 565     LLVMRustMetadataRef File, unsigned Line, unsigned Col) {
 566   return wrap(Builder->createLexicalBlock(unwrapDI<DIDescriptor>(Scope),
 567                                           unwrapDI<DIFile>(File), Line, Col));
 568 }
 569
 570 extern "C" LLVMRustMetadataRef
 571 LLVMRustDIBuilderCreateLexicalBlockFile(LLVMRustDIBuilderRef Builder,
 572                                         LLVMRustMetadataRef Scope,
 573                                         LLVMRustMetadataRef File) {
 574   return wrap(Builder->createLexicalBlockFile(unwrapDI<DIDescriptor>(Scope),
 575                                               unwrapDI<DIFile>(File)));
 576 }
 577
 578 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateStaticVariable(
 579     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Context, const char *Name,
 580     const char *LinkageName, LLVMRustMetadataRef File, unsigned LineNo,
 581     LLVMRustMetadataRef Ty, bool IsLocalToUnit, LLVMValueRef V,
 582     LLVMRustMetadataRef Decl = nullptr, uint32_t AlignInBits = 0) {
 583   Constant *InitVal = cast<Constant>(unwrap(V));
 584
 585 #if LLVM_VERSION_GE(4, 0)
 586   llvm::DIExpression *InitExpr = nullptr;
 587   if (llvm::ConstantInt *IntVal = llvm::dyn_cast<llvm::ConstantInt>(InitVal)) {
 588     InitExpr = Builder->createConstantValueExpression(
 589         IntVal->getValue().getSExtValue());
 590   } else if (llvm::ConstantFP *FPVal =
 591                  llvm::dyn_cast<llvm::ConstantFP>(InitVal)) {
 592     InitExpr = Builder->createConstantValueExpression(
 593         FPVal->getValueAPF().bitcastToAPInt().getZExtValue());
 594   }
 595 #endif
 596
 597 #if LLVM_VERSION_GE(4, 0)
 598   return wrap(Builder->createGlobalVariableExpression(
 599 #else
 600   return wrap(Builder->createGlobalVariable(
 601 #endif
 602       unwrapDI<DIDescriptor>(Context), Name, LinkageName,
 603       unwrapDI<DIFile>(File), LineNo, unwrapDI<DIType>(Ty), IsLocalToUnit,
 604 #if LLVM_VERSION_GE(4, 0)
 605       InitExpr,
 606 #else
 607       InitVal,
 608 #endif
 609       unwrapDIPtr<MDNode>(Decl)
 610 #if LLVM_VERSION_GE(4, 0)
 611       ,
 612       AlignInBits
 613 #endif
 614       ));
 615 }
 616
 617 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateVariable(
 618     LLVMRustDIBuilderRef Builder, unsigned Tag, LLVMRustMetadataRef Scope,
 619     const char *Name, LLVMRustMetadataRef File, unsigned LineNo,
 620     LLVMRustMetadataRef Ty, bool AlwaysPreserve, LLVMRustDIFlags Flags,
 621     unsigned ArgNo, uint32_t AlignInBits) {
 622 #if LLVM_VERSION_GE(3, 8)
 623   if (Tag == 0x100) { // DW_TAG_auto_variable
 624     return wrap(Builder->createAutoVariable(
 625         unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIFile>(File), LineNo,
 626         unwrapDI<DIType>(Ty), AlwaysPreserve, fromRust(Flags)
 627 #if LLVM_VERSION_GE(4, 0)
 628         ,
 629         AlignInBits
 630 #endif
 631         ));
 632   } else {
 633     return wrap(Builder->createParameterVariable(
 634         unwrapDI<DIDescriptor>(Scope), Name, ArgNo, unwrapDI<DIFile>(File),
 635         LineNo, unwrapDI<DIType>(Ty), AlwaysPreserve, fromRust(Flags)));
 636   }
 637 #else
 638   return wrap(Builder->createLocalVariable(
 639       Tag, unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIFile>(File), LineNo,
 640       unwrapDI<DIType>(Ty), AlwaysPreserve, fromRust(Flags), ArgNo));
 641 #endif
 642 }
 643
 644 extern "C" LLVMRustMetadataRef
 645 LLVMRustDIBuilderCreateArrayType(LLVMRustDIBuilderRef Builder, uint64_t Size,
 646                                  uint32_t AlignInBits, LLVMRustMetadataRef Ty,
 647                                  LLVMRustMetadataRef Subscripts) {
 648   return wrap(
 649       Builder->createArrayType(Size, AlignInBits, unwrapDI<DIType>(Ty),
 650                                DINodeArray(unwrapDI<MDTuple>(Subscripts))));
 651 }
 652
 653 extern "C" LLVMRustMetadataRef
 654 LLVMRustDIBuilderCreateVectorType(LLVMRustDIBuilderRef Builder, uint64_t Size,
 655                                   uint32_t AlignInBits, LLVMRustMetadataRef Ty,
 656                                   LLVMRustMetadataRef Subscripts) {
 657   return wrap(
 658       Builder->createVectorType(Size, AlignInBits, unwrapDI<DIType>(Ty),
 659                                 DINodeArray(unwrapDI<MDTuple>(Subscripts))));
 660 }
 661
 662 extern "C" LLVMRustMetadataRef
 663 LLVMRustDIBuilderGetOrCreateSubrange(LLVMRustDIBuilderRef Builder, int64_t Lo,
 664                                      int64_t Count) {
 665   return wrap(Builder->getOrCreateSubrange(Lo, Count));
 666 }
 667
 668 extern "C" LLVMRustMetadataRef
 669 LLVMRustDIBuilderGetOrCreateArray(LLVMRustDIBuilderRef Builder,
 670                                   LLVMRustMetadataRef *Ptr, unsigned Count) {
 671   Metadata **DataValue = unwrap(Ptr);
 672   return wrap(
 673       Builder->getOrCreateArray(ArrayRef<Metadata *>(DataValue, Count)).get());
 674 }
 675
 676 extern "C" LLVMValueRef LLVMRustDIBuilderInsertDeclareAtEnd(
 677     LLVMRustDIBuilderRef Builder, LLVMValueRef V, LLVMRustMetadataRef VarInfo,
 678     int64_t *AddrOps, unsigned AddrOpsCount, LLVMValueRef DL,
 679     LLVMBasicBlockRef InsertAtEnd) {
 680   return wrap(Builder->insertDeclare(
 681       unwrap(V), unwrap<DILocalVariable>(VarInfo),
 682       Builder->createExpression(llvm::ArrayRef<int64_t>(AddrOps, AddrOpsCount)),
 683       DebugLoc(cast<MDNode>(unwrap<MetadataAsValue>(DL)->getMetadata())),
 684       unwrap(InsertAtEnd)));
 685 }
 686
 687 extern "C" LLVMRustMetadataRef
 688 LLVMRustDIBuilderCreateEnumerator(LLVMRustDIBuilderRef Builder,
 689                                   const char *Name, uint64_t Val) {
 690   return wrap(Builder->createEnumerator(Name, Val));
 691 }
 692
 693 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateEnumerationType(
 694     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope, const char *Name,
 695     LLVMRustMetadataRef File, unsigned LineNumber, uint64_t SizeInBits,
 696     uint32_t AlignInBits, LLVMRustMetadataRef Elements,
 697     LLVMRustMetadataRef ClassTy) {
 698   return wrap(Builder->createEnumerationType(
 699       unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIFile>(File), LineNumber,
 700       SizeInBits, AlignInBits, DINodeArray(unwrapDI<MDTuple>(Elements)),
 701       unwrapDI<DIType>(ClassTy)));
 702 }
 703
 704 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateUnionType(
 705     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope, const char *Name,
 706     LLVMRustMetadataRef File, unsigned LineNumber, uint64_t SizeInBits,
 707     uint32_t AlignInBits, LLVMRustDIFlags Flags, LLVMRustMetadataRef Elements,
 708     unsigned RunTimeLang, const char *UniqueId) {
 709   return wrap(Builder->createUnionType(
 710       unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIFile>(File), LineNumber,
 711       SizeInBits, AlignInBits, fromRust(Flags),
 712       DINodeArray(unwrapDI<MDTuple>(Elements)), RunTimeLang, UniqueId));
 713 }
 714
 715 extern "C" LLVMRustMetadataRef LLVMRustDIBuilderCreateTemplateTypeParameter(
 716     LLVMRustDIBuilderRef Builder, LLVMRustMetadataRef Scope, const char *Name,
 717     LLVMRustMetadataRef Ty, LLVMRustMetadataRef File, unsigned LineNo,
 718     unsigned ColumnNo) {
 719   return wrap(Builder->createTemplateTypeParameter(
 720       unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIType>(Ty)));
 721 }
 722
 723 extern "C" LLVMRustMetadataRef
 724 LLVMRustDIBuilderCreateNameSpace(LLVMRustDIBuilderRef Builder,
 725                                  LLVMRustMetadataRef Scope, const char *Name,
 726                                  LLVMRustMetadataRef File, unsigned LineNo) {
 727   return wrap(Builder->createNameSpace(
 728       unwrapDI<DIDescriptor>(Scope), Name, unwrapDI<DIFile>(File), LineNo
 729 #if LLVM_VERSION_GE(4, 0)
 730       ,
 731       false // ExportSymbols (only relevant for C++ anonymous namespaces)
 732 #endif
 733       ));
 734 }
 735
 736 extern "C" void
 737 LLVMRustDICompositeTypeSetTypeArray(LLVMRustDIBuilderRef Builder,
 738                                     LLVMRustMetadataRef CompositeTy,
 739                                     LLVMRustMetadataRef TyArray) {
 740   DICompositeType *Tmp = unwrapDI<DICompositeType>(CompositeTy);
 741   Builder->replaceArrays(Tmp, DINodeArray(unwrap<MDTuple>(TyArray)));
 742 }
 743
 744 extern "C" LLVMValueRef
 745 LLVMRustDIBuilderCreateDebugLocation(LLVMContextRef ContextRef, unsigned Line,
 746                                      unsigned Column, LLVMRustMetadataRef Scope,
 747                                      LLVMRustMetadataRef InlinedAt) {
 748   LLVMContext &Context = *unwrap(ContextRef);
 749
 750   DebugLoc debug_loc = DebugLoc::get(Line, Column, unwrapDIPtr<MDNode>(Scope),
 751                                      unwrapDIPtr<MDNode>(InlinedAt));
 752
 753   return wrap(MetadataAsValue::get(Context, debug_loc.getAsMDNode()));
 754 }
 755
 756 extern "C" int64_t LLVMRustDIBuilderCreateOpDeref() {
 757   return dwarf::DW_OP_deref;
 758 }
 759
 760 extern "C" int64_t LLVMRustDIBuilderCreateOpPlus() { return dwarf::DW_OP_plus; }
 761
 762 extern "C" void LLVMRustWriteTypeToString(LLVMTypeRef Ty, RustStringRef Str) {
 763   RawRustStringOstream OS(Str);
 764   unwrap<llvm::Type>(Ty)->print(OS);
 765 }
 766
 767 extern "C" void LLVMRustWriteValueToString(LLVMValueRef V,
 768                                            RustStringRef Str) {
 769   RawRustStringOstream OS(Str);
 770   OS << "(";
 771   unwrap<llvm::Value>(V)->getType()->print(OS);
 772   OS << ":";
 773   unwrap<llvm::Value>(V)->print(OS);
 774   OS << ")";
 775 }
 776
 777 extern "C" bool LLVMRustLinkInExternalBitcode(LLVMModuleRef DstRef, char *BC,
 778                                               size_t Len) {
 779   Module *Dst = unwrap(DstRef);
 780
 781   std::unique_ptr<MemoryBuffer> Buf =
 782       MemoryBuffer::getMemBufferCopy(StringRef(BC, Len));
 783
 784 #if LLVM_VERSION_GE(4, 0)
 785   Expected<std::unique_ptr<Module>> SrcOrError =
 786       llvm::getLazyBitcodeModule(Buf->getMemBufferRef(), Dst->getContext());
 787   if (!SrcOrError) {
 788     LLVMRustSetLastError(toString(SrcOrError.takeError()).c_str());
 789     return false;
 790   }
 791
 792   auto Src = std::move(*SrcOrError);
 793 #else
 794   ErrorOr<std::unique_ptr<Module>> Src =
 795       llvm::getLazyBitcodeModule(std::move(Buf), Dst->getContext());
 796   if (!Src) {
 797     LLVMRustSetLastError(Src.getError().message().c_str());
 798     return false;
 799   }
 800 #endif
 801
 802   std::string Err;
 803
 804   raw_string_ostream Stream(Err);
 805   DiagnosticPrinterRawOStream DP(Stream);
 806 #if LLVM_VERSION_GE(4, 0)
 807   if (Linker::linkModules(*Dst, std::move(Src))) {
 808 #elif LLVM_VERSION_GE(3, 8)
 809   if (Linker::linkModules(*Dst, std::move(Src.get()))) {
 810 #else
 811   if (Linker::LinkModules(Dst, Src->get(),
 812                           [&](const DiagnosticInfo &DI) { DI.print(DP); })) {
 813 #endif
 814     LLVMRustSetLastError(Err.c_str());
 815     return false;
 816   }
 817   return true;
 818 }
 819
 820 // Note that the two following functions look quite similar to the
 821 // LLVMGetSectionName function. Sadly, it appears that this function only
 822 // returns a char* pointer, which isn't guaranteed to be null-terminated. The
 823 // function provided by LLVM doesn't return the length, so we've created our own
 824 // function which returns the length as well as the data pointer.
 825 //
 826 // For an example of this not returning a null terminated string, see
 827 // lib/Object/COFFObjectFile.cpp in the getSectionName function. One of the
 828 // branches explicitly creates a StringRef without a null terminator, and then
 829 // that's returned.
 830
 831 inline section_iterator *unwrap(LLVMSectionIteratorRef SI) {
 832   return reinterpret_cast<section_iterator *>(SI);
 833 }
 834
 835 extern "C" size_t LLVMRustGetSectionName(LLVMSectionIteratorRef SI,
 836                                          const char **Ptr) {
 837   StringRef Ret;
 838   if (std::error_code EC = (*unwrap(SI))->getName(Ret))
 839     report_fatal_error(EC.message());
 840   *Ptr = Ret.data();
 841   return Ret.size();
 842 }
 843
 844 // LLVMArrayType function does not support 64-bit ElementCount
 845 extern "C" LLVMTypeRef LLVMRustArrayType(LLVMTypeRef ElementTy,
 846                                          uint64_t ElementCount) {
 847   return wrap(ArrayType::get(unwrap(ElementTy), ElementCount));
 848 }
 849
 850 DEFINE_SIMPLE_CONVERSION_FUNCTIONS(Twine, LLVMTwineRef)
 851 DEFINE_SIMPLE_CONVERSION_FUNCTIONS(DebugLoc, LLVMDebugLocRef)
 852
 853 extern "C" void LLVMRustWriteTwineToString(LLVMTwineRef T, RustStringRef Str) {
 854   RawRustStringOstream OS(Str);
 855   unwrap(T)->print(OS);
 856 }
 857
 858 extern "C" void LLVMRustUnpackOptimizationDiagnostic(
 859     LLVMDiagnosticInfoRef DI, RustStringRef PassNameOut,
 860     LLVMValueRef *FunctionOut, LLVMDebugLocRef *DebugLocOut,
 861     RustStringRef MessageOut) {
 862   // Undefined to call this not on an optimization diagnostic!
 863   llvm::DiagnosticInfoOptimizationBase *Opt =
 864       static_cast<llvm::DiagnosticInfoOptimizationBase *>(unwrap(DI));
 865
 866   RawRustStringOstream PassNameOS(PassNameOut);
 867   PassNameOS << Opt->getPassName();
 868   *FunctionOut = wrap(&Opt->getFunction());
 869   *DebugLocOut = wrap(&Opt->getDebugLoc());
 870   RawRustStringOstream MessageOS(MessageOut);
 871   MessageOS << Opt->getMsg();
 872 }
 873
 874 extern "C" void
 875 LLVMRustUnpackInlineAsmDiagnostic(LLVMDiagnosticInfoRef DI, unsigned *CookieOut,
 876                                   LLVMTwineRef *MessageOut,
 877                                   LLVMValueRef *InstructionOut) {
 878   // Undefined to call this not on an inline assembly diagnostic!
 879   llvm::DiagnosticInfoInlineAsm *IA =
 880       static_cast<llvm::DiagnosticInfoInlineAsm *>(unwrap(DI));
 881
 882   *CookieOut = IA->getLocCookie();
 883   *MessageOut = wrap(&IA->getMsgStr());
 884   *InstructionOut = wrap(IA->getInstruction());
 885 }
 886
 887 extern "C" void LLVMRustWriteDiagnosticInfoToString(LLVMDiagnosticInfoRef DI,
 888                                                     RustStringRef Str) {
 889   RawRustStringOstream OS(Str);
 890   DiagnosticPrinterRawOStream DP(OS);
 891   unwrap(DI)->print(DP);
 892 }
 893
 894 enum class LLVMRustDiagnosticKind {
 895   Other,
 896   InlineAsm,
 897   StackSize,
 898   DebugMetadataVersion,
 899   SampleProfile,
 900   OptimizationRemark,
 901   OptimizationRemarkMissed,
 902   OptimizationRemarkAnalysis,
 903   OptimizationRemarkAnalysisFPCommute,
 904   OptimizationRemarkAnalysisAliasing,
 905   OptimizationRemarkOther,
 906   OptimizationFailure,
 907 };
 908
 909 static LLVMRustDiagnosticKind toRust(DiagnosticKind Kind) {
 910   switch (Kind) {
 911   case DK_InlineAsm:
 912     return LLVMRustDiagnosticKind::InlineAsm;
 913   case DK_StackSize:
 914     return LLVMRustDiagnosticKind::StackSize;
 915   case DK_DebugMetadataVersion:
 916     return LLVMRustDiagnosticKind::DebugMetadataVersion;
 917   case DK_SampleProfile:
 918     return LLVMRustDiagnosticKind::SampleProfile;
 919   case DK_OptimizationRemark:
 920     return LLVMRustDiagnosticKind::OptimizationRemark;
 921   case DK_OptimizationRemarkMissed:
 922     return LLVMRustDiagnosticKind::OptimizationRemarkMissed;
 923   case DK_OptimizationRemarkAnalysis:
 924     return LLVMRustDiagnosticKind::OptimizationRemarkAnalysis;
 925 #if LLVM_VERSION_GE(3, 8)
 926   case DK_OptimizationRemarkAnalysisFPCommute:
 927     return LLVMRustDiagnosticKind::OptimizationRemarkAnalysisFPCommute;
 928   case DK_OptimizationRemarkAnalysisAliasing:
 929     return LLVMRustDiagnosticKind::OptimizationRemarkAnalysisAliasing;
 930 #endif
 931   default:
 932 #if LLVM_VERSION_GE(3, 9)
 933     return (Kind >= DK_FirstRemark && Kind <= DK_LastRemark)
 934                ? LLVMRustDiagnosticKind::OptimizationRemarkOther
 935                : LLVMRustDiagnosticKind::Other;
 936 #else
 937     return LLVMRustDiagnosticKind::Other;
 938 #endif
 939   }
 940 }
 941
 942 extern "C" LLVMRustDiagnosticKind
 943 LLVMRustGetDiagInfoKind(LLVMDiagnosticInfoRef DI) {
 944   return toRust((DiagnosticKind)unwrap(DI)->getKind());
 945 }
 946 // This is kept distinct from LLVMGetTypeKind, because when
 947 // a new type kind is added, the Rust-side enum must be
 948 // updated or UB will result.
 949 extern "C" LLVMTypeKind LLVMRustGetTypeKind(LLVMTypeRef Ty) {
 950   switch (unwrap(Ty)->getTypeID()) {
 951   case Type::VoidTyID:
 952     return LLVMVoidTypeKind;
 953   case Type::HalfTyID:
 954     return LLVMHalfTypeKind;
 955   case Type::FloatTyID:
 956     return LLVMFloatTypeKind;
 957   case Type::DoubleTyID:
 958     return LLVMDoubleTypeKind;
 959   case Type::X86_FP80TyID:
 960     return LLVMX86_FP80TypeKind;
 961   case Type::FP128TyID:
 962     return LLVMFP128TypeKind;
 963   case Type::PPC_FP128TyID:
 964     return LLVMPPC_FP128TypeKind;
 965   case Type::LabelTyID:
 966     return LLVMLabelTypeKind;
 967   case Type::MetadataTyID:
 968     return LLVMMetadataTypeKind;
 969   case Type::IntegerTyID:
 970     return LLVMIntegerTypeKind;
 971   case Type::FunctionTyID:
 972     return LLVMFunctionTypeKind;
 973   case Type::StructTyID:
 974     return LLVMStructTypeKind;
 975   case Type::ArrayTyID:
 976     return LLVMArrayTypeKind;
 977   case Type::PointerTyID:
 978     return LLVMPointerTypeKind;
 979   case Type::VectorTyID:
 980     return LLVMVectorTypeKind;
 981   case Type::X86_MMXTyID:
 982     return LLVMX86_MMXTypeKind;
 983 #if LLVM_VERSION_GE(3, 8)
 984   case Type::TokenTyID:
 985     return LLVMTokenTypeKind;
 986 #endif
 987   }
 988   llvm_unreachable("Unhandled TypeID.");
 989 }
 990
 991 extern "C" void LLVMRustWriteDebugLocToString(LLVMContextRef C,
 992                                               LLVMDebugLocRef DL,
 993                                               RustStringRef Str) {
 994   RawRustStringOstream OS(Str);
 995   unwrap(DL)->print(OS);
 996 }
 997
 998 DEFINE_SIMPLE_CONVERSION_FUNCTIONS(SMDiagnostic, LLVMSMDiagnosticRef)
 999
1000 extern "C" void LLVMRustSetInlineAsmDiagnosticHandler(
1001     LLVMContextRef C, LLVMContext::InlineAsmDiagHandlerTy H, void *CX) {
1002   unwrap(C)->setInlineAsmDiagnosticHandler(H, CX);
1003 }
1004
1005 extern "C" void LLVMRustWriteSMDiagnosticToString(LLVMSMDiagnosticRef D,
1006                                                   RustStringRef Str) {
1007   RawRustStringOstream OS(Str);
1008   unwrap(D)->print("", OS);
1009 }
1010
1011 extern "C" LLVMValueRef
1012 LLVMRustBuildLandingPad(LLVMBuilderRef B, LLVMTypeRef Ty,
1013                         LLVMValueRef PersFn, unsigned NumClauses,
1014                         const char *Name, LLVMValueRef F) {
1015   return LLVMBuildLandingPad(B, Ty, PersFn, NumClauses, Name);
1016 }
1017
1018 extern "C" LLVMValueRef LLVMRustBuildCleanupPad(LLVMBuilderRef B,
1019                                                 LLVMValueRef ParentPad,
1020                                                 unsigned ArgCount,
1021                                                 LLVMValueRef *LLArgs,
1022                                                 const char *Name) {
1023 #if LLVM_VERSION_GE(3, 8)
1024   Value **Args = unwrap(LLArgs);
1025   if (ParentPad == nullptr) {
1026     Type *Ty = Type::getTokenTy(unwrap(B)->getContext());
1027     ParentPad = wrap(Constant::getNullValue(Ty));
1028   }
1029   return wrap(unwrap(B)->CreateCleanupPad(
1030       unwrap(ParentPad), ArrayRef<Value *>(Args, ArgCount), Name));
1031 #else
1032   return nullptr;
1033 #endif
1034 }
1035
1036 extern "C" LLVMValueRef LLVMRustBuildCleanupRet(LLVMBuilderRef B,
1037                                                 LLVMValueRef CleanupPad,
1038                                                 LLVMBasicBlockRef UnwindBB) {
1039 #if LLVM_VERSION_GE(3, 8)
1040   CleanupPadInst *Inst = cast<CleanupPadInst>(unwrap(CleanupPad));
1041   return wrap(unwrap(B)->CreateCleanupRet(Inst, unwrap(UnwindBB)));
1042 #else
1043   return nullptr;
1044 #endif
1045 }
1046
1047 extern "C" LLVMValueRef
1048 LLVMRustBuildCatchPad(LLVMBuilderRef B, LLVMValueRef ParentPad,
1049                       unsigned ArgCount, LLVMValueRef *LLArgs, const char *Name) {
1050 #if LLVM_VERSION_GE(3, 8)
1051   Value **Args = unwrap(LLArgs);
1052   return wrap(unwrap(B)->CreateCatchPad(
1053       unwrap(ParentPad), ArrayRef<Value *>(Args, ArgCount), Name));
1054 #else
1055   return nullptr;
1056 #endif
1057 }
1058
1059 extern "C" LLVMValueRef LLVMRustBuildCatchRet(LLVMBuilderRef B,
1060                                               LLVMValueRef Pad,
1061                                               LLVMBasicBlockRef BB) {
1062 #if LLVM_VERSION_GE(3, 8)
1063   return wrap(unwrap(B)->CreateCatchRet(cast<CatchPadInst>(unwrap(Pad)),
1064                                               unwrap(BB)));
1065 #else
1066   return nullptr;
1067 #endif
1068 }
1069
1070 extern "C" LLVMValueRef LLVMRustBuildCatchSwitch(LLVMBuilderRef B,
1071                                                  LLVMValueRef ParentPad,
1072                                                  LLVMBasicBlockRef BB,
1073                                                  unsigned NumHandlers,
1074                                                  const char *Name) {
1075 #if LLVM_VERSION_GE(3, 8)
1076   if (ParentPad == nullptr) {
1077     Type *Ty = Type::getTokenTy(unwrap(B)->getContext());
1078     ParentPad = wrap(Constant::getNullValue(Ty));
1079   }
1080   return wrap(unwrap(B)->CreateCatchSwitch(unwrap(ParentPad), unwrap(BB),
1081                                                  NumHandlers, Name));
1082 #else
1083   return nullptr;
1084 #endif
1085 }
1086
1087 extern "C" void LLVMRustAddHandler(LLVMValueRef CatchSwitchRef,
1088                                    LLVMBasicBlockRef Handler) {
1089 #if LLVM_VERSION_GE(3, 8)
1090   Value *CatchSwitch = unwrap(CatchSwitchRef);
1091   cast<CatchSwitchInst>(CatchSwitch)->addHandler(unwrap(Handler));
1092 #endif
1093 }
1094
1095 #if LLVM_VERSION_GE(3, 8)
1096 extern "C" OperandBundleDef *LLVMRustBuildOperandBundleDef(const char *Name,
1097                                                            LLVMValueRef *Inputs,
1098                                                            unsigned NumInputs) {
1099   return new OperandBundleDef(Name, makeArrayRef(unwrap(Inputs), NumInputs));
1100 }
1101
1102 extern "C" void LLVMRustFreeOperandBundleDef(OperandBundleDef *Bundle) {
1103   delete Bundle;
1104 }
1105
1106 extern "C" LLVMValueRef LLVMRustBuildCall(LLVMBuilderRef B, LLVMValueRef Fn,
1107                                           LLVMValueRef *Args, unsigned NumArgs,
1108                                           OperandBundleDef *Bundle,
1109                                           const char *Name) {
1110   unsigned Len = Bundle ? 1 : 0;
1111   ArrayRef<OperandBundleDef> Bundles = makeArrayRef(Bundle, Len);
1112   return wrap(unwrap(B)->CreateCall(
1113       unwrap(Fn), makeArrayRef(unwrap(Args), NumArgs), Bundles, Name));
1114 }
1115
1116 extern "C" LLVMValueRef
1117 LLVMRustBuildInvoke(LLVMBuilderRef B, LLVMValueRef Fn, LLVMValueRef *Args,
1118                     unsigned NumArgs, LLVMBasicBlockRef Then,
1119                     LLVMBasicBlockRef Catch, OperandBundleDef *Bundle,
1120                     const char *Name) {
1121   unsigned Len = Bundle ? 1 : 0;
1122   ArrayRef<OperandBundleDef> Bundles = makeArrayRef(Bundle, Len);
1123   return wrap(unwrap(B)->CreateInvoke(unwrap(Fn), unwrap(Then), unwrap(Catch),
1124                                       makeArrayRef(unwrap(Args), NumArgs),
1125                                       Bundles, Name));
1126 }
1127 #else
1128 extern "C" void *LLVMRustBuildOperandBundleDef(const char *Name,
1129                                                LLVMValueRef *Inputs,
1130                                                unsigned NumInputs) {
1131   return nullptr;
1132 }
1133
1134 extern "C" void LLVMRustFreeOperandBundleDef(void *Bundle) {}
1135
1136 extern "C" LLVMValueRef LLVMRustBuildCall(LLVMBuilderRef B, LLVMValueRef Fn,
1137                                           LLVMValueRef *Args, unsigned NumArgs,
1138                                           void *Bundle, const char *Name) {
1139   return LLVMBuildCall(B, Fn, Args, NumArgs, Name);
1140 }
1141
1142 extern "C" LLVMValueRef
1143 LLVMRustBuildInvoke(LLVMBuilderRef B, LLVMValueRef Fn, LLVMValueRef *Args,
1144                     unsigned NumArgs, LLVMBasicBlockRef Then,
1145                     LLVMBasicBlockRef Catch, void *Bundle, const char *Name) {
1146   return LLVMBuildInvoke(B, Fn, Args, NumArgs, Then, Catch, Name);
1147 }
1148 #endif
1149
1150 extern "C" void LLVMRustPositionBuilderAtStart(LLVMBuilderRef B,
1151                                                LLVMBasicBlockRef BB) {
1152   auto Point = unwrap(BB)->getFirstInsertionPt();
1153   unwrap(B)->SetInsertPoint(unwrap(BB), Point);
1154 }
1155
1156 extern "C" void LLVMRustSetComdat(LLVMModuleRef M, LLVMValueRef V,
1157                                   const char *Name) {
1158   Triple TargetTriple(unwrap(M)->getTargetTriple());
1159   GlobalObject *GV = unwrap<GlobalObject>(V);
1160   if (!TargetTriple.isOSBinFormatMachO()) {
1161     GV->setComdat(unwrap(M)->getOrInsertComdat(Name));
1162   }
1163 }
1164
1165 extern "C" void LLVMRustUnsetComdat(LLVMValueRef V) {
1166   GlobalObject *GV = unwrap<GlobalObject>(V);
1167   GV->setComdat(nullptr);
1168 }
1169
1170 enum class LLVMRustLinkage {
1171   ExternalLinkage = 0,
1172   AvailableExternallyLinkage = 1,
1173   LinkOnceAnyLinkage = 2,
1174   LinkOnceODRLinkage = 3,
1175   WeakAnyLinkage = 4,
1176   WeakODRLinkage = 5,
1177   AppendingLinkage = 6,
1178   InternalLinkage = 7,
1179   PrivateLinkage = 8,
1180   ExternalWeakLinkage = 9,
1181   CommonLinkage = 10,
1182 };
1183
1184 static LLVMRustLinkage toRust(LLVMLinkage Linkage) {
1185   switch (Linkage) {
1186   case LLVMExternalLinkage:
1187     return LLVMRustLinkage::ExternalLinkage;
1188   case LLVMAvailableExternallyLinkage:
1189     return LLVMRustLinkage::AvailableExternallyLinkage;
1190   case LLVMLinkOnceAnyLinkage:
1191     return LLVMRustLinkage::LinkOnceAnyLinkage;
1192   case LLVMLinkOnceODRLinkage:
1193     return LLVMRustLinkage::LinkOnceODRLinkage;
1194   case LLVMWeakAnyLinkage:
1195     return LLVMRustLinkage::WeakAnyLinkage;
1196   case LLVMWeakODRLinkage:
1197     return LLVMRustLinkage::WeakODRLinkage;
1198   case LLVMAppendingLinkage:
1199     return LLVMRustLinkage::AppendingLinkage;
1200   case LLVMInternalLinkage:
1201     return LLVMRustLinkage::InternalLinkage;
1202   case LLVMPrivateLinkage:
1203     return LLVMRustLinkage::PrivateLinkage;
1204   case LLVMExternalWeakLinkage:
1205     return LLVMRustLinkage::ExternalWeakLinkage;
1206   case LLVMCommonLinkage:
1207     return LLVMRustLinkage::CommonLinkage;
1208   default:
1209     llvm_unreachable("Invalid LLVMRustLinkage value!");
1210   }
1211 }
1212
1213 static LLVMLinkage fromRust(LLVMRustLinkage Linkage) {
1214   switch (Linkage) {
1215   case LLVMRustLinkage::ExternalLinkage:
1216     return LLVMExternalLinkage;
1217   case LLVMRustLinkage::AvailableExternallyLinkage:
1218     return LLVMAvailableExternallyLinkage;
1219   case LLVMRustLinkage::LinkOnceAnyLinkage:
1220     return LLVMLinkOnceAnyLinkage;
1221   case LLVMRustLinkage::LinkOnceODRLinkage:
1222     return LLVMLinkOnceODRLinkage;
1223   case LLVMRustLinkage::WeakAnyLinkage:
1224     return LLVMWeakAnyLinkage;
1225   case LLVMRustLinkage::WeakODRLinkage:
1226     return LLVMWeakODRLinkage;
1227   case LLVMRustLinkage::AppendingLinkage:
1228     return LLVMAppendingLinkage;
1229   case LLVMRustLinkage::InternalLinkage:
1230     return LLVMInternalLinkage;
1231   case LLVMRustLinkage::PrivateLinkage:
1232     return LLVMPrivateLinkage;
1233   case LLVMRustLinkage::ExternalWeakLinkage:
1234     return LLVMExternalWeakLinkage;
1235   case LLVMRustLinkage::CommonLinkage:
1236     return LLVMCommonLinkage;
1237   }
1238   llvm_unreachable("Invalid LLVMRustLinkage value!");
1239 }
1240
1241 extern "C" LLVMRustLinkage LLVMRustGetLinkage(LLVMValueRef V) {
1242   return toRust(LLVMGetLinkage(V));
1243 }
1244
1245 extern "C" void LLVMRustSetLinkage(LLVMValueRef V,
1246                                    LLVMRustLinkage RustLinkage) {
1247   LLVMSetLinkage(V, fromRust(RustLinkage));
1248 }
1249
1250 // Returns true if both high and low were successfully set. Fails in case constant wasn’t any of
1251 // the common sizes (1, 8, 16, 32, 64, 128 bits)
1252 extern "C" bool LLVMRustConstInt128Get(LLVMValueRef CV, bool sext, uint64_t *high, uint64_t *low)
1253 {
1254     auto C = unwrap<llvm::ConstantInt>(CV);
1255     if (C->getBitWidth() > 128) { return false; }
1256     APInt AP;
1257     if (sext) {
1258         AP = C->getValue().sextOrSelf(128);
1259     } else {
1260         AP = C->getValue().zextOrSelf(128);
1261     }
1262     *low = AP.getLoBits(64).getZExtValue();
1263     *high = AP.getHiBits(64).getZExtValue();
1264     return true;
1265 }
1266
1267 extern "C" LLVMContextRef LLVMRustGetValueContext(LLVMValueRef V) {
1268   return wrap(&unwrap(V)->getContext());
1269 }
1270
1271 enum class LLVMRustVisibility {
1272   Default = 0,
1273   Hidden = 1,
1274   Protected = 2,
1275 };
1276
1277 static LLVMRustVisibility toRust(LLVMVisibility Vis) {
1278   switch (Vis) {
1279   case LLVMDefaultVisibility:
1280     return LLVMRustVisibility::Default;
1281   case LLVMHiddenVisibility:
1282     return LLVMRustVisibility::Hidden;
1283   case LLVMProtectedVisibility:
1284     return LLVMRustVisibility::Protected;
1285   }
1286   llvm_unreachable("Invalid LLVMRustVisibility value!");
1287 }
1288
1289 static LLVMVisibility fromRust(LLVMRustVisibility Vis) {
1290   switch (Vis) {
1291   case LLVMRustVisibility::Default:
1292     return LLVMDefaultVisibility;
1293   case LLVMRustVisibility::Hidden:
1294     return LLVMHiddenVisibility;
1295   case LLVMRustVisibility::Protected:
1296     return LLVMProtectedVisibility;
1297   }
1298   llvm_unreachable("Invalid LLVMRustVisibility value!");
1299 }
1300
1301 extern "C" LLVMRustVisibility LLVMRustGetVisibility(LLVMValueRef V) {
1302   return toRust(LLVMGetVisibility(V));
1303 }
1304
1305 extern "C" void LLVMRustSetVisibility(LLVMValueRef V,
1306                                       LLVMRustVisibility RustVisibility) {
1307   LLVMSetVisibility(V, fromRust(RustVisibility));
1308 }