在Linux下使用C语言编写mybash

mybash:通常shell中执行命令的流程都是bash进程创建一个子进程,然后子进程替换,替换成可执行的命令文件.

实质:不断进行fork()+exec();

bash的输出格式:[用户名 @主机名所在文件]$ ,在代码中实现了动态提示符,使用getuid()函数获取用户的ID,然后再使用getpwuid()函数根据用户的UID获取当前用户的passwd信息,在从中找到用户名字;而主机所在文件名的获取,需要使用getcwd()函数来获得文件的绝对路径,之后在通过strtok()函数进行划分,即可得到.

主要的问题就是如何将输入的命令进行拆解,这里主要使用到strtok()函数:

1.定义
分解字符串为一组字符串。s为要分解的字符,delim为分隔符字符(如果传入字符串,则传入的字符串中每个字符均为分割符)。首次调用时,s指向要分解的字符串,之后再次调用要把s设成NULL。在头文件#include<string.h>中。
2.原型
char *strtok(char s[], const char *delim);
3.说明
(1)当strtok()在参数s的字符串中发现参数delim中包含的分割字符时,则会将该字符改为\0 字符。在第一次调用时,strtok()必需给予参数s字符串,往后的调用则将参数s设置成NULL。每次调用成功则返回指向被分割出片段的指针。
(2)返回值:从s开头开始的一个个被分割的串。当s中的字符查找到末尾时,返回NULL。如果查找不到delim中的字符时,返回当前strtok的字符串的指针。所有delim中包含的字符都会被滤掉,并将被滤掉的地方设为一处分割的节点。

(3)需要注意的是,使用该函数进行字符串分割时,会破坏被分解字符串的完整,调用前和调用后的s已经不一样了。第一次分割之后,原字符串str是分割完成之后的第一个字符串,剩余的字符串存储在一个静态变量中,因此多线程同时访问该静态变量时,则会出现错误。

4.使用
strtok函数会破坏被分解字符串的完整,调用前和调用后的s已经不一样了。如果要保持原字符串的完整,可以使用strchr和sscanf的组合等。
 

因为所输入的命令会带有参数,所以使用execvp()为最佳;为了在程序运行时有别于系统运行,所以在提示符处改变了颜色,如有需要百度即可;

以下是实现代码:

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>
#include<assert.h>
#include<signal.h>
#include<pwd.h>
#include<unistd.h>



int main()
{
    while(1)
    {
        struct passwd *pwd;
        pwd = getpwuid(getuid());
        printf("\033[33m[\033[0m");
        printf("\033[33m%s\033[0m",pwd->pw_name);
        printf("\033[33m@localhost \033[0m");

        char name[80];
        getcwd(name,sizeof(name));
    if(name[0]=='/' && name[1] =='\0')
    {
        printf("\033[33m/]$ \033[0m");
    }
    else
    {
        char *nm = strtok(name,"/");
        char *realNM[20] = {0};
        while(nm != NULL)
        {
            realNM[0] = nm;
            nm = strtok(NULL,"/");
        }

        if(strncmp(realNM[0],"wb",2)==0)
        {
            printf("\033[33m~]$ \033[0m");
        }
        else
        {
            printf("\033[33m%s\033[0m",realNM[0]);
            printf("\033[33m]$ \033[0m");
        }
    }
        fflush(stdout);

        char buff[128] = {0};
        fgets(buff,128,stdin);

        buff[strlen(buff)-1] = 0;

        char *s = strtok(buff," ");
        if(s == NULL)
        {
            continue;
        }
        char* myargv[10] = {0};
        int i = 0;
        while(s != NULL)
        {
            myargv[i++] = s;
            s = strtok(NULL," ");
        }

        if(myargv[0]==0)
        {
            continue;
        }
        if(strcmp(myargv[0],"exit")==0)
        {
            break;
        }
        else if(strcmp(myargv[0],"cd")==0)
        {
            if(myargv[1] == NULL)
            {
                continue;
            }
            chdir(myargv[1]);
            continue;
        }
        pid_t pid = fork();
        assert(pid != -1);
        if(pid ==0)
        {

            execvp(myargv[0],myargv);
            exit(0);
        }
        wait(NULL);
    }
    exit(0);
}

 

上一篇:matlab学习(1)strsplit与strtok


下一篇:等快递无聊--旋转字符串