Changeset 44355 in vbox for trunk/src/VBox/VMM

Timestamp:

Jan 24, 2013 1:27:28 PM (12 years ago)

Author:

vboxsync

Message:

PDMNetShaper: Cleanups & fixes (untested). PVM -> PUVM. Note: cs is a register within the VMM, not critical section. :)

Location:

trunk/src/VBox/VMM

Files:

• Makefile.kmk (modified) (2 diffs)
• VMMR0/PDMNetShaperR0.cpp (deleted)
• VMMR3/PDMDriver.cpp (modified) (2 diffs)
• VMMR3/PDMNetShaper.cpp (modified) (16 diffs)
• include/PDMInternal.h (modified) (1 diff)
• include/PDMNetShaperInternal.h (moved) (moved from trunk/include/VBox/vmm/pdmnetshaperint.h ) (3 diffs)

trunk/src/VBox/VMM/Makefile.kmk

@@ -214 +214 @@
 ifdef VBOX_WITH_NETSHAPER
  VBoxVMM_SOURCES += \
-        VMMR3/PDMNetShaper.cpp
+        VMMR3/PDMNetShaper.cpp \
+        VMMAll/PDMAllNetShaper.cpp
 endif
 
@@ -548 +549 @@
  ifdef VBOX_WITH_NETSHAPER
   VMMR0_SOURCES += \
-        VMMR0/PDMNetShaperR0.cpp
+        VMMAll/PDMAllNetShaper.cpp
  endif
  VMMR0_SOURCES.amd64 = \

trunk/src/VBox/VMM/VMMR3/PDMDriver.cpp

@@ -1561 +1561 @@
              pDrvIns->pReg->szName, pDrvIns->iInstance, pFilter, pszBwGroup, pszBwGroup));
 
-    int rc = PDMR3NsAttach(pDrvIns->Internal.s.pVMR3, pDrvIns, pszBwGroup, pFilter);
+    int rc = PDMR3NsAttach(pDrvIns->Internal.s.pVMR3->pUVM, pDrvIns, pszBwGroup, pFilter);
 
     LogFlow(("pdmR3DrvHlp_NetShaperAttach: caller='%s'/%d: returns %Rrc\n", pDrvIns->pReg->szName,
@@ -1576 +1576 @@
              pDrvIns->pReg->szName, pDrvIns->iInstance, pFilter));
 
