Linux 写一个自己的bash2.0版本

Linux 专栏收录该内容
23 篇文章 0 订阅

写一个自己的bash2.0版本


在上一次我们写了一个mybash程序,生成了一个自己的简单的bash。通过执行mybash可执行程序,调用命令,如ps,ls等。原理是mybash调用fork()产生一个子进程然后再替换该子进程,详细请见上一篇博客 写一个自己的bash1.0版本但是仍有一些缺陷,比如, 1、我们的命令提示符是写死的,是通过printf()函数直接打印出来的 printf(“wys@DESKTOP-2OU3HRV:~$”);那么现在来回顾一下Linux终端的命令提示符各部分的含义
在这里插入图片描述
明白了各部分含义后,在
*今天的2.0版本中首先得任务是改进命令提示符***,不要写死,让程序自己获取用户名,主机名,还有当前目录并识别当前是普通用户还是管理员用户。 2、像ps,ls,cp等等命令是在/usr/bin/目录下的,我们可以通过fork()一个子进程进而替换子进程来执行,但是还有一些命令是内置命令,比如cd ,jobs,exit等是在bin目录下找不到的,所以只能通过我们自己来实现, 今天2.0版本第二个任务就是自己简单实现cd命令
在这里插入图片描述

1、改进命令提示符

1.1获取用户名

参考Linux程序设计中文第四版 4.5用户信息
每一个用户都有一个唯一的用户标识符UID(管理员用户UID为0)。既然UID是用户身份的关键,那么我们就从UID开始吧!!!
UID有自己的类型–uid_t;
获取UID方法 uid_t getuid(void);
得到了用户UID后就可以通过UID来获取用户信息,方法:*struct passwd getpwuid(uid_t uid);
代码如下:

 30     int id = getuid();//获取用户id
 31     char* s = "$";
 32     if( id  == 0 )
 33     {
 34        s = "#";
 35     }
 36
 37     struct passwd* ptr = getpwuid(id);//通过uid获取用户详细信息
 38     if( ptr == NULL )
 39     {
 40       printf("$");
 41       fflush(stdout);
 42       return;
 43     }

1.2获取主机名

参考Linux程序设计中文第四版 4.6主机信息
通过gethostname函数获取
*int gethostname(char name,size_t namelean);
gethostname函数把机器的网络名写入name字符串。该字符串至少有namelen个字长,成功时,gethostbyname返回0,否则返回-1。
代码如下:

 45     char hostname[64] = { 0 };
 46     if (gethostname(hostname,64) == -1 )
 47     {
 48       printf("$");
 49       fflush(stdout);
 50       return;
 51     }

1.3获取当前位置

参考Linux程序设计中文第四版 3.7.5chdir系统调用和getcwd函数
程序可以通过调用getcwd函数来确定自己的当前工作目录
**#include <unistd.h>
char getcwd(char buff,size_t size);
getcwd函数把当前目录的名字写到给定的缓冲区buff里。如果目录名的长度超出了参数size的缓冲区长度(一个ERANGE错误),比如你所在的目录特别深,它就返回NULL,如果成功,它返回指针buf。
代码如下:

 54     char pwd_buff[128] = {0};
 55     if(getcwd(pwd_buff,128) == NULL)
 56     {
 57       printf("$");
 58       fflush(stdout);
 59       return;
 60     }

1.4整个void print_info()函数完整代码

 27 void print_info()
 28 {
 29
 30     int id = getuid();//获取用户id
 31     char* s = "$";
 32     if( id  == 0 )
 33     {
 34        s = "#";
 35     }
 36
 37     struct passwd* ptr = getpwuid(id);//通过uid获取用户详细信息
 38     if( ptr == NULL )
 39     {
 40       printf("$");
 41       fflush(stdout);
 42       return;
 43     }
 44
 45     char hostname[64] = { 0 };
 46     if (gethostname(hostname,64) == -1 )
 47     {
 48       printf("$");
 49       fflush(stdout);
 50       return;
 51     }
 52
 53     //获取当前位置
 54     char pwd_buff[128] = {0};
 55     if(getcwd(pwd_buff,128) == NULL)
 56     {
 57       printf("$");
 58       fflush(stdout);
 59       return;
 60     }
 61
 62     //打印命令提示行信息
 63     printf("\033[1;32;40m %s@%s:\033[0m \033[1;34;40m%s%s\033[0m",ptr->pw_name,hostname,pwd_buff,s);
 64     fflush(stdout);
 65 }

