adding rdata trait, tightening question parsing

This commit is contained in:
Andy Pack 2024-02-02 20:21:57 +00:00
parent 26d1a1881a
commit 0ae64f674c
Signed by: sarsoo
GPG Key ID: A55BA3536A5E0ED7
12 changed files with 448 additions and 305 deletions

View File

@ -0,0 +1,30 @@
use std::fmt::{Debug, Formatter};
use std::net::{IpAddr, Ipv4Addr};
use crate::message::answer::RData;
pub struct IpRData {
pub rdata: Ipv4Addr
}
impl Debug for IpRData {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
f.debug_struct("IP")
.field("data", &self.rdata)
.finish()
}
}
impl RData for IpRData {
fn to_bytes(&self) -> Vec<u8> {
return self.rdata.octets().to_vec();
}
}
impl IpRData {
pub fn from(rdata: Ipv4Addr) -> IpRData
{
IpRData {
rdata
}
}
}

View File

@ -1,15 +1,30 @@
mod raw_rdata;
pub use raw_rdata::RawRData;
mod ip_address;
pub use ip_address::IpRData;
#[cfg(test)]
mod tests;
use std::fmt::{Debug, Display};
use crate::byte::{four_byte_split, two_byte_split};
use crate::message::question::{DNSQuestion, QClass, QType};
use crate::message::question::{DNSQuestion, QClass, QType, QuestionParseError};
use crate::string::encode_domain_name;
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
pub trait RData: Debug {
fn to_bytes(&self) -> Vec<u8>;
}
#[derive(Debug)]
pub struct DNSAnswer {
pub name: String,
pub answer_type: QType,
pub class: QClass,
pub ttl: u32,
pub rd_length: u16,
pub r_data: Vec<u8>
pub r_data: Box<dyn RData>
}
impl DNSAnswer {
@ -36,10 +51,22 @@ impl DNSAnswer {
ret.push(rd_length_split.0);
ret.push(rd_length_split.1);
ret.append(&mut self.r_data.clone());
ret.append(&mut self.r_data.to_bytes());
return ret
}
pub fn from_query(query: &DNSQuestion, data: Box<dyn RData>, ttl: Option<u32>) -> DNSAnswer
{
DNSAnswer {
name: query.qname.clone(),
answer_type: query.qtype,
class: query.qclass,
ttl: ttl.unwrap_or(0),
rd_length: data.to_bytes().len() as u16,
r_data: data
}
}
}
pub fn answers_to_bytes(answers: &Vec<DNSAnswer>) -> Vec<u8>
@ -52,4 +79,16 @@ pub fn answers_to_bytes(answers: &Vec<DNSAnswer>) -> Vec<u8>
}
ret
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
pub enum AnswerParseError {
ShortLength(usize),
QTypeParse(u8),
QClassParse(u8)
}
pub fn answers_from_bytes(bytes: Vec<u8>, total_answers: u16) -> Result<(i32, Vec<DNSAnswer>), AnswerParseError>
{
Ok((0, vec![]))
}

View File

@ -0,0 +1,29 @@
use std::fmt::{Debug, Formatter};
use crate::message::answer::RData;
pub struct RawRData {
pub rdata: Vec<u8>
}
impl Debug for RawRData {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
f.debug_struct("RawRData")
.field("data", &self.rdata)
.finish()
}
}
impl RData for RawRData {
fn to_bytes(&self) -> Vec<u8> {
return self.rdata.clone();
}
}
impl RawRData {
pub fn from(rdata: Vec<u8>) -> RawRData
{
RawRData {
rdata
}
}
}

View File

