如何區分以下兩種情況下用C

Question

我目前工作的一個貓計劃，必須區分以下情況：

./cat 

和

./cat <file1> file2.txt 

這是代碼

while((fileFD=read(STDIN_FILENO,str,1000))!=0)// If we just execute the command cat, it will receive input from the stdin and put it in the stdout. 
    { 
    write(STDOUT_FILENO,str,fileFD);// Write it to the standard output 
    if(fileFD==-1) 
    { 
    perror("error"); 
    exit(1); 
    } 
    } 
    break;  

我使用的參數的argv和argc個跟蹤的參數。在這兩種情況下ARGC = 1，因爲對<權東西沒有考慮到。你會如何做到這一點？

Answer 1

給定兩個文件描述符，一個用於輸入，一個用於輸出，下面的函數將數據從輸入到輸出合理可靠。

#include <unistd.h> 

/* Copy file from input file descriptor ifd to output file descriptor ofd */ 
extern int copy_file(int ifd, int ofd); 

int copy_file(int ifd, int ofd) 
{ 
    char buffer[65536]; // Resize to suit your requirements 
    ssize_t i_bytes; 
    while ((i_bytes = read(ifd, buffer, sizeof(buffer))) > 0) 
    { 
     ssize_t bytes_left = i_bytes; 
     char *posn = buffer; 
     ssize_t o_bytes; 
     while ((o_bytes = write(ofd, posn, bytes_left)) > 0) 
     { 
      bytes_left -= o_bytes; 
      posn += o_bytes; 
     } 
     if (o_bytes < 0) 
      return -1; 
    } 
    if (i_bytes < 0) 
     return -1; 
    return 0; 
} 

裸最小測試程序可能是：

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

…declaration or definition of file_copy()… 

int main(void) 
{ 
    int rc = EXIT_SUCCESS; 
    if (copy_file(STDIN_FILENO, STDOUT_FILENO) != 0) 
    { 
     int errnum = errno; 
     rc = EXIT_FAILURE; 
     fprintf(stderr, "copy_file failed: (%d) %s\n", errnum, strerror(errnum)); 
    } 
    return rc; 
} 

從標準輸入到標準輸出這正確地複製文件。我還沒有明確測試過「短寫」。這是非常罕見的情況，最可能的上下文是輸出文件描述符是套接字的時候。測試錯誤路徑也很困難。我打電話給cf89。一個試驗：

$ (trap '' 13; ./cf89 < /dev/zero) | (sleep 1) 
copy_file failed: (32) Broken pipe 
$ 

的trap忽略SIGPIPE（13），所以寫塊，因爲沒有什麼是閱讀的管道。當sleep退出時，寫入處理獲得從write()錯誤 - 由於信號被忽略 - 這生成錯誤消息。

順便說一句，還有那些誰反對都報告錯誤號和基於安全理由的錯誤信息 - 它告訴攻擊者太多的系統，如果他們設法觸發錯誤。我不在這個營地;攻擊者在攻擊這樣的程序時已經知道系統類型。它可能與其他環境相關，比如Web服務器。

Answer 2

此命令行：

cat <file1> file2.txt 

是簡單地將標準輸入和標準輸出和程序需要做什麼特別的。

然而，如果在命令行是：

cat file1 

然後代碼需要重新打開（）標準輸入到文件

並且可以存在，如果重新打開（到stderr的消息）失敗

在任一組命令行參數中，所有I/O都將從標準輸入和標準輸出

Answer 3

您忘記了其他情況。什麼pipelines，像

date | ./cat | wc 

也許你想測試，如果標準輸入是一個僞TTY（或沒有）。閱讀tty demystified第一（與termios(3) ...）。然後考慮使用isatty(3)作爲isatty(STDIN_FILENO)來測試。但是你應該提供一些選項來覆蓋測試。見L. Serrni's deleted answer。您可能需要使用poll(2)。閱讀Advanced Linux Programming。

這個問題看起來像一些XY problem。