loaddrive.cpp

Windows CE上的加载驱动的类

// LoadDrive.cpp: implementation of the CLoadDrive class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "swt.h"
#include "LoadDrive.h"

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

extern HANDLE	m_hSendReadIdEvent;//读传感器事件
extern HANDLE	m_hSetFreqNumEvent;
extern HANDLE	m_hSendReadNetEvent;//读网络id事件
extern HANDLE	m_hSendSequReadEvent;//连续读事件，
extern HANDLE	m_hSendStopEvent;//停止事件
extern HANDLE	m_hSendPCEvent;
extern HANDLE	m_hEventArray[5];
extern HANDLE	hUSBFile;

/*
; Simple Driver 注册表信息
[HKEY_LOCAL_MACHINE\Drivers\BuiltIn\SPL]
"Prefix"="SPL"
"Dll"="\\SimpleDriver.dll"
"FriendlyName"="Simple Driver"
"Index"=dword:0
"Order"=dword:0
*/

// 向注册表写入一个字符串
#define WRITE_REG_SZ(Name, Value) RegSetValueEx(hDrvKey, Name, 0, REG_SZ, (LPBYTE)Value, \
(wcslen(Value)+1)*sizeof(WCHAR))

// 向注册表写入一个 DWORD 型的值
#define WRITE_REG_DWORD(Name, Value) {DWORD dwValue = Value; RegSetValueEx( hDrvKey, Name, \
									  0, REG_DWORD, (LPBYTE)&dwValue, sizeof(DWORD)); }
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CLoadDrive::CLoadDrive()
{

	hUSBFile = INVALID_HANDLE_VALUE;	
	HandleDriver = INVALID_HANDLE_VALUE;		/* 驱动程序句柄     */
	hDrvKey = NULL;							/* 注册表操作句柄   */
	szRegKey = L"Drivers\\BuiltIn\\CPL";
	szPrefix = L"CPL";
	szDLL = L"\\cpldDriver.dll";
	szFriendlyName = L"Key Driver";
	DrvIndex = 1;
	DrvOrder = 0;

	/*************************************
			初始化消息事件
	*************************************/
	
	m_hSendReadIdEvent=CreateEvent(NULL,FALSE,FALSE,NULL);
	m_hSendReadNetEvent=CreateEvent(NULL,FALSE,FALSE,NULL);
	m_hSetFreqNumEvent=CreateEvent(NULL,FALSE,FALSE,NULL);
	m_hSendSequReadEvent=CreateEvent(NULL,FALSE,FALSE,NULL);
	m_hSendStopEvent=CreateEvent(NULL,FALSE,FALSE,NULL);
	m_hSendPCEvent=CreateEvent(NULL,FALSE,FALSE,NULL);
	m_hEventArray[0]=m_hSendReadIdEvent;
	m_hEventArray[1]=m_hSendReadNetEvent;
	m_hEventArray[2]=m_hSendSequReadEvent;
	m_hEventArray[3]=m_hSendStopEvent;
	m_hEventArray[4]=m_hSendPCEvent;

	
}

CLoadDrive::~CLoadDrive()
{

}

bool CLoadDrive::AutoLoad()
{

	LONG nErr;
	DWORD dwDisp;
	// 创建注册表键HKEY_LOCAL_MACHINE\Drivers\BuiltIn\SPL
    nErr = RegCreateKeyEx(HKEY_LOCAL_MACHINE, szRegKey, 0, L"", 0, 0, NULL, &hDrvKey, &dwDisp);
	if (nErr != ERROR_SUCCESS)
	{
		AfxMessageBox(_T("创建驱动注册表子键失败!"));	
		return false;
	}
	else
	{	
		// 往注册表添加内容
        WRITE_REG_SZ( L"Prefix", szPrefix);						/* 写入 "Prefix"="SPL" */ 
        WRITE_REG_SZ( L"Dll", szDLL);							/* 写入 "Dll"="\\CPLDriver.dll" */
        WRITE_REG_SZ( L"FriendlyName", szFriendlyName);			/* 写入 "FriendlyName"="Simple Driver" */
        WRITE_REG_DWORD( L"Order", DrvOrder);					/* 写入 "Index"=dword:0 */
        WRITE_REG_DWORD( L"Index", DrvIndex);					/* 写入 "Order"=dword:0 */
	}
	
	HandleDriver = ActivateDeviceEx(szRegKey, NULL, 0, NULL);	/* 加载 CPLDriver 驱动 */
	if (HandleDriver == INVALID_HANDLE_VALUE)
		AfxMessageBox(_T("加载CPLDriver failed."));
}

bool CLoadDrive::SendComd(BYTE *comd, DWORD writetonum, DWORD *writenum)
{

	if (hUSBFile == INVALID_HANDLE_VALUE)							/* 检查驱动是否已打开 */
	{
		AfxMessageBox(_T("CPLDriver 驱动未打开!"));
		return false;
	}
	BOOL ret = WriteFile(hUSBFile, comd, writetonum, writenum, NULL);	/* 将字节写入驱动 */
// 	Sleep(3);
// 	ret =WriteFile(hUSBFile, comd, writetonum, writenum, NULL);	/* 将字节写入驱动 */
// 	Sleep(3);
// 	ret =WriteFile(hUSBFile, comd, writetonum, writenum, NULL);	/* 将字节写入驱动 */
	if ((ret == FALSE)/* || (*writenum != writetonum)*/)	
	{
		AfxMessageBox(_T("WriteDataError"));
		return false;
	}
	else
	{	
		return true;
	}
		
}
void CLoadDrive::RecvComd(BYTE *szBuffer,DWORD readtonum,DWORD readnum)
{
	if (hUSBFile == INVALID_HANDLE_VALUE)							/* 检查驱动是否已打开 */
	{
		AfxMessageBox(_T("CPLDriver 驱动未打开!"));
		return;
	}
	BOOL ret = ReadFile(hUSBFile, szBuffer, readtonum, &readnum, NULL);		/* 从驱动中读取一个字节 */
	if ((ret == FALSE))
	{
		AfxMessageBox(_T("error"));
		return;
	}	
	else
	{
		CString str;
		str.Format(_T("%d",actlen));
		AfxMessageBox(str);
	}

}

bool CLoadDrive::OpenDrive(CWnd *pPortOwner)
{
	m_pOwner=pPortOwner;
	// 打开 CPLDriver 驱动
	hUSBFile = CreateFile(TEXT("CPL1:"), GENERIC_READ | GENERIC_WRITE, 0, 
					   NULL, OPEN_EXISTING, 0, 0);
	if (hUSBFile == INVALID_HANDLE_VALUE)							/* 检查驱动是否已打开 */
	{
		AfxMessageBox(_T("CPLDriver 驱动未打开!"));
		return false;
	}
}

bool CLoadDrive::Unload()
{

	if (hUSBFile != INVALID_HANDLE_VALUE)
	{
		CloseHandle(hUSBFile);											/* 关闭驱动 */
		BOOL ret = DeactivateDevice(hUSBFile);					/* 卸载 CPLDriver 驱动 */
		if (ret == TRUE)
		{	
			RegCloseKey(hDrvKey);									/* 关闭注册表键		  */
			RegDeleteKey(HKEY_LOCAL_MACHINE, szRegKey);				/* 删除注册表键的内容 */
			HandleDriver = INVALID_HANDLE_VALUE;
			return true;
		}
		else
		{
			AfxMessageBox(_T("卸载CPLDriver failed."));
			return false;
		}
	}
}