2、实现简单的cd命令操作

参考Linux程序设计中文第四版 3.7.5chdir系统调用和getcwd函数
程序可以像用户在文件系统里那样来浏览目录。就像你在shell里使用cd命令来切换目录一样,程序使用的是chdir系统调用。
#include<unistd.h>
int chdir(const char *path);

如果是cd命令,传入的第二个参数,即要切换的目录为空,则结束本次循环,再次打印出命令提示符(Linux bash中当cd后没有目录时,则切换到家目录,自己的mybash没有必要和bash相同,也可以自己设计),不为空时,使用chdir系统调用,操作成功则continue结束本次循环,重新打印当前路径。
代码如下:

 86     if (strcmp(cmd,"cd") == 0)
 87     {
 88         if( myargv[1] == NULL )
 89         {
 90             continue;
 91         }
 92         if ( chdir(myargv[1]) == -1)
 93         {
 94             perror("cd err");
 95         }
 96         continue;
 97     }

3、mybash完整代码

 1 #include<stdio.h>
  2 #include<string.h>
  3 #include<stdlib.h>
  4 #include<unistd.h>
  5 #include<sys/wait.h>
  6 #include<pwd.h>
  7 #include<errno.h>
  8
  9 char* get_cmd (char buff[],char* myargv[])
 10 {
 11
 12    if( buff == 0 || myargv == 0)
 13    {
 14       return NULL;
 15
 16    }
 17    char*s = strtok(buff," ");
 18    int i = 0;
 19    while( s != NULL )
 20    {
 21       myargv[i++] = s;
 22       s = strtok(NULL," ");
 23    }
 24    return myargv[0];
 25 }
 26
 27 void print_info()
 28 {
 29
 30     int id = getuid();//获取用户id
 31     char* s = "$";
 32     if( id  == 0 )
 33     {
 34        s = "#";
 35     }
 36
 37     struct passwd* ptr = getpwuid(id);//通过uid获取用户详细信息
 38     if( ptr == NULL )
 39     {
 40       printf("$");
 41       fflush(stdout);
 42       return;
 43     }
 44
 45     char hostname[64] = { 0 };
 46     if (gethostname(hostname,64) == -1 )
 47     {
 48       printf("$");
 49       fflush(stdout);
  50       return;
 51     }
 52
 53     //获取当前位置
 54     char pwd_buff[128] = {0};
 55     if(getcwd(pwd_buff,128) == NULL)
 56     {
 57       printf("$");
 58       fflush(stdout);
 59       return;
 60     }
 61
 62     //打印命令提示行信息
 63     printf("\033[1;32;40m %s@%s:\033[0m \033[1;34;40m%s%s\033[0m",ptr->pw_name,hostname,pwd_buff,s);
 64     fflush(stdout);
 65 }
     
 66 int main()
 67
 68 {
 69   while(1)
 70   {
 71     char buff[128] = { 0 };
 72     print_info();
 73     fgets(buff,128,stdin);
 74     buff[strlen(buff) - 1] = 0;
 75     char* myargv[10] = { 0 };
 76     char* cmd = get_cmd(buff,myargv);
 77     if(cmd == NULL)
 78     {
 79         continue;
 80     }
 81
 82     else if (strcmp(cmd,"exit") == 0 )
 83     {
 84        exit(0);
 85     }
 86     else if (strcmp(cmd,"cd") == 0)
 87     {
 88         if( myargv[1] == NULL )
 89         {
 90             continue;
 91         }
 92         if ( chdir(myargv[1]) == -1)
 93         {
 94             perror("cd err");
 95         }
 96         continue;
 97     }
 98     else
 99     {
100        pid_t pid = fork();
101        if(pid == -1 )
102        {
103          printf("fork error\n");
104        }
105
106        if(pid == 0 )
107        {
108          execvp(cmd,myargv);
109          printf("exec error\n");
110          exit(0);
111        }
112       wait(NULL);
113     }
114
115
116   }
117   exit(0);
118 }

4、编译运行程序

在这里插入图片描述
通过运行,我们发现mybash已经自动获取用户名,主机名,当前目录,以及是普通用户还是管理员用户,也实现了cd命令。就此mybash2.0版本结束了!!!

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 打赏
    打赏
  • 扫一扫,分享海报

©️2022 CSDN 皮肤主题:数字20 设计师:CSDN官方博客 返回首页

打赏作者

月笙€

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

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

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

打赏作者

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

抵扣说明:

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

余额充值