普通派发函数

read，write

R3

  1. HANDLE deviceHandle = NULL;
  2. void CDeviceUserCtlDlg::OnBnClickedButton1()
  3. {
  4.     // TODO: Add your control notification handler code here
  5.     if (deviceHandle!=INVALID_HANDLE_VALUE)
  6.     {
  7.         CloseHandle(deviceHandle);
  9.     }
  10. }
  14. void CDeviceUserCtlDlg::OnBnClickedButton2()
  15. {
  16.     // TODO: Add your control notification handler code here
  17.     deviceHandle =CreateFile(L"\\\\.\\mydevicelink123",GENERIC_ALL,0,0,OPEN_EXISTING,FILE_ATTRIBUTE_SYSTEM,0);
  18.     if (deviceHandle==INVALID_HANDLE_VALUE)
  19.     {
  20.         MessageBox(L"vaild value",0,0);
  21.         return;
  22.     }
  23.     MessageBox(L"not valid value",0,0);

image.png

R0

  1. #include <ntddk.h>
  3. UNICODE_STRING DeviceName = RTL_CONSTANT_STRING(L"\\Device\\mydevice123");
  4. UNICODE_STRING SymLinkName = RTL_CONSTANT_STRING(L"\\??\\mydevicelink123");
  5. PDEVICE_OBJECT DeviceObject = NULL;
  7. VOID Unload(IN PDRIVER_OBJECT DriverObject) {
  9.     IoDeleteSymbolicLink(&SymLinkName);
  10.     IoDeleteDevice(DeviceObject);
  11.     KdPrint(("driver unload \r\n"));
  13. }
  16. NTSTATUS DispatchPassThru(PDEVICE_OBJECT DeviceObject,PIRP Irp) {
  17.     PIO_STACK_LOCATION irpsp = IoGetCurrentIrpStackLocation(Irp);
  18.     NTSTATUS status = STATUS_SUCCESS;
  20.     switch (irpsp->MajorFunction)
  21.     {
  22.         case IRP_MJ_CREATE:
  23.             KdPrint(("create request \r\n"));
  24.             break;
  25.         case IRP_MJ_CLOSE:
  26.             KdPrint(("close request\r\n"));
  27.             break;
  28.         case IRP_MJ_READ:
  29.             KdPrint(("read quest\r\n"));
  30.             break;
  31.         default:
  32.             break;
  33.     }
  34.     Irp->IoStatus.Information = 0;
  35.     Irp->IoStatus.Status = status;
  36.     IoCompleteRequest(Irp,IO_NO_INCREMENT);
  37.     return status;
  38. }
  41. NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject,IN PUNICODE_STRING RegisteryPath) {
  43.     int i;
  44.     NTSTATUS status;
  45.     DriverObject->DriverUnload = Unload;
  47.     status = IoCreateDevice(DriverObject,0,&DeviceName,FILE_DEVICE_UNKNOWN,FILE_DEVICE_SECURE_OPEN,FALSE,&DeviceObject);
  48.     if (!NT_SUCCESS(status))
  49.     {
  50.         KdPrint(("creating devce failed \r\n"));
  51.         return status;
  52.     }
  55.     IoCreateSymbolicLink(&SymLinkName,&DeviceName);
  56.     if (!NT_SUCCESS(status))
  57.     {
  58.         KdPrint(("creating symbolic link failed\r\n"));
  59.         IoDeleteDevice(DeviceObject);
  60.         return status;
  62.     }
  65.     for ( i = 0; i < IRP_MJ_MAXIMUM_FUNCTION; i++)
  66.     {
  67.         DriverObject->MajorFunction[i] = DispatchPassThru;
  69.     }
  70.     KdPrint(("driver load \r\n"));
  74.     return status;
  75. }

deviceControl

