- Don't call winsock initialiser for each seperate server thread
authorGed Murphy <gedmurphy@reactos.org>
Mon, 3 Oct 2005 18:32:30 +0000 (18:32 +0000)
committerGed Murphy <gedmurphy@reactos.org>
Mon, 3 Oct 2005 18:32:30 +0000 (18:32 +0000)
- Add more code for tcpsvcs to be controled via services. Still #if 0'd out until fully implemented.
- Make server more informative with what is happening. (this info will be output to a log file once implemented as a service.

svn path=/trunk/; revision=18248

reactos/apps/utils/net/tcpsvcs/chargen.c
reactos/apps/utils/net/tcpsvcs/daytime.c
reactos/apps/utils/net/tcpsvcs/discard.c
reactos/apps/utils/net/tcpsvcs/echo.c
reactos/apps/utils/net/tcpsvcs/qotd.c
reactos/apps/utils/net/tcpsvcs/skelserver.c
reactos/apps/utils/net/tcpsvcs/tcpsvcs.c
reactos/apps/utils/net/tcpsvcs/tcpsvcs.h

index 8ce5825..4b9429e 100644 (file)
@@ -22,7 +22,7 @@ DWORD WINAPI ChargenHandler(VOID* Sock_)
         _tprintf(_T("Connection shutdown failed\n"));
         Retval = 3;
     }
-    _tprintf(_T("Terminating thread\n"));
+    _tprintf(_T("Terminating chargen thread\n"));
     ExitThread(0);
 
     return Retval;
index 271afdf..83f7a7e 100644 (file)
@@ -26,7 +26,7 @@ DWORD WINAPI DaytimeHandler(VOID* Sock_)
         _tprintf(_T("Connection shutdown failed\n"));
         Retval = 3;
     }
-    _tprintf(_T("Terminating thread\n"));
+    _tprintf(_T("Terminating daytime thread\n"));
     ExitThread(0);
 
     return Retval;
index aa2ff5f..aab07f6 100644 (file)
@@ -24,7 +24,7 @@ DWORD WINAPI DiscardHandler(VOID* Sock_)
         _tprintf(_T("Connection shutdown failed\n"));\r
         Retval = 3;\r
     }\r
-    _tprintf(_T("Terminating thread\n"));\r
+    _tprintf(_T("Terminating discard thread\n"));\r
     ExitThread(0);\r
 \r
     return Retval;\r
index 1acc731..3ec8e2b 100644 (file)
@@ -22,7 +22,7 @@ DWORD WINAPI EchoHandler(VOID* Sock_)
         _tprintf(_T("Connection shutdown failed\n"));
         Retval = 3;
     }
-    _tprintf(_T("Terminating thread\n"));
+    _tprintf(_T("Terminating echo thread\n"));
     ExitThread(0);
 
     return Retval;
index 3c45a56..81484e4 100644 (file)
@@ -33,7 +33,7 @@ DWORD WINAPI QotdHandler(VOID* Sock_)
         _tprintf(_T("Connection shutdown failed\n"));
         Retval = 3;
     }
-    _tprintf(_T("Terminating thread\n"));
+    _tprintf(_T("Terminating qotd thread\n"));
     ExitThread(0);
 
     return Retval;
