在shell中對程序進行重定向很簡單,用<和>符號就可以了,但在自己的程序中怎么實現輸入輸出重定向呢?
先來看看Linux內核中,文件(還是設備)是通過哪些數據結構保存的:
每個進程都保存一份文件描述符的表格,每一行又指向file table,然后file table再指向v-node table。v-node table我們可以暫不考慮,暫且把它當作文件的內容。而從process table entry中的file pointer可以指向不同或者相同的file table。原本標準輸入輸出是指向“鍵盤”和“屏幕”這兩個設備的,如果可以將它們指向我們指定的文件,就可以實現重定向了。
先用open()打開需要重定向到的文件,獲取去文件描述符fd,在用dup2()把進程中原先的輸入輸出文件描述符STDIN_FILENO和STDOUT_FILENO重定向至fd,這樣就可以實現輸入輸出重定向了。我想,shell實現重定向也應該是類似的思想,關鍵代碼如下:
#include <stdio.h> #include <stdlib.h> #include <ctype.h> #include <unistd.h> #include <sys/wait.h> #include <sys/resource.h> #include <sys/types.h> #include <sys/stat.h> #include <errno.h> #include <fcntl.h> #include <signal.h> int main (int argc, char** argv) { if ( argc != 3 ) { printf("usage: inputFile outputFile\n"); return 1; } int inFd, outFd; //open file descriptor inFd = open(argv[1], O_RDONLY); if ( inFd < 0 ) { printf("inFd open error!\n%s\n",strerror(errno)); return 1; } outFd = open(argv[2], O_CREAT | O_TRUNC | O_RDWR, S_IRWXU | S_IRGRP | S_IROTH); if ( outFd < 0 ) { printf("outFd open error!\n%s\n", strerror(errno)); return 1; } //change standard input and output if ( dup2(inFd, STDIN_FILENO) < 0 ) { printf("inFd dup2 error!\n"); return 1; } if ( dup2(outFd, STDOUT_FILENO) < 0 ) { printf("outFd dup2 error!\n"); return 1; } char line[128]; while ( scanf("%s", &line) != EOF ) { printf("%s\n", line); } return 0; }


