【Linux】自制shell

已于 2023-09-13 08:03:32 修改
阅读量513 收藏 2
点赞数
分类专栏： Linux 文章标签： linux 运维 服务器
于 2023-09-12 19:06:02 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_70811813/article/details/132818821
版权

本期我们利用之前学过的知识，写一个shell命令行程序

目录

一、初始代码

二、使用户输入的ls指令带有颜色分类

三、解决cd指令后用户所在路径不变化问题

3.1 chdir函数

四、关于环境变量的问题

 

一、初始代码

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

#define MAX 1024    
#define ARGc 512    

void shift_commend(char** s, char* c, int n)
{
    assert(s);
    assert(c);
    int num = 0;
    *s = c;
    ++s;
    while (num < n)
    {
        if (*(c + num) == ' ')//将每个空格字符替换为'\0',再将选项的首地址存入s
        {
            *(c + num) = '\0';
            if (*(c + num + 1) != ' ')//防止用户输入多个空字符
            {
                *s = c + 1 + num;
                ++s;
            }
        }
        ++num;
    }
}
int main()
{
    while (1)
    {
        printf("[%s@MyShell]#", getenv("USER"));
        fflush(stdout);
        char comment[MAX] = { 0 };
        char* p = fgets(comment, sizeof(comment), stdin);
        assert(p);//assert函数在release版本下会被编译器删去    
        (void)p;//防止在release版本下assert函数被删除导致p指针从未被使用而报错    
        comment[strlen(comment) - 1] = '\0';//去除fgets函数最后获取的\n字符    
        char* s[ARGc] = { NULL };
        shift_commend(s, comment, strlen(comment));//分割输入的命令行字符串,将每一个选项的地址存入指针数组s    
        pid_t id = fork();
        if (id == 0)
        {
            execvp(*s, s);
            exit(1);
        }
        int status = 0;
        waitpid(id, &status, 0);
    }
}

上述代码我们实现了一个基本的shell命令行程序，其实现基本思路为：

使用fgets函数获取用户输入的命令行，获取后切除字符串最后的换行符（用户输入指令后必须使用换行符来输入缓冲区中，而fgets并不会自动切除其换行符），然后将输入的指令进行切割（将指令与选项一个个切割开，方便传入execvp函数（例如fgets函数获取到用户输入的“ls -a -l”，经过切割会变成“ls/0-a/0-l/0”））。最后再创建子进程，用子进程调用execvp函数，传入用户指令最后实现shell命令行程序。

运行效果：af43d80d76c04fb09bede967d98cddc4.png

我们拿xshell来对比一下我们自己实现的shell命令行：

704fb0baae0a4f509308a4d20373baae.png

咦？xshell的ls指令有颜色变化，为什么我们自己实现的shell就没有呢？

fca1f4bc7b28448da2fd549a427b38c7.png

这是因为xshell下的ls指令多了一行颜色配置：--color=auto

如果我们要想自己的shell命令行的ls指令有颜色分类的话，我们只需要在切割用户输入的指令后判断其是否为ls指令，如果是，在该指针数组的最后一项加上"--color=auto"字符串即可：

 

二、使用户输入的ls指令带有颜色分类

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

#define MAX 1024    
#define ARGc 512    

