Added dummy support

1 files changed, 18 insertions, 3 deletions

diff --git a/src/lib.rs b/src/lib.rs
index af104d8..37a9075 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -37,11 +37,21 @@ pub struct WireGuard {
 impl WireGuard {
     /// Creates a new `WireGuard` instance
     pub fn new(addr: &str) -> WgResult<Self> {
+        WireGuard::create(addr, false)
+    }
+
+    /// Create a new dummy `WireGuard` instance listening on "127.0.0.1:8080"
+    pub fn dummy() -> WgResult<Self> {
+        WireGuard::create("127.0.0.1:8080", true)
+    }
+
+    /// Internal `WireGuard` creation method
+    fn create(addr: &str, dummy: bool) -> WgResult<Self> {
         // Create a new tokio core
         let core = Core::new()?;
 
         /// Create a tunnel instance
-        let tunnel = WireGuardFuture::new(&core.handle(), addr)?;
+        let tunnel = WireGuardFuture::new(&core.handle(), addr, dummy)?;
 
         /// Return the `WireGuard` instance
         Ok(WireGuard {
@@ -83,9 +93,14 @@ pub struct WireGuardFuture {
 
 impl WireGuardFuture {
     /// Creates a new `WireGuardFuture`
-    pub fn new(handle: &Handle, addr: &str) -> WgResult<Self> {
+    pub fn new(handle: &Handle, addr: &str, dummy: bool) -> WgResult<Self> {
         // Create a tunneling device
-        let device = Device::new("wg")?;
+        let device_name = "wg";
+        let device = if dummy {
+            Device::dummy(device_name)?
+        } else {
+            Device::new(device_name)?
+        };
 
         // Create a server for the tunnel
         let sock_addr = SocketAddr::from_str(addr)?;