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);
}