1 files changed, 141 insertions, 0 deletions
diff --git a/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/ArmJunoDxe.c b/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/ArmJunoDxe.c
index b97f044004..333872ef05 100644
--- a/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/ArmJunoDxe.c
+++ b/ArmPlatformPkg/ArmJunoPkg/Drivers/ArmJunoDxe/ArmJunoDxe.c
@@ -17,6 +17,8 @@
 
 #include <Protocol/DevicePathFromText.h>
 #include <Protocol/PciRootBridgeIo.h>
+#include <Protocol/PciIo.h>
+#include <IndustryStandard/Pci.h>
 
 #include <Guid/EventGroup.h>
 #include <Guid/GlobalVariable.h>
@@ -68,6 +70,142 @@ STATIC CONST EFI_PCI_ROOT_BRIDGE_DEVICE_PATH mPciRootComplexDevicePath = {
 
 EFI_EVENT mAcpiRegistration = NULL;
 
+UINT32 SwapUINT32(UINT32 value)
+{
+  value = ((value << 8) & 0xFF00FF00 ) | ((value >> 8) & 0xFF00FF );
+  return (value << 16) | (value >> 16);
+}
+
+/**
+  The function reads MAC address from Juno IOFPGA registers and writes it
+  into Marvell Yukon NIC.
+**/
+STATIC
+EFI_STATUS
+ArmJunoSetNetworkMAC()
+{
+
+  EFI_STATUS                          Status;
+  UINTN                               HandleCount;
+  EFI_HANDLE                          *HandleBuffer;
+  UINTN                               HIndex;
+  EFI_PCI_IO_PROTOCOL*                PciIo;
+  UINT64                              PciID;
+  UINT32                              MacHigh;
+  UINT32                              MacLow;
+  UINT32                              PciRegBase;
+  UINT64                              OldPciAttributes;
+  UINT64                              AttrSupports;
+  UINT8                               *PciBarAttributes;
+
+  Status = gBS->LocateHandleBuffer (ByProtocol,
+                                    &gEfiPciIoProtocolGuid,
+                                    NULL, &HandleCount, &HandleBuffer);
+
+  if (!EFI_ERROR (Status)) {
+    for (HIndex = 0; HIndex < HandleCount; ++HIndex) {
+      Status = gBS->OpenProtocol (
+                      HandleBuffer[HIndex],
+                      &gEfiPciIoProtocolGuid,
+                      (VOID **) &PciIo,
+                      NULL,
+                      NULL,
+                      EFI_OPEN_PROTOCOL_GET_PROTOCOL);
+
+      if (EFI_ERROR (Status)) {
+        continue;
+      }
+
+      Status = PciIo->Pci.Read (
+            PciIo,
+            EfiPciIoWidthUint32,
+            PCI_VENDOR_ID_OFFSET,
+            1,
+            &PciID
+            );
+
+      if (EFI_ERROR (Status)) {
+        continue;
+      }
+
+      if ((PciID & 0xFFFFFFFF) == JUNO_MARVELL_YUKON_ID) {
+
+        // Read MAC address from IOFPGA
+        MacHigh= MmioRead32 (ARM_JUNO_SYS_PCIGBE_H);
+        MacLow = MmioRead32 (ARM_JUNO_SYS_PCIGBE_L);
+
+        Status = PciIo->Attributes (
+              PciIo,
+              EfiPciIoAttributeOperationGet,
+              0,
+              &OldPciAttributes
+              );
+
+        if (EFI_ERROR (Status)) {
+          continue;
+        }
+
+        Status = PciIo->Attributes (
+              PciIo,
+              EfiPciIoAttributeOperationSupported,
+              0,
+              &AttrSupports
+              );
+
+        if (!EFI_ERROR (Status)) {
+          AttrSupports &= EFI_PCI_DEVICE_ENABLE;
+          Status = PciIo->Attributes (
+                PciIo,
+                EfiPciIoAttributeOperationEnable,
+                AttrSupports,
+                NULL
+                );
+
+          Status = PciIo->GetBarAttributes (PciIo, 0, &AttrSupports, (VOID**)&PciBarAttributes);
+          if (!EFI_ERROR (Status) && (((EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *)PciBarAttributes)->Desc == ACPI_ADDRESS_SPACE_DESCRIPTOR)) {
+            if (((EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *)PciBarAttributes)->ResType == ACPI_ADDRESS_SPACE_TYPE_MEM) {
+              if (!(((EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *)PciBarAttributes)->SpecificFlag & ACPI_SPECFLAG_PREFETCHABLE)) {
+                PciRegBase = ((EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR *)PciBarAttributes)->AddrRangeMin;
+
+                // Clear Software Reset
+                MmioWrite16 (PciRegBase + R_CONTROL_STATUS, CS_RESET_CLR);
+
+                // Convert to Marvell MAC Address register format
+                MacHigh = SwapUINT32 ((MacHigh & 0xFFFF) << 16 |
+                                     (MacLow & 0xFFFF0000) >> 16);
+                MacLow = SwapUINT32 (MacLow) >> 16;
+
+                // Set MAC Address
+                MmioWrite8 (PciRegBase + R_TST_CTRL_1, TST_CFG_WRITE_ENABLE);
+                MmioWrite32 (PciRegBase + R_MAC, MacHigh);
+                MmioWrite32 (PciRegBase + R_MAC_MAINT, MacHigh);
+                MmioWrite32 (PciRegBase + R_MAC + 4, MacLow);
+                MmioWrite32 (PciRegBase + R_MAC_MAINT + 4, MacLow);
+                MmioWrite8 (PciRegBase + R_TST_CTRL_1, TST_CFG_WRITE_DISABLE);
+
+                // Soft reset
+                MmioWrite16 (PciRegBase + R_CONTROL_STATUS, CS_RESET_SET);
+                MmioWrite16 (PciRegBase + R_CONTROL_STATUS, CS_RESET_CLR);
+
+                Status = EFI_SUCCESS;
+              }
+            }
+          }
+
+          PciIo->Attributes (
+                PciIo,
+                EfiPciIoAttributeOperationSet,
+                OldPciAttributes,
+                NULL
+                );
+        }
+      }
+    }
+  }
+
+  return Status;
+}
+
 /**
   Notification function of the event defined as belonging to the
   EFI_END_OF_DXE_EVENT_GROUP_GUID event group that was created in
@@ -106,6 +244,9 @@ OnEndOfDxe (
 
   Status = gBS->ConnectController (Handle, NULL, PciRootComplexDevicePath, FALSE);
   ASSERT_EFI_ERROR (Status);
+
+  Status = ArmJunoSetNetworkMAC();
+  ASSERT_EFI_ERROR (Status);
 }
 
 STATIC