void shift_commend(char** s, char* c, int n)
{
    assert(s);
    assert(c);
    int num = 0;
    *s = c;
    ++s;
    while (num < n)
    {
        if (*(c + num) == ' ')//将每个空格字符替换为'\0',再将选项的首地址存入s
        {
            *(c + num) = '\0';
            if (*(c + num + 1) != ' ')//防止用户输入多个空字符
            {
                *s = c + 1 + num;
                ++s;
            }
        }
        ++num;
    }
}
int main()
{
    while (1)
    {
        printf("[%s@MyShell]#", getenv("USER"));
        fflush(stdout);
        char comment[MAX] = { 0 };
        char* p = fgets(comment, sizeof(comment), stdin);
        assert(p);//assert函数在release版本下会被编译器删去    
        (void)p;//防止在release版本下assert函数被删除导致p指针从未被使用而报错    
        comment[strlen(comment) - 1] = '\0';//去除fgets函数最后获取的\n字符    
        char* s[ARGc] = { NULL };
        shift_commend(s, comment, strlen(comment));//分割输入的命令行字符串,将每一个选项的地址存入指针数组s
        if (strcmp(s[0], "ls") == 0)//使用户输入的ls指令带有颜色分类
        {
            int n = 0;
            while (*(s + n))
            {
                ++n;
            }
            *(s + n) = (char*)"--color=auto";
        }
        pid_t id = fork();
        if (id == 0)
        {
            execvp(*s, s);
            exit(1);
        }
        int status = 0;
        waitpid(id, &status, 0);
    }
}

运行效果： 

9f8f50e77750433a8fa66ae3d1d1b9bc.png

 

三、解决cd指令后用户所在路径不变化问题

上面代码还有一个问题，在我们使用cd指令后，再使用pwd来查看所在路径时，会发现没有任何变化：

31dca27a1aeb40a89f3c0402071815ea.png

这是因为当用户输入cd指令后，通过子进程调用的execvp函数来执行cd指令，改变的是子进程的目录路径和父进程压根没有关系！

所以当类似cd这样要改变父进程环境的指令（内建指令），就要父进程自己来执行，我们需要特殊判断做特殊处理：

 

3.1 chdir函数

chdir函数（包含在头文件unistd.h中）可以改变进程所在路径：

69b55cd89ec74fb7bbdfc19ca87cc4a4.png

我们可以向path传入想要进入的路径，这样子chdir函数就会自动帮我们改变当前进程的所处路径：

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

#define MAX 1024    
#define ARGc 512    

void shift_commend(char** s, char* c, int n)
{
    assert(s);
    assert(c);
    int num = 0;
    *s = c;
    ++s;
    while (num < n)
    {
        if (*(c + num) == ' ')//将每个空格字符替换为'\0',再将选项的首地址存入s
        {
            *(c + num) = '\0';
            if (*(c + num + 1) != ' ')//防止用户输入多个空字符
            {
                *s = c + 1 + num;
                ++s;
            }
        }
        ++num;
    }
}
int main()
{
    while (1)
    {
        printf("[%s@MyShell]#", getenv("USER"));
        fflush(stdout);
        char comment[MAX] = { 0 };
        char* p = fgets(comment, sizeof(comment), stdin);
        assert(p);//assert函数在release版本下会被编译器删去    
        (void)p;//防止在release版本下assert函数被删除导致p指针从未被使用而报错    
        comment[strlen(comment) - 1] = '\0';//去除fgets函数最后获取的\n字符    
        char* s[ARGc] = { NULL };
        shift_commend(s, comment, strlen(comment));//分割输入的命令行字符串,将每一个选项的地址存入指针数组s
        if (strcmp(s[0], "cd") == 0)//cd指令直接使用chdir函数改变父进程所在路径
        {
            if(s[1]!=NULL)
            chdir(s[1]);
            continue;
        }
        if (strcmp(s[0], "ls") == 0)//使用户输入的ls指令带有颜色分类
        {
            int n = 0;
            while (*(s + n))
            {
                ++n;
            }
            *(s + n) = (char*)"--color=auto";
        }
        pid_t id = fork();
        if (id == 0)
        {
            execvp(*s, s);
            exit(1);
        }
        int status = 0;
        waitpid(id, &status, 0);
    }
}

2d1a274b401d4bfdab9ba28cd5a2895e.png

 

四、关于环境变量的问题

在xshell中我们可以使用export导入自定义的环境变量，但是在我们自实现的代码中还没有这个功能，下面我们来实现一下：

