【Linux】自己实现一个bash进程

嘛弄

已于 2024-06-04 18:37:59 修改

阅读量987

点赞数 9

文章标签： linux bash 运维

于 2024-05-28 20:40:15 首次发布

本文链接：https://blog.csdn.net/2201_76024104/article/details/139219885

版权

bash就是命令行解释器，就是Linux操作系统让我们看到的，与用户进行交互的一种外壳（shell），当然了bash也是一个进程，它有时候就是通过创建子进程来执行我们输入的命令的。这无疑就离不开我们上篇博客所说的进程程序替换，就是让子进程去替换我们的命令进程，知道了它的原理，我们就可以试着自己写一个bash进程。

我们可以大体上对于整个过程来细分一下：

1.bash就是先打印命令行提示符，就是下面这个东西，获取用户输入的命令

命令行提示符由 [用户名@主机名路径]# 这些构成

2.对于用户输入的字符串进行分割成命令

3.执行这个命令

其实大体上就是这三步，下面还有很多的细节问题，我们遇到再说

读入用户输入

我们目前写的这个代码的结果是

为什么打印了两个换行？我们了解一下fgets这个函数

因为fgets会读入一个换行，就是我们敲完abc后的换行，它是不应该有的，我们要给它去掉

这样就可以把换行改成0了。

因为bash就是一只在等待用户输入指令，所以我们要把我们的程序写成一个循环。

还有一个问题，就是当我们什么都不输入，直接回车时，我们其实就没必要在执行下边了，直接回到循环最开始就可以，于是我们可以做一个判断就是如果输入的命令的长度为0，那么就直接回到循环最开始处，这里的长度可以让Interactive的返回值来给。

对字符串进行分割

下一步就是对于字符串进行分割：

我们分割的话，函数可以将收到的字符串分割后放到一个全局变量中，方便后续的使用，我们C语言有一个分割字符串的函数，叫strtok

分割完了之后，我们就可以开始执行命令了，就是让子进程执行，父进程等待。我们这里先是一些普通的需要子进程去执行的命令，因为有的命令需要父进程去执行，比如cd，需要父进程去切换路径，子进程是不行的

处理内建命令

那么截止到现在，我们的程序已经可以像bash一样处理一些命令了，但是还不够，因为有一些内建命令，就是需要父进程去执行的：

这些命令，我们要在代码被子进程执行前先行判断，如果是内建命令，那么让父进程去执行，否则再让子进程去执行

cd命令

我们这里先以cd命令为例，如果判断出来了用户就是要输入cd命令，如果后边什么都没有，那么默认是回到家目录，如果有，那就chdir到那个目录，并且不要忘记命令行提示符可是一直通过环境变量PWD来打印我们当前所在目录的，所以我们还要通过putenv把PWD环境变量改一下，它默认是会覆盖的。

export

之后我们要知道export也是一个内建命令，export的作用是给自己设置一个环境变量，如果给子进程设置，那显然是不合理的，所以我们也需要处理一下。

echo

下一个就是echo，我们的echo通常会有这么几种用法：

1.echo后什么都不加

2.后加$?表示打印最近一次进程的退出码

3.后随便打印一串字符

4.后加$环境变量，就打印环境变量的内容

处理重定向

我们应该如何处理重定向呢？

首先，在分割命令时，我们就应该判断这个命令中是否有重定向，包括输入重定向(<)，输出重定向(>)，追加重定向(>>)，我们做好标记

然后，在执行命令的时候如果有上面的标记，那么就用dup2系统调用改掉相对应的文件描述符表中的内容

我们测试输入和追加重定向好测试，测试输入重定向可以写一个代码，这个代码负责从stdin中读入并输出数据，这时我们用输入重定向就可以将stdin改成普通文件，这时就会打印出普通文件中的内容

代码汇总

下面是所有的代码，有不足的可以添加：

