1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#![allow(clippy::use_self)]
use crate::alloc::{AllocRef, BuildAllocRef, DeallocRef, NonZeroLayout, ReallocRef};
use core::ptr::NonNull;
use liballoc::alloc::handle_alloc_error;
#[derive(Default, Copy, Clone)]
pub struct AbortAlloc<A>(pub A);
impl<A: BuildAllocRef> BuildAllocRef for AbortAlloc<A> {
type Ref = AbortAlloc<A::Ref>;
unsafe fn build_alloc_ref(
&mut self,
ptr: NonNull<u8>,
layout: Option<NonZeroLayout>,
) -> Self::Ref {
Self(self.0.build_alloc_ref(ptr, layout))
}
}
impl<A: DeallocRef> DeallocRef for AbortAlloc<A> {
type BuildAlloc = AbortAlloc<A::BuildAlloc>;
fn get_build_alloc(&mut self) -> Self::BuildAlloc {
Self(self.0.get_build_alloc())
}
unsafe fn dealloc(&mut self, ptr: NonNull<u8>, layout: NonZeroLayout) {
self.0.dealloc(ptr, layout)
}
}
impl<A: AllocRef> AllocRef for AbortAlloc<A> {
type Error = crate::Never;
fn alloc(&mut self, layout: NonZeroLayout) -> Result<NonNull<u8>, Self::Error> {
self.0
.alloc(layout)
.map_err(|_| handle_alloc_error(layout.into()))
}
fn alloc_zeroed(&mut self, layout: NonZeroLayout) -> Result<NonNull<u8>, Self::Error> {
self.0
.alloc_zeroed(layout)
.map_err(|_| handle_alloc_error(layout.into()))
}
}
impl<A: ReallocRef> ReallocRef for AbortAlloc<A> {
unsafe fn realloc(
&mut self,
ptr: NonNull<u8>,
old_layout: NonZeroLayout,
new_layout: NonZeroLayout,
) -> Result<NonNull<u8>, Self::Error> {
self.0
.realloc(ptr, old_layout, new_layout)
.map_err(|_| handle_alloc_error(new_layout.into()))
}
}