【Linux】编写一个简易的shell

简介: 【Linux】编写一个简易的shell

思维导图

学习目标

      将简易的shell代码进行编写。

一、阐述shell的基本思路

      在进程程序替换中,我们可以将一个指令交给子进程,让子进程去完成这个指令。如果这个命令是一个内建命令,我们需要将这个命令交给bash进行处理。

      大致思路是:首先,我们先打印出来一行命令行,代表我们的主机名,名字和当前路径;之后捕获一行指令命令,将指令命令进行分割,存储在字符串指针数组中;然后,将这个字符串指针数组交给exec*函数进行程序替换。

二、输出一个命令行

2.1 思路和代码

         

      我们可以观察xshell中的这个命令行中的内容,我们可以仿效这个命令行中的内容打印出自己的xshell的命令行。现在我们应该思考从哪里获取这个内容呢??在环境变量中,我们可以发现有这些内容:

      我们可以使用getenv函数来分别获取USER、HOSTNAME、PWD的内容,之后使用snprintf函数将这个内容串联起来打印到一个字符串数组,以便将这个命名行打印出来。

const char* Getname()
{
  const char* name = getenv("USER");
  if(name == NULL) return "None";
  return name;
}
 
const char* Gethostname()
{
  const char* hostname = getenv("HOSTNAME");
  if(hostname == NULL) return "None";
  return hostname;
}
 
const char* Getpwd()
{
  const char* pwd = getenv("PWD");
  if(pwd == NULL) return "None";
  return pwd;
}
 
void MakeCommendLine(char commend[], size_t size)
{
  const char* name = Getname();
  const char* hostname = Gethostname();
  const char* pwd = Getpwd();
  SkipPath(pwd);
  snprintf(commend, size, "[%s@%s %s]>" , name, hostname, strlen(pwd) == 1 ? "/" : pwd + 1);
  printf("%s", commend);
  fflush(stdout);
}

2.2 简要介绍一下snprintf函数

char *getenv(const char *name)

      函数的用途:该函数返回一个以 null 结尾的字符串,该字符串为被请求环境变量的值。如果该环境变量不存在,则返回 NULL。

2.3 简要介绍一下getenv函数

int snprintf(char *str, size_t size, const char *format, ...)
  1. 如果格式化后的字符串长度 < size,则将此字符串全部复制到str中,并给其后添加一个字符串结束符('\0');
  2. 如果格式化后的字符串长度 >= size,则只将其中的(size-1)个字符复制到str中,并给其后添加一个字符串结束符('\0'),返回值为欲写入的字符串长度。
  3. snprintf的返回值n,当调用失败时,n为负数,当调用成功时,n为格式化的字符串的总长度(不包括\0),当然这个字符串有可能被截断,因为buf的长度不够放下整个字符串。

三、获取用户命令字符串

      我们在输入指令命令时,会有空格,我们不能使用scanf函数,所以我们应该使用fgets函数,将指令命令进行接收。

int getecho(char* commend, size_t n)
{
  char* s = fgets(commend, n, stdin);
  if(s == NULL) return -1;
  commend[strlen(commend) - 1] = '\0';
  return strlen(commend);
}

简要介绍一下fgets函数

char *fgets(char *restrict str, int size, FILE *restrict stream)

函数的用途:fgets函数就是用来读取一行数据的,从第三个参数指定的流中读取最多第二个参数大小的字符到第一个参数指定的容器地址中。

函数的返回值:在正常情况下fgets()函数的返回值和它第一个参数相同。即读取到数据后存储的容器地址。但是如果读取出错或读取文件时文件为空,则返回一个空指针。

函数的注意事项:fgets()函数的眼里,换行符’\n’也是它要读取的一个普通字符而已。在读取键盘输入的时候会把最后输入的回车符也存进数组里面,即会把’\n’也存进数组里面,而又由于字符串本身会是以’\0’结尾的。所以在输入字符个数没有超过第二个参数指定大小之前,你输入n个字符按下回车输入,fgets()存储进第一个参数指定内存地址的是n+2个字节。最后面会多出一个’\n’和一个’\0’,而且’\n’是在’\0’的前面一个(\n\0)。其余部分请看大佬写的:fgets函数详解

四、切割命令字符串

      在获取到输入的指令字符串后,我们需要将指令进行切割。因为指令间隔是空格,我们可以使用strtok函数进行分割指令。

char* gargv[SIZE];
 
void slashecho(char commend[], size_t n)
{
  gargv[0] = strtok(commend, SEP);
  int cnt = 1;
  while ((gargv[cnt++] = strtok(NULL, SEP))); // 故意写出赋值,
}

简要介绍一下strtok函数

char *strtok(char s[], const char *delim);

函数的用途:分解字符串为一组字符串。s为要分解的字符,delim为分隔符字符(如果传入字符串,则传入的字符串中每个字符均为分割符)。首次调用时,s指向要分解的字符串,之后再次调用要把s设成NULL。

函数的返回值:从s开头开始的一个个被分割的串。当s中的字符查找到末尾时,返回NULL。如果查找不到delim中的字符时,返回当前strtok的字符串的指针。所有delim中包含的字符都会被滤掉,并将被滤掉的地方设为一处分割的节点。

五、创建子进程进行进程替换

5.1 检查指令是否为内建命令

      比较草率,直接利用if语句进行逐一的判断,如果成功,则是内建命令;如果失败,则是普通命令。如果是内建命令,我们可以重新创建一个函数来单独的进行内建命令的执行。

