c - 处理文件或标准输入

Question

我正在创建一个处理存档的小程序；其运行方式如下:

"./program file.txt".

如果没有找到第二个参数，它将从 stdin 处理

"cat file.txt | ./program"

但是在代码内部......我知道我可以做到这一点:

if (argc==2){
    FILE* archive = argv[1];
    function_1(archive);
    function_2(archive);
    ...
    function_n(archive);
else{
    function_1(stdin);
    function_2(stdin);
    ...
    function_n(stdin);
}

但是我可以这样做吗？

file* archive;
if (argc==2){
    archive = argv[1];
else{
    archive = stdin;
}
function_1(archive);
function_2(archive);
...
function_n(archive);

因为我想少写代码!

Answer 1

由于 stdin 是 FILE * 类型的变量，因此您可以这样做:

FILE *archive;
if (argc==2){
    archive = fopen(argv[1], "r");
else{
    archive = stdin;
}