public void SendTo(string sessionKey, byte[] data, int offset, int count)
{
GuardRunning();
if (string.IsNullOrEmpty(sessionKey))
{
throw new ArgumentNullException("sessionKey");
}
if (data == null)
{
throw new ArgumentNullException("data");
}
TcpSocketSession session = null;
if (_sessions.TryGetValue(sessionKey, out session))
{
session.Send(data, offset, count);
}
else
{
_log.WarnFormat("Cannot find session [{0}].", sessionKey);
}
}
public TcpClientDataReceivedEventArgs(TcpSocketSession session, byte[] data, int dataOffset, int dataLength)
{
Session = session;
Data = data;
DataOffset = dataOffset;
DataLength = dataLength;
}
public TcpClientConnectedEventArgs(TcpSocketSession session)
{
if (session == null)
throw new ArgumentNullException("session");
this.Session = session;
}
public TcpClientDataReceivedEventArgs(TcpSocketSession session, byte[] data, int dataOffset, int dataLength)
{
Session = session;
Data = data;
DataOffset = dataOffset;
DataLength = dataLength;
}
public IAsyncResult BeginSendTo(TcpSocketSession session, byte[] data, int offset, int count, AsyncCallback callback, object state)
{
GuardRunning();
if (session == null)
{
throw new ArgumentNullException("session");
}
if (data == null)
{
throw new ArgumentNullException("data");
}
TcpSocketSession writeSession = null;
if (_sessions.TryGetValue(session.SessionKey, out writeSession))
{
return(session.BeginSend(data, offset, count, callback, state));
}
else
{
_log.WarnFormat("Cannot find session [{0}].", session);
}
return(null);
}
public void BeginSendTo(TcpSocketSession session, byte[] data, int offset, int count)
{
GuardRunning();
if (session == null)
{
throw new ArgumentNullException("session");
}
if (data == null)
{
throw new ArgumentNullException("data");
}
TcpSocketSession writeSession = null;
if (_sessions.TryGetValue(session.SessionKey, out writeSession))
{
session.BeginSend(data, offset, count);
}
else
{
_log.WarnFormat("Cannot find session [{0}].", session);
}
}
public TcpSocketSession GetSession(string sessionKey)
{
TcpSocketSession session = null;
_sessions.TryGetValue(sessionKey, out session);
return(session);
}
public TcpClientDisconnectedEventArgs(TcpSocketSession session)
{
if (session == null)
{
throw new ArgumentNullException("session");
}
this.Session = session;
}
public void CloseSession(string sessionKey)
{
TcpSocketSession session = null;
if (_sessions.TryGetValue(sessionKey, out session))
{
session.Close(); // parent server close session by session-key
}
}
public void CloseSession(string sessionKey)
{
TcpSocketSession session = null;
if (_sessions.TryGetValue(sessionKey, out session))
{
session.Close();
}
}
private void CloseSession(TcpSocketSession session)
{
if (session == null)
{
return;
}
_sessions.TryRemove(session.SessionKey, out var sessionToBeThrowAway);
session.Close(); // parent server close session
}
private void HandleTcpClientAccepted(IAsyncResult ar)
{
if (!_isListening)
{
return;
}
TcpListener listener = (TcpListener)ar.AsyncState;
try
{
TcpClient tcpClient = listener.EndAcceptTcpClient(ar);
if (!tcpClient.Connected)
{
return;
}
var session = new TcpSocketSession(tcpClient, _configuration, _configuration.BufferManager, this);
bool isSessionStarted = false;
try
{
_sessions.AddOrUpdate(session.SessionKey, session, (n, o) => { return(o); });
session.Start();
isSessionStarted = true;
}
catch (Exception ex)
{
_log.Error(ex.Message, ex);
}
if (!isSessionStarted)
{
CloseSession(session); // session was not started
}
}
catch (Exception ex)
{
if (!ShouldThrow(ex))
{
_log.Error(ex.Message, ex);
}
else
{
throw;
}
}
finally
{
try
{
ContinueAcceptSession(listener);
}
catch { }
}
}
private void CloseSession(TcpSocketSession session)
{
TcpSocketSession sessionToBeThrowAway;
_sessions.TryRemove(session.SessionKey, out sessionToBeThrowAway);
if (session != null)
{
session.Close(); // parent server close session
}
}
internal void RaiseClientConnected(TcpSocketSession session)
{
try
{
if (ClientConnected != null)
{
ClientConnected(this, new TcpClientConnectedEventArgs(session));
}
}
catch (Exception ex)
{
HandleUserSideError(session, ex);
}
}
internal void RaiseClientDataReceived(TcpSocketSession session, byte[] data, int dataOffset, int dataLength)
{
try
{
if (ClientDataReceived != null)
{
ClientDataReceived(this, new TcpClientDataReceivedEventArgs(session, data, dataOffset, dataLength));
}
}
catch (Exception ex)
{
HandleUserSideError(session, ex);
}
}
public void EndSendTo(TcpSocketSession session, IAsyncResult asyncResult)
{
GuardRunning();
if (session == null)
{
throw new ArgumentNullException("session");
}
if (_sessions.TryGetValue(session.SessionKey, out var writeSession))
{
session.EndSend(asyncResult);
}
}
public void EndSendTo(string sessionKey, IAsyncResult asyncResult)
{
GuardRunning();
if (string.IsNullOrEmpty(sessionKey))
{
throw new ArgumentNullException("sessionKey");
}
TcpSocketSession session = null;
if (_sessions.TryGetValue(sessionKey, out session))
{
session.EndSend(asyncResult);
}
}
public IAsyncResult BeginSendTo(TcpSocketSession session, byte[] data, AsyncCallback callback, object state)
{
GuardRunning();
if (session == null)
{
throw new ArgumentNullException("session");
}
if (data == null)
{
throw new ArgumentNullException("data");
}
return(BeginSendTo(session, data, 0, data.Length, callback, state));
}
public void BeginSendTo(TcpSocketSession session, byte[] data)
{
GuardRunning();
if (session == null)
{
throw new ArgumentNullException("session");
}
if (data == null)
{
throw new ArgumentNullException("data");
}
BeginSendTo(session, data, 0, data.Length);
}
internal void RaiseClientDisconnected(TcpSocketSession session)
{
try
{
if (ClientDisconnected != null)
{
ClientDisconnected(this, new TcpClientDisconnectedEventArgs(session));
}
}
catch (Exception ex)
{
HandleUserSideError(session, ex);
}
finally
{
TcpSocketSession sessionToBeThrowAway;
_sessions.TryRemove(session.SessionKey, out sessionToBeThrowAway);
}
}
public TcpClientDataReceivedEventArgs(TcpSocketSession session, byte[] data)
: this(session, data, 0, data.Length)
{
}
public void SendToAsync(TcpSocketSession session, byte[] data, int offset, int count)
{
GuardRunning();
if (session == null)
throw new ArgumentNullException("session");
if (data == null)
throw new ArgumentNullException("data");
TcpSocketSession writeSession = null;
if (_sessions.TryGetValue(session.SessionKey, out writeSession))
{
session.SendAsync(data, offset, count);
}
else
{
_log.WarnFormat("Cannot find session [{0}].", session);
}
}
public void SendToAsync(TcpSocketSession session, byte[] data)
{
GuardRunning();
if (session == null)
throw new ArgumentNullException("session");
if (data == null)
throw new ArgumentNullException("data");
SendToAsync(session, data, 0, data.Length);
}
internal void RaiseClientConnected(TcpSocketSession session)
{
try
{
if (ClientConnected != null)
{
ClientConnected(this, new TcpClientConnectedEventArgs(session));
}
}
catch (Exception ex)
{
HandleUserSideError(ex);
}
}
internal void RaiseClientDisconnected(TcpSocketSession session)
{
try
{
if (ClientDisconnected != null)
{
ClientDisconnected(this, new TcpClientDisconnectedEventArgs(session));
}
}
catch (Exception ex)
{
HandleUserSideError(ex);
}
finally
{
TcpSocketSession sessionToBeThrowAway;
_sessions.TryRemove(session.SessionKey, out sessionToBeThrowAway);
}
}
internal void RaiseClientDataReceived(TcpSocketSession session, byte[] data, int dataOffset, int dataLength)
{
try
{
if (ClientDataReceived != null)
{
ClientDataReceived(this, new TcpClientDataReceivedEventArgs(session, data, dataOffset, dataLength));
}
}
catch (Exception ex)
{
HandleUserSideError(ex);
}
}
public TcpClientDataReceivedEventArgs(TcpSocketSession session, byte[] data)
: this(session, data, 0, data.Length)
{
}
private void CloseSession(TcpSocketSession session)
{
TcpSocketSession sessionToBeThrowAway;
_sessions.TryRemove(session.SessionKey, out sessionToBeThrowAway);
if (session != null)
{
session.Close();
}
}
private void HandleUserSideError(TcpSocketSession session, Exception ex)
{
_log.Error(string.Format("Session [{0}] error occurred in user side [{1}].", session, ex.Message), ex);
}
private void HandleTcpClientAccepted(IAsyncResult ar)
{
if (!_isListening) return;
try
{
TcpListener listener = (TcpListener)ar.AsyncState;
TcpClient tcpClient = listener.EndAcceptTcpClient(ar);
if (!tcpClient.Connected) return;
var session = new TcpSocketSession(tcpClient, _configuration, _bufferManager, this);
bool isSessionStarted = false;
try
{
_sessions.AddOrUpdate(session.SessionKey, session, (n, o) => { return o; });
session.Start();
isSessionStarted = true;
}
catch (Exception ex)
{
_log.Error(ex.Message, ex);
}
if (isSessionStarted)
{
ContinueAcceptSession(listener);
}
else
{
CloseSession(session);
}
}
catch (Exception ex)
{
if (!ShouldThrow(ex))
{
_log.Error(ex.Message, ex);
}
else throw;
}
}
