注册表Demo

一、获取安装程序信息

 #include <windows.h>
 #include <iostream>
 #include <string>
 #include <vector>

 using namespace std;

 //记录安装软件信息的结构体
 struct ApplicationInfoA
 {
     string strName;                //软件名
     string strDisplayName;        //显示的软件名
     string strPublisher;        //发布者
     string strVersion;            //版本
     string strDisPlayVersion;    //显示的版本
     string strInstallLocation;    //安装的位置
 };

 int GetAppInfoA(HKEY hKey, LPCSTR lpszAppName, LPCSTR lpszKeyValueName, string &strKeyValue)
 {
     int iRet;
     HKEY hInstallAppKey;

     iRet = RegOpenKeyExA(hKey, lpszAppName, , KEY_ALL_ACCESS, &hInstallAppKey);
     if (iRet != ERROR_SUCCESS)
     {
         return -;
     }

     DWORD dwKeyValueType = REG_SZ;
     DWORD dwKeyValueDataSize = ;

     iRet = RegQueryValueExA(hInstallAppKey, lpszKeyValueName, NULL, &dwKeyValueType, NULL, &dwKeyValueDataSize);
     if (iRet == ERROR_FILE_NOT_FOUND)
     {
         RegCloseKey(hInstallAppKey);
         return ;
     }
     else if (iRet != ERROR_SUCCESS)
     {
         RegCloseKey(hInstallAppKey);
         return -;
     }

     if (dwKeyValueType != REG_SZ)
     {
         RegCloseKey(hInstallAppKey);
         return ;
     }

     LPSTR lpszKeyValueData = new char[dwKeyValueDataSize + ];
     memset(lpszKeyValueData, , dwKeyValueDataSize + );

     iRet = RegQueryValueExA(hInstallAppKey, lpszKeyValueName, NULL, &dwKeyValueType, (LPBYTE)lpszKeyValueData, &dwKeyValueDataSize);
     if (iRet != ERROR_SUCCESS)
     {
         delete []lpszKeyValueData;
         RegCloseKey(hInstallAppKey);
         return -;
     }

     strKeyValue = lpszKeyValueData;
     delete []lpszKeyValueData;
     RegCloseKey(hInstallAppKey);

     return ;
 }

 int GetAllInstalledAppInfoA(vector<ApplicationInfoA> &vecAppInfo)
 {
     int iRet;
     HKEY hKey;
     LPCSTR lpszSubKey = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall";
     iRet = RegOpenKeyExA(HKEY_LOCAL_MACHINE, lpszSubKey, , KEY_ALL_ACCESS, &hKey);
     if (iRet != ERROR_SUCCESS)
     {
         return -;
     }

     DWORD dwSubKeysCnt;
     DWORD dwMaxSubKeyNameLen;
     DWORD dwKeyValueCnt;
     DWORD dwMaxKeyValueNameLen;
     DWORD dwMaxKeyValueDataLen;

     iRet = RegQueryInfoKeyA(
         hKey,
         NULL,
         NULL,
         NULL,
         &dwSubKeysCnt,
         &dwMaxSubKeyNameLen,
         NULL,
         &dwKeyValueCnt,
         &dwMaxKeyValueNameLen,
         &dwMaxKeyValueDataLen,
         NULL,
         NULL);
     if (iRet != ERROR_SUCCESS)
     {
         RegCloseKey(hKey);
         return -;
     }

     DWORD dwIndex;
     DWORD dwNameLen = dwMaxSubKeyNameLen + ;
     LPSTR lpszSubKeyName = new char[dwNameLen];

     for (dwIndex = ; dwIndex < dwSubKeysCnt; ++dwIndex)
     {
         dwNameLen = dwMaxSubKeyNameLen + ;
         memset(lpszSubKeyName, , dwNameLen);

         iRet = RegEnumKeyExA(hKey, dwIndex, lpszSubKeyName, &dwNameLen, NULL, NULL, NULL, NULL);
         if (iRet != ERROR_SUCCESS)
         {
             RegCloseKey(hKey);
             delete []lpszSubKeyName;
             return -;
         }

         ApplicationInfoA appInfo;
         appInfo.strName = lpszSubKeyName;
         GetAppInfoA(hKey, lpszSubKeyName, "DisplayName", appInfo.strDisplayName);
         GetAppInfoA(hKey, lpszSubKeyName, "Publisher", appInfo.strPublisher);
         GetAppInfoA(hKey, lpszSubKeyName, "Version", appInfo.strVersion);
         GetAppInfoA(hKey, lpszSubKeyName, "DisplayVersion", appInfo.strDisPlayVersion);
         GetAppInfoA(hKey, lpszSubKeyName, "InstallLocation", appInfo.strInstallLocation);
         vecAppInfo.push_back(appInfo);
     }

     delete []lpszSubKeyName;
     RegCloseKey(hKey);

     return ;
 }

 int main(int argc, char **argv)
 {
     cout << "Application Information" << endl;

     vector<ApplicationInfoA> vecAppInfo;
     cout << GetAllInstalledAppInfoA(vecAppInfo) << endl;

     for (vector<ApplicationInfoA>::const_iterator iter = vecAppInfo.begin(); iter != vecAppInfo.end(); ++iter)
     {
         cout << "-------------------------------------------------------------------------------" << endl;
         cout << "Name:" << iter->strName << endl;
         cout << "DisplayName:" << iter->strDisplayName << endl;
         cout << "Publisher:" << iter->strPublisher << endl;
         cout << "Version:" << iter->strVersion << endl;
         cout << "DisplayVersion:" << iter->strDisPlayVersion << endl;
         cout << "InstallLocation:" << iter->strInstallLocation << endl;
         cout << "-------------------------------------------------------------------------------" << endl;
         cout << endl;
         cout << endl;
     }

     return ;
 }

