[Linux - C] 自主Shell

已于 2024-04-13 10:04:16 修改
阅读量957 收藏 30
点赞数 23
分类专栏： Linux 文章标签： linux c语言 服务器 shell
于 2024-04-09 23:47:50 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/GenshiN__IMPACt_/article/details/137558015
版权

[Linux - C] 自主Shell

[Linux - C语言] 自主Shell

逻辑策划 main()

先打印命令行，然后再获取用户字符串，并且判断字符串是否有效，在进行重定向检查，然后再切割字符串。将整行的命令切割为命令与命令选项，在判断是否为内键命令，如果是内建命令则不需要利用子进程替换，如果不是内建命令则利用子进程替换执行命令。
main函数代码如下：

int main()
{
    while(1)
    {
        //命令行
        MakeCommandLineAndPrint();
        
        //获取用户字符串
        int n = GetUserCommand();
        if(n<=0) continue;

        //检查重定向
        CheckRedir();

        //切割字符
        SplitCommand();
        
        //内建命令
        n = CheckBuildin();
        if(n) continue;

        //执行命令
        ExecuteCommand();
        filename = NULL;
    }
    return 0;
}

打印命令行 void MakeCommandLineAndPrint()

在Linux系统中的命令行中，包括了用户名USER，主机名HOSTNAME，当前目录PWD，还有身份标识符，如下图所示：
在这里插入图片描述
下面就让我们来一一实现吧！

用户名 USER

获取用户名，我们需要从环境变量中获取USER，这里需要使用char* getenv(const char* name)函数，参数name代表我们要查询的变量，查询后，如果成功找到了返回该环境变量，否则返回NULL。
代码参考如下：

const char* GetUserName()
{
	const char* UserName = getenv("USER");
	if(UserName == NULL) return "None";
	return UserName;
}
主机名 HOSTNAME

同理。

const char* GetHostName()
{
    const char* name = getenv("HOSTNAME");
    if(name==NULL) return "None";
    return name;
}
当前目录 PWD
const char* GetCwd()
{
    const char* cwd = getenv("PWD");
    if(cwd==NULL) return "None";
    return cwd;
}
SkipPath 切割目录

因为我们PWD获取的环境变量，是由用户的家目录到当前目录的绝对地址，而命令行中只是当前目录，所以我们需要对目录进行切割。
因为代码量较小，所以我们利用宏函数，用do - while结构对函数内部进行封装，利用尾指针向右移的方法切割。

#define SkipPath(p) do{ p+=strlen(p)-1,while(*p != '/') p++; }while(0)

例如：”/home/Ang/Test“ 切割后为 “/Test”

打印命令行

利用前面的函数再将其组合即可。
需要注意的是用户标识符，在bash进程的命令行中 ‘$’ 代表普通用户，'#'代表超级用户。

我们这里为了和bash的命令行区分，我们就用 ‘>’ 代表普通用户，'->'代表超级用户。
如果返回的name，为root就表示为超级用户，所以在这进行判断即可。

还需要注意的是如果我们当前位置为根目录，即honstname = "/"的时候，
如果直接使用cwd+1，就没有字符可打印了，而在根目录下命令行的此时这个位置为"/"，所以此时直接打印"/"。

void MakeCommandLineAndPrint()
{
    const char* name = GetUsrName();
    const char* hostname = GetHostName();
    const char* cwd = GetCwd();
    
    SkipPath(cwd);
    const char* ManageSign = "->";
	const char* CommonSign = ">";

    printf("[%s@%s %s]%s ",name,hostname,cwd[1]=='\0'?"/":cwd+1,
    !strcmp("root",name)?ManageSign:CommonSign);
    fflush(stdout);
}

获取用户字符串 int GetUserCommand()

在定义这个函数之前，我们现在全局定义一个字符数组，用于储存我们输入的命令，如下：

#define NUM 500
char* command[NUM];

然后利用fgets函数获取用户的行输入，如果fgets读取失败返回NULL，所以这里要进行判断。
还需要注意的是'\n'也会被读取的command中，而此时的’\n’会影响后续的命令执行，所以要将尾部的'\n'改为'\0'，再返回有效地输入个数。
而在main函数中根据读取的状态来确定是否继续执行下一步。
代码如下：

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

检查重定向 void CheckRedir()

重定向的本质：是在内核中改变文件描述符表特定下标的内容，如下：
在这里插入图片描述
在定义这个函数之前，我们现在全局定义几个常量来表示重定向的状态，再定义一个全局的字符串指针用来指向command中的目标文件，再定义一个int变量来表示当前命令重定向的状态，如下：

