myservice.cpp

email client and support pop3、smtp、imap protocol.

// myservice.cpp

#include "NTServApp.h"
#include "myservice.h"


CMyService::CMyService()
:CNTService("InetServ Windows NT Service")
{
	m_iStartParam = 0;
	m_iIncParam = 1;
	m_iState = m_iStartParam;
}

BOOL CMyService::OnInit()
{
	// Read the registry parameters
    // Try opening the registry key:
    // HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\<AppName>\Parameters
    HKEY hkey;
	char szKey[1024];
	strcpy(szKey, "SYSTEM\\CurrentControlSet\\Services\\");
	strcat(szKey, m_szServiceName);
	strcat(szKey, "\\Parameters");
    if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                     szKey,
                     0,
                     KEY_QUERY_VALUE,
                     &hkey) == ERROR_SUCCESS) {
        // Yes we are installed
        DWORD dwType = 0;
        DWORD dwSize = sizeof(m_iStartParam);
        RegQueryValueEx(hkey,
                        "Start",
                        NULL,
                        &dwType,
                        (BYTE*)&m_iStartParam,
                        &dwSize);
        dwSize = sizeof(m_iIncParam);
        RegQueryValueEx(hkey,
                        "Inc",
                        NULL,
                        &dwType,
                        (BYTE*)&m_iIncParam,
                        &dwSize);
        RegCloseKey(hkey);
    }

	// Set the initial state
	m_iState = m_iStartParam;

	return TRUE;
}

void CMyService::Run()
{

	
	char szPath[1024];
	::GetModuleFileName(NULL, szPath, sizeof(szPath));

	for (int i = strlen(szPath);i>=0;i--)
	{
	 if (szPath[i] == '\\') 
	 {
		 szPath[i] = '\0';
		 break;
	 }

	}





	  tdoPID  =0;

 	  PROCESS_INFORMATION pi = {0};
      STARTUPINFO si = {0};
	  char tempstr[255];
	  sprintf(tempstr,"inetserv.exe");
      si.cb = sizeof(STARTUPINFO);
      si.lpReserved = NULL;
      si.lpReserved2 = NULL;
      si.cbReserved2 = 0;
      si.lpDesktop = NULL;
      si.dwFlags = 0;
 	  




      if (CreateProcess( NULL,tempstr,NULL,NULL,TRUE,
		  NORMAL_PRIORITY_CLASS,NULL,szPath,&si,&pi )  != 0)
	  {
		tdoPID= pi.dwProcessId;
	  }
	  else
	  {
		m_bIsRunning = 0;
	  }
   
     
   

while (m_bIsRunning) 
	{
 
        Sleep(1000);

			if (GetProcessVersion(tdoPID) == 0) 
			{
			Run();
			}

		m_iState += m_iIncParam;
    }
}

// Process user control requests
BOOL CMyService::OnUserControl(DWORD dwOpcode)
{
    switch (dwOpcode) {
    case SERVICE_CONTROL_USER + 0:

        // Save the current status in the registry
        SaveStatus();
        return TRUE;

    default:
        break;
    }
    return FALSE; // say not handled
}

// Save the current status in the registry
void CMyService::SaveStatus()
{
 
}

void CMyService::OnStop()
{
HANDLE hPr = OpenProcess(PROCESS_TERMINATE, TRUE,tdoPID );
TerminateProcess(hPr, 0);
CloseHandle(hPr);
}