111 lines
4.3 KiB
C#
Executable File
111 lines
4.3 KiB
C#
Executable File
using System.Windows.Forms;
|
|
using Microsoft.Win32;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
using System.IO;
|
|
using Shadowsocks.Model;
|
|
|
|
namespace Shadowsocks.Controller
|
|
{
|
|
public class SystemProxy
|
|
{
|
|
|
|
[DllImport("wininet.dll")]
|
|
public static extern bool InternetSetOption(IntPtr hInternet, int dwOption, IntPtr lpBuffer, int dwBufferLength);
|
|
public const int INTERNET_OPTION_SETTINGS_CHANGED = 39;
|
|
public const int INTERNET_OPTION_REFRESH = 37;
|
|
static bool _settingsReturn, _refreshReturn;
|
|
|
|
public static void NotifyIE()
|
|
{
|
|
// These lines implement the Interface in the beginning of program
|
|
// They cause the OS to refresh the settings, causing IP to realy update
|
|
_settingsReturn = InternetSetOption(IntPtr.Zero, INTERNET_OPTION_SETTINGS_CHANGED, IntPtr.Zero, 0);
|
|
_refreshReturn = InternetSetOption(IntPtr.Zero, INTERNET_OPTION_REFRESH, IntPtr.Zero, 0);
|
|
}
|
|
|
|
public static void Update(Configuration config, bool forceDisable)
|
|
{
|
|
bool global = config.global;
|
|
bool enabled = config.enabled;
|
|
if (forceDisable)
|
|
{
|
|
enabled = false;
|
|
}
|
|
try
|
|
{
|
|
RegistryKey registry =
|
|
Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
|
|
true);
|
|
if (enabled)
|
|
{
|
|
if (global)
|
|
{
|
|
registry.SetValue("ProxyEnable", 1);
|
|
registry.SetValue("ProxyServer", "127.0.0.1:" + config.localPort.ToString());
|
|
registry.SetValue("AutoConfigURL", "");
|
|
}
|
|
else
|
|
{
|
|
string pacUrl;
|
|
if (config.useOnlinePac && !string.IsNullOrEmpty(config.pacUrl))
|
|
pacUrl = config.pacUrl;
|
|
else
|
|
pacUrl = "http://127.0.0.1:" + config.localPort.ToString() + "/pac?t=" + GetTimestamp(DateTime.Now);
|
|
registry.SetValue("ProxyEnable", 0);
|
|
registry.SetValue("ProxyServer", "");
|
|
registry.SetValue("AutoConfigURL", pacUrl);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
registry.SetValue("ProxyEnable", 0);
|
|
registry.SetValue("ProxyServer", "");
|
|
registry.SetValue("AutoConfigURL", "");
|
|
}
|
|
SystemProxy.NotifyIE();
|
|
//Must Notify IE first, or the connections do not chanage
|
|
CopyProxySettingFromLan();
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logging.LogUsefulException(e);
|
|
// TODO this should be moved into views
|
|
MessageBox.Show(I18N.GetString("Failed to update registry"));
|
|
}
|
|
}
|
|
|
|
private static void CopyProxySettingFromLan()
|
|
{
|
|
RegistryKey registry =
|
|
Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\Connections",
|
|
true);
|
|
var defaultValue = registry.GetValue("DefaultConnectionSettings");
|
|
try
|
|
{
|
|
var connections = registry.GetValueNames();
|
|
foreach (String each in connections)
|
|
{
|
|
if (!(each.Equals("DefaultConnectionSettings")
|
|
|| each.Equals("LAN Connection")
|
|
|| each.Equals("SavedLegacySettings")))
|
|
{
|
|
//set all the connections's proxy as the lan
|
|
registry.SetValue(each, defaultValue);
|
|
}
|
|
}
|
|
SystemProxy.NotifyIE();
|
|
} catch (IOException e) {
|
|
Logging.LogUsefulException(e);
|
|
}
|
|
}
|
|
|
|
private static String GetTimestamp(DateTime value)
|
|
{
|
|
return value.ToString("yyyyMMddHHmmssffff");
|
|
}
|
|
}
|
|
}
|