#include "stdafx.h"
#include "GrobalFunc.h"
#include "OperateReg.h"

COperateReg::COperateReg()
{
}

COperateReg::~COperateReg()
{
}

int COperateReg::GetRegKey()
{
	//打開路徑
	psKey="SOFTWARE\\qsards\\";
	//訪問注冊表，lKey則保存此函數所打開的鍵的句柄
	lResult=(::RegOpenKeyEx(HKEY_LOCAL_MACHINE,psKey,0,KEY_READ,&lKey));
	if(lResult!=ERROR_SUCCESS)//如果無法打開lKey,則中止程序的執行
	{
		return -2;
	}
    //查詢有關的數據
	value=new BYTE[20]; //查詢得到鍵值存放處
	DWORD type_1=REG_SZ;//定義數據類型
	DWORD cbData_1=20;  //定義數據長度

	long lResult=::RegQueryValueEx(lKey,keyname,NULL,&type_1,value,&cbData_1);
	if(lResult!=ERROR_SUCCESS)
	{
		return -1;
	}
	keyvalue=CString(value);
    delete[] value;

	//程序結束，關閉打開的lKey
	::RegCloseKey(lKey);
	return 0;
}

int COperateReg::SetRegKey()
{
	//打開路徑
	psKey="SOFTWARE\\qsards\\";
	//訪問注冊表，lKey則保存此函數所打開的鍵的句柄
	lResult=(::RegOpenKeyEx(HKEY_LOCAL_MACHINE,psKey,0,KEY_WRITE,&lKey));
	if(lResult!=ERROR_SUCCESS)//如果無法打開lKey,則中止程序的執行
	{
		return -2;
	}

    //設置有關的數據
    value=CString_To_LPBYTE(CString(keyvalue));//
	DWORD type_1=REG_SZ;//定義數據類型

	long lResult=::RegSetValueEx(lKey,keyname,NULL,type_1,value,CString(value).GetLength());
	if(lResult!=ERROR_SUCCESS)
	{
		return -1;
	}	
	//程序結束，關閉打開的lKey
	::RegCloseKey(lKey);
	return 0;
}