PDMAsyncCompletionFileCache.cpp@ 23673

Last change on this file since 23673 was 23673, checked in by vboxsync, 15 years ago
typo
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 53.9 KB

Line
1	/* $Id: PDMAsyncCompletionFileCache.cpp 23673 2009-10-10 21:26:43Z vboxsync $ */
2	/** @file
3	* PDM Async I/O - Transport data asynchronous in R3 using EMT.
4	* File data cache.
5	*/
6
7	/*
8	* Copyright (C) 2006-2008 Sun Microsystems, Inc.
9	*
10	* This file is part of VirtualBox Open Source Edition (OSE), as
11	* available from http://www.virtualbox.org. This file is free software;
12	* you can redistribute it and/or modify it under the terms of the GNU
13	* General Public License (GPL) as published by the Free Software
14	* Foundation, in version 2 as it comes in the "COPYING" file of the
15	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
16	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
17	*
18	* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
19	* Clara, CA 95054 USA or visit http://www.sun.com if you need
20	* additional information or have any questions.
21	*/
22
23	/** @page pg_pdm_async_completion_cache PDM Async Completion Cache - The file I/O cache
24	* This component implements an I/O cache for file endpoints based on the ARC algorithm.
25	* http://en.wikipedia.org/wiki/Adaptive_Replacement_Cache
26	*
27	* The algorithm uses four LRU (Least frequently used) lists to store data in the cache.
28	* Two of them contain data where one stores entries which were accessed recently and one
29	* which is used for frequently accessed data.
30	* The other two lists are called ghost lists and store information about the accessed range
31	* but do not contain data. They are used to track data access. If these entries are accessed
32	* they will push the data to a higher position in the cache preventing it from getting removed
33	* quickly again.
34	*
35	* The algorithm needs to be modified to meet our requirements. Like the implementation
36	* for the ZFS filesystem we need to handle pages with a variable size. It would
37	* be possible to use a fixed size but would increase the computational
38	* and memory overhead.
39	* Because we do I/O asynchronously we also need to mark entries which are currently accessed
40	* as non evictable to prevent removal of the entry while the data is being accessed.
41	*/
42
43	/*******************************************************************************
44	* Header Files *
45	*******************************************************************************/
46	#define LOG_GROUP LOG_GROUP_PDM_ASYNC_COMPLETION
47	#include <iprt/types.h>
48	#include <iprt/mem.h>
49	#include <VBox/log.h>
50	#include <VBox/stam.h>
51
52	#include "PDMAsyncCompletionFileInternal.h"
53
54	#ifdef VBOX_STRICT
55	# define PDMACFILECACHE_IS_CRITSECT_OWNER(Cache) \
56	do \
57	{ \
58	AssertMsg(RTCritSectIsOwner(&pCache->CritSect), \
59	("Thread does not own critical section\n"));\
60	} while(0);
61	#else
62	# define PDMACFILECACHE_IS_CRITSECT_OWNER(Cache) do { } while(0);
63	#endif
64
65	/*******************************************************************************
66	* Internal Functions *
67	*******************************************************************************/
68	static void pdmacFileCacheTaskCompleted(PPDMACTASKFILE pTask, void *pvUser);
69
70	DECLINLINE(void) pdmacFileEpCacheEntryRelease(PPDMACFILECACHEENTRY pEntry)
71	{
72	AssertMsg(pEntry->cRefs > 0, ("Trying to release a not referenced entry\n"));
73	ASMAtomicDecU32(&pEntry->cRefs);
74	}
75
76	DECLINLINE(void) pdmacFileEpCacheEntryRef(PPDMACFILECACHEENTRY pEntry)
77	{
78	ASMAtomicIncU32(&pEntry->cRefs);
79	}
80
81	/**
82	* Checks consistency of a LRU list.
83	*
84	* @returns nothing
85	* @param pList The LRU list to check.
86	* @param pNotInList Element which is not allowed to occur in the list.
87	*/
88	static void pdmacFileCacheCheckList(PPDMACFILELRULIST pList, PPDMACFILECACHEENTRY pNotInList)
89	{
90	#ifdef PDMACFILECACHE_WITH_LRULIST_CHECKS
91	PPDMACFILECACHEENTRY pCurr = pList->pHead;
92
93	/* Check that there are no double entries and no cycles in the list. */
94	while (pCurr)
95	{
96	PPDMACFILECACHEENTRY pNext = pCurr->pNext;
97
98	while (pNext)
99	{
100	AssertMsg(pCurr != pNext,
101	("Entry %#p is at least two times in list %#p or there is a cycle in the list\n",
102	pCurr, pList));
103	pNext = pNext->pNext;
104	}
105
106	AssertMsg(pCurr != pNotInList, ("Not allowed entry %#p is in list\n", pCurr));
107
108	if (!pCurr->pNext)
109	AssertMsg(pCurr == pList->pTail, ("End of list reached but last element is not list tail\n"));
110
111	pCurr = pCurr->pNext;
112	}
113	#endif
114	}
115
116	/**
117	* Unlinks a cache entry from the LRU list it is assigned to.
118	*
119	* @returns nothing.
120	* @param pEntry The entry to unlink.
121	*/
122	static void pdmacFileCacheEntryRemoveFromList(PPDMACFILECACHEENTRY pEntry)
123	{
124	PPDMACFILELRULIST pList = pEntry->pList;
125	PPDMACFILECACHEENTRY pPrev, pNext;
126
127	LogFlowFunc((": Deleting entry %#p from list %#p\n", pEntry, pList));
128
129	AssertPtr(pList);
130	pdmacFileCacheCheckList(pList, NULL);
131
132	pPrev = pEntry->pPrev;
133	pNext = pEntry->pNext;
134
135	AssertMsg(pEntry != pPrev, ("Entry links to itself as previous element\n"));
136	AssertMsg(pEntry != pNext, ("Entry links to itself as next element\n"));
137
138	if (pPrev)
139	pPrev->pNext = pNext;
140	else
141	{
142	pList->pHead = pNext;
143
144	if (pNext)
145	pNext->pPrev = NULL;
146	}
147
148	if (pNext)
149	pNext->pPrev = pPrev;
150	else
151	{
152	pList->pTail = pPrev;
153
154	if (pPrev)
155	pPrev->pNext = NULL;
156	}
157
158	pEntry->pList = NULL;
159	pEntry->pPrev = NULL;
160	pEntry->pNext = NULL;
161	pList->cbCached -= pEntry->cbData;
162	pdmacFileCacheCheckList(pList, pEntry);
163	}
164
165	/**
166	* Adds a cache entry to the given LRU list unlinking it from the currently
167	* assigned list if needed.
168	*
169	* @returns nothing.
170	* @param pList List to the add entry to.
171	* @param pEntry Entry to add.
172	*/
173	static void pdmacFileCacheEntryAddToList(PPDMACFILELRULIST pList, PPDMACFILECACHEENTRY pEntry)
174	{
175	LogFlowFunc((": Adding entry %#p to list %#p\n", pEntry, pList));
176	pdmacFileCacheCheckList(pList, NULL);
177
178	/* Remove from old list if needed */
179	if (pEntry->pList)
180	pdmacFileCacheEntryRemoveFromList(pEntry);
181
182	pEntry->pNext = pList->pHead;
183	if (pList->pHead)
184	pList->pHead->pPrev = pEntry;
185	else
186	{
187	Assert(!pList->pTail);
188	pList->pTail = pEntry;
189	}
190
191	pEntry->pPrev = NULL;
192	pList->pHead = pEntry;
193	pList->cbCached += pEntry->cbData;
194	pEntry->pList = pList;
195	pdmacFileCacheCheckList(pList, NULL);
196	}
197
198	/**
199	* Destroys a LRU list freeing all entries.
200	*
201	* @returns nothing
202	* @param pList Pointer to the LRU list to destroy.
203	*
204	* @note The caller must own the critical section of the cache.
205	*/
206	static void pdmacFileCacheDestroyList(PPDMACFILELRULIST pList)
207	{
208	while (pList->pHead)
209	{
210	PPDMACFILECACHEENTRY pEntry = pList->pHead;
211
212	pList->pHead = pEntry->pNext;
213
214	AssertMsg(!(pEntry->fFlags & (PDMACFILECACHE_ENTRY_IO_IN_PROGRESS \| PDMACFILECACHE_ENTRY_IS_DIRTY)),
215	("Entry is dirty and/or still in progress fFlags=%#x\n", pEntry->fFlags));
216
217	RTMemPageFree(pEntry->pbData);
218	RTMemFree(pEntry);
219	}
220	}
221
222	/**
223	* Tries to remove the given amount of bytes from a given list in the cache
224	* moving the entries to one of the given ghosts lists
225	*
226	* @returns Amount of data which could be freed.
227	* @param pCache Pointer to the global cache data.
228	* @param cbData The amount of the data to free.
229	* @param pListSrc The source list to evict data from.
230	* @param pGhostListSrc The ghost list removed entries should be moved to
231	* NULL if the entry should be freed.
232	*
233	* @notes This function may return fewer bytes than requested because entries
234	* may be marked as non evictable if they are used for I/O at the moment.
235	*/
236	static size_t pdmacFileCacheEvictPagesFrom(PPDMACFILECACHEGLOBAL pCache, size_t cbData,
237	PPDMACFILELRULIST pListSrc, PPDMACFILELRULIST pGhostListDst)
238	{
239	size_t cbEvicted = 0;
240
241	PDMACFILECACHE_IS_CRITSECT_OWNER(pCache);
242
243	AssertMsg(cbData > 0, ("Evicting 0 bytes not possible\n"));
244	AssertMsg( !pGhostListDst
245	\|\| (pGhostListDst == &pCache->LruRecentlyGhost)
246	\|\| (pGhostListDst == &pCache->LruFrequentlyGhost),
247	("Destination list must be NULL or one of the ghost lists\n"));
248
249	/* Start deleting from the tail. */
250	PPDMACFILECACHEENTRY pEntry = pListSrc->pTail;
251
252	while ((cbEvicted < cbData) && pEntry)
253	{
254	PPDMACFILECACHEENTRY pCurr = pEntry;
255
256	pEntry = pEntry->pPrev;
257
258	/* We can't evict pages which are currently in progress */
259	if (!(pCurr->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS)
260	&& (ASMAtomicReadU32(&pCurr->cRefs) == 0))
261	{
262	/* Ok eviction candidate. Grab the endpoint semaphore and check again
263	* because somebody else might have raced us. */
264	PPDMACFILEENDPOINTCACHE pEndpointCache = &pCurr->pEndpoint->DataCache;
265	RTSemRWRequestWrite(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
266
267	if (!(pCurr->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS)
268	&& (ASMAtomicReadU32(&pCurr->cRefs) == 0))
269	{
270	LogFlow(("Evicting entry %#p (%u bytes)\n", pCurr, pCurr->cbData));
271	if (pCurr->pbData)
272	{
273	RTMemPageFree(pCurr->pbData);
274	pCurr->pbData = NULL;
275	}
276
277	cbEvicted += pCurr->cbData;
278
279	if (pGhostListDst)
280	{
281	pdmacFileCacheEntryAddToList(pGhostListDst, pCurr);
282	}
283	else
284	{
285	/* Delete the entry from the AVL tree it is assigned to. */
286	STAM_PROFILE_ADV_START(&pCache->StatTreeRemove, Cache);
287	RTAvlrFileOffsetRemove(pCurr->pEndpoint->DataCache.pTree, pCurr->Core.Key);
288	STAM_PROFILE_ADV_STOP(&pCache->StatTreeRemove, Cache);
289
290	pdmacFileCacheEntryRemoveFromList(pCurr);
291	pCache->cbCached -= pCurr->cbData;
292	RTMemFree(pCurr);
293	}
294	}
295	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
296	}
297	else
298	LogFlow(("Entry %#p (%u bytes) is still in progress and can't be evicted\n", pCurr, pCurr->cbData));
299	}
300
301	return cbEvicted;
302	}
303
304	static size_t pdmacFileCacheReplace(PPDMACFILECACHEGLOBAL pCache, size_t cbData, PPDMACFILELRULIST pEntryList)
305	{
306	PDMACFILECACHE_IS_CRITSECT_OWNER(pCache);
307
308	if ( (pCache->LruRecentlyUsed.cbCached)
309	&& ( (pCache->LruRecentlyUsed.cbCached > pCache->uAdaptVal)
310	\|\| ( (pEntryList == &pCache->LruFrequentlyGhost)
311	&& (pCache->LruRecentlyUsed.cbCached == pCache->uAdaptVal))))
312	{
313	/* We need to remove entry size pages from T1 and move the entries to B1 */
314	return pdmacFileCacheEvictPagesFrom(pCache, cbData,
315	&pCache->LruRecentlyUsed,
316	&pCache->LruRecentlyGhost);
317	}
318	else
319	{
320	/* We need to remove entry size pages from T2 and move the entries to B2 */
321	return pdmacFileCacheEvictPagesFrom(pCache, cbData,
322	&pCache->LruFrequentlyUsed,
323	&pCache->LruFrequentlyGhost);
324	}
325	}
326
327	/**
328	* Tries to evict the given amount of the data from the cache.
329	*
330	* @returns Bytes removed.
331	* @param pCache The global cache data.
332	* @param cbData Number of bytes to evict.
333	*/
334	static size_t pdmacFileCacheEvict(PPDMACFILECACHEGLOBAL pCache, size_t cbData)
335	{
336	size_t cbRemoved = ~0;
337
338	PDMACFILECACHE_IS_CRITSECT_OWNER(pCache);
339
340	if ((pCache->LruRecentlyUsed.cbCached + pCache->LruRecentlyGhost.cbCached) >= pCache->cbMax)
341	{
342	/* Delete desired pages from the cache. */
343	if (pCache->LruRecentlyUsed.cbCached < pCache->cbMax)
344	{
345	cbRemoved = pdmacFileCacheEvictPagesFrom(pCache, cbData,
346	&pCache->LruRecentlyGhost,
347	NULL);
348	}
349	else
350	{
351	cbRemoved = pdmacFileCacheEvictPagesFrom(pCache, cbData,
352	&pCache->LruRecentlyUsed,
353	NULL);
354	}
355	}
356	else
357	{
358	uint32_t cbUsed = pCache->LruRecentlyUsed.cbCached + pCache->LruRecentlyGhost.cbCached +
359	pCache->LruFrequentlyUsed.cbCached + pCache->LruFrequentlyGhost.cbCached;
360
361	if (cbUsed >= pCache->cbMax)
362	{
363	if (cbUsed == 2*pCache->cbMax)
364	cbRemoved = pdmacFileCacheEvictPagesFrom(pCache, cbData,
365	&pCache->LruFrequentlyGhost,
366	NULL);
367
368	if (cbRemoved >= cbData)
369	cbRemoved = pdmacFileCacheReplace(pCache, cbData, NULL);
370	}
371	}
372
373	return cbRemoved;
374	}
375
376	/**
377	* Updates the cache parameters
378	*
379	* @returns nothing.
380	* @param pCache The global cache data.
381	* @param pEntry The entry usign for the update.
382	*/
383	static void pdmacFileCacheUpdate(PPDMACFILECACHEGLOBAL pCache, PPDMACFILECACHEENTRY pEntry)
384	{
385	int32_t uUpdateVal = 0;
386
387	PDMACFILECACHE_IS_CRITSECT_OWNER(pCache);
388
389	/* Update parameters */
390	if (pEntry->pList == &pCache->LruRecentlyGhost)
391	{
392	if (pCache->LruRecentlyGhost.cbCached >= pCache->LruFrequentlyGhost.cbCached)
393	uUpdateVal = 1;
394	else
395	uUpdateVal = pCache->LruFrequentlyGhost.cbCached / pCache->LruRecentlyGhost.cbCached;
396
397	pCache->uAdaptVal = RT_MIN(pCache->uAdaptVal + uUpdateVal, pCache->cbMax);
398	}
399	else if (pEntry->pList == &pCache->LruFrequentlyGhost)
400	{
401	if (pCache->LruFrequentlyGhost.cbCached >= pCache->LruRecentlyGhost.cbCached)
402	uUpdateVal = 1;
403	else
404	uUpdateVal = pCache->LruRecentlyGhost.cbCached / pCache->LruFrequentlyGhost.cbCached;
405
406	pCache->uAdaptVal = RT_MIN(pCache->uAdaptVal - uUpdateVal, 0);
407	}
408	else
409	AssertMsgFailed(("Invalid list type\n"));
410	}
411
412	/**
413	* Initiates a read I/O task for the given entry.
414	*
415	* @returns nothing.
416	* @param pEntry The entry to fetch the data to.
417	*/
418	static void pdmacFileCacheReadFromEndpoint(PPDMACFILECACHEENTRY pEntry)
419	{
420	LogFlowFunc((": Reading data into cache entry %#p\n", pEntry));
421
422	/* Make sure no one evicts the entry while it is accessed. */
423	pEntry->fFlags \|= PDMACFILECACHE_ENTRY_IO_IN_PROGRESS;
424
425	PPDMACTASKFILE pIoTask = pdmacFileTaskAlloc(pEntry->pEndpoint);
426	AssertPtr(pIoTask);
427
428	AssertMsg(pEntry->pbData, ("Entry is in ghost state\n"));
429
430	pIoTask->pEndpoint = pEntry->pEndpoint;
431	pIoTask->enmTransferType = PDMACTASKFILETRANSFER_READ;
432	pIoTask->Off = pEntry->Core.Key;
433	pIoTask->DataSeg.cbSeg = pEntry->cbData;
434	pIoTask->DataSeg.pvSeg = pEntry->pbData;
435	pIoTask->pvUser = pEntry;
436	pIoTask->pfnCompleted = pdmacFileCacheTaskCompleted;
437
438	/* Send it off to the I/O manager. */
439	pdmacFileEpAddTask(pEntry->pEndpoint, pIoTask);
440	}
441
442	/**
443	* Initiates a write I/O task for the given entry.
444	*
445	* @returns nothing.
446	* @param pEntry The entry to read the data from.
447	*/
448	static void pdmacFileCacheWriteToEndpoint(PPDMACFILECACHEENTRY pEntry)
449	{
450	LogFlowFunc((": Writing data from cache entry %#p\n", pEntry));
451
452	/* Make sure no one evicts the entry while it is accessed. */
453	pEntry->fFlags \|= PDMACFILECACHE_ENTRY_IO_IN_PROGRESS;
454
455	PPDMACTASKFILE pIoTask = pdmacFileTaskAlloc(pEntry->pEndpoint);
456	AssertPtr(pIoTask);
457
458	AssertMsg(pEntry->pbData, ("Entry is in ghost state\n"));
459
460	pIoTask->pEndpoint = pEntry->pEndpoint;
461	pIoTask->enmTransferType = PDMACTASKFILETRANSFER_WRITE;
462	pIoTask->Off = pEntry->Core.Key;
463	pIoTask->DataSeg.cbSeg = pEntry->cbData;
464	pIoTask->DataSeg.pvSeg = pEntry->pbData;
465	pIoTask->pvUser = pEntry;
466	pIoTask->pfnCompleted = pdmacFileCacheTaskCompleted;
467
468	/* Send it off to the I/O manager. */
469	pdmacFileEpAddTask(pEntry->pEndpoint, pIoTask);
470	}
471
472	/**
473	* Completion callback for I/O tasks.
474	*
475	* @returns nothing.
476	* @param pTask The completed task.
477	* @param pvUser Opaque user data.
478	*/
479	static void pdmacFileCacheTaskCompleted(PPDMACTASKFILE pTask, void *pvUser)
480	{
481	PPDMACFILECACHEENTRY pEntry = (PPDMACFILECACHEENTRY)pvUser;
482	PPDMACFILECACHEGLOBAL pCache = pEntry->pCache;
483	PPDMASYNCCOMPLETIONENDPOINTFILE pEndpoint = pEntry->pEndpoint;
484
485	/* Reference the entry now as we are clearing the I/O in progres flag
486	* which protects the entry till now. */
487	pdmacFileEpCacheEntryRef(pEntry);
488
489	RTSemRWRequestWrite(pEndpoint->DataCache.SemRWEntries, RT_INDEFINITE_WAIT);
490	pEntry->fFlags &= ~PDMACFILECACHE_ENTRY_IO_IN_PROGRESS;
491
492	if (pTask->enmTransferType == PDMACTASKFILETRANSFER_WRITE)
493	{
494	pEntry->fFlags &= ~PDMACFILECACHE_ENTRY_IS_DIRTY;
495
496	/* Process waiting segment list. The data in entry might have changed inbetween. */
497	PPDMACFILETASKSEG pCurr = pEntry->pHead;
498
499	while (pCurr)
500	{
501	AssertMsg(pCurr->fWrite, ("Completed write entries should never have read tasks attached\n"));
502
503	memcpy(pEntry->pbData + pCurr->uBufOffset, pCurr->pvBuf, pCurr->cbTransfer);
504	pEntry->fFlags \|= PDMACFILECACHE_ENTRY_IS_DIRTY;
505
506	uint32_t uOld = ASMAtomicSubU32(&pCurr->pTask->cbTransferLeft, pCurr->cbTransfer);
507	AssertMsg(uOld >= pCurr->cbTransfer, ("New value would overflow\n"));
508	if (!(uOld - pCurr->cbTransfer)
509	&& !ASMAtomicXchgBool(&pCurr->pTask->fCompleted, true))
510	pdmR3AsyncCompletionCompleteTask(&pCurr->pTask->Core);
511
512	PPDMACFILETASKSEG pFree = pCurr;
513	pCurr = pCurr->pNext;
514
515	RTMemFree(pFree);
516	}
517	}
518	else
519	{
520	AssertMsg(pTask->enmTransferType == PDMACTASKFILETRANSFER_READ, ("Invalid transfer type\n"));
521	AssertMsg(!(pEntry->fFlags & PDMACFILECACHE_ENTRY_IS_DIRTY),("Invalid flags set\n"));
522
523	/* Process waiting segment list. */
524	PPDMACFILETASKSEG pCurr = pEntry->pHead;
525
526	while (pCurr)
527	{
528	if (pCurr->fWrite)
529	{
530	memcpy(pEntry->pbData + pCurr->uBufOffset, pCurr->pvBuf, pCurr->cbTransfer);
531	pEntry->fFlags \|= PDMACFILECACHE_ENTRY_IS_DIRTY;
532	}
533	else
534	memcpy(pCurr->pvBuf, pEntry->pbData + pCurr->uBufOffset, pCurr->cbTransfer);
535
536	uint32_t uOld = ASMAtomicSubU32(&pCurr->pTask->cbTransferLeft, pCurr->cbTransfer);
537	AssertMsg(uOld >= pCurr->cbTransfer, ("New value would overflow\n"));
538	if (!(uOld - pCurr->cbTransfer)
539	&& !ASMAtomicXchgBool(&pCurr->pTask->fCompleted, true))
540	pdmR3AsyncCompletionCompleteTask(&pCurr->pTask->Core);
541
542	PPDMACFILETASKSEG pFree = pCurr;
543	pCurr = pCurr->pNext;
544
545	RTMemFree(pFree);
546	}
547	}
548
549	pEntry->pHead = NULL;
550
551	if (pEntry->fFlags & PDMACFILECACHE_ENTRY_IS_DIRTY)
552	pdmacFileCacheWriteToEndpoint(pEntry);
553
554	RTSemRWReleaseWrite(pEndpoint->DataCache.SemRWEntries);
555
556	/* Dereference so that it isn't protected anymore except we issued anyother write for it. */
557	pdmacFileEpCacheEntryRelease(pEntry);
558	}
559
560	/**
561	* Initializies the I/O cache.
562	*
563	* returns VBox status code.
564	* @param pClassFile The global class data for file endpoints.
565	* @param pCfgNode CFGM node to query configuration data from.
566	*/
567	int pdmacFileCacheInit(PPDMASYNCCOMPLETIONEPCLASSFILE pClassFile, PCFGMNODE pCfgNode)
568	{
569	int rc = VINF_SUCCESS;
570	PPDMACFILECACHEGLOBAL pCache = &pClassFile->Cache;
571
572	/* Initialize members */
573	pCache->LruRecentlyUsed.pHead = NULL;
574	pCache->LruRecentlyUsed.pTail = NULL;
575	pCache->LruRecentlyUsed.cbCached = 0;
576
577	pCache->LruFrequentlyUsed.pHead = NULL;
578	pCache->LruFrequentlyUsed.pTail = NULL;
579	pCache->LruFrequentlyUsed.cbCached = 0;
580
581	pCache->LruRecentlyGhost.pHead = NULL;
582	pCache->LruRecentlyGhost.pTail = NULL;
583	pCache->LruRecentlyGhost.cbCached = 0;
584
585	pCache->LruFrequentlyGhost.pHead = NULL;
586	pCache->LruFrequentlyGhost.pTail = NULL;
587	pCache->LruFrequentlyGhost.cbCached = 0;
588
589	rc = CFGMR3QueryU32Def(pCfgNode, "CacheSize", &pCache->cbMax, 5 * _1M);
590	AssertLogRelRCReturn(rc, rc);
591
592	pCache->cbCached = 0;
593	pCache->uAdaptVal = 0;
594	LogFlowFunc((": Maximum number of bytes cached %u\n", pCache->cbCached));
595
596	STAMR3Register(pClassFile->Core.pVM, &pCache->cbMax,
597	STAMTYPE_U32, STAMVISIBILITY_ALWAYS,
598	"/PDM/AsyncCompletion/File/cbMax",
599	STAMUNIT_BYTES,
600	"Maximum cache size");
601	STAMR3Register(pClassFile->Core.pVM, &pCache->cbCached,
602	STAMTYPE_U32, STAMVISIBILITY_ALWAYS,
603	"/PDM/AsyncCompletion/File/cbCached",
604	STAMUNIT_BYTES,
605	"Currently used cache");
606	STAMR3Register(pClassFile->Core.pVM, &pCache->LruRecentlyUsed.cbCached,
607	STAMTYPE_U32, STAMVISIBILITY_ALWAYS,
608	"/PDM/AsyncCompletion/File/cbCachedMru",
609	STAMUNIT_BYTES,
610	"Number of bytes cached in Mru list");
611	STAMR3Register(pClassFile->Core.pVM, &pCache->LruFrequentlyUsed.cbCached,
612	STAMTYPE_U32, STAMVISIBILITY_ALWAYS,
613	"/PDM/AsyncCompletion/File/cbCachedFru",
614	STAMUNIT_BYTES,
615	"Number of bytes cached in Fru list");
616	STAMR3Register(pClassFile->Core.pVM, &pCache->LruRecentlyGhost.cbCached,
617	STAMTYPE_U32, STAMVISIBILITY_ALWAYS,
618	"/PDM/AsyncCompletion/File/cbCachedMruGhost",
619	STAMUNIT_BYTES,
620	"Number of bytes cached in Mru ghost list");
621	STAMR3Register(pClassFile->Core.pVM, &pCache->LruFrequentlyGhost.cbCached,
622	STAMTYPE_U32, STAMVISIBILITY_ALWAYS,
623	"/PDM/AsyncCompletion/File/cbCachedFruGhost",
624	STAMUNIT_BYTES, "Number of bytes cached in Fru ghost list");
625
626	#ifdef VBOX_WITH_STATISTICS
627	STAMR3Register(pClassFile->Core.pVM, &pCache->cHits,
628	STAMTYPE_COUNTER, STAMVISIBILITY_ALWAYS,
629	"/PDM/AsyncCompletion/File/CacheHits",
630	STAMUNIT_COUNT, "Number of hits in the cache");
631	STAMR3Register(pClassFile->Core.pVM, &pCache->cPartialHits,
632	STAMTYPE_COUNTER, STAMVISIBILITY_ALWAYS,
633	"/PDM/AsyncCompletion/File/CachePartialHits",
634	STAMUNIT_COUNT, "Number of partial hits in the cache");
635	STAMR3Register(pClassFile->Core.pVM, &pCache->cMisses,
636	STAMTYPE_COUNTER, STAMVISIBILITY_ALWAYS,
637	"/PDM/AsyncCompletion/File/CacheMisses",
638	STAMUNIT_COUNT, "Number of misses when accessing the cache");
639	STAMR3Register(pClassFile->Core.pVM, &pCache->StatRead,
640	STAMTYPE_COUNTER, STAMVISIBILITY_ALWAYS,
641	"/PDM/AsyncCompletion/File/CacheRead",
642	STAMUNIT_BYTES, "Number of bytes read from the cache");
643	STAMR3Register(pClassFile->Core.pVM, &pCache->StatWritten,
644	STAMTYPE_COUNTER, STAMVISIBILITY_ALWAYS,
645	"/PDM/AsyncCompletion/File/CacheWritten",
646	STAMUNIT_BYTES, "Number of bytes written to the cache");
647	STAMR3Register(pClassFile->Core.pVM, &pCache->StatTreeGet,
648	STAMTYPE_PROFILE_ADV, STAMVISIBILITY_ALWAYS,
649	"/PDM/AsyncCompletion/File/CacheTreeGet",
650	STAMUNIT_TICKS_PER_CALL, "Time taken to access an entry in the tree");
651	STAMR3Register(pClassFile->Core.pVM, &pCache->StatTreeInsert,
652	STAMTYPE_PROFILE_ADV, STAMVISIBILITY_ALWAYS,
653	"/PDM/AsyncCompletion/File/CacheTreeInsert",
654	STAMUNIT_TICKS_PER_CALL, "Time taken to insert an entry in the tree");
655	STAMR3Register(pClassFile->Core.pVM, &pCache->StatTreeRemove,
656	STAMTYPE_PROFILE_ADV, STAMVISIBILITY_ALWAYS,
657	"/PDM/AsyncCompletion/File/CacheTreeRemove",
658	STAMUNIT_TICKS_PER_CALL, "Time taken to remove an entry an the tree");
659	#endif
660
661	/* Initialize the critical section */
662	rc = RTCritSectInit(&pCache->CritSect);
663	return rc;
664	}
665
666	/**
667	* Destroysthe cache freeing all data.
668	*
669	* returns nothing.
670	* @param pClassFile The global class data for file endpoints.
671	*/
672	void pdmacFileCacheDestroy(PPDMASYNCCOMPLETIONEPCLASSFILE pClassFile)
673	{
674	PPDMACFILECACHEGLOBAL pCache = &pClassFile->Cache;
675
676	/* Make sure no one else uses the cache now */
677	RTCritSectEnter(&pCache->CritSect);
678
679	/* Cleanup deleting all cache entries waiting for in progress entries to finish. */
680	pdmacFileCacheDestroyList(&pCache->LruRecentlyUsed);
681	pdmacFileCacheDestroyList(&pCache->LruFrequentlyUsed);
682	pdmacFileCacheDestroyList(&pCache->LruRecentlyGhost);
683	pdmacFileCacheDestroyList(&pCache->LruFrequentlyGhost);
684
685	RTCritSectLeave(&pCache->CritSect);
686
687	RTCritSectDelete(&pCache->CritSect);
688	}
689
690	/**
691	* Initializes per endpoint cache data
692	* like the AVL tree used to access cached entries.
693	*
694	* @returns VBox status code.
695	* @param pEndpoint The endpoint to init the cache for,
696	* @param pClassFile The global class data for file endpoints.
697	*/
698	int pdmacFileEpCacheInit(PPDMASYNCCOMPLETIONENDPOINTFILE pEndpoint, PPDMASYNCCOMPLETIONEPCLASSFILE pClassFile)
699	{
700	PPDMACFILEENDPOINTCACHE pEndpointCache = &pEndpoint->DataCache;
701
702	pEndpointCache->pCache = &pClassFile->Cache;
703
704	int rc = RTSemRWCreate(&pEndpointCache->SemRWEntries);
705	if (RT_SUCCESS(rc))
706	{
707	pEndpointCache->pTree = (PAVLRFOFFTREE)RTMemAllocZ(sizeof(AVLRFOFFTREE));
708	if (!pEndpointCache->pTree)
709	{
710	rc = VERR_NO_MEMORY;
711	RTSemRWDestroy(pEndpointCache->SemRWEntries);
712	}
713	}
714
715	return rc;
716	}
717
718	/**
719	* Callback for the AVL destroy routine. Frees a cache entry for this endpoint.
720	*
721	* @returns IPRT status code.
722	* @param pNode The node to destroy.
723	* @param pvUser Opaque user data.
724	*/
725	static int pdmacFileEpCacheEntryDestroy(PAVLRFOFFNODECORE pNode, void *pvUser)
726	{
727	PPDMACFILECACHEENTRY pEntry = (PPDMACFILECACHEENTRY)pNode;
728	PPDMACFILECACHEGLOBAL pCache = (PPDMACFILECACHEGLOBAL)pvUser;
729	PPDMACFILEENDPOINTCACHE pEndpointCache = &pEntry->pEndpoint->DataCache;
730
731	while (pEntry->fFlags & (PDMACFILECACHE_ENTRY_IO_IN_PROGRESS \| PDMACFILECACHE_ENTRY_IS_DIRTY))
732	{
733	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
734	RTThreadSleep(250);
735	RTSemRWRequestWrite(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
736	}
737
738	AssertMsg(!(pEntry->fFlags & (PDMACFILECACHE_ENTRY_IO_IN_PROGRESS \| PDMACFILECACHE_ENTRY_IS_DIRTY)),
739	("Entry is dirty and/or still in progress fFlags=%#x\n", pEntry->fFlags));
740
741	pdmacFileCacheEntryRemoveFromList(pEntry);
742	pCache->cbCached -= pEntry->cbData;
743
744	RTMemPageFree(pEntry->pbData);
745	RTMemFree(pEntry);
746
747	return VINF_SUCCESS;
748	}
749
750	/**
751	* Destroys all cache ressources used by the given endpoint.
752	*
753	* @returns nothing.
754	* @param pEndpoint The endpoint to the destroy.
755	*/
756	void pdmacFileEpCacheDestroy(PPDMASYNCCOMPLETIONENDPOINTFILE pEndpoint)
757	{
758	PPDMACFILEENDPOINTCACHE pEndpointCache = &pEndpoint->DataCache;
759	PPDMACFILECACHEGLOBAL pCache = pEndpointCache->pCache;
760
761	/* Make sure nobody is accessing the cache while we delete the tree. */
762	RTSemRWRequestWrite(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
763	RTCritSectEnter(&pCache->CritSect);
764	RTAvlrFileOffsetDestroy(pEndpointCache->pTree, pdmacFileEpCacheEntryDestroy, pCache);
765	RTCritSectLeave(&pCache->CritSect);
766	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
767
768	RTSemRWDestroy(pEndpointCache->SemRWEntries);
769	}
770
771	static PPDMACFILECACHEENTRY pdmacFileEpCacheGetCacheEntryByOffset(PPDMACFILEENDPOINTCACHE pEndpointCache, RTFOFF off)
772	{
773	PPDMACFILECACHEGLOBAL pCache = pEndpointCache->pCache;
774	PPDMACFILECACHEENTRY pEntry = NULL;
775
776	STAM_PROFILE_ADV_START(&pCache->StatTreeGet, Cache);
777
778	RTSemRWRequestRead(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
779	pEntry = (PPDMACFILECACHEENTRY)RTAvlrFileOffsetRangeGet(pEndpointCache->pTree, off);
780	if (pEntry)
781	pdmacFileEpCacheEntryRef(pEntry);
782	RTSemRWReleaseRead(pEndpointCache->SemRWEntries);
783
784	STAM_PROFILE_ADV_STOP(&pCache->StatTreeGet, Cache);
785
786	return pEntry;
787	}
788
789	static PPDMACFILECACHEENTRY pdmacFileEpCacheGetCacheBestFitEntryByOffset(PPDMACFILEENDPOINTCACHE pEndpointCache, RTFOFF off)
790	{
791	PPDMACFILECACHEGLOBAL pCache = pEndpointCache->pCache;
792	PPDMACFILECACHEENTRY pEntry = NULL;
793
794	STAM_PROFILE_ADV_START(&pCache->StatTreeGet, Cache);
795
796	RTSemRWRequestRead(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
797	pEntry = (PPDMACFILECACHEENTRY)RTAvlrFileOffsetGetBestFit(pEndpointCache->pTree, off, true);
798	if (pEntry)
799	pdmacFileEpCacheEntryRef(pEntry);
800	RTSemRWReleaseRead(pEndpointCache->SemRWEntries);
801
802	STAM_PROFILE_ADV_STOP(&pCache->StatTreeGet, Cache);
803
804	return pEntry;
805	}
806
807	static void pdmacFileEpCacheInsertEntry(PPDMACFILEENDPOINTCACHE pEndpointCache, PPDMACFILECACHEENTRY pEntry)
808	{
809	PPDMACFILECACHEGLOBAL pCache = pEndpointCache->pCache;
810
811	STAM_PROFILE_ADV_START(&pCache->StatTreeInsert, Cache);
812	RTSemRWRequestWrite(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
813	bool fInserted = RTAvlrFileOffsetInsert(pEndpointCache->pTree, &pEntry->Core);
814	AssertMsg(fInserted, ("Node was not inserted into tree\n"));
815	STAM_PROFILE_ADV_STOP(&pCache->StatTreeInsert, Cache);
816	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
817	}
818
819	/**
820	* Allocates and initializes a new entry for the cache.
821	* The entry has a reference count of 1.
822	*
823	* @returns Pointer to the new cache entry or NULL if out of memory.
824	* @param pCache The cache the entry belongs to.
825	* @param pEndoint The endpoint the entry holds data for.
826	* @param off Start offset.
827	* @param cbData Size of the cache entry.
828	*/
829	static PPDMACFILECACHEENTRY pdmacFileCacheEntryAlloc(PPDMACFILECACHEGLOBAL pCache,
830	PPDMASYNCCOMPLETIONENDPOINTFILE pEndpoint,
831	RTFOFF off, size_t cbData)
832	{
833	PPDMACFILECACHEENTRY pEntryNew = (PPDMACFILECACHEENTRY)RTMemAllocZ(sizeof(PDMACFILECACHEENTRY));
834
835	if (RT_UNLIKELY(!pEntryNew))
836	return NULL;
837
838	pEntryNew->Core.Key = off;
839	pEntryNew->Core.KeyLast = off + cbData - 1;
840	pEntryNew->pEndpoint = pEndpoint;
841	pEntryNew->pCache = pCache;
842	pEntryNew->fFlags = 0;
843	pEntryNew->cRefs = 1; /* We are using it now. */
844	pEntryNew->pList = NULL;
845	pEntryNew->cbData = cbData;
846	pEntryNew->pHead = NULL;
847	pEntryNew->pbData = (uint8_t *)RTMemPageAlloc(cbData);
848
849	if (RT_UNLIKELY(!pEntryNew->pbData))
850	{
851	RTMemFree(pEntryNew);
852	return NULL;
853	}
854
855	return pEntryNew;
856	}
857
858	/**
859	* Advances the current segment buffer by the number of bytes transfered
860	* or gets the next segment.
861	*/
862	#define ADVANCE_SEGMENT_BUFFER(BytesTransfered) \
863	do \
864	{ \
865	cbSegLeft -= BytesTransfered; \
866	if (!cbSegLeft) \
867	{ \
868	iSegCurr++; \
869	cbSegLeft = paSegments[iSegCurr].cbSeg; \
870	pbSegBuf = (uint8_t *)paSegments[iSegCurr].pvSeg; \
871	} \
872	else \
873	pbSegBuf += BytesTransfered; \
874	} \
875	while (0);
876
877	/**
878	* Reads the specified data from the endpoint using the cache if possible.
879	*
880	* @returns VBox status code.
881	* @param pEndpoint The endpoint to read from.
882	* @param pTask The task structure used as identifier for this request.
883	* @param off The offset to start reading from.
884	* @param paSegments Pointer to the array holding the destination buffers.
885	* @param cSegments Number of segments in the array.
886	* @param cbRead Number of bytes to read.
887	*/
888	int pdmacFileEpCacheRead(PPDMASYNCCOMPLETIONENDPOINTFILE pEndpoint, PPDMASYNCCOMPLETIONTASKFILE pTask,
889	RTFOFF off, PCPDMDATASEG paSegments, size_t cSegments,
890	size_t cbRead)
891	{
892	int rc = VINF_SUCCESS;
893	PPDMACFILEENDPOINTCACHE pEndpointCache = &pEndpoint->DataCache;
894	PPDMACFILECACHEGLOBAL pCache = pEndpointCache->pCache;
895	PPDMACFILECACHEENTRY pEntry;
896
897	LogFlowFunc((": pEndpoint=%#p{%s} pTask=%#p off=%RTfoff paSegments=%#p cSegments=%u cbRead=%u\n",
898	pEndpoint, pEndpoint->Core.pszUri, pTask, off, paSegments, cSegments, cbRead));
899
900	pTask->cbTransferLeft = cbRead;
901	/* Set to completed to make sure that the task is valid while we access it. */
902	ASMAtomicWriteBool(&pTask->fCompleted, true);
903
904	int iSegCurr = 0;
905	uint8_t pbSegBuf = (uint8_t )paSegments[iSegCurr].pvSeg;
906	size_t cbSegLeft = paSegments[iSegCurr].cbSeg;
907
908	while (cbRead)
909	{
910	size_t cbToRead;
911
912	pEntry = pdmacFileEpCacheGetCacheEntryByOffset(pEndpointCache, off);
913
914	/*
915	* If there is no entry we try to create a new one eviciting unused pages
916	* if the cache is full. If this is not possible we will pass the request through
917	* and skip the caching (all entries may be still in progress so they can't
918	* be evicted)
919	* If we have an entry it can be in one of the LRU lists where the entry
920	* contains data (recently used or frequently used LRU) so we can just read
921	* the data we need and put the entry at the head of the frequently used LRU list.
922	* In case the entry is in one of the ghost lists it doesn't contain any data.
923	* We have to fetch it again evicting pages from either T1 or T2 to make room.
924	*/
925	if (pEntry)
926	{
927	RTFOFF OffDiff = off - pEntry->Core.Key;
928
929	AssertMsg(off >= pEntry->Core.Key,
930	("Overflow in calculation off=%RTfoff OffsetAligned=%RTfoff\n",
931	off, pEntry->Core.Key));
932
933	AssertPtr(pEntry->pList);
934
935	cbToRead = RT_MIN(pEntry->cbData - OffDiff, cbRead);
936	cbRead -= cbToRead;
937
938	if (!cbRead)
939	STAM_COUNTER_INC(&pCache->cHits);
940	else
941	STAM_COUNTER_INC(&pCache->cPartialHits);
942
943	STAM_COUNTER_ADD(&pCache->StatRead, cbToRead);
944
945	/* Ghost lists contain no data. */
946	if ( (pEntry->pList == &pCache->LruRecentlyUsed)
947	\|\| (pEntry->pList == &pCache->LruFrequentlyUsed))
948	{
949	LogFlow(("Fetching data for ghost entry %#p from file\n", pEntry));
950
951	if ( (pEntry->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS)
952	&& !(pEntry->fFlags & PDMACFILECACHE_ENTRY_IS_DIRTY))
953	{
954	RTSemRWRequestWrite(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
955	/* Check again. The completion callback might have raced us. */
956
957	if ( (pEntry->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS)
958	&& !(pEntry->fFlags & PDMACFILECACHE_ENTRY_IS_DIRTY))
959	{
960	/* Entry didn't completed yet. Append to the list */
961	while (cbToRead)
962	{
963	PPDMACFILETASKSEG pSeg = (PPDMACFILETASKSEG)RTMemAllocZ(sizeof(PDMACFILETASKSEG));
964
965	pSeg->pTask = pTask;
966	pSeg->uBufOffset = OffDiff;
967	pSeg->cbTransfer = RT_MIN(cbToRead, cbSegLeft);
968	pSeg->pvBuf = pbSegBuf;
969	pSeg->fWrite = false;
970
971	ADVANCE_SEGMENT_BUFFER(pSeg->cbTransfer);
972
973	pSeg->pNext = pEntry->pHead;
974	pEntry->pHead = pSeg;
975
976	off += pSeg->cbTransfer;
977	cbToRead -= pSeg->cbTransfer;
978	OffDiff += pSeg->cbTransfer;
979	}
980	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
981	}
982	else
983	{
984	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
985
986	/* Read as much as we can from the entry. */
987	while (cbToRead)
988	{
989	size_t cbCopy = RT_MIN(cbSegLeft, cbToRead);
990
991	memcpy(pbSegBuf, pEntry->pbData + OffDiff, cbCopy);
992
993	ADVANCE_SEGMENT_BUFFER(cbCopy);
994
995	cbToRead -= cbCopy;
996	off += cbCopy;
997	OffDiff += cbCopy;
998	ASMAtomicSubS32(&pTask->cbTransferLeft, cbCopy);
999	}
1000	}
1001	}
1002	else
1003	{
1004	/* Read as much as we can from the entry. */
1005	while (cbToRead)
1006	{
1007	size_t cbCopy = RT_MIN(cbSegLeft, cbToRead);
1008
1009	memcpy(pbSegBuf, pEntry->pbData + OffDiff, cbCopy);
1010
1011	ADVANCE_SEGMENT_BUFFER(cbCopy);
1012
1013	cbToRead -= cbCopy;
1014	off += cbCopy;
1015	OffDiff += cbCopy;
1016	ASMAtomicSubS32(&pTask->cbTransferLeft, cbCopy);
1017	}
1018	}
1019
1020	/* Move this entry to the top position */
1021	RTCritSectEnter(&pCache->CritSect);
1022	pdmacFileCacheEntryAddToList(&pCache->LruFrequentlyUsed, pEntry);
1023	RTCritSectLeave(&pCache->CritSect);
1024	}
1025	else
1026	{
1027	RTCritSectEnter(&pCache->CritSect);
1028	pdmacFileCacheUpdate(pCache, pEntry);
1029	pdmacFileCacheReplace(pCache, pEntry->cbData, pEntry->pList);
1030
1031	/* Move the entry to T2 and fetch it to the cache. */
1032	pdmacFileCacheEntryAddToList(&pCache->LruFrequentlyUsed, pEntry);
1033	RTCritSectLeave(&pCache->CritSect);
1034
1035	pEntry->pbData = (uint8_t *)RTMemPageAlloc(pEntry->cbData);
1036	AssertPtr(pEntry->pbData);
1037
1038	while (cbToRead)
1039	{
1040	PPDMACFILETASKSEG pSeg = (PPDMACFILETASKSEG)RTMemAllocZ(sizeof(PDMACFILETASKSEG));
1041
1042	AssertMsg(off >= pEntry->Core.Key,
1043	("Overflow in calculation off=%RTfoff OffsetAligned=%RTfoff\n",
1044	off, pEntry->Core.Key));
1045
1046	pSeg->pTask = pTask;
1047	pSeg->uBufOffset = OffDiff;
1048	pSeg->cbTransfer = RT_MIN(cbToRead, cbSegLeft);
1049	pSeg->pvBuf = pbSegBuf;
1050
1051	ADVANCE_SEGMENT_BUFFER(pSeg->cbTransfer);
1052
1053	pSeg->pNext = pEntry->pHead;
1054	pEntry->pHead = pSeg;
1055
1056	off += pSeg->cbTransfer;
1057	OffDiff += pSeg->cbTransfer;
1058	cbToRead -= pSeg->cbTransfer;
1059	}
1060
1061	pdmacFileCacheReadFromEndpoint(pEntry);
1062	}
1063	pdmacFileEpCacheEntryRelease(pEntry);
1064	}
1065	else
1066	{
1067	/* No entry found for this offset. Get best fit entry and fetch the data to the cache. */
1068	size_t cbToReadAligned;
1069	PPDMACFILECACHEENTRY pEntryBestFit = pdmacFileEpCacheGetCacheBestFitEntryByOffset(pEndpointCache, off);
1070
1071	LogFlow(("%sbest fit entry for off=%RTfoff (BestFit=%RTfoff BestFitEnd=%RTfoff BestFitSize=%u)\n",
1072	pEntryBestFit ? "" : "No ",
1073	off,
1074	pEntryBestFit ? pEntryBestFit->Core.Key : 0,
1075	pEntryBestFit ? pEntryBestFit->Core.KeyLast : 0,
1076	pEntryBestFit ? pEntryBestFit->cbData : 0));
1077
1078	if (pEntryBestFit && ((off + (RTFOFF)cbRead) > pEntryBestFit->Core.Key))
1079	{
1080	cbToRead = pEntryBestFit->Core.Key - off;
1081	pdmacFileEpCacheEntryRelease(pEntryBestFit);
1082	cbToReadAligned = cbToRead;
1083	}
1084	else
1085	{
1086	/*
1087	* Align the size to a 4KB boundary.
1088	* Memory size is aligned to a page boundary
1089	* and memory is wasted if the size is rahter small.
1090	* (For example reads with a size of 512 bytes.
1091	*/
1092	cbToRead = cbRead;
1093	cbToReadAligned = RT_ALIGN_Z(cbRead, PAGE_SIZE);
1094
1095	/* Clip read to file size */
1096	cbToReadAligned = RT_MIN(pEndpoint->cbFile - off, cbToReadAligned);
1097	if (pEntryBestFit)
1098	cbToReadAligned = RT_MIN(cbToReadAligned, pEntryBestFit->Core.Key - off);
1099	}
1100
1101	cbRead -= cbToRead;
1102
1103	if (!cbRead)
1104	STAM_COUNTER_INC(&pCache->cMisses);
1105	else
1106	STAM_COUNTER_INC(&pCache->cPartialHits);
1107
1108	RTCritSectEnter(&pCache->CritSect);
1109	size_t cbRemoved = pdmacFileCacheEvict(pCache, cbToReadAligned);
1110	RTCritSectLeave(&pCache->CritSect);
1111
1112	if (cbRemoved >= cbToRead)
1113	{
1114	LogFlow(("Evicted %u bytes (%u requested). Creating new cache entry\n", cbRemoved, cbToReadAligned));
1115	PPDMACFILECACHEENTRY pEntryNew = pdmacFileCacheEntryAlloc(pCache, pEndpoint, off, cbToReadAligned);
1116	AssertPtr(pEntryNew);
1117
1118	RTCritSectEnter(&pCache->CritSect);
1119	pdmacFileCacheEntryAddToList(&pCache->LruRecentlyUsed, pEntryNew);
1120	RTCritSectLeave(&pCache->CritSect);
1121
1122	pdmacFileEpCacheInsertEntry(pEndpointCache, pEntryNew);
1123	uint32_t uBufOffset = 0;
1124
1125	pCache->cbCached += cbToReadAligned;
1126
1127	while (cbToRead)
1128	{
1129	PPDMACFILETASKSEG pSeg = (PPDMACFILETASKSEG)RTMemAllocZ(sizeof(PDMACFILETASKSEG));
1130
1131	pSeg->pTask = pTask;
1132	pSeg->uBufOffset = uBufOffset;
1133	pSeg->cbTransfer = RT_MIN(cbToRead, cbSegLeft);
1134	pSeg->pvBuf = pbSegBuf;
1135
1136	ADVANCE_SEGMENT_BUFFER(pSeg->cbTransfer);
1137
1138	pSeg->pNext = pEntryNew->pHead;
1139	pEntryNew->pHead = pSeg;
1140
1141	off += pSeg->cbTransfer;
1142	cbToRead -= pSeg->cbTransfer;
1143	uBufOffset += pSeg->cbTransfer;
1144	}
1145
1146	pdmacFileCacheReadFromEndpoint(pEntryNew);
1147	pdmacFileEpCacheEntryRelease(pEntryNew); /* it is protected by the I/O in progress flag now. */
1148	}
1149	else
1150	{
1151	/*
1152	* There is not enough free space in the cache.
1153	* Pass the request directly to the I/O manager.
1154	*/
1155	LogFlow(("Couldn't evict %u bytes from the cache (%u actually removed). Remaining request will be passed through\n", cbToRead, cbRemoved));
1156
1157	while (cbToRead)
1158	{
1159	PPDMACTASKFILE pIoTask = pdmacFileTaskAlloc(pEndpoint);
1160	AssertPtr(pIoTask);
1161
1162	pIoTask->pEndpoint = pEndpoint;
1163	pIoTask->enmTransferType = PDMACTASKFILETRANSFER_READ;
1164	pIoTask->Off = off;
1165	pIoTask->DataSeg.cbSeg = RT_MIN(cbToRead, cbSegLeft);
1166	pIoTask->DataSeg.pvSeg = pbSegBuf;
1167	pIoTask->pvUser = pTask;
1168	pIoTask->pfnCompleted = pdmacFileEpTaskCompleted;
1169
1170	off += pIoTask->DataSeg.cbSeg;
1171	cbToRead -= pIoTask->DataSeg.cbSeg;
1172
1173	ADVANCE_SEGMENT_BUFFER(pIoTask->DataSeg.cbSeg);
1174
1175	/* Send it off to the I/O manager. */
1176	pdmacFileEpAddTask(pEndpoint, pIoTask);
1177	}
1178	}
1179	}
1180	}
1181
1182	ASMAtomicWriteBool(&pTask->fCompleted, false);
1183
1184	if (ASMAtomicReadS32(&pTask->cbTransferLeft) == 0
1185	&& !ASMAtomicXchgBool(&pTask->fCompleted, true))
1186	pdmR3AsyncCompletionCompleteTask(&pTask->Core);
1187
1188	return rc;
1189	}
1190
1191	/**
1192	* Writes the given data to the endpoint using the cache if possible.
1193	*
1194	* @returns VBox status code.
1195	* @param pEndpoint The endpoint to write to.
1196	* @param pTask The task structure used as identifier for this request.
1197	* @param off The offset to start writing to
1198	* @param paSegments Pointer to the array holding the source buffers.
1199	* @param cSegments Number of segments in the array.
1200	* @param cbWrite Number of bytes to write.
1201	*/
1202	int pdmacFileEpCacheWrite(PPDMASYNCCOMPLETIONENDPOINTFILE pEndpoint, PPDMASYNCCOMPLETIONTASKFILE pTask,
1203	RTFOFF off, PCPDMDATASEG paSegments, size_t cSegments,
1204	size_t cbWrite)
1205	{
1206	int rc = VINF_SUCCESS;
1207	PPDMACFILEENDPOINTCACHE pEndpointCache = &pEndpoint->DataCache;
1208	PPDMACFILECACHEGLOBAL pCache = pEndpointCache->pCache;
1209	PPDMACFILECACHEENTRY pEntry;
1210
1211	LogFlowFunc((": pEndpoint=%#p{%s} pTask=%#p off=%RTfoff paSegments=%#p cSegments=%u cbWrite=%u\n",
1212	pEndpoint, pEndpoint->Core.pszUri, pTask, off, paSegments, cSegments, cbWrite));
1213
1214	pTask->cbTransferLeft = cbWrite;
1215	/* Set to completed to make sure that the task is valid while we access it. */
1216	ASMAtomicWriteBool(&pTask->fCompleted, true);
1217
1218	int iSegCurr = 0;
1219	uint8_t pbSegBuf = (uint8_t )paSegments[iSegCurr].pvSeg;
1220	size_t cbSegLeft = paSegments[iSegCurr].cbSeg;
1221
1222	while (cbWrite)
1223	{
1224	size_t cbToWrite;
1225
1226	pEntry = pdmacFileEpCacheGetCacheEntryByOffset(pEndpointCache, off);
1227
1228	if (pEntry)
1229	{
1230	/* Write the data into the entry and mark it as dirty */
1231	AssertPtr(pEntry->pList);
1232
1233	RTFOFF OffDiff = off - pEntry->Core.Key;
1234
1235	AssertMsg(off >= pEntry->Core.Key,
1236	("Overflow in calculation off=%RTfoff OffsetAligned=%RTfoff\n",
1237	off, pEntry->Core.Key));
1238
1239	cbToWrite = RT_MIN(pEntry->cbData - OffDiff, cbWrite);
1240	cbWrite -= cbToWrite;
1241
1242	if (!cbWrite)
1243	STAM_COUNTER_INC(&pCache->cHits);
1244	else
1245	STAM_COUNTER_INC(&pCache->cPartialHits);
1246
1247	STAM_COUNTER_ADD(&pCache->StatWritten, cbToWrite);
1248
1249	/* Ghost lists contain no data. */
1250	if ( (pEntry->pList == &pCache->LruRecentlyUsed)
1251	\|\| (pEntry->pList == &pCache->LruFrequentlyUsed))
1252	{
1253	if (pEntry->fFlags & PDMACFILECACHE_ENTRY_IS_DIRTY)
1254	{
1255	RTSemRWRequestWrite(pEndpointCache->SemRWEntries, RT_INDEFINITE_WAIT);
1256	/* Check again. The completion callback might have raced us. */
1257
1258	if (pEntry->fFlags & PDMACFILECACHE_ENTRY_IS_DIRTY)
1259	{
1260	AssertMsg(pEntry->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS,
1261	("Entry is dirty but not in progress\n"));
1262
1263	/* The data isn't written to the file yet */
1264	while (cbToWrite)
1265	{
1266	PPDMACFILETASKSEG pSeg = (PPDMACFILETASKSEG)RTMemAllocZ(sizeof(PDMACFILETASKSEG));
1267
1268	pSeg->pTask = pTask;
1269	pSeg->uBufOffset = OffDiff;
1270	pSeg->cbTransfer = RT_MIN(cbToWrite, cbSegLeft);
1271	pSeg->pvBuf = pbSegBuf;
1272	pSeg->fWrite = true;
1273
1274	ADVANCE_SEGMENT_BUFFER(pSeg->cbTransfer);
1275
1276	pSeg->pNext = pEntry->pHead;
1277	pEntry->pHead = pSeg;
1278
1279	off += pSeg->cbTransfer;
1280	OffDiff += pSeg->cbTransfer;
1281	cbToWrite -= pSeg->cbTransfer;
1282	}
1283	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
1284	}
1285	else
1286	{
1287	RTSemRWReleaseWrite(pEndpointCache->SemRWEntries);
1288
1289	AssertMsg(!(pEntry->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS),
1290	("Entry is not dirty but in progress\n"));
1291
1292	/* Write as much as we can into the entry and update the file. */
1293	while (cbToWrite)
1294	{
1295	size_t cbCopy = RT_MIN(cbSegLeft, cbToWrite);
1296
1297	memcpy(pEntry->pbData + OffDiff, pbSegBuf, cbCopy);
1298
1299	ADVANCE_SEGMENT_BUFFER(cbCopy);
1300
1301	cbToWrite-= cbCopy;
1302	off += cbCopy;
1303	OffDiff += cbCopy;
1304	ASMAtomicSubS32(&pTask->cbTransferLeft, cbCopy);
1305	}
1306
1307	pEntry->fFlags \|= PDMACFILECACHE_ENTRY_IS_DIRTY;
1308	pdmacFileCacheWriteToEndpoint(pEntry);
1309	}
1310	}
1311	else
1312	{
1313	AssertMsg(!(pEntry->fFlags & PDMACFILECACHE_ENTRY_IO_IN_PROGRESS),
1314	("Entry is not dirty but in progress\n"));
1315
1316	/* Write as much as we can into the entry and update the file. */
1317	while (cbToWrite)
1318	{
1319	size_t cbCopy = RT_MIN(cbSegLeft, cbToWrite);
1320
1321	memcpy(pEntry->pbData + OffDiff, pbSegBuf, cbCopy);
1322
1323	ADVANCE_SEGMENT_BUFFER(cbCopy);
1324
1325	cbToWrite-= cbCopy;
1326	off += cbCopy;
1327	OffDiff += cbCopy;
1328	ASMAtomicSubS32(&pTask->cbTransferLeft, cbCopy);
1329	}
1330
1331	pEntry->fFlags \|= PDMACFILECACHE_ENTRY_IS_DIRTY;
1332	pdmacFileCacheWriteToEndpoint(pEntry);
1333	}
1334
1335	/* Move this entry to the top position */
1336	RTCritSectEnter(&pCache->CritSect);
1337	pdmacFileCacheEntryAddToList(&pCache->LruFrequentlyUsed, pEntry);
1338	RTCritSectLeave(&pCache->CritSect);
1339	}
1340	else
1341	{
1342	RTCritSectEnter(&pCache->CritSect);
1343	pdmacFileCacheUpdate(pCache, pEntry);
1344	pdmacFileCacheReplace(pCache, pEntry->cbData, pEntry->pList);
1345
1346	/* Move the entry to T2 and fetch it to the cache. */
1347	pdmacFileCacheEntryAddToList(&pCache->LruFrequentlyUsed, pEntry);
1348	RTCritSectLeave(&pCache->CritSect);
1349
1350	pEntry->pbData = (uint8_t *)RTMemPageAlloc(pEntry->cbData);
1351	AssertPtr(pEntry->pbData);
1352
1353	while (cbToWrite)
1354	{
1355	PPDMACFILETASKSEG pSeg = (PPDMACFILETASKSEG)RTMemAllocZ(sizeof(PDMACFILETASKSEG));
1356
1357	AssertMsg(off >= pEntry->Core.Key,
1358	("Overflow in calculation off=%RTfoff OffsetAligned=%RTfoff\n",
1359	off, pEntry->Core.Key));
1360
1361	pSeg->pTask = pTask;
1362	pSeg->uBufOffset = OffDiff;
1363	pSeg->cbTransfer = RT_MIN(cbToWrite, cbSegLeft);
1364	pSeg->pvBuf = pbSegBuf;
1365	pSeg->fWrite = true;
1366
1367	ADVANCE_SEGMENT_BUFFER(pSeg->cbTransfer);
1368
1369	pSeg->pNext = pEntry->pHead;
1370	pEntry->pHead = pSeg;
1371
1372	off += pSeg->cbTransfer;
1373	OffDiff += pSeg->cbTransfer;
1374	cbToWrite -= pSeg->cbTransfer;
1375	}
1376
1377	pdmacFileCacheReadFromEndpoint(pEntry);
1378	}
1379
1380	/* Release the reference. If it is still needed the I/O in progress flag should protect it now. */
1381	pdmacFileEpCacheEntryRelease(pEntry);
1382	}
1383	else
1384	{
1385	/*
1386	* No entry found. Write directly into file.
1387	*/
1388	PPDMACFILECACHEENTRY pEntryBestFit = pdmacFileEpCacheGetCacheBestFitEntryByOffset(pEndpointCache, off);
1389
1390	LogFlow(("%sbest fit entry for off=%RTfoff (BestFit=%RTfoff BestFitEnd=%RTfoff BestFitSize=%u)\n",
1391	pEntryBestFit ? "" : "No ",
1392	off,
1393	pEntryBestFit ? pEntryBestFit->Core.Key : 0,
1394	pEntryBestFit ? pEntryBestFit->Core.KeyLast : 0,
1395	pEntryBestFit ? pEntryBestFit->cbData : 0));
1396
1397	if (pEntryBestFit && ((off + (RTFOFF)cbWrite) > pEntryBestFit->Core.Key))
1398	{
1399	cbToWrite = pEntryBestFit->Core.Key - off;
1400	pdmacFileEpCacheEntryRelease(pEntryBestFit);
1401	}
1402	else
1403	cbToWrite = cbWrite;
1404
1405	cbWrite -= cbToWrite;
1406
1407	while (cbToWrite)
1408	{
1409	PPDMACTASKFILE pIoTask = pdmacFileTaskAlloc(pEndpoint);
1410	AssertPtr(pIoTask);
1411
1412	pIoTask->pEndpoint = pEndpoint;
1413	pIoTask->enmTransferType = PDMACTASKFILETRANSFER_WRITE;
1414	pIoTask->Off = off;
1415	pIoTask->DataSeg.cbSeg = RT_MIN(cbToWrite, cbSegLeft);
1416	pIoTask->DataSeg.pvSeg = pbSegBuf;
1417	pIoTask->pvUser = pTask;
1418	pIoTask->pfnCompleted = pdmacFileEpTaskCompleted;
1419
1420	off += pIoTask->DataSeg.cbSeg;
1421	cbToWrite -= pIoTask->DataSeg.cbSeg;
1422
1423	ADVANCE_SEGMENT_BUFFER(pIoTask->DataSeg.cbSeg);
1424
1425	/* Send it off to the I/O manager. */
1426	pdmacFileEpAddTask(pEndpoint, pIoTask);
1427	}
1428	}
1429	}
1430
1431	ASMAtomicWriteBool(&pTask->fCompleted, false);
1432
1433	if (ASMAtomicReadS32(&pTask->cbTransferLeft) == 0
1434	&& !ASMAtomicXchgBool(&pTask->fCompleted, true))
1435	pdmR3AsyncCompletionCompleteTask(&pTask->Core);
1436
1437	return VINF_SUCCESS;
1438	}
1439
1440	#undef ADVANCE_SEGMENT_BUFFER
1441

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

source: vbox/trunk/src/VBox/VMM/PDMAsyncCompletionFileCache.cpp@ 23673

Download in other formats: