Kernel: Better handling of allocation failure in profiling

If we can't allocate a PerformanceEventBuffer to store the profiling
events, we now fail sys$profiling_enable() and sys$perf_event()
with ENOMEM instead of carrying on with a broken buffer.

2 files changed, 5 insertions, 2 deletions

diff --git a/Kernel/Syscalls/perf_event.cpp b/Kernel/Syscalls/perf_event.cpp
index 36229bc232..3e90b95a77 100644
--- a/Kernel/Syscalls/perf_event.cpp
+++ b/Kernel/Syscalls/perf_event.cpp
@@ -31,7 +31,9 @@ namespace Kernel {
 
 KResultOr<int> Process::sys$perf_event(int type, FlatPtr arg1, FlatPtr arg2)
 {
-    return ensure_perf_events().append(type, arg1, arg2);
+    if (!create_perf_events_buffer_if_needed())
+        return ENOMEM;
+    return perf_events()->append(type, arg1, arg2);
 }
 
 }
diff --git a/Kernel/Syscalls/profiling.cpp b/Kernel/Syscalls/profiling.cpp
index c02cf73215..42e1381697 100644
--- a/Kernel/Syscalls/profiling.cpp
+++ b/Kernel/Syscalls/profiling.cpp
@@ -43,7 +43,8 @@ KResultOr<int> Process::sys$profiling_enable(pid_t pid)
         return ESRCH;
     if (!is_superuser() && process->uid() != m_euid)
         return EPERM;
-    process->ensure_perf_events();
+    if (!process->create_perf_events_buffer_if_needed())
+        return ENOMEM;
     process->set_profiling(true);
     return 0;
 }