二、添加环境变量

 #define _CRT_SECURE_NO_WARNINGS

 #include <windows.h>
 #include <iostream>
 #include <string>

 using namespace std;

 int AddPathEnvValue(LPCSTR lpszPathValue)
 {
     int iRet;
     HKEY hKey;
     LPCSTR lpszSubKey = "SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment";

     iRet = RegOpenKeyExA(HKEY_LOCAL_MACHINE, lpszSubKey, , KEY_ALL_ACCESS, &hKey);
     if (ERROR_SUCCESS != iRet)
     {
         cout << "RegOpenKeyExA Error!" << endl;
         return -;
     }

     LPCSTR lpszKeyValueName = "Path";
     DWORD dwKeyValueType = REG_EXPAND_SZ;
     DWORD dwKeyValueDataSize = ;

     iRet = RegQueryValueExA(hKey, lpszKeyValueName, NULL, &dwKeyValueType, NULL, &dwKeyValueDataSize);
     if (ERROR_FILE_NOT_FOUND != iRet)
     {
         iRet = RegSetValueExA(hKey, lpszKeyValueName, , REG_EXPAND_SZ, (const BYTE*)"", );
         if (ERROR_SUCCESS != iRet)
         {
             cout << "RegSetValueExA Error!" << endl;
             RegCloseKey(hKey);
             return -;
         }
     }
     else if (ERROR_SUCCESS != iRet)
     {
         cout << "RegSetValueExA Error!" << endl;
         RegCloseKey(hKey);
         return -;
     }
     else if (REG_EXPAND_SZ != dwKeyValueType)
     {
         cout << "It is impossible!" << endl;
         cout << dwKeyValueType << endl;
         RegCloseKey(hKey);
         return -;
     }

     char *lpszKeyValueData = new char[dwKeyValueDataSize + ];
     memset(lpszKeyValueData, , dwKeyValueDataSize + );

     iRet = RegQueryValueExA(hKey, lpszKeyValueName, NULL, &dwKeyValueType, (LPBYTE)lpszKeyValueData, &dwKeyValueDataSize);
     if (ERROR_SUCCESS != iRet)
     {
         cout << "RegQueryValueExA Error!" << endl;
         RegCloseKey(hKey);
         delete []lpszKeyValueData;
         return -;
     }

     unsigned int iLen = strlen(lpszPathValue);
     iLen += strlen(lpszKeyValueData);
     char *lpszKeyValueData_New = new char[iLen + ];
     memset(lpszKeyValueData_New, , iLen + );
     sprintf(lpszKeyValueData_New, "%s%s", lpszKeyValueData, lpszPathValue);
     iRet = RegSetValueExA(hKey, lpszKeyValueName, , REG_EXPAND_SZ, (const BYTE*)lpszKeyValueData_New, iLen + );
     if (ERROR_SUCCESS != iRet)
     {
         cout << "RegSetValueExA Error!" << endl;
         RegCloseKey(hKey);
         delete []lpszKeyValueData;
         delete []lpszKeyValueData_New;
         return -;
     }

     delete []lpszKeyValueData_New;
     delete []lpszKeyValueData;

     RegCloseKey(hKey);

     return ;
 }

 int main(int argc, char *argv)
 {
     cout << AddPathEnvValue(";HelloKitty") << endl;
     return ;
 }