4 use rustc::session::config;
6 use rustc::middle::cstore::{EncodedMetadata, MetadataLoader};
7 use rustc_codegen_ssa::METADATA_FILENAME;
8 use rustc_data_structures::owning_ref::{self, OwningRef};
9 use rustc_data_structures::rustc_erase_owner;
10 use rustc_target::spec::Target;
12 pub struct CraneliftMetadataLoader;
14 impl MetadataLoader for CraneliftMetadataLoader {
19 ) -> Result<owning_ref::ErasedBoxRef<[u8]>, String> {
20 let mut archive = ar::Archive::new(File::open(path).map_err(|e| format!("{:?}", e))?);
21 // Iterate over all entries in the archive:
22 while let Some(entry_result) = archive.next_entry() {
23 let mut entry = entry_result.map_err(|e| format!("{:?}", e))?;
24 if entry.header().identifier() == METADATA_FILENAME.as_bytes() {
25 let mut buf = Vec::new();
26 ::std::io::copy(&mut entry, &mut buf).map_err(|e| format!("{:?}", e))?;
27 let buf: OwningRef<Vec<u8>, [u8]> = OwningRef::new(buf).into();
28 return Ok(rustc_erase_owner!(buf.map_owner_box()));
32 Err("couldn't find metadata entry".to_string())
33 //self.get_dylib_metadata(target, path)
36 fn get_dylib_metadata(
40 ) -> Result<owning_ref::ErasedBoxRef<[u8]>, String> {
42 let file = std::fs::read(path).map_err(|e| format!("read:{:?}", e))?;
43 let file = object::File::parse(&file).map_err(|e| format!("parse: {:?}", e))?;
44 let buf = file.section_data_by_name(".rustc").ok_or("no .rustc section")?.into_owned();
45 let buf: OwningRef<Vec<u8>, [u8]> = OwningRef::new(buf).into();
46 Ok(rustc_erase_owner!(buf.map_owner_box()))
50 // Adapted from https://github.com/rust-lang/rust/blob/da573206f87b5510de4b0ee1a9c044127e409bd3/src/librustc_codegen_llvm/base.rs#L47-L112
51 pub fn write_metadata<'a, 'gcx>(
52 tcx: TyCtxt<'gcx, 'gcx>,
53 artifact: &mut faerie::Artifact
54 ) -> EncodedMetadata {
56 use flate2::Compression;
57 use flate2::write::DeflateEncoder;
59 #[derive(PartialEq, Eq, PartialOrd, Ord)]
66 let kind = tcx.sess.crate_types.borrow().iter().map(|ty| {
68 config::CrateType::Executable |
69 config::CrateType::Staticlib |
70 config::CrateType::Cdylib => MetadataKind::None,
72 config::CrateType::Rlib => MetadataKind::Uncompressed,
74 config::CrateType::Dylib |
75 config::CrateType::ProcMacro => MetadataKind::Compressed,
77 }).max().unwrap_or(MetadataKind::None);
79 if kind == MetadataKind::None {
80 return EncodedMetadata::new();
83 let metadata = tcx.encode_metadata();
84 if kind == MetadataKind::Uncompressed {
88 assert!(kind == MetadataKind::Compressed);
89 let mut compressed = tcx.metadata_encoding_version();
90 DeflateEncoder::new(&mut compressed, Compression::fast())
91 .write_all(&metadata.raw_data).unwrap();
93 artifact.declare(".rustc", faerie::Decl::section(faerie::SectionKind::Data)).unwrap();
94 artifact.define_with_symbols(".rustc", compressed, {
95 let mut map = std::collections::BTreeMap::new();
96 // FIXME implement faerie elf backend section custom symbols
97 // For MachO this is necessary to prevent the linker from throwing away the .rustc section,
98 // but for ELF it isn't.
99 if tcx.sess.target.target.options.is_like_osx {
100 map.insert(rustc::middle::exported_symbols::metadata_symbol_name(tcx), 0);