Changeset 28208 in vbox for trunk/src/VBox

Timestamp:

Apr 12, 2010 2:07:33 PM (15 years ago)

Author:

vboxsync

Message:

intnet,VBoxNetFlt-linux: Added INTNETTRUNKSWPORT::pfnPreRecv for doing filtering and switching in difficult contexts. It currently returns broadcast all the time, but that will be addressed before long.

Location:

trunk/src/VBox

Files:

• Devices/Network/DrvIntNet.cpp (modified) (2 diffs)
• Devices/Network/SrvIntNetR0.cpp (modified) (2 diffs)
• HostDrivers/VBoxNetFlt/linux/VBoxNetFlt-linux.c (modified) (4 diffs)

trunk/src/VBox/Devices/Network/DrvIntNet.cpp

@@ -135 +135 @@
 #ifdef IN_RING3
 
-/* -=-=-=-=- PDMINETWORKUP -=-=-=-=- */
-
 
 /**
@@ -192 +190 @@
 }
 
+/* -=-=-=-=- PDMINETWORKUP -=-=-=-=- */
 
 /**

trunk/src/VBox/Devices/Network/SrvIntNetR0.cpp

@@ -3329 +3329 @@
 
 
+/** @copydoc INTNETTRUNKSWPORT::pfnPreRecv */
+static DECLCALLBACK(INTNETSWDECISION) intnetR0TrunkIfPortPreRecv(PINTNETTRUNKSWPORT pSwitchPort,
+                                                                 void const *pvSrc, size_t cbSrc, uint32_t fSrc)
+{
+    PINTNETTRUNKIF pThis = INTNET_SWITCHPORT_2_TRUNKIF(pSwitchPort);
+    PINTNETNETWORK pNetwork = pThis->pNetwork;
+
+    /* assert some sanity */
+    AssertPtrReturn(pNetwork, INTNETSWDECISION_TRUNK);
+    AssertReturn(pNetwork->FastMutex != NIL_RTSEMFASTMUTEX, INTNETSWDECISION_TRUNK);
+    AssertPtr(pvSrc);
+    AssertPtr(cbSrc >= 6);
+    Assert(fSrc);
+
+    /** @todo implement the switch table. */
+
+    return INTNETSWDECISION_BROADCAST;
+}
+
+
 /** @copydoc INTNETTRUNKSWPORT::pfnRecv */
 static DECLCALLBACK(bool) intnetR0TrunkIfPortRecv(PINTNETTRUNKSWPORT pSwitchPort, PINTNETSG pSG, uint32_t fSrc)
@@ -3612 +3632 @@
         return VERR_NO_MEMORY;
     pTrunkIF->SwitchPort.u32Version                 = INTNETTRUNKSWPORT_VERSION;
+    pTrunkIF->SwitchPort.pfnPreRecv                 = intnetR0TrunkIfPortPreRecv;
     pTrunkIF->SwitchPort.pfnRecv                    = intnetR0TrunkIfPortRecv;
     pTrunkIF->SwitchPort.pfnSGRetain                = intnetR0TrunkIfPortSGRetain;

trunk/src/VBox/HostDrivers/VBoxNetFlt/linux/VBoxNetFlt-linux.c

@@ -250 +250 @@
     /** Our overridden ops. */
     struct net_device_ops           Ops;
+    /** Magic word. */
+    uint32_t                        u32Magic;
     /** Pointer to the original ops. */
     struct net_device_ops const    *pOrgOps;
-    /** Magic word. */
-    uint32_t                        u32Magic;
+    /** Pointer to the net filter instance. */
+    PVBOXNETFLTINS                  pVBoxNetFlt;
     /** The number of filtered packages. */
     uint64_t                        cFiltered;
@@ -273 +275 @@
 {
     PVBOXNETDEVICEOPSOVERRIDE   pOverride = (PVBOXNETDEVICEOPSOVERRIDE)pDev->netdev_ops;
-    RTNETETHERHDR               EtherHdrBuf;
+    uint8_t                     abHdrBuf[sizeof(RTNETETHERHDR) + sizeof(uint32_t) + RTNETIPV4_MIN_LEN];
     PCRTNETETHERHDR             pEtherHdr;
+    PINTNETTRUNKSWPORT          pSwitchPort;
+
 
     /*
@@ -300 +304 @@
      *       drop it.
      */
-    pEtherHdr = (PCRTNETETHERHDR)skb_header_pointer(pSkb, 0, sizeof(EtherHdrBuf), &EtherHdrBuf);
-    if (    pEtherHdr
-        &&  pEtherHdr->DstMac.au8[0] == 0x08
-        &&  pEtherHdr->DstMac.au8[1] == 0x00
-        &&  pEtherHdr->DstMac.au8[2] == 0x27
+    pEtherHdr = (PCRTNETETHERHDR)skb_header_pointer(pSkb, 0, sizeof(abHdrBuf), &abHdrBuf[0]);
+    if (   pEtherHdr
+        && VALID_PTR(pOverride->pVBoxNetFlt)
+        && (pSwitchPort = pOverride->pVBoxNetFlt->pSwitchPort) != NULL
+        && VALID_PTR(pSwitchPort)
        )
     {
-        dev_kfree_skb(pSkb);
-        pOverride->cFiltered++;
-        return NETDEV_TX_OK;
+        INTNETSWDECISION    enmDecision;
+        uint32_t            cbHdrs = skb_headlen(pSkb);
+        cbHdrs = RT_MAX(cbHdrs, sizeof(abHdrBuf));
+
+        /** @todo consider reference counting, etc. */
+        enmDecision = pSwitchPort->pfnPreRecv(pSwitchPort, pEtherHdr, cbHdrs, INTNETTRUNKDIR_HOST);
+        if (enmDecision == INTNETSWDECISION_INTNET)
+        {
+            dev_kfree_skb(pSkb);
+            pOverride->cFiltered++;
+            return NETDEV_TX_OK;
+        }
     }
 
@@ -333 +346 @@
     pOverride->Ops.ndo_start_xmit   = vboxNetFltLinuxStartXmitFilter;
     pOverride->u32Magic             = VBOXNETDEVICEOPSOVERRIDE_MAGIC;
+    pOverride->cTotal               = 0;
+    pOverride->cFiltered            = 0;
+    pOverride->pVBoxNetFlt          = pThis;
 
     RTSpinlockAcquire(pThis->hSpinlock, &Tmp); /* (this isn't necessary, but so what) */