1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
/*++
Module Name: application.c
Abstract:This file contains application which communicates with driversoft, including main and GetDevicePath function.
Environment: User-mode Driver Framework
Time: 20181015
--*/

#include <windows.h>
#include <winioctl.h>
#include <setupapi.h>
#include <stdio.h>
#include <stdlib.h>
#include <conio.h>
#include <string.h>

#include "tchar.h"
#include "../PCIe/Public.h"


HANDLE hDevice = INVALID_HANDLE_VALUE;
PCHAR GetDevicePath(IN LPGUID InterfaceGuid);

int _tmain(int argc, _TCHAR* argv[])
{
	PCHAR DevicePath;
	ULONG nOutput;
	//LPDWORD count = 0;
	UNREFERENCED_PARAMETER(argc);
	UNREFERENCED_PARAMETER(argv);

	printf("Application control test starting...n");
	//利用GUID号找到设备路径
	DevicePath = GetDevicePath((LPGUID)&GUID_DEVINTERFACE_PCIe);
	if (DevicePath == 0)
	{
		printf("ERROR: Get DevicePath failed! ErrNo:%0xn", GetLastError());
		return 0;
	}
	printf("DevicePath gets successfullyn");

	//打开设备，创建相关句柄
	//LPCWSTR name = _T(*DevicePath);
	hDevice = CreateFile(DevicePath,
		GENERIC_READ | GENERIC_WRITE,
		FILE_SHARE_READ | FILE_SHARE_WRITE,
		NULL,
		OPEN_EXISTING,
		FILE_ATTRIBUTE_NORMAL,
		NULL);
	if (hDevice == INVALID_HANDLE_VALUE) {
		//KdPrint("Error opening device:(%0x) returned from createfilen", GetLastError());
		printf("ERROR: hDevice create failed！ErrNo:%0xn", GetLastError());
		return 0;
	}
	printf("Open device successfullyn");
	fprintf(stderr, "n");

	
	//调用WDF框架下DeviceIoControl函数与驱动程序通信
	ULONG x = 0,y;
	if (!DeviceIoControl(hDevice, WDF_DMA, &x, sizeof(ULONG), &y, sizeof(ULONG), &nOutput, NULL))
	{
		printf("ERROR:DeviceIoControl 'Single_Measurement' returns falied! ErrNo: %0x.", GetLastError());
		goto exit;
	}
	printf("the returned data is %dn", y);

exit:
	if (hDevice != INVALID_HANDLE_VALUE)
	{
		CloseHandle(hDevice);
	}

	return 0;
}

//利用GUID号找到设备路径的具体函数
PCHAR GetDevicePath(IN LPGUID InterfaceGuid)
{
	//设别信息集
	HDEVINFO DeviceInfoSet;
	//设备接口信息
	SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;
	//指向设备接口详细信息的指针
	PSP_DEVICE_INTERFACE_DETAIL_DATA DeviceInterfaceDetailData = NULL;
	//函数指向结果返回值
	BOOL bResult;
	//遍历
	DWORD dwIndex = 0 ;

	ULONG Length, RequiredLength = 0;

	//获取指定设备接口类的设备信息集
	DeviceInfoSet = SetupDiGetClassDevs(
		InterfaceGuid,
		NULL,   //Enumerator
		NULL,
		(DIGCF_PRESENT | DIGCF_DEVICEINTERFACE));
	printf("**************************************************n");
	printf("Getting into function 'GetDevicePath'n");
	if (DeviceInfoSet == INVALID_HANDLE_VALUE)
	{
		printf("SetupDiGetClassDevs faild! ErrNo:%dn", GetLastError());
		return 0;
	}

	//枚举设备信息集下的所有设备的设备接口
	//for (dwIndex = 0;;dwIndex++)
	//{
		DeviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);
		
		bResult = SetupDiEnumDeviceInterfaces(
			DeviceInfoSet,
			NULL,
			InterfaceGuid,
			dwIndex, // 第dwIndex个设备
			&DeviceInterfaceData);
		if (!bResult)
		{
			//259错误表示遍历完成
			printf("SetupDiEnumDeviceInterfaces faild! ErrNo:%dn", GetLastError());
			SetupDiDestroyDeviceInfoList(DeviceInfoSet);
			return FALSE;
			//break;
		}
		//获取设备接口详细信息
		//第一次调用，DeviceInterfaceDetailData置为NULL,获取待分配缓冲区的大小
		//根据MSDN,此次调用返回值为FALSE,而且GetLastError为122，如果进行错误检查，需要注意
		SetupDiGetDeviceInterfaceDetail(
			DeviceInfoSet,
			&DeviceInterfaceData, //待获取详细信息的设备接口
			NULL,
			0,
			&RequiredLength,
			NULL);

		DeviceInterfaceDetailData = (PSP_DEVICE_INTERFACE_DETAIL_DATA)LocalAlloc(LMEM_FIXED, RequiredLength);
		if (DeviceInterfaceDetailData == NULL)
		{
			printf("Failed to allocate memory!n");

			SetupDiDestroyDeviceInfoList(DeviceInfoSet);
			return 0;
		}

		DeviceInterfaceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
		Length = RequiredLength;

		//第二次调用，返回设备接口详细信息
		bResult = SetupDiGetDeviceInterfaceDetail(
			DeviceInfoSet,
			&DeviceInterfaceData,
			DeviceInterfaceDetailData,
			Length,
			&RequiredLength,
			NULL);

		if (!bResult)
		{
			printf("SetupDiGetDeviceInterfaceDetail failed(second)！ErrNo:%dn", GetLastError());

			SetupDiDestroyDeviceInfoList(DeviceInfoSet);
			LocalFree(DeviceInterfaceDetailData);
			return 0;
		}
		//PCHAR devPath;
		//PCHAR devPath = DeviceInterfaceDetailData->DevicePath;
		//printf(DeviceInterfaceDetailData->DevicePath);
		printf("Completed working in function 'GetDevicePath'n");
		printf("**************************************************n");
		return DeviceInterfaceDetailData->DevicePath;

		//LocalFree(DeviceInterfaceDetailData);
	//}

	//return NULL;
}

1
2
DeviceIoControl(hDevice, WDF_DMA, &x, sizeof(ULONG), &y, sizeof(ULONG), &nOutput, NULL)