Enable privileged only on linux

This commit is contained in:
Prabhpreet Dua
2024-06-13 17:34:48 +05:30
parent a75d7a53e0
commit 96bed38ad0
2 changed files with 54 additions and 48 deletions

View File

@@ -1,13 +1,19 @@
#![cfg(target_os = "linux")]
use std::io::{stdin, stdout, Read, Write};
use std::result::Result;
fn main() {
#[cfg(target_os = "linux")]
linux::main().unwrap();
}
use rosenpass_wireguard_broker::api::msgs;
use rosenpass_wireguard_broker::api::server::BrokerServer;
use rosenpass_wireguard_broker::brokers::netlink as wg;
#[cfg(target_os = "linux")]
pub mod linux {
use std::io::{stdin, stdout, Read, Write};
use std::result::Result;
#[derive(thiserror::Error, Debug)]
pub enum BrokerAppError {
use rosenpass_wireguard_broker::api::msgs;
use rosenpass_wireguard_broker::api::server::BrokerServer;
use rosenpass_wireguard_broker::brokers::netlink as wg;
#[derive(thiserror::Error, Debug)]
pub enum BrokerAppError {
#[error(transparent)]
IoError(#[from] std::io::Error),
#[error(transparent)]
@@ -16,9 +22,9 @@ pub enum BrokerAppError {
WgSetPskError(#[from] wg::SetPskError),
#[error("Oversized message {}; something about the request is fatally wrong", .0)]
OversizedMessage(u64),
}
}
fn main() -> Result<(), BrokerAppError> {
pub fn main() -> Result<(), BrokerAppError> {
let mut broker = BrokerServer::new(wg::NetlinkWireGuardBroker::new()?);
let mut stdin = stdin().lock();
@@ -54,4 +60,5 @@ fn main() -> Result<(), BrokerAppError> {
stdout.write_all(&res)?;
stdout.flush()?;
}
}
}

View File

@@ -1,4 +1,3 @@
#![cfg(target_os = "linux")]
use std::process::Stdio;
use tokio::io::{AsyncReadExt, AsyncWriteExt};