Skip to content

Commit

Permalink
fix: try fix windows issue
Browse files Browse the repository at this point in the history
  • Loading branch information
@zzzgydi
zzzgydi committed Jan 19, 2024
1 parent 215e018 commit 95a2ba9
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 142 deletions.
2 changes: 1 addition & 1 deletion src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ pub enum Error {

#[cfg(target_os = "windows")]
#[error("system call failed")]
SystemCallFailed(#[from] windows::SystemCallFailed),
SystemCall(#[from] windows::Win32Error),
}

pub type Result<T> = std::result::Result<T, Error>;
Expand Down
167 changes: 26 additions & 141 deletions src/windows.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,30 +2,17 @@ use crate::{Error, Result, Sysproxy};
use std::ffi::c_void;
use std::{mem::size_of, mem::ManuallyDrop, net::SocketAddr, str::FromStr};
use windows::core::PWSTR;
use windows::Win32::NetworkManagement::Rras::{
RasEnumEntriesW, ERROR_BUFFER_TOO_SMALL, RASENTRYNAMEW,
};

use windows::Win32::Networking::WinInet::{
InternetSetOptionW, INTERNET_OPTION_PER_CONNECTION_OPTION,
INTERNET_OPTION_PROXY_SETTINGS_CHANGED, INTERNET_OPTION_REFRESH,
INTERNET_PER_CONN_AUTOCONFIG_URL, INTERNET_PER_CONN_FLAGS, INTERNET_PER_CONN_OPTIONW,
INTERNET_PER_CONN_OPTIONW_0, INTERNET_PER_CONN_OPTION_LISTW, INTERNET_PER_CONN_PROXY_BYPASS,
INTERNET_PER_CONN_PROXY_SERVER, PROXY_TYPE_AUTO_DETECT, PROXY_TYPE_AUTO_PROXY_URL,
PROXY_TYPE_DIRECT, PROXY_TYPE_PROXY,
INTERNET_OPTION_PROXY_SETTINGS_CHANGED, INTERNET_OPTION_REFRESH, INTERNET_PER_CONN_FLAGS,
INTERNET_PER_CONN_OPTIONW, INTERNET_PER_CONN_OPTIONW_0, INTERNET_PER_CONN_OPTION_LISTW,
INTERNET_PER_CONN_PROXY_BYPASS, INTERNET_PER_CONN_PROXY_SERVER, PROXY_TYPE_DIRECT,
PROXY_TYPE_PROXY,
};
use winreg::{enums, RegKey};

pub use windows::core::Error as Win32Error;

#[derive(thiserror::Error, Debug)]
pub enum SystemCallFailed {
#[error("operation failed: {0}")]
Raw(String),
#[error("operation failed")]
Win32Error(#[from] Win32Error),
}

const SUB_KEY: &str = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings";

/// unset proxy
Expand All @@ -35,7 +22,7 @@ fn unset_proxy() -> Result<()> {
dwOption: INTERNET_PER_CONN_FLAGS,
Value: {
let mut v = INTERNET_PER_CONN_OPTIONW_0::default();
v.dwValue = PROXY_TYPE_AUTO_DETECT | PROXY_TYPE_DIRECT;
v.dwValue = PROXY_TYPE_DIRECT;
v
},
});
Expand All @@ -53,43 +40,6 @@ fn unset_proxy() -> Result<()> {
res
}

#[allow(dead_code)]
/// set auto detect proxy, aka PAC
fn set_auto_proxy(url: &str) -> Result<()> {
let mut p_opts = ManuallyDrop::new(Vec::<INTERNET_PER_CONN_OPTIONW>::with_capacity(2));
p_opts.push(INTERNET_PER_CONN_OPTIONW {
dwOption: INTERNET_PER_CONN_FLAGS,
Value: {
let mut v = INTERNET_PER_CONN_OPTIONW_0::default();
v.dwValue = PROXY_TYPE_AUTO_PROXY_URL | PROXY_TYPE_DIRECT;
v
},
});
let mut url = ManuallyDrop::new(url.encode_utf16().chain([0u16]).collect::<Vec<u16>>());
p_opts.push(INTERNET_PER_CONN_OPTIONW {
dwOption: INTERNET_PER_CONN_AUTOCONFIG_URL,
Value: {
let mut v = INTERNET_PER_CONN_OPTIONW_0::default();
v.pszValue = PWSTR::from_raw(url.as_ptr() as *mut u16);
v
},
});
let opts = INTERNET_PER_CONN_OPTION_LISTW {
dwSize: size_of::<INTERNET_PER_CONN_OPTION_LISTW>() as u32,
dwOptionCount: 2,
dwOptionError: 0,
pOptions: p_opts.as_mut_ptr() as *mut INTERNET_PER_CONN_OPTIONW,
pszConnection: PWSTR::null(),
};

let res = apply(&opts);
unsafe {
ManuallyDrop::drop(&mut url);
ManuallyDrop::drop(&mut p_opts);
}
res
}

/// set global proxy
fn set_global_proxy(server: String, bypass: String) -> Result<()> {
let mut p_opts = ManuallyDrop::new(Vec::<INTERNET_PER_CONN_OPTIONW>::with_capacity(3));
Expand All @@ -108,15 +58,13 @@ fn set_global_proxy(server: String, bypass: String) -> Result<()> {
},
});

let mut b = ManuallyDrop::new(if bypass.is_empty() {
"<local>".encode_utf16().chain([0u16]).collect::<Vec<u16>>()
} else {
let mut b = ManuallyDrop::new(
bypass
.clone()
.encode_utf16()
.chain([0u16])
.collect::<Vec<u16>>()
});
.collect::<Vec<u16>>(),
);
p_opts.push(INTERNET_PER_CONN_OPTIONW {
dwOption: INTERNET_PER_CONN_PROXY_BYPASS,
Value: INTERNET_PER_CONN_OPTIONW_0 {
Expand All @@ -142,63 +90,9 @@ fn set_global_proxy(server: String, bypass: String) -> Result<()> {
}

fn apply(options: &INTERNET_PER_CONN_OPTION_LISTW) -> Result<()> {
let mut dw_cb = 0;
let mut dw_entries = 0;
let mut ret;
unsafe {
ret = RasEnumEntriesW(None, None, None, &mut dw_cb, &mut dw_entries);
}
if ret == ERROR_BUFFER_TOO_SMALL {
let mut entries = Vec::<RASENTRYNAMEW>::with_capacity(dw_cb as usize);
for _ in 0..dw_cb {
entries.push(RASENTRYNAMEW {
dwSize: size_of::<RASENTRYNAMEW>() as u32,
..Default::default()
});
}
unsafe {
ret = RasEnumEntriesW(
None,
None,
Some(entries.as_mut_ptr()),
&mut dw_cb,
&mut dw_entries,
);
}
match ret {
0 => {
apply_connect(options, PWSTR::null())?;
for entry in entries.iter() {
apply_connect(
options,
PWSTR::from_raw(entry.szEntryName.as_ptr() as *mut u16),
)?;
}
return Ok(());
}
_ => return Err(SystemCallFailed::Raw(format!("RasEnumEntriesW: {}", ret)).into()),
}
}
if dw_entries > 1 {
return Err(SystemCallFailed::Raw("acquire buffer size".into()).into());
}

// No ras entry, set default only.
match apply_connect(options, PWSTR::null()) {
Ok(_) => Ok(()),
Err(e) => Err(e.into()),
}
}

fn apply_connect(
options: &INTERNET_PER_CONN_OPTION_LISTW,
conn: PWSTR,
) -> std::result::Result<(), SystemCallFailed> {
let opts = &mut options.clone();
opts.pszConnection = conn;
unsafe {
// setting options
let opts = opts as *const INTERNET_PER_CONN_OPTION_LISTW as *const c_void;
let opts = options as *const INTERNET_PER_CONN_OPTION_LISTW as *const c_void;
InternetSetOptionW(
None,
INTERNET_OPTION_PER_CONNECTION_OPTION,
Expand All @@ -213,38 +107,29 @@ fn apply_connect(
Ok(())
}

/// get_system_proxy_with_registry is intended to get system proxy from registry.
fn get_system_proxy_with_registry() -> Result<Sysproxy> {
let hkcu = RegKey::predef(enums::HKEY_CURRENT_USER);
let cur_var = hkcu.open_subkey_with_flags(SUB_KEY, enums::KEY_READ)?;
impl Sysproxy {
pub fn get_system_proxy() -> Result<Sysproxy> {
let hkcu = RegKey::predef(enums::HKEY_CURRENT_USER);
let cur_var = hkcu.open_subkey_with_flags(SUB_KEY, enums::KEY_READ)?;

let enable = cur_var.get_value::<u32, _>("ProxyEnable").unwrap_or(0u32) == 1u32;
let server = cur_var
.get_value::<String, _>("ProxyServer")
.unwrap_or("".into());
let server = server.as_str();

let enable = cur_var.get_value::<u32, _>("ProxyEnable").unwrap_or(0u32) == 1u32;
let server = cur_var
.get_value::<String, _>("ProxyServer")
.unwrap_or("".into());
let server = server.as_str();
let (host, port) = if server.is_empty() {
("".into(), 0)
} else {
let socket = SocketAddr::from_str(server).or(Err(Error::ParseStr(server.to_string())))?;
let host = socket.ip().to_string();
let port = socket.port();
(host, port)
};

let bypass = cur_var.get_value("ProxyOverride").unwrap_or("".into());

Ok(Sysproxy {
enable,
host,
port,
bypass,
})
}
let bypass = cur_var.get_value("ProxyOverride").unwrap_or("".into());

impl Sysproxy {
pub fn get_system_proxy() -> Result<Sysproxy> {
get_system_proxy_with_registry()
Ok(Sysproxy {
enable,
host,
port,
bypass,
})
}

pub fn set_system_proxy(&self) -> Result<()> {
Expand Down

0 comments on commit 95a2ba9

Please sign in to comment.