Do not make NtSecureConnectPort fail: for now direct the call to NtConnectPort.
[reactos.git] / reactos / ntoskrnl / lpc / connect.c
index 59d0756..9a35d70 100644 (file)
@@ -1,4 +1,4 @@
-/* $Id: connect.c,v 1.8 2001/12/02 23:34:42 dwelch Exp $
+/* $Id: connect.c,v 1.23 2004/01/22 21:56:48 ea Exp $
  * 
  * COPYRIGHT:       See COPYING in the top level directory
  * PROJECT:         ReactOS kernel
 
 /* INCLUDES *****************************************************************/
 
-#include <ddk/ntddk.h>
+#define NTOS_MODE_KERNEL
+#include <ntos.h>
 #include <internal/ob.h>
 #include <internal/port.h>
 #include <internal/dbg.h>
 #include <internal/pool.h>
-#include <internal/mm.h>
 #include <internal/safe.h>
+#include <internal/mm.h>
 
 #define NDEBUG
 #include <internal/debug.h>
@@ -62,10 +63,14 @@ EiConnectPort(IN PEPORT* ConnectedPort,
   /*
    * Create a port to represent our side of the connection
    */
-  Status = ObCreateObject (NULL,
-                          PORT_ALL_ACCESS,
-                          NULL,
+  Status = ObCreateObject (KernelMode,
                           ExPortType,
+                          NULL,
+                          KernelMode,
+                          NULL,
+                          sizeof(EPORT),
+                          0,
+                          0,
                           (PVOID*)&OurPort);
   if (!NT_SUCCESS(Status))
     {
@@ -90,12 +95,12 @@ EiConnectPort(IN PEPORT* ConnectedPort,
    */
   RequestMessage->MessageHeader.DataSize = 
     sizeof(EPORT_CONNECT_REQUEST_MESSAGE) + RequestConnectDataLength -
-    sizeof(LPC_MESSAGE_HEADER);
+    sizeof(LPC_MESSAGE);
   RequestMessage->MessageHeader.MessageSize = 
     sizeof(EPORT_CONNECT_REQUEST_MESSAGE) + RequestConnectDataLength;
   DPRINT("RequestMessageSize %d\n",
         RequestMessage->MessageHeader.MessageSize);
-  RequestMessage->MessageHeader.SharedSectionSize = 0;
+  RequestMessage->MessageHeader.SectionSize = 0;
   RequestMessage->ConnectingProcess = PsGetCurrentProcess();
   ObReferenceObjectByPointer(RequestMessage->ConnectingProcess,
                             PROCESS_VM_OPERATION,
@@ -232,6 +237,7 @@ EiConnectPort(IN PEPORT* ConnectedPort,
  * 
  * RETURN VALUE
  * 
+ * @unimplemented
  */
 NTSTATUS STDCALL
 NtConnectPort (PHANDLE                         UnsafeConnectedPortHandle,
@@ -260,7 +266,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
    */
   if (UnsafeWriteMap != NULL)
     {
-      Status = MmCopyFromCaller(&WriteMap, UnsafeWriteMap, 
+      Status = MmCopyFromCaller(&WriteMap,
+                               UnsafeWriteMap,
                                sizeof(LPC_SECTION_WRITE));
       if (!NT_SUCCESS(Status))
        {
@@ -308,7 +315,7 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
            }
          Status = MmCopyFromCaller(ConnectData,
                                    UnsafeConnectData,
-                                   ConnectDataLength); 
+                                   ConnectDataLength);
          if (!NT_SUCCESS(Status))
            {
              ExFreePool(ConnectData);
@@ -385,11 +392,13 @@ NtConnectPort (PHANDLE                            UnsafeConnectedPortHandle,
        {
          if (ExGetPreviousMode() != KernelMode)
            {
-             MmCopyToCaller(UnsafeConnectData, ConnectData,
+             MmCopyToCaller(UnsafeConnectData,
+                            ConnectData,
                             ConnectDataLength);
              ExFreePool(ConnectData);
            }
-         MmCopyToCaller(UnsafeConnectDataLength, &ConnectDataLength,
+         MmCopyToCaller(UnsafeConnectDataLength,
+                        &ConnectDataLength,
                         sizeof(ULONG));
        }
       return(Status);
@@ -415,7 +424,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
        {
          if (ExGetPreviousMode() != KernelMode)
            {
-             Status = MmCopyToCaller(UnsafeConnectData, ConnectData,
+             Status = MmCopyToCaller(UnsafeConnectData,
+                                     ConnectData,
                                      ConnectDataLength);
              ExFreePool(ConnectData);
              if (!NT_SUCCESS(Status))
@@ -423,7 +433,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
                  return(Status);
                }
            }
-         Status = MmCopyToCaller(UnsafeConnectDataLength, &ConnectDataLength,
+         Status = MmCopyToCaller(UnsafeConnectDataLength,
+                                 &ConnectDataLength,
                                  sizeof(ULONG));
          if (!NT_SUCCESS(Status))
            {
@@ -441,7 +452,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
     {
       return(Status);
     }
-  Status = MmCopyToCaller(UnsafeConnectedPortHandle, &ConnectedPortHandle,
+  Status = MmCopyToCaller(UnsafeConnectedPortHandle,
+                         &ConnectedPortHandle,
                          sizeof(HANDLE));
   if (!NT_SUCCESS(Status))
     {
@@ -449,7 +461,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
     }
   if (UnsafeWriteMap != NULL)
     {
-      Status = MmCopyToCaller(UnsafeWriteMap, &WriteMap, 
+      Status = MmCopyToCaller(UnsafeWriteMap,
+                             &WriteMap,
                              sizeof(LPC_SECTION_WRITE));
       if (!NT_SUCCESS(Status))
        {
@@ -458,7 +471,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
     }
   if (UnsafeReadMap != NULL)
     {
-      Status = MmCopyToCaller(UnsafeReadMap, &ReadMap,
+      Status = MmCopyToCaller(UnsafeReadMap,
+                             &ReadMap,
                              sizeof(LPC_SECTION_READ));
       if (!NT_SUCCESS(Status))
        {
@@ -467,9 +481,9 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
     }
   if (UnsafeMaximumMessageSize != NULL)
     {
-      Status = MmCopyToCaller(UnsafeMaximumMessageSize, 
+      Status = MmCopyToCaller(UnsafeMaximumMessageSize,
                              &MaximumMessageSize,
-                             sizeof(LPC_SECTION_WRITE));
+                             sizeof(ULONG));
       if (!NT_SUCCESS(Status))
        {
          return(Status);
@@ -479,7 +493,6 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
   /*
    * All done.
    */
-  ObDereferenceObject(ConnectedPort);
 
   return(STATUS_SUCCESS);
 }
@@ -500,9 +513,8 @@ NtConnectPort (PHANDLE                              UnsafeConnectedPortHandle,
  *     ReadMap
  *
  * RETURN VALUE
- *
  */
-EXPORTED NTSTATUS STDCALL
+/*EXPORTED*/ NTSTATUS STDCALL
 NtAcceptConnectPort (PHANDLE                   ServerPortHandle,
                     HANDLE                     NamedPortHandle,
                     PLPC_MESSAGE               LpcMessage,
@@ -517,9 +529,15 @@ NtAcceptConnectPort (PHANDLE                       ServerPortHandle,
   KIRQL                oldIrql;
   PEPORT_CONNECT_REQUEST_MESSAGE CRequest;
   PEPORT_CONNECT_REPLY_MESSAGE CReply;
+  ULONG Size;
+
+  Size = sizeof(EPORT_CONNECT_REPLY_MESSAGE);
+  if (LpcMessage)
+  {
+     Size += LpcMessage->DataSize;
+  }
 
-  CReply = ExAllocatePool(NonPagedPool, 
-                sizeof(EPORT_CONNECT_REPLY_MESSAGE) + LpcMessage->DataSize);
+  CReply = ExAllocatePool(NonPagedPool, Size);
   if (CReply == NULL)
     {
       return(STATUS_NO_MEMORY);
@@ -540,21 +558,41 @@ NtAcceptConnectPort (PHANDLE                      ServerPortHandle,
   /*
    * Create a port object for our side of the connection
    */
-  if (AcceptIt == 1)
+  if (AcceptIt)
     {
-      Status = ObCreateObject(ServerPortHandle,
-                             PORT_ALL_ACCESS,
-                             NULL,
+      Status = ObCreateObject(ExGetPreviousMode(),
                              ExPortType,
+                             NULL,
+                             ExGetPreviousMode(),
+                             NULL,
+                             sizeof(EPORT),
+                             0,
+                             0,
                              (PVOID*)&OurPort);
       if (!NT_SUCCESS(Status))
        {
+         ExFreePool(CReply);
+         ObDereferenceObject(NamedPort);
+         return(Status);
+       }
+
+      Status = ObInsertObject ((PVOID)OurPort,
+                              NULL,
+                              PORT_ALL_ACCESS,
+                              0,
+                              NULL,
+                              ServerPortHandle);
+      if (!NT_SUCCESS(Status))
+       {
+         ObDereferenceObject(OurPort);
+         ExFreePool(CReply);
          ObDereferenceObject(NamedPort);
          return(Status);
        }
+
       NiInitializePort(OurPort);
     }
-  
+
   /*
    * Dequeue the connection request
    */
@@ -568,24 +606,23 @@ NtAcceptConnectPort (PHANDLE                      ServerPortHandle,
    */
   if (LpcMessage != NULL)
     {
-      memcpy(&CReply->MessageHeader, LpcMessage, sizeof(LPC_MESSAGE_HEADER));
+      memcpy(&CReply->MessageHeader, LpcMessage, sizeof(LPC_MESSAGE));
       memcpy(&CReply->ConnectData, (PVOID)(LpcMessage + 1), 
             LpcMessage->DataSize);
       CReply->MessageHeader.MessageSize =
        sizeof(EPORT_CONNECT_REPLY_MESSAGE) + LpcMessage->DataSize;
       CReply->MessageHeader.DataSize = CReply->MessageHeader.MessageSize -
-       sizeof(LPC_MESSAGE_HEADER);
+       sizeof(LPC_MESSAGE);
       CReply->ConnectDataLength = LpcMessage->DataSize;
     }
   else
     {
       CReply->MessageHeader.MessageSize = sizeof(EPORT_CONNECT_REPLY_MESSAGE);
       CReply->MessageHeader.DataSize = sizeof(EPORT_CONNECT_REPLY_MESSAGE) -
-       sizeof(LPC_MESSAGE_HEADER);
+       sizeof(LPC_MESSAGE);
       CReply->ConnectDataLength = 0;
     }
-
-  if (AcceptIt != 1)
+  if (!AcceptIt)
     {  
       EiReplyOrRequestPort(ConnectionRequest->Sender,
                           &CReply->MessageHeader,
@@ -697,16 +734,18 @@ NtAcceptConnectPort (PHANDLE                      ServerPortHandle,
     }
   CReply->MaximumMessageSize = 0x148;
 
+
   /*
    * Connect the two ports
    */
   OurPort->OtherPort = ConnectionRequest->Sender;
   OurPort->OtherPort->OtherPort = OurPort;
   EiReplyOrRequestPort(ConnectionRequest->Sender,
-                      LpcMessage,
+                      (PLPC_MESSAGE)CReply,
                       LPC_REPLY,
                       OurPort);
   ExFreePool(ConnectionRequest);
+  ExFreePool(CReply);
    
   ObDereferenceObject(OurPort);
   ObDereferenceObject(NamedPort);
@@ -716,7 +755,7 @@ NtAcceptConnectPort (PHANDLE                        ServerPortHandle,
 
 /**********************************************************************
  * NAME                                                        EXPORTED
- *     NtSecureConnectPort@36
+ *     NtSecureConnectPort/9
  *     
  * DESCRIPTION
  *     Connect to a named port and wait for the other side to 
@@ -726,7 +765,7 @@ NtAcceptConnectPort (PHANDLE                        ServerPortHandle,
  *
  * ARGUMENTS
  *     ConnectedPort
- *     PortName
+ *     PortName: fully qualified name in the Ob name space;
  *     Qos
  *     WriteMap
  *     ServerSid
@@ -736,7 +775,6 @@ NtAcceptConnectPort (PHANDLE                        ServerPortHandle,
  *     UserConnectInfoLength
  * 
  * RETURN VALUE
- * 
  */
 NTSTATUS STDCALL
 NtSecureConnectPort (OUT    PHANDLE                            ConnectedPort,
@@ -749,7 +787,16 @@ NtSecureConnectPort (OUT    PHANDLE                                ConnectedPort,
                     IN OUT PVOID                               ConnectInfo             OPTIONAL,
                     IN OUT PULONG                              UserConnectInfoLength   OPTIONAL)
 {
-       return (STATUS_NOT_IMPLEMENTED);
+  /* TODO: implement a new object type: SecurePort */
+  /* TODO: verify the process' SID that hosts the rendez-vous port equals ServerSid */
+  return NtConnectPort (ConnectedPort,
+                       PortName,
+                       Qos,
+                       WriteMap,
+                       ReadMap,
+                       MaxMessageSize,
+                       ConnectInfo,
+                       UserConnectInfoLength);
 }
 
 /* EOF */