#define _XOPEN_SOURCE
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/wait.h>
#include<stdlib.h>
#include<sys/stat.h>
#include<fcntl.h>
#define SIZE 1024
#define MAX_ARGC 30//最大命令行字符串个数
#define SEP " "//设置分隔符为空格
char*argv[MAX_ARGC];
int lastcode;//最近一次进程退出码
enum redirection{
  No_redir,
  In_redir,
  Out_redir,
  App_redir
};
int redir=No_redir;
char redirfile[30];
void Extractfile(char*str)
{
  int cur=0;
  while(str[cur]==' ')cur++;
  strcpy(redirfile,str+cur);
}
int Interactive(char commandline[],int size)
{
  printf("[%s@%s %s]$ ",getenv("USER"),getenv("HOSTNAME"),getenv("PWD"));
  fgets(commandline,size,stdin);
  commandline[strlen(commandline)-1]='\0';
  int end=strlen(commandline)-1;
  while(end>=0)
  {
    if(commandline[end]=='>')
    {
      if(commandline[end-1]=='>')
      {
        redir=App_redir;
        commandline[end-1]='\0';
        Extractfile(commandline+end+1);
        break;
      }
      else
      {
        redir=Out_redir;
        commandline[end]='\0';
        Extractfile(commandline+end+1);
        break;
      }

    }
    else if(commandline[end]=='<')
    {
      redir=In_redir;
      commandline[end]='\0';
      Extractfile(commandline+end+1);
      break;
    }
    else end--;
  }
  return strlen(commandline);
}
void Splitstr(char commandline[])
{
  int i=0;
  argv[i++]=strtok(commandline,SEP);
  while(argv[i++]=strtok(NULL,SEP));
  if(strcmp("ls",argv[0])==0)
  {
    argv[i-1]="--color";
    argv[i]=NULL;
  }
}
void Execute()
{
  if(strcmp("ll",argv[0])==0&&!argv[1])//特别处理ll
  {
    argv[0]="ls";
    argv[1]="-l";
  }
  pid_t id=fork();
  if(id==0)
  {
    if(redir==In_redir)
    {
      int fp=open(redirfile,O_RDONLY);
      dup2(fp,0);
    }
    else if(redir==Out_redir)
    {
      int fp=open(redirfile,O_WRONLY|O_CREAT|O_TRUNC,0666);
      dup2(fp,1);
    }
    else if(redir==App_redir)
    {
      int fp=open(redirfile,O_WRONLY|O_CREAT|O_APPEND,0666);
      dup2(fp,1); 
    }
    execvp(argv[0],argv);
    printf("mybash: ");//只有进程程序替换失败才会走到下边
    for(int i=0;argv[i];i++)printf("%s ",argv[i]);
    printf(": not found your command\n");
    exit(2);
  }
  redir=No_redir;
  int status=0;
  pid_t rid=waitpid(id,&status,0);
  if(rid>0)lastcode=WEXITSTATUS(status);
}
int Bulidincmd()
{
  int ret=0;
  if(strcmp("cd",argv[0])==0)
  {
    ret=1;
    char*target=argv[1];
    if(!target)target=getenv("HOME");
    chdir(target);
    char tmp[SIZE];
    snprintf(tmp,SIZE,"PWD=%s",target);
    putenv(tmp);
  }
  else if(strcmp("export",argv[0])==0)
  {
    ret=1;
    if(argv[1])
    {
      char tmp[SIZE];
      strncpy(tmp,argv[1]+1,strlen(argv[1])-2);
      putenv(tmp);
    }
  }
  else if(strcmp("echo",argv[0])==0)
  {
    ret=1;
    if(argv[1]&&argv[1][0]=='$')
    {
      if(argv[1][1]=='?'&&!argv[2])
      {
        printf("%d\n",lastcode);
        lastcode=0;
      }
      else
      {
        char* tmp=getenv(argv[1]+1);
        if(tmp)printf("%s\n",tmp);
      }
    }
    else
    {
        for(int i=1;argv[i];i++)printf("%s ",argv[i]);
        printf("\n");
    }
  }
  lastcode=0;
  return ret;
}
int main()
{
  while(1)
  {
  char commandline[SIZE];
  //打印命令行提示符，获取用户输入的命令字符串
  int n=Interactive(commandline,SIZE);
  if(n==0)continue;
  //分割字符串成命令行参数
  Splitstr(commandline);
  //处理内建命令
  n=Bulidincmd();
  if(n==1)continue;
  //执行命令
  Execute(); 
 }
  return 0;
}

嘛弄

关注

9
点赞
踩
16

收藏

觉得还不错? 一键收藏
0
评论
【Linux】自己实现一个bash进程

还有一个问题，就是当我们什么都不输入，直接回车时，我们其实就没必要在执行下边了，直接回到循环最开始就可以，于是我们可以做一个判断就是如果输入的命令的长度为0，那么就直接回到循环最开始处，这里的长度可以让Interactive的返回值来给。为例，如果判断出来了用户就是要输入cd命令，如果后边什么都没有，那么默认是回到家目录，如果有，那就chdir到那个目录，并且不要忘记命令行提示符可是一直通过环境变量PWD来打印我们当前所在目录的，所以我们还要通过putenv把PWD。了，就是让子进程执行，父进程等待。
复制链接

扫一扫