public static Protocol Create(Buffer reader, Version version)
{
if (BASE_SIZE > reader.BytesLeft)
{
return(null);
}
byte command_flags = reader.ReadByte();
var channel = reader.ReadByte();
var reliableSequenceNumber = reader.ReadUInt16();
Protocol result;
switch ((ProtocolCommand)(command_flags & 0x0Fu))
{
case ProtocolCommand.NONE: result = null; break;
case ProtocolCommand.ACKNOWLEDGE: result = new Acknowledge(); break;
case ProtocolCommand.CONNECT: result = new Connect(); break;
case ProtocolCommand.VERIFY_CONNECT: result = new VerifyConnect(); break;
case ProtocolCommand.DISCONNECT: result = new Disconnect(); break;
case ProtocolCommand.PING: result = new Ping(); break;
case ProtocolCommand.SEND_FRAGMENT: result = new Send.Fragment(); break;
case ProtocolCommand.SEND_RELIABLE: result = new Send.Reliable(); break;
case ProtocolCommand.SEND_UNRELIABLE: result = new Send.Unreliable(); break;
case ProtocolCommand.SEND_UNSEQUENCED: result = new Send.Unsequenced(); break;
case ProtocolCommand.BANDWIDTH_LIMIT: result = new BandwidthLimit(); break;
case ProtocolCommand.THROTTLE_CONFIGURE: result = new ThrottleConfigure(); break;
default: result = null; break;
}
;
if (result == null || (result.Size - BASE_SIZE) > reader.BytesLeft)
{
return(null);
}
result.ChannelID = channel;
result.Flags = (ProtocolFlag)(command_flags & 0xF0);
result.ReliableSequenceNumber = reliableSequenceNumber;
result.ReadInternal(reader, version);
return(result);
}
public BaseConnection(Cap cap, BandwidthLimit bandwidthLimit, int maxReceiveCount, BufferManager bufferManager)
{
_cap = cap;
_bandwidthLimit = bandwidthLimit;
_maxReceiveCount = maxReceiveCount;
_bufferManager = bufferManager;
if (_bandwidthLimit != null) _bandwidthLimit.Join(this);
_aliveTimer = new System.Threading.Timer(this.AliveTimer, null, 1000 * 30, 1000 * 30);
_aliveStopwatch.Start();
_connect = true;
}
public Connection AcceptConnection(out string uri, BandwidthLimit bandwidthLimit)
{
uri = null;
List<IDisposable> garbages = new List<IDisposable>();
try
{
Connection connection = null;
foreach (var type in (new int[] { 0, 1 }).Randomize())
{
if (this.State == ManagerState.Stop) return null;
if (type == 0)
{
lock (this.ThisLock)
{
foreach (var item in _tcpListeners)
{
if (item.Value.Pending())
{
var socket = item.Value.AcceptTcpClient().Client;
garbages.Add(socket);
{
var remoteEndPoint = (IPEndPoint)socket.RemoteEndPoint;
uri = string.Format("tcp:{0}:{1}", remoteEndPoint.Address, remoteEndPoint.Port);
}
if (!this.OnCheckUriEvent(uri))
{
_blockedCount.Increment();
continue;
}
var cap = new SocketCap(socket);
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
}
}
}
else if (type == 1)
{
// Overlay network
var cap = this.OnAcceptCapEvent(out uri);
if (cap == null) continue;
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
if (connection != null) break;
}
if (connection == null) return null;
var secureConnection = new SecureConnection(SecureConnectionVersion.Version3, SecureConnectionType.Accept, connection, null, _bufferManager);
garbages.Add(secureConnection);
secureConnection.Connect(new TimeSpan(0, 0, 30));
var compressConnection = new CompressConnection(secureConnection, _maxReceiveCount, _bufferManager);
garbages.Add(compressConnection);
compressConnection.Connect(new TimeSpan(0, 0, 10));
return compressConnection;
}
catch (Exception)
{
foreach (var item in garbages)
{
item.Dispose();
}
}
return null;
}
public Connection CreateConnection(string uri, BandwidthLimit bandwidthLimit)
{
List <IDisposable> garbages = new List <IDisposable>();
try
{
Connection connection = null;
if (connection == null)
{
// Overlay network
var cap = this.OnCreateCapEvent(uri);
if (cap == null)
{
goto End;
}
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
End :;
}
if (connection == null)
{
ConnectionFilter connectionFilter = null;
lock (this.ThisLock)
{
foreach (var filter in this.Filters)
{
if (filter.UriCondition.IsMatch(uri))
{
if (filter.ConnectionType != ConnectionType.None)
{
connectionFilter = filter.Clone();
}
break;
}
}
}
if (connectionFilter == null)
{
return(null);
}
string scheme = null;
string host = null;
int port = -1;
{
var match = _regex.Match(uri);
if (match.Success)
{
scheme = match.Groups[1].Value;
host = match.Groups[2].Value;
port = int.Parse(match.Groups[3].Value);
}
else
{
var match2 = _regex2.Match(uri);
if (match2.Success)
{
scheme = match2.Groups[1].Value;
host = match2.Groups[2].Value;
port = 4050;
}
}
}
if (host == null)
{
return(null);
}
IList <KeyValuePair <string, string> > options = null;
if (!string.IsNullOrWhiteSpace(connectionFilter.Option))
{
options = ClientManager.Decode(connectionFilter.Option).OfType <KeyValuePair <string, string> >().ToList();
}
if (connectionFilter.ConnectionType == ConnectionType.Tcp)
{
var ipAddress = ClientManager.GetIpAddress(host);
if (ipAddress == null)
{
return(null);
}
host = ipAddress.ToString();
uri = string.Format("{0}:{1}:{2}", scheme, host, port);
if (!this.OnCheckUriEvent(uri))
{
return(null);
}
#if !DEBUG
// Check
{
Uri url = new Uri(string.Format("{0}://{1}:{2}", scheme, host, port));
if (url.HostNameType == UriHostNameType.IPv4)
{
if (IPAddress.Any.ToString() == ipAddress.ToString() ||
IPAddress.Loopback.ToString() == ipAddress.ToString() ||
IPAddress.Broadcast.ToString() == ipAddress.ToString())
{
return(null);
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("10.0.0.0").GetAddressBytes()) >= 0 &&
CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("10.255.255.255").GetAddressBytes()) <= 0)
{
return(null);
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("172.16.0.0").GetAddressBytes()) >= 0 &&
CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("172.31.255.255").GetAddressBytes()) <= 0)
{
return(null);
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("127.0.0.0").GetAddressBytes()) >= 0 &&
CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("127.255.255.255").GetAddressBytes()) <= 0)
{
return(null);
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("192.168.0.0").GetAddressBytes()) >= 0 &&
CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("192.168.255.255").GetAddressBytes()) <= 0)
{
return(null);
}
}
else if (url.HostNameType == UriHostNameType.IPv6)
{
if (IPAddress.IPv6Any.ToString() == ipAddress.ToString() ||
IPAddress.IPv6Loopback.ToString() == ipAddress.ToString() ||
IPAddress.IPv6None.ToString() == ipAddress.ToString())
{
return(null);
}
if (ipAddress.ToString().ToLower().StartsWith("fe80:"))
{
return(null);
}
}
}
#endif
var socket = ClientManager.Connect(new IPEndPoint(ipAddress, port), new TimeSpan(0, 0, 10));
garbages.Add(socket);
var cap = new SocketCap(socket);
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
else
{
if (!this.OnCheckUriEvent(uri))
{
return(null);
}
string proxyScheme = null;
string proxyHost = null;
int proxyPort = -1;
{
var match = _regex.Match(connectionFilter.ProxyUri);
if (match.Success)
{
proxyScheme = match.Groups[1].Value;
proxyHost = match.Groups[2].Value;
proxyPort = int.Parse(match.Groups[3].Value);
}
else
{
var match2 = _regex2.Match(connectionFilter.ProxyUri);
if (match2.Success)
{
proxyScheme = match2.Groups[1].Value;
proxyHost = match2.Groups[2].Value;
if (connectionFilter.ConnectionType == ConnectionType.Socks4Proxy ||
connectionFilter.ConnectionType == ConnectionType.Socks4aProxy ||
connectionFilter.ConnectionType == ConnectionType.Socks5Proxy)
{
proxyPort = 1080;
}
else if (connectionFilter.ConnectionType == ConnectionType.HttpProxy)
{
proxyPort = 80;
}
}
}
}
if (proxyHost == null)
{
return(null);
}
if (connectionFilter.ConnectionType == ConnectionType.Socks4Proxy ||
connectionFilter.ConnectionType == ConnectionType.Socks4aProxy ||
connectionFilter.ConnectionType == ConnectionType.Socks5Proxy ||
connectionFilter.ConnectionType == ConnectionType.HttpProxy)
{
var socket = ClientManager.Connect(new IPEndPoint(ClientManager.GetIpAddress(proxyHost), proxyPort), new TimeSpan(0, 0, 10));
garbages.Add(socket);
ProxyClientBase proxy = null;
if (connectionFilter.ConnectionType == ConnectionType.Socks4Proxy)
{
var user = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("user")).Select(n => n.Value).FirstOrDefault() : null;
proxy = new Socks4ProxyClient(socket, user, host, port);
}
else if (connectionFilter.ConnectionType == ConnectionType.Socks4aProxy)
{
var user = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("user")).Select(n => n.Value).FirstOrDefault() : null;
proxy = new Socks4aProxyClient(socket, user, host, port);
}
else if (connectionFilter.ConnectionType == ConnectionType.Socks5Proxy)
{
var user = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("user")).Select(n => n.Value).FirstOrDefault() : null;
var pass = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("pass")).Select(n => n.Value).FirstOrDefault() : null;
proxy = new Socks5ProxyClient(socket, user, pass, host, port);
}
else if (connectionFilter.ConnectionType == ConnectionType.HttpProxy)
{
proxy = new HttpProxyClient(socket, host, port);
}
var cap = new SocketCap(proxy.Create(new TimeSpan(0, 0, 30)));
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
}
}
if (connection == null)
{
return(null);
}
var secureConnection = new SecureConnection(SecureConnectionVersion.Version3, SecureConnectionType.Connect, connection, null, _bufferManager);
garbages.Add(secureConnection);
secureConnection.Connect(new TimeSpan(0, 0, 30));
var compressConnection = new CompressConnection(secureConnection, _maxReceiveCount, _bufferManager);
garbages.Add(compressConnection);
compressConnection.Connect(new TimeSpan(0, 0, 10));
return(compressConnection);
}
catch (Exception)
{
foreach (var item in garbages)
{
item.Dispose();
}
}
return(null);
}
public Connection AcceptConnection(out string uri, BandwidthLimit bandwidthLimit)
{
uri = null;
List <IDisposable> garbages = new List <IDisposable>();
try
{
Connection connection = null;
foreach (var type in (new int[] { 0, 1 }).Randomize())
{
if (this.State == ManagerState.Stop)
{
return(null);
}
if (type == 0)
{
lock (this.ThisLock)
{
foreach (var item in _tcpListeners)
{
if (item.Value.Pending())
{
var socket = item.Value.AcceptTcpClient().Client;
garbages.Add(socket);
{
var remoteEndPoint = (IPEndPoint)socket.RemoteEndPoint;
uri = string.Format("tcp:{0}:{1}", remoteEndPoint.Address, remoteEndPoint.Port);
}
if (!this.OnCheckUriEvent(uri))
{
_blockedCount.Increment();
continue;
}
var cap = new SocketCap(socket);
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
}
}
}
else if (type == 1)
{
// Overlay network
var cap = this.OnAcceptCapEvent(out uri);
if (cap == null)
{
continue;
}
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
if (connection != null)
{
break;
}
}
if (connection == null)
{
return(null);
}
var secureConnection = new SecureConnection(SecureConnectionVersion.Version3, SecureConnectionType.Accept, connection, null, _bufferManager);
garbages.Add(secureConnection);
secureConnection.Connect(new TimeSpan(0, 0, 30));
var compressConnection = new CompressConnection(secureConnection, _maxReceiveCount, _bufferManager);
garbages.Add(compressConnection);
compressConnection.Connect(new TimeSpan(0, 0, 10));
return(compressConnection);
}
catch (Exception)
{
foreach (var item in garbages)
{
item.Dispose();
}
}
return(null);
}
protected override void Dispose(bool disposing)
{
if (_disposed) return;
_disposed = true;
if (disposing)
{
if (_aliveTimer != null)
{
try
{
_aliveTimer.Dispose();
}
catch (Exception)
{
}
_aliveTimer = null;
}
if (_cap != null)
{
try
{
_cap.Dispose();
}
catch (Exception)
{
}
_cap = null;
}
if (_bandwidthLimit != null)
{
try
{
_bandwidthLimit.Leave(this);
}
catch (Exception)
{
}
_bandwidthLimit = null;
}
_connect = false;
}
}
public Connection CreateConnection(string uri, BandwidthLimit bandwidthLimit)
{
List<IDisposable> garbages = new List<IDisposable>();
try
{
Connection connection = null;
if (connection == null)
{
// Overlay network
var cap = this.OnCreateCapEvent(uri);
if (cap == null) goto End;
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
End: ;
}
if (connection == null)
{
ConnectionFilter connectionFilter = null;
lock (this.ThisLock)
{
foreach (var filter in this.Filters)
{
if (filter.UriCondition.IsMatch(uri))
{
if (filter.ConnectionType != ConnectionType.None)
{
connectionFilter = filter.Clone();
}
break;
}
}
}
if (connectionFilter == null) return null;
string scheme = null;
string host = null;
int port = -1;
{
var match = _regex.Match(uri);
if (match.Success)
{
scheme = match.Groups[1].Value;
host = match.Groups[2].Value;
port = int.Parse(match.Groups[3].Value);
}
else
{
var match2 = _regex2.Match(uri);
if (match2.Success)
{
scheme = match2.Groups[1].Value;
host = match2.Groups[2].Value;
port = 4050;
}
}
}
if (host == null) return null;
IList<KeyValuePair<string, string>> options = null;
if (!string.IsNullOrWhiteSpace(connectionFilter.Option))
{
options = ClientManager.Decode(connectionFilter.Option).OfType<KeyValuePair<string, string>>().ToList();
}
if (connectionFilter.ConnectionType == ConnectionType.Tcp)
{
var ipAddress = ClientManager.GetIpAddress(host);
if (ipAddress == null) return null;
host = ipAddress.ToString();
uri = string.Format("{0}:{1}:{2}", scheme, host, port);
if (!this.OnCheckUriEvent(uri))
{
return null;
}
#if !DEBUG
// Check
{
Uri url = new Uri(string.Format("{0}://{1}:{2}", scheme, host, port));
if (url.HostNameType == UriHostNameType.IPv4)
{
if (IPAddress.Any.ToString() == ipAddress.ToString()
|| IPAddress.Loopback.ToString() == ipAddress.ToString()
|| IPAddress.Broadcast.ToString() == ipAddress.ToString())
{
return null;
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("10.0.0.0").GetAddressBytes()) >= 0
&& CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("10.255.255.255").GetAddressBytes()) <= 0)
{
return null;
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("172.16.0.0").GetAddressBytes()) >= 0
&& CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("172.31.255.255").GetAddressBytes()) <= 0)
{
return null;
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("127.0.0.0").GetAddressBytes()) >= 0
&& CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("127.255.255.255").GetAddressBytes()) <= 0)
{
return null;
}
if (CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("192.168.0.0").GetAddressBytes()) >= 0
&& CollectionUtilities.Compare(ipAddress.GetAddressBytes(), IPAddress.Parse("192.168.255.255").GetAddressBytes()) <= 0)
{
return null;
}
}
else if (url.HostNameType == UriHostNameType.IPv6)
{
if (IPAddress.IPv6Any.ToString() == ipAddress.ToString()
|| IPAddress.IPv6Loopback.ToString() == ipAddress.ToString()
|| IPAddress.IPv6None.ToString() == ipAddress.ToString())
{
return null;
}
if (ipAddress.ToString().ToLower().StartsWith("fe80:"))
{
return null;
}
}
}
#endif
var socket = ClientManager.Connect(new IPEndPoint(ipAddress, port), new TimeSpan(0, 0, 10));
garbages.Add(socket);
var cap = new SocketCap(socket);
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
else
{
if (!this.OnCheckUriEvent(uri))
{
return null;
}
string proxyScheme = null;
string proxyHost = null;
int proxyPort = -1;
{
var match = _regex.Match(connectionFilter.ProxyUri);
if (match.Success)
{
proxyScheme = match.Groups[1].Value;
proxyHost = match.Groups[2].Value;
proxyPort = int.Parse(match.Groups[3].Value);
}
else
{
var match2 = _regex2.Match(connectionFilter.ProxyUri);
if (match2.Success)
{
proxyScheme = match2.Groups[1].Value;
proxyHost = match2.Groups[2].Value;
if (connectionFilter.ConnectionType == ConnectionType.Socks4Proxy
|| connectionFilter.ConnectionType == ConnectionType.Socks4aProxy
|| connectionFilter.ConnectionType == ConnectionType.Socks5Proxy)
{
proxyPort = 1080;
}
else if (connectionFilter.ConnectionType == ConnectionType.HttpProxy)
{
proxyPort = 80;
}
}
}
}
if (proxyHost == null) return null;
if (connectionFilter.ConnectionType == ConnectionType.Socks4Proxy
|| connectionFilter.ConnectionType == ConnectionType.Socks4aProxy
|| connectionFilter.ConnectionType == ConnectionType.Socks5Proxy
|| connectionFilter.ConnectionType == ConnectionType.HttpProxy)
{
var socket = ClientManager.Connect(new IPEndPoint(ClientManager.GetIpAddress(proxyHost), proxyPort), new TimeSpan(0, 0, 10));
garbages.Add(socket);
ProxyClientBase proxy = null;
if (connectionFilter.ConnectionType == ConnectionType.Socks4Proxy)
{
var user = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("user")).Select(n => n.Value).FirstOrDefault() : null;
proxy = new Socks4ProxyClient(socket, user, host, port);
}
else if (connectionFilter.ConnectionType == ConnectionType.Socks4aProxy)
{
var user = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("user")).Select(n => n.Value).FirstOrDefault() : null;
proxy = new Socks4aProxyClient(socket, user, host, port);
}
else if (connectionFilter.ConnectionType == ConnectionType.Socks5Proxy)
{
var user = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("user")).Select(n => n.Value).FirstOrDefault() : null;
var pass = (options != null) ? options.Where(n => n.Key.ToLower().StartsWith("pass")).Select(n => n.Value).FirstOrDefault() : null;
proxy = new Socks5ProxyClient(socket, user, pass, host, port);
}
else if (connectionFilter.ConnectionType == ConnectionType.HttpProxy)
{
proxy = new HttpProxyClient(socket, host, port);
}
var cap = new SocketCap(proxy.Create(new TimeSpan(0, 0, 30)));
garbages.Add(cap);
connection = new BaseConnection(cap, bandwidthLimit, _maxReceiveCount, _bufferManager);
garbages.Add(connection);
}
}
}
if (connection == null) return null;
var secureConnection = new SecureConnection(SecureConnectionVersion.Version3, SecureConnectionType.Connect, connection, null, _bufferManager);
garbages.Add(secureConnection);
secureConnection.Connect(new TimeSpan(0, 0, 30));
var compressConnection = new CompressConnection(secureConnection, _maxReceiveCount, _bufferManager);
garbages.Add(compressConnection);
compressConnection.Connect(new TimeSpan(0, 0, 10));
return compressConnection;
}
catch (Exception)
{
foreach (var item in garbages)
{
item.Dispose();
}
}
return null;
}
protected override void Dispose(bool disposing)
{
if (_disposed) return;
_disposed = true;
if (disposing)
{
_blockUploadedEventQueue.Dispose();
if (_refreshTimer != null)
{
try
{
_refreshTimer.Dispose();
}
catch (Exception)
{
}
_refreshTimer = null;
}
if (_mediateTimer != null)
{
try
{
_mediateTimer.Dispose();
}
catch (Exception)
{
}
_mediateTimer = null;
}
if (_reduceTimer != null)
{
try
{
_reduceTimer.Dispose();
}
catch (Exception)
{
}
_reduceTimer = null;
}
if (_packetControlManager != null)
{
try
{
_packetControlManager.Dispose();
}
catch (Exception)
{
}
_packetControlManager = null;
}
if (_bandwidthLimit != null)
{
try
{
_bandwidthLimit.Dispose();
}
catch (Exception)
{
}
_bandwidthLimit = null;
}
}
}
protected override void Dispose(bool disposing)
{
if (_disposed) return;
_disposed = true;
if (disposing)
{
if (_refreshTimer != null)
{
try
{
_refreshTimer.Dispose();
}
catch (Exception)
{
}
_refreshTimer = null;
}
if (_messagesManager != null)
{
try
{
_messagesManager.Dispose();
}
catch (Exception)
{
}
_messagesManager = null;
}
if (_bandwidthLimit != null)
{
try
{
_bandwidthLimit.Dispose();
}
catch (Exception)
{
}
_bandwidthLimit = null;
}
}
}
