7.利用getopt()实现tee命令

前言

上一篇解析了getopt()的用法，这一节就来实战一下，实现tee命令。而且这还是书上的练习题。

纯净代码

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <unistd.h>
#include <fcntl.h>
#define BUF_SIZE 1024
#define MAX_FILE_NUM 128

void errQuit(const char *format, ...);

int main(int argc, char **argv)
{
    int opt;
    bool append = false;
    char helpMsg[100];
    sprintf(helpMsg, "usage: %s <file1> <file2>... <fileN>", argv[0]);
	opterr = 0;
    for (; (opt = getopt(argc, argv, "+a")) != -1;)
    {
        switch (opt)
        {
        case '?':
            errQuit("%s\n", helpMsg);
        case 'a':
            append = true;
            break;
        }
    }
    if (optind >= argc)
        errQuit("%s\n", helpMsg);
    int flag = O_CREAT | O_WRONLY;
    if (append)
        flag |= O_APPEND;
    else
        flag |= O_TRUNC;
    mode_t filePerms = S_IRUSR | S_IWUSR; // rw-------
    int i, fileNum, fd, fds[MAX_FILE_NUM];
    for (i = optind, fileNum = 0; i < argc; ++i)
    {
        fds[i - optind] = fd = open(argv[i], flag, filePerms);
        if (fd == -1)
            errQuit("open");
        fileNum++;
    }
    int numRead;
    char buf[BUF_SIZE];
    for (; (numRead = read(STDIN_FILENO, buf, sizeof(buf) - 1)) > 0;)
    {
        if (write(STDOUT_FILENO, buf, numRead) != numRead)
            errQuit("write");
        for (i = 0; i < fileNum; ++i)
        {
            if (write(fds[i], buf, numRead) != numRead)
                errQuit("write");
        }
    }
    if (numRead == -1)
        errQuit("read");
    for (i = 0; i < fileNum; ++i)
    {
        if (close(fds[i]) == -1)
            errQuit("close");
    }
    exit(EXIT_SUCCESS);
}
void errQuit(const char *format, ...)
{
    va_list argptr;
    va_start(argptr, format);
    char buf[BUF_SIZE];
    vsnprintf(buf, sizeof(buf) - 1, format, argptr);
    va_end(argptr);
    printf("%s", buf);
    exit(EXIT_FAILURE);
}

运行效果

注释代码

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <unistd.h>
#include <fcntl.h>
#define BUF_SIZE 1024	//缓冲区大小
#define MAX_FILE_NUM 128//最大处理文件数

void errQuit(const char *format, ...);//报错函数

int main(int argc, char **argv)
{
    int opt;//存储getopt()返回值
    bool append = false;//是否追加模式写入
    char helpMsg[100];//报错消息
    sprintf(helpMsg, "usage: %s <file1> <file2>... <fileN>", argv[0]);
		//连续调用，正常解析完毕才退出循环。
	opterr = 0;//关闭getopt()自动显示报错消息
    for (; (opt = getopt(argc, argv, "+a")) != -1;)//加号表示关闭GNU重排列扩展。
    {
        switch (opt)
        {
        case '?'://返回问号是报错：非法选项，或者选项缺少参数。这里是非法选项。
            errQuit("%s\n", helpMsg);
        case 'a'://解析到了-a选项，说明开启追加模式。
            append = true;
            break;
        }
    }
	//缺少参数。这里的缺少是指没有我们期待的参数，而不是选项带参数，所以要自己
	//判定一下。比如：./tee，或者是 ./tee -a，都没有我们想要的参数（要写入的文件名）。
    if (optind >= argc)
        errQuit("%s\n", helpMsg);
    int flag = O_CREAT | O_WRONLY;//基本操作
    if (append)//追加模式判定
        flag |= O_APPEND;
    else
        flag |= O_TRUNC;
    mode_t filePerms = S_IRUSR | S_IWUSR; //rw权限创建打开
    int i, fileNum, fd, fds[MAX_FILE_NUM];//fds是文件描述符数组
    for (i = optind, fileNum = 0; i < argc; ++i)
    {
        fds[i - optind] = fd = open(argv[i], flag, filePerms);
        if (fd == -1)
            errQuit("open");
        fileNum++;//记录文件数
    }
    int numRead;
    char buf[BUF_SIZE];
    for (; (numRead = read(STDIN_FILENO, buf, sizeof(buf) - 1)) > 0;)
    {
        if (write(STDOUT_FILENO, buf, numRead) != numRead)//控制台打印
            errQuit("write");
        for (i = 0; i < fileNum; ++i)
        {
            if (write(fds[i], buf, numRead) != numRead)
                errQuit("write");
        }
    }
    if (numRead == -1)
        errQuit("read");
    for (i = 0; i < fileNum; ++i)
    {
        if (close(fds[i]) == -1)
            errQuit("close");
    }
    exit(EXIT_SUCCESS);
}
void errQuit(const char *format, ...)
{
    va_list argptr;
    va_start(argptr, format);
    char buf[BUF_SIZE];
    vsnprintf(buf, sizeof(buf) - 1, format, argptr);
    va_end(argptr);
    printf("%s", buf);
    exit(EXIT_FAILURE);
}

你学会了吗？😃