2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS kernel
4 * FILE: drivers/fs/np/fsctrl.c
5 * PURPOSE: Named pipe filesystem
6 * PROGRAMMER: David Welch <welch@cwcom.net>
11 /* INCLUDES ******************************************************************/
18 /* FUNCTIONS *****************************************************************/
20 static DRIVER_CANCEL NpfsListeningCancelRoutine
;
22 NpfsListeningCancelRoutine(IN PDEVICE_OBJECT DeviceObject
,
25 PNPFS_WAITER_ENTRY Waiter
;
27 Waiter
= (PNPFS_WAITER_ENTRY
)&Irp
->Tail
.Overlay
.DriverContext
;
29 DPRINT("NpfsListeningCancelRoutine() called for <%wZ>\n",
30 &Waiter
->Ccb
->Fcb
->PipeName
);
32 IoReleaseCancelSpinLock(Irp
->CancelIrql
);
35 KeLockMutex(&Waiter
->Ccb
->Fcb
->CcbListLock
);
36 RemoveEntryList(&Waiter
->Entry
);
37 KeUnlockMutex(&Waiter
->Ccb
->Fcb
->CcbListLock
);
39 Irp
->IoStatus
.Status
= STATUS_CANCELLED
;
40 Irp
->IoStatus
.Information
= 0;
41 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
46 NpfsAddListeningServerInstance(PIRP Irp
,
49 PNPFS_WAITER_ENTRY Entry
;
52 Entry
= (PNPFS_WAITER_ENTRY
)&Irp
->Tail
.Overlay
.DriverContext
;
56 KeLockMutex(&Ccb
->Fcb
->CcbListLock
);
58 IoMarkIrpPending(Irp
);
59 InsertTailList(&Ccb
->Fcb
->WaiterListHead
, &Entry
->Entry
);
61 IoAcquireCancelSpinLock(&oldIrql
);
64 (void)IoSetCancelRoutine(Irp
, NpfsListeningCancelRoutine
);
65 IoReleaseCancelSpinLock(oldIrql
);
66 KeUnlockMutex(&Ccb
->Fcb
->CcbListLock
);
67 return STATUS_PENDING
;
69 IoReleaseCancelSpinLock(oldIrql
);
71 RemoveEntryList(&Entry
->Entry
);
73 Irp
->IoStatus
.Status
= STATUS_CANCELLED
;
74 Irp
->IoStatus
.Information
= 0;
75 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
76 KeUnlockMutex(&Ccb
->Fcb
->CcbListLock
);
78 return STATUS_CANCELLED
;
83 NpfsConnectPipe(PIRP Irp
,
86 PIO_STACK_LOCATION IoStack
;
87 PFILE_OBJECT FileObject
;
89 PLIST_ENTRY current_entry
;
94 DPRINT("NpfsConnectPipe()\n");
96 if (Ccb
->PipeState
== FILE_PIPE_CONNECTED_STATE
)
98 KeResetEvent(&Ccb
->ConnectEvent
);
99 return STATUS_PIPE_CONNECTED
;
102 if (Ccb
->PipeState
== FILE_PIPE_CLOSING_STATE
)
103 return STATUS_PIPE_CLOSING
;
105 DPRINT("Waiting for connection...\n");
108 IoStack
= IoGetCurrentIrpStackLocation(Irp
);
109 FileObject
= IoStack
->FileObject
;
110 Flags
= FileObject
->Flags
;
112 /* search for a listening client fcb */
113 KeLockMutex(&Fcb
->CcbListLock
);
115 current_entry
= Fcb
->ClientCcbListHead
.Flink
;
116 while (current_entry
!= &Fcb
->ClientCcbListHead
)
118 ClientCcb
= CONTAINING_RECORD(current_entry
,
122 if (ClientCcb
->PipeState
== 0)
124 /* found a passive (waiting) client CCB */
125 DPRINT("Passive (waiting) client CCB found -- wake the client\n");
126 KeSetEvent(&ClientCcb
->ConnectEvent
, IO_NO_INCREMENT
, FALSE
);
131 if (ClientCcb
->PipeState
== FILE_PIPE_LISTENING_STATE
)
133 /* found a listening client CCB */
134 DPRINT("Listening client CCB found -- connecting\n");
136 /* connect client and server CCBs */
137 Ccb
->OtherSide
= ClientCcb
;
138 ClientCcb
->OtherSide
= Ccb
;
140 /* set connected state */
141 Ccb
->PipeState
= FILE_PIPE_CONNECTED_STATE
;
142 ClientCcb
->PipeState
= FILE_PIPE_CONNECTED_STATE
;
144 KeUnlockMutex(&Fcb
->CcbListLock
);
146 /* FIXME: create and initialize data queues */
148 /* signal client's connect event */
149 DPRINT("Setting the ConnectEvent for %x\n", ClientCcb
);
150 KeSetEvent(&ClientCcb
->ConnectEvent
, IO_NO_INCREMENT
, FALSE
);
152 return STATUS_PIPE_CONNECTED
;
156 current_entry
= current_entry
->Flink
;
159 /* no listening client fcb found */
160 DPRINT("No listening client fcb found -- waiting for client\n");
162 Ccb
->PipeState
= FILE_PIPE_LISTENING_STATE
;
164 Status
= NpfsAddListeningServerInstance(Irp
, Ccb
);
166 KeUnlockMutex(&Fcb
->CcbListLock
);
168 if (Flags
& FO_SYNCHRONOUS_IO
)
170 KeWaitForSingleObject(&Ccb
->ConnectEvent
,
177 DPRINT("NpfsConnectPipe() done (Status %lx)\n", Status
);
184 NpfsDisconnectPipe(PNPFS_CCB Ccb
)
191 DPRINT("NpfsDisconnectPipe()\n");
194 KeLockMutex(&Fcb
->CcbListLock
);
196 if (Ccb
->PipeState
== FILE_PIPE_DISCONNECTED_STATE
)
198 DPRINT("Pipe is already disconnected\n");
199 Status
= STATUS_PIPE_DISCONNECTED
;
201 else if ((!Ccb
->OtherSide
) && (Ccb
->PipeState
== FILE_PIPE_CONNECTED_STATE
))
203 ExAcquireFastMutex(&Ccb
->DataListLock
);
204 Ccb
->PipeState
= FILE_PIPE_DISCONNECTED_STATE
;
205 ExReleaseFastMutex(&Ccb
->DataListLock
);
206 Status
= STATUS_SUCCESS
;
208 else if (Ccb
->PipeState
== FILE_PIPE_CONNECTED_STATE
)
210 Server
= (Ccb
->PipeEnd
== FILE_PIPE_SERVER_END
);
211 OtherSide
= Ccb
->OtherSide
;
212 //Ccb->OtherSide = NULL;
213 Ccb
->PipeState
= FILE_PIPE_DISCONNECTED_STATE
;
214 /* Lock the server first */
217 ExAcquireFastMutex(&Ccb
->DataListLock
);
218 ExAcquireFastMutex(&OtherSide
->DataListLock
);
222 ExAcquireFastMutex(&OtherSide
->DataListLock
);
223 ExAcquireFastMutex(&Ccb
->DataListLock
);
225 OtherSide
->PipeState
= FILE_PIPE_DISCONNECTED_STATE
;
226 //OtherSide->OtherSide = NULL;
228 * Signaling the write event. If is possible that an other
229 * thread waits for an empty buffer.
231 KeSetEvent(&OtherSide
->ReadEvent
, IO_NO_INCREMENT
, FALSE
);
232 KeSetEvent(&OtherSide
->WriteEvent
, IO_NO_INCREMENT
, FALSE
);
235 ExReleaseFastMutex(&OtherSide
->DataListLock
);
236 ExReleaseFastMutex(&Ccb
->DataListLock
);
240 ExReleaseFastMutex(&Ccb
->DataListLock
);
241 ExReleaseFastMutex(&OtherSide
->DataListLock
);
243 Status
= STATUS_SUCCESS
;
245 else if (Ccb
->PipeState
== FILE_PIPE_LISTENING_STATE
)
248 PNPFS_WAITER_ENTRY WaitEntry
= NULL
;
249 BOOLEAN Complete
= FALSE
;
252 Entry
= Ccb
->Fcb
->WaiterListHead
.Flink
;
253 while (Entry
!= &Ccb
->Fcb
->WaiterListHead
)
255 WaitEntry
= CONTAINING_RECORD(Entry
, NPFS_WAITER_ENTRY
, Entry
);
256 if (WaitEntry
->Ccb
== Ccb
)
258 RemoveEntryList(Entry
);
259 Irp
= CONTAINING_RECORD(Entry
, IRP
, Tail
.Overlay
.DriverContext
);
260 Complete
= (NULL
== IoSetCancelRoutine(Irp
, NULL
));
263 Entry
= Entry
->Flink
;
270 Irp
->IoStatus
.Status
= STATUS_PIPE_BROKEN
;
271 Irp
->IoStatus
.Information
= 0;
272 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
275 Ccb
->PipeState
= FILE_PIPE_DISCONNECTED_STATE
;
276 Status
= STATUS_SUCCESS
;
278 else if (Ccb
->PipeState
== FILE_PIPE_CLOSING_STATE
)
280 Status
= STATUS_PIPE_CLOSING
;
284 Status
= STATUS_UNSUCCESSFUL
;
286 KeUnlockMutex(&Fcb
->CcbListLock
);
292 NpfsWaitPipe(PIRP Irp
,
295 PLIST_ENTRY current_entry
;
298 PFILE_PIPE_WAIT_FOR_BUFFER WaitPipe
;
300 LARGE_INTEGER TimeOut
;
302 DPRINT("NpfsWaitPipe\n");
304 WaitPipe
= (PFILE_PIPE_WAIT_FOR_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
307 if (Ccb
->PipeState
!= 0)
309 DPRINT("Pipe is not in passive (waiting) state!\n");
310 return STATUS_UNSUCCESSFUL
;
313 /* search for listening server */
314 current_entry
= Fcb
->ServerCcbListHead
.Flink
;
315 while (current_entry
!= &Fcb
->ServerCcbListHead
)
317 ServerCcb
= CONTAINING_RECORD(current_entry
,
321 if (ServerCcb
->PipeState
== FILE_PIPE_LISTENING_STATE
)
323 /* found a listening server CCB */
324 DPRINT("Listening server CCB found -- connecting\n");
326 return STATUS_SUCCESS
;
329 current_entry
= current_entry
->Flink
;
332 /* No listening server fcb found */
334 /* If no timeout specified, use the default one */
335 if (WaitPipe
->TimeoutSpecified
)
336 TimeOut
= WaitPipe
->Timeout
;
338 TimeOut
= Fcb
->TimeOut
;
341 Status
= KeWaitForSingleObject(&Ccb
->ConnectEvent
,
347 DPRINT("KeWaitForSingleObject() returned (Status %lx)\n", Status
);
354 * FUNCTION: Return current state of a pipe
356 * Irp = Pointer to I/O request packet
357 * IrpSp = Pointer to current stack location of Irp
359 * Status of operation
363 * FUNCTION: Peek at a pipe (get information about messages)
365 * Irp = Pointer to I/O request packet
366 * IoStack = Pointer to current stack location of Irp
368 * Status of operation
371 NpfsPeekPipe(PIRP Irp
,
372 PIO_STACK_LOCATION IoStack
)
374 ULONG OutputBufferLength
;
375 ULONG ReturnLength
= 0;
376 PFILE_PIPE_PEEK_BUFFER Reply
;
380 ULONG MessageCount
= 0;
382 ULONG ReadDataAvailable
;
385 DPRINT("NpfsPeekPipe\n");
387 OutputBufferLength
= IoStack
->Parameters
.DeviceIoControl
.OutputBufferLength
;
388 DPRINT("OutputBufferLength: %lu\n", OutputBufferLength
);
390 /* Validate parameters */
391 if (OutputBufferLength
< sizeof(FILE_PIPE_PEEK_BUFFER
))
393 DPRINT1("Buffer too small\n");
394 return STATUS_INVALID_PARAMETER
;
397 Ccb
= IoStack
->FileObject
->FsContext2
;
398 Reply
= (PFILE_PIPE_PEEK_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
402 Reply
->NamedPipeState
= Ccb
->PipeState
;
404 Reply
->ReadDataAvailable
= Ccb
->ReadDataAvailable
;
405 DPRINT("ReadDataAvailable: %lu\n", Ccb
->ReadDataAvailable
);
407 ExAcquireFastMutex(&Ccb
->DataListLock
);
408 BufferPtr
= Ccb
->ReadPtr
;
409 DPRINT("BufferPtr = %x\n", BufferPtr
);
410 if (Ccb
->Fcb
->PipeType
== FILE_PIPE_BYTE_STREAM_TYPE
)
412 DPRINT("Byte Stream Mode\n");
413 Reply
->MessageLength
= Ccb
->ReadDataAvailable
;
414 DPRINT("Reply->MessageLength %lu\n",Reply
->MessageLength
);
417 if (Reply
->Data
[0] && (OutputBufferLength
>= Ccb
->ReadDataAvailable
+ FIELD_OFFSET(FILE_PIPE_PEEK_BUFFER
, Data
[0])))
419 ReturnLength
= Ccb
->ReadDataAvailable
;
420 memcpy(&Reply
->Data
[0], (PVOID
)BufferPtr
, Ccb
->ReadDataAvailable
);
425 DPRINT("Message Mode\n");
426 ReadDataAvailable
=Ccb
->ReadDataAvailable
;
428 if (ReadDataAvailable
> 0)
430 memcpy(&Reply
->MessageLength
, BufferPtr
, sizeof(ULONG
));
432 while ((ReadDataAvailable
> 0) && (BufferPtr
< Ccb
->WritePtr
))
434 memcpy(&MessageLength
, BufferPtr
, sizeof(MessageLength
));
436 ASSERT(MessageLength
> 0);
438 DPRINT("MessageLength = %lu\n",MessageLength
);
439 ReadDataAvailable
-= MessageLength
;
442 /* If its the first message, copy the Message if the size of buffer is large enough */
446 && (OutputBufferLength
>= (MessageLength
+ FIELD_OFFSET(FILE_PIPE_PEEK_BUFFER
, Data
[0]))))
448 memcpy(&Reply
->Data
[0], (PVOID
)((ULONG_PTR
)BufferPtr
+ sizeof(MessageLength
)), MessageLength
);
449 ReturnLength
= MessageLength
;
453 BufferPtr
=(PVOID
)((ULONG_PTR
)BufferPtr
+ MessageLength
+ sizeof(MessageLength
));
454 DPRINT("BufferPtr = %x\n", BufferPtr
);
455 DPRINT("ReadDataAvailable: %lu\n", ReadDataAvailable
);
458 if (ReadDataAvailable
!= 0)
460 DPRINT1("Possible memory corruption.\n");
465 ExReleaseFastMutex(&Ccb
->DataListLock
);
467 Reply
->NumberOfMessages
= MessageCount
;
469 Irp
->IoStatus
.Information
= ReturnLength
+ FIELD_OFFSET(FILE_PIPE_PEEK_BUFFER
, Data
[0]);
470 Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
472 Status
= STATUS_SUCCESS
;
474 DPRINT("NpfsPeekPipe done\n");
481 NpfsFileSystemControl(PDEVICE_OBJECT DeviceObject
,
484 PIO_STACK_LOCATION IoStack
;
485 PFILE_OBJECT FileObject
;
487 PNPFS_DEVICE_EXTENSION DeviceExt
;
491 DPRINT("NpfsFileSystemContol(DeviceObject %p Irp %p)\n", DeviceObject
, Irp
);
493 DeviceExt
= (PNPFS_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
;
494 IoStack
= IoGetCurrentIrpStackLocation(Irp
);
495 DPRINT("IoStack: %p\n", IoStack
);
496 FileObject
= IoStack
->FileObject
;
497 DPRINT("FileObject: %p\n", FileObject
);
498 Ccb
= FileObject
->FsContext2
;
499 DPRINT("CCB: %p\n", Ccb
);
501 DPRINT("Pipe: %p\n", Fcb
);
502 DPRINT("PipeName: %wZ\n", &Fcb
->PipeName
);
504 Irp
->IoStatus
.Information
= 0;
506 switch (IoStack
->Parameters
.FileSystemControl
.FsControlCode
)
508 case FSCTL_PIPE_ASSIGN_EVENT
:
509 DPRINT1("Assign event not implemented\n");
510 Status
= STATUS_NOT_IMPLEMENTED
;
513 case FSCTL_PIPE_DISCONNECT
:
514 DPRINT("Disconnecting pipe %wZ\n", &Fcb
->PipeName
);
515 Status
= NpfsDisconnectPipe(Ccb
);
518 case FSCTL_PIPE_LISTEN
:
519 DPRINT("Connecting pipe %wZ\n", &Fcb
->PipeName
);
520 Status
= NpfsConnectPipe(Irp
, Ccb
);
523 case FSCTL_PIPE_PEEK
:
524 DPRINT("Peeking pipe %wZ\n", &Fcb
->PipeName
);
525 Status
= NpfsPeekPipe(Irp
, (PIO_STACK_LOCATION
)IoStack
);
528 case FSCTL_PIPE_QUERY_EVENT
:
529 DPRINT1("Query event not implemented\n");
530 Status
= STATUS_NOT_IMPLEMENTED
;
533 case FSCTL_PIPE_TRANSCEIVE
:
534 /* If you implement this, please remove the workaround in
535 lib/kernel32/file/npipe.c function TransactNamedPipe() */
536 DPRINT1("Transceive not implemented\n");
537 Status
= STATUS_NOT_IMPLEMENTED
;
540 case FSCTL_PIPE_WAIT
:
541 DPRINT("Waiting for pipe %wZ\n", &Fcb
->PipeName
);
542 Status
= NpfsWaitPipe(Irp
, Ccb
);
545 case FSCTL_PIPE_IMPERSONATE
:
546 DPRINT1("Impersonate not implemented\n");
547 Status
= STATUS_NOT_IMPLEMENTED
;
550 case FSCTL_PIPE_SET_CLIENT_PROCESS
:
551 DPRINT1("Set client process not implemented\n");
552 Status
= STATUS_NOT_IMPLEMENTED
;
555 case FSCTL_PIPE_QUERY_CLIENT_PROCESS
:
556 DPRINT1("Query client process not implemented\n");
557 Status
= STATUS_NOT_IMPLEMENTED
;
560 case FSCTL_PIPE_INTERNAL_READ
:
561 DPRINT1("Internal read not implemented\n");
562 Status
= STATUS_NOT_IMPLEMENTED
;
565 case FSCTL_PIPE_INTERNAL_WRITE
:
566 DPRINT1("Internal write not implemented\n");
567 Status
= STATUS_NOT_IMPLEMENTED
;
570 case FSCTL_PIPE_INTERNAL_TRANSCEIVE
:
571 DPRINT1("Internal transceive not implemented\n");
572 Status
= STATUS_NOT_IMPLEMENTED
;
575 case FSCTL_PIPE_INTERNAL_READ_OVFLOW
:
576 DPRINT1("Internal read overflow not implemented\n");
577 Status
= STATUS_NOT_IMPLEMENTED
;
581 DPRINT1("Unrecognized IoControlCode: %x\n",
582 IoStack
->Parameters
.FileSystemControl
.FsControlCode
);
583 Status
= STATUS_UNSUCCESSFUL
;
586 if (Status
!= STATUS_PENDING
)
588 Irp
->IoStatus
.Status
= Status
;
590 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
598 NpfsFlushBuffers(PDEVICE_OBJECT DeviceObject
,
601 /* FIXME: Implement */
603 Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
604 Irp
->IoStatus
.Information
= 0;
606 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
608 return STATUS_SUCCESS
;