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 7f297bdeaea35308c98ce25a45d55ca5f0ae24f7..679fe3b592e5274d13d0a2aa2d954d50c3e4bf68 100644 (file)
--- a/OvmfPkg/XenBusDxe/XenBusDxe.c
+++ b/OvmfPkg/XenBusDxe/XenBusDxe.c
@@ -29,6 +29,10 @@
 \r
 #include "XenBusDxe.h"\r
 \r
+#include "XenHypercall.h"\r
+#include "GrantTable.h"\r
+#include "XenStore.h"\r
+\r
 \r
 ///\r
 /// Driver Binding Protocol instance\r
@@ -43,6 +47,9 @@ EFI_DRIVER_BINDING_PROTOCOL gXenBusDxeDriverBinding = {
 };\r
 \r
 \r
+STATIC EFI_LOCK       mMyDeviceLock = EFI_INITIALIZE_LOCK_VARIABLE (TPL_CALLBACK);\r
+STATIC XENBUS_DEVICE *mMyDevice = NULL;\r
+\r
 /**\r
   Unloads an image.\r
 \r
@@ -216,6 +223,19 @@ XenBusDxeDriverBindingSupported (
   return Status;\r
 }\r
 \r
+VOID\r
+EFIAPI\r
+NotifyExitBoot (\r
+  IN EFI_EVENT Event,\r
+  IN VOID *Context\r
+  )\r
+{\r
+  XENBUS_DEVICE *Dev = Context;\r
+\r
+  gBS->DisconnectController(Dev->ControllerHandle,\r
+                            Dev->This->DriverBindingHandle, NULL);\r
+}\r
+\r
 /**\r
   Starts a bus controller.\r
 \r
@@ -248,6 +268,8 @@ XenBusDxeDriverBindingSupported (
   @retval EFI_SUCCESS              The device was started.\r
   @retval EFI_DEVICE_ERROR         The device could not be started due to a device error.Currently not implemented.\r
   @retval EFI_OUT_OF_RESOURCES     The request could not be completed due to a lack of resources.\r
+  @retval EFI_UNSUPPORTED          Something is missing on the system that\r
+                                   prevent to start the edvice.\r
   @retval Others                   The driver failded to start the device.\r
 \r
 **/\r
@@ -259,7 +281,88 @@ XenBusDxeDriverBindingStart (
   IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath OPTIONAL\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\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
+    EfiReleaseLock (&mMyDeviceLock);\r
+    //\r
+    // There is already a XenBus running, only one can be used at a time.\r
+    //\r
+    Status = EFI_ALREADY_STARTED;\r
+    goto ErrorAllocated;\r
+  }\r
+  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
+    Status = EFI_UNSUPPORTED;\r
+    goto ErrorAllocated;\r
+  }\r
+\r
+  Status = XenGetSharedInfoPage (Dev);\r
+  if (EFI_ERROR (Status)) {\r
+    DEBUG ((EFI_D_ERROR, "XenBus: Unable to get the shared info page.\n"));\r
+    Status = EFI_UNSUPPORTED;\r
+    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
+                             &Dev->ExitBootEvent);\r
+  ASSERT_EFI_ERROR (Status);\r
+\r
+  return EFI_SUCCESS;\r
+\r
+ErrorAllocated:\r
+  FreePool (Dev);\r
+  gBS->CloseProtocol (ControllerHandle, &gEfiPciIoProtocolGuid,\r
+                      This->DriverBindingHandle, ControllerHandle);\r
+  return Status;\r
 }\r
 \r
 /**\r
@@ -297,5 +400,16 @@ XenBusDxeDriverBindingStop (
   IN EFI_HANDLE                   *ChildHandleBuffer OPTIONAL\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\r
+  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
+  return EFI_SUCCESS;\r
 }\r