zoukankan      html  css  js  c++  java
  • 驱动问题代码01

    蓝屏,待调试

    #include <ntddk.h>
    
    typedef struct _DEVICE_EXTENSION
    {
        PDEVICE_OBJECT pDevObj;
        UNICODE_STRING ustrDeviceName;
        UNICODE_STRING ustrSymbolLinkName;
    }DEVICE_EXTENSION,*PDEVICE_EXTENSION;
    
    
    NTSTATUS DeviceCreate(PDEVICE_OBJECT pDeviceObject, PIRP pIrp);
    NTSTATUS DeviceClose(PDEVICE_OBJECT pDeviceObject, PIRP pIrp);
    NTSTATUS DeviceRead(PDEVICE_OBJECT pDeviceObject, PIRP pIrp);
    
    NTSTATUS DemoCreateDevice(PDRIVER_OBJECT pDriver,PCWSTR devName,PCWSTR devSymName);
    
    NTSTATUS unload(PDRIVER_OBJECT driver)
    {
        PDEVICE_OBJECT pDev;
        DbgPrint("driver :%ws unload", driver->DriverName.Buffer);
        pDev = driver->DeviceObject;
        while (pDev != NULL)
        {
            PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)pDev->DeviceExtension;
    
            //删除符号链接
            UNICODE_STRING pLinkName = pDevExt->ustrSymbolLinkName;
            IoDeleteSymbolicLink(&pLinkName);
            pDev = pDev->NextDevice;
            IoDeleteDevice(pDevExt->pDevObj);
        }
        DbgPrint("driver unload success...");
        return STATUS_SUCCESS;
    }
    
    NTSTATUS DriverEntry(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path)
    {
        NTSTATUS ntstatus = STATUS_SUCCESS;
        driver->DriverUnload = unload;
        driver->MajorFunction[IRP_MJ_CREATE] = DeviceCreate;//创建
        driver->MajorFunction[IRP_MJ_READ] = DeviceRead;//通信
        driver->MajorFunction[IRP_MJ_CLOSE] = DeviceClose;//关闭
    
        ntstatus = DemoCreateDevice(driver, L"\Device\MyDevice01", L"\??\MyDeviceSymbolLinkName01");
        if (!NT_SUCCESS(ntstatus))
        {
            DbgPrint("IoCreateDevice Failed");
            return ntstatus;
        }
        ntstatus = DemoCreateDevice(driver, L"\Device\MyDevice02", L"\??\MyDeviceSymbolLinkName02");
        if (!NT_SUCCESS(ntstatus))
        {
            DbgPrint("IoCreateDevice Failed");
            return ntstatus;
        }
    
        DbgPrint("%ws", reg_path->Buffer);
        DbgPrint("driver load success...");
        return STATUS_SUCCESS;
    }
    
    NTSTATUS DeviceCreate(PDEVICE_OBJECT pDeviceObject, PIRP pIrp)
    {
        //业务代码区
    
        //设置返回状态
        pIrp->IoStatus.Status = STATUS_SUCCESS;
        pIrp->IoStatus.Information = 0;
        IoCompleteRequest(pIrp, IO_NO_INCREMENT);
        DbgPrint("create device success...%d", pDeviceObject->ActiveThreadCount);
        return STATUS_SUCCESS;
    }
    
    NTSTATUS DeviceClose(PDEVICE_OBJECT pDeviceObject, PIRP pIrp)
    {
        //业务代码区
    
        //设置返回状态
        pIrp->IoStatus.Status = STATUS_SUCCESS;//getLastError()得到的值
        pIrp->IoStatus.Information = 0;            //返回给3环多少数据,没有填0
        IoCompleteRequest(pIrp, IO_NO_INCREMENT);
        DbgPrint("create device success...%d", pDeviceObject->ActiveThreadCount);
        return STATUS_SUCCESS;
    }
    
    NTSTATUS DeviceRead(PDEVICE_OBJECT pDeviceObject, PIRP pIrp)
    {
        PIO_STACK_LOCATION iostack;
        iostack = IoGetCurrentIrpStackLocation(pIrp);
        ULONG length = iostack->Parameters.Read.Length;
        PVOID pBuffer = pIrp->AssociatedIrp.SystemBuffer;
    
        pIrp->IoStatus.Status = STATUS_SUCCESS;//getLastError()得到的值
        pIrp->IoStatus.Information = length;            //返回给3环多少数据,没有填0
        RtlFillMemory(pBuffer, length, 0xAA);
        IoCompleteRequest(pIrp, IO_NO_INCREMENT);
        DbgPrint("read device success...%d", pDeviceObject->ActiveThreadCount);
        return STATUS_SUCCESS;
    }
    
    NTSTATUS DemoCreateDevice(PDRIVER_OBJECT pDriver,PCWSTR devName,PCWSTR devSymName)
    {
        PDEVICE_OBJECT pDevice;
        PDEVICE_EXTENSION pDevExt;
    
        UNICODE_STRING DeviceName;
        UNICODE_STRING SymbolLinkName;
    
        RtlInitUnicodeString(&DeviceName, devName);
        RtlInitUnicodeString(&SymbolLinkName, devSymName);
    
        NTSTATUS ntstatus = STATUS_SUCCESS;
        ntstatus = IoCreateDevice(pDriver, 0, &DeviceName, FILE_DEVICE_UNKNOWN, FILE_DEVICE_SECURE_OPEN, FALSE, &pDevice);
        if (!NT_SUCCESS(ntstatus))
        {
            DbgPrint("IoCreateDevice Failed");
            return ntstatus;
        }
    
        ntstatus = IoCreateSymbolicLink(&SymbolLinkName, &DeviceName);
        if (!NT_SUCCESS(ntstatus))
        {
            DbgPrint("IoCreateSymbolicLink Failed");
            IoDeleteDevice(pDevice);
            return ntstatus;
        }
    
        pDevice->Flags |= DO_BUFFERED_IO;
    
        pDevExt = (PDEVICE_EXTENSION)pDevice->DeviceExtension;
        pDevExt->pDevObj = pDevice;
        pDevExt->ustrDeviceName = DeviceName;
        pDevExt->ustrSymbolLinkName = SymbolLinkName;
        return ntstatus;
    }
  • 相关阅读:
    13.Convert BST to Greater Tree(将树转为更大树)
    13.调用数组顺序使奇数位于偶数前面
    12.数值的整数次方
    11.二进制中1的个数
    12.Hamming Distance(汉明距离)
    11.Find All Numbers Disappeared in an Array(找出数组中缺失的数)
    10.Find All Anagrams in a String(在一个字符串中发现所有的目标串排列)
    垃圾收集器与内存分配策略---垃圾收集器
    线程之间的协作
    1287. Mars Canals(DP)
  • 原文地址:https://www.cnblogs.com/a-s-m/p/12343213.html
Copyright © 2011-2022 走看看