/** @file
*
*  Copyright (c) 2015 - 2020, Hisilicon Limited. All rights reserved.
*  Copyright (c) 2015, Linaro Limited. All rights reserved.
*
*  SPDX-License-Identifier: BSD-2-Clause-Patent
*
**/

#include "AddSmbiosType9.h"

extern SMBIOS_TABLE_TYPE9 gPcieSlotInfo[];
extern UINT8 OemGetPcieSlotNumber ();

VOID
EFIAPI
UpdateSmbiosType9Info (
  IN OUT SMBIOS_TABLE_TYPE9             *Type9Record
)
{
  EFI_STATUS                         Status;
  UINTN                              HandleIndex;
  EFI_HANDLE                        *HandleBuffer;
  UINTN                              HandleCount;
  EFI_PCI_IO_PROTOCOL               *PciIo;
  UINTN                              SegmentNumber;
  UINTN                              BusNumber;
  UINTN                              DeviceNumber;
  UINTN                              FunctionNumber;
  UINTN                              Index;
  REPORT_PCIEDIDVID2BMC              ReportPcieDidVid[PCIEDEVICE_REPORT_MAX];

  GetPciDidVid ((VOID *)ReportPcieDidVid);

  Status = gBS->LocateHandleBuffer (
                  ByProtocol,
                  &gEfiPciIoProtocolGuid,
                  NULL,
                  &HandleCount,
                  &HandleBuffer
                  );
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_ERROR, " Locate gEfiPciIoProtocol Failed.\n"));
    gBS->FreePool ((VOID *)HandleBuffer);
    return;
  }

  for (HandleIndex = 0; HandleIndex < HandleCount; HandleIndex++) {
    Status = gBS->HandleProtocol (
                    HandleBuffer[HandleIndex],
                    &gEfiPciIoProtocolGuid,
                    (VOID **)&PciIo
                    );
    if (EFI_ERROR (Status)) {
      DEBUG ((DEBUG_ERROR, "[%a]:[%dL] Status : %r\n", __FUNCTION__, __LINE__, Status));
      continue;
    }
    (VOID)PciIo->GetLocation (PciIo, &SegmentNumber, &BusNumber, &DeviceNumber, &FunctionNumber);
    for (Index = 0; Index < sizeof(ReportPcieDidVid) / sizeof(REPORT_PCIEDIDVID2BMC); Index++) {
      if (Type9Record->SlotID == ReportPcieDidVid[Index].Slot + 1) {
        if ((BusNumber == ReportPcieDidVid[Index].Bus) && (DeviceNumber == ReportPcieDidVid[Index].Device)) {
          DEBUG ((DEBUG_ERROR, "PCIe device plot in slot Seg %d  bdf %d %d %d\r\n",
            SegmentNumber, BusNumber, DeviceNumber, FunctionNumber));
          Type9Record->SegmentGroupNum   = SegmentNumber;
          Type9Record->BusNum            = BusNumber;
          Type9Record->DevFuncNum        = (DeviceNumber << 3) | FunctionNumber;
          Type9Record->CurrentUsage      = SlotUsageInUse;
          break;
        }
      }
    }
  }

  gBS->FreePool ((VOID *)HandleBuffer);
  return;
}

STATIC
VOID
EmptySmbiosType9 (
  EFI_SMBIOS_PROTOCOL                 *Smbios
  )
{
  EFI_STATUS                          Status;
  EFI_SMBIOS_TYPE                     SmbiosType;
  EFI_SMBIOS_HANDLE                   SmbiosHandle;
  EFI_SMBIOS_TABLE_HEADER             *Record;

  do {
    SmbiosHandle = SMBIOS_HANDLE_PI_RESERVED;
    SmbiosType = EFI_SMBIOS_TYPE_SYSTEM_SLOTS;
    Status = Smbios->GetNext (Smbios, &SmbiosHandle, &SmbiosType, &Record, NULL);
    if (!EFI_ERROR (Status)) {
      Status = Smbios->Remove (Smbios, SmbiosHandle);
      if (EFI_ERROR (Status)) {
        DEBUG ((DEBUG_ERROR, "[%a]:[%dL] Remove System Slot Failed. Status : %r\n",
          __FUNCTION__, __LINE__, Status));
        break;
      }
    }
  } while (SmbiosHandle != SMBIOS_HANDLE_PI_RESERVED);

  return;
}

