DrvRamDisk.cpp@ 106297

Last change on this file since 106297 was 106061, checked in by vboxsync, 2 months ago
Copyright year updates by scm.
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 68.6 KB

Line
1	/* $Id: DrvRamDisk.cpp 106061 2024-09-16 14:03:52Z vboxsync $ */
2	/** @file
3	* VBox storage devices: RAM disk driver.
4	*/
5
6	/*
7	* Copyright (C) 2016-2024 Oracle and/or its affiliates.
8	*
9	* This file is part of VirtualBox base platform packages, as
10	* available from https://www.virtualbox.org.
11	*
12	* This program is free software; you can redistribute it and/or
13	* modify it under the terms of the GNU General Public License
14	* as published by the Free Software Foundation, in version 3 of the
15	* License.
16	*
17	* This program is distributed in the hope that it will be useful, but
18	* WITHOUT ANY WARRANTY; without even the implied warranty of
19	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20	* General Public License for more details.
21	*
22	* You should have received a copy of the GNU General Public License
23	* along with this program; if not, see <https://www.gnu.org/licenses>.
24	*
25	* SPDX-License-Identifier: GPL-3.0-only
26	*/
27
28
29	/*********************************************************************************************************************************
30	* Header Files *
31	*********************************************************************************************************************************/
32	#define LOG_GROUP LOG_GROUP_DRV_DISK_INTEGRITY
33	#include <VBox/vmm/pdmdrv.h>
34	#include <VBox/vmm/pdmstorageifs.h>
35	#include <iprt/assert.h>
36	#include <iprt/string.h>
37	#include <iprt/uuid.h>
38	#include <iprt/avl.h>
39	#include <iprt/list.h>
40	#include <iprt/mem.h>
41	#include <iprt/memcache.h>
42	#include <iprt/message.h>
43	#include <iprt/sg.h>
44	#include <iprt/time.h>
45	#include <iprt/semaphore.h>
46	#include <iprt/asm.h>
47	#include <iprt/req.h>
48	#include <iprt/thread.h>
49
50	#include "VBoxDD.h"
51	#include "IOBufMgmt.h"
52
53
54	/*********************************************************************************************************************************
55	* Structures and Typedefs *
56	*********************************************************************************************************************************/
57
58	/** Pointer to a ramdisk driver instance. */
59	typedef struct DRVRAMDISK *PDRVRAMDISK;
60
61	/**
62	* Disk segment.
63	*/
64	typedef struct DRVDISKSEGMENT
65	{
66	/** AVL core. */
67	AVLRFOFFNODECORE Core;
68	/** Size of the segment */
69	size_t cbSeg;
70	/** Data for this segment */
71	uint8_t *pbSeg;
72	} DRVDISKSEGMENT, *PDRVDISKSEGMENT;
73
74	/**
75	* VD I/O request state.
76	*/
77	typedef enum VDIOREQSTATE
78	{
79	/** Invalid. */
80	VDIOREQSTATE_INVALID = 0,
81	/** The request is not in use and resides on the free list. */
82	VDIOREQSTATE_FREE,
83	/** The request was just allocated and is not active. */
84	VDIOREQSTATE_ALLOCATED,
85	/** The request was allocated and is in use. */
86	VDIOREQSTATE_ACTIVE,
87	/** The request was suspended and is not actively processed. */
88	VDIOREQSTATE_SUSPENDED,
89	/** The request is in the last step of completion and syncs memory. */
90	VDIOREQSTATE_COMPLETING,
91	/** The request completed. */
92	VDIOREQSTATE_COMPLETED,
93	/** The request was aborted but wasn't returned as complete from the storage
94	* layer below us. */
95	VDIOREQSTATE_CANCELED,
96	/** 32bit hack. */
97	VDIOREQSTATE_32BIT_HACK = 0x7fffffff
98	} VDIOREQSTATE;
99
100	/**
101	* VD I/O Request.
102	*/
103	typedef struct PDMMEDIAEXIOREQINT
104	{
105	/** List node for the list of allocated requests. */
106	RTLISTNODE NdAllocatedList;
107	/** List for requests waiting for I/O memory or on the redo list. */
108	RTLISTNODE NdLstWait;
109	/** I/O request type. */
110	PDMMEDIAEXIOREQTYPE enmType;
111	/** Request state. */
112	volatile VDIOREQSTATE enmState;
113	/** I/O request ID. */
114	PDMMEDIAEXIOREQID uIoReqId;
115	/** Pointer to the disk container. */
116	PDRVRAMDISK pDisk;
117	/** Flags. */
118	uint32_t fFlags;
119	/** Timestamp when the request was submitted. */
120	uint64_t tsSubmit;
121	/** Type dependent data. */
122	union
123	{
124	/** Read/Write request sepcific data. */
125	struct
126	{
127	/** Start offset of the request. */
128	uint64_t offStart;
129	/** Size of the request. */
130	size_t cbReq;
131	/** Size left for this request. */
132	size_t cbReqLeft;
133	/** Size of the allocated I/O buffer. */
134	size_t cbIoBuf;
135	/** I/O buffer descriptor. */
136	IOBUFDESC IoBuf;
137	} ReadWrite;
138	/** Discard specific data. */
139	struct
140	{
141	/** Pointer to array of ranges to discard. */
142	PRTRANGE paRanges;
143	/** Number of ranges to discard. */
144	unsigned cRanges;
145	} Discard;
146	};
147	/** Allocator specific memory - variable size. */
148	uint8_t abAlloc[1];
149	} PDMMEDIAEXIOREQINT;
150	/** Pointer to a VD I/O request. */
151	typedef PDMMEDIAEXIOREQINT *PPDMMEDIAEXIOREQINT;
152
153	/**
154	* Structure for holding a list of allocated requests.
155	*/
156	typedef struct VDLSTIOREQALLOC
157	{
158	/** Mutex protecting the table of allocated requests. */
159	RTSEMFASTMUTEX hMtxLstIoReqAlloc;
160	/** List anchor. */
161	RTLISTANCHOR LstIoReqAlloc;
162	} VDLSTIOREQALLOC;
163	typedef VDLSTIOREQALLOC *PVDLSTIOREQALLOC;
164
165	/** Number of bins for allocated requests. */
166	#define DRVVD_VDIOREQ_ALLOC_BINS 8
167
168	/**
169	* Disk integrity driver instance data.
170	*
171	* @implements PDMIMEDIA
172	*/
173	typedef struct DRVRAMDISK
174	{
175	/** Pointer driver instance. */
176	PPDMDRVINS pDrvIns;
177	/** Pointer to the media driver below us.
178	* This is NULL if the media is not mounted. */
179	PPDMIMEDIA pDrvMedia;
180	/** Our media interface */
181	PDMIMEDIA IMedia;
182
183	/** The media port interface above. */
184	PPDMIMEDIAPORT pDrvMediaPort;
185	/** Media port interface */
186	PDMIMEDIAPORT IMediaPort;
187
188	/** Flag whether the RAM disk was pre allocated. */
189	bool fPreallocRamDisk;
190	/** Flag whether to report a non totating medium. */
191	bool fNonRotational;
192	/** AVL tree containing the disk blocks to check. */
193	PAVLRFOFFTREE pTreeSegments;
194	/** Size of the disk. */
195	uint64_t cbDisk;
196	/** Size of one sector. */
197	uint32_t cbSector;
198
199	/** Worker request queue. */
200	RTREQQUEUE hReqQ;
201	/** Worker thread for async requests. */
202	RTTHREAD hThrdWrk;
203
204	/** @name IMEDIAEX interface support specific members.
205	* @{ */
206	/** Pointer to the IMEDIAEXPORT interface above us. */
207	PPDMIMEDIAEXPORT pDrvMediaExPort;
208	/** Our extended media interface. */
209	PDMIMEDIAEX IMediaEx;
210	/** Memory cache for the I/O requests. */
211	RTMEMCACHE hIoReqCache;
212	/** I/O buffer manager. */
213	IOBUFMGR hIoBufMgr;
214	/** Active request counter. */
215	volatile uint32_t cIoReqsActive;
216	/** Bins for allocated requests. */
217	VDLSTIOREQALLOC aIoReqAllocBins[DRVVD_VDIOREQ_ALLOC_BINS];
218	/** List of requests for I/O memory to be available - VDIOREQ::NdLstWait. */
219	RTLISTANCHOR LstIoReqIoBufWait;
220	/** Critical section protecting the list of requests waiting for I/O memory. */
221	RTCRITSECT CritSectIoReqsIoBufWait;
222	/** Number of requests waiting for a I/O buffer. */
223	volatile uint32_t cIoReqsWaiting;
224	/** Flag whether we have to resubmit requests on resume because the
225	* VM was suspended due to a recoverable I/O error.
226	*/
227	volatile bool fRedo;
228	/** List of requests we have to redo. */
229	RTLISTANCHOR LstIoReqRedo;
230	/** Criticial section protecting the list of waiting requests. */
231	RTCRITSECT CritSectIoReqRedo;
232	/** Number of errors logged so far. */
233	unsigned cErrors;
234	/** @} */
235
236	} DRVRAMDISK;
237
238
239	static void drvramdiskMediaExIoReqComplete(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq,
240	int rcReq);
241
242	/**
243	* Record a successful write to the virtual disk.
244	*
245	* @returns VBox status code.
246	* @param pThis Disk integrity driver instance data.
247	* @param pSgBuf The S/G buffer holding the data to write.
248	* @param off Start offset.
249	* @param cbWrite Number of bytes to record.
250	*/
251	static int drvramdiskWriteWorker(PDRVRAMDISK pThis, PRTSGBUF pSgBuf,
252	uint64_t off, size_t cbWrite)
253	{
254	int rc = VINF_SUCCESS;
255
256	LogFlowFunc(("pThis=%#p pSgBuf=%#p off=%llx cbWrite=%u\n",
257	pThis, pSgBuf, off, cbWrite));
258
259	/* Update the segments */
260	size_t cbLeft = cbWrite;
261	RTFOFF offCurr = (RTFOFF)off;
262
263	while (cbLeft)
264	{
265	PDRVDISKSEGMENT pSeg = (PDRVDISKSEGMENT)RTAvlrFileOffsetRangeGet(pThis->pTreeSegments, offCurr);
266	size_t cbRange = 0;
267	bool fSet = false;
268	unsigned offSeg = 0;
269
270	if (!pSeg)
271	{
272	/* Get next segment */
273	pSeg = (PDRVDISKSEGMENT)RTAvlrFileOffsetGetBestFit(pThis->pTreeSegments, offCurr, true);
274	if ( !pSeg
275	\|\| offCurr + (RTFOFF)cbLeft <= pSeg->Core.Key)
276	cbRange = cbLeft;
277	else
278	cbRange = pSeg->Core.Key - offCurr;
279
280	Assert(cbRange % 512 == 0);
281
282	/* Create new segment */
283	pSeg = (PDRVDISKSEGMENT)RTMemAllocZ(sizeof(DRVDISKSEGMENT));
284	if (pSeg)
285	{
286	pSeg->Core.Key = offCurr;
287	pSeg->Core.KeyLast = offCurr + (RTFOFF)cbRange - 1;
288	pSeg->cbSeg = cbRange;
289	pSeg->pbSeg = (uint8_t *)RTMemAllocZ(cbRange);
290	if (!pSeg->pbSeg)
291	RTMemFree(pSeg);
292	else
293	{
294	bool fInserted = RTAvlrFileOffsetInsert(pThis->pTreeSegments, &pSeg->Core);
295	AssertMsg(fInserted, ("Bug!\n")); RT_NOREF(fInserted);
296	fSet = true;
297	}
298	}
299	}
300	else
301	{
302	fSet = true;
303	offSeg = offCurr - pSeg->Core.Key;
304	cbRange = RT_MIN(cbLeft, (size_t)(pSeg->Core.KeyLast + 1 - offCurr));
305	}
306
307	if (fSet)
308	{
309	AssertPtr(pSeg);
310	size_t cbCopied = RTSgBufCopyToBuf(pSgBuf, pSeg->pbSeg + offSeg, cbRange);
311	Assert(cbCopied == cbRange); RT_NOREF(cbCopied);
312	}
313	else
314	RTSgBufAdvance(pSgBuf, cbRange);
315
316	offCurr += cbRange;
317	cbLeft -= cbRange;
318	}
319
320	return rc;
321	}
322
323	/**
324	* Read data from the ram disk.
325	*
326	* @returns VBox status code.
327	* @param pThis RAM disk driver instance data.
328	* @param pSgBuf The S/G buffer to store the data.
329	* @param off Start offset.
330	* @param cbRead Number of bytes to read.
331	*/
332	static int drvramdiskReadWorker(PDRVRAMDISK pThis, PRTSGBUF pSgBuf,
333	uint64_t off, size_t cbRead)
334	{
335	int rc = VINF_SUCCESS;
336
337	LogFlowFunc(("pThis=%#p pSgBuf=%#p off=%llx cbRead=%u\n",
338	pThis, pSgBuf, off, cbRead));
339
340	Assert(off % 512 == 0);
341	Assert(cbRead % 512 == 0);
342
343	/* Compare read data */
344	size_t cbLeft = cbRead;
345	RTFOFF offCurr = (RTFOFF)off;
346
347	while (cbLeft)
348	{
349	PDRVDISKSEGMENT pSeg = (PDRVDISKSEGMENT)RTAvlrFileOffsetRangeGet(pThis->pTreeSegments, offCurr);
350	size_t cbRange = 0;
351	bool fCmp = false;
352	unsigned offSeg = 0;
353
354	if (!pSeg)
355	{
356	/* Get next segment */
357	pSeg = (PDRVDISKSEGMENT)RTAvlrFileOffsetGetBestFit(pThis->pTreeSegments, offCurr, true);
358	if ( !pSeg
359	\|\| offCurr + (RTFOFF)cbLeft <= pSeg->Core.Key)
360	cbRange = cbLeft;
361	else
362	cbRange = pSeg->Core.Key - offCurr;
363
364	/* No segment means everything should be 0 for this part. */
365	RTSgBufSet(pSgBuf, 0, cbRange);
366	}
367	else
368	{
369	fCmp = true;
370	offSeg = offCurr - pSeg->Core.Key;
371	cbRange = RT_MIN(cbLeft, (size_t)(pSeg->Core.KeyLast + 1 - offCurr));
372
373	RTSGSEG Seg;
374	RTSGBUF SgBufSrc;
375
376	Seg.cbSeg = cbRange;
377	Seg.pvSeg = pSeg->pbSeg + offSeg;
378
379	RTSgBufInit(&SgBufSrc, &Seg, 1);
380	RTSgBufCopy(pSgBuf, &SgBufSrc, cbRange);
381	}
382
383	offCurr += cbRange;
384	cbLeft -= cbRange;
385	}
386
387	return rc;
388	}
389
390	/**
391	* Discards the given ranges from the disk.
392	*
393	* @returns VBox status code.
394	* @param pThis Disk integrity driver instance data.
395	* @param paRanges Array of ranges to discard.
396	* @param cRanges Number of ranges in the array.
397	*/
398	static int drvramdiskDiscardRecords(PDRVRAMDISK pThis, PCRTRANGE paRanges, unsigned cRanges)
399	{
400	int rc = VINF_SUCCESS;
401
402	LogFlowFunc(("pThis=%#p paRanges=%#p cRanges=%u\n", pThis, paRanges, cRanges));
403
404	for (unsigned i = 0; i < cRanges; i++)
405	{
406	uint64_t offStart = paRanges[i].offStart;
407	size_t cbLeft = paRanges[i].cbRange;
408
409	LogFlowFunc(("Discarding off=%llu cbRange=%zu\n", offStart, cbLeft));
410
411	while (cbLeft)
412	{
413	size_t cbRange;
414	PDRVDISKSEGMENT pSeg = (PDRVDISKSEGMENT)RTAvlrFileOffsetRangeGet(pThis->pTreeSegments, offStart);
415
416	if (!pSeg)
417	{
418	/* Get next segment */
419	pSeg = (PDRVDISKSEGMENT)RTAvlrFileOffsetGetBestFit(pThis->pTreeSegments, offStart, true);
420	if ( !pSeg
421	\|\| (RTFOFF)offStart + (RTFOFF)cbLeft <= pSeg->Core.Key)
422	cbRange = cbLeft;
423	else
424	cbRange = pSeg->Core.Key - offStart;
425
426	Assert(!(cbRange % 512));
427	}
428	else
429	{
430	size_t cbPreLeft, cbPostLeft;
431
432	cbRange = RT_MIN(cbLeft, pSeg->Core.KeyLast - offStart + 1);
433	cbPreLeft = offStart - pSeg->Core.Key;
434	cbPostLeft = pSeg->cbSeg - cbRange - cbPreLeft;
435
436	Assert(!(cbRange % 512));
437	Assert(!(cbPreLeft % 512));
438	Assert(!(cbPostLeft % 512));
439
440	LogFlowFunc(("cbRange=%zu cbPreLeft=%zu cbPostLeft=%zu\n",
441	cbRange, cbPreLeft, cbPostLeft));
442
443	RTAvlrFileOffsetRemove(pThis->pTreeSegments, pSeg->Core.Key);
444
445	if (!cbPreLeft && !cbPostLeft)
446	{
447	/* Just free the whole segment. */
448	LogFlowFunc(("Freeing whole segment pSeg=%#p\n", pSeg));
449	RTMemFree(pSeg->pbSeg);
450	RTMemFree(pSeg);
451	}
452	else if (cbPreLeft && !cbPostLeft)
453	{
454	/* Realloc to new size and insert. */
455	LogFlowFunc(("Realloc segment pSeg=%#p\n", pSeg));
456	pSeg->pbSeg = (uint8_t *)RTMemRealloc(pSeg->pbSeg, cbPreLeft);
457	pSeg = (PDRVDISKSEGMENT)RTMemRealloc(pSeg, sizeof(DRVDISKSEGMENT));
458	pSeg->Core.KeyLast = pSeg->Core.Key + cbPreLeft - 1;
459	pSeg->cbSeg = cbPreLeft;
460	bool fInserted = RTAvlrFileOffsetInsert(pThis->pTreeSegments, &pSeg->Core);
461	Assert(fInserted); RT_NOREF(fInserted);
462	}
463	else if (!cbPreLeft && cbPostLeft)
464	{
465	/* Move data to the front and realloc. */
466	LogFlowFunc(("Move data and realloc segment pSeg=%#p\n", pSeg));
467	memmove(pSeg->pbSeg, pSeg->pbSeg + cbRange, cbPostLeft);
468	pSeg = (PDRVDISKSEGMENT)RTMemRealloc(pSeg, sizeof(DRVDISKSEGMENT));
469	pSeg->pbSeg = (uint8_t *)RTMemRealloc(pSeg->pbSeg, cbPostLeft);
470	pSeg->Core.Key += cbRange;
471	pSeg->cbSeg = cbPostLeft;
472	bool fInserted = RTAvlrFileOffsetInsert(pThis->pTreeSegments, &pSeg->Core);
473	Assert(fInserted); RT_NOREF(fInserted);
474	}
475	else
476	{
477	/* Split the segment into 2 new segments. */
478	LogFlowFunc(("Split segment pSeg=%#p\n", pSeg));
479	PDRVDISKSEGMENT pSegPost = (PDRVDISKSEGMENT)RTMemAllocZ(sizeof(DRVDISKSEGMENT));
480	if (pSegPost)
481	{
482	pSegPost->Core.Key = pSeg->Core.Key + cbPreLeft + cbRange;
483	pSegPost->Core.KeyLast = pSeg->Core.KeyLast;
484	pSegPost->cbSeg = cbPostLeft;
485	pSegPost->pbSeg = (uint8_t *)RTMemAllocZ(cbPostLeft);
486	if (!pSegPost->pbSeg)
487	RTMemFree(pSegPost);
488	else
489	{
490	memcpy(pSegPost->pbSeg, pSeg->pbSeg + cbPreLeft + cbRange, cbPostLeft);
491	bool fInserted = RTAvlrFileOffsetInsert(pThis->pTreeSegments, &pSegPost->Core);
492	Assert(fInserted); RT_NOREF(fInserted);
493	}
494	}
495
496	/* Shrink the current segment. */
497	pSeg->pbSeg = (uint8_t *)RTMemRealloc(pSeg->pbSeg, cbPreLeft);
498	pSeg = (PDRVDISKSEGMENT)RTMemRealloc(pSeg, sizeof(DRVDISKSEGMENT));
499	pSeg->Core.KeyLast = pSeg->Core.Key + cbPreLeft - 1;
500	pSeg->cbSeg = cbPreLeft;
501	bool fInserted = RTAvlrFileOffsetInsert(pThis->pTreeSegments, &pSeg->Core);
502	Assert(fInserted); RT_NOREF(fInserted);
503	} /* if (cbPreLeft && cbPostLeft) */
504	}
505
506	offStart += cbRange;
507	cbLeft -= cbRange;
508	}
509	}
510
511	LogFlowFunc(("returns rc=%Rrc\n", rc));
512	return rc;
513	}
514
515	/* -=-=-=-=- IMedia -=-=-=-=- */
516
517
518	/*********************************************************************************************************************************
519	* Media interface methods *
520	*********************************************************************************************************************************/
521
522	/** @copydoc PDMIMEDIA::pfnRead */
523	static DECLCALLBACK(int) drvramdiskRead(PPDMIMEDIA pInterface,
524	uint64_t off, void *pvBuf, size_t cbRead)
525	{
526	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
527	RTSGSEG Seg;
528	RTSGBUF SgBuf;
529
530	Seg.cbSeg = cbRead;
531	Seg.pvSeg = pvBuf;
532	RTSgBufInit(&SgBuf, &Seg, 1);
533	return drvramdiskReadWorker(pThis, &SgBuf, off, cbRead);
534	}
535
536	/** @copydoc PDMIMEDIA::pfnWrite */
537	static DECLCALLBACK(int) drvramdiskWrite(PPDMIMEDIA pInterface,
538	uint64_t off, const void *pvBuf,
539	size_t cbWrite)
540	{
541	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
542	RTSGSEG Seg;
543	RTSGBUF SgBuf;
544
545	Seg.cbSeg = cbWrite;
546	Seg.pvSeg = (void *)pvBuf;
547	RTSgBufInit(&SgBuf, &Seg, 1);
548	return drvramdiskWriteWorker(pThis, &SgBuf, off, cbWrite);
549	}
550
551	/** @copydoc PDMIMEDIA::pfnFlush */
552	static DECLCALLBACK(int) drvramdiskFlush(PPDMIMEDIA pInterface)
553	{
554	RT_NOREF1(pInterface);
555	/* Nothing to do here. */
556	return VINF_SUCCESS;
557	}
558
559	/** @copydoc PDMIMEDIA::pfnGetSize */
560	static DECLCALLBACK(uint64_t) drvramdiskGetSize(PPDMIMEDIA pInterface)
561	{
562	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
563	return pThis->cbDisk;
564	}
565
566	/** @interface_method_impl{PDMIMEDIA,pfnBiosIsVisible} */
567	static DECLCALLBACK(bool) drvramdiskBiosIsVisible(PPDMIMEDIA pInterface)
568	{
569	RT_NOREF1(pInterface);
570	return false;
571	}
572
573	/** @copydoc PDMIMEDIA::pfnGetType */
574	static DECLCALLBACK(PDMMEDIATYPE) drvramdiskGetType(PPDMIMEDIA pInterface)
575	{
576	RT_NOREF1(pInterface);
577	return PDMMEDIATYPE_HARD_DISK;
578	}
579
580	/** @copydoc PDMIMEDIA::pfnIsReadOnly */
581	static DECLCALLBACK(bool) drvramdiskIsReadOnly(PPDMIMEDIA pInterface)
582	{
583	RT_NOREF1(pInterface);
584	return false; /** @todo */
585	}
586
587	/** @copydoc PDMIMEDIA::pfnBiosGetPCHSGeometry */
588	static DECLCALLBACK(int) drvramdiskBiosGetPCHSGeometry(PPDMIMEDIA pInterface,
589	PPDMMEDIAGEOMETRY pPCHSGeometry)
590	{
591	RT_NOREF2(pInterface, pPCHSGeometry);
592	return VERR_PDM_GEOMETRY_NOT_SET;
593	}
594
595	/** @copydoc PDMIMEDIA::pfnBiosSetPCHSGeometry */
596	static DECLCALLBACK(int) drvramdiskBiosSetPCHSGeometry(PPDMIMEDIA pInterface,
597	PCPDMMEDIAGEOMETRY pPCHSGeometry)
598	{
599	RT_NOREF2(pInterface, pPCHSGeometry);
600	return VERR_NOT_IMPLEMENTED;
601	}
602
603	/** @copydoc PDMIMEDIA::pfnBiosGetLCHSGeometry */
604	static DECLCALLBACK(int) drvramdiskBiosGetLCHSGeometry(PPDMIMEDIA pInterface,
605	PPDMMEDIAGEOMETRY pLCHSGeometry)
606	{
607	RT_NOREF2(pInterface, pLCHSGeometry);
608	return VERR_PDM_GEOMETRY_NOT_SET;
609	}
610
611	/** @copydoc PDMIMEDIA::pfnBiosSetLCHSGeometry */
612	static DECLCALLBACK(int) drvramdiskBiosSetLCHSGeometry(PPDMIMEDIA pInterface,
613	PCPDMMEDIAGEOMETRY pLCHSGeometry)
614	{
615	RT_NOREF2(pInterface, pLCHSGeometry);
616	return VERR_NOT_IMPLEMENTED;
617	}
618
619	/** @copydoc PDMIMEDIA::pfnGetUuid */
620	static DECLCALLBACK(int) drvramdiskGetUuid(PPDMIMEDIA pInterface, PRTUUID pUuid)
621	{
622	RT_NOREF1(pInterface);
623	return RTUuidClear(pUuid);
624	}
625
626	/** @copydoc PDMIMEDIA::pfnGetSectorSize */
627	static DECLCALLBACK(uint32_t) drvramdiskGetSectorSize(PPDMIMEDIA pInterface)
628	{
629	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
630	return pThis->cbSector;
631	}
632
633	/** @copydoc PDMIMEDIA::pfnDiscard */
634	static DECLCALLBACK(int) drvramdiskDiscard(PPDMIMEDIA pInterface, PCRTRANGE paRanges, unsigned cRanges)
635	{
636	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
637	return drvramdiskDiscardRecords(pThis, paRanges, cRanges);
638	}
639
640	/** @copydoc PDMIMEDIA::pfnReadPcBios */
641	static DECLCALLBACK(int) drvramdiskReadPcBios(PPDMIMEDIA pInterface,
642	uint64_t off, void *pvBuf, size_t cbRead)
643	{
644	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
645	RTSGSEG Seg;
646	RTSGBUF SgBuf;
647
648	Seg.cbSeg = cbRead;
649	Seg.pvSeg = pvBuf;
650	RTSgBufInit(&SgBuf, &Seg, 1);
651	return drvramdiskReadWorker(pThis, &SgBuf, off, cbRead);
652	}
653
654	/** @interface_method_impl{PDMIMEDIA,pfnIsNonRotational} */
655	static DECLCALLBACK(bool) drvramdiskIsNonRotational(PPDMIMEDIA pInterface)
656	{
657	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
658	return pThis->fNonRotational;
659	}
660
661	/** @interface_method_impl{PDMIMEDIA,pfnGetRegionCount} */
662	static DECLCALLBACK(uint32_t) drvramdiskGetRegionCount(PPDMIMEDIA pInterface)
663	{
664	RT_NOREF(pInterface);
665	return 1;
666	}
667
668	/** @interface_method_impl{PDMIMEDIA,pfnQueryRegionProperties} */
669	static DECLCALLBACK(int) drvramdiskQueryRegionProperties(PPDMIMEDIA pInterface, uint32_t uRegion, uint64_t *pu64LbaStart,
670	uint64_t pcBlocks, uint64_t pcbBlock,
671	PVDREGIONDATAFORM penmDataForm)
672	{
673	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
674	AssertReturn(uRegion == 0, VERR_INVALID_PARAMETER);
675	if (pu64LbaStart)
676	*pu64LbaStart = 0;
677	if (pcBlocks)
678	*pcBlocks = pThis->cbDisk / pThis->cbSector;
679	if (pcbBlock)
680	*pcbBlock = pThis->cbSector;
681	if (penmDataForm)
682	*penmDataForm = VDREGIONDATAFORM_RAW;
683	return VINF_SUCCESS;
684	}
685
686	/** @interface_method_impl{PDMIMEDIA,pfnQueryRegionPropertiesForLba} */
687	static DECLCALLBACK(int) drvramdiskQueryRegionPropertiesForLba(PPDMIMEDIA pInterface, uint64_t u64LbaStart,
688	uint32_t puRegion, uint64_t pcBlocks,
689	uint64_t *pcbBlock, PVDREGIONDATAFORM penmDataForm)
690	{
691	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMedia);
692	RT_NOREF(u64LbaStart);
693	if (puRegion)
694	*puRegion = 0;
695	if (pcBlocks)
696	*pcBlocks = pThis->cbDisk / pThis->cbSector;
697	if (pcbBlock)
698	*pcbBlock = pThis->cbSector;
699	if (penmDataForm)
700	*penmDataForm = VDREGIONDATAFORM_RAW;
701	return VINF_SUCCESS;
702	}
703
704
705	/*********************************************************************************************************************************
706	* Extended media interface methods *
707	*********************************************************************************************************************************/
708
709	static void drvramdiskMediaExIoReqWarningOutOfMemory(PPDMDRVINS pDrvIns)
710	{
711	int rc;
712	LogRel(("RamDisk#%u: Out of memory\n", pDrvIns->iInstance));
713	rc = PDMDrvHlpVMSetRuntimeError(pDrvIns, VMSETRTERR_FLAGS_SUSPEND \| VMSETRTERR_FLAGS_NO_WAIT, "DrvRamDisk_OOM",
714	N_("There is not enough free memory for the ramdisk"));
715	AssertRC(rc);
716	}
717
718	/**
719	* Checks whether a given status code indicates a recoverable error
720	* suspending the VM if it is.
721	*
722	* @returns Flag indicating whether the status code is a recoverable error
723	* (full disk, broken network connection).
724	* @param pThis VBox disk container instance data.
725	* @param rc Status code to check.
726	*/
727	static bool drvramdiskMediaExIoReqIsRedoSetWarning(PDRVRAMDISK pThis, int rc)
728	{
729	if (rc == VERR_NO_MEMORY)
730	{
731	if (ASMAtomicCmpXchgBool(&pThis->fRedo, true, false))
732	drvramdiskMediaExIoReqWarningOutOfMemory(pThis->pDrvIns);
733	return true;
734	}
735
736	return false;
737	}
738
739	/**
740	* Syncs the memory buffers between the I/O request allocator and the internal buffer.
741	*
742	* @returns VBox status code.
743	* @param pThis VBox disk container instance data.
744	* @param pIoReq I/O request to sync.
745	* @param fToIoBuf Flag indicating the sync direction.
746	* true to copy data from the allocators buffer to our internal buffer.
747	* false for the other direction.
748	*/
749	DECLINLINE(int) drvramdiskMediaExIoReqBufSync(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq, bool fToIoBuf)
750	{
751	int rc = VINF_SUCCESS;
752
753	Assert(pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ \|\| pIoReq->enmType == PDMMEDIAEXIOREQTYPE_WRITE);
754
755	/* Make sure the buffer is reset. */
756	RTSgBufReset(&pIoReq->ReadWrite.IoBuf.SgBuf);
757
758	if (fToIoBuf)
759	rc = pThis->pDrvMediaExPort->pfnIoReqCopyToBuf(pThis->pDrvMediaExPort, pIoReq, &pIoReq->abAlloc[0],
760	(uint32_t)(pIoReq->ReadWrite.cbReq - pIoReq->ReadWrite.cbReqLeft),
761	&pIoReq->ReadWrite.IoBuf.SgBuf,
762	RT_MIN(pIoReq->ReadWrite.cbIoBuf, pIoReq->ReadWrite.cbReqLeft));
763	else
764	rc = pThis->pDrvMediaExPort->pfnIoReqCopyFromBuf(pThis->pDrvMediaExPort, pIoReq, &pIoReq->abAlloc[0],
765	(uint32_t)(pIoReq->ReadWrite.cbReq - pIoReq->ReadWrite.cbReqLeft),
766	&pIoReq->ReadWrite.IoBuf.SgBuf,
767	RT_MIN(pIoReq->ReadWrite.cbIoBuf, pIoReq->ReadWrite.cbReqLeft));
768
769	RTSgBufReset(&pIoReq->ReadWrite.IoBuf.SgBuf);
770	return rc;
771	}
772
773	/**
774	* Hashes the I/O request ID to an index for the allocated I/O request bin.
775	*/
776	DECLINLINE(unsigned) drvramdiskMediaExIoReqIdHash(PDMMEDIAEXIOREQID uIoReqId)
777	{
778	return uIoReqId % DRVVD_VDIOREQ_ALLOC_BINS; /** @todo Find something better? */
779	}
780
781	/**
782	* Inserts the given I/O request in to the list of allocated I/O requests.
783	*
784	* @returns VBox status code.
785	* @param pThis VBox disk container instance data.
786	* @param pIoReq I/O request to insert.
787	*/
788	static int drvramdiskMediaExIoReqInsert(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
789	{
790	int rc = VINF_SUCCESS;
791	unsigned idxBin = drvramdiskMediaExIoReqIdHash(pIoReq->uIoReqId);
792
793	rc = RTSemFastMutexRequest(pThis->aIoReqAllocBins[idxBin].hMtxLstIoReqAlloc);
794	if (RT_SUCCESS(rc))
795	{
796	/* Search for conflicting I/O request ID. */
797	PPDMMEDIAEXIOREQINT pIt;
798	RTListForEach(&pThis->aIoReqAllocBins[idxBin].LstIoReqAlloc, pIt, PDMMEDIAEXIOREQINT, NdAllocatedList)
799	{
800	if (RT_UNLIKELY(pIt->uIoReqId == pIoReq->uIoReqId))
801	{
802	rc = VERR_PDM_MEDIAEX_IOREQID_CONFLICT;
803	break;
804	}
805	}
806	if (RT_SUCCESS(rc))
807	RTListAppend(&pThis->aIoReqAllocBins[idxBin].LstIoReqAlloc, &pIoReq->NdAllocatedList);
808	RTSemFastMutexRelease(pThis->aIoReqAllocBins[idxBin].hMtxLstIoReqAlloc);
809	}
810
811	return rc;
812	}
813
814	/**
815	* Removes the given I/O request from the list of allocated I/O requests.
816	*
817	* @returns VBox status code.
818	* @param pThis VBox disk container instance data.
819	* @param pIoReq I/O request to insert.
820	*/
821	static int drvramdiskMediaExIoReqRemove(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
822	{
823	int rc = VINF_SUCCESS;
824	unsigned idxBin = drvramdiskMediaExIoReqIdHash(pIoReq->uIoReqId);
825
826	rc = RTSemFastMutexRequest(pThis->aIoReqAllocBins[idxBin].hMtxLstIoReqAlloc);
827	if (RT_SUCCESS(rc))
828	{
829	RTListNodeRemove(&pIoReq->NdAllocatedList);
830	RTSemFastMutexRelease(pThis->aIoReqAllocBins[idxBin].hMtxLstIoReqAlloc);
831	}
832
833	return rc;
834	}
835
836	/**
837	* I/O request completion worker.
838	*
839	* @returns VBox status code.
840	* @param pThis VBox disk container instance data.
841	* @param pIoReq I/O request to complete.
842	* @param rcReq The status code the request completed with.
843	* @param fUpNotify Flag whether to notify the driver/device above us about the completion.
844	*/
845	static int drvramdiskMediaExIoReqCompleteWorker(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq, int rcReq, bool fUpNotify)
846	{
847	int rc;
848	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_COMPLETING, VDIOREQSTATE_ACTIVE);
849	if (fXchg)
850	ASMAtomicDecU32(&pThis->cIoReqsActive);
851	else
852	{
853	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
854	rcReq = VERR_PDM_MEDIAEX_IOREQ_CANCELED;
855	}
856
857	ASMAtomicXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_COMPLETED);
858
859	/*
860	* Leave a release log entry if the request was active for more than 25 seconds
861	* (30 seconds is the timeout of the guest).
862	*/
863	uint64_t tsNow = RTTimeMilliTS();
864	if (tsNow - pIoReq->tsSubmit >= 25 * 1000)
865	{
866	const char *pcszReq = NULL;
867
868	switch (pIoReq->enmType)
869	{
870	case PDMMEDIAEXIOREQTYPE_READ:
871	pcszReq = "Read";
872	break;
873	case PDMMEDIAEXIOREQTYPE_WRITE:
874	pcszReq = "Write";
875	break;
876	case PDMMEDIAEXIOREQTYPE_FLUSH:
877	pcszReq = "Flush";
878	break;
879	case PDMMEDIAEXIOREQTYPE_DISCARD:
880	pcszReq = "Discard";
881	break;
882	default:
883	pcszReq = "<Invalid>";
884	}
885
886	LogRel(("RamDisk#%u: %s request was active for %llu seconds\n",
887	pThis->pDrvIns->iInstance, pcszReq, (tsNow - pIoReq->tsSubmit) / 1000));
888	}
889
890	if (RT_FAILURE(rcReq))
891	{
892	/* Log the error. */
893	if (pThis->cErrors++ < 100)
894	{
895	if (rcReq == VERR_PDM_MEDIAEX_IOREQ_CANCELED)
896	{
897	if (pIoReq->enmType == PDMMEDIAEXIOREQTYPE_FLUSH)
898	LogRel(("RamDisk#%u: Aborted flush returned rc=%Rrc\n",
899	pThis->pDrvIns->iInstance, rcReq));
900	else
901	LogRel(("RamDisk#%u: Aborted %s (%u bytes left) returned rc=%Rrc\n",
902	pThis->pDrvIns->iInstance,
903	pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ
904	? "read"
905	: "write",
906	pIoReq->ReadWrite.cbReqLeft, rcReq));
907	}
908	else
909	{
910	if (pIoReq->enmType == PDMMEDIAEXIOREQTYPE_FLUSH)
911	LogRel(("RamDisk#%u: Flush returned rc=%Rrc\n",
912	pThis->pDrvIns->iInstance, rcReq));
913	else
914	LogRel(("RamDisk#%u: %s (%u bytes left) returned rc=%Rrc\n",
915	pThis->pDrvIns->iInstance,
916	pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ
917	? "Read"
918	: "Write",
919	pIoReq->ReadWrite.cbReqLeft, rcReq));
920	}
921	}
922	}
923
924	if (fUpNotify)
925	{
926	rc = pThis->pDrvMediaExPort->pfnIoReqCompleteNotify(pThis->pDrvMediaExPort,
927	pIoReq, &pIoReq->abAlloc[0], rcReq);
928	AssertRC(rc);
929	}
930
931	return rcReq;
932	}
933
934	/**
935	* Allocates a memory buffer suitable for I/O for the given request.
936	*
937	* @returns VBox status code.
938	* @retval VINF_PDM_MEDIAEX_IOREQ_IN_PROGRESS if there is no I/O memory available to allocate and
939	* the request was placed on a waiting list.
940	* @param pThis VBox disk container instance data.
941	* @param pIoReq I/O request to allocate memory for.
942	* @param cb Size of the buffer.
943	*/
944	DECLINLINE(int) drvramdiskMediaExIoReqBufAlloc(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq, size_t cb)
945	{
946	int rc = IOBUFMgrAllocBuf(pThis->hIoBufMgr, &pIoReq->ReadWrite.IoBuf, cb, &pIoReq->ReadWrite.cbIoBuf);
947	if (rc == VERR_NO_MEMORY)
948	{
949	RTCritSectEnter(&pThis->CritSectIoReqsIoBufWait);
950	RTListAppend(&pThis->LstIoReqIoBufWait, &pIoReq->NdLstWait);
951	RTCritSectLeave(&pThis->CritSectIoReqsIoBufWait);
952	ASMAtomicIncU32(&pThis->cIoReqsWaiting);
953	rc = VINF_PDM_MEDIAEX_IOREQ_IN_PROGRESS;
954	}
955
956	return rc;
957	}
958
959	/**
960	* Worker for a read request.
961	*
962	* @returns VBox status code.
963	* @param pThis RAM disk container instance data.
964	* @param pIoReq The read request.
965	*/
966	static DECLCALLBACK(int) drvramdiskIoReqReadWorker(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
967	{
968	size_t cbReqIo = RT_MIN(pIoReq->ReadWrite.cbReqLeft, pIoReq->ReadWrite.cbIoBuf);
969	int rc = drvramdiskReadWorker(pThis, &pIoReq->ReadWrite.IoBuf.SgBuf, pIoReq->ReadWrite.offStart,
970	cbReqIo);
971	drvramdiskMediaExIoReqComplete(pThis, pIoReq, rc);
972	return VINF_SUCCESS;
973	}
974
975	/**
976	* Worker for a read request.
977	*
978	* @returns VBox status code.
979	* @param pThis RAM disk container instance data.
980	* @param pIoReq The read request.
981	*/
982	static DECLCALLBACK(int) drvramdiskIoReqWriteWorker(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
983	{
984	size_t cbReqIo = RT_MIN(pIoReq->ReadWrite.cbReqLeft, pIoReq->ReadWrite.cbIoBuf);
985	int rc = drvramdiskWriteWorker(pThis, &pIoReq->ReadWrite.IoBuf.SgBuf, pIoReq->ReadWrite.offStart,
986	cbReqIo);
987	drvramdiskMediaExIoReqComplete(pThis, pIoReq, rc);
988	return VINF_SUCCESS;
989	}
990
991	/**
992	* Kicks the worker thread out of the loop.
993	*
994	* @returns VBox status code.
995	* @retval VERR_CANCELLED;
996	*/
997	static DECLCALLBACK(int) drvramdiskDestructQueue(void)
998	{
999	return VERR_CANCELLED;
1000	}
1001
1002	/**
1003	* Processes a read/write request.
1004	*
1005	* @returns VBox status code.
1006	* @param pThis VBox disk container instance data.
1007	* @param pIoReq I/O request to process.
1008	* @param fUpNotify Flag whether to notify the driver/device above us about the completion.
1009	*/
1010	static int drvramdiskMediaExIoReqReadWriteProcess(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq, bool fUpNotify)
1011	{
1012	int rc = VINF_SUCCESS;
1013
1014	Assert(pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ \|\| pIoReq->enmType == PDMMEDIAEXIOREQTYPE_WRITE);
1015
1016	while ( pIoReq->ReadWrite.cbReqLeft
1017	&& rc == VINF_SUCCESS)
1018	{
1019	if (pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ)
1020	rc = RTReqQueueCallEx(pThis->hReqQ, NULL, 0, RTREQFLAGS_NO_WAIT, (PFNRT)drvramdiskIoReqReadWorker, 2, pThis, pIoReq);
1021	else
1022	{
1023	/* Sync memory buffer from the request initiator. */
1024	rc = drvramdiskMediaExIoReqBufSync(pThis, pIoReq, true /* fToIoBuf */);
1025	if (RT_SUCCESS(rc))
1026	rc = RTReqQueueCallEx(pThis->hReqQ, NULL, 0, RTREQFLAGS_NO_WAIT,
1027	(PFNRT)drvramdiskIoReqWriteWorker, 2, pThis, pIoReq);
1028	}
1029
1030	if (rc == VINF_SUCCESS)
1031	rc = VINF_PDM_MEDIAEX_IOREQ_IN_PROGRESS;
1032	}
1033
1034	if (rc != VINF_PDM_MEDIAEX_IOREQ_IN_PROGRESS)
1035	{
1036	Assert(!pIoReq->ReadWrite.cbReqLeft \|\| RT_FAILURE(rc));
1037	rc = drvramdiskMediaExIoReqCompleteWorker(pThis, pIoReq, rc, fUpNotify);
1038	}
1039
1040	return rc;
1041	}
1042
1043
1044	/**
1045	* Frees a I/O memory buffer allocated previously.
1046	*
1047	* @param pThis VBox disk container instance data.
1048	* @param pIoReq I/O request for which to free memory.
1049	*/
1050	DECLINLINE(void) drvramdiskMediaExIoReqBufFree(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
1051	{
1052	if ( pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ
1053	\|\| pIoReq->enmType == PDMMEDIAEXIOREQTYPE_WRITE)
1054	{
1055	IOBUFMgrFreeBuf(&pIoReq->ReadWrite.IoBuf);
1056
1057	if (ASMAtomicReadU32(&pThis->cIoReqsWaiting) > 0)
1058	{
1059	/* Try to process as many requests as possible. */
1060	RTCritSectEnter(&pThis->CritSectIoReqsIoBufWait);
1061	PPDMMEDIAEXIOREQINT pIoReqCur, pIoReqNext;
1062
1063	RTListForEachSafe(&pThis->LstIoReqIoBufWait, pIoReqCur, pIoReqNext, PDMMEDIAEXIOREQINT, NdLstWait)
1064	{
1065	/* Allocate a suitable I/O buffer for this request. */
1066	int rc = IOBUFMgrAllocBuf(pThis->hIoBufMgr, &pIoReqCur->ReadWrite.IoBuf, pIoReqCur->ReadWrite.cbReq,
1067	&pIoReqCur->ReadWrite.cbIoBuf);
1068	if (rc == VINF_SUCCESS)
1069	{
1070	ASMAtomicDecU32(&pThis->cIoReqsWaiting);
1071	RTListNodeRemove(&pIoReqCur->NdLstWait);
1072
1073	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReqCur->enmState, VDIOREQSTATE_ACTIVE, VDIOREQSTATE_ALLOCATED);
1074	if (RT_UNLIKELY(!fXchg))
1075	{
1076	/* Must have been canceled inbetween. */
1077	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
1078	drvramdiskMediaExIoReqCompleteWorker(pThis, pIoReqCur, VERR_PDM_MEDIAEX_IOREQ_CANCELED, true /* fUpNotify */);
1079	}
1080	ASMAtomicIncU32(&pThis->cIoReqsActive);
1081	rc = drvramdiskMediaExIoReqReadWriteProcess(pThis, pIoReqCur, true /* fUpNotify */);
1082	}
1083	else
1084	{
1085	Assert(rc == VERR_NO_MEMORY);
1086	break;
1087	}
1088	}
1089	RTCritSectLeave(&pThis->CritSectIoReqsIoBufWait);
1090	}
1091	}
1092	}
1093
1094
1095	/**
1096	* Returns whether the VM is in a running state.
1097	*
1098	* @returns Flag indicating whether the VM is currently in a running state.
1099	* @param pThis VBox disk container instance data.
1100	*/
1101	DECLINLINE(bool) drvramdiskMediaExIoReqIsVmRunning(PDRVRAMDISK pThis)
1102	{
1103	VMSTATE enmVmState = PDMDrvHlpVMState(pThis->pDrvIns);
1104	if ( enmVmState == VMSTATE_RESUMING
1105	\|\| enmVmState == VMSTATE_RUNNING
1106	\|\| enmVmState == VMSTATE_RUNNING_LS
1107	\|\| enmVmState == VMSTATE_RESETTING
1108	\|\| enmVmState == VMSTATE_RESETTING_LS
1109	\|\| enmVmState == VMSTATE_SOFT_RESETTING
1110	\|\| enmVmState == VMSTATE_SOFT_RESETTING_LS
1111	\|\| enmVmState == VMSTATE_SUSPENDING
1112	\|\| enmVmState == VMSTATE_SUSPENDING_LS
1113	\|\| enmVmState == VMSTATE_SUSPENDING_EXT_LS)
1114	return true;
1115
1116	return false;
1117	}
1118
1119	/**
1120	* @copydoc FNVDASYNCTRANSFERCOMPLETE
1121	*/
1122	static void drvramdiskMediaExIoReqComplete(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq,
1123	int rcReq)
1124	{
1125	/*
1126	* For a read we need to sync the memory before continuing to process
1127	* the request further.
1128	*/
1129	if ( RT_SUCCESS(rcReq)
1130	&& pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ)
1131	rcReq = drvramdiskMediaExIoReqBufSync(pThis, pIoReq, false /* fToIoBuf */);
1132
1133	/*
1134	* When the request owner instructs us to handle recoverable errors like full disks
1135	* do it. Mark the request as suspended, notify the owner and put the request on the
1136	* redo list.
1137	*/
1138	if ( RT_FAILURE(rcReq)
1139	&& (pIoReq->fFlags & PDMIMEDIAEX_F_SUSPEND_ON_RECOVERABLE_ERR)
1140	&& drvramdiskMediaExIoReqIsRedoSetWarning(pThis, rcReq))
1141	{
1142	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_SUSPENDED, VDIOREQSTATE_ACTIVE);
1143	if (fXchg)
1144	{
1145	/* Put on redo list and adjust active request counter. */
1146	RTCritSectEnter(&pThis->CritSectIoReqRedo);
1147	RTListAppend(&pThis->LstIoReqRedo, &pIoReq->NdLstWait);
1148	RTCritSectLeave(&pThis->CritSectIoReqRedo);
1149	ASMAtomicDecU32(&pThis->cIoReqsActive);
1150	pThis->pDrvMediaExPort->pfnIoReqStateChanged(pThis->pDrvMediaExPort, pIoReq, &pIoReq->abAlloc[0],
1151	PDMMEDIAEXIOREQSTATE_SUSPENDED);
1152	}
1153	else
1154	{
1155	/* Request was canceled inbetween, so don't care and notify the owner about the completed request. */
1156	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
1157	drvramdiskMediaExIoReqCompleteWorker(pThis, pIoReq, rcReq, true /* fUpNotify */);
1158	}
1159	}
1160	else
1161	{
1162	/* Adjust the remaining amount to transfer. */
1163	size_t cbReqIo = RT_MIN(pIoReq->ReadWrite.cbReqLeft, pIoReq->ReadWrite.cbIoBuf);
1164	pIoReq->ReadWrite.offStart += cbReqIo;
1165	pIoReq->ReadWrite.cbReqLeft -= cbReqIo;
1166
1167	if ( RT_FAILURE(rcReq)
1168	\|\| !pIoReq->ReadWrite.cbReqLeft
1169	\|\| ( pIoReq->enmType != PDMMEDIAEXIOREQTYPE_READ
1170	&& pIoReq->enmType != PDMMEDIAEXIOREQTYPE_WRITE))
1171	drvramdiskMediaExIoReqCompleteWorker(pThis, pIoReq, rcReq, true /* fUpNotify */);
1172	else
1173	drvramdiskMediaExIoReqReadWriteProcess(pThis, pIoReq, true /* fUpNotify */);
1174	}
1175	}
1176
1177	/**
1178	* Worker for a flush request.
1179	*
1180	* @returns VBox status code.
1181	* @param pThis RAM disk container instance data.
1182	* @param pIoReq The flush request.
1183	*/
1184	static DECLCALLBACK(int) drvramdiskIoReqFlushWorker(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
1185	{
1186	/* Nothing to do for a ram disk. */
1187	drvramdiskMediaExIoReqComplete(pThis, pIoReq, VINF_SUCCESS);
1188	return VINF_SUCCESS;
1189	}
1190
1191	/**
1192	* Worker for a discard request.
1193	*
1194	* @returns VBox status code.
1195	* @param pThis RAM disk container instance data.
1196	* @param pIoReq The discard request.
1197	*/
1198	static DECLCALLBACK(int) drvramdiskIoReqDiscardWorker(PDRVRAMDISK pThis, PPDMMEDIAEXIOREQINT pIoReq)
1199	{
1200	int rc = drvramdiskDiscardRecords(pThis, pIoReq->Discard.paRanges, pIoReq->Discard.cRanges);
1201	drvramdiskMediaExIoReqComplete(pThis, pIoReq, rc);
1202	return VINF_SUCCESS;
1203	}
1204
1205	/**
1206	* @interface_method_impl{PDMIMEDIAEX,pfnQueryFeatures}
1207	*/
1208	static DECLCALLBACK(int) drvramdiskQueryFeatures(PPDMIMEDIAEX pInterface, uint32_t *pfFeatures)
1209	{
1210	RT_NOREF1(pInterface);
1211	*pfFeatures = PDMIMEDIAEX_FEATURE_F_ASYNC \| PDMIMEDIAEX_FEATURE_F_DISCARD;
1212	return VINF_SUCCESS;
1213	}
1214
1215
1216	/**
1217	* @interface_method_impl{PDMIMEDIAEX,pfnNotifySuspend}
1218	*/
1219	static DECLCALLBACK(void) drvramdiskNotifySuspend(PPDMIMEDIAEX pInterface)
1220	{
1221	RT_NOREF(pInterface);
1222	}
1223
1224
1225	/**
1226	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqAllocSizeSet}
1227	*/
1228	static DECLCALLBACK(int) drvramdiskIoReqAllocSizeSet(PPDMIMEDIAEX pInterface, size_t cbIoReqAlloc)
1229	{
1230	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1231
1232	if (RT_UNLIKELY(pThis->hIoReqCache != NIL_RTMEMCACHE))
1233	return VERR_INVALID_STATE;
1234
1235	return RTMemCacheCreate(&pThis->hIoReqCache, sizeof(PDMMEDIAEXIOREQINT) + cbIoReqAlloc, 0, UINT32_MAX,
1236	NULL, NULL, NULL, 0);
1237	}
1238
1239	/**
1240	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqAlloc}
1241	*/
1242	static DECLCALLBACK(int) drvramdiskIoReqAlloc(PPDMIMEDIAEX pInterface, PPDMMEDIAEXIOREQ phIoReq, void **ppvIoReqAlloc,
1243	PDMMEDIAEXIOREQID uIoReqId, uint32_t fFlags)
1244	{
1245	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1246
1247	AssertReturn(!(fFlags & ~PDMIMEDIAEX_F_VALID), VERR_INVALID_PARAMETER);
1248
1249	PPDMMEDIAEXIOREQINT pIoReq = (PPDMMEDIAEXIOREQINT)RTMemCacheAlloc(pThis->hIoReqCache);
1250
1251	if (RT_UNLIKELY(!pIoReq))
1252	return VERR_NO_MEMORY;
1253
1254	pIoReq->uIoReqId = uIoReqId;
1255	pIoReq->fFlags = fFlags;
1256	pIoReq->pDisk = pThis;
1257	pIoReq->enmState = VDIOREQSTATE_ALLOCATED;
1258	pIoReq->enmType = PDMMEDIAEXIOREQTYPE_INVALID;
1259
1260	int rc = drvramdiskMediaExIoReqInsert(pThis, pIoReq);
1261	if (RT_SUCCESS(rc))
1262	{
1263	*phIoReq = pIoReq;
1264	*ppvIoReqAlloc = &pIoReq->abAlloc[0];
1265	}
1266	else
1267	RTMemCacheFree(pThis->hIoReqCache, pIoReq);
1268
1269	return rc;
1270	}
1271
1272	/**
1273	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqFree}
1274	*/
1275	static DECLCALLBACK(int) drvramdiskIoReqFree(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq)
1276	{
1277	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1278	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1279
1280	if ( pIoReq->enmState != VDIOREQSTATE_COMPLETED
1281	&& pIoReq->enmState != VDIOREQSTATE_ALLOCATED)
1282	return VERR_PDM_MEDIAEX_IOREQ_INVALID_STATE;
1283
1284	/* Remove from allocated list. */
1285	int rc = drvramdiskMediaExIoReqRemove(pThis, pIoReq);
1286	if (RT_FAILURE(rc))
1287	return rc;
1288
1289	/* Free any associated I/O memory. */
1290	drvramdiskMediaExIoReqBufFree(pThis, pIoReq);
1291
1292	/* For discard request discard the range array. */
1293	if ( pIoReq->enmType == PDMMEDIAEXIOREQTYPE_DISCARD
1294	&& pIoReq->Discard.paRanges)
1295	{
1296	RTMemFree(pIoReq->Discard.paRanges);
1297	pIoReq->Discard.paRanges = NULL;
1298	}
1299
1300	pIoReq->enmState = VDIOREQSTATE_FREE;
1301	RTMemCacheFree(pThis->hIoReqCache, pIoReq);
1302	return VINF_SUCCESS;
1303	}
1304
1305	/**
1306	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqQueryResidual}
1307	*/
1308	static DECLCALLBACK(int) drvramdiskIoReqQueryResidual(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq, size_t *pcbResidual)
1309	{
1310	RT_NOREF2(pInterface, hIoReq);
1311
1312	*pcbResidual = 0;
1313	return VINF_SUCCESS;
1314	}
1315
1316	/**
1317	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqQueryXferSize}
1318	*/
1319	static DECLCALLBACK(int) drvramdiskIoReqQueryXferSize(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq, size_t *pcbXfer)
1320	{
1321	RT_NOREF1(pInterface);
1322	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1323
1324	if ( pIoReq->enmType == PDMMEDIAEXIOREQTYPE_READ
1325	\|\| pIoReq->enmType == PDMMEDIAEXIOREQTYPE_WRITE)
1326	*pcbXfer = pIoReq->ReadWrite.cbReq;
1327	else
1328	*pcbXfer = 0;
1329
1330	return VINF_SUCCESS;
1331	}
1332
1333	/**
1334	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqCancelAll}
1335	*/
1336	static DECLCALLBACK(int) drvramdiskIoReqCancelAll(PPDMIMEDIAEX pInterface)
1337	{
1338	RT_NOREF1(pInterface);
1339	return VINF_SUCCESS; /** @todo */
1340	}
1341
1342	/**
1343	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqCancel}
1344	*/
1345	static DECLCALLBACK(int) drvramdiskIoReqCancel(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQID uIoReqId)
1346	{
1347	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1348	unsigned idxBin = drvramdiskMediaExIoReqIdHash(uIoReqId);
1349
1350	int rc = RTSemFastMutexRequest(pThis->aIoReqAllocBins[idxBin].hMtxLstIoReqAlloc);
1351	if (RT_SUCCESS(rc))
1352	{
1353	/* Search for I/O request with ID. */
1354	PPDMMEDIAEXIOREQINT pIt;
1355	rc = VERR_PDM_MEDIAEX_IOREQID_NOT_FOUND;
1356
1357	RTListForEach(&pThis->aIoReqAllocBins[idxBin].LstIoReqAlloc, pIt, PDMMEDIAEXIOREQINT, NdAllocatedList)
1358	{
1359	if (pIt->uIoReqId == uIoReqId)
1360	{
1361	bool fXchg = true;
1362	VDIOREQSTATE enmStateOld = (VDIOREQSTATE)ASMAtomicReadU32((volatile uint32_t *)&pIt->enmState);
1363
1364	/*
1365	* We might have to try canceling the request multiple times if it transitioned from
1366	* ALLOCATED to ACTIVE or to SUSPENDED between reading the state and trying to change it.
1367	*/
1368	while ( ( enmStateOld == VDIOREQSTATE_ALLOCATED
1369	\|\| enmStateOld == VDIOREQSTATE_ACTIVE
1370	\|\| enmStateOld == VDIOREQSTATE_SUSPENDED)
1371	&& !fXchg)
1372	{
1373	fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIt->enmState, VDIOREQSTATE_CANCELED, enmStateOld);
1374	if (!fXchg)
1375	enmStateOld = (VDIOREQSTATE)ASMAtomicReadU32((volatile uint32_t *)&pIt->enmState);
1376	}
1377
1378	if (fXchg)
1379	{
1380	ASMAtomicDecU32(&pThis->cIoReqsActive);
1381	rc = VINF_SUCCESS;
1382	}
1383	break;
1384	}
1385	}
1386	RTSemFastMutexRelease(pThis->aIoReqAllocBins[idxBin].hMtxLstIoReqAlloc);
1387	}
1388
1389	return rc;
1390	}
1391
1392	/**
1393	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqRead}
1394	*/
1395	static DECLCALLBACK(int) drvramdiskIoReqRead(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq, uint64_t off, size_t cbRead)
1396	{
1397	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1398	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1399	VDIOREQSTATE enmState = (VDIOREQSTATE)ASMAtomicReadU32((volatile uint32_t *)&pIoReq->enmState);
1400
1401	if (RT_UNLIKELY(enmState == VDIOREQSTATE_CANCELED))
1402	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1403
1404	if (RT_UNLIKELY(enmState != VDIOREQSTATE_ALLOCATED))
1405	return VERR_PDM_MEDIAEX_IOREQ_INVALID_STATE;
1406
1407	pIoReq->enmType = PDMMEDIAEXIOREQTYPE_READ;
1408	pIoReq->tsSubmit = RTTimeMilliTS();
1409	pIoReq->ReadWrite.offStart = off;
1410	pIoReq->ReadWrite.cbReq = cbRead;
1411	pIoReq->ReadWrite.cbReqLeft = cbRead;
1412	/* Allocate a suitable I/O buffer for this request. */
1413	int rc = drvramdiskMediaExIoReqBufAlloc(pThis, pIoReq, cbRead);
1414	if (rc == VINF_SUCCESS)
1415	{
1416	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_ACTIVE, VDIOREQSTATE_ALLOCATED);
1417	if (RT_UNLIKELY(!fXchg))
1418	{
1419	/* Must have been canceled inbetween. */
1420	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
1421	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1422	}
1423	ASMAtomicIncU32(&pThis->cIoReqsActive);
1424
1425	rc = drvramdiskMediaExIoReqReadWriteProcess(pThis, pIoReq, false /* fUpNotify */);
1426	}
1427
1428	return rc;
1429	}
1430
1431	/**
1432	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqWrite}
1433	*/
1434	static DECLCALLBACK(int) drvramdiskIoReqWrite(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq, uint64_t off, size_t cbWrite)
1435	{
1436	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1437	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1438	VDIOREQSTATE enmState = (VDIOREQSTATE)ASMAtomicReadU32((volatile uint32_t *)&pIoReq->enmState);
1439
1440	if (RT_UNLIKELY(enmState == VDIOREQSTATE_CANCELED))
1441	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1442
1443	if (RT_UNLIKELY(enmState != VDIOREQSTATE_ALLOCATED))
1444	return VERR_PDM_MEDIAEX_IOREQ_INVALID_STATE;
1445
1446	pIoReq->enmType = PDMMEDIAEXIOREQTYPE_WRITE;
1447	pIoReq->tsSubmit = RTTimeMilliTS();
1448	pIoReq->ReadWrite.offStart = off;
1449	pIoReq->ReadWrite.cbReq = cbWrite;
1450	pIoReq->ReadWrite.cbReqLeft = cbWrite;
1451	/* Allocate a suitable I/O buffer for this request. */
1452	int rc = drvramdiskMediaExIoReqBufAlloc(pThis, pIoReq, cbWrite);
1453	if (rc == VINF_SUCCESS)
1454	{
1455	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_ACTIVE, VDIOREQSTATE_ALLOCATED);
1456	if (RT_UNLIKELY(!fXchg))
1457	{
1458	/* Must have been canceled inbetween. */
1459	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
1460	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1461	}
1462	ASMAtomicIncU32(&pThis->cIoReqsActive);
1463
1464	rc = drvramdiskMediaExIoReqReadWriteProcess(pThis, pIoReq, false /* fUpNotify */);
1465	}
1466
1467	return rc;
1468	}
1469
1470	/**
1471	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqFlush}
1472	*/
1473	static DECLCALLBACK(int) drvramdiskIoReqFlush(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq)
1474	{
1475	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1476	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1477	VDIOREQSTATE enmState = (VDIOREQSTATE)ASMAtomicReadU32((volatile uint32_t *)&pIoReq->enmState);
1478
1479	if (RT_UNLIKELY(enmState == VDIOREQSTATE_CANCELED))
1480	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1481
1482	if (RT_UNLIKELY(enmState != VDIOREQSTATE_ALLOCATED))
1483	return VERR_PDM_MEDIAEX_IOREQ_INVALID_STATE;
1484
1485	pIoReq->enmType = PDMMEDIAEXIOREQTYPE_FLUSH;
1486	pIoReq->tsSubmit = RTTimeMilliTS();
1487	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_ACTIVE, VDIOREQSTATE_ALLOCATED);
1488	if (RT_UNLIKELY(!fXchg))
1489	{
1490	/* Must have been canceled inbetween. */
1491	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
1492	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1493	}
1494
1495	ASMAtomicIncU32(&pThis->cIoReqsActive);
1496	return RTReqQueueCallEx(pThis->hReqQ, NULL, 0, RTREQFLAGS_NO_WAIT, (PFNRT)drvramdiskIoReqFlushWorker, 2, pThis, pIoReq);
1497	}
1498
1499	/**
1500	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqDiscard}
1501	*/
1502	static DECLCALLBACK(int) drvramdiskIoReqDiscard(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq, unsigned cRangesMax)
1503	{
1504	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1505	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1506	VDIOREQSTATE enmState = (VDIOREQSTATE)ASMAtomicReadU32((volatile uint32_t *)&pIoReq->enmState);
1507
1508	if (RT_UNLIKELY(enmState == VDIOREQSTATE_CANCELED))
1509	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1510
1511	if (RT_UNLIKELY(enmState != VDIOREQSTATE_ALLOCATED))
1512	return VERR_PDM_MEDIAEX_IOREQ_INVALID_STATE;
1513
1514	/* Copy the ranges over now, this can be optimized in the future. */
1515	pIoReq->Discard.paRanges = (PRTRANGE)RTMemAllocZ(cRangesMax * sizeof(RTRANGE));
1516	if (RT_UNLIKELY(!pIoReq->Discard.paRanges))
1517	return VERR_NO_MEMORY;
1518
1519	int rc = pThis->pDrvMediaExPort->pfnIoReqQueryDiscardRanges(pThis->pDrvMediaExPort, pIoReq, &pIoReq->abAlloc[0],
1520	0, cRangesMax, pIoReq->Discard.paRanges,
1521	&pIoReq->Discard.cRanges);
1522	if (RT_SUCCESS(rc))
1523	{
1524	pIoReq->enmType = PDMMEDIAEXIOREQTYPE_DISCARD;
1525	pIoReq->tsSubmit = RTTimeMilliTS();
1526
1527	bool fXchg = ASMAtomicCmpXchgU32((volatile uint32_t *)&pIoReq->enmState, VDIOREQSTATE_ACTIVE, VDIOREQSTATE_ALLOCATED);
1528	if (RT_UNLIKELY(!fXchg))
1529	{
1530	/* Must have been canceled inbetween. */
1531	Assert(pIoReq->enmState == VDIOREQSTATE_CANCELED);
1532	return VERR_PDM_MEDIAEX_IOREQ_CANCELED;
1533	}
1534
1535	ASMAtomicIncU32(&pThis->cIoReqsActive);
1536
1537	rc = RTReqQueueCallEx(pThis->hReqQ, NULL, 0, RTREQFLAGS_NO_WAIT, (PFNRT)drvramdiskIoReqDiscardWorker, 2, pThis, pIoReq);
1538	if (rc == VINF_SUCCESS)
1539	rc = VINF_PDM_MEDIAEX_IOREQ_IN_PROGRESS;
1540	}
1541
1542	return rc;
1543	}
1544
1545	/**
1546	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqGetActiveCount}
1547	*/
1548	static DECLCALLBACK(uint32_t) drvramdiskIoReqGetActiveCount(PPDMIMEDIAEX pInterface)
1549	{
1550	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1551	return ASMAtomicReadU32(&pThis->cIoReqsActive);
1552	}
1553
1554	/**
1555	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqGetSuspendedCount}
1556	*/
1557	static DECLCALLBACK(uint32_t) drvramdiskIoReqGetSuspendedCount(PPDMIMEDIAEX pInterface)
1558	{
1559	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1560
1561	AssertReturn(!drvramdiskMediaExIoReqIsVmRunning(pThis), 0);
1562
1563	uint32_t cIoReqSuspended = 0;
1564	PPDMMEDIAEXIOREQINT pIoReq;
1565	RTCritSectEnter(&pThis->CritSectIoReqRedo);
1566	RTListForEach(&pThis->LstIoReqRedo, pIoReq, PDMMEDIAEXIOREQINT, NdLstWait)
1567	{
1568	cIoReqSuspended++;
1569	}
1570	RTCritSectLeave(&pThis->CritSectIoReqRedo);
1571
1572	return cIoReqSuspended;
1573	}
1574
1575	/**
1576	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqQuerySuspendedStart}
1577	*/
1578	static DECLCALLBACK(int) drvramdiskIoReqQuerySuspendedStart(PPDMIMEDIAEX pInterface, PPDMMEDIAEXIOREQ phIoReq,
1579	void **ppvIoReqAlloc)
1580	{
1581	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1582
1583	AssertReturn(!drvramdiskMediaExIoReqIsVmRunning(pThis), VERR_INVALID_STATE);
1584	AssertReturn(!RTListIsEmpty(&pThis->LstIoReqRedo), VERR_NOT_FOUND);
1585
1586	RTCritSectEnter(&pThis->CritSectIoReqRedo);
1587	PPDMMEDIAEXIOREQINT pIoReq = RTListGetFirst(&pThis->LstIoReqRedo, PDMMEDIAEXIOREQINT, NdLstWait);
1588	*phIoReq = pIoReq;
1589	*ppvIoReqAlloc = &pIoReq->abAlloc[0];
1590	RTCritSectLeave(&pThis->CritSectIoReqRedo);
1591
1592	return VINF_SUCCESS;
1593	}
1594
1595	/**
1596	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqQuerySuspendedNext}
1597	*/
1598	static DECLCALLBACK(int) drvramdiskIoReqQuerySuspendedNext(PPDMIMEDIAEX pInterface, PDMMEDIAEXIOREQ hIoReq,
1599	PPDMMEDIAEXIOREQ phIoReqNext, void **ppvIoReqAllocNext)
1600	{
1601	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1602	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1603
1604	AssertReturn(!drvramdiskMediaExIoReqIsVmRunning(pThis), VERR_INVALID_STATE);
1605	AssertPtrReturn(pIoReq, VERR_INVALID_HANDLE);
1606	AssertReturn(!RTListNodeIsLast(&pThis->LstIoReqRedo, &pIoReq->NdLstWait), VERR_NOT_FOUND);
1607
1608	RTCritSectEnter(&pThis->CritSectIoReqRedo);
1609	PPDMMEDIAEXIOREQINT pIoReqNext = RTListNodeGetNext(&pIoReq->NdLstWait, PDMMEDIAEXIOREQINT, NdLstWait);
1610	*phIoReqNext = pIoReqNext;
1611	*ppvIoReqAllocNext = &pIoReqNext->abAlloc[0];
1612	RTCritSectLeave(&pThis->CritSectIoReqRedo);
1613
1614	return VINF_SUCCESS;
1615	}
1616
1617	/**
1618	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqSuspendedSave}
1619	*/
1620	static DECLCALLBACK(int) drvramdiskIoReqSuspendedSave(PPDMIMEDIAEX pInterface, PSSMHANDLE pSSM, PDMMEDIAEXIOREQ hIoReq)
1621	{
1622	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1623	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1624
1625	RT_NOREF1(pSSM);
1626
1627	AssertReturn(!drvramdiskMediaExIoReqIsVmRunning(pThis), VERR_INVALID_STATE);
1628	AssertPtrReturn(pIoReq, VERR_INVALID_HANDLE);
1629	AssertReturn(pIoReq->enmState == VDIOREQSTATE_SUSPENDED, VERR_INVALID_STATE);
1630
1631	return VERR_NOT_IMPLEMENTED;
1632	}
1633
1634	/**
1635	* @interface_method_impl{PDMIMEDIAEX,pfnIoReqSuspendedLoad}
1636	*/
1637	static DECLCALLBACK(int) drvramdiskIoReqSuspendedLoad(PPDMIMEDIAEX pInterface, PSSMHANDLE pSSM, PDMMEDIAEXIOREQ hIoReq)
1638	{
1639	PDRVRAMDISK pThis = RT_FROM_MEMBER(pInterface, DRVRAMDISK, IMediaEx);
1640	PPDMMEDIAEXIOREQINT pIoReq = hIoReq;
1641
1642	RT_NOREF1(pSSM);
1643
1644	AssertReturn(!drvramdiskMediaExIoReqIsVmRunning(pThis), VERR_INVALID_STATE);
1645	AssertPtrReturn(pIoReq, VERR_INVALID_HANDLE);
1646	AssertReturn(pIoReq->enmState == VDIOREQSTATE_ALLOCATED, VERR_INVALID_STATE);
1647
1648	return VERR_NOT_IMPLEMENTED;
1649	}
1650
1651	static DECLCALLBACK(int) drvramdiskIoReqWorker(RTTHREAD hThrdSelf, void *pvUser)
1652	{
1653	int rc = VINF_SUCCESS;
1654	PDRVRAMDISK pThis = (PDRVRAMDISK)pvUser;
1655
1656	RT_NOREF1(hThrdSelf);
1657
1658	do
1659	{
1660	rc = RTReqQueueProcess(pThis->hReqQ, RT_INDEFINITE_WAIT);
1661	} while (RT_SUCCESS(rc));
1662
1663	return VINF_SUCCESS;
1664	}
1665
1666	/* -=-=-=-=- IBase -=-=-=-=- */
1667
1668	/**
1669	* @interface_method_impl{PDMIBASE,pfnQueryInterface}
1670	*/
1671	static DECLCALLBACK(void ) drvramdiskQueryInterface(PPDMIBASE pInterface, const char pszIID)
1672	{
1673	PPDMDRVINS pDrvIns = PDMIBASE_2_PDMDRV(pInterface);
1674	PDRVRAMDISK pThis = PDMINS_2_DATA(pDrvIns, PDRVRAMDISK);
1675
1676	PDMIBASE_RETURN_INTERFACE(pszIID, PDMIBASE, &pDrvIns->IBase);
1677	PDMIBASE_RETURN_INTERFACE(pszIID, PDMIMEDIA, &pThis->IMedia);
1678	PDMIBASE_RETURN_INTERFACE(pszIID, PDMIMEDIAEX, &pThis->IMediaEx);
1679
1680	return NULL;
1681	}
1682
1683
1684	/* -=-=-=-=- driver interface -=-=-=-=- */
1685
1686	static DECLCALLBACK(int) drvramdiskTreeDestroy(PAVLRFOFFNODECORE pNode, void *pvUser)
1687	{
1688	PDRVDISKSEGMENT pSeg = (PDRVDISKSEGMENT)pNode;
1689
1690	RT_NOREF1(pvUser);
1691
1692	RTMemFree(pSeg->pbSeg);
1693	RTMemFree(pSeg);
1694	return VINF_SUCCESS;
1695	}
1696
1697	/**
1698	* @copydoc FNPDMDRVDESTRUCT
1699	*/
1700	static DECLCALLBACK(void) drvramdiskDestruct(PPDMDRVINS pDrvIns)
1701	{
1702	PDRVRAMDISK pThis = PDMINS_2_DATA(pDrvIns, PDRVRAMDISK);
1703
1704	PRTREQ pReq = NULL;
1705	int rc = RTReqQueueCallEx(pThis->hReqQ, &pReq, RT_INDEFINITE_WAIT, RTREQFLAGS_IPRT_STATUS, (PFNRT)drvramdiskDestructQueue, 0);
1706	AssertRC(rc);
1707	RTReqRelease(pReq);
1708
1709	if (pThis->pTreeSegments)
1710	{
1711	RTAvlrFileOffsetDestroy(pThis->pTreeSegments, drvramdiskTreeDestroy, NULL);
1712	RTMemFree(pThis->pTreeSegments);
1713	}
1714	if (pThis->hIoBufMgr)
1715	IOBUFMgrDestroy(pThis->hIoBufMgr);
1716
1717	for (unsigned i = 0; i < RT_ELEMENTS(pThis->aIoReqAllocBins); i++)
1718	if (pThis->aIoReqAllocBins[i].hMtxLstIoReqAlloc != NIL_RTSEMFASTMUTEX)
1719	RTSemFastMutexDestroy(pThis->aIoReqAllocBins[i].hMtxLstIoReqAlloc);
1720
1721	if (RTCritSectIsInitialized(&pThis->CritSectIoReqsIoBufWait))
1722	RTCritSectDelete(&pThis->CritSectIoReqsIoBufWait);
1723
1724	if (RTCritSectIsInitialized(&pThis->CritSectIoReqRedo))
1725	RTCritSectDelete(&pThis->CritSectIoReqRedo);
1726
1727	if (pThis->hIoReqCache != NIL_RTMEMCACHE)
1728	RTMemCacheDestroy(pThis->hIoReqCache);
1729
1730	RTReqQueueDestroy(pThis->hReqQ);
1731	}
1732
1733	/**
1734	* Construct a disk integrity driver instance.
1735	*
1736	* @copydoc FNPDMDRVCONSTRUCT
1737	*/
1738	static DECLCALLBACK(int) drvramdiskConstruct(PPDMDRVINS pDrvIns, PCFGMNODE pCfg, uint32_t fFlags)
1739	{
1740	RT_NOREF1(fFlags);
1741	PDMDRV_CHECK_VERSIONS_RETURN(pDrvIns);
1742	PDRVRAMDISK pThis = PDMINS_2_DATA(pDrvIns, PDRVRAMDISK);
1743	PCPDMDRVHLPR3 pHlp = pDrvIns->pHlpR3;
1744
1745	LogFlow(("drvdiskintConstruct: iInstance=%d\n", pDrvIns->iInstance));
1746
1747	/*
1748	* Initialize most of the data members.
1749	*/
1750	pThis->pDrvIns = pDrvIns;
1751
1752	/* IBase. */
1753	pDrvIns->IBase.pfnQueryInterface = drvramdiskQueryInterface;
1754
1755	/* IMedia */
1756	pThis->IMedia.pfnRead = drvramdiskRead;
1757	pThis->IMedia.pfnWrite = drvramdiskWrite;
1758	pThis->IMedia.pfnFlush = drvramdiskFlush;
1759	pThis->IMedia.pfnGetSize = drvramdiskGetSize;
1760	pThis->IMedia.pfnBiosIsVisible = drvramdiskBiosIsVisible;
1761	pThis->IMedia.pfnGetType = drvramdiskGetType;
1762	pThis->IMedia.pfnIsReadOnly = drvramdiskIsReadOnly;
1763	pThis->IMedia.pfnBiosGetPCHSGeometry = drvramdiskBiosGetPCHSGeometry;
1764	pThis->IMedia.pfnBiosSetPCHSGeometry = drvramdiskBiosSetPCHSGeometry;
1765	pThis->IMedia.pfnBiosGetLCHSGeometry = drvramdiskBiosGetLCHSGeometry;
1766	pThis->IMedia.pfnBiosSetLCHSGeometry = drvramdiskBiosSetLCHSGeometry;
1767	pThis->IMedia.pfnGetUuid = drvramdiskGetUuid;
1768	pThis->IMedia.pfnGetSectorSize = drvramdiskGetSectorSize;
1769	pThis->IMedia.pfnReadPcBios = drvramdiskReadPcBios;
1770	pThis->IMedia.pfnDiscard = drvramdiskDiscard;
1771	pThis->IMedia.pfnIsNonRotational = drvramdiskIsNonRotational;
1772	pThis->IMedia.pfnSendCmd = NULL;
1773	pThis->IMedia.pfnGetRegionCount = drvramdiskGetRegionCount;
1774	pThis->IMedia.pfnQueryRegionProperties = drvramdiskQueryRegionProperties;
1775	pThis->IMedia.pfnQueryRegionPropertiesForLba = drvramdiskQueryRegionPropertiesForLba;
1776
1777	/* IMediaEx */
1778	pThis->IMediaEx.pfnQueryFeatures = drvramdiskQueryFeatures;
1779	pThis->IMediaEx.pfnNotifySuspend = drvramdiskNotifySuspend;
1780	pThis->IMediaEx.pfnIoReqAllocSizeSet = drvramdiskIoReqAllocSizeSet;
1781	pThis->IMediaEx.pfnIoReqAlloc = drvramdiskIoReqAlloc;
1782	pThis->IMediaEx.pfnIoReqFree = drvramdiskIoReqFree;
1783	pThis->IMediaEx.pfnIoReqQueryResidual = drvramdiskIoReqQueryResidual;
1784	pThis->IMediaEx.pfnIoReqQueryXferSize = drvramdiskIoReqQueryXferSize;
1785	pThis->IMediaEx.pfnIoReqCancelAll = drvramdiskIoReqCancelAll;
1786	pThis->IMediaEx.pfnIoReqCancel = drvramdiskIoReqCancel;
1787	pThis->IMediaEx.pfnIoReqRead = drvramdiskIoReqRead;
1788	pThis->IMediaEx.pfnIoReqWrite = drvramdiskIoReqWrite;
1789	pThis->IMediaEx.pfnIoReqFlush = drvramdiskIoReqFlush;
1790	pThis->IMediaEx.pfnIoReqDiscard = drvramdiskIoReqDiscard;
1791	pThis->IMediaEx.pfnIoReqGetActiveCount = drvramdiskIoReqGetActiveCount;
1792	pThis->IMediaEx.pfnIoReqGetSuspendedCount = drvramdiskIoReqGetSuspendedCount;
1793	pThis->IMediaEx.pfnIoReqQuerySuspendedStart = drvramdiskIoReqQuerySuspendedStart;
1794	pThis->IMediaEx.pfnIoReqQuerySuspendedNext = drvramdiskIoReqQuerySuspendedNext;
1795	pThis->IMediaEx.pfnIoReqSuspendedSave = drvramdiskIoReqSuspendedSave;
1796	pThis->IMediaEx.pfnIoReqSuspendedLoad = drvramdiskIoReqSuspendedLoad;
1797
1798	/*
1799	* Validate configuration.
1800	*/
1801	PDMDRV_VALIDATE_CONFIG_RETURN(pDrvIns, "Size"
1802	"\|PreAlloc"
1803	"\|IoBufMax"
1804	"\|SectorSize"
1805	"\|NonRotational",
1806	"");
1807
1808	int rc = pHlp->pfnCFGMQueryU64(pCfg, "Size", &pThis->cbDisk);
1809	if (RT_FAILURE(rc))
1810	return PDMDRV_SET_ERROR(pDrvIns, rc,
1811	N_("RamDisk: Error querying the media size"));
1812	rc = pHlp->pfnCFGMQueryBoolDef(pCfg, "PreAlloc", &pThis->fPreallocRamDisk, false);
1813	if (RT_FAILURE(rc))
1814	return PDMDRV_SET_ERROR(pDrvIns, rc,
1815	N_("RamDisk: Error querying \"PreAlloc\""));
1816	rc = pHlp->pfnCFGMQueryBoolDef(pCfg, "NonRotational", &pThis->fNonRotational, true);
1817	if (RT_FAILURE(rc))
1818	return PDMDRV_SET_ERROR(pDrvIns, rc,
1819	N_("RamDisk: Error querying \"NonRotational\""));
1820
1821	uint32_t cbIoBufMax;
1822	rc = pHlp->pfnCFGMQueryU32Def(pCfg, "IoBufMax", &cbIoBufMax, 5 * _1M);
1823	if (RT_FAILURE(rc))
1824	return PDMDRV_SET_ERROR(pDrvIns, rc, N_("Failed to query \"IoBufMax\" from the config"));
1825	rc = pHlp->pfnCFGMQueryU32Def(pCfg, "SectorSize", &pThis->cbSector, 512);
1826	if (RT_FAILURE(rc))
1827	return PDMDRV_SET_ERROR(pDrvIns, rc, N_("Failed to query \"SectorSize\" from the config"));
1828
1829	/* Query the media port interface above us. */
1830	pThis->pDrvMediaPort = PDMIBASE_QUERY_INTERFACE(pDrvIns->pUpBase, PDMIMEDIAPORT);
1831	if (!pThis->pDrvMediaPort)
1832	return PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_MISSING_INTERFACE_BELOW,
1833	N_("No media port interface above"));
1834
1835	/* Try to attach extended media port interface above.*/
1836	pThis->pDrvMediaExPort = PDMIBASE_QUERY_INTERFACE(pDrvIns->pUpBase, PDMIMEDIAEXPORT);
1837	if (pThis->pDrvMediaExPort)
1838	{
1839	for (unsigned i = 0; i < RT_ELEMENTS(pThis->aIoReqAllocBins); i++)
1840	{
1841	rc = RTSemFastMutexCreate(&pThis->aIoReqAllocBins[i].hMtxLstIoReqAlloc);
1842	if (RT_FAILURE(rc))
1843	break;
1844	RTListInit(&pThis->aIoReqAllocBins[i].LstIoReqAlloc);
1845	}
1846
1847	if (RT_SUCCESS(rc))
1848	rc = RTCritSectInit(&pThis->CritSectIoReqsIoBufWait);
1849
1850	if (RT_SUCCESS(rc))
1851	rc = RTCritSectInit(&pThis->CritSectIoReqRedo);
1852
1853	if (RT_FAILURE(rc))
1854	return PDMDRV_SET_ERROR(pDrvIns, rc, N_("Creating Mutex failed"));
1855
1856	RTListInit(&pThis->LstIoReqIoBufWait);
1857	RTListInit(&pThis->LstIoReqRedo);
1858	}
1859
1860	/* Create the AVL tree. */
1861	pThis->pTreeSegments = (PAVLRFOFFTREE)RTMemAllocZ(sizeof(AVLRFOFFTREE));
1862	if (!pThis->pTreeSegments)
1863	rc = VERR_NO_MEMORY;
1864
1865	if (pThis->pDrvMediaExPort)
1866	{
1867	rc = RTReqQueueCreate(&pThis->hReqQ);
1868	if (RT_SUCCESS(rc))
1869	{
1870	/* Spin up the worker thread. */
1871	rc = RTThreadCreate(&pThis->hThrdWrk, drvramdiskIoReqWorker, pThis, 0,
1872	RTTHREADTYPE_IO, 0, "RAMDSK");
1873	}
1874	}
1875
1876	if (pThis->pDrvMediaExPort)
1877	rc = IOBUFMgrCreate(&pThis->hIoBufMgr, cbIoBufMax, IOBUFMGR_F_DEFAULT);
1878
1879	/* Read in all data before the start if requested. */
1880	if ( RT_SUCCESS(rc)
1881	&& pThis->fPreallocRamDisk)
1882	{
1883	LogRel(("RamDisk: Preallocating RAM disk...\n"));
1884	return VERR_NOT_IMPLEMENTED;
1885	}
1886
1887	return rc;
1888	}
1889
1890
1891	/**
1892	* Block driver registration record.
1893	*/
1894	const PDMDRVREG g_DrvRamDisk =
1895	{
1896	/* u32Version */
1897	PDM_DRVREG_VERSION,
1898	/* szName */
1899	"RamDisk",
1900	/* szRCMod */
1901	"",
1902	/* szR0Mod */
1903	"",
1904	/* pszDescription */
1905	"RAM disk driver.",
1906	/* fFlags */
1907	PDM_DRVREG_FLAGS_HOST_BITS_DEFAULT,
1908	/* fClass. */
1909	PDM_DRVREG_CLASS_BLOCK,
1910	/* cMaxInstances */
1911	~0U,
1912	/* cbInstance */
1913	sizeof(DRVRAMDISK),
1914	/* pfnConstruct */
1915	drvramdiskConstruct,
1916	/* pfnDestruct */
1917	drvramdiskDestruct,
1918	/* pfnRelocate */
1919	NULL,
1920	/* pfnIOCtl */
1921	NULL,
1922	/* pfnPowerOn */
1923	NULL,
1924	/* pfnReset */
1925	NULL,
1926	/* pfnSuspend */
1927	NULL,
1928	/* pfnResume */
1929	NULL,
1930	/* pfnAttach */
1931	NULL,
1932	/* pfnDetach */
1933	NULL,
1934	/* pfnPowerOff */
1935	NULL,
1936	/* pfnSoftReset */
1937	NULL,
1938	/* u32EndVersion */
1939	PDM_DRVREG_VERSION
1940	};
1941

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Devices/Storage/DrvRamDisk.cpp@ 106297

Download in other formats: