simono41/prometheus_wireguard_exporter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Started wireguard config file parse

Browse source
This commit is contained in:
Francesco Cogno 2019-05-15 16:55:07 +02:00
parent 0c13d4bff7
commit 1b4d57871e
2 changed files with 86 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/main.rs
View file

@ -20,6 +20,7 @@ use std::convert::TryFrom;
use std::process::Command; use std::process::Command;
use std::string::String; use std::string::String;
use wireguard::WireGuard; use wireguard::WireGuard;
mod wireguard_config;
fn check_compliance(req: &Request<Body>) -> Result<(), Response<Body>> { fn check_compliance(req: &Request<Body>) -> Result<(), Response<Body>> {
if req.uri() != "/metrics" { if req.uri() != "/metrics" {

85
src/wireguard_config.rs Normal file
View file

@ -0,0 +1,85 @@
use std::collections::HashMap;
#[derive(Debug, Default, Clone)]
pub(crate) struct PeerEntry {
pub public_key: String,
pub name: Option<String>,
}
pub(crate) fn parse<'a>(txt: &'a str) -> HashMap<String, PeerEntry> {
let mut ht = HashMap::new();
let mut name = "";
txt.lines().fold("", |prev, cur| {
if cur == "[Peer]" {
if prev.chars().next() == Some('#') {
name = prev;
}
} else if cur.starts_with("PublicKey") {
// public key found, use it as key
// TODO we must strip the PublicKey = first !
ht.insert(
cur.to_owned(),
PeerEntry {
public_key: cur.to_owned(),
name: Some(name.to_owned()),
},
);
}
cur
});
ht
}
#[cfg(test)]
mod tests {
use super::*;
const TEXT: &'static str = "
ListenPort = 51820
PrivateKey = my_super_secret_private_key
# PreUp = iptables -t nat -A POSTROUTING -s 10.70.0.0/24 -o enp7s0 -j MASQUERADE
# PostDown = iptables -t nat -D POSTROUTING -s 10.70.0.0/24 -o enp7s0 -j MASQUERADE
# OnePlus 6T
[Peer]
PublicKey = 2S7mA0vEMethCNQrJpJKE81/JmhgtB+tHHLYQhgM6kk=
AllowedIPs = 10.70.0.2/32
# varch.local (laptop)
[Peer]
PublicKey = qnoxQoQI8KKMupLnSSureORV0wMmH7JryZNsmGVISzU=
AllowedIPs = 10.70.0.3/32
# cantarch
[Peer]
PublicKey = L2UoJZN7RmEKsMmqaJgKG0m1S2Zs2wd2ptAf+kb3008=
AllowedIPs = 10.70.0.4/32
# frcognoarch
[Peer]
PublicKey = MdVOIPKt9K2MPj/sO2NlWQbOnFJ6L/qX80mmhQwsUlA=
AllowedIPs = 10.70.0.50/32
# frcognowin10
[Peer]
PublicKey = lqYcojJMsIZXMUw1heAFbQHBoKjCEaeo7M1WXDh/KWc=
AllowedIPs = 10.70.0.40/32
# OnePlus 5T
[Peer]
PublicKey = 928vO9Lf4+Mo84cWu4k1oRyzf0AR7FTGoPKHGoTMSHk=
AllowedIPs = 10.70.0.80/32
";
#[test]
fn test_parse() {
let a = parse(TEXT);
println!("{:?}", a);
}
#[test]
fn test_parse_and_serialize() {}
}