Never inline naked functions
The `#[naked]` attribute disabled prologue / epilogue emission for the function and it is responsibility of a developer to provide them. The compiler is no position to inline such functions correctly. Disable inlining of naked functions at LLVM and MIR level.
This commit is contained in:
parent
fe982319aa
commit
c2fb99984c
@ -67,7 +67,7 @@ fn handle_errors(sess: &ParseSess, span: Span, error: AttrError) {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Encodable, Decodable)]
|
||||
#[derive(Copy, Clone, PartialEq, Encodable, Decodable)]
|
||||
pub enum InlineAttr {
|
||||
None,
|
||||
Hint,
|
||||
|
@ -25,7 +25,7 @@
|
||||
|
||||
/// Mark LLVM function to use provided inline heuristic.
|
||||
#[inline]
|
||||
fn inline(cx: &CodegenCx<'ll, '_>, val: &'ll Value, inline: InlineAttr, requires_inline: bool) {
|
||||
fn inline(cx: &CodegenCx<'ll, '_>, val: &'ll Value, inline: InlineAttr) {
|
||||
use self::InlineAttr::*;
|
||||
match inline {
|
||||
Hint => Attribute::InlineHint.apply_llfn(Function, val),
|
||||
@ -35,7 +35,6 @@ fn inline(cx: &CodegenCx<'ll, '_>, val: &'ll Value, inline: InlineAttr, requires
|
||||
Attribute::NoInline.apply_llfn(Function, val);
|
||||
}
|
||||
}
|
||||
None if requires_inline => Attribute::InlineHint.apply_llfn(Function, val),
|
||||
None => {}
|
||||
};
|
||||
}
|
||||
@ -226,7 +225,14 @@ pub fn from_fn_attrs(cx: &CodegenCx<'ll, 'tcx>, llfn: &'ll Value, instance: ty::
|
||||
}
|
||||
}
|
||||
|
||||
inline(cx, llfn, codegen_fn_attrs.inline.clone(), instance.def.requires_inline(cx.tcx));
|
||||
let inline_attr = if codegen_fn_attrs.flags.contains(CodegenFnAttrFlags::NAKED) {
|
||||
InlineAttr::Never
|
||||
} else if codegen_fn_attrs.inline == InlineAttr::None && instance.def.requires_inline(cx.tcx) {
|
||||
InlineAttr::Hint
|
||||
} else {
|
||||
codegen_fn_attrs.inline
|
||||
};
|
||||
inline(cx, llfn, inline_attr);
|
||||
|
||||
// The `uwtable` attribute according to LLVM is:
|
||||
//
|
||||
|
@ -254,6 +254,11 @@ fn should_inline(&self, callsite: CallSite<'tcx>, callee_body: &Body<'tcx>) -> b
|
||||
self.tcx.sess.opts.debugging_opts.inline_mir_threshold
|
||||
};
|
||||
|
||||
if codegen_fn_attrs.flags.contains(CodegenFnAttrFlags::NAKED) {
|
||||
debug!("#[naked] present - not inlining");
|
||||
return false;
|
||||
}
|
||||
|
||||
if codegen_fn_attrs.flags.contains(CodegenFnAttrFlags::COLD) {
|
||||
debug!("#[cold] present - not inlining");
|
||||
return false;
|
||||
|
30
src/test/codegen/naked-noinline.rs
Normal file
30
src/test/codegen/naked-noinline.rs
Normal file
@ -0,0 +1,30 @@
|
||||
// Checks that naked functions are never inlined.
|
||||
// compile-flags: -O -Zmir-opt-level=2
|
||||
// ignore-wasm32
|
||||
#![crate_type = "lib"]
|
||||
#![feature(asm)]
|
||||
#![feature(naked_functions)]
|
||||
|
||||
#[inline(always)]
|
||||
#[naked]
|
||||
#[no_mangle]
|
||||
pub unsafe extern "C" fn f() {
|
||||
// Check that f has naked and noinline attributes.
|
||||
//
|
||||
// CHECK: define void @f() unnamed_addr [[ATTR:#[0-9]+]]
|
||||
// CHECK-NEXT: start:
|
||||
// CHECK-NEXT: call void asm
|
||||
asm!("", options(noreturn));
|
||||
}
|
||||
|
||||
#[no_mangle]
|
||||
pub unsafe fn g() {
|
||||
// Check that call to f is not inlined.
|
||||
//
|
||||
// CHECK-LABEL: define void @g()
|
||||
// CHECK-NEXT: start:
|
||||
// CHECK-NEXT: call void @f()
|
||||
f();
|
||||
}
|
||||
|
||||
// CHECK: attributes [[ATTR]] = { naked noinline{{.*}} }
|
Loading…
Reference in New Issue
Block a user