use crate::backtrace::Backtrace;
use crate::chain::Chain;
use crate::location::Location;
use crate::ptr::MutPtr;
use crate::ptr::{OwnPtr, RefPtr};
use crate::{Error, StdError};
use alloc::boxed::Box;
use core::any::TypeId;
use core::fmt::{self, Debug, Display};
use core::mem::ManuallyDrop;
use core::ptr;
use core::ptr::NonNull;
#[cfg(error_generic_member_access)]
use std::error::{self, Request};
use core::ops::{Deref, DerefMut};
impl Error {
#[cold]
#[must_use]
#[track_caller]
pub fn new<E>(error: E) -> Self
where
E: StdError + Send + Sync + 'static,
{
let backtrace = backtrace_if_absent!(&error);
Error::from_std(error, backtrace)
}
#[cold]
#[must_use]
#[track_caller]
pub fn msg<M>(message: M) -> Self
where
M: Display + Debug + Send + Sync + 'static,
{
Error::from_debug(message, backtrace!())
}
#[cold]
#[track_caller]
pub(crate) fn from_std<E>(error: E, backtrace: Option<Backtrace>) -> Self
where
E: StdError + Send + Sync + 'static,
{
let vtable = &ErrorVTable {
object_drop: object_drop::<E>,
object_ref: object_ref::<E>,
object_mut: object_mut::<E>,
object_boxed: object_boxed::<E>,
object_downcast: object_downcast::<E>,
object_drop_rest: object_drop_front::<E>,
object_backtrace: no_backtrace,
};
unsafe { Error::construct(error, vtable, backtrace) }
}
#[cold]
#[track_caller]
pub(crate) fn from_debug<M>(message: M, backtrace: Option<Backtrace>) -> Self
where
M: Display + Debug + Send + Sync + 'static,
{
use crate::wrapper::MessageError;
let error: MessageError<M> = MessageError(message);
let vtable = &ErrorVTable {
object_drop: object_drop::<MessageError<M>>,
object_ref: object_ref::<MessageError<M>>,
object_mut: object_mut::<MessageError<M>>,
object_boxed: object_boxed::<MessageError<M>>,
object_downcast: object_downcast::<M>,
object_drop_rest: object_drop_front::<M>,
object_backtrace: no_backtrace,
};
unsafe { Error::construct(error, vtable, backtrace) }
}
#[cold]
#[track_caller]
pub(crate) fn from_display<M>(message: M, backtrace: Option<Backtrace>) -> Self
where
M: Display + Send + Sync + 'static,
{
use crate::wrapper::DisplayError;
let error: DisplayError<M> = DisplayError(message);
let vtable = &ErrorVTable {
object_drop: object_drop::<DisplayError<M>>,
object_ref: object_ref::<DisplayError<M>>,
object_mut: object_mut::<DisplayError<M>>,
object_boxed: object_boxed::<DisplayError<M>>,
object_downcast: object_downcast::<M>,
object_drop_rest: object_drop_front::<M>,
object_backtrace: no_backtrace,
};
unsafe { Error::construct(error, vtable, backtrace) }
}
#[cold]
#[track_caller]
pub(crate) fn from_context<C, E>(context: C, error: E, backtrace: Option<Backtrace>) -> Self
where
C: Display + Send + Sync + 'static,
E: StdError + Send + Sync + 'static,
{
let error: ContextError<C, E> = ContextError { context, error };
let vtable = &ErrorVTable {
object_drop: object_drop::<ContextError<C, E>>,
object_ref: object_ref::<ContextError<C, E>>,
object_mut: object_mut::<ContextError<C, E>>,
object_boxed: object_boxed::<ContextError<C, E>>,
object_downcast: context_downcast::<C, E>,
object_drop_rest: context_drop_rest::<C, E>,
object_backtrace: no_backtrace,
};
unsafe { Error::construct(error, vtable, backtrace) }
}
#[cold]
#[track_caller]
pub(crate) fn from_boxed(
error: Box<dyn StdError + Send + Sync>,
backtrace: Option<Backtrace>,
) -> Self {
use crate::wrapper::BoxedError;
let error = BoxedError(error);
let vtable = &ErrorVTable {
object_drop: object_drop::<BoxedError>,
object_ref: object_ref::<BoxedError>,
object_mut: object_mut::<BoxedError>,
object_boxed: object_boxed::<BoxedError>,
object_downcast: object_downcast::<Box<dyn StdError + Send + Sync>>,
object_drop_rest: object_drop_front::<Box<dyn StdError + Send + Sync>>,
object_backtrace: no_backtrace,
};
unsafe { Error::construct(error, vtable, backtrace) }
}
#[cold]
#[track_caller]
unsafe fn construct<E>(
error: E,
vtable: &'static ErrorVTable,
backtrace: Option<Backtrace>,
) -> Self
where
E: StdError + Send + Sync + 'static,
{
let location = caller!();
let inner: Box<ErrorImpl<E>> = Box::new(ErrorImpl {
vtable,
backtrace,
location,
_object: error,
});
let inner = OwnPtr::new(inner).cast::<ErrorImpl>();
Error { inner }
}
#[cold]
#[must_use]
#[track_caller]
pub fn context<C>(self, context: C) -> Self
where
C: Display + Send + Sync + 'static,
{
let error: ContextError<C, Error> = ContextError {
context,
error: self,
};
let vtable = &ErrorVTable {
object_drop: object_drop::<ContextError<C, Error>>,
object_ref: object_ref::<ContextError<C, Error>>,
object_mut: object_mut::<ContextError<C, Error>>,
object_boxed: object_boxed::<ContextError<C, Error>>,
object_downcast: context_chain_downcast::<C>,
object_drop_rest: context_chain_drop_rest::<C>,
object_backtrace: context_backtrace::<C>,
};
let backtrace = None;
unsafe { Error::construct(error, vtable, backtrace) }
}
pub fn backtrace(&self) -> &impl_backtrace!() {
unsafe { ErrorImpl::backtrace(self.inner.as_ref()) }
}
pub fn file(&self) -> &str {
unsafe { ErrorImpl::location(self.inner.as_ref()).file() }
}
pub fn line(&self) -> u32 {
unsafe { ErrorImpl::location(self.inner.as_ref()).line() }
}
pub fn column(&self) -> u32 {
unsafe { ErrorImpl::location(self.inner.as_ref()).column() }
}
#[cold]
pub fn chain(&'_ self) -> Chain<'_> {
unsafe { ErrorImpl::chain(self.inner.as_ref()) }
}
pub fn root_cause(&self) -> &(dyn StdError + 'static) {
self.chain().last().unwrap()
}
pub fn is<E>(&self) -> bool
where
E: Display + Debug + Send + Sync + 'static,
{
self.downcast_ref::<E>().is_some()
}
pub fn downcast<E>(self) -> Result<E, Self>
where
E: Display + Debug + Send + Sync + 'static,
{
let target = TypeId::of::<E>();
let inner = self.inner;
unsafe {
let addr = match (vtable(inner.ptr).object_downcast)(inner, target) {
Some(addr) => addr,
None => return Err(self),
};
let outer = ManuallyDrop::new(self);
let error = addr.cast::<E>().as_ptr().read();
(vtable(outer.inner.ptr).object_drop_rest)(outer.inner, target);
Ok(error)
}
}
pub fn downcast_ref<E>(&self) -> Option<&E>
where
E: Display + Debug + Send + Sync + 'static,
{
let target = TypeId::of::<E>();
unsafe {
let addr = (vtable(self.inner.ptr).object_downcast)(self.inner, target)?;
Some(addr.cast::<E>().deref())
}
}
pub fn downcast_mut<E>(&mut self) -> Option<&mut E>
where
E: Display + Debug + Send + Sync + 'static,
{
let target = TypeId::of::<E>();
unsafe {
let addr = (vtable(self.inner.ptr).object_downcast)(self.inner, target)?;
Some(addr.cast::<E>().deref_mut())
}
}
#[cfg(error_generic_member_access)]
pub(crate) fn provide<'a>(&'a self, request: &mut Request<'a>) {
unsafe { ErrorImpl::provide(self.inner.by_ref(), request) }
}
#[cfg(error_generic_member_access)]
#[doc(hidden)]
pub fn thiserror_provide<'a>(&'a self, request: &mut Request<'a>) {
Self::provide(self, request);
}
}
impl<E> From<E> for Error
where
E: StdError + Send + Sync + 'static,
{
#[cold]
#[track_caller]
fn from(error: E) -> Self {
let backtrace = backtrace_if_absent!(&error);
Error::from_std(error, backtrace)
}
}
impl Deref for Error {
type Target = dyn StdError + Send + Sync + 'static;
fn deref(&self) -> &Self::Target {
unsafe { ErrorImpl::error(self.inner.as_ref()) }
}
}
impl DerefMut for Error {
fn deref_mut(&mut self) -> &mut Self::Target {
unsafe { ErrorImpl::error_mut(self.inner.as_mut()) }
}
}
impl Display for Error {
fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
unsafe { ErrorImpl::display(self.inner.as_ref(), formatter) }
}
}
impl Debug for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
unsafe { ErrorImpl::debug(self.inner.as_ref(), f) }
}
}
impl Drop for Error {
fn drop(&mut self) {
unsafe {
(vtable(self.inner.ptr).object_drop)(self.inner);
}
}
}
struct ErrorVTable {
object_drop: unsafe fn(OwnPtr<ErrorImpl>),
object_ref: unsafe fn(RefPtr<ErrorImpl>) -> &(dyn StdError + Send + Sync + 'static),
object_mut: unsafe fn(MutPtr<ErrorImpl>) -> &mut (dyn StdError + Send + Sync + 'static),
object_boxed: unsafe fn(OwnPtr<ErrorImpl>) -> Box<dyn StdError + Send + Sync + 'static>,
object_downcast: unsafe fn(OwnPtr<ErrorImpl>, TypeId) -> Option<OwnPtr<()>>,
object_drop_rest: unsafe fn(OwnPtr<ErrorImpl>, TypeId),
object_backtrace: unsafe fn(RefPtr<ErrorImpl>) -> Option<&Backtrace>,
}
unsafe fn object_drop<E>(e: OwnPtr<ErrorImpl>) {
let unerased_own = e.cast::<ErrorImpl<E>>();
drop(unsafe { unerased_own.boxed() });
}
unsafe fn object_drop_front<E>(e: OwnPtr<ErrorImpl>, target: TypeId) {
let _ = target;
let unerased_own = e.cast::<ErrorImpl<ManuallyDrop<E>>>();
drop(unsafe { unerased_own.boxed() });
}
unsafe fn object_ref<E>(e: RefPtr<'_, ErrorImpl>) -> &'_ (dyn StdError + Send + Sync + 'static)
where
E: StdError + Send + Sync + 'static,
{
&e.cast::<ErrorImpl<E>>().as_ref()._object
}
unsafe fn object_mut<E>(e: MutPtr<'_, ErrorImpl>) -> &'_ mut (dyn StdError + Send + Sync + 'static)
where
E: StdError + Send + Sync + 'static,
{
&mut e.cast::<ErrorImpl<E>>().as_mut()._object
}
unsafe fn object_boxed<E>(e: OwnPtr<ErrorImpl>) -> Box<dyn StdError + Send + Sync + 'static>
where
E: StdError + Send + Sync + 'static,
{
let unerased_own = e.cast::<ErrorImpl<E>>();
unsafe { unerased_own.boxed() }
}
unsafe fn object_downcast<E>(e: OwnPtr<ErrorImpl>, target: TypeId) -> Option<OwnPtr<()>>
where
E: 'static,
{
if TypeId::of::<E>() == target {
let unerased_ref = e.cast::<ErrorImpl<E>>();
Some(
OwnPtr::from_raw(unsafe {
NonNull::new_unchecked(ptr::addr_of!((*unerased_ref.as_ptr())._object) as *mut E)
})
.cast::<()>(),
)
} else {
None
}
}
fn no_backtrace(e: RefPtr<'_, ErrorImpl>) -> Option<&'_ Backtrace> {
let _ = e;
None
}
unsafe fn context_downcast<C, E>(e: OwnPtr<ErrorImpl>, target: TypeId) -> Option<OwnPtr<()>>
where
C: 'static,
E: 'static,
{
if TypeId::of::<C>() == target {
let unerased_ref = e.cast::<ErrorImpl<ContextError<C, E>>>();
let unerased = unerased_ref.deref();
Some(OwnPtr::from_raw(NonNull::from(&unerased._object.context)).cast::<()>())
} else if TypeId::of::<E>() == target {
let unerased_ref = e.cast::<ErrorImpl<ContextError<C, E>>>();
let unerased = unerased_ref.deref();
Some(OwnPtr::from_raw(NonNull::from(&unerased._object.error)).cast::<()>())
} else {
None
}
}
unsafe fn context_drop_rest<C, E>(e: OwnPtr<ErrorImpl>, target: TypeId)
where
C: 'static,
E: 'static,
{
if TypeId::of::<C>() == target {
let unerased_own = e.cast::<ErrorImpl<ContextError<ManuallyDrop<C>, E>>>();
drop(unsafe { unerased_own.boxed() });
} else {
let unerased_own = e.cast::<ErrorImpl<ContextError<C, ManuallyDrop<E>>>>();
drop(unsafe { unerased_own.boxed() });
}
}
unsafe fn context_chain_downcast<C>(e: OwnPtr<ErrorImpl>, target: TypeId) -> Option<OwnPtr<()>>
where
C: 'static,
{
let unerased_ref = e.cast::<ErrorImpl<ContextError<C, Error>>>();
let unerased = unerased_ref.deref();
if TypeId::of::<C>() == target {
Some(OwnPtr::from_raw(NonNull::from(&unerased._object.context)).cast::<()>())
} else {
let source = &unerased._object.error;
unsafe { (vtable(source.inner.ptr).object_downcast)(source.inner, target) }
}
}
unsafe fn context_chain_drop_rest<C>(e: OwnPtr<ErrorImpl>, target: TypeId)
where
C: 'static,
{
if TypeId::of::<C>() == target {
let unerased_own = e.cast::<ErrorImpl<ContextError<ManuallyDrop<C>, Error>>>();
drop(unsafe { unerased_own.boxed() });
} else {
let unerased_own = e.cast::<ErrorImpl<ContextError<C, ManuallyDrop<Error>>>>();
let unerased = unsafe { unerased_own.boxed() };
let inner = unerased._object.error.inner;
drop(unerased);
let vtable = unsafe { vtable(inner.ptr) };
unsafe { (vtable.object_drop_rest)(inner, target) };
}
}
#[allow(clippy::unnecessary_wraps)]
unsafe fn context_backtrace<C>(e: RefPtr<'_, ErrorImpl>) -> Option<&'_ Backtrace>
where
C: 'static,
{
let unerased = e.cast::<ErrorImpl<ContextError<C, Error>>>().as_ref();
let backtrace = unsafe { ErrorImpl::backtrace(unerased._object.error.inner.as_ref()) };
Some(backtrace)
}
#[repr(C)]
pub(crate) struct ErrorImpl<E = ()> {
vtable: &'static ErrorVTable,
backtrace: Option<Backtrace>,
location: Location,
_object: E,
}
unsafe fn vtable(p: NonNull<ErrorImpl>) -> &'static ErrorVTable {
unsafe { *(p.as_ptr() as *const &'static ErrorVTable) }
}
#[repr(C)]
pub(crate) struct ContextError<C, E> {
pub context: C,
pub error: E,
}
impl<E> ErrorImpl<E> {
fn erase(&'_ self) -> RefPtr<'_, ErrorImpl> {
RefPtr::new(self).cast::<ErrorImpl>()
}
}
impl ErrorImpl {
pub(crate) unsafe fn error(
this: RefPtr<'_, Self>,
) -> &'_ (dyn StdError + Send + Sync + 'static) {
unsafe { (vtable(this.ptr).object_ref)(this) }
}
pub(crate) unsafe fn error_mut(
this: MutPtr<'_, Self>,
) -> &'_ mut (dyn StdError + Send + Sync + 'static) {
unsafe { (vtable(this.ptr).object_mut)(this) }
}
pub(crate) unsafe fn backtrace(this: RefPtr<'_, Self>) -> &'_ Backtrace {
this.as_ref()
.backtrace
.as_ref()
.or_else(|| {
#[cfg(error_generic_member_access)]
return error::request_ref::<Backtrace>(unsafe { Self::error(this) });
#[cfg(not(error_generic_member_access))]
return unsafe { (vtable(this.ptr).object_backtrace)(this) };
})
.expect("backtrace capture failed")
}
pub(crate) unsafe fn location(this: RefPtr<'_, Self>) -> &'_ Location {
&this.as_ref().location
}
#[cfg(error_generic_member_access)]
unsafe fn provide<'a>(this: RefPtr<'a, Self>, request: &mut Request<'a>) {
if let Some(backtrace) = unsafe { &this.deref().backtrace } {
request.provide_ref(backtrace);
}
unsafe { Self::error(this) }.provide(request);
}
#[cold]
pub(crate) unsafe fn chain(this: RefPtr<Self>) -> Chain {
Chain::new(unsafe { Self::error(this) })
}
}
impl<E> StdError for ErrorImpl<E>
where
E: StdError,
{
fn source(&self) -> Option<&(dyn StdError + 'static)> {
unsafe { ErrorImpl::error(self.erase()).source() }
}
}
impl<E> Debug for ErrorImpl<E>
where
E: Debug,
{
fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
unsafe { ErrorImpl::debug(self.erase(), formatter) }
}
}
impl<E> Display for ErrorImpl<E>
where
E: Display,
{
fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
unsafe { ErrorImpl::display(self.erase(), formatter) }
}
}
impl From<Error> for Box<dyn StdError + Send + Sync + 'static> {
#[cold]
fn from(error: Error) -> Self {
let outer = ManuallyDrop::new(error);
unsafe {
(vtable(outer.inner.ptr).object_boxed)(outer.inner)
}
}
}
impl From<Error> for Box<dyn StdError + Send + 'static> {
fn from(error: Error) -> Self {
Box::<dyn StdError + Send + Sync>::from(error)
}
}
impl From<Error> for Box<dyn StdError + 'static> {
fn from(error: Error) -> Self {
Box::<dyn StdError + Send + Sync>::from(error)
}
}
impl AsRef<dyn StdError + Send + Sync> for Error {
fn as_ref(&self) -> &(dyn StdError + Send + Sync + 'static) {
&**self
}
}
impl AsRef<dyn StdError> for Error {
fn as_ref(&self) -> &(dyn StdError + 'static) {
&**self
}
}