一个驱动加载器代码

最新推荐文章于 2024-08-10 22:57:55 发布
最新推荐文章于 2024-08-10 22:57:55 发布
阅读量1.4k 收藏 1
点赞数
分类专栏： 驱动相关
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/late0001/article/details/45618493
版权 
//原作者：不明（来源于网络）

/*
使用方法：
修改DRV_NAME和DRV_FILENAME定义
HANDLE openDriver(void)：打开驱动建立的符号链接的句柄
void uninstallDriver(void)：卸载驱动（只有把openDriver打开的句柄关闭才可以成功卸载）
*/

#pragma comment(lib,"user32.lib")
#pragma comment(lib,"Advapi32.lib")

#define DRV_NAME		"myDriver"//驱动名
#define DRV_FILENAME	"myDriver.sys"//驱动文件
#define DRV_SYMBOL_LINK  "\\\\.\\LenfSys"              // \??\LenfSys

#define STATUS_SUCCESS     ((NTSTATUS)0x00000000L)

typedef LONG NTSTATUS;

typedef struct _STRING
{
    USHORT  Length;
    USHORT  MaximumLength;
    PCHAR  Buffer;
} ANSI_STRING, *PANSI_STRING;

typedef struct _UNICODE_STRING
{
    USHORT  Length;
    USHORT  MaximumLength;
    PWSTR  Buffer;
} UNICODE_STRING, *PUNICODE_STRING;

VOID AfxMessageBox(char *sz)
{
	LoadLibraryA("user32.dll");
    MessageBoxA(0,sz,DRV_NAME,0);
}

//*********************************************************************************************
// Assign loaddriver priviledge to our process, so we can load our support driver.
//*********************************************************************************************

BOOL getLoadDriverPriv()
{
    HANDLE hToken;

    if(OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES, &hToken))
    {
        LUID huid;
        if(LookupPrivilegeValueA(NULL, "SeLoadDriverPrivilege", &huid))
        {
            LUID_AND_ATTRIBUTES priv;
            priv.Attributes = SE_PRIVILEGE_ENABLED;
            priv.Luid = huid;

            TOKEN_PRIVILEGES tp;
            tp.PrivilegeCount = 1;
            tp.Privileges[0] = priv;

            if(AdjustTokenPrivileges(hToken, FALSE, &tp, 0, NULL, NULL))
            {
                return TRUE;
            }
         }
     }
    return FALSE;
}