index cfce714..14fbdf1 100644 (file)
@@ -7,18 +7,10 @@
 DWORD WINAPI StartServer(LPVOID lpParam)
 {
     const TCHAR* HostIP = "127.0.0.1";
-    DWORD RetVal;
-    WSADATA wsaData;
     PSERVICES pServices;
 
     pServices = (PSERVICES)lpParam;
 
-    if ((RetVal = WSAStartup(MAKEWORD(2, 2), &wsaData)) != 0)
-    {
-        _tprintf(_T("WSAStartup() failed : %lu\n"), RetVal);
-        return -1;
-    }
-
     SOCKET ListeningSocket = SetUpListener(HostIP, htons(pServices->Port));
     if (ListeningSocket == INVALID_SOCKET)
     {
@@ -26,10 +18,11 @@ DWORD WINAPI StartServer(LPVOID lpParam)
         return 3;
     }
 
-    _tprintf(_T("Waiting for connections...\n"));
+    _tprintf(_T("%s is waiting for connections on port %d...\n"),
+        pServices->Name, pServices->Port);
     while (1)
     {
-        AcceptConnections(ListeningSocket, pServices->Service);
+        AcceptConnections(ListeningSocket, pServices->Service, pServices->Name);
         printf("Acceptor restarting...\n");
     }
 
@@ -66,7 +59,8 @@ SOCKET SetUpListener(const char* ServAddr, int Port)
 
 
 
-VOID AcceptConnections(SOCKET ListeningSocket, LPTHREAD_START_ROUTINE Service)
+VOID AcceptConnections(SOCKET ListeningSocket,
+    LPTHREAD_START_ROUTINE Service, TCHAR *Name)
 {
     SOCKADDR_IN Client;
     SOCKET Sock;
@@ -78,9 +72,9 @@ VOID AcceptConnections(SOCKET ListeningSocket, LPTHREAD_START_ROUTINE Service)
         Sock = accept(ListeningSocket, (SOCKADDR*)&Client, &nAddrSize);
         if (Sock != INVALID_SOCKET)
         {
-            _tprintf(_T("Accepted connection from %s:%d\n"),
-                inet_ntoa(Client.sin_addr), ntohs(Client.sin_port));
-            _tprintf(_T("About to create thread\n"));
+            _tprintf(_T("Accepted connection to %s server from %s:%d\n"),
+                Name, inet_ntoa(Client.sin_addr), ntohs(Client.sin_port));
+            _tprintf(_T("Creating new thread for %s\n"), Name);
             CreateThread(0, 0, Service, (void*)Sock, 0, &ThreadID);
         }
         else
index 3db7aad..1d5cc92 100644 (file)
@@ -3,68 +3,67 @@
 #include <tchar.h>
 #include "tcpsvcs.h"
 
-static
-LPTHREAD_START_ROUTINE
-ServiceHandler[NUM_SERVICES] =
-{
-    EchoHandler,
-    DiscardHandler,
-    DaytimeHandler,
-    QotdHandler,
-    ChargenHandler
-};
-
-static int
-ServicePort[NUM_SERVICES] =
-{
-    ECHO_PORT,
-    DISCARD_PORT,
-    DAYTIME_PORT,
-    QOTD_PORT,
-    CHARGEN_PORT
-};
-
 #if 0
-static
-SERVICE_TABLE_ENTRY
+/*
+ * globals
+ */
+static SERVICE_STATUS hServStatus;
+static SERVICE_STATUS_HANDLE hSStat;
+FILE *hLogFile;
+BOOL bLogEvents = TRUE;
+BOOL ShutDown, PauseFlag;
+LPTSTR LogFileName = "tcpsvcs_log.txt";
+
+static SERVICE_TABLE_ENTRY
 ServiceTable[2] =
 {
-    {TEXT("tcpsvcs"), ServiceMain},
+    {_T("tcpsvcs"), ServiceMain},
     {NULL, NULL}
 };
 #endif
 
+static SERVICES
+Services[NUM_SERVICES] =
+{
+    {ECHO_PORT, _T("Echo"), EchoHandler},
+    {DISCARD_PORT, _T("Discard"), DiscardHandler},
+    {DAYTIME_PORT, _T("Daytime"), DaytimeHandler},
+    {QOTD_PORT, _T("QOTD"), QotdHandler},
+    {CHARGEN_PORT, _T("Chargen"), ChargenHandler}
+};
+
+
 int main(void)
 {
-    PSERVICES pServices[NUM_SERVICES];
     DWORD dwThreadId[NUM_SERVICES];
     HANDLE hThread[NUM_SERVICES];
+    WSADATA wsaData;
+    DWORD RetVal;
     INT i;
 
+    if ((RetVal = WSAStartup(MAKEWORD(2, 2), &wsaData)) != 0)
+    {
+        _tprintf(_T("WSAStartup() failed : %lu\n"), RetVal);
+        return -1;
+    }
+
     /* Create MAX_THREADS worker threads. */
     for( i=0; i<NUM_SERVICES; i++ )
     {
-        /* Allocate memory for thread data. */
-        pServices[i] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(SERVICES));
-
-        if( pServices[i] == NULL )
-            ExitProcess(2);
-
-        /* Generate unique data for each thread. */
-        pServices[i]->Service = ServiceHandler[i];
-        pServices[i]->Port = ServicePort[i];
+        _tprintf(_T("Starting %s server....\n"), Services[i].Name);
 
         hThread[i] = CreateThread(
             NULL,              // default security attributes
             0,                 // use default stack size
             StartServer,       // thread function
-            pServices[i],      // argument to thread function
+            &Services[i],     // argument to thread function
             0,                 // use default creation flags
             &dwThreadId[i]);   // returns the thread identifier
 
         /* Check the return value for success. */
         if (hThread[i] == NULL)
         {
+            _tprintf(_T("Failed to start %s server....\n"), Services[i].Name);
             ExitProcess(i);
         }
     }
@@ -80,39 +79,134 @@ int main(void)
     return 0;
 }
 
+
+
+/* code to run tcpsvcs as a service */
 #if 0
-static VOID CALLBACK
-ServiceMain(DWORD argc, LPTSTR *argv)
+int
+main(int argc, char *argv[])
+{
+    //DPRINT("tcpsvcs: main() started. See tcpsvcs_log.txt for info\n");
+
+    if (!StartServiceCtrlDispatcher(ServiceTable))
+        _tprintf(_T("failed to start the service control dispatcher\n"));
+
+    //DPRINT("tcpsvcs: main() done\n");
+
+    return 0;
+}
+
+
+static VOID WINAPI
+ServiceMain(DWORD argc, LPTSTR argv[])
+{
+    DWORD i;
+
+    hLogFile = fopen(LogFileName, _T("w+"));
+    if (hLogFile == NULL)
+        return;
+
+    LogEvent(_T("Entering ServiceMain"), 0, FALSE);
+
+    hServStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
+    hServStatus.dwCurrentState = SERVICE_START_PENDING;
+    hServStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP |
+        SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_PAUSE_CONTINUE;
+    hServStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
+    hServStatus.dwServiceSpecificExitCode = 0;
+    hServStatus.dwCheckPoint = 0;
+    hServStatus.dwWaitHint = 2*CS_TIMEOUT;
+
+    hSStat = RegisterServiceCtrlHandler("tcpsvcs", ServerCtrlHandler);
+    if (hSStat == 0)
+        LogEvent(_T("Failed to register service\n"), 100, TRUE);
+
+    LogEvent(_T("Control handler registered successfully"), 0, FALSE);
+    SetServiceStatus (hSStat, &hServStatus);
+    LogEvent(_T("Service status set to SERVICE_START_PENDING"), 0, FALSE);
+
+    if (CreateServers() != 0)
+    {
+        hServStatus.dwCurrentState = SERVICE_STOPPED;
+        hServStatus.dwServiceSpecificExitCode = 1;
+        SetServiceStatus(hSStat, &hServStatus);
+        return;
+    }
+
+    LogEvent(_T("Service threads shut down. Set SERVICE_STOPPED status"), 0, FALSE);
+    /*  We will only return here when the ServiceSpecific function
+        completes, indicating system shutdown. */
+    UpdateStatus (SERVICE_STOPPED, 0);
+    LogEvent(_T("Service status set to SERVICE_STOPPED"), 0, FALSE);
+    fclose(hLogFile);  /*  Clean up everything, in general */
+    return;
+
+}
+
+VOID WINAPI
+ServerCtrlHandler(DWORD Control)
+{
+    switch (Control)
+    {
+        case SERVICE_CONTROL_SHUTDOWN: /* fall through */
+        case SERVICE_CONTROL_STOP:
+            ShutDown = TRUE;
+            UpdateStatus(SERVICE_STOP_PENDING, -1);
+            break;
+        case SERVICE_CONTROL_PAUSE:
+            PauseFlag = TRUE;
+            break;
+        case SERVICE_CONTROL_CONTINUE:
+            PauseFlag = FALSE;
+            break;
+        case SERVICE_CONTROL_INTERROGATE:
+            break;
+        default:
+            if (Control > 127 && Control < 256) /* user defined */
+            break;
+    }
+    UpdateStatus(-1, -1); /* increment checkpoint */
+    return;
+}
+
+
+void UpdateStatus (int NewStatus, int Check)
+/*  Set a new service status and checkpoint (either specific value or increment) */
+{
+    if (Check < 0 ) hServStatus.dwCheckPoint++;
+    else            hServStatus.dwCheckPoint = Check;
+    if (NewStatus >= 0) hServStatus.dwCurrentState = NewStatus;
+    if (!SetServiceStatus (hSStat, &hServStatus))
+        LogEvent (_T("Cannot set service status"), 101, TRUE);
+    return;
+}
+
+INT
+CreateServers()
 {
-    PSERVICES pServices[NUM_SERVICES];
     DWORD dwThreadId[NUM_SERVICES];
     HANDLE hThread[NUM_SERVICES];
     INT i;
 
+    UpdateStatus(-1, -1); /* increment checkpoint */
+
     /* Create MAX_THREADS worker threads. */
     for( i=0; i<NUM_SERVICES; i++ )
     {
-        /* Allocate memory for thread data. */
-        pServices[i] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(SERVICES));
-
-        if( pServices[i] == NULL )
-            ExitProcess(2);
-
-        /* Generate unique data for each thread. */
-        pServices[i]->Service = ServiceHandler[i];
-        pServices[i]->Port = ServicePort[i];
+        _tprintf(_T("Starting %s server....\n"), Services[i].Name);
 
         hThread[i] = CreateThread(
             NULL,              // default security attributes
             0,                 // use default stack size
             StartServer,       // thread function
-            pServices[i],      // argument to thread function
+            &Services[i],     // argument to thread function
             0,                 // use default creation flags
             &dwThreadId[i]);   // returns the thread identifier
 
         /* Check the return value for success. */
         if (hThread[i] == NULL)
         {
+            _tprintf(_T("Failed to start %s server....\n"), Services[i].Name);
             ExitProcess(i);
         }
     }
@@ -125,20 +219,42 @@ ServiceMain(DWORD argc, LPTSTR *argv)
     {
         CloseHandle(hThread[i]);
     }
-
+    return 0;
 }
 
