Fix nullsafe operator on delayed oplines

Author: iluuu1994

Zend/tests/nullsafe_operator/034.phpt

@@ -0,0 +1,28 @@
+--TEST--
+Test nullsafe operator on delayed dim
+--FILE--
+<?php
+
+$arr = [
+    'foo' => null,
+    'bar' => [
+        'baz' => null,
+    ],
+];
+
+var_dump($arr['foo']?->something);
+var_dump($arr['invalid']?->something);
+
+var_dump($arr['bar']['baz']?->something);
+var_dump($arr['bar']['invalid']?->something);
+
+?>
+--EXPECTF--
+NULL
+
+Warning: Undefined array key "invalid" in %s.php on line 11
+NULL
+NULL
+
+Warning: Undefined array key "invalid" in %s.php on line 14
+NULL

Zend/tests/nullsafe_operator/035.phpt

@@ -0,0 +1,27 @@
+--TEST--
+Test nullsafe operator on delayed var
+--FILE--
+<?php
+
+class Foo {
+    public ?Bar $bar;
+}
+
+class Bar {
+    public string $baz;
+}
+
+$foo = new Foo();
+
+$foo->bar = null;
+var_dump($foo->bar?->baz);
+
+$bar = new Bar();
+$bar->baz = 'baz';
+$foo->bar = $bar;
+var_dump($foo->bar?->baz);
+
+?>
+--EXPECT--
+NULL
+string(3) "baz"

Zend/tests/nullsafe_operator/036.phpt

@@ -0,0 +1,28 @@
+--TEST--
+Test nullsafe method call on delayed var
+--FILE--
+<?php
+
+class Foo {
+    public ?Bar $bar;
+}
+
+class Bar {
+    public function baz() {
+        return 'baz';
+    }
+}
+
+$foo = new Foo();
+
+$foo->bar = null;
+var_dump($foo->bar?->baz());
+
+$bar = new Bar();
+$foo->bar = $bar;
+var_dump($foo->bar?->baz());
+
+?>
+--EXPECT--
+NULL
+string(3) "baz"

Zend/tests/nullsafe_operator/037.phpt

@@ -0,0 +1,15 @@
+--TEST--
+Test nullsafe operator in netsted delayed dims
+--FILE--
+<?php
+
+$foo = new stdClass();
+$foo->bar = 'bar';
+
+$array = ['foo' => ['bar' => 'baz']];
+
+var_dump($array['foo'][$foo?->bar]);
+
+?>
+--EXPECT--
+string(3) "baz"

Zend/zend.c

