OvmfPkg: Make the VirtIo devices use the new VIRTIO_DEVICE_PROTOCOL

[mirror_edk2.git] / OvmfPkg / VirtioNetDxe / SnpTransmit.c

diff --git a/OvmfPkg/VirtioNetDxe/SnpTransmit.c b/OvmfPkg/VirtioNetDxe/SnpTransmit.c
index ff922ca021646200ee5acc6c71608865280a79ee..7ca40d5d06504b75e870aeab68c33cf3d9f047fc 100644 (file)
--- a/OvmfPkg/VirtioNetDxe/SnpTransmit.c
+++ b/OvmfPkg/VirtioNetDxe/SnpTransmit.c
@@ -127,15 +127,15 @@ VirtioNetTransmit (
       goto Exit;\r
     }\r
     Ptr = Buffer;\r
-    ASSERT (SIZE_OF_VNET (VhdrMac) <= sizeof (EFI_MAC_ADDRESS));\r
+    ASSERT (SIZE_OF_VNET (Mac) <= sizeof (EFI_MAC_ADDRESS));\r
 \r
-    CopyMem (Ptr, DestAddr, SIZE_OF_VNET (VhdrMac));\r
-    Ptr += SIZE_OF_VNET (VhdrMac);\r
+    CopyMem (Ptr, DestAddr, SIZE_OF_VNET (Mac));\r
+    Ptr += SIZE_OF_VNET (Mac);\r
 \r
     CopyMem (Ptr,\r
       (SrcAddr == NULL) ? &Dev->Snm.CurrentAddress : SrcAddr,\r
-      SIZE_OF_VNET (VhdrMac));\r
-    Ptr += SIZE_OF_VNET (VhdrMac);\r
+      SIZE_OF_VNET (Mac));\r
+    Ptr += SIZE_OF_VNET (Mac);\r
 \r
     *Ptr++ = (UINT8) (*Protocol >> 8);\r
     *Ptr++ = (UINT8) *Protocol;\r
@@ -161,7 +161,7 @@ VirtioNetTransmit (
   *Dev->TxRing.Avail.Idx = AvailIdx;\r
 \r
   MemoryFence ();\r
-  Status = VIRTIO_CFG_WRITE (Dev, Generic.VhdrQueueNotify, VIRTIO_NET_Q_TX);\r
+  Status = Dev->VirtIo->SetQueueNotify (Dev->VirtIo, VIRTIO_NET_Q_TX);\r
 \r
 Exit:\r
   gBS->RestoreTPL (OldTpl);\r