使用我们在往期博客中介绍过导入环境变量的函数：putenv

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

#define MAX 1024
#define ARGc 512

void shift_commend(char** s, char* c, int n)
{
    assert(s);
    assert(c);
    int num = 0;
    *s = c;
    ++s;
    while (num < n)
    {
        if (*(c + num) == ' ')//将每个空格字符替换为'\0',再将选项的首地址存入s
        {
            *(c + num) = '\0';
            if (*(c + num + 1) != ' ')//防止用户输入多个空字符
            {
                *s = c + 1 + num;
                ++s;
            }
        }
        ++num;
    }
}
int main()
{
    while (1)
    {
        printf("[%s@MyShell]#", getenv("USER"));
        fflush(stdout);
        char comment[MAX] = { 0 };
        char* p = fgets(comment, sizeof(comment), stdin);
        assert(p);//assert函数在release版本下会被编译器删去    
        (void)p;//防止在release版本下assert函数被删除导致p指针从未被使用而报错    
        comment[strlen(comment) - 1] = '\0';//去除fgets函数最后获取的\n字符    
        char* s[ARGc] = { NULL };
        shift_commend(s, comment, strlen(comment));//分割输入的命令行字符串,将每一个选项的地址存入指针数组s
        if (strcmp(s[0], "cd") == 0)//cd指令直接使用chdir函数改变父进程所在路径
        {
            if(s[1]!=NULL)
                chdir(s[1]);
            continue;
        }
        if (strcmp(s[0], "export") == 0)//导入用户自定义的环境变量
        {
            if (s[1] != NULL)
                putenv(s[1]);
            continue;
        }
        if (strcmp(s[0], "ls") == 0)//使用户输入的ls指令带有颜色分类
        {
            int n = 0;
            while (*(s + n))
            {
                ++n;
            }
            *(s + n) = (char*)"--color=auto";
        }
        pid_t id = fork();
        if (id == 0)
        {
            execvp(*s, s);
            exit(1);
        }
        int status = 0;
        waitpid(id, &status, 0);
    }
}

运行结果： 

4a32bf8f5e924a41bfc4696486e3b732.png

这张图不太好看，博主仔细找过，在子进程调用env指令查找环境变量时，并没有看到导入的环境变量“Myenv=100”

这是为什么呢？子进程的环境变量不应该继承父进程的嘛？

这点没错，子进程确实基础了父进程的环境变量，我们再来仔细看看可以看到，在我们导入环境变量后，再一次调用env指令，屏幕上在最后多打印了一行空行。为什么会这样？

因为使用putenv函数导入环境变量时，该函数只是把形参所获取到的地址添加到进程中的环境变量表中了，我们仔细看看代码中存储环境变量的是一个变量s[1]，当我们下一次在调用其他指令时该地址的数据就发生了变化！也就是说自定义环境变量是由我们自己维护的，我们应该将其放在一个不会被覆盖的空间里：

下面我们改写一下代码：

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

#define MAX 1024
#define ARGc 512

