142 lines
4.3 KiB
Rust
142 lines
4.3 KiB
Rust
extern crate serde_json;
|
|
#[macro_use]
|
|
extern crate failure;
|
|
use clap;
|
|
use clap::{crate_authors, crate_name, crate_version, Arg};
|
|
use futures::future::{done, ok, Either, Future};
|
|
use hyper::{Body, Request, Response};
|
|
use log::{info, trace};
|
|
use std::env;
|
|
mod options;
|
|
use options::Options;
|
|
mod wireguard;
|
|
use std::convert::TryFrom;
|
|
use std::process::Command;
|
|
use std::string::String;
|
|
use wireguard::WireGuard;
|
|
mod exporter_error;
|
|
mod wireguard_config;
|
|
use wireguard_config::peer_entry_hashmap_try_from;
|
|
extern crate prometheus_exporter_base;
|
|
use crate::exporter_error::ExporterError;
|
|
use prometheus_exporter_base::render_prometheus;
|
|
use std::sync::Arc;
|
|
|
|
fn wg_with_text(
|
|
wg_config_str: &str,
|
|
wg_output: ::std::process::Output,
|
|
options: Arc<Options>,
|
|
) -> Result<Response<Body>, ExporterError> {
|
|
let pehm = peer_entry_hashmap_try_from(wg_config_str)?;
|
|
trace!("pehm == {:?}", pehm);
|
|
|
|
let wg_output_string = String::from_utf8(wg_output.stdout)?;
|
|
let wg = WireGuard::try_from(&wg_output_string as &str)?;
|
|
Ok(Response::new(Body::from(wg.render_with_names(
|
|
Some(&pehm),
|
|
options.separate_allowed_ips,
|
|
))))
|
|
}
|
|
|
|
fn perform_request(
|
|
_req: Request<Body>,
|
|
options: &Arc<Options>,
|
|
) -> impl Future<Item = Response<Body>, Error = failure::Error> {
|
|
trace!("perform_request");
|
|
|
|
// this is needed to satisfy the borrow checker
|
|
let options = options.clone();
|
|
|
|
done(
|
|
Command::new("wg")
|
|
.arg("show")
|
|
.arg("all")
|
|
.arg("dump")
|
|
.output(),
|
|
)
|
|
.from_err()
|
|
.and_then(move |output| {
|
|
if let Some(extract_names_config_file) = &options.extract_names_config_file {
|
|
Either::A(
|
|
done(::std::fs::read_to_string(extract_names_config_file))
|
|
.from_err()
|
|
.and_then(|wg_config_string| {
|
|
wg_with_text(&wg_config_string as &str, output, options)
|
|
}),
|
|
)
|
|
} else {
|
|
Either::B(done(String::from_utf8(output.stdout)).from_err().and_then(
|
|
move |output_str| {
|
|
trace!("{}", output_str);
|
|
done(WireGuard::try_from(&output_str as &str))
|
|
.from_err()
|
|
.and_then(move |wg| {
|
|
ok(Response::new(Body::from(
|
|
wg.render_with_names(None, options.separate_allowed_ips),
|
|
)))
|
|
})
|
|
},
|
|
))
|
|
}
|
|
})
|
|
.from_err()
|
|
}
|
|
|
|
fn main() {
|
|
let matches = clap::App::new(crate_name!())
|
|
.version(crate_version!())
|
|
.author(crate_authors!("\n"))
|
|
.arg(
|
|
Arg::with_name("port")
|
|
.short("p")
|
|
.help("exporter port")
|
|
.default_value("9586")
|
|
.takes_value(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("verbose")
|
|
.short("v")
|
|
.help("verbose logging")
|
|
.takes_value(false),
|
|
)
|
|
.arg(
|
|
Arg::with_name("separate_allowed_ips")
|
|
.short("s")
|
|
.help("separate allowed ips and ports")
|
|
.takes_value(false),
|
|
)
|
|
.arg(
|
|
Arg::with_name("extract_names_config_file")
|
|
.short("n")
|
|
.help("If set, the exporter will look in the specified WireGuard config file for peer names (must be in [Peer] definition and be a comment)")
|
|
.takes_value(true))
|
|
.get_matches();
|
|
|
|
let options = Options::from_claps(&matches);
|
|
|
|
if options.verbose {
|
|
env::set_var(
|
|
"RUST_LOG",
|
|
format!("{}=trace,prometheus_exporter_base=trace", crate_name!()),
|
|
);
|
|
} else {
|
|
env::set_var(
|
|
"RUST_LOG",
|
|
format!("{}=info,prometheus_exporter_base=info", crate_name!()),
|
|
);
|
|
}
|
|
env_logger::init();
|
|
|
|
info!("using options: {:?}", options);
|
|
|
|
let bind = matches.value_of("port").unwrap();
|
|
let bind = u16::from_str_radix(&bind, 10).expect("port must be a valid number");
|
|
let addr = ([0, 0, 0, 0], bind).into();
|
|
|
|
info!("starting exporter on {}", addr);
|
|
|
|
render_prometheus(&addr, options, |request, options| {
|
|
Box::new(perform_request(request, options))
|
|
});
|
|
}
|