从标准输入连续读取

Question

我刚开始学习 C 很抱歉，如果我错过了一些明显的东西。作为一种实践，我想编写自己的 posix cat 并且几乎可以正常工作，但是当没有参数时，它不会从 stdin 读取多个文件。

这是预期的行为（忽略这个，因为这是我不知道的 Zsh 功能）：

$ cat < hello.txt < world.txt
hello
world
$ 

但是我的cat 因“分段错误”而崩溃（我有一个没有崩溃但也没有输出任何内容的版本）

代码如下：

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>

int main(int argc, char *argv[])
{
    char opt;
    while ((opt = getopt(argc, argv, "u")) != EOF) {
        switch(opt) {
            case 'u':
                /* Make the output un-buffered */
                setbuf(stdout, NULL);
                break;
            default:
                break;
        }
    }

    argc -= optind;
    argv += optind;

    int i = 0, fildes, fs = 0;

    do {
        /* Check for operands, if none or operand = "-". Read from stdin */
        if (argc == 0 || !strcmp(argv[i], "-")) {
            fildes = STDIN_FILENO;
        } else {
            fildes = open(argv[i], O_RDONLY);
        }

        /* Check for directories */
        struct stat fb;
        if (!fstat(fildes, &fb) && S_ISDIR(fb.st_mode)) {
            fprintf(stderr, "cat: %s: Is a directory\n", argv[i]);
            i++;
            continue;
        }

        /* Get file size */
        fs = fb.st_size;

        /* If bytes are read, write them to stdout */
        char buf[fs];
        while ((read(fildes, buf, fs)) > 0)
            write(STDOUT_FILENO, buf, fs);

        /* Close file if it's not stdin */
        if (fildes != STDIN_FILENO)
            close(fildes);

        i++;
    } while (argv[i] != NULL || argc == 0);

    return 0;
}

Answer 1

当您使用 gcc 编译代码时使用 -fsanitize=address 代码显示：

AddressSanitizer:DEADLYSIGNAL
=================================================================
==604950==ERROR: AddressSanitizer: stack-overflow on address 0x7ffea6d94000 (pc 0x55b9a63dfb57 bp 0x7ffea6d90d20 sp 0x7ffea6d90b80 T0)
    #0 0x55b9a63dfb57 in main /tmp/10/1.c:56
    #1 0x7f1403c17b24 in __libc_start_main (/usr/lib/libc.so.6+0x27b24)
    #2 0x55b9a63df3dd in _start (/tmp/.ccrun.604893.exefile+0x23dd)

SUMMARY: AddressSanitizer: stack-overflow /tmp/10/1.c:56 in main
==604950==ABORTING

Th sn-p:

do {
   ...
   ++i;
} while (argv[i] != NULL || argc == 0);       // HERE!

argc 总是0，所以while(...) 总是在执行，而++i 总是在递增。 argc[...] 被越界访问，直到出现段错误。

我觉得你应该改成} while (argv[i] != NULL && argc != 0);