@ -0,0 +1,27 @@
use crate::message::question::{DNSQuestion, QClass, QType, questions_from_bytes};
use super::*;
#[test]
#[ignore]
fn one_answer_back_and_forth() {
let q = DNSAnswer {
name: "google.com".to_string(),
answer_type: QType::A,
class: QClass::Internet,
ttl: 0,
rd_length: 0,
r_data: Box::from(RawRData::from(vec![]))
};
let mut q_bytes = q.to_bytes();
q_bytes.append(&mut vec![0, 0, 0, 0, 0, 0]);
let (q_read, q_reconstructed) = answers_from_bytes(q_bytes, 0).unwrap();
assert_eq!(q.name, q_reconstructed[0].name);
assert_eq!(q.answer_type, q_reconstructed[0].answer_type);
assert_eq!(q.class, q_reconstructed[0].class);
assert_eq!(q.ttl, q_reconstructed[0].ttl);
assert_eq!(q.rd_length, q_reconstructed[0].rd_length);
}

View File

@ -73,7 +73,7 @@ impl TryFrom<u16> for ResponseCode {
}
/// Represents a header including flag fields and record counts
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Clone)]
pub struct DNSHeader {
/// Random ID for associating responses with requests
pub id: u16,

View File

@ -1,291 +0,0 @@
use std::ops::Sub;
use urlencoding::{encode, decode};
use crate::string::encode_domain_name;
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Copy, Clone)]
pub enum QType {
A = 1,
NS = 2,
CNAME = 5,
SOA = 6,
WKS = 11,
PTR = 12,
HINFO = 13,
MINFO = 14,
MX = 15,
TXT = 16,
RP = 17,
AAAA = 28,
SRV = 33
}
impl TryFrom<u8> for QType {
type Error = u8;
fn try_from(v: u8) -> Result<Self, Self::Error> {
match v {
x if x == QType::A as u8 => Ok(QType::A),
x if x == QType::NS as u8 => Ok(QType::NS),
x if x == QType::CNAME as u8 => Ok(QType::CNAME),
x if x == QType::SOA as u8 => Ok(QType::SOA),
x if x == QType::WKS as u8 => Ok(QType::WKS),
x if x == QType::PTR as u8 => Ok(QType::PTR),
x if x == QType::HINFO as u8 => Ok(QType::HINFO),
x if x == QType::MINFO as u8 => Ok(QType::MINFO),
x if x == QType::MX as u8 => Ok(QType::MX),
x if x == QType::TXT as u8 => Ok(QType::TXT),
x if x == QType::RP as u8 => Ok(QType::RP),
x if x == QType::AAAA as u8 => Ok(QType::AAAA),
x if x == QType::SRV as u8 => Ok(QType::SRV),
_ => Err(v),
}
}
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Copy, Clone)]
pub enum QClass {
Internet = 1,
Chaos = 3,
Hesiod = 4,
}
impl TryFrom<u8> for QClass {
type Error = u8;
fn try_from(v: u8) -> Result<Self, Self::Error> {
match v {
x if x == QClass::Internet as u8 => Ok(QClass::Internet),
x if x == QClass::Chaos as u8 => Ok(QClass::Chaos),
x if x == QClass::Hesiod as u8 => Ok(QClass::Hesiod),
_ => Err(v),
}
}
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
pub struct DNSQuestion {
pub qname: String,
pub qtype: QType,
pub qclass: QClass
}
impl DNSQuestion {
pub fn new(qname: String, qtype: QType, qclass: QClass) -> DNSQuestion
{
DNSQuestion {
qname,
qtype,
qclass
}
}
pub fn to_bytes(&self) -> Vec<u8>
{
let mut ret = encode_domain_name(&self.qname);
ret.push(self.qtype as u8);
ret.push(self.qclass as u8);
ret
}
}
pub fn questions_to_bytes(questions: &Vec<DNSQuestion>) -> Vec<u8>
{
let mut ret = Vec::with_capacity(20);
for q in questions
{
ret.append(&mut q.to_bytes());
}
ret
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
pub enum QuestionParseError {
ShortLength(usize),
QTypeParse(u8),
QClassParse(u8)
}
pub fn questions_from_bytes(bytes: Vec<u8>, total_questions: u16) -> Result<Vec<DNSQuestion>, QuestionParseError>
{
if bytes.len() < 4
{
return Err(QuestionParseError::ShortLength(bytes.len()));
}
let mut questions: Vec<DNSQuestion> = Vec::with_capacity(total_questions as usize);
let mut current_query: Option<Vec<u8>> = None;
let mut current_length: Option<u8> = None;
let mut remaining_length: Box<u8> = Box::from(0);
let mut current_qtype: Option<u8> = None;
let mut current_qclass: Option<u8> = None;
let mut trailers_reached = false;
// let mut finished = false;
for byte in bytes {
if questions.len() != total_questions as usize {
match current_length {
None => { // next question, init lengths
current_length = Some(byte);
remaining_length = Box::from(byte);
current_query = Some(Vec::with_capacity(10));
}
Some(_) => {
if byte == 0 {
trailers_reached = true;
continue
}
if *remaining_length == 0 && !trailers_reached {
current_query.as_mut().unwrap().push('.' as u8);
current_length = Some(byte);
remaining_length = Box::from(byte);
}
else if trailers_reached { // trailer fields
match current_qtype {
None => {
current_qtype = Some(byte);
}
Some(qtype_b) => {
match current_qclass {
None => {
// current_qclass = Some(byte);
match (qtype_b.try_into(), byte.try_into()) {
(Ok(qtype), Ok(qclass)) => {
questions.push(DNSQuestion {
qname: decode(String::from_utf8(current_query.unwrap()).unwrap().as_str()).unwrap().to_string(),
qtype,
qclass
});
current_length = None;
remaining_length = Box::from(byte);
current_query = None;
current_qtype = None;
current_qclass = None;
trailers_reached = false;
}
(Err(qtype_e), _) => {
return Err(QuestionParseError::QTypeParse(qtype_e));
}
(_, Err(qclass_e)) => {
return Err(QuestionParseError::QClassParse(qclass_e));
}
}
}
Some(_) => {
}
}
}
}
} else {
current_query.as_mut().unwrap().push(byte);
*remaining_length = remaining_length.sub(1);
}
}
}
}
}
Ok(questions)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn one_question_back_and_forth() {
let q = DNSQuestion {
qname: "google.com".to_string(),
qclass: QClass::Internet,
qtype: QType::A
};
let mut q_bytes = q.to_bytes();
q_bytes.append(&mut vec![0, 0, 0, 0, 0, 0]);
let q_reconstructed = questions_from_bytes(q_bytes, 1).unwrap();
assert_eq!(q.qname, q_reconstructed[0].qname);
assert_eq!(q.qclass, q_reconstructed[0].qclass);
assert_eq!(q.qtype, q_reconstructed[0].qtype);
}
#[test]
fn two_questions_back_and_forth() {
let q = DNSQuestion {
qname: "google.com".to_string(),
qclass: QClass::Internet,
qtype: QType::A
};
let q2 = DNSQuestion {
qname: "duck.com".to_string(),
qclass: QClass::Internet,
qtype: QType::AAAA
};
let mut q_bytes = q.to_bytes();
let mut q2_bytes = q2.to_bytes();
q_bytes.append(&mut q2_bytes);
let q_reconstructed = questions_from_bytes(q_bytes, 2).unwrap();
assert_eq!(q.qname, q_reconstructed[0].qname);
assert_eq!(q.qclass, q_reconstructed[0].qclass);
assert_eq!(q.qtype, q_reconstructed[0].qtype);
assert_eq!(q2.qname, q_reconstructed[1].qname);
assert_eq!(q2.qclass, q_reconstructed[1].qclass);
assert_eq!(q2.qtype, q_reconstructed[1].qtype);
}
#[test]
fn three_questions_back_and_forth() {
let q = DNSQuestion {
qname: "google.com".to_string(),
qclass: QClass::Internet,
qtype: QType::A
};
let q2 = DNSQuestion {
qname: "duck.com".to_string(),
qclass: QClass::Internet,
qtype: QType::AAAA
};
let q3 = DNSQuestion {
qname: "facebook.com".to_string(),
qclass: QClass::Hesiod,
qtype: QType::CNAME
};
let mut q_bytes = q.to_bytes();
let mut q2_bytes = q2.to_bytes();
let mut q3_bytes = q3.to_bytes();
q_bytes.append(&mut q2_bytes);
q_bytes.append(&mut q3_bytes);
let q_reconstructed = questions_from_bytes(q_bytes, 3).unwrap();
assert_eq!(q.qname, q_reconstructed[0].qname);
assert_eq!(q.qclass, q_reconstructed[0].qclass);
assert_eq!(q.qtype, q_reconstructed[0].qtype);
assert_eq!(q2.qname, q_reconstructed[1].qname);
assert_eq!(q2.qclass, q_reconstructed[1].qclass);
assert_eq!(q2.qtype, q_reconstructed[1].qtype);
assert_eq!(q3.qname, q_reconstructed[2].qname);
assert_eq!(q3.qclass, q_reconstructed[2].qclass);
assert_eq!(q3.qtype, q_reconstructed[2].qtype);
}
}

View File

@ -0,0 +1,193 @@
#[cfg(test)]
mod tests;
use urlencoding::decode;
use crate::string::encode_domain_name;
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Copy, Clone)]
pub enum QType {
A = 1,
NS = 2,
CNAME = 5,
SOA = 6,
WKS = 11,
PTR = 12,
HINFO = 13,
MINFO = 14,
MX = 15,
TXT = 16,
RP = 17,
AAAA = 28,
SRV = 33
}
impl TryFrom<u8> for QType {
type Error = u8;
fn try_from(v: u8) -> Result<Self, Self::Error> {
match v {
x if x == QType::A as u8 => Ok(QType::A),
x if x == QType::NS as u8 => Ok(QType::NS),
x if x == QType::CNAME as u8 => Ok(QType::CNAME),
x if x == QType::SOA as u8 => Ok(QType::SOA),
x if x == QType::WKS as u8 => Ok(QType::WKS),
x if x == QType::PTR as u8 => Ok(QType::PTR),
x if x == QType::HINFO as u8 => Ok(QType::HINFO),
x if x == QType::MINFO as u8 => Ok(QType::MINFO),
x if x == QType::MX as u8 => Ok(QType::MX),
x if x == QType::TXT as u8 => Ok(QType::TXT),
x if x == QType::RP as u8 => Ok(QType::RP),
x if x == QType::AAAA as u8 => Ok(QType::AAAA),
x if x == QType::SRV as u8 => Ok(QType::SRV),
_ => Err(v),
}
}
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Copy, Clone)]
pub enum QClass {
Internet = 1,
Chaos = 3,
Hesiod = 4,
}
impl TryFrom<u8> for QClass {
type Error = u8;
fn try_from(v: u8) -> Result<Self, Self::Error> {
match v {
x if x == QClass::Internet as u8 => Ok(QClass::Internet),
x if x == QClass::Chaos as u8 => Ok(QClass::Chaos),
x if x == QClass::Hesiod as u8 => Ok(QClass::Hesiod),
_ => Err(v),
}
}
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Clone)]
pub struct DNSQuestion {
pub qname: String,
pub qtype: QType,
pub qclass: QClass
}
impl DNSQuestion {
pub fn new(qname: String, qtype: QType, qclass: QClass) -> DNSQuestion
{
DNSQuestion {
qname,
qtype,
qclass
}
}
pub fn to_bytes(&self) -> Vec<u8>
{
let mut ret = encode_domain_name(&self.qname);
ret.push(self.qtype as u8);
ret.push(self.qclass as u8);
ret
}
}
pub fn questions_to_bytes(questions: &Vec<DNSQuestion>) -> Vec<u8>
{
let mut ret = Vec::with_capacity(20);
for q in questions
{
ret.append(&mut q.to_bytes());
}
ret
}
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
pub enum QuestionParseError {
ShortLength(usize),
QTypeParse(u8),
QClassParse(u8)
}
pub fn questions_from_bytes(bytes: Vec<u8>, total_questions: u16) -> Result<(i32, Vec<DNSQuestion>), QuestionParseError>
{
if bytes.len() < 4
{
return Err(QuestionParseError::ShortLength(bytes.len()));
}
let mut questions: Vec<DNSQuestion> = Vec::with_capacity(total_questions as usize);
let mut current_query: Vec<u8> = Vec::with_capacity(10);
let mut current_length: Option<u8> = None;
let mut remaining_length: u8 = 0;
let mut current_qtype: Option<u8> = None;
let mut trailers_reached = false;
let mut byte_counter = 0;
for byte in bytes {
byte_counter += 1;
match current_length {
None => { // next question, init lengths
current_length = Some(byte);
remaining_length = byte;
current_query.clear();
}
Some(_) => {
if byte == 0 {
trailers_reached = true;
continue
}
if remaining_length == 0 && !trailers_reached {
current_query.push('.' as u8);
current_length = Some(byte);
remaining_length = byte;
}
else if trailers_reached { // trailer fields
match current_qtype {
None => {
current_qtype = Some(byte);
}
Some(qtype_b) => {
match (qtype_b.try_into(), byte.try_into()) {
(Ok(qtype), Ok(qclass)) => {
questions.push(DNSQuestion {
qname: decode(String::from_utf8(current_query.clone()).unwrap().as_str()).unwrap().to_string(),
qtype,
qclass
});
if questions.len() == total_questions as usize {
break
}
current_length = None;
remaining_length = byte;
current_query.clear();
current_qtype = None;
trailers_reached = false;
}
(Err(qtype_e), _) => {
return Err(QuestionParseError::QTypeParse(qtype_e));
}
(_, Err(qclass_e)) => {
return Err(QuestionParseError::QClassParse(qclass_e));
}
}
}
}
}
else {
current_query.push(byte);
remaining_length -= 1;
}
}
}
}
Ok((byte_counter, questions))
}

