ch4 0.12.0

DNS Client Tool
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use anyhow::Result;
use base64::{engine::general_purpose::STANDARD as Base64Engine, Engine as _};
use chrono::{DateTime, NaiveDateTime, Utc};
use rsdns::records::Type;
use serde::{Deserialize, Serialize};
use std::{
    fs::OpenOptions,
    net::SocketAddr,
    str::FromStr,
    time::{Duration, SystemTime},
};

#[derive(Serialize, Deserialize, Debug, Copy, Clone)]
pub struct EncodedTime {
    pub secs: i64,
    pub nanos: u32,
}

impl From<EncodedTime> for NaiveDateTime {
    fn from(t: EncodedTime) -> Self {
        NaiveDateTime::from_timestamp_opt(t.secs, t.nanos).expect("failed to convert a timestamp")
    }
}

impl From<SystemTime> for EncodedTime {
    fn from(st: SystemTime) -> Self {
        let dt: DateTime<Utc> = DateTime::from(st);
        Self {
            secs: dt.timestamp(),
            nanos: dt.timestamp_subsec_nanos(),
        }
    }
}

impl From<EncodedTime> for SystemTime {
    fn from(et: EncodedTime) -> Self {
        let ndt: NaiveDateTime = et.into();
        let dt: DateTime<Utc> = DateTime::from_naive_utc_and_offset(ndt, Utc);
        dt.into()
    }
}

#[derive(Serialize, Deserialize, Debug, Copy, Clone)]
pub struct EncodedDuration {
    pub secs: u64,
    pub nanos: u32,
}

impl From<Duration> for EncodedDuration {
    fn from(d: Duration) -> Self {
        Self {
            secs: d.as_secs(),
            nanos: d.subsec_nanos(),
        }
    }
}

impl From<EncodedDuration> for Duration {
    fn from(ed: EncodedDuration) -> Self {
        Duration::new(ed.secs, ed.nanos)
    }
}

#[derive(Serialize, Deserialize, Debug)]
pub struct EncodedMessage {
    pub data: String,
    pub qname: Option<String>,
    pub qtype: Option<String>,
    pub nameserver: Option<String>,
    pub timestamp: Option<EncodedTime>,
    pub duration: Option<EncodedDuration>,
}

impl EncodedMessage {
    pub fn encode(
        msg: &[u8],
        qname: Option<&str>,
        qtype: Option<Type>,
        nameserver: Option<SocketAddr>,
        ts: Option<SystemTime>,
        elapsed: Option<Duration>,
    ) -> Result<serde_json::Value> {
        let res = Self {
            data: Base64Engine.encode(msg),
            qname: qname.map(|s| s.to_string()),
            qtype: qtype.map(|t| t.to_string()),
            nameserver: nameserver.map(|ns| ns.to_string()),
            timestamp: ts.map(EncodedTime::from),
            duration: elapsed.map(EncodedDuration::from),
        };
        Ok(serde_json::to_value(res)?)
    }

    pub fn save_all(v: &[serde_json::Value], path: &str) -> Result<()> {
        let f = OpenOptions::new()
            .write(true)
            .create(true)
            .truncate(true)
            .open(path)?;
        serde_json::to_writer_pretty(f, v)?;
        Ok(())
    }

    pub fn load_all(path: &str) -> Result<Vec<EncodedMessage>> {
        let f = OpenOptions::new().read(true).open(path)?;
        let v: Vec<EncodedMessage> = serde_json::from_reader(f)?;
        Ok(v)
    }

    pub fn qname(&self) -> Option<&str> {
        self.qname.as_deref()
    }

    pub fn qtype(&self) -> Option<Type> {
        if let Some(ref s) = self.qtype {
            if let Ok(t) = Type::from_str(s) {
                return Some(t);
            }
        }
        None
    }

    pub fn nameserver(&self) -> Option<SocketAddr> {
        if let Some(ref ns) = self.nameserver {
            if let Ok(sa) = SocketAddr::from_str(ns) {
                return Some(sa);
            }
        }
        None
    }

    pub fn msg(&self) -> Vec<u8> {
        if let Ok(v) = Base64Engine.decode(&self.data) {
            v
        } else {
            Vec::new()
        }
    }

    pub fn time(&self) -> Option<SystemTime> {
        self.timestamp.map(|et| et.into())
    }

    pub fn elapsed(&self) -> Option<Duration> {
        self.duration.map(|d| d.into())
    }
}