use lib::{mem, ptr};
use util::*;
#[inline(always)]
pub(crate) unsafe fn parse_digit<T: Integer>(digit: &mut T, c: u8)
-> T
{
let x = char_to_digit(c);
*digit = as_cast(x);
*digit
}
#[inline]
unsafe fn unchecked_unsafe<T>(value: &mut T, state: &mut ParseState, radix: T, last: *const u8)
where T: Integer
{
let mut digit: T = mem::uninitialized();
while state.curr < last && parse_digit(&mut digit, *state.curr) < radix {
let (v, o1) = value.overflowing_mul(radix);
let (v, o2) = v.overflowing_add(digit);
*value = v;
if state.trunc.is_null() && (o1 | o2) {
state.trunc = state.curr;
}
state.increment();
}
}
#[inline]
pub(crate) fn unchecked<T>(value: &mut T, state: &mut ParseState, radix: u32, last: *const u8)
where T: Integer
{
unsafe { unchecked_unsafe::<T>(value, state, as_cast(radix), last) }
}
#[inline]
unsafe fn checked_unsafe<T>(value: &mut T, state: &mut ParseState, radix: T, last: *const u8)
where T: Integer
{
let mut digit: T = mem::uninitialized();
while state.curr < last && parse_digit(&mut digit, *state.curr) < radix {
if state.trunc.is_null() {
match value.checked_mul(radix).and_then(|v| v.checked_add(digit)) {
Some(v) => *value = v,
None => state.trunc = state.curr,
}
}
state.increment();
}
}
#[inline]
#[allow(dead_code)]
pub(crate) fn checked<T>(value: &mut T, state: &mut ParseState, radix: u32, last: *const u8)
where T: Integer
{
unsafe { checked_unsafe::<T>(value, state, as_cast(radix), last) }
}
#[inline]
pub(crate) unsafe fn value<T, Cb>(state: &mut ParseState, radix: u32, last: *const u8, cb: Cb)
-> T
where T: Integer,
Cb: FnOnce(&mut T, &mut ParseState, u32, *const u8)
{
debug_assert_radix!(radix);
state.ltrim_char(last, b'0');
let mut value: T = T::ZERO;
cb(&mut value, state, radix, last);
value
}
#[inline]
pub(crate) unsafe fn filter_sign<T, Cb>(state: &mut ParseState, radix: u32, last: *const u8, cb: Cb)
-> (T, i32)
where T: Integer,
Cb: FnOnce(&mut T, &mut ParseState, u32, *const u8)
{
if state.curr == last {
(T::ZERO, 1)
} else {
match *state.curr {
b'+' => {
state.increment();
let value = value::<T, Cb>(state, radix, last, cb);
(value, 1)
},
b'-' => {
state.increment();
let value = value::<T, Cb>(state, radix, last, cb);
(value, -1)
},
_ => {
let value = value::<T, Cb>(state, radix, last, cb);
(value, 1)
},
}
}
}
#[inline]
pub(crate) unsafe fn unsigned<T, Cb>(radix: u32, first: *const u8, last: *const u8, cb: Cb)
-> (T, *const u8, bool)
where T: UnsignedInteger,
Cb: FnOnce(&mut T, &mut ParseState, u32, *const u8)
{
if first == last {
(T::ZERO, ptr::null(), false)
} else {
let mut state = ParseState::new(first);
let (value, sign) = filter_sign::<T, Cb>(&mut state, radix, last, cb);
match sign {
-1 => (value.wrapping_neg(), state.curr, true),
1 => (value, state.curr, state.is_truncated()),
_ => unreachable!(),
}
}
}
#[inline]
pub(crate) unsafe fn signed<T, Cb>(radix: u32, first: *const u8, last: *const u8, cb: Cb)
-> (T, *const u8, bool)
where T: SignedInteger,
Cb: FnOnce(&mut T, &mut ParseState, u32, *const u8)
{
if first == last {
(T::ZERO, ptr::null(), false)
} else {
let mut state = ParseState::new(first);
let (value, sign) = filter_sign::<T, Cb>(&mut state, radix, last, cb);
match sign {
-1 => (-value, state.curr, state.is_truncated()),
1 => (value, state.curr, state.is_truncated()),
_ => unreachable!(),
}
}
}
macro_rules! generate_unsafe_unsigned {
($func:ident, $t:tt) => (
#[inline]
unsafe fn $func(radix: u8, first: *const u8, last: *const u8)
-> ($t, *const u8, bool)
{
unsigned::<$t, _>(radix.into(), first, last, unchecked::<$t>)
}
)
}
generate_unsafe_unsigned!(atou8_unsafe, u8);
generate_unsafe_unsigned!(atou16_unsafe, u16);
generate_unsafe_unsigned!(atou32_unsafe, u32);
generate_unsafe_unsigned!(atou64_unsafe, u64);
generate_unsafe_unsigned!(atou128_unsafe, u128);
generate_unsafe_unsigned!(atousize_unsafe, usize);
macro_rules! generate_unsafe_signed {
($func:ident, $t:tt) => (
#[inline]
unsafe fn $func(radix: u8, first: *const u8, last: *const u8)
-> ($t, *const u8, bool)
{
signed::<$t, _>(radix.into(), first, last, unchecked::<$t>)
}
)
}
generate_unsafe_signed!(atoi8_unsafe, i8);
generate_unsafe_signed!(atoi16_unsafe, i16);
generate_unsafe_signed!(atoi32_unsafe, i32);
generate_unsafe_signed!(atoi64_unsafe, i64);
generate_unsafe_signed!(atoi128_unsafe, i128);
generate_unsafe_signed!(atoisize_unsafe, isize);
generate_from_bytes_local!(atou8_local, u8, atou8_unsafe);
generate_from_bytes_local!(atou16_local, u16, atou16_unsafe);
generate_from_bytes_local!(atou32_local, u32, atou32_unsafe);
generate_from_bytes_local!(atou64_local, u64, atou64_unsafe);
generate_from_bytes_local!(atou128_local, u128, atou128_unsafe);
generate_from_bytes_local!(atousize_local, usize, atousize_unsafe);
generate_from_bytes_local!(atoi8_local, i8, atoi8_unsafe);
generate_from_bytes_local!(atoi16_local, i16, atoi16_unsafe);
generate_from_bytes_local!(atoi32_local, i32, atoi32_unsafe);
generate_from_bytes_local!(atoi64_local, i64, atoi64_unsafe);
generate_from_bytes_local!(atoi128_local, i128, atoi128_unsafe);
generate_from_bytes_local!(atoisize_local, isize, atoisize_unsafe);
generate_from_range_api!(atou8_range, u8, atou8_local);
generate_from_range_api!(atou16_range, u16, atou16_local);
generate_from_range_api!(atou32_range, u32, atou32_local);
generate_from_range_api!(atou64_range, u64, atou64_local);
generate_from_range_api!(atou128_range, u128, atou128_local);
generate_from_range_api!(atousize_range, usize, atousize_local);
generate_from_range_api!(atoi8_range, i8, atoi8_local);
generate_from_range_api!(atoi16_range, i16, atoi16_local);
generate_from_range_api!(atoi32_range, i32, atoi32_local);
generate_from_range_api!(atoi64_range, i64, atoi64_local);
generate_from_range_api!(atoi128_range, i128, atoi128_local);
generate_from_range_api!(atoisize_range, isize, atoisize_local);
generate_try_from_range_api!(try_atou8_range, u8, atou8_local);
generate_try_from_range_api!(try_atou16_range, u16, atou16_local);
generate_try_from_range_api!(try_atou32_range, u32, atou32_local);
generate_try_from_range_api!(try_atou64_range, u64, atou64_local);
generate_try_from_range_api!(try_atou128_range, u128, atou128_local);
generate_try_from_range_api!(try_atousize_range, usize, atousize_local);
generate_try_from_range_api!(try_atoi8_range, i8, atoi8_local);
generate_try_from_range_api!(try_atoi16_range, i16, atoi16_local);
generate_try_from_range_api!(try_atoi32_range, i32, atoi32_local);
generate_try_from_range_api!(try_atoi64_range, i64, atoi64_local);
generate_try_from_range_api!(try_atoi128_range, i128, atoi128_local);
generate_try_from_range_api!(try_atoisize_range, isize, atoisize_local);
generate_from_slice_api!(atou8_slice, u8, atou8_local);
generate_from_slice_api!(atou16_slice, u16, atou16_local);
generate_from_slice_api!(atou32_slice, u32, atou32_local);
generate_from_slice_api!(atou64_slice, u64, atou64_local);
generate_from_slice_api!(atou128_slice, u128, atou128_local);
generate_from_slice_api!(atousize_slice, usize, atousize_local);
generate_from_slice_api!(atoi8_slice, i8, atoi8_local);
generate_from_slice_api!(atoi16_slice, i16, atoi16_local);
generate_from_slice_api!(atoi32_slice, i32, atoi32_local);
generate_from_slice_api!(atoi64_slice, i64, atoi64_local);
generate_from_slice_api!(atoi128_slice, i128, atoi128_local);
generate_from_slice_api!(atoisize_slice, isize, atoisize_local);
generate_try_from_slice_api!(try_atou8_slice, u8, atou8_local);
generate_try_from_slice_api!(try_atou16_slice, u16, atou16_local);
generate_try_from_slice_api!(try_atou32_slice, u32, atou32_local);
generate_try_from_slice_api!(try_atou64_slice, u64, atou64_local);
generate_try_from_slice_api!(try_atou128_slice, u128, atou128_local);
generate_try_from_slice_api!(try_atousize_slice, usize, atousize_local);
generate_try_from_slice_api!(try_atoi8_slice, i8, atoi8_local);
generate_try_from_slice_api!(try_atoi16_slice, i16, atoi16_local);
generate_try_from_slice_api!(try_atoi32_slice, i32, atoi32_local);
generate_try_from_slice_api!(try_atoi64_slice, i64, atoi64_local);
generate_try_from_slice_api!(try_atoi128_slice, i128, atoi128_local);
generate_try_from_slice_api!(try_atoisize_slice, isize, atoisize_local);
#[cfg(test)]
mod tests {
use super::*;
#[cfg(feature = "radix")]
const DATA: [(u8, &'static str); 35] = [
(2, "100101"),
(3, "1101"),
(4, "211"),
(5, "122"),
(6, "101"),
(7, "52"),
(8, "45"),
(9, "41"),
(10, "37"),
(11, "34"),
(12, "31"),
(13, "2B"),
(14, "29"),
(15, "27"),
(16, "25"),
(17, "23"),
(18, "21"),
(19, "1I"),
(20, "1H"),
(21, "1G"),
(22, "1F"),
(23, "1E"),
(24, "1D"),
(25, "1C"),
(26, "1B"),
(27, "1A"),
(28, "19"),
(29, "18"),
(30, "17"),
(31, "16"),
(32, "15"),
(33, "14"),
(34, "13"),
(35, "12"),
(36, "11"),
];
#[test]
fn checked_test() {
let s = "1234567891234567890123";
unsafe {
let first = s.as_ptr();
let last = first.add(s.len());
let mut value: u64 = 0;
let mut state = ParseState::new(first);
checked(&mut value, &mut state, 10, last);
assert_eq!(value, 12345678912345678901);
assert_eq!(state.curr, last);
assert_eq!(state.truncated_bytes(), 2);
}
}
#[test]
fn unchecked_test() {
let s = "1234567891234567890123";
unsafe {
let first = s.as_ptr();
let last = first.add(s.len());
let mut value: u64 = 0;
let mut state = ParseState::new(first);
unchecked(&mut value, &mut state, 10, last);
assert_eq!(value, 17082782369737483467);
assert_eq!(state.curr, last);
assert_eq!(state.truncated_bytes(), 2);
}
}
#[test]
fn atou8_base10_test() {
assert_eq!(0, atou8_slice(10, b"0"));
assert_eq!(127, atou8_slice(10, b"127"));
assert_eq!(128, atou8_slice(10, b"128"));
assert_eq!(255, atou8_slice(10, b"255"));
assert_eq!(255, atou8_slice(10, b"-1"));
assert_eq!(1, atou8_slice(10, b"1a"));
}
#[cfg(feature = "radix")]
#[test]
fn atou8_basen_test() {
for (b, s) in DATA.iter() {
assert_eq!(atou8_slice(*b, s.as_bytes()), 37);
}
}
#[test]
fn atoi8_base10_test() {
assert_eq!(0, atoi8_slice(10, b"0"));
assert_eq!(127, atoi8_slice(10, b"127"));
assert_eq!(-128, atoi8_slice(10, b"128"));
assert_eq!(-1, atoi8_slice(10, b"255"));
assert_eq!(-1, atoi8_slice(10, b"-1"));
assert_eq!(1, atoi8_slice(10, b"1a"));
}
#[test]
fn atou16_base10_test() {
assert_eq!(0, atou16_slice(10, b"0"));
assert_eq!(32767, atou16_slice(10, b"32767"));
assert_eq!(32768, atou16_slice(10, b"32768"));
assert_eq!(65535, atou16_slice(10, b"65535"));
assert_eq!(65535, atou16_slice(10, b"-1"));
assert_eq!(1, atou16_slice(10, b"1a"));
}
#[test]
fn atoi16_base10_test() {
assert_eq!(0, atoi16_slice(10, b"0"));
assert_eq!(32767, atoi16_slice(10, b"32767"));
assert_eq!(-32768, atoi16_slice(10, b"32768"));
assert_eq!(-1, atoi16_slice(10, b"65535"));
assert_eq!(-1, atoi16_slice(10, b"-1"));
assert_eq!(1, atoi16_slice(10, b"1a"));
}
#[cfg(feature = "radix")]
#[test]
fn atoi16_basen_test() {
assert_eq!(atoi16_slice(36, b"YA"), 1234);
}
#[test]
fn atou32_base10_test() {
assert_eq!(0, atou32_slice(10, b"0"));
assert_eq!(2147483647, atou32_slice(10, b"2147483647"));
assert_eq!(2147483648, atou32_slice(10, b"2147483648"));
assert_eq!(4294967295, atou32_slice(10, b"4294967295"));
assert_eq!(4294967295, atou32_slice(10, b"-1"));
assert_eq!(1, atou32_slice(10, b"1a"));
}
#[test]
fn atoi32_base10_test() {
assert_eq!(0, atoi32_slice(10, b"0"));
assert_eq!(2147483647, atoi32_slice(10, b"2147483647"));
assert_eq!(-2147483648, atoi32_slice(10, b"2147483648"));
assert_eq!(-1, atoi32_slice(10, b"4294967295"));
assert_eq!(-1, atoi32_slice(10, b"-1"));
assert_eq!(1, atoi32_slice(10, b"1a"));
}
#[test]
fn atou64_base10_test() {
assert_eq!(0, atou64_slice(10, b"0"));
assert_eq!(9223372036854775807, atou64_slice(10, b"9223372036854775807"));
assert_eq!(9223372036854775808, atou64_slice(10, b"9223372036854775808"));
assert_eq!(18446744073709551615, atou64_slice(10, b"18446744073709551615"));
assert_eq!(18446744073709551615, atou64_slice(10, b"-1"));
assert_eq!(1, atou64_slice(10, b"1a"));
}
#[test]
fn atoi64_base10_test() {
assert_eq!(0, atoi64_slice(10, b"0"));
assert_eq!(9223372036854775807, atoi64_slice(10, b"9223372036854775807"));
assert_eq!(-9223372036854775808, atoi64_slice(10, b"9223372036854775808"));
assert_eq!(-1, atoi64_slice(10, b"18446744073709551615"));
assert_eq!(-1, atoi64_slice(10, b"-1"));
assert_eq!(1, atoi64_slice(10, b"1a"));
}
#[test]
fn try_atou8_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atou8_slice(10, b""));
assert_eq!(success(0), try_atou8_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atou8_slice(10, b"1a"));
assert_eq!(overflow_error(0), try_atou8_slice(10, b"256"));
}
#[test]
fn try_atoi8_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atoi8_slice(10, b""));
assert_eq!(success(0), try_atoi8_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atoi8_slice(10, b"1a"));
assert_eq!(overflow_error(-128), try_atoi8_slice(10, b"128"));
}
#[test]
fn try_atou16_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atou16_slice(10, b""));
assert_eq!(success(0), try_atou16_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atou16_slice(10, b"1a"));
assert_eq!(overflow_error(0), try_atou16_slice(10, b"65536"));
}
#[test]
fn try_atoi16_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atoi16_slice(10, b""));
assert_eq!(success(0), try_atoi16_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atoi16_slice(10, b"1a"));
assert_eq!(overflow_error(-32768), try_atoi16_slice(10, b"32768"));
}
#[test]
fn try_atou32_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atou32_slice(10, b""));
assert_eq!(success(0), try_atou32_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atou32_slice(10, b"1a"));
assert_eq!(overflow_error(0), try_atou32_slice(10, b"4294967296"));
}
#[test]
fn try_atoi32_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atoi32_slice(10, b""));
assert_eq!(success(0), try_atoi32_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atoi32_slice(10, b"1a"));
assert_eq!(overflow_error(-2147483648), try_atoi32_slice(10, b"2147483648"));
}
#[test]
fn try_atou64_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atou64_slice(10, b""));
assert_eq!(success(0), try_atou64_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atou64_slice(10, b"1a"));
assert_eq!(overflow_error(0), try_atou64_slice(10, b"18446744073709551616"));
}
#[test]
fn try_atoi64_base10_test() {
assert_eq!(invalid_digit_error(0, 0), try_atoi64_slice(10, b""));
assert_eq!(success(0), try_atoi64_slice(10, b"0"));
assert_eq!(invalid_digit_error(1, 1), try_atoi64_slice(10, b"1a"));
assert_eq!(overflow_error(-9223372036854775808), try_atoi64_slice(10, b"9223372036854775808"));
}
}