View File

@ -0,0 +1,91 @@
use super::*;
#[test]
fn one_question_back_and_forth() {
let q = DNSQuestion {
qname: "google.com".to_string(),
qclass: QClass::Internet,
qtype: QType::A
};
let mut q_bytes = q.to_bytes();
q_bytes.append(&mut vec![0, 0, 0, 0, 0, 0]);
let (q_read, q_reconstructed) = questions_from_bytes(q_bytes, 1).unwrap();
assert_eq!(q.qname, q_reconstructed[0].qname);
assert_eq!(q.qclass, q_reconstructed[0].qclass);
assert_eq!(q.qtype, q_reconstructed[0].qtype);
}
#[test]
fn two_questions_back_and_forth() {
let q = DNSQuestion {
qname: "google.com".to_string(),
qclass: QClass::Internet,
qtype: QType::A
};
let q2 = DNSQuestion {
qname: "duck.com".to_string(),
qclass: QClass::Internet,
qtype: QType::AAAA
};
let mut q_bytes = q.to_bytes();
let mut q2_bytes = q2.to_bytes();
q_bytes.append(&mut q2_bytes);
let (q_read, q_reconstructed) = questions_from_bytes(q_bytes, 2).unwrap();
assert_eq!(q.qname, q_reconstructed[0].qname);
assert_eq!(q.qclass, q_reconstructed[0].qclass);
assert_eq!(q.qtype, q_reconstructed[0].qtype);
assert_eq!(q2.qname, q_reconstructed[1].qname);
assert_eq!(q2.qclass, q_reconstructed[1].qclass);
assert_eq!(q2.qtype, q_reconstructed[1].qtype);
}
#[test]
fn three_questions_back_and_forth() {
let q = DNSQuestion {
qname: "google.com".to_string(),
qclass: QClass::Internet,
qtype: QType::A
};
let q2 = DNSQuestion {
qname: "duck.com".to_string(),
qclass: QClass::Internet,
qtype: QType::AAAA
};
let q3 = DNSQuestion {
qname: "facebook.com".to_string(),
qclass: QClass::Hesiod,
qtype: QType::CNAME
};
let mut q_bytes = q.to_bytes();
let mut q2_bytes = q2.to_bytes();
let mut q3_bytes = q3.to_bytes();
q_bytes.append(&mut q2_bytes);
q_bytes.append(&mut q3_bytes);
let (q_read, q_reconstructed) = questions_from_bytes(q_bytes, 3).unwrap();
assert_eq!(q.qname, q_reconstructed[0].qname);
assert_eq!(q.qclass, q_reconstructed[0].qclass);
assert_eq!(q.qtype, q_reconstructed[0].qtype);
assert_eq!(q2.qname, q_reconstructed[1].qname);
assert_eq!(q2.qclass, q_reconstructed[1].qclass);
assert_eq!(q2.qtype, q_reconstructed[1].qtype);
assert_eq!(q3.qname, q_reconstructed[2].qname);
assert_eq!(q3.qclass, q_reconstructed[2].qclass);
assert_eq!(q3.qtype, q_reconstructed[2].qtype);
}

