Add missing processor architecture cases
[reactos.git] / reactos / ntoskrnl / se / priv.c
index cfa4c3f..752279e 100644 (file)
@@ -1,21 +1,23 @@
-/* $Id: priv.c,v 1.6 2003/05/31 11:10:30 ekohl Exp $
+/* $Id$
  *
- * COPYRIGHT:         See COPYING in the top level directory
- * PROJECT:           ReactOS kernel
- * PURPOSE:           Security manager
- * FILE:              kernel/se/priv.c
- * PROGRAMER:         ?
- * REVISION HISTORY:
- *                 26/07/98: Added stubs for security functions
+ * COPYRIGHT:       See COPYING in the top level directory
+ * PROJECT:         ReactOS kernel
+ * FILE:            ntoskrnl/se/priv.c
+ * PURPOSE:         Security manager
+ *
+ * PROGRAMMERS:     No programmer listed.
  */
 
 /* INCLUDES *****************************************************************/
 
-#include <ddk/ntddk.h>
-#include <internal/se.h>
-
+#include <ntoskrnl.h>
+#define NDEBUG
 #include <internal/debug.h>
 
+#if defined (ALLOC_PRAGMA)
+#pragma alloc_text(INIT, SepInitPrivileges)
+#endif
+
 
 /* GLOBALS *******************************************************************/
 
@@ -42,11 +44,16 @@ LUID SeAuditPrivilege;
 LUID SeSystemEnvironmentPrivilege;
 LUID SeChangeNotifyPrivilege;
 LUID SeRemoteShutdownPrivilege;
+LUID SeUndockPrivilege;
+LUID SeSyncAgentPrivilege;
+LUID SeEnableDelegationPrivilege;
 
 
 /* FUNCTIONS ***************************************************************/
 
 VOID
+INIT_FUNCTION
+NTAPI
 SepInitPrivileges (VOID)
 {
   SeCreateTokenPrivilege.LowPart = SE_CREATE_TOKEN_PRIVILEGE;
@@ -95,57 +102,67 @@ SepInitPrivileges (VOID)
   SeChangeNotifyPrivilege.HighPart = 0;
   SeRemoteShutdownPrivilege.LowPart = SE_REMOTE_SHUTDOWN_PRIVILEGE;
   SeRemoteShutdownPrivilege.HighPart = 0;
+  SeUndockPrivilege.LowPart = SE_UNDOCK_PRIVILEGE;
+  SeUndockPrivilege.HighPart = 0;
+  SeSyncAgentPrivilege.LowPart = SE_SYNC_AGENT_PRIVILEGE;
+  SeSyncAgentPrivilege.HighPart = 0;
+  SeEnableDelegationPrivilege.LowPart = SE_ENABLE_DELEGATION_PRIVILEGE;
+  SeEnableDelegationPrivilege.HighPart = 0;
 }
 
 
 BOOLEAN
