C++读取注册表的实现方法

转载  2017-10-11   作者:luoweifu   我要评论

这篇文章主要介绍了C++读取注册表的实现方法的相关资料,希望通过本文能帮助到大家,让大家理解掌握实现这样的功能,需要的朋友可以参考下

C++读取注册表

GetRegValue.h:

#ifndef __GETREGVALUE_H__
#define __GETREGVALUE_H__

#include <string>

//---------------------------------------------------------------
//function: 
//     GetRegValue 获取注册表中指定键的值
//Access:
//      public 
//Parameter:
//     [in] int nKeyType - 注册表项的类型,传入的参数只可能是以下数值:
//               0:HKEY_CLASSES_ROOT
//               1:HKEY_CURRENT_USER
//               2:HKEY_LOCAL_MACHINE
//               3:HKEY_USERS
//               4:HKEY_PERFORMANCE_DATA
//               5:HKEY_CURRENT_CONFIG
//               6:HKEY_DYN_DATA
//               7:HKEY_CURRENT_USER_LOCAL_SETTINGS
//               8:HKEY_PERFORMANCE_TEXT
//               9:HKEY_PERFORMANCE_NLSTEXT
//     [in] const std::string & strUrl - 要查找 的键的路径
//     [in] const std::string & strKey - 指定的键
//Returns:
//     std::string - 指定键的值
//Remarks:
//     ...
//author:  luoweifu
//---------------------------------------------------------------
std::string GetRegValue(int nKeyType, const std::string& strUrl, const std::string& strKey);

//可移植版本 wstring => string
std::string ws2s(const std::wstring& ws);

//可移植版本 string => wstring
std::wstring s2ws(const std::string& s);


#endif //__GETREGVALUE_H__

GetRegValue.cpp

#include "stdafx.h"
#include <Windows.h>
#include "GetRegValue.h"

//可移植版本 wstring => string
std::string ws2s(const std::wstring& ws)
{
  std::string curLocale = setlocale(LC_ALL, "");
  const wchar_t* _Source = ws.c_str();
  size_t _Dsize = wcstombs(NULL, _Source, 0) + 1;
  char *_Dest = new char[_Dsize];
  memset(_Dest,0,_Dsize);
  wcstombs(_Dest,_Source,_Dsize);
  std::string result = _Dest;
  delete []_Dest;
  setlocale(LC_ALL, curLocale.c_str());
  return result;
}

//可移植版本 string => wstring
std::wstring s2ws(const std::string& s)
{
  std::string curLocale = setlocale(LC_ALL, ""); 
  const char* _Source = s.c_str();
  size_t _Dsize = mbstowcs(NULL, _Source, 0) + 1;
  wchar_t *_Dest = new wchar_t[_Dsize];
  wmemset(_Dest, 0, _Dsize);
  mbstowcs(_Dest,_Source,_Dsize);
  std::wstring result = _Dest;
  delete []_Dest;
  setlocale(LC_ALL, curLocale.c_str());
  return result;
}

std::string GetRegValue(int nKeyType, const std::string& strUrl, const std::string& strKey)
{
  std::string strValue("");
  HKEY hKey = NULL;
  HKEY hKeyResult = NULL;
  DWORD dwSize   = 0;
  DWORD dwDataType = 0;
  std::wstring wstrUrl = s2ws(strUrl);
  std::wstring wstrKey = s2ws(strKey);

  switch(nKeyType)
  {
  case 0:
    {
      hKey = HKEY_CLASSES_ROOT;
      break;
    }
  case 1:
    {
      hKey = HKEY_CURRENT_USER;
      break;
    }
  case 2:
    {
      hKey = HKEY_LOCAL_MACHINE;
      break;
    }
  case 3:
    {
      hKey = HKEY_USERS;
      break;
    }
  case 4:
    {
      hKey = HKEY_PERFORMANCE_DATA;
      break;
    }
  case 5:
    {
      hKey = HKEY_CURRENT_CONFIG;
      break;
    }
  case 6:
    {
      hKey = HKEY_DYN_DATA;
      break;
    }
  case 7:
    {
      hKey = HKEY_CURRENT_USER_LOCAL_SETTINGS;
      break;
    }
  case 8:
    {
      hKey = HKEY_PERFORMANCE_TEXT;
      break;
    }
  case 9:
    {
      hKey = HKEY_PERFORMANCE_NLSTEXT;
      break;
    }
  default:
    {
      return strValue;
    }
  }

  //打开注册表
  if(ERROR_SUCCESS == ::RegOpenKeyEx(hKey, wstrUrl.c_str(), 0, KEY_QUERY_VALUE, &hKeyResult))
  {
    // 获取缓存的长度dwSize及类型dwDataType
    ::RegQueryValueEx(hKeyResult, wstrKey.c_str(), 0, &dwDataType, NULL, &dwSize); 
    switch (dwDataType)
    {
    case REG_MULTI_SZ:
      {
        //分配内存大小
        BYTE* lpValue = new BYTE[dwSize];
        //获取注册表中指定的键所对应的值
        LONG lRet = ::RegQueryValueEx(hKeyResult, wstrKey.c_str(), 0, &dwDataType, lpValue, &dwSize);
        delete[] lpValue;
        break;
      }
    case REG_SZ:
      {
        //分配内存大小
        wchar_t* lpValue = new wchar_t[dwSize];
        memset(lpValue, 0, dwSize * sizeof(wchar_t));
        //获取注册表中指定的键所对应的值
        if (ERROR_SUCCESS == ::RegQueryValueEx(hKeyResult, wstrKey.c_str(), 0, &dwDataType, (LPBYTE)lpValue, &dwSize))
        {
          std::wstring wstrValue(lpValue);
          strValue = ws2s(wstrValue);
        }
        delete[] lpValue;
        break;
      }
    default:
      break;
    }
  }

  //关闭注册表
  ::RegCloseKey(hKeyResult);


  return strValue;
}

