#[macro_export]
macro_rules! tuple (
($i:expr, $($rest:tt)*) => (
{
tuple_parser!($i, (), $($rest)*)
}
);
);
#[doc(hidden)]
#[macro_export]
macro_rules! tuple_parser (
($i:expr, ($($parsed:tt),*), $e:path, $($rest:tt)*) => (
tuple_parser!($i, ($($parsed),*), call!($e), $($rest)*);
);
($i:expr, (), $submac:ident!( $($args:tt)* ), $($rest:tt)*) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
let i_ = i.clone();
tuple_parser!(i_, (o), $($rest)*)
}
}
}
);
($i:expr, ($($parsed:tt)*), $submac:ident!( $($args:tt)* ), $($rest:tt)*) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
let i_ = i.clone();
tuple_parser!(i_, ($($parsed)* , o), $($rest)*)
}
}
}
);
($i:expr, ($($parsed:tt),*), $e:path) => (
tuple_parser!($i, ($($parsed),*), call!($e));
);
($i:expr, (), $submac:ident!( $($args:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
Ok((i, (o)))
}
}
}
);
($i:expr, ($($parsed:expr),*), $submac:ident!( $($args:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match $submac!($i, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
Ok((i, ($($parsed),* , o)))
}
}
}
);
($i:expr, ($($parsed:expr),*)) => (
{
::std::result::Result::Ok(($i, ($($parsed),*)))
}
);
);
#[macro_export]
macro_rules! pair(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
tuple!($i, $submac!($($args)*), $submac2!($($args2)*))
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
pair!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
pair!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
pair!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! separated_pair(
($i:expr, $submac:ident!( $($args:tt)* ), $($rest:tt)+) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match tuple_parser!($i, (), $submac!($($args)*), $($rest)*) {
Err(e) => Err(Err::convert(e)),
Ok((i1, (o1, _, o2))) => {
Ok((i1, (o1, o2)))
}
}
}
);
($i:expr, $f:expr, $($rest:tt)+) => (
separated_pair!($i, call!($f), $($rest)*);
);
);
#[macro_export]
macro_rules! preceded(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match tuple!($i, $submac!($($args)*), $submac2!($($args2)*)) {
Err(e) => Err(Err::convert(e)),
Ok((remaining, (_,o))) => {
Ok((remaining, o))
}
}
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
preceded!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
preceded!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
preceded!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! terminated(
($i:expr, $submac:ident!( $($args:tt)* ), $submac2:ident!( $($args2:tt)* )) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match tuple!($i, $submac!($($args)*), $submac2!($($args2)*)) {
Err(e) => Err(Err::convert(e)),
Ok((remaining, (o,_))) => {
Ok((remaining, o))
}
}
}
);
($i:expr, $submac:ident!( $($args:tt)* ), $g:expr) => (
terminated!($i, $submac!($($args)*), call!($g));
);
($i:expr, $f:expr, $submac:ident!( $($args:tt)* )) => (
terminated!($i, call!($f), $submac!($($args)*));
);
($i:expr, $f:expr, $g:expr) => (
terminated!($i, call!($f), call!($g));
);
);
#[macro_export]
macro_rules! delimited(
($i:expr, $submac:ident!( $($args:tt)* ), $($rest:tt)+) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
match tuple_parser!($i, (), $submac!($($args)*), $($rest)*) {
Err(e) => Err(Err::convert(e)),
Ok((i1, (_, o, _))) => {
Ok((i1, o))
}
}
}
);
($i:expr, $f:expr, $($rest:tt)+) => (
delimited!($i, call!($f), $($rest)*);
);
);
#[macro_export]
macro_rules! do_parse (
(__impl $i:expr, ( $($rest:expr),* )) => (
::std::result::Result::Ok(($i, ( $($rest),* )))
);
(__impl $i:expr, $field:ident : $submac:ident!( $($args:tt)* ) ) => (
do_parse!(__impl $i, $submac!( $($args)* ))
);
(__impl $i:expr, $submac:ident!( $($args:tt)* ) ) => (
compile_error!("do_parse is missing the return value. A do_parse call must end
with a return value between parenthesis, as follows:
do_parse!(
a: tag!(\"abcd\") >>
b: tag!(\"efgh\") >>
( Value { a: a, b: b } )
");
);
(__impl $i:expr, $field:ident : $submac:ident!( $($args:tt)* ) ~ $($rest:tt)* ) => (
compile_error!("do_parse uses >> as separator, not ~");
);
(__impl $i:expr, $submac:ident!( $($args:tt)* ) ~ $($rest:tt)* ) => (
compile_error!("do_parse uses >> as separator, not ~");
);
(__impl $i:expr, $field:ident : $e:ident ~ $($rest:tt)*) => (
do_parse!(__impl $i, $field: call!($e) ~ $($rest)*);
);
(__impl $i:expr, $e:ident ~ $($rest:tt)*) => (
do_parse!(__impl $i, call!($e) ~ $($rest)*);
);
(__impl $i:expr, $e:ident >> $($rest:tt)*) => (
do_parse!(__impl $i, call!($e) >> $($rest)*);
);
(__impl $i:expr, $submac:ident!( $($args:tt)* ) >> $($rest:tt)*) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,_)) => {
let i_ = i.clone();
do_parse!(__impl i_, $($rest)*)
},
}
}
);
(__impl $i:expr, $field:ident : $e:ident >> $($rest:tt)*) => (
do_parse!(__impl $i, $field: call!($e) >> $($rest)*);
);
(__impl $i:expr, $field:ident : $submac:ident!( $($args:tt)* ) >> $($rest:tt)*) => (
{
use ::std::result::Result::*;
use $crate::{Err,Convert};
let i_ = $i.clone();
match $submac!(i_, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
let $field = o;
let i_ = i.clone();
do_parse!(__impl i_, $($rest)*)
},
}
}
);
(__impl $i:expr, $e:ident >> ( $($rest:tt)* )) => (
do_parse!(__impl $i, call!($e) >> ( $($rest)* ));
);
(__impl $i:expr, $submac:ident!( $($args:tt)* ) >> ( $($rest:tt)* )) => ({
use ::std::result::Result::*;
use $crate::{Err,Convert};
match $submac!($i, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,_)) => {
Ok((i, ( $($rest)* )))
},
}
});
(__impl $i:expr, $field:ident : $e:ident >> ( $($rest:tt)* )) => (
do_parse!(__impl $i, $field: call!($e) >> ( $($rest)* ) );
);
(__impl $i:expr, $field:ident : $submac:ident!( $($args:tt)* ) >> ( $($rest:tt)* )) => ({
use ::std::result::Result::*;
use $crate::{Err,Convert};
match $submac!($i, $($args)*) {
Err(e) => Err(Err::convert(e)),
Ok((i,o)) => {
let $field = o;
Ok((i, ( $($rest)* )))
},
}
});
($i:expr, $($rest:tt)*) => (
{
do_parse!(__impl $i, $($rest)*)
}
);
($submac:ident!( $($args:tt)* ) >> $($rest:tt)* ) => (
compile_error!("if you are using do_parse outside of a named! macro, you must
pass the input data as first argument, like this:
let res = do_parse!(input,
a: tag!(\"abcd\") >>
b: tag!(\"efgh\") >>
( Value { a: a, b: b } )
);");
);
($e:ident! >> $($rest:tt)* ) => (
do_parse!( call!($e) >> $($rest)*);
);
);
#[cfg(test)]
mod tests {
use internal::{Err,Needed,IResult};
use nom::be_u16;
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::need_more;
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 {
need_more($i, Needed::Size(blen))
} else {
Ok((&$i[blen..], reduced))
};
res
}
);
);
macro_rules! take (
($i:expr, $count:expr) => (
{
use $crate::need_more;
let cnt = $count as usize;
let res:IResult<&[u8],&[u8],u32> = if $i.len() < cnt {
need_more($i, Needed::Size(cnt))
} else {
Ok((&$i[cnt..],&$i[0..cnt]))
};
res
}
);
);
#[derive(PartialEq,Eq,Debug)]
struct B {
a: u8,
b: u8
}
#[derive(PartialEq,Eq,Debug)]
struct C {
a: u8,
b: Option<u8>
}
#[cfg(feature = "verbose-errors")]
use util::{error_to_list, add_error_pattern, print_error};
#[cfg(feature = "verbose-errors")]
use verbose_errors::Context;
#[cfg(feature = "verbose-errors")]
use util::ErrorKind;
#[cfg(feature = "verbose-errors")]
fn error_to_string<P:Clone+PartialEq>(e: &Context<P,u32>) -> &'static str {
let v:Vec<(P,ErrorKind<u32>)> = error_to_list(e);
let collected: Vec<ErrorKind<u32>> = v.iter().map(|&(_, ref e)| e.clone()).collect();
if &collected[..] == [ErrorKind::Custom(42),ErrorKind::Tag] {
"missing `ijkl` tag"
} else if &collected[..] == [ErrorKind::Custom(42), ErrorKind::Custom(128), ErrorKind::Tag] {
"missing `mnop` tag after `ijkl`"
} else {
"unrecognized error"
}
}
#[cfg(feature = "verbose-errors")]
use std::collections;
#[cfg(feature = "verbose-errors")]
#[test]
fn err() {
named!(err_test, alt!(
tag!("abcd") |
preceded!(tag!("efgh"), return_error!(ErrorKind::Custom(42),
do_parse!(
tag!("ijkl") >>
res: return_error!(ErrorKind::Custom(128), tag!("mnop")) >>
(res)
)
)
)
));
let a = &b"efghblah"[..];
let b = &b"efghijklblah"[..];
let c = &b"efghijklmnop"[..];
let blah = &b"blah"[..];
let res_a = err_test(a);
let res_b = err_test(b);
let res_c = err_test(c);
assert_eq!(res_a, Err(Err::Failure(error_node_position!(ErrorKind::Custom(42), blah, error_position!(ErrorKind::Tag, blah)))));
assert_eq!(res_b, Err(Err::Failure(error_node_position!(ErrorKind::Custom(42), &b"ijklblah"[..], error_node_position!(ErrorKind::Custom(128), blah, error_position!(ErrorKind::Tag, blah))))));
assert_eq!(res_c, Ok((&b""[..], &b"mnop"[..])));
let mut err_map = collections::HashMap::new();
assert!(add_error_pattern(&mut err_map, err_test(&b"efghpouet"[..]), "missing `ijkl` tag"));
assert!(add_error_pattern(&mut err_map, err_test(&b"efghijklpouet"[..]), "missing `mnop` tag after `ijkl`"));
let res_a2 = res_a.clone();
match res_a {
Err(Err::Error(e)) | Err(Err::Failure(e)) => {
let collected: Vec<ErrorKind<u32>> = error_to_list(&e).iter().map(|&(_,ref e)| e.clone()).collect();
assert_eq!(collected, [ErrorKind::Custom(42), ErrorKind::Tag]);
assert_eq!(error_to_string(&e), "missing `ijkl` tag");
assert_eq!(err_map.get(&error_to_list(&e)), None);
},
_ => panic!()
};
let res_b2 = res_b.clone();
match res_b {
Err(Err::Error(e)) | Err(Err::Failure(e)) => {
let collected: Vec<ErrorKind<u32>> = error_to_list(&e).iter().map(|&(_,ref e)| e.clone()).collect();
assert_eq!(collected, [ErrorKind::Custom(42), ErrorKind::Custom(128), ErrorKind::Tag]);
assert_eq!(error_to_string(&e), "missing `mnop` tag after `ijkl`");
assert_eq!(err_map.get(&error_to_list(&e)), None);
},
_ => panic!()
};
print_error(a, res_a2);
print_error(b, res_b2);
}
#[allow(unused_variables)]
#[test]
fn add_err() {
named!(err_test,
preceded!(tag!("efgh"), add_return_error!(ErrorKind::Custom(42),
do_parse!(
tag!("ijkl") >>
res: add_return_error!(ErrorKind::Custom(128), tag!("mnop")) >>
(res)
)
)
));
let a = &b"efghblah"[..];
let b = &b"efghijklblah"[..];
let c = &b"efghijklmnop"[..];
let blah = &b"blah"[..];
let res_a = err_test(a);
let res_b = err_test(b);
let res_c = err_test(c);
assert_eq!(res_a, Err(Err::Error(error_node_position!(ErrorKind::Custom(42), blah, error_position!(ErrorKind::Tag, blah)))));
assert_eq!(res_b, Err(Err::Error(error_node_position!(ErrorKind::Custom(42), &b"ijklblah"[..], error_node_position!(ErrorKind::Custom(128), blah, error_position!(ErrorKind::Tag, blah))))));
assert_eq!(res_c, Ok((&b""[..], &b"mnop"[..])));
}
#[test]
fn complete() {
named!(err_test,
do_parse!(
tag!("ijkl") >>
res: complete!(tag!("mnop")) >>
(res)
)
);
let a = &b"ijklmn"[..];
let res_a = err_test(a);
assert_eq!(res_a, Err(Err::Error(error_position!(ErrorKind::Complete, &b"mn"[..]))));
}
#[test]
fn pair() {
named!( tag_abc, tag!("abc") );
named!( tag_def, tag!("def") );
named!( pair_abc_def<&[u8],(&[u8], &[u8])>, pair!(tag_abc, tag_def) );
assert_eq!(pair_abc_def(&b"abcdefghijkl"[..]), Ok((&b"ghijkl"[..], (&b"abc"[..], &b"def"[..]))));
assert_eq!(pair_abc_def(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(pair_abc_def(&b"abcd"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(pair_abc_def(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
assert_eq!(pair_abc_def(&b"xxxdef"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxxdef"[..]))));
assert_eq!(pair_abc_def(&b"abcxxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
}
#[test]
fn separated_pair() {
named!( tag_abc, tag!("abc") );
named!( tag_def, tag!("def") );
named!( tag_separator, tag!(",") );
named!( sep_pair_abc_def<&[u8],(&[u8], &[u8])>, separated_pair!(tag_abc, tag_separator, tag_def) );
assert_eq!(sep_pair_abc_def(&b"abc,defghijkl"[..]), Ok((&b"ghijkl"[..], (&b"abc"[..], &b"def"[..]))));
assert_eq!(sep_pair_abc_def(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(sep_pair_abc_def(&b"abc,d"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(sep_pair_abc_def(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
assert_eq!(sep_pair_abc_def(&b"xxx,def"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx,def"[..]))));
assert_eq!(sep_pair_abc_def(&b"abc,xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
}
#[test]
fn preceded() {
named!( tag_abcd, tag!("abcd") );
named!( tag_efgh, tag!("efgh") );
named!( preceded_abcd_efgh<&[u8], &[u8]>, preceded!(tag_abcd, tag_efgh) );
assert_eq!(preceded_abcd_efgh(&b"abcdefghijkl"[..]), Ok((&b"ijkl"[..], &b"efgh"[..])));
assert_eq!(preceded_abcd_efgh(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(preceded_abcd_efgh(&b"abcde"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(preceded_abcd_efgh(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
assert_eq!(preceded_abcd_efgh(&b"xxxxdef"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxxxdef"[..]))));
assert_eq!(preceded_abcd_efgh(&b"abcdxxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
}
#[test]
fn terminated() {
named!( tag_abcd, tag!("abcd") );
named!( tag_efgh, tag!("efgh") );
named!( terminated_abcd_efgh<&[u8], &[u8]>, terminated!(tag_abcd, tag_efgh) );
assert_eq!(terminated_abcd_efgh(&b"abcdefghijkl"[..]), Ok((&b"ijkl"[..], &b"abcd"[..])));
assert_eq!(terminated_abcd_efgh(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(terminated_abcd_efgh(&b"abcde"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(terminated_abcd_efgh(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
assert_eq!(terminated_abcd_efgh(&b"xxxxdef"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxxxdef"[..]))));
assert_eq!(terminated_abcd_efgh(&b"abcdxxxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxxx"[..]))));
}
#[test]
fn delimited() {
named!( tag_abc, tag!("abc") );
named!( tag_def, tag!("def") );
named!( tag_ghi, tag!("ghi") );
named!( delimited_abc_def_ghi<&[u8], &[u8]>, delimited!(tag_abc, tag_def, tag_ghi) );
assert_eq!(delimited_abc_def_ghi(&b"abcdefghijkl"[..]), Ok((&b"jkl"[..], &b"def"[..])));
assert_eq!(delimited_abc_def_ghi(&b"ab"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(delimited_abc_def_ghi(&b"abcde"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(delimited_abc_def_ghi(&b"abcdefgh"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(delimited_abc_def_ghi(&b"xxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
assert_eq!(delimited_abc_def_ghi(&b"xxxdefghi"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxxdefghi"[..]))));
assert_eq!(delimited_abc_def_ghi(&b"abcxxxghi"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxxghi"[..]))));
assert_eq!(delimited_abc_def_ghi(&b"abcdefxxx"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"xxx"[..]))));
}
#[test]
fn tuple_test() {
named!(tuple_3<&[u8], (u16, &[u8], &[u8]) >,
tuple!( be_u16 , take!(3), tag!("fg") ) );
assert_eq!(tuple_3(&b"abcdefgh"[..]), Ok((&b"h"[..], (0x6162u16, &b"cde"[..], &b"fg"[..]))));
assert_eq!(tuple_3(&b"abcd"[..]), Err(Err::Incomplete(Needed::Size(3))));
assert_eq!(tuple_3(&b"abcde"[..]), Err(Err::Incomplete(Needed::Size(2))));
assert_eq!(tuple_3(&b"abcdejk"[..]), Err(Err::Error(error_position!(ErrorKind::Tag, &b"jk"[..]))));
}
#[test]
fn do_parse() {
fn ret_int1(i:&[u8]) -> IResult<&[u8], u8> { Ok((i,1)) };
fn ret_int2(i:&[u8]) -> IResult<&[u8], u8> { Ok((i,2)) };
named!(do_parser<&[u8], (u8, u8)>,
do_parse!(
tag!("abcd") >>
opt!(tag!("abcd")) >>
aa: ret_int1 >>
tag!("efgh") >>
bb: ret_int2 >>
tag!("efgh") >>
(aa, bb)
)
);
assert_eq!(do_parser(&b"abcdabcdefghefghX"[..]), Ok((&b"X"[..], (1, 2))));
assert_eq!(do_parser(&b"abcdefghefghX"[..]), Ok((&b"X"[..], (1, 2))));
assert_eq!(do_parser(&b"abcdab"[..]), Err(Err::Incomplete(Needed::Size(4))));
assert_eq!(do_parser(&b"abcdefghef"[..]), Err(Err::Incomplete(Needed::Size(4))));
}
#[test]
fn do_parse_dependency() {
use nom::be_u8;
named!(length_value,
do_parse!(
length: be_u8 >>
bytes: take!(length) >>
(bytes)
)
);
let a = [2u8, 3, 4, 5];
let res_a = [3u8, 4];
assert_eq!(length_value(&a[..]), Ok((&a[3..], &res_a[..])));
let b = [5u8, 3, 4, 5];
assert_eq!(length_value(&b[..]), Err(Err::Incomplete(Needed::Size(5))));
}
}