linuxc 实现my_shell

其他 2020-09-11 10:33:41 阅读次数: 0

问题

1.cd命令的实现(cd ,cd ~,cd -,cd path)

由于cd命令是shell 内建的命令,是不能调用系统的命令实现cd,所以就得手写

所用函数:

char * getcwd(char * buf, size_t size); //取得当前工作目录
buf 为所取得的路径,   size为所取的大小
int chdir(const char * path); //改变当前目录
path 为要切换的路径

因为要实现 cd -,所以要将每次的路径保存下来,(这里用cd_pathnametemp)保存

                getcwd(cd_pathname,100);
                if((argcount == 1)  || strcmp(arg[1],"~") == 0)  
                {   
                        strcpy(cd_pathnametemp,cd_pathname);
                        my_chdir();   //更改家目录
                }   
                else if(strcmp(arg[1],"-") == 0)
                {   
                //      strcpy(cd_pathnametemp,cd_pathname);
                        chdir(cd_pathnametemp);
                        strcpy(cd_pathnametemp,cd_pathname);
                }   
                else
                {   
                        strcpy(cd_pathnametemp,cd_pathname);
                        chdir(arg[1]);    //更改当前工作目录
                }
2.ls输出加颜色

因为ls 本身是不输出颜色的,而我们的shell 中能输出颜色是因为 用了别名

 alias ls='ls --color=auto'

所以在处理 ls 的时候,给它加上 - -color=auto参数就ok

3.屏蔽ctrl + c

一个优秀的程序是不会轻易的挂掉,所以不屏蔽ctrl + c 的话,用户一个ctrl + c 就会让程序挂掉,那样会有多尴尬.
而屏蔽ctrl + c也不难

 signal(SIGINT, SIG_IGN);

SIGINT :当用户按下了<ctrl + c >时,用户终端向正在运行中的由该终端启动的程序发出此信号,默认为终止进程.而进程终止后,程序自然也就被终止了
SIG_IGN : 忽略该信号

4.命令自动补全

了解了之后,发现readline 这个库是真的强大,有许多特别好用,方便的函数,下面的history也是这个库里的函数

自动补全这个功能需要用Readline 库 ,没有的话需要安装

			sudo apt-get install libreadline6-dev

因为readline 是动态链接库 所以 ,编译的时候 需要加上 -lreadline
要加头文件

#include<readline/readline.h>

        char * str = readline(" ");
5.通过上下查找历史命令

头文件

#include<readline/history.h>

同样,编译的时候需要加上 -lreadline

        add_history(str);

每次将所读取的命令加到 add_history()中就ok了.

对readline还不是很熟悉的可以做个小测试熟悉一下

6.将my_shell 设置环境变量，使程序可以像bash，zsh这种shell一样运行;

这个问题就将my_shell 的可执行程序 加在 /bin 下就搞定了.

以上就是这次shell 的主要问题.
其他问题
    1. 支持输入输出重定向(< > >>)
    2. 支持管道(|)
    3. 支持后台运行程序(&)
看源码就可以搞定,我就不再赘述.

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>
#include<sys/types.h>
#include<sys/wait.h>
#include<fcntl.h>
#include<sys/stat.h>
#include<dirent.h>
#include<pwd.h>
#include<readline/readline.h>
#include<readline/history.h>


#define normal  0 //一般命令
#define out_redirect  1  //输出重定向
#define in_redirect 2 //输入重定向
#define have_pipe   3  //命令中有管道
#define add_out_redirect 4
#define add_in_redirect 5
#define CLOSE "\001\033[0m\002"                 // 关闭所有属性
char cd_pathnametemp[PATH_MAX] = "/home/tt";    //默认家目录
char cd_pathname[PATH_MAX];


void print_prompt();   //打印提示符  
void get_input(char *);   //得到输入的命令
void explain_input(char *,int *,char a[ ][256]);  //对输入命令进行解析
void do_cmd(int ,char a[ ][256]);  //执行命令
int find_command(char *);    //查找命令中的可执行程序
void my_dir();//cd 到家目录
int main(int argc,char **argv)
{
	 signal(SIGINT, SIG_IGN);
	int i;
	int argcount = 0; //记录 命令的个数
	char arglist[100][256];   //存储命令
	char **arg = NULL;
	char *buf = NULL;

	buf = (char *)malloc(256);
	if(buf == NULL)
	{
		perror("malloc failed\n");
		exit(-1);
	}


	while(1)
	{
		memset(buf,0,sizeof((buf)));
		print_prompt();   //输出命令提示符
		get_input(buf);   //获取输入
		//若输入的命令为 exit 或 logout 则退出本程序
		if(strcmp("exit\n",buf) == 0 || strcmp("logout\n",buf) == 0)  break;
		if(strcmp(buf,"\n") == 0) continue;
		//清空 arglist 
		for(i = 0;i < 100;i++)    arglist[i][0] = '\0';
		argcount = 0;//命令个数 清0


		explain_input(buf,&argcount,arglist);

		do_cmd(argcount,arglist);

	}
	if(buf == NULL)
	{
		free(buf);
		buf = NULL;
	}
	exit(0);

}
void my_chdir()
{
/*	int i,flag = 0,j;
	char name[30];
	char pathname[100];
	char pathnametemp[100];
	int uid;
	struct passwd *data;
	//uid_t uid;
	uid = getuid();  //获取uid 
	data = getpwuid(uid);
	getcwd(pathname,100);
	//处理路径  
	int len = strlen(pathname);
	for(i = 0;i < len;i++)
	{
		if(pathname[i] == '/') flag++;
		if(flag == 3)  break;
	}
	for(j = 0;j < i;j++)
	{
		pathnametemp[j] = pathname[j];
	}

	pathnametemp[j] = '\0';
*/
	//printf( "%s\n",pathnametemp);
	//chdir(pathnametemp);
	chdir("/home/tt");
}

//输出 命令提示符
void print_prompt()
{
	int i,flag = 0,j;
	char name[30];
	char pathname[100];
	char pathnametemp[100];
	int uid;
	struct passwd *data;

	//uid_t uid;
	uid = getuid();
	data = getpwuid(uid);
	printf("\033[43;35m%s@\033[0m",data->pw_name);
	gethostname(name,30);
	printf( "\033[43;35m%s:\033[0m",name);
	getcwd(pathname,100);
	
	if(pathname[1] != 'h' && pathname[2] != 'o')
	{
		printf( "\033[35;43m%s\033[0m",pathname);
		return ;
	}

	//处理路径  
	int len = strlen(pathname);
	for(i = 0;i < len;i++)
	{
		if(pathname[i] == '/') flag++;
		if(flag == 3)  break;
	}
	for(j = i;j < len;j++)
	{
		pathnametemp[j-i] = pathname[j];
	}
	pathnametemp[len-i] = '\0';
	strcpy(pathname,"~");
	strcat(pathname,pathnametemp);
	printf( "\033[35;43m%s\033[0m",pathname);

	//打印用户提示符
	if(0 == uid)  printf( "\033[40;32m#\033[0m");
	else printf( "\033[40;32m$\033[0m");
	
	return ;
}

//获取用户输入  
void get_input(char *buf)
{
	int len = 0;
	int ch;

	char * str = readline(" "CLOSE);
	add_history(str);
	strcpy(buf,str);
	buf[strlen(buf)] = '\n';


	/*free(str);
	ch = getchar();
	while(len < 256 && ch != '\n')
	{
		//char * str = readline("");
		//free(str);
		buf[len++] = ch;
		ch = getchar( );
	}

	if(len == 256)
	{
		printf( "command is too long \n");
		exit(-1);  //若输入的命令太长,则退出程序
	}

	buf[len] = '\n';
	len++;
	buf[len] = '\0';
	*/

}

//解析buf 中的命令,将结果存入 arglist 中,命令以回车符号'\n'结束
//如 输入命令"ls -l /temp" 则arglist[0],arglist[1],arglist[2],分别为 ls,-l,/temp

void explain_input(char *buf,int *argcount,char arglist[100][256])
{
	char *p = buf;
	char *q = buf;
	int number = 0;

	while(1)
	{
		if(p[0] == '\n')  break;
		if(p[0] == ' ') p++;
		else
		{
			q = p;
			number = 0;  //记录每个命令的长度

			while(q[0] != ' ' && (q[0] != '\n'))
			{
				number++;
				q++;
			}

			strncpy(arglist[*argcount],p,number+1);  //*argcount 为命令的个数
			arglist[*argcount][number] = '\0';
			*argcount = *argcount + 1;
			p = q;
		}
	}
}

//执行arglist 中存放的命令 ,argcount 为待执行命令的参数个数  
void do_cmd(int argcount,char arglist[100][256])
{
	int flag = 0;
	int how = 0;   //用于只是命令中是否含有 < , > , | 
	int background = 0; //表示命令中是否有后台运行表示符
	int status;
	int i;
	int fd;
	char *arg[argcount + 1];
	char *argnext[argcount + 1];
	char *file;  //保存文件名
	pid_t pid;
	int cdflag = 0;

	//将命令取出 
	for(i = 0;i < argcount;i++)   arg[i] = (char *)arglist[i];
	//给ls 加颜色
	if(strcmp(arg[0],"ls") == 0)
	{
		arg[argcount] = "--color=auto";
		arg[argcount + 1] = NULL;
	}
	else 	arg[argcount] = NULL;


	//cd   
	if(strcmp(arg[0],"cd") == 0)
	{
		getcwd(cd_pathname,100);
		if((argcount == 1)  || strcmp(arg[1],"~") == 0)  
		{
			strcpy(cd_pathnametemp,cd_pathname);
			my_chdir();   //更改家目录
		}
		else if(strcmp(arg[1],"-") == 0)
		{
		//	strcpy(cd_pathnametemp,cd_pathname);
			chdir(cd_pathnametemp);
			strcpy(cd_pathnametemp,cd_pathname);
		}
		else
		{
			strcpy(cd_pathnametemp,cd_pathname);
			chdir(arg[1]);    //更改当前工作目录
		}


		return ;
	}

	//查看命令行是否有后台运行符
	for(i = 0;i < argcount; i++)
	{
		if(strncmp(arg[i],"&",1) == 0)
		{
			if(i == argcount -1)
			{
				background = 1;
				arg[argcount - 1] = NULL;
				break;
			}
			else
			{
				printf( "wrong command\n");
				return ;
			}
		}
	}

	//查看命令行 是否 有 重定向 和管道符
	for(i = 0; arg[i] != NULL;i++)
	{
		if(strcmp(arg[i],">") == 0)
		{
			flag++;
			how = out_redirect;
			if(arg[i+1] == NULL) flag++;
		}
		if(strcmp(arg[i],"<") == 0)
		{
			flag++;
			how = in_redirect;
			if(i == 0) flag++;
		}
		if(strcmp(arg[i],"|") == 0)
		{
			flag++;
			how = have_pipe;
			if(arg[i+1] == NULL) flag++;
			if(i == 0)  flag++;
		}
		if(strcmp(arg[i],">>") == 0)
		{
			flag++;
			how = add_out_redirect;
			if(arg[i+1] == NULL)  flag++;
		}
		if(strcmp(arg[i],"<<") == 0)
		{
			flag++;
			how = add_in_redirect;
			if(i == 0) flag++;
		}

	}
	//若 flag == 1, 则有
	//若 flag > 1,则格式错误,不支持
	if(flag > 1)
	{
		printf( "wrong command\n");
		return ;
	}

	//命令中 只含一个输出替换重定向符号
	if(how == out_redirect)
	{
		for(i = 0;arg[i] != NULL;i++)
		{
			if(strcmp(arg[i],">") == 0)
			{
				file = arg[i+1];
				arg[i] = NULL;
			}
		}
	}

	//命令中只含有 一个输入替换重定向	
	if(how == in_redirect)
	{
		for(i = 0;arg[i] != NULL;i++)
		{
			if(strcmp(arg[i],"<") == 0)
			{
				file = arg[i+1];
				arg[i] = NULL;
			}
		}
	}
	//命令中只含有 输出追加重定向
	if(how == add_out_redirect)
	{
		for(i = 0;arg[i] != NULL;i++)
		{
			if(strcmp(arg[i],">>") == 0)
			{
				file = arg[i+1];
				arg[i] = NULL;
				arg[i] = NULL;
			}
		}
	}

	//命令中只含有 输入追加重定向
	if(how == add_in_redirect)
	{
		for(i = 0;arg[i] != NULL;i++)
		{
			if(strcmp(arg[i],"<<") == 0)
			{
				file = arg[i+1];
				arg[i] = NULL;
			}
		}
	}
	//命令中只含有 一管道符号
	//把管道符号后面的部分存入argnext中,管道后面的部分是一个可执行 的shell 命令
	if(how == have_pipe)
	{
		for(i = 0;arg[i] != NULL;i++)
		{
			if(strcmp(arg[i],"|") == 0)
			{
				arg[i] = NULL;
				int j;
				//将| 后面的 命令存入argnext 中
				//j - i -1   就是 从零开始,往后加
				for(j = i+1;arg[j] != NULL;j++)  argnext[j-i-1] = arg[j];
				argnext[j-i-1] = arg[j];
				break;
			}
		}
	}

	if((pid = fork()) < 0)
	{
		printf( "fork error\n");
		return ;
	}

 	    //0  一般命令
	    //1  输出重定向
	    //2  输入重定向
	    //3  命令中有管道
	//pid 为 0 说明是子进程 ,在子进程中执行命令
	switch(how)
	{
		case 0: //一般命令
			if(pid == 0)
			{
				if(!(find_command(arg[0])))
				{
					printf( "%s : command not found\n",arg[0]);
					exit(0);
				}

				execvp(arg[0],arg);
				exit(0);
			}
			break;
		case 1:  //替换输出重定向
			if(pid == 0)
			{
				if(!(find_command(arg[0])))
				{
					printf( "%s : command not found\n",arg[0]);
					exit(0);
				}
				fd = open(file,O_RDWR | O_CREAT | O_TRUNC,0644);  //可读可写 ,若文件不存在就自动建立该文件  ,将文件长度清 0
				dup2(fd,1);  //赋值 文件描述符描述符   将本来 的文件描述符 改为 1 标准输入
				execvp(arg[0],arg);
				exit(0);
			}
			break;
		case 2:  //替换输入重定向
			if(pid == 0)
			{
				if(!(find_command(arg[0])))
				{
					printf( "%s : command not found\n",arg[0]);
					exit(0);
				}
				fd = open(file,O_RDONLY);
				dup2(fd,0);
				execvp(arg[0],arg);
				exit(0);
			}
			break;
		case 3:  //含有管道
			if(pid == 0)
			{
				int pid2;
				int status2;
				int fd2;

				if((pid2 = fork()) < 0)
				{
					printf( "fork2 error\n");
					return ;
				}
				else if(pid2 == 0)
				{
					if(!(find_command(arg[0])))
					{
						printf( "%s : command not found\n",arg[0]);
						exit(0);
					}

					fd2 = open("/tmp/youdonotknowfile",O_WRONLY | O_CREAT | O_TRUNC,0644);
					dup2(fd2,1);
					execvp(arg[0],arg);
					exit(0);
				}
				if(waitpid(pid2,&status2,0) == -1)  printf( "wait for child procrss error\n");

				if(!(find_command(argnext[0])))
				{
					printf( "%s : command not found\n",argnext[0]);
					exit(0);
				}

				fd2 = open("/tmp/youdonotknowfile",O_RDONLY);
				dup2(fd2,0);
				execvp(argnext[0],argnext);

				if(remove( "/tmp/youdonotknowfile"))  printf( "remove error\n");
				exit(0);

			}
			break;
		case 4://追加输出重定向
			if(pid == 0)
			{
				if(!(find_command(arg[0])))
				{
					printf( "%s : command not found\n",arg[0]);
					exit(0);
				}

				fd = open(file,O_RDWR | O_CREAT | O_APPEND);
				dup2(fd,1);
				execvp(arg[0],arg);
				exit(0);
			} break;
		default : break;
	}

	//若命令中有 &,表示后台运行,父进程直接返回,不等待子进程结束
	if(background == 1)
	{
		printf( "[process id %d ]\n",pid);
		return ;
	}

	//父进程等待子进程结束
	if(waitpid(pid,&status,0) == -1)  printf( "wait for child process error\n");

}

//查找命令中的可执行文件
int find_command(char *command)
{
	DIR* dp;
	struct dirent * dirp;
	char * path[] = {"./","/bin","/usr/bin",NULL};

	//使用当前目录下的程序可以运行,如命令"./fork "可以被正确解释和执行 
	if(strncmp(command,"./",2) == 0)   command = command + 2;

	//分别在当前目录 /bin ,/usr/bin 目录查找要执行的程序
	//
	
	int i = 0;
	while(path[i] != NULL)
	{
		if((dp = opendir(path[i])) == NULL)   printf( "can not open /bin\n");

		while((dirp = readdir(dp)) != NULL)
		{
			if(strcmp(dirp->d_name,command) == 0)
			{
				closedir(dp);
				return 1;
			}
		}

		closedir(dp);
		i++;
	}

	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43701555/article/details/96986997
今日推荐
周排行
成为C++高手之宏与枚举
在CAD二次开发中使用进度条
Js插件ECharts，HighCharts学习网址整理
Celery提交任务出错(on windows.)
cephfs内核客户端性能追踪
thinkphp中PHPExcel用法
EntityFramework动态组合多排序字段
汇编语言（八）实验9 根据材料编程
安装ubuntu后必须做的事情（对我而言）
JS函数式编程
每日归档
更多
2024-10-22(0)
2024-10-21(0)
2024-10-20(0)
2024-10-19(0)
2024-10-18(0)
2024-10-17(0)
2024-10-16(0)
2024-10-15(0)
2024-10-14(0)
2024-10-13(0)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022