高级I/O函数(3)-tee、fcntl函数

tee函数使用

功能描述:tee函数在两个管道文件描述符之间复制数据,也是零拷贝操作.它不消耗数据,因此源文件描述符仍然可以用于后续的操作.

函数原型:

#include <fcntl.h>
 ssize_t tee(int fd_in,int fd_out,size_t len,unsigned int flags);

函数参数:fd_in和fd_out必须都是管道文件描述符。

返回值:成功时返回在两个文件描述符之间复制的数据字节数，返回0表示没有复制任何数据.失败时返回-1并设置errno.

下面是利用tee函数和splice函数,实现同时输出数据到终端与文件的程序)的基本功能.

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <assert.h>
#include <errno.h>
#include <fcntl.h>

int main(int argc,const char* argv[]){
    if(argc<=2){
        printf("usage:%s <file>\n",argv[0]);
        exit(-1);
    }

    int filefd=open(argv[1],O_WRONLY|O_CREAT|O_TRUNC,0666);
    assert(filefd>=0);

    int pipefd_stdout[2];
    int ret=pipe(pipefd_stdout);
    assert(ret!=-1);

    int pipefd_file[2];
    ret=pipe(pipefd_file);
    assert(ret!=-1);

    /*将标准输入内容输入管道文件pipefd_stdout*/
    ret=splice(STDIN_FILENO,NULL,pipefd_stdout[1],NULL,32768,SPLICE_F_MORE
                                    |SPLICE_F_MOVE);
    assert(ret!=-1);

    /*将管道pipefd_stdout[0]输出复制到管道pipefd_file的输入端*/
    ret=tee(pipefd_stdout[0],pipefd_file[1],32768,SPLICE_F_NONBLOCK);
    assert(ret!=-1);
    /*将管道pipefd_file的输出定向到文件描述符filefd上*/
    ret=splice(pipefd_file[0],NULL,filefd,NULL,32768,SPLICE_F_MORE|
                                    SPLICE_F_MORE);
    assert(ret!=-1);
    /*将管道pipefd_stdout的输出定向到标准输出,其内容与文件中完全一致*/
    ret=splice(pipefd_stdout[0],NULL,STDOUT_FILENO,NULL,32768,SPLICE_F_MORE|
                                        SPLICE_F_MOVE);
    assert(ret!=-1);

    close(filefd);
    close(pipefd_file[0]);
    close(pipefd_file[1]);
    close(pipefd_stdout[0]);
    close(pipefd_stdout[1]);
    return 0;
}

fcntl函数的使用

功能描述:常用于控制文件描述符的属性和行为

函数原型:

#include <fcntl.h>
int fcntl(int fd,int cmd,...);

fd是被操作的文件描述符，cmd参数指定执行何种类型的操作.根据操作类型的不同,该函数可能还需要第三个可选参数.

fcntl支持的常用操作及其参数