//*********************************************************************************************
// Sets up the necessary registry settings to load the support driver
//*********************************************************************************************
BOOL setupRegistry()
{
    HKEY hkey;
    if(RegCreateKeyA(HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\"DRV_NAME, &hkey) != ERROR_SUCCESS)
        return FALSE;

    DWORD val;
    val = 1;
    if(RegSetValueExA(hkey, "Type", 0, REG_DWORD, (PBYTE)&val, sizeof(val)) != ERROR_SUCCESS)
        return FALSE;

    if(RegSetValueExA(hkey, "ErrorControl", 0, REG_DWORD, (PBYTE)&val, sizeof(val)) != ERROR_SUCCESS)
        return FALSE;

    val = 3;
    if(RegSetValueExA(hkey, "Start", 0, REG_DWORD, (PBYTE)&val, sizeof(val)) != ERROR_SUCCESS)
        return FALSE;

    char *imgName = "System32\\DRIVERS\\"DRV_FILENAME;

    if(RegSetValueExA(hkey, "ImagePath", 0, REG_EXPAND_SZ, (PBYTE)imgName, strlen(imgName)) != ERROR_SUCCESS)
        return FALSE;

    return TRUE;
}


//*********************************************************************************************
// Actual code to load our driver into memory
//
//*********************************************************************************************

BOOL loadDriver()
{
    HMODULE hntdll;

    NTSTATUS (WINAPI *_RtlAnsiStringToUnicodeString)
		(PUNICODE_STRING  DestinationString,IN PANSI_STRING  SourceString,
		IN BOOLEAN b);

    VOID (WINAPI *_RtlInitAnsiString)
    (IN OUT PANSI_STRING  DestinationString,
     IN PCHAR  SourceString);

    NTSTATUS (WINAPI * _ZwLoadDriver)
    (IN PUNICODE_STRING DriverServiceName);

    NTSTATUS (WINAPI * _ZwUnloadDriver)
    (IN PUNICODE_STRING DriverServiceName);

    VOID (WINAPI * _RtlFreeUnicodeString)
    (IN PUNICODE_STRING  UnicodeString);


    hntdll = GetModuleHandleA("ntdll.dll");

    *(FARPROC *)&_ZwLoadDriver = GetProcAddress(hntdll, "NtLoadDriver");

    *(FARPROC *)&_ZwUnloadDriver = GetProcAddress(hntdll, "NtUnloadDriver");

    *(FARPROC *)&_RtlAnsiStringToUnicodeString =
        GetProcAddress(hntdll, "RtlAnsiStringToUnicodeString");

    *(FARPROC *)&_RtlInitAnsiString =
        GetProcAddress(hntdll, "RtlInitAnsiString");

    *(FARPROC *)&_RtlFreeUnicodeString =
        GetProcAddress(hntdll, "RtlFreeUnicodeString");

    if(_ZwLoadDriver && _ZwUnloadDriver && _RtlAnsiStringToUnicodeString &&
            _RtlInitAnsiString && _RtlFreeUnicodeString)
    {
        ANSI_STRING aStr;

        _RtlInitAnsiString(&aStr,
                           "\\Registry\\Machine\\System\\CurrentControlSet\\Services\\"DRV_NAME);

        UNICODE_STRING uStr;

        if(_RtlAnsiStringToUnicodeString(&uStr, &aStr, TRUE) != STATUS_SUCCESS)
            return FALSE;
        else
        {
            if(_ZwLoadDriver(&uStr) == STATUS_SUCCESS)
            {
                _RtlFreeUnicodeString(&uStr);
                return TRUE;
            }
            _RtlFreeUnicodeString(&uStr);
        }
    }

    return FALSE;
}


//*********************************************************************************************
// Actual code to remove our driver from memory
//
//*********************************************************************************************

BOOL unloadDriver()
{
// call ntdll APIs
    HMODULE hntdll;
    NTSTATUS (WINAPI * _RtlAnsiStringToUnicodeString)
    (PUNICODE_STRING  DestinationString,
     IN PANSI_STRING  SourceString,
     IN BOOLEAN);

    VOID (WINAPI *_RtlInitAnsiString)
    (IN OUT PANSI_STRING  DestinationString,
     IN PCHAR  SourceString);

    NTSTATUS (WINAPI * _ZwLoadDriver)
    (IN PUNICODE_STRING DriverServiceName);

    NTSTATUS (WINAPI * _ZwUnloadDriver)
    (IN PUNICODE_STRING DriverServiceName);

    VOID (WINAPI * _RtlFreeUnicodeString)
    (IN PUNICODE_STRING  UnicodeString);


    hntdll = GetModuleHandleA("ntdll.dll");

    *(FARPROC *)&_ZwLoadDriver = GetProcAddress(hntdll, "NtLoadDriver");

    *(FARPROC *)&_ZwUnloadDriver = GetProcAddress(hntdll, "NtUnloadDriver");

    *(FARPROC *)&_RtlAnsiStringToUnicodeString =
        GetProcAddress(hntdll, "RtlAnsiStringToUnicodeString");

    *(FARPROC *)&_RtlInitAnsiString =
        GetProcAddress(hntdll, "RtlInitAnsiString");

    *(FARPROC *)&_RtlFreeUnicodeString =
        GetProcAddress(hntdll, "RtlFreeUnicodeString");

    if(_ZwLoadDriver && _ZwUnloadDriver && _RtlAnsiStringToUnicodeString &&
            _RtlInitAnsiString && _RtlFreeUnicodeString)
    {
        ANSI_STRING aStr;

        _RtlInitAnsiString(&aStr,
                           "\\Registry\\Machine\\System\\CurrentControlSet\\Services\\"DRV_NAME);

        UNICODE_STRING uStr;

        if(_RtlAnsiStringToUnicodeString(&uStr, &aStr, TRUE) != STATUS_SUCCESS)
            return FALSE;
        else
        {
            if(_ZwUnloadDriver(&uStr) == STATUS_SUCCESS)
            {
                _RtlFreeUnicodeString(&uStr);
                return TRUE;
            }
            _RtlFreeUnicodeString(&uStr);
        }
    }

    return FALSE;
}


//*********************************************************************************************
// Removes our driver file and registry settings
//
//*********************************************************************************************
void cleanupReg(void)
{
    char sysDir[MAX_PATH + 1];
    GetSystemDirectoryA(sysDir, MAX_PATH); //C:\Windows\system32
    strncat(sysDir, "\\drivers\\"DRV_FILENAME, MAX_PATH); //C:\Windows\system32\驱动文件名
    DeleteFileA(sysDir);//删除该文件
	//删除键值
    RegDeleteKeyA(HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\"DRV_NAME"\\Enum");
    RegDeleteKeyA(HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\"DRV_NAME);
}


//*********************************************************************************************
// Attempts to get a handle to our kernel driver.  If fails, try to install the driver.
//
//*********************************************************************************************
HANDLE openDriver(void)
{
    HANDLE hDevice;
	//CreateFile打开驱动建立的符号链接, 符号连接可以设为，例如："\\\\.\\"DRV_NAME
    hDevice = CreateFileA(DRV_SYMBOL_LINK, GENERIC_WRITE | GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
                          NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if(hDevice == INVALID_HANDLE_VALUE)
    {
        char drvFullPath[MAX_PATH+1];
        char *filePart;

        ZeroMemory(drvFullPath, MAX_PATH);
        GetFullPathNameA(DRV_FILENAME, MAX_PATH, drvFullPath, &filePart);
		//打开驱动文件句柄
        HANDLE hFile = CreateFileA(drvFullPath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
                                   FILE_ATTRIBUTE_NORMAL, 0);
        if(hFile == INVALID_HANDLE_VALUE)
        {
            AfxMessageBox("Cannot find required driver file");
            return INVALID_HANDLE_VALUE;
        }
        else
        {
            CloseHandle(hFile);

            char sysDir[MAX_PATH + 1];
            GetSystemDirectoryA(sysDir, MAX_PATH);
            strncat(sysDir, "\\drivers\\"DRV_FILENAME, MAX_PATH);
            CopyFileA(drvFullPath, sysDir, TRUE);

            if(!getLoadDriverPriv())//获取加载驱动权限
            {
                AfxMessageBox("Error getting load driver privilege! ");
            }
            else
            {
                if(!setupRegistry()) //写入注册表
                {
                    AfxMessageBox("Error setting driver registry keys! Make sure you are running this as Administrator. ");
                }
                else
                {
                    loadDriver(); //加载驱动
					//打开符号链接 
                    hDevice = CreateFileA(DRV_SYMBOL_LINK, GENERIC_WRITE | GENERIC_READ,FILE_SHARE_READ | FILE_SHARE_WRITE,
                                          NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
                    if(hDevice == INVALID_HANDLE_VALUE)
                    {
                        AfxMessageBox("Error loading kernel support driver! Make sure you are running this as Administrator. ");
                    }
                    else
                    {
                        AfxMessageBox("loading kernel support driver success");
                    }
                }
            }
            cleanupReg();//清除该驱动在注册表里的相关键值
        }
    }

    return hDevice;
}


//*********************************************************************************************
// Remove our kernel driver from memory
//
//*********************************************************************************************

void uninstallDriver(void)
{
    char drvFullPath[MAX_PATH+1];
    char *filePart;

    ZeroMemory(drvFullPath, MAX_PATH);
    GetFullPathNameA(DRV_FILENAME, MAX_PATH, drvFullPath, &filePart);
	//打开驱动文件句柄
    HANDLE hFile = CreateFileA(drvFullPath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
                               FILE_ATTRIBUTE_NORMAL, 0);
    if(hFile == INVALID_HANDLE_VALUE)
    {
        AfxMessageBox("Cannot find required driver file ");
        return;
    }
    else
    {
        CloseHandle(hFile);

        char sysDir[MAX_PATH + 1];
        GetSystemDirectoryA(sysDir, MAX_PATH);
        strncat(sysDir, "\\drivers\\"DRV_FILENAME, MAX_PATH);//复制驱动文件到C:\Windows\System32\drivers下
        CopyFileA(drvFullPath, sysDir, TRUE);

        if(!getLoadDriverPriv()) //获取加载权限
        {
            AfxMessageBox("Error getting load driver privilege! ");
        }
        else
        {
            if(!setupRegistry())//建立注册表项
            {
                AfxMessageBox("Error setting driver registry keys! Make sure you are running this as Administrator. ");
            }
            else
            {
                if(unloadDriver())//卸载驱动
                    AfxMessageBox("Support driver successfully unloaded. ");
                else
                    AfxMessageBox("Unload support driver failed.  It is probably not loaded. ");
            }
        }
        cleanupReg(); //清理注册表项
    }
}

DWORD CTL_CODE_GEN(DWORD dwFunction)
{
    return (FILE_DEVICE_UNKNOWN * 65536) | (FILE_ANY_ACCESS * 16384) | (dwFunction * 4) | METHOD_BUFFERED;
}

BOOL IoControl(HANDLE hDrvHandle, DWORD dwIoControlCode, PVOID lpInBuffer, DWORD nInBufferSize, 
	PVOID lpOutBuffer, DWORD nOutBufferSize)
{
    DWORD dwRetSize;
    return DeviceIoControl(hDrvHandle, dwIoControlCode, lpInBuffer, nInBufferSize,
		lpOutBuffer, nOutBufferSize, &dwRetSize, 0);
}

late0001
关注
专栏目录
驱动加载源码
01-08
InstDrv驱动加载源码。
[源码和文档分享]编程实现驱动加载器
qq_38474647的博客
12-30 180
背景 自己在钻研Windows内核编程的时候，经常需要使用驱动加载器加载自己写的小驱动程序进行测试。刚开始的时候，使用的是别人写的驱动加载器，但是，越用越不顺眼，因为感觉驱动加载器功能包括：创建服务、启动服务、停止服务、删除服务，这四个功能而已。但是，别人写的功能太过复杂，还增加了其它额外的功能，或者是界面太丑，按钮太小了。所以，就决定自己写一个驱动加载器...
一个驱动加载器
最新发布
baidu_39410688的博客
08-10 112
【代码】一个驱动加载器。
驱动加载工具(附源码)
08-18
根据著名软件InstDvr驱动加载工具 模仿写的在学习驱动编程时很有用 同时提供的源码 供大家学习之用 软件用Win API SDK编写 所用语言是C语言 编译环境VC++ 6.0
驱动加载器
RedLobster的博客
02-28 623
#欢迎使用Markdown编辑器写博客本Markdown编辑器使用StackEdit修改而来，用它写博客，将会带来全新的体验哦： Markdown和扩展Markdown简洁的语法 代码块高亮 图片链接和图片上传 LaTex数学公式 UML序列图和流程图 离线写博客 导入导出Markdown文件 丰富的快捷键 快捷键 加粗 Ctrl + B 斜体 Ctrl + I 引用 Ctrl
OLED驱动完整程序_1.3寸oled驱动_oled驱动代码_OLED驱动
09-10
- 字符库支持：对于文本显示，驱动代码可能包含内置的ASCII字符集，或者支持自定义字体加载。 4. OLED驱动代码优化： - 位操作：为了提高效率，驱动代码往往利用位操作进行数据打包和解包，减少对内存和CPU资源的...
WINDOWS系统中出现无法加载这个硬件的设备驱动程序(代码39)的解决方法介绍.pdf
07-14
找到并删除`upperfilters`和`lowfilters`键（如果存在），这两个键可能因某些原因导致驱动加载失败。删除后，重启计算机。 4. **使用系统还原**：如果你之前创建过系统还原点，可以尝试恢复到出问题前的状态，这...
高创伺服驱动器.pdf
08-01
【高创伺服驱动器】是高创传动科技开发有限公司（Servotronix Motion Control Ltd.）生产的一款高性能伺服驱动器，广泛应用于自动化领域。本用户手册详细介绍了CDHD2伺服驱动器的各项特性和操作方法，旨在帮助用户...
高创伺服驱动器故障维修(8.闪烁)
12-16
闪烁)"这个主题聚焦于高创品牌的伺服驱动器在遇到特定故障模式——闪烁问题时的诊断与修复。 高创伺服驱动器是一款高性能的电机控制器，广泛应用于各种工业应用中，如机器人、精密机床和自动化生产线。当驱动器显示...
安川伺服驱动器的常用故障代码
03-20
安川伺服驱动器是一种广泛应用于自动化控制系统中的伺服电机驱动装置，它负责接收控制命令并驱动电机按照要求运行。在使用过程中，可能会出现各种故障，安川伺服驱动器通过特定的故障代码来指示发生的异常情况。这些...
动态加载驱动程序源代码
06-14
源代码包含三种不同的驱动加载方法,使用ZwSetSystemInformation函数加载驱动,使用NtLoadDriver函数加载驱动,使用服务控制管理器加载驱动,还包括三种线程注入技术,使用RtlCreateUserThread 函数注入线程,使用CreateRemoteThread函数注入线程,使用NtCreateThreadEx函数注入线程,源代码包含C,C#的demo
SYS驱动文件安装加载启动工具源码 支持文件拖放 Driver Loading Tool
06-01
资源介绍：。源码里面有个GetDeviceName(获取设备名称)的子程序。它其实就是读入驱动文件后 在文件中搜索特征 然后再进行取文本  这样做是为了做个通用的加载 通讯工具 。因为别的加载工具里都是安装 启动 停止 卸载 则少了通讯获取到设备名称就能建立设备连接 通过调用DeviceIoControl和驱动进行通讯。但在后来发现构建通讯结构没有那么简单 加上时间的关系 所以没加这功能..资源作者：。guazi。资源界面：。资源下载：。
驱动加载工具(InstDrv - 中文版)
04-10
>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>软件名称：驱动加载工具(InstDrv - 中文版)软件版本：V1.0版软件类型：个人免费版(无插件)运行环境：Windows操作系统开发作者：Coderui更新日期：2008年04月08日联系邮箱：coderui@163.com作者博客：http://hi.baidu.com/coderui>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>-------------------------------------------------介绍：　　该软件和网上大家常用的“InstDrv”程序功能几乎一样，只是新加入了详细的中文提示信息和拖拽打开文件等，这样可以方便用户的使用。“InstDrv”是一个比较好用的驱动安装、卸载工具。可以帮助您方便的把驱动程序安装上或卸载掉。本程序更多应用于动态调式驱动前的加载操作中。功能：1：执行加载、启动、停止、卸载驱动操作。2：支持文件拖拽打开。3：支持程序窗口总在最前显示。4：操作后有详细的中文提示信息。说明：压缩包中的“MySYS.sys”文件为测试驱动程序，支持加载、启动、停止、卸载驱动等功能，可以使用“Dbgview”工具查看驱动输出的提示信息。-------------------------------------------------详细的中文提示信息如下：状态：支持拖拽获取文件路径.状态：只能一次拖拽一个文件!状态：拖拽获取文件路径成功!状态：请输入驱动文件的路径名!状态：创建句柄失败!状态：因为不是一个有效文件，所以驱动服务安装失败!状态：因为驱动服务已经存在，所以驱动服务安装失败!状态：因为驱动服务已经启动，所以驱动服务安装失败!状态：驱动服务安装失败!状态：驱动服务安装成功!状态：打开驱动服务失败,可能该驱动服务不存在!状态：驱动服务启动成功!状态：因为驱动服务已经启动，所以驱动服务启动失败!状态：因为不是一个有效驱动，所以驱动服务启动失败!状态：因为驱动服务未被停止，所以驱动服务启动失败!状态：驱动服务启动失败!状态：驱动服务停止成功!状态：因为驱动服务没有启动，所以驱动服务停止失败!状态：驱动服务停止失败!状态：驱动服务卸栽成功!状态：因为驱动服务已经启动，所以驱动服务卸栽失败!状态：驱动服务卸栽失败!-------------------------------------------------下载说明：如果当前网页连接地址下载不了该软件，您可以发送邮件到coderui@163.com中，向程序作者所要该软件，谢谢您的支持!-------------------------------------------------解压后EXE程序全文MD5校验值为：591d784d24f5aead3a383535a4100733专用MD5校验器程序下载地址为：http://hi.baidu.com/coderui/blog/item/e3a1282354c128519822ed24.html>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
Windows驱动加载工具DriverMonitor
01-10
Windwos驱动加载工具DriverMonitor，非常的好用。DriverStudio下的DriverMonitor.exe驱动加载调试工具。密码123
64位驱动加载工具(NT式驱动)
05-28
64位驱动加载工具NT式驱动，直接把驱动拖入窗体，点击确认即可加载,
Ring3加载驱动源码
ChinaPrc
11-12 807
头文件 Instdrv.h #ifndef _INSTDRV_H_ #define _INSTDRV_H_ BOOL LoadDeviceDriver( IN const TCHAR * Name, IN const TCHAR * Path, OUT HANDLE * lphDevice, OUT PDWORD Error ); BOOL UnloadDeviceDr...
驱动加载工具(VC7源码)
guoyi987的专栏
02-21 588
开发环境VC7 + DDK 3790.1830       XP SP3             用来做些简单的驱动测试不错。   CSDN下载地址：http://download.csdn.net/detail/guoyi987/4077345
OTIS驱动器故障代码详解
2. 000POWERON: 表示驱动器供电正常，这是启动过程的一个正常状态。 3. 001NEWRUN: 意味着系统正在重新初始化运行，这可能是由于某些操作或电源中断后恢复的结果。 4. 002GOTOSLEEP: 变频器进入了节能模式，这是为了...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值