Limit czasu gniazda asynchronicznego .net sprawdź bezpieczeństwo wątku

http://msdn.microsoft.com/en-us/library/system.net.sockets.socketasynceventargs.aspx

Począwszy od powyższego przykładu msdn, próbuję napisać test limitu czasu, który zamknie nieaktywne gniazda klienta i zwolni zasoby.

Właśnie to wymyśliłem. Ale nie jestem pewien, czy jest całkowicie bezpieczny dla wątków i czy istnieje lepszy sposób, aby to zrobić. Mam nadzieję, że ktoś może udzielić kilku rad.

void IO_Completed(object sender, SocketAsyncEventArgs e)
{
    if (e.SocketError != SocketError.Success)
    {
        CloseClientSocket(e);
        return;
    }

    if (1 < Interlocked.CompareExchange(ref token.Status, 1, 0))
        return;

    switch (e.LastOperation)
    {
        case SocketAsyncOperation.Receive:
            ProcessReceive(e);
            break;
        ...
    }

    token.LastActive = Environment.TickCount;
    Interlocked.CompareExchange(ref token.Status, 0, 1);
}

void ProcessReceive(SocketAsyncEventArgs e)
{
    AsyncUserToken token = (AsyncUserToken)e.UserToken;
    if (e.BytesTransferred > 0)
        if (!token.Socket.SendAsync(e))
            ProcessSend(e);
    else
        CloseClientSocket(e);
}

void ProcessSend(SocketAsyncEventArgs e)
{
    AsyncUserToken token = (AsyncUserToken)e.UserToken;
    if (!token.Socket.ReceiveAsync(e))
        ProcessReceive(e);
}

TimeoutCheck zostanie wykonany raz na 20 sekund. allReadWriteArgs jest tablicą zawierającą wszystkie SocketAsyncEventArgs. Po zamknięciu gniazda IO_Completed zostanie wywołany z SocketError.OperationAborted.

void TimeoutCheck(object state)
{
    AsyncUserToken token;
    int timeout = Environment.TickCount - 20000;
    for (int i = 0; i < allReadWriteArgs.Length; i++)
    {
        token = (AsyncUserToken)allReadWriteArgs[i].UserToken;
        if (token.LastActive < timeout)
            if (0 == Interlocked.CompareExchange(ref token.Status, 2, 0))
                Interlocked.Exchange(ref token.Socket, null).Close();
    }
}


void CloseClientSocket(SocketAsyncEventArgs e)
{
    AsyncUserToken token = e.UserToken as AsyncUserToken;

    if (token.Socket != null)
    {
        try
        {
            token.Socket.Shutdown(SocketShutdown.Both);
        }
        catch (SocketException) { }
        token.Socket.Close();
    }

    token.Status = 2;
    bufferManager.FreeBuffer(e);
    readWritePool.Push(e);
    ...
}

questionAnswers(1)

yourAnswerToTheQuestion