View File

@ -2,7 +2,7 @@ use std::net::SocketAddr;
use crate::message::header::DNSHeader;
use crate::message::question::{DNSQuestion, questions_to_bytes};
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug, Clone)]
pub struct DNSRequest {
pub header: DNSHeader,
pub questions: Vec<DNSQuestion>,

View File

@ -3,7 +3,7 @@ use crate::message::answer::{answers_to_bytes, DNSAnswer};
use crate::message::header::DNSHeader;
use crate::message::question::{DNSQuestion, questions_to_bytes};
#[derive(Ord, PartialOrd, Eq, PartialEq, Debug)]
#[derive(Debug)]
pub struct DNSResponse {
pub header: DNSHeader,
pub questions: Vec<DNSQuestion>,

View File

@ -1,9 +1,13 @@
use std::net::{IpAddr, Ipv4Addr};
use std::sync::mpsc;
use std::sync::mpsc::{Receiver, Sender};
use std::thread;
use log::{error, info};
use crate::message::answer::{DNSAnswer, IpRData, RawRData};
use crate::message::header::{Direction, ResponseCode};
use crate::message::question::QuestionParseError;
use crate::net::raw_request::NetworkMessagePtr;
use crate::message::response::DNSResponse;
use crate::net::raw_request::{NetworkMessage, NetworkMessagePtr};
use crate::request_parser::{HeaderParseError, parse_request, RequestParseError};
pub struct RequestProcesor {
@ -31,6 +35,32 @@ impl RequestProcesor {
match parse_request(*m) {
Ok(r) => {
info!("received dns message: {:?}", r);
let mut response = DNSResponse{
header: r.header.clone(),
questions: r.questions.clone(),
answers: vec![],
peer: r.peer
};
// response.answers = r.questions.iter().map(|x| DNSAnswer::from_query(x, Box::from(IpRData::from(Ipv4Addr::from([127, 0, 0, 1]))), None)).collect();
response.header.direction = Direction::Response;
response.header.response = ResponseCode::NameError;
response.header.answer_record_count = 0;
response.header.authority_record_count = 0;
response.header.additional_record_count = 0;
if response.header.recursion_desired {
response.header.recursion_available = true;
}
sending_channel.send(Box::from(
NetworkMessage {
buffer: Box::from(response.to_bytes()),
peer: response.peer
}
));
}
Err(e) => {
match e {
@ -60,11 +90,6 @@ impl RequestProcesor {
}
}
}
// match sending_channel.send(m) {
// Ok(_) => {}
// Err(_) => {}
// }
}
info!("message processing thread finishing")

View File

@ -92,7 +92,7 @@ pub fn parse_request(msg: NetworkMessage) -> Result<DNSRequest, RequestParseErro
trimmed.drain(0 .. 12);
match questions_from_bytes(trimmed, header.question_count)
{
Ok(questions) => {
Ok((bytes_read, questions)) => {
Ok(DNSRequest {
header,
questions,