操作系统上机作业--实现shell（2）（多进程）

sh2.c: 实现shell程序，要求在第1版的基础上，添加如下功能

•   实现文件重定向
•   $ echo hello >log
•   $ cat log
•   Hello

实现思路：
和sh1.c相比，主要是修改了cmd函数的实现过程。通过循环找出重定向符号">"的位置，记录下来，并记录下重定向符号后面的文件名，open该文件，然后创建一个子进程，在子进程中，使用dup2()函数将本来要输出到屏幕的数据，重定向到我们刚打开的文件中，然后使用execvp()函数装载外部命令。
代码：

#include<stdio.h>
#include<sys/wait.h>
#include<unistd.h>
#include<sys/types.h>
#include<stdlib.h>
#include<string.h>
#include<pwd.h>
#include<fcntl.h>#define MAX_PATH_LEN 200
#define CMD_LEN 200
int cmd_num;
char *cmd_array[100];void cmd(){pid_t pid;int i;int newFd;for(i=0;i<cmd_num;i++)if(strcmp(cmd_array[i],">")==0){cmd_array[i]=NULL;newFd=open(cmd_array[i+1],O_RDWR|O_CREAT,400,200);if(newFd==-1){printf("Error:can't open the file\n");exit(0);}break;}pid=fork();if(pid==0){dup2(newFd,1);close(newFd);if(execvp(cmd_array[0],cmd_array)<0){printf("Error:command not found\n");}}wait(NULL);
}void print_user(){struct passwd *pwd;char path_name[MAX_PATH_LEN];getcwd(path_name,MAX_PATH_LEN);pwd=getpwuid(getuid());printf("%s@kali:%s$ ",pwd->pw_name,path_name);
}void get_cmd(){int i=0;char str[CMD_LEN];char *buf;cmd_num=0;memset(cmd_array,0,100);fgets(str,CMD_LEN,stdin);buf=strtok(str," \n");while(buf!=NULL){cmd_array[i]=(char *)malloc(sizeof(*buf));strcpy(cmd_array[i++],buf);buf=strtok(NULL," \n");}cmd_array[i]=NULL;cmd_num=i;
}void mycd(){struct passwd *pwd;char path_name[MAX_PATH_LEN];pwd=getpwuid(getuid());if(cmd_num==1){strcpy(path_name,pwd->pw_dir);if(chdir(path_name)==-1){perror("mycd:chdir..");exit(1);}}else{if(chdir(cmd_array[1])==-1){printf("mycd: cd:%s:No such directory\n",cmd_array[1]);}}
}void myexit(){exit(0);
}void mypwd(){char path_name[MAX_PATH_LEN];if(getcwd(path_name,MAX_PATH_LEN)){printf("%s\n",path_name);}else{perror("Error:getcwd...\n");exit(0);}
}int main(){while(1){print_user();get_cmd();if(cmd_array[0]==NULL)return 0;else if(strcmp(cmd_array[0],"exit")==0)myexit();else if(strcmp(cmd_array[0],"pwd")==0)mypwd();else if(strcmp(cmd_array[0],"cd")==0)mycd();else cmd();}return 0;
}