C/C++ 实现读写注册表

原创

LyShark 2022-05-23 13:51:43 ©著作权

文章标签 C/C++ 基础知识 microsoft 键值数据 文章分类 C/C++ 后端开发

©著作权归作者所有：来自51CTO博客作者LyShark的原创作品，请联系作者获取转载授权，否则将追究法律责任

读注册表

// 初始化
    HKEY hKey_return = NULL;

    // 打开注册表失败
    if(ERROR_SUCCESS != RegOpenKeyEx(
        HKEY_LOCAL_MACHINE,             // 注册表根键句柄（也可以是 HKEY_CLASSES_ROOT、HKEY_CURRENT_USER、HKEY_USERS、HKEY_CURRENT_CONFIG） 
        去掉第一个参数后的注册表路径,     // 注册表路径（开头无'\'，结尾有'\'）
        0,                              // 保留
        KEY_READ,                       // 只读
        &hKey_return                    // 接收打开键的指针（RegQueryValueEx 会用到）
    )){
        cout << "RegOpenKeyEx failed." << endl;
        return false;
    }

    // 打开键失败
    char keyValue[256];
    DWORD keySzType;
    DWORD keySize;
    if(ERROR_SUCCESS != RegQueryValueEx(
        hKey_return,                    // 由 RegOpenKeyEx 返回
        注册表键名,                      // 键名
        0,                              // 不定义值类型
        &keySzType,                     // 接收键类型
        (LPBYTE)&keyValue,              // 接收键值
        &keySize                        // 接收键值的字节大小
    )){
        cout << "RegQueryValueEx failed." << endl;
        return false;
    }

    // 打印注册表键值
    cout << keyValue << endl;

写注册表: 这是一个例子，有多余的代码。功能大概是遍历了某个注册表键下的所有键值，酌情修改其中的某个数据 ????

BOOL CDesktopController::CheckShellFloders()
{
  std::map<std::wstring, std::wstring> mapShellFolders = {
  {L"AppData",      L"%USERPROFILE%\\AppData\\Roaming"},
  {L"Cache",        L"%USERPROFILE%\\AppData\\Local\\Microsoft\\Windows\\INetCache"},
  {L"Cookies",      L"%USERPROFILE%\\AppData\\Local\\Microsoft\\Windows\\INetCookies"},
  {L"Desktop",      L"%USERPROFILE%\\Desktop"},
  {L"Favorites",      L"%USERPROFILE%\\Favorites"},
  {L"History",      L"%USERPROFILE%\\AppData\\Local\\Microsoft\\Windows\\History"},
  {L"Local AppData",    L"%USERPROFILE%\\AppData\\Local"},
  {L"My Music",     L"%USERPROFILE%\\Music"},
  {L"My Pictures",    L"%USERPROFILE%\\Pictures"},
  {L"My Video",     L"%USERPROFILE%\\Videos"},
  {L"NetHood",      L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Network Shortcuts"},
  {L"Personal",     L"%USERPROFILE%\\Documents"},
  {L"PrintHood",      L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Printer Shortcuts"},
  {L"Programs",     L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Start Menu\\Programs"},
  {L"Recent",       L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Recent"},
  {L"SendTo",       L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\SendTo"},
  {L"Start Menu",     L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Start Menu"},
  {L"Startup",      L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Start Menu\\Programs\\Startup"},
  {L"Templates",      L"%USERPROFILE%\\AppData\\Roaming\\Microsoft\\Windows\\Templates"}
  };

  BOOL ret = FALSE;
  BOOL needKillExplorer = FALSE;
  HKEY hTestKey = NULL;
  DWORD cSubKeys = 0;         // 子健个数
  DWORD cValues = 0;          // 键值个数
  TCHAR achValue[MAX_VALUE_NAME];   // 键值名
  DWORD cchValue = MAX_VALUE_NAME;  // 键值最大长度
  TCHAR achData[MAX_VALUE_DATA];    // 键数据
  DWORD cchData = MAX_VALUE_DATA;   // 键数据最大长度
  DWORD cbName = 0;

  // 遍历 HKEY_CURRENT_USER/Software/Microsoft/Windows/CurrentVersion/Explorer/User Shell Folders 
  if (ERROR_SUCCESS != RegOpenKeyEx(
    HKEY_CURRENT_USER,
    L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\User Shell Folders",
    0,
    KEY_ALL_ACCESS,
    &hTestKey))
  {
    LOG_INFO_A("RegOpenKeyEx failed. \n");
    goto end;
  }
  if (ERROR_SUCCESS != RegQueryInfoKey(
    hTestKey,
    NULL,
    NULL,
    NULL,
    &cSubKeys,
    NULL,
    NULL,
    &cValues,
    NULL,
    NULL,
    NULL,
    NULL
  ))
  {
    LOG_INFO_A("RegQueryInfoKey failed. \n");
    goto end;
  }
  for (UINT i = 0; i < cValues; ++i)
  {
    cchValue = MAX_VALUE_NAME;
    cchData = MAX_VALUE_DATA;
    achValue[0] = '\0';
    achValue[0] = '\0';
    if (ERROR_SUCCESS != RegEnumValue(
      hTestKey,
      i,
      achValue,
      &cchValue,
      NULL,
      NULL,
      (LPBYTE)achData,
      &cchData
    ))
    {
      LOG_INFO_A("RegEnumValue failed. \n");
      continue; 
    }

    // 如果有 \\ 开头的网络路径，就恢复
    if (achData[0] == '\\' && achData[1] == '\\')
    {
      if (mapShellFolders.find(achValue) == mapShellFolders.end())
      {
        continue;
      }
      if (ERROR_SUCCESS != RegSetValueEx(
        hTestKey,
        achValue,
        0,
        REG_EXPAND_SZ,
        (const BYTE*)mapShellFolders.find(achValue)->second.c_str(),
        MAX_VALUE_DATA))
      {
        LOG_INFO_A("[LYSM] RegSetValueEx failed. \n");
        continue;
      }
      needKillExplorer = TRUE;
    }
  }

  // 杀安全桌面 - explorer.exe 进程，使注册表改动生效 
  if (needKillExplorer)
  {
    if (!killSecureExplorer())
    {
      goto end;
    }
  }

  ret = TRUE;
end:
  if (hTestKey)
  {
    RegCloseKey(hTestKey);
  }
  return ret;
}