驱动开发：内核封装TDI网络通信接口

2024-02-16 09:20
文章标签 封装 开发 驱动 网络 内核 通信接口 tdi

本文主要是介绍驱动开发：内核封装TDI网络通信接口，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

在上一篇文章《驱动开发：内核封装WSK网络通信接口》中，LyShark已经带大家看过了如何通过WSK接口实现套接字通信，但WSK实现的通信是内核与内核模块之间的，而如果需要内核与应用层之间通信则使用TDK会更好一些因为它更接近应用层，本章将使用TDK实现，TDI全称传输驱动接口，其主要负责连接Socket和协议驱动，用于实现访问传输层的功能，该接口比NDIS更接近于应用层，在早期Win系统中常用于实现过滤防火墙，同样经过封装后也可实现通信功能，本章将运用TDI接口实现驱动与应用层之间传输字符串，结构体，多线程收发等技术。

  • TDI传输字符串
  • TDI多线程收发
  • TDI传数结构实现认证

TDI 传输字符串: 服务端在应用层侦听，客户端是驱动程序，驱动程序加载后自动连接应用层并发送消息。

首先来看应用层(服务端)代码，具体我就不说了，来看教程的都是有基础的。

// 署名权
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: me@lyshark.com#define _CRT_SECURE_NO_WARNINGS
#include <iostream>  
#include <winsock2.h>  #pragma comment(lib,"ws2_32.lib")
#define PORT 8888 int main(int argc, char *argv[])
{printf("hello lyshark.com \n");WSADATA WSAData;SOCKET sock, msgsock;struct sockaddr_in ServerAddr;if (WSAStartup(MAKEWORD(2, 0), &WSAData) != SOCKET_ERROR){ServerAddr.sin_family = AF_INET;ServerAddr.sin_port = htons(PORT);ServerAddr.sin_addr.s_addr = INADDR_ANY;sock = socket(AF_INET, SOCK_STREAM, 0);int BindRet = bind(sock, (LPSOCKADDR)&ServerAddr, sizeof(ServerAddr));int LinsRet = listen(sock, 10);}while (1){char buf[1024] = { 0 };msgsock = accept(sock, (LPSOCKADDR)0, (int *)0);memset(buf, 0, sizeof(buf));recv(msgsock, buf, 1024, 0);printf("内核返回: %s \n", buf);char send_buffer[1024] = { 0 };memset(send_buffer, 0, 1024);strcpy(send_buffer, "Hi,R0 !");send(msgsock, send_buffer, strlen(send_buffer), 0);closesocket(msgsock);}closesocket(sock);WSACleanup();return 0;
}

再来是驱动层代码，如下所示；

// 署名权
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: me@lyshark.com#include "MyTDI.hpp"// 发送接收数据
NTSTATUS SendOnRecv()
{NTSTATUS status = STATUS_SUCCESS;HANDLE hTdiAddress = NULL;HANDLE hTdiEndPoint = NULL;PDEVICE_OBJECT pTdiAddressDevObj = NULL;PFILE_OBJECT pTdiEndPointFileObject = NULL;LONG pServerIp[4] = { 127, 0, 0, 1 };LONG lServerPort = 8888;UCHAR szSendData[] = "hello lyshark";ULONG ulSendDataLength = 1 + strlen(szSendData);HANDLE hThread = NULL;// TDI初始化status = TdiOpen(&pTdiAddressDevObj, &pTdiEndPointFileObject, &hTdiAddress, &hTdiEndPoint);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}// TDI TCP连接服务器status = TdiConnection(pTdiAddressDevObj, pTdiEndPointFileObject, pServerIp, lServerPort);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}// TDI TCP发送信息status = TdiSend(pTdiAddressDevObj, pTdiEndPointFileObject, szSendData, ulSendDataLength);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}DbgPrint("发送: %s\n", szSendData);// 创建接收信息多线程, 循环接收信息char szRecvData[1024] = { 0 };ULONG ulRecvDataLenngth = 1024;RtlZeroMemory(szRecvData, ulRecvDataLenngth);// TDI TCP接收信息do{ulRecvDataLenngth = TdiRecv(pTdiAddressDevObj, pTdiEndPointFileObject, szRecvData, ulRecvDataLenngth);if (0 < ulRecvDataLenngth){DbgPrint("接收数据: %s\n", szRecvData);break;;}} while (TRUE);// 释放TdiClose(pTdiEndPointFileObject, hTdiAddress, hTdiEndPoint);return STATUS_SUCCESS;
}VOID UnDriver(PDRIVER_OBJECT driver)
{DbgPrint("驱动卸载成功 \n");
}NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{for (int x = 0; x < 10; x++){SendOnRecv();}DbgPrint("驱动加载成功 \n");Driver->DriverUnload = UnDriver;return STATUS_SUCCESS;
}