-    int rc = PDMR3NsDetach(pDrvIns->Internal.s.pVMR3, pDrvIns, pFilter);
+    int rc = PDMR3NsDetach(pDrvIns->Internal.s.pVMR3->pUVM, pDrvIns, pFilter);
 
     LogFlow(("pdmR3DrvHlp_NetShaperDetach: caller='%s'/%d: returns %Rrc\n", pDrvIns->pReg->szName,

trunk/src/VBox/VMM/VMMR3/PDMNetShaper.cpp

@@ -42 +42 @@
 
 #include <VBox/vmm/pdmnetshaper.h>
-#include <VBox/vmm/pdmnetshaperint.h>
+#include "PDMNetShaperInternal.h"
 
 
@@ -57 +57 @@
     PVM                      pVM;
     /** Critical section protecting all members below. */
-    RTCRITSECT               cs;
+    RTCRITSECT               Lock;
     /** Pending TX thread. */
-    PPDMTHREAD               hTxThread;
+    PPDMTHREAD               pTxThread;
     /** Pointer to the first bandwidth group. */
     PPDMNSBWGROUP            pBwGroupsHead;
@@ -65 +65 @@
 
 
-
-
-
-
-static PPDMNSBWGROUP pdmNsBwGroupFindById(PPDMNETSHAPER pShaper, const char *pcszId)
+/** Takes the shaper lock (asserts but doesn't return or anything on
+ *  failure). */
+#define LOCK_NETSHAPER(a_pShaper) do { int rcShaper = RTCritSectEnter(&(a_pShaper)->Lock); AssertRC(rcShaper); } while (0)
+
+/** Takes the shaper lock, returns + asserts on failure. */
+#define LOCK_NETSHAPER_RETURN(a_pShaper) \
+    do { int rcShaper = RTCritSectEnter(&(a_pShaper)->Lock); AssertRCReturn(rcShaper, rcShaper); } while (0)
+
+/** Releases the shaper lock (asserts on failure). */
+#define UNLOCK_NETSHAPER(a_pShaper) do { int rcShaper = RTCritSectLeave(&(a_pShaper)->Lock); AssertRC(rcShaper); } while (0)
+
+
+
+
+static PPDMNSBWGROUP pdmNsBwGroupFindById(PPDMNETSHAPER pShaper, const char *pszId)
 {
     PPDMNSBWGROUP pBwGroup = NULL;
 
-    if (RT_VALID_PTR(pcszId))
-    {
-        int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
+    if (RT_VALID_PTR(pszId))
+    {
+        LOCK_NETSHAPER(pShaper);
 
         pBwGroup = pShaper->pBwGroupsHead;
         while (   pBwGroup
-               && RTStrCmp(pBwGroup->pszName, pcszId))
-            pBwGroup = pBwGroup->pNext;
-
-        rc = RTCritSectLeave(&pShaper->cs); AssertRC(rc);
+               && RTStrCmp(pBwGroup->pszNameR3, pszId))
+            pBwGroup = pBwGroup->pNextR3;
+
+        UNLOCK_NETSHAPER(pShaper);
     }
 
@@ -91 +101 @@
 static void pdmNsBwGroupLink(PPDMNSBWGROUP pBwGroup)
 {
-    PPDMNETSHAPER pShaper = pBwGroup->pShaper;
-    int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
-
-    pBwGroup->pNext = pShaper->pBwGroupsHead;
+    PPDMNETSHAPER pShaper = pBwGroup->pShaperR3;
+    LOCK_NETSHAPER(pShaper);
+
+    pBwGroup->pNextR3 = pShaper->pBwGroupsHead;
     pShaper->pBwGroupsHead = pBwGroup;
 
-    rc = RTCritSectLeave(&pShaper->cs); AssertRC(rc);
+    UNLOCK_NETSHAPER(pShaper);
 }
 
@@ -105 +115 @@
 {
     PPDMNETSHAPER pShaper = pBwGroup->pShaper;
-    int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
+    LOCK_NETSHAPER(pShaper);
 
     if (pBwGroup == pShaper->pBwGroupsHead)
@@ -120 +130 @@
     }
 
-    rc = RTCritSectLeave(&pShaper->cs); AssertRC(rc);
+    UNLOCK_NETSHAPER(pShaper);
 }
 #endif
 
 
-static void pdmNsBwGroupSetLimit(PPDMNSBWGROUP pBwGroup, uint64_t cbTransferPerSecMax)
-{
-    pBwGroup->cbTransferPerSecMax = cbTransferPerSecMax;
-    pBwGroup->cbBucketSize        = RT_MAX(PDM_NETSHAPER_MIN_BUCKET_SIZE,
-                                           cbTransferPerSecMax * PDM_NETSHAPER_MAX_LATENCY / 1000);
-    LogFlowFunc(("New rate limit is %llu bytes per second, adjusted bucket size to %d bytes\n",
-                 pBwGroup->cbTransferPerSecMax, pBwGroup->cbBucketSize));
-}
-
-
-static int pdmNsBwGroupCreate(PPDMNETSHAPER pShaper, const char *pcszBwGroup, uint64_t cbTransferPerSecMax)
-{
-    LogFlowFunc(("pShaper=%#p pcszBwGroup=%#p{%s} cbTransferPerSecMax=%llu\n",
-                 pShaper, pcszBwGroup, pcszBwGroup, cbTransferPerSecMax));
+static void pdmNsBwGroupSetLimit(PPDMNSBWGROUP pBwGroup, uint64_t cbPerSecMax)
+{
+    pBwGroup->cbPerSecMax = cbPerSecMax;
+    pBwGroup->cbBucket    = RT_MAX(PDM_NETSHAPER_MIN_BUCKET_SIZE, cbPerSecMax * PDM_NETSHAPER_MAX_LATENCY / 1000);
+    LogFlow(("pdmNsBwGroupSetLimit: New rate limit is %llu bytes per second, adjusted bucket size to %u bytes\n",
+             pBwGroup->cbPerSecMax, pBwGroup->cbBucket));
+}
+
+
+static int pdmNsBwGroupCreate(PPDMNETSHAPER pShaper, const char *pszBwGroup, uint64_t cbPerSecMax)
+{
+    LogFlow(("pdmNsBwGroupCreate: pShaper=%#p pszBwGroup=%#p{%s} cbPerSecMax=%llu\n", pShaper, pszBwGroup, pszBwGroup, cbPerSecMax));
 
     AssertPtrReturn(pShaper, VERR_INVALID_POINTER);
-    AssertPtrReturn(pcszBwGroup, VERR_INVALID_POINTER);
-    AssertReturn(*pcszBwGroup != '\0', VERR_INVALID_PARAMETER);
+    AssertPtrReturn(pszBwGroup, VERR_INVALID_POINTER);
+    AssertReturn(*pszBwGroup != '\0', VERR_INVALID_PARAMETER);
 
     int         rc;
-    PPDMNSBWGROUP pBwGroup = pdmNsBwGroupFindById(pShaper, pcszBwGroup);
+    PPDMNSBWGROUP pBwGroup = pdmNsBwGroupFindById(pShaper, pszBwGroup);
     if (!pBwGroup)
     {
@@ -152 +160 @@
         if (RT_SUCCESS(rc))
         {
-            rc = PDMR3CritSectInit(pShaper->pVM, &pBwGroup->cs, RT_SRC_POS, "BWGRP");
+            rc = PDMR3CritSectInit(pShaper->pVM, &pBwGroup->Lock, RT_SRC_POS, "BWGRP");
             if (RT_SUCCESS(rc))
             {
-                pBwGroup->pszName = RTStrDup(pcszBwGroup);
-                if (pBwGroup->pszName)
+                pBwGroup->pszNameR3 = MMR3HeapStrDup(pShaper->pVM, MM_TAG_PDM_NET_SHAPER, pszBwGroup);
+                if (pBwGroup->pszNameR3)
                 {
-                    pBwGroup->pShaper               = pShaper;
+                    pBwGroup->pShaperR3             = pShaper;
                     pBwGroup->cRefs                 = 0;
 
-                    pdmNsBwGroupSetLimit(pBwGroup, cbTransferPerSecMax);
-
-                    pBwGroup->cbTokensLast          = pBwGroup->cbBucketSize;
+                    pdmNsBwGroupSetLimit(pBwGroup, cbPerSecMax);
+
+                    pBwGroup->cbTokensLast          = pBwGroup->cbBucket;
                     pBwGroup->tsUpdatedLast         = RTTimeSystemNanoTS();
 
-                    LogFlowFunc(("pcszBwGroup={%s} cbBucketSize=%u\n",
-                                 pcszBwGroup, pBwGroup->cbBucketSize));
+                    LogFlowFunc(("pszBwGroup={%s} cbBucket=%u\n",
+                                 pszBwGroup, pBwGroup->cbBucket));
                     pdmNsBwGroupLink(pBwGroup);
                     return VINF_SUCCESS;
                 }
-                PDMR3CritSectDelete(&pBwGroup->cs);
+                PDMR3CritSectDelete(&pBwGroup->Lock);
             }
             MMHyperFree(pShaper->pVM, pBwGroup);
@@ -189 +197 @@
 {
     Assert(pBwGroup->cRefs == 0);
-    if (PDMCritSectIsInitialized(&pBwGroup->cs))
-        PDMR3CritSectDelete(&pBwGroup->cs);
+    if (PDMCritSectIsInitialized(&pBwGroup->Lock))
+        PDMR3CritSectDelete(&pBwGroup->Lock);
 }
 
@@ -215 +223 @@
      */
     AssertPtr(pBwGroup);
-    AssertPtr(pBwGroup->pShaper);
-    Assert(RTCritSectIsOwner(&pBwGroup->pShaper->cs));
-    //int rc = RTCritSectEnter(&pBwGroup->cs); AssertRC(rc);
+    AssertPtr(pBwGroup->pShaperR3);
+    Assert(RTCritSectIsOwner(&pBwGroup->pShaperR3->Lock));
+    //LOCK_NETSHAPER(pShaper);
 
     /* Check if the group is disabled. */
-    if (pBwGroup->cbTransferPerSecMax == 0)
+    if (pBwGroup->cbPerSecMax == 0)
         return;
 
-    PPDMNSFILTER pFilter = pBwGroup->pFiltersHead;
+    PPDMNSFILTER pFilter = pBwGroup->pFiltersHeadR3;
     while (pFilter)
     {
         bool fChoked = ASMAtomicXchgBool(&pFilter->fChoked, false);
         Log3((LOG_FN_FMT ": pFilter=%#p fChoked=%RTbool\n", __PRETTY_FUNCTION__, pFilter, fChoked));
-        if (fChoked && pFilter->pIDrvNet)
+        if (fChoked && pFilter->pIDrvNetR3)
         {
             LogFlowFunc(("Calling pfnXmitPending for pFilter=%#p\n", pFilter));
-            pFilter->pIDrvNet->pfnXmitPending(pFilter->pIDrvNet);
+            pFilter->pIDrvNetR3->pfnXmitPending(pFilter->pIDrvNetR3);
         }
 
-        pFilter = pFilter->pNext;
-    }
-
-    //rc = RTCritSectLeave(&pBwGroup->cs); AssertRC(rc);
+        pFilter = pFilter->pNextR3;
+    }
+
+    //UNLOCK_NETSHAPER(pShaper);
 }
 
@@ -244 +252 @@
 {
     PPDMNSBWGROUP pBwGroup = pFilter->pBwGroupR3;
-    int rc = PDMCritSectEnter(&pBwGroup->cs, VERR_SEM_BUSY); AssertRC(rc);
-
-    pFilter->pNext = pBwGroup->pFiltersHead;
-    pBwGroup->pFiltersHead = pFilter;
-
-    rc = PDMCritSectLeave(&pBwGroup->cs); AssertRC(rc);
+    int rc = PDMCritSectEnter(&pBwGroup->Lock, VERR_SEM_BUSY); AssertRC(rc);
+
+    pFilter->pNextR3 = pBwGroup->pFiltersHeadR3;
+    pBwGroup->pFiltersHeadR3 = pFilter;
+
+    rc = PDMCritSectLeave(&pBwGroup->Lock); AssertRC(rc);
 }
 
@@ -262 +270 @@
      */
     AssertPtr(pBwGroup);
-    AssertPtr(pBwGroup->pShaper);
-    Assert(RTCritSectIsOwner(&pBwGroup->pShaper->cs));
-    int rc = PDMCritSectEnter(&pBwGroup->cs, VERR_SEM_BUSY); AssertRC(rc);
-
-    if (pFilter == pBwGroup->pFiltersHead)
-        pBwGroup->pFiltersHead = pFilter->pNext;
+    AssertPtr(pBwGroup->pShaperR3);
+    Assert(RTCritSectIsOwner(&pBwGroup->pShaperR3->Lock));
+    int rc = PDMCritSectEnter(&pBwGroup->Lock, VERR_SEM_BUSY); AssertRC(rc);
+
+    if (pFilter == pBwGroup->pFiltersHeadR3)
+        pBwGroup->pFiltersHeadR3 = pFilter->pNextR3;
     else
     {
-        PPDMNSFILTER pPrev = pBwGroup->pFiltersHead;
+        PPDMNSFILTER pPrev = pBwGroup->pFiltersHeadR3;
         while (   pPrev
-               && pPrev->pNext != pFilter)
-            pPrev = pPrev->pNext;
+               && pPrev->pNextR3 != pFilter)
+            pPrev = pPrev->pNextR3;
 
         AssertPtr(pPrev);
-        pPrev->pNext = pFilter->pNext;
-    }
-
-    rc = PDMCritSectLeave(&pBwGroup->cs); AssertRC(rc);
-}
-
-
-VMMR3DECL(int) PDMR3NsAttach(PVM pVM, PPDMDRVINS pDrvIns, const char *pcszBwGroup,
-                             PPDMNSFILTER pFilter)
-{
-    VM_ASSERT_EMT(pVM);
+        pPrev->pNextR3 = pFilter->pNextR3;
+    }
+
+    rc = PDMCritSectLeave(&pBwGroup->Lock); AssertRC(rc);
+}
+
+
+/**
+ * Attach network filter driver from bandwidth group.
+ *
+ * @returns VBox status code.
+ * @param   pVM             Handle of VM.
+ * @param   pDrvIns         The driver instance.
+ * @param   pszBwGroup     Name of the bandwidth group to attach to.
+ * @param   pFilter         Pointer to the filter we attach.
+ */
+VMMR3_INT_DECL(int) PDMR3NsAttach(PUVM pUVM, PPDMDRVINS pDrvIns, const char *pszBwGroup, PPDMNSFILTER pFilter)
+{
+    VM_ASSERT_EMT(pUVM->pVM);
     AssertPtrReturn(pFilter, VERR_INVALID_POINTER);
     AssertReturn(pFilter->pBwGroupR3 == NULL, VERR_ALREADY_EXISTS);
 
-
-    PUVM pUVM = pVM->pUVM;
     PPDMNETSHAPER pShaper = pUVM->pdm.s.pNetShaper;
-
-    PPDMNSBWGROUP pBwGroupOld = NULL;
-    PPDMNSBWGROUP pBwGroupNew = NULL;
-
-    int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
+    LOCK_NETSHAPER_RETURN(pShaper);
+
+    int             rc          = VINF_SUCCESS;
+    PPDMNSBWGROUP   pBwGroupNew = NULL;
+    if (pszBwGroup)
+    {
+        pBwGroupNew = pdmNsBwGroupFindById(pShaper, pszBwGroup);
+        if (pBwGroupNew)
+            pdmNsBwGroupRef(pBwGroupNew);
+        else
+            rc = VERR_NOT_FOUND;
+    }
+
     if (RT_SUCCESS(rc))
     {
-        if (pcszBwGroup)
-        {
-            pBwGroupNew = pdmNsBwGroupFindById(pShaper, pcszBwGroup);
-            if (pBwGroupNew)
-                pdmNsBwGroupRef(pBwGroupNew);
-            else
-                rc = VERR_NOT_FOUND;
-        }
-
+        PPDMNSBWGROUP pBwGroupOld = ASMAtomicXchgPtrT(&pFilter->pBwGroupR3, pBwGroupNew, PPDMNSBWGROUP);
+        ASMAtomicWritePtr(&pFilter->pBwGroupR0, MMHyperR3ToR0(pUVM->pVM, pBwGroupNew));
+        if (pBwGroupOld)
+            pdmNsBwGroupUnref(pBwGroupOld);
+        pdmNsFilterLink(pFilter);
+    }
+
+    UNLOCK_NETSHAPER(pShaper);
+    return rc;
+}
+
+
+/**
+ * Detach network filter driver from bandwidth group.
+ *
+ * @returns VBox status code.
+ * @param   pUVM            The user mode VM handle.
+ * @param   pDrvIns         The driver instance.
+ * @param   pFilter         Pointer to the filter we detach.
+ */
+VMMR3_INT_DECL(int) PDMR3NsDetach(PUVM pUVM, PPDMDRVINS pDrvIns, PPDMNSFILTER pFilter)
+{
+    VM_ASSERT_EMT(pUVM->pVM);
+    AssertPtrReturn(pFilter, VERR_INVALID_POINTER);
+    AssertPtrReturn(pFilter->pBwGroupR3, VERR_INVALID_POINTER);
+
+    PPDMNETSHAPER pShaper = pUVM->pdm.s.pNetShaper;
+    LOCK_NETSHAPER_RETURN(pShaper);
+
+    pdmNsFilterUnlink(pFilter);
+    PPDMNSBWGROUP pBwGroup = ASMAtomicXchgPtrT(&pFilter->pBwGroupR3, NULL, PPDMNSBWGROUP);
+    if (pBwGroup)
+        pdmNsBwGroupUnref(pBwGroup);
+
+    UNLOCK_NETSHAPER(pShaper);
+    return VINF_SUCCESS;
+}
+
+
+/**
+ * Adjusts the maximum rate for the bandwidth group.
+ *
+ * @returns VBox status code.
+ * @param   pUVM            The user mode VM handle.
+ * @param   pszBwGroup      Name of the bandwidth group to attach to.
+ * @param   cbPerSecMax     Maximum number of bytes per second to be transmitted.
+ */
+VMMR3DECL(int) PDMR3NsBwGroupSetLimit(PUVM pUVM, const char *pszBwGroup, uint64_t cbPerSecMax)
+{
+    UVM_ASSERT_VALID_EXT_RETURN(pUVM, VERR_INVALID_VM_HANDLE);
+    PPDMNETSHAPER pShaper = pUVM->pdm.s.pNetShaper;
+    LOCK_NETSHAPER_RETURN(pShaper);
+
+    int           rc;
+    PPDMNSBWGROUP pBwGroup = pdmNsBwGroupFindById(pShaper, pszBwGroup);
+    if (pBwGroup)
+    {
+        rc = PDMCritSectEnter(&pBwGroup->Lock, VERR_SEM_BUSY); AssertRC(rc);
         if (RT_SUCCESS(rc))
         {
-            pBwGroupOld = ASMAtomicXchgPtrT(&pFilter->pBwGroupR3, pBwGroupNew, PPDMNSBWGROUP);
-            ASMAtomicWritePtr(&pFilter->pBwGroupR0, MMHyperR3ToR0(pVM, pBwGroupNew));
-            if (pBwGroupOld)
-                pdmNsBwGroupUnref(pBwGroupOld);
-            pdmNsFilterLink(pFilter);
+            pdmNsBwGroupSetLimit(pBwGroup, cbPerSecMax);
+
+            /* Drop extra tokens */
+            if (pBwGroup->cbTokensLast > pBwGroup->cbBucket)
+                pBwGroup->cbTokensLast = pBwGroup->cbBucket;
+
+            int rc2 = PDMCritSectLeave(&pBwGroup->Lock); AssertRC(rc2);
         }
-        int rc2 = RTCritSectLeave(&pShaper->cs); AssertRC(rc2);
-    }
-
-    return rc;
-}
-
-
-VMMR3DECL(int) PDMR3NsDetach(PVM pVM, PPDMDRVINS pDrvIns, PPDMNSFILTER pFilter)
-{
-    VM_ASSERT_EMT(pVM);
-    AssertPtrReturn(pFilter, VERR_INVALID_POINTER);
-    AssertPtrReturn(pFilter->pBwGroupR3, VERR_INVALID_POINTER);
-
-    PUVM pUVM = pVM->pUVM;
-    PPDMNETSHAPER pShaper = pUVM->pdm.s.pNetShaper;
-
-    int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
-    if (RT_SUCCESS(rc))
-    {
-        pdmNsFilterUnlink(pFilter);
-        PPDMNSBWGROUP pBwGroup = NULL;
-        pBwGroup = ASMAtomicXchgPtrT(&pFilter->pBwGroupR3, NULL, PPDMNSBWGROUP);
-        if (pBwGroup)
-            pdmNsBwGroupUnref(pBwGroup);
-        int rc2 = RTCritSectLeave(&pShaper->cs); AssertRC(rc2);
-    }
-    return rc;
-}
-
-
-VMMR3DECL(bool) PDMR3NsAllocateBandwidth(PPDMNSFILTER pFilter, size_t cbTransfer)
-{
-    return pdmNsAllocateBandwidth(pFilter, cbTransfer);
-}
-
-
-VMMR3DECL(int) PDMR3NsBwGroupSetLimit(PVM pVM, const char *pcszBwGroup, uint64_t cbTransferPerSecMax)
-{
-    PUVM pUVM = pVM->pUVM;
-    PPDMNETSHAPER pShaper = pUVM->pdm.s.pNetShaper;
-
-    int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
-    if (RT_SUCCESS(rc))
-    {
-        PPDMNSBWGROUP pBwGroup = pdmNsBwGroupFindById(pShaper, pcszBwGroup);
-        if (pBwGroup)
-        {
-            rc = PDMCritSectEnter(&pBwGroup->cs, VERR_SEM_BUSY); AssertRC(rc);
-            pdmNsBwGroupSetLimit(pBwGroup, cbTransferPerSecMax);
-            /* Drop extra tokens */
-            if (pBwGroup->cbTokensLast > pBwGroup->cbBucketSize)
-                pBwGroup->cbTokensLast = pBwGroup->cbBucketSize;
-            rc = PDMCritSectLeave(&pBwGroup->cs); AssertRC(rc);
-        }
-        rc = RTCritSectLeave(&pShaper->cs); AssertRC(rc);
-    }
+    }
+    else
+        rc = VERR_NOT_FOUND;
+
+    UNLOCK_NETSHAPER(pShaper);
     return rc;
 }
