registrydlg.cpp

来自「《突破Visual C++.NET编程实例五十讲+源文件,初学者学习的好东东！」· C++ 代码 · 共 266 行

// RegistryDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "Registry.h"
#include "RegistryDlg.h"
#include "Modify.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CRegistryDlg 对话框



CRegistryDlg::CRegistryDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CRegistryDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CRegistryDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CRegistryDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_MODIFY, OnBnClickedModify)
	ON_BN_CLICKED(IDC_QUERY, OnBnClickedQuery)
END_MESSAGE_MAP()


// CRegistryDlg 消息处理程序

BOOL CRegistryDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将\“关于...\”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO：在此添加额外的初始化代码
	
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

void CRegistryDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CRegistryDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CRegistryDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CRegistryDlg::OnBnClickedModify()
{
	// TODO: 在此添加控件通知处理程序代码
	CModify dlg;
	if (dlg.DoModal() ==IDOK){//按给定的值修改注册表信息
		CString str_username = dlg.m_Username;
		CString str_company  = dlg.m_Company;
		//定义有关的hKEY，程序结束时关闭
		HKEY hKEY;
		LPCTSTR data_Set = "Software\\Microsoft\\Ms Setup (ACME)\\User Info\\";
		long ret0 = ::RegOpenKeyEx(HKEY_CURRENT_USER,//根键名称
				data_Set,	//要访问的键的位置；
				0,			//必须为0；
				KEY_WRITE,	//以写方式访问注册表；
				&hKEY);	//hKEY保存此函数所打开的键的句柄。
		if (ret0 != ERROR_SUCCESS){//若无法打开，则终止程序运行
			MessageBox("错误：无法打开相关的hKEY!");
			return;
		}
		//修改用户姓名
		LPBYTE username_Set = CString_To_LPBYTE(str_username);
		DWORD type_1 = REG_SZ;
		DWORD cbData_1 = str_username.GetLength()+1;
		long ret1 = ::RegSetValueEx(hKEY,//已打开的键的句柄
			"DefName",		//要修改的键值名
			NULL,			
			type_1,		//新数据的类型
			username_Set,	//新的键值
			cbData_1);		//新值的数据长度
		if (ret1 != ERROR_SUCCESS){//若无法修改，则终止程序运行
			MessageBox("错误：无法修改用户姓名");
			return;
		}
		//修改用户年龄
		LPBYTE company_Set = CString_To_LPBYTE(str_company);
		DWORD type_2 = REG_SZ;
		DWORD cbData_2 = str_company.GetLength()+1;
		long ret2 = ::RegSetValueEx(hKEY,//已打开的键的句柄
			"DefCompany",		//要修改的键值名
			NULL,			
			type_2,		//新数据的类型
			company_Set,	//新的键值
			cbData_2);		//新值的数据长度
		if (ret2 != ERROR_SUCCESS){//若无法修改，则终止程序运行
			MessageBox("错误：无法修改用户年龄");
			return;
		}
	}
}

void CRegistryDlg::OnBnClickedQuery()
{
	// TODO: 在此添加控件通知处理程序代码
	HKEY hKEY;//定义有关的hKEY，查询结束时关闭
	LPCTSTR data_Set = "Software\\Microsoft\\Ms Setup (ACME)\\User Info\\";
	//打开与路径data_Set相关的hKEY
	long ret0 = ::RegOpenKeyEx(HKEY_CURRENT_USER,//根键名称
			data_Set,	//要访问的键的位置；
			0,			//必须为0；
			KEY_READ,	//以查询的方式访问注册表；
			&hKEY);	//hKEY保存此函数所打开的键的句柄。
	if (ret0 != ERROR_SUCCESS){//若无法打开，则终止程序运行
		MessageBox("错误：无法打开相关的hKEY!");
		return;
	}
	//查询用户名称信息
	LPBYTE username_Get = new BYTE[40];
	DWORD type_1 = REG_SZ;
	DWORD cbData_1 = 40;
	long ret1 = ::RegQueryValueEx(hKEY,//所打开的键的句柄
			"DefName",		//要查询的键值名
			NULL,			
			&type_1,		//查询数据的类型
			username_Get,	//保存所查询的数据
			&cbData_1);		//预设置的数据长度
	if (ret1 != ERROR_SUCCESS){
		MessageBox("错误：无法查询有关注册表信息!");
		return;
	}
	//查询用户年龄信息数据
	LPBYTE company_Get = new BYTE[40];
	DWORD type_2 = REG_SZ;
	DWORD cbData_2 = 40;
	long ret2 = ::RegQueryValueEx(hKEY,//所打开的键的句柄
			"DefCompany",		//要查询的键值名
			NULL,			
			&type_2,		//查询数据的类型
			company_Get,	//保存所查询的数据
			&cbData_2);		//预设置的数据长度
	if (ret1 != ERROR_SUCCESS){
		MessageBox("错误：无法查询有关注册表信息!");
		return;
	}
	//将查询结果转换为CString类型
	CString str_username = CString(username_Get);
	CString str_company = CString(company_Get);
	delete[] username_Get;
	delete[] company_Get;
	//关闭打开的hKEY
	::RegCloseKey(hKEY);
	//显示查询信息
	CString value = "用户姓名" + str_username + "，用户年龄" + str_company;
	AfxMessageBox(value,MB_OK|MB_ICONINFORMATION);
}

LPBYTE CRegistryDlg::CString_To_LPBYTE(CString str)
{
	LPBYTE lpb = new BYTE[str.GetLength()+1];
	for (int i = 0;i <str.GetLength();i++)
		lpb[i] = str[i];
	lpb[str.GetLength()] = 0;
	return lpb;
	//return LPBYTE();
}