首先运行应用层开启服务端侦听，然后运行驱动程序，会输出如下信息；

TDI 多线程收发包: 实现驱动内部发送数据包后开启一个线程用于等待应用层返回并输出结果，多线程收发在发送数据包后需要创建新的线程等待接收。

首先是服务端代码。

// 署名权
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: me@lyshark.com#define _CRT_SECURE_NO_WARNINGS
#include <iostream>  
#include <winsock2.h>  #pragma comment(lib,"ws2_32.lib")
#define PORT 8888 int main(int argc, char *argv[])
{printf("hello lyshark.com \n");WSADATA WSAData;SOCKET sock, msgsock;struct sockaddr_in ServerAddr;if (WSAStartup(MAKEWORD(2, 0), &WSAData) != SOCKET_ERROR){ServerAddr.sin_family = AF_INET;ServerAddr.sin_port = htons(PORT);ServerAddr.sin_addr.s_addr = INADDR_ANY;sock = socket(AF_INET, SOCK_STREAM, 0);int BindRet = bind(sock, (LPSOCKADDR)&ServerAddr, sizeof(ServerAddr));int LinsRet = listen(sock, 10);}while (1){char buf[1024] = { 0 };msgsock = accept(sock, (LPSOCKADDR)0, (int *)0);memset(buf, 0, sizeof(buf));recv(msgsock, buf, 1024, 0);printf("内核返回: %s \n", buf);char send_buffer[1024] = { 0 };memset(send_buffer, 0, 1024);strcpy(send_buffer, "Hi,R0 !");send(msgsock, send_buffer, strlen(send_buffer), 0);closesocket(msgsock);}closesocket(sock);WSACleanup();return 0;
}

驱动程序代码如下，RecvThreadProc主要负责数据接收，SendThreadData负责数据发送。

