学习理解shell的好办法--编写自己的shell 之三

青旅半醒 2022-01-15 07:13 142阅读 0赞

[为什么80%的码农都做不了架构师？>>> ][80_] ![hot3.png][]

加入if...then后，then作用的部分，在条件失败时，里面的命令不执行，不传给子进程，条件成功才传给子进程。因此，if,then,fi是控制命令，要和普通命令区分开。在splitline之后，在fork之前，应加一层判断，看解析后的命令是不是控制命令。这里的判断不是用if直接判断，否则就无穷的递归了....而是用原始的字符比较，如果命令字符是"if","then","fi"那就是属于控制.  
  
增加的这层叫process，它负责管理流程。流程图是这样的（略去了信号处理）：  
  
取得命令  
\---------->解析命令---------->exit  
| |  
| \*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*  
| \* y | \*  
|<\*--------是否为控制？ \*  
| \* | \*  
| \* |n \*  
| \* | \*  
| \* -----fork------ \*  
| \* | | \*  
| \* wait execvp \*  
| \* | | \*  
| \* | | \* 星号方框内是process函数  
\-\*----<-----------exit \*  
\* \*  
\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*  
  
process将脚本分成四种区域:if之前和fi之后的中立区neutral; if和then之间的want-then; then到else之间的then-block； else到fi的else-block.  
中立区代码挨个执行；  
want-then中，每执行一条就记录其退出状态，一般是一条需要if判断执行正确与否的指令；  
else-block有时候可以没有。  
  
shell记录当前区域，还记录want-then区的执行结果  
  
基于chicken\_sh.c修改后的main函数只改一个地方，把调用，execute的地方换成process，这样就用process封装了命令执行部分。增加了两个文件process.c和controlflow.c ( [源码下载][Link 1]), 后者中有这三个处理区域的函数：  
is\_control\_command: 判断读入的这个命令是普通命令还是"if","then"这种属于shell的关键字  
do\_control\_command:处理关键字  
ok\_to\_execute:根据状态和条件命令的结果返回一个真假值，说明能否执行当前命令  
controlflow.c的代码中，不带else的控制共有三个状态,NEUTRAL, WANT\_THEN, THEN\_BLOCK，用if\_state表示, if\_result有SUCCESS，FAIL两种结果，表示if后的命令是否成功执行。加上else后多了WANT\_ELSE, ELSE\_BLOCK  
do\_control\_command函数如下：

#include<stdio.h>
    #include "chicken_sh.h"
    
    enum states { NEUTRAL, WANT_THEN, THEN_BLOCK, WANT_ELSE, ELSE_BLOCK };
    enum results { SUCCESS, FAIL };
    
    static int if_state = NEUTRAL;
    static int if_result = SUCCESS;
    static int last_stat = 0;
    
    int syn_err(char *);
    
    int ok_to_execute()
    /*决定shell是否执行一条命令
     * 是则返回1,否则返回0
     */
    {
    	int rv = 1;
    
    	if (if_state == WANT_THEN || if_state == WANT_ELSE) {
    		syn_err("then or else expected");
    		rv = 0;
    	} else if (if_state == THEN_BLOCK && if_result == SUCCESS)
    		rv = 1;
    	else if (if_state == THEN_BLOCK && if_result == FAIL)
    		rv = 0;
    	else if (if_state == ELSE_BLOCK && if_result == FAIL)
    		rv = 1;
    	else if (if_state == ELSE_BLOCK && if_result == SUCCESS)
    		rv = 0;
    	return rv;
    }
    
    int is_control_command(char *s)
    {
    	return (strcmp(s, "if") == 0 || strcmp(s, "then") == 0 ||
    		strcmp(s, "else") == 0 || strcmp(s, "fi") == 0);
    }
    
    int do_control_command(char **args)
    {
    	char *cmd = args[0];
    	int rv = -1;
    
    	if (strcmp(cmd, "if") == 0) {
    		if (if_state != NEUTRAL)
    			rv = syn_err("if unexpected");
    		else {
    			last_stat = process(args + 1);
    			if (last_stat == 0) {
    				if_result = SUCCESS;
    				if_state = WANT_THEN;
    			} else {
    				if_result = FAIL;
    				if_state = WANT_ELSE;
    			}
    			rv = 0;
    		}
    	} else if (strcmp(cmd, "then") == 0) {
    		if (if_state != WANT_THEN)
    			rv = syn_err("then unexpected");
    		else {
    			if_state = THEN_BLOCK;
    			rv = 0;
    		}
    	} else if (strcmp(cmd, "else") == 0) {
    		if (if_state != WANT_ELSE)
    			rv = syn_err("else unexpected");
    		else {
    			if_state = ELSE_BLOCK;
    			rv = 0;
    		}
    	} else if (strcmp(cmd, "fi") == 0) {
    		if (if_state != THEN_BLOCK || if_state != ELSE_BLOCK)
    			rv = syn_err("fi unexpected");
    		else {
    			if_state = NEUTRAL;
    			rv = 0;
    		}
    	} else
    		fatal("internal error processing:", cmd, 2);
    	return rv;
    }
    
    int syn_err(char *msg)
    {
    	if_state = NEUTRAL;
    	fprintf(stderr, "syntax error: %s\n", msg);
    	return -1;
    }

