rustem_proxy 0.1.5

A rust library to set system proxy on different operating systems
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use crate::{Protocol, SystemProxy};
use std::{net::SocketAddr, str::FromStr};
use winapi::shared::ntdef::NULL;
use winapi::um::wininet::{
    InternetSetOptionA, INTERNET_OPTION_REFRESH, INTERNET_OPTION_SETTINGS_CHANGED,
};
use winreg::{enums, RegKey};

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

impl SystemProxy {
    pub fn get() -> SystemProxy {
        let current_user = RegKey::predef(enums::HKEY_CURRENT_USER);
        let key = current_user
            .open_subkey_with_flags(SUB_KEY, enums::KEY_READ)
            .map_err(|e| eprintln!("ERROR: Could not open subkey with flags: {}", e.to_string()))
            .unwrap();
        let is_enabled = key
            .get_value::<u32, _>("ProxyEnable")
            .expect("ERROR: Could not get key value")
            == 1u32;
        let server = key
            .get_value::<String, _>("ProxyServer")
            .expect("ERROR: Could not get key value");
        let server = server.as_str();
        let socket = SocketAddr::from_str(server)
            .map_err(|e| {
                eprintln!("ERROR: Could not parse from str: {}", e.to_string());
            })
            .unwrap();
        let host = socket.ip().to_string();
        let port = socket.port();
        let bypass: String = key.get_value("ProxyOverride").unwrap_or("".into());

        SystemProxy {
            is_enabled,
            host: host,
            port: port,
            bypass: bypass,
            protocol: Protocol::ALL,
        }
    }

    pub fn set(proxy: SystemProxy) {
        let hkcu = RegKey::predef(enums::HKEY_CURRENT_USER);
        let cur_var = hkcu
            .open_subkey_with_flags(SUB_KEY, enums::KEY_SET_VALUE)
            .expect("ERROR: Could not open subkey with flags");

        let enable = if proxy.is_enabled { 1u32 } else { 0u32 };
        let server = format!("{}:{}", proxy.host, proxy.port);
        let bypass = proxy.bypass.as_str();

        cur_var
            .set_value("ProxyEnable", &enable)
            .expect("ERROR: Could not set value ProxyEnable");
        cur_var
            .set_value("ProxyServer", &server)
            .expect("ERROR: Could not set value ProxyServer");
        cur_var
            .set_value("ProxyOverride", &bypass)
            .expect("ERROR: Could not set value ProxyOverride");

        unsafe {
            InternetSetOptionA(NULL, INTERNET_OPTION_SETTINGS_CHANGED, NULL, 0);
            InternetSetOptionA(NULL, INTERNET_OPTION_REFRESH, NULL, 0);
        }
    }

    pub fn unset() {
        let hkcu = RegKey::predef(enums::HKEY_CURRENT_USER);
        let cur_var = hkcu
            .open_subkey_with_flags(SUB_KEY, enums::KEY_SET_VALUE)
            .expect("ERROR: Could not open subkey with flags");

        let enable = 0u32;

        cur_var
            .set_value("ProxyEnable", &enable)
            .expect("ERROR: Could not set value ProxyEnable");

        unsafe {
            InternetSetOptionA(NULL, INTERNET_OPTION_SETTINGS_CHANGED, NULL, 0);
            InternetSetOptionA(NULL, INTERNET_OPTION_REFRESH, NULL, 0);
        }
    }
}