Browse Source

- better FileStream handling

- restore WPAD
tags/3.0
Licshee 8 years ago
parent
commit
0a71dc6f00
2 changed files with 47 additions and 38 deletions
  1. +8
    -10
      shadowsocks-csharp/Controller/FileManager.cs
  2. +39
    -28
      shadowsocks-csharp/Controller/System/SystemProxy.cs

+ 8
- 10
shadowsocks-csharp/Controller/FileManager.cs View File

@@ -10,37 +10,35 @@ namespace Shadowsocks.Controller
{ {
try try
{ {
FileStream _FileStream = new FileStream(fileName, FileMode.Create, FileAccess.Write);
_FileStream.Write(content, 0, content.Length);
_FileStream.Close();
using (var fs = new FileStream(fileName, FileMode.Create, FileAccess.Write))
fs.Write(content, 0, content.Length);
return true; return true;
} }
catch (Exception _Exception)
catch (Exception ex)
{ {
Console.WriteLine("Exception caught in process: {0}", Console.WriteLine("Exception caught in process: {0}",
_Exception.ToString());
ex.ToString());
} }
return false; return false;
} }
public static void UncompressFile(string fileName, byte[] content) public static void UncompressFile(string fileName, byte[] content)
{ {
FileStream destinationFile = File.Create(fileName);
// Because the uncompressed size of the file is unknown, // Because the uncompressed size of the file is unknown,
// we are using an arbitrary buffer size. // we are using an arbitrary buffer size.
byte[] buffer = new byte[4096]; byte[] buffer = new byte[4096];
int n; int n;
using (GZipStream input = new GZipStream(new MemoryStream(content),
using(var fs = File.Create(fileName))
using (var input = new GZipStream(
new MemoryStream(content),
CompressionMode.Decompress, false)) CompressionMode.Decompress, false))
{ {
while ((n = input.Read(buffer, 0, buffer.Length)) > 0) while ((n = input.Read(buffer, 0, buffer.Length)) > 0)
{ {
destinationFile.Write(buffer, 0, n);
fs.Write(buffer, 0, n);
} }
} }
destinationFile.Close();
} }
} }


+ 39
- 28
shadowsocks-csharp/Controller/System/SystemProxy.cs View File

