zoukankan      html  css  js  c++  java
  • Linux/UNIX编程:实现简单 tee 命令

    思路很简单,从标准输入文件描述符读入数据,然后同时向标准输出和参数指定的文件写出数据;如果加了 -a 选项,则以追加的方式向文件写出数据。还没了解 getopt() 函数就没判断参数是否合法。

    #include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <unistd.h>
    #include <fcntl.h>
    #include <sys/stat.h>
    #include <errno.h>
    
    #ifndef BUF_SIZE
    #define BUF_SIZE 1024
    #endif
    
    int main(int argc, char const *argv[])
    {
        int outputFd;
        /* 命令格式要求为 `tee [-a] file`,没有判断参数是否合法*/
        if(argc == 3){ //加了 -a 选项,则以追加的方式写数据
            outputFd = open(argv[2], O_WRONLY | O_CREAT | O_APPEND, S_IRUSR | S_IWUSR | S_IRGRP); //rw-r-----
        } else if (argc == 2){
            outputFd = open(argv[1], O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP); //rw-r-----
        }
    
        if(outputFd == -1){
            perror(strerror(errno));
            exit(EXIT_FAILURE);
        }
    
        char buffer[BUF_SIZE];
        int readNum;
    
        while((readNum = read(STDIN_FILENO, buffer, BUF_SIZE)) > 0){
            ssize_t fileWrNum = write(outputFd, buffer, readNum); //向指定文件写出数据
            ssize_t stdoutWrNum = write(STDOUT_FILENO, buffer, readNum); //向标准输出写入数据
    
            if(fileWrNum != readNum || stdoutWrNum != readNum){ //判断是否写成功
                perror("couldn't write the whole buffer
    ");
                exit(EXIT_FAILURE);
            }
        }
        if(readNum == -1){ //判断是否读取成功
            perror(strerror(errno));
            exit(EXIT_FAILURE);
        }
    
        if(close(outputFd) == -1){ //判断是否成功关闭
            perror(strerror(errno));
            exit(EXIT_FAILURE);
        }
       exit(EXIT_SUCCESS);
    }

      

  • 相关阅读:
    1066 Bash 游戏
    1070 Bash 游戏 V4
    codevs 2928 你缺什么
    分块、线段树练习
    Father Christmas flymouse
    codevs 2494 Vani和Cl2捉迷藏
    codevs 2144 砝码称重2
    国王游戏
    codevs 1664 清凉冷水
    2015ACM/ICPC亚洲区沈阳站 Pagodas
  • 原文地址:https://www.cnblogs.com/yuanyb/p/11000972.html
Copyright © 2011-2022 走看看