// 署名权
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: me@lyshark.com#include "LySocket.hpp"typedef struct _MY_DATA
{PDEVICE_OBJECT pTdiAddressDevObj;PFILE_OBJECT pTdiEndPointFileObject;HANDLE hTdiAddress;HANDLE hTdiEndPoint;
}MY_DATA, *PMY_DATA;// 接收信息多线程
VOID RecvThreadProc(_In_ PVOID StartContext)
{PMY_DATA pMyData = (PMY_DATA)StartContext;NTSTATUS status = STATUS_SUCCESS;char szRecvData[1024] = { 0 };ULONG ulRecvDataLenngth = 1024;RtlZeroMemory(szRecvData, ulRecvDataLenngth);// TDI TCP接收信息do{ulRecvDataLenngth = TdiRecv(pMyData->pTdiAddressDevObj, pMyData->pTdiEndPointFileObject, szRecvData, ulRecvDataLenngth);if (0 < ulRecvDataLenngth){DbgPrint("线程句柄:%x --> 接收数据包: %s\n", pMyData->hTdiEndPoint, szRecvData);break;;}} while (TRUE);// 释放TdiClose(pMyData->pTdiEndPointFileObject, pMyData->hTdiAddress, pMyData->hTdiEndPoint);ExFreePool(pMyData);
}// 多线程发送
NTSTATUS SendThreadData()
{NTSTATUS status = STATUS_SUCCESS;HANDLE hTdiAddress = NULL;HANDLE hTdiEndPoint = NULL;PDEVICE_OBJECT pTdiAddressDevObj = NULL;PFILE_OBJECT pTdiEndPointFileObject = NULL;LONG pServerIp[4] = { 127, 0, 0, 1 };LONG lServerPort = 8888;UCHAR szSendData[] = "hello lyshark";ULONG ulSendDataLength = 1 + strlen(szSendData);HANDLE hThread = NULL;// TDI初始化status = TdiOpen(&pTdiAddressDevObj, &pTdiEndPointFileObject, &hTdiAddress, &hTdiEndPoint);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}// TDI TCP连接服务器status = TdiConnection(pTdiAddressDevObj, pTdiEndPointFileObject, pServerIp, lServerPort);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}// TDI TCP发送信息status = TdiSend(pTdiAddressDevObj, pTdiEndPointFileObject, szSendData, ulSendDataLength);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}DbgPrint("发送 %s\n", szSendData);// 创建接收信息多线程, 循环接收信息PMY_DATA pMyData = ExAllocatePool(NonPagedPool, sizeof(MY_DATA));pMyData->pTdiAddressDevObj = pTdiAddressDevObj;pMyData->pTdiEndPointFileObject = pTdiEndPointFileObject;pMyData->hTdiAddress = hTdiAddress;pMyData->hTdiEndPoint = hTdiEndPoint;PsCreateSystemThread(&hThread, 0, NULL, NtCurrentProcess(), NULL, RecvThreadProc, pMyData);
}VOID UnDriver(PDRIVER_OBJECT driver)
{DbgPrint("驱动卸载成功 \n");
}NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{DbgPrint("hello lyshark.com \n");for (int x = 0; x < 10; x++){SendThreadData();}Driver->DriverUnload = UnDriver;return STATUS_SUCCESS;
}

运行应用层服务端等待侦听，运行驱动程序输出如下效果；

TDI 传数结构实现认证: 驱动内部发送结构体给应用层，应用层验证结构体成员，此功能可实现对驱动程序的控制机制，例如是否允许驱动加载卸载等，通常用于驱动辅助认证。

应用层代码