void shift_commend(char** s, char* c, int n)
{
    assert(s);
    assert(c);
    int num = 0;
    *s = c;
    ++s;
    while (num < n)
    {
        if (*(c + num) == ' ')//将每个空格字符替换为'\0',再将选项的首地址存入s
        {
            *(c + num) = '\0';
            if (*(c + num + 1) != ' ')//防止用户输入多个空字符
            {
                *s = c + 1 + num;
                ++s;
            }
        }
        ++num;
    }
}
int main()
{
    char user_env[32][256];//存储自定义环境变量
    int env_num = 0;
    while (1)
    {
        printf("[%s@MyShell]#", getenv("USER"));
        fflush(stdout);
        char comment[MAX] = { 0 };
        char* p = fgets(comment, sizeof(comment), stdin);
        assert(p);//assert函数在release版本下会被编译器删去    
        (void)p;//防止在release版本下assert函数被删除导致p指针从未被使用而报错    
        comment[strlen(comment) - 1] = '\0';//去除fgets函数最后获取的\n字符    
        char* s[ARGc] = { NULL };
        shift_commend(s, comment, strlen(comment));//分割输入的命令行字符串,将每一个选项的地址存入指针数组s
        if (strcmp(s[0], "cd") == 0)//cd指令直接使用chdir函数改变父进程所在路径
        {
            if(s[1]!=NULL)
                chdir(s[1]);
            continue;
        }
        if (strcmp(s[0], "export") == 0)//导入用户自定义的环境变量
        {
            if (s[1] != NULL)
            {
                strcpy(*user_env, s[1]);//将用户输入的环境变量复制到自定义存储空间里
                putenv(user_env[env_num++]);
            }
            continue;
        }
        if (strcmp(s[0], "ls") == 0)//使用户输入的ls指令带有颜色分类
        {
            int n = 0;
            while (*(s + n))
            {
                ++n;
            }
            *(s + n) = (char*)"--color=auto";
        }
        pid_t id = fork();
        if (id == 0)
        {
            execvp(*s, s);
            exit(1);
        }
        int status = 0;
        waitpid(id, &status, 0);
    }
}

运行效果： 

a9bec864d04d4af0b1a6afc7cb181368.png

但是这还不够，我们最终是用子进程调用env来打印环境变量的，但是不排除子进程因为要进行某些操作会修改环境变量，所以最保险的方法还是自己实现一个打印父进程的环境变量的函数：

void show_env()
{
    extern char** environ;//使用environ前先声明  
    int i = 0;
    for (i; environ[i]; ++i)
    {
        printf("environ[%d]:%s\n", i, environ[i]);
    }
}

检测到env指令时调用一下该函数即可：

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

#define MAX 1024
#define ARGc 512

void shift_commend(char** s, char* c, int n)
{
    assert(s);
    assert(c);
    int num = 0;
    *s = c;
    ++s;
    while (num < n)
    {
        if (*(c + num) == ' ')//将每个空格字符替换为'\0',再将选项的首地址存入s
        {
            *(c + num) = '\0';
            if (*(c + num + 1) != ' ')//防止用户输入多个空字符
            {
                *s = c + 1 + num;
                ++s;
            }
        }
        ++num;
    }
}
void show_env()
{
    extern char** environ;//使用environ前先声明  
    int i = 0;
    for (i; environ[i]; ++i)
    {
        printf("environ[%d]:%s\n", i, environ[i]);
    }
}
int main()
{
    char user_env[32][256];//存储自定义环境变量
    int env_num = 0;
    while (1)
    {
        printf("[%s@MyShell]#", getenv("USER"));
        fflush(stdout);
        char comment[MAX] = { 0 };
        char* p = fgets(comment, sizeof(comment), stdin);
        assert(p);//assert函数在release版本下会被编译器删去    
        (void)p;//防止在release版本下assert函数被删除导致p指针从未被使用而报错    
        comment[strlen(comment) - 1] = '\0';//去除fgets函数最后获取的\n字符    
        char* s[ARGc] = { NULL };
        shift_commend(s, comment, strlen(comment));//分割输入的命令行字符串,将每一个选项的地址存入指针数组s
        if (strcmp(s[0], "cd") == 0)//cd指令直接使用chdir函数改变父进程所在路径
        {
            if(s[1]!=NULL)
                chdir(s[1]);
            continue;
        }
        if (strcmp(s[0], "export") == 0)//导入用户自定义的环境变量
        {
            if (s[1] != NULL)
            {
                strcpy(*user_env, s[1]);//将用户输入的环境变量复制到自定义存储空间里
                putenv(user_env[env_num++]);
            }
            continue;
        }
        if (strcmp(s[0], "env") == 0)//直接打印父进程环境变量
        {
            show_env();
            continue;
        }
        if (strcmp(s[0], "ls") == 0)//使用户输入的ls指令带有颜色分类
        {
            int n = 0;
            while (*(s + n))
            {
                ++n;
            }
            *(s + n) = (char*)"--color=auto";
        }
        pid_t id = fork();
        if (id == 0)
        {
            execvp(*s, s);
            exit(1);
        }
        int status = 0;
        waitpid(id, &status, 0);
    }
}

