开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 文章/文档 > 通讯编程文档 > 查看源码
Test.C
资源名称：虚拟串口驱动程序.rar　[点击查看]
上传用户：zanmei2
上传日期：2010-03-06
资源大小：775k
文件大小：11k
源码类别：

通讯编程文档

开发平台：

C/C++

Test.C：源码内容
  1. #include <windows.h>
  2. #include <initguid.h>
  3. #include "winioctl.h"
  5. #include <stdio.h>
  6. #include "tchar.h"
  8. #include <setupapi.h>
  9. #include "objbase.h"
  10. #include "..incwdmioctl.h"
  11. #include "conio.h"
  12. #include "test.h"
  14. /*++
  15. 打印使用信息
  16. --*/
  17. void Usage()
  18. {
  19. printf("Usage: test [-e] [-r IP:Port] [-n N] serial device numbern");
  20. printf(" -e:    enumerate serial devicen");
  21. printf(" -r:    set remote IP address and Portn");
  22. printf(" -n:    create virtual device DOS name, eg. COM5n");
  23. printf(" -d:    delete virtual device DOS namen");
  24. }
  25. /*++
  26. 格式化输入的参数
  27. argc - 参数个数
  28. argv - 参数字符串数组
  29. lpCmd - 格式化后的命令结构
  31. Return value:
  32. >=0: Success, 返回将要操作的目标设备序号
  33. -1:  Error, 输入参数格式错误
  34. -2:  Success, 没有返回特定的目标设备号
  35. --*/
  36. int Parse( int argc, char* argv[], PCMDS lpCmd )
  37. {
  38. int n;
  39. int nCmd;
  40. int Ret;
  41. nCmd = 0;
  42. Ret = -2;
  43. lpCmd->TargetDevice = -1;
  44. for ( n = 1; n < argc; n++)
  45. {
  46. if ( argv[ n ][ 0 ] == '-' )
  47. {
  48. switch ( argv[ n ][ 1 ] )
  49. {
  50. case 'e':
  51. lpCmd->bIsEnum = TRUE;
  52. break;
  53. case 'd':
  54. lpCmd->bDeleteDosName = TRUE;
  55. break;
  56. case 'r':
  57. n++;
  58. if ( n >= argc )
  59. return -1;
  60. if ( -1 == ParseAddress ( argv[ n ], &lpCmd->IPAddress, &lpCmd->Port ) )
  61. {
  62. printf("IPAddress and Port errorn");
  63. return -1;
  64. }
  65. lpCmd->bSetRmt = TRUE;
  66. break;
  67. case 'n':
  68. n++;
  69. if ( n >= argc )
  70. return -1;
  71. {
  72. int ComX;
  73. ComX = atoi ( argv[ n ] );
  74. if ( ComX > 255 ) 
  75. {
  76. printf("-n: create DOS name: Com%dn", ComX );
  77. printf("                        ~~--> error!n");
  78. return -1;
  79. }
  80. lpCmd->ComX = ComX;
  81. lpCmd->bCreateDosName = TRUE;
  82. }
  83. break;
  84. default:
  85. return -1;
  86. }
  87. }else
  88. {
  89. if ( Ret == -2 )
  90. {
  91. Ret = atoi( argv[ n ] );
  92. if ( Ret < 0 )
  93. return -1;
  94. else
  95. lpCmd->TargetDevice = Ret;
  96. }else
  97. return -1;
  98. }
  99. }
  100. return Ret;
  101. }
  102. /*++
  103. 格式化IP地址和端口参数字符串
  104. szAddress - 输入的IP地址和端口参数字符串，形如：192.168.0.3:1025
  105. IPAddress - 输出的IP地址，符合网络字节序，即 0x0301a8c0
  106. Port - 输出的端口，符合网络字节序，即 0x0104
  107. --*/
  108. int ParseAddress( const char* szAddress, PULONG IPAddress, PUSHORT Port)
  109. {
  110. int Ret;
  111. char* pdest;
  112. char  buf[ 16 ];
  113. int length;
  114. PUCHAR pa;
  115. UCHAR temp;
  116. int i, n, lastdot;
  117. Ret = -1;
  118. memset ( buf, 0, 16 );
  119. pdest = strchr ( szAddress, ':' );
  120. if ( pdest == NULL )
  121. return Ret;
  122. if ( pdest != strrchr ( szAddress, ':' ) )
  123. return Ret;
  124. strcpy ( buf, (const char* )( pdest + 1 ) );
  125. *Port = (USHORT)atoi( (const char*)buf );
  126. // change *Port from Intel order to Network Order
  127. pa = (PUCHAR)Port;
  128. temp = pa[ 0 ];
  129. pa[ 0 ] = pa[ 1 ];
  130. pa[ 1 ] = temp;
  131. if( ( pdest - szAddress ) > 15 )
  132. return Ret;
  133. memset ( buf, 0, 16 );
  134. strncpy ( buf, szAddress, ( pdest - szAddress ) );
  135. length = strlen ( buf );
  136. n = 0;
  137. lastdot = -1;
  138. if ( buf[ length - 1 ] == '.' )
  139. return Ret;
  140. for ( i = 0; i < length; i++)
  141. {
  142. if ( buf[ i ] == '.' )
  143. {
  144. if (( ( i - lastdot ) > 4 ) || ( ( i - lastdot ) < 2 ) )
  145. return Ret;
  146. buf[ i ] = 0;
  147. lastdot = i;
  148. n++;
  149. }
  150. }
  151. if ( n != 3 )
  152. return Ret;
  153. pdest = buf;
  154. pa = (PUCHAR)IPAddress;
  155. for ( i = 0; i < 4; i++ )
  156. {
  157. length = strlen ( pdest );
  158. if ( length > 3 )
  159. return Ret;
  160. n = atoi ( pdest );
  161. if ( n > 255 )
  162. return Ret;
  163. pa[ i ] = (UCHAR)n;
  164. pdest += strlen ( pdest ) + 1;
  165. }
  166. return 0;
  167. }
  168. int main(int argc, char* argv[])
  169. {
  170. CMDS cmd;
  171. int nDevice;
  173. if ( argc == 1 )
  174. {
  175. Usage();
  176. return 0;
  177. }
  178. memset ( (void*)&cmd, 0, sizeof ( CMDS ) );
  179. nDevice = Parse( argc, argv, &cmd );
  180. if ( nDevice == -1 )
  181. {
  182. printf("Error parameters!nn");
  183. Usage();
  184. return 0;
  185. }
  186. if ( cmd.bIsEnum )
  187. {
  188. printf("n- Serial ports classn--------------n");
  189. EnumDevice ( (LPGUID)&GUID_CLASS_COMPORT );
  190. printf("nn");
  191. }
  192. if ( cmd.bDeleteDosName && ( cmd.TargetDevice >= 0 ) )
  193. {
  194. printf("- Delete DOS namen-------------n");
  195. DeleteDosName ( (GUID*)&GUID_CLASS_COMPORT, (int)cmd.TargetDevice );
  196. printf("nn");
  197. }
  198. if( ( cmd.bCreateDosName ) && ( cmd.TargetDevice >= 0 ) )
  199. {
  200. printf("- Create DOS namen-------------n");
  201. CreateDosName( (GUID*)&GUID_CLASS_COMPORT, (int)cmd.TargetDevice, (ULONG)cmd.ComX );
  202. printf("nn");
  203. }
  204. if( ( cmd.bSetRmt ) && ( cmd.TargetDevice >= 0 ) )
  205. {
  206. printf("- Set remote IP infon-------------n");
  207. printf("IPAddress = 0x%x,  Port = 0x%xn", cmd.IPAddress, cmd.Port );
  208. SetRmt( (LPGUID) &GUID_CLASS_COMPORT, cmd.TargetDevice, cmd.IPAddress, cmd.Port );
  209. printf("nn");
  210. }
  211. return 0;
  212. }
  213. /*++
  214. 向虚拟串口设备发送命令，要求创建指定的DOS name，即 ComX
  216. lpGuid - 设备类GUID, 此处传入 GUID_CLASS_COMPORT
  217. TargetDevice- 目标设备的序号，此序号由 EnumDevice函数枚举设备类得来。
  218. ComX - 指定的DOS name的序列号，若要创建COM10，则ComX = 10
  219. --*/
  220. void CreateDosName( LPGUID lpGuid, int TargetDevice, ULONG ComX )
  221. {
  222. HANDLE hfile;
  223. BOOL bRet;
  224. DWORD dwRet;
  225. hfile = OpenDevice( lpGuid, TargetDevice );
  226. if ( hfile == INVALID_HANDLE_VALUE )
  227. return;
  228. bRet = DeviceIoControl( hfile, 
  229. IOCTL_CREATE_DOS_NAME,
  230. &ComX, sizeof ( ULONG ), NULL, 0, &dwRet, NULL );
  231. if ( !bRet )
  232. printf("IOCTL_CREATE_DOS_NAME          Failed (0x%x)n", GetLastError() );
  233. else
  234. printf("IOCTL_CREATE_DOS_NAME          OKn");
  235. CloseHandle ( hfile );
  236. }
  237. /*++
  238. 向虚拟串口设备发送命令，要求删除DOS name
  240. lpGuid - 设备类GUID, 此处传入 GUID_CLASS_COMPORT
  241. TargetDevice- 目标设备的序号，此序号由 EnumDevice函数枚举设备类得来。
  243. --*/
  244. void DeleteDosName ( LPGUID lpGuid, int TargetDevice )
  245. {
  246. HANDLE hfile;
  247. BOOL bRet;
  248. DWORD dwRet;
  249. hfile = OpenDevice( lpGuid, TargetDevice );
  250. if ( hfile == INVALID_HANDLE_VALUE )
  251. return;
  252. bRet = DeviceIoControl( hfile, 
  253. IOCTL_DELETE_DOS_NAME,
  254. NULL, 0, NULL, 0, &dwRet, NULL );
  255. if ( !bRet )
  256. printf("IOCTL_DELETE_DOS_NAME          Failed (0x%x)n", GetLastError() );
  257. else
  258. printf("IOCTL_DELETE_DOS_NAME          OKn");
  259. CloseHandle ( hfile );
  260. }
  261. /*++
  262. 向虚拟串口设备发送命令，指定远端目标机器的地址和端口，用于以后发送UDP数据。
  264. lpGuid - 设备类GUID, 此处传入 GUID_CLASS_COMPORT
  265. TargetDevice- 目标设备的序号，此序号由 EnumDevice函数枚举设备类得来
  266. IPAddress - 远端目标机器的IP地址
  267. Port - 远端目标机器的端口
  268. --*/
  269. void SetRmt( LPGUID lpGuid, int TargetDevice,  ULONG IPAddress, USHORT Port )
  270. {
  271. HANDLE hfile;
  272. BOOL bRet;
  273. DWORD dwRet;
  274. ULONG buf[ 2 ];
  275. hfile = OpenDevice( lpGuid, TargetDevice );
  276. if ( hfile == INVALID_HANDLE_VALUE )
  277. return;
  278. buf[ 0 ] = IPAddress;
  279. buf[ 1 ] = (ULONG)Port;
  280. bRet = DeviceIoControl( hfile, 
  281. IOCTL_SET_REMOTE_INFO,
  282. buf, sizeof ( ULONG ) * 2, NULL, 0, &dwRet, NULL );
  283. if ( !bRet )
  284. printf("IOCTL_SET_REMOTE_INFO           Failed (0x%x)n", GetLastError() );
  285. else
  286. printf("IOCTL_SET_REMOTE_INFO           OKn");
  287. CloseHandle ( hfile );
  288. }
  289. /*++
  290. 打开TargetDevice指定的设备
  292. lpGuid - 设备类GUID, 此处传入 GUID_CLASS_COMPORT
  293. TargetDevice- 目标设备的序号，此序号由 EnumDevice函数枚举设备类得来。
  294. --*/
  295. HANDLE OpenDevice( LPGUID lpGuid , int TargetDevice)
  296. {
  297. HANDLE hfile;
  298. TCHAR szDevName[ MAX_PATH ];
  299. hfile = INVALID_HANDLE_VALUE;
  300. if ( TRUE != GetDeviceName( lpGuid, TargetDevice, szDevName ) )
  301. {
  302. return hfile;
  303. }
  304. hfile = CreateFile(
  305. szDevName,
  306. GENERIC_READ | GENERIC_WRITE,
  307. 0,
  308. NULL,
  309. OPEN_EXISTING,
  310. FILE_ATTRIBUTE_NORMAL, //FILE_FLAG_OVERLAPPED ,
  311. NULL
  312. );
  313.     if (hfile == INVALID_HANDLE_VALUE)
  314.         printf("Open serial device             Failed (0x%x)n", GetLastError());
  315. else
  316. printf("Open serial device             OKn", hfile );
  317. return hfile;
  318. }
  319. /*++
  320. 枚举lpGuid指定的设备类下面所有当前存在的设备，并且打印出来
  321. lpGuid - 设备类GUID, 此处传入 GUID_CLASS_COMPORT
  322. --*/
  323. BOOL EnumDevice( const LPGUID lpGUID )
  324. {
  325. HDEVINFO info;
  326. SP_INTERFACE_DEVICE_DATA ifdata;
  327. DWORD needed;
  328.     PSP_INTERFACE_DEVICE_DETAIL_DATA detail;
  329. BYTE buf[ 4096 ];
  330. int i;
  332. info = SetupDiGetClassDevs( lpGUID,
  333. NULL, NULL, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE);
  334. if( info == INVALID_HANDLE_VALUE ){
  335. printf("Enumeration device               Failed (0x%x)n", GetLastError() );
  336. return 0;
  337. }
  338. detail = (PSP_INTERFACE_DEVICE_DETAIL_DATA )buf;
  339. for( i = 0;; i++)
  340. {
  341. memset( &ifdata, 0, sizeof ( SP_INTERFACE_DEVICE_DATA ) );
  342. memset ( detail, 0, 4096 );
  343. ifdata.cbSize = sizeof(ifdata);
  345. if (!SetupDiEnumDeviceInterfaces(info, NULL, (LPGUID) &GUID_CLASS_COMPORT,
  346. i, &ifdata))
  347. {
  348. SetupDiDestroyDeviceInfoList(info);
  349. return 1;
  350. }
  351. detail->cbSize = sizeof(SP_INTERFACE_DEVICE_DETAIL_DATA);
  352. needed = 4096;
  353. if (!SetupDiGetDeviceInterfaceDetail(info, &ifdata, detail, needed, NULL, NULL))
  354. {
  355. printf("Error %d getting interface detailn", GetLastError());
  356. SetupDiDestroyDeviceInfoList(info);
  357. return FALSE;
  358. }
  359. printf("No. %d  %sn", i, detail->DevicePath );
  360. }
  361.     SetupDiDestroyDeviceInfoList( info );
  362. return TRUE;
  363. }
  364. /*++
  365. 得到指定设备的访问路径
  367. lpGuid - 设备类GUID, 此处传入 GUID_CLASS_COMPORT
  368. TargetDevice- 目标设备的序号，此序号由 EnumDevice函数枚举设备类得来。
  369. szDevName - 输出目标设备的访问路径，以后可以使用CreateFile来打开该设备
  370. --*/
  372. BOOL GetDeviceName( LPGUID lpGuid, int TargetDevice, TCHAR* szDevName )
  373. {
  374. HDEVINFO info;
  375. SP_INTERFACE_DEVICE_DATA ifdata;
  376. DWORD needed;
  377.     PSP_INTERFACE_DEVICE_DETAIL_DATA detail;
  379. info = SetupDiGetClassDevs( lpGuid,
  380. NULL, NULL, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE);
  381. if( info == INVALID_HANDLE_VALUE ){
  382. printf("Enumeration device                Failed (0x%x)n", GetLastError() );
  383. return FALSE;
  384. }
  385.     ifdata.cbSize = sizeof(ifdata);
  386.     
  387. if (!SetupDiEnumDeviceInterfaces(info, NULL, lpGuid,
  388.                                      TargetDevice, &ifdata))
  389.     {
  390.         printf("Enumerate interfaces             Failed (0x%x)n", GetLastError());
  391.         SetupDiDestroyDeviceInfoList(info);
  392.         return FALSE;
  393.      }
  395.     SetupDiGetDeviceInterfaceDetail(info, &ifdata, NULL, 0, &needed, NULL);
  397.     detail = (PSP_INTERFACE_DEVICE_DETAIL_DATA) malloc(needed);
  399. if (!detail)
  400.     {
  401.         printf("Error %d trying to get memory for interface detailn", GetLastError());
  402.         SetupDiDestroyDeviceInfoList(info);
  403.         exit(1);
  404.     }
  406.     detail->cbSize = sizeof(SP_INTERFACE_DEVICE_DETAIL_DATA);
  407.     if (!SetupDiGetDeviceInterfaceDetail(info, &ifdata, detail, needed, NULL, NULL))
  408.     {
  409.         printf("Error %d getting interface detailn", GetLastError());
  410.         free((PVOID) detail);
  411.         SetupDiDestroyDeviceInfoList(info);
  412. return FALSE;
  413.     }
  414.     _tcscpy( szDevName, detail->DevicePath );
  415.     free( (PVOID)detail );
  416.     SetupDiDestroyDeviceInfoList( info );
  417. return TRUE;
  418. }