First commit: config file parsing

5 files changed, 198 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..ea8c4bf
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+/target
diff --git a/Cargo.lock b/Cargo.lock
new file mode 100644
index 0000000..7ee9b3d
--- /dev/null
+++ b/Cargo.lock
@@ -0,0 +1,7 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "wgmgr"
+version = "0.1.0"
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..433e4a6
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,8 @@
+[package]
+name = "wgmgr"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..c9bce25
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,179 @@
+use std::env::join_paths;
+use std::fs::read_to_string;
+use std::{net::Ipv4Addr, fs, error::Error};
+use std::fmt::{Formatter, Display};
+
+#[derive(Debug)]
+struct WgMgrError {
+    msg: String
+}
+
+impl WgMgrError {
+    fn new(msg: String) -> WgMgrError {
+        WgMgrError { 
+            msg
+        }
+    }
+}
+
+impl Error for WgMgrError {
+    fn description(&self) -> &str {
+        return self.msg.as_str()
+    }
+
+    fn cause(&self) -> Option<&dyn Error> {
+        None
+    }
+}
+
+impl Display for WgMgrError {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+            write!(f, "{}", self.msg)
+    }
+}
+
+#[derive(Debug)]
+pub struct WireguardConfig {
+    private_key: String,
+    listen_port: u32,
+    address: Ipv4Addr,
+    net_range: u32,
+    peers: Vec<Peer>
+}
+
+#[derive(Debug)]
+struct Peer {
+    name: String,
+    public_key: String,
+    ip: Ipv4Addr
+}
+
+impl WireguardConfig {
+    pub fn new(config_path: &str) -> Result<WireguardConfig, Box<dyn Error>> {
+        let f = fs::read_to_string(config_path)?;
+
+        let mut conf = WireguardConfig{
+            private_key: String::new(),
+            listen_port: 0,
+            address: Ipv4Addr::UNSPECIFIED,
+            net_range: 0,
+            peers: vec![]
+        };
+        let mut current_peer = Peer {
+            ip: Ipv4Addr::UNSPECIFIED,
+            name: String::from("N/A"),
+            public_key: String::new()
+        };
+
+        let mut in_interface = false;
+        let mut in_peer = false;
+
+        let mut prev_line: &str = "N/A";
+
+        for line in f.lines() {
+            match line.trim() {
+                "[Interface]" => {
+                    in_interface = true;
+                    in_peer = false;
+                    continue;
+                }
+                "[Peer]" => {
+                    in_interface = false;
+                    in_peer = true;
+
+                    if current_peer.ip != Ipv4Addr::UNSPECIFIED {
+                        conf.peers.push(current_peer);
+                    }
+
+                    current_peer = Peer {
+                        ip: Ipv4Addr::UNSPECIFIED,
+                        name: String::from(prev_line.trim_start_matches(&['#', ' '])),
+                        public_key: String::new()
+                    };
+                    continue;
+                }
+                "" => {
+                    continue;
+                }
+                _ => {}
+            }
+
+            if in_interface {
+                if line.starts_with("PrivateKey") {
+                    conf.private_key = String::from(config_value(line)?);
+                }
+                else if line.starts_with("ListenPort") {
+                    conf.listen_port = config_value(line)?.parse()?;
+                }
+                else if line.starts_with("Address") {
+                    let addr = config_value(line)?;
+                    match addr.find("/") {
+                        Some(idx) => {
+                            conf.address = addr[..idx].parse()?;
+                            conf.net_range = addr[idx+1..].parse()?;
+                        }
+                        None => {
+                            return Err(Box::new(WgMgrError::new(String::from("'Interface' > 'Address' is missing a subnet"))));
+                        }
+                    }
+                }
+            }
+
+            else if in_peer {
+                if line.starts_with("AllowedIPs") {
+                    let addr = config_value(line)?;
+                    match addr.find("/") {
+                        Some(idx) => {
+                            current_peer.ip = addr[..idx].parse()?;
+                            let net_range: i32 = addr[idx+1..].parse()?;
+
+                            if net_range != 32 {
+                                let msg = format!("Peer '{}' has invalid net range {}", current_peer.name, net_range);
+                                return Err(Box::new(WgMgrError::new(msg)));
+                            }
+                        }
+                        None => {
+                            let msg = format!("Peer '{}' is missing net range", current_peer.name);
+                            return Err(Box::new(WgMgrError::new(msg)));
+                        }
+                    }
+                }
+                else if line.starts_with("PublicKey") {
+                    current_peer.public_key = String::from(config_value(line)?);
+                }
+            }
+
+            prev_line = line;
+        }
+
+        if current_peer.ip != Ipv4Addr::UNSPECIFIED {
+            conf.peers.push(current_peer);
+        }
+
+        Ok(conf)
+    }
+}
+
+impl Peer {
+    pub fn private_key(&self) -> Result<String, Box<dyn Error>> {
+        // TODO: do not hardcode where private keys are stored
+
+        let pk_folder = String::from("./private_keys/");
+        let pk_path = join_paths(&[pk_folder, self.name.clone()])?;
+
+        let pk = read_to_string(pk_path)?;
+
+        Ok(pk)
+    }
+}
+
+fn config_value(line: &str) -> Result<&str, &str> {
+    match line.find("=") {
+        Some(i) => {
+            Ok(line[i+1..].trim())
+        },
+        None => {
+            Err("line does not seem to contain a key-value pair")
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..028e0e2
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,3 @@
+fn main() {
+    println!("{:?}", wgmgr::WireguardConfig::new("wg.conf").unwrap())
+}