当前位置：首页 > 编程日记 > 正文

一种注册表沙箱的思路、实现——研究Reactos中注册表函数的实现2

编程日记 2024-12-12 11:50:00

上一篇博文中主要介绍了Reactos中大部分函数的思路和HKEY和HANDLE之间的关系，本文将介绍一些Reactos中有意思的函数和存在bug的函数。（转载请指明出处）

CreateNestedKey是一个辅助创建键的函数，比如我们要创建\Regsitry\User\3\2\1，而我们只是存在\Regsitry\User\，我们最终将调用此函数将在User下创建键3，然后在3键下创建键2，最后在2键下创建键1。假如你是这个函数的实现者，你要思考这个函数的实现思路。最容易的办法是将\Regsitry\User\3\2\1路径拆成一系列键名（\Regsitry\User、3、2、1），然后从根键一步一步的调用NtCreateKey，这样就可以保证路径上的所有键都被创建。这个方法的优点是简单。但是Reactos实现的思路却不是这样的。我列一下我改写的该函数。

NTSTATUS OriCreateNestedKey(PHANDLE pKeyHandle,POBJECT_ATTRIBUTES ObjectAttributes,PUNICODE_STRING ClassString,DWORD dwOptions,REGSAM samDesired,DWORD *lpdwDisposition )
{UNICODE_STRING LocalKeyName;NTSTATUS lRes = STATUS_INVALID_PARAMETER;do {if ( NULL == ObjectAttributes || NULL == ObjectAttributes->ObjectName ) {break;}lRes = RtlCreateUnicodeString_( &LocalKeyName, ObjectAttributes->ObjectName->Buffer );CHECKRESULT(lRes);lRes = GETORIFUNC(CreateKey)( pKeyHandle, samDesired,ObjectAttributes, 0, ClassString, dwOptions, (PULONG)lpdwDisposition );if ( STATUS_OBJECT_NAME_NOT_FOUND != lRes ) {return lRes;}OBJECT_ATTRIBUTES LocalObjectAttributes;RtlCopyMemory( &LocalObjectAttributes, ObjectAttributes, sizeof(OBJECT_ATTRIBUTES));LocalObjectAttributes.ObjectName = &LocalKeyName;ULONG FullNameLength = LocalKeyName.Length / sizeof(WCHAR);HANDLE LocalKeyHandle = NULL;PWCHAR Ptr = NULL;ULONG Disposition = 0;// 通过将\\改成结尾符，逐个去掉最后一个键名，查看什么路径的注册表键存在while ( STATUS_OBJECT_NAME_NOT_FOUND == lRes ) {Ptr = wcsrchr(LocalKeyName.Buffer, '\\');if ( NULL == Ptr || Ptr == LocalKeyName.Buffer ) {lRes = STATUS_UNSUCCESSFUL;break;}// 通过将\\改成结尾符，通过长度去掉键名*Ptr = (WCHAR)0;LocalKeyName.Length = wcslen(LocalKeyName.Buffer) * sizeof(WCHAR);lRes = GETORIFUNC(CreateKey)( &LocalKeyHandle, KEY_CREATE_SUB_KEY,&LocalObjectAttributes, 0, NULL, 0, &Disposition );}CHECKRESULT(lRes);// 通过将结尾符改成\\，重新拼接上键名，逐个创建键ULONG Length = wcslen(LocalKeyName.Buffer);while ( TRUE ) {if ( NULL != LocalKeyHandle ) {GETORIFUNC(Close) (LocalKeyHandle);}// 通过将结尾符改成\\，重新拼接上键名LocalKeyName.Buffer[Length] = L'\\';Length = wcslen (LocalKeyName.Buffer);LocalKeyName.Length = Length * sizeof(WCHAR);if ( Length == FullNameLength ) {// 如果已经拼接回以前的键名了，则执行完后就退出lRes = GETORIFUNC(CreateKey)( pKeyHandle, samDesired,ObjectAttributes, 0, ClassString, dwOptions, (PULONG)lpdwDisposition);break;}// 创建祖宗键lRes = GETORIFUNC(CreateKey)( &LocalKeyHandle, KEY_CREATE_SUB_KEY,&LocalObjectAttributes, 0, NULL, 0, &Disposition);if ( NT_FAILED(lRes) ) {break;}}} while (0);RtlFreeUnicodeString_(&LocalKeyName);return lRes;
}

大致说下这个函数的流程：

使用NtCreateKey函数调用传入的参数。如果不是返回STATUS_OBJECT_NAME_NOT_FOUND，则说明要创建的路径上的键的父键是存在的，直接返回创建的结果；否则说明父键就不存在，得依赖之后的步骤将父键创建起来。
声明个变量OBJECT_ATTRIBUTES LocalObjectAttributes;将作为参数传入的ObjectAttributes全部拷贝到LocalObjectAttributes，同时记录下路径的长度FullNameLength。
将LocalObjectAttributes.LocalKeyName中的记录路径的Buffer字段，从后向前寻找“\”符号，找到一个，就将其改为空，同时再计算LocalObjectAttributes中LocalKeyName的长度。得到修改后的LocalObjectAttributes后，再调用NtCreateKey。以上面的例子为例，就是用NtCreateKey创建\Regsitry\User\3\2。查看返回结果，如果结果还是STATUS_OBJECT_NAME_NOT_FOUND，说明父键还是不存在的，于是要重复3的步骤，一直到NtCreateKey执行成功。如例子中，一直到用NtCreateKey创建\Regsitry\User\3就停止这个循环，因为3键可以创建成功。
比较LocalObjectAttributes.LocalKeyName的长度和FullNameLength，如果相等，则说明我们应该创建的键都创建完了。如果不等，则我们将LocalObjectAttributes.LocalKeyName的最前一个空改成“\”，得到修改后的LocalObjectAttributes，再调用NtCreateKey。在我们的例子中，就是创建\Regsitry\User\3\2。一直重复4，一直到LocalObjectAttributes.LocalKeyName的长度和FullNameLength相等。

我很欣赏这样的思路，通过改改内存达到目的，效率方面会好很多。

https://www.dkcj.cn/info/31497.html

一种注册表沙箱的思路、实现——研究Reactos中注册表函数的实现2

相关文章：

云计算安全解决方案白皮书(一)

一种注册表沙箱的思路、实现——研究Reactos中注册表函数的实现3

腾讯Angel升级：加入图算法，支持十亿节点、千亿边规模！中国首个毕业于Linux AI基金会的开源项目...

如何在JSP页面中获取当前系统时间转

如何用Python实现超级玛丽的界面和状态机？

一种注册表沙箱的思路、实现——研究Reactos中注册表函数的实现4

Netscaler 认证，访问报http 5000 内部错误

解读 | 2019年10篇计算机视觉精选论文（中）

PE文件和COFF文件格式分析--概述

2015第22周六Java反射、泛型、容器简介

中服公司企业信息化的ERP系统选择

PE文件和COFF文件格式分析--MS-DOS 2.0兼容Exe文件段

时间可以是二维的？基于二维时间图的视频内容片段检测 | AAAI 2020

《燃烧的岁月》

PE文件和COFF文件格式分析——签名、COFF文件头和可选文件头1

遇到bug心寒了？用Enter键即可解决！

8月第1周安全回顾 0Day漏洞成企业最大威胁应重视网络监听

最大匹配、最小顶点覆盖、最大独立集、最小路径覆盖（转）

一种在注入进程中使用WTL创建无焦点不在任务栏出现“吸附”窗口的方法和思路

转:ASP.NET状态保存方法

时至今日，NLP怎么还这么难！

Quartz定时任务学习（四）调度器

反汇编算法介绍和应用——线性扫描算法分析

项目计划书的内容

不用写代码就能学用Pandas，适合新老程序员的神器Bamboolib

后海日记（8）

反汇编算法介绍和应用——递归下降算法分析

如何快速get到AI工程师面试重点，这12道题必备！

金邦黑金刚4G内存 VS Vista系统

程序员的量化交易之路（25）--Cointrader之MarketData市场数据实体（12）