OvmfPkg/VirtioMmioDeviceLib: list "VirtioMmioDevice.h" in the INF file

[mirror_edk2.git] / OvmfPkg / Library / VirtioMmioDeviceLib / VirtioMmioDevice.c

diff --git a/OvmfPkg/Library/VirtioMmioDeviceLib/VirtioMmioDevice.c b/OvmfPkg/Library/VirtioMmioDeviceLib/VirtioMmioDevice.c
index 4af9dd0ac2964874d5bb68f369967ac974545a8f..fce934e1e9532c316aaa3c33a930b6880e4e120d 100644 (file)
--- a/OvmfPkg/Library/VirtioMmioDeviceLib/VirtioMmioDevice.c
+++ b/OvmfPkg/Library/VirtioMmioDeviceLib/VirtioMmioDevice.c
@@ -3,6 +3,7 @@
   This driver produces Virtio Device Protocol instances for Virtio Mmio devices.\r
 \r
   Copyright (C) 2013, ARM Ltd.\r
+  Copyright (C) 2017, AMD Inc. All rights reserved.<BR>\r
 \r
   This program and the accompanying materials are licensed and made available\r
   under the terms and conditions of the BSD License which accompanies this\r
@@ -15,7 +16,6 @@
 **/\r
 \r
 #include <Library/BaseMemoryLib.h>\r
-#include <Library/MemoryAllocationLib.h>\r
 #include <Library/UefiBootServicesTableLib.h>\r
 \r
 #include "VirtioMmioDevice.h"\r
@@ -25,7 +25,6 @@ static VIRTIO_DEVICE_PROTOCOL mMmioDeviceProtocolTemplate = {
     0,                                     // SubSystemDeviceId\r
     VirtioMmioGetDeviceFeatures,           // GetDeviceFeatures\r
     VirtioMmioSetGuestFeatures,            // SetGuestFeatures\r
-    VirtioMmioGetQueueAddress,             // GetQueueAddress\r
     VirtioMmioSetQueueAddress,             // SetQueueAddress\r
     VirtioMmioSetQueueSel,                 // SetQueueSel\r
     VirtioMmioSetQueueNotify,              // SetQueueNotify\r
@@ -36,7 +35,11 @@ static VIRTIO_DEVICE_PROTOCOL mMmioDeviceProtocolTemplate = {
     VirtioMmioGetDeviceStatus,             // GetDeviceStatus\r
     VirtioMmioSetDeviceStatus,             // SetDeviceStatus\r
     VirtioMmioDeviceWrite,                 // WriteDevice\r
-    VirtioMmioDeviceRead                   // ReadDevice\r
+    VirtioMmioDeviceRead,                  // ReadDevice\r
+    VirtioMmioAllocateSharedPages,         // AllocateSharedPages\r
+    VirtioMmioFreeSharedPages,             // FreeSharedPages\r
+    VirtioMmioMapSharedBuffer,             // MapSharedBuffer\r
+    VirtioMmioUnmapSharedBuffer            // UnmapSharedBuffer\r
 };\r
 \r
 /**\r