非阻塞調用waitpid
這樣父進程就不會阻塞,此時循環使用我們可以讓父進程執行其他任務而不是阻塞等待
進程程序替換
進程=PCB+加載到內存中的代碼和數據
替換就是完全替換當前進程的代碼段、數據段、堆和棧,保存當前的PCB
代碼指的是二進制代碼不是源碼!!!
#include <unistd.h>extern char **environ;int execl(const char *pathname, const char *arg, .../* (char *) NULL */);int execlp(const char *file, const char *arg, .../* (char *) NULL */);int execle(const char *pathname, const char *arg, .../*, (char *) NULL, char *const envp[] */);int execv(const char *pathname, char *const argv[]);int execvp(const char *file, char *const argv[]);int execvpe(const char *file, char *const argv[],char *const envp[]);
pathname是要執行的可執行文件的完整路徑? /bin/ls
file程序名,不帶路徑在環境變量PATH查找
l就是list,以可變參數的形式傳遞"ls",“-l”,“NULL”
p就是會從環境變量查找,只要程序名即可
e就是環境變量數組
v就是vector以指針數組的形式傳遞
自定義shell的編寫
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <cstring>
#include <unordered_map>
#include <string>#define COMMAD_SIZE 1024
#define FORMAT "[%s@%s %s]# "// 下面是shell定義的全局數據// 1.命令行參數
#define MAXARGC 128
char *g_argv[MAXARGC];
int g_argc = 0;// 2.環境變量表
#define MAX_ENVS 100
char *g_env[MAX_ENVS];
int g_envs = 0;// 3.別名映射表
std::unordered_map<std::string, std::string> alias_list; // la對于ls -a// for test
char cwd[1024];
char cwdenv[1024];
char oldpwd[1024];
// last exit code
int lastcode = 0; // 最新子進程退出碼const char *GetUserName()
{const char *name = getenv("USER");return name == NULL ? "None" : name;
}const char *GetHostName() // 變化不大,可以不用系統調用。其實最好也用系統調用
{const char *hostname = getenv("HOSTNAME");return hostname == NULL ? "None" : hostname;
}// 切換路徑時,環境變量表也要切換。因此我們要更新環境變量。
// 注意我們自定義shell的環境變量表繼承自shell
// 路徑先變,環境變量才變,調用系統調用
const char *GetPwd()
{const char *pwd = getcwd(cwd, sizeof(cwd));if (pwd){snprintf(cwdenv, sizeof(cwdenv), "PWD=%s", cwd);putenv(cwdenv);}return pwd == NULL ? "None" : pwd;
}const char *GetOldPwd()
{return getenv("OLDPWD");
}
const char *GetHome()
{return getenv("HOME");
}void InitEnv()
{extern char **environ; // 頭文件#include <unistd.h>定義了這個,這里是聲明memset(g_env, 0, sizeof(g_env));// 本來從父進程繼承,現在直接從配置文件(操作系統)來// 1.獲取環境變量for (int i = 0; environ[i]; i++){// 申請空間// 別用sizeof,sizeof得到的是指針大小g_env[i] = (char *)malloc(strlen(environ[i]) + 1);strcpy(g_env[i], environ[i]);g_envs++;}g_env[g_envs] = NULL;// 2.導入環境變量// 增量修改for (int i = 0; g_env[i]; i++){putenv(g_env[i]);}// environ=g_env;完全重置
}// command 內建命令
bool Cd()
{memset(oldpwd, 0, sizeof(oldpwd));snprintf(oldpwd, sizeof(oldpwd), "OLDPWD=%s", GetPwd());putenv(oldpwd);if (g_argc == 1){std::string home = GetHome();if (home.empty())return true;chdir(home.c_str());}else{std::string where = g_argv[1];if (where == "-")chdir(GetOldPwd());else if (where == "~")chdir(GetHome());elsechdir(where.c_str());}return true;// cd argc = 1if (g_argc == 1){std::string home = GetHome();if (home.empty())return true;chdir(home.c_str());}else{std::string where = g_argv[1];// cd - / cd ~if (where == "-"){// Todu}else if (where == "~"){// Todu}else{chdir(where.c_str());}}return true;
}void Echo()
{if (g_argc == 2){std::string opt = g_argv[1];if (opt == "$?"){std::cout << lastcode << std::endl;lastcode = 0; // echo執行完,退出碼應該是0}else if (opt[0] == '$'){std::string env_name = opt.substr(1);const char *env_value = getenv(env_name.c_str());if (env_value)std::cout << env_value << std::endl;}else{std::cout << opt << std::endl;}}
}
// 防止自定義bash路徑名太長
/*std::string DirName(const char *pwd)
{
#define SLASH "/"std::string dir = pwd;if (dir == SLASH)return SLASH; // 只有根目錄直接返回auto pos = dir.rfind(SLASH);return dir.substr(pos + 1);
}*/
// 命令行提示符
void MakeCommandLine(char cmd_prompt[], int size)
{snprintf(cmd_prompt, size, FORMAT, GetUserName(), GetHostName(), GetPwd());
}
// 打印命令行提示符
void PrintCommandPrompt()
{char prompt[COMMAD_SIZE];MakeCommandLine(prompt, sizeof(prompt));std::cout << prompt;fflush(stdout);
}
bool GetCommandLine(char *out, int size)
{char *c = fgets(out, size, stdin);if (c == NULL)return false;out[strlen(out) - 1] = 0; // 清理\nif (strlen(out) == 0)return false; // 只輸入了/nreturn true;
}
// 3.命令行分析 "ls -a -l" 解析為 "ls" "-a" "-l"
bool CommandParse(char *commandline)
{
#define SEP " "g_argc = 0;g_argv[g_argc++] = strtok(commandline, SEP);while(g_argv[g_argc++] = strtok(nullptr, SEP));g_argc--;return g_argc > 0 ? true : false;
}void PrintArgv()
{for (int i = 0; g_argv[i]; i++){printf("argv[%d]->%s\n", i, g_argv[i]);}printf("argc: %d\n", g_argc);
}bool CheckAndExecBuiltin()
{std::string cmd = g_argv[0];if (cmd == "cd"){Cd();return true;}else if (cmd == "echo"){Echo();return true;}else if (cmd == "export")return true;else if (cmd == "alias")return true;return false;
}int Execute()
{pid_t id = fork();if (id == 0){// childexecvp(g_argv[0], g_argv);exit(1); // 一旦替換exit不會執行}// fatherint status = 0;pid_t rid = waitpid(id, &status, 0);if (rid > 0) // 等待成功lastcode = WEXITSTATUS(status); // 獲取子進程退出碼return 0;
}int main()
{InitEnv();while (true) // 死循環{// 1.輸出命令行提示符PrintCommandPrompt();// 2.獲取用戶輸入的命令char commandline[COMMAD_SIZE];if (!GetCommandLine(commandline, COMMAD_SIZE))continue;// 3.命令行分析,并將分析后的命令行導入全面變量表中if (!CommandParse(commandline))continue;// 檢測是否是內建命令,若是直接調用然后continue,若不是則執行下面的if (CheckAndExecBuiltin())continue;;Execute();}return 0;
}