QuXiao

在shell中對(duì)程序進(jìn)行重定向很簡(jiǎn)單，用<和>符號(hào)就可以了，但在自己的程序中怎么實(shí)現(xiàn)輸入輸出重定向呢？

先來(lái)看看Linux內(nèi)核中，文件（還是設(shè)備）是通過(guò)哪些數(shù)據(jù)結(jié)構(gòu)保存的：

每個(gè)進(jìn)程都保存一份文件描述符的表格，每一行又指向file table，然后file table再指向v-node table。v-node table我們可以暫不考慮，暫且把它當(dāng)作文件的內(nèi)容。而從process table entry中的file pointer可以指向不同或者相同的file table。原本標(biāo)準(zhǔn)輸入輸出是指向“鍵盤(pán)”和“屏幕”這兩個(gè)設(shè)備的，如果可以將它們指向我們指定的文件，就可以實(shí)現(xiàn)重定向了。

先用open()打開(kāi)需要重定向到的文件，獲取去文件描述符fd，在用dup2()把進(jìn)程中原先的輸入輸出文件描述符STDIN_FILENO和STDOUT_FILENO重定向至fd，這樣就可以實(shí)現(xiàn)輸入輸出重定向了。我想，shell實(shí)現(xiàn)重定向也應(yīng)該是類(lèi)似的思想，關(guān)鍵代碼如下：

#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;
}