C / C++ / MFC

Kushagra Tiwari wrote:

Can you detail me exactly what steps have to be followed in achieving the same.

Covean wrote:

An other possibility is to use the SetConsoleCtrlHandler function...

Covean wrote:

. You have to use SetServiceStatus with a SERVICE_STATUS where dwControlsAccepted is a least SERVICE_ACCEPT_SHUTDOWN.

void ...SetServiceStatus(DWORD dwState)
{
    if (::SetServiceStatus(m_hServiceStatus, &m_status) == TRUE)
    {
        // ...
        LogEvent(TEXT("SetServiceStatus Failed"));
    }
    else
    {
        // ...
        LogEvent(TEXT("SetServiceStatus Failed"));
    }
}

switch (dwOpcode)
{
    case SERVICE_CONTROL_STOP:
        LogEvent(_T("SERVICE_CONTROL_STOP"));
        //...
        break;
    case SERVICE_CONTROL_PAUSE:
        LogEvent(_T("SERVICE_CONTROL_PAUSE"));
        break;
    case SERVICE_CONTROL_CONTINUE:
        LogEvent(_T("SERVICE_CONTROL_CONTINUE"));
        break;
    case SERVICE_CONTROL_INTERROGATE:
        LogEvent(_T("SERVICE_CONTROL_INTERROGATE"));
        break;
    case SERVICE_CONTROL_SHUTDOWN:
        LogEvent(_T("SERVICE_CONTROL_SHUTDOWN"));

        AbortShutdown(chComputerName);
        break;
    //...
}

BOOL ...::AbortShutdown(LPTSTR lpMachineName)
{
	HANDLE hToken;              // handle to process token 
	TOKEN_PRIVILEGES tkp;       // pointer to token structure 
	 
	BOOL fResult;               // system shutdown flag 

 // Get the current process token handle  so we can get shutdown 
    // privilege. 
	if (!OpenProcessToken(GetCurrentProcess(), 
            TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) 
    {
       // TRACE("OpenProcessToken failed.\n"); 
        LogEvent(TEXT("OpenProcessToken failed"));

		return false;
    }

 
    // Get the LUID for shutdown privilege. 
    if(!LookupPrivilegeValue(lpMachineName, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid))
    {
        LogEvent(TEXT("LookupPrivilegeValue failed"));
		return false;
    }


 
    tkp.PrivilegeCount = 1;  // one privilege to set    
    tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 
 
    // Get shutdown privilege for this process. 
    if(!AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0))
    {
        LogEvent(TEXT("AdjustTokenPrivileges failed"));
		return false;
    }
 
    // Cannot test the return value of AdjustTokenPrivileges. 
    /*if (GetLastError() != ERROR_SUCCESS) 
    {
        //TRACE("AdjustTokenPrivileges(setting) enable failed.\n"); 
		return false;
    }*/
 
    // Prevent the system from shutting down. 
    LogEvent(TEXT("AbortSystemShutdown"));
    fResult = AbortSystemShutdown(NULL); // <---- changed to NULL for local pc

 
    if (!fResult) 
    { 
        //TRACE("AbortSystemShutdown failed.\n"); 

        wchar_t test[512];
        swprintf((wchar_t*)&test, TEXT("AbortSystemShutdown failed! Error=%d"), (DWORD)GetLastError());
        LogEvent((LPCWSTR)&test);
		return false;
    }
	else
	{
		//Cxcutter objexe;
		//objexe.Write_W3C_Log ("ZDEVSERVICE" , "1.0" , "" , "AbortSystemShutdown Successful" , "In CtrlHandler" ,  "" , "");
        LogEvent(TEXT("AbortSystemShutdown succeed"));
	}
		//m_bIsShuttingDown = false; // Reset shut down flag
 

    // Disable shutdown privilege. 
    tkp.Privileges[0].Attributes = 0; 
    AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, 
        (PTOKEN_PRIVILEGES) NULL, 0); 
 
    if (GetLastError() != ERROR_SUCCESS) 
    {
       // TRACE("AdjustTokenPrivileges(re-setting) disable failed.\n"); 
		return false;
    }
	return true;
}