peel-ip 0.2.0

Packet parsing for the Internet Protocol Suite
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

//! Transport Layer packets
pub mod tcp;
pub mod tls;
pub mod udp;

use prelude::*;
use path::error::PathResult;

/// Track a connection based in the current parsing result and return the connection data
pub fn track_connection<'a>(path: Option<&'a mut PathIp>,
                            result: Option<&ParserResultVec>,
                            src_port: u16,
                            dst_port: u16)
                            -> PathResult<()> {
    // Get the identifier
    let identifier = match result {
        Some(vector) => {
            match vector.get(1) {

                Some(ref any) => {
                    match (any.downcast_ref::<Ipv4Packet>(), any.downcast_ref::<Ipv6Packet>()) {
                        // IPv4
                        (Some(p), _) => {
                            Some(Identifier::new(IpAddr::V4(p.src),
                                                 src_port,
                                                 IpAddr::V4(p.dst),
                                                 dst_port,
                                                 p.protocol))
                        }

                        // IPv6
                        (_, Some(p)) => {
                            Some(Identifier::new(IpAddr::V6(p.src),
                                                 src_port,
                                                 IpAddr::V6(p.dst),
                                                 dst_port,
                                                 p.next_header))
                        }

                        _ => None,
                    }
                }

                // Previous result found, but not the correct one
                _ => None,
            }
        }
        None => None,
    };

    // Just track the connection, do nothing additional with the data
    if let (Some(path), Some(identifier)) = (path, identifier) {
        path.track(identifier)?;
    }

    Ok(())
}