riscv: set fdtfile on VisionFive 2

[platform/kernel/u-boot.git] / board / starfive / visionfive2 / starfive_visionfive2.c

diff --git a/board/starfive/visionfive2/starfive_visionfive2.c b/board/starfive/visionfive2/starfive_visionfive2.c
index 07dcca2..19b7b44 100644 (file)
--- a/board/starfive/visionfive2/starfive_visionfive2.c
+++ b/board/starfive/visionfive2/starfive_visionfive2.c
@@ -5,14 +5,22 @@
  */
 
 #include <common.h>
-#include <asm/io.h>
-#include <asm/sections.h>
 #include <cpu_func.h>
 #include <dm.h>
+#include <fdt_support.h>
+#include <env.h>
+#include <asm/arch/eeprom.h>
+#include <asm/io.h>
+#include <asm/sections.h>
 #include <linux/bitops.h>
 
+DECLARE_GLOBAL_DATA_PTR;
 #define JH7110_L2_PREFETCHER_BASE_ADDR         0x2030000
 #define JH7110_L2_PREFETCHER_HART_OFFSET       0x2000
+#define FDTFILE_VISIONFIVE2_1_2A \
+       "starfive/jh7110-starfive-visionfive-2-v1.2a.dtb"
+#define FDTFILE_VISIONFIVE2_1_3B \
+       "starfive/jh7110-starfive-visionfive-2-v1.3b.dtb"
 
 /* enable U74-mc hart1~hart4 prefetcher */
 static void enable_prefetcher(void)
@@ -33,6 +41,31 @@ static void enable_prefetcher(void)
        }
 }
 
+/**
+ * set_fdtfile() - set the $fdtfile variable based on the board revision
+ */
+static void set_fdtfile(void)
+{
+       u8 version;
+       const char *fdtfile;
+
+       version = get_pcb_revision_from_eeprom();
+       switch (version) {
+       case 'a':
+       case 'A':
+               fdtfile = FDTFILE_VISIONFIVE2_1_2A;
+               break;
+
+       case 'b':
+       case 'B':
+       default:
+               fdtfile = FDTFILE_VISIONFIVE2_1_3B;
+               break;
+       };
+
+       env_set("fdtfile", fdtfile);
+}
+
 int board_init(void)
 {
        enable_caches();
@@ -41,6 +74,14 @@ int board_init(void)
        return 0;
 }
 
+int board_late_init(void)
+{
+       if (CONFIG_IS_ENABLED(ID_EEPROM))
+               set_fdtfile();
+
+       return 0;
+}
+
 void *board_fdt_blob_setup(int *err)
 {
        *err = 0;
@@ -51,3 +92,8 @@ void *board_fdt_blob_setup(int *err)
 
        return (ulong *)&_end;
 }
+
+int ft_board_setup(void *blob, struct bd_info *bd)
+{
+       return fdt_fixup_memory(blob, 0x40000000, gd->ram_size);
+}