shell的模拟实现

最新推荐文章于 2024-07-20 16:36:14 发布

三分苦

最新推荐文章于 2024-07-20 16:36:14 发布

阅读量1.5k

点赞数 2

分类专栏： Linux 文章标签： bash linux 开发语言

本文链接：https://blog.csdn.net/bit_zyx/article/details/128539614

版权

Linux 专栏收录该内容

23 篇文章 6 订阅

订阅专栏

整体框架分析

代码演示

代码分析

整体框架分析

考虑下面这个与shell典型的互动：
[xzy@ecs-333953 date16]$ ls
makefile  mycmd  mycmd.cpp  myexec  myexec.c  test.py
[xzy@ecs-333953 date16]$ ps
  PID TTY          TIME CMD
21919 pts/0    00:00:00 bash
21947 pts/0    00:00:00 ps
[xzy@ecs-333953 date16]$ 
用下图的时间轴来表示事件的发生次序。其中时间从左到右。shell由表示为bash的方块代表，它随着时间的流逝从左向右移动。shell从用户读入字符串"ls"。shell建立一个新的进程，然后在那个进程中运行ls程序并等待那个进程结束。

然后shell读取新的一行输入，建立一个新的进程，在这个进程中运行程序并等待这个进程结束，所以要写一个shell，需要循环以下过程：

获取命令行
解析命令行
建立一个子进程（fork）
替换子进程（execvp）
等待子进程退出（wait）

根据这些思路，和我们前面学习的技术，就可以自己来实现一个shell了。

代码演示

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

#define SEP " "//分隔符
#define NUM 1024
#define SIZE 128

char command_line[NUM];
char* command_args[SIZE];
char env_buffer[NUM];//for test

//对应上层的内建命令
int ChangeDir(const char* new_path)
{
	chdir(new_path);
	return 0;//调用成功
}

void PutEnvInMyShell(char* new_env)
{
	putenv(new_env);
}
int main()
{
	//shell本质就是一个死循环
	while (1)
	{
		//1、显示提示符
		printf("[张三@我的主机名 当前目录]# ");
		fflush(stdout);

		//2、获取用户输入
		memset(command_line, '\0', sizeof(command_line) * sizeof(char));
		fgets(command_line, NUM, stdin);//键盘，标准输入，stdin，获取到的是C风格的字符串，'\0'结尾
		command_line[strlen(command_line) - 1] = '\0';//清空回车键的\n

		//3、"ls -a -l -i" -> "ls" "-a" "-l" "-i" 字符串切分
		command_args[0] = strtok(command_line, SEP);
		int index = 1;
		//给ls命令添加颜色
		if (strcmp(command_args[0]/*程序名*/, "ls") == 0)
		{
			command_args[index++] = (char*)"--color=auto";
		}
		// = 是故意这么写的
		// strtok截取成功，返回字符串起始地址;截取失败，返回NULL
		while (command_args[index++] = strtok(NULL, SEP));
		//for debug
	    //for (int i = 0; i < index; i++)
	    //{
	    //  printf("%d : %s\n", i, command_args[i]);
	    //}

		//4、TOOD，编写后面的逻辑，内建命令
		if (strcmp(command_args[0], "cd") == 0 && command_args[1] != NULL)
		{
			ChangeDir(command_args[1]);//让调用方进行路径切换，父进程
			continue;
		}
		if (strcmp(command_args[0], "export") == 0 && command_args[1] != NULL)
		{
			//目前，环境变量信息在command_line，会被清空
			//此处我们需要自己保存一下环境变量内容
			strcpy(env_buffer, command_args[1]);
			PutEnvInMyShell(env_buffer);//export myval=100;
			continue;
		}

		//5、创建进程，执行
		pid_t id = fork();
		if (id == 0)
		{
			//child
			//6、程序替换
			execvp(command_args[0]/*此处下标0就是保存的我们要执行的程序名字*/, command_args);
			exit(1);//执行到这里，子进程一定替换失败
		}
		int status = 0;
		pid_t ret = waitpid(id, &status, 0);
		if (ret > 0)
		{
			printf("等待子进程成功：sig: %d, code: %d\n", status & 0x7F, (status >> 8) & 0xFF);
		}
	}//end while 
	return 0;
}

运行gif动图演示：

代码分析

首先，一个shell本质就是一个死循环，所以写成while(1)死循环。接下来进入如下几大步骤：

一、显示提示符：

这里只是简易模拟，不关心属性的接口，所以用“张三@我的主机名当前目录”来简易表示，为了让输出的内容在#的后面，要加上fflush(stdout)刷新缓冲区。
//1、显示提示符
printf("[张三@我的主机名 当前目录]# ");
fflush(stdout);
二、获取用户输入

先宏定义NUM，再定义一个command_line[NUM]数组，利用mmset把数组设置为'\0'，我们平时在输入指令时或有带空格的场景（ls -a -l -i），需要将其整体当成字符串获取，用fgets来解决，注意输出指令后会按下回车键，回车也是字符，要把其去掉，否则会多输出一行，只需要把数组里最后一个字符设为'\0'即可。
#define NUM 1024
char command_line[NUM];
//2、获取用户输入
memset(command_line, '\0', sizeof(command_line) * sizeof(char));
fgets(command_line, NUM, stdin);//键盘，标准输入，stdin，获取到的是C风格的字符串，'\0'结尾
command_line[strlen(command_line) - 1] = '\0';//清空回车键的\n
三、字符串切分 + 给ls命令添加颜色