R0

  1. #include <ntddk.h>
  2. #define DEVICE_SEND CTL_CODE(FILE_DEVICE_UNKNOWN,0X801,METHOD_BUFFERED,FILE_WRITE_DATA)
  3. #define DEVICE_REC CTL_CODE(FILE_DEVICE_UNKNOWN,0X802,METHOD_BUFFERED,FILE_READ_DATA)
  4. //0X800-0XFFF  buffercopy,
  6. UNICODE_STRING DeviceName = RTL_CONSTANT_STRING(L"\\Device\\mydevice123");
  7. UNICODE_STRING SymLinkName = RTL_CONSTANT_STRING(L"\\??\\mydevicelink123");
  8. PDEVICE_OBJECT DeviceObject = NULL;
  10. VOID Unload(IN PDRIVER_OBJECT DriverObject) {
  12.     IoDeleteSymbolicLink(&SymLinkName);
  13.     IoDeleteDevice(DeviceObject);
  14.     KdPrint(("driver unload \r\n"));
  16. }
  19. NTSTATUS DispatchPassThru(PDEVICE_OBJECT DeviceObject,PIRP Irp) {
  20.     PIO_STACK_LOCATION irpsp = IoGetCurrentIrpStackLocation(Irp);
  21.     NTSTATUS status = STATUS_SUCCESS;
  23.     switch (irpsp->MajorFunction)
  24.     {
  25.         case IRP_MJ_CREATE:
  26.             KdPrint(("create request \r\n"));
  27.             break;
  28.         case IRP_MJ_CLOSE:
  29.             KdPrint(("close request\r\n"));
  30.             break;
  33.         default:
  34.             status = STATUS_INVALID_PARAMETER;
  35.             break;
  36.     }
  40.     Irp->IoStatus.Information = 0;
  41.     Irp->IoStatus.Status = status;
  42.     IoCompleteRequest(Irp,IO_NO_INCREMENT);
  43.     return status;
  44. }
  47. NTSTATUS DispatchDevCTL(PDEVICE_OBJECT DeviceObject,PIRP Irp) 
  48. {
  49.     PIO_STACK_LOCATION irpsp = IoGetCurrentIrpStackLocation(Irp);
  50.     NTSTATUS status = STATUS_SUCCESS;
  51.     ULONG returnLength;
  52.     PVOID buffer = Irp->AssociatedIrp.SystemBuffer;
  53.     ULONG inLength = irpsp->Parameters.DeviceIoControl.InputBufferLength;
  54.     ULONG outLength = irpsp->Parameters.DeviceIoControl.OutputBufferLength;
  55.     WCHAR* demo = L"sample returned from driver";
  57.     switch (irpsp->Parameters.DeviceIoControl.IoControlCode)
  58.     {
  59.     case DEVICE_SEND:
  60.         KdPrint(("recv data is%ws \r\n", buffer));
  61.         returnLength = (wcsnlen(buffer, 511) + 1) * 2;
  62.         break;
  64.     case DEVICE_REC:
  65.         KdPrint(("sending data\r\n "));
  66.         wcsncpy(buffer, demo, 511);
  67.         returnLength = (wcsnlen(buffer,511)+1)*2;
  69.         break;
  71.     default:
  72.         status = STATUS_INVALID_PARAMETER;
  73.         break;
  74.     }
  75.     Irp->IoStatus.Status = status;
  76.     Irp->IoStatus.Information = returnLength;
  77.     IoCompleteRequest(Irp, IO_NO_INCREMENT);
  78.     return status;
  81. }
  85. NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject,IN PUNICODE_STRING RegisteryPath) {
  87.     int i;
  88.     NTSTATUS status;
  89.     DriverObject->DriverUnload = Unload;
  91.     status = IoCreateDevice(DriverObject,0,&DeviceName,FILE_DEVICE_UNKNOWN,FILE_DEVICE_SECURE_OPEN,FALSE,&DeviceObject);
  92.     if (!NT_SUCCESS(status))
  93.     {
  94.         KdPrint(("creating devce failed \r\n"));
  95.         return status;
  96.     }
  99.     IoCreateSymbolicLink(&SymLinkName,&DeviceName);
  100.     if (!NT_SUCCESS(status))
  101.     {
  102.         KdPrint(("creating symbolic link failed\r\n"));
  103.         IoDeleteDevice(DeviceObject);
  104.         return status;
  106.     }
  109.     for ( i = 0; i < IRP_MJ_MAXIMUM_FUNCTION; i++)
  110.     {
  111.         DriverObject->MajorFunction[i] = DispatchPassThru;
  113.     }
  116.     DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = DispatchDevCTL;
  117.     KdPrint(("driver load \r\n"));
  121.     return status;
  122. }

