2019-01-11 16:32:31 +00:00
|
|
|
// Original implementation taken from rust-memchr.
|
2016-09-22 00:10:37 +00:00
|
|
|
// Copyright 2015 Andrew Gallant, bluss and Nicolas Koch
|
|
|
|
|
|
|
|
pub fn memchr(needle: u8, haystack: &[u8]) -> Option<usize> {
|
|
|
|
let p = unsafe {
|
|
|
|
libc::memchr(
|
|
|
|
haystack.as_ptr() as *const libc::c_void,
|
|
|
|
needle as libc::c_int,
|
2016-10-08 13:48:28 +00:00
|
|
|
haystack.len())
|
2016-09-22 00:10:37 +00:00
|
|
|
};
|
|
|
|
if p.is_null() {
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
Some(p as usize - (haystack.as_ptr() as usize))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn memrchr(needle: u8, haystack: &[u8]) -> Option<usize> {
|
|
|
|
|
|
|
|
#[cfg(target_os = "linux")]
|
|
|
|
fn memrchr_specific(needle: u8, haystack: &[u8]) -> Option<usize> {
|
|
|
|
// GNU's memrchr() will - unlike memchr() - error if haystack is empty.
|
|
|
|
if haystack.is_empty() {return None}
|
|
|
|
let p = unsafe {
|
|
|
|
libc::memrchr(
|
|
|
|
haystack.as_ptr() as *const libc::c_void,
|
|
|
|
needle as libc::c_int,
|
2016-10-08 13:48:28 +00:00
|
|
|
haystack.len())
|
2016-09-22 00:10:37 +00:00
|
|
|
};
|
|
|
|
if p.is_null() {
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
Some(p as usize - (haystack.as_ptr() as usize))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os = "linux"))]
|
|
|
|
fn memrchr_specific(needle: u8, haystack: &[u8]) -> Option<usize> {
|
2019-02-10 19:23:21 +00:00
|
|
|
core::slice::memchr::memrchr(needle, haystack)
|
2016-09-22 00:10:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
memrchr_specific(needle, haystack)
|
|
|
|
}
|