6 use rustc_ast::expand::allocator::{AllocatorKind, AllocatorTy, ALLOCATOR_METHODS};
7 use rustc_session::config::OomStrategy;
8 use rustc_span::symbol::sym;
10 /// Returns whether an allocator shim was created
11 pub(crate) fn codegen(
13 module: &mut impl Module,
14 unwind_context: &mut UnwindContext,
16 let any_dynamic_crate = tcx.dependency_formats(()).iter().any(|(_, list)| {
17 use rustc_middle::middle::dependency_format::Linkage;
18 list.iter().any(|&linkage| linkage == Linkage::Dynamic)
20 if any_dynamic_crate {
22 } else if let Some(kind) = tcx.allocator_kind(()) {
27 tcx.alloc_error_handler_kind(()).unwrap(),
28 tcx.sess.opts.unstable_opts.oom,
37 module: &mut impl Module,
38 unwind_context: &mut UnwindContext,
40 alloc_error_handler_kind: AllocatorKind,
41 oom_strategy: OomStrategy,
43 let usize_ty = module.target_config().pointer_type();
45 for method in ALLOCATOR_METHODS {
46 let mut arg_tys = Vec::with_capacity(method.inputs.len());
47 for ty in method.inputs.iter() {
49 AllocatorTy::Layout => {
50 arg_tys.push(usize_ty); // size
51 arg_tys.push(usize_ty); // align
53 AllocatorTy::Ptr => arg_tys.push(usize_ty),
54 AllocatorTy::Usize => arg_tys.push(usize_ty),
56 AllocatorTy::ResultPtr | AllocatorTy::Unit => panic!("invalid allocator arg"),
59 let output = match method.output {
60 AllocatorTy::ResultPtr => Some(usize_ty),
61 AllocatorTy::Unit => None,
63 AllocatorTy::Layout | AllocatorTy::Usize | AllocatorTy::Ptr => {
64 panic!("invalid allocator output")
69 call_conv: module.target_config().default_call_conv,
70 params: arg_tys.iter().cloned().map(AbiParam::new).collect(),
71 returns: output.into_iter().map(AbiParam::new).collect(),
73 crate::common::create_wrapper_function(
77 &format!("__rust_{}", method.name),
78 &kind.fn_name(method.name),
83 call_conv: module.target_config().default_call_conv,
84 params: vec![AbiParam::new(usize_ty), AbiParam::new(usize_ty)],
87 crate::common::create_wrapper_function(
91 "__rust_alloc_error_handler",
92 &alloc_error_handler_kind.fn_name(sym::oom),
95 let data_id = module.declare_data(OomStrategy::SYMBOL, Linkage::Export, false, false).unwrap();
96 let mut data_ctx = DataContext::new();
97 data_ctx.set_align(1);
98 let val = oom_strategy.should_panic();
99 data_ctx.define(Box::new([val]));
100 module.define_data(data_id, &data_ctx).unwrap();