Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-02 05:45:50 +00:00
Code Issues Projects Releases Packages Wiki Activity

Clean up the start code

Browse source 
svn path=/trunk/; revision=44963
This commit is contained in:
Ged Murphy 2010-01-05 21:14:10 +00:00
parent d5bc8625fb
commit 5dfb2cc0c3
1 changed files with 84 additions and 80 deletions
Download patch file Download diff file Expand all files Collapse all files

164
reactos/base/applications/mscutils/servman/start.c
View file

@ -3,7 +3,7 @@
* LICENSE: GPL - See COPYING in the top level directory * LICENSE: GPL - See COPYING in the top level directory
* FILE: base/applications/mscutils/servman/start.c * FILE: base/applications/mscutils/servman/start.c
* PURPOSE: Start a service * PURPOSE: Start a service
* COPYRIGHT: Copyright 2005-2009 Ged Murphy <gedmurphy@reactos.org> * COPYRIGHT: Copyright 2005-2010 Ged Murphy <gedmurphy@reactos.org>
* *
*/ */
@ -11,7 +11,7 @@
static BOOL static BOOL
DoStartService(PMAIN_WND_INFO Info, DoStartService(PMAIN_WND_INFO Info,
HWND hProgDlg) HWND hProgress)
{ {
SC_HANDLE hSCManager; SC_HANDLE hSCManager;
SC_HANDLE hService; SC_HANDLE hService;
@ -25,72 +25,86 @@ DoStartService(PMAIN_WND_INFO Info,
hSCManager = OpenSCManager(NULL, hSCManager = OpenSCManager(NULL,
NULL, NULL,
SC_MANAGER_ALL_ACCESS); SC_MANAGER_CONNECT);
if (!hSCManager) if (hSCManager)
{ {
return FALSE; hService = OpenService(hSCManager,
} Info->pCurrentService->lpServiceName,
SERVICE_START | SERVICE_QUERY_STATUS);
hService = OpenService(hSCManager, if (hService)
Info->pCurrentService->lpServiceName,
SERVICE_START | SERVICE_QUERY_STATUS);
if (hService)
{
bRet = StartService(hService,
0,
NULL);
if (!bRet && GetLastError() == ERROR_SERVICE_ALREADY_RUNNING)
{ {
bRet = TRUE; if (hProgress)
}
else if (bRet)
{
bRet = FALSE;
if (QueryServiceStatusEx(hService,
SC_STATUS_PROCESS_INFO,
(LPBYTE)&ServiceStatus,
sizeof(SERVICE_STATUS_PROCESS),
&BytesNeeded))
{ {
dwStartTickCount = GetTickCount(); /* Increment the progress bar */
dwOldCheckPoint = ServiceStatus.dwCheckPoint; IncrementProgressBar(hProgress, DEFAULT_STEP);
dwMaxWait = 30000; // 30 secs }
while (ServiceStatus.dwCurrentState != SERVICE_RUNNING) bRet = StartService(hService,
0,
NULL);
if (!bRet && GetLastError() == ERROR_SERVICE_ALREADY_RUNNING)
{
/* If it's already running, just return TRUE */
bRet = TRUE;
}
else if (bRet)
{
bRet = FALSE;
/* Get the service status to check if it's running */
if (QueryServiceStatusEx(hService,
SC_STATUS_PROCESS_INFO,
(LPBYTE)&ServiceStatus,
sizeof(SERVICE_STATUS_PROCESS),
&BytesNeeded))
{ {
dwWaitTime = ServiceStatus.dwWaitHint / 10; /* We don't want to wait for more than 30 seconds */
dwMaxWait = 30000;
dwStartTickCount = GetTickCount();
if (!QueryServiceStatusEx(hService, /* Loop until it's running */
SC_STATUS_PROCESS_INFO, while (ServiceStatus.dwCurrentState != SERVICE_RUNNING)
(LPBYTE)&ServiceStatus,
sizeof(SERVICE_STATUS_PROCESS),
&BytesNeeded))
{ {
break;
}
if (ServiceStatus.dwCheckPoint > dwOldCheckPoint)
{
/* The service is making progress*/
dwStartTickCount = GetTickCount();
dwOldCheckPoint = ServiceStatus.dwCheckPoint; dwOldCheckPoint = ServiceStatus.dwCheckPoint;
} dwWaitTime = ServiceStatus.dwWaitHint / 10;
else
{ /* Get the latest status info */
if (GetTickCount() >= dwStartTickCount + dwMaxWait) if (!QueryServiceStatusEx(hService,
SC_STATUS_PROCESS_INFO,
(LPBYTE)&ServiceStatus,
sizeof(SERVICE_STATUS_PROCESS),
&BytesNeeded))
{ {
/* We exceeded our max wait time, give up */ /* Something went wrong... */
break; break;
} }
/* Is the service making progress? */
if (ServiceStatus.dwCheckPoint > dwOldCheckPoint)
{
/* It is, get the latest tickcount to reset the max wait time */
dwStartTickCount = GetTickCount();
dwOldCheckPoint = ServiceStatus.dwCheckPoint;
}
else
{
/* It's not, make sure we haven't exceeded our wait time */
if (GetTickCount() >= dwStartTickCount + dwMaxWait)
{
/* We have, give up */
break;
}
}
/* Adjust the wait hint times */
if (dwWaitTime < 200)
dwWaitTime = 200;
else if (dwWaitTime > 10000)
dwWaitTime = 10000;
/* Wait before trying again */
Sleep(dwWaitTime);
} }
if (dwWaitTime < 200)
dwWaitTime = 200;
else if (dwWaitTime > 10000)
dwWaitTime = 10000;
Sleep(dwWaitTime);
} }
if (ServiceStatus.dwCurrentState == SERVICE_RUNNING) if (ServiceStatus.dwCurrentState == SERVICE_RUNNING)
@ -98,45 +112,35 @@ DoStartService(PMAIN_WND_INFO Info,
bRet = TRUE; bRet = TRUE;
} }
} }
CloseServiceHandle(hService);
} }
CloseServiceHandle(hService); CloseServiceHandle(hSCManager);
} }
CloseServiceHandle(hSCManager);
return bRet; return bRet;
} }
BOOL BOOL
DoStart(PMAIN_WND_INFO Info) DoStart(PMAIN_WND_INFO Info)
{ {
HWND hProgDlg; HWND hProgress;
BOOL bRet = FALSE; BOOL bRet = FALSE;
hProgDlg = CreateProgressDialog(Info->hMainWnd, /* Create a progress window to track the progress of the stopping service */
IDS_PROGRESS_INFO_START); hProgress = CreateProgressDialog(Info->hMainWnd,
IDS_PROGRESS_INFO_START);
if (hProgDlg) if (hProgress)
{ {
InitializeProgressDialog(hProgDlg, Info->pCurrentService->lpServiceName); /* Set the service name and reset the progress bag */
InitializeProgressDialog(hProgress, Info->pCurrentService->lpServiceName);
bRet = DoStartService(Info, /* Start the requested service */
hProgDlg); bRet = DoStartService(Info, hProgress);
if (bRet) /* Complete and destroy the progress bar */
{ DestroyProgressDialog(hProgress, bRet);
CompleteProgressBar(hProgDlg);
Sleep(500);
bRet = TRUE;
}
else
{
GetError();
}
DestroyWindow(hProgDlg);
} }
return bRet; return bRet;