]> git.proxmox.com Git - mirror_edk2.git/blobdiff - OvmfPkg/PvScsiDxe/PvScsi.c
OvmfPkg/PvScsiDxe: Backup/Restore PCI attributes on Init/UnInit
[mirror_edk2.git] / OvmfPkg / PvScsiDxe / PvScsi.c
index 04c08036b799be6adfb74f85ca179a3f98f90061..5566b4cce46725743ce2f2aa2981278a5ceac06c 100644 (file)
@@ -11,6 +11,7 @@
 \r
 #include <IndustryStandard/Pci.h>\r
 #include <IndustryStandard/PvScsi.h>\r
+#include <Library/BaseMemoryLib.h>\r
 #include <Library/MemoryAllocationLib.h>\r
 #include <Library/UefiBootServicesTableLib.h>\r
 #include <Library/UefiLib.h>\r
 //\r
 #define PVSCSI_BINDING_VERSION      0x10\r
 \r
+//\r
+// Ext SCSI Pass Thru utilities\r
+//\r
+\r
+/**\r
+  Check if Target argument to EXT_SCSI_PASS_THRU.GetNextTarget() and\r
+  EXT_SCSI_PASS_THRU.GetNextTargetLun() is initialized\r
+**/\r
+STATIC\r
+BOOLEAN\r
+IsTargetInitialized (\r
+  IN UINT8                                          *Target\r
+  )\r
+{\r
+  UINTN Idx;\r
+\r
+  for (Idx = 0; Idx < TARGET_MAX_BYTES; ++Idx) {\r
+    if (Target[Idx] != 0xFF) {\r
+      return TRUE;\r
+    }\r
+  }\r
+  return FALSE;\r
+}\r
+\r
 //\r
 // Ext SCSI Pass Thru\r
 //\r
