pm: Only register privileged processes with FS as needed

Author: SciresM

stratosphere/pm/pm.json

@@ -65,6 +65,7 @@
                 "svcReplyAndReceiveWithUserBuffer":	"0x44",
                 "svcCreateEvent":	"0x45",
                 "svcSetUnsafeLimit":	"0x4a",
+                "svcGetProcessList":	"0x65",
                 "svcStartProcess":	"0x7a",
                 "svcTerminateProcess":	"0x7b",
                 "svcGetProcessInfo":	"0x7c",

stratosphere/pm/source/pm_main.cpp

@@ -65,9 +65,21 @@ void RegisterPrivilegedProcessesWithFs() {
     const u32 PRIVILEGED_FAH[0x1C/sizeof(u32)] = {0x00000001, 0x00000000, 0x80000000, 0x0000001C, 0x00000000, 0x0000001C, 0x00000000};
     const u32 PRIVILEGED_FAC[0x2C/sizeof(u32)] = {0x00000001, 0x00000000, 0x80000000, 0x00000000, 0x00000000, 0xFFFFFFFF, 0xFFFFFFFF, 0x00000000, 0x00000000, 0xFFFFFFFF, 0xFFFFFFFF};
 
-    for (u64 pid = PRIVILEGED_PROCESS_MIN; pid <= PRIVILEGED_PROCESS_MAX; pid++) {
-        fsprUnregisterProgram(pid);
-        fsprRegisterProgram(pid, pid, FsStorageId_NandSystem,  PRIVILEGED_FAH, sizeof(PRIVILEGED_FAH), PRIVILEGED_FAC, sizeof(PRIVILEGED_FAC));
+    u32 num_pids;
+    u64 pids[PRIVILEGED_PROCESS_MAX+1];
+    if (R_SUCCEEDED(svcGetProcessList(&num_pids, pids, sizeof(pids)/sizeof(pids[0])))) {
+        for (u32 i = 0; i < num_pids; i++) {
+            const u64 pid = pids[i];
+            if (PRIVILEGED_PROCESS_MIN <= pid && pid <= PRIVILEGED_PROCESS_MAX) {
+                fsprUnregisterProgram(pid);
+                fsprRegisterProgram(pid, pid, FsStorageId_NandSystem,  PRIVILEGED_FAH, sizeof(PRIVILEGED_FAH), PRIVILEGED_FAC, sizeof(PRIVILEGED_FAC));
+            }
+        }
+    } else {
+        for (u64 pid = PRIVILEGED_PROCESS_MIN; pid <= PRIVILEGED_PROCESS_MAX; pid++) {
+            fsprUnregisterProgram(pid);
+            fsprRegisterProgram(pid, pid, FsStorageId_NandSystem,  PRIVILEGED_FAH, sizeof(PRIVILEGED_FAH), PRIVILEGED_FAC, sizeof(PRIVILEGED_FAC));
+        }
     }
 }
 

stratosphere/pm/source/pm_registration.cpp

@@ -292,15 +292,15 @@ void Registration::FinalizeExitedProcess(std::shared_ptr<Registration::Process>
     bool signal_debug_process_5x = kernelAbove500() && process->flags & PROCESSFLAGS_NOTIFYWHENEXITED;
     /* Unregister with FS. */
     if (R_FAILED(fsprUnregisterProgram(process->pid))) {
-        /* TODO: Panic. */
+        std::abort();
     }
     /* Unregister with SM. */
     if (R_FAILED(smManagerUnregisterProcess(process->pid))) {
-        /* TODO: Panic. */
+        std::abort();
     }
     /* Unregister with LDR. */
     if (R_FAILED(ldrPmUnregisterTitle(process->ldr_queue_index))) {
-        /* TODO: Panic. */
+        std::abort();
     }
 
     /* Close the process's handle. */