home *** CD-ROM | disk | FTP | other *** search
/ PC World 2004 November / PCWorld_2004-11_cd.bin / software / temacd / poweroff / pwroff30.exe / src / service.c < prev    next >
C/C++ Source or Header  |  2002-02-16  |  12KB  |  421 lines
  1. #include "poweroff.h"
  2.  
  3. SERVICE_STATUS_HANDLE sh;
  4. SERVICE_STATUS status;
  5.  
  6. int CopyPoweroffToSystem(HWND hWnd,PowerSettings *ps,char *system_poweroff)
  7. {
  8.   char system_dir[MAX_PATH],path[1024]="",original_poweroff[MAX_PATH];
  9.   char *pos,*lastpos;
  10.   HANDLE f=INVALID_HANDLE_VALUE;
  11.  
  12.   Log("CopyPoweroffToSystem start, executable=%s, system_poweroff=%s",ps->executable,system_poweroff);
  13.   if (strlen(ps->executable)<4 || _stricmp(&ps->executable[strlen(ps->executable)-4],".exe"))
  14.     strcat(ps->executable,".exe");
  15.  
  16.   if (!strchr(ps->executable,'\\'))
  17.   {
  18.     strcpy(original_poweroff,ps->executable);
  19.     if ((f=CreateFile(ps->executable,
  20.                       0,
  21.                       FILE_SHARE_READ,
  22.                       NULL,
  23.                       OPEN_EXISTING,
  24.                       0,
  25.                       NULL))==INVALID_HANDLE_VALUE)
  26.     {
  27.       GetEnvironmentVariable("PATH",path,1024);
  28.       lastpos=path;
  29.       pos=strchr(lastpos,';');
  30.       while (pos)
  31.       {
  32.         *pos='\0';
  33.         sprintf(original_poweroff,"%s\\%s",lastpos,ps->executable);
  34.         if ((f=CreateFile(original_poweroff,
  35.                           0,
  36.                           FILE_SHARE_READ,
  37.                           NULL,
  38.                           OPEN_EXISTING,
  39.                           0,
  40.                           NULL))!=INVALID_HANDLE_VALUE)
  41.           break;
  42.         lastpos=pos+1;
  43.         pos=strchr(lastpos,';');
  44.       }
  45.       if (f==INVALID_HANDLE_VALUE)
  46.       {
  47.         sprintf(original_poweroff,"%s\\%s",lastpos,ps->executable);
  48.         f=CreateFile(original_poweroff,
  49.                      0,
  50.                      FILE_SHARE_READ,
  51.                      NULL,
  52.                      OPEN_EXISTING,
  53.                      0,
  54.                      NULL);
  55.       }
  56.     }
  57.   }
  58.   else
  59.   {
  60.     strcpy(original_poweroff,ps->executable);
  61.     f=CreateFile(original_poweroff,
  62.                  0,
  63.                  FILE_SHARE_READ,
  64.                  NULL,
  65.                  OPEN_EXISTING,
  66.                  0,
  67.                  NULL);
  68.   }
  69.   if (f==INVALID_HANDLE_VALUE)
  70.   {
  71.     Error(ps,NULL,"Can't find poweroff.exe!");
  72.     return 0;
  73.   }
  74.   CloseHandle(f);
  75.   
  76.   if (GetSystemDirectory(system_dir,MAX_PATH)==0)
  77.   {
  78.     DisplayLastError(ps,hWnd);
  79.     return 0;
  80.   }
  81.   sprintf(system_poweroff,"%s\\poweroff.exe",system_dir);
  82.  
  83.   if (_stricmp(original_poweroff,system_poweroff))
  84.   {
  85.     Log("Copying %s to %s",original_poweroff,system_poweroff);
  86.     if (CopyFile(original_poweroff,system_poweroff,FALSE)==0)
  87.     {
  88.       DisplayLastError(ps,hWnd);
  89.       CloseHandle(f);
  90.       return 0;
  91.     }
  92.   }
  93.   sprintf(path,"\"%s\" -service",system_poweroff);
  94.   strcpy(system_poweroff,path);
  95.   Log("CopyPoweroffToSystem end");
  96.   return 1;
  97. }
  98.  
  99. void CreatePoweroffService(HWND hWnd,PowerSettings *ps)
  100. {
  101.   char system_poweroff[MAX_PATH];
  102.   SC_HANDLE hsm,sh;
  103.  
  104.   Log("CreatePoweroffService start");
  105.   if (!CopyPoweroffToSystem(hWnd,ps,system_poweroff))
  106.     return;
  107.   SaveSettings(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\Services\\Poweroff\\Parameters",ps);
  108.  
  109.   if (ps->windowsversion.dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
  110.   {
  111.     Log("Creating WIN98 service");
  112.     if (!WriteRegistryString(HKEY_LOCAL_MACHINE,"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunServices","Poweroff",system_poweroff))
  113.     {
  114.       DisplayLastError(ps,hWnd);
  115.       return;
  116.     }
  117.     DisplayInformation(ps,hWnd,"Please restart your computer to start the service");
  118.   }
  119.   else /* Create a real service */
  120.   {
  121.     Log("Creating real NT service");
  122.     hsm=OpenSCManager(NULL,NULL,SC_MANAGER_CREATE_SERVICE);
  123.     if (hsm==NULL)
  124.     {
  125.       DisplayLastError(ps,hWnd);
  126.       return;
  127.     }
  128.  
  129.     sh=CreateService(hsm,
  130.                     "Poweroff",
  131.                     "Poweroff",
  132.                     STANDARD_RIGHTS_REQUIRED|SERVICE_START,
  133.                     SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS,
  134.                     SERVICE_AUTO_START,
  135.                     SERVICE_ERROR_NORMAL,
  136.                     system_poweroff,
  137.                     NULL,
  138.                     NULL,
  139.                     NULL,
  140.                     NULL,
  141.                     NULL);
  142.     if (sh==NULL)
  143.     {
  144.       DisplayLastError(ps,hWnd);
  145.       CloseServiceHandle(hsm);
  146.       return;
  147.     }
  148.     Log("Starting service");
  149.     if (StartService(sh,0,NULL)==0)
  150.     {
  151.       DisplayLastError(ps,hWnd);
  152.       CloseServiceHandle(sh);
  153.       CloseServiceHandle(hsm);
  154.       return;
  155.     }
  156.     CloseServiceHandle(hsm);
  157.     DisplayInformation(ps,hWnd,"Poweroff service succesfully installed");
  158.   }
  159.   Log("CreatePoweroffService end");
  160. }
  161.  
  162. void RemovePoweroffService(HWND hWnd,PowerSettings *ps)
  163. {
  164.   SC_HANDLE hsm,sh;
  165.  
  166.   Log("RemovePoweroffService start");
  167.   if (ps->windowsversion.dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
  168.   {
  169.     Log("Deleting WIN98 service");
  170.     if (!DeleteRegistryValue(HKEY_LOCAL_MACHINE,"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunServices","Poweroff"))
  171.     {
  172.       DisplayLastError(ps,hWnd);
  173.       return;
  174.     }
  175.     DisplayInformation(ps,hWnd,"Please restart your computer to stop the service");
  176.   }
  177.   else
  178.   {
  179.     Log("Deleting NT service");
  180.     hsm=OpenSCManager(NULL,NULL,GENERIC_WRITE);
  181.     if (hsm==NULL)
  182.     {
  183.       DisplayLastError(ps,hWnd);
  184.       return;
  185.     }
  186.     sh=OpenService(hsm,"Poweroff",DELETE);
  187.     if (sh==NULL)
  188.     {
  189.       DisplayLastError(ps,hWnd);
  190.       CloseServiceHandle(hsm);
  191.       return;
  192.     }
  193.     if (!DeleteService(sh))
  194.     {
  195.       DisplayLastError(ps,hWnd);
  196.     }
  197.     CloseServiceHandle(sh);
  198.     CloseServiceHandle(hsm);
  199.     DisplayInformation(ps,hWnd,"Poweroff service succesfully deleted");
  200.   }
  201.   DeleteRegistryKey(HKEY_LOCAL_MACHINE,"SYSTEM\\CurrentControlSet\\Services\\Poweroff\\Parameters");
  202.   Log("RemovePoweroffService end");
  203. }
  204.  
  205. VOID WINAPI ServiceHandlerProcedure(DWORD fdwControl)
  206. {
  207.   Log("ServiceHandlerProcedure start, fdwControl=%d",fdwControl);
  208.   switch (fdwControl)
  209.   {
  210.     case SERVICE_CONTROL_STOP:
  211.       status.dwServiceType=SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS;
  212.       status.dwCurrentState=SERVICE_STOPPED;
  213.       status.dwControlsAccepted=SERVICE_ACCEPT_STOP;
  214.       status.dwWin32ExitCode=NO_ERROR;
  215.       status.dwCheckPoint++;
  216.       status.dwWaitHint=1000;
  217.       SetServiceStatus(sh,&status);
  218.       RemoveFromTray(g_hWnd);
  219.       Cleanup(NULL);
  220.       SetTimer(g_hWnd,ID_STOP_TIMER,1000,NULL);
  221.       break;
  222.     case SERVICE_CONTROL_PAUSE:
  223.       break;
  224.     case SERVICE_CONTROL_CONTINUE:
  225.       break;
  226.     case SERVICE_CONTROL_INTERROGATE:
  227.       status.dwServiceType=SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS;
  228.       status.dwCurrentState=SERVICE_RUNNING;
  229.       status.dwControlsAccepted=SERVICE_ACCEPT_STOP;
  230.       status.dwWin32ExitCode=NO_ERROR;
  231.       status.dwCheckPoint++;
  232.       status.dwWaitHint=1000;
  233.       SetServiceStatus(sh,&status);
  234.       break;
  235.     case SERVICE_CONTROL_SHUTDOWN:
  236.       status.dwServiceType=SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS;
  237.       status.dwCurrentState=SERVICE_STOPPED;
  238.       status.dwControlsAccepted=SERVICE_ACCEPT_STOP;
  239.       status.dwWin32ExitCode=NO_ERROR;
  240.       status.dwCheckPoint++;
  241.       status.dwWaitHint=1000;
  242.       SetServiceStatus(sh,&status);
  243.       SendMessage(g_hWnd,WM_QUIT,0,0);
  244.       break;
  245.   }
  246.   Log("ServiceHandlerProcedure end");
  247. }
  248.  
  249. VOID WINAPI ServiceProcedure(DWORD dwArgc,LPTSTR *lpszArgv)
  250. {
  251.   Log("ServiceProcedure start");
  252.   sh=RegisterServiceCtrlHandler("Poweroff",ServiceHandlerProcedure);
  253.   if (sh==0)
  254.   {
  255.     return;
  256.   }
  257.   status.dwServiceType=SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS;
  258.   status.dwCurrentState=SERVICE_START_PENDING;
  259.   status.dwControlsAccepted=SERVICE_ACCEPT_STOP;
  260.   status.dwWin32ExitCode=NO_ERROR;
  261.   status.dwCheckPoint=0;
  262.   status.dwWaitHint=1000;
  263.   SetServiceStatus(sh,&status);
  264.   status.dwServiceType=SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS;
  265.   status.dwCurrentState=SERVICE_RUNNING;
  266.   status.dwControlsAccepted=SERVICE_ACCEPT_STOP;
  267.   status.dwWin32ExitCode=NO_ERROR;
  268.   status.dwCheckPoint=0;
  269.   status.dwWaitHint=1000;
  270.   SetServiceStatus(sh,&status);
  271.   Log("Waiting for service to start");
  272.   while (status.dwCurrentState==SERVICE_RUNNING)
  273.     Sleep(500);
  274.   Log("ServiceProcedure end");
  275. } 
  276.  
  277. DWORD WINAPI CheckService(PowerSettings *ps)
  278. {
  279.   SERVICE_TABLE_ENTRY service_table[2];
  280.  
  281.   Log("CheckService start");
  282.   if (ps->run_as_service && ps->windowsversion.dwPlatformId!=VER_PLATFORM_WIN32_WINDOWS)
  283.   {
  284.     Log("Starting as NT service");
  285.     service_table[0].lpServiceName="Poweroff";
  286.     service_table[0].lpServiceProc=ServiceProcedure;
  287.     service_table[1].lpServiceName=NULL;
  288.     service_table[1].lpServiceProc=NULL;
  289.     StartServiceCtrlDispatcher(service_table);
  290.   }
  291.   else if (ps->run_as_service)
  292.   {
  293.     HMODULE hm;
  294.     DWORD (WINAPI * RegisterServiceProcess)(DWORD dwProcessId,DWORD dwType);
  295.  
  296.     Log("Starting as WIN98 service");
  297.     hm=LoadLibrary("kernel32");
  298.     if (hm==NULL)
  299.     {
  300.       DisplayLastError(ps,NULL);
  301.       return 0;
  302.     }
  303.     RegisterServiceProcess=(void*)GetProcAddress(hm,"RegisterServiceProcess");
  304.     if (RegisterServiceProcess==NULL)
  305.     {
  306.       DisplayLastError(ps,NULL);
  307.       return 0;
  308.     }
  309.     RegisterServiceProcess(0,1);
  310.     FreeLibrary(hm);
  311.   }
  312.   Log("CheckService end");
  313.   return 0;
  314. }
  315.  
  316. int IsPoweroffServiceRunning(HWND hWnd,PowerSettings *ps)
  317. {
  318.   Log("IsPoweroffServiceRunning start");
  319.   if (ps->windowsversion.dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
  320.   {
  321.     return 0;
  322.   }
  323.   else
  324.   {
  325.     SERVICE_STATUS ss;
  326.     SC_HANDLE hsm,sh;
  327.  
  328.     hsm=OpenSCManager(NULL,NULL,GENERIC_READ);
  329.     if (hsm==NULL)
  330.     {
  331.       return 0;
  332.     }
  333.     sh=OpenService(hsm,"Poweroff",SERVICE_QUERY_STATUS);
  334.     if (sh==NULL)
  335.     {
  336.       CloseServiceHandle(hsm);
  337.       return 0;
  338.     }
  339.     if (!QueryServiceStatus(sh,&ss))
  340.     {
  341.       CloseServiceHandle(sh);
  342.       CloseServiceHandle(hsm);
  343.       return 0;
  344.     }
  345.     CloseServiceHandle(sh);
  346.     CloseServiceHandle(hsm);
  347.     if (ss.dwCurrentState==SERVICE_RUNNING)
  348.       return 1;
  349.   }
  350.   Log("IsPoweroffServiceRunning stop");
  351.   return 0;
  352. }
  353.  
  354. void StartPoweroffService(HWND hWnd,PowerSettings *ps)
  355. {
  356.   SC_HANDLE hsm,sh;
  357.  
  358.   Log("StartPoweroffService start");
  359.   if (ps->windowsversion.dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
  360.   {
  361.   }
  362.   else
  363.   {
  364.     hsm=OpenSCManager(NULL,NULL,GENERIC_WRITE);
  365.     if (hsm==NULL)
  366.     {
  367.       DisplayLastError(ps,hWnd);
  368.       return;
  369.     }
  370.     sh=OpenService(hsm,"Poweroff",SERVICE_START);
  371.     if (sh==NULL)
  372.     {
  373.       DisplayLastError(ps,hWnd);
  374.       CloseServiceHandle(hsm);
  375.       return;
  376.     }
  377.     if (!StartService(sh,0,NULL))
  378.     {
  379.       DisplayLastError(ps,hWnd);
  380.     }
  381.     CloseServiceHandle(sh);
  382.     CloseServiceHandle(hsm);
  383.   }
  384.   Log("StartPoweroffService end");
  385. }
  386.  
  387. void StopPoweroffService(HWND hWnd,PowerSettings *ps)
  388. {
  389.   SC_HANDLE hsm,sh;
  390.  
  391.   Log("StopPoweroffService start");
  392.   if (ps->windowsversion.dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
  393.   {
  394.   }
  395.   else
  396.   {
  397.     SERVICE_STATUS ss;
  398.  
  399.     hsm=OpenSCManager(NULL,NULL,GENERIC_WRITE);
  400.     if (hsm==NULL)
  401.     {
  402.       DisplayLastError(ps,hWnd);
  403.       return;
  404.     }
  405.     sh=OpenService(hsm,"Poweroff",SERVICE_STOP);
  406.     if (sh==NULL)
  407.     {
  408.       DisplayLastError(ps,hWnd);
  409.       CloseServiceHandle(hsm);
  410.       return;
  411.     }
  412.     if (!ControlService(sh,SERVICE_CONTROL_STOP,&ss))
  413.     {
  414.       DisplayLastError(ps,hWnd);
  415.     }
  416.     CloseServiceHandle(sh);
  417.     CloseServiceHandle(hsm);
  418.   }
  419.   Log("StopPoweroffService end");
  420. }
  421.