#[cfg(feature = "alloc")]
#[macro_export(local_inner_macros)]
macro_rules! separated_list(
($i:expr, $sep:ident!( $($args:tt)* ), $submac:ident!( $($args2:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::Err;
use $crate::InputLength;
let mut res = $crate::lib::std::vec::Vec::new();
let mut input = $i.clone();
let input_ = input.clone();
match $submac!(input_, $($args2)*) {
Err(Err::Error(_)) => Ok((input, res)),
Err(e) => Err(e),
Ok((i,o)) => {
if i.input_len() == input.input_len() {
Err(Err::Error(error_position!(input, $crate::ErrorKind::SeparatedList)))
} 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(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(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(local_inner_macros)]
macro_rules! separated_nonempty_list(
($i:expr, $sep:ident!( $($args:tt)* ), $submac:ident!( $($args2:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,ErrorKind};
use $crate::InputLength;
let mut res = $crate::lib::std::vec::Vec::new();
let mut input = $i.clone();
let input_ = input.clone();
match $submac!(input_, $($args2)*) {
Err(e) => Err(e),
Ok((i,o)) => {
if i.input_len() == input.input_len() {
let e = ErrorKind::SeparatedNonEmptyList;
Err(Err::Error(error_position!(input, e)))
} 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(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(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(local_inner_macros)]
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(local_inner_macros)]
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));
);
}
#[cfg(feature = "alloc")]
#[macro_export(local_inner_macros)]
macro_rules! many0(
($i:expr, $submac:ident!( $($args:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,AtEof};
let ret;
let mut res = $crate::lib::std::vec::Vec::new();
let mut input = $i.clone();
loop {
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Ok((i, o)) => {
if i == input {
if i.at_eof() {
ret = Ok((input, res));
} else {
ret = Err(Err::Error(error_position!(input, $crate::ErrorKind::Many0)));
}
break;
}
res.push(o);
input = i;
},
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
},
Err(e) => {
ret = Err(e);
break;
},
}
}
ret
}
);
($i:expr, $f:expr) => (
many0!($i, call!($f));
);
);
#[cfg(feature = "alloc")]
#[macro_export(local_inner_macros)]
macro_rules! many1(
($i:expr, $submac:ident!( $($args:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::Err;
use $crate::InputLength;
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(Err::Error(_)) => Err(Err::Error(
error_position!(i_, $crate::ErrorKind::Many1)
)),
Err(Err::Failure(_)) => Err(Err::Failure(
error_position!(i_, $crate::ErrorKind::Many1)
)),
Err(i) => Err(i),
Ok((i1,o1)) => {
let mut res = $crate::lib::std::vec::Vec::with_capacity(4);
res.push(o1);
let mut input = i1;
let mut error = $crate::lib::std::option::Option::None;
loop {
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Err(Err::Error(_)) => {
break;
},
Err(e) => {
error = $crate::lib::std::option::Option::Some(e);
break;
},
Ok((i, o)) => {
if i.input_len() == input.input_len() {
break;
}
res.push(o);
input = i;
}
}
}
match error {
$crate::lib::std::option::Option::Some(e) => Err(e),
$crate::lib::std::option::Option::None => Ok((input, res))
}
}
}
}
);
($i:expr, $f:expr) => (
many1!($i, call!($f));
);
);
#[cfg(feature = "alloc")]
#[macro_export(local_inner_macros)]
macro_rules! many_till(
(__impl $i:expr, $submac1:ident!( $($args1:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,ErrorKind};
let ret;
let mut res = $crate::lib::std::vec::Vec::new();
let mut input = $i.clone();
loop {
match $submac2!(input, $($args2)*) {
Ok((i, o)) => {
ret = Ok((i, (res, o)));
break;
},
Err(e1) => {
match $submac1!(input, $($args1)*) {
Err(Err::Error(err)) => {
fn unify_types<T>(_: &T, _: &T) {}
let e = Err::Error(error_node_position!(input, ErrorKind::ManyTill, err));
unify_types(&e1, &e);
ret = Err(e);
break;
},
Err(e) => {
ret = Err(e);
break;
},
Ok((i, o)) => {
if i == input {
ret = Err(Err::Error(error_position!(input, $crate::ErrorKind::ManyTill)));
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));
);
);
#[cfg(feature = "alloc")]
#[macro_export(local_inner_macros)]
macro_rules! many_m_n(
($i:expr, $m:expr, $n: expr, $submac:ident!( $($args:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Context,Err,Needed};
use $crate::InputLength;
let mut res = $crate::lib::std::vec::Vec::with_capacity($m);
let mut input = $i.clone();
let mut count: usize = 0;
let mut err = false;
let mut incomplete: $crate::lib::std::option::Option<Needed> = $crate::lib::std::option::Option::None;
let mut failure: $crate::lib::std::option::Option<Context<_,_>> = $crate::lib::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 = $crate::lib::std::option::Option::Some(i);
break;
},
Err(Err::Failure(e)) => {
failure = $crate::lib::std::option::Option::Some(e);
break;
},
}
}
if count < $m {
if err {
Err(Err::Error(error_position!($i, $crate::ErrorKind::ManyMN)))
} else {
match failure {
$crate::lib::std::option::Option::Some(i) => Err(Err::Failure(i)),
$crate::lib::std::option::Option::None => match incomplete {
$crate::lib::std::option::Option::Some(i) => $crate::need_more($i, i),
$crate::lib::std::option::Option::None => $crate::need_more($i, Needed::Unknown)
}
}
}
} else {
match failure {
$crate::lib::std::option::Option::Some(i) => Err(Err::Failure(i)),
$crate::lib::std::option::Option::None => match incomplete {
$crate::lib::std::option::Option::Some(i) => $crate::need_more($i, i),
$crate::lib::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! many0_count {
($i:expr, $submac:ident!( $($args:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err, AtEof};
let ret;
let mut count: usize = 0;
let mut input = $i.clone();
loop {
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Ok((i, _)) => {
if i == input {
if i.at_eof() {
ret = Ok((input, count));
}
else {
ret = Err(Err::Error(error_position!(input, $crate::ErrorKind::Many0Count)));
}
break;
}
count += 1;
input = i;
},
Err(Err::Error(_)) => {
ret = Ok((input, count));
break;
},
Err(e) => {
ret = Err(e);
break;
},
}
}
ret
}
);
($i:expr, $f:expr) => (
many0_count!($i, call!($f));
);
}
#[macro_export]
macro_rules! many1_count {
($i:expr, $submac:ident!( $($args:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::Err;
use $crate::InputLength;
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(Err::Error(_)) => Err(Err::Error(
error_position!(i_, $crate::ErrorKind::Many1Count)
)),
Err(Err::Failure(_)) => Err(Err::Failure(
error_position!(i_, $crate::ErrorKind::Many1Count)
)),
Err(i) => Err(i),
Ok((i1, _)) => {
let mut count: usize = 1;
let mut input = i1;
let mut error = $crate::lib::std::option::Option::None;
loop {
let input_ = input.clone();
match $submac!(input_, $($args)*) {
Err(Err::Error(_)) => {
break;
},
Err(e) => {
error = $crate::lib::std::option::Option::Some(e);
break;
},
Ok((i, _)) => {
if i.input_len() == input.input_len() {
break;
}
count += 1;
input = i;
},
}
}
match error {
$crate::lib::std::option::Option::Some(e) => Err(e),
$crate::lib::std::option::Option::None => Ok((input, count)),
}
},
}
}
);
($i:expr, $f:expr) => (
many1_count!($i, call!($f));
);
}
#[cfg(feature = "alloc")]
#[macro_export(local_inner_macros)]
macro_rules! count(
($i:expr, $submac:ident!( $($args:tt)* ), $count: expr) => (
{
use $crate::lib::std::result::Result::*;
use $crate::Err;
let ret;
let mut input = $i.clone();
let mut res = $crate::lib::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(e)) => {
fn unify_types<T>(_: &T, _: &T) {}
let e2 = error_position!($i, $crate::ErrorKind::Count);
unify_types(&e, &e2);
ret = Err(Err::Error(e2));
break;
},
Err(e) => {
ret = Err(e);
break;
},
}
}
ret
}
);
($i:expr, $f:expr, $count: expr) => (
count!($i, call!($f), $count);
);
);
#[macro_export(local_inner_macros)]
macro_rules! count_fixed (
($i:expr, $typ:ty, $submac:ident!( $($args:tt)* ), $count: expr) => (
{
use $crate::lib::std::result::Result::*;
use $crate::Err;
let ret;
let mut input = $i.clone();
let mut res: [$typ; $count] = unsafe{[$crate::lib::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(e)) => {
fn unify_types<T>(_: &T, _: &T) {}
let e2 = error_position!($i, $crate::ErrorKind::Count);
unify_types(&e, &e2);
ret = Err(Err::Error(e2));
break;
},
Err(e) => {
ret = Err(e);
break;
},
}
}
ret
}
);
($i:expr, $typ: ty, $f:expr, $count: expr) => (
count_fixed!($i, $typ, call!($f), $count);
);
);
#[macro_export(local_inner_macros)]
macro_rules! length_count(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use $crate::lib::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(local_inner_macros)]
macro_rules! length_data(
($i:expr, $submac:ident!( $($args:tt)* )) => ({
use $crate::lib::std::result::Result::*;
use $crate::{Convert,Err};
match $submac!($i, $($args)*) {
Err(e) => Err(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(local_inner_macros)]
macro_rules! length_value(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,Convert};
match $submac!($i, $($args)*) {
Err(e) => Err(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(local_inner_macros)]
macro_rules! fold_many0(
($i:expr, $submac:ident!( $($args:tt)* ), $init:expr, $f:expr) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,AtEof};
let ret;
let f = $f;
let mut res = $init;
let mut input = $i.clone();
loop {
match $submac!(input, $($args)*) {
Ok((i, o)) => {
if i == input {
if i.at_eof() {
ret = Ok((input, res));
} else {
ret = Err(Err::Error(error_position!(input, $crate::ErrorKind::Many0)));
}
break;
}
res = f(res, o);
input = i;
},
Err(Err::Error(_)) => {
ret = Ok((input, res));
break;
},
Err(e) => {
ret = Err(e);
break;
},
}
}
ret
}
);
($i:expr, $f:expr, $init:expr, $fold_f:expr) => (
fold_many0!($i, call!($f), $init, $fold_f);
);
);
#[macro_export(local_inner_macros)]
macro_rules! fold_many1(
($i:expr, $submac:ident!( $($args:tt)* ), $init:expr, $f:expr) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,Needed,InputLength,Context,AtEof};
match $submac!($i, $($args)*) {
Err(Err::Error(_)) => Err(Err::Error(
error_position!($i, $crate::ErrorKind::Many1)
)),
Err(Err::Failure(_)) => Err(Err::Failure(
error_position!($i, $crate::ErrorKind::Many1)
)),
Err(Err::Incomplete(i)) => Err(Err::Incomplete(i)),
Ok((i1,o1)) => {
let f = $f;
let mut acc = f($init, o1);
let mut input = i1;
let mut incomplete: $crate::lib::std::option::Option<Needed> =
$crate::lib::std::option::Option::None;
let mut failure: $crate::lib::std::option::Option<Context<_,_>> =
$crate::lib::std::option::Option::None;
loop {
match $submac!(input, $($args)*) {
Err(Err::Error(_)) => {
break;
},
Err(Err::Incomplete(i)) => {
incomplete = $crate::lib::std::option::Option::Some(i);
break;
},
Err(Err::Failure(e)) => {
failure = $crate::lib::std::option::Option::Some(e);
break;
},
Ok((i, o)) => {
if i.input_len() == input.input_len() {
if !i.at_eof() {
failure = $crate::lib::std::option::Option::Some(error_position!(i, $crate::ErrorKind::Many1));
}
break;
}
acc = f(acc, o);
input = i;
}
}
}
match failure {
$crate::lib::std::option::Option::Some(e) => Err(Err::Failure(e)),
$crate::lib::std::option::Option::None => match incomplete {
$crate::lib::std::option::Option::Some(i) => $crate::need_more($i, i),
$crate::lib::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(local_inner_macros)]
macro_rules! fold_many_m_n(
($i:expr, $m:expr, $n: expr, $submac:ident!( $($args:tt)* ), $init:expr, $f:expr) => (
{
use $crate::lib::std::result::Result::*;
use $crate::{Err,Needed};
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: $crate::lib::std::option::Option<Needed> = $crate::lib::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 = $crate::lib::std::option::Option::Some(i);
break;
},
}
}
if count < $m {
if err {
Err(Err::Error(error_position!($i, $crate::ErrorKind::ManyMN)))
} else {
match incomplete {
$crate::lib::std::option::Option::Some(i) => Err(Err::Incomplete(i)),
$crate::lib::std::option::Option::None => Err(Err::Incomplete(Needed::Unknown))
}
}
} else {
match incomplete {
$crate::lib::std::option::Option::Some(i) => Err(Err::Incomplete(i)),
$crate::lib::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, IResult, Needed};
use nom::{digit, be_u16, be_u8, le_u16};
use lib::std::str::{self, FromStr};
#[cfg(feature = "alloc")]
use lib::std::vec::Vec;
use util::ErrorKind;
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 $crate::lib::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($crate::Context::Code($i, $crate::ErrorKind::Tag::<u32>)))
} 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 = "alloc")]
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!(d, ErrorKind::SeparatedList)))
);
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 = "alloc")]
fn separated_list_complete() {
use nom::alpha;
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[..a.len() - 1]])));
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 = "alloc")]
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!(c, ErrorKind::Tag)))
);
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 = "alloc")]
fn separated_nonempty_list_complete() {
use nom::alpha;
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[..a.len() - 1]])));
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!(&b"123"[..], ErrorKind::Alpha)))
);
}
#[test]
#[cfg(feature = "alloc")]
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"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(multi(&b""[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(
multi_empty(&b"abcdef"[..]),
Err(Err::Error(error_position!(
&b"abcdef"[..],
ErrorKind::Many0
)))
);
}
#[cfg(nightly)]
use test::Bencher;
#[cfg(nightly)]
#[bench]
fn many0_bench(b: &mut Bencher) {
named!(multi<&[u8],Vec<&[u8]> >, many0!(tag!("abcd")));
b.iter(|| multi(&b"abcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcd"[..]));
}
#[test]
#[cfg(feature = "alloc")]
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!(c, ErrorKind::Many1)))
);
assert_eq!(multi(d), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "alloc")]
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!(
&c[..],
ErrorKind::ManyTill,
error_position!(&c[..], ErrorKind::Tag)
)))
);
}
#[test]
#[cfg(feature = "std")]
fn infinite_many() {
fn tst(input: &[u8]) -> IResult<&[u8], &[u8]> {
println!("input: {:?}", input);
Err(Err::Error(error_position!(input, ErrorKind::Custom(0u32))))
}
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!(a, ErrorKind::Many1)))
);
}
#[test]
#[cfg(feature = "alloc")]
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!(a, ErrorKind::ManyMN)))
);
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 = "alloc")]
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!(&b"xxx"[..], ErrorKind::Count)))
);
assert_eq!(
cnt_2(&b"xxxabcabcdef"[..]),
Err(Err::Error(error_position!(
&b"xxxabcabcdef"[..],
ErrorKind::Count
)))
);
assert_eq!(
cnt_2(&b"abcxxxabcdef"[..]),
Err(Err::Error(error_position!(
&b"abcxxxabcdef"[..],
ErrorKind::Count
)))
);
}
#[test]
#[cfg(feature = "alloc")]
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!(&b"xxx"[..], ErrorKind::Count)))
);
assert_eq!(
cnt_2(&b"xxxabcabcdef"[..]),
Err(Err::Error(error_position!(
&b"xxxabcabcdef"[..],
ErrorKind::Count
)))
);
assert_eq!(
cnt_2(&b"abcxxxabcdef"[..]),
Err(Err::Error(error_position!(
&b"abcxxxabcdef"[..],
ErrorKind::Count
)))
);
}
#[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], fix_error!(NilError, 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!(error, ErrorKind::Count)))
);
assert_eq!(
counter_2(error_1),
Err(Err::Error(error_position!(
error_1_remain,
ErrorKind::Count
)))
);
assert_eq!(
counter_2(error_2),
Err(Err::Error(error_position!(
error_2_remain,
ErrorKind::Count
)))
);
}
named!(pub number<u32>, map_res!(
map_res!(
digit,
str::from_utf8
),
FromStr::from_str
));
#[test]
#[cfg(feature = "alloc")]
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!(&b"xxx"[..], ErrorKind::Digit)))
);
assert_eq!(
cnt(&b"2abcxxx"[..]),
Err(Err::Error(error_position!(
&b"abcxxx"[..],
ErrorKind::Count
)))
);
}
#[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!(&b"xxx"[..], ErrorKind::Digit)))
);
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!(&b""[..], ErrorKind::Complete)))
);
assert_eq!(
length_value_2(&i1),
Err(Err::Error(error_position!(&b""[..], ErrorKind::Complete)))
);
let i2 = [1, 5, 6, 3];
assert_eq!(
length_value_1(&i2),
Err(Err::Error(error_position!(&i2[1..2], ErrorKind::Complete)))
);
assert_eq!(
length_value_2(&i2),
Err(Err::Error(error_position!(&i2[1..2], ErrorKind::Complete)))
);
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 = "alloc")]
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"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(multi(&b""[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(
multi_empty(&b"abcdef"[..]),
Err(Err::Error(error_position!(
&b"abcdef"[..],
ErrorKind::Many0
)))
);
}
#[test]
#[cfg(feature = "alloc")]
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!(c, ErrorKind::Many1)))
);
assert_eq!(multi(d), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
#[cfg(feature = "alloc")]
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!(a, ErrorKind::ManyMN)))
);
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]
fn many0_count() {
named!(
count0_nums(&[u8]) -> usize,
many0_count!(pair!(digit, tag!(",")))
);
assert_eq!(
count0_nums(&b"123,junk"[..]),
Ok((&b"junk"[..], 1))
);
assert_eq!(
count0_nums(&b"123,45,junk"[..]),
Ok((&b"junk"[..], 2))
);
assert_eq!(
count0_nums(&b"1,2,3,4,5,6,7,8,9,0,junk"[..]),
Ok((&b"junk"[..], 10))
);
assert_eq!(
count0_nums(&b"hello"[..]),
Ok((&b"hello"[..], 0))
);
}
#[test]
fn many1_count() {
named!(
count1_nums(&[u8]) -> usize,
many1_count!(pair!(digit, tag!(",")))
);
assert_eq!(
count1_nums(&b"123,45,junk"[..]),
Ok((&b"junk"[..], 2))
);
assert_eq!(
count1_nums(&b"1,2,3,4,5,6,7,8,9,0,junk"[..]),
Ok((&b"junk"[..], 10))
);
assert_eq!(
count1_nums(&b"hello"[..]),
Err(Err::Error(error_position!(
&b"hello"[..],
ErrorKind::Many1Count
)))
);
}
}