@@ -391 +413 @@
     {
         RTThreadSleep(PDM_NETSHAPER_MAX_LATENCY);
+
         /* Go over all bandwidth groups/filters calling pfnXmitPending */
-        int rc = RTCritSectEnter(&pShaper->cs); AssertRC(rc);
+        LOCK_NETSHAPER(pShaper);
         PPDMNSBWGROUP pBwGroup = pShaper->pBwGroupsHead;
         while (pBwGroup)
         {
             pdmNsBwGroupXmitPending(pBwGroup);
-            pBwGroup = pBwGroup->pNext;
+            pBwGroup = pBwGroup->pNextR3;
         }
-        rc = RTCritSectLeave(&pShaper->cs); AssertRC(rc);
+        UNLOCK_NETSHAPER(pShaper);
     }
     return VINF_SUCCESS;
@@ -437 +460 @@
     {
         PPDMNSBWGROUP pFree = pBwGroup;
-        pBwGroup = pBwGroup->pNext;
+        pBwGroup = pBwGroup->pNextR3;
         pdmNsBwGroupTerminate(pFree);
+        MMR3HeapFree(pFree->pszNameR3);
         MMHyperFree(pVM, pFree);
     }
 
-    RTCritSectDelete(&pShaper->cs);
+    RTCritSectDelete(&pShaper->Lock);
     return VINF_SUCCESS;
 }
