如何在Go中捕获C函数调用输出的stderr内容？

问题场景 #

现有Go程序调用无法修改的第三方C库函数，该函数直接通过fprintf(stderr, ...)输出内容。当前尝试替换os.Stderr的方式无法捕获C函数的输出，需要实现：

• 将C函数输出到stderr的内容捕获到内存缓冲区
• 调用完成后恢复os.Stderr的正常输出功能
• 支持Windows及类UNIX平台

问题原因 #

Go的os.Stderr是对系统文件描述符的封装，但C标准库的stderr在程序启动时就已绑定到对应的文件描述符（类UNIX为2，Windows为STD_ERROR_HANDLE）。替换os.Stderr不会改变C库已打开的stderr指向，因此C函数仍会往原始的标准错误流输出。

解决方案 #

需要在文件描述符/句柄级别替换标准错误流，保存原始句柄，调用C函数后再恢复。以下是跨平台实现代码：

package main

// #include <stdio.h>
// #include <stdlib.h>
//
// // 无法修改的第三方C函数
// static void library_call(char* s) {
//   fprintf(stderr, "%s\n", s);
// }
import "C"

import (
	"bytes"
	"fmt"
	"io"
	"os"
	"sync"
	"unsafe"
	"syscall"
)

// 保存原始stderr句柄/文件描述符
var originalStderr interface{}

// 类UNIX平台初始化：复制原始stderr文件描述符
//go:build !windows
func init() {
	newFD, err := C.dup(C.int(2))
	if err != nil {
		panic(err)
	}
	originalStderr = int(newFD)
}

// Windows平台初始化：复制原始标准错误句柄
//go:build windows
func init() {
	stdErrHandle, err := syscall.GetStdHandle(syscall.STD_ERROR_HANDLE)
	if err != nil {
		panic(err)
	}
	var originalHandle syscall.Handle
	err = syscall.DuplicateHandle(syscall.CurrentProcess(), stdErrHandle, syscall.CurrentProcess(), &originalHandle, 0, false, syscall.DUPLICATE_SAME_ACCESS)
	if err != nil {
		panic(err)
	}
	originalStderr = originalHandle
}

// 类UNIX平台：重定向stderr到指定文件
//go:build !windows
func redirectStderr(w *os.File) error {
	_, err := C.dup2(C.int(w.Fd()), C.int(2))
	return err
}

// Windows平台：重定向stderr到指定文件
//go:build windows
func redirectStderr(w *os.File) error {
	handle, err := syscall.Open(syscall.StringToUTF16Ptr(w.Name()), syscall.O_WRONLY, 0)
	if err != nil {
		return err
	}
	defer syscall.Close(handle)
	return syscall.SetStdHandle(syscall.STD_ERROR_HANDLE, handle)
}

// 类UNIX平台：恢复原始stderr
//go:build !windows
func restoreStderr() error {
	_, err := C.dup2(C.int(originalStderr.(int)), C.int(2))
	return err
}

// Windows平台：恢复原始stderr
//go:build windows
func restoreStderr() error {
	return syscall.SetStdHandle(syscall.STD_ERROR_HANDLE, originalStderr.(syscall.Handle))
}

func main() {
	var buf bytes.Buffer
	var wg sync.WaitGroup

	// 创建管道
	r, w, err := os.Pipe()
	if err != nil {
		panic(err)
	}
	defer r.Close()
	defer w.Close()

	// 启动goroutine异步读取管道内容到缓冲区
	wg.Add(1)
	go func() {
		defer wg.Done()
		io.Copy(&buf, r)
	}()

	// 重定向stderr到管道写端
	if err := redirectStderr(w); err != nil {
		panic(err)
	}

	// 调用目标C函数
	cs := C.CString("This should be the contents of the buffer")
	C.library_call(cs)
	C.free(unsafe.Pointer(cs))

	// 恢复原始stderr
	if err := restoreStderr(); err != nil {
		panic(err)
	}

	// 关闭管道写端，触发goroutine读取完成
	w.Close()
	wg.Wait()

	// 输出捕获的内容
	fmt.Printf("got output: %s\n", buf.String())

	// 验证stderr恢复正常
	fmt.Fprintln(os.Stderr, "stderr has been restored successfully")
}

关键逻辑说明 #

1. 保存原始句柄：程序启动时复制原始stderr的文件描述符/句柄，为后续恢复做准备。
2. 管道重定向：创建匿名管道，将stderr的底层句柄重定向到管道写端，使C函数输出流入管道。
3. 异步读取：启动goroutine读取管道读端内容，避免因管道缓冲区满导致C函数阻塞。
4. 恢复流：调用完C函数后立即恢复原始stderr句柄，保证后续程序的标准错误输出正常。
5. 跨平台兼容：通过Go的build标签区分平台，分别实现对应系统的句柄操作逻辑。

编译运行 #

类UNIX平台：

go build test.go && ./test

Windows平台：

go build test.go && test.exe

内容的提问来源于stack exchange，提问作者llimllib