myapp.cpp

实战DeviceIoControl：通过API访问设备驱动程序

// MyApp.cpp : Defines the entry point for the application.
//

#include "stdafx.h"
#include "resource.h"
#include <commctrl.h>

#pragma comment(lib, "commctrl")

#define MAX_LOADSTRING 100

#define IDC_CMDBAR 0x101
HWND hWndCmdBar;

void WriteToDriver( );
void ReadFromDriver( );
void HandleIOCTL( );

#define IOCTL_DRIVER_DEMO   42


// Global Variables:
HINSTANCE hInst;								// current instance
TCHAR szTitle[MAX_LOADSTRING];								// The title bar text
TCHAR szWindowClass[MAX_LOADSTRING];								// The title bar text

// Forward declarations of functions included in this code module:
ATOM				MyRegisterClass(HINSTANCE hInstance);
BOOL				InitInstance(HINSTANCE, int);
LRESULT CALLBACK	WndProc(HWND, UINT, WPARAM, LPARAM);





void WriteToDriver( )
{
	DWORD dwWritten;
	TCHAR *tcString=L"Demo String...";
	HANDLE hDrv=CreateFile(L"DEM1:",GENERIC_WRITE,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
	if (INVALID_HANDLE_VALUE == hDrv) {
		OutputDebugString(L"Failed to open Driver...\n");
	} else {
		WriteFile(hDrv,(LPVOID)tcString,lstrlen(tcString)*sizeof(TCHAR),&dwWritten,NULL);
	}
	CloseHandle(hDrv);
}

void ReadFromDriver( )
{
	DWORD dwRead;
	TCHAR tcTemp[30];
	HANDLE hDrv=CreateFile(L"DEM1:",GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
	if (INVALID_HANDLE_VALUE == hDrv) {
		OutputDebugString(L"Failed to open Driver...\n");
	} else {
		memset(tcTemp,0x00,30*sizeof(TCHAR));
		ReadFile(hDrv,tcTemp,30,&dwRead,NULL);
		MessageBox(NULL,tcTemp,L"Demo Data",MB_OK);
	}
	CloseHandle(hDrv);
}

void HandleIOCTL( )
{
	HANDLE hDrv=CreateFile(L"DEM1:",GENERIC_WRITE,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
	TCHAR tcBuffer[10];
	DWORD dwBytesReturned;
	
	lstrcpy(tcBuffer,L"Hello");
	
	BOOL bRet=DeviceIoControl(
		hDrv, 
		IOCTL_DRIVER_DEMO, 
		tcBuffer, 
		lstrlen(tcBuffer)*sizeof(TCHAR), 
		tcBuffer, 
		lstrlen(tcBuffer)*sizeof(TCHAR),
		&dwBytesReturned, 
		NULL);
	
	MessageBox(NULL,tcBuffer,L"IOCTL Test",MB_OK);
	CloseHandle(hDrv);
	
}






int WINAPI WinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPTSTR     lpCmdLine,
                     int       nCmdShow)
{
 	// TODO: Place code here.
	MSG msg;
	HACCEL hAccelTable;

	// Initialize global strings
	LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
	LoadString(hInstance, IDC_MYAPP, szWindowClass, MAX_LOADSTRING);
	MyRegisterClass(hInstance);

	// Perform application initialization:
	if (!InitInstance (hInstance, nCmdShow)) 
	{
		return FALSE;
	}

	hAccelTable = LoadAccelerators(hInstance, (LPCTSTR)IDC_MYAPP);

	// Main message loop:
	while (GetMessage(&msg, NULL, 0, 0)) 
	{
		if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg)) 
		{
			TranslateMessage(&msg);
			DispatchMessage(&msg);
		}
	}

	return msg.wParam;
}



//
//  FUNCTION: MyRegisterClass()
//
//  PURPOSE: Registers the window class.
//
//  COMMENTS:
//
//    This function and its usage is only necessary if you want this code
//    to be compatible with Win32 systems prior to the 'RegisterClassEx'
//    function that was added to Windows 95. It is important to call this function
//    so that the application will get 'well formed' small icons associated
//    with it.
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
	WNDCLASS wc;

    wc.style = CS_HREDRAW | CS_VREDRAW;
    wc.lpfnWndProc = (WNDPROC) WndProc;
    wc.cbClsExtra = 0;
    wc.cbWndExtra = 0;
    wc.hInstance = hInstance;
    wc.hIcon = 0;
    wc.hCursor = 0;
    wc.hbrBackground = (HBRUSH) GetStockObject(WHITE_BRUSH);
    wc.lpszMenuName = 0;
    wc.lpszClassName = szWindowClass;

	return RegisterClass(&wc);
}

//
//   FUNCTION: InitInstance(HANDLE, int)
//
//   PURPOSE: Saves instance handle and creates main window
//
//   COMMENTS:
//
//        In this function, we save the instance handle in a global variable and
//        create and display the main program window.
//
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
   HWND hWnd;

   hInst = hInstance; // Store instance handle in our global variable

   hWnd = CreateWindow(szWindowClass, szTitle, WS_VISIBLE,
      0, 0, CW_USEDEFAULT, CW_USEDEFAULT, NULL, NULL, hInstance, NULL);

   if (!hWnd)
   {
      return FALSE;
   }

   ShowWindow(hWnd, nCmdShow);
   UpdateWindow(hWnd);

   return TRUE;
}

//
//  FUNCTION: WndProc(HWND, unsigned, WORD, LONG)
//
//  PURPOSE:  Processes messages for the main window.
//
//  WM_COMMAND	- process the application menu
//  WM_PAINT	- Paint the main window
//  WM_DESTROY	- post a quit message and return
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
	PAINTSTRUCT ps;
	HDC hdc;
	TCHAR szHello[MAX_LOADSTRING];
	LoadString(hInst, IDS_HELLO, szHello, MAX_LOADSTRING);

	switch (message) 
	{
	case WM_COMMAND:
		switch(LOWORD(wParam)) {
		case ID_FILE_EXIT:
			PostQuitMessage(0);
            break;
		case ID_DRIVER_WRITE:
			WriteToDriver( );
            break;
		case ID_DRIVER_READ:
			ReadFromDriver( );
            break;
		case ID_DRIVER_IOCTL:
			HandleIOCTL( );
            break;
		}		
		break;
		case WM_CREATE:
			hWndCmdBar=CommandBar_Create(hInst,hWnd,IDC_CMDBAR);
			CommandBar_InsertMenubar(hWndCmdBar,hInst,IDR_MENU,0);
			break;
		case WM_PAINT:
			hdc = BeginPaint(hWnd, &ps);
			// TODO: Add any drawing code here...
			RECT rt;
			GetClientRect(hWnd, &rt);
			DrawText(hdc, szHello, _tcslen(szHello), &rt, DT_CENTER);
			EndPaint(hWnd, &ps);
			break;
		case WM_DESTROY:
			PostQuitMessage(0);
			break;
		default:
			return DefWindowProc(hWnd, message, wParam, lParam);
   }
   return 0;
}