@@ -455 +479 @@
 int pdmR3NetShaperInit(PVM pVM)
 {
-    LogFlowFunc((": pVM=%p\n", pVM));
-
+    LogFlow(("pdmR3NetShaperInit: pVM=%p\n", pVM));
     VM_ASSERT_EMT(pVM);
-
-    PPDMNETSHAPER pNetShaper = NULL;
-
-    int rc = MMR3HeapAllocZEx(pVM, MM_TAG_PDM_NET_SHAPER,
-                              sizeof(PDMNETSHAPER),
-                              (void **)&pNetShaper);
+    PUVM pUVM = pVM->pUVM;
+    AssertMsgReturn(!pUVM->pdm.s.pNetShaper, ("Network shaper was already initialized\n"), VERR_WRONG_ORDER);
+
+    PPDMNETSHAPER pShaper;
+    int rc = MMR3HeapAllocZEx(pVM, MM_TAG_PDM_NET_SHAPER, sizeof(PDMNETSHAPER), (void **)&pShaper);
     if (RT_SUCCESS(rc))
     {
-        PCFGMNODE pCfgRoot      = CFGMR3GetRoot(pVM);
-        PCFGMNODE pCfgNetShaper = CFGMR3GetChild(CFGMR3GetChild(pCfgRoot, "PDM"), "NetworkShaper");
-
-        pNetShaper->pVM = pVM;
-        rc = RTCritSectInit(&pNetShaper->cs);
+        PCFGMNODE pCfgNetShaper = CFGMR3GetChild(CFGMR3GetChild(CFGMR3GetRoot(pVM), "PDM"), "NetworkShaper");
+
+        pShaper->pVM = pVM;
+        rc = RTCritSectInit(&pShaper->Lock);
         if (RT_SUCCESS(rc))
         {
             /* Create all bandwidth groups. */
             PCFGMNODE pCfgBwGrp = CFGMR3GetChild(pCfgNetShaper, "BwGroups");
-
             if (pCfgBwGrp)
             {
@@ -496 +516 @@
                         rc = CFGMR3QueryU64(pCur, "Max", &cbMax);
                     if (RT_SUCCESS(rc))
-                        rc = pdmNsBwGroupCreate(pNetShaper, pszBwGrpId, cbMax);
+                        rc = pdmNsBwGroupCreate(pShaper, pszBwGrpId, cbMax);
 
                     RTMemFree(pszBwGrpId);
@@ -507 +527 @@
             if (RT_SUCCESS(rc))
             {
-                PUVM pUVM = pVM->pUVM;
-                AssertMsg(!pUVM->pdm.s.pNetShaper, ("Network shaper was already initialized\n"));
-
-                char szDesc[64];
-                static unsigned s_iThread;
-
-                RTStrPrintf(szDesc, sizeof(szDesc), "PDMNsTx-%d", ++s_iThread);
-                rc = PDMR3ThreadCreate(pVM, &pNetShaper->hTxThread, pNetShaper,
-                                       pdmR3NsTxThread, pdmR3NsTxWakeUp, 0,
-                                       RTTHREADTYPE_IO, szDesc);
+                rc = PDMR3ThreadCreate(pVM, &pShaper->pTxThread, pShaper, pdmR3NsTxThread, pdmR3NsTxWakeUp,
+                                       0 /*cbStack*/, RTTHREADTYPE_IO, "PDMNsTx");
                 if (RT_SUCCESS(rc))
                 {
-                    pUVM->pdm.s.pNetShaper = pNetShaper;
+                    pUVM->pdm.s.pNetShaper = pShaper;
                     return VINF_SUCCESS;
                 }
             }
 
-            RTCritSectDelete(&pNetShaper->cs);
+            RTCritSectDelete(&pShaper->Lock);
         }
-        MMR3HeapFree(pNetShaper);
-    }
-
-    LogFlowFunc((": pVM=%p rc=%Rrc\n", pVM, rc));
+
+        MMR3HeapFree(pShaper);
+    }
+
+    LogFlow(("pdmR3NetShaperInit: pVM=%p rc=%Rrc\n", pVM, rc));
     return rc;
 }

