OvmfPkg/XenBusDxe: Introduce XenBus support itself.

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

diff --git a/OvmfPkg/XenBusDxe/XenBusDxe.c b/OvmfPkg/XenBusDxe/XenBusDxe.c
index 679fe3b592e5274d13d0a2aa2d954d50c3e4bf68..7a7fd82d559d1a1604d8b4fa4ddc82798c72840d 100644 (file)
--- a/OvmfPkg/XenBusDxe/XenBusDxe.c
+++ b/OvmfPkg/XenBusDxe/XenBusDxe.c
@@ -32,6 +32,7 @@
 #include "XenHypercall.h"\r
 #include "GrantTable.h"\r
 #include "XenStore.h"\r
+#include "XenBus.h"\r
 \r
 \r
 ///\r
@@ -286,6 +287,7 @@ XenBusDxeDriverBindingStart (
   EFI_PCI_IO_PROTOCOL *PciIo;\r
   EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *BarDesc;\r
   UINT64 MmioAddr;\r
+  EFI_DEVICE_PATH_PROTOCOL *DevicePath;\r
 \r
   Status = gBS->OpenProtocol (\r
                      ControllerHandle,\r
@@ -299,11 +301,26 @@ XenBusDxeDriverBindingStart (
     return Status;\r
   }\r
 \r
+  Status = gBS->OpenProtocol (\r
+                  ControllerHandle,\r
+                  &gEfiDevicePathProtocolGuid,\r
+                  (VOID **) &DevicePath,\r
+                  This->DriverBindingHandle,\r
+                  ControllerHandle,\r
+                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
+                  );\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    goto ErrorOpenningProtocol;\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
+  Dev->DevicePath = DevicePath;\r
+  InitializeListHead (&Dev->ChildList);\r
 \r
   EfiAcquireLock (&mMyDeviceLock);\r
   if (mMyDevice != NULL) {\r
@@ -350,6 +367,8 @@ XenBusDxeDriverBindingStart (
   Status = XenStoreInit (Dev);\r
   ASSERT_EFI_ERROR (Status);\r
 \r
+  XenBusEnumerateBus (Dev);\r
+\r
   Status = gBS->CreateEvent (EVT_SIGNAL_EXIT_BOOT_SERVICES, TPL_CALLBACK,\r
                              NotifyExitBoot,\r
                              (VOID*) Dev,\r
@@ -360,6 +379,9 @@ XenBusDxeDriverBindingStart (
 \r
 ErrorAllocated:\r
   FreePool (Dev);\r
+  gBS->CloseProtocol (ControllerHandle, &gEfiDevicePathProtocolGuid,\r
+                      This->DriverBindingHandle, ControllerHandle);\r
+ErrorOpenningProtocol:\r
   gBS->CloseProtocol (ControllerHandle, &gEfiPciIoProtocolGuid,\r
                       This->DriverBindingHandle, ControllerHandle);\r
   return Status;\r
@@ -400,12 +422,56 @@ XenBusDxeDriverBindingStop (
   IN EFI_HANDLE                   *ChildHandleBuffer OPTIONAL\r
   )\r
 {\r
+  UINTN Index;\r
+  XENBUS_PROTOCOL *XenBusIo;\r
+  XENBUS_PRIVATE_DATA *ChildData;\r
+  EFI_STATUS Status;\r
   XENBUS_DEVICE *Dev = mMyDevice;\r
 \r
+  for (Index = 0; Index < NumberOfChildren; Index++) {\r
+    Status = gBS->OpenProtocol (\r
+               ChildHandleBuffer[Index],\r
+               &gXenBusProtocolGuid,\r
+               (VOID **) &XenBusIo,\r
+               This->DriverBindingHandle,\r
+               ControllerHandle,\r
+               EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
+    if (EFI_ERROR (Status)) {\r
+      DEBUG ((EFI_D_ERROR, "XenBusDxe: get children protocol failed: %r\n", Status));\r
+      continue;\r
+    }\r
+    ChildData = XENBUS_PRIVATE_DATA_FROM_THIS (XenBusIo);\r
+    Status = gBS->DisconnectController (ChildData->Handle, NULL, NULL);\r
+    if (EFI_ERROR (Status)) {\r
+      DEBUG ((EFI_D_ERROR, "XenBusDxe: error disconnecting child: %r\n",\r
+              Status));\r
+      continue;\r
+    }\r
+\r
+    Status = gBS->UninstallMultipleProtocolInterfaces (\r
+               ChildData->Handle,\r
+               &gEfiDevicePathProtocolGuid, ChildData->DevicePath,\r
+               &gXenBusProtocolGuid, &ChildData->XenBusIo,\r
+               NULL);\r
+    ASSERT_EFI_ERROR (Status);\r
+\r
+    FreePool ((VOID*)ChildData->XenBusIo.Type);\r
+    FreePool ((VOID*)ChildData->XenBusIo.Node);\r
+    FreePool ((VOID*)ChildData->XenBusIo.Backend);\r
+    FreePool (ChildData->DevicePath);\r
+    RemoveEntryList (&ChildData->Link);\r
+    FreePool (ChildData);\r
+  }\r
+  if (NumberOfChildren > 0) {\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
   gBS->CloseEvent (Dev->ExitBootEvent);\r
   XenStoreDeinit (Dev);\r
   XenGrantTableDeinit (Dev);\r
 \r
+  gBS->CloseProtocol (ControllerHandle, &gEfiDevicePathProtocolGuid,\r
+         This->DriverBindingHandle, ControllerHandle);\r
   gBS->CloseProtocol (ControllerHandle, &gEfiPciIoProtocolGuid,\r
          This->DriverBindingHandle, ControllerHandle);\r
 \r