#[macro_export]
macro_rules! separated_list(
($i:expr, $sep:ident!( $($args:tt)* ), $submac:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert,ErrorKind};
use $crate::InputLength;
let mut res = ::std::vec::Vec::new();
let mut input = $i.clone();
let input_ = input.clone();
match $submac!(input_, $($args2)*) {
Err(Err::Error(_)) => Ok((input, ::std::vec::Vec::new())),
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
if i.input_len() == input.input_len() {
let e:ErrorKind<u32> = ErrorKind::SeparatedList;
Err(Err::Error(error_position!(e,input)))
} else {
res.push(o);
input = i;
let ret;
loop {
let input_ = input.clone();
match $sep!(input_, $($args)*) {
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
}
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i2,_)) => {
let i2_len = i2.input_len();
if i2_len == input.input_len() {
ret = Ok((input, res));
break;
}
match $submac!(i2, $($args2)*) {
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i3,o3)) => {
if i3.input_len() == i2_len {
ret = Ok((input, res));
break;
}
res.push(o3);
input = i3;
}
}
}
}
}
ret
}
},
}
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
separated_list!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
separated_list!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
separated_list!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! separated_nonempty_list(
($i:expr, $sep:ident!( $($args:tt)* ), $submac:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Convert,Err,ErrorKind};
use $crate::InputLength;
let mut res = ::std::vec::Vec::new();
let mut input = $i.clone();
let input_ = input.clone();
match $submac!(input_, $($args2)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
if i.input_len() == input.input_len() {
let e:ErrorKind<u32> = ErrorKind::SeparatedNonEmptyList;
Err(Err::Error(error_position!(e,input)))
} else {
res.push(o);
input = i;
let ret;
loop {
let input_ = input.clone();
match $sep!(input_, $($args)*) {
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
}
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i2,_)) => {
let i2_len = i2.input_len();
if i2_len == input.input_len() {
ret = Ok((input, res));
break;
}
match $submac!(i2, $($args2)*) {
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i3,o3)) => {
if i3.input_len() == i2_len {
ret = Ok((input, res));
break;
}
res.push(o3);
input = i3;
}
}
}
}
}
ret
}
},
}
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
separated_nonempty_list!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
separated_nonempty_list!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
separated_nonempty_list!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! separated_list_complete {
($i:expr, $sep:ident!( $($args:tt)* ), $submac:ident!( $($args2:tt)* )) => ({
separated_list!($i, complete!($sep!($($args)*)), complete!($submac!($($args2)*)))
});
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
separated_list_complete!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
separated_list_complete!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
separated_list_complete!($i, call!($f), call!($g));
);
}
#[macro_export]
macro_rules! separated_nonempty_list_complete {
($i:expr, $sep:ident!( $($args:tt)* ), $submac:ident!( $($args2:tt)* )) => ({
separated_nonempty_list!($i, complete!($sep!($($args)*)), complete!($submac!($($args2)*)))
});
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
separated_nonempty_list_complete!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
separated_nonempty_list_complete!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
separated_nonempty_list_complete!($i, call!($f), call!($g));
);
}
#[macro_export]
macro_rules! many0(
($i:expr, $submac:ident!( $($args:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert,InputLength};
let ret;
let mut res = ::std::vec::Vec::new();
let mut input = $i.clone();
loop {
if input.input_len() == 0 {
ret = Ok((input, res));
break;
}
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i, o)) => {
if i == input {
ret = Err(Err::Error(error_position!(ErrorKind::Many0,input)));
break;
}
res.push(o);
input = i;
}
}
}
ret
}
);
($i:expr, $f:expr) => (
many0!($i, call!($f));
);
);
#[macro_export]
macro_rules! many1(
($i:expr, $submac:ident!( $($args:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
use $crate::InputLength;
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(Err::Error(_)) => Err(Err::Error(
error_position!(ErrorKind::Many1,i_)
)),
Err(Err::Failure(_)) => Err(Err::Failure(
error_position!(ErrorKind::Many1,i_)
)),
Err(i) => Err(Err::convert(i)),
Ok((i1,o1)) => {
if i1.input_len() == 0 {
let mut res = ::std::vec::Vec::new();
res.push(o1);
Ok((i1,res))
} else {
let mut res = ::std::vec::Vec::with_capacity(4);
res.push(o1);
let mut input = i1;
let mut error = ::std::option::Option::None;
loop {
if input.input_len() == 0 {
break;
}
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Err(Err::Error(_)) => {
break;
},
Err(e) => {
error = ::std::option::Option::Some(e);
break;
},
Ok((i, o)) => {
if i.input_len() == input.input_len() {
break;
}
res.push(o);
input = i;
}
}
}
match error {
::std::option::Option::Some(e) => Err(Err::convert(e)),
::std::option::Option::None => Ok((input, res))
}
}
}
}
}
);
($i:expr, $f:expr) => (
many1!($i, call!($f));
);
);
#[macro_export]
macro_rules! many_till(
(__impl $i:expr, $submac1:ident!( $($args1:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let ret;
let mut res = ::std::vec::Vec::new();
let mut input = $i.clone();
loop {
match $submac2!(input, $($args2)*) {
Ok((i, o)) => {
ret = Ok((i, (res, o)));
break;
},
_ => {
match $submac1!(input, $($args1)*) {
Err(Err::Error(err)) => {
ret = Err(Err::Error(error_node_position!(ErrorKind::ManyTill,input, err)));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i, o)) => {
if i == input {
ret = Err(Err::Error(error_position!(ErrorKind::ManyTill,input)));
break;
}
res.push(o);
input = i;
},
}
},
}
}
ret
}
);
($i:expr, $submac1:ident!( $($args1:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
many_till!(__impl $i, $submac1!($($args1)*), $submac2!($($args2)*));
);
($i:expr, $submac1:ident!( $($args1:tt)* ), $g:expr) => (
many_till!(__impl $i, $submac1!($($args1)*), call!($g));
);
($i:expr, $f:expr, $submac2:ident!( $($args2:tt)* )) => (
many_till!(__impl $i, call!($f), $submac2!($($args2)*));
);
($i:expr, $f:expr, $g: expr) => (
many_till!(__impl $i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! many_m_n(
($i:expr, $m:expr, $n: expr, $submac:ident!( $($args:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Context,Err,Needed};
use $crate::InputLength;
let mut res = ::std::vec::Vec::with_capacity($m);
let mut input = $i.clone();
let mut count: usize = 0;
let mut err = false;
let mut incomplete: ::std::option::Option<Needed> = ::std::option::Option::None;
let mut failure: ::std::option::Option<Context<_,_>> = ::std::option::Option::None;
loop {
if count == $n { break }
let i_ = input.clone();
match $submac!(i_, $($args)*) {
Ok((i, o)) => {
if i.input_len() == input.input_len() {
break;
}
res.push(o);
input = i;
count += 1;
}
Err(Err::Error(_)) => {
err = true;
break;
},
Err(Err::Incomplete(i)) => {
incomplete = ::std::option::Option::Some(i);
break;
},
Err(Err::Failure(e)) => {
failure = ::std::option::Option::Some(e);
break;
},
}
if input.input_len() == 0 {
break;
}
}
if count < $m {
if err {
Err(Err::Error(error_position!(ErrorKind::ManyMN,$i)))
} else {
match failure {
::std::option::Option::Some(i) => Err(Err::Failure(i)),
::std::option::Option::None => match incomplete {
::std::option::Option::Some(i) => $crate::need_more($i, i),
::std::option::Option::None => $crate::need_more($i, Needed::Unknown)
}
}
}
} else {
match failure {
::std::option::Option::Some(i) => Err(Err::Failure(i)),
::std::option::Option::None => match incomplete {
::std::option::Option::Some(i) => $crate::need_more($i, i),
::std::option::Option::None => Ok((input, res))
}
}
}
}
);
($i:expr, $m:expr, $n: expr, $f:expr) => (
many_m_n!($i, $m, $n, call!($f));
);
);
#[macro_export]
macro_rules! count(
($i:expr, $submac:ident!( $($args:tt)* ), $count: expr) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let ret;
let mut input = $i.clone();
let mut res = ::std::vec::Vec::new();
loop {
if res.len() == $count {
ret = Ok((input, res));
break;
}
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Ok((i,o)) => {
res.push(o);
input = i;
},
Err(Err::Error(_)) => {
ret = Err(Err::Error(error_position!(ErrorKind::Count,$i)));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
}
}
ret
}
);
($i:expr, $f:expr, $count: expr) => (
count!($i, call!($f), $count);
);
);
#[macro_export]
macro_rules! count_fixed (
($i:expr, $typ:ty, $submac:ident!( $($args:tt)* ), $count: expr) => (
{
use ::std::result::Result::*;
use $crate::{Convert,Err};
let ret;
let mut input = $i.clone();
let mut res: [$typ; $count] = unsafe{[::std::mem::uninitialized(); $count as usize]};
let mut cnt: usize = 0;
loop {
if cnt == $count {
ret = Ok((input, res)); break;
}
match $submac!(input, $($args)*) {
Ok((i,o)) => {
res[cnt] = o;
cnt += 1;
input = i;
},
Err(Err::Error(_)) => {
ret = Err(Err::Error(error_position!(ErrorKind::Count,$i)));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
}
}
ret
}
);
($i:expr, $typ: ty, $f:expr, $count: expr) => (
count_fixed!($i, $typ, call!($f), $count);
);
);
#[macro_export]
macro_rules! length_count(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match $submac!($i, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i, o)) => {
match count!(i, $submac2!($($args2)*), o as usize) {
Err(e) => Err(Err::convert(e)),
Ok((i2, o2)) => Ok((i2, o2))
}
}
}
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
length_count!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
length_count!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
length_count!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! length_data(
($i:expr, $submac:ident!( $($args:tt)* )) => ({
use ::std::result::Result::*;
use $crate::{Convert,Err};
match $submac!($i, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i, o)) => {
match take!(i, o as usize) {
Err(e) => Err(Err::convert(e)),
Ok((i2, o2)) => Ok((i2, o2))
}
}
}
});
($i:expr, $f:expr) => (
length_data!($i, call!($f));
);
);
#[macro_export]
macro_rules! length_value(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match $submac!($i, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i, o)) => {
match take!(i, o as usize) {
Err(e) => Err(Err::convert(e)),
Ok((i2, o2)) => {
match complete!(o2, $submac2!($($args2)*)) {
Err(e) => Err(Err::convert(e)),
Ok((_, o3)) => Ok((i2, o3))
}
}
}
}
}
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
length_value!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
length_value!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
length_value!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! fold_many0(
($i:expr, $submac:ident!( $($args:tt)* ), $init:expr, $f:expr) => (
{
use ::std::result::Result::*;
use $crate::{Err,InputLength,Convert};
let ret;
let f = $f;
let mut res = $init;
let mut input = $i.clone();
loop {
if input.input_len() == 0 {
ret = Ok((input, res));
break;
}
match $submac!(input, $($args)*) {
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
},
Err(e) => {
ret = Err(Err::convert(e));
break;
},
Ok((i, o)) => {
if i == input {
ret = Err(Err::Error(
error_position!(ErrorKind::Many0,input)
));
break;
}
res = f(res, o);
input = i;
}
}
}
ret
}
);
($i:expr, $f:expr, $init:expr, $fold_f:expr) => (
fold_many0!($i, call!($f), $init, $fold_f);
);
);
#[macro_export]
macro_rules! fold_many1(
($i:expr, $submac:ident!( $($args:tt)* ), $init:expr, $f:expr) => (
{
use ::std::result::Result::*;
use $crate::{Err,Needed,IResult,InputLength,Context};
match $submac!($i, $($args)*) {
Err(Err::Error(_)) => Err(Err::Error(
error_position!(ErrorKind::Many1,$i)
)),
Err(Err::Failure(_)) => Err(Err::Failure(
error_position!(ErrorKind::Many1,$i)
)),
Err(Err::Incomplete(i)) => Err(Err::Incomplete(i)),
Ok((i1,o1)) => {
let acc = $init;
let f = $f;
if i1.input_len() == 0 {
let acc = f(acc, o1);
Ok((i1,acc))
} else {
let mut acc = f(acc, o1);
let mut input = i1;
let mut incomplete: ::std::option::Option<Needed> =
::std::option::Option::None;
let mut failure: ::std::option::Option<Context<_,_>> =
::std::option::Option::None;
loop {
if input.input_len() == 0 {
break;
}
match $submac!(input, $($args)*) {
Err(Err::Error(_)) => {
break;
},
Err(Err::Incomplete(i)) => {
incomplete = ::std::option::Option::Some(i);
break;
},
Err(Err::Failure(e)) => {
failure = ::std::option::Option::Some(e);
break;
},
Ok((i, o)) => {
if i.input_len() == input.input_len() {
break;
}
acc = f(acc, o);
input = i;
}
}
}
match failure {
::std::option::Option::Some(e) => Err(Err::Failure(e)),
::std::option::Option::None => match incomplete {
::std::option::Option::Some(i) => $crate::need_more($i, i),
::std::option::Option::None => Ok((input, acc))
}
}
}
}
}
}
);
($i:expr, $f:expr, $init:expr, $fold_f:expr) => (
fold_many1!($i, call!($f), $init, $fold_f);
);
);
#[macro_export]
macro_rules! fold_many_m_n(
($i:expr, $m:expr, $n: expr, $submac:ident!( $($args:tt)* ), $init:expr, $f:expr) => (
{
use ::std::result::Result::*;
use $crate::{Err,Needed,IResult};
use $crate::InputLength;
let mut acc = $init;
let f = $f;
let mut input = $i.clone();
let mut count: usize = 0;
let mut err = false;
let mut incomplete: ::std::option::Option<Needed> = ::std::option::Option::None;
loop {
if count == $n { break }
match $submac!(input, $($args)*) {
Ok((i, o)) => {
if i.input_len() == input.input_len() {
break;
}
acc = f(acc, o);
input = i;
count += 1;
}
Err(Err::Error(_)) | Err(Err::Failure(_)) => {
err = true;
break;
},
Err(Err::Incomplete(i)) => {
incomplete = ::std::option::Option::Some(i);
break;
},
}
if input.input_len() == 0 {
break;
}
}
if count < $m {
if err {
Err(Err::Error(error_position!(ErrorKind::ManyMN,$i)))
} else {
match incomplete {
::std::option::Option::Some(i) => Err(Err::Incomplete(i)),
::std::option::Option::None => Err(Err::Incomplete(Needed::Unknown))
}
}
} else {
match incomplete {
::std::option::Option::Some(i) => Err(Err::Incomplete(i)),
::std::option::Option::None => Ok((input, acc))
}
}
}
);
($i:expr, $m:expr, $n: expr, $f:expr, $init:expr, $fold_f:expr) => (
fold_many_m_n!($i, $m, $n, call!($f), $init, $fold_f);
);
);
#[cfg(test)]
mod tests {
use internal::{Err,Needed,IResult};
use nom::{alpha,be_u8,be_u16,le_u16,digit};
use std::str::{self,FromStr};
macro_rules! tag (
($i:expr, $inp: expr) => (
{
#[inline(always)]
fn as_bytes<T: $crate::AsBytes>(b: &T) -> &[u8] {
b.as_bytes()
}
let expected = $inp;
let bytes = as_bytes(&expected);
tag_bytes!($i,bytes)
}
);
);
macro_rules! tag_bytes (
($i:expr, $bytes: expr) => (
{
use std::cmp::min;
let len = $i.len();
let blen = $bytes.len();
let m = min(len, blen);
let reduced = &$i[..m];
let b = &$bytes[..m];
let res: IResult<_,_,u32> = if reduced != b {
Err($crate::Err::Error(error_position!(ErrorKind::Tag, $i)))
} else if m < blen {
Err($crate::Err::Incomplete(Needed::Size(blen)))
} else {
Ok((&$i[blen..], reduced))
};
res
}
);
);
macro_rules! take (
($i:expr, $count:expr) => (
{
let cnt = $count as usize;
let res:IResult<&[u8],&[u8],u32> = if $i.len() < cnt {
Err($crate::Err::Incomplete(Needed::Size(cnt)))
} else {
Ok((&$i[cnt..],&$i[0..cnt]))
};
res
}
)
);
#[test]
#[cfg(feature = "std")]
fn separated_list() {
named!(multi<&[u8],Vec<&[u8]> >, separated_list!(tag!(","), tag!("abcd")));
named!(multi_empty<&[u8],Vec<&[u8]> >, separated_list!(tag!(","), tag!("")));
named!(multi_longsep<&[u8],Vec<&[u8]> >, separated_list!(tag!(".."), tag!("abcd")));
let a = &b"abcdef"[..];
let b = &b"abcd,abcdef"[..];
let c = &b"azerty"[..];
let d = &b",,abc"[..];
let e = &b"abcd,abcd,ef"[..];
let f = &b"abc"[..];
let g = &b"abcd."[..];
let h = &b"abcd,abc"[..];
let res1 = vec![&b"abcd"[..]];
assert_eq!(multi(a),Ok((&b"ef"[..], res1)));
let res2 = vec![&b"abcd"[..], &b"abcd"[..]];
assert_eq!(multi(b),Ok((&b"ef"[..], res2)));
assert_eq!(multi(c),Ok((&b"azerty"[..], Vec::new())));
assert_eq!(multi_empty(d), Err(Err::Error(error_position!(ErrorKind::SeparatedList, d))));
let res4 = vec![&b"abcd"[..], &b"abcd"[..]];
assert_eq!(multi(e),Ok((&b",ef"[..], res4)));
assert_eq!(multi(f), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(multi_longsep(g), Err(Err::Incomplete(Needed::Size(2))));
assert_eq!(multi(h), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "std")]
fn separated_list_complete() {
named!(multi<&[u8],Vec<&[u8]> >, separated_list_complete!(tag!(","), alpha));
let a = &b"abcdef"[..];
let b = &b"abcd,abcdef"[..];
let c = &b"abcd,abcd,ef"[..];
let d = &b"abc."[..];
let e = &b"abcd,ef."[..];
let f = &b"123"[..];
assert_eq!(multi(a),Ok((&b""[..], vec!(a))));
assert_eq!(multi(b),Ok((&b""[..], vec!(&b"abcd"[..], &b"abcdef"[..]))));
assert_eq!(multi(c),Ok((&b""[..], vec!(&b"abcd"[..], &b"abcd"[..], &b"ef"[..]))));
assert_eq!(multi(d),Ok((&b"."[..], vec!(&b"abc"[..]))));
assert_eq!(multi(e),Ok((&b"."[..], vec!(&b"abcd"[..], &b"ef"[..]))));
assert_eq!(multi(f),Ok((&b"123"[..], Vec::new())));
}
#[test]
#[cfg(feature = "std")]
fn separated_nonempty_list() {
named!(multi<&[u8],Vec<&[u8]> >, separated_nonempty_list!(tag!(","), tag!("abcd")));
named!(multi_longsep<&[u8],Vec<&[u8]> >, separated_nonempty_list!(tag!(".."), tag!("abcd")));
let a = &b"abcdef"[..];
let b = &b"abcd,abcdef"[..];
let c = &b"azerty"[..];
let d = &b"abcd,abcd,ef"[..];
let f = &b"abc"[..];
let g = &b"abcd."[..];
let h = &b"abcd,abc"[..];
let res1 = vec![&b"abcd"[..]];
assert_eq!(multi(a),Ok((&b"ef"[..], res1)));
let res2 = vec![&b"abcd"[..], &b"abcd"[..]];
assert_eq!(multi(b),Ok((&b"ef"[..], res2)));
assert_eq!(multi(c), Err(Err::Error(error_position!(ErrorKind::Tag,c))));
let res3 = vec![&b"abcd"[..], &b"abcd"[..]];
assert_eq!(multi(d),Ok((&b",ef"[..], res3)));
assert_eq!(multi(f), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(multi_longsep(g), Err(Err::Incomplete(Needed::Size(2))));
assert_eq!(multi(h), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "std")]
fn separated_nonempty_list_complete() {
named!(multi<&[u8],Vec<&[u8]> >, separated_nonempty_list_complete!(tag!(","), alpha));
let a = &b"abcdef"[..];
let b = &b"abcd,abcdef"[..];
let c = &b"abcd,abcd,ef"[..];
let d = &b"abc."[..];
let e = &b"abcd,ef."[..];
let f = &b"123"[..];
assert_eq!(multi(a),Ok((&b""[..], vec!(a))));
assert_eq!(multi(b),Ok((&b""[..], vec!(&b"abcd"[..], &b"abcdef"[..]))));
assert_eq!(multi(c),Ok((&b""[..], vec!(&b"abcd"[..], &b"abcd"[..], &b"ef"[..]))));
assert_eq!(multi(d),Ok((&b"."[..], vec!(&b"abc"[..]))));
assert_eq!(multi(e),Ok((&b"."[..], vec!(&b"abcd"[..], &b"ef"[..]))));
assert_eq!(multi(f), Err(Err::Error(error_position!(ErrorKind::Alpha, &b"123"[..]))));
}
#[test]
#[cfg(feature = "std")]
fn many0() {
named!( tag_abcd, tag!("abcd") );
named!( tag_empty, tag!("") );
named!( multi<&[u8],Vec<&[u8]> >, many0!(tag_abcd) );
named!( multi_empty<&[u8],Vec<&[u8]> >, many0!(tag_empty) );
assert_eq!(multi(&b"abcdef"[..]),Ok((&b"ef"[..], vec![&b"abcd"[..]])));
assert_eq!(multi(&b"abcdabcdefgh"[..]),Ok((&b"efgh"[..], vec![&b"abcd"[..], &b"abcd"[..]])));
assert_eq!(multi(&b"azerty"[..]),Ok((&b"azerty"[..], Vec::new())));
assert_eq!(multi(&b"abcdab"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(multi(&b"abcd"[..]),Ok((&b""[..], vec![&b"abcd"[..]])));
assert_eq!(multi(&b""[..]),Ok((&b""[..], Vec::new())));
assert_eq!(multi_empty(&b"abcdef"[..]), Err(Err::Error(error_position!(ErrorKind::Many0, &b"abcdef"[..]))));
}
#[cfg(feature = "nightly")]
use test::Bencher;
#[cfg(feature = "nightly")]
#[bench]
fn many0_bench(b: &mut Bencher) {
named!(multi<&[u8],Vec<&[u8]> >, many0!(tag!("abcd")));
b.iter(|| {
multi(&b"abcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcd"[..])
});
}
#[test]
#[cfg(feature = "std")]
fn many1() {
named!(multi<&[u8],Vec<&[u8]> >, many1!(tag!("abcd")));
let a = &b"abcdef"[..];
let b = &b"abcdabcdefgh"[..];
let c = &b"azerty"[..];
let d = &b"abcdab"[..];
let res1 = vec![&b"abcd"[..]];
assert_eq!(multi(a),Ok((&b"ef"[..], res1)));
let res2 = vec![&b"abcd"[..], &b"abcd"[..]];
assert_eq!(multi(b),Ok((&b"efgh"[..], res2)));
assert_eq!(multi(c), Err(Err::Error(error_position!(ErrorKind::Many1,c))));
assert_eq!(multi(d), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "std")]
fn many_till() {
named!(multi<&[u8], (Vec<&[u8]>, &[u8]) >, many_till!( tag!( "abcd" ), tag!( "efgh" ) ) );
let a = b"abcdabcdefghabcd";
let b = b"efghabcd";
let c = b"azerty";
let res_a = (vec![&b"abcd"[..], &b"abcd"[..]], &b"efgh"[..]);
let res_b: (Vec<&[u8]>, &[u8]) = (Vec::new(), &b"efgh"[..]);
assert_eq!(multi(&a[..]),Ok((&b"abcd"[..], res_a)));
assert_eq!(multi(&b[..]),Ok((&b"abcd"[..], res_b)));
assert_eq!(multi(&c[..]), Err(Err::Error(error_node_position!(ErrorKind::ManyTill,&c[..], error_position!(ErrorKind::Tag,&c[..])))));
}
#[test]
#[cfg(feature = "std")]
fn infinite_many() {
fn tst(input: &[u8]) -> IResult<&[u8], &[u8]> {
println!("input: {:?}", input);
Err(Err::Error(error_position!(ErrorKind::Custom(0),input)))
}
named!(multi0<&[u8],Vec<&[u8]> >, many0!(tst));
let a = &b"abcdef"[..];
assert_eq!(multi0(a),Ok((a, Vec::new())));
named!(multi1<&[u8],Vec<&[u8]> >, many1!(tst));
let a = &b"abcdef"[..];
assert_eq!(multi1(a), Err(Err::Error(error_position!(ErrorKind::Many1,a))));
}
#[test]
#[cfg(feature = "std")]
fn many_m_n() {
named!(multi<&[u8],Vec<&[u8]> >, many_m_n!(2, 4, tag!("Abcd")));
let a = &b"Abcdef"[..];
let b = &b"AbcdAbcdefgh"[..];
let c = &b"AbcdAbcdAbcdAbcdefgh"[..];
let d = &b"AbcdAbcdAbcdAbcdAbcdefgh"[..];
let e = &b"AbcdAb"[..];
assert_eq!(multi(a), Err(Err::Error(error_position!(ErrorKind::ManyMN,a))));
let res1 = vec![&b"Abcd"[..], &b"Abcd"[..]];
assert_eq!(multi(b),Ok((&b"efgh"[..], res1)));
let res2 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]];
assert_eq!(multi(c),Ok((&b"efgh"[..], res2)));
let res3 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]];
assert_eq!(multi(d),Ok((&b"Abcdefgh"[..], res3)));
assert_eq!(multi(e), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "std")]
fn count() {
const TIMES: usize = 2;
named!( tag_abc, tag!("abc") );
named!( cnt_2<&[u8], Vec<&[u8]> >, count!(tag_abc, TIMES ) );
assert_eq!(cnt_2(&b"abcabcabcdef"[..]),Ok((&b"abcdef"[..], vec![&b"abc"[..], &b"abc"[..]])));
assert_eq!(cnt_2(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(cnt_2(&b"abcab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(cnt_2(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"xxx"[..]))));
assert_eq!(cnt_2(&b"xxxabcabcdef"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"xxxabcabcdef"[..]))));
assert_eq!(cnt_2(&b"abcxxxabcdef"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"abcxxxabcdef"[..]))));
}
#[test]
#[cfg(feature = "std")]
fn count_zero() {
const TIMES: usize = 0;
named!( tag_abc, tag!("abc") );
named!( counter_2<&[u8], Vec<&[u8]> >, count!(tag_abc, TIMES ) );
let done = &b"abcabcabcdef"[..];
let parsed_done = Vec::new();
let rest = done;
let incomplete_1 = &b"ab"[..];
let parsed_incompl_1 = Vec::new();
let incomplete_2 = &b"abcab"[..];
let parsed_incompl_2 = Vec::new();
let error = &b"xxx"[..];
let error_remain = &b"xxx"[..];
let parsed_err = Vec::new();
let error_1 = &b"xxxabcabcdef"[..];
let parsed_err_1 = Vec::new();
let error_1_remain = &b"xxxabcabcdef"[..];
let error_2 = &b"abcxxxabcdef"[..];
let parsed_err_2 = Vec::new();
let error_2_remain = &b"abcxxxabcdef"[..];
assert_eq!(counter_2(done),Ok((rest, parsed_done)));
assert_eq!(counter_2(incomplete_1),Ok((incomplete_1, parsed_incompl_1)));
assert_eq!(counter_2(incomplete_2),Ok((incomplete_2, parsed_incompl_2)));
assert_eq!(counter_2(error),Ok((error_remain, parsed_err)));
assert_eq!(counter_2(error_1),Ok((error_1_remain, parsed_err_1)));
assert_eq!(counter_2(error_2),Ok((error_2_remain, parsed_err_2)));
}
#[test]
fn count_fixed() {
const TIMES: usize = 2;
named!( tag_abc, tag!("abc") );
named!( cnt_2<&[u8], [&[u8]; TIMES] >, count_fixed!(&[u8], tag_abc, TIMES ) );
assert_eq!(cnt_2(&b"abcabcabcdef"[..]),Ok((&b"abcdef"[..], [&b"abc"[..], &b"abc"[..]])));
assert_eq!(cnt_2(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(cnt_2(&b"abcab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(cnt_2(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"xxx"[..]))));
assert_eq!(cnt_2(&b"xxxabcabcdef"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"xxxabcabcdef"[..]))));
assert_eq!(cnt_2(&b"abcxxxabcdef"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"abcxxxabcdef"[..]))));
}
#[allow(dead_code)]
pub fn compile_count_fixed(input: &[u8]) -> IResult<&[u8], ()> {
do_parse!(input,
tag!("abcd") >>
count_fixed!( u16, le_u16, 4 ) >>
eof!() >>
()
)
}
#[derive(Debug,Clone,PartialEq)]
pub struct NilError;
impl From<u32> for NilError {
fn from(_: u32) -> Self {
NilError
}
}
#[allow(unused_variables)]
#[test]
fn count_fixed_no_type() {
const TIMES: usize = 2;
named!( tag_abc, tag!("abc") );
named!( counter_2<&[u8], [&[u8]; TIMES], NilError >, count_fixed!(&[u8], tag_abc, TIMES ) );
let done = &b"abcabcabcdef"[..];
let parsed_main = [&b"abc"[..], &b"abc"[..]];
let rest = &b"abcdef"[..];
let incomplete_1 = &b"ab"[..];
let incomplete_2 = &b"abcab"[..];
let error = &b"xxx"[..];
let error_1 = &b"xxxabcabcdef"[..];
let error_1_remain = &b"xxxabcabcdef"[..];
let error_2 = &b"abcxxxabcdef"[..];
let error_2_remain = &b"abcxxxabcdef"[..];
assert_eq!(counter_2(done),Ok((rest, parsed_main)));
assert_eq!(counter_2(incomplete_1), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(counter_2(incomplete_2), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(counter_2(error), Err(Err::Error(error_position!(ErrorKind::Count, error))));
assert_eq!(counter_2(error_1), Err(Err::Error(error_position!(ErrorKind::Count, error_1_remain))));
assert_eq!(counter_2(error_2), Err(Err::Error(error_position!(ErrorKind::Count, error_2_remain))));
}
named!(pub number<u32>, map_res!(
map_res!(
digit,
str::from_utf8
),
FromStr::from_str
));
#[test]
#[cfg(feature = "std")]
fn length_count() {
named!(tag_abc, tag!(&b"abc"[..]) );
named!( cnt<&[u8], Vec<&[u8]> >, length_count!(number, tag_abc) );
assert_eq!(cnt(&b"2abcabcabcdef"[..]),Ok((&b"abcdef"[..], vec![&b"abc"[..], &b"abc"[..]])));
assert_eq!(cnt(&b"2ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(cnt(&b"3abcab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(cnt(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Digit, &b"xxx"[..]))));
assert_eq!(cnt(&b"2abcxxx"[..]), Err(Err::Error(error_position!(ErrorKind::Count, &b"abcxxx"[..]))));
}
#[test]
fn length_data() {
named!( take<&[u8], &[u8]>, length_data!(number) );
assert_eq!(take(&b"6abcabcabcdef"[..]),Ok((&b"abcdef"[..], &b"abcabc"[..])));
assert_eq!(take(&b"3ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(take(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Digit, &b"xxx"[..]))));
assert_eq!(take(&b"2abcxxx"[..]),Ok((&b"cxxx"[..], &b"ab"[..])));
}
#[test]
fn length_value_test() {
named!(length_value_1<&[u8], u16 >, length_value!(be_u8, be_u16));
named!(length_value_2<&[u8], (u8, u8) >, length_value!(be_u8, tuple!(be_u8, be_u8)));
let i1 = [0, 5, 6];
assert_eq!(length_value_1(&i1), Err(Err::Error(error_position!(ErrorKind::Complete, &b""[..]))));
assert_eq!(length_value_2(&i1), Err(Err::Error(error_position!(ErrorKind::Complete, &b""[..]))));
let i2 = [1, 5, 6, 3];
assert_eq!(length_value_1(&i2), Err(Err::Error(error_position!(ErrorKind::Complete, &i2[1..2]))));
assert_eq!(length_value_2(&i2), Err(Err::Error(error_position!(ErrorKind::Complete, &i2[1..2]))));
let i3 = [2, 5, 6, 3, 4, 5, 7];
assert_eq!(length_value_1(&i3), Ok((&i3[3..], 1286)));
assert_eq!(length_value_2(&i3), Ok((&i3[3..], (5, 6))));
let i4 = [3, 5, 6, 3, 4, 5];
assert_eq!(length_value_1(&i4), Ok((&i4[4..], 1286)));
assert_eq!(length_value_2(&i4), Ok((&i4[4..], (5, 6))));
}
#[test]
#[cfg(feature = "std")]
fn fold_many0() {
fn fold_into_vec<T>(mut acc: Vec<T>, item: T) -> Vec<T> {
acc.push(item);
acc
};
named!( tag_abcd, tag!("abcd") );
named!( tag_empty, tag!("") );
named!( multi<&[u8],Vec<&[u8]> >, fold_many0!(tag_abcd, Vec::new(), fold_into_vec) );
named!( multi_empty<&[u8],Vec<&[u8]> >, fold_many0!(tag_empty, Vec::new(), fold_into_vec) );
assert_eq!(multi(&b"abcdef"[..]),Ok((&b"ef"[..], vec![&b"abcd"[..]])));
assert_eq!(multi(&b"abcdabcdefgh"[..]),Ok((&b"efgh"[..], vec![&b"abcd"[..], &b"abcd"[..]])));
assert_eq!(multi(&b"azerty"[..]),Ok((&b"azerty"[..], Vec::new())));
assert_eq!(multi(&b"abcdab"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(multi(&b"abcd"[..]),Ok((&b""[..], vec![&b"abcd"[..]])));
assert_eq!(multi(&b""[..]),Ok((&b""[..], Vec::new())));
assert_eq!(multi_empty(&b"abcdef"[..]), Err(Err::Error(error_position!(ErrorKind::Many0, &b"abcdef"[..]))));
}
#[test]
#[cfg(feature = "std")]
fn fold_many1() {
fn fold_into_vec<T>(mut acc: Vec<T>, item: T) -> Vec<T> {
acc.push(item);
acc
};
named!(multi<&[u8],Vec<&[u8]> >, fold_many1!(tag!("abcd"), Vec::new(), fold_into_vec));
let a = &b"abcdef"[..];
let b = &b"abcdabcdefgh"[..];
let c = &b"azerty"[..];
let d = &b"abcdab"[..];
let res1 = vec![&b"abcd"[..]];
assert_eq!(multi(a),Ok((&b"ef"[..], res1)));
let res2 = vec![&b"abcd"[..], &b"abcd"[..]];
assert_eq!(multi(b),Ok((&b"efgh"[..], res2)));
assert_eq!(multi(c), Err(Err::Error(error_position!(ErrorKind::Many1,c))));
assert_eq!(multi(d), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "std")]
fn fold_many_m_n() {
fn fold_into_vec<T>(mut acc: Vec<T>, item: T) -> Vec<T> {
acc.push(item);
acc
};
named!(multi<&[u8],Vec<&[u8]> >, fold_many_m_n!(2, 4, tag!("Abcd"), Vec::new(), fold_into_vec));
let a = &b"Abcdef"[..];
let b = &b"AbcdAbcdefgh"[..];
let c = &b"AbcdAbcdAbcdAbcdefgh"[..];
let d = &b"AbcdAbcdAbcdAbcdAbcdefgh"[..];
let e = &b"AbcdAb"[..];
assert_eq!(multi(a), Err(Err::Error(error_position!(ErrorKind::ManyMN,a))));
let res1 = vec![&b"Abcd"[..], &b"Abcd"[..]];
assert_eq!(multi(b),Ok((&b"efgh"[..], res1)));
let res2 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]];
assert_eq!(multi(c),Ok((&b"efgh"[..], res2)));
let res3 = vec![&b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..], &b"Abcd"[..]];
assert_eq!(multi(d),Ok((&b"Abcdefgh"[..], res3)));
assert_eq!(multi(e), Err(Err::Incomplete(Needed::Size(4))));
}
}