@@ -52,7 +77,54 @@ PvScsiGetNextTargetLun (
   IN OUT UINT64                                     *Lun\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\r
+  UINT8      *TargetPtr;\r
+  UINT8      LastTarget;\r
+  PVSCSI_DEV *Dev;\r
+\r
+  if (Target == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  //\r
+  // The Target input parameter is unnecessarily a pointer-to-pointer\r
+  //\r
+  TargetPtr = *Target;\r
+\r
+  //\r
+  // If target not initialized, return first target & LUN\r
+  //\r
+  if (!IsTargetInitialized (TargetPtr)) {\r
+    ZeroMem (TargetPtr, TARGET_MAX_BYTES);\r
+    *Lun = 0;\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  //\r
+  // We only use first byte of target identifer\r
+  //\r
+  LastTarget = *TargetPtr;\r
+\r
+  //\r
+  // Increment (target, LUN) pair if valid on input\r
+  //\r
+  Dev = PVSCSI_FROM_PASS_THRU (This);\r
+  if (LastTarget > Dev->MaxTarget || *Lun > Dev->MaxLun) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  if (*Lun < Dev->MaxLun) {\r
+    ++*Lun;\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  if (LastTarget < Dev->MaxTarget) {\r
+    *Lun = 0;\r
+    ++LastTarget;\r
+    *TargetPtr = LastTarget;\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  return EFI_NOT_FOUND;\r
 }\r
 \r
 STATIC\r
@@ -65,7 +137,38 @@ PvScsiBuildDevicePath (
   IN OUT EFI_DEVICE_PATH_PROTOCOL                   **DevicePath\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\r
+  UINT8             TargetValue;\r
+  PVSCSI_DEV        *Dev;\r
+  SCSI_DEVICE_PATH  *ScsiDevicePath;\r
+\r
+  if (DevicePath == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  //\r
+  // We only use first byte of target identifer\r
+  //\r
+  TargetValue = *Target;\r
+\r
+  Dev = PVSCSI_FROM_PASS_THRU (This);\r
+  if (TargetValue > Dev->MaxTarget || Lun > Dev->MaxLun) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  ScsiDevicePath = AllocatePool (sizeof (*ScsiDevicePath));\r
+  if (ScsiDevicePath == NULL) {\r
+    return EFI_OUT_OF_RESOURCES;\r
+  }\r
+\r
+  ScsiDevicePath->Header.Type      = MESSAGING_DEVICE_PATH;\r
+  ScsiDevicePath->Header.SubType   = MSG_SCSI_DP;\r
+  ScsiDevicePath->Header.Length[0] = (UINT8)sizeof (*ScsiDevicePath);\r
+  ScsiDevicePath->Header.Length[1] = (UINT8)(sizeof (*ScsiDevicePath) >> 8);\r
+  ScsiDevicePath->Pun              = TargetValue;\r
+  ScsiDevicePath->Lun              = (UINT16)Lun;\r
+\r
+  *DevicePath = &ScsiDevicePath->Header;\r
+  return EFI_SUCCESS;\r
 }\r
 \r
 STATIC\r
@@ -78,7 +181,33 @@ PvScsiGetTargetLun (
   OUT UINT64                                        *Lun\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\r
+  SCSI_DEVICE_PATH *ScsiDevicePath;\r
+  PVSCSI_DEV       *Dev;\r
+\r
+  if (DevicePath == NULL || Target == NULL || *Target == NULL || Lun == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  if (DevicePath->Type    != MESSAGING_DEVICE_PATH ||\r
+      DevicePath->SubType != MSG_SCSI_DP) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+\r
+  ScsiDevicePath = (SCSI_DEVICE_PATH *)DevicePath;\r
+  Dev = PVSCSI_FROM_PASS_THRU (This);\r
+  if (ScsiDevicePath->Pun > Dev->MaxTarget ||\r
+      ScsiDevicePath->Lun > Dev->MaxLun) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  //\r
+  // We only use first byte of target identifer\r
+  //\r
+  **Target = (UINT8)ScsiDevicePath->Pun;\r
+  ZeroMem (*Target + 1, TARGET_MAX_BYTES - 1);\r
+  *Lun = ScsiDevicePath->Lun;\r
+\r
+  return EFI_SUCCESS;\r
 }\r
 \r
 STATIC\r
@@ -111,7 +240,89 @@ PvScsiGetNextTarget (
   IN OUT UINT8                                      **Target\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\r
+  UINT8      *TargetPtr;\r
+  UINT8      LastTarget;\r
+  PVSCSI_DEV *Dev;\r
+\r
+  if (Target == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  //\r
+  // The Target input parameter is unnecessarily a pointer-to-pointer\r
+  //\r
+  TargetPtr = *Target;\r
+\r
+  //\r
+  // If target not initialized, return first target\r
+  //\r
+  if (!IsTargetInitialized (TargetPtr)) {\r
+    ZeroMem (TargetPtr, TARGET_MAX_BYTES);\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  //\r
+  // We only use first byte of target identifer\r
+  //\r
+  LastTarget = *TargetPtr;\r
+\r
+  //\r
+  // Increment target if valid on input\r
+  //\r
+  Dev = PVSCSI_FROM_PASS_THRU (This);\r
+  if (LastTarget > Dev->MaxTarget) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  if (LastTarget < Dev->MaxTarget) {\r
+    ++LastTarget;\r
+    *TargetPtr = LastTarget;\r
+    return EFI_SUCCESS;\r
+  }\r
+\r
+  return EFI_NOT_FOUND;\r
+}\r
+\r
+STATIC\r
+EFI_STATUS\r
+PvScsiSetPciAttributes (\r
+  IN OUT PVSCSI_DEV *Dev\r
+  )\r
+{\r
+  EFI_STATUS Status;\r
+\r
+  //\r
+  // Backup original PCI Attributes\r
+  //\r
+  Status = Dev->PciIo->Attributes (\r
+                         Dev->PciIo,\r
+                         EfiPciIoAttributeOperationGet,\r
+                         0,\r
+                         &Dev->OriginalPciAttributes\r
+                         );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  //\r
+  // TODO: Change PCI Attributes\r
+  //\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+STATIC\r
+VOID\r
+PvScsiRestorePciAttributes (\r
+  IN PVSCSI_DEV *Dev\r
+  )\r
+{\r
+  Dev->PciIo->Attributes (\r
+                Dev->PciIo,\r
+                EfiPciIoAttributeOperationSet,\r
+                Dev->OriginalPciAttributes,\r
+                NULL\r
+                );\r
 }\r
 \r
 STATIC\r
@@ -120,6 +331,22 @@ PvScsiInit (
   IN OUT PVSCSI_DEV *Dev\r
   )\r
 {\r
+  EFI_STATUS Status;\r
+\r
+  //\r
+  // Init configuration\r
+  //\r
+  Dev->MaxTarget = PcdGet8 (PcdPvScsiMaxTargetLimit);\r
+  Dev->MaxLun = PcdGet8 (PcdPvScsiMaxLunLimit);\r
+\r
+  //\r
+  // Set PCI Attributes\r
+  //\r
+  Status = PvScsiSetPciAttributes (Dev);\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
   //\r
   // Populate the exported interface's attributes\r
   //\r
@@ -158,7 +385,7 @@ PvScsiUninit (
   IN OUT PVSCSI_DEV *Dev\r
   )\r
 {\r
-  // Currently nothing to do here\r
+  PvScsiRestorePciAttributes (Dev);\r
 }\r
 \r
 //\r
@@ -237,11 +464,23 @@ PvScsiDriverBindingStart (
     return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
-  Status = PvScsiInit (Dev);\r
+  Status = gBS->OpenProtocol (\r
+                  ControllerHandle,\r
+                  &gEfiPciIoProtocolGuid,\r
+                  (VOID **)&Dev->PciIo,\r
+                  This->DriverBindingHandle,\r
+                  ControllerHandle,\r
+                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
+                  );\r
   if (EFI_ERROR (Status)) {\r
     goto FreePvScsi;\r
   }\r
 \r
+  Status = PvScsiInit (Dev);\r
+  if (EFI_ERROR (Status)) {\r
+    goto ClosePciIo;\r
+  }\r
+\r
   //\r
   // Setup complete, attempt to export the driver instance's PassThru interface\r
   //\r
@@ -261,6 +500,14 @@ PvScsiDriverBindingStart (
 UninitDev:\r
   PvScsiUninit (Dev);\r
 \r
+ClosePciIo:\r
+  gBS->CloseProtocol (\r
+         ControllerHandle,\r
+         &gEfiPciIoProtocolGuid,\r
+         This->DriverBindingHandle,\r
+         ControllerHandle\r
+         );\r
+\r
 FreePvScsi:\r
   FreePool (Dev);\r
 \r
@@ -306,6 +553,13 @@ PvScsiDriverBindingStop (
 \r
   PvScsiUninit (Dev);\r
 \r
+  gBS->CloseProtocol (\r
+         ControllerHandle,\r
+         &gEfiPciIoProtocolGuid,\r
+         This->DriverBindingHandle,\r
+         ControllerHandle\r
+         );\r
+\r
   FreePool (Dev);\r
 \r
   return EFI_SUCCESS;\r