1 /*
2 * Copyright (C) 2008 The Android Open Source Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <boot/boot.h>
30 #include <boot/board.h>
31 #include <boot/flash.h>
32
33 #define FLASH_PAGE_SIZE 2048
34 #define FLASH_PAGE_BITS 11
35
startswith(const char * str,const char * prefix)36 int startswith(const char *str, const char *prefix)
37 {
38 while(*prefix){
39 if(*prefix++ != *str++) return 0;
40 }
41 return 1;
42 }
43
verify_flash(ptentry * p,void * addr,unsigned len,int extra)44 void verify_flash(ptentry *p, void *addr, unsigned len, int extra)
45 {
46 int offset = 0;
47 void *buf = alloc(FLASH_PAGE_SIZE + extra);
48 int verify_extra = extra;
49 if(verify_extra > 4)
50 verify_extra = 16;
51 while(len > 0) {
52 flash_read_ext(p, extra, offset, buf, FLASH_PAGE_SIZE);
53 if(memcmp(addr, buf, FLASH_PAGE_SIZE + verify_extra)) {
54 dprintf("verify failed at %x\n", offset);
55 jtag_fail("verify failed");
56 return;
57 }
58 offset += FLASH_PAGE_SIZE;
59 addr += FLASH_PAGE_SIZE;
60 len -= FLASH_PAGE_SIZE;
61 if(extra) {
62 addr += extra;
63 len -= extra;
64 }
65 }
66 dprintf("verify done %d extra bytes\n", verify_extra);
67 jtag_okay("verify done");
68 }
69
handle_flash(const char * name,unsigned addr,unsigned len)70 void handle_flash(const char *name, unsigned addr, unsigned len)
71 {
72 int r;
73 ptentry *p;
74
75 dprintf("image @ %x (%d bytes)\n", addr, len);
76 dprintf("write to '%s' partition\n", name);
77
78 p = flash_find_ptn(name);
79
80 if(p == 0) {
81 jtag_fail("partition not found");
82 return;
83 } else {
84 if(flash_init()) {
85 jtag_fail("flash_init() failed");
86 return;
87 }
88
89 dprintf("erasing flash @ %d (len=%d)\n", p->start, p->length);
90 flash_erase(p);
91
92 if(len) {
93 dprintf("writing flash at @ %d\n", p->start);
94
95 if(!strcmp(name, "system") || !strcmp(name, "userdata")) {
96 r = flash_write(p, 64, (void*) addr, len);
97 } else {
98 len = (len + FLASH_PAGE_SIZE - 1) & (~(FLASH_PAGE_SIZE-1));
99 r = flash_write(p, 0, (void*) addr, len);
100 }
101 //verify_flash(p, addr, len, (!strcmp(name, "system") || !strcmp(name, "userdata")) ? 64 : 0);
102 if(r) {
103 jtag_fail("partition write failed");
104 } else {
105 jtag_okay("partition written");
106 }
107 return;
108 } else {
109 jtag_okay("partition erased");
110 return;
111 }
112 }
113 }
114
hexdump(void * ptr,unsigned len)115 void hexdump(void *ptr, unsigned len)
116 {
117 unsigned char *b = ptr;
118 int count = 0;
119
120 dprintf("%x: ", (unsigned) b);
121 while(len-- > 0) {
122 dprintf("%b ", *b++);
123 if(++count == 16) {
124 dprintf("\n%x: ", (unsigned) b);
125 count = 0;
126 }
127 }
128 if(count != 0) dprintf("\n");
129 }
130
131 static unsigned char *tmpbuf = 0;
132
handle_dump(const char * name,unsigned offset)133 void handle_dump(const char *name, unsigned offset)
134 {
135 ptentry *p;
136
137 if(tmpbuf == 0) {
138 tmpbuf = alloc(4096);
139 }
140
141 dprintf("dump '%s' partition\n", name);
142 p = flash_find_ptn(name);
143
144 if(p == 0) {
145 jtag_fail("partition not found");
146 return;
147 }
148
149 if(flash_init()) {
150 jtag_fail("flash_init() failed");
151 return;
152 }
153
154 #if 0
155 /* XXX reimpl */
156 if(flash_read_page(p->start * 64, tmpbuf, tmpbuf + 2048)){
157 jtag_fail("flash_read() failed");
158 return;
159 }
160 #endif
161
162 dprintf("page %d data:\n", p->start * 64);
163 hexdump(tmpbuf, 256);
164 dprintf("page %d extra:\n", p->start * 64);
165 hexdump(tmpbuf, 16);
166 jtag_okay("done");
167 }
168
handle_command(const char * cmd,unsigned a0,unsigned a1,unsigned a2)169 void handle_command(const char *cmd, unsigned a0, unsigned a1, unsigned a2)
170 {
171 if(startswith(cmd,"flash:")){
172 handle_flash(cmd + 6, a0, a1);
173 return;
174 }
175
176 if(startswith(cmd,"dump:")){
177 handle_dump(cmd + 5, a0);
178 return;
179 }
180
181 jtag_fail("unknown command");
182 }
183
_main(void)184 int _main(void)
185 {
186 arm11_clock_init();
187
188 dprintf_set_putc(jtag_dputc);
189
190 board_init();
191
192 jtag_cmd_loop(handle_command);
193
194 return 0;
195 }
196