Skip to content

Commit

Permalink
fix: check executable for file
Browse files Browse the repository at this point in the history
  • Loading branch information
yuezk committed May 10, 2024
1 parent 1596736 commit 92b8588
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 19 deletions.
19 changes: 16 additions & 3 deletions crates/common/src/vpn_utils.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
use is_executable::IsExecutable;
use std::path::Path;
use std::{io, path::Path};

pub use is_executable::is_executable;
use is_executable::IsExecutable;

const VPNC_SCRIPT_LOCATIONS: [&str; 6] = [
"/usr/local/share/vpnc-scripts/vpnc-script",
Expand Down Expand Up @@ -39,3 +38,17 @@ pub fn find_vpnc_script() -> Option<String> {
pub fn find_csd_wrapper() -> Option<String> {
find_executable(&CSD_WRAPPER_LOCATIONS)
}

/// If file exists, check if it is executable
pub fn check_executable(file: &str) -> Result<(), io::Error> {
let path = Path::new(file);

if path.exists() && !path.is_executable() {
return Err(io::Error::new(
io::ErrorKind::PermissionDenied,
format!("{} is not executable", file),
));
}

Ok(())
}
28 changes: 12 additions & 16 deletions crates/openconnect/src/vpn.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ use std::{
sync::{Arc, RwLock},
};

use common::vpn_utils::{find_vpnc_script, is_executable};
use common::vpn_utils::{check_executable, find_vpnc_script};
use log::info;

use crate::ffi;
Expand Down Expand Up @@ -80,23 +80,23 @@ impl Vpn {
}

#[derive(Debug)]
pub struct VpnError<'a> {
message: &'a str,
pub struct VpnError {
message: String,
}

impl<'a> VpnError<'a> {
fn new(message: &'a str) -> Self {
impl VpnError {
fn new(message: String) -> Self {
Self { message }
}
}

impl fmt::Display for VpnError<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
impl fmt::Display for VpnError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}", self.message)
}
}

impl std::error::Error for VpnError<'_> {}
impl std::error::Error for VpnError {}

pub struct VpnBuilder {
server: String,
Expand Down Expand Up @@ -159,21 +159,17 @@ impl VpnBuilder {
self
}

pub fn build(self) -> Result<Vpn, VpnError<'static>> {
pub fn build(self) -> Result<Vpn, VpnError> {
let script = match self.script {
Some(script) => {
if !is_executable(&script) {
return Err(VpnError::new("vpnc script is not executable"));
}
check_executable(&script).map_err(|e| VpnError::new(e.to_string()))?;
script
}
None => find_vpnc_script().ok_or_else(|| VpnError::new("Failed to find vpnc-script"))?,
None => find_vpnc_script().ok_or_else(|| VpnError::new(String::from("Failed to find vpnc-script")))?,
};

if let Some(csd_wrapper) = &self.csd_wrapper {
if !is_executable(csd_wrapper) {
return Err(VpnError::new("CSD wrapper is not executable"));
}
check_executable(csd_wrapper).map_err(|e| VpnError::new(e.to_string()))?;
}

let user_agent = self.user_agent.unwrap_or_default();
Expand Down

0 comments on commit 92b8588

Please sign in to comment.