1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
|
/*
* (C) Copyright 2013 Amlogic, Inc
*
* This file is used to run commands from misc partition
* More detail to check the command "run bcb_cmd" usage
*
* cheng.wang@amlogic.com,
* 2015-04-23 @ Shenzhen
*
*/
#include <common.h>
#include <command.h>
#include <environment.h>
#include <malloc.h>
#include <asm/byteorder.h>
#include <config.h>
#include <asm/arch/io.h>
#ifdef CONFIG_BOOTLOADER_CONTROL_BLOCK
#if defined(CONFIG_STORE_COMPATIBLE)
extern int store_read_ops(
unsigned char *partition_name,
unsigned char * buf, uint64_t off, uint64_t size);
extern int store_write_ops(
unsigned char *partition_name,
unsigned char * buf, uint64_t off, uint64_t size);
#endif
#define COMMANDBUF_SIZE 32
#define STATUSBUF_SIZE 32
#define RECOVERYBUF_SIZE 768
#define BOOTINFO_OFFSET 864
#define SLOTBUF_SIZE 32
#define MISCBUF_SIZE 1088
#define CMD_WIPE_DATA "wipe_data"
#define CMD_SYSTEM_CRASH "system_crash"
#define CMD_RUN_RECOVERY "boot-recovery"
struct bootloader_message {
char command[32];
char status[32];
char recovery[768];
// The 'recovery' field used to be 1024 bytes. It has only ever
// been used to store the recovery command line, so 768 bytes
// should be plenty. We carve off the last 256 bytes to store the
// stage string (for multistage packages) and possible future
// expansion.
char stage[32];
char slot_suffix[32];
char reserved[192];
};
static int clear_misc_partition(char *clearbuf, int size)
{
#if defined(CONFIG_STORE_COMPATIBLE)
char *partition = "misc";
memset(clearbuf, 0, size);
if (store_write_ops((unsigned char *)partition,
(unsigned char *)clearbuf, 0, size) < 0) {
printf("failed to clear %s.\n", partition);
return -1;
}
#endif
return 0;
}
static int do_RunBcbCommand(
cmd_tbl_t * cmdtp,
int flag,
int argc,
char * const argv[])
{
int i = 0;
char command[COMMANDBUF_SIZE] = {0};
char status[STATUSBUF_SIZE] = {0};
char recovery[RECOVERYBUF_SIZE] = {0};
char miscbuf[MISCBUF_SIZE] = {0};
char clearbuf[COMMANDBUF_SIZE+STATUSBUF_SIZE+RECOVERYBUF_SIZE] = {0};
if (argc != 2) {
return cmd_usage(cmdtp);
}
printf("Command: ");
for (i = 0; i < argc; i++) {
printf("%s ", argv[i]);
}
printf("\n");
char *partition = "misc";
char *command_mark = (char *)argv[1];
if (strlen(command_mark) > sizeof(command)) {
//printf("Bcb command mark range out of length(%d > %d).\n",
//strlen(command_mark), sizeof(command));
goto ERR;
}
if (!memcmp(command_mark, CMD_WIPE_DATA, strlen(command_mark))) {
printf("Start to write --wipe_data to %s\n", partition);
memcpy(miscbuf, CMD_RUN_RECOVERY, sizeof(CMD_RUN_RECOVERY));
memcpy(miscbuf+sizeof(command)+sizeof(status), "recovery\n--wipe_data", sizeof("recovery\n--wipe_data"));
#if defined(CONFIG_STORE_COMPATIBLE)
store_write_ops((unsigned char *)partition, (unsigned char *)miscbuf, 0, sizeof(miscbuf));
#endif
} else if (!memcmp(command_mark, CMD_SYSTEM_CRASH, strlen(command_mark))) {
printf("Start to write --system_crash to %s\n", partition);
memcpy(miscbuf, CMD_RUN_RECOVERY, sizeof(CMD_RUN_RECOVERY));
memcpy(miscbuf+sizeof(command)+sizeof(status), "recovery\n--system_crash", sizeof("recovery\n--system_crash"));
#if defined(CONFIG_STORE_COMPATIBLE)
store_write_ops((unsigned char *)partition, (unsigned char *)miscbuf, 0, sizeof(miscbuf));
#endif
}
printf("Start read %s partition datas!\n", partition);
#if defined(CONFIG_STORE_COMPATIBLE)
if (store_read_ops((unsigned char *)partition,
(unsigned char *)miscbuf, 0, sizeof(miscbuf)) < 0) {
printf("failed to store read %s.\n", partition);
goto ERR;
}
#endif
// judge misc partition whether has datas
char tmpbuf[MISCBUF_SIZE];
memset(tmpbuf, 0, sizeof(tmpbuf));
if (!memcmp(tmpbuf, miscbuf, strlen(miscbuf))) {
printf("BCB hasn't any datas,exit!\n");
return 0;
}
memcpy(command, miscbuf, sizeof(command));
memcpy(status, miscbuf+sizeof(command), sizeof(status));
memcpy(recovery, miscbuf+sizeof(command)+sizeof(status), sizeof(recovery));
memcpy(clearbuf, miscbuf, sizeof(clearbuf));
printf("get bootloader message from misc partition:\n");
printf("[commannd:%s]\n[status:%s]\n[recovery:%s]\n",
command, status, recovery);
if (!memcmp(command, CMD_RUN_RECOVERY, strlen(CMD_RUN_RECOVERY))) {
if (run_command("run recovery_from_flash", 0) < 0) {
printf("run_command for cmd:run recovery_from_flash failed.\n");
return -1;
}
printf("run command:run recovery_from_flash successful.\n");
return 0;
}
if (!memcmp(command_mark, command, strlen(command_mark))) {
printf("%s\n", recovery);
if (run_command((char *)recovery, 0) < 0) {
printf("run_command for cmd:%s failed.\n", recovery);
goto ERR;
}
printf("run command successful.\n");
if (clear_misc_partition(clearbuf, sizeof(clearbuf)) < 0) {
printf("clear misc partition failed.\n");
goto ERR;
} else {
printf("clear misc partition successful.\n");
}
} else {
printf("command mark(%s) not match %s,don't execute.\n",
command_mark, command);
}
return 0;
ERR:
return -1;
}
#else
static int do_RunBcbCommand(
cmd_tbl_t * cmdtp,
int flag,
int argc,
char * const argv[]) {
if (argc != 2) {
return cmd_usage(cmdtp);
}
// Do-Nothing!
return 0;
}
#endif /* CONFIG_BOOTLOADER_CONTROL_BLOCK */
// BCB: Bootloader Control Block
U_BOOT_CMD(
bcb, 2, 0, do_RunBcbCommand,
"bcb",
"\nThis command will run some commands which saved in misc\n"
"partition by mark to decide whether execute command!\n"
"Command format:\n"
" bcb bcb_mark\n"
"Example:\n"
" /dev/block/misc partiton is saved some contents:\n"
" uboot-command\n" // command mark
" N/A\n"
" setenv aa 11;setenv bb 22;setenv cc 33;saveenv;\n" // command
"So you can execute command: bcb uboot-command"
);
|
