Linux 学习-模拟实现【简易版bash】

1、bash本质

在模拟实现前，先得了解 bash 的本质

bash 也是一个进程，并且是不断运行中的进程
证明：常显示的命令输入提示符就是 bash 不断打印输出的结果

输入指令后，bash 会创建子进程，并进行程序替换
证明：运行自己写的程序后，可以看到当前进程的 父进程 为 bash

 此时可以断定神秘的 bash 就是一个运行中的进程，因为进程间具有独立性，因此可以同时存在多个 bash，这也是多用户登录 Linux 可以同时使用 bash 的重要原因

系统自带的 bash 是一个庞然大物，我们只需根据其本质，实现一个简易版 bash 就行了

---

2、需求分析

bash 需要帮我们完成命令解释+程序替换的任务，因此它至少要具备以下功能：

• 接收指令（字符串）
• 对指令进行分割，构成有效信息
• 创建子进程，执行进程替换
• 子进程运行结束后，父进程回收僵尸进程
• 输入特殊指令时的处理

3、核心内容

核心内容主要为 读取、切割、替换 这三部分，逐一实现，首先从指令读取开始

 3.1、指令读取

读取指令前，首先要清楚待读取命令可能有多长

• 常见命令如 ls -a -l 长度不超过 10
• 为了避免极端情况，这里预设命令最大长度为 1024
• 使用数组进行指令存储（缓冲区）

char commandline[1024];//命令行

 考虑什么是指令？如何读取指令？

• Linux 中的大部分指令由 指令 [选项] 构成，在 指令 和 [选择] 间有空格
• 常规的 scanf 无法正常读取指令，因为空格会触发输入缓冲区刷新
• 这里主要使用 fgets 逐行读取，可以读取到空格

void interact(char* cline,int size)//输出命令行{getpwd();printf("[%s@%s%s]#  " ,getusrname(),gethostname(),pwd); char *s=fgets(cline,size,stdin);//输入指令，有可能什么也没有输入直接回车   assert(s);(void)s;//”abcd\n\0" cline[strlen(cline)-1]='\0';//原来\n，在输入的时候也会加入到字符串中;checkdir(cline);//检查重定向}

 注意： 可能存在读取失败的情况，assert 断言解决；因为 fgets 也会把最后的 '\n' 读进去，为了避免出错，手动置为 '\0';

3.2、指令分割

获得指令后，就需要将指令进行分割

 为何要分割指令？

• 程序替换时，需要使用 argv 表，这张表由 指令、选项、NULL 构成
• 利用指令间的空格进行分割

如何分割指令？

• C语言 提供了字符串分割函数 strtok，可以直接使用
• 当然也可以手动实现分割

指令分割后呢？

• 将分割好的指令段，依次存入 argv 表中，供后续程序替换使用
• argv 表实际为一个指针数组，可以存储字符串

如 command 一样，表 argv 也需要考虑大小，这里设置为 64，实际使用时也就分割为四五个指令段

strtok 是 C 语言中的一个字符串处理函数，用于将一个字符串分割成多个子字符串（tokens）。该函数定义在 string.h 头文件中。strtok 通常用于解析由分隔符（如空格、逗号等）分隔的字符串。

函数原型：

char *strtok(char *str, const char *delim);

参数说明：

• str：要分割的字符串。在第一次调用时，传入需要分割的原始字符串，之后的调用则传入 NULL，以继续分割上次 strtok 返回的部分。

• delim：一个包含所有分隔符字符的字符串。例如，如果分隔符是空格和逗号，delim 可以是 " ,"。

返回值：

• 成功：返回指向分割出的子字符串的指针（tokens）。子字符串会从原始字符串中分割出来，并且这个分割后的子字符串是原始字符串的一部分，它们将共享内存空间。

• 失败：如果没有更多的子字符串可供提取，strtok 返回 NULL。

使用说明：

1. 第一次调用：传入待分割的字符串。

2. 后续调用：每次调用时，传入 NULL 以继续分割上次 strtok 返回的部分，直到没有更多的子字符串为止（返回 NULL）。

#define DEF_CHAR " "	//预设分割项，需为字符串void split(char* argv[ARGV_SIZE], char* ps)
{assert(argv && ps);//调用 C语言 中的 strtok 函数分割字符串int pos = 0;argv[pos++] = strtok(ps, DEF_CHAR);  //有空格就分割while(argv[pos++] = strtok(NULL, DEF_CHAR));  //不断分割argv[pos] = NULL; //确保安全
}

注意： 指令分割结束后，需要在添加 argv 表结尾 NULL

3.3、程序替换

获得实际可用的 argv 表后，就可以开始子进程程序替换操作了

这里使用的是函数 execvp，理由：

