printf重定向

用到第三方庫使用printf打印調試信息的時候往往需要重定向日志到文件, 但如果沒有源碼或修改點較多時就比較麻煩. 這里提供兩個重定位方法:
1. 修改值為1的文件描述符
默認printf打印指向標准輸出stdout(fd=1), 最終指向終端. 因此可以關閉值為1的文件描述符再打開另一文件, 之后printf打印就被寫入該文件中. 注意這種方法必須保證關閉fd與打開文件之間不會有其它文件操作(內核分配fd是順序分配的, 如有第三個文件打開則第三個文件的fd被設為1), 且fork進程會保留對應的fd, 即新進程也會操作同一文件, 需要做好互斥(可以通過關閉子進程fd來解決這個問題).
2. 修改printf實現
定義同名同類型的printf並修改其實現. 這里要注意一點, 默認打印不帶參數的字符串時gcc會將printf優化為puts, 解決辦法有兩個: 一是同樣重定義puts, 二是編譯時添加-fno-builtin-printf.

#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <stdarg.h>
#define REDIRECT
#undef REDIRECT
#ifdef REDIRECT
int fd = -1;
#else
FILE *fp = NULL;
int printf(const char *fmt, ...)
{
    va_list args;
    int ret = 0;
    va_start(args, fmt);
    ret = vfprintf(fp, fmt, args);
    va_end(args);
    fflush(fp);
    return ret;
}
/*
int puts(const char *s)
{
    //add \n here to avoid buffering
    return printf("%s\n", s);
}
*/
#endif
int main()
{
#ifdef REDIRECT
    close(1);
    
    fd = open("./out", O_RDWR | O_CREAT, 0755);
    if (1 != fd)
    {
        fprintf(stderr, "open file fail with %d\n", fd);
        return -1;
    }
    printf("this line shall written into file!\n");
    //flush cache, man stdout for detail
    fsync(fd);
#else
    fp = fopen("./out", "w+");
    if (NULL == fp)
    {
        fprintf(stderr, "open file fail\n");
        return -1;
    }
    const char cmd[] = "this line shall written into file!\n";
    printf("this line shall written into file!\n");
    printf("this line shall written into file! more args %p\n", fp);
#endif
    return 0;
}