test_intrdemodlg.cpp

Windows 2000/XP WDM設備驅動程式開發 使用 Numega 公司出版的軟體快速建置驅動程式

// Test_intrdemoDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Test_intrdemo.h"
#include "Test_intrdemoDlg.h"

#include "..\common.h"
#include <winioctl.h>

HWND Print_Event::ViewHwnd = NULL;

BOOL KeepRunning;

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

DWORD WINAPI ServiceThread(PVOID lParm)
{
	HANDLE		m_hDevice,m_hEvent;
	ULONG		nItems, nBytesRead;
	TIMESTAMP	tsbuf[FIFOSIZE];
	PTIMESTAMP	pts;
	CString		msg;

	KeepRunning=FALSE;
	m_hDevice = CreateFile("\\\\.\\IntrDemo0",
			GENERIC_READ,
            FILE_SHARE_READ,
            NULL,
            OPEN_EXISTING,
            0,
            NULL);
	if (m_hDevice == INVALID_HANDLE_VALUE)
	{
		msg.Format("Cannot open device, error 0x%x\n", GetLastError());
		Print_Event::SendEvent(msg);
		return 1;
	}

	// Create an event for notification
	m_hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
	if( m_hEvent==NULL)
	{
		msg.Format("Cannot Create Event, error 0x%x\n", GetLastError());
		Print_Event::SendEvent(msg);
		CloseHandle(m_hDevice);
		return 1;
	}
	if (! DeviceIoControl(
			m_hDevice, 
			IOCTL_SET_NOTIFICATION_EVENT, 
			&m_hEvent, 
			sizeof(m_hEvent),
			NULL,
			0,
			&nBytesRead,
			NULL
			)	)
	{
		msg.Format("SET_NOTIFICATION_EVENT failed: %x\n", GetLastError());
		Print_Event::SendEvent(msg);
		CloseHandle(m_hDevice);
		CloseHandle(m_hEvent);
		return 1;
	}

	KeepRunning=TRUE;
	while (TRUE)
	{
		while (WaitForSingleObject(m_hEvent, 0)!=WAIT_OBJECT_0)
		{
			if (!KeepRunning) {
				CloseHandle(m_hDevice);
				CloseHandle(m_hEvent);
				return 0;
			}
		};
		msg.Format("New event:\n");
		if(KeepRunning) Print_Event::SendEvent(msg);

		if (!DeviceIoControl(
			m_hDevice, 
			IOCTL_GET_TIMESTAMP_DATA, 
			NULL, 
			0, 
			tsbuf, 
			sizeof(tsbuf),
			&nBytesRead,
			NULL) )
		{
			msg.Format("DeviceIoControl failed %x\n", GetLastError());
			if(KeepRunning) Print_Event::SendEvent(msg);
			CloseHandle(m_hDevice);
			CloseHandle(m_hEvent);
			return 1;
		}

		nItems = nBytesRead/sizeof(TIMESTAMP);

		for (pts = tsbuf; nItems; nItems--, pts++)
		{
			msg.Format("Interrupt time %08x (High=%08x) count %x\n", pts->ts_time, pts->ts_interrupt_count);
			if(KeepRunning) Print_Event::SendEvent(msg);
		}
	}
}

/////////////////////////////////////////////////////////////////////////////
// CTest_intrdemoDlg dialog
CTest_intrdemoDlg::CTest_intrdemoDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CTest_intrdemoDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CTest_intrdemoDlg)
		// 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);
}
CTest_intrdemoDlg::~CTest_intrdemoDlg()
{
}

void CTest_intrdemoDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CTest_intrdemoDlg)
	DDX_Control(pDX, IDC_LIST1, m_Event);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CTest_intrdemoDlg, CDialog)
	//{{AFX_MSG_MAP(CTest_intrdemoDlg)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_WM_DESTROY()
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_PRINTEVENT,OnPrintEvent)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTest_intrdemoDlg message handlers

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

	// 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
	KeepRunning=FALSE;
	CWnd* m_cWnd=AfxGetMainWnd();
	Print_Event::ViewHwnd = m_cWnd->m_hWnd;
	DWORD Tid;		  
	Thread1=CreateThread(0, 0x1000, ServiceThread, 0, 0, &Tid);

	return TRUE;  // return TRUE  unless you set the focus to a control
}

// 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 CTest_intrdemoDlg::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 CTest_intrdemoDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}
void CTest_intrdemoDlg::OnDestroy() 
{
	CDialog::OnDestroy();
	
	// TODO: Add your message handler code here
	if (KeepRunning)
	{
		KeepRunning=FALSE;
		WaitForSingleObject(Thread1,INFINITE);
	}
}

LONG CTest_intrdemoDlg::OnPrintEvent(UINT, LONG lParam)
{
	Print_Event* pEvent = (Print_Event*)lParam;
	if (pEvent) 
	{
		m_Event.AddString(pEvent->Message);
		delete pEvent;
	}
	return 0;
}