一个驱动加载器代码

最新推荐文章于 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 186
背景 自己在钻研Windows内核编程的时候，经常需要使用驱动加载器加载自己写的小驱动程序进行测试。刚开始的时候，使用的是别人写的驱动加载器，但是，越用越不顺眼，因为感觉驱动加载器功能包括：创建服务、启动服务、停止服务、删除服务，这四个功能而已。但是，别人写的功能太过复杂，还增加了其它额外的功能，或者是界面太丑，按钮太小了。所以，就决定自己写一个驱动加载器...
驱动加载工具(附源码)
08-18
根据著名软件InstDvr驱动加载工具 模仿写的在学习驱动编程时很有用 同时提供的源码 供大家学习之用 软件用Win API SDK编写 所用语言是C语言 编译环境VC++ 6.0
驱动加载器
09-26
在下载吧下载的驱动加载器，自己学习用的，也测试过，安全无毒，连广告都没....
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伺服驱动器的各项特性和操作方法，旨在帮助用户...
AHT20模拟IIC获取温湿度的驱动代码
09-28
4. **数据校验**：为了确保数据的准确性，AHT20返回的数据会包含一个校验和。驱动代码中会进行校验，如`AHT20_CheckData()`，以验证接收到的数据是否有效。 5. **数据处理**：获取到原始数据后，通常需要进行一定的...
高创伺服驱动器故障维修(8.闪烁)
12-16
闪烁)"这个主题聚焦于高创品牌的伺服驱动器在遇到特定故障模式——闪烁问题时的诊断与修复。 高创伺服驱动器是一款高性能的电机控制器，广泛应用于各种工业应用中，如机器人、精密机床和自动化生产线。当驱动器显示...
动态加载驱动程序源代码
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 824
头文件 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...
一个驱动加载器
baidu_39410688的博客
08-10 137
【代码】一个驱动加载器。
驱动加载工具(VC7源码)
guoyi987的专栏
02-21 602
开发环境VC7 + DDK 3790.1830       XP SP3             用来做些简单的驱动测试不错。   CSDN下载地址：http://download.csdn.net/detail/guoyi987/4077345
NT驱动加载器
superqiong2011的专栏
06-28 1230
1.windows驱动分类 由于windows版本发展，出现过几种不同驱动模型：vxd、NT、WDM、WDF。前面的vxd主要同之前的win95/98系统相匹配，后面的NT、WDM、WDF则是同后面的NT内核的windows系统匹配。 ps：个人猜测，其实应该只有2种类型的驱动模型vxd和NT，这2种驱动模型分别对应的是windows发展过程中使用到的2种不同内核：win95/98的wince
写一个uboot 霍尔驱动代码
最新发布
10-17
在嵌入式Linux环境中，编写U-Boot（统一固件加载器）的霍尔传感器驱动程序通常涉及以下几个步骤： 1. **头文件包含**： ```c #include <linux/hwmon.h> // 硬件监控设备头文件 #include "u-boot.h" // U-Boot...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值