1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
use crate::snmp::*;
use crate::snmpv3::*;
use der_parser::ber::BerTag;
use der_parser::der::*;
use nom::{Err,ErrorKind,IResult};
use std::str;

#[derive(Debug,PartialEq)]
pub enum SnmpGenericMessage<'a> {
    V1(SnmpMessage<'a>),
    V2(SnmpMessage<'a>),
    V3(SnmpV3Message<'a>),
}

pub fn parse_snmp_generic_message<'a>(i:&'a[u8]) -> IResult<&'a[u8],SnmpGenericMessage> {
    der_read_element_header(i).and_then(|(rem,hdr)| {
        if hdr.tag != BerTag::Sequence { return Err(Err::Error(error_position!(i, ErrorKind::Tag))); }
        take!(rem, hdr.len as usize)
    }).and_then(|(rem,data)| {
        parse_der_u32(data).and_then(|(r,version)| {
            match version {
                0 => {
                    do_parse!(
                        r,
                        community: map_res!(parse_der_octetstring_as_slice, str::from_utf8) >>
                        pdu:       parse_snmp_v1_pdu >>
                        (
                            SnmpGenericMessage::V1(SnmpMessage{
                                version,
                                community: community.to_string(),
                                pdu
                            })
                        )
                    )
                }
                1 => {
                    do_parse!(
                        r,
                        community: map_res!(parse_der_octetstring_as_slice, str::from_utf8) >>
                        pdu:       parse_snmp_v2c_pdu >>
                        (
                            SnmpGenericMessage::V2(SnmpMessage{
                                version,
                                community: community.to_string(),
                                pdu
                            })
                        )
                    )
                }
                3 => {
                    do_parse!(
                        r,
                        hdr:  parse_snmp_v3_headerdata >>
                        secp: map_res!(parse_der_octetstring, |x: DerObject<'a>| parse_secp(&x,&hdr)) >>
                        data: apply!(parse_snmp_v3_data,&hdr) >>
                        (
                            SnmpGenericMessage::V3(SnmpV3Message{
                                version,
                                header_data: hdr,
                                security_params: secp,
                                data
                            })
                        )
                    )
                },
                _ => { return Err(Err::Error(error_position!(i, ErrorKind::Tag))); }
            }
        }).map(|(_,res)| (rem,res)) // adjust remaining bytes
    })
}