summaryrefslogtreecommitdiffstats
path: root/src/wireguard/types/dummy.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/wireguard/types/dummy.rs')
-rw-r--r--src/wireguard/types/dummy.rs339
1 files changed, 0 insertions, 339 deletions
diff --git a/src/wireguard/types/dummy.rs b/src/wireguard/types/dummy.rs
deleted file mode 100644
index 384f123..0000000
--- a/src/wireguard/types/dummy.rs
+++ /dev/null
@@ -1,339 +0,0 @@
-use std::error::Error;
-use std::fmt;
-use std::marker;
-use std::net::SocketAddr;
-use std::sync::atomic::{AtomicUsize, Ordering};
-use std::sync::mpsc::{sync_channel, Receiver, SyncSender};
-use std::sync::Arc;
-use std::sync::Mutex;
-use std::time::Instant;
-
-use super::*;
-
-/* This submodule provides pure/dummy implementations of the IO interfaces
- * for use in unit tests thoughout the project.
- */
-
-/* Error implementation */
-
-#[derive(Debug)]
-pub enum BindError {
- Disconnected,
-}
-
-impl Error for BindError {
- fn description(&self) -> &str {
- "Generic Bind Error"
- }
-
- fn source(&self) -> Option<&(dyn Error + 'static)> {
- None
- }
-}
-
-impl fmt::Display for BindError {
- fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
- match self {
- BindError::Disconnected => write!(f, "PairBind disconnected"),
- }
- }
-}
-
-/* TUN implementation */
-
-#[derive(Debug)]
-pub enum TunError {
- Disconnected,
-}
-
-impl Error for TunError {
- fn description(&self) -> &str {
- "Generic Tun Error"
- }
-
- fn source(&self) -> Option<&(dyn Error + 'static)> {
- None
- }
-}
-
-impl fmt::Display for TunError {
- fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
- write!(f, "Not Possible")
- }
-}
-
-/* Endpoint implementation */
-
-#[derive(Clone, Copy)]
-pub struct UnitEndpoint {}
-
-impl Endpoint for UnitEndpoint {
- fn from_address(_: SocketAddr) -> UnitEndpoint {
- UnitEndpoint {}
- }
-
- fn into_address(&self) -> SocketAddr {
- "127.0.0.1:8080".parse().unwrap()
- }
-
- fn clear_src(&mut self) {}
-}
-
-impl UnitEndpoint {
- pub fn new() -> UnitEndpoint {
- UnitEndpoint {}
- }
-}
-
-/* */
-
-pub struct TunTest {}
-
-pub struct TunFakeIO {
- store: bool,
- tx: SyncSender<Vec<u8>>,
- rx: Receiver<Vec<u8>>,
-}
-
-pub struct TunReader {
- rx: Receiver<Vec<u8>>,
-}
-
-pub struct TunWriter {
- store: bool,
- tx: Mutex<SyncSender<Vec<u8>>>,
-}
-
-#[derive(Clone)]
-pub struct TunMTU {
- mtu: Arc<AtomicUsize>,
-}
-
-impl tun::Reader for TunReader {
- type Error = TunError;
-
- fn read(&self, buf: &mut [u8], offset: usize) -> Result<usize, Self::Error> {
- match self.rx.recv() {
- Ok(m) => {
- buf[offset..].copy_from_slice(&m[..]);
- Ok(m.len())
- }
- Err(_) => Err(TunError::Disconnected),
- }
- }
-}
-
-impl tun::Writer for TunWriter {
- type Error = TunError;
-
- fn write(&self, src: &[u8]) -> Result<(), Self::Error> {
- if self.store {
- let m = src.to_owned();
- match self.tx.lock().unwrap().send(m) {
- Ok(_) => Ok(()),
- Err(_) => Err(TunError::Disconnected),
- }
- } else {
- Ok(())
- }
- }
-}
-
-impl tun::MTU for TunMTU {
- fn mtu(&self) -> usize {
- self.mtu.load(Ordering::Acquire)
- }
-}
-
-impl tun::Tun for TunTest {
- type Writer = TunWriter;
- type Reader = TunReader;
- type MTU = TunMTU;
- type Error = TunError;
-}
-
-impl TunFakeIO {
- pub fn write(&self, msg: Vec<u8>) {
- if self.store {
- self.tx.send(msg).unwrap();
- }
- }
-
- pub fn read(&self) -> Vec<u8> {
- self.rx.recv().unwrap()
- }
-}
-
-impl TunTest {
- pub fn create(mtu: usize, store: bool) -> (TunFakeIO, TunReader, TunWriter, TunMTU) {
- let (tx1, rx1) = if store {
- sync_channel(32)
- } else {
- sync_channel(1)
- };
- let (tx2, rx2) = if store {
- sync_channel(32)
- } else {
- sync_channel(1)
- };
-
- let fake = TunFakeIO {
- tx: tx1,
- rx: rx2,
- store,
- };
- let reader = TunReader { rx: rx1 };
- let writer = TunWriter {
- tx: Mutex::new(tx2),
- store,
- };
- let mtu = TunMTU {
- mtu: Arc::new(AtomicUsize::new(mtu)),
- };
-
- (fake, reader, writer, mtu)
- }
-}
-
-/* Void Bind */
-
-#[derive(Clone, Copy)]
-pub struct VoidBind {}
-
-impl bind::Reader<UnitEndpoint> for VoidBind {
- type Error = BindError;
-
- fn read(&self, _buf: &mut [u8]) -> Result<(usize, UnitEndpoint), Self::Error> {
- Ok((0, UnitEndpoint {}))
- }
-}
-
-impl bind::Writer<UnitEndpoint> for VoidBind {
- type Error = BindError;
-
- fn write(&self, _buf: &[u8], _dst: &UnitEndpoint) -> Result<(), Self::Error> {
- Ok(())
- }
-}
-
-impl bind::Bind for VoidBind {
- type Error = BindError;
- type Endpoint = UnitEndpoint;
-
- type Reader = VoidBind;
- type Writer = VoidBind;
-}
-
-impl VoidBind {
- pub fn new() -> VoidBind {
- VoidBind {}
- }
-}
-
-/* Pair Bind */
-
-#[derive(Clone)]
-pub struct PairReader<E> {
- recv: Arc<Mutex<Receiver<Vec<u8>>>>,
- _marker: marker::PhantomData<E>,
-}
-
-impl bind::Reader<UnitEndpoint> for PairReader<UnitEndpoint> {
- type Error = BindError;
- fn read(&self, buf: &mut [u8]) -> Result<(usize, UnitEndpoint), Self::Error> {
- let vec = self
- .recv
- .lock()
- .unwrap()
- .recv()
- .map_err(|_| BindError::Disconnected)?;
- let len = vec.len();
- buf[..len].copy_from_slice(&vec[..]);
- Ok((vec.len(), UnitEndpoint {}))
- }
-}
-
-impl bind::Writer<UnitEndpoint> for PairWriter<UnitEndpoint> {
- type Error = BindError;
- fn write(&self, buf: &[u8], _dst: &UnitEndpoint) -> Result<(), Self::Error> {
- let owned = buf.to_owned();
- match self.send.lock().unwrap().send(owned) {
- Err(_) => Err(BindError::Disconnected),
- Ok(_) => Ok(()),
- }
- }
-}
-
-#[derive(Clone)]
-pub struct PairWriter<E> {
- send: Arc<Mutex<SyncSender<Vec<u8>>>>,
- _marker: marker::PhantomData<E>,
-}
-
-#[derive(Clone)]
-pub struct PairBind {}
-
-impl PairBind {
- pub fn pair<E>() -> (
- (PairReader<E>, PairWriter<E>),
- (PairReader<E>, PairWriter<E>),
- ) {
- let (tx1, rx1) = sync_channel(128);
- let (tx2, rx2) = sync_channel(128);
- (
- (
- PairReader {
- recv: Arc::new(Mutex::new(rx1)),
- _marker: marker::PhantomData,
- },
- PairWriter {
- send: Arc::new(Mutex::new(tx2)),
- _marker: marker::PhantomData,
- },
- ),
- (
- PairReader {
- recv: Arc::new(Mutex::new(rx2)),
- _marker: marker::PhantomData,
- },
- PairWriter {
- send: Arc::new(Mutex::new(tx1)),
- _marker: marker::PhantomData,
- },
- ),
- )
- }
-}
-
-impl bind::Bind for PairBind {
- type Error = BindError;
- type Endpoint = UnitEndpoint;
- type Reader = PairReader<Self::Endpoint>;
- type Writer = PairWriter<Self::Endpoint>;
-}
-
-pub fn keypair(initiator: bool) -> KeyPair {
- let k1 = Key {
- key: [0x53u8; 32],
- id: 0x646e6573,
- };
- let k2 = Key {
- key: [0x52u8; 32],
- id: 0x76636572,
- };
- if initiator {
- KeyPair {
- birth: Instant::now(),
- initiator: true,
- send: k1,
- recv: k2,
- }
- } else {
- KeyPair {
- birth: Instant::now(),
- initiator: false,
- send: k2,
- recv: k1,
- }
- }
-}