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