迭代用Marshal.AllocHGlobal()分配的内存

时间:2012-04-02 18:18:37

标签: c# pinvoke

我有一个第三方C库,其导出方法之一如下:

#define MAX_INDEX 8
int GetStuff(IN char* index[MAX_INDEX], OUT char* buf, IN size_t size);

第一个参数填充了指向存储特定字符串的buf参数的指针。大小表示每个字符串在buf缓冲区中的预期长度。我的C#P / Invoke方法目前如下所示:

[DllImport("Path/To/Dll", CharSet = CharSet.Ansi)]
private static extern int GetStuff(IntPtr indecies, IntPtr buf, Int32 size);

C#P / Invoke方法是私有的,因为我将它的功能包装在一个公共的“getter”方法中,该方法处理调用者的内存分配/释放。当我在C ++中使用这个方法时,迭代实际上非常简单。我只是做了类似的事情:

char* pIndecies[MAX_INDEX];
char* pBuffer = new char[MAX_INDEX * (256 + 1)]; // +1 for terminating NULL

GetStuff(pIndecies, pBuffer, 256);

// iterate over the items
for(int i(0); i < MAX_INDEX; i++) {
    if(pIndecies[i]) {
        std::cout << "String for index: " << i << " " << pIndecies[i] << std::endl;
    }
}

由于在C ++中如何使用这些,我决定我应该使用IntPtr对象,只需从堆中分配我需要的内存,调用本机代码,并按照我在C ++中的方式迭代它。然后我记得C#中的字符是unicode字符而不是ASCII字符。即使我将迭代放在一个不安全的代码块中,C#中的迭代也会一样吗?我的第一个想法是做以下事情:

IntPtr pIndecies = Marshal.AllocHGlobal(MAX_INDEX * 4); // the size of a 32-pointer
IntPtr pBuffer = Marshal.AllocHGlobal(MAX_INDEX * (256 + 1)); // should be the same
NativeMethods.GetStuff(pIndecies, pBuffer, 256);

unsafe {
    char* pCStrings = (char*)pIndecies.ToPointer();
    for(int i = 0; i < MAX_INDEX; i++) {
        if(pCStrings[i])
            string s = pCStrings[i];
    }
}

我的问题是,“我应该如何重复这个本机代码方法返回的内容?”这是编组此功能的正确方法吗?我应该为第二个参数使用StringBuilder对象吗?一个约束问题是第一个参数是GetStuff()方法后面的结构的1:1映射。每个索引的值对于理解您在第二个缓冲区参数中看到的内容至关重要。

我感谢任何建议。

谢谢, 安迪

1 个答案:

答案 0 :(得分:3)

我认为你走的正确,但我不使用不安全的代码就行。像这样:

[DllImport("Path/To/Dll", CharSet = CharSet.Ansi)]
private static extern int GetStuff(IntPtr[] index, IntPtr buf, Int32 size);
....
IntPtr[] index = new IntPtr[MAX_INDEX];
IntPtr pBuffer = Marshal.AllocHGlobal(MAX_INDEX * 256 + 1);
try
{
    int res = NativeMethods.GetStuff(index, pBuffer, 256);
    // check res for errors?
    foreach (IntPtr item in index)
    {
        if (item != IntPtr.Zero)
            string s = Marshal.PtrToStrAnsi(item);
    }
} 
finally
{
    Marshal.FreeHGlobal(pBuffer);
}

这是您的C ++版本的直接翻译。