从Go调用C/C++ DLL中返回字符串的函数-Golang-PHP中文网

从go调用c/c++ dll中返回字符串的函数

在Go语言中调用C/C++编写的动态链接库（DLL）是很常见的需求。当DLL中的函数返回整数类型时，可以直接通过syscall.Call获取返回值。然而，当函数返回字符串类型（char*）时，syscall.Call返回的是一个uintptr，它代表了指向字符串的指针地址。我们需要将其转换为Go语言中的字符串类型才能正确使用。

下面将详细介绍如何使用unsafe包进行转换：

1. 了解uintptr和unsafe.Pointer

uintptr：Go语言中的一种无符号整数类型，可以存储指针地址。
unsafe.Pointer：一种特殊的指针类型，可以指向任何类型的数据。它允许在不同类型之间进行指针转换，但需要谨慎使用，因为它绕过了Go语言的类型安全检查。

2. 使用unsafe包进行转换

假设我们有以下C/C++ DLL函数：

度加AI

度加AI官网入口，百度官方 AIGC 创作平台，支持 AI 成片、AI 生文、数字人、声音克隆、配音字幕与智能剪辑等在线创作能力。

下载

// C++ DLL
#ifdef _WIN32
#define WIN32_DLL_EXPORT __declspec(dllexport)
#else
#define WIN32_DLL_EXPORT
#endif

extern "C" {
    WIN32_DLL_EXPORT const char* FnRetString() {
        return "THIS IS A TEST STRING";
    }
}

在Go语言中，我们可以这样调用该函数并处理返回值：

package main

import (
    "fmt"
    "syscall"
    "unsafe"
)

func main() {
    dllPath := "your_dll_path.dll" // 替换为你的DLL路径
    dllFuncName := "FnRetString"

    hd, err := syscall.LoadLibrary(dllPath)
    if err != nil {
        fmt.Println("LoadLibrary error:", err)
        return
    }
    defer syscall.FreeLibrary(hd)

    proc, err := syscall.GetProcAddress(hd, dllFuncName)
    if err != nil {
        fmt.Println("GetProcAddress error:", err)
        return
    }

    ret, _, _ := syscall.SyscallN(proc, 0) // 调用函数，返回 uintptr

    // 将 uintptr 转换为 *uint8 (C中的 char*)
    charPtr := (*uint8)(unsafe.Pointer(ret))

    // 将 *uint8 转换为 Go 字符串
    str := cStringToGoString(charPtr)

    fmt.Println("Returned string:", str)
}

// cStringToGoString converts a C string (char*) to a Go string.
func cStringToGoString(cStr *uint8) string {
    if cStr == nil {
        return ""
    }

    var buf []byte
    for {
        b := *cStr
        if b == 0 {
            break
        }
        buf = append(buf, b)
        cStr = (*uint8)(unsafe.Pointer(uintptr(unsafe.Pointer(cStr)) + 1))
    }
    return string(buf)
}

代码解释:

加载DLL和函数: 首先，使用syscall.LoadLibrary加载DLL，然后使用syscall.GetProcAddress获取函数地址。
调用函数: 使用syscall.SyscallN调用DLL中的函数，得到返回的uintptr。
*转换为`uint8:** 使用unsafe.Pointer(ret)将uintptr转换为unsafe.Pointer，再将unsafe.Pointer转换为uint8，即C语言中的char`。
转换为Go字符串: 编写cStringToGoString函数，该函数接收一个*uint8类型的参数，循环读取直到遇到空字符\0，然后将读取到的字节转换为Go字符串。该函数确保了正确处理C字符串，避免因编码问题导致乱码。

注意事项:

内存管理: 如果DLL返回的字符串是在DLL内部动态分配的，你需要确保在Go程序中释放这块内存。否则，可能会导致内存泄漏。通常DLL会提供一个释放内存的函数，你需要调用该函数来释放内存。如果没有提供，则需要仔细分析DLL的内存管理机制，并采取相应的措施。
编码问题: C/C++和Go语言可能使用不同的字符编码。如果DLL返回的字符串不是UTF-8编码，你需要将其转换为UTF-8编码才能在Go程序中正确显示。可以使用golang.org/x/text/encoding包来进行编码转换。
unsafe包的使用: unsafe包绕过了Go语言的类型安全检查，因此需要谨慎使用。确保你理解指针转换的含义，避免出现错误。

总结:

通过使用unsafe包，我们可以将syscall.Call返回的uintptr转换为Go语言中的字符串类型，从而实现从Go语言调用C/C++ DLL并获取字符串返回值的功能。但是，需要注意内存管理和编码问题，并谨慎使用unsafe包，确保程序的正确性和安全性。在实际开发中，建议尽量避免直接使用unsafe包，而是选择更安全、更高级的解决方案，例如使用cgo。

C++免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 C++ 的入门与实战技巧！