2013-01-25 16:56:56 -06:00
|
|
|
// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
|
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
|
|
|
use lib::llvm::{llvm, TypeRef, ValueRef, Attribute, Void};
|
|
|
|
use middle::trans::base::*;
|
|
|
|
use middle::trans::build::*;
|
|
|
|
use middle::trans::common::*;
|
|
|
|
|
2013-02-25 13:11:21 -06:00
|
|
|
use core::libc::c_uint;
|
|
|
|
use core::option;
|
|
|
|
use core::vec;
|
|
|
|
|
2013-01-30 13:46:19 -06:00
|
|
|
pub trait ABIInfo {
|
2013-01-25 16:56:56 -06:00
|
|
|
fn compute_info(&self,
|
|
|
|
atys: &[TypeRef],
|
|
|
|
rty: TypeRef,
|
|
|
|
ret_def: bool) -> FnType;
|
|
|
|
}
|
|
|
|
|
2013-01-30 13:46:19 -06:00
|
|
|
pub struct LLVMType {
|
2013-01-25 16:56:56 -06:00
|
|
|
cast: bool,
|
|
|
|
ty: TypeRef
|
|
|
|
}
|
|
|
|
|
2013-01-30 13:46:19 -06:00
|
|
|
pub struct FnType {
|
2013-01-25 16:56:56 -06:00
|
|
|
arg_tys: ~[LLVMType],
|
|
|
|
ret_ty: LLVMType,
|
2013-02-25 13:11:21 -06:00
|
|
|
attrs: ~[option::Option<Attribute>],
|
2013-01-25 16:56:56 -06:00
|
|
|
sret: bool
|
|
|
|
}
|
|
|
|
|
2013-01-30 13:46:19 -06:00
|
|
|
pub impl FnType {
|
2013-03-07 16:38:38 -06:00
|
|
|
fn decl_fn(&self, decl: &fn(fnty: TypeRef) -> ValueRef) -> ValueRef {
|
2013-01-25 16:56:56 -06:00
|
|
|
let atys = vec::map(self.arg_tys, |t| t.ty);
|
|
|
|
let rty = self.ret_ty.ty;
|
|
|
|
let fnty = T_fn(atys, rty);
|
|
|
|
let llfn = decl(fnty);
|
|
|
|
|
|
|
|
for vec::eachi(self.attrs) |i, a| {
|
|
|
|
match *a {
|
|
|
|
option::Some(attr) => {
|
|
|
|
unsafe {
|
|
|
|
let llarg = get_param(llfn, i);
|
|
|
|
llvm::LLVMAddAttribute(llarg, attr as c_uint);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => ()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return llfn;
|
|
|
|
}
|
|
|
|
|
|
|
|
fn build_shim_args(&self, bcx: block,
|
|
|
|
arg_tys: &[TypeRef],
|
|
|
|
llargbundle: ValueRef) -> ~[ValueRef] {
|
|
|
|
let mut atys = /*bad*/copy self.arg_tys;
|
|
|
|
let mut attrs = /*bad*/copy self.attrs;
|
|
|
|
|
|
|
|
let mut llargvals = ~[];
|
|
|
|
let mut i = 0u;
|
2013-05-14 04:52:12 -05:00
|
|
|
let n = arg_tys.len();
|
2013-01-25 16:56:56 -06:00
|
|
|
|
|
|
|
if self.sret {
|
|
|
|
let llretptr = GEPi(bcx, llargbundle, [0u, n]);
|
|
|
|
let llretloc = Load(bcx, llretptr);
|
|
|
|
llargvals = ~[llretloc];
|
2013-05-10 04:38:54 -05:00
|
|
|
atys = vec::to_owned(atys.tail());
|
|
|
|
attrs = vec::to_owned(attrs.tail());
|
2013-01-25 16:56:56 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
while i < n {
|
|
|
|
let llargval = if atys[i].cast {
|
|
|
|
let arg_ptr = GEPi(bcx, llargbundle, [0u, i]);
|
|
|
|
let arg_ptr = BitCast(bcx, arg_ptr, T_ptr(atys[i].ty));
|
|
|
|
Load(bcx, arg_ptr)
|
|
|
|
} else if attrs[i].is_some() {
|
|
|
|
GEPi(bcx, llargbundle, [0u, i])
|
|
|
|
} else {
|
|
|
|
load_inbounds(bcx, llargbundle, [0u, i])
|
|
|
|
};
|
|
|
|
llargvals.push(llargval);
|
|
|
|
i += 1u;
|
|
|
|
}
|
|
|
|
|
|
|
|
return llargvals;
|
|
|
|
}
|
|
|
|
|
2013-04-18 17:53:29 -05:00
|
|
|
fn build_shim_ret(&self,
|
|
|
|
bcx: block,
|
|
|
|
arg_tys: &[TypeRef],
|
|
|
|
ret_def: bool,
|
|
|
|
llargbundle: ValueRef,
|
|
|
|
llretval: ValueRef) {
|
2013-01-25 16:56:56 -06:00
|
|
|
for vec::eachi(self.attrs) |i, a| {
|
|
|
|
match *a {
|
2013-02-25 13:11:21 -06:00
|
|
|
option::Some(attr) => {
|
2013-01-25 16:56:56 -06:00
|
|
|
unsafe {
|
2013-04-18 17:53:29 -05:00
|
|
|
llvm::LLVMAddInstrAttribute(llretval,
|
|
|
|
(i + 1u) as c_uint,
|
|
|
|
attr as c_uint);
|
2013-01-25 16:56:56 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => ()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if self.sret || !ret_def {
|
|
|
|
return;
|
|
|
|
}
|
2013-05-14 04:52:12 -05:00
|
|
|
let n = arg_tys.len();
|
2013-01-25 16:56:56 -06:00
|
|
|
// R** llretptr = &args->r;
|
|
|
|
let llretptr = GEPi(bcx, llargbundle, [0u, n]);
|
|
|
|
// R* llretloc = *llretptr; /* (args->r) */
|
|
|
|
let llretloc = Load(bcx, llretptr);
|
|
|
|
if self.ret_ty.cast {
|
|
|
|
let tmp_ptr = BitCast(bcx, llretloc, T_ptr(self.ret_ty.ty));
|
|
|
|
// *args->r = r;
|
|
|
|
Store(bcx, llretval, tmp_ptr);
|
|
|
|
} else {
|
|
|
|
// *args->r = r;
|
|
|
|
Store(bcx, llretval, llretloc);
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2013-04-18 17:53:29 -05:00
|
|
|
fn build_wrap_args(&self,
|
|
|
|
bcx: block,
|
|
|
|
ret_ty: TypeRef,
|
|
|
|
llwrapfn: ValueRef,
|
|
|
|
llargbundle: ValueRef) {
|
2013-01-25 16:56:56 -06:00
|
|
|
let mut atys = /*bad*/copy self.arg_tys;
|
|
|
|
let mut attrs = /*bad*/copy self.attrs;
|
|
|
|
let mut j = 0u;
|
|
|
|
let llretptr = if self.sret {
|
2013-05-10 04:38:54 -05:00
|
|
|
atys = vec::to_owned(atys.tail());
|
|
|
|
attrs = vec::to_owned(attrs.tail());
|
2013-01-25 16:56:56 -06:00
|
|
|
j = 1u;
|
|
|
|
get_param(llwrapfn, 0u)
|
|
|
|
} else if self.ret_ty.cast {
|
|
|
|
let retptr = alloca(bcx, self.ret_ty.ty);
|
|
|
|
BitCast(bcx, retptr, T_ptr(ret_ty))
|
|
|
|
} else {
|
|
|
|
alloca(bcx, ret_ty)
|
|
|
|
};
|
|
|
|
|
|
|
|
let mut i = 0u;
|
2013-05-14 04:52:12 -05:00
|
|
|
let n = atys.len();
|
2013-01-25 16:56:56 -06:00
|
|
|
while i < n {
|
|
|
|
let mut argval = get_param(llwrapfn, i + j);
|
|
|
|
if attrs[i].is_some() {
|
|
|
|
argval = Load(bcx, argval);
|
|
|
|
store_inbounds(bcx, argval, llargbundle, [0u, i]);
|
|
|
|
} else if atys[i].cast {
|
|
|
|
let argptr = GEPi(bcx, llargbundle, [0u, i]);
|
|
|
|
let argptr = BitCast(bcx, argptr, T_ptr(atys[i].ty));
|
|
|
|
Store(bcx, argval, argptr);
|
|
|
|
} else {
|
|
|
|
store_inbounds(bcx, argval, llargbundle, [0u, i]);
|
|
|
|
}
|
|
|
|
i += 1u;
|
|
|
|
}
|
|
|
|
store_inbounds(bcx, llretptr, llargbundle, [0u, n]);
|
|
|
|
}
|
|
|
|
|
2013-04-18 17:53:29 -05:00
|
|
|
fn build_wrap_ret(&self,
|
|
|
|
bcx: block,
|
|
|
|
arg_tys: &[TypeRef],
|
|
|
|
llargbundle: ValueRef) {
|
2013-01-25 16:56:56 -06:00
|
|
|
unsafe {
|
|
|
|
if llvm::LLVMGetTypeKind(self.ret_ty.ty) == Void {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2013-04-18 17:53:29 -05:00
|
|
|
|
|
|
|
let llretval = load_inbounds(bcx, llargbundle, ~[ 0, arg_tys.len() ]);
|
2013-01-25 16:56:56 -06:00
|
|
|
let llretval = if self.ret_ty.cast {
|
|
|
|
let retptr = BitCast(bcx, llretval, T_ptr(self.ret_ty.ty));
|
|
|
|
Load(bcx, retptr)
|
|
|
|
} else {
|
|
|
|
Load(bcx, llretval)
|
|
|
|
};
|
2013-04-18 17:53:29 -05:00
|
|
|
let llretptr = BitCast(bcx,
|
|
|
|
bcx.fcx.llretptr.get(),
|
|
|
|
T_ptr(self.ret_ty.ty));
|
|
|
|
Store(bcx, llretval, llretptr);
|
2013-01-25 16:56:56 -06:00
|
|
|
}
|
|
|
|
}
|