【C语言】之实现 shell 模拟终端

/*********************************************
 * 文件名：myShell.c
 * 功能描述：模拟终端shell并实现shell外部命令
 * 编辑人：王廷云
 * 编辑时间：2017-8-21
 * 修改时间：2018-1-23
*********************************************/

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <pwd.h>
#include <sys/types.h>
#include <sys/wait.h>

#define  SIZE  1024		// 缓冲区大小

/* 功能函数声明 */
void getCmd(char *buff, char *cmd[]);			 // 获取命令
void showShell(void);							 // 显示终端提示符
void getInput(char buff[]);						 // 获取输入
void insertChar(char buff[], char ch, int *idx); // 字符插入
void deleteChar(char buff[], int *idx);			 // 字符删除
void moveRight(char buff[], int *idx);			 // 光标右移
void moveLeft(char buff[], int *idx);			 // 光标左移

/*
 * 程序从主函数开始
 * 思路：
 * 	 1.打印终端提示符
 * 	 2.获取用户输入的命令
 * 	 3.创建子进程执行命令
 * 	 4.执行完毕回到终端提示符
*/
int main(void)
{
	char buff[SIZE] = {0};	// 用户输入
	char *cmd[20];			// 把输入转换为终端命令

	/* 配置界面 */
	system("stty -echo");	// 关闭回显
	system("stty -icanon");	// 关闭缓冲

	/* 进入终端循环 */
	while (1)
	{
		/* 显示终端提示符 */
		showShell();

		/* 获取用户输入 */
		getInput(buff);

		/* 把用户输入转换为终端命令 */
		getCmd(buff,cmd);
		if (NULL == cmd[0])	// 用户输入的回车
		{
			continue;
		}

		/* 命令为exit则退出终端 */
		else if (strcmp(cmd[0],"exit") == 0)
		{
			putchar('\n');			// 退出前换行
			system("stty echo");	// 恢复回显
			system("stty icanon");	// 恢复缓冲
			return 0;
		}

		/* 创建子进程执行用户输入命令 */
		if (0 == fork())
		{
			putchar('\n');		// 把命令执行结果换行显示
			execvp(cmd[0], cmd);
			perror(cmd[0]);		// 如果返回则说明执行错误
			exit(EXIT_FAILURE);
		}

		wait(NULL);	// 回收子进程
	}
}

/*
 * 函数名：getInput
 * 函数功能：获取用户输入，采用字符串保存方式使得支持光标移动
 * 参数：用于保存用户输入的字符数组
 * 返回值：无
*/
void getInput(char buff[])
{
	char ch;
	int idx = 0;	// 光标索引
	buff[0] = '\0';

	/* 保存光标位置 */
	printf("\033[s");
	fflush(stdout);

	/* 进入获取用户输入循环，直到输入完成 */
	while (1)
	{
		ch = getchar();
		if ('\033' == ch && getchar() == '[')  // 方向键
		{
			switch (ch = getchar())
			{
				case 'C': moveRight(buff,&idx); break;
				case 'D': moveLeft(buff,&idx);  break;
				default : break;
	 		}
			continue;
		}

		/* 其它键 */
		switch (ch)
		{
			case 127 : deleteChar(buff, &idx);    break;
			case '\n': 					  	     return;
			default  : insertChar(buff, ch, &idx);break; 
		}

		/* 恢复光标-清除光标至行末尾-打印命令-恢复光标位置 */
		printf("\033[u\033[K%s\033[u", buff);

		/* 光标与下标关联 */
		if (idx > 0)
		{
			printf("\033[%dC", idx);
		}
		fflush(stdout);
	}
}

/*
 * 函数名：insertChar
 * 函数功能：在光标出插入字符
 * 参数：1.存储用户输入的字符串 2.待插入的字符 3.光标索引值
 * 返回值：无
*/
void insertChar(char buff[], char ch, int *idx)
{
	/* 越界问题由光标移动解决 */
	if (strlen(buff) < SIZE-1)
	{
		/* 光标以后的字符右移 */
		memmove(buff+*idx+1,buff+*idx,strlen(buff)-*idx+1);
		buff[*idx] = ch; // 光标处插入字符
		(*idx)++;
	}	
}

/*
 * 函数名：deleteChar
 * 函数功能：删除光标前的字符
 * 参数：1.用户输入的字符串 2.光标索引值
 * 返回值：无
*/
void deleteChar(char buff[], int *idx)
{
	if (idx > 0)
	{
		/* 通过移动覆盖达到删除的目的 */
		memmove(buff+*idx-1,buff+*idx,strlen(buff)-*idx+1);
	}
	(*idx)--;
}

/*
 * 函数名：moveRight
 * 函数功能：光标右移
 * 参数：1.用户输入的字符串 2.光标索引值
 * 返回值：无
*/
void moveRight(char buff[], int *idx)
{
	if (*idx < strlen(buff))
		(*idx)++;
}

/*
 * 函数名：moveLeft
 * 函数功能：光标左移
 * 参数：1.用户输入的字符串 2.光标索引值
 * 返回值：无
*/
void moveLeft(char buff[], int *idx)
{
	if (*idx > 0)
		(*idx)--;
}

/*
 * 函数名：getCmd
 * 函数功能：把用户输入的字符串转换为终端命令
 * 参数：1.字符串地址 2.指针数组-保存转换后的命令
 * 返回值：无
*/
void getCmd(char *buff, char *cmd[])
{
	int idx = 0;
	char *ret;

	ret = strtok(buff," \t\n"); // 分割命令
	while (1)
	{
		cmd[idx++] = ret;
		if (NULL == ret)		// 分割命令结束
			return;
		ret = strtok(NULL," \t\n");
	}
}

/*
 * 函数名：showShell
 * 函数功能：打印终端提示符
 * 参数：无
 * 返回值：无
*/
void showShell(void)
{
	/* 终端提示符组成：[用户名+@+主机名+当前目录]+用户提示符 */
	
	uid_t uid;
	char *ret;
	struct  passwd *user;
	char hostname[100];
	char cwd[120];

	/* 获取用户id */
	uid = getuid();

	/* 根据uid获取用户的passwd结构体 */
	user = getpwuid(uid);
	printf("[%s@",user->pw_name); // 打印用户名

	/* 获取主机名 */
	gethostname(hostname, 100);
	printf("%s:", hostname);	  // 打印主机名

	/* 获取当前工作目录 */
	getcwd(cwd, 120);
	if (strcmp(cwd,user->pw_dir) == 0)
	{
		printf("~"); 			  // 家目录特殊处理
	}
	else
	{
		ret = strrchr(cwd,'/');   // 获取路径的最后一个目录
		if (ret[1] == '\0')
		{
			printf("/]");		  // 根目录
		}
		else
		{
			printf("%s]",ret+1);
		}
	}

	/* 打印用户提示符 */
	if (0 == uid)
	{
		printf("# ");		// 超级用户
	}
	else
	{
		printf("$ ");		// 普通用户
	}

	fflush(stdout);			// 刷新终端提示符
}