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 {
|
pub enum InlineAttr {
|
||||||
None,
|
None,
|
||||||
Hint,
|
Hint,
|
||||||
|
@ -25,7 +25,7 @@ use crate::value::Value;
|
|||||||
|
|
||||||
/// 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:
|
||||||
//
|
//
|
||||||
|
@ -254,6 +254,11 @@ impl Inliner<'tcx> {
|
|||||||
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;
|
||||||
|
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