@@ -1303,6 +1303,7 @@ static ZEND_COLD void zend_error_impl(
 	zend_class_entry *saved_class_entry;
 	zend_stack loop_var_stack;
 	zend_stack delayed_oplines_stack;
+	zend_stack delayed_oplines_offset_stack;
 	int type = orig_type & E_ALL;
 
 	/* Report about uncaught exception in case of fatal errors */
@@ -1381,6 +1382,7 @@ static ZEND_COLD void zend_error_impl(
 				CG(active_class_entry) = NULL;
 				SAVE_STACK(loop_var_stack);
 				SAVE_STACK(delayed_oplines_stack);
+				SAVE_STACK(delayed_oplines_offset_stack);
 				CG(in_compilation) = 0;
 			}
 
@@ -1400,6 +1402,7 @@ static ZEND_COLD void zend_error_impl(
 				CG(active_class_entry) = saved_class_entry;
 				RESTORE_STACK(loop_var_stack);
 				RESTORE_STACK(delayed_oplines_stack);
+				RESTORE_STACK(delayed_oplines_offset_stack);
 				CG(in_compilation) = 1;
 			}
 

Zend/zend_compile.c

@@ -368,6 +368,7 @@ void zend_init_compiler_data_structures(void) /* {{{ */
 {
 	zend_stack_init(&CG(loop_var_stack), sizeof(zend_loop_var));
 	zend_stack_init(&CG(delayed_oplines_stack), sizeof(zend_op));
+	zend_stack_init(&CG(delayed_oplines_offset_stack), sizeof(uint32_t));
 	zend_stack_init(&CG(short_circuiting_opnums), sizeof(uint32_t));
 	CG(active_class_entry) = NULL;
 	CG(in_compilation) = 0;
@@ -422,6 +423,7 @@ void shutdown_compiler(void) /* {{{ */
 {
 	zend_stack_destroy(&CG(loop_var_stack));
 	zend_stack_destroy(&CG(delayed_oplines_stack));
+	zend_stack_destroy(&CG(delayed_oplines_offset_stack));
 	zend_stack_destroy(&CG(short_circuiting_opnums));
 	zend_hash_destroy(&CG(filenames_table));
 	zend_arena_destroy(CG(arena));
@@ -2208,7 +2210,9 @@ static inline zend_op *zend_delayed_emit_op(znode *result, zend_uchar opcode, zn
 
 static inline uint32_t zend_delayed_compile_begin(void) /* {{{ */
 {
-	return zend_stack_count(&CG(delayed_oplines_stack));
+	uint32_t offset = zend_stack_count(&CG(delayed_oplines_stack));
+	zend_stack_push(&CG(delayed_oplines_offset_stack), &offset);
+	return offset;
 }
 /* }}} */
 
@@ -2223,6 +2227,7 @@ static zend_op *zend_delayed_compile_end(uint32_t offset) /* {{{ */
 		memcpy(opline, &oplines[i], sizeof(zend_op));
 	}
 	CG(delayed_oplines_stack).top = offset;
+	CG(delayed_oplines_offset_stack).top -= 1;
 	return opline;
 }
 /* }}} */
@@ -2305,10 +2310,20 @@ static void zend_short_circuiting_commit(uint32_t checkpoint, znode *result, zen
 	}
 }
 
-static void zend_emit_jmp_null(znode *obj_node)
+static void zend_emit_jmp_null(znode *obj_node, zend_bool delay)
 {
 	uint32_t jmp_null_opnum = get_next_op_number();
-	zend_op *opline = zend_emit_op(NULL, ZEND_JMP_NULL, obj_node, NULL);
+	zend_op *opline;
+
+	if (delay) {
+		uint32_t delayed_oplines_count = zend_stack_count(&CG(delayed_oplines_stack));
+		uint32_t delayed_oplines_offset = *(uint32_t *)zend_stack_top(&CG(delayed_oplines_offset_stack));
+		jmp_null_opnum += (delayed_oplines_count - delayed_oplines_offset);
+		opline = zend_delayed_emit_op(NULL, ZEND_JMP_NULL, obj_node, NULL);
+	} else {
+		opline = zend_emit_op(NULL, ZEND_JMP_NULL, obj_node, NULL);
+	}
+
 	if (opline->op1_type == IS_CONST) {
 		Z_TRY_ADDREF_P(CT_CONSTANT(opline->op1));
 	}
@@ -2796,7 +2811,7 @@ static zend_op *zend_delayed_compile_prop(znode *result, zend_ast *ast, uint32_t
 		opline = zend_delayed_compile_var(&obj_node, obj_ast, type, 0);
 		zend_separate_if_call_and_write(&obj_node, obj_ast, type);
 		if (nullsafe) {
-			zend_emit_jmp_null(&obj_node);
+			zend_emit_jmp_null(&obj_node, 1);
 		}
 	}
 
@@ -4356,7 +4371,7 @@ void zend_compile_method_call(znode *result, zend_ast *ast, uint32_t type) /* {{
 		zend_short_circuiting_mark_inner(obj_ast);
 		zend_compile_expr(&obj_node, obj_ast);
 		if (nullsafe) {
-			zend_emit_jmp_null(&obj_node);
+			zend_emit_jmp_null(&obj_node, 0);
 		}
 	}
 

Zend/zend_globals.h

@@ -119,6 +119,7 @@ struct _zend_compiler_globals {
 	zend_arena *ast_arena;
 
 	zend_stack delayed_oplines_stack;
+	zend_stack delayed_oplines_offset_stack;
 	HashTable *memoized_exprs;
 	int memoize_mode;