使用C语言。我有一个写入标准输出的函数。
我想捕获那个输出，稍微修改一下（替换一些字符串）。然后再次将其输出到标准输出。所以我想开始：
char huge_string_buf[MASSIVE_SIZE];
freopen("NUL", "a", stdout); -OR- freopen("/dev/null", "a", stdout);
setbuf(stdout, huge_string_buffer);
/* modify huge_string_buffer */
现在的问题是，如何将huge_string_buffer输出回原来的stdout？
【问题讨论】：
一个想法是模仿标准 Unix 实用程序 tee 的功能，但完全在您的程序中这样做，而不依赖于外部重定向。
所以我写了一个简单的函数mytee()，它似乎可以工作。它使用shmget(), pipe(), fork(), and dup2():
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/shm.h>
static char *mytee(int size) {
int shmid = shmget(IPC_PRIVATE, size + 1, 0660 | IPC_CREAT);
int pipe_fds[2];
pipe(pipe_fds);
switch (fork()) {
case -1: // = error
perror("fork");
exit(EXIT_FAILURE);
case 0: { // = child
char *out = shmat(shmid, 0, 0), c;
int i = 0;
out[0] = 0;
dup2(pipe_fds[0], 0); // redirect pipe to child's stdin
setvbuf(stdout, 0, _IONBF, 0);
while (read(0, &c, 1) == 1 && i < size) {
printf("<%c>", c); // pass parent's stdout to real stdout,
out[i++] = c; // and then buffer in mycapture buffer
out[i] = 0; // (the extra <> are just for clarity)
}
_exit(EXIT_SUCCESS);
}
default: // = parent
dup2(pipe_fds[1], 1); // replace stdout with output to child
setvbuf(stdout, 0, _IONBF, 0);
return shmat(shmid, 0, 0); // return the child's capture buffer
}
}
我的测试程序是：
int main(void) {
char *mycapture = mytee(100); // capture first 100 bytes
printf("Hello World"); // sample test string
sleep(1);
fprintf(stderr, "\nCaptured: <%s>\n", mycapture);
return 0;
}
输出是：
<H><e><l><l><o>< ><W><o><r><l><d> Captured: <Hello World>
要在您的应用程序中使用它，在mytee() 中，您需要将测试语句printf("<%c>", c) 替换为write(1, &c, 1)。您可能需要处理对read 的调用中的信号。在两个dup2() 之后，您可能需要添加：
close(pipe_fds[0]);
close(pipe_fds[1]);
有关此类内容的参考，请参阅 Dave Curry 撰写的 27 岁、220 页的优秀短篇 O'Reilly book 在 Unix 系统上使用 C。
【讨论】：
Unix 做到这一点的方法实际上就是编写一个小程序来处理您需要的输入，然后通过命令行将其他程序的输出通过管道传递给它。
如果你坚持把它全部保存在你的 C 程序中，我会改写那个函数，让它把它的输出发送到给定的字符缓冲区（最好是returning 缓冲区的char *），所以它可以发送到标准输出或根据客户的需要进行处理。
比如老办法：
void usage () {
printf ("usage: frob sourcefile [-options]\n");
}
...和新的方式：
char * usage(char * buffer) {
strcpy (buffer, "usage: frob sourcefile [-options]\n");
return buffer;
}
【讨论】：
我真的不喜欢带有文件描述符的棘手游戏。你不能修改函数，让它以其他方式返回数据，而不是写入标准输出？
如果您无权访问源代码，并且您不能这样做，那么我建议将写入标准输出的代码分解为一个小的单独程序，并将其作为另一个进程运行。从进程重定向输出（可能通过命名管道）既简单又干净，然后从接收数据的进程输出到标准输出就没有问题了。
此外，根据您希望进行的编辑类型，您最好使用 Python 等高级语言来编辑数据。
【讨论】：
char huge_string_buf[MASSIVE_SIZE];
FILE stdout_ori=fdopen(stdout,"a");
freopen("NUL", "a", stdout); -OR- freopen("/dev/null", "a", stdout);
setbuf(stdout, huge_string_buffer);
/* modify huge_string_buffer */
//Write to stdout_fdopen
【讨论】：