protected void Clean()
{
if (trp.KeepAlive)
{
targetSocket = null;
targetConnectionPool.ReuseSocket(crp, targetSocketInfo);
}
else
{
targetSocket = null;
targetConnectionPool.RemoveSocket(crp, targetSocketInfo);
}
targetSocket = null;
//clear both buffers
crp = null;
trp = null;
try
{
clientSocket.BeginReceive(clientBuffer, 0, BiConnection.BUFFER_SIZE,
SocketFlags.None, new AsyncCallback(OnClientReceived), null);
}
catch (SocketException)
{
RetryOrDispose(true, false, false);
}
}
public virtual void ReuseSocket(ClientRequestPacket crp, SocketInfo socketInfo)
{
lock (socketPool)
{
AddSocket(crp, socketInfo);
socketInfo.InUse = false;
}
}
protected void OnClientReceived(IAsyncResult ir)
{
int len = 0;
try
{
SocketError se;
len = clientSocket.EndReceive(ir, out se);
}
catch (SocketException)
{
//client failed.
RetryOrDispose(true, false, false);
return;
}
if (len <= 0)
{
RetryOrDispose(true, false, false);
return;
}
if (crp == null)
{
crp = new ClientRequestPacket();
}
crp.AppendData(clientBuffer, 0, len);
if (crp.HeaderReceived && !crp.HeaderShooted)
{
//shoot header;
this.listener.InvokeNewRequestEvent(new NewRequestEventHandlerArgs
{
Url = crp.Headers[0]
});
ConnectTarget();
}
else if (crp.HeaderShooted && !crp.ContentShooted)
{
if (crp.ContentBuffer != null && crp.ContentBuffer.Length > 0)
{
try
{
targetSocket.BeginSend(crp.ContentBuffer, 0, crp.ContentBuffer.Length, SocketFlags.None,
new AsyncCallback(OnTargetContentSent), targetSocket);
}
catch (SocketException)
{
RetryOrDispose(false, true, false);
return;
}
}
}
}
public bool NeedValidate(ClientRequestPacket crp)
{
string fullURL;
if (crp.Port == 80)
{
fullURL = crp.Protocol + "://" + crp.Host + crp.RelativePath;
}
else
{
fullURL = crp.Protocol + "://" + crp.Host + ":" + crp.Port + crp.RelativePath;
}
return(pacSetting.IsNeedValidate(fullURL));
}
public bool ValidateTargetResponse(ClientRequestPacket crp, TargetResponsePacket trp)
{
string fullURL;
if (crp.Port == 80)
{
fullURL = crp.Protocol + "://" + crp.Host + crp.RelativePath;
}
else
{
fullURL = crp.Protocol + "://" + crp.Host + ":" + crp.Port + crp.RelativePath;
}
string html = ReformHTML(trp);
return(pacSetting.ValidateHtml(fullURL, html));
}
public virtual void RemoveSocket(ClientRequestPacket crp, SocketInfo socketInfo)
{
string requestHost = crp.Host + ":" + crp.Host;
List <SocketInfo> sockets;
lock (socketLock)
{
if (socketPool.TryGetValue(requestHost, out sockets))
{
sockets.Remove(socketInfo);
if (sockets.Count == 0)
{
socketPool.Remove(requestHost);
}
}
}
}
protected virtual void AddSocket(ClientRequestPacket crp, SocketInfo socketInfo)
{
string requestHost = crp.Host + ":" + crp.Port;
List <SocketInfo> sockets;
Debug.Assert(socketInfo.Socket != null);
lock (socketLock)
{
if (!socketPool.TryGetValue(requestHost, out sockets))
{
sockets = new List <SocketInfo>(1);
socketPool[requestHost] = sockets;
}
if (!sockets.Contains(socketInfo))
{
sockets.Add(socketInfo);
}
}
}
public virtual bool RequestSocketReuse(ClientRequestPacket crp, out SocketInfo socketInfo)
{
string requestHost = crp.Host + ":" + crp.Port;
List <SocketInfo> sockets;
socketInfo = null;
lock (socketLock)
{
if (!socketPool.TryGetValue(requestHost, out sockets))
{
sockets = new List <SocketInfo>(1);
socketPool[requestHost] = sockets;
List <SocketInfo> toRemove = new List <SocketInfo>();
foreach (SocketInfo si in sockets)
{
if (!si.InUse)
{
if (TouchSocket(si))
{
socketInfo = si;
socketInfo.InUse = true;
break;
}
else
{
//remove
toRemove.Add(si);
}
}
}
foreach (SocketInfo si in toRemove)
{
//si.Socket.Shutdown(SocketShutdown.Both);
Console.WriteLine("Actively Removed {0}", ((IPEndPoint)si.Socket.RemoteEndPoint).Port);
si.Socket.Close();
sockets.Remove(si);
}
}
}
//if no connection available,return null;
return(socketInfo != null);
}