1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * (C) Copyright 2021
4 * Köry Maincent, Bootlin, <kory.maincent@bootlin.com>
5 */
6
7 #include <common.h>
8 #include <command.h>
9 #include <malloc.h>
10 #include <extension_board.h>
11 #include <mapmem.h>
12 #include <linux/libfdt.h>
13 #include <fdt_support.h>
14
15 static LIST_HEAD(extension_list);
16
extension_apply(struct extension * extension)17 static int extension_apply(struct extension *extension)
18 {
19 char *overlay_cmd;
20 ulong extrasize, overlay_addr;
21 struct fdt_header *blob;
22
23 if (!working_fdt) {
24 printf("No FDT memory address configured. Please configure\n"
25 "the FDT address via \"fdt addr <address>\" command.\n");
26 return CMD_RET_FAILURE;
27 }
28
29 overlay_cmd = env_get("extension_overlay_cmd");
30 if (!overlay_cmd) {
31 printf("Environment extension_overlay_cmd is missing\n");
32 return CMD_RET_FAILURE;
33 }
34
35 overlay_addr = env_get_hex("extension_overlay_addr", 0);
36 if (!overlay_addr) {
37 printf("Environment extension_overlay_addr is missing\n");
38 return CMD_RET_FAILURE;
39 }
40
41 env_set("extension_overlay_name", extension->overlay);
42 if (run_command(overlay_cmd, 0) != 0)
43 return CMD_RET_FAILURE;
44
45 extrasize = env_get_hex("filesize", 0);
46 if (!extrasize)
47 return CMD_RET_FAILURE;
48
49 fdt_shrink_to_minimum(working_fdt, extrasize);
50
51 blob = map_sysmem(overlay_addr, 0);
52 if (!fdt_valid(&blob))
53 return CMD_RET_FAILURE;
54
55 /* apply method prints messages on error */
56 if (fdt_overlay_apply_verbose(working_fdt, blob))
57 return CMD_RET_FAILURE;
58
59 return CMD_RET_SUCCESS;
60 }
61
do_extension_list(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])62 static int do_extension_list(struct cmd_tbl *cmdtp, int flag,
63 int argc, char *const argv[])
64 {
65 int i = 0;
66 struct extension *extension;
67
68 if (list_empty(&extension_list)) {
69 printf("No extension registered - Please run \"extension scan\"\n");
70 return CMD_RET_SUCCESS;
71 }
72
73 list_for_each_entry(extension, &extension_list, list) {
74 printf("Extension %d: %s\n", i++, extension->name);
75 printf("\tManufacturer: \t\t%s\n", extension->owner);
76 printf("\tVersion: \t\t%s\n", extension->version);
77 printf("\tDevicetree overlay: \t%s\n", extension->overlay);
78 printf("\tOther information: \t%s\n", extension->other);
79 }
80 return CMD_RET_SUCCESS;
81 }
82
do_extension_scan(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])83 static int do_extension_scan(struct cmd_tbl *cmdtp, int flag,
84 int argc, char *const argv[])
85 {
86 struct extension *extension, *next;
87 int extension_num;
88
89 list_for_each_entry_safe(extension, next, &extension_list, list) {
90 list_del(&extension->list);
91 free(extension);
92 }
93 extension_num = extension_board_scan(&extension_list);
94
95 if (extension_num < 0)
96 return CMD_RET_FAILURE;
97
98 printf("Found %d extension board(s).\n", extension_num);
99
100 return CMD_RET_SUCCESS;
101 }
102
do_extension_apply(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])103 static int do_extension_apply(struct cmd_tbl *cmdtp, int flag,
104 int argc, char *const argv[])
105 {
106 struct extension *extension = NULL;
107 struct list_head *entry;
108 int i = 0, extension_id, ret;
109
110 if (argc < 2)
111 return CMD_RET_USAGE;
112
113 if (strcmp(argv[1], "all") == 0) {
114 list_for_each_entry(extension, &extension_list, list) {
115 ret = extension_apply(extension);
116 if (ret != CMD_RET_SUCCESS)
117 break;
118 }
119 } else {
120 extension_id = simple_strtol(argv[1], NULL, 10);
121 list_for_each(entry, &extension_list) {
122 if (i == extension_id) {
123 extension = list_entry(entry, struct extension, list);
124 break;
125 }
126 i++;
127 }
128
129 if (!extension) {
130 printf("Wrong extension number\n");
131 return CMD_RET_FAILURE;
132 }
133
134 ret = extension_apply(extension);
135 }
136
137 return ret;
138 }
139
140 static struct cmd_tbl cmd_extension[] = {
141 U_BOOT_CMD_MKENT(scan, 1, 1, do_extension_scan, "", ""),
142 U_BOOT_CMD_MKENT(list, 1, 0, do_extension_list, "", ""),
143 U_BOOT_CMD_MKENT(apply, 2, 0, do_extension_apply, "", ""),
144 };
145
do_extensionops(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])146 static int do_extensionops(struct cmd_tbl *cmdtp, int flag, int argc,
147 char *const argv[])
148 {
149 struct cmd_tbl *cp;
150
151 /* Drop the extension command */
152 argc--;
153 argv++;
154
155 cp = find_cmd_tbl(argv[0], cmd_extension, ARRAY_SIZE(cmd_extension));
156 if (cp)
157 return cp->cmd(cmdtp, flag, argc, argv);
158
159 return CMD_RET_USAGE;
160 }
161
162 U_BOOT_CMD(extension, 3, 1, do_extensionops,
163 "Extension board management sub system",
164 "scan - scan plugged extension(s) board(s)\n"
165 "extension list - lists available extension(s) board(s)\n"
166 "extension apply <extension number|all> - applies DT overlays corresponding to extension boards\n"
167 );
168