defer推迟函数调用语法的实现
最后更新于：2022-04-02 05:19:53
# 附录2：defer推迟函数调用语法的实现
使用过Go语言的应该都知道defer这个语法，它用来推迟一个函数的执行，在函数执行返回前首先检查当前函数内是否有推迟执行的函数，如果有则执行，然后再返回。defer是一个非常有用的语法，这个功能可以很方便的在函数结束前执行一些清理工作，比如关闭打开的文件、关闭连接、释放资源、解锁等等。这样延迟一个函数有以下两个好处：
* (1) 靠近使用位置，避免漏掉清理工作，同时比放在函数结尾要清晰
* (2) 如果有多处返回的地方可以避免代码重复，比如函数中有很多处return
在一个函数中可以使用多个defer，其执行顺序与栈类似：后进先出，先定义的defer后执行。另外，在返回之后定义的defer将不会被执行，只有返回前定义的才会执行，通过exit退出程序的情况也不会执行任何defer。
在PHP中并没有实现类似的语法，本节我们将尝试在PHP中实现类似Go语言中defer的功能。此功能的实现需要对PHP的语法解析、抽象语法树/opcode的编译、opcode指令的执行等环节进行改造，涉及的地方比较多，但是改动点比较简单，可以很好的帮助大家完整的理解PHP编译、执行两个核心阶段的实现。总体实现思路：
* __(1)语法解析：__ defer本质上还是函数调用，只是将调用时机移到了函数的最后，所以编译时可以复用调用函数的规则，但是需要与普通的调用区分开，所以我们新增一个AST节点类型，其子节点为为正常函数调用编译的AST，语法我们定义为：`defer function_name()`;
* __(2)opcode编译：__ 编译opcode时也复用调用函数的编译逻辑，不同的地方在于把defer放在最后编译，另外需要在编译return前新增一条opcode，用于执行return前跳转到defer开始的位置，在defer的最后也需要新增一条opcode，用于执行完defer后跳回return的位置；
* __(3)执行阶段：__ 执行时如果发现是return前新增的opcode则跳转到defer开始的位置，同时把return的位置记录下来，执行完defer后再跳回return。
编译后的opcode指令如下图所示：
接下来我们详细介绍下各个环节的改动，一步步实现defer功能。
__(1)语法解析__
想让PHP支持`defer function_name()`的语法首先需要修改的是词法解析规则，将"defer"关键词解析为token：T_DEFER，这样词法扫描器在匹配token时遇到"defer"将告诉语法解析器这是一个T_DEFER。这一步改动比较简单，PHP的词法解析规则定义在zend_language_scanner.l中，加入以下代码即可：
```c
"defer" {
RETURN_TOKEN(T_DEFER);
}
```
完成词法解析规则的修改后接着需要定义语法解析规则，这是非常关键的一步，语法解析器会根据配置的语法规则将PHP代码解析为抽象语法树(AST)。普通函数调用会被解析为ZEND_AST_CALL类型的AST节点，我们新增一种节点类型：ZEND_AST_DEFER_CALL，抽象语法树的节点类型为enum，定义在zend_ast.h中，同时此节点只需要一个子节点，这个子节点用于保存ZEND_AST_CALL节点，因此zend_ast.h的修改如下：
```c
enum _zend_ast_kind {
...
/* 1 child node */
...
ZEND_AST_DEFER_CALL
....
}
```
定义完AST节点后就可以在配置语法解析规则了，把defer语法解析为ZEND_AST_DEFER_CALL节点，我们把这条语法规则定义在"statement:"节点下，if、echo、for等语法都定义在此节点下，语法解析规则文件为zend_language_parser.y：
```c
statement:
'{' inner_statement_list '}' { $$ = $2; }
...
| T_DEFER function_call ';' { $$ = zend_ast_create(ZEND_AST_DEFER_CALL, $2); }
;
```
修改完这两个文件后需要分别调用re2c、yacc生成对应的C文件，具体的生成命令可以在Makefile.frag中看到：
```sh
$ re2c --no-generation-date --case-inverted -cbdFt Zend/zend_language_scanner_defs.h -oZend/zend_language_scanner.c Zend/zend_language_scanner.l
$ yacc -p zend -v -d Zend/zend_language_parser.y -oZend/zend_language_parser.c
```
执行完以后将在Zend目录下重新生成zend_language_scanner.c、zend_language_parser.c两个文件。到这一步已经完成生成抽象语法树的工作了，重新编译PHP后已经能够解析defer语法了，将会生成以下节点：
__(2)编译ZEND_AST_DEFER_CALL__
生成抽象语法树后接下来就是编译生成opcodes的操作，即从AST->Opcodes。编译ZEND_AST_DEFER_CALL节点时不能立即进行编译，需要等到当前脚本或函数全部编译完以后再进行编译，所以在编译过程需要把ZEND_AST_DEFER_CALL节点先缓存下来，参考循环结构编译时生成的zend_brk_cont_element的存储位置，我们也把ZEND_AST_DEFER_CALL节点保存在zend_op_array中，通过数组进行存储，将ZEND_AST_DEFER_CALL节点依次存入该数组，zend_op_array中加入以下几个成员：
* __last_defer:__ 整形，记录当前编译的defer数
* __defer_start_op:__ 整形，用于记录defer编译生成opcode指令的起始位置
* __defer_call_array:__ 保存ZEND_AST_DEFER_CALL节点的数组，用于保存ast节点的地址
```c
struct _zend_op_array {
...
int last_defer;
uint32_t defer_start_op;
zend_ast **defer_call_array;
}
```
修改完数据结构后接着对应修改zend_op_array初始化的过程：
```c
//zend_opcode.c
void init_op_array(zend_op_array *op_array, zend_uchar type, int initial_ops_size)
{
...
op_array->last_defer = 0;
op_array->defer_start_op = 0;
op_array->defer_call_array = NULL;
...
}
```
完成依赖的这些数据结构的改造后接下来开始编写具体的编译逻辑，也就是编译ZEND_AST_DEFER_CALL的处理。抽象语法树的编译入口函数为zend_compile_top_stmt()，然后根据不同节点的类型进行相应的编译，我们在zend_compile_stmt()函数中对ZEND_AST_DEFER_CALL节点进行编译：
```c
void zend_compile_stmt(zend_ast *ast)
{
...
switch (ast->kind) {
...
case ZEND_AST_DEFER_CALL:
zend_compile_defer_call(ast);
break
...
}
}
```
编译过程只是将ZEND_AST_DEFER_CALL的子节点(即：ZEND_AST_CALL)保存到zend_op_array->defer_call_array数组中，注意这里defer_call_array数组还没有分配内存，参考循环结构的实现，这里我们定义了一个函数用于数组的分配：
```c
//zend_compile.c
void zend_compile_defer_call(zend_ast *ast)
{
if(!ast){
return;
}
zend_ast **call_ast = NULL;
//将普通函数调用的ast节点保存到defer_call_array数组中
call_ast = get_next_defer_call(CG(active_op_array));
*call_ast = ast->child[0];
}
//zend_opcode.c
zend_ast **get_next_defer_call(zend_op_array *op_array)
{
op_array->last_defer++;
op_array->defer_call_array = erealloc(op_array->defer_call_array, sizeof(zend_ast*)*op_array->last_defer);
return &op_array->defer_call_array[op_array->last_defer-1];
}
```
既然分配了defer_call_array数组的内存就需要在zend_op_array销毁时释放：
```c
//zend_opcode.c
ZEND_API void destroy_op_array(zend_op_array *op_array)
{
...
if (op_array->defer_call_array) {
efree(op_array->defer_call_array);
}
...
}
```
编译完整个脚本或函数后，最后还会编译一条ZEND_RETURN，也就是返回指令，相当于ret指令，注意：这条opcode并不是我们在脚本中定义的return语句的，而是PHP内核为我们加的一条指令，这就是为什么有些函数我们没有写return也能返回的原因，任何函数或脚本都会生成这样一条指令。我们缓存在zend_op_array->defer_call_array数组中defer就是要在这时进行编译，也就是把defer的指令编译在最后。内核最后编译返回的这条指令由zend_emit_final_return()方法完成，我们把defer的编译放在此方法的末尾：
```c
//zend_compile.c
void zend_emit_final_return(zval *zv)
{
...
ret = zend_emit_op(NULL, returns_reference ? ZEND_RETURN_BY_REF : ZEND_RETURN, &zn, NULL);
ret->extended_value = -1;
//编译推迟执行的函数调用
zend_emit_defer_call();
}
```
前面已经说过，defer本质上就是函数调用，所以编译的过程直接复用普通函数调用的即可。另外，在编译时把起始位置记录到zend_op_array->defer_start_op中，因为在执行return前需要知道跳转到什么位置，这个值就是在那时使用的，具体的用法稍后再作说明。编译时按照倒序的顺序进行编译：
```c
//zend_compile.c
void zend_emit_defer_call()
{
if (!CG(active_op_array)->defer_call_array) {
return;
}
zend_ast *call_ast;
zend_op *nop;
znode result;
uint32_t opnum = get_next_op_number(CG(active_op_array));
int defer_num = CG(active_op_array)->last_defer;
//记录推迟的函数调用指令开始位置
CG(active_op_array)->defer_start_op = opnum;
while(--defer_num >= 0){
call_ast = CG(active_op_array)->defer_call_array[defer_num];
if (call_ast == NULL) {
continue;
}
nop = zend_emit_op(NULL, ZEND_NOP, NULL, NULL);
nop->op1.var = -2;
//编译函数调用
zend_compile_call(&result, call_ast, BP_VAR_R);
}
//compile ZEND_DEFER_CALL_END
zend_emit_op(NULL, ZEND_DEFER_CALL_END, NULL, NULL);
}
```
编译完推迟的函数调用之后，编译一条ZEND_DEFER_CALL_END指令，该指令用于执行完推迟的函数后跳回return的位置进行返回，opcode定义在zend_vm_opcodes.h中：
```c
//zend_vm_opcodes.h
#define ZEND_DEFER_CALL_END 174
```
还有一个地方你可能已经注意到，在逐个编译defer的函数调用前都生成了一条ZEND_NOP的指令，这个的目的是什么呢？开始的时候已经介绍过defer语法的特点，函数中定义的defer并不是全部执行，在return之后定义的defer是不会执行的，比如：
```go
func main(){
defer fmt.Println("A")
if 1 == 1{
return
}
defer fmt.Println("B")
}
```
这种情况下第2个defer就不会生效，因此在return前跳转的位置就不一定是zend_op_array->defer_start_op，有可能会跳过几个函数的调用，所以这里我们通过ZEND_NOP这条空指令对多个defer call进行隔离，同时为避免与其它ZEND_NOP指令混淆，增加一个判断条件：op1.var=-2。这样在return前跳转时就根据此前定义的defer数跳过部分函数的调用，如下图所示。
到这一步我们已经完成defer函数调用的编译，此时重新编译PHP后可以看到通过defer推迟的函数调用已经被编译在最后了，只不过这个时候它们不能被执行。
__(3)编译return__
编译return时需要插入一条指令用于跳转到推迟执行的函数调用指令处，因此这里需要再定义一条opcode：ZEND_DEFER_CALL，在编译过程中defer call还未编译，因此此时还无法知道具体的跳转值。
```c
//zend_vm_opcodes.h
#define ZEND_DEFER_CALL 173
#define ZEND_DEFER_CALL_END 174
```
PHP脚本中声明的return语句由zend_compile_return()方法完成编译，在编译生成ZEND_DEFER_CALL指令时还需要将当前已定义的defer数(即在return前声明的defer)记录下来，用于计算具体的跳转值。
```c
void zend_compile_return(zend_ast *ast)
{
...
//在return前编译ZEND_DEFER_CALL：用于在执行retur前跳转到defer call
if (CG(active_op_array)->defer_call_array) {
defer_zn.op_type = IS_UNUSED;
defer_zn.u.op.num = CG(active_op_array)->last_defer;
zend_emit_op(NULL, ZEND_DEFER_CALL, NULL, &defer_zn);
}
//编译正常返回的指令
opline = zend_emit_op(NULL, by_ref ? ZEND_RETURN_BY_REF : ZEND_RETURN,
&expr_node, NULL);
...
}
```
除了这种return外还有一种我们上面已经提过的return，即PHP内核编译的return指令，当PHP脚本中没有声明return语句时将执行内核添加的那条指令，因此也需要在zend_emit_final_return()加上上面的逻辑。
```c
void zend_emit_final_return(zval *zv)
{
...
//在return前编译ZEND_DEFER_CALL：用于在执行retur前跳转到defer call
if (CG(active_op_array)->defer_call_array) {
//当前return之前定义的defer数
defer_zn.op_type = IS_UNUSED;
defer_zn.u.op.num = CG(active_op_array)->last_defer;
zend_emit_op(NULL, ZEND_DEFER_CALL, NULL, &defer_zn);
}
//编译返回指令
ret = zend_emit_op(NULL, returns_reference ? ZEND_RETURN_BY_REF : ZEND_RETURN, &zn, NULL);
ret->extended_value = -1;
//编译推迟执行的函数调用
zend_emit_defer_call();
}
```
__(4)计算ZEND_DEFER_CALL指令的跳转位置__
前面我们已经完成了推迟调用函数以及return编译过程的改造，在编译完成后ZEND_DEFER_CALL指令已经能够知道具体的跳转位置了，因为推迟调用的函数已经编译完成了，所以下一步就是为全部的ZEND_DEFER_CALL指令计算跳转值。前面曾介绍过，在编译完成有一个pass_two()的环节，我们就在这里完成具体跳转位置的计算，并把跳转位置保存到ZEND_DEFER_CALL指令的操作数中，在执行阶段直接跳转到对应位置。
```c
ZEND_API int pass_two(zend_op_array *op_array)
{
zend_op *opline, *end;
...
//遍历opcode
opline = op_array->opcodes;
end = opline + op_array->last;
while (opline < end) {
switch (opline->opcode) {
...
case ZEND_DEFER_CALL: //设置jmp
{
uint32_t defer_start = op_array->defer_start_op;
//skip_defer为当前return之后声明的defer数，也就是不需要执行的defer
uint32_t skip_defer = op_array->last_defer - opline->op2.num;
//defer_opline为推迟的函数调用起始位置
zend_op *defer_opline = op_array->opcodes + defer_start;
uint32_t n = 0;
while(n <= skip_defer){
if (defer_opline->opcode == ZEND_NOP && defer_opline->op1.var == -2) {
n++;
}
defer_opline++;
defer_start++;
}
//defer_start为opcode在op_array->opcodes数组中的位置
opline->op1.opline_num = defer_start;
//将跳转位置保存到操作数op1中
ZEND_PASS_TWO_UPDATE_JMP_TARGET(op_array, opline, opline->op1);
}
break;
}
...
}
...
}
```
这里我们并没有直接编译为ZEND_JMP跳转指令，虽然ZEND_JMP可以跳转到后面的指令位置，但是最后的那条跳回return位置的指令(即：ZEND_DEFER_CALL_END)由于可能存在多个return的原因无法在编译期间确定具体的跳转值，只能在运行期间执行ZEND_DEFER_CALL时才能确定，所以需要在ZEND_DEFER_CALL指令的handler中将return的位置记录下来，执行ZEND_DEFER_CALL_END时根据这个值跳回。
__(5)定义ZEND_DEFER_CALL、ZEND_DEFER_CALL_END指令的handler__
ZEND_DEFER_CALL指令执行时需要将return的位置保存下来，我们把这个值保存到zend_execute_data结构中：
```c
//zend_compile.h
struct _zend_execute_data {
...
const zend_op *return_opline;
...
}
```
opcode的handler定义在zend_vm_def.h文件中，定义完成后需要执行`php zend_vm_gen.php`脚本生成具体的handler函数。
```c
ZEND_VM_HANDLER(173, ZEND_DEFER_CALL, ANY, ANY)
{
USE_OPLINE
//1) 将return指令的位置保存到EX(return_opline)
EX(return_opline) = opline + 1;
//2) 跳转
ZEND_VM_SET_OPCODE(OP_JMP_ADDR(opline, opline->op1));
ZEND_VM_CONTINUE();
}
ZEND_VM_HANDLER(174, ZEND_DEFER_CALL_END, ANY, ANY)
{
USE_OPLINE
ZEND_VM_SET_OPCODE(EX(return_opline));
ZEND_VM_CONTINUE();
}
```
到目前为止我们已经完成了全部的修改，重新编译PHP后就可以使用defer语法了：
```php
function shutdown($a){
echo $a."\n";
}
function test(){
$a = 1234;
defer shutdown($a);
$a = 8888;
if(1){
return "mid end\n";
}
defer shutdown("9999");
return "last end\n";
}
echo test();
```
执行后将显示：
```sh
8888
mid end
```
这里我们只实现了普通函数调用的方式，关于成员方法、静态方法、匿名函数等调用方式并未实现，留给有兴趣的读者自己去实现。
完整代码：[https://github.com/pangudashu/php-7.0.12](https://github.com/pangudashu/php-7.0.12)
';