下面我们需要把获取到的一整个字符串("ls -a -l -i"）拆分成("ls" "-a" "-l" "-i"），因为后续要一个一个或以数组的形式把这一个个命令传给替换函数。这里我们利用strtok函数（把一串字符串按照分隔符来切割）来解决，先定义一个command_args指针数组，把下表0的位置利用strtok截取到以空格" '为分隔符的位置，这就让下表0为ls了，提取成功了，接下来截取剩下的，利用while语句，注意如果第一个已经截取了，那么strtok截取剩下的字符串的函数的第一个参数要设为NULL。

我们利用which指令看下正常的ls里的内容：

这里面的--color=auto就是能够让ls命令有颜色的选项，并且利用alias将其取别名为ls指令，我们需要给自己写的ls指令加上--color=auto这个选项，首先利用strcmp函数判断当command_args数组下标0程序名的位置和"ls"是一样的，那么就在数组下标1处添加--color=auto即可。
#define SEP " "//分隔符
#define SIZE 128
char* command_args[SIZE];
//3、字符串切分"ls -a -l -i" -> "ls" "-a" "-l" "-i" 
command_args[0] = strtok(command_line, SEP);
int index = 1;
//给ls命令添加颜色
if (strcmp(command_args[0]/*程序名*/, "ls") == 0)
{
	command_args[index++] = (char*)"--color=auto";
}
// = 是故意这么写的
// strtok截取成功，返回字符串起始地址;截取失败，返回NULL
while (command_args[index++] = strtok(NULL, SEP));
4、内建命令

内建命令1：

理想状态下，使用cd ..可以回退上级目录，但是就现在模拟的程序中是执行不了的：

如果exec*执行，发生了程序替换，只影响子进程，最多只是让子进程进行程序路径切换，但是子进程是一运行就完毕的进程！那么路径切换就没有意义了，所以我们在shell中更希望父进程的路径shell本身发生变化，而不是让子进程路径发生变化。
如果有些行为，是必须让父进程shell执行的，不想让子进程执行，在这样的场景下是绝对不能创建子进程。所以只能是父进程自己实现对应的代码，这种由shell自己执行的命令，我们称之为内建（内置 build -in）命令。

为了解决上述路径切换的问题，我们需要用到chdir函数，其作用是更改工作路径，哪个进程调用此函数，那么它的工作路径就要发生变化。

这里我们手写一个ChangeDir函数，内部调用系统的chdir，接下来利用strcmp函数判断当command_args[0]的位置为"cd"指令时，直接调用ChangeDir函数切换成父进程，路径就是command_args[1]，并加上continue，目的是不进行后续的创建进程。
//4、内建命令完成路径切换
if (strcmp(command_args[0], "cd") == 0 && command_args[1] != NULL)
{
	ChangeDir(command_args[1]);//让调用方进行路径切换，父进程
	continue;
}
内建命令2：

在我们之前的学习中，清楚使用export可以将变量导入环境变量，如下的hello.c程序即可帮我们演示前后的变化：

我们运行此程序，接着执行export xxx=123后，环境变量应该会多一个：

但是在我已经使用export添加xxx环境变量后，运行myshell程序，再次执行hello_env程序，并不能在我写的shell里看到添加过的环境变量：

产生此现象的原因不奇怪：

因为myshell的父进程系统中的shell，所以继承的是系统中的环境变量，我们使用的是execvp的程序替换，这也足以证明默认的不带e的程序替换是默认会把环境变量传给所有子进程的，契合了环境变量具有全局属性。当我们使用execvpe，并且手动传入环境变量后，就能看到我们导入的xxx=123环境变量了。

如果我现在就想用execvp替换函数，并且导出环境变量还要能让子进程看到，解决过程如下：

运行我自己写的myshell，利用export导入环境变量：

我们调整hello.c的代码：

为了解决上述问题，又要用到内建命令了，需要用到putenv函数，作用是把特定的环境变量导入当前进程的上下文环境中。

这里我们手写一个PutEnvInMshell的函数，内部调用putenv函数，利用strcmp函数判断当comand_args[0]的位置为export命令时，直接调用PutEnvInMshell函数，但是注意目前的环境变量信息在command_line中，会被清空，所以需要提前保存一下环境变量内容。
char env_buffer[NUM];//for test
void PutEnvInMyShell(char* new_env)
{
	putenv(new_env);
}
if (strcmp(command_args[0], "export") == 0 && command_args[1] != NULL)
{
	//目前，环境变量信息在command_line，会被清空
	//此处我们需要自己保存一下环境变量内容
	strcpy(env_buffer, command_args[1]);
	PutEnvInMyShell(env_buffer);//export myval=100;
	continue;
}
综上，我们根据环境变量的数据在进程的上下文中，总结出如下两点：

环境变量会被子进程继承下去，所以它会有全局属性
当我们进行程序替换的时候，当前进程的环境变量非但不会被替换，而且是继承父进程的

5、创建进程，执行 && 6、程序替换

这里的目的是让子进程去执行程序替换，而不是让自己本身去执行，所以要创建进程，创建的过程都是基操
程序替换函数这里要选择execvp最为方便，因为我们输入的命令行参数全部被打散到了command_args数组里头，何不利用起来呢，此外执行的程序名字就是此数组的下标0位置
//5、创建进程，执行
pid_t id = fork();
if (id == 0)
{
	//child
	//6、程序替换
	execvp(command_args[0]/*此处下标0就是保存的我们要执行的程序名字*/, command_args);
	exit(1);//执行到这里，子进程一定替换失败
}
int status = 0;
pid_t ret = waitpid(id, &status, 0);
if (ret > 0)
{
	printf("等待子进程成功：sig: %d, code: %d\n", status & 0x7F, (status >> 8) & 0xFF);
}