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
use crate::sys::{c_char, enif_snprintf};
use crate::wrapper::NIF_TERM;
use std::fmt;
pub fn fmt(term: NIF_TERM, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
const SIZE: usize = 1024;
let mut bytes: Vec<u8> = Vec::with_capacity(SIZE);
let mut n = 0;
for _ in 0..10 {
let i = unsafe {
enif_snprintf!(
bytes.as_mut_ptr() as *mut c_char,
bytes.capacity(),
b"%T\x00" as *const u8 as *const c_char,
term
)
};
if i < 0 {
// Do not propagate an error, because string formatting is
// supposed to be infallible.
break;
}
n = i as usize;
if n >= bytes.capacity() {
// Bizarrely, enif_snprintf consistently underestimates the
// amount of memory it will need to write long lists. To try to
// avoid going around the loop again, double the estimate.
bytes.reserve_exact(2 * n + 1);
// Ensure that the `set_len` call below does not expose
// uninitialized bytes if we give up after 10 attempts.
n = 0;
} else {
break;
}
}
unsafe {
bytes.set_len(n);
}
f.write_str(&String::from_utf8_lossy(&bytes))
}