现在的位置: 首页 > 综合 > 正文

[代码]文栏化ring3 API列举驱动(Console和GUI版)

2012年08月10日 ⁄ 综合 ⁄ 共 12499字 ⁄ 字号 小 中 大 ⁄ 评论关闭
 console版如下:

  1. /*++ 
  2. Module Name:
  3.     ListDrvCon.cpp
  5. Enviroment:
  6.     All Windows NT Platfrom;Console
  8. Abstract:
  9.     List all the driver's name & baseaddr & fileaddr
  11. Note:
  12.     Using documented API in psapi.h
  14. Revision:
  15.     23-Nov-2008 Created.
  17. Author:
  18.     benyanwk
  20. --*/
  22. //
  23. // directives
  24. //
  25. #define MAXNUM 255
  26. #define UNICODE // to support chinese 
  27. #include <windows.h>
  28. #include <psapi.h>
  29. #include <stdio.h>
  30. #pragma comment(lib,"psapi.lib")
  32. //
  33. // strcut defintion
  34. //
  36. typedef struct _DRIVER_INFO {
  37.     WCHAR BaseName[MAX_PATH];
  38.     WCHAR FileName[MAX_PATH];
  39.     DWORD BaseAddr;
  40. } DRIVER_INFO,*PDRIVER_INFO ;
  42. typedef struct _ALL_DRIVER_INFO {
  43.     DWORD cbNum;
  44.     DRIVER_INFO DrvInfo[1]; // define variable length structure
  45. } ALL_DRIVER_INFO,*PALL_DRIVER_INFO;
  47. //
  48. // function declaration
  49. //
  50. int main();
  51. int myGetDriverInfo(
  52.     IN PVOID &pDrvInfo,
  53.     IN BOOLEAN bAlloc
  54.     );
  56. //
  57. // function definition
  58. //
  59. int myGetDriverInfo(
  60.     IN PALL_DRIVER_INFO* pDrvInfo,  
  61.     IN BOOLEAN bAlloc
  62.     )
  63. {
  64. /*++
  65. Arguments:
  66.     pDrvInfo-->the buffer to store the driver information
  67.     bAlloc-->alloc the global memory by the function or not
  68.     return 0 indicate success otherwise error 
  69. --*/
  70.     DWORD cbNum = 0;
  71.     PDWORD pBaseAddr = NULL ;
  72.     PWCHAR pFileName = NULL;
  73.     PWCHAR pBaseName = NULL;
  74.     PALL_DRIVER_INFO pAllDrvInfo;
  76.         
  77.     pBaseAddr = (PDWORD)GlobalAlloc( GMEM_FIXED, sizeof(DWORD)*MAXNUM );
  80.     if( EnumDeviceDrivers( (LPVOID*)pBaseAddr, sizeof(DWORD)*MAXNUM, &cbNum ) != TRUE )
  81.     {
  82.         //
  83.         // indicate EnumDeviceDriver failed!
  84.         //
  85.         wprintf( L"EnumDeviceDriver failed! ErrorCode = %8x/n", GetLastError() );
  86.         return 1; 
  87.     }
  88.     
  89.     cbNum /=4; // cbNum return the bytes
  91.     if( bAlloc == TRUE )
  92.         pAllDrvInfo = (PALL_DRIVER_INFO)GlobalAlloc( GMEM_FIXED, sizeof(DRIVER_INFO)*cbNum + 4 );
  93.     else
  94.         pAllDrvInfo = (PALL_DRIVER_INFO)pDrvInfo;
  96.     pAllDrvInfo->cbNum = cbNum;
  98.     for( int i = 0; i < cbNum; i++ )
  99.     {
  100.         pAllDrvInfo->DrvInfo[i].BaseAddr = *pBaseAddr;
  102.         GetDeviceDriverBaseName( (LPVOID)*pBaseAddr, (LPWSTR)&pAllDrvInfo->DrvInfo[i].BaseName, MAX_PATH );
  103.         GetDeviceDriverFileName( (LPVOID)*pBaseAddr, (LPWSTR)&pAllDrvInfo->DrvInfo[i].FileName, MAX_PATH );
  105.         pBaseAddr++;
  106.     }
  108.     *pDrvInfo = pAllDrvInfo; // return the drv info buffer pointer 
  109.     return 0;   // indicate success
  110. }
  112. int main()
  113. {
  114.     PALL_DRIVER_INFO pDrvInfo = NULL;
  115.     myGetDriverInfo(
  116.         (PALL_DRIVER_INFO*)&pDrvInfo,
  117.         TRUE
  118.         );
  119.     wprintf(L"=====the list of driver as follows === /n");
  120.     wprintf(L"Order/tBaseAddr /t BaseName /t FileName /t /n");
  121.     for( int i = 0; i<pDrvInfo->cbNum; i++ )
  122.     {
  123.         wprintf(L"%d/t%8x/t%s/t%s/n",
  124.                 i,
  125.                 pDrvInfo->DrvInfo[i].BaseAddr,
  126.                 pDrvInfo->DrvInfo[i].BaseName,
  127.                 pDrvInfo->DrvInfo[i].FileName
  128.                 );
  129.     }
  130.     return 0;
  131. }

 