增加变量的这几种处理：赋值，引用，列出，导出（export），都为shell内置命令。

变量保存在一个数据中，global这个布尔变量表示已导出。

修改process函数，在execute执行前，增加一个判断，看是否为内置命令: if(!builtin\_command(args,&rv)).

该函数在builtin.c文件中([source code][Link 1])

#include<stdio.h>
    #include<string.h>
    #include<ctype.h>
    #include"chicken_sh.h"
    #define MAXVARS 200
    
    
    int assign(char *);
    int okname(char *);
    
    
    
    struct var{
        char *str;
        int global;
    }
    
    static struct var tab[MAXVARS]
    
    
     int builtin_command(char **args, int *resultp)
    /* run a builtin command
       return: 1 if args[0] is builtin, 0 if not
    */
    {
      int rv = 0;
    
      if (strcmp(args[0],"set") == 0) { /* 'set' command? */
        VList();
        *resultp = 0;
        rv = 1;
      }else if (strchr(args[0], '=') != NULL) { /* assignment cmd */
        *resultp = assign(args[0]);
        if (*resultp != -1)
          rv = 1;
      }else if (strcmp(args[0], "export") == 0) {
         if (args[1] != NULL && okname(args[1]))
          *resultp = VLexport(args[1]);
        else
          *resultp = 1;
        rv = 1;
      }
      return rv;
    }
    
    int assign(char *str)
    /* 
     * 赋值：name=value
     */
    {
      char *cp;
      int rv;
    
      cp = strchr(str, '=');
      *cp = '\0';
      rv = (okname(str)?VLstore(str, cp + 1):-1);
      *cp = '=';
      return rv;
    }
    
    int okname(char *str)
    /* 
       determines if a string is a legal variable name */
    {
      char *cp;
    
      for (cp = str; *cp; cp++ ) {
        if ((isdigit(*cp) && cp == str) || !(isalnum(*cp) || *cp == '_'))
          return 0;
      }
      return (cp != str);		/* no empty strings, either */
    }

用到的VLstore增加更新名值对，VLookup取得var的值，VList输出列表在varlib.c中，此文件将环境一起实现

另外big\_chicken\_sh.c中，setup要增加VLenviron2table，将环境复制到变量表，即继承环境的变量。

转载于:https://my.oschina.net/sukai/blog/296805

[80_]: https://my.oschina.net/u/2663968/blog/3051541
[hot3.png]: /images/20220115/c6dcdebf95aa439da358c01758ee8309.png
[Link 1]: https://github.com/hotea/esh/tree/master/big_chicken