shell命令解析器

最新推荐文章于 2025-06-20 19:36:48 发布

#欲速则不达#

最新推荐文章于 2025-06-20 19:36:48 发布

阅读量786

点赞数 29

CC 4.0 BY-SA版权

文章标签： linux 运维服务器 c++ vim bash 开发语言

本文链接：https://blog.csdn.net/m0_69323023/article/details/139030413

一、原理

我们前面学习了进程创建，进程终止，进程等待，进程替换，通过这些内容我们可以来进行实现简

单的shell命令行解释器。下面我们直接来看一看如何去实现shell命令行解释器。

1.1、什么是shell

shell就是一个应用程序，我们从键盘上给shell发送命令，回车之后shell就会去执行这些命令。

shell可以接收键盘上的输入数据并回显，当我们使用键盘给shell程序发送回车时，shell就知道字符输入结束了，它会解析输入的字符串，把这些字符串用空格分为几个部分，我们以ls -h为例，第1部分就是程序名，其它部分就是参数，然后寻找程序然后执行程序。那么shell会去哪里找程序呢？shell会去PATH环境变量所指示的位置找。我们也可以指定绝对、相对路径，shell就会直接去这些路径找到程序，找到程序后，会启动该程序，并传入参数。

我们输入env命令

等号左边都是环境变量，等号右边是环境变量的值

我们可以用echo $PATH来显示某个环境变量，里面都是各个字符串，各个字符串都是用冒号隔开，shell程序就是去这些目录里面逐个寻找看看有没有ls这个程序名，找到之后就去执行它。

也可以自己指定绝对路径和相对路径，这样shell程序就不会去PATH环境变量里面寻找。

1.2、基本原理

1、我们知道，在打开shell解释器后，除非我们自己关闭它，那么它将一直不退出，所以命令行解释器一定是一个死循环。

2、输入：我们需要获取命令行上一行的内容，利用fgets函数获取，同时，可以定义一个cmd_line[NUM]数组来保存用户输入的内容。

3、解析：输入之后，我们自然需要去进行字符串的解析，我们需要分割字符串，将其分成命令+选项等内容，这个地方用strtok函数，把字符串切割成若干个子串，并将分开的各个子串保存至char* g_val[SIZE]数组中。strtok：第一次直接传递参数，第二次则必须传NULL。且在最终strtok会返回NULL。

4、我们在输入命令后，发现界面还是维持的原样，这是因为父进程会创建子进程去帮助其实现命令，同时父进程需要等待子进程退出返回结果。

5、子进程在执行命令的时候，我们需要使用进程替换的知识，将代码换成命令执行的代码。

6、有些命令，如cd命令，如果我们是交给子进程去完成的话，只会改变子进程的路径，所以cd命令必须由父进程亲自来完成。

不需要创建子进程执行，让shell自己执行命令，称为内建命令。本质就是执行系统接口，我们可以调用一个系统接口chdir，可以让父进程自己改变路径。

7、我们在输入完命令后，通常会按回车建使其执行，而fgets则会将它当成输入的一个字符，在给用户呈现结果时会多换一行，而不会紧挨着上一行去显示结果。所以我们要执行下面的操作：

cmd_line[strlen(line_cmd)-1] = '\0';

8、再比如，像ll这种命令，其等效于ls -l，因此我们需要将其在数组char* g_val[SIZE]中换成 g_val[0] = "ls"，g_val[1] = "-l"即可。

二、代码实现

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

#define NUM 1024
#define SIZE 64
#define SEP " "
//#define Debug 1

//redir
#define NoneRedir   0
#define OutputRedir 1
#define AppendRedir 2
#define InputRedir  3

int redir = NoneRedir;
char *filename = NULL;

char cwd[1024];
char enval[1024]; // for test
int lastcode = 0;

char *homepath()
{
    char *home = getenv("HOME");
    if(home) return home;
    else return (char*)".";
}

const char *getUsername()
{
    const char *name = getenv("USER");
    if(name) return name;
    else return "none";
}
const char *getHostname()
{
    const char *hostname = getenv("HOSTNAME");
    if(hostname) return hostname;
    else return "none";
}
const char *getCwd()
{
    const char *cwd = getenv("PWD");
    if(cwd) return cwd;
    else return "none";
}
int getUserCommand(char *command, int num)
{
    printf("[%s@%s %s]# ", getUsername(), getHostname(), getCwd());
    char *r = fgets(command, num, stdin); // 最终你还是会输入\n
    if(r == NULL) return -1;
    // "abcd\n" "\n"
    command[strlen(command) - 1] = '\0'; // 有没有可能越界？不会
    return strlen(command);
}

