#include "stdafx.h"
#include "SearchExeFolder.h"
#include <iostream>
#define MAINKEY HKEY_LOCAL_MACHINE
using namespace std;
CSearchExeFolder::CSearchExeFolder(void)
{
}
CSearchExeFolder::~CSearchExeFolder(void)
{
}
void String2LPWSTR(std::string str, LPWSTR lp)
{
size_t size = str.length();
MultiByteToWideChar(CP_ACP, 0, str.c_str(), size, lp, size * sizeof(wchar_t));
lp[size] = 0; //确保以 '\0' 结尾
}
//wstring转换成string
std::string WChar2Ansi(LPCWSTR pwszSrc)
{
int nLen = WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, NULL, 0, NULL, NULL);
if (nLen <= 0) return std::string("");
char* pszDst = new char[nLen];
if (NULL == pszDst) return std::string("");
WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, pszDst, nLen, NULL, NULL);
pszDst[nLen - 1] = 0;
std::string strTemp(pszDst);
delete[] pszDst;
return strTemp;
}
std::string CSearchExeFolder::EnumReg(const std::string& sSubKey)
{
string sResult;;
HKEY hKey = NULL;
DWORD num = MAX_PATH;
DWORD index = 0;
WCHAR arrSubKey[MAX_PATH] = {'\0'};
String2LPWSTR(sSubKey, arrSubKey);
/// MAINKEY与arrSubKey的值都是软件安装时写入注册表的,需要手动去注册表中找
DWORD rc = ::RegOpenKeyEx(MAINKEY, arrSubKey, 0, KEY_READ, &hKey);
if(rc == ERROR_SUCCESS)
{
WCHAR buffer[MAX_PATH] = {'\0'};
while(RegEnumValue(hKey,index, buffer ,&num, NULL,NULL, NULL, NULL) == 0 )
{
string sTargeKeyName = "Inno Setup: App Path";// 设置值的键(软件安装时写死的)
string sKeyName = WChar2Ansi(buffer);
if(sKeyName.find(sTargeKeyName) != string::npos)
{
WCHAR arrKeyName [MAX_PATH] = {'\0'};
WCHAR arrResult[MAX_PATH] = {'\0'};
DWORD dwSize = sizeof(DWORD);
DWORD dwType = REG_SZ;
LONG ret;
String2LPWSTR(sTargeKeyName, arrKeyName);
// 试探获取dwSize的大小
ret = RegQueryValueEx(hKey, arrKeyName, 0, &dwType, NULL, &dwSize);
ret = RegQueryValueEx(hKey, arrKeyName, 0, &dwType, (LPBYTE)&arrResult, &dwSize);
if (ERROR_SUCCESS == ret)
{
sResult = WChar2Ansi(arrResult);
};
}
index++;
num = MAX_PATH;
}
}
RegCloseKey(hKey);
return sResult;
}
std::string CSearchExeFolder::FindFolder()
{
string sFilePath = EnumReg("SOFTWARE\\Wow6432Node\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\{C5D16AB8-FEDE-4686-87A8-350C68EB24FA}_is1");
return sFilePath;
}
拼接上启动项的文件名
CSearchExeFolder searchRegKey;
std::string sFolder = searchRegKey.FindFolder();
std::string sFilePath = sFolder + "\\Access.exe";
StartExe(sFilePath);