关机重启后运行父进程

目的：进程在进行某些操作后需要系统重启，并且在系统重启后调用原进程的父进程进行某些操作。

BOOL ParentProcessRunonce()
{
    DWORD pid = GetCurrentProcessId();
    CString strParentPath;
    if(!GetParentProcessFileName(pid,strParentPath))   //传出至strParentPath
    {
        WRITE_LOG(_T("无法获取父进程的路径"));
        return FALSE;
    }

    if(SetRunonce(strParentPath,L""))
    {
        WRITE_LOG(_T("SetRunonce [%s] SUC"),strParentPath);
        return TRUE;
    }
    else
    {
        WRITE_LOG(_T("SetRunonce [%s] FAIL"),strParentPath);
        return FALSE;
    }
}

1.通过当前进程ID获得父进程ID 

BOOL GetParentProcessFileName( DWORD dwProcessId, CString& szFileName )
{
    szFileName.Empty();
    HANDLE hProcessSnap = ::CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
    if( hProcessSnap == INVALID_HANDLE_VALUE )
        return FALSE;

    PROCESSENTRY32 pe32 = { 0 };
    pe32.dwSize = sizeof(pe32);
    BOOL bContinue = ::Process32First( hProcessSnap, &pe32 );
    DWORD dwParentProcessId = 0;
    while( bContinue )
    {
        if( pe32.th32ProcessID == dwProcessId )
        {
            dwParentProcessId = pe32.th32ParentProcessID;
            break;
        }

        bContinue = ::Process32Next( hProcessSnap, &pe32 );
    }
    ::CloseHandle( hProcessSnap );

    return GetProcessFileName( dwParentProcessId, szFileName );
}

2.通过父进程ID获得进程路径 

BOOL GetProcessFileName( DWORD dwPID, CString& szFileName )
{
    szFileName.Empty();
    if( dwPID != 0 )
    {
        HANDLE hProcessParent = ::OpenProcess( PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, dwPID );
        if( hProcessParent != NULL )
        {
            HMODULE hModule = NULL;
            DWORD dwNeeded = 0;
            TCHAR szBuff[MAX_PATH] = { 0 };
            EnumProcessModules( hProcessParent, &hModule, sizeof( hModule ), &dwNeeded );    //获得指定进程中所有模块的句柄
            GetModuleFileNameEx( hProcessParent, hModule, szBuff, MAX_PATH );
            szFileName = szBuff;
            ::CloseHandle( hProcessParent );
        }    
    }    
    return ( szFileName.GetLength() > 0 );
}

3.将父进程路径写入注册表（重启后自动运行父进程）

BOOL SetRunonce( LPCTSTR lpPath, LPCTSTR lpParam )
{
    //CRegKey提供了对系统注册表的操作方法，通过CRegKey类，可以方便的打开注册表的某个分支或子键（CRegKey::Open）
    //可以方便的修改一个键的键值（CRegKey::SetValue），也可以查询某个键的键值（CRegKey::QueryValue），操作完成之后，可以关闭子键（CRegKey::Close）
    CRegKey key;
    if (ERROR_SUCCESS == key.Create(HKEY_CURRENT_USER, L"SOFTWARE\Microsoft\Windows\CurrentVersion\RunOnce"))
    {
        CString strPath = lpPath;

        //判断路径中是否有空格，有的话，就是用“”引号把整个路径包含起来
        PathQuoteSpaces(CStrBuf(strPath, MAX_PATH));
        CString strParam = lpParam;
        if (!strParam.IsEmpty())
        {
            strPath += _T(" ");
            strPath += lpParam;
        }
        return key.SetStringValue(L"***设置的值的名称", strPath, REG_SZ) == ERROR_SUCCESS;
    }
    else
    {
        //HKEY_CURRENT_USER、HKEY_LOCAL_MACHINE代表注册表中不同分支
        if (ERROR_SUCCESS == key.Create(HKEY_LOCAL_MACHINE, L"SOFTWARE\Microsoft\Windows\CurrentVersion\RunOnce"))
        {
            CString strPath = lpPath;

            PathQuoteSpaces(CStrBuf(strPath, MAX_PATH));    
            CString strParam = lpParam;
            if (!strParam.IsEmpty())
            {
                strPath += _T(" ");
                strPath += lpParam;
            }
            return key.SetStringValue(L"***设置的值的名称", strPath, REG_SZ) == ERROR_SUCCESS;
        }
    }
    return FALSE;
}