mirror of
https://github.com/reactos/reactos.git
synced 2024-12-28 01:55:19 +00:00
- Implement LpcRequestWaitReplyPort as a copy of NtRequestWaitReplyPort, but remove object referencing/dereferencing (we already have a pointer to the port object), and SEH (this function is internal to the kernel).
svn path=/trunk/; revision=31360
This commit is contained in:
Aleksey Bragin
parent
8673adbeb8
commit
a9853e5a80
1 changed files with 251 additions and 6 deletions
|
|
@ -167,16 +167,261 @@ LpcRequestPort(IN PVOID PortObject,
|
|||
}
|
||||
|
||||
/*
|
||||
* @unimplemented
|
||||
* @implemented
|
||||
*/
|
||||
NTSTATUS
|
||||
NTAPI
|
||||
LpcRequestWaitReplyPort(IN PVOID Port,
|
||||
IN PPORT_MESSAGE LpcMessageRequest,
|
||||
OUT PPORT_MESSAGE LpcMessageReply)
|
||||
LpcRequestWaitReplyPort(IN PVOID PortObject,
|
||||
IN PPORT_MESSAGE LpcRequest,
|
||||
OUT PPORT_MESSAGE LpcReply)
|
||||
{
|
||||
UNIMPLEMENTED;
|
||||
return STATUS_NOT_IMPLEMENTED;
|
||||
PLPCP_PORT_OBJECT Port, QueuePort, ReplyPort, ConnectionPort = NULL;
|
||||
KPROCESSOR_MODE PreviousMode = KeGetPreviousMode();
|
||||
NTSTATUS Status = STATUS_SUCCESS;
|
||||
PLPCP_MESSAGE Message;
|
||||
PETHREAD Thread = PsGetCurrentThread();
|
||||
BOOLEAN Callback;
|
||||
PKSEMAPHORE Semaphore;
|
||||
ULONG MessageType;
|
||||
PAGED_CODE();
|
||||
|
||||
Port = (PLPCP_PORT_OBJECT)PortObject;
|
||||
|
||||
LPCTRACE(LPC_SEND_DEBUG,
|
||||
"Port: %p. Messages: %p/%p. Type: %lx\n",
|
||||
Port,
|
||||
LpcRequest,
|
||||
LpcReply,
|
||||
LpcpGetMessageType(LpcRequest));
|
||||
|
||||
/* Check if the thread is dying */
|
||||
if (Thread->LpcExitThreadCalled) return STATUS_THREAD_IS_TERMINATING;
|
||||
|
||||
/* Check if this is an LPC Request */
|
||||
if (LpcpGetMessageType(LpcRequest) == LPC_REQUEST)
|
||||
{
|
||||
/* Then it's a callback */
|
||||
Callback = TRUE;
|
||||
}
|
||||
else if (LpcpGetMessageType(LpcRequest))
|
||||
{
|
||||
/* This is a not kernel-mode message */
|
||||
return STATUS_INVALID_PARAMETER;
|
||||
}
|
||||
else
|
||||
{
|
||||
/* This is a kernel-mode message without a callback */
|
||||
LpcRequest->u2.s2.Type |= LPC_REQUEST;
|
||||
Callback = FALSE;
|
||||
}
|
||||
|
||||
/* Get the message type */
|
||||
MessageType = LpcRequest->u2.s2.Type;
|
||||
|
||||
/* Validate the length */
|
||||
if (((ULONG)LpcRequest->u1.s1.DataLength + sizeof(PORT_MESSAGE)) >
|
||||
(ULONG)LpcRequest->u1.s1.TotalLength)
|
||||
{
|
||||
/* Fail */
|
||||
return STATUS_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
/* Validate the message length */
|
||||
if (((ULONG)LpcRequest->u1.s1.TotalLength > Port->MaxMessageLength) ||
|
||||
((ULONG)LpcRequest->u1.s1.TotalLength <= (ULONG)LpcRequest->u1.s1.DataLength))
|
||||
{
|
||||
/* Fail */
|
||||
return STATUS_PORT_MESSAGE_TOO_LONG;
|
||||
}
|
||||
|
||||
/* Allocate a message from the port zone */
|
||||
Message = LpcpAllocateFromPortZone();
|
||||
if (!Message)
|
||||
{
|
||||
/* Fail if we couldn't allocate a message */
|
||||
return STATUS_NO_MEMORY;
|
||||
}
|
||||
|
||||
/* Check if this is a callback */
|
||||
if (Callback)
|
||||
{
|
||||
/* FIXME: TODO */
|
||||
Semaphore = NULL; // we'd use the Thread Semaphore here
|
||||
ASSERT(FALSE);
|
||||
}
|
||||
else
|
||||
{
|
||||
/* No callback, just copy the message */
|
||||
LpcpMoveMessage(&Message->Request,
|
||||
LpcRequest,
|
||||
LpcRequest + 1,
|
||||
MessageType,
|
||||
&Thread->Cid);
|
||||
|
||||
/* Acquire the LPC lock */
|
||||
KeAcquireGuardedMutex(&LpcpLock);
|
||||
|
||||
/* Right now clear the port context */
|
||||
Message->PortContext = NULL;
|
||||
|
||||
/* Check if this is a not connection port */
|
||||
if ((Port->Flags & LPCP_PORT_TYPE_MASK) != LPCP_CONNECTION_PORT)
|
||||
{
|
||||
/* We want the connected port */
|
||||
QueuePort = Port->ConnectedPort;
|
||||
if (!QueuePort)
|
||||
{
|
||||
/* We have no connected port, fail */
|
||||
LpcpFreeToPortZone(Message, 3);
|
||||
return STATUS_PORT_DISCONNECTED;
|
||||
}
|
||||
|
||||
/* This will be the rundown port */
|
||||
ReplyPort = QueuePort;
|
||||
|
||||
/* Check if this is a communication port */
|
||||
if ((Port->Flags & LPCP_PORT_TYPE_MASK) == LPCP_CLIENT_PORT)
|
||||
{
|
||||
/* Copy the port context and use the connection port */
|
||||
Message->PortContext = QueuePort->PortContext;
|
||||
ConnectionPort = QueuePort = Port->ConnectionPort;
|
||||
if (!ConnectionPort)
|
||||
{
|
||||
/* Fail */
|
||||
LpcpFreeToPortZone(Message, 3);
|
||||
return STATUS_PORT_DISCONNECTED;
|
||||
}
|
||||
}
|
||||
else if ((Port->Flags & LPCP_PORT_TYPE_MASK) !=
|
||||
LPCP_COMMUNICATION_PORT)
|
||||
{
|
||||
/* Use the connection port for anything but communication ports */
|
||||
ConnectionPort = QueuePort = Port->ConnectionPort;
|
||||
if (!ConnectionPort)
|
||||
{
|
||||
/* Fail */
|
||||
LpcpFreeToPortZone(Message, 3);
|
||||
return STATUS_PORT_DISCONNECTED;
|
||||
}
|
||||
}
|
||||
|
||||
/* Reference the connection port if it exists */
|
||||
if (ConnectionPort) ObReferenceObject(ConnectionPort);
|
||||
}
|
||||
else
|
||||
{
|
||||
/* Otherwise, for a connection port, use the same port object */
|
||||
QueuePort = ReplyPort = Port;
|
||||
}
|
||||
|
||||
/* No reply thread */
|
||||
Message->RepliedToThread = NULL;
|
||||
Message->SenderPort = Port;
|
||||
|
||||
/* Generate the Message ID and set it */
|
||||
Message->Request.MessageId = LpcpNextMessageId++;
|
||||
if (!LpcpNextMessageId) LpcpNextMessageId = 1;
|
||||
Message->Request.CallbackId = 0;
|
||||
|
||||
/* Set the message ID for our thread now */
|
||||
Thread->LpcReplyMessageId = Message->Request.MessageId;
|
||||
Thread->LpcReplyMessage = NULL;
|
||||
|
||||
/* Insert the message in our chain */
|
||||
InsertTailList(&QueuePort->MsgQueue.ReceiveHead, &Message->Entry);
|
||||
InsertTailList(&ReplyPort->LpcReplyChainHead, &Thread->LpcReplyChain);
|
||||
LpcpSetPortToThread(Thread, Port);
|
||||
|
||||
/* Release the lock and get the semaphore we'll use later */
|
||||
KeEnterCriticalRegion();
|
||||
KeReleaseGuardedMutex(&LpcpLock);
|
||||
Semaphore = QueuePort->MsgQueue.Semaphore;
|
||||
|
||||
/* If this is a waitable port, wake it up */
|
||||
if (QueuePort->Flags & LPCP_WAITABLE_PORT)
|
||||
{
|
||||
/* Wake it */
|
||||
KeSetEvent(&QueuePort->WaitEvent, IO_NO_INCREMENT, FALSE);
|
||||
}
|
||||
}
|
||||
|
||||
/* Now release the semaphore */
|
||||
LpcpCompleteWait(Semaphore);
|
||||
KeLeaveCriticalRegion();
|
||||
|
||||
/* And let's wait for the reply */
|
||||
LpcpReplyWait(&Thread->LpcReplySemaphore, PreviousMode);
|
||||
|
||||
/* Acquire the LPC lock */
|
||||
KeAcquireGuardedMutex(&LpcpLock);
|
||||
|
||||
/* Get the LPC Message and clear our thread's reply data */
|
||||
Message = LpcpGetMessageFromThread(Thread);
|
||||
Thread->LpcReplyMessage = NULL;
|
||||
Thread->LpcReplyMessageId = 0;
|
||||
|
||||
/* Check if we have anything on the reply chain*/
|
||||
if (!IsListEmpty(&Thread->LpcReplyChain))
|
||||
{
|
||||
/* Remove this thread and reinitialize the list */
|
||||
RemoveEntryList(&Thread->LpcReplyChain);
|
||||
InitializeListHead(&Thread->LpcReplyChain);
|
||||
}
|
||||
|
||||
/* Release the lock */
|
||||
KeReleaseGuardedMutex(&LpcpLock);
|
||||
|
||||
/* Check if we got a reply */
|
||||
if (Status == STATUS_SUCCESS)
|
||||
{
|
||||
/* Check if we have a valid message */
|
||||
if (Message)
|
||||
{
|
||||
LPCTRACE(LPC_SEND_DEBUG,
|
||||
"Reply Messages: %p/%p\n",
|
||||
&Message->Request,
|
||||
(&Message->Request) + 1);
|
||||
|
||||
/* Move the message */
|
||||
LpcpMoveMessage(LpcReply,
|
||||
&Message->Request,
|
||||
(&Message->Request) + 1,
|
||||
0,
|
||||
NULL);
|
||||
|
||||
/* Check if this is an LPC request with data information */
|
||||
if ((LpcpGetMessageType(&Message->Request) == LPC_REQUEST) &&
|
||||
(Message->Request.u2.s2.DataInfoOffset))
|
||||
{
|
||||
/* Save the data information */
|
||||
LpcpSaveDataInfoMessage(Port, Message, 0);
|
||||
}
|
||||
else
|
||||
{
|
||||
/* Otherwise, just free it */
|
||||
LpcpFreeToPortZone(Message, 0);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
/* We don't have a reply */
|
||||
Status = STATUS_LPC_REPLY_LOST;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
/* The wait failed, free the message */
|
||||
if (Message) LpcpFreeToPortZone(Message, 0);
|
||||
}
|
||||
|
||||
/* All done */
|
||||
LPCTRACE(LPC_SEND_DEBUG,
|
||||
"Port: %p. Status: %p\n",
|
||||
Port,
|
||||
Status);
|
||||
|
||||
if (ConnectionPort) ObDereferenceObject(ConnectionPort);
|
||||
return Status;
|
||||
}
|
||||
|
||||
/*
|
||||
|
|
|
|||
Loading…
Reference in a new issue