GUI版如下:

  1. /*++ 
  2. Module Name:
  3.     ListDrvGui.cpp
  5. Enviroment:
  6.     All Windows NT Platfrom;GUI
  8. Abstract:
  9.     List all the driver's name & baseaddr & fileaddr
  11. Note:
  12.     (1) Using documented API in psapi and 
  13.         ListView control to show the result
  14.     (2) Wanted upgraded:
  15.         
  16.         + ListView control sort functionality(IMPORTANT)
  17.         + Item should can be selected
  18.         + Support the right click menu
  20. Revision:
  21.     23-Nov-2008 Created.
  23. Author:
  24.     benyanwk
  26. --*/
  29. //////////////////////////////////////////////////
  30. // directives
  31. /////////////////////////////////////////////////
  32. #define MAXNUM 255
  33. #define UNICODE             // to support chinese 
  34. #include <windows.h>
  35. #include <commctrl.h>   
  36. #include <psapi.h>
  37. #include <stdio.h>
  39. #define C_COLUMNS 4  // 4 columns to be added
  40. #include "resource.h"   // the resource file
  41. #pragma comment(lib,"psapi.lib")
  42. #pragma comment(lib,"comctl32.lib")
  44. ////////////////////////////////////////////////
  45. // strcut defintion
  46. /////////////////////////////////////////////////
  48. typedef struct _DRIVER_INFO {
  49.     WCHAR BaseName[MAX_PATH];
  50.     WCHAR FileName[MAX_PATH];
  51.     DWORD BaseAddr;
  52. } DRIVER_INFO,*PDRIVER_INFO ;
  54. typedef struct _ALL_DRIVER_INFO {
  55.     DWORD cnNum;
  56.     DRIVER_INFO DrvInfo[1]; // define variable length structure
  57. } ALL_DRIVER_INFO,*PALL_DRIVER_INFO;
  59. ////////////////////////////////////////////////
  60. // function declaration
  61. ///////////////////////////////////////////////
  63. INT_PTR CALLBACK DlgProc(          
  64.     HWND hwndDlg,
  65.     UINT uMsg,
  66.     WPARAM wParam,
  67.     LPARAM lParam
  68. );
  70. int WINAPI WinMain(          
  71.     HINSTANCE hInstance,
  72.     HINSTANCE hPrevInstance,
  73.     LPSTR lpCmdLine,
  74.     int nCmdShow
  75. );
  76. int myGetDriverInfo(
  77.     IN PVOID &pDrvInfo,
  78.     IN BOOLEAN bAlloc
  79.     );
  81. BOOLEAN SetListView(
  82.       IN HWND hListView
  83.       );
  85. BOOLEAN InitListView(
  86.       IN HWND hListView
  87.       );
  88. //////////////////////////////////////////////
  89. // Global variable declaration
  90. ////////////////////////////////////////////
  92. PALL_DRIVER_INFO g_pDrvInfo = NULL; // the global variable used to store the driver information
  93. HWND g_hInst = NULL;
  94. HWND g_hListView = NULL;
  97. ///////////////////////////////////////////////
  98. // function definition
  99. //////////////////////////////////////////////
  100. int myGetDriverInfo(
  101.     IN PALL_DRIVER_INFO* pDrvInfo,  
  102.     IN BOOLEAN bAlloc
  103.     )
  104. {
  105. /*++
  106. Arguments:
  107.     pDrvInfo-->the buffer to store the driver information
  109.     bAlloc-->alloc the global memory by the function or not
  110.     return 0 indicate success otherwise error 
  111. --*/
  112.     DWORD cbNum = 0;
  113.     PDWORD pBaseAddr = NULL ;
  114.     PWCHAR pFileName = NULL;
  115.     PWCHAR pBaseName = NULL;
  116.     PALL_DRIVER_INFO pAllDrvInfo = NULL;
  118.         
  119.     pBaseAddr = (PDWORD)GlobalAlloc( GMEM_FIXED, sizeof(DWORD)*MAXNUM );
  122.     if( EnumDeviceDrivers( (LPVOID*)pBaseAddr, sizeof(DWORD)*MAXNUM, &cbNum ) != TRUE )
  123.     {
  124.         //
  125.         // indicate EnumDeviceDriver failed!
  126.         //
  127.         wprintf( L"EnumDeviceDriver failed! ErrorCode = %8x/n", GetLastError() );
  128.         return 1; 
  129.     }
  130.     
  131.     cbNum /=4; // cbNum return the bytes
  133.     if( bAlloc == TRUE )
  134.         pAllDrvInfo = (PALL_DRIVER_INFO)GlobalAlloc( GMEM_FIXED, sizeof(DRIVER_INFO)*cbNum + 4 );
  135.     else
  136.         pAllDrvInfo = (PALL_DRIVER_INFO)pDrvInfo;
  138.     pAllDrvInfo->cnNum = cbNum;
  140.     for( int i = 0; i < cbNum; i++ )
  141.     {
  142.         pAllDrvInfo->DrvInfo[i].BaseAddr = *pBaseAddr;
  144.         GetDeviceDriverBaseName( (LPVOID)*pBaseAddr, (LPWSTR)&pAllDrvInfo->DrvInfo[i].BaseName, MAX_PATH );
  145.         GetDeviceDriverFileName( (LPVOID)*pBaseAddr, (LPWSTR)&pAllDrvInfo->DrvInfo[i].FileName, MAX_PATH );
  147.         pBaseAddr++;
  148.     }
  150.     GlobalFree( pBaseAddr );
  152.     *pDrvInfo = pAllDrvInfo; // return the drv info buffer pointer 
  153.     return 0;   // indicate success
  154. }
  156. int WINAPI WinMain(          
  157.     HINSTANCE hInstance,
  158.     HINSTANCE hPrevInstance,
  159.     LPSTR lpCmdLine,
  160.     int nCmdShow
  161. )
  162. {
  163.     g_hInst = (HWND)GetModuleHandle( NULL );
  165.     InitCommonControls();
  168.     //
  169.     // Create the main dialog
  170.     //
  171.     DialogBoxParam( 
  172.         (HINSTANCE)g_hInst, // hinstance
  173.         MAKEINTRESOURCE(IDD_DIALOGBAR), // temple name
  174.         NULL,   // parent hwnd
  175.         DlgProc, // dialog functoin
  176.         NULL    // initialize parameter
  177.         );  
  179.     ExitProcess(0);
  180. }
  182. INT_PTR CALLBACK DlgProc(          
  183.     HWND hwndDlg,
  184.     UINT uMsg,
  185.     WPARAM wParam,
  186.     LPARAM lParam
  187. )
  188. {
  189.     RECT rect;
  190.     HWND hwndOwner;
  191.     RECT rc,rcDlg,rcOwner;
  192.     int  iSelCol;
  193.     HICON hIcon;
  197.     switch( uMsg )
  198.     {
  199.     case WM_CLOSE:
  200.         GlobalFree( g_pDrvInfo);
  201.         EndDialog( hwndDlg, NULL );
  202.         break;
  203.     case WM_INITDIALOG:
  204.         //
  205.         // set the icon
  206.         //
  207.         hIcon = LoadIcon( (HINSTANCE)g_hInst, (LPCWSTR)IDI_ICON1 );
  208.         SendMessage( hwndDlg, WM_SETICON, ICON_BIG, (LPARAM)hIcon );
  209.         SendMessage( hwndDlg, WM_SETICON, ICON_SMALL, (LPARAM)hIcon );
  211.         // 
  212.         // centre the dialog
  213.         //
  215.         g_hListView = GetDlgItem( hwndDlg, IDC_LIST2 );
  216.         
  217.         hwndOwner = GetDesktopWindow();
  219.         GetWindowRect( hwndOwner, &rcOwner );
  220.         GetWindowRect( hwndDlg, &rcDlg );
  221.         CopyRect( &rc, &rcOwner );  
  223.         // Offset the owner and dialog box rectangle so that
  224.         // right and bottom values represent the width and 
  225.         // height, and then offset the owner again to discard 
  226.         // space taken up by the dialog
  227.         OffsetRect( &rcDlg, -rcDlg.left, -rcDlg.top );
  228.         OffsetRect( &rc, -rc.left, -rc.top );
  229.         OffsetRect( &rc, -rcDlg.right, -rcDlg.bottom );
  231.         SetWindowPos(
  232.             hwndDlg,
  233.             HWND_TOP,
  234.             rcOwner.left + ( rc.right / 2 ),
  235.             rcOwner.top + ( rc.bottom / 2 ),
  236.             rcDlg.right,
  237.             rcDlg.bottom,
  238.             SWP_NOSIZE );
  239.         
  240.         //
  241.         // set the default control to the refresh button
  242.         //
  244.         SetFocus( GetDlgItem( hwndDlg, IDC_REFRESH ) );
  246.         //
  247.         // set the listview control
  248.         //
  249.         
  250.         SetWindowPos( 
  251.             g_hListView,
  252.             HWND_TOP,
  253.             LOWORD(lParam)*0.5/10,
  254.             HIWORD(lParam)*0.5/10,
  255.             LOWORD(lParam)*9/10,
  256.             HIWORD(lParam)*8.5/10,
  257.             SWP_SHOWWINDOW
  258.             );
  259.         InitListView( g_hListView );
  261.         SetListView( g_hListView );
  262.         break;
  263.     case WM_COMMAND:
  264.         if( wParam == IDC_REFRESH )
  265.         {
  266.             // 
  267.             // indicate push refresh button
  268.             //
  269.             SetListView( g_hListView );
  270.         }
  271.         else if( wParam == IDC_EXIT )
  272.         {
  273.             // 
  274.             // indicate exit
  275.             // 
  276.             GlobalFree( g_pDrvInfo);
  278.             EndDialog( hwndDlg, NULL );
  279.         }
  280.         break;
  281.     case WM_SIZE:
  283.         // 
  284.         // reset the button control
  285.         //
  287.         SetWindowPos(
  288.             GetDlgItem( hwndDlg, IDC_REFRESH ),
  289.             HWND_TOP,
  290.             LOWORD(lParam)*1/8,
  291.             HIWORD(lParam)*9.2/10,
  292.             LOWORD(lParam)*1/8,
  293.             HIWORD(lParam)*0.5/10,
  294.             SWP_SHOWWINDOW
  295.             );
  296.         SetWindowPos(
  297.             GetDlgItem( hwndDlg, IDC_EXIT ),
  298.             HWND_TOP,
  299.             LOWORD(lParam)*5/8,
  300.             HIWORD(lParam)*9.2/10,
  301.             LOWORD(lParam)*1/8,
  302.             HIWORD(lParam)*0.5/10,
  303.             SWP_SHOWWINDOW 
  304.             );
  306.         //
  307.         // reset the listview control
  308.         //
  310.         SetWindowPos( 
  311.             g_hListView,
  312.             HWND_TOP,
  313.             LOWORD(lParam)*0.5/10,
  314.             HIWORD(lParam)*0.5/10,
  315.             LOWORD(lParam)*9/10,
  316.             HIWORD(lParam)*8.5/10,
  317.             SWP_SHOWWINDOW
  318.             );
  320.         //InitListView( g_hListView );
  322.         SetListView( g_hListView );
  323.             
  324.         return FALSE;
  326.         default:
  327.             return FALSE;
  328.     }
  329.     return TRUE;
  330.             
  331. }
  335. BOOLEAN InitListView(
  336.       IN HWND hListView
  337.       )
  338. /*++
  339.  Note: this routine totally copy from MSDN
  340. --*/
  341. {
  342.     LVCOLUMN lvc = {0};
  343.     int iCol;
  344.     WCHAR szText[256];
  345.     
  346.     lvc.mask = LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM;
  348.     for( iCol = 0; iCol < C_COLUMNS; iCol++ )
  349.     {
  350.         lvc.iSubItem = 0;
  351.         lvc.pszText = szText; 
  352.         
  353.         switch( iCol )
  354.         {
  355.         case 0:
  356.             lvc.cx = 60;
  357.             break;
  358.         case 1:
  359.             lvc.cx = 100;
  360.             break;
  361.         case 2:
  362.             lvc.cx = 150;
  363.             break;
  364.         case 3:
  365.             lvc.cx = 450;
  366.             break;
  367.         }
  369.         LoadString((HINSTANCE)g_hInst, IDS_FIRSTCOLUMN + iCol,
  370.             szText, sizeof(szText)/sizeof(szText[0]));
  372.         if (ListView_InsertColumn( hListView, iCol,
  373.                             &lvc) == -1 )
  374.             return FALSE;
  375.     }
  376.     return TRUE;
  377. }
  380. BOOLEAN SetListView(
  381.       IN HWND hListView
  382.       )
  383. {
  384.     
  385.     LVITEM lvi = {0};
  386.     int iItem;
  387.     WCHAR temp[13] = {0};
  389.     //
  390.     // get the driver information
  391.     //
  392.     myGetDriverInfo(
  393.         &g_pDrvInfo,
  394.         TRUE
  395.         );
  397.     //
  398.     // first clean the orignal text
  399.     //
  401.     ListView_DeleteAllItems( g_hListView );
  403.     //
  404.     // now insert the subitems
  405.     //
  406.     lvi.mask = LVIF_TEXT;
  407.     lvi.cchTextMax = MAX_PATH;
  410.     for( iItem = 0; iItem < g_pDrvInfo->cnNum; iItem++ )
  411.     {
  412.         lvi.iItem = iItem;
  414.         // insert the odrer
  415.         lvi.iSubItem = 0;
  416.         wsprintf( temp,  L"%d", iItem );
  417.         lvi.pszText = temp;
  419.         ListView_InsertItem( hListView, &lvi ); 
  421.         // insert the base addr
  422.         lvi.iSubItem = 1;
  423.         wsprintf( temp, L"0x%8x", g_pDrvInfo->DrvInfo[iItem].BaseAddr );
  424.         lvi.pszText = temp;
  425.         ListView_SetItem( hListView, &lvi );
  427.         // insert the base name
  428.         lvi.iSubItem = 2;
  429.         lvi.pszText = (LPWSTR)g_pDrvInfo->DrvInfo[iItem].BaseName;
  430.         ListView_SetItem( hListView, &lvi );
  432.         // insert the file name
  433.         lvi.iSubItem = 3;
  434.         lvi.pszText = (LPWSTR)g_pDrvInfo->DrvInfo[iItem].FileName;
  435.         ListView_SetItem( hListView, &lvi );
  437.     }
  439.     return TRUE;
  440. }

相关文件下载:http://www.live-share.com/files/368301/C1_ListDrv.rar.html

 

返回
【上篇】
【下篇】

作者:

抱歉!评论已关闭.

返回首页

Copyright © 2013-2018 学步园  保留所有权利.
软文销售 QQ客服：2265327166 点击这里给我发消息（其他合作也可洽谈）
必威体育
必威电竞