shell编程实战（shell编程实例详解）

在构建简易Shell的时候我们首先就是要获取输入获取环境变量：能够像shell一样运行会出现部分环境变量获取用户输入：获取用户输入的指令

// 获取环境变量 user,hostname,pwd,homechar *homepath(){ char *home= getenv("HOME"); if(home) return home; else return (char*)".";}const char *getUsername(){ const char *name = getenv("USER"); if(name) return name; else return "none"; }const char *getHostname(){ const char *hostname = getenv("HOSTNAME"); if(hostname) return hostname; else return "none";}const char *getCwd(){ const char *cwd = getenv("PWD"); if(cwd) return cwd; else return "none";}printf("[%s@%s %s]$ ",getUsername(), getHostname(), getCwd());

#define NUM 1024char usercommand[NUM];

// 对用户输入进行封装int getUserCommand(char *command, int num){ printf("[%s@%s %s]$ ",getUsername(), getHostname(), getCwd());  char *r = fgets(command, num, stdin); // 最后还是会输入'\n'，回车 if(r == NULL) return -1; // TODO // 将最后一次输入修改成'\0' command[strlen(command)-1] = '\0'; // 这里不会越界 return strlen(command); }

#define SIZE 64 // 设置argv的大小#define SEP " " // 分隔符 " "#define debug 1 // 用来测试是否能够成功分割字符串char *argv[SIZE]; // 用来储存分割后的字符串

char *strtok(char *str, const char *delim);

commandSplit：void commandSplit(char *in, char *out[]) // in -> usercommand{ // out -> argv int argc = 0; out[argc++] = strtok(in, SEP); while(out[argc++] = strtok(NULL, SEP)); // 我们只需要将用户输入全部读取即可// 用来测试是否能够成功分割字符串#ifdef debug int i = 0; for(i = 0; out[i]; i++) { printf("%d:%s\n", i, out[i]); } #endif}

doBuildin：char cwd[1024]; // 存储char enval[1024]; // for testint lastcode = 0; // 存储上一条指令的返回信息void cd(const char *pash) { chdir(pash); char tmp[1024]; getcwd(tmp, sizeof(tmp)); sprintf(cwd, "PWD=%s", tmp); // 改变当前的路径 putenv(cwd); // 改变环境变量}int doBuildin(char *argv[]){ if(strcmp(argv[0], "cd") == 0) { char *pash = NULL; if(argv[1] == NULL) pash = homepath(); // 当我们cd之后不更任何输入时， //我们直接返回家目录 else { pash = argv[1]; } cd(pash); return 1; }else if(strcmp(argv[0], "exprot") == 0) { if(argv[1] == NULL) return 1;  strcpy(enval, argv[1]); putenv(enval); // 用于增加环境变量内容 return 1; } else if(strcmp(argv[0], "echo") == 0){if(argv[1] == NULL) { printf("\n"); return 1; } if(*(argv[1]) == '$' && strlen(argv[1]) > 1) { char *val = argv[1]+1; // argv[1]为'$',argv[1]+1则为'$'后的字符 if(strcmp(val, "?") == 0) { // lastcode 存储上一条指令的返回信息，初始状态为0 // 当程序执行时，出错时会改变lastcode的值，打印后，重新赋值为0 printf("%d\n", lastcode); lastcode = 0; } else // '$'后不跟"?"，则获取环境变量 { const char *enval = getenv(val); if(enval) printf("%s\n", enval); else printf("\n"); } return 1; }else // 直接输出字符{ printf("%s\n",argv[1]); return 1; } } return 0;}

getcwd：用于获取当前工作目录的绝对路径chdir：用于改变当前工作目录chdir命令通常通过cd命令来实现，因为cd是chdir的别名，两者具有相同的功能。不过，在编程语言中（如C、PHP等），chdir则是一个具体的函数，用于在程序中动态改变当前工作目录sprintf：用于将格式化的数据写入字符数组中putenv：用于改变或增加环境变量内容的函数

#include<unistd.h>#include<sys/types.h>#include<wait.h>

int execute(char *argv[]){ pid_t id = fork(); if(id < 0) return -1; else if(id == 0) { // 子进程 execvp(argv[0], argv); exit(1); } else{ // 父进程 int status = 0; pid_t rid = waitpid(id, &status, 0); if(rid > 0) { // 将现在状态提供个lastcode lastcode = WEXITSTATUS(status); } } return 0; }

int main(){// shell是一个一直循环的程序 while(1) { char usercommand[NUM]; char *argv[SIZE];  // 获取输入 int n = getUserCommand(usercommand, sizeof(usercommand)); // 当获取输入时，返回一个小于0的数时，我们直接continue返回，不用往下继续走了 if(n <= 0) continue; // 分割字符串 (strtok) commandSplit(usercommand, argv); // check Buildin n = doBuildin(argv); // 一般内建命令不会出错 if(n) continue; // 执行命令 execute(argv);  }  return 0;}

注意：当我们输入指令输出字符时，Backspace键是不能直接删除的，我们需要Ctrl + Backspace 组合键用于删除光标前的一个单词

exec/exit就像call/return一个C程序有很多函数组成。一个函数可以调用另外一个函数，同时传递给它一些参数。被调用的函数执行一定的操作，然后返回一个值。每个函数都有他的局部变量，不同的函数通过call/return系统进行通信