home *** CD-ROM | disk | FTP | other *** search
/ DEFCON 15 / DefCon15.bin / Speakers / Jennings / Extras / incognito / incognito_service.c < prev    next >
C/C++ Source or Header  |  2007-03-12  |  4KB  |  137 lines
  1. #define _CRT_SECURE_NO_DEPRECATE 1
  2. #include <windows.h>
  3. #include <stdio.h>
  4. #include "handle_arguments.h"
  5.  
  6. #define BUFFER_SIZE        500
  7.  
  8. SERVICE_STATUS ServiceStatus; 
  9. SERVICE_STATUS_HANDLE hStatus; 
  10.  
  11. void ServiceMain(int argc, char** argv); 
  12. void ControlHandler(DWORD request); 
  13. BOOL InitializePipe(LPCTSTR lpszPipe);
  14.  
  15. HANDLE hPipeR, hPipeW;
  16.  
  17. void main() 
  18. { 
  19.     SERVICE_TABLE_ENTRY ServiceTable[2];
  20.     ServiceTable[0].lpServiceName = "incognito_service";
  21.     ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;
  22.  
  23.     ServiceTable[1].lpServiceName = NULL;
  24.     ServiceTable[1].lpServiceProc = NULL;
  25.  
  26.     // Start the control dispatcher thread for our service
  27.     StartServiceCtrlDispatcher(ServiceTable);  
  28. }
  29.  
  30. void ServiceMain(int argc, char** argv) 
  31. { 
  32.     ServiceStatus.dwServiceType        = SERVICE_WIN32; 
  33.     ServiceStatus.dwCurrentState       = SERVICE_START_PENDING; 
  34.     ServiceStatus.dwControlsAccepted   = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
  35.     ServiceStatus.dwWin32ExitCode      = 0; 
  36.     ServiceStatus.dwServiceSpecificExitCode = 0; 
  37.     ServiceStatus.dwCheckPoint         = 0; 
  38.     ServiceStatus.dwWaitHint           = 0; 
  39.  
  40.     hStatus = RegisterServiceCtrlHandler(
  41.         "incognito_service", 
  42.         (LPHANDLER_FUNCTION)ControlHandler); 
  43.  
  44.     if (hStatus == (SERVICE_STATUS_HANDLE)0) 
  45.     { 
  46.         // Registering Control Handler failed
  47.         return; 
  48.     }  
  49.  
  50.     // We report the running status to SCM. 
  51.     ServiceStatus.dwCurrentState = SERVICE_RUNNING; 
  52.     SetServiceStatus (hStatus, &ServiceStatus);
  53.  
  54.     InitializePipe(argv[1]);
  55.     Sleep(100);
  56.  
  57.     OUTPUT = hPipeW;
  58.     INPUT = hPipeR;
  59.  
  60.     handle_options(argc-1, ++argv);
  61.  
  62.     ServiceStatus.dwCurrentState = SERVICE_STOPPED; 
  63.     SetServiceStatus (hStatus, &ServiceStatus);
  64.  
  65.     Sleep(100);
  66.     return; 
  67. }
  68.  
  69. // Control handler function
  70. void ControlHandler(DWORD request) 
  71. { 
  72.     switch(request) 
  73.     { 
  74.         case SERVICE_CONTROL_STOP: 
  75.             ServiceStatus.dwWin32ExitCode = 0; 
  76.             ServiceStatus.dwCurrentState  = SERVICE_STOPPED; 
  77.             SetServiceStatus (hStatus, &ServiceStatus);
  78.             return; 
  79.  
  80.         case SERVICE_CONTROL_SHUTDOWN: 
  81.             ServiceStatus.dwWin32ExitCode = 0; 
  82.             ServiceStatus.dwCurrentState  = SERVICE_STOPPED; 
  83.             SetServiceStatus (hStatus, &ServiceStatus);
  84.             return; 
  85.         
  86.         default:
  87.             break;
  88.     } 
  89.  
  90.     // Report current status
  91.     SetServiceStatus (hStatus,  &ServiceStatus);
  92.  
  93.     return; 
  94. } 
  95.  
  96. BOOL InitializePipe(LPCTSTR lpszPipe)
  97. {
  98.     char szPipeName[MAX_PATH];
  99.  
  100.     memset(szPipeName, 0, MAX_PATH);
  101.     if (_snprintf(szPipeName, MAX_PATH, "\\\\.\\pipe\\%s", lpszPipe) == -1)
  102.         szPipeName[sizeof(szPipeName)-1] = '\0';
  103.     
  104.     hPipeW = CreateNamedPipeA(szPipeName, 
  105.                             PIPE_ACCESS_DUPLEX, // read/write access 
  106.                             PIPE_TYPE_MESSAGE | // message type pipe 
  107.                             PIPE_READMODE_MESSAGE | // message-read mode 
  108.                             PIPE_WAIT, // blocking mode 
  109.                             PIPE_UNLIMITED_INSTANCES, // max. instances 
  110.                             BUFFER_SIZE, // output buffer size 
  111.                             BUFFER_SIZE, // input buffer size 
  112.                             10000, // client time-out 
  113.                             NULL); // no security attribute 
  114.  
  115.     ConnectNamedPipe(hPipeW, NULL);
  116.  
  117.     hPipeR = CreateNamedPipeA(szPipeName, 
  118.                             PIPE_ACCESS_DUPLEX, // read/write access 
  119.                             PIPE_TYPE_MESSAGE | // message type pipe 
  120.                             PIPE_READMODE_MESSAGE | // message-read mode 
  121.                             PIPE_WAIT, // blocking mode 
  122.                             PIPE_UNLIMITED_INSTANCES, // max. instances 
  123.                             BUFFER_SIZE, // output buffer size 
  124.                             BUFFER_SIZE, // input buffer size 
  125.                             10000, // client time-out 
  126.                             NULL); // no security attribute 
  127.  
  128.     ConnectNamedPipe(hPipeR, NULL);
  129.  
  130.     if (hPipeR == INVALID_HANDLE_VALUE || hPipeW == INVALID_HANDLE_VALUE) 
  131.     {
  132.         return FALSE;
  133.     }
  134.     
  135.     return TRUE;
  136. }
  137.