-int
-main(int argc, char *argv[])
+
+/*  LogEvent is similar to the ReportError function used elsewhere
+    For a service, however, we ReportEvent rather than write to standard
+    error. Eventually, this function should go into the utility
+    library.  */
+VOID
+LogEvent (LPCTSTR UserMessage, DWORD ExitCode, BOOL PrintErrorMsg)
 {
-    DPRINT("tcpsvcs: main() started\n");
+    DWORD eMsgLen, ErrNum = GetLastError ();
+    LPTSTR lpvSysMsg;
+    TCHAR MessageBuffer[512];
 
-    StartServiceCtrlDispatcher(ServiceTable);
+    if (PrintErrorMsg) {
+        eMsgLen = FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER |
+            FORMAT_MESSAGE_FROM_SYSTEM, NULL,
+            ErrNum, MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT),
+            (LPTSTR)&lpvSysMsg, 0, NULL);
 
-    DPRINT("Umpnpmgr: main() done\n");
+        _stprintf (MessageBuffer, _T("\n%s %s ErrNum = %d. ExitCode = %d."),
+            UserMessage, lpvSysMsg, ErrNum, ExitCode);
+        HeapFree (GetProcessHeap (), 0, lpvSysMsg);
+                /* Explained in Chapter 6. */
+    } else {
+        _stprintf (MessageBuffer, _T("\n%s ExitCode = %d."),
+            UserMessage, ExitCode);
+    }
 
