private void HeaderReceivedCallBack(IAsyncResult result)
{
ITransferContext context = (ITransferContext)result.AsyncState;
int receive = context.Listener.EndReceive(result);
if (receive > 0)
{
context.IncomingHeader(receive);
}
if (context.DeserialPacketSize > 0)
{
int buffersize = (context.DeserialPacketSize < context.BufferSize) ? (int)context.DeserialPacketSize : context.BufferSize;
context.Listener.BeginReceive(context.HeaderBuffer, 0, buffersize, SocketFlags.None, HeaderReceivedCallBack, context);
}
else
{
TransferOperation request = new TransferOperation(context.Transfer, MessagePart.Header, DirectionType.Receive);
request.Resolve(context.DeserialPacket);
if (!context.ReceiveMessage &&
!context.SendMessage)
{
context.Close = true;
}
context.HeaderReceivedNotice.Set();
HeaderReceived.Execute(this);
}
}
private void MessageSentCallback(IAsyncResult result)
{
ITransferContext context = (ITransferContext)result.AsyncState;
try
{
int sendcount = context.Listener.EndSend(result);
}
catch (SocketException) { }
catch (ObjectDisposedException) { }
if (context.SerialPacketId >= 0)
{
TransferOperation request = new TransferOperation(context.Transfer, MessagePart.Message, DirectionType.Send);
request.Resolve();
context.Listener.BeginSend(context.SerialPacket, 0, context.SerialPacket.Length, SocketFlags.None, MessageSentCallback, context);
}
else
{
if (!context.ReceiveMessage)
{
context.Close = true;
}
context.MessageSentNotice.Set();
MessageSent.Execute(this);
}
}
public void Send(MessagePart messagePart)
{
if (!IsConnected())
{
throw new Exception("Destination socket is not connected.");
}
AsyncCallback callback = HeaderSentCallback;
if (messagePart == MessagePart.Header)
{
callback = HeaderSentCallback;
TransferOperation request = new TransferOperation(Context.Transfer, MessagePart.Header, DirectionType.Send);
request.Resolve();
}
else if (Context.SendMessage)
{
callback = MessageSentCallback;
context.SerialPacketId = 0;
TransferOperation request = new TransferOperation(context.Transfer, MessagePart.Message, DirectionType.Send);
request.Resolve();
}
else
{
return;
}
context.Listener.BeginSend(context.SerialPacket, 0, context.SerialPacket.Length, SocketFlags.None, callback, context);
}
public void DealHeaderReceived(ITransferContext context)
{
if (context.DeserialPacketSize > 0)
{
int buffersize = (context.DeserialPacketSize < context.BufferSize) ? (int)context.DeserialPacketSize : context.BufferSize;
context.Listener.BeginReceive(context.HeaderBuffer, 0, buffersize, SocketFlags.None, HeaderReceivedCallback, context);
}
else
{
TransferOperation request = new TransferOperation(context.Transfer, MessagePart.Header, DirectionType.Receive);
request.Resolve(context.DeserialPacket);
context.HeaderReceivedNotice.Set();
try
{
HeaderReceived.Execute(context);
}
catch (Exception ex)
{
Echo(ex.Message);
CloseClient(context.Id);
}
}
}
public void MessageReceivedCallback(IAsyncResult result)
{
ITransferContext context = (ITransferContext)result.AsyncState;
MarkupType noiseKind = MarkupType.None;
int receive = context.Listener.EndReceive(result);
if (receive > 0)
{
noiseKind = context.IncomingMessage(receive);
}
if (context.DeserialPacketSize > 0)
{
int buffersize = (context.DeserialPacketSize < context.BufferSize) ? (int)context.DeserialPacketSize : context.BufferSize;
context.Listener.BeginReceive(context.MessageBuffer, 0, buffersize, SocketFlags.None, MessageReceivedCallback, context);
}
else
{
object received = context.DeserialPacket;
object readPosition = context.DeserialPacketId;
if (noiseKind == MarkupType.Block || (noiseKind == MarkupType.End && (int)readPosition < (context.Transfer.HeaderReceived.Context.ObjectsCount - 1)))
{
context.Listener.BeginReceive(context.MessageBuffer, 0, context.BufferSize, SocketFlags.None, MessageReceivedCallback, context);
}
TransferOperation request = new TransferOperation(context.Transfer, MessagePart.Message, DirectionType.Receive);
request.Resolve(received, readPosition);
if (context.ObjectsLeft <= 0 && !context.BatchesReceivedNotice.SafeWaitHandle.IsClosed)
{
context.BatchesReceivedNotice.Set();
}
if (noiseKind == MarkupType.End && (int)readPosition >= (context.Transfer.HeaderReceived.Context.ObjectsCount - 1))
{
context.BatchesReceivedNotice.WaitOne();
context.MessageReceivedNotice.Set();
try
{
MessageReceived.Execute(context);
}
catch (Exception ex)
{
Echo(ex.Message);
CloseClient(context.Id);
}
}
}
}
public void MessageSentCallback(IAsyncResult result)
{
ITransferContext context = (ITransferContext)result.AsyncState;
try
{
int sendcount = context.Listener.EndSend(result);
}
catch (SocketException) { }
catch (ObjectDisposedException) { }
if (context.SerialPacketId >= 0 || context.ObjectPosition < (context.Transfer.MyHeader.Context.ObjectsCount - 1))
{
TransferOperation request = new TransferOperation(context.Transfer, MessagePart.Message, DirectionType.Send);
request.Resolve();
context.Listener.BeginSend(context.SerialPacket, 0, context.SerialPacket.Length, SocketFlags.None, MessageSentCallback, context);
}
else
{
if (context.ReceiveMessage)
{
context.MessageReceivedNotice.WaitOne();
}
//int _timeout = 0;
// while (IsConnected(context.Id) && timeout < 10) _timeout++;
context.Close = true;
context.MessageSentNotice.Set();
try
{
MessageSent.Execute(context);
}
catch (Exception ex)
{
Echo(ex.Message);
CloseClient(context.Id);
}
}
}
