wasmi/src/host.rs

use std::any::TypeId;
use value::{RuntimeValue, TryInto};
use Error;

/// Safe wrapper for list of arguments.
#[derive(Debug)]
pub struct RuntimeArgs<'a>(&'a [RuntimeValue]);

impl<'a> From<&'a [RuntimeValue]> for RuntimeArgs<'a> {
	fn from(inner: &'a [RuntimeValue]) -> Self {
		RuntimeArgs(inner)
	}
}

impl<'a> RuntimeArgs<'a> {

	/// Extract argument by index `idx` returning error if cast is invalid or not enough arguments
	pub fn nth<T>(&self, idx: usize) -> Result<T, Error> where RuntimeValue: TryInto<T, Error> {
		Ok(self.nth_value(idx)?.try_into().map_err(|_| Error::Value("Invalid argument cast".to_owned()))?)
	}

	/// Extract argument as a runtime value by index `idx` returning error is not enough arguments
	pub fn nth_value(&self, idx: usize) -> Result<RuntimeValue, Error> {
		if self.0.len() <= idx {
			return Err(Error::Value("Invalid argument index".to_owned()));
		}
		Ok(self.0[idx])
	}

	/// Total number of arguments
	pub fn len(&self) -> usize {
		self.0.len()
	}
}

/// Custom user error.
pub trait HostError: 'static + ::std::fmt::Display + ::std::fmt::Debug + Send + Sync {
	#[doc(hidden)]
	fn __private_get_type_id__(&self) -> TypeId {
		TypeId::of::<Self>()
	}
}

impl HostError {
	/// Attempt to downcast this `HostError` to a concrete type by reference.
	pub fn downcast_ref<T: HostError>(&self) -> Option<&T> {
		if self.__private_get_type_id__() == TypeId::of::<T>() {
			unsafe { Some(&*(self as *const HostError as *const T)) }
		} else {
			None
		}
	}

	/// Attempt to downcast this `HostError` to a concrete type by mutable
	/// reference.
	pub fn downcast_mut<T: HostError>(&mut self) -> Option<&mut T> {
		if self.__private_get_type_id__() == TypeId::of::<T>() {
			unsafe { Some(&mut *(self as *mut HostError as *mut T)) }
		} else {
			None
		}
	}
}

pub trait Externals {
	fn invoke_index(
		&mut self,
		index: usize,
		args: RuntimeArgs,
	) -> Result<Option<RuntimeValue>, Error>;
}

pub struct NopExternals;

impl Externals for NopExternals {
	fn invoke_index(
		&mut self,
		_index: usize,
		_args: RuntimeArgs,
	) -> Result<Option<RuntimeValue>, Error> {
		Err(Error::Trap("invoke index on no-op externals".into()))
	}
}

#[cfg(test)]
mod tests {

	use value::RuntimeValue;
	use super::RuntimeArgs;

	#[test]
	fn i32_runtime_args() {
		let args: RuntimeArgs = (&[RuntimeValue::I32(0)][..]).into();
		let val: i32 = args.nth(0).unwrap();
		assert_eq!(val, 0);
	}

	#[test]
	fn i64_invalid_arg_cast() {
		let args: RuntimeArgs = (&[RuntimeValue::I64(90534534545322)][..]).into();
		assert!(args.nth::<i32>(0).is_err());
	}
}