src/test/run-make-fulldeps/hotplug_codegen_backend/the_backend.rs

   1 #![feature(rustc_private)]
   2
   3 extern crate rustc_codegen_ssa;
   4 extern crate rustc_errors;
   5 extern crate rustc_middle;
   6 #[macro_use]
   7 extern crate rustc_data_structures;
   8 extern crate rustc_driver;
   9 extern crate rustc_hir;
  10 extern crate rustc_session;
  11 extern crate rustc_span;
  12 extern crate rustc_symbol_mangling;
  13 extern crate rustc_target;
  14
  15 use rustc_codegen_ssa::traits::CodegenBackend;
  16 use rustc_data_structures::owning_ref::OwningRef;
  17 use rustc_data_structures::sync::MetadataRef;
  18 use rustc_errors::ErrorReported;
  19 use rustc_middle::dep_graph::DepGraph;
  20 use rustc_middle::middle::cstore::{EncodedMetadata, MetadataLoader, MetadataLoaderDyn};
  21 use rustc_middle::ty::query::Providers;
  22 use rustc_middle::ty::TyCtxt;
  23 use rustc_session::config::OutputFilenames;
  24 use rustc_session::Session;
  25 use rustc_span::symbol::Symbol;
  26 use rustc_target::spec::Target;
  27 use std::any::Any;
  28 use std::path::Path;
  29
  30 pub struct NoLlvmMetadataLoader;
  31
  32 impl MetadataLoader for NoLlvmMetadataLoader {
  33     fn get_rlib_metadata(&self, _: &Target, filename: &Path) -> Result<MetadataRef, String> {
  34         let buf =
  35             std::fs::read(filename).map_err(|e| format!("metadata file open err: {:?}", e))?;
  36         let buf: OwningRef<Vec<u8>, [u8]> = OwningRef::new(buf);
  37         Ok(rustc_erase_owner!(buf.map_owner_box()))
  38     }
  39
  40     fn get_dylib_metadata(&self, target: &Target, filename: &Path) -> Result<MetadataRef, String> {
  41         self.get_rlib_metadata(target, filename)
  42     }
  43 }
  44
  45 struct TheBackend;
  46
  47 impl CodegenBackend for TheBackend {
  48     fn metadata_loader(&self) -> Box<MetadataLoaderDyn> {
  49         Box::new(NoLlvmMetadataLoader)
  50     }
  51
  52     fn provide(&self, providers: &mut Providers) {
  53         rustc_symbol_mangling::provide(providers);
  54
  55         providers.target_features_whitelist = |tcx, _cnum| {
  56             Default::default() // Just a dummy
  57         };
  58         providers.is_reachable_non_generic = |_tcx, _defid| true;
  59         providers.exported_symbols = |_tcx, _crate| &[];
  60     }
  61
  62     fn provide_extern(&self, providers: &mut Providers) {
  63         providers.is_reachable_non_generic = |_tcx, _defid| true;
  64     }
  65
  66     fn codegen_crate<'a, 'tcx>(
  67         &self,
  68         tcx: TyCtxt<'tcx>,
  69         _metadata: EncodedMetadata,
  70         _need_metadata_module: bool,
  71     ) -> Box<dyn Any> {
  72         use rustc_hir::def_id::LOCAL_CRATE;
  73
  74         Box::new(tcx.crate_name(LOCAL_CRATE) as Symbol)
  75     }
  76
  77     fn join_codegen(
  78         &self,
  79         ongoing_codegen: Box<dyn Any>,
  80         _sess: &Session,
  81         _dep_graph: &DepGraph,
  82     ) -> Result<Box<dyn Any>, ErrorReported> {
  83         let crate_name = ongoing_codegen
  84             .downcast::<Symbol>()
  85             .expect("in join_codegen: ongoing_codegen is not a Symbol");
  86         Ok(crate_name)
  87     }
  88
  89     fn link(
  90         &self,
  91         sess: &Session,
  92         codegen_results: Box<dyn Any>,
  93         outputs: &OutputFilenames,
  94     ) -> Result<(), ErrorReported> {
  95         use rustc_session::{config::CrateType, output::out_filename};
  96         use std::io::Write;
  97         let crate_name =
  98             codegen_results.downcast::<Symbol>().expect("in link: codegen_results is not a Symbol");
  99         for &crate_type in sess.opts.crate_types.iter() {
 100             if crate_type != CrateType::Rlib {
 101                 sess.fatal(&format!("Crate type is {:?}", crate_type));
 102             }
 103             let output_name = out_filename(sess, crate_type, &outputs, &*crate_name.as_str());
 104             let mut out_file = ::std::fs::File::create(output_name).unwrap();
 105             write!(out_file, "This has been \"compiled\" successfully.").unwrap();
 106         }
 107         Ok(())
 108     }
 109 }
 110
 111 /// This is the entrypoint for a hot plugged rustc_codegen_llvm
 112 #[no_mangle]
 113 pub fn __rustc_codegen_backend() -> Box<dyn CodegenBackend> {
 114     Box::new(TheBackend)
 115 }