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