-SepPrivilegeCheck (PACCESS_TOKEN Token,
+NTAPI
+SepPrivilegeCheck (PTOKEN Token,
                   PLUID_AND_ATTRIBUTES Privileges,
                   ULONG PrivilegeCount,
                   ULONG PrivilegeControl,
                   KPROCESSOR_MODE PreviousMode)
 {
-  PLUID_AND_ATTRIBUTES Current;
   ULONG i;
   ULONG j;
   ULONG k;
 
+  DPRINT ("SepPrivilegeCheck() called\n");
+
+  PAGED_CODE();
+
   if (PreviousMode == KernelMode)
     {
       return TRUE;
     }
 
-  j = 0;
-  if (PrivilegeCount != 0)
+  k = 0;
+  if (PrivilegeCount > 0)
     {
-      k = PrivilegeCount;
-      do
+      for (i = 0; i < Token->PrivilegeCount; i++)
        {
-         i = Token->PrivilegeCount;
-         Current = Token->Privileges;
-         for (i = 0; i < Token->PrivilegeCount; i++)
+         for (j = 0; j < PrivilegeCount; j++)
            {
-             if (!(Current[i].Attributes & SE_PRIVILEGE_ENABLED) &&
-                 Privileges[i].Luid.LowPart == Current[i].Luid.LowPart &&
-                 Privileges[i].Luid.HighPart == Current[i].Luid.HighPart)
+             if (Token->Privileges[i].Luid.LowPart == Privileges[j].Luid.LowPart &&
+                 Token->Privileges[i].Luid.HighPart == Privileges[j].Luid.HighPart)
                {
-                 Privileges[i].Attributes |= SE_PRIVILEGE_USED_FOR_ACCESS;
-                 j++;
-                 break;
+                 DPRINT ("Found privilege\n");
+                 DPRINT ("Privilege attributes %lx\n",
+                         Token->Privileges[i].Attributes);
+
+                 if (Token->Privileges[i].Attributes & SE_PRIVILEGE_ENABLED)
+                   {
+                     Privileges[j].Attributes |= SE_PRIVILEGE_USED_FOR_ACCESS;
+                     k++;
+                   }
                }
            }
-         k--;
        }
-      while (k > 0);
     }
 
   if ((PrivilegeControl & PRIVILEGE_SET_ALL_NECESSARY) &&
-      PrivilegeCount == j)
+      PrivilegeCount == k)
     {
       return TRUE;
     }
 
-  if (j > 0 &&
+  if (k > 0 &&
       !(PrivilegeControl & PRIVILEGE_SET_ALL_NECESSARY))
     {
       return TRUE;
@@ -156,6 +173,7 @@ SepPrivilegeCheck (PACCESS_TOKEN Token,
 
 
 NTSTATUS
+NTAPI
 SeCaptureLuidAndAttributesArray (PLUID_AND_ATTRIBUTES Src,
                                 ULONG PrivilegeCount,
                                 KPROCESSOR_MODE PreviousMode,
@@ -166,168 +184,202 @@ SeCaptureLuidAndAttributesArray (PLUID_AND_ATTRIBUTES Src,
                                 PLUID_AND_ATTRIBUTES* Dest,
                                 PULONG Length)
 {
-   PLUID_AND_ATTRIBUTES* NewMem;
-   ULONG SrcLength;
-   
-   if (PrivilegeCount == 0)
-     {
-       *Dest = 0;
-       *Length = 0;
-       return(STATUS_SUCCESS);
-     }
-   if (PreviousMode == 0 && d == 0)
-     {
-       *Dest = Src;
-       return(STATUS_SUCCESS);
-     }
-   SrcLength = ((PrivilegeCount * sizeof(LUID_AND_ATTRIBUTES)) + 3) & 0xfc;
-   *Length = SrcLength;
-   if (AllocatedMem == NULL)
-     {
-       NewMem = ExAllocatePool(PoolType, SrcLength);
-       *Dest =  (PLUID_AND_ATTRIBUTES)NewMem;
-       if (NewMem == NULL)
-         {
-            return(STATUS_UNSUCCESSFUL);
-         }
-     }
-   else
-     {
-       if (SrcLength > AllocatedLength)
-         {
-            return(STATUS_UNSUCCESSFUL);
-         }
-       *Dest = AllocatedMem;
-     }
-   memmove(*Dest, Src, SrcLength);
-   return(STATUS_SUCCESS);
+  PLUID_AND_ATTRIBUTES* NewMem;
+  ULONG SrcLength;
+
+  PAGED_CODE();
+
+  if (PrivilegeCount == 0)
+    {
+      *Dest = 0;
+      *Length = 0;
+      return STATUS_SUCCESS;
+    }
+
+  if (PreviousMode == KernelMode && d == 0)
+    {
+      *Dest = Src;
+      return STATUS_SUCCESS;
+    }
+
+  SrcLength = ((PrivilegeCount * sizeof(LUID_AND_ATTRIBUTES)) + 3) & 0xfc;
+  *Length = SrcLength;
+  if (AllocatedMem == NULL)
+    {
+      NewMem = ExAllocatePool (PoolType,
+                              SrcLength);
+      *Dest = (PLUID_AND_ATTRIBUTES)NewMem;
+      if (NewMem == NULL)
+       {
+         return STATUS_UNSUCCESSFUL;
+       }
+    }
+  else
+    {
+      if (SrcLength > AllocatedLength)
+       {
+         return STATUS_UNSUCCESSFUL;
+       }
+      *Dest = AllocatedMem;
+    }
+  memmove (*Dest, Src, SrcLength);
+
+  return STATUS_SUCCESS;
 }
 
 
 VOID
-SeReleaseLuidAndAttributesArray(PLUID_AND_ATTRIBUTES Privilege,
-                               KPROCESSOR_MODE PreviousMode,
-                               ULONG a)
+NTAPI
+SeReleaseLuidAndAttributesArray (PLUID_AND_ATTRIBUTES Privilege,
+                                KPROCESSOR_MODE PreviousMode,
+                                ULONG a)
 {
-  ExFreePool(Privilege);
+  PAGED_CODE();
+
+  ExFreePool (Privilege);
 }
 
 
 NTSTATUS STDCALL
-NtPrivilegeCheck(IN HANDLE ClientToken,
-                IN PPRIVILEGE_SET RequiredPrivileges,
-                IN PBOOLEAN Result)
+NtPrivilegeCheck (IN HANDLE ClientToken,
+                 IN PPRIVILEGE_SET RequiredPrivileges,
+                 IN PBOOLEAN Result)
 {
-   NTSTATUS Status;
-   PACCESS_TOKEN Token;
-   ULONG PrivilegeCount;
-   BOOLEAN TResult;
-   ULONG PrivilegeControl;
-   PLUID_AND_ATTRIBUTES Privilege;
-   ULONG Length;
-   
-   Status = ObReferenceObjectByHandle(ClientToken,
-                                     0,
+  PLUID_AND_ATTRIBUTES Privilege;
+  PTOKEN Token;
+  ULONG PrivilegeCount;
+  ULONG PrivilegeControl;
+  ULONG Length;
+  KPROCESSOR_MODE PreviousMode;
+  NTSTATUS Status;
+
+  PAGED_CODE();
+
+  PreviousMode = KeGetPreviousMode();
+
+  Status = ObReferenceObjectByHandle (ClientToken,
+                                     TOKEN_QUERY,
                                      SepTokenObjectType,
-                                     UserMode,
+                                     PreviousMode,
                                      (PVOID*)&Token,
                                      NULL);
-   if (!NT_SUCCESS(Status))
-     {
-       return(Status);
-     }
-   if (Token->TokenType == TokenImpersonation &&
-       Token->ImpersonationLevel < SecurityAnonymous)
-     {
-       ObDereferenceObject(Token);
-       return(STATUS_UNSUCCESSFUL);
-     }
-   PrivilegeCount = RequiredPrivileges->PrivilegeCount;
-   PrivilegeControl = RequiredPrivileges->Control;
-   Privilege = 0;
-   Status = SeCaptureLuidAndAttributesArray(RequiredPrivileges->Privilege,
+  if (!NT_SUCCESS(Status))
+    {
+      return Status;
+    }
+
+  if (Token->TokenType == TokenImpersonation &&
+      Token->ImpersonationLevel < SecurityAnonymous)
+    {
+      ObDereferenceObject (Token);
+      return STATUS_UNSUCCESSFUL;
+    }
+
+  PrivilegeCount = RequiredPrivileges->PrivilegeCount;
+  PrivilegeControl = RequiredPrivileges->Control;
+  Privilege = 0;
+  Status = SeCaptureLuidAndAttributesArray (RequiredPrivileges->Privilege,
                                            PrivilegeCount,
-                                           1,
+                                           UserMode,
+                                           NULL,
                                            0,
-                                           0,
-                                           1,
+                                           PagedPool,
                                            1,
                                            &Privilege,
                                            &Length);
-   if (!NT_SUCCESS(Status))
-     {
-       ObDereferenceObject(Token);
-       return(STATUS_UNSUCCESSFUL);
-     }
-   TResult = SepPrivilegeCheck(Token,
+  if (!NT_SUCCESS(Status))
+    {
+      ObDereferenceObject (Token);
+      return STATUS_UNSUCCESSFUL;
+    }
+
+  *Result = SepPrivilegeCheck (Token,
                               Privilege,
                               PrivilegeCount,
                               PrivilegeControl,
                               UserMode);
-   memmove(RequiredPrivileges->Privilege, Privilege, Length);
-   *Result = TResult;
-   SeReleaseLuidAndAttributesArray(Privilege, UserMode, 1);
-   return(STATUS_SUCCESS);
+
+  memmove (RequiredPrivileges->Privilege,
+          Privilege,
+          Length);
+
+  SeReleaseLuidAndAttributesArray (Privilege,
+                                  UserMode,
+                                  1);
+
+  return STATUS_SUCCESS;
 }
 
 
+/*
+ * @implemented
+ */
 BOOLEAN STDCALL
-SePrivilegeCheck(PPRIVILEGE_SET Privileges,
-                PSECURITY_SUBJECT_CONTEXT SubjectContext,
-                KPROCESSOR_MODE PreviousMode)
+SePrivilegeCheck (PPRIVILEGE_SET Privileges,
+                 PSECURITY_SUBJECT_CONTEXT SubjectContext,
+                 KPROCESSOR_MODE PreviousMode)
 {
-   PACCESS_TOKEN Token = NULL;
-   
-   if (SubjectContext->ClientToken == NULL)
-     {
-       Token = SubjectContext->PrimaryToken;
-     }
-   else
-     {
-       Token = SubjectContext->ClientToken;
-       if (SubjectContext->ImpersonationLevel < 2)
-         {
-            return(FALSE);
-         }
-     }
-   
-   return(SepPrivilegeCheck(Token,
+  PACCESS_TOKEN Token = NULL;
+
+  PAGED_CODE();
+
+  if (SubjectContext->ClientToken == NULL)
+    {
+      Token = SubjectContext->PrimaryToken;
+    }
+  else
+    {
+      Token = SubjectContext->ClientToken;
+      if (SubjectContext->ImpersonationLevel < 2)
+       {
+         return FALSE;
+       }
+    }
+
+  return SepPrivilegeCheck (Token,
                            Privileges->Privilege,
                            Privileges->PrivilegeCount,
                            Privileges->Control,
-                           PreviousMode));
+                           PreviousMode);
 }
 
 
+/*
+ * @implemented
+ */
 BOOLEAN STDCALL
-SeSinglePrivilegeCheck(IN LUID PrivilegeValue,
-                      IN KPROCESSOR_MODE PreviousMode)
+SeSinglePrivilegeCheck (IN LUID PrivilegeValue,
+                       IN KPROCESSOR_MODE PreviousMode)
 {
-   SECURITY_SUBJECT_CONTEXT SubjectContext;
-   BOOLEAN r;
-   PRIVILEGE_SET Priv;
-   
-   SeCaptureSubjectContext(&SubjectContext);
-   
-   Priv.PrivilegeCount = 1;
-   Priv.Control = 1;
-   Priv.Privilege[0].Luid = PrivilegeValue;
-   Priv.Privilege[0].Attributes = 0;
-   
-   r = SePrivilegeCheck(&Priv,
-                       &SubjectContext,
-                       PreviousMode);
-      
-   if (PreviousMode != KernelMode)
-     {
+  SECURITY_SUBJECT_CONTEXT SubjectContext;
+  PRIVILEGE_SET Priv;
+  BOOLEAN Result;
+
+  PAGED_CODE();
+
+  SeCaptureSubjectContext (&SubjectContext);
+
+  Priv.PrivilegeCount = 1;
+  Priv.Control = PRIVILEGE_SET_ALL_NECESSARY;
+  Priv.Privilege[0].Luid = PrivilegeValue;
+  Priv.Privilege[0].Attributes = SE_PRIVILEGE_ENABLED;
+
+  Result = SePrivilegeCheck (&Priv,
+                            &SubjectContext,
+                            PreviousMode);
+
+  if (PreviousMode != KernelMode)
+    {
 #if 0
-       SePrivilegeServiceAuditAlarm(0,
+      SePrivilegedServiceAuditAlarm (0,
                                     &SubjectContext,
                                     &PrivilegeValue);
 #endif
-     }
-   SeReleaseSubjectContext(&SubjectContext);
-   return(r);
+    }
+
+  SeReleaseSubjectContext (&SubjectContext);
+
+  return Result;
 }
 
 /* EOF */