STATIC
EFI_STATUS
AddSmbiosType9Record (
  EFI_SMBIOS_PROTOCOL                 *Smbios,
  SMBIOS_TABLE_TYPE9                  *Type9Record
  )
{
  EFI_STATUS                          Status;
  EFI_SMBIOS_HANDLE                   SmbiosHandle;
  SMBIOS_TABLE_TYPE9                  *SmbiosRecord;
  CHAR8                               *OptionalStrStart;
  CHAR16                              SlotDesignation[SMBIOS_STRING_MAX_LENGTH];
  UINTN                               SlotStrLen;

  SlotStrLen = UnicodeSPrint (
                 SlotDesignation,
                 SMBIOS_STRING_MAX_LENGTH * 2,
                 L"PCIE Slot%d",
                 Type9Record->SlotID);

  //
  // Two zeros following the last string.
  //
  SmbiosRecord = AllocateZeroPool (sizeof (SMBIOS_TABLE_TYPE9) + SlotStrLen + 1 + 1);
  if (SmbiosRecord == NULL) {
    DEBUG ((DEBUG_ERROR, "AllocateZeroPool Failed for SmbiosRecord.\n"));
    return EFI_OUT_OF_RESOURCES;
  }

  (VOID)CopyMem (SmbiosRecord, Type9Record, sizeof (SMBIOS_TABLE_TYPE9));
  SmbiosRecord->Hdr.Length = sizeof (SMBIOS_TABLE_TYPE9);
  OptionalStrStart = (CHAR8 *)(SmbiosRecord + 1);
  Status = UnicodeStrToAsciiStrS (SlotDesignation, OptionalStrStart, SlotStrLen + 1);
  ASSERT_EFI_ERROR (Status);

  //
  // Now we have got the full smbios record, call smbios protocol to add this record.
  //
  SmbiosHandle = SMBIOS_HANDLE_PI_RESERVED;
  Status = Smbios->Add (Smbios, NULL, &SmbiosHandle, (EFI_SMBIOS_TABLE_HEADER *)SmbiosRecord);
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_ERROR, "Add Smbios Type09 Failed! %r\n", Status));
  }

  FreePool (SmbiosRecord);
  return Status;
}

STATIC
VOID
HandleSmbiosType9 (
  EFI_SMBIOS_PROTOCOL                 *Smbios
  )
{
  EFI_STATUS                          Status;
  SMBIOS_TABLE_TYPE9                  *Type9Record;
  UINT8                               RecordCount;
  UINT8                               Index;

  RecordCount = OemGetPcieSlotNumber ();
  if (RecordCount == 0) {
    return;
  }

  EmptySmbiosType9 (Smbios);
  Status = EFI_SUCCESS;
  for (Index = 0; Index < RecordCount; Index++) {
    if (gPcieSlotInfo[Index].Hdr.Type != EFI_SMBIOS_TYPE_SYSTEM_SLOTS) {
      continue;
    }

    Type9Record = &gPcieSlotInfo[Index];

    UpdateSmbiosType9Info (Type9Record);

    Status = AddSmbiosType9Record (Smbios, Type9Record);
    if (EFI_ERROR (Status)) {
      break;
    }
  }

  return;
}

EFI_STATUS
EFIAPI
AddSmbiosType9Entry (
  IN EFI_HANDLE            ImageHandle,
  IN EFI_SYSTEM_TABLE      *SystemTable
  )
{
  EFI_STATUS                          Status;
  EFI_SMBIOS_PROTOCOL                 *Smbios;

  Status = gBS->LocateProtocol (
                  &gEfiSmbiosProtocolGuid,
                  NULL,
                  (VOID **) &Smbios
                  );
  if (EFI_ERROR (Status)) {
      DEBUG ((DEBUG_ERROR, "[%a]:[%dL] LocateProtocol Failed. Status : %r\n",
        __FUNCTION__, __LINE__, Status));
      return Status;
  }

  HandleSmbiosType9 (Smbios);

  return Status;
}