1 // Copyright 2012-2015 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.
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.
11 #![allow(non_upper_case_globals)]
12 #![allow(non_camel_case_types)]
13 #![allow(non_snake_case)]
16 #![doc(html_logo_url = "https://www.rust-lang.org/logos/rust-logo-128x128-blk-v2.png",
17 html_favicon_url = "https://doc.rust-lang.org/favicon.ico",
18 html_root_url = "https://doc.rust-lang.org/nightly/")]
21 #![feature(box_syntax)]
22 #![feature(concat_idents)]
24 #![feature(link_args)]
25 #![feature(static_nobundle)]
27 // See librustc_cratesio_shim/Cargo.toml for a comment explaining this.
28 #[allow(unused_extern_crates)]
29 extern crate rustc_cratesio_shim;
32 extern crate bitflags;
35 pub use self::IntPredicate::*;
36 pub use self::RealPredicate::*;
37 pub use self::TypeKind::*;
38 pub use self::AtomicRmwBinOp::*;
39 pub use self::MetadataType::*;
40 pub use self::CodeGenOptSize::*;
41 pub use self::CallConv::*;
42 pub use self::Linkage::*;
44 use std::str::FromStr;
46 use std::ffi::{CString, CStr};
47 use std::cell::RefCell;
48 use libc::{c_uint, c_char, size_t};
57 pub fn into_result(self) -> Result<(), ()> {
59 LLVMRustResult::Success => Ok(()),
60 LLVMRustResult::Failure => Err(()),
65 pub fn AddFunctionAttrStringValue(llfn: ValueRef,
70 LLVMRustAddFunctionAttrStringValue(llfn,
77 #[derive(Copy, Clone)]
78 pub enum AttributePlace {
85 pub fn as_uint(self) -> c_uint {
87 AttributePlace::ReturnValue => 0,
88 AttributePlace::Argument(i) => 1 + i,
89 AttributePlace::Function => !0,
94 #[derive(Copy, Clone, PartialEq)]
96 pub enum CodeGenOptSize {
97 CodeGenOptSizeNone = 0,
98 CodeGenOptSizeDefault = 1,
99 CodeGenOptSizeAggressive = 2,
102 impl FromStr for ArchiveKind {
105 fn from_str(s: &str) -> Result<Self, Self::Err> {
107 "gnu" => Ok(ArchiveKind::K_GNU),
108 "mips64" => Ok(ArchiveKind::K_MIPS64),
109 "bsd" => Ok(ArchiveKind::K_BSD),
110 "coff" => Ok(ArchiveKind::K_COFF),
116 #[allow(missing_copy_implementations)]
117 pub enum RustString_opaque {}
118 type RustStringRef = *mut RustString_opaque;
119 type RustStringRepr = *mut RefCell<Vec<u8>>;
121 /// Appending to a Rust string -- used by RawRustStringOstream.
123 pub unsafe extern "C" fn LLVMRustStringWriteImpl(sr: RustStringRef,
126 let slice = slice::from_raw_parts(ptr as *const u8, size as usize);
128 let sr = sr as RustStringRepr;
129 (*sr).borrow_mut().extend_from_slice(slice);
132 pub fn SetInstructionCallConv(instr: ValueRef, cc: CallConv) {
134 LLVMSetInstructionCallConv(instr, cc as c_uint);
137 pub fn SetFunctionCallConv(fn_: ValueRef, cc: CallConv) {
139 LLVMSetFunctionCallConv(fn_, cc as c_uint);
143 // Externally visible symbols that might appear in multiple translation units need to appear in
144 // their own comdat section so that the duplicates can be discarded at link time. This can for
145 // example happen for generics when using multiple codegen units. This function simply uses the
146 // value's name as the comdat value to make sure that it is in a 1-to-1 relationship to the
148 // For more details on COMDAT sections see e.g. http://www.airs.com/blog/archives/52
149 pub fn SetUniqueComdat(llmod: ModuleRef, val: ValueRef) {
151 LLVMRustSetComdat(llmod, val, LLVMGetValueName(val));
155 pub fn UnsetComdat(val: ValueRef) {
157 LLVMRustUnsetComdat(val);
161 pub fn SetUnnamedAddr(global: ValueRef, unnamed: bool) {
163 LLVMSetUnnamedAddr(global, unnamed as Bool);
167 pub fn set_thread_local(global: ValueRef, is_thread_local: bool) {
169 LLVMSetThreadLocal(global, is_thread_local as Bool);
172 pub fn set_thread_local_mode(global: ValueRef, mode: ThreadLocalMode) {
174 LLVMSetThreadLocalMode(global, mode);
179 pub fn apply_llfn(&self, idx: AttributePlace, llfn: ValueRef) {
180 unsafe { LLVMRustAddFunctionAttribute(llfn, idx.as_uint(), *self) }
183 pub fn apply_callsite(&self, idx: AttributePlace, callsite: ValueRef) {
184 unsafe { LLVMRustAddCallSiteAttribute(callsite, idx.as_uint(), *self) }
187 pub fn unapply_llfn(&self, idx: AttributePlace, llfn: ValueRef) {
188 unsafe { LLVMRustRemoveFunctionAttributes(llfn, idx.as_uint(), *self) }
191 pub fn toggle_llfn(&self, idx: AttributePlace, llfn: ValueRef, set: bool) {
193 self.apply_llfn(idx, llfn);
195 self.unapply_llfn(idx, llfn);
200 // Memory-managed interface to target data.
206 impl Drop for TargetData {
209 LLVMDisposeTargetData(self.lltd);
214 fn mk_target_data(string_rep: &str) -> TargetData {
215 let string_rep = CString::new(string_rep).unwrap();
216 TargetData { lltd: unsafe { LLVMCreateTargetData(string_rep.as_ptr()) } }
219 // Memory-managed interface to object files.
221 pub struct ObjectFile {
222 pub llof: ObjectFileRef,
226 // This will take ownership of llmb
227 pub fn new(llmb: MemoryBufferRef) -> Option<ObjectFile> {
229 let llof = LLVMCreateObjectFile(llmb);
230 if llof as isize == 0 {
231 // LLVMCreateObjectFile took ownership of llmb
235 Some(ObjectFile { llof: llof })
240 impl Drop for ObjectFile {
243 LLVMDisposeObjectFile(self.llof);
248 // Memory-managed interface to section iterators.
250 pub struct SectionIter {
251 pub llsi: SectionIteratorRef,
254 impl Drop for SectionIter {
257 LLVMDisposeSectionIterator(self.llsi);
262 pub fn mk_section_iter(llof: ObjectFileRef) -> SectionIter {
263 unsafe { SectionIter { llsi: LLVMGetSections(llof) } }
266 /// Safe wrapper around `LLVMGetParam`, because segfaults are no fun.
267 pub fn get_param(llfn: ValueRef, index: c_uint) -> ValueRef {
269 assert!(index < LLVMCountParams(llfn),
270 "out of bounds argument access: {} out of {} arguments", index, LLVMCountParams(llfn));
271 LLVMGetParam(llfn, index)
275 fn get_params(llfn: ValueRef) -> Vec<ValueRef> {
277 let num_params = LLVMCountParams(llfn);
278 let mut params = Vec::with_capacity(num_params as usize);
279 for idx in 0..num_params {
280 params.push(LLVMGetParam(llfn, idx));
287 pub fn build_string<F>(f: F) -> Option<String>
288 where F: FnOnce(RustStringRef)
290 let mut buf = RefCell::new(Vec::new());
291 f(&mut buf as RustStringRepr as RustStringRef);
292 String::from_utf8(buf.into_inner()).ok()
295 pub unsafe fn twine_to_string(tr: TwineRef) -> String {
296 build_string(|s| LLVMRustWriteTwineToString(tr, s)).expect("got a non-UTF8 Twine from LLVM")
299 pub fn initialize_available_targets() {
300 macro_rules! init_target(
301 ($cfg:meta, $($method:ident),*) => { {
316 init_target!(llvm_component = "x86",
317 LLVMInitializeX86TargetInfo,
318 LLVMInitializeX86Target,
319 LLVMInitializeX86TargetMC,
320 LLVMInitializeX86AsmPrinter,
321 LLVMInitializeX86AsmParser);
322 init_target!(llvm_component = "arm",
323 LLVMInitializeARMTargetInfo,
324 LLVMInitializeARMTarget,
325 LLVMInitializeARMTargetMC,
326 LLVMInitializeARMAsmPrinter,
327 LLVMInitializeARMAsmParser);
328 init_target!(llvm_component = "aarch64",
329 LLVMInitializeAArch64TargetInfo,
330 LLVMInitializeAArch64Target,
331 LLVMInitializeAArch64TargetMC,
332 LLVMInitializeAArch64AsmPrinter,
333 LLVMInitializeAArch64AsmParser);
334 init_target!(llvm_component = "mips",
335 LLVMInitializeMipsTargetInfo,
336 LLVMInitializeMipsTarget,
337 LLVMInitializeMipsTargetMC,
338 LLVMInitializeMipsAsmPrinter,
339 LLVMInitializeMipsAsmParser);
340 init_target!(llvm_component = "powerpc",
341 LLVMInitializePowerPCTargetInfo,
342 LLVMInitializePowerPCTarget,
343 LLVMInitializePowerPCTargetMC,
344 LLVMInitializePowerPCAsmPrinter,
345 LLVMInitializePowerPCAsmParser);
346 init_target!(llvm_component = "systemz",
347 LLVMInitializeSystemZTargetInfo,
348 LLVMInitializeSystemZTarget,
349 LLVMInitializeSystemZTargetMC,
350 LLVMInitializeSystemZAsmPrinter,
351 LLVMInitializeSystemZAsmParser);
352 init_target!(llvm_component = "jsbackend",
353 LLVMInitializeJSBackendTargetInfo,
354 LLVMInitializeJSBackendTarget,
355 LLVMInitializeJSBackendTargetMC);
356 init_target!(llvm_component = "msp430",
357 LLVMInitializeMSP430TargetInfo,
358 LLVMInitializeMSP430Target,
359 LLVMInitializeMSP430TargetMC,
360 LLVMInitializeMSP430AsmPrinter);
361 init_target!(llvm_component = "sparc",
362 LLVMInitializeSparcTargetInfo,
363 LLVMInitializeSparcTarget,
364 LLVMInitializeSparcTargetMC,
365 LLVMInitializeSparcAsmPrinter,
366 LLVMInitializeSparcAsmParser);
367 init_target!(llvm_component = "nvptx",
368 LLVMInitializeNVPTXTargetInfo,
369 LLVMInitializeNVPTXTarget,
370 LLVMInitializeNVPTXTargetMC,
371 LLVMInitializeNVPTXAsmPrinter);
372 init_target!(llvm_component = "hexagon",
373 LLVMInitializeHexagonTargetInfo,
374 LLVMInitializeHexagonTarget,
375 LLVMInitializeHexagonTargetMC,
376 LLVMInitializeHexagonAsmPrinter,
377 LLVMInitializeHexagonAsmParser);
378 init_target!(llvm_component = "webassembly",
379 LLVMInitializeWebAssemblyTargetInfo,
380 LLVMInitializeWebAssemblyTarget,
381 LLVMInitializeWebAssemblyTargetMC,
382 LLVMInitializeWebAssemblyAsmPrinter);
385 pub fn last_error() -> Option<String> {
387 let cstr = LLVMRustGetLastError();
391 let err = CStr::from_ptr(cstr).to_bytes();
392 let err = String::from_utf8_lossy(err).to_string();
393 libc::free(cstr as *mut _);
399 pub struct OperandBundleDef {
400 inner: OperandBundleDefRef,
403 impl OperandBundleDef {
404 pub fn new(name: &str, vals: &[ValueRef]) -> OperandBundleDef {
405 let name = CString::new(name).unwrap();
407 LLVMRustBuildOperandBundleDef(name.as_ptr(), vals.as_ptr(), vals.len() as c_uint)
409 OperandBundleDef { inner: def }
412 pub fn raw(&self) -> OperandBundleDefRef {
417 impl Drop for OperandBundleDef {
420 LLVMRustFreeOperandBundleDef(self.inner);