#define None_Redir 0 //没有重定向
#define In_Redir   1 //输入重定向
#define Out_Refir  2 //输出重定向
#define App_Redir  3 //追加重定向

int redir_type = None_Redir;
char* filename = NULL;

在此之前，我们还需要写一个宏函数，功能是：当我们找到重定向符号后，让filename指向后面的文件名称.

#define SkipSpace(cmd, pos) do{\
	while(1)\
	{\
		if(isspace(cmd[pos])) pos++;\
		else break;\
	}\
}while(0)

在实现了上述功能之后，这个函数的编写变得异常简单了。只需要再确认重定向符号后，将filename指向正确的文件名，并且将redir_type变为正确的重定向状态即可，需要注意的是在检测到重定向符号后，要将其改为'\0'，因为后边的内容已经不是指令了。

void CheckRedir()
{
    int pos = 0;
    int n = strlen(command);
    while(pos < n)
    {
        if(command[pos]=='>')
        {
            if(command[pos+1]=='>')
            {
                redir_type = App_Redir;
                command[pos++]='\0';
                pos++;
                SkipSpace(command,pos);
                filename = command + pos;
            }
            else 
            {
                command[pos++]='\0';
                redir_type = Out_Redir;
                SkipSpace(command,pos);
                filename = command + pos;
            }
        }
        else if(command[pos]=='<')
        {
            command[pos++]='\0';
            redir_type = In_Redir;
            SkipSpace(command,pos);
            filename = command + pos;
        }
        else pos++;
    }
}

切割字符 void SplitCommand()

因为我们输入的一行指令中间是由空格所分开的，在切割字符的时候，利用strtok函数进行切割，所以我们需要定义一个全局变量表示只有一个空格的字符串。
我们在后续利用execvp进行程序替换，所以我们需要一个全局的可以存放指令的参数。

#define SEP " "
char* gArgv[NUM];

利用strtok进行切割即可。

void SplitCommand()
{
	int index = 0;
	gArgv[index++] = strtok(command, SEP);
	while(gArgv[index++] =strtok(NULL, SEP));
}

内建命令 bool CheckBuildin()

内建命令不需要子进程来执行，是由bash进程直接执行的命令，所以这一类的命令不能利用子进程替换，实现需要自己编写。

直接利用strcmp来判断是否为内建命令，如果是实现相应功能并返回1，表示为内建命令。
代码如下：

bool CheckBuildin()
{
    int yes = 0;
    if(strcmp("cd",gArgv[0]) == 0)
    {
        yes = 1;
        Cd();
    }
    else if(!strcmp("echo",gArgv[0])&&!strcmp(gArgv[1],"$?"))
    {
        yes=1;
        printf("%d",lastcode);
        lastcode=0;
    }
    return yes;
}
Cd

利用chird函数来变换当前目录，再定义一个temp来存放当前目录（此时目录已经变了），再利用snprintf重写cwd，使其符合putenv的参数标准，再让putenv感谢环境变量即可。
代码如下：

void Cd()
{
    const char* path = gArgv[1];
    if(path == NULL) path = GetHome();
    chdir(path);

    char temp[SIZE*2];
    getcwd(temp, sizeof(temp));
    snprintf(cwd,sizeof(cwd),"PWD=%s",temp);
    putenv(cwd);
}

执行命令 void ExecuteCommand()

利用fork创建子进程，利用返回值id判断进程，如果id<0，进程就可以直接去死了，如果id=0说明是子进程，如果id>0，说明是父进程。

在子进程中，我们先用filename判断命令是否使用了重定向，如果使用了则打开相关文件，利用dup2函数，更改相关文件操作符下标，然后再执行程序替换即可。如果替换失败，利用exit退出并报告错误。

在父进程中，我们需要用waitpid等待子进程。等待子进程结束后，利用waitpid的返回值判断等待是否成功，如果返回值为负数，说明waitpid调用失败，可能是子进程不存在。再利用输出型参数status。判断子进程的运行状态，用lastcode进行更新 ，如果lastcode不为零，说明子进程有异常，打印错误即可。

