echo.c File Reference

#include "tcpsvcs.h"

Include dependency graph for echo.c:

Go to the source code of this file.

Macros
#define	RECV_BUF   1024

Functions
static BOOL	EchoIncomingPackets (SOCKET sock)
DWORD WINAPI	EchoHandler (VOID *sock_)

Macro Definition Documentation

RECV_BUF

#define RECV_BUF   1024

Definition at line 12 of file echo.c.

Referenced by EchoIncomingPackets().

Function Documentation

EchoHandler()

DWORD WINAPI EchoHandler

(

VOID *

sock_

)

Definition at line 69 of file echo.c.

{
    DWORD retVal = 0;
    SOCKET sock = (SOCKET)sock_;

    if (!EchoIncomingPackets(sock))
    {
        LogEvent(L"Echo: EchoIncomingPackets failed", 0, 0, LOG_FILE);
        retVal = 1;
    }

    LogEvent(L"Echo: Shutting connection down", 0, 0, LOG_FILE);

    if (ShutdownConnection(sock, TRUE))
    {
        LogEvent(L"Echo: Connection is down", 0, 0, LOG_FILE);
    }
    else
    {
        LogEvent(L"Echo: Connection shutdown failed", 0, 0, LOG_FILE);
        retVal = 1;
    }

    LogEvent(L"Echo: Terminating thread", 0, 0, LOG_FILE);
    ExitThread(retVal);
}

EchoIncomingPackets()

static BOOL EchoIncomingPackets ( SOCKET  sock )

static

Definition at line 15 of file echo.c.

Referenced by EchoHandler().

{
    CHAR readBuffer[RECV_BUF];
    WCHAR logBuf[256];
    INT totalSentBytes;
    INT readBytes;
    INT retVal;

    do 
    {
        readBytes = recv(sock, readBuffer, RECV_BUF, 0);
        if (readBytes > 0)
        {
            swprintf(logBuf, L"Received %d bytes from client", readBytes);
            LogEvent(logBuf, 0, 0, LOG_FILE);

            totalSentBytes = 0;
            while (!bShutdown && totalSentBytes < readBytes)
            {
                retVal = send(sock, readBuffer + totalSentBytes, readBytes - totalSentBytes, 0);
                if (retVal > 0)
                {
                    swprintf(logBuf, L"Sent %d bytes back to client", retVal);
                    LogEvent(logBuf, 0, 0, LOG_FILE);
                    totalSentBytes += retVal;
                }
                else if (retVal == SOCKET_ERROR)
                {
                    LogEvent(L"Echo: socket error", WSAGetLastError(), 0, LOG_ERROR);
                    return FALSE;
                }
                else
                {
                    /* Client closed connection before we could reply to
                       all the data it sent, so quit early. */
                    LogEvent(L"Peer unexpectedly dropped connection!", 0, 0, LOG_FILE);
                    return FALSE;
                }
            }
        }
        else if (readBytes == SOCKET_ERROR)
        {
            LogEvent(L"Echo: socket error", WSAGetLastError(), 0, LOG_ERROR);
            return FALSE;
        }
    } while ((readBytes != 0) && (!bShutdown));

    if (!bShutdown)
        LogEvent(L"Echo: Connection closed by peer", 0, 0, LOG_FILE);

    return TRUE;
}