2

我正在尝试将我的本机 dll 通过 CoTaskMemAlloc 分配的一些数据编组到我的 c# 应用程序中，并想知道我这样做的方式是否完全错误，或者我错过了方法 c# 方面的一些微妙装饰。

目前我有 c++ 方面。

extern "C" __declspec(dllexport) bool __stdcall CompressData(  unsigned char* pInputData, unsigned int inSize, unsigned char*& pOutputBuffer, unsigned int& uOutputSize)
{ ...
    pOutputBuffer = static_cast<unsigned char*>(CoTaskMemAlloc(60000));
    uOutputSize = 60000;

在 C# 方面。

    private const string dllName = "TestDll.dll";

    [System.Security.SuppressUnmanagedCodeSecurity]
    [DllImport(dllName)]
    public static extern bool CompressData(byte[] inputData, uint inputSize, out byte[] outputData, out uint outputSize );
    ...
    byte[] outputData;
    uint outputSize;
    bool ret = CompressData(packEntry.uncompressedData, (uint)packEntry.uncompressedData.Length, out outputData, out outputSize);

这里 outputSize 是 60000 正如预期的那样，但 outputData 的大小为 1，当我对缓冲区 c++ 端进行 memset 时，它似乎只复制了 1 个字节，所以这是错误的，我需要使用IntPtr + outputSize，还是我缺少一些微妙的东西来开始我已经拥有的工作？

谢谢。

4

2 回答 2

4

有两件事。

首先，P/Invoke 层不处理 C++ 中的引用参数，它只能处理指针。最后两个参数 (pOutputBuffer和uOutputSize) 特别不能保证正确编组。

我建议您将 C++ 方法声明更改为（或创建表单的包装器）：

extern "C" __declspec(dllexport) bool __stdcall CompressData(  
    unsigned char* pInputData, unsigned int inSize, 
    unsigned char** pOutputBuffer, unsigned int* uOutputSize)

也就是说，第二个问题来自这样一个事实，即 P/Invoke 层也不知道如何封送SAFEARRAY在非托管代码中分配的“原始”数组（而不是说，COM 中知道它的大小） .

这意味着在 .NET 方面，您必须编组返回的指针，然后手动编组数组中的元素（以及处置它，如果这是您的责任，看起来确实如此）。

您的 .NET 声明如下所示：

[System.Security.SuppressUnmanagedCodeSecurity]
[DllImport(dllName)]
public static extern bool CompressData(byte[] inputData, uint inputSize, 
    ref IntPtr outputData, ref uint outputSize);

一旦你有了outputDataas IntPtr（这将指向非托管内存），你可以通过调用类上的Copy方法将其转换为字节数组，如下所示：Marshal

var bytes = new byte[(int) outputSize];

// Copy.
Marshal.Copy(outputData, bytes, 0, (int) outputSize);

请注意，如果您有责任释放内存，则可以调用FreeCoTaskMem方法，如下所示：

Marshal.FreeCoTaskMem(outputData);

当然，你可以把它包装成更好的东西，像这样：

static byte[] CompressData(byte[] input, int size)
{
    // The output buffer.
    IntPtr output = IntPtr.Zero;

    // Wrap in a try/finally, to make sure unmanaged array
    // is cleaned up.
    try
    {
        // Length.
        uint length = 0;

        // Make the call.
        CompressData(input, size, ref output, ref length);

        // Allocate the bytes.
        var bytes = new byte[(int) length)];

        // Copy.
        Marshal.Copy(output, bytes, 0, bytes.Length);

        // Return the byte array.
        return bytes;
    }
    finally
    {
        // If the pointer is not zero, free.
        if (output != IntPtr.Zero) Marshal.FreeCoTaskMem(output);
    }
}
于 2012-10-29T14:48:32.267 回答
1

pinvoke 编组器无法猜测返回的 byte[] 可能有多大。C++ 中指向内存的原始指针没有指向内存块的可发现大小。这就是您添加 uOutputSize 参数的原因。对客户端程序很好，但对 pinvoke 编组器来说还不够好。您必须帮助并将 [MarshalAs] 属性应用于 pOutputBuffer，指定SizeParamIndex属性。

请注意，编组器正在复制数组。这不是那么理想，您可以通过允许客户端代码传递数组来避免它。编组器将固定它并将指针传递给托管数组。唯一的麻烦是客户端代码没有像样的方法来猜测数组的大小。典型的解决方案是让客户端调用它两次，第一次用 uOutputSize = 0，函数返回所需的数组大小。这将使 C++ 函数看起来像这样：

extern "C" __declspec(dllexport) 
int __stdcall CompressData(
     const unsigned char* pInputData, unsigned int inSize, 
     [Out]unsigned char* pOutputBuffer, unsigned int uOutputSize)
于 2012-10-29T17:14:19.767 回答