-    ExitThread(0);
+    fputs (MessageBuffer, hLogFile);
 
-    return 0;
+    if (ExitCode > 0)
+        ExitProcess (ExitCode);
+    else
+        return;
 }
+
 #endif
index 3e8121f..3a17cb6 100644 (file)
@@ -5,9 +5,10 @@
 #define QOTD_PORT 17
 #define CHARGEN_PORT 19
 
-#define NUM_SERVICES 6
+#define NUM_SERVICES 5
 #define BUF_SIZE 255
 #define BUF 1024
+#define CS_TIMEOUT 1000
 
 /* RFC865 states no more than 512 chars per line */
 #define MAX_QUOTE_BUF 512
 /* data structure to pass to threads */
 typedef struct _Services {
     INT Port;
+    TCHAR *Name;
     LPTHREAD_START_ROUTINE Service;
 } SERVICES, *PSERVICES;
 
-/* tcpsvcs functions */
-//static VOID CALLBACK ServiceMain(DWORD argc, LPTSTR *argv);
+
+/* tcpsvcs functions * /
+static VOID WINAPI ServiceMain(DWORD argc, LPTSTR argv[]);
+VOID WINAPI ServerCtrlHandler(DWORD control);
+INT CreateServers(VOID);
+VOID LogEvent (LPCTSTR UserMessage, DWORD ExitCode, BOOL PrintErrorMsg);
+void UpdateStatus (int NewStatus, int Check);
+*/
 
 /* skelserver functions */
 DWORD WINAPI StartServer(LPVOID lpParam);
 SOCKET SetUpListener(const char* ServAddr, int Port);
-VOID AcceptConnections(SOCKET ListeningSocket, LPTHREAD_START_ROUTINE Service);
+VOID AcceptConnections(SOCKET ListeningSocket,
+    LPTHREAD_START_ROUTINE Service, TCHAR *Name);
 BOOL EchoIncomingPackets(SOCKET sd);
 BOOL ShutdownConnection(SOCKET Sock, BOOL bRec);