@@ -9,7 +9,7 @@ using Shadowsocks.Model;
namespace Shadowsocks.Controller namespace Shadowsocks.Controller
{ {
public class SystemProxy
public static class SystemProxy
{ {
[DllImport("wininet.dll")] [DllImport("wininet.dll")]
@@ -26,19 +26,29 @@ namespace Shadowsocks.Controller
_refreshReturn = InternetSetOption(IntPtr.Zero, INTERNET_OPTION_REFRESH, IntPtr.Zero, 0); _refreshReturn = InternetSetOption(IntPtr.Zero, INTERNET_OPTION_REFRESH, IntPtr.Zero, 0);
} }
private static readonly DateTime UnixEpoch
= new DateTime(1970, 1, 1, 0, 0, 0, DateTimeKind.Utc);
public static long ToUnixEpochMilliseconds(this DateTime dt)
=> (long)(dt - UnixEpoch).TotalMilliseconds;
private static string GetTimestamp(DateTime value)
{
return value.ToString("yyyyMMddHHmmssfff");
}
public static void Update(Configuration config, bool forceDisable) public static void Update(Configuration config, bool forceDisable)
{ {
bool global = config.global; bool global = config.global;
bool enabled = config.enabled; bool enabled = config.enabled;
if (forceDisable) if (forceDisable)
{ {
enabled = false; enabled = false;
} }
try try
{ {
RegistryKey registry =
Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
true);
var registry = Registry.CurrentUser
.OpenSubKey(@"Software\Microsoft\Windows\CurrentVersion\Internet Settings", true);
if (enabled) if (enabled)
{ {
if (global) if (global)
@@ -53,7 +63,7 @@ namespace Shadowsocks.Controller
if (config.useOnlinePac && !string.IsNullOrEmpty(config.pacUrl)) if (config.useOnlinePac && !string.IsNullOrEmpty(config.pacUrl))
pacUrl = config.pacUrl; pacUrl = config.pacUrl;
else else
pacUrl = "http://127.0.0.1:" + config.localPort.ToString() + "/pac?t=" + GetTimestamp(DateTime.Now);
pacUrl = $"http://127.0.0.1:{config.localPort}/pac?t={GetTimestamp(DateTime.Now)}";
registry.SetValue("ProxyEnable", 0); registry.SetValue("ProxyEnable", 0);
var readProxyServer = registry.GetValue("ProxyServer"); var readProxyServer = registry.GetValue("ProxyServer");
registry.SetValue("ProxyServer", ""); registry.SetValue("ProxyServer", "");
@@ -66,9 +76,11 @@ namespace Shadowsocks.Controller
registry.SetValue("ProxyServer", ""); registry.SetValue("ProxyServer", "");
registry.SetValue("AutoConfigURL", ""); registry.SetValue("AutoConfigURL", "");
} }
//Set AutoDetectProxy Off
IEAutoDetectProxy(false);
SystemProxy.NotifyIE();
//Set AutoDetectProxy
IEAutoDetectProxy(!enabled);
NotifyIE();
//Must Notify IE first, or the connections do not chanage //Must Notify IE first, or the connections do not chanage
CopyProxySettingFromLan(); CopyProxySettingFromLan();
} }
@@ -82,14 +94,13 @@ namespace Shadowsocks.Controller
private static void CopyProxySettingFromLan() private static void CopyProxySettingFromLan()
{ {
RegistryKey registry =
Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\Connections",
true);
var registry = Registry.CurrentUser
.OpenSubKey(@"Software\Microsoft\Windows\CurrentVersion\Internet Settings\Connections", true);
var defaultValue = registry.GetValue("DefaultConnectionSettings"); var defaultValue = registry.GetValue("DefaultConnectionSettings");
try try
{ {
var connections = registry.GetValueNames(); var connections = registry.GetValueNames();
foreach (String each in connections)
foreach (var each in connections)
{ {
if (!(each.Equals("DefaultConnectionSettings") if (!(each.Equals("DefaultConnectionSettings")
|| each.Equals("LAN Connection") || each.Equals("LAN Connection")
@@ -99,38 +110,38 @@ namespace Shadowsocks.Controller
registry.SetValue(each, defaultValue); registry.SetValue(each, defaultValue);
} }
} }
SystemProxy.NotifyIE();
} catch (IOException e) {
NotifyIE();
}
catch (IOException e)
{
Logging.LogUsefulException(e); Logging.LogUsefulException(e);
} }
} }
private static String GetTimestamp(DateTime value)
{
return value.ToString("yyyyMMddHHmmssffff");
}
/// <summary> /// <summary>
/// Checks or unchecks the IE Options Connection setting of "Automatically detect Proxy" /// Checks or unchecks the IE Options Connection setting of "Automatically detect Proxy"
/// </summary> /// </summary>
/// <param name="set">Provide 'true' if you want to check the 'Automatically detect Proxy' check box. To uncheck, pass 'false'</param> /// <param name="set">Provide 'true' if you want to check the 'Automatically detect Proxy' check box. To uncheck, pass 'false'</param>
private static void IEAutoDetectProxy(bool set) private static void IEAutoDetectProxy(bool set)
{ {
RegistryKey registry =
Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\Connections",
true);
byte[] defConnection = (byte[])registry.GetValue("DefaultConnectionSettings");
byte[] savedLegacySetting = (byte[])registry.GetValue("SavedLegacySettings");
var registry = Registry.CurrentUser
.OpenSubKey(@"Software\Microsoft\Windows\CurrentVersion\Internet Settings\Connections", true);
var defConnection = (byte[])registry.GetValue("DefaultConnectionSettings");
var savedLegacySetting = (byte[])registry.GetValue("SavedLegacySettings");
if (set) if (set)
{ {
defConnection[8] = Convert.ToByte(defConnection[8] & 8);
savedLegacySetting[8] = Convert.ToByte(savedLegacySetting[8] & 8);
defConnection[8] = (byte)(defConnection[8] | 8);
savedLegacySetting[8] = (byte)(savedLegacySetting[8] | 8);
} }
else else
{ {
defConnection[8] = Convert.ToByte(defConnection[8] & ~8);
savedLegacySetting[8] = Convert.ToByte(savedLegacySetting[8] & ~8);
defConnection[8] = (byte)(defConnection[8] & ~8);
savedLegacySetting[8] = (byte)(savedLegacySetting[8] & ~8);
} }
BitConverter.GetBytes(unchecked(BitConverter.ToUInt32(defConnection, 4) + 1)).CopyTo(defConnection, 4);
BitConverter.GetBytes(unchecked(BitConverter.ToUInt32(savedLegacySetting, 4) + 1)).CopyTo(savedLegacySetting, 4);
registry.SetValue("DefaultConnectionSettings", defConnection); registry.SetValue("DefaultConnectionSettings", defConnection);
registry.SetValue("SavedLegacySettings", savedLegacySetting); registry.SetValue("SavedLegacySettings", savedLegacySetting);
} }


Loading…
Cancel
Save