Make ongoing_codegen a query

Author: Zoxc

src/librustc/query/mod.rs

@@ -101,6 +101,12 @@ rustc_queries! {
             desc { "running analysis passes on this crate" }
         }
 
+        query ongoing_codegen(_: CrateNum) -> Result<Lrc<ty::OngoingCodegen>, ErrorReported> {
+            no_hash
+            eval_always
+            desc { "starting code generation" }
+        }
+
         /// Records the type of every item.
         query type_of(key: DefId) -> Ty<'tcx> {
             cache { key.is_local() }

src/librustc/ty/context.rs

@@ -52,7 +52,7 @@ use rustc_data_structures::stable_hasher::{HashStable, hash_stable_hashmap,
                                            StableVec};
 use arena::{TypedArena, SyncDroplessArena};
 use rustc_data_structures::indexed_vec::{Idx, IndexVec};
-use rustc_data_structures::sync::{self, Lrc, Lock, WorkerLocal, AtomicOnce, Once};
+use rustc_data_structures::sync::{self, Lrc, Lock, WorkerLocal, AtomicOnce, Once, OneThread};
 use std::any::Any;
 use std::borrow::Borrow;
 use std::cmp::Ordering;
@@ -1029,6 +1029,9 @@ pub struct GlobalCtxt<'tcx> {
 
     pub sess_rc: Lrc<Session>,
 
+    // This stores a `Arc<dyn CodegenBackend + Send + Sync>`.
+    pub codegen_backend: Box<dyn Any + Send + Sync>,
+
     lowered_hir: AtomicOnce<&'tcx hir::LoweredHir>,
     hir_map: AtomicOnce<&'tcx hir_map::Map<'tcx>>,
 
@@ -1071,6 +1074,7 @@ pub struct GlobalCtxt<'tcx> {
     /// when satisfying the query for a particular codegen unit. Internally in
     /// the query it'll send data along this channel to get processed later.
     pub tx_to_llvm_workers: Lock<mpsc::Sender<Box<dyn Any + Send>>>,
