2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Kernel Streaming
4 * FILE: drivers/wdm/audio/backpln/portcls/irpstream.cpp
5 * PURPOSE: IRP Stream handling
6 * PROGRAMMER: Johannes Anderwald
12 class CIrpQueue
: public IIrpQueue
15 STDMETHODIMP
QueryInterface( REFIID InterfaceId
, PVOID
* Interface
);
17 STDMETHODIMP_(ULONG
) AddRef()
19 InterlockedIncrement(&m_Ref
);
22 STDMETHODIMP_(ULONG
) Release()
24 InterlockedDecrement(&m_Ref
);
34 CIrpQueue(IUnknown
*OuterUnknown
){}
35 virtual ~CIrpQueue(){}
38 volatile ULONG m_CurrentOffset
;
40 ULONG m_NumDataAvailable
;
41 PKSPIN_CONNECT m_ConnectDetails
;
42 KSPIN_LOCK m_IrpListLock
;
44 LIST_ENTRY m_FreeIrpList
;
55 #define OFFSET_HEADERINDEX (0)
56 #define OFFSET_STREAMHEADER (2)
57 #define OFFSET_HEADERCOUNT (3)
60 #define STREAMHEADER_INDEX(Irp) (PtrToUlong(Irp->Tail.Overlay.DriverContext[OFFSET_HEADERINDEX]))
61 #define STREAMHEADER_COUNT(Irp) (PtrToUlong(Irp->Tail.Overlay.DriverContext[OFFSET_HEADERCOUNT]))
62 #define STREAMHEADER_CURRENT(Irp) (Irp->Tail.Overlay.DriverContext[OFFSET_STREAMHEADER])
67 CIrpQueue::QueryInterface(
71 if (IsEqualGUIDAligned(refiid
, IID_IUnknown
))
73 *Output
= PVOID(PUNKNOWN(this));
74 PUNKNOWN(*Output
)->AddRef();
75 return STATUS_SUCCESS
;
78 return STATUS_UNSUCCESSFUL
;
84 IN KSPIN_CONNECT
*ConnectDetails
,
88 m_ConnectDetails
= ConnectDetails
;
89 m_MaxFrameSize
= FrameSize
;
90 m_Alignment
= Alignment
;
92 InitializeListHead(&m_IrpList
);
93 InitializeListHead(&m_FreeIrpList
);
94 KeInitializeSpinLock(&m_IrpListLock
);
96 return STATUS_SUCCESS
;
101 CIrpQueue::AddMapping(
105 PKSSTREAM_HEADER Header
;
106 NTSTATUS Status
= STATUS_SUCCESS
;
107 PIO_STACK_LOCATION IoStack
;
108 ULONG NumHeaders
, NumData
, Index
;
111 PC_ASSERT(KeGetCurrentIrql() == PASSIVE_LEVEL
);
113 // get current irp stack location
114 IoStack
= IoGetCurrentIrpStackLocation(Irp
);
116 if (!Irp
->MdlAddress
)
118 // ioctl from KsStudio
119 // Wdmaud already probes buffers, therefore no need to probe it again
120 // probe the stream irp
121 if (IoStack
->Parameters
.DeviceIoControl
.IoControlCode
== IOCTL_KS_WRITE_STREAM
)
122 Status
= KsProbeStreamIrp(Irp
, KSSTREAM_WRITE
| KSPROBE_ALLOCATEMDL
| KSPROBE_PROBEANDLOCK
| KSPROBE_SYSTEMADDRESS
, 0);
123 else if (IoStack
->Parameters
.DeviceIoControl
.IoControlCode
== IOCTL_KS_READ_STREAM
)
124 Status
= KsProbeStreamIrp(Irp
, KSSTREAM_READ
| KSPROBE_ALLOCATEMDL
| KSPROBE_PROBEANDLOCK
| KSPROBE_SYSTEMADDRESS
, 0);
129 if (!NT_SUCCESS(Status
))
131 DPRINT("KsProbeStreamIrp failed with %x\n", Status
);
136 // get first stream header
138 if (Irp
->RequestorMode
== UserMode
)
139 Header
= (PKSSTREAM_HEADER
)Irp
->AssociatedIrp
.SystemBuffer
;
141 Header
= (PKSSTREAM_HEADER
)Irp
->UserBuffer
;
146 // calculate num headers
147 NumHeaders
= IoStack
->Parameters
.DeviceIoControl
.OutputBufferLength
/ Header
->Size
;
149 // assume headers of same length
150 PC_ASSERT(IoStack
->Parameters
.DeviceIoControl
.OutputBufferLength
% Header
->Size
== 0);
153 // get first audio buffer
154 Mdl
= Irp
->MdlAddress
;
158 // store the current stream header
159 Irp
->Tail
.Overlay
.DriverContext
[OFFSET_STREAMHEADER
] = (PVOID
)Header
;
160 // store header count
161 Irp
->Tail
.Overlay
.DriverContext
[OFFSET_HEADERCOUNT
] = UlongToPtr(NumHeaders
);
163 // store current header index
164 Irp
->Tail
.Overlay
.DriverContext
[OFFSET_HEADERINDEX
] = UlongToPtr(0);
167 // prepare all headers
168 for(Index
= 0; Index
< NumHeaders
; Index
++)
174 if (Irp
->RequestorMode
== UserMode
)
176 Header
->Data
= MmGetSystemAddressForMdlSafe(Mdl
, NormalPagePriority
);
181 // insufficient resources
182 ExFreePool(Irp
->AssociatedIrp
.SystemBuffer
);
183 Irp
->AssociatedIrp
.SystemBuffer
= NULL
;
184 // complete and forget request
185 Irp
->IoStatus
.Status
= STATUS_INSUFFICIENT_RESOURCES
;
186 Irp
->IoStatus
.Information
= 0;
188 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
189 return STATUS_INSUFFICIENT_RESOURCES
;
192 // increment num mappings
193 InterlockedIncrement(&m_NumMappings
);
195 // increment available data
196 InterlockedExchangeAdd((PLONG
)&m_NumDataAvailable
,
197 (max(Header
->DataUsed
, Header
->FrameExtent
)));
199 NumData
+= max(Header
->DataUsed
, Header
->FrameExtent
);
201 // move to next header
202 Header
= (PKSSTREAM_HEADER
)((ULONG_PTR
)Header
+ Header
->Size
);
208 DPRINT("StreamHeaders %u NumData %u FrameSize %u NumDataAvailable %u\n", NumHeaders
, NumData
, m_MaxFrameSize
, m_NumDataAvailable
);
211 // mark irp as pending
212 IoMarkIrpPending(Irp
);
214 // add irp to cancelable queue
215 KsAddIrpToCancelableQueue(&m_IrpList
, &m_IrpListLock
, Irp
, KsListEntryTail
, NULL
);
217 // disable mapping failed status
218 m_OutOfMapping
= FALSE
;
226 CIrpQueue::GetMapping(
228 OUT PULONG BufferSize
)
232 //PIO_STACK_LOCATION IoStack;
233 PKSSTREAM_HEADER StreamHeader
;
235 // check if there is an irp in the partially processed
239 if (m_Irp
->Cancel
== FALSE
)
242 Offset
= m_CurrentOffset
;
246 // irp has been cancelled
247 m_Irp
->IoStatus
.Status
= STATUS_CANCELLED
;
248 IoCompleteRequest(m_Irp
, IO_NO_INCREMENT
);
254 // get a fresh new irp from the queue
255 m_Irp
= Irp
= KsRemoveIrpFromCancelableQueue(&m_IrpList
, &m_IrpListLock
, KsListEntryHead
, KsAcquireAndRemoveOnlySingleItem
);
256 m_CurrentOffset
= Offset
= 0;
261 // no irp buffer available
263 return STATUS_UNSUCCESSFUL
;
267 StreamHeader
= (PKSSTREAM_HEADER
)Irp
->Tail
.Overlay
.DriverContext
[2];
270 PC_ASSERT(StreamHeader
);
273 if (StreamHeader
->DataUsed
)
274 *BufferSize
= StreamHeader
->DataUsed
- Offset
;
276 *BufferSize
= StreamHeader
->FrameExtent
- Offset
;
278 PC_ASSERT(*BufferSize
);
281 *Buffer
= &((PUCHAR
)StreamHeader
->Data
)[Offset
];
283 // unset flag that no irps are available
284 m_OutOfMapping
= FALSE
;
286 return STATUS_SUCCESS
;
291 CIrpQueue::UpdateMapping(
292 IN ULONG BytesWritten
)
294 PKSSTREAM_HEADER StreamHeader
;
295 ULONG Size
, NumData
, Index
;
299 // silence buffer was used
304 StreamHeader
= (PKSSTREAM_HEADER
)STREAMHEADER_CURRENT(m_Irp
);
307 // ASSERT(StreamHeader);
309 // add to current offset
310 InterlockedExchangeAdd((volatile PLONG
)&m_CurrentOffset
, (LONG
)BytesWritten
);
312 // decrement available data counter
313 m_NumDataAvailable
-= BytesWritten
;
315 if (StreamHeader
->DataUsed
)
316 Size
= StreamHeader
->DataUsed
;
318 Size
= StreamHeader
->FrameExtent
;
322 if (m_CurrentOffset
>= Size
)
324 if (STREAMHEADER_INDEX(m_Irp
) + 1 < STREAMHEADER_COUNT(m_Irp
))
326 // the irp has at least one more stream header
327 m_Irp
->Tail
.Overlay
.DriverContext
[OFFSET_HEADERINDEX
] = UlongToPtr(STREAMHEADER_INDEX(m_Irp
) + 1);
329 // get next stream header
330 StreamHeader
= (PKSSTREAM_HEADER
)((ULONG_PTR
)StreamHeader
+ StreamHeader
->Size
);
332 // store next stream header
333 STREAMHEADER_CURRENT(m_Irp
) = (PVOID
)StreamHeader
;
335 // reset current offset
342 // irp has been processed completly
344 if (m_Irp
->RequestorMode
== KernelMode
)
345 StreamHeader
= (PKSSTREAM_HEADER
)m_Irp
->UserBuffer
;
347 StreamHeader
= (PKSSTREAM_HEADER
)m_Irp
->AssociatedIrp
.SystemBuffer
;
349 // loop all stream headers
350 for(Index
= 0; Index
< STREAMHEADER_COUNT(m_Irp
); Index
++)
352 PC_ASSERT(StreamHeader
);
354 // add size of buffer
355 // depends on if the buffer is input / output
356 if (StreamHeader
->DataUsed
)
357 Size
= StreamHeader
->DataUsed
;
359 Size
= StreamHeader
->FrameExtent
;
364 // get next stream header
365 StreamHeader
= (PKSSTREAM_HEADER
)((ULONG_PTR
)StreamHeader
+ StreamHeader
->Size
);
368 if (m_ConnectDetails
->Interface
.Id
== KSINTERFACE_STANDARD_LOOPED_STREAMING
)
370 // looped streaming repeat the buffers untill
371 // the caller decides to stop the streams
373 // reset stream header index
374 m_Irp
->Tail
.Overlay
.DriverContext
[OFFSET_HEADERINDEX
] = UlongToPtr(0);
376 KsAddIrpToCancelableQueue(&m_IrpList
, &m_IrpListLock
, m_Irp
, KsListEntryTail
, NULL
);
381 // increment available data
382 InterlockedExchangeAdd((PLONG
)&m_NumDataAvailable
, NumData
);
387 // store operation status
388 m_Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
389 m_Irp
->IoStatus
.Information
= 0;
391 // complete the request
392 IoCompleteRequest(m_Irp
, IO_SOUND_INCREMENT
);
394 // remove irp as it is complete
406 // returns the amount of audio stream data available
407 return m_NumDataAvailable
;
412 CIrpQueue::CancelBuffers()
414 // is there an active irp
417 // re-insert it to cancelable queue
418 KsAddIrpToCancelableQueue(&m_IrpList
, &m_IrpListLock
, m_Irp
, KsListEntryTail
, NULL
);
424 KsCancelIo(&m_IrpList
, &m_IrpListLock
);
425 // reset number of mappings
427 // reset number of data available
428 m_NumDataAvailable
= 0;
436 CIrpQueue::GetMappingWithTag(
438 OUT PPHYSICAL_ADDRESS PhysicalAddress
,
439 OUT PVOID
*VirtualAddress
,
440 OUT PULONG ByteCount
,
443 PKSSTREAM_HEADER StreamHeader
;
447 PC_ASSERT(Tag
!= NULL
);
449 // get an irp from the queue
450 Irp
= KsRemoveIrpFromCancelableQueue(&m_IrpList
, &m_IrpListLock
, KsListEntryHead
, KsAcquireAndRemoveOnlySingleItem
);
452 // check if there is an irp
456 m_OutOfMapping
= TRUE
;
457 return STATUS_NOT_FOUND
;
460 //FIXME support more than one stream header
461 PC_ASSERT(STREAMHEADER_COUNT(Irp
) == 1);
463 // HACK get stream header
464 StreamHeader
= (PKSSTREAM_HEADER
)Irp
->Tail
.Overlay
.DriverContext
[2];
466 // store mapping in the free list
467 ExInterlockedInsertTailList(&m_FreeIrpList
, &Irp
->Tail
.Overlay
.ListEntry
, &m_IrpListLock
);
470 *PhysicalAddress
= MmGetPhysicalAddress(StreamHeader
->Data
);
471 *VirtualAddress
= StreamHeader
->Data
;
472 *ByteCount
= StreamHeader
->DataUsed
;
474 // decrement mapping count
475 InterlockedDecrement(&m_NumMappings
);
476 // decrement num data available
477 m_NumDataAvailable
-= StreamHeader
->DataUsed
;
480 Irp
->Tail
.Overlay
.DriverContext
[3] = Tag
;
483 return STATUS_SUCCESS
;
488 CIrpQueue::ReleaseMappingWithTag(
492 PLIST_ENTRY CurEntry
;
493 PKSSTREAM_HEADER StreamHeader
;
495 DPRINT("CIrpQueue::ReleaseMappingWithTag Tag %p\n", Tag
);
497 // remove irp from used list
498 CurEntry
= ExInterlockedRemoveHeadList(&m_FreeIrpList
, &m_IrpListLock
);
502 // get irp from list entry
503 Irp
= (PIRP
)CONTAINING_RECORD(CurEntry
, IRP
, Tail
.Overlay
.ListEntry
);
505 // HACK get stream header
506 StreamHeader
= (PKSSTREAM_HEADER
)Irp
->Tail
.Overlay
.DriverContext
[2];
508 // driver must release items in the same order
509 PC_ASSERT(Irp
->Tail
.Overlay
.DriverContext
[3] == Tag
);
511 // irp has been processed completly
512 Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
514 // frame extend contains the original request size, DataUsed contains the real buffer size
515 // is different when kmixer performs channel conversion, upsampling etc
517 Irp
->IoStatus
.Information
= StreamHeader
->FrameExtent
;
519 // complete the request
520 IoCompleteRequest(Irp
, IO_SOUND_INCREMENT
);
522 return STATUS_SUCCESS
;
527 CIrpQueue::HasLastMappingFailed()
529 return m_OutOfMapping
;
534 CIrpQueue::GetCurrentIrpOffset()
537 return m_CurrentOffset
;
542 CIrpQueue::GetAcquiredTagRange(
549 PLIST_ENTRY CurEntry
;
551 KeAcquireSpinLock(&m_IrpListLock
, &OldLevel
);
553 if (!IsListEmpty(&m_FreeIrpList
))
556 CurEntry
= RemoveHeadList(&m_FreeIrpList
);
557 // get irp from list entry
558 Irp
= (PIRP
)CONTAINING_RECORD(CurEntry
, IRP
, Tail
.Overlay
.ListEntry
);
560 // get tag of first acquired buffer
561 *FirstTag
= Irp
->Tail
.Overlay
.DriverContext
[3];
564 InsertHeadList(&m_FreeIrpList
, &Irp
->Tail
.Overlay
.ListEntry
);
567 CurEntry
= RemoveTailList(&m_FreeIrpList
);
568 // get irp from list entry
569 Irp
= (PIRP
)CONTAINING_RECORD(CurEntry
, IRP
, Tail
.Overlay
.ListEntry
);
571 // get tag of first acquired buffer
572 *LastTag
= Irp
->Tail
.Overlay
.DriverContext
[3];
575 InsertTailList(&m_FreeIrpList
, &Irp
->Tail
.Overlay
.ListEntry
);
582 KeReleaseSpinLock(&m_IrpListLock
, OldLevel
);
590 IN IIrpQueue
**Queue
)
592 CIrpQueue
*This
= new(NonPagedPool
, TAG_PORTCLASS
)CIrpQueue(NULL
);
594 return STATUS_INSUFFICIENT_RESOURCES
;
598 *Queue
= (IIrpQueue
*)This
;
599 return STATUS_SUCCESS
;