trunk/src/VBox/VMM/include/PDMInternal.h

@@ -1086 +1086 @@
     R3PTRTYPE(PPDMASYNCCOMPLETIONTEMPLATE) pAsyncCompletionTemplates;
     /** @} */
+
+    /** Global block cache data. */
+    R3PTRTYPE(PPDMBLKCACHEGLOBAL)   pBlkCacheGlobal;
 #ifdef VBOX_WITH_NETSHAPER
     /** Pointer to network shaper instance. */
     R3PTRTYPE(PPDMNETSHAPER)        pNetShaper;
 #endif /* VBOX_WITH_NETSHAPER */
-
-    R3PTRTYPE(PPDMBLKCACHEGLOBAL)   pBlkCacheGlobal;
 
 } PDMUSERPERVM;

trunk/src/VBox/VMM/include/PDMNetShaperInternal.h

@@ -1 +1 @@
 /* $Id$ */
 /** @file
- * PDM Network Shaper - Internal data structures and functions common for both
- * R0 and R3 parts.
+ * PDM Network Shaper - Internal data structures and functions common for both R0 and R3 parts.
  */
 
 /*
- * Copyright (C) 2011-2012 Oracle Corporation
+ * Copyright (C) 2011-2013 Oracle Corporation
  *
  * This file is part of VirtualBox Open Source Edition (OSE), as
@@ -32 +31 @@
 {
     /** Pointer to the next group in the list. */
-    struct PDMNSBWGROUP                        *pNext;
+    R3PTRTYPE(struct PDMNSBWGROUP *)            pNextR3;
     /** Pointer to the shared UVM structure. */
-    struct PDMNETSHAPER                        *pShaper;
+    R3PTRTYPE(struct PDMNETSHAPER *)            pShaperR3;
     /** Critical section protecting all members below. */
-    PDMCRITSECT               cs;
+    PDMCRITSECT                                 Lock;
     /** Pointer to the first filter attached to this group. */
-    struct PDMNSFILTER                         *pFiltersHead;
+    R3PTRTYPE(struct PDMNSFILTER *)             pFiltersHeadR3;
     /** Bandwidth group name. */
-    char                                       *pszName;
+    R3PTRTYPE(char *)                           pszNameR3;
     /** Maximum number of bytes filters are allowed to transfer. */
-    volatile uint64_t                           cbTransferPerSecMax;
+    volatile uint64_t                           cbPerSecMax;
     /** Number of bytes we are allowed to transfer in one burst. */
-    volatile uint32_t                           cbBucketSize;
+    volatile uint32_t                           cbBucket;
     /** Number of bytes we were allowed to transfer at the last update. */
     volatile uint32_t                           cbTokensLast;
@@ -55 +54 @@
 typedef PDMNSBWGROUP *PPDMNSBWGROUP;
 