void commandSplit(char *in, char *out[])
{
    int argc = 0;
    out[argc++] = strtok(in, SEP);
    while( out[argc++] = strtok(NULL, SEP));

#ifdef Debug
    for(int i = 0; out[i]; i++)
    {
        printf("%d:%s\n", i, out[i]);
    }
#endif
}

int execute(char *argv[])
{
    pid_t id = fork();
    if(id < 0) return -1;
    else if(id == 0) //child
    {
        // 程序替换会不会影响曾经的重定向呢？？不会！！ 为什么？如何理解？？
        int fd = 0;
        if(redir == InputRedir)
        {
            fd = open(filename, O_RDONLY); // 差错处理我们不做了
            dup2(fd, 0);
        }
        else if(redir == OutputRedir)
        {
            fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0666);
            dup2(fd, 1);
        }
        else if(redir == AppendRedir)
        {
            fd = open(filename, O_WRONLY | O_CREAT | O_APPEND, 0666);
            dup2(fd, 1);
        }
        else
        {
            //do nothing
        }
        // exec command
        execvp(argv[0], argv); // cd ..
        exit(1);
    }
    else // father
    {
        int status = 0;
        pid_t rid = waitpid(id, &status, 0);
        if(rid > 0){
            lastcode = WEXITSTATUS(status);
        }
    }

    return 0;
}

void cd(const char *path)
{
    chdir(path);
    char tmp[1024];
    getcwd(tmp, sizeof(tmp));
    sprintf(cwd, "PWD=%s", tmp); // bug
    putenv(cwd);
}

// 什么叫做内键命令: 内建命令就是bash自己执行的，类似于自己内部的一个函数！
// 1->yes, 0->no, -1->err
int doBuildin(char *argv[])
{
    if(strcmp(argv[0], "cd") == 0)
    {
        char *path = NULL;
        if(argv[1] == NULL) path=homepath();
        else path = argv[1];
        cd(path);
        return 1;
    }
    else if(strcmp(argv[0], "export") == 0)
    {
        if(argv[1] == NULL) return 1;
        strcpy(enval, argv[1]);
        putenv(enval); // ???
        return 1;
    }
    else if(strcmp(argv[0], "echo") == 0)
    {
        if(argv[1] == NULL){
            printf("\n");
            return 1;
        }
        if(*(argv[1]) == '$' && strlen(argv[1]) > 1){ 
            char *val = argv[1]+1; // $PATH $?
            if(strcmp(val, "?") == 0)
            {
                printf("%d\n", lastcode);
                lastcode = 0;
            }
            else{
                const char *enval = getenv(val);
                if(enval) printf("%s\n", enval);
                else printf("\n");
            }
            return 1;
        }
        else {
            printf("%s\n", argv[1]);
            return 1;
        }
    }
    else if(0){}

    return 0;
}

#define SkipSpace(pos) do{ while(isspace(*pos)) pos++; }while(0)

void checkRedir(char usercommand[], int len)
{
    // ls -a -l > log.txt
    // ls -a -l >> log.txt
    char *end = usercommand + len - 1;
    char *start = usercommand;
    while(end>start)
    {
        if(*end == '>')
        {
            if(*(end-1) == '>')
            {
                *(end-1) = '\0';
                filename = end+1;
                SkipSpace(filename);
                redir = AppendRedir;
                break;
            }
            else
            {
                *end = '\0';
                filename = end+1;
                SkipSpace(filename);
                redir = OutputRedir;
                break;
            }
        }
        else if(*end == '<')
        {
            *end = '\0';
            filename = end+1;
            SkipSpace(filename); // 如果有空格，就跳过
            redir = InputRedir;
            break;
        }
        else
        {
            end--;
        }
    }
}
int main()
{
    while(1){
        redir = NoneRedir;
        filename = NULL;
        char usercommand[NUM];
        char *argv[SIZE];
        // 1. 打印提示符&&获取用户命令字符串获取成功
        int n = getUserCommand(usercommand, sizeof(usercommand));
        if(n <= 0) continue;
        // "ls -a -l > log.txt" -> 判断 -> "ls -a -l"  redir_type   "log.txt"
        // 1.1 检测是否发生了重定向
        checkRedir(usercommand, strlen(usercommand));
        // 2. 分割字符串
        // "ls -a -l" -> "ls" "-a" "-l"
        commandSplit(usercommand, argv);
        // 3. check build-in command
        n = doBuildin(argv);
        if(n) continue;
        // 4. 执行对应的命令
        execute(argv);
    }
}