void ExecuteCommand()
{
    pid_t id = fork();
    if(id < 0) Die();
    if(id == 0)
    {
        if(filename != NULL)
        {
            if(redir_type == In_Redir)
            {
                int fd = open(filename,O_RDONLY);
                dup2(fd,0);
            }
            else if(redir_type == Out_Redir)
            {
                int fd = open(filename,O_WRONLY|O_CREAT|O_TRUNC,0666);
                dup2(fd,1);
            }
            else if(redir_type == App_Redir)
            {
                int fd = open(filename,O_WRONLY|O_CREAT|O_APPEND,0666);
                dup2(fd,1);
            }
            else {}
        }
        execvp(gArgv[0],gArgv);
        exit(errno);
    }
    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);
            }
        }
    }
}

ps: lastcode是一个全局变量，用来记录子进程的退出状态，配合echo内建命令。

源代码

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <sys/wait.h>
#include <errno.h>
#include <stdbool.h>
#include <ctype.h>

#define NUM 1000
#define SIZE 64
#define SkipPath(p) do{ p+=strlen(p)-1; while(*p!='/') p--;  }while(0)
#define SkipSpace(cmd, pos) do {\
    while(1)\
    {\
        if(isspace(cmd[pos]))\
            pos++;\
        else break;\
    }\
}while(0)

#define None_Redir 0
#define In_Redir   1
#define Out_Redir  2
#define App_Redir  3

char cwd[SIZE*2];
char* gArgv[NUM];
char command[NUM];
char* filename = NULL; 
const char* SEP =" ";


int redir_type = 0;
int lastcode = 0;

const char* GetCwd()
{
    const char* cwd = getenv("PWD");
    if(cwd==NULL) return "None";
    return cwd;
}

const char* GetUsrName()
{
    const char* name = getenv("USER");
    if(name==NULL) return "None";
    return name;
}

const char* GetHostName()
{
    const char* name = getenv("HOSTNAME");
    if(name==NULL) return "None";
    return name;
}

void MakeCommandLineAndPrint()
{
    const char* name = GetUsrName();
    const char* hostname = GetHostName();
    const char* cwd = GetCwd();
    
    SkipPath(cwd);
    const char* ManageSign = "->";
	const char* CommonSign = ">";

    printf("[%s@%s %s]%s ",name,hostname,cwd[1]=='\0'?"/":cwd+1,
    !strcmp("root",name)?ManageSign:CommonSign);
    fflush(stdout);
}

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

void SplitCommand()
{
    int index = 0;
    gArgv[index++] = strtok(command,SEP);
    while(gArgv[index++] = strtok(NULL,SEP));
}

void Die()
{
    exit(1);
}

void ExecuteCommand()
{
    pid_t id = fork();
    if(id < 0) Die();
    if(id == 0)
    {
        if(filename != NULL)
        {
            if(redir_type == In_Redir)
            {
                int fd = open(filename,O_RDONLY);
                dup2(fd,0);
            }
            else if(redir_type == Out_Redir)
            {
                int fd = open(filename,O_WRONLY|O_CREAT|O_TRUNC,0666);
                dup2(fd,1);
            }
            else if(redir_type == App_Redir)
            {
                int fd = open(filename,O_WRONLY|O_CREAT|O_APPEND,0666);
                dup2(fd,1);
            }
            else {}
        }
        execvp(gArgv[0],gArgv);
        exit(errno);
    }
    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);
            }
        }
    }
}

const char* GetHome()
{
    const char* home = getenv("HOME");
    if(home== NULL) return "/";
    return home;
}

void Cd()
{
    const char* path = gArgv[1];
    if(path == NULL) path = GetHome();
    chdir(path);

    char temp[SIZE*2];
    getcwd(temp, sizeof(temp));
    snprintf(cwd,sizeof(cwd),"PWD=%s",temp);
    putenv(cwd);

}

bool CheckBuildin()
{
    int yes = 0;
    if(strcmp("cd",gArgv[0]) == 0)
    {
        yes = 1;
        Cd();
    }
    else if(!strcmp("echo",gArgv[0])&&!strcmp(gArgv[1],"$?"))
    {
        yes=1;
        printf("%d",lastcode);
        lastcode=0;
    }
    return yes;
}

void CheckRedir()
{
    int pos = 0;
    int n = strlen(command);
    while(pos < n)
    {
        if(command[pos]=='>')
        {
            if(command[pos+1]=='>')
            {
                redir_type = App_Redir;
                command[pos++]='\0';
                pos++;
                SkipSpace(command,pos);
                filename = command + pos;
            }
            else 
            {
                command[pos++]='\0';
                redir_type = Out_Redir;
                SkipSpace(command,pos);
                filename = command + pos;
            }
        }
        else if(command[pos]=='<')
        {
            command[pos++]='\0';
            redir_type = In_Redir;
            SkipSpace(command,pos);
            filename = command + pos;
        }
        else pos++;
    }
}

