+ ZEND_ASYNC_WAKER_APPLY_ERROR

Author: EdmondDantes

Zend/zend_async_API.c

@@ -711,6 +711,32 @@ ZEND_API zend_async_waker_t * zend_async_waker_new_with_timeout(
 	return waker;
 }
 
+ZEND_API bool zend_async_waker_apply_error(zend_async_waker_t *waker, zend_object *error, bool override, bool for_cancellation)
+{
+	if (UNEXPECTED(waker == NULL)) {
+		return false;
+	}
+
+	if (EXPECTED(waker->error == NULL)) {
+		waker->error = error;
+		return true;
+	}
+
+	if (for_cancellation && instanceof_function(waker->error->ce, zend_ce_cancellation_exception)) {
+		// If the waker already has a cancellation exception, we do not override it
+		return false;
+	}
+
+	if (override) {
+		zend_exception_set_previous(error, waker->error);
+		waker->error = error;
+	} else {
+		zend_exception_set_previous(waker->error, error);
+	}
+
+	return true;
+}
+
 //////////////////////////////////////////////////////////////////////
 /* Waker API end */
 //////////////////////////////////////////////////////////////////////

Zend/zend_async_API.h

@@ -1124,9 +1124,14 @@ ZEND_API zend_async_waker_t *zend_async_waker_new(zend_coroutine_t *coroutine);
 ZEND_API zend_async_waker_t * zend_async_waker_new_with_timeout(
 	zend_coroutine_t * coroutine, const zend_ulong timeout, zend_async_event_t *cancellation
 );
+ZEND_API bool zend_async_waker_apply_error(zend_async_waker_t *waker, zend_object *error, bool override, bool for_cancellation);
 ZEND_API void zend_async_waker_destroy(zend_coroutine_t *coroutine);
 ZEND_API void zend_async_waker_add_triggered_event(zend_coroutine_t *coroutine, zend_async_event_t *event);
 
+#define ZEND_ASYNC_WAKER_APPLY_ERROR(waker, error) zend_async_waker_apply_error((waker), (error), true, false)
+#define ZEND_ASYNC_WAKER_APPEND_ERROR(waker, error) zend_async_waker_apply_error((waker), (error), false, false)
+#define ZEND_ASYNC_WAKER_APPLY_CANCELLATION(waker, error) zend_async_waker_apply_error((waker), (error), true, true)
+
 ZEND_API void zend_async_resume_when(
 		zend_coroutine_t			*coroutine,
 		zend_async_event_t			*event,