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

//! A DNS transaction.
//! Subscribable alias for [`retina_core::protocols::stream::dns::Dns`]

use retina_core::protocols::stream::dns::Dns;
use retina_core::protocols::stream::{Session, SessionData};

use super::{FromSession, SessionList};

pub type DnsTransaction = Box<Dns>;

impl FromSession for DnsTransaction {
    fn stream_protocols() -> Vec<&'static str> {
        vec!["dns"]
    }

    fn from_session(session: &Session) -> Option<&Self> {
        if let SessionData::Dns(dns) = &session.data {
            return Some(dns);
        }
        None
    }

    fn from_sessionlist(session_list: &SessionList) -> Option<&Self> {
        for session in session_list {
            if let SessionData::Dns(dns) = &session.data {
                return Some(dns);
            }
        }
        None
    }
}