• v 表示 vector，正好和我们的 argv 表对应
• p 为 path，可以根据 argv[0]（指令），在 PATH 中寻找该程序并替换

当然也可以使用 execve 系统级替换函数

//子进程进行程序替换
pid_t id = fork();
if(id == 0)
{//直接执行程序替换，这里使用 execvpexecvp(argv[0], argv);exit(168); //替换失败后返回
}

注意： 程序替换成功后，exit(168) 语句不会执行. 

4、特殊情况处理

对特殊情况进行处理，使 myBash 更加完善

4.1、ls 显示高亮

系统中的 bash 在面对 ls 等文件显示指令时，不仅会显示内容，还会将特殊文件做颜色高亮处理，比如在我的环境下，可执行文件显示为绿色

实现原理

• 在指令结尾加上 --color=auto 语句，即可实现高亮处理这个问题很简单，在指令分割结束后，判断是否为 ls，如果是，就在 argv 表后尾插入语句 --color=auto 即可

//特殊处理
//颜色高亮处理，识别是否为 ls 指令
if(strcmp(argv[0], "ls") == 0)
{int pos = 0;while(argv[pos++]); //找到尾argv[pos - 1] = (char*)"--color=auto"; //添加此字段argv[pos] = NULL; //结新尾
}

 注意：

• 因为 argv 表中的元素类型为 char*，所以在尾插语句时，需要进行类型转换
• 尾插语句后，需要再次添加结尾，确保安全

4.2、内建命令

内建命令是比较特殊的命令，不同于普通命令直接进行程序替换，内建命令需要进行特殊处理，比如 cd 命令调用系统级接口 chdir 让 父进程(myBash) 进行目录间的移动

 5.3、cd

首先实现不同目录间的切换

切换的本质：令当前 bash 移动至另一个目录下，不能直接使用 子进程 ，因为需要移动的是 父进程(bash)

对于当前的 myBash 来说，cd 没有丝毫效果，因为此时 指令会被拆分后交给子进程处理，这个方向本身就是错误的

 特殊情况特殊处理，同 ls 高亮一样，对指令进行识别，如果识别到 cd 命令，就直接调用 chdir 函数令当前进程 myBash 移动至指定目录即可（不必再创建子进程进行替换）

//目录间移动处理
if(strcmp(argv[0], "cd") == 0)
{//直接调用接口，然后 continue 不再执行后续代码if(strcmp(argv[1], "~") == 0)chdir("/home");  //回到家目录else if(strcmp(argv[1], "-") == 0)chdir(getenv("OLDPWD"));else if(argv[1])chdir(argv[1]);  //argv[1] 中就是路径continue;  //终止此次循环
}

4.3、export

当添加环境变量时，环境变量具有全局属性，需要持久存在，所以要定义一个全局的数组存储环境变量的值。myenv 是一个全局的数组。

strcpy(myenv[count],_argv[1]);
putenv(myenv[count++]);

4.4、重定向

 重定向的本质：关闭默认输出/输入流，打开新的文件流，从其中写入/读取数据

重定向的三种情况：

• echo 字符串 > 文件 向文件中写入数据，写入前会先清空内容
• echo 字符串 >> 文件 向文件中追加数据，追加前不会先清空内容
• 可执行程序 < 文件 从文件中读取数据给可执行程序

所以实现重定向的关键在于判断指令中是否含有 >、>>、< 这三个字符，如果有，就具体问题具体分析，完成重定向

具体实现步骤：

• 判断字符串中是否含有目标字符，如果有，就置当前位置为 '\0‘，其后半部分不参与指令分割 
• 后半部分就是文件名，在打开文件时需要使用
• 根据不同的字符，设置不同的标记位，用于判断打开文件的方式（只写、追加、只读）
• 判断是否需要进行重定向，如果需要，在子进程创建后，打开目标文件，并调用 dup2 函数进行标准流的替换

open 函数的打开选项

O_RDONLY	//只读
O_WRONLY | O_CREAT | O_TRUNC	//只写
O_WRONLY | O_CREAT | O_APPEND	//追加

 标准流交换函数 dup2

//给参数1传打开文件后的文件描述符，给参数2传递待关闭的标准流
//读取：关闭0号流
//写入、追加：关闭1号流
int dup2(int oldfd, int newfd);