// 署名权
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: me@lyshark.com#define _CRT_SECURE_NO_WARNINGS
#include <iostream>  
#include <winsock2.h>  #pragma comment(lib,"ws2_32.lib")
#define PORT 8888// 传输结构体
typedef struct
{int uuid;char username[256];char password[256];
}SocketData;int main(int argc, char *argv[])
{printf("hello lyshark.com \n");WSADATA WSAData;SOCKET sock, msgsock;struct sockaddr_in ServerAddr;if (WSAStartup(MAKEWORD(2, 0), &WSAData) != SOCKET_ERROR){ServerAddr.sin_family = AF_INET;ServerAddr.sin_port = htons(PORT);ServerAddr.sin_addr.s_addr = INADDR_ANY;sock = socket(AF_INET, SOCK_STREAM, 0);int BindRet = bind(sock, (LPSOCKADDR)&ServerAddr, sizeof(ServerAddr));int LinsRet = listen(sock, 10);}while (1){char buf[8192] = { 0 };msgsock = accept(sock, (LPSOCKADDR)0, (int *)0);memset(buf, 0, sizeof(buf));// 接收返回数据recv(msgsock, buf, sizeof(SocketData), 0);// 强转结构体SocketData* msg = (SocketData*)buf;printf("UUID = %d \n", msg->uuid);printf("名字 = %s \n", msg->username);printf("密码 = %s \n", msg->password);// 验证通过则继续使用if ((strcmp(msg->username, "lyshark") == 0) && (strcmp(msg->password, "123") == 0)){char send_buffer[8192] = { 0 };memset(send_buffer, 0, 8192);strcpy(send_buffer, "success");send(msgsock, send_buffer, strlen(send_buffer), 0);closesocket(msgsock);}// 不通过则禁止驱动加载else{char send_buffer[8192] = { 0 };memset(send_buffer, 0, 8192);strcpy(send_buffer, "error");send(msgsock, send_buffer, strlen(send_buffer), 0);closesocket(msgsock);}}closesocket(sock);WSACleanup();return 0;
}

驱动层代码

// 署名权
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: me@lyshark.com
#include "LySocket.hpp"// 传输结构体
typedef struct
{int uuid;char username[256];char password[256];
}SocketData;// 验证账号密码是否正确
BOOLEAN CheckDriver()
{NTSTATUS status = STATUS_SUCCESS;HANDLE hTdiAddress = NULL;HANDLE hTdiEndPoint = NULL;PDEVICE_OBJECT pTdiAddressDevObj = NULL;PFILE_OBJECT pTdiEndPointFileObject = NULL;LONG pServerIp[4] = { 127, 0, 0, 1 };LONG lServerPort = 8888;// TDI初始化status = TdiOpen(&pTdiAddressDevObj, &pTdiEndPointFileObject, &hTdiAddress, &hTdiEndPoint);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}// TDI TCP连接服务器status = TdiConnection(pTdiAddressDevObj, pTdiEndPointFileObject, pServerIp, lServerPort);if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}SocketData ptr;RtlZeroMemory(&ptr, sizeof(SocketData));// 填充结构ptr.uuid = 1001;RtlCopyMemory(ptr.username, "lyshark", strlen("xxxxxxx"));RtlCopyMemory(ptr.password, "123123", strlen("xxxxxx"));// TDI TCP发送信息status = TdiSend(pTdiAddressDevObj, pTdiEndPointFileObject, &ptr, sizeof(SocketData));if (!NT_SUCCESS(status)){return STATUS_SUCCESS;}// 创建接收信息多线程, 循环接收信息char szRecvData[8192] = { 0 };ULONG ulRecvDataLenngth = 8192;RtlZeroMemory(szRecvData, ulRecvDataLenngth);// TDI TCP接收信息do{ulRecvDataLenngth = TdiRecv(pTdiAddressDevObj, pTdiEndPointFileObject, szRecvData, ulRecvDataLenngth);if (0 < ulRecvDataLenngth){DbgPrint("接收数据: %s\n", szRecvData);if (strncmp(szRecvData, "success", 7) == 0){// 释放TdiClose(pTdiEndPointFileObject, hTdiAddress, hTdiEndPoint);return TRUE;}else if (strncmp(szRecvData, "error", 5) == 0){// 释放TdiClose(pTdiEndPointFileObject, hTdiAddress, hTdiEndPoint);return FALSE;}break;;}} while (TRUE);// 释放TdiClose(pTdiEndPointFileObject, hTdiAddress, hTdiEndPoint);return STATUS_SUCCESS;
}VOID UnDriver(PDRIVER_OBJECT driver)
{DbgPrint("驱动卸载成功 \n");
}NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{DbgPrint("hello lyshark.com \n");BOOLEAN ref = CheckDriver();if (ref == FALSE){DbgPrint("[LyShark.com] 驱动已过期,无法加载 \n");Driver->DriverUnload = UnDriver;return STATUS_SUCCESS;}DbgPrint("[*] 驱动正常使用 \n");Driver->DriverUnload = UnDriver;return STATUS_SUCCESS;
}

运行应用层服务端，并运行驱动程序，则会验证该驱动是否合法，如果合法则加载不合法则拒绝；

这篇关于驱动开发：内核封装TDI网络通信接口的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/714130。

相关文章

如何在Ubuntu上安装NVIDIA显卡驱动? Ubuntu安装英伟达显卡驱动教程

如何在Ubuntu上安装NVIDIA显卡驱动? Ubuntu安装英伟达显卡驱动教程

《如何在Ubuntu上安装NVIDIA显卡驱动?Ubuntu安装英伟达显卡驱动教程》Windows系统不同，Linux系统通常不会自动安装专有显卡驱动，今天我们就来看看Ubuntu系统安装英伟达显卡... 对于使用NVIDIA显卡的Ubuntu用户来说，正确安装显卡驱动是获得最佳图形性能的关键。与Windo
阅读更多...
Linux高并发场景下的网络参数调优实战指南

Linux高并发场景下的网络参数调优实战指南

《Linux高并发场景下的网络参数调优实战指南》在高并发网络服务场景中,Linux内核的默认网络参数往往无法满足需求,导致性能瓶颈、连接超时甚至服务崩溃,本文基于真实案例分析,从参数解读、问题诊断到优... 目录一、问题背景：当并发连接遇上性能瓶颈1.1 案例环境1.2 初始参数分析二、深度诊断：连接状态与
阅读更多...
一文教你如何解决Python开发总是import出错的问题

一文教你如何解决Python开发总是import出错的问题

《一文教你如何解决Python开发总是import出错的问题》经常朋友碰到Python开发的过程中import包报错的问题,所以本文将和大家介绍一下可编辑安装（EditableInstall）模式,可... 目录摘要1. 可编辑安装（Editable Install）模式到底在解决什么问题？2. 原理3.
阅读更多...
一文详解如何在Vue3中封装API请求

一文详解如何在Vue3中封装API请求

《一文详解如何在Vue3中封装API请求》在现代前端开发中,API请求是不可避免的一部分,尤其是与后端交互时,下面我们来看看如何在Vue3项目中封装API请求,让你在实现功能时更加高效吧... 目录为什么要封装API请求1. vue 3项目结构2. 安装axIOS3. 创建API封装模块4. 封装API请求
阅读更多...
Python+PyQt5开发一个Windows电脑启动项管理神器

Python+PyQt5开发一个Windows电脑启动项管理神器

《Python+PyQt5开发一个Windows电脑启动项管理神器》：本文主要介绍如何使用PyQt5开发一款颜值与功能并存的Windows启动项管理工具,不仅能查看/删除现有启动项,还能智能添加新... 目录开篇：为什么我们需要启动项管理工具功能全景图核心技术解析1. Windows注册表操作2. 启动文件
阅读更多...
嵌入式Linux之使用设备树驱动GPIO的实现方式

嵌入式Linux之使用设备树驱动GPIO的实现方式

《嵌入式Linux之使用设备树驱动GPIO的实现方式》：本文主要介绍嵌入式Linux之使用设备树驱动GPIO的实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐... 目录一、设备树配置1.1 添加 pinctrl 节点1.2 添加 LED 设备节点二、编写驱动程序2.1
阅读更多...
嵌入式Linux驱动中的异步通知机制详解

嵌入式Linux驱动中的异步通知机制详解

《嵌入式Linux驱动中的异步通知机制详解》：本文主要介绍嵌入式Linux驱动中的异步通知机制,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录前言一、异步通知的核心概念1. 什么是异步通知2. 异步通知的关键组件二、异步通知的实现原理三、代码示例分析1. 设备结构
阅读更多...
使用Python开发Markdown兼容公式格式转换工具

使用Python开发Markdown兼容公式格式转换工具

《使用Python开发Markdown兼容公式格式转换工具》在技术写作中我们经常遇到公式格式问题,例如MathML无法显示,LaTeX格式错乱等,所以本文我们将使用Python开发Markdown兼容... 目录一、工具背景二、环境配置（Windows 10/11）1. 创建conda环境2. 获取XSLT
阅读更多...
Android开发环境配置避坑指南

Android开发环境配置避坑指南

《Android开发环境配置避坑指南》本文主要介绍了Android开发环境配置过程中遇到的问题及解决方案,包括VPN注意事项、工具版本统一、Gerrit邮箱配置、Git拉取和提交代码、MergevsR... 目录网络环境：VPN 注意事项工具版本统一：android Studio & JDKGerrit的邮
阅读更多...
Python开发文字版随机事件游戏的项目实例

Python开发文字版随机事件游戏的项目实例

《Python开发文字版随机事件游戏的项目实例》随机事件游戏是一种通过生成不可预测的事件来增强游戏体验的类型,在这篇博文中,我们将使用Python开发一款文字版随机事件游戏,通过这个项目,读者不仅能够... 目录项目概述2.1 游戏概念2.2 游戏特色2.3 目标玩家群体技术选择与环境准备3.1 开发环境3
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2