mirror of
https://github.com/reactos/reactos.git
synced 2025-08-03 22:56:00 +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
|
NTSTATUS
|
||||||
NTAPI
|
NTAPI
|
||||||
LpcRequestWaitReplyPort(IN PVOID Port,
|
LpcRequestWaitReplyPort(IN PVOID PortObject,
|
||||||
IN PPORT_MESSAGE LpcMessageRequest,
|
IN PPORT_MESSAGE LpcRequest,
|
||||||
OUT PPORT_MESSAGE LpcMessageReply)
|
OUT PPORT_MESSAGE LpcReply)
|
||||||
{
|
{
|
||||||
UNIMPLEMENTED;
|
PLPCP_PORT_OBJECT Port, QueuePort, ReplyPort, ConnectionPort = NULL;
|
||||||
return STATUS_NOT_IMPLEMENTED;
|
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…
Add table
Add a link
Reference in a new issue