[ros-diffs] [weiden] 13524: fixed returning the reference to the token in SeCopyClientToken() and probe pointer when neccessary

fixed returning the reference to the token in SeCopyClientToken() and
probe pointer when neccessary
Modified: trunk/reactos/ntoskrnl/se/token.c
  _____

Modified: trunk/reactos/ntoskrnl/se/token.c
--- trunk/reactos/ntoskrnl/se/token.c	2005-02-13 09:56:12 UTC (rev
13523)
+++ trunk/reactos/ntoskrnl/se/token.c	2005-02-13 13:03:44 UTC (rev
13524)
@@ -131,13 +131,32 @@

KPROCESSOR_MODE PreviousMode,
    	  PTOKEN* NewAccessToken)
 {
-  NTSTATUS Status;
   ULONG uLength;
   ULONG i;
-  
   PVOID EndMem;
-
   PTOKEN AccessToken;
+  NTSTATUS Status;
+  
+  if(PreviousMode != KernelMode)
+  {
+    Status = STATUS_SUCCESS;
+    _SEH_TRY
+    {
+      ProbeForWrite(NewAccessToken,
+                    sizeof(TOKEN),
+                    sizeof(ULONG));
+    }
+    _SEH_HANDLE
+    {
+      Status = _SEH_GetExceptionCode();
+    }
+    _SEH_END;
+    
+    if(!NT_SUCCESS(Status))
+    {
+      return Status;
+    }
+  }

Status = ObCreateObject(PreviousMode,
    		  SepTokenObjectType,
@@ -244,8 +263,17 @@

if ( NT_SUCCESS(Status) )
     {
-      *NewAccessToken = AccessToken;
-      return(STATUS_SUCCESS);
+      _SEH_TRY
+      {
+        *NewAccessToken = AccessToken;
+        Status = STATUS_SUCCESS;
+      }
+      _SEH_HANDLE
+      {
+        Status = _SEH_GetExceptionCode();
+      }
+      _SEH_END;
+      return Status;
     }

ObDereferenceObject(AccessToken);
@@ -320,7 +348,7 @@
    			TokenImpersonation,
    			Level,
    			PreviousMode,
-			    (PTOKEN*)&NewToken);
+			        (PTOKEN*)NewToken);

return(Status);
 }