❀Linux进程
前言:在Linux的浩瀚宇宙中,Shell脚本无疑是连接用户与系统之间的桥梁,它赋予了用户强大的自动化处理能力,使得繁琐的重复性任务变得轻松高效。对于每一位Linux爱好者、系统管理员或是开发人员而言,掌握Shell脚本编写技能无疑是一项不可或缺的宝贵财富
让我们一起,在Shell脚本的海洋中扬帆起航,探索更多未知的精彩吧!
📒1. 获取输入
🌞获取环境变量
在运行shell时就会出现一些环境变量,我们自定义构建的shell中,也可以实现这一步
用户名:pxt
主机名:hecs - 198213
当前目录:myshell
// 获取环境变量 user,hostname,pwd,home
char *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());
这里我们直接将绝对路径展示了出来,当然没什么影响
在我们完成最基础的一步之后,我们要开始模拟我们使用的shell的使用方式,接下来一步就是获取用户输入
⭐获取用户输入
获取用户输入时,我们可以创建一个字符数组用来存储用户的输入
#define NUM 1024
char usercommand[NUM];
在获取用户输入时,输完一个指令后不换行,进行下一次输入。
但是在我们输入完成时,总会回车确认,因此我们需要修改最后一次输入
getUserCommand:
// 对用户输入进行封装
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);
}
📚2. 分割字符串
在我们完成用户输入指令的读取之后,我们需要将字符串进行分割,让我们的指令能够被正确的读取并且实现出来,通常我们的分隔符是' '
(空格)
#define SIZE 64 // 设置argv的大小
#define SEP " " // 分隔符 " "
#define debug 1 // 用来测试是否能够成功分割字符串
char *argv[SIZE]; // 用来储存分割后的字符串
而关于分割字符串,我们在C语言的学习中可能提到过一个字符串函数strtok()
,它的头文件是<string.h>
,用于分割字符串。它基于一系列的分隔符来分割字符串,并返回指向被分割出的字符串的指针
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
}
在完成分割字符串之后,我们就可以用很久之前学习的进程的知识,让它运行起来的,但是仅仅如此我们的很多命令是根本无法实现的,所以我们还需要进行一步,检查内建命令!
📜3. 检查内建命令
我们简单实现3个内建命令:cd,exprot,echo
doBuildin:
char cwd[1024]; // 存储
char enval[1024]; // for test
int 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
:用于改变或增加环境变量内容的函数
📝4. 执行程序
在完成前面的铺垫之后,我们就可以创建进程来实现我们的程序了,在之前学习进程时,我们需要用到3个头文件
#include<unistd.h>
#include<sys/types.h>
#include<wait.h>
execute:
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;
}
以上就是对一些基本操作的封装,让我们看一下主函数main
main:
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;
}
我们来思考函数和进程之间的相似性
这种通过参数和返回值在拥有私有数据的函数间通信的模式是结构化程序设计的基础。Linux鼓励将这种应用于程序之内的模式扩展到程序之间
一个C程序可以fork/exec另一个程序,并传给它一些参数。这个被调用的程序执行一定的操作,然后通过exit(n)来返回值。调用它的进程可以通过wait(&ret)来获取exit的返回值
📖5. 总结
在探索和学习编写Linux中简易shell脚本的旅程即将告一段落之际,我们不禁回望这段充满挑战与收获的时光。Shell脚本,作为Linux系统中不可或缺的一部分,以其强大的自动化能力和灵活的语法结构,成为了系统管理员、开发者以及任何希望提高工作效率的用户的得力助手
最后,让我们以一颗永不停歇的求知之心,继续前行在Linux与shell脚本编程的广阔天地中,共同书写属于我们的精彩篇章!
希望本文能够为你提供有益的参考和启示,让我们一起在编程的道路上不断前行!
谢谢大家支持本篇到这里就结束了,祝大家天天开心!