Initial commit
This commit is contained in:
commit
32f496d3d0
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
/target
|
||||
49
2
Normal file
49
2
Normal file
@ -0,0 +1,49 @@
|
||||
use std::net::Ipv4Addr;
|
||||
|
||||
use tun_tap::Mode;
|
||||
|
||||
use crate::ethernet::{EtherType, EthernetFrame, InvalidEtherType, MacAddr};
|
||||
|
||||
pub struct Iface {
|
||||
iface: tun_tap::Iface,
|
||||
mac: MacAddr,
|
||||
ip: Option<Ipv4Addr>,
|
||||
}
|
||||
|
||||
impl Iface {
|
||||
pub fn new() -> Iface {
|
||||
Self {
|
||||
iface: tun_tap::Iface::without_packet_info("tap0", Mode::Tap).unwrap(),
|
||||
mac: MacAddr::new([0x9A, 0xB1, 0xB5, 0x22, 0x38, 0xAB]),
|
||||
ip: None,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn recv(&self) -> Result<EthernetFrame, InvalidEtherType> {
|
||||
let mut frame_buf = [0u8; 2048];
|
||||
let frame_len = self.iface.recv(&mut frame_buf).unwrap();
|
||||
EthernetFrame::parse(&frame_buf[0..frame_len])
|
||||
}
|
||||
|
||||
pub fn mac_to_self(&self, mac: MacAddr) -> bool {
|
||||
mac == self.mac || mac.is_broadcast()
|
||||
}
|
||||
|
||||
pub fn send(&self, dst: MacAddr, typ: EtherType, data: &[u8]) {
|
||||
let frame = EthernetFrame::new(dst, self.mac, typ, data).serialize();
|
||||
println!("{}", frame);
|
||||
self.iface.send(&frame).unwrap();
|
||||
}
|
||||
|
||||
pub fn mac(&self) -> MacAddr {
|
||||
self.mac
|
||||
}
|
||||
|
||||
pub fn ip(&self) -> Option<Ipv4Addr> {
|
||||
self.ip
|
||||
}
|
||||
|
||||
pub fn set_ip(&mut self, ip: Ipv4Addr) {
|
||||
self.ip = Some(ip);
|
||||
}
|
||||
}
|
||||
686
Cargo.lock
generated
Normal file
686
Cargo.lock
generated
Normal file
@ -0,0 +1,686 @@
|
||||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d468802bab17cbc0cc575e9b053f41e72aa36bfa6b7f55e3529ffa43161b97fa"
|
||||
|
||||
[[package]]
|
||||
name = "bitflags"
|
||||
version = "1.3.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
|
||||
|
||||
[[package]]
|
||||
name = "byteorder"
|
||||
version = "1.4.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "14c189c53d098945499cdfa7ecc63567cf3886b3332b312a5b4585d8d3a6a610"
|
||||
|
||||
[[package]]
|
||||
name = "bytes"
|
||||
version = "0.4.12"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "206fdffcfa2df7cbe15601ef46c813fce0965eb3286db6b56c583b814b51c81c"
|
||||
dependencies = [
|
||||
"byteorder",
|
||||
"iovec",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.0.73"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2fff2a6927b3bb87f9595d67196a70493f627687a71d87a0d692242c33f58c11"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "0.1.10"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4785bdd1c96b2a846b2bd7cc02e86b6b3dbf14e7e53446c4f54c92a361040822"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||
|
||||
[[package]]
|
||||
name = "cloudabi"
|
||||
version = "0.0.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ddfc5b9aa5d4507acaf872de71051dfd0e309860e88966e1051e462a077aac4f"
|
||||
dependencies = [
|
||||
"bitflags",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-deque"
|
||||
version = "0.7.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c20ff29ded3204c5106278a81a38f4b482636ed4fa1e6cfbeef193291beb29ed"
|
||||
dependencies = [
|
||||
"crossbeam-epoch",
|
||||
"crossbeam-utils",
|
||||
"maybe-uninit",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-epoch"
|
||||
version = "0.8.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "058ed274caafc1f60c4997b5fc07bf7dc7cca454af7c6e81edffe5f33f70dace"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"cfg-if 0.1.10",
|
||||
"crossbeam-utils",
|
||||
"lazy_static",
|
||||
"maybe-uninit",
|
||||
"memoffset",
|
||||
"scopeguard",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-queue"
|
||||
version = "0.2.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "774ba60a54c213d409d5353bda12d49cd68d14e45036a285234c8d6f91f92570"
|
||||
dependencies = [
|
||||
"cfg-if 0.1.10",
|
||||
"crossbeam-utils",
|
||||
"maybe-uninit",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-utils"
|
||||
version = "0.7.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c3c7c73a2d1e9fc0886a08b93e98eb643461230d5f1925e4036204d5f2e261a8"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"cfg-if 0.1.10",
|
||||
"lazy_static",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "derive-try-from-primitive"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "302ccf094df1151173bb6f5a2282fcd2f45accd5eae1bdf82dcbfefbc501ad5c"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "fnv"
|
||||
version = "1.0.7"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
|
||||
|
||||
[[package]]
|
||||
name = "fuchsia-zircon"
|
||||
version = "0.3.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2e9763c69ebaae630ba35f74888db465e49e259ba1bc0eda7d06f4a067615d82"
|
||||
dependencies = [
|
||||
"bitflags",
|
||||
"fuchsia-zircon-sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "fuchsia-zircon-sys"
|
||||
version = "0.3.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3dcaa9ae7725d12cdb85b3ad99a434db70b468c09ded17e012d86b5c1010f7a7"
|
||||
|
||||
[[package]]
|
||||
name = "futures"
|
||||
version = "0.1.31"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3a471a38ef8ed83cd6e40aa59c1ffe17db6855c18e3604d9c4ed8c08ebc28678"
|
||||
|
||||
[[package]]
|
||||
name = "hermit-abi"
|
||||
version = "0.1.19"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "62b467343b94ba476dcb2500d242dadbb39557df889310ac77c5d99100aaac33"
|
||||
dependencies = [
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "internet-checksum"
|
||||
version = "0.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc6d6206008e25125b1f97fbe5d309eb7b85141cf9199d52dbd3729a1584dd16"
|
||||
|
||||
[[package]]
|
||||
name = "iovec"
|
||||
version = "0.1.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b2b3ea6ff95e175473f8ffe6a7eb7c00d054240321b84c57051175fe3c1e075e"
|
||||
dependencies = [
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "kernel32-sys"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7507624b29483431c0ba2d82aece8ca6cdba9382bff4ddd0f7490560c056098d"
|
||||
dependencies = [
|
||||
"winapi 0.2.8",
|
||||
"winapi-build",
|
||||
]
|
||||
|
||||
[[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.126"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "349d5a591cd28b49e1d1037471617a32ddcda5731b99419008085f72d5a53836"
|
||||
|
||||
[[package]]
|
||||
name = "lock_api"
|
||||
version = "0.3.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4da24a77a3d8a6d4862d95f72e6fdb9c09a643ecdb402d754004a557f2bec75"
|
||||
dependencies = [
|
||||
"scopeguard",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "log"
|
||||
version = "0.4.17"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "abb12e687cfb44aa40f41fc3978ef76448f9b6038cad6aef4259d3c095a2382e"
|
||||
dependencies = [
|
||||
"cfg-if 1.0.0",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "maybe-uninit"
|
||||
version = "2.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "60302e4db3a61da70c0cb7991976248362f30319e88850c487b9b95bbf059e00"
|
||||
|
||||
[[package]]
|
||||
name = "memoffset"
|
||||
version = "0.5.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "043175f069eda7b85febe4a74abbaeff828d9f8b448515d3151a14a3542811aa"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "mio"
|
||||
version = "0.6.23"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4afd66f5b91bf2a3bc13fad0e21caedac168ca4c707504e75585648ae80e4cc4"
|
||||
dependencies = [
|
||||
"cfg-if 0.1.10",
|
||||
"fuchsia-zircon",
|
||||
"fuchsia-zircon-sys",
|
||||
"iovec",
|
||||
"kernel32-sys",
|
||||
"libc",
|
||||
"log",
|
||||
"miow",
|
||||
"net2",
|
||||
"slab",
|
||||
"winapi 0.2.8",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "mio-uds"
|
||||
version = "0.6.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "afcb699eb26d4332647cc848492bbc15eafb26f08d0304550d5aa1f612e066f0"
|
||||
dependencies = [
|
||||
"iovec",
|
||||
"libc",
|
||||
"mio",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "miow"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ebd808424166322d4a38da87083bfddd3ac4c131334ed55856112eb06d46944d"
|
||||
dependencies = [
|
||||
"kernel32-sys",
|
||||
"net2",
|
||||
"winapi 0.2.8",
|
||||
"ws2_32-sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "mopa"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a785740271256c230f57462d3b83e52f998433a7062fc18f96d5999474a9f915"
|
||||
|
||||
[[package]]
|
||||
name = "net2"
|
||||
version = "0.2.37"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "391630d12b68002ae1e25e8f974306474966550ad82dac6886fb8910c19568ae"
|
||||
dependencies = [
|
||||
"cfg-if 0.1.10",
|
||||
"libc",
|
||||
"winapi 0.3.9",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "nstack-rs"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"derive-try-from-primitive",
|
||||
"internet-checksum",
|
||||
"mopa",
|
||||
"rhexdump",
|
||||
"tun-tap",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "num_cpus"
|
||||
version = "1.13.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "19e64526ebdee182341572e50e9ad03965aa510cd94427a4549448f285e957a1"
|
||||
dependencies = [
|
||||
"hermit-abi",
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "parking_lot"
|
||||
version = "0.9.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f842b1982eb6c2fe34036a4fbfb06dd185a3f5c8edfaacdf7d1ea10b07de6252"
|
||||
dependencies = [
|
||||
"lock_api",
|
||||
"parking_lot_core",
|
||||
"rustc_version",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "parking_lot_core"
|
||||
version = "0.6.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b876b1b9e7ac6e1a74a6da34d25c42e17e8862aa409cbbbdcfc8d86c6f3bc62b"
|
||||
dependencies = [
|
||||
"cfg-if 0.1.10",
|
||||
"cloudabi",
|
||||
"libc",
|
||||
"redox_syscall",
|
||||
"rustc_version",
|
||||
"smallvec",
|
||||
"winapi 0.3.9",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "proc-macro2"
|
||||
version = "1.0.39"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c54b25569025b7fc9651de43004ae593a75ad88543b17178aa5e1b9c4f15f56f"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "quote"
|
||||
version = "1.0.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a1feb54ed693b93a84e14094943b84b7c4eae204c512b7ccb95ab0c66d278ad1"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "redox_syscall"
|
||||
version = "0.1.57"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "41cc0f7e4d5d4544e8861606a285bb08d3e70712ccc7d2b84d7c0ccfaf4b05ce"
|
||||
|
||||
[[package]]
|
||||
name = "rhexdump"
|
||||
version = "0.1.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c5e9af64574935e39f24d1c0313a997c8b880ca0e087c888bc6af8af31579847"
|
||||
|
||||
[[package]]
|
||||
name = "rustc_version"
|
||||
version = "0.2.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "138e3e0acb6c9fb258b19b67cb8abd63c00679d2851805ea151465464fe9030a"
|
||||
dependencies = [
|
||||
"semver",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "scoped-tls"
|
||||
version = "0.1.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "332ffa32bf586782a3efaeb58f127980944bbc8c4d6913a86107ac2a5ab24b28"
|
||||
|
||||
[[package]]
|
||||
name = "scopeguard"
|
||||
version = "1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d29ab0c6d3fc0ee92fe66e2d99f700eab17a8d57d1c1d3b748380fb20baa78cd"
|
||||
|
||||
[[package]]
|
||||
name = "semver"
|
||||
version = "0.9.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1d7eb9ef2c18661902cc47e535f9bc51b78acd254da71d375c2f6720d9a40403"
|
||||
dependencies = [
|
||||
"semver-parser",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "semver-parser"
|
||||
version = "0.7.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "388a1df253eca08550bef6c72392cfe7c30914bf41df5269b68cbd6ff8f570a3"
|
||||
|
||||
[[package]]
|
||||
name = "slab"
|
||||
version = "0.4.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "eb703cfe953bccee95685111adeedb76fabe4e97549a58d16f03ea7b9367bb32"
|
||||
|
||||
[[package]]
|
||||
name = "smallvec"
|
||||
version = "0.6.14"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b97fcaeba89edba30f044a10c6a3cc39df9c3f17d7cd829dd1446cab35f890e0"
|
||||
dependencies = [
|
||||
"maybe-uninit",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "syn"
|
||||
version = "1.0.96"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "0748dd251e24453cb8717f0354206b91557e4ec8703673a4b30208f2abaf1ebf"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio"
|
||||
version = "0.1.22"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5a09c0b5bb588872ab2f09afa13ee6e9dac11e10a0ec9e8e3ba39a5a5d530af6"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"mio",
|
||||
"num_cpus",
|
||||
"tokio-codec",
|
||||
"tokio-current-thread",
|
||||
"tokio-executor",
|
||||
"tokio-fs",
|
||||
"tokio-io",
|
||||
"tokio-reactor",
|
||||
"tokio-sync",
|
||||
"tokio-tcp",
|
||||
"tokio-threadpool",
|
||||
"tokio-timer",
|
||||
"tokio-udp",
|
||||
"tokio-uds",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-codec"
|
||||
version = "0.1.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "25b2998660ba0e70d18684de5d06b70b70a3a747469af9dea7618cc59e75976b"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"tokio-io",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-core"
|
||||
version = "0.1.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "87b1395334443abca552f63d4f61d0486f12377c2ba8b368e523f89e828cffd4"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"iovec",
|
||||
"log",
|
||||
"mio",
|
||||
"scoped-tls",
|
||||
"tokio",
|
||||
"tokio-executor",
|
||||
"tokio-io",
|
||||
"tokio-reactor",
|
||||
"tokio-timer",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-current-thread"
|
||||
version = "0.1.7"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b1de0e32a83f131e002238d7ccde18211c0a5397f60cbfffcb112868c2e0e20e"
|
||||
dependencies = [
|
||||
"futures",
|
||||
"tokio-executor",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-executor"
|
||||
version = "0.1.10"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fb2d1b8f4548dbf5e1f7818512e9c406860678f29c300cdf0ebac72d1a3a1671"
|
||||
dependencies = [
|
||||
"crossbeam-utils",
|
||||
"futures",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-fs"
|
||||
version = "0.1.7"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "297a1206e0ca6302a0eed35b700d292b275256f596e2f3fea7729d5e629b6ff4"
|
||||
dependencies = [
|
||||
"futures",
|
||||
"tokio-io",
|
||||
"tokio-threadpool",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-io"
|
||||
version = "0.1.13"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "57fc868aae093479e3131e3d165c93b1c7474109d13c90ec0dda2a1bbfff0674"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"log",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-reactor"
|
||||
version = "0.1.12"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "09bc590ec4ba8ba87652da2068d150dcada2cfa2e07faae270a5e0409aa51351"
|
||||
dependencies = [
|
||||
"crossbeam-utils",
|
||||
"futures",
|
||||
"lazy_static",
|
||||
"log",
|
||||
"mio",
|
||||
"num_cpus",
|
||||
"parking_lot",
|
||||
"slab",
|
||||
"tokio-executor",
|
||||
"tokio-io",
|
||||
"tokio-sync",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-sync"
|
||||
version = "0.1.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "edfe50152bc8164fcc456dab7891fa9bf8beaf01c5ee7e1dd43a397c3cf87dee"
|
||||
dependencies = [
|
||||
"fnv",
|
||||
"futures",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-tcp"
|
||||
version = "0.1.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "98df18ed66e3b72e742f185882a9e201892407957e45fbff8da17ae7a7c51f72"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"iovec",
|
||||
"mio",
|
||||
"tokio-io",
|
||||
"tokio-reactor",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-threadpool"
|
||||
version = "0.1.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "df720b6581784c118f0eb4310796b12b1d242a7eb95f716a8367855325c25f89"
|
||||
dependencies = [
|
||||
"crossbeam-deque",
|
||||
"crossbeam-queue",
|
||||
"crossbeam-utils",
|
||||
"futures",
|
||||
"lazy_static",
|
||||
"log",
|
||||
"num_cpus",
|
||||
"slab",
|
||||
"tokio-executor",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-timer"
|
||||
version = "0.2.13"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "93044f2d313c95ff1cb7809ce9a7a05735b012288a888b62d4434fd58c94f296"
|
||||
dependencies = [
|
||||
"crossbeam-utils",
|
||||
"futures",
|
||||
"slab",
|
||||
"tokio-executor",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-udp"
|
||||
version = "0.1.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e2a0b10e610b39c38b031a2fcab08e4b82f16ece36504988dcbd81dbba650d82"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"log",
|
||||
"mio",
|
||||
"tokio-codec",
|
||||
"tokio-io",
|
||||
"tokio-reactor",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tokio-uds"
|
||||
version = "0.2.7"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ab57a4ac4111c8c9dbcf70779f6fc8bc35ae4b2454809febac840ad19bd7e4e0"
|
||||
dependencies = [
|
||||
"bytes",
|
||||
"futures",
|
||||
"iovec",
|
||||
"libc",
|
||||
"log",
|
||||
"mio",
|
||||
"mio-uds",
|
||||
"tokio-codec",
|
||||
"tokio-io",
|
||||
"tokio-reactor",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "tun-tap"
|
||||
version = "0.1.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "53ccbe9cfffdaa7eefd36538bb26f228d4bd319a8aeca044d54377612a646bcd"
|
||||
dependencies = [
|
||||
"cc",
|
||||
"futures",
|
||||
"libc",
|
||||
"mio",
|
||||
"tokio-core",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unicode-ident"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d22af068fba1eb5edcb4aea19d382b2a3deb4c8f9d475c589b6ada9e0fd493ee"
|
||||
|
||||
[[package]]
|
||||
name = "winapi"
|
||||
version = "0.2.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "167dc9d6949a9b857f3451275e911c3f44255842c1f7a76f33c55103a909087a"
|
||||
|
||||
[[package]]
|
||||
name = "winapi"
|
||||
version = "0.3.9"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
|
||||
dependencies = [
|
||||
"winapi-i686-pc-windows-gnu",
|
||||
"winapi-x86_64-pc-windows-gnu",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "winapi-build"
|
||||
version = "0.1.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2d315eee3b34aca4797b2da6b13ed88266e6d612562a0c46390af8299fc699bc"
|
||||
|
||||
[[package]]
|
||||
name = "winapi-i686-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
|
||||
|
||||
[[package]]
|
||||
name = "winapi-x86_64-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
|
||||
|
||||
[[package]]
|
||||
name = "ws2_32-sys"
|
||||
version = "0.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d59cefebd0c892fa2dd6de581e937301d8552cb44489cdff035c6187cb63fa5e"
|
||||
dependencies = [
|
||||
"winapi 0.2.8",
|
||||
"winapi-build",
|
||||
]
|
||||
13
Cargo.toml
Normal file
13
Cargo.toml
Normal file
@ -0,0 +1,13 @@
|
||||
[package]
|
||||
name = "nstack-rs"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
derive-try-from-primitive = "1.0.0"
|
||||
internet-checksum = "0.2.1"
|
||||
mopa = "0.2.2"
|
||||
rhexdump = "0.1.1"
|
||||
tun-tap = "0.1.2"
|
||||
1
nstack-runner
Symbolic link
1
nstack-runner
Symbolic link
@ -0,0 +1 @@
|
||||
runner/target/debug/runner
|
||||
47
runner/Cargo.lock
generated
Normal file
47
runner/Cargo.lock
generated
Normal file
@ -0,0 +1,47 @@
|
||||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "bitflags"
|
||||
version = "1.3.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
|
||||
|
||||
[[package]]
|
||||
name = "capctl"
|
||||
version = "0.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "526c6a8746a7cfb052c15d20259c4f5c021966affdc7c960c71ca640f824c801"
|
||||
dependencies = [
|
||||
"bitflags",
|
||||
"cfg-if",
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||
|
||||
[[package]]
|
||||
name = "iptool"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.126"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "349d5a591cd28b49e1d1037471617a32ddcda5731b99419008085f72d5a53836"
|
||||
|
||||
[[package]]
|
||||
name = "runner"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"capctl",
|
||||
"iptool",
|
||||
]
|
||||
10
runner/Cargo.toml
Normal file
10
runner/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
||||
[package]
|
||||
name = "runner"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
capctl = "0.2.1"
|
||||
iptool = { path = "iptool" }
|
||||
17
runner/iptool/.github/workflows/build.yml
vendored
Normal file
17
runner/iptool/.github/workflows/build.yml
vendored
Normal file
@ -0,0 +1,17 @@
|
||||
name: "Build"
|
||||
on:
|
||||
pull_request:
|
||||
push:
|
||||
jobs:
|
||||
build:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v2.3.4
|
||||
- uses: cachix/install-nix-action@v13
|
||||
with:
|
||||
nix_path: ""
|
||||
- uses: cachix/cachix-action@v10
|
||||
with:
|
||||
name: nyantec
|
||||
authToken: '${{ secrets.CACHIX_AUTH_TOKEN }}'
|
||||
- run: nix-build -A iptool
|
||||
1
runner/iptool/.gitignore
vendored
Normal file
1
runner/iptool/.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
/target
|
||||
215
runner/iptool/Cargo.lock
generated
Normal file
215
runner/iptool/Cargo.lock
generated
Normal file
@ -0,0 +1,215 @@
|
||||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "aho-corasick"
|
||||
version = "0.7.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1e37cfd5e7657ada45f742d6e99ca5788580b5c529dc78faf11ece6dc702656f"
|
||||
dependencies = [
|
||||
"memchr",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "glob"
|
||||
version = "0.3.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9b919933a397b79c37e33b77bb2aa3dc8eb6e165ad809e58ff75bc7db2e34574"
|
||||
|
||||
[[package]]
|
||||
name = "ipnetwork"
|
||||
version = "0.18.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4088d739b183546b239688ddbc79891831df421773df95e236daf7867866d355"
|
||||
dependencies = [
|
||||
"serde",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "iptool"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"libc",
|
||||
"pnet",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.104"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7b2f96d100e1cf1929e7719b7edb3b90ab5298072638fccd77be9ce942ecdfce"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.4.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "308cc39be01b73d0d18f82a0e7b2a3df85245f84af96fdddc5d202d27e47b86a"
|
||||
|
||||
[[package]]
|
||||
name = "pnet"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4b6d2a0409666964722368ef5fb74b9f93fac11c18bef3308693c16c6733f103"
|
||||
dependencies = [
|
||||
"ipnetwork",
|
||||
"pnet_base",
|
||||
"pnet_datalink",
|
||||
"pnet_packet",
|
||||
"pnet_sys",
|
||||
"pnet_transport",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pnet_base"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "25488cd551a753dcaaa6fffc9f69a7610a412dd8954425bf7ffad5f7d1156fb8"
|
||||
|
||||
[[package]]
|
||||
name = "pnet_datalink"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d4d1f8ab1ef6c914cf51dc5dfe0be64088ea5f3b08bbf5a31abc70356d271198"
|
||||
dependencies = [
|
||||
"ipnetwork",
|
||||
"libc",
|
||||
"pnet_base",
|
||||
"pnet_sys",
|
||||
"winapi",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pnet_macros"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "30490e0852e58402b8fae0d39897b08a24f493023a4d6cf56b2e30f31ed57548"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"regex",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pnet_macros_support"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d4714e10f30cab023005adce048f2d30dd4ac4f093662abf2220855655ef8f90"
|
||||
dependencies = [
|
||||
"pnet_base",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pnet_packet"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8588067671d03c9f4254b2e66fecb4d8b93b5d3e703195b84f311cd137e32130"
|
||||
dependencies = [
|
||||
"glob",
|
||||
"pnet_base",
|
||||
"pnet_macros",
|
||||
"pnet_macros_support",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pnet_sys"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d9a3f32b0df45515befd19eed04616f6b56a488da92afc61164ef455e955f07f"
|
||||
dependencies = [
|
||||
"libc",
|
||||
"winapi",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "pnet_transport"
|
||||
version = "0.28.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "932b2916d693bcc5fa18443dc99142e0a6fd31a6ce75a511868f7174c17e2bce"
|
||||
dependencies = [
|
||||
"libc",
|
||||
"pnet_base",
|
||||
"pnet_packet",
|
||||
"pnet_sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "proc-macro2"
|
||||
version = "1.0.30"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "edc3358ebc67bc8b7fa0c007f945b0b18226f78437d61bec735a9eb96b61ee70"
|
||||
dependencies = [
|
||||
"unicode-xid",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "quote"
|
||||
version = "1.0.10"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "38bc8cc6a5f2e3655e0899c1b848643b2562f853f114bfec7be120678e3ace05"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "regex"
|
||||
version = "1.5.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d07a8629359eb56f1e2fb1652bb04212c072a87ba68546a04065d525673ac461"
|
||||
dependencies = [
|
||||
"aho-corasick",
|
||||
"memchr",
|
||||
"regex-syntax",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "regex-syntax"
|
||||
version = "0.6.25"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f497285884f3fcff424ffc933e56d7cbca511def0c9831a7f9b5f6153e3cc89b"
|
||||
|
||||
[[package]]
|
||||
name = "serde"
|
||||
version = "1.0.130"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f12d06de37cf59146fbdecab66aa99f9fe4f78722e3607577a5375d66bd0c913"
|
||||
|
||||
[[package]]
|
||||
name = "syn"
|
||||
version = "1.0.80"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d010a1623fbd906d51d650a9916aaefc05ffa0e4053ff7fe601167f3e715d194"
|
||||
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 = "winapi"
|
||||
version = "0.3.9"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
|
||||
dependencies = [
|
||||
"winapi-i686-pc-windows-gnu",
|
||||
"winapi-x86_64-pc-windows-gnu",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "winapi-i686-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
|
||||
|
||||
[[package]]
|
||||
name = "winapi-x86_64-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
|
||||
21
runner/iptool/Cargo.toml
Normal file
21
runner/iptool/Cargo.toml
Normal file
@ -0,0 +1,21 @@
|
||||
[package]
|
||||
name = "iptool"
|
||||
version = "0.1.0"
|
||||
authors = ["Finn Behrens <fin@nyantec.com>"]
|
||||
edition = "2018"
|
||||
repository = "https://github.com/nyantec/iptool"
|
||||
license = "MirOS"
|
||||
keywords = [ "linux", "network" ]
|
||||
categories = [ "hardware-support" ]
|
||||
description = "Rust linux iptool helpers for network interfaces"
|
||||
homepage = "https://github.com/nyantec/iptool"
|
||||
readme = "README.md"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
[features]
|
||||
default = []
|
||||
|
||||
[dependencies]
|
||||
libc = "0.2"
|
||||
|
||||
pnet = { version = "0.28", optional = true }
|
||||
8
runner/iptool/LICENSE
Normal file
8
runner/iptool/LICENSE
Normal file
@ -0,0 +1,8 @@
|
||||
Copyright 2020 nyantec GmbH <oss@nyantec.com>
|
||||
|
||||
Authors:
|
||||
Finn Behrens <fin@nyantec.com>
|
||||
|
||||
Provided that these terms and disclaimer and all copyright notices are retained or reproduced in an accompanying document, permission is granted to deal in this work without restriction, including unlimited rights to use, publicly perform, distribute, sell, modify, merge, give away, or sublicence.
|
||||
|
||||
This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to the utmost extent permitted by applicable law, neither express nor implied; without malicious intent or gross negligence. In no event may a licensor, author or contributor be held liable for indirect, direct, other damage, loss, or other issues arising in any way out of dealing in the work, even if advised of the possibility of such damage or existence of a defect, except proven that it results out of said person's immediate fault when using the work as intended.
|
||||
3
runner/iptool/README.md
Normal file
3
runner/iptool/README.md
Normal file
@ -0,0 +1,3 @@
|
||||
# IpTool
|
||||
|
||||
Rust linux iptool helpers for network interfaces.
|
||||
8
runner/iptool/default.nix
Normal file
8
runner/iptool/default.nix
Normal file
@ -0,0 +1,8 @@
|
||||
(import (
|
||||
fetchTarball {
|
||||
url = "https://github.com/edolstra/flake-compat/archive/12c64ca55c1014cdc1b16ed5a804aa8576601ff2.tar.gz";
|
||||
sha256 = "0jm6nzb83wa6ai17ly9fzpqc40wg1viib8klq8lby54agpl213w5";
|
||||
}
|
||||
) {
|
||||
src = ./.;
|
||||
}).defaultNix.packages.x86_64-linux
|
||||
25
runner/iptool/flake.lock
Normal file
25
runner/iptool/flake.lock
Normal file
@ -0,0 +1,25 @@
|
||||
{
|
||||
"nodes": {
|
||||
"nixpkgs": {
|
||||
"locked": {
|
||||
"lastModified": 1635150109,
|
||||
"narHash": "sha256-KTtyOdjFY8AqQRKytamEuLgPJYRjQSDivyNlZtT/2Zs=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "8e18c70837aa01ade3718cd0fd35b649b3a2cf52",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"id": "nixpkgs",
|
||||
"type": "indirect"
|
||||
}
|
||||
},
|
||||
"root": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs"
|
||||
}
|
||||
}
|
||||
},
|
||||
"root": "root",
|
||||
"version": 7
|
||||
}
|
||||
28
runner/iptool/flake.nix
Normal file
28
runner/iptool/flake.nix
Normal file
@ -0,0 +1,28 @@
|
||||
{
|
||||
description = "iptool";
|
||||
|
||||
outputs = { self, nixpkgs }: let
|
||||
version = self.shortRev or (toString self.lastModifiedDate);
|
||||
overlay = final: prev: {
|
||||
iptool = final.callPackage (
|
||||
{ rustPlatform }: rustPlatform.buildRustPackage {
|
||||
pname = "iptool";
|
||||
inherit version;
|
||||
src = self;
|
||||
cargoLock.lockFile = ./Cargo.lock;
|
||||
cargoBuildFlags = [ "--all-features" ];
|
||||
}
|
||||
) {};
|
||||
};
|
||||
pkgs = import nixpkgs {
|
||||
system = "x86_64-linux";
|
||||
overlays = [ overlay ];
|
||||
};
|
||||
in {
|
||||
inherit overlay;
|
||||
packages.x86_64-linux = {
|
||||
inherit (pkgs) iptool;
|
||||
};
|
||||
defaultPackage.x86_64-linux = self.packages.x86_64-linux.iptool;
|
||||
};
|
||||
}
|
||||
133
runner/iptool/src/lib.rs
Normal file
133
runner/iptool/src/lib.rs
Normal file
@ -0,0 +1,133 @@
|
||||
use std::io::{Error, Result};
|
||||
|
||||
#[cfg(target_family = "unix")]
|
||||
use std::os::unix::io::{AsRawFd, FromRawFd, RawFd};
|
||||
|
||||
#[cfg(feature = "pnet")]
|
||||
use pnet::datalink::MacAddr;
|
||||
|
||||
#[cfg(target_os = "linux")]
|
||||
mod linux;
|
||||
|
||||
#[cfg(target_os = "linux")]
|
||||
pub use linux::{Ifreq, SIOCGIFINDEX};
|
||||
|
||||
// TODO: macOS
|
||||
|
||||
/// Only use it for a short amount of time, as it does not close it's ioctl socket
|
||||
pub struct IpTool {
|
||||
#[cfg(target_family = "unix")]
|
||||
fd: RawFd,
|
||||
}
|
||||
|
||||
/*
|
||||
TODO: is it already non send?
|
||||
impl !Send for IpTool {}
|
||||
impl !Sync for IpTool {}
|
||||
*/
|
||||
|
||||
impl AsRawFd for IpTool {
|
||||
fn as_raw_fd(&self) -> RawFd {
|
||||
self.fd
|
||||
}
|
||||
}
|
||||
|
||||
impl FromRawFd for IpTool {
|
||||
unsafe fn from_raw_fd(fd: RawFd) -> Self {
|
||||
Self { fd }
|
||||
}
|
||||
}
|
||||
|
||||
// Helper function
|
||||
#[allow(dead_code)] // not used yet on non linux
|
||||
pub(crate) fn copy_slice(dst: &mut [u8], src: &[u8]) -> usize {
|
||||
let mut c = 0;
|
||||
|
||||
for (d, s) in dst.iter_mut().zip(src.iter()) {
|
||||
*d = *s;
|
||||
c += 1;
|
||||
}
|
||||
|
||||
c
|
||||
}
|
||||
|
||||
pub fn parse_mac_addr(mac: &str) -> Result<[libc::c_char; 14]> {
|
||||
let mut addr: [libc::c_char; 14] = [0; 14];
|
||||
let mac_vec: Vec<&str> = mac.split(':').collect();
|
||||
if mac_vec.len() != 6 {
|
||||
// TODO: unlikly (https://doc.rust-lang.org/nightly/std/intrinsics/fn.unlikely.html)
|
||||
return Err(Error::from_raw_os_error(libc::EINVAL));
|
||||
}
|
||||
for x in 0..6 {
|
||||
if let Ok(data) = u8::from_str_radix(mac_vec[x], 16) {
|
||||
addr[x] = data as i8;
|
||||
} else {
|
||||
// TODO: unlikly (https://doc.rust-lang.org/nightly/std/intrinsics/fn.unlikely.html)
|
||||
return Err(Error::from_raw_os_error(libc::EINVAL));
|
||||
}
|
||||
}
|
||||
|
||||
Ok(addr)
|
||||
}
|
||||
|
||||
// Ext traits
|
||||
#[cfg(feature = "pnet")]
|
||||
pub trait MacAddrLinxExt: From<[u8; 6]> {
|
||||
fn from_interface(interface: &str) -> Result<Self>;
|
||||
}
|
||||
|
||||
#[cfg(feature = "pnet")]
|
||||
impl MacAddrLinxExt for MacAddr {
|
||||
fn from_interface(interface: &str) -> Result<Self> {
|
||||
let tool = IpTool::new()?;
|
||||
|
||||
let hwaddr = tool.get_mac_sa_data(interface)?;
|
||||
//let hwaddr: [u8; 6] = hwaddr.try_into()?;
|
||||
let hwaddr = unsafe { *(&hwaddr as *const _ as *const [u8; 6]) };
|
||||
|
||||
let hwaddr: [u8; 6] = hwaddr.into();
|
||||
|
||||
Ok(hwaddr.into())
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(target_os = "linux")]
|
||||
pub use linux::IpAddrLinkExt;
|
||||
|
||||
#[cold]
|
||||
#[inline]
|
||||
#[allow(dead_code)] // not used yet on non linux
|
||||
/// Own (cold) function to optimize if statements
|
||||
fn last_err() -> Error {
|
||||
Error::last_os_error()
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
|
||||
#[test]
|
||||
#[allow(overflowing_literals)]
|
||||
fn parse_mac_addr() {
|
||||
let addr = "5A:E6:60:8F:5F:DE";
|
||||
let mut addr_vec: [libc::c_char; 14] = [0; 14];
|
||||
addr_vec[0] = 0x5A;
|
||||
addr_vec[1] = 0xE6;
|
||||
addr_vec[2] = 0x60;
|
||||
addr_vec[3] = 0x8F;
|
||||
addr_vec[4] = 0x5F;
|
||||
addr_vec[5] = 0xDE;
|
||||
|
||||
assert_eq!(super::parse_mac_addr(addr).unwrap(), addr_vec);
|
||||
|
||||
// not long enough address
|
||||
super::parse_mac_addr("5A:3B:2D").unwrap_err();
|
||||
}
|
||||
|
||||
#[cfg(feature = "pnet")]
|
||||
#[test]
|
||||
fn macaddr_from_interface() {
|
||||
use super::MacAddrLinxExt;
|
||||
|
||||
assert!(pnet::util::MacAddr::from_interface("lo").unwrap().is_zero());
|
||||
}
|
||||
}
|
||||
369
runner/iptool/src/linux.rs
Normal file
369
runner/iptool/src/linux.rs
Normal file
@ -0,0 +1,369 @@
|
||||
use std::io::{Error, Result};
|
||||
use std::net::{IpAddr, Ipv4Addr};
|
||||
|
||||
use libc::{
|
||||
c_int, c_short, c_uchar, close, in6_addr, ioctl, sockaddr, sockaddr_in, sockaddr_in6, socket,
|
||||
};
|
||||
|
||||
use super::{copy_slice, last_err, parse_mac_addr, IpTool};
|
||||
|
||||
pub const SIOCGIFINDEX: u64 = 0x8933;
|
||||
|
||||
// Helper traits
|
||||
pub trait IpAddrLinkExt
|
||||
where
|
||||
Self: Sized,
|
||||
{
|
||||
fn from_interface(dev: &str) -> Result<Self>;
|
||||
}
|
||||
|
||||
impl IpAddrLinkExt for Ipv4Addr {
|
||||
fn from_interface(dev: &str) -> Result<Self> {
|
||||
let iptool = IpTool::new()?;
|
||||
|
||||
iptool.get_address(dev)
|
||||
}
|
||||
}
|
||||
|
||||
impl IpTool {
|
||||
pub fn new() -> Result<Self> {
|
||||
let fd = Self::get_ctl_fd()?;
|
||||
|
||||
Ok(Self { fd })
|
||||
}
|
||||
|
||||
pub fn set_up(&self, dev: &str, up: bool) -> Result<()> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCGIFFLAGS as _)?;
|
||||
|
||||
let flag_val = libc::IFF_UP as i16;
|
||||
|
||||
// SAFETY: union
|
||||
unsafe {
|
||||
ifr.ifr_ifru.ifru_flags = if up {
|
||||
ifr.ifr_ifru.ifru_flags | flag_val
|
||||
} else {
|
||||
ifr.ifr_ifru.ifru_flags & (!flag_val)
|
||||
};
|
||||
}
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCSIFFLAGS as _)?;
|
||||
|
||||
if self.get_up(dev)? != up {
|
||||
return Err(Error::from_raw_os_error(libc::ENOTRECOVERABLE));
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn get_up(&self, dev: &str) -> Result<bool> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCGIFFLAGS as _)?;
|
||||
|
||||
// unions
|
||||
let flags: i16 = unsafe { ifr.ifr_ifru.ifru_flags };
|
||||
Ok((flags & libc::IFF_UP as i16) == 1)
|
||||
}
|
||||
|
||||
pub fn set_mtu(&self, dev: &str, mtu: u32) -> Result<()> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
ifr.ifr_ifru.ifru_mtu = mtu as i32;
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCSIFMTU as _)?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn get_mtu(&self, dev: &str) -> Result<u32> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCGIFMTU as _)?;
|
||||
|
||||
let mtu = unsafe { ifr.ifr_ifru.ifru_mtu as u32 };
|
||||
Ok(mtu)
|
||||
}
|
||||
|
||||
pub fn get_index(&self, dev: &str) -> Result<c_int> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
|
||||
ifr.ioctl(&self, SIOCGIFINDEX as _)?;
|
||||
|
||||
Ok(unsafe { ifr.ifr_ifru.ifru_ivalue })
|
||||
}
|
||||
|
||||
pub fn set_address(&self, dev: &str, address: &IpAddr, prefix_length: u32) -> Result<()> {
|
||||
let index = self.get_index(dev)?;
|
||||
match address {
|
||||
IpAddr::V4(addr) => {
|
||||
if prefix_length > 32 {
|
||||
return Err(Error::from_raw_os_error(libc::EINVAL));
|
||||
}
|
||||
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
ifr.ifr_ifru.ifru_addr_v4.sin_family = libc::AF_INET as _;
|
||||
ifr.ifr_ifru.ifru_addr_v4.sin_addr.s_addr = u32::from_ne_bytes(addr.octets());
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCSIFADDR as _)?;
|
||||
|
||||
ifr.ifr_ifru.ifru_addr_v4.sin_addr.s_addr = u32::MAX >> (32 - prefix_length);
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCSIFNETMASK as _)?;
|
||||
}
|
||||
IpAddr::V6(addr) => {
|
||||
let mut ifr = Ifreq6 {
|
||||
prefix_length,
|
||||
ifindex: index as _,
|
||||
addr: in6_addr {
|
||||
s6_addr: addr.octets(),
|
||||
},
|
||||
};
|
||||
ifr.ioctl(&self, libc::SIOCSIFADDR as _)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
//let mut ifr = Ifreq::new(dev);
|
||||
/*match address {
|
||||
IpAddr::V4(addr) => {
|
||||
ifr.ifr_ifru.ifru_addr_v4.sin_family = libc::AF_INET as libc::sa_family_t;
|
||||
ifr.ifr_ifru.ifru_addr_v4.sin_addr.s_addr = u32::from_ne_bytes(addr.octets());
|
||||
}
|
||||
}*/
|
||||
|
||||
/*let res = unsafe { libc::ioctl(self.fd, libc::SIOCSIFADDR as _, &mut ifr) };*/
|
||||
}
|
||||
|
||||
pub fn get_address(&self, dev: &str) -> Result<Ipv4Addr> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCGIFADDR as _)?;
|
||||
|
||||
// SAFETY: union
|
||||
let addr: Ipv4Addr = unsafe {
|
||||
if ifr.ifr_ifru.ifru_addr_v4.sin_family != libc::AF_INET as _ {
|
||||
return Err(Error::from_raw_os_error(libc::ENOTRECOVERABLE));
|
||||
}
|
||||
ifr.ifr_ifru.ifru_addr_v4.sin_addr.s_addr
|
||||
}
|
||||
.into();
|
||||
|
||||
Ok(addr)
|
||||
}
|
||||
|
||||
pub fn set_mac(&self, dev: &str, mac: &str) -> Result<()> {
|
||||
self.set_mac_sa_data(dev, parse_mac_addr(mac)?)
|
||||
}
|
||||
pub fn set_mac_sa_data(&self, dev: &str, mac: [libc::c_char; 14]) -> Result<()> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
ifr.ifr_ifru.ifru_hwaddr.sa_family = libc::ARPHRD_ETHER;
|
||||
ifr.ifr_ifru.ifru_hwaddr.sa_data = mac;
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCSIFHWADDR as _)
|
||||
}
|
||||
|
||||
pub fn get_mac_sa_data(&self, dev: &str) -> Result<[libc::c_char; 14]> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
ifr.ifr_ifru.ifru_hwaddr.sa_family = libc::ARPHRD_ETHER;
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCGIFHWADDR as _)?;
|
||||
|
||||
let sa_data = unsafe { ifr.ifr_ifru.ifru_hwaddr.sa_data };
|
||||
Ok(sa_data)
|
||||
}
|
||||
// TODO: get_mac -> String
|
||||
|
||||
/// Get the hardware type of the given interface
|
||||
pub fn get_arptype(&self, dev: &str) -> Result<libc::sa_family_t> {
|
||||
let mut ifr = Ifreq::new(dev);
|
||||
|
||||
ifr.ioctl(&self, libc::SIOCGIFHWADDR as _)?;
|
||||
|
||||
let arptype = unsafe { ifr.ifr_ifru.ifru_hwaddr.sa_family };
|
||||
return Ok(arptype);
|
||||
}
|
||||
|
||||
fn get_ctl_fd() -> Result<c_int> {
|
||||
let fd = unsafe { socket(libc::PF_INET, libc::SOCK_DGRAM, 0) };
|
||||
if fd >= 0 {
|
||||
return Ok(fd);
|
||||
}
|
||||
let error = std::io::Error::last_os_error();
|
||||
let fd = unsafe { socket(libc::PF_PACKET, libc::SOCK_DGRAM, 0) };
|
||||
if fd >= 0 {
|
||||
return Ok(fd);
|
||||
}
|
||||
let fd = unsafe { socket(libc::PF_INET6, libc::SOCK_DGRAM, 0) };
|
||||
if fd >= 0 {
|
||||
return Ok(fd);
|
||||
}
|
||||
Err(error)
|
||||
}
|
||||
}
|
||||
|
||||
impl Drop for IpTool {
|
||||
fn drop(&mut self) {
|
||||
unsafe { close(self.fd) };
|
||||
}
|
||||
}
|
||||
|
||||
#[repr(C)]
|
||||
union IfrIfru {
|
||||
ifru_addr: sockaddr,
|
||||
ifru_hwaddr: sockaddr,
|
||||
ifru_addr_v4: sockaddr_in,
|
||||
ifru_addr_v6: sockaddr_in6,
|
||||
ifru_dstaddr: sockaddr,
|
||||
ifru_broadaddr: sockaddr,
|
||||
ifru_flags: c_short,
|
||||
ifru_metric: c_int,
|
||||
ifru_ivalue: c_int,
|
||||
ifru_mtu: c_int,
|
||||
ifru_phys: c_int,
|
||||
ifru_media: c_int,
|
||||
ifru_intval: c_int,
|
||||
//ifru_data: caddr_t,
|
||||
//ifru_devmtu: ifdevmtu,
|
||||
//ifru_kpi: ifkpi,
|
||||
ifru_wake_flags: u32,
|
||||
ifru_route_refcnt: u32,
|
||||
ifru_cap: [c_int; 2],
|
||||
ifru_functional_type: u32,
|
||||
}
|
||||
|
||||
trait IoctlReq {
|
||||
fn ioctl(&mut self, iptool: &IpTool, request: libc::c_ulong) -> Result<()> {
|
||||
let res = unsafe { ioctl(iptool.fd, request as _, self) };
|
||||
if res < 0 {
|
||||
Err(last_err())
|
||||
} else {
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[repr(C)]
|
||||
pub struct Ifreq {
|
||||
ifr_name: [c_uchar; libc::IFNAMSIZ],
|
||||
ifr_ifru: IfrIfru,
|
||||
}
|
||||
|
||||
impl Ifreq {
|
||||
pub fn new(dev: &str) -> Self {
|
||||
//let mut ifr_name = [0; libc::IF_NAMESIZE];
|
||||
|
||||
//ifr_name[..dev.len()].copy_from_slice(dev.as_bytes().as_ref());
|
||||
|
||||
let s: [u8; core::mem::size_of::<Self>()] = [0; core::mem::size_of::<Self>()];
|
||||
let mut s: Self = unsafe { core::mem::transmute(s) };
|
||||
|
||||
copy_slice(&mut s.ifr_name, dev.as_bytes());
|
||||
|
||||
s
|
||||
/*Self {
|
||||
ifr_name,
|
||||
ifr_ifru: IfrIfru { ifru_flags: 0 },
|
||||
}*/
|
||||
}
|
||||
}
|
||||
|
||||
impl IoctlReq for Ifreq {}
|
||||
|
||||
#[repr(C)]
|
||||
pub struct Ifreq6 {
|
||||
addr: in6_addr,
|
||||
prefix_length: u32,
|
||||
ifindex: libc::c_uint,
|
||||
}
|
||||
|
||||
impl IoctlReq for Ifreq6 {}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
use super::IpTool;
|
||||
use std::net::{IpAddr, Ipv4Addr};
|
||||
|
||||
const TEST_INTERFACE: &str = "loop1";
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn down() {
|
||||
let ip_tool = IpTool::new().unwrap();
|
||||
|
||||
ip_tool.set_up(TEST_INTERFACE, false).unwrap();
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn up() {
|
||||
let ip_tool = IpTool::new().unwrap();
|
||||
|
||||
ip_tool.set_up(TEST_INTERFACE, true).unwrap();
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn sleep_down_and_up() {
|
||||
let ip_tool = IpTool::new().unwrap();
|
||||
|
||||
ip_tool.set_up(TEST_INTERFACE, false).unwrap();
|
||||
|
||||
std::thread::sleep(std::time::Duration::from_secs(5));
|
||||
|
||||
ip_tool.set_up(TEST_INTERFACE, true).unwrap();
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn mtu() {
|
||||
let ip_tool = IpTool::new().unwrap();
|
||||
|
||||
ip_tool.set_mtu(TEST_INTERFACE, 1420).unwrap();
|
||||
|
||||
assert_eq!(ip_tool.get_mtu(TEST_INTERFACE).unwrap(), 1420);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn mac() {
|
||||
let ip_tool = IpTool::new().unwrap();
|
||||
let mac = "5A:E6:60:8F:5F:DE";
|
||||
|
||||
ip_tool.set_mac(TEST_INTERFACE, mac).unwrap();
|
||||
|
||||
let sa_data = ip_tool.get_mac_sa_data(TEST_INTERFACE).unwrap();
|
||||
assert_eq!(sa_data, super::parse_mac_addr(mac).unwrap());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn set_ipv4() {
|
||||
let ip_tool = IpTool::new().unwrap();
|
||||
let address: Ipv4Addr = "10.23.42.1".parse().unwrap();
|
||||
|
||||
ip_tool
|
||||
.set_address(TEST_INTERFACE, &IpAddr::V4(address), 24)
|
||||
.unwrap();
|
||||
|
||||
let addr_on_link = ip_tool.get_address(TEST_INTERFACE).unwrap();
|
||||
assert_eq!(address, addr_on_link);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[ignore]
|
||||
fn ipv4addr_link_ext() {
|
||||
use super::IpAddrLinkExt;
|
||||
let address: Ipv4Addr = "10.23.42.1".parse().unwrap();
|
||||
|
||||
let addr_if = Ipv4Addr::from_interface(TEST_INTERFACE).unwrap();
|
||||
|
||||
assert_eq!(address, addr_if);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn get_arptype() {
|
||||
let iptool = IpTool::new().unwrap();
|
||||
let arptype = iptool.get_arptype("lo").unwrap();
|
||||
assert_eq!(arptype, libc::ARPHRD_LOOPBACK);
|
||||
}
|
||||
}
|
||||
2
runner/rust-toolchain.toml
Normal file
2
runner/rust-toolchain.toml
Normal file
@ -0,0 +1,2 @@
|
||||
[toolchain]
|
||||
channel = "nightly"
|
||||
33
runner/src/main.rs
Normal file
33
runner/src/main.rs
Normal file
@ -0,0 +1,33 @@
|
||||
#![feature(io_error_uncategorized)]
|
||||
|
||||
use iptool::IpTool;
|
||||
use std::{io, net::Ipv4Addr, process::Command, str::FromStr};
|
||||
|
||||
use capctl::{Cap, FileCaps};
|
||||
|
||||
fn main() {
|
||||
Command::new("cargo")
|
||||
.arg("build")
|
||||
.spawn()
|
||||
.unwrap()
|
||||
.wait()
|
||||
.unwrap();
|
||||
let mut caps = FileCaps::empty();
|
||||
caps.effective = true;
|
||||
caps.permitted.add(Cap::NET_ADMIN);
|
||||
caps.permitted.add(Cap::NET_RAW);
|
||||
caps.set_for_file("target/debug/nstack-rs").unwrap();
|
||||
let mut nstack = Command::new("target/debug/nstack-rs").spawn().unwrap();
|
||||
let iptool = IpTool::new().unwrap();
|
||||
while let Err(x) =
|
||||
iptool.set_address("tap0", &(Ipv4Addr::from_str("10.0.0.2").unwrap().into()), 8)
|
||||
{
|
||||
if x.kind() == io::ErrorKind::Uncategorized {
|
||||
continue;
|
||||
} else {
|
||||
Err::<(), _>(x).unwrap();
|
||||
}
|
||||
}
|
||||
iptool.set_up("tap0", true).unwrap();
|
||||
nstack.wait().unwrap();
|
||||
}
|
||||
217
src/arp.rs
Normal file
217
src/arp.rs
Normal file
@ -0,0 +1,217 @@
|
||||
use std::{
|
||||
cell::RefCell,
|
||||
collections::{hash_map::Entry, HashMap},
|
||||
fmt::Display,
|
||||
net::Ipv4Addr,
|
||||
};
|
||||
|
||||
use derive_try_from_primitive::TryFromPrimitive;
|
||||
|
||||
use crate::{
|
||||
ethernet::{EthernetHeader, MacAddr},
|
||||
iface::Iface,
|
||||
packet::{Header, HeaderParse, Packet},
|
||||
};
|
||||
|
||||
#[derive(Copy, Clone, Debug)]
|
||||
pub enum ArpParseError {
|
||||
HardwareType(u16),
|
||||
ProtocolType(u16),
|
||||
ProtocolAddressLen(u8),
|
||||
HardwareAddressLen(u8),
|
||||
Opcode(u16),
|
||||
}
|
||||
|
||||
#[derive(Copy, Clone, Debug, TryFromPrimitive, PartialEq, Eq)]
|
||||
#[repr(u16)]
|
||||
pub enum ArpOpcode {
|
||||
Request = 1,
|
||||
Reply = 2,
|
||||
}
|
||||
|
||||
#[derive(Copy, Clone, Debug)]
|
||||
pub struct ArpMessage {
|
||||
op: ArpOpcode,
|
||||
sha: MacAddr,
|
||||
spa: Ipv4Addr,
|
||||
tha: MacAddr,
|
||||
tpa: Ipv4Addr,
|
||||
}
|
||||
|
||||
impl Display for ArpMessage {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self.op {
|
||||
ArpOpcode::Request => f.write_fmt(format_args!(
|
||||
"ARP request from {} ({}) for {}",
|
||||
self.spa, self.sha, self.tpa,
|
||||
)),
|
||||
ArpOpcode::Reply => f.write_fmt(format_args!(
|
||||
"ARP response for {} ({}) to {} ({})",
|
||||
self.spa, self.sha, self.tpa, self.tha,
|
||||
)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl ArpMessage {
|
||||
pub fn new(op: ArpOpcode, sha: MacAddr, spa: Ipv4Addr, tha: MacAddr, tpa: Ipv4Addr) -> Self {
|
||||
Self {
|
||||
op,
|
||||
sha,
|
||||
spa,
|
||||
tha,
|
||||
tpa,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn op(&self) -> ArpOpcode {
|
||||
self.op
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn sha(&self) -> MacAddr {
|
||||
self.sha
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn spa(&self) -> Ipv4Addr {
|
||||
self.spa
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn tha(&self) -> MacAddr {
|
||||
self.tha
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn tpa(&self) -> Ipv4Addr {
|
||||
self.tpa
|
||||
}
|
||||
}
|
||||
|
||||
impl Header for ArpMessage {
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>) {
|
||||
buf.extend_from_slice(&1u16.to_be_bytes());
|
||||
buf.extend_from_slice(&0x800u16.to_be_bytes());
|
||||
buf.push(6);
|
||||
buf.push(4);
|
||||
buf.extend_from_slice(&(self.op as u16).to_be_bytes());
|
||||
buf.extend_from_slice(self.sha.octets());
|
||||
buf.extend_from_slice(&self.spa.octets());
|
||||
buf.extend_from_slice(self.tha.octets());
|
||||
buf.extend_from_slice(&self.tpa.octets());
|
||||
}
|
||||
}
|
||||
|
||||
impl HeaderParse for ArpMessage {
|
||||
type ParseError = ArpParseError;
|
||||
fn parse(message: &[u8]) -> Result<(Self, &[u8]), ArpParseError> {
|
||||
let hrd = u16::from_be_bytes((&message[0..2]).try_into().unwrap());
|
||||
if hrd != 1 {
|
||||
return Err(ArpParseError::HardwareType(hrd));
|
||||
}
|
||||
let pro = u16::from_be_bytes((&message[2..4]).try_into().unwrap());
|
||||
if pro != 0x800 {
|
||||
return Err(ArpParseError::ProtocolType(pro));
|
||||
}
|
||||
let hln = u8::from_be_bytes((&message[4..5]).try_into().unwrap());
|
||||
if hln != 6 {
|
||||
return Err(ArpParseError::HardwareAddressLen(hln));
|
||||
}
|
||||
let pln = u8::from_be_bytes((&message[5..6]).try_into().unwrap());
|
||||
if pln != 4 {
|
||||
return Err(ArpParseError::ProtocolAddressLen(pln));
|
||||
}
|
||||
let op = ArpOpcode::try_from(u16::from_be_bytes((&message[6..8]).try_into().unwrap()))
|
||||
.map_err(ArpParseError::Opcode)?;
|
||||
let sha = MacAddr::new((&message[8..14]).try_into().unwrap());
|
||||
let spa = Ipv4Addr::from(<[u8; 4]>::try_from(&message[14..18]).unwrap());
|
||||
let tha = MacAddr::new((&message[18..24]).try_into().unwrap());
|
||||
let tpa = Ipv4Addr::from(<[u8; 4]>::try_from(&message[24..28]).unwrap());
|
||||
Ok((
|
||||
Self {
|
||||
op,
|
||||
sha,
|
||||
spa,
|
||||
tha,
|
||||
tpa,
|
||||
},
|
||||
&[],
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
type ArpCallback<'a> = Box<dyn FnOnce(MacAddr) + 'a>;
|
||||
|
||||
pub struct ArpClient<'a> {
|
||||
iface: &'a Iface,
|
||||
cache: RefCell<HashMap<Ipv4Addr, MacAddr>>,
|
||||
waiting: RefCell<HashMap<Ipv4Addr, Vec<ArpCallback<'a>>>>,
|
||||
}
|
||||
|
||||
impl<'a> ArpClient<'a> {
|
||||
pub fn new(iface: &'a Iface) -> Self {
|
||||
Self {
|
||||
iface,
|
||||
cache: RefCell::new(HashMap::new()),
|
||||
waiting: RefCell::new(HashMap::new()),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn handle_message(&self, message: Packet<EthernetHeader, ArpMessage>) {
|
||||
match message.l3().op() {
|
||||
ArpOpcode::Request => {
|
||||
if let Some(ip) = self.iface.ip() {
|
||||
if message.l3().tpa() == ip {
|
||||
self.iface.send(
|
||||
message.l3().sha(),
|
||||
Packet::new(Vec::new()).set_l3(ArpMessage::new(
|
||||
ArpOpcode::Reply,
|
||||
self.iface.mac(),
|
||||
ip,
|
||||
message.l3().sha(),
|
||||
message.l3().spa(),
|
||||
)),
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
ArpOpcode::Reply => {
|
||||
self.cache
|
||||
.borrow_mut()
|
||||
.insert(message.l3().spa(), message.l3().sha());
|
||||
if let Some(callback_vec) = self.waiting.borrow_mut().remove(&message.l3().spa()) {
|
||||
for callback in callback_vec {
|
||||
callback(message.l3().sha());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get_mac(&self, ip: Ipv4Addr, callback: Box<dyn FnOnce(MacAddr) + 'a>) {
|
||||
if let Some(&mac) = self.cache.borrow().get(&ip) {
|
||||
callback(mac);
|
||||
} else {
|
||||
match self.waiting.borrow_mut().entry(ip) {
|
||||
Entry::Vacant(e) => {
|
||||
e.insert(vec![callback]);
|
||||
self.iface.send(
|
||||
MacAddr::broadcast(),
|
||||
Packet::new(Vec::new()).set_l3(ArpMessage::new(
|
||||
ArpOpcode::Request,
|
||||
self.iface.mac(),
|
||||
self.iface.ip().unwrap_or(Ipv4Addr::new(0, 0, 0, 0)),
|
||||
MacAddr::broadcast(),
|
||||
ip,
|
||||
)),
|
||||
);
|
||||
}
|
||||
Entry::Occupied(mut e) => {
|
||||
e.get_mut().push(callback);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
124
src/ethernet.rs
Normal file
124
src/ethernet.rs
Normal file
@ -0,0 +1,124 @@
|
||||
use std::fmt::Display;
|
||||
|
||||
use derive_try_from_primitive::TryFromPrimitive;
|
||||
|
||||
use crate::{
|
||||
arp::ArpMessage,
|
||||
from_header_type_impl,
|
||||
ip::IpHeader,
|
||||
packet::{Header, HeaderParse},
|
||||
};
|
||||
|
||||
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
|
||||
pub struct MacAddr([u8; 6]);
|
||||
|
||||
impl Display for MacAddr {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
for byte in &self.0[0..5] {
|
||||
f.write_fmt(format_args!("{:0>2x}:", byte))?;
|
||||
}
|
||||
f.write_fmt(format_args!("{:0>2x}", self.0[5]))?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl MacAddr {
|
||||
pub fn new(mac: [u8; 6]) -> MacAddr {
|
||||
Self(mac)
|
||||
}
|
||||
|
||||
pub fn is_broadcast(&self) -> bool {
|
||||
self == &Self::broadcast()
|
||||
}
|
||||
|
||||
pub fn octets(&self) -> &[u8] {
|
||||
&self.0
|
||||
}
|
||||
|
||||
pub fn broadcast() -> MacAddr {
|
||||
Self([0xFFu8; 6])
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(TryFromPrimitive)]
|
||||
#[repr(u16)]
|
||||
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
|
||||
pub enum EtherType {
|
||||
Ipv4 = 0x800,
|
||||
Arp = 0x806,
|
||||
}
|
||||
|
||||
impl Display for EtherType {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
Self::Ipv4 => f.write_str("IPv4"),
|
||||
Self::Arp => f.write_str("ARP"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
from_header_type_impl!(EtherType, IpHeader => Self::Ipv4, ArpMessage => Self::Arp);
|
||||
|
||||
#[derive(Copy, Clone, Debug)]
|
||||
pub struct InvalidEtherType(pub u16);
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct EthernetHeader {
|
||||
dst: MacAddr,
|
||||
src: MacAddr,
|
||||
typ: EtherType,
|
||||
}
|
||||
|
||||
impl Display for EthernetHeader {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
f.write_fmt(format_args!(
|
||||
"{} from {} to {}",
|
||||
self.typ, self.src, self.dst,
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
impl EthernetHeader {
|
||||
#[allow(dead_code)]
|
||||
pub fn new(dst: MacAddr, src: MacAddr, typ: EtherType) -> Self {
|
||||
Self { dst, src, typ }
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn dst(&self) -> MacAddr {
|
||||
self.dst
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn src(&self) -> MacAddr {
|
||||
self.src
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn typ(&self) -> EtherType {
|
||||
self.typ
|
||||
}
|
||||
}
|
||||
|
||||
impl Header for EthernetHeader {
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>) {
|
||||
buf.extend_from_slice(self.dst.octets());
|
||||
buf.extend_from_slice(self.src.octets());
|
||||
buf.extend_from_slice(&(self.typ as u16).to_be_bytes());
|
||||
}
|
||||
}
|
||||
|
||||
impl HeaderParse for EthernetHeader {
|
||||
type ParseError = InvalidEtherType;
|
||||
fn parse(frame: &[u8]) -> Result<(Self, &[u8]), InvalidEtherType> {
|
||||
Ok((
|
||||
Self {
|
||||
dst: MacAddr(frame[0..6].try_into().unwrap()),
|
||||
src: MacAddr(frame[6..12].try_into().unwrap()),
|
||||
typ: EtherType::try_from(u16::from_be_bytes(frame[12..14].try_into().unwrap()))
|
||||
.map_err(InvalidEtherType)?,
|
||||
},
|
||||
&frame[14..],
|
||||
))
|
||||
}
|
||||
}
|
||||
94
src/icmp.rs
Normal file
94
src/icmp.rs
Normal file
@ -0,0 +1,94 @@
|
||||
use crate::packet::{Header, HeaderParse};
|
||||
use derive_try_from_primitive::TryFromPrimitive;
|
||||
use internet_checksum::Checksum;
|
||||
use std::fmt::Display;
|
||||
|
||||
#[derive(Debug, Copy, Clone)]
|
||||
pub struct InvalidType(pub u8);
|
||||
|
||||
#[derive(Debug, Copy, Clone, TryFromPrimitive)]
|
||||
#[repr(u8)]
|
||||
enum IcmpHeaderType {
|
||||
EchoRequest = 8,
|
||||
EchoReply = 0,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum IcmpHeader {
|
||||
EchoRequest { id: u16, seq: u16 },
|
||||
EchoReply { id: u16, seq: u16 },
|
||||
}
|
||||
|
||||
impl IcmpHeader {
|
||||
fn typ(&self) -> IcmpHeaderType {
|
||||
match self {
|
||||
Self::EchoRequest { .. } => IcmpHeaderType::EchoRequest,
|
||||
Self::EchoReply { .. } => IcmpHeaderType::EchoReply,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for IcmpHeader {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
Self::EchoRequest { id, seq } => f.write_fmt(format_args!(
|
||||
"ICMP Echo Request with ID {} and seq number {}",
|
||||
id, seq,
|
||||
)),
|
||||
Self::EchoReply { id, seq } => f.write_fmt(format_args!(
|
||||
"ICMP Echo Reply with ID {} and seq number {}.",
|
||||
id, seq,
|
||||
)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Header for IcmpHeader {
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>) {
|
||||
buf.extend_from_slice(&[self.typ() as u8, 0]);
|
||||
buf.extend_from_slice(&[0, 0]); // Filled with checksum in stage 2
|
||||
match self {
|
||||
Self::EchoRequest { id, seq } => {
|
||||
buf.extend_from_slice(&id.to_be_bytes());
|
||||
buf.extend_from_slice(&seq.to_be_bytes());
|
||||
}
|
||||
Self::EchoReply { id, seq } => {
|
||||
buf.extend_from_slice(&id.to_be_bytes());
|
||||
buf.extend_from_slice(&seq.to_be_bytes());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn serialize_stage2(hdr: &mut [u8], data: &[u8]) {
|
||||
let mut checksum = Checksum::new();
|
||||
checksum.add_bytes(hdr);
|
||||
checksum.add_bytes(data);
|
||||
hdr[2..4].copy_from_slice(&checksum.checksum());
|
||||
}
|
||||
}
|
||||
|
||||
impl HeaderParse for IcmpHeader {
|
||||
type ParseError = InvalidType;
|
||||
fn parse(message: &[u8]) -> Result<(Self, &[u8]), InvalidType> {
|
||||
let typ = IcmpHeaderType::try_from(message[0]).map_err(InvalidType)?;
|
||||
let _code = message[1];
|
||||
let rest = &message[4..];
|
||||
match typ {
|
||||
IcmpHeaderType::EchoRequest => Ok((
|
||||
Self::EchoRequest {
|
||||
id: u16::from_be_bytes(rest[0..2].try_into().unwrap()),
|
||||
seq: u16::from_be_bytes(rest[2..4].try_into().unwrap()),
|
||||
},
|
||||
&rest[4..],
|
||||
)),
|
||||
|
||||
IcmpHeaderType::EchoReply => Ok((
|
||||
Self::EchoReply {
|
||||
id: u16::from_be_bytes(rest[0..2].try_into().unwrap()),
|
||||
seq: u16::from_be_bytes(rest[2..4].try_into().unwrap()),
|
||||
},
|
||||
&rest[4..],
|
||||
)),
|
||||
}
|
||||
}
|
||||
}
|
||||
62
src/iface.rs
Normal file
62
src/iface.rs
Normal file
@ -0,0 +1,62 @@
|
||||
use std::{cell::Cell, net::Ipv4Addr};
|
||||
|
||||
use tun_tap::Mode;
|
||||
|
||||
use crate::{
|
||||
ethernet::{EtherType, EthernetHeader, InvalidEtherType, MacAddr},
|
||||
packet::{FromHeaderType, Header, NoHeader, OptionalHeader, Packet},
|
||||
};
|
||||
|
||||
pub struct Iface {
|
||||
iface: tun_tap::Iface,
|
||||
mac: MacAddr,
|
||||
ip: Cell<Option<Ipv4Addr>>,
|
||||
}
|
||||
|
||||
impl Iface {
|
||||
pub fn new() -> Iface {
|
||||
Self {
|
||||
iface: tun_tap::Iface::without_packet_info("tap0", Mode::Tap).unwrap(),
|
||||
mac: MacAddr::new([0x9A, 0xB1, 0xB5, 0x22, 0x38, 0xAB]),
|
||||
ip: Cell::new(None),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn recv(&self) -> Result<Packet<EthernetHeader>, InvalidEtherType> {
|
||||
let mut frame_buf = [0u8; 2048];
|
||||
let frame_len = self.iface.recv(&mut frame_buf).unwrap();
|
||||
let packet = Packet::new(frame_buf[0..frame_len].to_vec());
|
||||
packet.parse::<EthernetHeader>().map_err(|(_pkt, err)| err)
|
||||
}
|
||||
|
||||
pub fn mac_to_self(&self, mac: MacAddr) -> bool {
|
||||
mac == self.mac || mac.is_broadcast()
|
||||
}
|
||||
|
||||
pub fn send<L3, L4, L7>(&self, dst: MacAddr, packet: Packet<NoHeader, L3, L4, L7>)
|
||||
where
|
||||
L3: Header,
|
||||
L4: OptionalHeader,
|
||||
L7: OptionalHeader,
|
||||
EtherType: FromHeaderType<L3>,
|
||||
{
|
||||
let eth_header = EthernetHeader::new(dst, self.mac, EtherType::from_hdr_type());
|
||||
let packet = packet.set_l2(eth_header);
|
||||
println!("------------------");
|
||||
println!("{}", packet);
|
||||
println!("------------------");
|
||||
self.iface.send(&packet.serialize()).unwrap();
|
||||
}
|
||||
|
||||
pub fn mac(&self) -> MacAddr {
|
||||
self.mac
|
||||
}
|
||||
|
||||
pub fn ip(&self) -> Option<Ipv4Addr> {
|
||||
self.ip.get()
|
||||
}
|
||||
|
||||
pub fn set_ip(&self, ip: Ipv4Addr) {
|
||||
self.ip.set(Some(ip))
|
||||
}
|
||||
}
|
||||
170
src/ip.rs
Normal file
170
src/ip.rs
Normal file
@ -0,0 +1,170 @@
|
||||
use std::{fmt::Display, net::Ipv4Addr, rc::Rc};
|
||||
|
||||
use derive_try_from_primitive::TryFromPrimitive;
|
||||
|
||||
use crate::{
|
||||
arp::ArpClient,
|
||||
from_header_type_impl,
|
||||
icmp::IcmpHeader,
|
||||
iface::Iface,
|
||||
packet::{FromHeaderType, Header, HeaderParse, NoHeader, OptionalHeader, Packet},
|
||||
};
|
||||
|
||||
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
|
||||
pub enum ParseError {
|
||||
FragmentsUnsupported,
|
||||
OptionsUnsupported,
|
||||
InvalidVersion,
|
||||
InvaliidProtocol(u8),
|
||||
}
|
||||
|
||||
#[repr(u8)]
|
||||
#[derive(Debug, Copy, Clone, PartialEq, Eq, TryFromPrimitive)]
|
||||
pub enum IpProtocolType {
|
||||
Icmp = 1,
|
||||
Tcp = 6,
|
||||
Udp = 17,
|
||||
}
|
||||
|
||||
impl Display for IpProtocolType {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
Self::Icmp => f.write_str("ICMP"),
|
||||
Self::Tcp => f.write_str("TCP"),
|
||||
Self::Udp => f.write_str("UDP"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
from_header_type_impl!(IpProtocolType, IcmpHeader => Self::Icmp);
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct IpHeader {
|
||||
ttl: u8,
|
||||
protocol: IpProtocolType,
|
||||
src: Ipv4Addr,
|
||||
dst: Ipv4Addr,
|
||||
}
|
||||
|
||||
impl Display for IpHeader {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
f.write_fmt(format_args!(
|
||||
"IP datagram from {} to {} ({})",
|
||||
self.src, self.dst, self.protocol,
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
impl IpHeader {
|
||||
pub fn new(protocol: IpProtocolType, src: Ipv4Addr, dst: Ipv4Addr) -> Self {
|
||||
Self {
|
||||
ttl: 255,
|
||||
protocol,
|
||||
src,
|
||||
dst,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn ttl(&self) -> u8 {
|
||||
self.ttl
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn protocol(&self) -> IpProtocolType {
|
||||
self.protocol
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn dst(&self) -> Ipv4Addr {
|
||||
self.dst
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn src(&self) -> Ipv4Addr {
|
||||
self.src
|
||||
}
|
||||
}
|
||||
|
||||
impl Header for IpHeader {
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>) {
|
||||
buf.extend_from_slice(&[0x45, 0]);
|
||||
buf.extend_from_slice(&[0, 0]); // Filled with length in stage 2
|
||||
buf.extend_from_slice(&[0, 0]);
|
||||
buf.extend_from_slice(&[0, 0]);
|
||||
buf.push(self.ttl);
|
||||
buf.push(self.protocol as u8);
|
||||
buf.extend_from_slice(&[0, 0]); // Filled with checksum in stage 2
|
||||
buf.extend_from_slice(&self.src.octets());
|
||||
buf.extend_from_slice(&self.dst.octets());
|
||||
}
|
||||
|
||||
fn serialize_stage2(hdr: &mut [u8], data: &[u8]) {
|
||||
hdr[2..4].copy_from_slice(&(20u16 + data.len() as u16).to_be_bytes());
|
||||
let checksum = internet_checksum::checksum(hdr);
|
||||
hdr[10..12].copy_from_slice(&checksum);
|
||||
}
|
||||
}
|
||||
|
||||
impl HeaderParse for IpHeader {
|
||||
type ParseError = ParseError;
|
||||
fn parse(datagram: &[u8]) -> Result<(Self, &[u8]), ParseError> {
|
||||
if datagram[0] & 0xf0 != 0x40 {
|
||||
return Err(ParseError::InvalidVersion);
|
||||
}
|
||||
if datagram[0] & 0xf != 0x5 {
|
||||
return Err(ParseError::OptionsUnsupported);
|
||||
}
|
||||
if datagram[6] & 0x20 != 0x0 {
|
||||
return Err(ParseError::FragmentsUnsupported);
|
||||
}
|
||||
let ttl = datagram[8];
|
||||
let protocol =
|
||||
IpProtocolType::try_from(datagram[9]).map_err(ParseError::InvaliidProtocol)?;
|
||||
let src = Ipv4Addr::from(<[u8; 4]>::try_from(&datagram[12..16]).unwrap());
|
||||
let dst = Ipv4Addr::from(<[u8; 4]>::try_from(&datagram[16..20]).unwrap());
|
||||
Ok((
|
||||
Self {
|
||||
ttl,
|
||||
protocol,
|
||||
src,
|
||||
dst,
|
||||
},
|
||||
&datagram[20..],
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
pub struct IpSender<'a, 'b> {
|
||||
iface: Rc<Iface>,
|
||||
arp_client: &'a ArpClient<'b>,
|
||||
}
|
||||
|
||||
impl<'a, 'b> IpSender<'a, 'b> {
|
||||
pub fn new(iface: Rc<Iface>, arp_client: &'a ArpClient<'b>) -> Self {
|
||||
Self { iface, arp_client }
|
||||
}
|
||||
|
||||
pub fn send<L4: 'b, L7: 'b>(&self, dst: Ipv4Addr, packet: Packet<NoHeader, NoHeader, L4, L7>)
|
||||
where
|
||||
L4: Header,
|
||||
L7: OptionalHeader,
|
||||
IpProtocolType: FromHeaderType<L4>,
|
||||
{
|
||||
let header = IpHeader::new(
|
||||
IpProtocolType::from_hdr_type(),
|
||||
self.iface.ip().unwrap(),
|
||||
dst,
|
||||
);
|
||||
let packet = packet.set_l3(header);
|
||||
let iface = Rc::downgrade(&self.iface);
|
||||
self.arp_client.get_mac(
|
||||
dst,
|
||||
Box::new(move |mac| {
|
||||
if let Some(iface) = iface.upgrade() {
|
||||
iface.send(mac, packet);
|
||||
}
|
||||
}),
|
||||
);
|
||||
}
|
||||
}
|
||||
67
src/main.rs
Normal file
67
src/main.rs
Normal file
@ -0,0 +1,67 @@
|
||||
mod arp;
|
||||
mod ethernet;
|
||||
mod icmp;
|
||||
mod iface;
|
||||
mod ip;
|
||||
mod packet;
|
||||
|
||||
use std::{net::Ipv4Addr, rc::Rc};
|
||||
|
||||
use arp::ArpClient;
|
||||
use iface::Iface;
|
||||
use ip::{IpHeader, IpSender};
|
||||
|
||||
use crate::{
|
||||
arp::ArpMessage,
|
||||
ethernet::{EtherType, InvalidEtherType},
|
||||
icmp::IcmpHeader,
|
||||
ip::IpProtocolType,
|
||||
packet::Packet,
|
||||
};
|
||||
|
||||
fn main() {
|
||||
let iface = Rc::new(Iface::new());
|
||||
iface.set_ip(Ipv4Addr::new(10, 0, 0, 1));
|
||||
let arp_client = ArpClient::new(&iface);
|
||||
let ip_sender = IpSender::new(iface.clone(), &arp_client);
|
||||
loop {
|
||||
let packet = match iface.recv() {
|
||||
Ok(x) => x,
|
||||
Err(InvalidEtherType(_typ)) => {
|
||||
continue;
|
||||
}
|
||||
};
|
||||
if !iface.mac_to_self(packet.l2().dst()) {
|
||||
continue;
|
||||
}
|
||||
match packet.l2().typ() {
|
||||
EtherType::Arp => {
|
||||
let packet = packet.parse::<ArpMessage>().unwrap();
|
||||
println!("------------------");
|
||||
println!("{}", packet);
|
||||
println!("------------------");
|
||||
arp_client.handle_message(packet);
|
||||
}
|
||||
EtherType::Ipv4 => {
|
||||
let packet = packet.parse::<IpHeader>().unwrap();
|
||||
#[allow(clippy::single_match)]
|
||||
match packet.l3().protocol() {
|
||||
IpProtocolType::Icmp => {
|
||||
let packet = packet.parse::<IcmpHeader>().unwrap();
|
||||
println!("------------------");
|
||||
println!("{}", packet);
|
||||
println!("------------------");
|
||||
if let &IcmpHeader::EchoRequest { id, seq } = packet.l4() {
|
||||
ip_sender.send(
|
||||
packet.l3().src(),
|
||||
Packet::new(packet.body().to_vec())
|
||||
.set_l4(IcmpHeader::EchoReply { id, seq }),
|
||||
);
|
||||
}
|
||||
}
|
||||
_ => (),
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
||||
306
src/packet.rs
Normal file
306
src/packet.rs
Normal file
@ -0,0 +1,306 @@
|
||||
#![allow(clippy::transmute_ptr_to_ref)] // Shut up a warning generated by mopafy!
|
||||
|
||||
use std::fmt::{Debug, Display};
|
||||
|
||||
pub trait FromHeaderType<T: Header> {
|
||||
fn from_hdr_type() -> Self;
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! from_header_type_impl {
|
||||
($t: ty, $($ht: ty => $ex: expr),+) => {
|
||||
$(
|
||||
impl $crate::packet::FromHeaderType<$ht> for $t {
|
||||
fn from_hdr_type() -> Self {
|
||||
$ex
|
||||
}
|
||||
}
|
||||
)+
|
||||
}
|
||||
}
|
||||
|
||||
pub trait Header: Display + Debug {
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>);
|
||||
fn serialize_stage2(_hdr: &mut [u8], _data: &[u8]) {}
|
||||
}
|
||||
|
||||
pub trait HeaderParse: Header {
|
||||
type ParseError;
|
||||
fn parse(message: &[u8]) -> Result<(Self, &[u8]), Self::ParseError>
|
||||
where
|
||||
Self: Sized;
|
||||
}
|
||||
|
||||
pub trait OptionalHeader: Display + Debug {
|
||||
fn is_some(&self) -> bool;
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>);
|
||||
fn serialize_stage2(_hdr: &mut [u8], _data: &[u8]) {}
|
||||
}
|
||||
|
||||
impl<T: Header> OptionalHeader for T {
|
||||
fn is_some(&self) -> bool {
|
||||
true
|
||||
}
|
||||
fn serialize_stage1(&self, buf: &mut Vec<u8>) {
|
||||
<Self as Header>::serialize_stage1(self, buf)
|
||||
}
|
||||
|
||||
fn serialize_stage2(hdr: &mut [u8], data: &[u8]) {
|
||||
<Self as Header>::serialize_stage2(hdr, data)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct NoHeader;
|
||||
|
||||
impl Display for NoHeader {
|
||||
fn fmt(&self, _f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl OptionalHeader for NoHeader {
|
||||
fn is_some(&self) -> bool {
|
||||
false
|
||||
}
|
||||
fn serialize_stage1(&self, _buf: &mut Vec<u8>) {}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Packet<
|
||||
L2: OptionalHeader = NoHeader,
|
||||
L3: OptionalHeader = NoHeader,
|
||||
L4: OptionalHeader = NoHeader,
|
||||
L7: OptionalHeader = NoHeader,
|
||||
> {
|
||||
l2: L2,
|
||||
l3: L3,
|
||||
l4: L4,
|
||||
l7: L7,
|
||||
body: Vec<u8>,
|
||||
}
|
||||
|
||||
impl<L2: OptionalHeader, L3: OptionalHeader, L4: OptionalHeader, L7: OptionalHeader> Display
|
||||
for Packet<L2, L3, L4, L7>
|
||||
{
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
let mut printed = false;
|
||||
if self.l2.is_some() {
|
||||
f.write_fmt(format_args!("{}", self.l2))?;
|
||||
printed = true;
|
||||
}
|
||||
if self.l3.is_some() {
|
||||
if printed {
|
||||
f.write_str("\n")?;
|
||||
}
|
||||
f.write_fmt(format_args!("{}", self.l3))?;
|
||||
printed = true;
|
||||
}
|
||||
if self.l4.is_some() {
|
||||
if printed {
|
||||
f.write_str("\n")?;
|
||||
}
|
||||
f.write_fmt(format_args!("{}", self.l4))?;
|
||||
printed = true;
|
||||
}
|
||||
if self.l7.is_some() {
|
||||
if printed {
|
||||
f.write_str("\n")?;
|
||||
}
|
||||
f.write_fmt(format_args!("{}", self.l7))?;
|
||||
printed = true;
|
||||
}
|
||||
if !self.body.is_empty() {
|
||||
if printed {
|
||||
f.write_str("\n")?;
|
||||
}
|
||||
f.write_fmt(format_args!("{}", rhexdump::hexdump(&self.body)))?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl Packet {
|
||||
pub fn new(body: Vec<u8>) -> Self {
|
||||
Self {
|
||||
l2: NoHeader,
|
||||
l3: NoHeader,
|
||||
l4: NoHeader,
|
||||
l7: NoHeader,
|
||||
body,
|
||||
}
|
||||
}
|
||||
}
|
||||
impl<L2: OptionalHeader, L3: OptionalHeader, L4: OptionalHeader, L7: OptionalHeader>
|
||||
Packet<L2, L3, L4, L7>
|
||||
{
|
||||
#[allow(dead_code)]
|
||||
pub fn body(&self) -> &[u8] {
|
||||
self.body.as_ref()
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn set_l2<T: Header>(self, l2: T) -> Packet<T, L3, L4, L7> {
|
||||
Packet {
|
||||
l2,
|
||||
l3: self.l3,
|
||||
l4: self.l4,
|
||||
l7: self.l7,
|
||||
body: self.body,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn set_l3<T: Header>(self, l3: T) -> Packet<L2, T, L4, L7> {
|
||||
Packet {
|
||||
l2: self.l2,
|
||||
l3,
|
||||
l4: self.l4,
|
||||
l7: self.l7,
|
||||
body: self.body,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn set_l4<T: Header>(self, l4: T) -> Packet<L2, L3, T, L7> {
|
||||
Packet {
|
||||
l2: self.l2,
|
||||
l3: self.l3,
|
||||
l4,
|
||||
l7: self.l7,
|
||||
body: self.body,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn set_l7<T: Header>(self, l7: T) -> Packet<L2, L3, L4, T> {
|
||||
Packet {
|
||||
l2: self.l2,
|
||||
l3: self.l3,
|
||||
l4: self.l4,
|
||||
l7,
|
||||
body: self.body,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn serialize(&self) -> Vec<u8> {
|
||||
let mut buf = Vec::new();
|
||||
self.l2.serialize_stage1(&mut buf);
|
||||
let l3_start = buf.len();
|
||||
self.l3.serialize_stage1(&mut buf);
|
||||
let l4_start = buf.len();
|
||||
self.l4.serialize_stage1(&mut buf);
|
||||
let l7_start = buf.len();
|
||||
self.l7.serialize_stage1(&mut buf);
|
||||
let body_start = buf.len();
|
||||
buf.extend_from_slice(&self.body);
|
||||
let (l7_hdr, l7_body) = &mut buf[l7_start..].split_at_mut(body_start - l7_start);
|
||||
L7::serialize_stage2(l7_hdr, l7_body);
|
||||
let (l4_hdr, l4_body) = &mut buf[l4_start..].split_at_mut(l7_start - l4_start);
|
||||
L4::serialize_stage2(l4_hdr, l4_body);
|
||||
let (l3_hdr, l3_body) = &mut buf[l3_start..].split_at_mut(l4_start - l3_start);
|
||||
L3::serialize_stage2(l3_hdr, l3_body);
|
||||
let (l2_hdr, l2_body) = &mut buf.split_at_mut(l3_start);
|
||||
L2::serialize_stage2(l2_hdr, l2_body);
|
||||
buf
|
||||
}
|
||||
}
|
||||
|
||||
impl<L2: Header, L3: OptionalHeader, L4: OptionalHeader, L7: OptionalHeader>
|
||||
Packet<L2, L3, L4, L7>
|
||||
{
|
||||
#[allow(dead_code)]
|
||||
pub fn l2(&self) -> &L2 {
|
||||
&self.l2
|
||||
}
|
||||
}
|
||||
impl<L2: OptionalHeader, L3: Header, L4: OptionalHeader, L7: OptionalHeader>
|
||||
Packet<L2, L3, L4, L7>
|
||||
{
|
||||
#[allow(dead_code)]
|
||||
pub fn l3(&self) -> &L3 {
|
||||
&self.l3
|
||||
}
|
||||
}
|
||||
impl<L2: OptionalHeader, L3: OptionalHeader, L4: Header, L7: OptionalHeader>
|
||||
Packet<L2, L3, L4, L7>
|
||||
{
|
||||
#[allow(dead_code)]
|
||||
pub fn l4(&self) -> &L4 {
|
||||
&self.l4
|
||||
}
|
||||
}
|
||||
impl<L2: OptionalHeader, L3: OptionalHeader, L4: OptionalHeader, L7: Header>
|
||||
Packet<L2, L3, L4, L7>
|
||||
{
|
||||
#[allow(dead_code)]
|
||||
pub fn l7(&self) -> &L7 {
|
||||
&self.l7
|
||||
}
|
||||
}
|
||||
|
||||
impl Packet {
|
||||
#[allow(dead_code)]
|
||||
pub fn parse<T: HeaderParse>(self) -> Result<Packet<T>, (Self, T::ParseError)> {
|
||||
match T::parse(&self.body) {
|
||||
Ok((header, body)) => Ok(Packet {
|
||||
l2: header,
|
||||
l3: self.l3,
|
||||
l4: self.l4,
|
||||
l7: self.l7,
|
||||
body: body.to_vec(),
|
||||
}),
|
||||
Err(err) => Err((self, err)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<L2: Header> Packet<L2> {
|
||||
#[allow(dead_code)]
|
||||
pub fn parse<T: HeaderParse>(self) -> Result<Packet<L2, T>, (Self, T::ParseError)> {
|
||||
match T::parse(&self.body) {
|
||||
Ok((header, body)) => Ok(Packet {
|
||||
l2: self.l2,
|
||||
l3: header,
|
||||
l4: self.l4,
|
||||
l7: self.l7,
|
||||
body: body.to_vec(),
|
||||
}),
|
||||
Err(err) => Err((self, err)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<L2: Header, L3: Header> Packet<L2, L3> {
|
||||
#[allow(dead_code)]
|
||||
pub fn parse<T: HeaderParse>(self) -> Result<Packet<L2, L3, T>, (Self, T::ParseError)> {
|
||||
match T::parse(&self.body) {
|
||||
Ok((header, body)) => Ok(Packet {
|
||||
l2: self.l2,
|
||||
l3: self.l3,
|
||||
l4: header,
|
||||
l7: self.l7,
|
||||
body: body.to_vec(),
|
||||
}),
|
||||
Err(err) => Err((self, err)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<L2: Header, L3: Header, L4: Header> Packet<L2, L3, L4> {
|
||||
#[allow(dead_code)]
|
||||
#[allow(clippy::type_complexity)]
|
||||
pub fn parse<T: HeaderParse>(self) -> Result<Packet<L2, L3, L4, T>, (Self, T::ParseError)> {
|
||||
match T::parse(&self.body) {
|
||||
Ok((header, body)) => Ok(Packet {
|
||||
l2: self.l2,
|
||||
l3: self.l3,
|
||||
l4: self.l4,
|
||||
l7: header,
|
||||
body: body.to_vec(),
|
||||
}),
|
||||
Err(err) => Err((self, err)),
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user