Skip to content

Commit

Permalink
Auto merge of #59577 - dlrobertson:fix_58881, r=nagisa
Browse files Browse the repository at this point in the history
Fix LLVM IR generated for C-variadic arguments

It is possible to create malformed LLVM IR given variadic arguments that
are aggregate types. This occurs due to improper tracking of the current
argument in the functions list of arguments.

Fixes: #58881
  • Loading branch information
bors committed Mar 31, 2019
2 parents 4fac5c9 + a9d62be commit e3428db
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 13 deletions.
15 changes: 2 additions & 13 deletions src/librustc_codegen_ssa/mir/block.rs
Expand Up @@ -20,7 +20,7 @@ use syntax_pos::Pos;

use super::{FunctionCx, LocalRef};
use super::place::PlaceRef;
use super::operand::{OperandValue, OperandRef};
use super::operand::OperandRef;
use super::operand::OperandValue::{Pair, Ref, Immediate};

/// Used by `FunctionCx::codegen_terminator` for emitting common patterns
Expand Down Expand Up @@ -695,18 +695,7 @@ impl<'a, 'tcx: 'a, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> {
// an "spoofed" `VaList`. This argument is ignored, but we need to
// populate it with a dummy operand so that the users real arguments
// are not overwritten.
let i = if sig.c_variadic && last_arg_idx.map(|x| x == i).unwrap_or(false) {
let layout = match self.cx.tcx().lang_items().va_list() {
Some(did) => bx.cx().layout_of(bx.tcx().type_of(did)),
None => bug!("`va_list` language item required for C-variadics"),
};
let op = OperandRef {
val: OperandValue::Immediate(
bx.cx().const_undef(bx.cx().immediate_backend_type(layout)
)),
layout: layout,
};
self.codegen_argument(&mut bx, op, &mut llargs, &fn_ty.args[i]);
let i = if sig.c_variadic && last_arg_idx.map(|x| i >= x).unwrap_or(false) {
if i + 1 < fn_ty.args.len() {
i + 1
} else {
Expand Down
21 changes: 21 additions & 0 deletions src/test/codegen/issue-58881.rs
@@ -0,0 +1,21 @@
// compile-flags: -C no-prepopulate-passes
//
// only-x86_64
// ignore-windows

#![crate_type = "lib"]

extern "C" {
fn variadic_fn(_: i32, ...);
}

#[repr(C)]
struct Foo(u8);
#[repr(C)]
struct Bar(u64, u64, u64);

// Ensure that emit arguments of the correct type.
pub unsafe fn test_call_variadic() {
// CHECK: call void (i32, ...) @variadic_fn(i32 0, i8 {{.*}}, %Bar* {{.*}})
variadic_fn(0, Foo(0), Bar(0, 0, 0))
}

0 comments on commit e3428db

Please sign in to comment.