-DECLINLINE(bool) pdmNsAllocateBandwidth(PPDMNSFILTER pFilter, size_t cbTransfer)
-{
-    AssertPtrReturn(pFilter, true);
-    if (!VALID_PTR(pFilter->CTX_SUFF(pBwGroup)))
-        return true;
 
-    PPDMNSBWGROUP pBwGroup = ASMAtomicReadPtrT(&pFilter->CTX_SUFF(pBwGroup), PPDMNSBWGROUP);
-    int rc = PDMCritSectEnter(&pBwGroup->cs, VERR_SEM_BUSY); AssertRC(rc);
-    if (RT_UNLIKELY(rc == VERR_SEM_BUSY))
-        return true;
-    bool fAllowed = true;
-    if (pBwGroup->cbTransferPerSecMax)
-    {
-        /* Re-fill the bucket first */
-        uint64_t tsNow = RTTimeSystemNanoTS();
-        uint32_t uTokensAdded = (tsNow - pBwGroup->tsUpdatedLast)*pBwGroup->cbTransferPerSecMax/(1000*1000*1000);
-        uint32_t uTokens = RT_MIN(pBwGroup->cbBucketSize, uTokensAdded + pBwGroup->cbTokensLast);
-
-        if (cbTransfer > uTokens)
-        {
-            fAllowed = false;
-            ASMAtomicWriteBool(&pFilter->fChoked, true);
-        }
-        else
-        {
-            pBwGroup->tsUpdatedLast = tsNow;
-            pBwGroup->cbTokensLast = uTokens - (uint32_t)cbTransfer;
-        }
-        Log2((LOG_FN_FMT "BwGroup=%#p{%s} cbTransfer=%u uTokens=%u uTokensAdded=%u fAllowed=%RTbool\n",
-              __PRETTY_FUNCTION__, pBwGroup, pBwGroup->pszName, cbTransfer, uTokens, uTokensAdded, fAllowed));
-    }
-    else
-        Log2((LOG_FN_FMT "BwGroup=%#p{%s} disabled fAllowed=%RTbool\n",
-              __PRETTY_FUNCTION__, pBwGroup, pBwGroup->pszName, fAllowed));
-
-    rc = PDMCritSectLeave(&pBwGroup->cs); AssertRC(rc);
-    return fAllowed;
-}