int ChickBuliding()
{
  int yes = 0;
  const char* entercommend = gargv[0];
  if(strcmp(entercommend, "cd") == 0)
  {
    yes = 1;
    Cd();
  }
  else if(strcmp(entercommend, "echo") == 0 && strcmp(gargv[1], "$?") == 0)
  {
    yes = 1;
    printf("%d\n", lastcode);
    lastcode = 0;
  }
  return yes;
}

      比如,说cd命令,我们可以利用chdir函数改变当前工作目录,getcwd函数将当前工作目录的绝对路径复制到参数buffer所指的内存空间中,参数size为buf的空间大小。在将获取到的路径写入cwd中,最后利用putenv函数将cwd写入环境变量中。

void Cd()
{
  const char* path = gargv[1];
  if(path == NULL) path = Home();
  chdir(path);
  // 刷新环境变量
  char temp[SIZE * 2];
  // 获取当前路径
  getcwd(temp, sizeof temp);
  // 将当前路径写入cwd中
  snprintf(cwd, sizeof cwd, "PWD=%s", temp);
  // 将cwd写入环境变量中
  putenv(cwd);
}

      还有一个echo $?命令,直接判断是否为这个命令,如果是这个命令,直接将lastcode返回,并将lastcode重新置为0。

5.2 指令是普通命令

      我们可以创建一个子进程,利用exec*函数进行程序进程替换。最后,让父进程进行等待,如果父进程等待成功,则检查退出码是否为0,如果不为0,将错误信息打印出来。

void executecommend()
{
  pid_t id = fork();
  if(id < 0) 
  {
    Die();
  }
  else if(id == 0)
  {
    execvp(gargv[0], gargv);
    exit(1);
  }
  else 
  {
    int status = 0;
    pid_t rid = waitpid(id, &status, 0);
    if(rid > 0)
    {
      lastcode = WEXITSTATUS(status);
      if(lastcode != 0) printf("%s:%s:%d\n", gargv[0], strerror(lastcode), lastcode);
    }
  }
}
相关文章
|
7月前
|
存储 安全 Unix
七、Linux Shell 与脚本基础
别再一遍遍地敲重复的命令了,把它们写进Shell脚本,就能一键搞定。脚本本质上就是个存着一堆命令的文本文件,但要让它“活”起来,有几个关键点:文件开头最好用#!/usr/bin/env bash来指定解释器,并用chmod +x给它执行权限。执行时也有讲究:./script.sh是在一个新“房间”(子Shell)里跑,不影响你;而source script.sh是在当前“房间”里跑,适合用来加载环境变量和配置文件。
627 10
|
7月前
|
算法 Linux Shell
Linux实用技能:打包压缩、热键、Shell与权限管理
本文详解Linux打包压缩技巧、常用命令与原理,涵盖.zip与.tgz格式操作、跨系统传文件方法、Shell运行机制及权限管理,助你高效使用Linux系统。
Linux实用技能:打包压缩、热键、Shell与权限管理
|
7月前
|
存储 Shell Linux
八、Linux Shell 脚本:变量与字符串
Shell脚本里的变量就像一个个贴着标签的“箱子”。装东西(赋值)时,=两边千万不能有空格。用单引号''装进去的东西会原封不动,用双引号""则会让里面的$变量先“变身”再装箱。默认箱子只能在当前“房间”(Shell进程)用,想让隔壁房间(子进程)也能看到,就得给箱子盖个export的“出口”戳。此外,Shell还自带了$?(上条命令的成绩单)和$1(别人递进来的第一个包裹)等许多特殊箱子,非常有用。
650 2
|
9月前
|
Web App开发 缓存 安全
Linux一键清理系统垃圾:释放30GB空间的Shell脚本实战​
这篇博客介绍了一个实用的Linux系统盘清理脚本,主要功能包括: 安全权限检查和旧内核清理,保留当前使用内核 7天以上日志文件清理和系统日志压缩 浏览器缓存(Chrome/Firefox)、APT缓存、临时文件清理 智能清理Snap旧版本和Docker无用数据 提供磁盘空间使用前后对比和大文件查找功能 脚本采用交互式设计确保安全性,适合定期维护开发环境、服务器和个人电脑。文章详细解析了脚本的关键功能代码,并给出了使用建议。完整脚本已开源,用户可根据需求自定义调整清理策略。
1076 1
|
Shell Linux
Linux shell编程学习笔记30:打造彩色的选项菜单
Linux shell编程学习笔记30:打造彩色的选项菜单
|
12月前
|
Linux Shell
在Linux、CentOS7中设置shell脚本开机自启动服务
以上就是在CentOS 7中设置shell脚本开机自启动服务的全部步骤。希望这个指南能帮助你更好地管理你的Linux系统。
1593 25
|
11月前
|
Linux Shell
Centos或Linux编写一键式Shell脚本删除用户、组指导手册
Centos或Linux编写一键式Shell脚本删除用户、组指导手册
323 4
|
11月前
|
Linux Shell 数据安全/隐私保护
Centos或Linux编写一键式Shell脚本创建用户、组、目录分配权限指导手册
Centos或Linux编写一键式Shell脚本创建用户、组、目录分配权限指导手册
580 3
|
12月前
|
Linux Shell
shell_42:Linux参数移动
总的来说,参数移动是Linux shell脚本中的一个重要概念,掌握它可以帮助我们更好地处理和管理脚本中的参数。希望这个解释能帮助你理解和使用参数移动。
268 18
|
Shell Linux
【linux】Shell脚本中basename和dirname的详细用法教程
本文详细介绍了Linux Shell脚本中 `basename`和 `dirname`命令的用法,包括去除路径信息、去除后缀、批量处理文件名和路径等。同时,通过文件备份和日志文件分离的实践应用,展示了这两个命令在实际脚本中的应用场景。希望本文能帮助您更好地理解和应用 `basename`和 `dirname`命令,提高Shell脚本编写的效率和灵活性。
1236 32
下一篇
开通oss服务