int main()
{
    while(1)
    {
        //命令行
        MakeCommandLineAndPrint();
        
        //获取用户字符串
        int n = GetUserCommand();
        if(n<=0) continue;

        //检查重定向
        CheckRedir();

        //切割字符
        SplitCommand();
        
        //内建命令
        n = CheckBuildin();
        if(n) continue;

        //执行命令
        ExecuteCommand();
        filename = NULL;
    }
    return 0;
}
结语

以上就是本期的全部内容了，喜欢就多多关注吧！！！
下期会继续完善的捏！
请添加图片描述

蒋志昂
关注
  • 23
    点赞
  • 踩
  • 30
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
专栏目录
linux命令中-c代表什么,linux – 命令“ls -l / dev / tty”结果中’c’的含义是什么？...
weixin_33158374的博客
05-01 2656
它是基于字符的(而不是基于块的)设备文件.基于块的设备是在(令人惊讶的)块中传输数据有意义的任何事物.通过这个,我的意思是像磁盘.基于字符的设备(同样,这应该不足为奇)倾向于一次传输字符.诸如终端,串口,打印机等等.如果您正在运行一个不错的Linux发行版,那么可以使用以下命令获取该信息(加上您可能需要的信息)：info ls其中包含这个小片段：The file type is one of th...
Linux_C_常用命令
秋阳2018的博客
03-08 448
安装与登陆：login：登陆系统；shutdown [-option]：关闭计算机，root用户可使用；            -k：不真正关机，向每位登陆者发送关机警告；            -h：关机后关闭电源；            -c:[解释讯息]：取消目前正在执行的关机程序，并向每位登陆者发送解释讯息；            -time：设定关机前的时间；halt [-option]...
Linux-C语言程序的编译、链接
qq_61167424的博客
01-04 1288
详细介绍了翻译的过程，包括预处理、编译、汇编、链接的过程，还包括效果的演示
Linux-C之shell常用命令
wyl的博客
08-13 2117
linux中常用的shell命令
linux-C语言内存管理精讲
最新发布
m0_74166341的博客
06-29 1100
任何一个程序，正常运行都需要内存资源，用来存放诸如变量、常量、函数代码等等。这些不同的内容，所存储的内存区域是不同的，且不同的区域有不同的特性。因此我们需要研究内存布局，逐个了解不同内存区域的特性。每个C语言进程都拥有一片结构相同的虚拟内存，例如三个完全不相干的进程p1、p2、p3，它们很显然会占据不同区段的物理内存，但经过系统的变换和映射，它们的虚拟内存的布局是完全一样的。PM：Physical Memory，物理内存。VM：Virtual Memory，虚拟内存。
Shell外壳运行.rar
04-05
在Linux、Unix以及类Unix系统中，我们常见的Shell有Bash（Bourne Again SHell）、Sh、Csh、Zsh等；而在Windows系统中，CMD.exe和PowerShell是常见的Shell环境。本文将主要围绕"Shell外壳运行"这一主题，深入探讨其...
大二下Linux全部实验报告
07-10
在最后一个实验中，学生将进入更高级的主题，学习如何使用C语言进行系统调用，例如打开文件、读写数据、创建进程等。这将帮助他们理解Linux内核与应用程序之间的交互。 通过这六个实验，学生们不仅能够熟练掌握...
MOOS-ivp学习教程，从头开始教如何进行学习MOOS编程进行C++/C语言的开发
02-25
MOOS-IVP（Mobile Outdoor Operating System - Intelligent Vehicle Platform）是一种开源软件框架，主要用于自主机器人和无人驾驶系统的控制与通信。这个教程将引导你通过使用Ubuntu操作系统来学习MOOS-IVP，掌握...
Linux操作系统实践-课程设计任务书.pdf
12-06
课程设计的目标是深化学生对Linux基础命令的理解，增强他们在实际环境中编写Shell脚本和系统级程序的能力，同时提高C/C++编程技能和文档编写技巧。 课程设计的主要内容分为两部分：一是根据教师分配的Shell编程和...
基于web的“linux操作系统”课程演练平台设计与应用.pdf
09-06
该系统允许学生通过浏览器访问教学网站，使用Web终端登录到Linux服务器进行实际操作，从而实现远程的Linux操作练习，包括Shell命令和C语言编程等。 在设计这个教学平台时，作者考虑到了大学生学习的特点，强调自主...
Linux C常用头文件及函数
雨日微风
09-04 4461
#include stdio 就是指 “standard input & output"（标准输入输出）,例如c语言中的 printf("%d",i); scanf("%d",&i);等函数。 #include 在使用到字符数组时需要使用. #include stdlib 头文件即standard library标准库头文件。stdlib.h里面定义了五种类型、一些宏和
LINUX-C成长之路（一）：LINUX下C编程概要
林世霖的技术博客
12-10 6599
由于C语言是进入LINUX开发，尤其是LINUX底层开发的必备技能，因此掌握C语言也成为很多想从事LINUX相关开发工作的朋友的一道坎，C语言较之其他编程语言，特点是简洁高效，没有过多冗余鸡肋的概念，本博客打算用连载的方式，将LINUX下的C语言编程所涉及的所有技术点一一剖析和展现，为了避免陷入教条式讲解的歧途，我将采用各个击破的方式，每一篇博客集中讨论关键点问题，让读者一目了然。 打算
linux下编写C/C++代码须知———串讲
DavidZh
05-24 1万+
一、C/C++的编辑编译与链接 敲代码的过程就是编辑过程，在编辑完成程序后需要经过编译和链接两个步骤才可以把程序变成我们所需要的可执行程序。 编译过程(包括预处理、编译，汇编（test.s到test.o）三个过程）最终在linux系统上生成.o文件，在windows系统上生成.obj文件。这个过程会将.cpp文件中包含的头文件添加到.cpp文件中，意思大概就是用头文件里面的具体内容替代incl...
Linux-C
交汇的理想与现实的博客
08-28 1072
C基础 类型 基础类型 值不可再分解为其他类型的数据类型。 整型 有不同的进制表示形式，可分为基本整型、短整型、长整型几种。 实型（浮点型） 既小数形式或指数形式 字符型 仅储存一个字符数据的常量或变量 各类型内存空间占用大小 枚举 用enum声明枚举类型，用枚举定义的变量其取值只能是在枚举的范围内取。 构造类型 可以分解为若干个“成员”和“元素”。每一个成员可以是一个基本...
Linux-C下C/S架构实例,实现文件传输功能
bgylde的博客
11-09 3698
多线程的server以及client，支持多个客户端的访问，支持客户端从服务器下载文件
Linux-C 文件操作
热门推荐
Genven_Liang的博客
08-20 4万+
Linux-C 文件操作 一、简述 记--Linux环境下C语言编程的文件操作。 两种操作文件的方式： 1、系统I/O：系统调用接口，open(), read(), write(), lseek(), close()。是操作系统直接提供的编程接口（API）。 2、标准/IO：标准库的I/O函数，fopen(), frea...
Linux-C网络编程之select函数
YiShi's Blog
08-06 4238
开门见山，如果我们要对多个客户端连接的多个事件进行操作，首先会想到建立多个线程或进程让其去各自进行，这也是最简单的模式。 但对每一个线程或进程而言，无论连接是否有事件发生，都必须随时待命，也就是说，每一个对象都必须有一个线程或进程与之一一对应，直到对象销毁。 可想而知，当连接量规模变大后，系统需要在很多个线程或进程之间进行切换，时间与空间上的开销巨大，也就是说，这种模式下，程序能承载对象
Linux+C 入门之路
Being__的博客
07-01 6984
当初次接触嵌入式时，我在虚拟机上安装了一个CentOS 6.8的操作系统，然后看《鸟哥的Linux私房菜》基础篇一步一步地练习敲命令。当然，也开始用vim练习C语言。其实也没有什么特别的提升捷径，那时就是了解->练习->熟练->闭着眼睛都会。 如果一定要说有什么快速提升的方法，就是不要偷懒，俗话说“药不能停”就是这个道理，如果在学习的中途放下了一段时间，那么下次再敲的时候肯定会生疏，然后又得从头想
Linux C语言高级学习第一天（Linux操作系统与操作）
nan_lei的博客
07-17 6506
Linux系统&amp;C语言高级 （3+5）天 Linux系统         操作系统的发展简史（命令行交互--&gt;鼠标交互--&gt;触摸屏交互--&gt;语音交互）         内核                ：7个组件         shell             shell命令                 通配符                 ...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

打赏作者

蒋志昂

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

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

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

打赏作者

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

抵扣说明：

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

余额充值