Save and restore serialized and deserialized global variables during coroutine switching

Author: matyhtf

ext-src/php_swoole_coroutine.h

@@ -68,6 +68,19 @@ struct PHPContext {
     /* for array_walk non-reentrancy */
     zend::Function *array_walk_fci;
 #endif
+	/*
+	 * for var serialize/unserialize,
+	 * coroutine switching may occur in the __sleep/__wakeup magic method of the object
+	 */
+    unsigned serialize_lock;
+	struct {
+		struct php_serialize_data *data;
+		unsigned level;
+	} serialize;
+	struct {
+		struct php_unserialize_data *data;
+		unsigned level;
+	} unserialize;
     /* for error control `@` */
     bool in_silence;
     bool enable_scheduler;
@@ -279,6 +292,8 @@ class PHPCoroutine {
     static void restore_vm_stack(PHPContext *ctx);
     static void save_og(PHPContext *ctx);
     static void restore_og(PHPContext *ctx);
+    static void save_bg(PHPContext *ctx);
+    static void restore_bg(PHPContext *ctx);
     static void save_context(PHPContext *ctx);
     static void restore_context(PHPContext *ctx);
     static void destroy_context(PHPContext *ctx);

ext-src/swoole_coroutine.cc

@@ -25,6 +25,7 @@
 #include "swoole_signal.h"
 
 #include "zend_builtin_functions.h"
+#include "ext/standard/basic_functions.h"
 #include "ext/spl/spl_array.h"
 
 #ifdef SWOOLE_COROUTINE_MOCK_FIBER_CONTEXT
@@ -585,6 +586,30 @@ inline void PHPCoroutine::restore_og(PHPContext *ctx) {
     }
 }
 
+void PHPCoroutine::save_bg(PHPContext *ctx) {
+	if (BG(serialize_lock)) {
+		ctx->serialize_lock = BG(serialize_lock);
+	}
+	if (BG(serialize).data) {
+        memcpy(&ctx->serialize, &BG(serialize), sizeof(BG(serialize)));
+	}
+	if (BG(unserialize).data) {
+		memcpy(&ctx->unserialize, &BG(unserialize), sizeof(BG(unserialize)));
+	}
+}
+
+void PHPCoroutine::restore_bg(PHPContext *ctx) {
+	if (ctx->serialize_lock) {
+		BG(serialize_lock) = ctx->serialize_lock;
+	}
+	if (ctx->serialize.data) {
+		memcpy(&BG(serialize), &ctx->serialize, sizeof(BG(serialize)));
+	}
+	if (ctx->unserialize.data) {
+		memcpy(&BG(unserialize), &ctx->unserialize, sizeof(BG(unserialize)));
+	}
+}
+
 void PHPCoroutine::set_hook_flags(uint32_t flags) {
     zval options;
     array_init(&options);
@@ -603,11 +628,13 @@ void PHPCoroutine::set_hook_flags(uint32_t flags) {
 void PHPCoroutine::save_context(PHPContext *ctx) {
     save_vm_stack(ctx);
     save_og(ctx);
+    save_bg(ctx);
 }
 
 void PHPCoroutine::restore_context(PHPContext *ctx) {
     restore_vm_stack(ctx);
     restore_og(ctx);
+    restore_bg(ctx);
 }
 
 void PHPCoroutine::on_yield(void *arg) {