lyfz_repos/经典版源码/kernelcode/lyfzClassicSvc/lyfzClassicSvc/MainWork.cpp

#include "StdAfx.h"
#include ".\mainwork.h"

HANDLE g_hCtrlThread = NULL;	
CRITICAL_SECTION  g_CSLimitMod;

DWORD g_nTickCount = 0;
char  g_szCurIniName[MAX_PATH];			// 当前配置文件名;

CMainWork::CMainWork(void)
{
	m_hMWConvertThread = NULL;
	InitializeCriticalSection(&g_CSLimitMod);

	g_hCtrlThread = CreateEvent(NULL,TRUE,FALSE,NULL);
}

CMainWork::~CMainWork(void)
{
	if (g_hCtrlThread)
		SetEvent(g_hCtrlThread);

	if (WaitForSingleObject(g_hCtrlThread,INFINITE) != WAIT_TIMEOUT)
	{
		CloseHandle(m_hMWConvertThread);
		m_hMWConvertThread = NULL;
	}

	if (g_hCtrlThread != NULL)
	{
		CloseHandle(g_hCtrlThread);
		g_hCtrlThread = NULL;
	}

	DeleteCriticalSection(&g_CSLimitMod);
	WSACleanup();
}

void CMainWork::MainWorkStart()
{
	sprintf_s(g_szCurIniName,260,"%s\\ControlService.ini",g_szModulePath);
	m_hMWConvertThread = CreateThread(NULL,0,ModeThreadProc,this,0,NULL);
	if (m_hMWConvertThread == NULL)
	{
		LOG4C((LOG_NOTICE,"创建模式线程失败"));
		return ;
	}
}

DWORD CMainWork::ModeThreadProc(LPVOID lpvoid)
{
	CMainWork *pThis = (CMainWork*)lpvoid;
	do 
	{
		pThis->StartSpecialService();
		pThis->StartSpecialProcess();
	} while (WaitForSingleObject(g_hCtrlThread, 150) == WAIT_TIMEOUT);

	return 0;
}
void CMainWork::StartSpecialService()
{
	CIniReader IniReader(g_szCurIniName);
	IniReader.GetSectionData("Service");
	int iCount = IniReader.GetKeyCount();

	IniReader.GetSectionNames();
	// 遍历段下所有键值;
	for(int i = 0; i < iCount; i++)
	{
		CString strKey = "";
		strKey.Format("Svr%d",i+1);

		CString strKeyValue = "";
		strKeyValue = IniReader.GetKeyValueText(strKey,"Service");

		CString strSvrProName = GetIniSvrProName(strKeyValue);
		if ( !m_Procsee.SrvRunState(strSvrProName) )// 服务未启动;
		{
			LOG4C((LOG_NOTICE,"启动服务%s",strSvrProName));
			m_Procsee.SC_StartSvr(strSvrProName);
			Sleep(150);
		}
	}
}

void CMainWork::StartSpecialProcess()
{
	CIniReader IniReader(g_szCurIniName);
	IniReader.GetSectionData("Process");
	int iCount = IniReader.GetKeyCount();
	IniReader.GetSectionNames();
	// 遍历段下所有键值;
	for(int i = 0; i < iCount; i++)
	{
		CString strKey = "";
		strKey.Format("Pro%d",i+1);

		CString strKeyValue = "";
		strKeyValue = IniReader.GetKeyValueText(strKey,"Process");

		CString strSvrProPath = GetIniSvrProPath(strKeyValue);
		CString strSvrProName = GetIniSvrProName(strKeyValue);
		// 启动指定进程;
		if ( !m_Procsee.IsProExist(strSvrProName) )
		{
			LOG4C((LOG_NOTICE,"启动进程%s",strSvrProName));
			m_Procsee.StartProcess(strSvrProPath);
			Sleep(1500);
		}
	}
}

// 获取 串时间;
CString CMainWork::GetKeyValueTime(CString strKeyValue)
{
	CString strTiem;
	int nIndex = strKeyValue.Find('|');
	strTiem = strKeyValue.Right(strKeyValue.GetLength() - nIndex -1);
	return strTiem;
}

