gthread: Check if memory allocation was successful

devkitPro · Sep 3, 2022 · 815dff3 · 815dff3
1 parent 0edd2e9
commit 815dff3
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/libraries/wutstdc++/wut_gthread_thread.cpp b/libraries/wutstdc++/wut_gthread_thread.cpp
@@ -24,9 +24,16 @@ __wut_thread_create(OSThread **outThread,
                     void *entryArgs)
 {
    OSThread *thread = (OSThread *)memalign(16, sizeof(OSThread));
-   char *stack = (char *)memalign(16, __WUT_STACK_SIZE);
+   if (!thread) {
+      return ENOMEM;
+   }   
    memset(thread, 0, sizeof(OSThread));
 
+   char *stack = (char *)memalign(16, __WUT_STACK_SIZE);
+   if (!stack) {
+      return ENOMEM;
+   }
+
    if (!OSCreateThread(thread,
                        (OSThreadEntryPointFn)entryPoint,
                        (int)entryArgs,