src/rustllvm/RustWrapper.cpp

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