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:
Tomasz Miąsko 2020-11-20 00:00:00 +00:00
parent fe982319aa
commit c2fb99984c
4 changed files with 45 additions and 4 deletions

View File

@ -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 { pub enum InlineAttr {
None, None,
Hint, Hint,

View File

@ -25,7 +25,7 @@
/// Mark LLVM function to use provided inline heuristic. /// Mark LLVM function to use provided inline heuristic.
#[inline] #[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::*; use self::InlineAttr::*;
match inline { match inline {
Hint => Attribute::InlineHint.apply_llfn(Function, val), 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); Attribute::NoInline.apply_llfn(Function, val);
} }
} }
None if requires_inline => Attribute::InlineHint.apply_llfn(Function, val),
None => {} 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: // The `uwtable` attribute according to LLVM is:
// //

View File

@ -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 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) { if codegen_fn_attrs.flags.contains(CodegenFnAttrFlags::COLD) {
debug!("#[cold] present - not inlining"); debug!("#[cold] present - not inlining");
return false; return false;

View 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{{.*}} }