blob: 4b67f56a5cec02a07f982ccde8cafd3921158da9 (
plain)
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
|
#[cfg(no_str_strip_prefix)] // rustc <1.45
pub(crate) trait StripPrefixExt {
fn strip_prefix(&self, ch: char) -> Option<&str>;
}
#[cfg(no_str_strip_prefix)]
impl StripPrefixExt for str {
fn strip_prefix(&self, ch: char) -> Option<&str> {
if self.starts_with(ch) {
Some(&self[ch.len_utf8()..])
} else {
None
}
}
}
pub(crate) use crate::alloc::vec::Vec;
#[cfg(no_alloc_crate)] // rustc <1.36
pub(crate) mod alloc {
pub use std::vec;
pub mod alloc {
use std::mem;
use std::process;
#[derive(Copy, Clone)]
pub struct Layout {
size: usize,
}
impl Layout {
pub unsafe fn from_size_align_unchecked(size: usize, align: usize) -> Self {
assert_eq!(align, 2);
Layout { size }
}
}
pub unsafe fn alloc(layout: Layout) -> *mut u8 {
let len_u16 = (layout.size + 1) / 2;
let mut vec = Vec::new();
vec.reserve_exact(len_u16);
let ptr: *mut u16 = vec.as_mut_ptr();
mem::forget(vec);
ptr as *mut u8
}
pub unsafe fn dealloc(ptr: *mut u8, layout: Layout) {
let len_u16 = (layout.size + 1) / 2;
unsafe { Vec::from_raw_parts(ptr as *mut u16, 0, len_u16) };
}
pub fn handle_alloc_error(_layout: Layout) -> ! {
// This is unreachable because the alloc implementation above never
// returns null; Vec::reserve_exact would already have called std's
// internal handle_alloc_error.
process::abort();
}
}
}
|
