加载驱动程序

一个ＫＭＤ都需要创建设备和符号链接，它们分别为kernel和win32提供了名字，这样kernel和win32就可以正确地使用KMD
相关例程　IoCreateDevice　IoDeleteDevice　IoCreateSymbolicLink　IoDeleteSymbolicLink　查阅ＤＤＫ文档

---

/*

driver.c

*/

#include <ntddk.h>



#define FILE_DEVICE_HELLO 0x00008010



/*函数声明*/

VOID DriverUnload(PDRIVER_OBJECT DriverObject);

/*DriverEntry例程*/

NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject,IN PUNICODE_STRING RegistryPath)

{

    UNICODE_STRING  DeviceNameUnicodeString;

    UNICODE_STRING  DeviceLinkUnicodeString;

    NTSTATUS        ntStatus;

    PDEVICE_OBJECT  DeviceObject = NULL;





    RtlInitUnicodeString (&DeviceNameUnicodeString, L"\\Device\\dload");    

    

    /*创建设备*/

    ntStatus = IoCreateDevice (DriverObject,

                                0,

                                &DeviceNameUnicodeString,

                                FILE_DEVICE_HELLO,

                                0,

                                TRUE,

                                &DeviceObject);



    if (NT_SUCCESS(ntStatus))

    {

        /*卸载例程*/

        DriverObject->DriverUnload                         = DriverUnload;



        /*创建符号链接*/

        RtlInitUnicodeString (&DeviceLinkUnicodeString, L"\\DosDevices\\dload");

        ntStatus = IoCreateSymbolicLink (&DeviceLinkUnicodeString,&DeviceNameUnicodeString);



        if (!NT_SUCCESS(ntStatus))

        {

            DbgPrint ("ERROR: IoCreateSymbolicLink failed");

            IoDeleteDevice (DeviceObject);

        }

    }

    else

    {

        DbgPrint ("ERROR: IoCreateDevice failed");

    }



    DbgPrint("dload");



    return ntStatus;



}

/*DriverUnload*/

VOID DriverUnload(PDRIVER_OBJECT DriverObject)

{

    UNICODE_STRING DeviceLinkUnicodeString;

    NTSTATUS ntStatus;





    RtlInitUnicodeString (&DeviceLinkUnicodeString, L"\\DosDevices\\dload");

    

    ntStatus = IoDeleteSymbolicLink (&DeviceLinkUnicodeString);



    if (NT_SUCCESS(ntStatus))

    {

        IoDeleteDevice (DriverObject->DeviceObject);

    }

    else

    {

        DbgPrint ("ERROR: IoDeleteSymbolicLink");

    }



}




 
   




#include <windows.h>

#include <Winsvc.h>

#include <stdio.h>



int main()

{

    SC_HANDLE hSrvMgr; /*服务管理器句柄*/

    SC_HANDLE hSrvDrv;/*KMD服务名柄*/

    SERVICE_STATUS SvrSta;

    char szDir[256];

    DWORD dwSize;

    GetCurrentDirectory( dwSize, szDir );/*取当前目录 */

    strcat(szDir,"\\dload.sys");

    LPCTSTR lpszBinaryPathName=TEXT(szDir);





    puts("任意键开始");

    getchar();

    hSrvMgr=OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS ); //打开服务控制管理器

    if(hSrvMgr)

    {

        puts("打开服务管理器：成功！");

    }

    else

    {

        puts("打开服务管理器：失败，任意键退出");

        getchar();

        exit(1);

    }



    puts("任意键继续");

    getchar();

    hSrvDrv= CreateService( hSrvMgr,

                            TEXT("dload"), //SYSTEM\CurrentControlSet\Services 驱动程序的在注册表中的名字

                            TEXT("dload"), // 注册表驱动程序的 DisplayName 值

                            SERVICE_ALL_ACCESS, // 加载驱动程序的访问权限

                            SERVICE_KERNEL_DRIVER,// 表示加载的服务是驱动程序

                            SERVICE_DEMAND_START, // 注册表驱动程序的 Start 值

                            SERVICE_ERROR_IGNORE, // 注册表驱动程序的 ErrorControl 值

                            lpszBinaryPathName, // 注册表驱动程序的 ImagePath 值

                            NULL,

                            NULL,

                            NULL,

                            NULL,

                            NULL);

    if(hSrvDrv)

    {

        puts("服务创建：成功！");

    }

    else

    {

        puts("服务创建：失败，任意键退出");

        getchar();

        exit(1);

    }

    puts("任意键继续");

    getchar();

    if(StartService( hSrvDrv, NULL, NULL ))

    {

        puts("服务启动：成功，ＫＭＤ加载成功");

    }

    else

    {

        puts("服务启动：失败，任意键退出");

        getchar();

        exit(1);

    }



    puts("任意键继续");

    getchar();

    if( ControlService(hSrvDrv, SERVICE_CONTROL_STOP , &SvrSta ) )

    {

        puts("服务停止：成功，ＫＭＤ不工作");

    }

    else

    {

        puts("服务停止：失败，需要重启再次加载，任意键退出");

        getchar();

        exit(1);

    }



    puts("任意键继续");

    getchar();

    if( DeleteService( hSrvDrv ) )

    {

        puts("服务卸载：成功");

    }

    else

    {

        puts("服务卸载：失败，任意键退出");

        getchar();

        exit(1);

    }



    CloseServiceHandle( hSrvDrv );

    CloseServiceHandle( hSrvMgr );

    puts("任意键退出");

    getchar();

    return 0;

}