Linux 用C语言实现简单的shell(2)

不知不觉两周没有发文了，因为“一万美金的福特奖学金答辩”，ACM比赛，网络论文阅读和网络大作业一大堆事把时间冲散了，所以先写一篇博文补上之前一坑。

之前发了一篇关于linux 用C语言实现简单shell的博文，当时因为刚刚接触linux，只是处理了：

1）外部命令

2）pwd,cd,exit内置命令

3）输入输出重定向

并且代码相比较而言是一步一步添加的，代码相对来讲比较丑QAQ，所以在学完管道之后，相信不得不重新写代码才能实现了。

相比较之前的版本我对代码进行了相关的修改：

1）对于shell指令采用结构体存储，方便管道的切割与执行；

2）采用strtok_r对于指令进行了更加合理的切割；

3）增加了多管道处理；

4）处理了一些简单的异常问题，包括文件路径问题，空行问题，指令错误等。

评论中有朋友想让我加入后台运行‘&’功能的，其实基本上没什么区别，只是对于增加了‘&’后台运行的命令主进程不再需要等待子进程结束再操作了，我已经对我的代码进行了一些修改并加上一定的注释。

/*author:Samsons
  date:2015.4.28
  1)对于shell指令采用结构体存储，方便管道的切割与执行；
  2)采用strtok_r对于指令进行了更加合理的切割；
  3)增加了多管道处理；
  4)处理了一些简单的异常问题，包括文件路径问题，空行问题，指令错误等。  
  date:2015.4.30  
  增加了后台处理命令“&”的处理
  */
#include <stdio.h>
#include <signal.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/wait.h>

#define MAX 100
#define LEN 100

//shell指令单个管道结构体
struct cmd_list{
    int argc;  //单个管道参数个数
    char *argv[MAX];
};

struct cmd_list *cmdv[MAX];  //shell指令
int num;//shell管道个数
int flagdo;//是否为后台处理命令标记

//执行外部命令
void execute(char *argv[])
{
        int error;
        error=execvp(argv[],argv);
        if (error==-)  printf("failed!\n");
        exit();
}

//切分单个管道
void split_cmd(char *line)
{
     struct cmd_list * cmd = (struct cmd_list *)malloc(sizeof(struct cmd_list));
     cmdv[num++] = cmd;
     cmd->argc = ;
     char *save;
     char *arg = strtok_r(line, " \t", &save);//切分空格
     while (arg)
     {
        cmd->argv[cmd->argc] = arg;
        arg = strtok_r(NULL, " \t", &save);
        cmd->argc++;
     }
     cmd->argv[cmd->argc] = NULL;
}

//切分管道
void split_pipe(char *line)
{
    char *save;
    char * cmd = strtok_r(line, "|", &save);
    while (cmd) {
        split_cmd(cmd);
        cmd = strtok_r(NULL, "|", &save);
    }
}

//执行管道命令
void do_pipe(int index)
{
    if (index == num - )
        execute(cmdv[index]->argv);
    int fd[];
    pipe(fd);//创建管道，0读，1写
    if (fork() == )
    {
        dup2(fd[], );
        close(fd[]);
        close(fd[]);
        execute(cmdv[index]->argv);
    }
    dup2(fd[], );
    close(fd[]);
    close(fd[]);
    do_pipe(index + );
}

//执行内部指令
int inner(char *line)
{
    char *save,*tmp[MAX];
    char t[LEN],p[LEN];
    strcpy(t,line);
    char *arg = strtok_r(line, " \t", &save);//切分空格
    int i=;
    while (arg) {
        tmp[i] = arg;
        i++;//记录命令个数
        arg = strtok_r(NULL, " \t", &save);
     }
    tmp[i] = NULL;
    if (strcmp(tmp[i-],"&")==)//判断是否为后台处理命令
    {
        flagdo=;
        i--;
    }
    if (strcmp(tmp[],"exit")==)//exit
    {
        exit();
        return ;
    }
    else
    if (strcmp(tmp[],"pwd")==)//pwd
    {
        char buf[LEN];
        getcwd(buf,sizeof(buf));//得到当前路径
        printf("Current dir is:%s\n",buf);
        return ;
    }
    else
    if (strcmp(tmp[],"cd")==)//cd
    {
        char buf[LEN];
        if (chdir(tmp[])>=)
        {
            getcwd(buf,sizeof(buf));
            printf("Current dir is:%s\n",buf);
        }
        else
        {
            printf("Error path!\n");
        }
        return ;
    }
    else return ;
}

//输入重定向
void cat_in(char *q)
{
    char t[];
    int fd;
    if (q[]=='<')
    {
        strcpy(t,q+);
        fd=open(t,O_RDONLY);
        cmdv[]->argv[cmdv[]->argc-]=NULL;  //默认重定向为参数的最后一个
        cmdv[]->argc--;
        if (fd==-)
        {
            printf("file open failed\n");
            return;
        }
        dup2(fd,);
        close(fd);
    }
}

//输出重定向
void cat_out(char *q)
{
    char t[];
    int fd;
    if (q[]=='>')
    {
        strcpy(t,q+);
        cmdv[num-]->argv[cmdv[num-]->argc-]=NULL;
        cmdv[num-]->argc--;
        fd=open(t,O_CREAT|O_RDWR,); //0666为权限
        if (fd==-)
        {
            printf("file open failed\n");
            return;
        }
        dup2(fd,);
        close(fd);
    }
}

int main()
{
    int i,pid;
    char buf[LEN],p[LEN];
    while ()
    {
        fgets(buf,LEN,stdin);//读入shell指令
        if (buf[]=='\n') continue;
        buf[strlen(buf)-]='\0';
        strcpy(p,buf);
        int inner_flag;
        inner_flag=inner(buf);//内置指令执行
        if (inner_flag==)
        {
            pid=fork();//建立新的进程
            if (pid==)
            {
                split_pipe(p);//管道的切割
                //如果是后台处理命令将&符号删除
                if (strcmp(cmdv[num-]->argv[cmdv[num-]->argc-],"&")==)
                {
                    cmdv[num-]->argc--;
                    cmdv[num-]->argv[cmdv[num-]->argc]=NULL;
                }
                //默认输入输出重定向都是最后一个参数，输入时第一个管道，输出是最后一个管道
                if (cmdv[]->argv[cmdv[]->argc-]!=NULL)
                {
                    char q[LEN];
                    strcpy(q,cmdv[]->argv[cmdv[]->argc-]);
                    cat_in(q);//输入重定向
                }
                if (cmdv[num-]->argv[cmdv[num-]->argc-]!=NULL)
                {
                    char q[LEN];
                    strcpy(q,cmdv[num-]->argv[cmdv[num-]->argc-]);
                    cat_out(q);//输出重定向
                }
                do_pipe();//执行管道
                exit();
            }
            if (flagdo==)//非后台处理命令主进程才需等待子进程处理
                waitpid(pid,NULL,);
        }
    }
    return ;
}