更改 IntPtr 指向的字符串

Question

在我的 C# 应用程序中，我有一个 IntPtr 类型的变量 lpData（从对非托管代码的调用中接收），它指向一个字符串。

我必须用另一个值替换这个字符串。

我试过了：

int RegQueryValueExW_Hooked(
        IntPtr hKey,
        string lpValueName,
        int lpReserved,
        ref Microsoft.Win32.RegistryValueKind lpType,
        IntPtr lpData,
        ref int lpcbData)
{ 
    lpData = Marshal.StringToHGlobalUni("new string"); 
    ...
}

但这似乎并不能取代实际的字符串。

有人可以为我指出正确的方向吗？

谢谢

Answer 1

当然，它不会替换字符串 - 您将获得指向调用者具有值的字符串的指针。您只是替换“局部变量”（参数）的值，这不会改变调用方的任何内容。

如果您想修改原始指针上的值（并确保您确实需要，这就是“奇怪的错误”潜伏的地方 - 很容易覆盖周围的变量，忘记空终止符等） ，可以使用Marshal.Copy，例如：

var bytes = Encoding.Unicode.GetBytes("your string\0");

Marshal.Copy(bytes, 0, lpData, bytes.Length);

再说一遍 - 这是一种非常危险的行为，您不应该这样做。您违反了参数传递等隐含的几项合同。

现在我已经回答了你的问题，让我谈谈你实际上需要这样做有多么错误（这与你关于使用 StringBuilder 的另一篇文章非常相关）。

您正在尝试修改作为参数传递给您的值。但是，该字符串是由调用者分配的。你甚至不知道它有多长！如果您开始将数据复制到该指针，您将覆盖例如的数据。完全不同的变量，只是随机恰好在字符串之后分配。这被认为是“非常糟糕”。

相反，您想要做的是遵循 RegQueryValueEx 拥有的正确流程 (http://msdn.microsoft.com/en-us/library/windows/desktop/ms724911(v=vs.85).aspx)。这意味着您首先必须检查 lpcbData 值。如果它大到足以容纳您要写入的所有字节，您只需将数据写入 lpData 并将 lpcbData 值设置为适当的长度。如果没有，您仍然设置 lpcbData，但返回 ERROR_MORE_DATA。然后调用者应再次调用 RegQueryValueEx，并使用更大的缓冲区。

示例代码如下所示：

string yourString = "Your string";

int RegQueryValueExW_Hooked(
        IntPtr hKey,
        string lpValueName,
        int lpReserved,
        ref Microsoft.Win32.RegistryValueKind lpType,
        IntPtr lpData,
        ref int lpcbData)
{ 
    var byteCount = Encoding.Unicode.GetByteCount(yourString);

    if (byteCount > lpcbData)
    {
        lpcbData = byteCount;

        return ERROR_MORE_DATA;
    }

    if (lpData == IntPtr.Zero)
    {
        return ERROR_SUCCESS;
    }

    lpcbData = byteCount;

    var bytes = Encoding.Unicode.GetBytes(yourString);   
    Marshal.Copy(bytes, 0, lpData, bytes.Length);

    return ERROR_SUCCESS;
}

请注意，这只是我在快速浏览文档后所写的内容 - 您应该进一步调查，并确保您处理了所有可能的情况。在这种情况下，.NET 无法保护您，您可能造成重大问题！