OvmfPkg/XenBusDxe: Add XenStore client implementation

[mirror_edk2.git] / OvmfPkg / XenBusDxe / XenBusDxe.c

diff --git a/OvmfPkg/XenBusDxe/XenBusDxe.c b/OvmfPkg/XenBusDxe/XenBusDxe.c
index 4c638b85f1abbe14aab3121a19da15e85035a69a..679fe3b592e5274d13d0a2aa2d954d50c3e4bf68 100644 (file)
--- a/OvmfPkg/XenBusDxe/XenBusDxe.c
+++ b/OvmfPkg/XenBusDxe/XenBusDxe.c
@@ -30,6 +30,8 @@
 #include "XenBusDxe.h"\r
 \r
 #include "XenHypercall.h"\r
+#include "GrantTable.h"\r
+#include "XenStore.h"\r
 \r
 \r
 ///\r
@@ -281,11 +283,27 @@ XenBusDxeDriverBindingStart (
 {\r
   EFI_STATUS Status;\r
   XENBUS_DEVICE *Dev;\r
+  EFI_PCI_IO_PROTOCOL *PciIo;\r
+  EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *BarDesc;\r
+  UINT64 MmioAddr;\r
+\r
+  Status = gBS->OpenProtocol (\r
+                     ControllerHandle,\r
+                     &gEfiPciIoProtocolGuid,\r
+                     (VOID **) &PciIo,\r
+                     This->DriverBindingHandle,\r
+                     ControllerHandle,\r
+                     EFI_OPEN_PROTOCOL_BY_DRIVER\r
+                     );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
 \r
   Dev = AllocateZeroPool (sizeof (*Dev));\r
   Dev->Signature = XENBUS_DEVICE_SIGNATURE;\r
   Dev->This = This;\r
   Dev->ControllerHandle = ControllerHandle;\r
+  Dev->PciIo = PciIo;\r
 \r
   EfiAcquireLock (&mMyDeviceLock);\r
   if (mMyDevice != NULL) {\r
@@ -299,6 +317,20 @@ XenBusDxeDriverBindingStart (
   mMyDevice = Dev;\r
   EfiReleaseLock (&mMyDeviceLock);\r
 \r
+  //\r
+  // The BAR1 of this PCI device is used for shared memory and is supposed to\r
+  // look like MMIO. The address space of the BAR1 will be used to map the\r
+  // Grant Table.\r
+  //\r
+  Status = PciIo->GetBarAttributes (PciIo, PCI_BAR_IDX1, NULL, (VOID**) &BarDesc);\r
+  ASSERT_EFI_ERROR (Status);\r
+  ASSERT (BarDesc->ResType == ACPI_ADDRESS_SPACE_TYPE_MEM);\r
+\r
+  /* Get a Memory address for mapping the Grant Table. */\r
+  DEBUG ((EFI_D_INFO, "XenBus: BAR at %LX\n", BarDesc->AddrRangeMin));\r
+  MmioAddr = BarDesc->AddrRangeMin;\r
+  FreePool (BarDesc);\r
+\r
   Status = XenHyperpageInit (Dev);\r
   if (EFI_ERROR (Status)) {\r
     DEBUG ((EFI_D_ERROR, "XenBus: Unable to retrieve the hyperpage.\n"));\r
@@ -313,6 +345,11 @@ XenBusDxeDriverBindingStart (
     goto ErrorAllocated;\r
   }\r
 \r
+  XenGrantTableInit (Dev, MmioAddr);\r
+\r
+  Status = XenStoreInit (Dev);\r
+  ASSERT_EFI_ERROR (Status);\r
+\r
   Status = gBS->CreateEvent (EVT_SIGNAL_EXIT_BOOT_SERVICES, TPL_CALLBACK,\r
                              NotifyExitBoot,\r
                              (VOID*) Dev,\r
@@ -323,6 +360,8 @@ XenBusDxeDriverBindingStart (
 \r
 ErrorAllocated:\r
   FreePool (Dev);\r
+  gBS->CloseProtocol (ControllerHandle, &gEfiPciIoProtocolGuid,\r
+                      This->DriverBindingHandle, ControllerHandle);\r
   return Status;\r
 }\r
 \r
@@ -364,6 +403,11 @@ XenBusDxeDriverBindingStop (
   XENBUS_DEVICE *Dev = mMyDevice;\r
 \r
   gBS->CloseEvent (Dev->ExitBootEvent);\r
+  XenStoreDeinit (Dev);\r
+  XenGrantTableDeinit (Dev);\r
+\r
+  gBS->CloseProtocol (ControllerHandle, &gEfiPciIoProtocolGuid,\r
+         This->DriverBindingHandle, ControllerHandle);\r
 \r
   mMyDevice = NULL;\r
   FreePool (Dev);\r