【Linux】自主实现shell

大概流程图

 普通minishell具体的代码实现

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <ctype.h>
#include <sys/wait.h>

int main()
{
    while(1)
    {
        //先做一个界面， 循环打印
        printf("[qujiale@minishell]$ ");
        fflush(stdout);
        char buf[1024] = {0};
        //%[^\n] : 获取字符串直到遇到\n  
        //%*c : 将一个字符从缓冲区中取出
        if(scanf("%[^\n]%*c",buf) != 1)
        {
            //如果输入只有一个\n, 就把\n获取到
            getchar();
            continue;
        }

        char* argv[32];
        int argc = 0;
        char* ptr = buf;
        while(*ptr != '\0')
        {
            if(!isspace(*ptr))
            {
                //当前位置是非空白字符
                argv[argc++] = ptr;
                while(!isspace(*ptr) && *ptr != '\0')
                {
                    ptr++;
                }
            }
            *ptr = '\0';
            ptr++;
        }
        //参数以NULL结尾
        argv[argc] = NULL;

        if(strcmp(argv[0], "cd") == 0)
        {
            //改变当前工作路径
            chdir(argv[1]);
            continue;
        }
        //创建子进程， 执行任务
        int pid = fork();
        if(pid < 0)
        {
            continue;
        }
        else if(pid == 0)
        {
            //子进程中程序替换
            if(execvp(argv[0],argv) < 0)
            {
                perror("");
            }
            //替换失败就退出
            exit(0);
        }

        //等待子进程退出--避免僵尸进程
        wait(NULL);
    }
    return 0;
}

运行结果 :

 

增加重定向功能的minishell代码

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <ctype.h>
#include <fcntl.h>
#include <sys/wait.h>

char buf[1024] = {0};
char* argv[32];
int argc = 0;
int redirect_flag = 0;
char* redirect_file = NULL;

int do_face()
{
    //先做一个界面， 循环打印
    printf("[qujiale@minishell]$ ");
    fflush(stdout);
    //%[^\n] : 获取字符串直到遇到\n  
    //%*c : 将一个字符从缓冲区中取出
    if(scanf("%[^\n]%*c",buf) != 1)
    {
        //如果输入只有一个\n, 就把\n获取到
        getchar();
        return -1;
    }
    return 0;
}

int do_parse()
{
    argc = 0;
    char* ptr = buf;
    while(*ptr != '\0')
    {
        if(!isspace(*ptr))
        {
            //当前位置是非空白字符
            argv[argc++] = ptr;
            while(!isspace(*ptr) && *ptr != '\0')
            {
                ptr++;
            }
        }
        *ptr = '\0';
        ptr++;
    }
    //参数以NULL结尾
    argv[argc] = NULL;

    return 0;
}

int build_in()
{
    if(strcmp(argv[0], "cd") == 0)
    {
        //改变当前工作路径
        chdir(argv[1]);
        return -1;
    }
    return 0;
}

int do_redirect()
{
    redirect_flag = 0;
    redirect_file = NULL;
    char* ptr = buf;
    while(*ptr != '\0')
    {
        if(*ptr == '>')
        {
            redirect_flag = 1;
            *ptr = '\0';
            ptr++;
            if(*ptr == '>')
            {
                redirect_flag = 2;
                *ptr = '\0';
                ptr++;
            }
            while(*ptr != '\0' && isspace(*ptr))
            {
                ptr++;
            }
            redirect_file = ptr;
            while(*ptr != '\0' && !isspace(*ptr))
            {
                ptr++;
            }
            *ptr = '\0';
        }
        ptr++;
    }
    return 0;
}

int main()
{
    int ret = 0;
    while(1)
    {
        ret = do_face();
        if(ret < 0)
        {
            continue;
        }
        ret = do_redirect();
        if(ret < 0)
        {
            continue;
        }
        ret = do_parse();
        if(ret < 0)
        {
            continue;
        }
        ret = build_in();
        if(ret < 0)
        {
            continue;
        }
        //创建子进程， 执行任务
        int pid = fork();
        if(pid < 0)
        {
            continue;
        }
        else if(pid == 0)
        {
            int fd = 1;
            //子进程中程序替换
            if(redirect_flag == 1)
            {
                //清空重定向
                fd = open(redirect_file,O_RDWR | O_CREAT | O_TRUNC, 0664);
                if(fd < 0)
                {
                    exit(-1);
                }
            }
            else if(redirect_flag == 2)
            {
                //追加重定向
                fd = open(redirect_file,O_RDWR | O_CREAT | O_APPEND, 0664);
                if(fd < 0)
                {
                    exit(-1);
                }
            }
            dup2(fd, 1);
            if(execvp(argv[0],argv) < 0)
            {
                perror("replace error");
            }
            //替换失败就退出
            exit(0);
        }

        //等待子进程退出--避免僵尸进程
        wait(NULL);
    }
    return 0;
}

 运行结果