如何在 C 中将字符串添加到 PUNICODE_STRING 的末尾

Question

我正在尝试修改现有的 C 迷你过滤器驱动程序，但对 C 的经验非常有限。我的函数接收到文件系统上文件的两个 PUNICODE_STRING 的两个路径。

 _In_  PUNICODE_STRING       SourceFile,
 _In_  PUNICODE_STRING       TargetFile,

我需要创建一个新的 PUNICODE_STRING（称为 NewTargetFile），然后在 NewTargetFile 的末尾附加“.xy”，而不是覆盖 TargetFile。

我认为一个明智的方法是定义一个新的 PUNICODE_STRING 并将 TargetFile 复制到 NewTargetFile。然后将字符串附加到末尾。但我显然在这里超出了我的深度！因为我只是得到一个空白值。

PUNICODE_STRING NewTargetFile = { 0 };

NewTargetFile = TargetFile;
NT_IF_FAIL_LEAVE(RtlAppendUnicodeToString(NewTargetFile, (PCWSTR)'.ta'));

因此，简而言之，如果我的函数收到“filename.txt”的 TargetFile，我需要生成一个 PUNICODE_STRING，其值为“filename.txt.xy”

Answer 1

通过分配NewTargetFile = TargetFile;，您只是在复制指针。在NewTargetFile 上执行的每个操作都将在TargetFile 上执行。

您需要为NewTargetFile创建一个单独的UNICODE_STRING并使用RtlUnicodeStringInit()函数对其进行初始化，然后您需要使用RtlUnicodeStringCopy()函数将TargetFile复制到NewTargetFile中。最后，您可以使用RtlAppendUnicodeToString()。

类似：

PUNICODE_STRING NewTargetFile = { 0 };
NTSTRSAFEDDI status; 
// NewTargetFile = TargetFile;
status = RtlUnicodeStringInit(
    NewTargetFile,
    (NTSTRSAFE_PCWSTR) "\0"
);
// check status ...
status = RtlUnicodeStringCopy(
    NewTargetFile,
    TargetFile
);
// check status ...
NT_IF_FAIL_LEAVE(RtlAppendUnicodeToString(NewTargetFile, (PCWSTR)'.ta'));

我没有尝试代码，但这似乎是问题所在。

希望对您有所帮助。