// RegistryDlg.cpp : implementation file
//

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

#include "Modify.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 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::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 (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;
}