运行效果： 

820121e12ebc4808b9782aa4f688f58f.png

从上面的实现过程我们可以得出一个结论：其实我们之前学习到的几乎所有的环境变量命令，都是内建命令！

这就是本期博客的全部内容，如有纰漏还请各位大佬指点~

 

1e-12
关注
  • 0
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
Linux一学就会——编写自己的shell
weixin_43635473的博客
05-06 916
也就是说可变参数是放在传入参数最后，放在中间必须在输入结束之后再输入一个NULL，而且可变参数和前面放的参数类型一致。
【Linux】实现一个专属定制的简单shell命令行
sea18323的博客
04-26 356
本文主要介绍如何实现一个简单shell命令行
极简shell制作
最新发布
欢迎来到阿熊的blog~~
04-28 1871
通过我们之前所学Linux知识以及C语言的知识，到目前为止，我们完全可以独立完成简易shell的制作，那么话不多说，开始今天的话题！
Linux下自定义shell（简单shell）
m0_63037816的博客
05-10 1639
一.原理 咱们首先运用fgets函数来获取stdin输入流，即获取你要输入的命令，然后把你输入的命令进行解析，不过在解析命令前要先把命令进行组织起来，这是咱们可以通过数组来把命令进行组织（下文会说明为什么要进行组织），然后就是解析，在Linux中咱们输入命令的过程是不会终止的，即当你输入一个命令并按下回车键后，咱们还可以接着输入命令，那为了实现这个功能，咱们可以通过循环语句还有进程的特性（独立性）来进行持续性解析命令，当咱们把输入的命令组织到一个数组中后（如args），咱们可以通过exec函数这个系统调用
linux 自定义shell脚本
zhaoyang10的专栏
05-17 672
目录备份文件脚本函数用法输入交互用法if条件用法数值计算用法系统参数用法发送消息脚本 备份文件脚本 daily_archive.sh #!/bin/bash # 首先判断输入参数个数是否为1 if [ $# -ne 1 ] then echo "参数个数错误！应该输入一个参数，作为归档目录名" exit fi # 从参数中获取目录名称 if [ -d $1 ] then echo else echo echo "目录不存在！" echo exit fi DIR_NAME=$(basena
使用shell封装Linux命令实现自定义Linux命令
cz1803472613的博客
04-18 1584
在日常工作中，尤其是在Linux上做开发的同学或者运维的同学们肯定会遇到过如下场景，比如在Linxu下通过find查找一个文件并且想看这个文件的详细信息，如果直接使用命令可能会一时想不起来，或者想起来了但是有个别参数忘记了。或者一些很频繁的操作，但是需要脚本来完成。于是一个目录下出现了很多脚本，真正用的时候又忘记了哪个脚本是干什么的。导致工作效率低下。
操作系统课设--自制shell
10-11
操作系统的课程设计，自己用C写的一个shell程序
自制ssh连接管理脚本 linux
09-06
自制ssh连接管理脚本 linux
自制Linux终端锁屏工具
01-10
 •Shell语言 ： 我使用的是默认的Bash Shell  •其他小工具 :   ◦fortune：系统随机的从语库中选出一句英文成语。  ◦cowsay : 在终端界面上显示出一个奶牛的语句框，配合管道连接上fortune，效果完美！  ...
