如何捕获共享C库的STDOUT而非仅在程序退出前输出？

你当前的问题核心是C库的stdout缓冲区未被及时刷新，导致恢复Python的stdout后，C库的输出才在程序退出时被打印，晚于"After!"消息。以下是具体解决思路和修改后的代码：

问题原因 #

C标准库的stdout在重定向到文件（包括你用dup2指向的日志文件）时，默认采用全缓冲模式——只有缓冲区满或程序退出时才会将内容写入文件描述符。而你之前尝试的unbuffer、fsync或Python层的flush()都无法触及C库内部的缓冲区：

• unbuffer是针对整个进程的输出缓冲控制，不适用Python进程内部的文件描述符重定向场景；
• fsync仅刷新内核级的文件缓冲区，无法强制C库将用户空间的缓冲内容写入内核；
• Python的sys.stdout.flush()只处理Python自身的输出缓冲，对C库的stdout缓冲区无效。

解决方法 #

在恢复stdout之前，通过ctypes调用C标准库的fflush()函数，强制刷新所有C层的输出缓冲区（包括stdout）。这样C库的输出会被立即写入日志文件，不会延迟到程序退出。

修改后的代码 #

import sys
import os
import ctypes

fd = sys.stdout.fileno()
old_stdout = os.fdopen(os.dup(fd), 'w')    
with open("_capture_stdout.log", 'w') as f:
    sys.stdout.close()
    os.dup2(f.fileno(), fd)
    sys.stdout = os.fdopen(fd, 'w')
    yield                                                                           
    # 强制刷新C库的所有输出缓冲区
    ctypes.CDLL(None).fflush(None)
    # 恢复原stdout
    os.dup2(old_stdout.fileno(), fd)
    sys.stdout = os.fdopen(fd, 'w')
    sys.stdout.write("After!\n")
    sys.stdout.flush()  # 确保Python的输出立即打印
    sys.exit()

额外说明 #

• ctypes.CDLL(None)会加载当前进程中的所有共享库（包括C标准库），因此可以直接调用fflush()；
• fflush(None)会刷新所有打开的输出流，如果你只想针对stdout，可以传入ctypes.c_void_p(sys.stdout.fileno())，但fflush(None)更简单可靠；
• 恢复Python的stdout后，建议调用sys.stdout.flush()确保"After!"消息立即输出，避免Python自身的缓冲延迟。

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