[NTOSKRNL] Add a raw implementation of !irpfind in kdbg

[reactos.git] / ntoskrnl / ps / quota.c

diff --git a/ntoskrnl/ps/quota.c b/ntoskrnl/ps/quota.c
index 09701ba..b813190 100644 (file)
--- a/ntoskrnl/ps/quota.c
+++ b/ntoskrnl/ps/quota.c
@@ -95,9 +95,9 @@ PspReturnProcessQuotaSpecifiedPool(IN PEPROCESS Process,
 
 /* FUNCTIONS ***************************************************************/
 
+INIT_FUNCTION
 VOID
 NTAPI
-INIT_FUNCTION
 PsInitializeQuotaSystem(VOID)
 {
     RtlZeroMemory(&PspDefaultQuotaBlock, sizeof(PspDefaultQuotaBlock));
@@ -335,7 +335,7 @@ PspSetQuotaLimits(
                  (CapturedQuotaLimits.Flags & QUOTA_LIMITS_HARDWS_MAX_DISABLE)))
             {
                 DPRINT1("Invalid quota flags: 0x%lx\n", CapturedQuotaLimits.Flags);
-                return STATUS_INVALID_PARAMETER;
+                _SEH2_YIELD(return STATUS_INVALID_PARAMETER);
             }
 
             /* Verify that the caller didn't pass reserved values */
@@ -351,19 +351,19 @@ PspSetQuotaLimits(
                         CapturedQuotaLimits.Reserved3,
                         CapturedQuotaLimits.Reserved4,
                         CapturedQuotaLimits.CpuRateLimit.RateData);
-                return STATUS_INVALID_PARAMETER;
+                _SEH2_YIELD(return STATUS_INVALID_PARAMETER);
             }
         }
         else
         {
             DPRINT1("Invalid quota size: 0x%lx\n", QuotaLimitsLength);
-            return STATUS_INFO_LENGTH_MISMATCH;
+            _SEH2_YIELD(return STATUS_INFO_LENGTH_MISMATCH);
         }
     }
     _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
     {
         DPRINT1("Exception while copying data\n");
-        return _SEH2_GetExceptionCode();
+        _SEH2_YIELD(return _SEH2_GetExceptionCode());
     }
     _SEH2_END;
 
@@ -444,6 +444,10 @@ PspSetQuotaLimits(
 
         Status = STATUS_SUCCESS;
     }
+    else
+    {
+        Status = STATUS_SUCCESS;
+    }
 
     return Status;
 }