Go语言通过Cgo调用C变长参数函数的策略与实践

#ifndef WRAPPER_H
#define WRAPPER_H

#include <curl/curl.h> // 假设已安装libcurl

// 声明一个包装函数，用于批量设置CURLoption
// curl_handle: CURL句柄
// options: 指向CURLoption数组的指针
// count: 数组中CURLoption的数量
CURLcode my_setopt_wrapper(CURL *curl_handle, CURLoption *options, int count);

#endif // WRAPPER_H

#include "wrapper.h"
#include <stdio.h> // For potential debugging output

// 实现包装函数
CURLcode my_setopt_wrapper(CURL *curl_handle, CURLoption *options, int count) {
    CURLcode res = CURLE_OK;
    for (int i = 0; i < count; ++i) {
        // 注意：这里简化了curl_easy_setopt的第三个参数。
        // 实际的curl_easy_setopt根据CURLoption不同，第三个参数类型也不同。
        // 对于真正的curl_easy_setopt调用，需要更复杂的包装逻辑来处理不同类型的参数。
        // 此处仅为演示如何通过包装函数传递数组。
        res = curl_easy_setopt(curl_handle, options[i], options[i]); // 假设第三个参数也是CURLoption
        if (res != CURLE_OK) {
            fprintf(stderr, "Error setting option %d: %s\n", options[i], curl_easy_strerror(res));
            break;
        }
    }
    return res;
}

package main

/*
#cgo pkg-config: libcurl
#include <stdlib.h> // For malloc and free
#include <curl/curl.h>
#include "wrapper.h" // 引入我们自己的C包装函数头文件

// 导入CURLcode和CURLoption类型
typedef CURLcode MyCURLcode;
typedef CURLoption MyCURLoption;
*/
import "C"
import (
    "fmt"
    "unsafe"
)

// 定义Go类型的CURLcode和CURLoption，用于公共API
type CURLcode C.MyCURLcode
type CURLoption C.MyCURLoption

// Easy 结构体用于管理CURL句柄和错误码
type Easy struct {
    curl unsafe.Pointer // C.CURL* 类型在Go中通常表示为unsafe.Pointer
    code CURLcode
}

// NewEasy 创建一个新的Easy实例
func NewEasy() *Easy {
    curl := C.curl_easy_init()
    if curl == nil {
        return nil
    }
    return &Easy{
        curl: unsafe.Pointer(curl),
        code: CURLcode(C.CURLE_OK),
    }
}

// Cleanup 释放CURL句柄
func (e *Easy) Cleanup() {
    if e.curl != nil {
        C.curl_easy_cleanup((*C.CURL)(e.curl))
        e.curl = nil
    }
}

// SetOptions 批量设置CURL选项
// 接收可变参数，每个参数都是CURLoption类型
func (e *Easy) SetOptions(options ...CURLoption) {
    if len(options) == 0 {
        e.code = CURLcode(C.CURLE_OK)
        return // 没有选项需要设置
    }

    // 1. 计算单个CURLoption在C中的大小
    // 使用C.sizeof_MyCURLoption 或 unsafe.Sizeof(C.MyCURLoption(0))
    // 这里假设C.MyCURLoption是一个整数类型，与CURLoption对齐
    // 实际项目中，如果C类型复杂，建议使用Cgo的C.sizeof_XXX宏
    size := int(unsafe.Sizeof(C.MyCURLoption(0))) // 获取C.MyCURLoption类型的大小

    // 2. 在C堆上分配内存，用于存储选项数组
    // C.malloc 返回的是C.void*，需要转换为unsafe.Pointer
    listPtr := C.malloc(C.size_t(size * len(options)))
    // 确保在函数返回前释放C堆上分配的内存，防止内存泄漏
    defer C.free(unsafe.Pointer(listPtr))

    // 3. 将Go切片中的选项复制到C堆上的数组中
    for i, opt := range options {
        // 计算当前选项在C数组中的内存地址
        // uintptr(listPtr) 将C指针转换为Go的uintptr，便于进行指针算术
        // uintptr(size * i) 计算偏移量
        // 然后再转换回unsafe.Pointer，最后转换为C.MyCURLoption的指针
        ptr := unsafe.Pointer(uintptr(listPtr) + uintptr(size*i))
        // 将Go的CURLoption值复制到C内存地址指向的位置
        *(*C.MyCURLoption)(ptr) = C.MyCURLoption(opt)
    }

    // 4. 调用C包装函数
    // 将CURL句柄、C数组指针和数组长度传递给C包装函数
    e.code = CURLcode(C.my_setopt_wrapper(
        (*C.CURL)(e.curl), // 将unsafe.Pointer转换回C.CURL*
        (*C.MyCURLoption)(listPtr), // 将C数组指针转换为C.MyCURLoption*
        C.int(len(options)),        // 传递数组长度
    ))
}

// GetCode 获取最后一次操作的错误码
func (e *Easy) GetCode() CURLcode {
    return e.code
}

func main() {
    easy := NewEasy()
    if easy == nil {
        fmt.Println("Failed to initialize CURL easy handle.")
        return
    }
    defer easy.Cleanup()

    // 示例：设置多个选项
    // 注意：这里的CURLoption值是示例，实际CURLoption常量来自libcurl
    // 并且如前所述，curl_easy_setopt的第三个参数类型是可变的。
    // 这个例子仅演示传递CURLoption枚举值数组。
    fmt.Println("Attempting to set options...")
    easy.SetOptions(
        CURLoption(C.CURLOPT_VERBOSE), // 启用详细输出
        CURLoption(C.CURLOPT_NOPROGRESS), // 禁用进度条
        // CURLoption(C.CURLOPT_URL), // 无法直接传递URL字符串，需要更复杂的包装
    )

    if easy.GetCode() != CURLcode(C.CURLE_OK) {
        fmt.Printf("Error setting options: %s\n", C.GoString(C.curl_easy_strerror(C.CURLcode(easy.GetCode()))))
    } else {
        fmt.Println("Options set successfully (conceptually).")
    }

    // 实际执行（需要设置URL等）
    // C.curl_easy_setopt((*C.CURL)(easy.curl), C.CURLOPT_URL, C.CString("http://example.com"))
    // C.curl_easy_perform((*C.CURL)(easy.curl))

    fmt.Println("Program finished.")
}