操作系统上机作业--实现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;
}

欢迎留言交流。。。