R3

需要包含 winio.h
定义control码

  2. // DeviceUserCtlDlg.cpp : implementation file
  3. //
  5. #include "pch.h"
  6. #include "framework.h"
  7. #include "DeviceUserCtl.h"
  8. #include "DeviceUserCtlDlg.h"
  9. #include "afxdialogex.h"
  10. //引入winio.H
  11. #include "Winioctl.h"
  13. #ifdef _DEBUG
  14. #define new DEBUG_NEW
  15. #endif
  17. #define DEVICE_SEND CTL_CODE(FILE_DEVICE_UNKNOWN,0X801,METHOD_BUFFERED,FILE_WRITE_DATA)
  18. #define DEVICE_REC CTL_CODE(FILE_DEVICE_UNKNOWN,0X802,METHOD_BUFFERED,FILE_READ_DATA)
  19. HANDLE deviceHandle = NULL;
  20. // CAboutDlg dialog used for App About
  22. class CAboutDlg : public CDialogEx
  23. {
  24. public:
  25.     CAboutDlg();
  27. // Dialog Data
  28. #ifdef AFX_DESIGN_TIME
  29.     enum { IDD = IDD_ABOUTBOX };
  30. #endif
  32.     protected:
  33.     virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  35. // Implementation
  36. protected:
  37.     DECLARE_MESSAGE_MAP()
  38. };
  40. CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
  41. {
  42. }
  44. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  45. {
  46.     CDialogEx::DoDataExchange(pDX);
  47. }
  49. BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
  50. END_MESSAGE_MAP()
  53. // CDeviceUserCtlDlg dialog
  57. CDeviceUserCtlDlg::CDeviceUserCtlDlg(CWnd* pParent /*=nullptr*/)
  58.     : CDialogEx(IDD_DEVICEUSERCTL_DIALOG, pParent)
  59. {
  60.     m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  61. }
  63. void CDeviceUserCtlDlg::DoDataExchange(CDataExchange* pDX)
  64. {
  65.     CDialogEx::DoDataExchange(pDX);
  66. }
  68. BEGIN_MESSAGE_MAP(CDeviceUserCtlDlg, CDialogEx)
  69.     ON_WM_SYSCOMMAND()
  70.     ON_WM_PAINT()
  71.     ON_WM_QUERYDRAGICON()
  72.     ON_BN_CLICKED(IDC_BUTTON1, &CDeviceUserCtlDlg::OnBnClickedButton1)
  73.     ON_BN_CLICKED(IDC_BUTTON2, &CDeviceUserCtlDlg::OnBnClickedButton2)
  74.     ON_BN_CLICKED(IDC_BUTTON3, &CDeviceUserCtlDlg::OnBnClickedButton3)
  75.     ON_BN_CLICKED(IDC_BUTTON4, &CDeviceUserCtlDlg::OnBnClickedButton4)
  76. END_MESSAGE_MAP()
  79. // CDeviceUserCtlDlg message handlers
  81. BOOL CDeviceUserCtlDlg::OnInitDialog()
  82. {
  83.     CDialogEx::OnInitDialog();
  85.     // Add "About..." menu item to system menu.
  87.     // IDM_ABOUTBOX must be in the system command range.
  88.     ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  89.     ASSERT(IDM_ABOUTBOX < 0xF000);
  91.     CMenu* pSysMenu = GetSystemMenu(FALSE);
  92.     if (pSysMenu != nullptr)
  93.     {
  94.         BOOL bNameValid;
  95.         CString strAboutMenu;
  96.         bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
  97.         ASSERT(bNameValid);
  98.         if (!strAboutMenu.IsEmpty())
  99.         {
  100.             pSysMenu->AppendMenu(MF_SEPARATOR);
  101.             pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  102.         }
  103.     }
  105.     // Set the icon for this dialog.  The framework does this automatically
  106.     //  when the application's main window is not a dialog
  107.     SetIcon(m_hIcon, TRUE);            // Set big icon
  108.     SetIcon(m_hIcon, FALSE);        // Set small icon
  110.     // TODO: Add extra initialization here
  112.     return TRUE;  // return TRUE  unless you set the focus to a control
  113. }
  115. void CDeviceUserCtlDlg::OnSysCommand(UINT nID, LPARAM lParam)
  116. {
  117.     if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  118.     {
  119.         CAboutDlg dlgAbout;
  120.         dlgAbout.DoModal();
  121.     }
  122.     else
  123.     {
  124.         CDialogEx::OnSysCommand(nID, lParam);
  125.     }
  126. }
  128. // If you add a minimize button to your dialog, you will need the code below
  129. //  to draw the icon.  For MFC applications using the document/view model,
  130. //  this is automatically done for you by the framework.
  132. void CDeviceUserCtlDlg::OnPaint()
  133. {
  134.     if (IsIconic())
  135.     {
  136.         CPaintDC dc(this); // device context for painting
  138.         SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  140.         // Center icon in client rectangle
  141.         int cxIcon = GetSystemMetrics(SM_CXICON);
  142.         int cyIcon = GetSystemMetrics(SM_CYICON);
  143.         CRect rect;
  144.         GetClientRect(&rect);
  145.         int x = (rect.Width() - cxIcon + 1) / 2;
  146.         int y = (rect.Height() - cyIcon + 1) / 2;
  148.         // Draw the icon
  149.         dc.DrawIcon(x, y, m_hIcon);
  150.     }
  151.     else
  152.     {
  153.         CDialogEx::OnPaint();
  154.     }
  155. }
  157. // The system calls this function to obtain the cursor to display while the user drags
  158. //  the minimized window.
  159. HCURSOR CDeviceUserCtlDlg::OnQueryDragIcon()
  160. {
  161.     return static_cast<HCURSOR>(m_hIcon);
  162. }
  166. void CDeviceUserCtlDlg::OnBnClickedButton1()
  167. {
  168.     // TODO: Add your control notification handler code here
  169.     if (deviceHandle!=INVALID_HANDLE_VALUE)
  170.     {
  171.         CloseHandle(deviceHandle);
  173.     }
  174. }
  178. void CDeviceUserCtlDlg::OnBnClickedButton2()
  179. {
  180.     // TODO: Add your control notification handler code here
  181.     deviceHandle =CreateFile(L"\\\\.\\mydevicelink123",GENERIC_ALL,0,0,OPEN_EXISTING,FILE_ATTRIBUTE_SYSTEM,0);
  182.     if (deviceHandle==INVALID_HANDLE_VALUE)
  183.     {
  184.         MessageBox(L"vaild value",0,0);
  185.         return;
  186.     }
  187.     MessageBox(L"not valid value",0,0);
  189. }
  192. void CDeviceUserCtlDlg::OnBnClickedButton3()
  193. {
  194.     // TODO: Add your control notification handler code here
  196.     //send data;
  197.     WCHAR* message = L"hello message from userApp";
  198.     ULONG returnLength = 0;
  199.     char wr[4] = {0};
  200.     if (deviceHandle != INVALID_HANDLE_VALUE && deviceHandle!=NULL)
  201.     {
  202.         if (!DeviceIoControl(deviceHandle,DEVICE_SEND,message,(wcslen(message)+1)*2,NULL,0,&returnLength,0))
  203.         {
  206.             MessageBox(L"device control error",0,0);
  208.         }
  209.         else
  210.         {
  211.             _itoa_s(returnLength, wr, 10);
  212.             MessageBoxA(0,wr,0,0);
  213.         }
  215.     }
  216. }
  219. void CDeviceUserCtlDlg::OnBnClickedButton4()
  220. {
  221.     // TODO: Add your control notification handler code here
  223.     WCHAR message[1024] = {0};
  224.     ULONG returnLength = 0;
  225.     if (deviceHandle != INVALID_HANDLE_VALUE && deviceHandle != NULL)
  226.     {
  227.         if (!DeviceIoControl(deviceHandle, DEVICE_REC, NULL, 0, message, 1024, &returnLength, 0))
  228.         {
  230.             MessageBox(L"device control error", 0, 0);
  232.         }
  233.         else
  234.         {
  236.             MessageBox(message,0,0);
  237.         }
  239.     }
  240. }

image.png

cpu温度监控