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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#![cfg_attr(not(test), no_std)]
pub trait Zeroable {
fn is_zero(self) -> bool;
}
macro_rules! impl_zeroable_num {
( $type:ty ) => {
impl Zeroable for $type {
fn is_zero(self) -> bool {
self == 0
}
}
}
}
impl_zeroable_num!(u8);
impl_zeroable_num!(i8);
impl_zeroable_num!(u16);
impl_zeroable_num!(i16);
impl_zeroable_num!(u32);
impl_zeroable_num!(i32);
impl_zeroable_num!(u64);
impl_zeroable_num!(i64);
impl<T> Zeroable for *const T {
fn is_zero(self) -> bool {
self.is_null()
}
}
impl<T> Zeroable for *mut T {
fn is_zero(self) -> bool {
self.is_null()
}
}
pub struct NullTermArrayIter<T: Zeroable + Copy> {
ptr: *const T,
}
impl<T: Zeroable + Copy> NullTermArrayIter<T> {
pub unsafe fn new(ptr: *const T) -> Self {
Self { ptr }
}
}
impl<T: Zeroable + Copy> Iterator for NullTermArrayIter<T> {
type Item = T;
fn next(&mut self) -> Option<T> {
let value = unsafe { *self.ptr };
if value.is_zero() {
None
} else {
self.ptr = unsafe { self.ptr.offset(1) };
Some(value)
}
}
}
#[test]
fn test_nulltermarray_strings() {
use std::ptr;
let strlist = vec!["array\0", "of\0", "strings\0"];
let mut ptrlist: Vec<_> = strlist.iter().map(|x| x.as_ptr()).collect();
ptrlist.push(ptr::null());
let mut iter = unsafe { NullTermArrayIter::new(ptrlist.as_ptr()) };
for i in strlist.iter() {
assert_eq!(iter.next(), Some(i.as_ptr()));
}
assert_eq!(iter.next(), None);
}
#[test]
fn test_nulltermarray_string() {
let text = "string\0";
let mut iter = unsafe { NullTermArrayIter::new(text.as_ptr()) };
for i in text[..text.len()-1].as_bytes().iter() {
assert_eq!(iter.next(), Some(*i));
}
assert_eq!(iter.next(), None);
}