registrydlg.cpp

访问和修改注册表中的用户姓名和公司名称

// RegistryDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Registry.h"
#include "RegistryDlg.h"
#include "modify.h"
#include "winreg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

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

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

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

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CRegistryDlg dialog

CRegistryDlg::CRegistryDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CRegistryDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CRegistryDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CRegistryDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CRegistryDlg)
		// NOTE: the ClassWizard will add DDX and DDV calls here
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CRegistryDlg, CDialog)
	//{{AFX_MSG_MAP(CRegistryDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_MODIFY, OnModify)
	ON_BN_CLICKED(IDC_QUERY, OnQuery)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CRegistryDlg message handlers

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

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	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);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

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

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CRegistryDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CRegistryDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CRegistryDlg::OnModify() 
{
	// TODO: Add your control notification handler code here
	CModify dlg;
	if(dlg.DoModal()==IDOK)
	{//按给定的值修改注册表
	   CString str_username=dlg.m_Username;
	   CString str_company=dlg.m_Company;

	   HKEY hKEY;
       LPCTSTR data_Set="Software\\Microsoft\\Ms Setup(ACME)\\User Info\\";
	   //打开与路径data_Set相关的hKEY
		long ret0=::RegOpenKeyEx(HKEY_CURRENT_USER,//根键名称
					data_Set,//要访问的键的位置;
					0,//必须为零;
					KEY_WRITE,//以写方式访问注册表
					&hKEY);//hKEY保存此函数所打开的键的句柄

		if(ret0!=ERROR_SUCCESS)
		{//若无法打开,则终止程序运行
			MessageBox("错误:无法打开相应的hKEY");
			return;
		}
		//修改用户姓名
		LPBYTE username_Set =CStringToLPBYTE(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 = CStringToLPBYTE(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;
		}
	}


}
//LPBYTE CRegistryDlg::CStringToLPBYTE(CString str)
LPBYTE CRegistryDlg::CStringToLPBYTE(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;
}


void CRegistryDlg::OnQuery() 
{
	// TODO: Add your control notification handler code here
		// TODO: Add your control notification handler code here
	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(ret2!=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);
	
}