1 // SPDX-License-Identifier: BSD-2-Clause
2 /*
3 * Copyright (C) 2016 The Android Open Source Project
4 */
5
6 #include <common.h>
7 #include <fastboot.h>
8 #include <fastboot-internal.h>
9 #include <fb_mmc.h>
10 #include <fb_nand.h>
11 #include <fs.h>
12 #include <version.h>
13
14 static void getvar_version(char *var_parameter, char *response);
15 static void getvar_bootloader_version(char *var_parameter, char *response);
16 static void getvar_downloadsize(char *var_parameter, char *response);
17 static void getvar_serialno(char *var_parameter, char *response);
18 static void getvar_version_baseband(char *var_parameter, char *response);
19 static void getvar_product(char *var_parameter, char *response);
20 static void getvar_current_slot(char *var_parameter, char *response);
21 static void getvar_slot_suffixes(char *var_parameter, char *response);
22 static void getvar_has_slot(char *var_parameter, char *response);
23 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH_MMC)
24 static void getvar_partition_type(char *part_name, char *response);
25 #endif
26 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH)
27 static void getvar_partition_size(char *part_name, char *response);
28 #endif
29
30 static const struct {
31 const char *variable;
32 void (*dispatch)(char *var_parameter, char *response);
33 } getvar_dispatch[] = {
34 {
35 .variable = "version",
36 .dispatch = getvar_version
37 }, {
38 .variable = "bootloader-version",
39 .dispatch = getvar_bootloader_version
40 }, {
41 .variable = "version-bootloader",
42 .dispatch = getvar_bootloader_version
43 }, {
44 .variable = "downloadsize",
45 .dispatch = getvar_downloadsize
46 }, {
47 .variable = "max-download-size",
48 .dispatch = getvar_downloadsize
49 }, {
50 .variable = "serialno",
51 .dispatch = getvar_serialno
52 }, {
53 .variable = "version-baseband",
54 .dispatch = getvar_version_baseband
55 }, {
56 .variable = "product",
57 .dispatch = getvar_product
58 }, {
59 .variable = "current-slot",
60 .dispatch = getvar_current_slot
61 }, {
62 .variable = "slot-suffixes",
63 .dispatch = getvar_slot_suffixes
64 }, {
65 .variable = "has_slot",
66 .dispatch = getvar_has_slot
67 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH_MMC)
68 }, {
69 .variable = "partition-type",
70 .dispatch = getvar_partition_type
71 #endif
72 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH)
73 }, {
74 .variable = "partition-size",
75 .dispatch = getvar_partition_size
76 #endif
77 }
78 };
79
getvar_version(char * var_parameter,char * response)80 static void getvar_version(char *var_parameter, char *response)
81 {
82 fastboot_okay(FASTBOOT_VERSION, response);
83 }
84
getvar_bootloader_version(char * var_parameter,char * response)85 static void getvar_bootloader_version(char *var_parameter, char *response)
86 {
87 fastboot_okay(U_BOOT_VERSION, response);
88 }
89
getvar_downloadsize(char * var_parameter,char * response)90 static void getvar_downloadsize(char *var_parameter, char *response)
91 {
92 fastboot_response("OKAY", response, "0x%08x", fastboot_buf_size);
93 }
94
getvar_serialno(char * var_parameter,char * response)95 static void getvar_serialno(char *var_parameter, char *response)
96 {
97 const char *tmp = env_get("serial#");
98
99 if (tmp)
100 fastboot_okay(tmp, response);
101 else
102 fastboot_fail("Value not set", response);
103 }
104
getvar_version_baseband(char * var_parameter,char * response)105 static void getvar_version_baseband(char *var_parameter, char *response)
106 {
107 fastboot_okay("N/A", response);
108 }
109
getvar_product(char * var_parameter,char * response)110 static void getvar_product(char *var_parameter, char *response)
111 {
112 const char *board = env_get("board");
113
114 if (board)
115 fastboot_okay(board, response);
116 else
117 fastboot_fail("Board not set", response);
118 }
119
getvar_current_slot(char * var_parameter,char * response)120 static void getvar_current_slot(char *var_parameter, char *response)
121 {
122 /* A/B not implemented, for now always return _a */
123 fastboot_okay("_a", response);
124 }
125
getvar_slot_suffixes(char * var_parameter,char * response)126 static void getvar_slot_suffixes(char *var_parameter, char *response)
127 {
128 fastboot_okay("_a,_b", response);
129 }
130
getvar_has_slot(char * part_name,char * response)131 static void getvar_has_slot(char *part_name, char *response)
132 {
133 if (part_name && (!strcmp(part_name, "boot") ||
134 !strcmp(part_name, "system")))
135 fastboot_okay("yes", response);
136 else
137 fastboot_okay("no", response);
138 }
139
140 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH_MMC)
getvar_partition_type(char * part_name,char * response)141 static void getvar_partition_type(char *part_name, char *response)
142 {
143 int r;
144 struct blk_desc *dev_desc;
145 disk_partition_t part_info;
146
147 r = fastboot_mmc_get_part_info(part_name, &dev_desc, &part_info,
148 response);
149 if (r >= 0) {
150 r = fs_set_blk_dev_with_part(dev_desc, r);
151 if (r < 0)
152 fastboot_fail("failed to set partition", response);
153 else
154 fastboot_okay(fs_get_type_name(), response);
155 }
156 }
157 #endif
158
159 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH)
getvar_partition_size(char * part_name,char * response)160 static void getvar_partition_size(char *part_name, char *response)
161 {
162 int r;
163 size_t size;
164
165 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH_MMC)
166 struct blk_desc *dev_desc;
167 disk_partition_t part_info;
168
169 r = fastboot_mmc_get_part_info(part_name, &dev_desc, &part_info,
170 response);
171 if (r >= 0)
172 size = part_info.size;
173 #endif
174 #if CONFIG_IS_ENABLED(FASTBOOT_FLASH_NAND)
175 struct part_info *part_info;
176
177 r = fastboot_nand_get_part_info(part_name, &part_info, response);
178 if (r >= 0)
179 size = part_info->size;
180 #endif
181 if (r >= 0)
182 fastboot_response("OKAY", response, "0x%016zx", size);
183 }
184 #endif
185
186 /**
187 * fastboot_getvar() - Writes variable indicated by cmd_parameter to response.
188 *
189 * @cmd_parameter: Pointer to command parameter
190 * @response: Pointer to fastboot response buffer
191 *
192 * Look up cmd_parameter first as an environment variable of the form
193 * fastboot.<cmd_parameter>, if that exists return use its value to set
194 * response.
195 *
196 * Otherwise lookup the name of variable and execute the appropriate
197 * function to return the requested value.
198 */
fastboot_getvar(char * cmd_parameter,char * response)199 void fastboot_getvar(char *cmd_parameter, char *response)
200 {
201 if (!cmd_parameter) {
202 fastboot_fail("missing var", response);
203 } else {
204 #define FASTBOOT_ENV_PREFIX "fastboot."
205 int i;
206 char *var_parameter = cmd_parameter;
207 char envstr[FASTBOOT_RESPONSE_LEN];
208 const char *s;
209
210 snprintf(envstr, sizeof(envstr) - 1,
211 FASTBOOT_ENV_PREFIX "%s", cmd_parameter);
212 s = env_get(envstr);
213 if (s) {
214 fastboot_response("OKAY", response, "%s", s);
215 return;
216 }
217
218 strsep(&var_parameter, ":");
219 for (i = 0; i < ARRAY_SIZE(getvar_dispatch); ++i) {
220 if (!strcmp(getvar_dispatch[i].variable,
221 cmd_parameter)) {
222 getvar_dispatch[i].dispatch(var_parameter,
223 response);
224 return;
225 }
226 }
227 pr_warn("WARNING: unknown variable: %s\n", cmd_parameter);
228 fastboot_fail("Variable not implemented", response);
229 }
230 }
231