private void HandleCompletionStatus(out CompletionStatus completionStatus, IntPtr overlappedAddress, IntPtr completionKey, int bytesTransferred)
{
if (completionKey.Equals(SignalPostCompletionKey))
{
object state;
m_signalQueue.TryDequeue(out state);
completionStatus = new CompletionStatus(null, state, OperationType.Signal, SocketError.Success, 0);
}
else
{
var overlapped = Overlapped.CompleteOperation(overlappedAddress);
if (completionKey.Equals(SocketCompletionKey))
{
// if the overlapped ntstatus is zero we assume success and don't call get overlapped result for optimization
if (overlapped.Success)
{
SocketError socketError = SocketError.Success;
try
{
if (overlapped.OperationType == OperationType.Accept)
{
overlapped.AsyncSocket.UpdateAccept();
}
else if (overlapped.OperationType == OperationType.Connect)
{
overlapped.AsyncSocket.UpdateConnect();
}
}
catch (SocketException ex)
{
socketError = ex.SocketErrorCode;
}
completionStatus = new CompletionStatus(overlapped.AsyncSocket, overlapped.State,
overlapped.OperationType, socketError,
bytesTransferred);
}
else
{
SocketError socketError = SocketError.Success;
SocketFlags socketFlags;
bool operationSucceed = UnsafeMethods.WSAGetOverlappedResult(overlapped.AsyncSocket.Handle,
overlappedAddress,
out bytesTransferred, false, out socketFlags);
if (!operationSucceed)
{
socketError = (SocketError)Marshal.GetLastWin32Error();
}
completionStatus = new CompletionStatus(overlapped.AsyncSocket, overlapped.State,
overlapped.OperationType, socketError,
bytesTransferred);
}
}
else
{
completionStatus = new CompletionStatus(overlapped.AsyncSocket, overlapped.State,
overlapped.OperationType, SocketError.Success, 0);
}
}
}
