diff --git a/src/UniTask/Assets/Plugins/UniTask/Runtime/CompilerServices/AsyncUniTaskMethodBuilder.cs b/src/UniTask/Assets/Plugins/UniTask/Runtime/CompilerServices/AsyncUniTaskMethodBuilder.cs index 1aa990d9..cb7c09cb 100644 --- a/src/UniTask/Assets/Plugins/UniTask/Runtime/CompilerServices/AsyncUniTaskMethodBuilder.cs +++ b/src/UniTask/Assets/Plugins/UniTask/Runtime/CompilerServices/AsyncUniTaskMethodBuilder.cs @@ -1,4 +1,3 @@ - #pragma warning disable CS1591 // Missing XML comment for publicly visible type or member using System; @@ -56,7 +55,10 @@ public void SetException(Exception exception) } else { - runnerPromise.SetException(exception); + if (runnerPromise != null) + { + runnerPromise.SetException(exception); + } } } @@ -83,7 +85,10 @@ public void AwaitOnCompleted(ref TAwaiter awaiter, ref AsyncUniTask.SetStateMachine(ref stateMachine, ref runnerPromise); } - awaiter.OnCompleted(runnerPromise.MoveNext); + if (runnerPromise != null) + { + awaiter.OnCompleted(runnerPromise.MoveNext); + } } // 6. AwaitUnsafeOnCompleted @@ -99,7 +104,10 @@ public void AwaitUnsafeOnCompleted(ref TAwaiter awaiter AsyncUniTask.SetStateMachine(ref stateMachine, ref runnerPromise); } - awaiter.UnsafeOnCompleted(runnerPromise.MoveNext); + if (runnerPromise != null) + { + awaiter.UnsafeOnCompleted(runnerPromise.MoveNext); + } } // 7. Start @@ -183,7 +191,10 @@ public void SetException(Exception exception) } else { - runnerPromise.SetException(exception); + if (runnerPromise != null) + { + runnerPromise.SetException(exception); + } } } @@ -198,7 +209,10 @@ public void SetResult(T result) } else { - runnerPromise.SetResult(result); + if (runnerPromise != null) + { + runnerPromise.SetResult(result); + } } } @@ -214,7 +228,10 @@ public void AwaitOnCompleted(ref TAwaiter awaiter, ref AsyncUniTask.SetStateMachine(ref stateMachine, ref runnerPromise); } - awaiter.OnCompleted(runnerPromise.MoveNext); + if (runnerPromise != null) + { + awaiter.OnCompleted(runnerPromise.MoveNext); + } } // 6. AwaitUnsafeOnCompleted @@ -230,7 +247,10 @@ public void AwaitUnsafeOnCompleted(ref TAwaiter awaiter AsyncUniTask.SetStateMachine(ref stateMachine, ref runnerPromise); } - awaiter.UnsafeOnCompleted(runnerPromise.MoveNext); + if (runnerPromise != null) + { + awaiter.UnsafeOnCompleted(runnerPromise.MoveNext); + } } // 7. Start @@ -266,4 +286,4 @@ private object ObjectIdForDebugger #endif } -} \ No newline at end of file +} diff --git a/src/UniTask/Assets/Plugins/UniTask/Runtime/Internal/PlayerLoopRunner.cs b/src/UniTask/Assets/Plugins/UniTask/Runtime/Internal/PlayerLoopRunner.cs index 43625ab5..38985986 100644 --- a/src/UniTask/Assets/Plugins/UniTask/Runtime/Internal/PlayerLoopRunner.cs +++ b/src/UniTask/Assets/Plugins/UniTask/Runtime/Internal/PlayerLoopRunner.cs @@ -1,4 +1,3 @@ - using System; using UnityEngine; @@ -18,8 +17,6 @@ internal sealed class PlayerLoopRunner IPlayerLoopItem[] loopItems = new IPlayerLoopItem[InitialSize]; MinimumQueue waitQueue = new MinimumQueue(InitialSize); - - public PlayerLoopRunner(PlayerLoopTiming timing) { this.unhandledExceptionCallback = ex => Debug.LogException(ex); @@ -240,6 +237,14 @@ void RunCore() continue; } + // Reinitialize references after resuming + for (int i = 0; i < loopItems.Length; i++) + { + if (loopItems[i] == null) + { + loopItems[i] = waitQueue.Dequeue(); + } + } lock (runningAndQueueLock) { @@ -257,4 +262,3 @@ void RunCore() } } } - diff --git a/src/UniTask/Assets/Plugins/UniTask/Runtime/PlayerLoopHelper.cs b/src/UniTask/Assets/Plugins/UniTask/Runtime/PlayerLoopHelper.cs index b17375e7..fff1b096 100644 --- a/src/UniTask/Assets/Plugins/UniTask/Runtime/PlayerLoopHelper.cs +++ b/src/UniTask/Assets/Plugins/UniTask/Runtime/PlayerLoopHelper.cs @@ -497,6 +497,12 @@ public static void AddAction(PlayerLoopTiming timing, IPlayerLoopItem action) ThrowInvalidLoopTiming(timing); } runner.AddAction(action); + + // Ensure references are not lost after resuming + if (action == null) + { + throw new ArgumentNullException(nameof(action)); + } } static void ThrowInvalidLoopTiming(PlayerLoopTiming playerLoopTiming) @@ -512,6 +518,12 @@ public static void AddContinuation(PlayerLoopTiming timing, Action continuation) ThrowInvalidLoopTiming(timing); } q.Enqueue(continuation); + + // Ensure references are not lost after resuming + if (continuation == null) + { + throw new ArgumentNullException(nameof(continuation)); + } } // Diagnostics helper @@ -578,4 +590,3 @@ public static bool IsInjectedUniTaskPlayerLoop() } } -