// 获取 串文件名;
CString CMainWork::GetKeyValueFileName(CString strKeyValue)
{
	CString strFileName;
	int nIndex = strKeyValue.Find('|');
	strFileName = strKeyValue.Left(nIndex);

	return strFileName;
}
// 注意:IOServer.exe这个服务,在系统服务里,暂停是无效;
// 所以,要停止IOServer这服务,只能从进程里结束;
void CMainWork::CloseSpecialService()
{
	CIniReader IniReader(g_szCurIniName);
	IniReader.GetSectionData("Service");
	int iCount = IniReader.GetKeyCount();
	IniReader.GetSectionNames();
	// 遍历段下所有键值;
	for(int i = 0; i < iCount; i++)
	{
		Sleep(1000);
		CString strKey = "";
		strKey.Format("Svr%d",i+1);

		CString strKeyValue = "";
		strKeyValue = IniReader.GetKeyValueText(strKey,"Service");

		CString strSvrProName = GetIniSvrProName(strKeyValue);

		// 停止指定服务;
		m_Procsee.SC_StopSvr(strSvrProName);
		//if (!m_Procsee.StopSrv(strSvrProName))
		//if (!m_Procsee.SC_StopSvr(strSvrProName))
			//LOG4C((LOG_NOTICE,"服务停止 失败 = %s" ,strSvrProName));

		// 并设置服务为手动,或禁止或卸载服务;
		if(!m_Procsee.SetSrvState(strSvrProName,SERVICE_DEMAND_START,NULL))
			LOG4C((LOG_NOTICE,"服务设置 失败 = %s",strSvrProName));
	}
}

void CMainWork::CloseSpecialProcess()
{
	CIniReader IniReader(g_szCurIniName);
	IniReader.GetSectionData("Process");
	int iCount = IniReader.GetKeyCount();
	IniReader.GetSectionNames();
	// 遍历段下所有键值;
	for(int i = 0; i < iCount; i++)
	{
		Sleep(1000);
		CString strKey = "";
		strKey.Format("Pro%d",i+1);

		CString strKeyValue = "";
		strKeyValue = IniReader.GetKeyValueText(strKey,"Process");

		CString strSvrProName = GetIniSvrProName(strKeyValue);
		// 结束指定进程;
		if(!m_Procsee.CloseProcess(strSvrProName))
			LOG4C((LOG_NOTICE,"结束进程 失败 = %s",strSvrProName));
	}
}

void CMainWork::InstallSpecialService()
{
	CIniReader IniReader(g_szCurIniName);
	IniReader.GetSectionData("Service");
	int iCount = IniReader.GetKeyCount();
	LOG4C((LOG_NOTICE,"iCount = %d",iCount));
	IniReader.GetSectionNames();
	// 遍历段下所有键值;
	for(int i = 0; i < iCount; i++)
	{
		Sleep(1000);
		CString strKey = "";
		strKey.Format("Svr%d",i+1);

		CString strKeyValue = "";
		strKeyValue = IniReader.GetKeyValueText(strKey,"Service");

		CString strSvrProName = GetIniSvrProName(strKeyValue);
		CString strSvrProPath = GetIniSvrProPath(strKeyValue);

		// .安装服务前,要检测下服务是否存在,这操作放在process类里做;
		// 安装指定服务(该安装方式是通过批处理安装的,非正常方式);
		if(!m_Procsee.StartProcess(strSvrProPath))
			LOG4C((LOG_NOTICE,"安装服务 失败 = %s", strSvrProPath));

		// 启动服务;
		m_Procsee.SC_StartSvr(strSvrProName);
	}
}

CString CMainWork::GetIniSvrProName(CString strKeyValue)
{
	CString strSvrProName;
	int nIndex = strKeyValue.Find('|');
	strSvrProName = strKeyValue.Left(nIndex);
	return strSvrProName;
}

CString CMainWork::GetIniSvrProPath(CString strKeyValue)
{
	CString strSvrProPath;
	int nIndex = strKeyValue.Find('|');
	strSvrProPath = strKeyValue.Right(strKeyValue.GetLength() - nIndex -1);
	return strSvrProPath;
}