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
use core::marker::PhantomData;
use crate::kernel::{cfg::CfgBuilder, hunk, Port};
impl<System: Port> hunk::Hunk<System> {
pub const fn build() -> CfgHunkBuilder<System> {
CfgHunkBuilder {
_phantom: PhantomData,
len: 1,
align: 1,
}
}
}
#[must_use = "must call `finish()` to complete registration"]
pub struct CfgHunkBuilder<System> {
_phantom: PhantomData<System>,
len: usize,
align: usize,
}
impl<System: Port> CfgHunkBuilder<System> {
pub const fn len(self, len: usize) -> Self {
Self { len, ..self }
}
pub const fn align(self, align: usize) -> Self {
Self { align, ..self }
}
}
impl<System: Port> CfgHunkBuilder<System> {
pub const fn finish(self, cfg: &mut CfgBuilder<System>) -> hunk::Hunk<System> {
let Self { align, len, .. } = self;
let inner = &mut cfg.inner;
inner.hunk_pool_len = (inner.hunk_pool_len + align - 1) / align * align;
let start = inner.hunk_pool_len;
inner.hunk_pool_len += len;
if align > inner.hunk_pool_align {
inner.hunk_pool_align = align;
}
hunk::Hunk::from_offset(start)
}
}