自己写的shell源码
07-14
这是我自制的linux shell 想学shell的可以看一下，里面包含了许多有关linux命令，目录切换，多进程的操作。。。
Linux 重命名命令自制详细介绍
01-10
Linux 重命名命令 相比于Windows上的ren命名，Linux还真的是没有一个特定的重命名的命令。（虽然可以间接的使用mv来实现）。下面我就来自己写一个简单的重命名命令。 准备工作 操作系统： Linux内核的系统都可以 ...
C例子：DIY shell
12-06
该程序是我写的博客“一起talk C栗子吧（第七十一回：C语言实例--DIY shell）”的配套程序，共享给大家使用
Linux Shell学习笔记：exit退出状态代码
互联网
11-14 1307
inux提供$?特殊变量来保存最后一条命令执行结束的退出状态。执行完一条命令后，立即执行echo$?，可以查看最后一条命令的退出状态值。 正常的情况下，命令成功执行完成的退出状态是0，如果非0，则命令执行有错。 该命令可以用于检查命令是否正确执行，比如在解压包的时候，检查解压包是否成功十分有效。 自定义退出状态码，可以在脚本中定义自己的退出状态代码，然后使用echo $?检查。 ...
Shell语法入门与实例——函数、exit命令
jakelihua
08-03 390
Shell是一种命令行解释器，允许用户直接与操作系统进行交互。Shell脚本由一系列的命令组成，可以自动执行任务、处理数据和控制流程。
shell内置命令
m0_56642224的博客
06-19 1157
Shell 内置命令，就是由 Bash Shell 自身提供的命令，而不是文件系统中的可执行文件。使用type 来确定一个命令是否是内置命令：从上往下看，第一行的ifconfig是可执行文件，第二行也是可执行文件，第三行的cd才是内嵌命令通常来说，内置命令会比外部命令执行得更快，执行外部命令时不但会触发磁盘 I/O，还需要 fork 出一个单独的进程来执行，执行完成后再退出。而执行内置命令相当于调用当前 Shell 进程的一个函数, 还是一个进程, 减少了上下文切换。
linux shell脚本 exit0,Bash exit命令和exit退出代码
weixin_36176546的博客
05-01 2490
通常，在编写Bash脚本时，需要在满足特定条件时终止脚本，或者根据命令的退出代码采取措施。在本文中，我们将介绍Bash exit内置命令和已执行命令的退出状态。退出状态每个shell命令成功终止或失败均返回退出代码。按照惯例，退出代码为零表示命令成功完成，而非零表示遇到错误。特殊变量$?返回最后执行的命令的退出状态：date &> /dev/nulldate命令成功完成，并且退出代码...
Linux自定义命令(shell脚本)并启动应用
hiekay的博客
05-07 2138
1. 创建Shell脚本 例子: 在 ~/mysh 目录下新建 test.sh #!/bin/bash echo "test shell" 2. 配置别名（alias） 配置只对自己生效的别名则编辑~/.bashrc文件 配置全局生效的别名则编辑/etc/profile文件 修改内容: $ sudo vim ~/.bashrc (/etc/profile) #加入内容 alias t="~...
[shell脚本]Linux自定义命令并启用应用
程序员进阶之路
03-03 8543
前言之前做过一个Java Web系统，由很多的子系统组成，由尽管其子系统颇多，但是要在服务器上管理这些子系统（启动、停止、重启等）却很方便，只需要输入tt 便能看见每个应用的名称、状态等，然后输入对应的应用名+操作代码，便能进行对应的操作，这一切都源于它使用了自定义的命令，来调用对应的shell脚本或命令。正好目前我做的一个系统，还处于起步阶段，每次部署都太麻烦，所以就做了一个类似的功能。实现这个功
linux cshell
12-16
Linux cshell（C Shell）是一种Unix操作系统的命令行解释器，它在Unix系统上提供了一种C语言风格的命令行界面。它的语法和用法与C语言非常相似，因此受到许多程序员和系统管理员的喜爱。 C Shell具有强大的脚本编程...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

1e-12

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值