summaryrefslogtreecommitdiff
path: root/src/dns/packet/question.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/dns/packet/question.rs')
-rw-r--r--src/dns/packet/question.rs31
1 files changed, 31 insertions, 0 deletions
diff --git a/src/dns/packet/question.rs b/src/dns/packet/question.rs
new file mode 100644
index 0000000..9042e1c
--- /dev/null
+++ b/src/dns/packet/question.rs
@@ -0,0 +1,31 @@
+use super::{buffer::PacketBuffer, query::QueryType, Result};
+
+#[derive(Debug, Clone, PartialEq, Eq, Hash)]
+pub struct DnsQuestion {
+ pub name: String,
+ pub qtype: QueryType,
+}
+
+impl DnsQuestion {
+ pub fn new(name: String, qtype: QueryType) -> Self {
+ Self { name, qtype }
+ }
+
+ pub fn read(&mut self, buffer: &mut PacketBuffer) -> Result<()> {
+ buffer.read_qname(&mut self.name)?;
+ self.qtype = QueryType::from_num(buffer.read_u16()?); // qtype
+ let _ = buffer.read_u16()?; // class
+
+ Ok(())
+ }
+
+ pub fn write(&self, buffer: &mut PacketBuffer) -> Result<()> {
+ buffer.write_qname(&self.name)?;
+
+ let typenum = self.qtype.to_num();
+ buffer.write_u16(typenum)?;
+ buffer.write_u16(1)?;
+
+ Ok(())
+ }
+}