内核复制文件

最新推荐文章于 2023-11-25 22:36:31 发布

原创最新推荐文章于 2023-11-25 22:36:31 发布 · 808 阅读

0 ·

CC 4.0 BY-SA版权

驱动入门专栏收录该内容

29 篇文章

订阅专栏

本文介绍在Windows 7 x64环境下，使用内核模式驱动进行文件复制的方法。通过示例代码展示了如何创建文件句柄、读取源文件、写入目标文件，并确保操作的正确性和资源释放。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在上一篇的时候我学会了读取文件还有查询文件大小写入文件等然后想写一些代码比如文件复制的什么的

然后我昨天决心学 X64的驱动了然后昨天也把win7 X64的双机调试搞定了然后开始了写win7 X64驱动的第一天

其实 x64 和x86都是差不多的只不过 x64有些数据结构重写了如果还用 x86 的可能就凉了

然后先看个 win7x86 的文件复制的效果把

效果就是里面的log 1 多出了 log 2 说明是成功的

然后看看x64 的效果

win 7 x64 也是完美运行的

那么下面就是代码

#include <ntddk.h>

VOID DriverUnload(PDRIVER_OBJECT driver)
{
	KdPrint(("goodbye"));
}
BOOLEAN copyfile(UNICODE_STRING destfile, UNICODE_STRING srcfile)
{
	HANDLE hdfile=NULL, hsrcfile=NULL;
	PVOID buffer = NULL;
	ULONG len = 0;
	LARGE_INTEGER offset = { 0 };
	IO_STATUS_BLOCK IO = { 0 };
	OBJECT_ATTRIBUTES ob;
	NTSTATUS status;
	BOOLEAN bret = FALSE;
	InitializeObjectAttributes(&ob,
		&srcfile,
		OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
		NULL,
		NULL);
	status = ZwCreateFile(&hsrcfile,
		GENERIC_READ,
		&ob,
		&IO,
		NULL,
		FILE_ATTRIBUTE_NORMAL,
		FILE_SHARE_READ,
		FILE_OPEN,
		FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT,
		NULL,
		0);
	if (!NT_SUCCESS(status))
	{
		if (hsrcfile)
			ZwClose(hsrcfile);
		if (hdfile)
			ZwClose(hdfile);
		if (buffer != NULL)
			ExFreePool(buffer);
		KdPrint(("获取文件句柄失败 清观察一下是否存在此文件!\n"));
		return FALSE;
	}
	InitializeObjectAttributes(&ob,
		&destfile,
		OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
		NULL,
		NULL);
	status = ZwCreateFile(&hdfile,
		GENERIC_WRITE,
		&ob,
		&IO,
		NULL,
		FILE_ATTRIBUTE_NORMAL,
		FILE_SHARE_READ,
		FILE_OPEN_IF,
		FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT,
		NULL,
		0);
	if (!NT_SUCCESS(status))
	{ 
		if (hsrcfile)
			ZwClose(hsrcfile);
		if (hdfile)
			ZwClose(hdfile);
		if (buffer != NULL)
			ExFreePool(buffer);
		KdPrint(("获取文件句柄失败 清观察一下是否存在此文件!\n"));
		return FALSE;
	}
	buffer = ExAllocatePool(NonPagedPool, 1024*4);
	if (buffer == NULL)
	{
		KdPrint(("申请内存失败！\n"));
		return FALSE;
	}
	while (1)
	{
		len = 4 * 1024;
		status = ZwReadFile(hsrcfile,
			NULL,
			NULL,
			NULL,
			&IO,
			buffer,
			len,
			&offset,
			NULL);
		if (!NT_SUCCESS(status))
		{
			if (status == STATUS_END_OF_FILE)
			{
				if (hsrcfile)
					ZwClose(hsrcfile);
				if (hdfile)
					ZwClose(hdfile);
				if (buffer != NULL)
					ExFreePool(buffer);
				KdPrint(("成功!\n"));
				return TRUE;
			}
		}
		len = (ULONG)IO.Information;
		status = ZwWriteFile(hdfile,
			NULL,
			NULL,
			NULL,
			&IO,
			buffer,
			len,
			&offset,
			NULL);
		if (!NT_SUCCESS(status))
		{
			if (hsrcfile)
				ZwClose(hsrcfile);
			if (hdfile)
				ZwClose(hdfile);
			if (buffer != NULL)
				ExFreePool(buffer);
			KdPrint(("写入失败!\n"));
			return FALSE;
		}
		offset.QuadPart += len;
	}
}
NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING RegistryPath)
{
	KdPrint(("first: Hello, my salary!"));
	UNICODE_STRING log, log1;
	RtlInitUnicodeString(&log, L"\\??\\C:\\2.log");
	RtlInitUnicodeString(&log1, L"\\??\\C:\\1.log");
	if (copyfile(log, log1))
	{
		KdPrint(("好嗨哟"));
	}
	else
	{
		KdPrint(("OMG"));
	}
	DriverObject->DriverUnload = DriverUnload;
	return STATUS_SUCCESS;
}

其实也不难一边写一边读