内核中获取操作系统安装盘路径

NTSTATUS GetSystemRootPath(PUNICODE_STRING pusSystemRoot)
{
	NTSTATUS status = STATUS_UNSUCCESSFUL;
	PFILE_OBJECT pFileObject = NULL;
	POBJECT_NAME_INFORMATION pObjectNameInformation = NULL;
	HANDLE hFile = NULL;
	IO_STATUS_BLOCK ioStausBlock = { 0 };
	OBJECT_ATTRIBUTES oa = { 0 };
	UNICODE_STRING usFileName = { 0 };

	do
	{
		if (!pusSystemRoot)
		{
			status = STATUS_INVALID_PARAMETER;
			break;
		}

		// Get SystemRoot via ZwOpenFile
		RtlInitUnicodeString(&usFileName, L"\\SystemRoot");
		InitializeObjectAttributes(&oa, &usFileName, OBJ_KERNEL_HANDLE, NULL, NULL);
		status = ZwOpenFile(&hFile, SYNCHRONIZE, &oa, &ioStausBlock, FILE_SHARE_READ, FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT);
		if (!NT_SUCCESS(status))
		{
			KdPrint(("ZwOpenFile failed! status = 0x%X\n", status));
			break;
		}

		status = ObReferenceObjectByHandle(hFile, FILE_ALL_ACCESS, *IoFileObjectType, KernelMode, &pFileObject, NULL);
		if (!NT_SUCCESS(status))
		{
			KdPrint(("***(%s line: %d***)\n\t%s failed with code %#x\n\n", __FILE__, __LINE__, "ObReferenceObjectByHandle", status));
			break;
		}

		status = IoQueryFileDosDeviceName(pFileObject, &pObjectNameInformation);
		if (!NT_SUCCESS(status))
		{
			KdPrint(("***(%s line: %d***)\n\t%s failed with code %#x\n\n", __FILE__, __LINE__, "IoQueryFileDosDeviceName", status));
			break;
		}

		if (pObjectNameInformation->Name.Length > pusSystemRoot->Length)
		{
			KdPrint(("***(%s line: %d***)\n\t%s failed with code %#x\n\n", __FILE__, __LINE__, "pwsSystemRoot Buffer is too small", status));
			status = STATUS_BUFFER_TOO_SMALL;
			break;
		}

		KdPrint(("SystemRoot Path:  %wZ\n", &pObjectNameInformation->Name));
		RtlCopyUnicodeString(pusSystemRoot, &pObjectNameInformation->Name);
		

	} while (FALSE);

	if (hFile)
	{
		ZwClose(hFile);
	}

	if (pFileObject)
	{
		ObDereferenceObject(pFileObject);
	}

	if (pObjectNameInformation)
	{
		ExFreePool(pObjectNameInformation);
	}

	return status;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值