void checkdir(char * cmd)48 {49   char *pos =cmd;50   while(*pos)51   {52     if(*pos=='>')53     {54         if(*(pos+1)=='>')//'>>'55         {56            *(pos++)='\0';57            *(pos++)='\0';58           while(*pos==' ') pos++;59 60           rdirfilename=pos;61           rdir = APPEND_RDIR;62           break;63         }64         else //'>'65         {66          *(pos++)='\0';                                                                                                                                              67        while(*pos==' ') pos++;68        rdirfilename=pos;69        rdir=OUT_RDIR;70         }72     }73     else if(*pos=='<')74     {75        *pos='\0';76         pos++;77         while(*pos==' ') pos++;78                                                                                                                                                                      79         rdirfilename=pos;80         rdir=IN_RDIR;81          break;82     }83     else{}84 85     pos++;86    }   87   88 }

5.源码：好好理解

#include<iostream>
#include<stdlib.h>
#include<unistd.h>
#include<sys/wait.h>
#include<sys/types.h>
#include<assert.h>
#include<string.h>
#include<stdlib.h>#include <sys/stat.h>#include <fcntl.h>extern char** environ;#define  NONE -1
#define  IN_RDIR 0 //输入
#define  OUT_RDIR 1//stdout
#define  APPEND_RDIR 2//stderrchar commandline[1024];//命令行
char *argv[32];//参数表
char pwd[1024];//路径长
char myenv[10][10];//环境变量表
int count=0;
int lastcode=0;//退出码char * rdirfilename=NULL; //重定向的文件
int rdir =NONE;const char* getusrname()
{ const  char* str=getenv("USER");return str;
}const char* gethostname()
{return getenv("HOSTNAME");
}void getpwd()
{getcwd(pwd,sizeof(pwd));//是一个接口函数，将路径写到pwd里面
}void checkdir(char * cmd)
{char *pos =cmd;while(*pos){if(*pos=='>'){if(*(pos+1)=='>')//'>>'{*(pos++)='\0';*(pos++)='\0';while(*pos==' ') pos++;rdirfilename=pos;rdir = APPEND_RDIR;break;}else //'>'{*(pos++)='\0';                                                                                                                  while(*pos==' ') pos++;rdirfilename=pos;rdir=OUT_RDIR;}}else if(*pos=='<'){*pos='\0';pos++;while(*pos==' ') pos++;rdirfilename=pos;rdir=IN_RDIR;break;}else{}pos++;}   }void interact(char* cline,int size)//输出命令行
{getpwd();printf("[%s@%s%s]#  " ,getusrname(),gethostname(),pwd);char *s=fgets(cline,size,stdin);//输入指令，有可能什么也没有输入直接回车assert(s);(void)s;//”abcd\n\0" cline[strlen(cline)-1]='\0';//原来\n，在输入的时候也会加入到字符串中;checkdir(cline);//检查重定向
}int splitstring(char * cline,char *_argv[])
{int i=0;argv[i++]=strtok(cline," ");//字符串分割while(_argv[i++]=strtok(NULL," "));//如果截取失败就会返回NULL,正好是参数表尾;return i-1; //含回指令的参数个数。NULL不算
}int buildCommand(char*_argv[],int _argc)
{if(_argc==2&&strcmp(_argv[0],"cd")==0){chdir(argv[1]);//改变当前进程的路径，但是并不影响环境变量当中的路径getpwd();sprintf(getenv("PWD"),"%s",pwd);return 1;}else if(_argc==2&&strcmp(_argv[0],"export")==0){strcpy(myenv[count],_argv[1]);putenv(myenv[count++]);return 1;}else if(_argc==2&&strcmp(_argv[0],"echo")==0){if(strcmp(_argv[1],"$?")==0){printf("%d\n",lastcode);lastcode=0;//查看完后置为0;}else if(*_argv[1]=='$'){char* val=getenv(_argv[1]+1);if(val) printf("%s\n",val);}else printf("%s\n",_argv[1]);return 1;}if(strcmp(_argv[0],"ls")==0){ _argv[_argc++]="--color=auto";_argv[_argc]=NULL;// return 1;}return 0;
}void NormalExcute(char* _argv[]){pid_t id=fork();if(id<0){perror("fork");return ;}else if(id==0){int fd=0;if(rdir==IN_RDIR){fd=open(rdirfilename,O_RDONLY);dup2(fd ,0);}else if(rdir==OUT_RDIR){fd=open(rdirfilename,O_CREAT|O_WRONLY|O_TRUNC,0666);dup2(fd,1);}else if(rdir==APPEND_RDIR){fd=open(rdirfilename, O_CREAT|O_WRONLY|O_APPEND,0666);dup2(fd,1);}execvp(_argv[0],_argv);exit(1);}else {int status=0;pid_t rid=waitpid(id,&status,0);//正常含回子进程的PIDif(rid==id){lastcode=WEXITSTATUS(status);}}
}int main()
{while(1){rdirfilename=NULL;rdir=NONE;    interact(commandline,sizeof(commandline));int argc=splitstring(commandline,argv);if(argc==0) continue;//for(int i=0;i<argc;i++) printf("argv[%d]:%s\n",i,argv[i]);int n=buildCommand(argv,argc);if(!n) NormalExcute(argv);}return 0;
}