[Linux]自定义shell详解

自定义shell

前言

写之前我们先看看一个完整的shell都包括了什么

$符号前面(包括这个符号)就是命令行提示符,后面就是命令行字符串了

1.命令行提示符,字符串的打印

1.1命令行提示符

命令行字符串的格式:

用户名@主机名 路径\]$

这三个其实就是环境变量,这就很简单了,getenv()获取环境变量

c 复制代码
const char *UserName()
{
    char *username = getenv("USER");
    if (username)
        return username;
    else
        return "None";
}
const char *HostName()
{
    char *hostname = getenv("HOME");
    if (hostname)
        return hostname;
    else
        return "None";
}
const char *CurrentWorkDir()
{
    char *current = getenv("PWD");
    if (current)
        return current;
    else
        return "None";
}

2.命令行字符串

这里输入命令行字符串,我们不能用scanf,当我们的命令行字符串带有空格的时候scanf就无法控制了。所以这里我们用fgets()

文件的操作下节会说,先跟着用;注意这里fgets会有一个自带的换行符,所以我们要对他做一下去换行的操作。

c 复制代码
int Interactive(char out[], int size)
{
    printf("[%s@%s %s]$ ", UserName(), HostName(), CurrentWorkDir());
    fgets(out, size, stdin);
    out[strlen(out) - 1] = '\0';
    return strlen(out);
}

2.0对命令行字符串进行切割

c 复制代码
void Split(char in[])
{
    int i=0;
    argv[i++]=strtok(in,SEP);
    while(argv[i++]=strtok(NULL,SEP));
}

2.执行命令

上个文章提到了进程切换的接口,接下面我们就要选择一个来使用,我们有了文件名,又有argv数组,所以我们用execvp

c 复制代码
void Execute()
{
    pid_t id=fork();
    if(id==0)
    {
        execvp(argv[0],argv);
        exit(1);
    }
    int status=0;
    pid_t rid=waitpid(id,&status,0);
}

3.有趣的小问题

完成了主体部分我们先测试一下代码

大部分指令都可以完成但是cd指令无法成功运行

类似于cd1这种子进程无法执行的指令我们叫做内建命令,这种指令只有bash自己才可以执行。

引入函数chdir

c 复制代码
int BuildinCmd()
{
    int ret=0;
    //如果是内建命令就是1执行,不是就是0继续向下执行
    if(strcmp("cd",argv[0])==0)
    {
        ret=1;
        char* target=argv[1];
        if(!target)
        {
            target=Home();
        }
        chdir(target);
    }
    return ret;
}

再次做测试,cd之后,我们的pwd确实改变了,但是命令行提示符并没有随之改变

再次修改


但是这里又出现了一个新的问题,我们输入...显示的就还是...我们想让他以绝对路径的形式让我们看。

c 复制代码
int BuildinCmd()
{
    int ret = 0;
    // 如果是内建命令就是1执行,不是就是0继续向下执行
    if (strcmp("cd", argv[0]) == 0)
    {
        // 2. 执行
        ret = 1;
        char *target = argv[1]; // cd XXX or cd
        if (!target)
            target = Home();
        chdir(target);
        char temp[1000];
        getcwd(temp, 1000);
        snprintf(pwd, SIZE, "PWD=%s", temp);
        putenv(pwd);
    }
    return ret;
}

这次改完就符合我们的预期了。

完整代码

c 复制代码
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>

#define SIZE 1024
#define SEP " "
#define MAX_ARGC 64

char *argv[MAX_ARGC];
char pwd[SIZE];

const char *UserName()
{
    char *username = getenv("USER");
    if (username)
        return username;
    else
        return "None";
}
const char *HostName()
{
    char *hostname = getenv("HOME");
    if (hostname)
        return hostname;
    else
        return "None";
}
const char *CurrentWorkDir()
{
    char *current = getenv("PWD");
    if (current)
        return current;
    else
        return "None";
}
char *Home()
{
    return getenv("HOME");
}
// 打印命令行提示符
int Interactive(char out[], int size)
{
    printf("[%s@%s %s]$ ", UserName(), HostName(), CurrentWorkDir());
    fgets(out, size, stdin);
    out[strlen(out) - 1] = '\0';
    return strlen(out);
}

// 分割
void Split(char in[])
{
    int i = 0;
    argv[i++] = strtok(in, SEP);
    while (argv[i++] = strtok(NULL, SEP))
        ;
}
int BuildinCmd()
{
    int ret = 0;
    // 如果是内建命令就是1执行,不是就是0继续向下执行
    if (strcmp("cd", argv[0]) == 0)
    {
        // 2. 执行
        ret = 1;
        char *target = argv[1]; // cd XXX or cd
        if (!target)
            target = Home();
        chdir(target);
        char temp[1000];
        getcwd(temp, 1000);
        snprintf(pwd, SIZE, "PWD=%s", temp);
        putenv(pwd);
    }
    return ret;
}
// 执行命令
void Execute()
{
    pid_t id = fork();
    if (id == 0)
    {
        execvp(argv[0], argv);
        exit(1);
    }
    int status = 0;
    pid_t rid = waitpid(id, &status, 0);
}
int main()
{
    while (1)
    {
        char commandline[SIZE];
        // 1. 命令行提示符[yjt@hece-265342 shell]
        int n = Interactive(commandline, SIZE);
        if (n == 0)
            continue;
        // 2.切割命令行字符串
        Split(commandline);
        // 2.0执行内建命令
        n = BuildinCmd();
        if (n)
            continue;
        // 3.执行命令
        Execute();
    }
    return 0;
}
相关推荐
小小不董8 分钟前
深入理解oracle ADG和RAC
linux·服务器·数据库·oracle·dba
AA陈超34 分钟前
虚幻引擎UE5专用服务器游戏开发-20 添加基础能力类与连招能力
c++·游戏·ue5·游戏引擎·虚幻
胚芽鞘68143 分钟前
查询依赖冲突工具maven Helper
java·数据库·maven
mit6.8241 小时前
[Meetily后端框架] AI摘要结构化 | `SummaryResponse`模型 | Pydantic库 | vs marshmallow库
c++·人工智能·后端
狄加山6751 小时前
Cadence模块复用
服务器·硬件架构·硬件工程·信号处理·智能硬件
R-G-B1 小时前
【02】MFC入门到精通——MFC 手动添加创建新的对话框模板
c++·mfc·mfc 手动添加创建新的对话框
宇钶宇夕1 小时前
SIMATIC S7-1200的以太网通信能力:协议与资源详细解析
运维·服务器·数据库·程序人生·自动化
该用户已不存在1 小时前
关于我把Mac Mini托管到机房,后续来了,还有更多玩法
服务器·前端·mac
linux kernel1 小时前
第七讲:C++中的string类
开发语言·c++