Code for parsing headers added.
This commit is contained in:
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
|||||||
|
/target
|
||||||
238
Cargo.lock
generated
Normal file
238
Cargo.lock
generated
Normal file
@ -0,0 +1,238 @@
|
|||||||
|
# This file is automatically @generated by Cargo.
|
||||||
|
# It is not intended for manual editing.
|
||||||
|
version = 3
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "bumpalo"
|
||||||
|
version = "3.9.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a4a45a46ab1f2412e53d3a0ade76ffad2025804294569aae387231a0cd6e0899"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "cfg-if"
|
||||||
|
version = "1.0.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "dns-parse"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"flume",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "flume"
|
||||||
|
version = "0.10.12"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "843c03199d0c0ca54bc1ea90ac0d507274c28abcc4f691ae8b4eaa375087c76a"
|
||||||
|
dependencies = [
|
||||||
|
"futures-core",
|
||||||
|
"futures-sink",
|
||||||
|
"nanorand",
|
||||||
|
"pin-project",
|
||||||
|
"spin",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "futures-core"
|
||||||
|
version = "0.3.21"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0c09fd04b7e4073ac7156a9539b57a484a8ea920f79c7c675d05d289ab6110d3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "futures-sink"
|
||||||
|
version = "0.3.21"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "21163e139fa306126e6eedaf49ecdb4588f939600f0b1e770f4205ee4b7fa868"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "getrandom"
|
||||||
|
version = "0.2.5"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d39cd93900197114fa1fcb7ae84ca742095eed9442088988ae74fa744e930e77"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
"js-sys",
|
||||||
|
"libc",
|
||||||
|
"wasi",
|
||||||
|
"wasm-bindgen",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "js-sys"
|
||||||
|
version = "0.3.56"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a38fc24e30fd564ce974c02bf1d337caddff65be6cc4735a1f7eab22a7440f04"
|
||||||
|
dependencies = [
|
||||||
|
"wasm-bindgen",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lazy_static"
|
||||||
|
version = "1.4.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "libc"
|
||||||
|
version = "0.2.119"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1bf2e165bb3457c8e098ea76f3e3bc9db55f87aa90d52d0e6be741470916aaa4"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "lock_api"
|
||||||
|
version = "0.4.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "88943dd7ef4a2e5a4bfa2753aaab3013e34ce2533d1996fb18ef591e315e2b3b"
|
||||||
|
dependencies = [
|
||||||
|
"scopeguard",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "log"
|
||||||
|
version = "0.4.14"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "51b9bbe6c47d51fc3e1a9b945965946b4c44142ab8792c50835a980d362c2710"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "nanorand"
|
||||||
|
version = "0.7.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "6a51313c5820b0b02bd422f4b44776fbf47961755c74ce64afc73bfad10226c3"
|
||||||
|
dependencies = [
|
||||||
|
"getrandom",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "pin-project"
|
||||||
|
version = "1.0.10"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "58ad3879ad3baf4e44784bc6a718a8698867bb991f8ce24d1bcbe2cfb4c3a75e"
|
||||||
|
dependencies = [
|
||||||
|
"pin-project-internal",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "pin-project-internal"
|
||||||
|
version = "1.0.10"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "744b6f092ba29c3650faf274db506afd39944f48420f6c86b17cfe0ee1cb36bb"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro2"
|
||||||
|
version = "1.0.36"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "c7342d5883fbccae1cc37a2353b09c87c9b0f3afd73f5fb9bba687a1f733b029"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-xid",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "quote"
|
||||||
|
version = "1.0.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "864d3e96a899863136fc6e99f3d7cae289dafe43bf2c5ac19b70df7210c0a145"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "scopeguard"
|
||||||
|
version = "1.1.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d29ab0c6d3fc0ee92fe66e2d99f700eab17a8d57d1c1d3b748380fb20baa78cd"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "spin"
|
||||||
|
version = "0.9.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "511254be0c5bcf062b019a6c89c01a664aa359ded62f78aa72c6fc137c0590e5"
|
||||||
|
dependencies = [
|
||||||
|
"lock_api",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "1.0.86"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8a65b3f4ffa0092e9887669db0eae07941f023991ab58ea44da8fe8e2d511c6b"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"unicode-xid",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-xid"
|
||||||
|
version = "0.2.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8ccb82d61f80a663efe1f787a51b16b5a51e3314d6ac365b08639f52387b33f3"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasi"
|
||||||
|
version = "0.10.2+wasi-snapshot-preview1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "fd6fbd9a79829dd1ad0cc20627bf1ed606756a7f77edff7b66b7064f9cb327c6"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen"
|
||||||
|
version = "0.2.79"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "25f1af7423d8588a3d840681122e72e6a24ddbcb3f0ec385cac0d12d24256c06"
|
||||||
|
dependencies = [
|
||||||
|
"cfg-if",
|
||||||
|
"wasm-bindgen-macro",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-backend"
|
||||||
|
version = "0.2.79"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8b21c0df030f5a177f3cba22e9bc4322695ec43e7257d865302900290bcdedca"
|
||||||
|
dependencies = [
|
||||||
|
"bumpalo",
|
||||||
|
"lazy_static",
|
||||||
|
"log",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
"wasm-bindgen-shared",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-macro"
|
||||||
|
version = "0.2.79"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2f4203d69e40a52ee523b2529a773d5ffc1dc0071801c87b3d270b471b80ed01"
|
||||||
|
dependencies = [
|
||||||
|
"quote",
|
||||||
|
"wasm-bindgen-macro-support",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-macro-support"
|
||||||
|
version = "0.2.79"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "bfa8a30d46208db204854cadbb5d4baf5fcf8071ba5bf48190c3e59937962ebc"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
"wasm-bindgen-backend",
|
||||||
|
"wasm-bindgen-shared",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "wasm-bindgen-shared"
|
||||||
|
version = "0.2.79"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "3d958d035c4438e28c70e4321a2911302f10135ce78a9c7834c0cab4123d06a2"
|
||||||
9
Cargo.toml
Normal file
9
Cargo.toml
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
[package]
|
||||||
|
name = "dns-parse"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
|
||||||
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
flume = "0.10"
|
||||||
96
src/main.rs
Normal file
96
src/main.rs
Normal file
@ -0,0 +1,96 @@
|
|||||||
|
use flume::{Receiver, Sender};
|
||||||
|
use std::net::UdpSocket;
|
||||||
|
use std::thread::{self, JoinHandle};
|
||||||
|
use std::convert::TryInto;
|
||||||
|
|
||||||
|
mod proto;
|
||||||
|
use proto::{DNSHeader, DNSOpCode, DNSRCode};
|
||||||
|
|
||||||
|
|
||||||
|
fn listen() -> (JoinHandle<()>, Sender<Vec<u8>>, Receiver<Vec<u8>>) {
|
||||||
|
let (tx, rx) = flume::unbounded();
|
||||||
|
|
||||||
|
let tx_clone = tx.clone();
|
||||||
|
let rx_clone = rx.clone();
|
||||||
|
|
||||||
|
(
|
||||||
|
thread::spawn(move || {
|
||||||
|
let socket = UdpSocket::bind("127.0.0.1:13337").unwrap();
|
||||||
|
|
||||||
|
let mut buf = [0; 512];
|
||||||
|
|
||||||
|
loop {
|
||||||
|
let (len, _src) = socket.recv_from(&mut buf).unwrap();
|
||||||
|
|
||||||
|
let buf = &mut buf[..len];
|
||||||
|
tx.send(Vec::from(buf)).unwrap();
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
tx_clone,
|
||||||
|
rx_clone,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
const QR_MASK:u16 = 0x0001;
|
||||||
|
const OPCODE_MASK:u16 = 0x000e;
|
||||||
|
const AA_MASK:u16 = 0x0020;
|
||||||
|
const TC_MASK:u16 = 0x0040;
|
||||||
|
const RD_MASK:u16 = 0x0080;
|
||||||
|
const RA_MASK:u16 = 0x0100;
|
||||||
|
const AD_MASK:u16 = 0x0400;
|
||||||
|
const CD_MASK:u16 = 0x0800;
|
||||||
|
const RCODE_MASK:u16 = 0xf000;
|
||||||
|
|
||||||
|
const OPCODE_OFFSET:u16 = 1;
|
||||||
|
const RCODE_OFFSET:u16 = 11;
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
let (thread_udp, _thread_udp_tx, thread_udp_rx) = listen();
|
||||||
|
|
||||||
|
for msg in thread_udp_rx.iter() {
|
||||||
|
println!("{:?}", msg);
|
||||||
|
|
||||||
|
// TODO: check for enough data in msg
|
||||||
|
|
||||||
|
let id = u16::from_be_bytes((&msg[..2]).try_into().unwrap());
|
||||||
|
|
||||||
|
let flags = u16::from_be_bytes((&msg[2..4]).try_into().unwrap());
|
||||||
|
let qr = (flags & QR_MASK) != 0;
|
||||||
|
let opcode:DNSOpCode = DNSOpCode::from((flags & OPCODE_MASK) >> OPCODE_OFFSET);
|
||||||
|
let authoritative_answer = (flags & AA_MASK) != 0;
|
||||||
|
let truncated = (flags & TC_MASK) != 0;
|
||||||
|
let recursion_desired= (flags & RD_MASK) != 0;
|
||||||
|
let recursion_available = (flags & RA_MASK) != 0;
|
||||||
|
let authentic_data = (flags & AD_MASK) != 0;
|
||||||
|
let checking_disabled = (flags & CD_MASK) != 0;
|
||||||
|
let response_code: DNSRCode = DNSRCode::from((flags & RCODE_MASK) >> RCODE_OFFSET);
|
||||||
|
|
||||||
|
let qd_zo_count = u16::from_be_bytes((&msg[4..6]).try_into().unwrap());
|
||||||
|
let an_pr_count = u16::from_be_bytes((&msg[6..8]).try_into().unwrap());
|
||||||
|
let ns_up_count = u16::from_be_bytes((&msg[8..10]).try_into().unwrap());
|
||||||
|
let ar_count = u16::from_be_bytes((&msg[10..12]).try_into().unwrap());
|
||||||
|
|
||||||
|
let hdr_struct = DNSHeader {
|
||||||
|
id,
|
||||||
|
qr,
|
||||||
|
opcode,
|
||||||
|
authoritative_answer,
|
||||||
|
truncated,
|
||||||
|
recursion_desired,
|
||||||
|
recursion_available,
|
||||||
|
authentic_data,
|
||||||
|
checking_disabled,
|
||||||
|
response_code,
|
||||||
|
qd_zo_count,
|
||||||
|
an_pr_count,
|
||||||
|
ns_up_count,
|
||||||
|
ar_count,
|
||||||
|
};
|
||||||
|
|
||||||
|
dbg!(hdr_struct);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
let _ = thread_udp.join();
|
||||||
|
}
|
||||||
84
src/proto.rs
Normal file
84
src/proto.rs
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
#[derive(Debug)]
|
||||||
|
pub enum DNSOpCode {
|
||||||
|
QUERY = 0,
|
||||||
|
IQUERY = 1, // obsolete
|
||||||
|
STATUS = 2,
|
||||||
|
NOTIFY = 4,
|
||||||
|
UPDATE = 5,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<u16> for DNSOpCode {
|
||||||
|
fn from(val: u16) -> Self {
|
||||||
|
match val {
|
||||||
|
0 => DNSOpCode::QUERY,
|
||||||
|
1 => DNSOpCode::IQUERY,
|
||||||
|
2 => DNSOpCode::STATUS,
|
||||||
|
4 => DNSOpCode::NOTIFY,
|
||||||
|
5 => DNSOpCode::UPDATE,
|
||||||
|
_ => panic!("KACKE"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: FIXME: TECHNISCHE SCHULD (alle pubs weg und gucken wie es richtig geht)
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum DNSRCode {
|
||||||
|
NOERROR = 0,
|
||||||
|
FORMERR = 1,
|
||||||
|
SERVFAIL = 2,
|
||||||
|
NXDOMAIN = 3,
|
||||||
|
NOTIMP = 4,
|
||||||
|
REFUSED = 5,
|
||||||
|
XYDOMAIN = 6,
|
||||||
|
XYRRSET = 7,
|
||||||
|
NXRRSET = 8,
|
||||||
|
NOTAUTH = 9,
|
||||||
|
NOTINZONE = 10,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<u16> for DNSRCode {
|
||||||
|
fn from(val: u16) -> Self {
|
||||||
|
match val {
|
||||||
|
0 => DNSRCode::NOERROR,
|
||||||
|
1 => DNSRCode::FORMERR,
|
||||||
|
2 => DNSRCode::SERVFAIL,
|
||||||
|
3 => DNSRCode::NXDOMAIN,
|
||||||
|
4 => DNSRCode::NOTIMP,
|
||||||
|
5 => DNSRCode::REFUSED,
|
||||||
|
6 => DNSRCode::XYDOMAIN,
|
||||||
|
7 => DNSRCode::XYRRSET,
|
||||||
|
8 => DNSRCode::NXRRSET,
|
||||||
|
9 => DNSRCode::NOTAUTH,
|
||||||
|
10 => DNSRCode::NOTINZONE,
|
||||||
|
_ => panic!("KACKE"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct DNSHeader {
|
||||||
|
/// used by the requester to match up replies to outstanding queries
|
||||||
|
pub id: u16,
|
||||||
|
/// specifies whether this message is a query (false), or a response (true)
|
||||||
|
pub qr: bool,
|
||||||
|
pub opcode: DNSOpCode,
|
||||||
|
pub authoritative_answer: bool,
|
||||||
|
pub truncated: bool,
|
||||||
|
pub recursion_desired: bool,
|
||||||
|
pub recursion_available: bool,
|
||||||
|
pub authentic_data: bool,
|
||||||
|
pub checking_disabled: bool,
|
||||||
|
pub response_code: DNSRCode,
|
||||||
|
pub qd_zo_count: u16,
|
||||||
|
pub an_pr_count: u16,
|
||||||
|
pub ns_up_count: u16,
|
||||||
|
pub ar_count: u16,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct DNSQuery {
|
||||||
|
pub hdr: DNSHeader,
|
||||||
|
pub name: String,
|
||||||
|
pub qclass: u16,
|
||||||
|
pub qtype: u16,
|
||||||
|
}
|
||||||
13
udp_sender.py
Executable file
13
udp_sender.py
Executable file
@ -0,0 +1,13 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
import socket
|
||||||
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
UDP_SIZE = int(sys.argv[1])
|
||||||
|
|
||||||
|
UDP_IP = "127.0.0.1"
|
||||||
|
UDP_PORT = 13337
|
||||||
|
MESSAGE = "A"
|
||||||
|
|
||||||
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||||
|
sock.sendto(bytes(MESSAGE*UDP_SIZE, "utf-8"), (UDP_IP, UDP_PORT))
|
||||||
Reference in New Issue
Block a user