+    pub rx_to_llvm_workers: Steal<OneThread<mpsc::Receiver<Box<dyn Any + Send>>>>,
 }
 
 impl<'a, 'gcx, 'tcx> TyCtxt<'a, 'gcx, 'tcx> {
@@ -1224,7 +1228,7 @@ impl<'a, 'gcx, 'tcx> TyCtxt<'a, 'gcx, 'tcx> {
         extern_providers: ty::query::Providers<'tcx>,
         arenas: &'tcx AllArenas<'tcx>,
         crate_name: Option<String>,
-        tx: mpsc::Sender<Box<dyn Any + Send>>,
+        codegen_backend: Box<dyn Any + Send + Sync>,
         io: InputsAndOutputs,
     ) -> GlobalCtxt<'tcx> {
         let data_layout = TargetDataLayout::parse(&s.target.target).unwrap_or_else(|err| {
@@ -1235,12 +1239,14 @@ impl<'a, 'gcx, 'tcx> TyCtxt<'a, 'gcx, 'tcx> {
         let max_cnum = cstore.crates_untracked().iter().map(|c| c.as_usize()).max().unwrap_or(0);
         let mut providers = IndexVec::from_elem_n(extern_providers, max_cnum + 1);
         providers[LOCAL_CRATE] = local_providers;
+        let (tx, rx) = mpsc::channel();
 
         GlobalCtxt {
             sess: &**s,
             cstore,
             cstore_rc,
             sess_rc: s.clone(),
+            codegen_backend,
             global_arenas: &arenas.global,
             global_interners: interners,
             dep_graph: AtomicOnce::new(),
@@ -1264,6 +1270,7 @@ impl<'a, 'gcx, 'tcx> TyCtxt<'a, 'gcx, 'tcx> {
             allocation_interner: Default::default(),
             alloc_map: Lock::new(interpret::AllocMap::new()),
             tx_to_llvm_workers: Lock::new(tx),
+            rx_to_llvm_workers: Steal::new(OneThread::new(rx)),
             io,
         }
     }

src/librustc/ty/mod.rs

@@ -21,12 +21,15 @@ use crate::mir::Mir;
 use crate::mir::interpret::{GlobalId, ErrorHandled};
 use crate::mir::GeneratorLayout;
 use crate::session::CrateDisambiguator;
+use crate::session::config::OutputFilenames;
+use crate::dep_graph::DepGraph;
 use crate::traits::{self, Reveal};
 use crate::ty;
 use crate::ty::layout::VariantIdx;
 use crate::ty::subst::{Subst, InternalSubsts, SubstsRef};
 use crate::ty::util::{IntTypeExt, Discr};
 use crate::ty::walk::TypeWalker;
+use crate::ty::steal::Steal;
 use crate::util::captures::Captures;
 use crate::util::nodemap::{NodeSet, DefIdMap, FxHashMap};
 use arena::SyncDroplessArena;
@@ -39,6 +42,7 @@ use std::fmt;
 use std::hash::{Hash, Hasher};
 use std::ops::Deref;
 use std::any::Any;
+use std::sync::Arc;
 use rustc_data_structures::sync::{self, Lrc, OneThread, ParallelIterator, par_iter};
 use std::slice;
 use std::{mem, ptr};
@@ -120,6 +124,12 @@ mod sty;
 
 // Data types
 
+pub struct OngoingCodegen {
+    pub outputs: Arc<OutputFilenames>,
+    pub dep_graph: DepGraph,
+    pub codegen_object: Steal<OneThread<Box<dyn Any>>>,
+}
+
 pub struct PluginInfo {
     pub syntax_exts: Vec<NamedSyntaxExtension>,
     pub attributes: Vec<(String, AttributeType)>,

src/librustc_codegen_llvm/lib.rs

@@ -222,8 +222,8 @@ unsafe impl Send for LlvmCodegenBackend {} // Llvm is on a per-thread basis
 unsafe impl Sync for LlvmCodegenBackend {}
 
 impl LlvmCodegenBackend {
-    pub fn new() -> Box<dyn CodegenBackend> {
-        box LlvmCodegenBackend(())
+    pub fn new() -> Arc<dyn CodegenBackend + Send + Sync> {
+        Arc::new(LlvmCodegenBackend(()))
     }
 }
 
@@ -347,7 +347,7 @@ impl CodegenBackend for LlvmCodegenBackend {
 
 /// This is the entrypoint for a hot plugged rustc_codegen_llvm
 #[no_mangle]
-pub fn __rustc_codegen_backend() -> Box<dyn CodegenBackend> {
+pub fn __rustc_codegen_backend() -> Arc<dyn CodegenBackend + Send + Sync> {
     LlvmCodegenBackend::new()
 }
 

src/librustc_driver/lib.rs

@@ -177,7 +177,7 @@ pub fn run_compiler(
                             return;
                         }
                         let should_stop = RustcDefaultCalls::print_crate_info(
-                            &***compiler.codegen_backend(),
+                            &**compiler.codegen_backend(),
                             compiler.session(),
                             None,
                             &odir,
@@ -231,7 +231,7 @@ pub fn run_compiler(
     interface::run_compiler(config, |compiler| {
         let sess = compiler.session();
         let should_stop = RustcDefaultCalls::print_crate_info(
-            &***compiler.codegen_backend(),
+            &**compiler.codegen_backend(),
             sess,
             Some(compiler.input()),
             compiler.output_dir(),

src/librustc_interface/interface.rs

@@ -27,7 +27,7 @@ pub type Result<T> = result::Result<T, ErrorReported>;
 /// Created by passing `Config` to `run_compiler`.
 pub struct Compiler {
     pub(crate) sess: Lrc<Session>,
-    codegen_backend: Lrc<Box<dyn CodegenBackend>>,
+    codegen_backend: Arc<dyn CodegenBackend + Send + Sync>,
     source_map: Lrc<SourceMap>,
     pub(crate) io: InputsAndOutputs,
     pub(crate) queries: Queries,
@@ -39,7 +39,7 @@ impl Compiler {
     pub fn session(&self) -> &Lrc<Session> {
         &self.sess
     }
-    pub fn codegen_backend(&self) -> &Lrc<Box<dyn CodegenBackend>> {
+    pub fn codegen_backend(&self) -> &Arc<dyn CodegenBackend + Send + Sync> {
         &self.codegen_backend
     }
     pub fn cstore(&self) -> &Lrc<CStore> {

src/librustc_interface/passes.rs

@@ -911,6 +911,7 @@ fn load_query_result_cache<'tcx>(
 }
 
 pub fn default_provide(providers: &mut ty::query::Providers<'_>) {
+    providers.ongoing_codegen = ongoing_codegen;
     providers.analysis = analysis;
     providers.hir_map = hir_map;
     providers.lower_ast_to_hir = lower_ast_to_hir;
@@ -968,7 +969,6 @@ impl BoxedGlobalCtxt {
 pub fn create_global_ctxt(
     compiler: &Compiler,
     io: InputsAndOutputs,
-    tx: mpsc::Sender<Box<dyn Any + Send>>,
 ) -> BoxedGlobalCtxt {
     let sess = compiler.session().clone();
     let cstore = compiler.cstore.clone();
@@ -990,6 +990,13 @@ pub fn create_global_ctxt(
         default_provide_extern(&mut extern_providers);
         codegen_backend.provide_extern(&mut extern_providers);
 
+        // Move the dyn Any coercion outside the generator to avoid lifetime issues
+        fn codegen_backend_any(
+            i: Arc<dyn CodegenBackend + Send + Sync>
+        ) -> Box<dyn Any + Send + Sync> {
+            Box::new(i)
+        }
+
         let gcx = TyCtxt::create_global_ctxt(
             sess,
             &**cstore,
@@ -998,7 +1005,7 @@ pub fn create_global_ctxt(
             extern_providers,
             &arenas,
             crate_name,
-            tx,
+            codegen_backend_any(codegen_backend.clone()),
             io,
         );
 
@@ -1164,12 +1171,23 @@ fn analysis<'tcx>(
 
 /// Runs the codegen backend, after which the AST and analysis can
 /// be discarded.
-pub fn start_codegen<'tcx>(
-    codegen_backend: &dyn CodegenBackend,
+fn ongoing_codegen<'tcx>(
     tcx: TyCtxt<'_, 'tcx, 'tcx>,
-    rx: mpsc::Receiver<Box<dyn Any + Send>>,
-    outputs: &OutputFilenames,
-) -> Box<dyn Any> {
+    cnum: CrateNum,
+) -> Result<Lrc<ty::OngoingCodegen>> {
+    tcx.analysis(cnum)?;
+
+    assert_eq!(cnum, LOCAL_CRATE);
+    // Don't do code generation if there were any errors
+    tcx.sess.compile_status()?;
+
+    let outputs = tcx.prepare_outputs(LocalCrate)?;
+
+    let rx = OneThread::into_inner(tcx.rx_to_llvm_workers.steal());
+    let codegen_backend: &dyn Any = &*tcx.codegen_backend;
+    let codegen_backend = codegen_backend.downcast_ref::<Arc<dyn CodegenBackend + Send + Sync>>()
+                                         .unwrap();
+
     if log_enabled!(::log::Level::Info) {
         println!("Pre-codegen");
         tcx.print_debug_stats();
@@ -1189,11 +1207,15 @@ pub fn start_codegen<'tcx>(
     }
 
     if tcx.sess.opts.output_types.contains_key(&OutputType::Mir) {
-        if let Err(e) = mir::transform::dump_mir::emit_mir(tcx, outputs) {
+        if let Err(e) = mir::transform::dump_mir::emit_mir(tcx, &outputs) {
             tcx.sess.err(&format!("could not emit MIR: {}", e));
             tcx.sess.abort_if_errors();
         }
     }
 
-    codegen
+    Ok(Lrc::new(ty::OngoingCodegen {
+        outputs,
+        dep_graph: tcx.dep_graph().clone(),
+        codegen_object: Steal::new(OneThread::new(codegen)),
+    }))
 }

src/librustc_interface/queries.rs

@@ -1,7 +1,7 @@
 use crate::interface::{Compiler, Result};
 use crate::passes::{self, BoxedResolver, ExpansionResult, BoxedGlobalCtxt};
 
-use rustc_data_structures::sync::{Lrc, Lock};
+use rustc_data_structures::sync::{Lrc, Lock, OneThread};
 use rustc::session::config::{Input, OutputFilenames, OutputType};
 use rustc::session::Session;
 use rustc::util::common::{time, ErrorReported};
@@ -83,67 +83,40 @@ impl<T> Default for Query<T> {
 
 #[derive(Default)]
 pub(crate) struct Queries {
-    codegen_channel: Query<(Steal<mpsc::Sender<Box<dyn Any + Send>>>,
-                            Steal<mpsc::Receiver<Box<dyn Any + Send>>>)>,
     global_ctxt: Query<BoxedGlobalCtxt>,
-    ongoing_codegen: Query<(Box<dyn Any>, Arc<OutputFilenames>, DepGraph)>,
+    ongoing_codegen: Query<Lrc<ty::OngoingCodegen>>,
     link: Query<()>,
 }
 
 impl Compiler {
-    pub fn codegen_channel(&self) -> Result<&Query<(Steal<mpsc::Sender<Box<dyn Any + Send>>>,
-                                                    Steal<mpsc::Receiver<Box<dyn Any + Send>>>)>> {
-        self.queries.codegen_channel.compute(|| {
-            let (tx, rx) = mpsc::channel();
-            Ok((Steal::new(tx), Steal::new(rx)))
-        })
-    }
-
     pub fn global_ctxt(&self) -> Result<&Query<BoxedGlobalCtxt>> {
         self.queries.global_ctxt.compute(|| {
-            let tx = self.codegen_channel()?.peek().0.steal();
             Ok(passes::create_global_ctxt(
                 self,
                 self.io.clone(),
-                tx
             ))
         })
     }
 
     pub fn ongoing_codegen(
         &self
-    ) -> Result<&Query<(Box<dyn Any>, Arc<OutputFilenames>, DepGraph)>> {
+    ) -> Result<&Query<Lrc<ty::OngoingCodegen>>> {
         self.queries.ongoing_codegen.compute(|| {
-            let rx = self.codegen_channel()?.peek().1.steal();
             self.global_ctxt()?.peek_mut().enter(|tcx| {
-                tcx.analysis(LOCAL_CRATE).ok();
-
-                // Don't do code generation if there were any errors
-                self.session().compile_status()?;
-
-                let outputs = tcx.prepare_outputs(LocalCrate)?;
-
-                Ok((passes::start_codegen(
-                    &***self.codegen_backend(),
-                    tcx,
-                    rx,
-                    &outputs,
-                ), outputs, tcx.dep_graph().clone()))
+                tcx.ongoing_codegen(LOCAL_CRATE)
             })
         })
     }
 
     pub fn link(&self) -> Result<&Query<()>> {
         self.queries.link.compute(|| {
-            let sess = self.session();
-
-            let (ongoing_codegen, outputs, dep_graph) = self.ongoing_codegen()?.take();
+            let ongoing_codegen = self.ongoing_codegen()?.take();
 
             self.codegen_backend().join_codegen_and_link(
-                ongoing_codegen,
-                sess,
-                &dep_graph,
-                &outputs,
+                OneThread::into_inner(ongoing_codegen.codegen_object.steal()),
+                self.session(),
+                &ongoing_codegen.dep_graph,
+                &ongoing_codegen.outputs,
             ).map_err(|_| ErrorReported)?;
 
             Ok(())

src/librustc_interface/util.rs

@@ -88,7 +88,7 @@ pub fn create_session(
     file_loader: Option<Box<dyn FileLoader + Send + Sync + 'static>>,
     input_path: Option<PathBuf>,
     lint_caps: FxHashMap<lint::LintId, lint::Level>,
-) -> (Lrc<Session>, Lrc<Box<dyn CodegenBackend>>, Lrc<SourceMap>) {
+) -> (Lrc<Session>, Arc<dyn CodegenBackend + Send + Sync>, Lrc<SourceMap>) {
     let descriptions = diagnostics_registry();
 
     let loader = file_loader.unwrap_or(box RealFileLoader);
@@ -116,7 +116,7 @@ pub fn create_session(
     add_configuration(&mut cfg, &sess, &*codegen_backend);
     sess.parse_sess.config = cfg;
 
-    (Lrc::new(sess), Lrc::new(codegen_backend), source_map)
+    (Lrc::new(sess), codegen_backend, source_map)
 }
 
 // Temporarily have stack size set to 32MB to deal with various crates with long method
@@ -239,7 +239,7 @@ pub fn spawn_thread_pool<F: FnOnce() -> R + Send, R: Send>(
     })
 }
 
-fn load_backend_from_dylib(path: &Path) -> fn() -> Box<dyn CodegenBackend> {
+fn load_backend_from_dylib(path: &Path) -> fn() -> Arc<dyn CodegenBackend + Send + Sync> {
     let lib = DynamicLibrary::open(Some(path)).unwrap_or_else(|err| {
         let err = format!("couldn't load codegen backend {:?}: {:?}", path, err);
         early_error(ErrorOutputType::default(), &err);
@@ -260,10 +260,10 @@ fn load_backend_from_dylib(path: &Path) -> fn() -> Box<dyn CodegenBackend> {
     }
 }
 
-pub fn get_codegen_backend(sess: &Session) -> Box<dyn CodegenBackend> {
+pub fn get_codegen_backend(sess: &Session) -> Arc<dyn CodegenBackend + Send + Sync> {
     static INIT: Once = Once::new();
 
-    static mut LOAD: fn() -> Box<dyn CodegenBackend> = || unreachable!();
+    static mut LOAD: fn() -> Arc<dyn CodegenBackend + Send + Sync> = || unreachable!();
 
     INIT.call_once(|| {
         let codegen_name = sess.opts.debugging_opts.codegen_backend.as_ref()
@@ -284,7 +284,7 @@ pub fn get_codegen_backend(sess: &Session) -> Box<dyn CodegenBackend> {
     backend
 }
 
-pub fn get_codegen_sysroot(backend_name: &str) -> fn() -> Box<dyn CodegenBackend> {
+pub fn get_codegen_sysroot(backend_name: &str) -> fn() -> Arc<dyn CodegenBackend + Send + Sync> {
     // For now we only allow this function to be called once as it'll dlopen a
     // few things, which seems to work best if we only do that once. In
     // general this assertion never trips due to the once guard in `get_codegen_backend`,

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

@@ -95,6 +95,6 @@ impl CodegenBackend for TheBackend {
 
 /// This is the entrypoint for a hot plugged rustc_codegen_llvm
 #[no_mangle]
-pub fn __rustc_codegen_backend() -> Box<CodegenBackend> {
-    Box::new(TheBackend)
+pub fn __rustc_codegen_backend() -> Arc<dyn CodegenBackend + Send + Sync> {
+    Arc::new(TheBackend)
 }