测试代码:

#include "stdafx.h"
#include <string>
#include "GetRegValue.h"

int _tmain(int argc, _TCHAR* argv[])
{
  std::string strValue = GetRegValue(2, "SOFTWARE\\360Safe\\Liveup", "mid");
  return 0;
}

结果:

strValue:

“ebd1360403764c9d48c585ef93a6eacbd89ded596f043f78e54eb0adeba7251d”

如有疑问请留言或者到本站社区交流讨论,感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!

相关文章

  • 深入ORACLE变量的定义与使用的详解

    深入ORACLE变量的定义与使用的详解

    本篇文章是对ORACLE变量的定义与使用进行了详细的分析与介绍,需要的朋友参考下
    2013-05-05
  • C语言获取Linux系统精确时间的方法

    C语言获取Linux系统精确时间的方法

    下面小编就为大家带来一篇C语言获取Linux系统精确时间的方法。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-09-09
  • C++构造函数深度学习

    C++构造函数深度学习

    这篇文章主要为大家详细介绍了C++构造函数,深度学习C++构造函数,感兴趣的小伙伴们可以参考一下
    2016-08-08
  • C++程序中使用Windows系统Native Wifi API的基本教程

    C++程序中使用Windows系统Native Wifi API的基本教程

    这篇文章主要介绍了C++程序中使用Windows系统Native Wifi API的基本教程,包括在程序中控制无线网卡开关的方法,需要的朋友可以参考下
    2016-03-03
  • C++二进制翻转实例分析

    C++二进制翻转实例分析

    这篇文章主要介绍了C++二进制翻转,通过几个实例分析二进制翻转算法的实现技巧,需要的朋友可以参考下
    2014-09-09
  • C++中继承与多态的基础虚函数类详解

    C++中继承与多态的基础虚函数类详解

    这篇文章主要给大家介绍了关于C++中继承与多态的基础虚函数类的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧。
    2017-09-09
  • C++中带空格字符串的输入问题解决

    C++中带空格字符串的输入问题解决

    这篇文章主要给大家介绍了关于如何解决C++中带空格字符串的输入问题,文中通过示例代码介绍的非常详细,需要的朋友可以参考借鉴,下面来一起看看吧。
    2017-04-04
  • C++基础入门教程(九):函数指针之回调

    C++基础入门教程(九):函数指针之回调

    这篇文章主要介绍了C++基础入门教程(九):函数指针之回调,本文讲解了函数的地址、声明函数指针、历史原因、typedef挽救复杂的函数指针等内容,需要的朋友可以参考下
    2014-11-11
  • 黑客帝国数字雨效果VC6源代码分享

    黑客帝国数字雨效果VC6源代码分享

    这篇文章主要介绍了黑客帝国数字雨效果VC6源代码分享,本文直接给出实现代码,Win7下编译通过,效果很酷,需要的朋友可以参考下
    2015-02-02
  • C++开发在IOS环境下运行的LRUCache缓存功能

    C++开发在IOS环境下运行的LRUCache缓存功能

    本文着重介绍如何在XCODE中,通过C++开发在IOS环境下运行的缓存功能。算法基于LRU,最近最少使用,需要的朋友可以参考下
    2012-11-11

最新评论