ar71xx: add support for TP-Link Archer C58 v1
[openwrt/staging/wigyori.git] / tools / firmware-utils / src / tplink-safeloader.c
1 /*
2 Copyright (c) 2014, Matthias Schiffer <mschiffer@universe-factory.net>
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 are met:
7
8 1. Redistributions of source code must retain the above copyright notice,
9 this list of conditions and the following disclaimer.
10 2. Redistributions in binary form must reproduce the above copyright notice,
11 this list of conditions and the following disclaimer in the documentation
12 and/or other materials provided with the distribution.
13
14 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
15 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17 DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
18 FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
20 SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
21 CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
22 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26
27 /*
28 tplink-safeloader
29
30 Image generation tool for the TP-LINK SafeLoader as seen on
31 TP-LINK Pharos devices (CPE210/220/510/520)
32 */
33
34
35 #include <assert.h>
36 #include <errno.h>
37 #include <stdbool.h>
38 #include <stdio.h>
39 #include <stdint.h>
40 #include <stdlib.h>
41 #include <string.h>
42 #include <time.h>
43 #include <unistd.h>
44
45 #include <arpa/inet.h>
46
47 #include <sys/types.h>
48 #include <sys/stat.h>
49
50 #include "md5.h"
51
52
53 #define ALIGN(x,a) ({ typeof(a) __a = (a); (((x) + __a - 1) & ~(__a - 1)); })
54
55
56 #define MAX_PARTITIONS 32
57
58 /** An image partition table entry */
59 struct image_partition_entry {
60 const char *name;
61 size_t size;
62 uint8_t *data;
63 };
64
65 /** A flash partition table entry */
66 struct flash_partition_entry {
67 const char *name;
68 uint32_t base;
69 uint32_t size;
70 };
71
72 /** Firmware layout description */
73 struct device_info {
74 const char *id;
75 const char *vendor;
76 const char *support_list;
77 char support_trail;
78 const char *soft_ver;
79 const struct flash_partition_entry partitions[MAX_PARTITIONS+1];
80 const char *first_sysupgrade_partition;
81 const char *last_sysupgrade_partition;
82 };
83
84 /** The content of the soft-version structure */
85 struct __attribute__((__packed__)) soft_version {
86 uint32_t magic;
87 uint32_t zero;
88 uint8_t pad1;
89 uint8_t version_major;
90 uint8_t version_minor;
91 uint8_t version_patch;
92 uint8_t year_hi;
93 uint8_t year_lo;
94 uint8_t month;
95 uint8_t day;
96 uint32_t rev;
97 uint8_t pad2;
98 };
99
100
101 static const uint8_t jffs2_eof_mark[4] = {0xde, 0xad, 0xc0, 0xde};
102
103
104 /**
105 Salt for the MD5 hash
106
107 Fortunately, TP-LINK seems to use the same salt for most devices which use
108 the new image format.
109 */
110 static const uint8_t md5_salt[16] = {
111 0x7a, 0x2b, 0x15, 0xed,
112 0x9b, 0x98, 0x59, 0x6d,
113 0xe5, 0x04, 0xab, 0x44,
114 0xac, 0x2a, 0x9f, 0x4e,
115 };
116
117
118 /** Firmware layout table */
119 static struct device_info boards[] = {
120 /** Firmware layout for the CPE210/220 */
121 {
122 .id = "CPE210",
123 .vendor = "CPE510(TP-LINK|UN|N300-5):1.0\r\n",
124 .support_list =
125 "SupportList:\r\n"
126 "CPE210(TP-LINK|UN|N300-2):1.0\r\n"
127 "CPE210(TP-LINK|UN|N300-2):1.1\r\n"
128 "CPE210(TP-LINK|US|N300-2):1.1\r\n"
129 "CPE210(TP-LINK|EU|N300-2):1.1\r\n"
130 "CPE220(TP-LINK|UN|N300-2):1.1\r\n"
131 "CPE220(TP-LINK|US|N300-2):1.1\r\n"
132 "CPE220(TP-LINK|EU|N300-2):1.1\r\n",
133 .support_trail = '\xff',
134 .soft_ver = NULL,
135
136 .partitions = {
137 {"fs-uboot", 0x00000, 0x20000},
138 {"partition-table", 0x20000, 0x02000},
139 {"default-mac", 0x30000, 0x00020},
140 {"product-info", 0x31100, 0x00100},
141 {"signature", 0x32000, 0x00400},
142 {"os-image", 0x40000, 0x170000},
143 {"soft-version", 0x1b0000, 0x00100},
144 {"support-list", 0x1b1000, 0x00400},
145 {"file-system", 0x1c0000, 0x600000},
146 {"user-config", 0x7c0000, 0x10000},
147 {"default-config", 0x7d0000, 0x10000},
148 {"log", 0x7e0000, 0x10000},
149 {"radio", 0x7f0000, 0x10000},
150 {NULL, 0, 0}
151 },
152
153 .first_sysupgrade_partition = "os-image",
154 .last_sysupgrade_partition = "file-system",
155 },
156
157 /** Firmware layout for the CPE510/520 */
158 {
159 .id = "CPE510",
160 .vendor = "CPE510(TP-LINK|UN|N300-5):1.0\r\n",
161 .support_list =
162 "SupportList:\r\n"
163 "CPE510(TP-LINK|UN|N300-5):1.0\r\n"
164 "CPE510(TP-LINK|UN|N300-5):1.1\r\n"
165 "CPE510(TP-LINK|UN|N300-5):1.1\r\n"
166 "CPE510(TP-LINK|US|N300-5):1.1\r\n"
167 "CPE510(TP-LINK|EU|N300-5):1.1\r\n"
168 "CPE520(TP-LINK|UN|N300-5):1.1\r\n"
169 "CPE520(TP-LINK|US|N300-5):1.1\r\n"
170 "CPE520(TP-LINK|EU|N300-5):1.1\r\n",
171 .support_trail = '\xff',
172 .soft_ver = NULL,
173
174 .partitions = {
175 {"fs-uboot", 0x00000, 0x20000},
176 {"partition-table", 0x20000, 0x02000},
177 {"default-mac", 0x30000, 0x00020},
178 {"product-info", 0x31100, 0x00100},
179 {"signature", 0x32000, 0x00400},
180 {"os-image", 0x40000, 0x170000},
181 {"soft-version", 0x1b0000, 0x00100},
182 {"support-list", 0x1b1000, 0x00400},
183 {"file-system", 0x1c0000, 0x600000},
184 {"user-config", 0x7c0000, 0x10000},
185 {"default-config", 0x7d0000, 0x10000},
186 {"log", 0x7e0000, 0x10000},
187 {"radio", 0x7f0000, 0x10000},
188 {NULL, 0, 0}
189 },
190
191 .first_sysupgrade_partition = "os-image",
192 .last_sysupgrade_partition = "file-system",
193 },
194
195 {
196 .id = "WBS210",
197 .vendor = "CPE510(TP-LINK|UN|N300-5):1.0\r\n",
198 .support_list =
199 "SupportList:\r\n"
200 "WBS210(TP-LINK|UN|N300-2):1.20\r\n"
201 "WBS210(TP-LINK|US|N300-2):1.20\r\n"
202 "WBS210(TP-LINK|EU|N300-2):1.20\r\n",
203 .support_trail = '\xff',
204 .soft_ver = NULL,
205
206 .partitions = {
207 {"fs-uboot", 0x00000, 0x20000},
208 {"partition-table", 0x20000, 0x02000},
209 {"default-mac", 0x30000, 0x00020},
210 {"product-info", 0x31100, 0x00100},
211 {"signature", 0x32000, 0x00400},
212 {"os-image", 0x40000, 0x170000},
213 {"soft-version", 0x1b0000, 0x00100},
214 {"support-list", 0x1b1000, 0x00400},
215 {"file-system", 0x1c0000, 0x600000},
216 {"user-config", 0x7c0000, 0x10000},
217 {"default-config", 0x7d0000, 0x10000},
218 {"log", 0x7e0000, 0x10000},
219 {"radio", 0x7f0000, 0x10000},
220 {NULL, 0, 0}
221 },
222
223 .first_sysupgrade_partition = "os-image",
224 .last_sysupgrade_partition = "file-system",
225 },
226
227 {
228 .id = "WBS510",
229 .vendor = "CPE510(TP-LINK|UN|N300-5):1.0\r\n",
230 .support_list =
231 "SupportList:\r\n"
232 "WBS510(TP-LINK|UN|N300-5):1.20\r\n"
233 "WBS510(TP-LINK|US|N300-5):1.20\r\n"
234 "WBS510(TP-LINK|EU|N300-5):1.20\r\n",
235 .support_trail = '\xff',
236 .soft_ver = NULL,
237
238 .partitions = {
239 {"fs-uboot", 0x00000, 0x20000},
240 {"partition-table", 0x20000, 0x02000},
241 {"default-mac", 0x30000, 0x00020},
242 {"product-info", 0x31100, 0x00100},
243 {"signature", 0x32000, 0x00400},
244 {"os-image", 0x40000, 0x170000},
245 {"soft-version", 0x1b0000, 0x00100},
246 {"support-list", 0x1b1000, 0x00400},
247 {"file-system", 0x1c0000, 0x600000},
248 {"user-config", 0x7c0000, 0x10000},
249 {"default-config", 0x7d0000, 0x10000},
250 {"log", 0x7e0000, 0x10000},
251 {"radio", 0x7f0000, 0x10000},
252 {NULL, 0, 0}
253 },
254
255 .first_sysupgrade_partition = "os-image",
256 .last_sysupgrade_partition = "file-system",
257 },
258
259 /** Firmware layout for the C2600 */
260 {
261 .id = "C2600",
262 .vendor = "",
263 .support_list =
264 "SupportList:\r\n"
265 "{product_name:Archer C2600,product_ver:1.0.0,special_id:00000000}\r\n",
266 .support_trail = '\x00',
267 .soft_ver = NULL,
268
269 .partitions = {
270 {"SBL1", 0x00000, 0x20000},
271 {"MIBIB", 0x20000, 0x20000},
272 {"SBL2", 0x40000, 0x20000},
273 {"SBL3", 0x60000, 0x30000},
274 {"DDRCONFIG", 0x90000, 0x10000},
275 {"SSD", 0xa0000, 0x10000},
276 {"TZ", 0xb0000, 0x30000},
277 {"RPM", 0xe0000, 0x20000},
278 {"fs-uboot", 0x100000, 0x70000},
279 {"uboot-env", 0x170000, 0x40000},
280 {"radio", 0x1b0000, 0x40000},
281 {"os-image", 0x1f0000, 0x200000},
282 {"file-system", 0x3f0000, 0x1b00000},
283 {"default-mac", 0x1ef0000, 0x00200},
284 {"pin", 0x1ef0200, 0x00200},
285 {"product-info", 0x1ef0400, 0x0fc00},
286 {"partition-table", 0x1f00000, 0x10000},
287 {"soft-version", 0x1f10000, 0x10000},
288 {"support-list", 0x1f20000, 0x10000},
289 {"profile", 0x1f30000, 0x10000},
290 {"default-config", 0x1f40000, 0x10000},
291 {"user-config", 0x1f50000, 0x40000},
292 {"qos-db", 0x1f90000, 0x40000},
293 {"usb-config", 0x1fd0000, 0x10000},
294 {"log", 0x1fe0000, 0x20000},
295 {NULL, 0, 0}
296 },
297
298 .first_sysupgrade_partition = "os-image",
299 .last_sysupgrade_partition = "file-system"
300 },
301
302 /** Firmware layout for the C25v1 */
303 {
304 .id = "ARCHER-C25-V1",
305 .support_list =
306 "SupportList:\n"
307 "{product_name:ArcherC25,product_ver:1.0.0,special_id:00000000}\n"
308 "{product_name:ArcherC25,product_ver:1.0.0,special_id:55530000}\n"
309 "{product_name:ArcherC25,product_ver:1.0.0,special_id:45550000}\n",
310 .support_trail = '\x00',
311 .soft_ver = "soft_ver:1.0.0\n",
312
313 /**
314 We use a bigger os-image partition than the stock images (and thus
315 smaller file-system), as our kernel doesn't fit in the stock firmware's
316 1MB os-image.
317 */
318 .partitions = {
319 {"factory-boot", 0x00000, 0x20000},
320 {"fs-uboot", 0x20000, 0x10000},
321 {"os-image", 0x30000, 0x180000}, /* Stock: base 0x30000 size 0x100000 */
322 {"file-system", 0x1b0000, 0x620000}, /* Stock: base 0x130000 size 0x6a0000 */
323 {"user-config", 0x7d0000, 0x04000},
324 {"default-mac", 0x7e0000, 0x00100},
325 {"device-id", 0x7e0100, 0x00100},
326 {"extra-para", 0x7e0200, 0x00100},
327 {"pin", 0x7e0300, 0x00100},
328 {"support-list", 0x7e0400, 0x00400},
329 {"soft-version", 0x7e0800, 0x00400},
330 {"product-info", 0x7e0c00, 0x01400},
331 {"partition-table", 0x7e2000, 0x01000},
332 {"profile", 0x7e3000, 0x01000},
333 {"default-config", 0x7e4000, 0x04000},
334 {"merge-config", 0x7ec000, 0x02000},
335 {"qos-db", 0x7ee000, 0x02000},
336 {"radio", 0x7f0000, 0x10000},
337 {NULL, 0, 0}
338 },
339
340 .first_sysupgrade_partition = "os-image",
341 .last_sysupgrade_partition = "file-system",
342 },
343
344 /** Firmware layout for the C58v1 */
345 {
346 .id = "ARCHER-C58-V1",
347 .vendor = "",
348 .support_list =
349 "SupportList:\r\n"
350 "{product_name:Archer C58,product_ver:1.0.0,special_id:00000000}\r\n"
351 "{product_name:Archer C58,product_ver:1.0.0,special_id:45550000}\r\n"
352 "{product_name:Archer C58,product_ver:1.0.0,special_id:55530000}\r\n",
353 .support_trail = '\x00',
354 .soft_ver = "soft_ver:1.0.0\n",
355
356 .partitions = {
357 {"fs-uboot", 0x00000, 0x10000},
358 {"default-mac", 0x10000, 0x00200},
359 {"pin", 0x10200, 0x00200},
360 {"product-info", 0x10400, 0x00100},
361 {"partition-table", 0x10500, 0x00800},
362 {"soft-version", 0x11300, 0x00200},
363 {"support-list", 0x11500, 0x00100},
364 {"device-id", 0x11600, 0x00100},
365 {"profile", 0x11700, 0x03900},
366 {"default-config", 0x15000, 0x04000},
367 {"user-config", 0x19000, 0x04000},
368 {"os-image", 0x20000, 0x150000},
369 {"file-system", 0x170000, 0x678000},
370 {"certyficate", 0x7e8000, 0x08000},
371 {"radio", 0x7f0000, 0x10000},
372 {NULL, 0, 0}
373 },
374
375 .first_sysupgrade_partition = "os-image",
376 .last_sysupgrade_partition = "file-system",
377 },
378
379 /** Firmware layout for the C59v1 */
380 {
381 .id = "ARCHER-C59-V1",
382 .vendor = "",
383 .support_list =
384 "SupportList:\r\n"
385 "{product_name:Archer C59,product_ver:1.0.0,special_id:00000000}\r\n"
386 "{product_name:Archer C59,product_ver:1.0.0,special_id:45550000}\r\n"
387 "{product_name:Archer C59,product_ver:1.0.0,special_id:52550000}\r\n"
388 "{product_name:Archer C59,product_ver:1.0.0,special_id:55530000}\r\n",
389 .support_trail = '\x00',
390 .soft_ver = "soft_ver:1.0.0\n",
391
392 .partitions = {
393 {"fs-uboot", 0x00000, 0x10000},
394 {"default-mac", 0x10000, 0x00200},
395 {"pin", 0x10200, 0x00200},
396 {"device-id", 0x10400, 0x00100},
397 {"product-info", 0x10500, 0x0fb00},
398 {"os-image", 0x20000, 0x180000},
399 {"file-system", 0x1a0000, 0xcb0000},
400 {"partition-table", 0xe50000, 0x10000},
401 {"soft-version", 0xe60000, 0x10000},
402 {"support-list", 0xe70000, 0x10000},
403 {"profile", 0xe80000, 0x10000},
404 {"default-config", 0xe90000, 0x10000},
405 {"user-config", 0xea0000, 0x40000},
406 {"usb-config", 0xee0000, 0x10000},
407 {"certificate", 0xef0000, 0x10000},
408 {"qos-db", 0xf00000, 0x40000},
409 {"log", 0xfe0000, 0x10000},
410 {"radio", 0xff0000, 0x10000},
411 {NULL, 0, 0}
412 },
413
414 .first_sysupgrade_partition = "os-image",
415 .last_sysupgrade_partition = "file-system",
416 },
417
418 /** Firmware layout for the C60v1 */
419 {
420 .id = "ARCHER-C60-V1",
421 .vendor = "",
422 .support_list =
423 "SupportList:\r\n"
424 "{product_name:Archer C60,product_ver:1.0.0,special_id:00000000}\r\n"
425 "{product_name:Archer C60,product_ver:1.0.0,special_id:45550000}\r\n"
426 "{product_name:Archer C60,product_ver:1.0.1,special_id:45550000}\r\n"
427 "{product_name:Archer C60,product_ver:1.0.0,special_id:52550000}\r\n"
428 "{product_name:Archer C60,product_ver:1.0.0,special_id:55530000}\r\n",
429 .support_trail = '\x00',
430 .soft_ver = "soft_ver:1.0.0\n",
431
432 .partitions = {
433 {"fs-uboot", 0x00000, 0x10000},
434 {"default-mac", 0x10000, 0x00200},
435 {"pin", 0x10200, 0x00200},
436 {"product-info", 0x10400, 0x00100},
437 {"partition-table", 0x10500, 0x00800},
438 {"soft-version", 0x11300, 0x00200},
439 {"support-list", 0x11500, 0x00100},
440 {"device-id", 0x11600, 0x00100},
441 {"profile", 0x11700, 0x03900},
442 {"default-config", 0x15000, 0x04000},
443 {"user-config", 0x19000, 0x04000},
444 {"os-image", 0x20000, 0x150000},
445 {"file-system", 0x170000, 0x678000},
446 {"certyficate", 0x7e8000, 0x08000},
447 {"radio", 0x7f0000, 0x10000},
448 {NULL, 0, 0}
449 },
450
451 .first_sysupgrade_partition = "os-image",
452 .last_sysupgrade_partition = "file-system",
453 },
454
455 /** Firmware layout for the C5 */
456 {
457 .id = "ARCHER-C5-V2",
458 .vendor = "",
459 .support_list =
460 "SupportList:\r\n"
461 "{product_name:ArcherC5,product_ver:2.0.0,special_id:00000000}\r\n"
462 "{product_name:ArcherC5,product_ver:2.0.0,special_id:55530000}\r\n"
463 "{product_name:ArcherC5,product_ver:2.0.0,special_id:4A500000}\r\n", /* JP version */
464 .support_trail = '\x00',
465 .soft_ver = NULL,
466
467 .partitions = {
468 {"fs-uboot", 0x00000, 0x40000},
469 {"os-image", 0x40000, 0x200000},
470 {"file-system", 0x240000, 0xc00000},
471 {"default-mac", 0xe40000, 0x00200},
472 {"pin", 0xe40200, 0x00200},
473 {"product-info", 0xe40400, 0x00200},
474 {"partition-table", 0xe50000, 0x10000},
475 {"soft-version", 0xe60000, 0x00200},
476 {"support-list", 0xe61000, 0x0f000},
477 {"profile", 0xe70000, 0x10000},
478 {"default-config", 0xe80000, 0x10000},
479 {"user-config", 0xe90000, 0x50000},
480 {"log", 0xee0000, 0x100000},
481 {"radio_bk", 0xfe0000, 0x10000},
482 {"radio", 0xff0000, 0x10000},
483 {NULL, 0, 0}
484 },
485
486 .first_sysupgrade_partition = "os-image",
487 .last_sysupgrade_partition = "file-system"
488 },
489
490 /** Firmware layout for the C9 */
491 {
492 .id = "ARCHERC9",
493 .vendor = "",
494 .support_list =
495 "SupportList:\n"
496 "{product_name:ArcherC9,"
497 "product_ver:1.0.0,"
498 "special_id:00000000}\n",
499 .support_trail = '\x00',
500 .soft_ver = NULL,
501
502 .partitions = {
503 {"fs-uboot", 0x00000, 0x40000},
504 {"os-image", 0x40000, 0x200000},
505 {"file-system", 0x240000, 0xc00000},
506 {"default-mac", 0xe40000, 0x00200},
507 {"pin", 0xe40200, 0x00200},
508 {"product-info", 0xe40400, 0x00200},
509 {"partition-table", 0xe50000, 0x10000},
510 {"soft-version", 0xe60000, 0x00200},
511 {"support-list", 0xe61000, 0x0f000},
512 {"profile", 0xe70000, 0x10000},
513 {"default-config", 0xe80000, 0x10000},
514 {"user-config", 0xe90000, 0x50000},
515 {"log", 0xee0000, 0x100000},
516 {"radio_bk", 0xfe0000, 0x10000},
517 {"radio", 0xff0000, 0x10000},
518 {NULL, 0, 0}
519 },
520
521 .first_sysupgrade_partition = "os-image",
522 .last_sysupgrade_partition = "file-system"
523 },
524
525 /** Firmware layout for the EAP120 */
526 {
527 .id = "EAP120",
528 .vendor = "EAP120(TP-LINK|UN|N300-2):1.0\r\n",
529 .support_list =
530 "SupportList:\r\n"
531 "EAP120(TP-LINK|UN|N300-2):1.0\r\n",
532 .support_trail = '\xff',
533 .soft_ver = NULL,
534
535 .partitions = {
536 {"fs-uboot", 0x00000, 0x20000},
537 {"partition-table", 0x20000, 0x02000},
538 {"default-mac", 0x30000, 0x00020},
539 {"support-list", 0x31000, 0x00100},
540 {"product-info", 0x31100, 0x00100},
541 {"soft-version", 0x32000, 0x00100},
542 {"os-image", 0x40000, 0x180000},
543 {"file-system", 0x1c0000, 0x600000},
544 {"user-config", 0x7c0000, 0x10000},
545 {"backup-config", 0x7d0000, 0x10000},
546 {"log", 0x7e0000, 0x10000},
547 {"radio", 0x7f0000, 0x10000},
548 {NULL, 0, 0}
549 },
550
551 .first_sysupgrade_partition = "os-image",
552 .last_sysupgrade_partition = "file-system"
553 },
554
555 /** Firmware layout for the TL-WA850RE v2 */
556 {
557 .id = "TLWA850REV2",
558 .vendor = "",
559 .support_list =
560 "SupportList:\n"
561 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:55530000}\n"
562 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:00000000}\n"
563 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:55534100}\n"
564 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:45550000}\n"
565 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:4B520000}\n"
566 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:42520000}\n"
567 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:4A500000}\n"
568 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:43410000}\n"
569 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:41550000}\n"
570 "{product_name:TL-WA850RE,product_ver:2.0.0,special_id:52550000}\n",
571 .support_trail = '\x00',
572 .soft_ver = NULL,
573
574 /**
575 576KB were moved from file-system to os-image
576 in comparison to the stock image
577 */
578 .partitions = {
579 {"fs-uboot", 0x00000, 0x20000},
580 {"os-image", 0x20000, 0x150000},
581 {"file-system", 0x170000, 0x240000},
582 {"partition-table", 0x3b0000, 0x02000},
583 {"default-mac", 0x3c0000, 0x00020},
584 {"pin", 0x3c0100, 0x00020},
585 {"product-info", 0x3c1000, 0x01000},
586 {"soft-version", 0x3c2000, 0x00100},
587 {"support-list", 0x3c3000, 0x01000},
588 {"profile", 0x3c4000, 0x08000},
589 {"user-config", 0x3d0000, 0x10000},
590 {"default-config", 0x3e0000, 0x10000},
591 {"radio", 0x3f0000, 0x10000},
592 {NULL, 0, 0}
593 },
594
595 .first_sysupgrade_partition = "os-image",
596 .last_sysupgrade_partition = "file-system"
597 },
598
599 /** Firmware layout for the TL-WA855RE v1 */
600 {
601 .id = "TLWA855REV1",
602 .vendor = "",
603 .support_list =
604 "SupportList:\n"
605 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:00000000}\n"
606 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:55530000}\n"
607 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:45550000}\n"
608 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:4B520000}\n"
609 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:42520000}\n"
610 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:4A500000}\n"
611 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:43410000}\n"
612 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:41550000}\n"
613 "{product_name:TL-WA855RE,product_ver:1.0.0,special_id:52550000}\n",
614 .support_trail = '\x00',
615 .soft_ver = NULL,
616
617 .partitions = {
618 {"fs-uboot", 0x00000, 0x20000},
619 {"os-image", 0x20000, 0x150000},
620 {"file-system", 0x170000, 0x240000},
621 {"partition-table", 0x3b0000, 0x02000},
622 {"default-mac", 0x3c0000, 0x00020},
623 {"pin", 0x3c0100, 0x00020},
624 {"product-info", 0x3c1000, 0x01000},
625 {"soft-version", 0x3c2000, 0x00100},
626 {"support-list", 0x3c3000, 0x01000},
627 {"profile", 0x3c4000, 0x08000},
628 {"user-config", 0x3d0000, 0x10000},
629 {"default-config", 0x3e0000, 0x10000},
630 {"radio", 0x3f0000, 0x10000},
631 {NULL, 0, 0}
632 },
633
634 .first_sysupgrade_partition = "os-image",
635 .last_sysupgrade_partition = "file-system"
636 },
637
638 /** Firmware layout for the TL-WR1043 v4 */
639 {
640 .id = "TLWR1043NDV4",
641 .vendor = "",
642 .support_list =
643 "SupportList:\n"
644 "{product_name:TL-WR1043ND,product_ver:4.0.0,special_id:45550000}\n",
645 .support_trail = '\x00',
646 .soft_ver = NULL,
647
648 /**
649 We use a bigger os-image partition than the stock images (and thus
650 smaller file-system), as our kernel doesn't fit in the stock firmware's
651 1MB os-image.
652 */
653 .partitions = {
654 {"fs-uboot", 0x00000, 0x20000},
655 {"os-image", 0x20000, 0x180000},
656 {"file-system", 0x1a0000, 0xdb0000},
657 {"default-mac", 0xf50000, 0x00200},
658 {"pin", 0xf50200, 0x00200},
659 {"product-info", 0xf50400, 0x0fc00},
660 {"soft-version", 0xf60000, 0x0b000},
661 {"support-list", 0xf6b000, 0x04000},
662 {"profile", 0xf70000, 0x04000},
663 {"default-config", 0xf74000, 0x0b000},
664 {"user-config", 0xf80000, 0x40000},
665 {"partition-table", 0xfc0000, 0x10000},
666 {"log", 0xfd0000, 0x20000},
667 {"radio", 0xff0000, 0x10000},
668 {NULL, 0, 0}
669 },
670
671 .first_sysupgrade_partition = "os-image",
672 .last_sysupgrade_partition = "file-system"
673 },
674
675 /** Firmware layout for the TL-WR942N V1 */
676 {
677 .id = "TLWR942NV1",
678 .vendor = "",
679 .support_list =
680 "SupportList:\r\n"
681 "{product_name:TL-WR942N,product_ver:1.0.0,special_id:00000000}\r\n"
682 "{product_name:TL-WR942N,product_ver:1.0.0,special_id:52550000}\r\n",
683 .support_trail = '\x00',
684 .soft_ver = NULL,
685
686 .partitions = {
687 {"fs-uboot", 0x00000, 0x20000},
688 {"os-image", 0x20000, 0x150000},
689 {"file-system", 0x170000, 0xcd0000},
690 {"default-mac", 0xe40000, 0x00200},
691 {"pin", 0xe40200, 0x00200},
692 {"product-info", 0xe40400, 0x0fc00},
693 {"partition-table", 0xe50000, 0x10000},
694 {"soft-version", 0xe60000, 0x10000},
695 {"support-list", 0xe70000, 0x10000},
696 {"profile", 0xe80000, 0x10000},
697 {"default-config", 0xe90000, 0x10000},
698 {"user-config", 0xea0000, 0x40000},
699 {"qos-db", 0xee0000, 0x40000},
700 {"certificate", 0xf20000, 0x10000},
701 {"usb-config", 0xfb0000, 0x10000},
702 {"log", 0xfc0000, 0x20000},
703 {"radio-bk", 0xfe0000, 0x10000},
704 {"radio", 0xff0000, 0x10000},
705 {NULL, 0, 0}
706 },
707
708 .first_sysupgrade_partition = "os-image",
709 .last_sysupgrade_partition = "file-system",
710 },
711
712 /** Firmware layout for the RE450 */
713 {
714 .id = "RE450",
715 .vendor = "",
716 .support_list =
717 "SupportList:\r\n"
718 "{product_name:RE450,product_ver:1.0.0,special_id:00000000}\r\n"
719 "{product_name:RE450,product_ver:1.0.0,special_id:55530000}\r\n"
720 "{product_name:RE450,product_ver:1.0.0,special_id:45550000}\r\n"
721 "{product_name:RE450,product_ver:1.0.0,special_id:4A500000}\r\n"
722 "{product_name:RE450,product_ver:1.0.0,special_id:43410000}\r\n"
723 "{product_name:RE450,product_ver:1.0.0,special_id:41550000}\r\n"
724 "{product_name:RE450,product_ver:1.0.0,special_id:4B520000}\r\n"
725 "{product_name:RE450,product_ver:1.0.0,special_id:55534100}\r\n",
726 .support_trail = '\x00',
727 .soft_ver = NULL,
728
729 /**
730 The flash partition table for RE450;
731 it is almost the same as the one used by the stock images,
732 576KB were moved from file-system to os-image.
733 */
734 .partitions = {
735 {"fs-uboot", 0x00000, 0x20000},
736 {"os-image", 0x20000, 0x150000},
737 {"file-system", 0x170000, 0x4a0000},
738 {"partition-table", 0x600000, 0x02000},
739 {"default-mac", 0x610000, 0x00020},
740 {"pin", 0x610100, 0x00020},
741 {"product-info", 0x611100, 0x01000},
742 {"soft-version", 0x620000, 0x01000},
743 {"support-list", 0x621000, 0x01000},
744 {"profile", 0x622000, 0x08000},
745 {"user-config", 0x630000, 0x10000},
746 {"default-config", 0x640000, 0x10000},
747 {"radio", 0x7f0000, 0x10000},
748 {NULL, 0, 0}
749 },
750
751 .first_sysupgrade_partition = "os-image",
752 .last_sysupgrade_partition = "file-system"
753 },
754
755 {}
756 };
757
758 #define error(_ret, _errno, _str, ...) \
759 do { \
760 fprintf(stderr, _str ": %s\n", ## __VA_ARGS__, \
761 strerror(_errno)); \
762 if (_ret) \
763 exit(_ret); \
764 } while (0)
765
766
767 /** Stores a uint32 as big endian */
768 static inline void put32(uint8_t *buf, uint32_t val) {
769 buf[0] = val >> 24;
770 buf[1] = val >> 16;
771 buf[2] = val >> 8;
772 buf[3] = val;
773 }
774
775 /** Allocates a new image partition */
776 static struct image_partition_entry alloc_image_partition(const char *name, size_t len) {
777 struct image_partition_entry entry = {name, len, malloc(len)};
778 if (!entry.data)
779 error(1, errno, "malloc");
780
781 return entry;
782 }
783
784 /** Frees an image partition */
785 static void free_image_partition(struct image_partition_entry entry) {
786 free(entry.data);
787 }
788
789 static time_t source_date_epoch = -1;
790 static void set_source_date_epoch() {
791 char *env = getenv("SOURCE_DATE_EPOCH");
792 char *endptr = env;
793 errno = 0;
794 if (env && *env) {
795 source_date_epoch = strtoull(env, &endptr, 10);
796 if (errno || (endptr && *endptr != '\0')) {
797 fprintf(stderr, "Invalid SOURCE_DATE_EPOCH");
798 exit(1);
799 }
800 }
801 }
802
803 /** Generates the partition-table partition */
804 static struct image_partition_entry make_partition_table(const struct flash_partition_entry *p) {
805 struct image_partition_entry entry = alloc_image_partition("partition-table", 0x800);
806
807 char *s = (char *)entry.data, *end = (char *)(s+entry.size);
808
809 *(s++) = 0x00;
810 *(s++) = 0x04;
811 *(s++) = 0x00;
812 *(s++) = 0x00;
813
814 size_t i;
815 for (i = 0; p[i].name; i++) {
816 size_t len = end-s;
817 size_t w = snprintf(s, len, "partition %s base 0x%05x size 0x%05x\n", p[i].name, p[i].base, p[i].size);
818
819 if (w > len-1)
820 error(1, 0, "flash partition table overflow?");
821
822 s += w;
823 }
824
825 s++;
826
827 memset(s, 0xff, end-s);
828
829 return entry;
830 }
831
832
833 /** Generates a binary-coded decimal representation of an integer in the range [0, 99] */
834 static inline uint8_t bcd(uint8_t v) {
835 return 0x10 * (v/10) + v%10;
836 }
837
838
839 /** Generates the soft-version partition */
840 static struct image_partition_entry make_soft_version(uint32_t rev) {
841 struct image_partition_entry entry = alloc_image_partition("soft-version", sizeof(struct soft_version));
842 struct soft_version *s = (struct soft_version *)entry.data;
843
844 time_t t;
845
846 if (source_date_epoch != -1)
847 t = source_date_epoch;
848 else if (time(&t) == (time_t)(-1))
849 error(1, errno, "time");
850
851 struct tm *tm = localtime(&t);
852
853 s->magic = htonl(0x0000000c);
854 s->zero = 0;
855 s->pad1 = 0xff;
856
857 s->version_major = 0;
858 s->version_minor = 0;
859 s->version_patch = 0;
860
861 s->year_hi = bcd((1900+tm->tm_year)/100);
862 s->year_lo = bcd(tm->tm_year%100);
863 s->month = bcd(tm->tm_mon+1);
864 s->day = bcd(tm->tm_mday);
865 s->rev = htonl(rev);
866
867 s->pad2 = 0xff;
868
869 return entry;
870 }
871
872 static struct image_partition_entry make_soft_version_from_string(const char *soft_ver) {
873 /** String length _including_ the terminating zero byte */
874 uint32_t ver_len = strlen(soft_ver) + 1;
875 /** Partition contains 64 bit header, the version string, and one additional null byte */
876 size_t partition_len = 2*sizeof(uint32_t) + ver_len + 1;
877 struct image_partition_entry entry = alloc_image_partition("soft-version", partition_len);
878
879 uint32_t *len = (uint32_t *)entry.data;
880 len[0] = htonl(ver_len);
881 len[1] = 0;
882 memcpy(&len[2], soft_ver, ver_len);
883
884 entry.data[partition_len - 1] = 0;
885
886 return entry;
887 }
888
889 /** Generates the support-list partition */
890 static struct image_partition_entry make_support_list(const struct device_info *info) {
891 size_t len = strlen(info->support_list);
892 struct image_partition_entry entry = alloc_image_partition("support-list", len + 9);
893
894 put32(entry.data, len);
895 memset(entry.data+4, 0, 4);
896 memcpy(entry.data+8, info->support_list, len);
897 entry.data[len+8] = info->support_trail;
898
899 return entry;
900 }
901
902 /** Creates a new image partition with an arbitrary name from a file */
903 static struct image_partition_entry read_file(const char *part_name, const char *filename, bool add_jffs2_eof) {
904 struct stat statbuf;
905
906 if (stat(filename, &statbuf) < 0)
907 error(1, errno, "unable to stat file `%s'", filename);
908
909 size_t len = statbuf.st_size;
910
911 if (add_jffs2_eof)
912 len = ALIGN(len, 0x10000) + sizeof(jffs2_eof_mark);
913
914 struct image_partition_entry entry = alloc_image_partition(part_name, len);
915
916 FILE *file = fopen(filename, "rb");
917 if (!file)
918 error(1, errno, "unable to open file `%s'", filename);
919
920 if (fread(entry.data, statbuf.st_size, 1, file) != 1)
921 error(1, errno, "unable to read file `%s'", filename);
922
923 if (add_jffs2_eof) {
924 uint8_t *eof = entry.data + statbuf.st_size, *end = entry.data+entry.size;
925
926 memset(eof, 0xff, end - eof - sizeof(jffs2_eof_mark));
927 memcpy(end - sizeof(jffs2_eof_mark), jffs2_eof_mark, sizeof(jffs2_eof_mark));
928 }
929
930 fclose(file);
931
932 return entry;
933 }
934
935 /** Creates a new image partition from arbitrary data */
936 static struct image_partition_entry put_data(const char *part_name, const char *datain, size_t len) {
937
938 struct image_partition_entry entry = alloc_image_partition(part_name, len);
939
940 memcpy(entry.data, datain, len);
941
942 return entry;
943 }
944
945 /**
946 Copies a list of image partitions into an image buffer and generates the image partition table while doing so
947
948 Example image partition table:
949
950 fwup-ptn partition-table base 0x00800 size 0x00800
951 fwup-ptn os-image base 0x01000 size 0x113b45
952 fwup-ptn file-system base 0x114b45 size 0x1d0004
953 fwup-ptn support-list base 0x2e4b49 size 0x000d1
954
955 Each line of the partition table is terminated with the bytes 09 0d 0a ("\t\r\n"),
956 the end of the partition table is marked with a zero byte.
957
958 The firmware image must contain at least the partition-table and support-list partitions
959 to be accepted. There aren't any alignment constraints for the image partitions.
960
961 The partition-table partition contains the actual flash layout; partitions
962 from the image partition table are mapped to the corresponding flash partitions during
963 the firmware upgrade. The support-list partition contains a list of devices supported by
964 the firmware image.
965
966 The base offsets in the firmware partition table are relative to the end
967 of the vendor information block, so the partition-table partition will
968 actually start at offset 0x1814 of the image.
969
970 I think partition-table must be the first partition in the firmware image.
971 */
972 static void put_partitions(uint8_t *buffer, const struct flash_partition_entry *flash_parts, const struct image_partition_entry *parts) {
973 size_t i, j;
974 char *image_pt = (char *)buffer, *end = image_pt + 0x800;
975
976 size_t base = 0x800;
977 for (i = 0; parts[i].name; i++) {
978 for (j = 0; flash_parts[j].name; j++) {
979 if (!strcmp(flash_parts[j].name, parts[i].name)) {
980 if (parts[i].size > flash_parts[j].size)
981 error(1, 0, "%s partition too big (more than %u bytes)", flash_parts[j].name, (unsigned)flash_parts[j].size);
982 break;
983 }
984 }
985
986 assert(flash_parts[j].name);
987
988 memcpy(buffer + base, parts[i].data, parts[i].size);
989
990 size_t len = end-image_pt;
991 size_t w = snprintf(image_pt, len, "fwup-ptn %s base 0x%05x size 0x%05x\t\r\n", parts[i].name, (unsigned)base, (unsigned)parts[i].size);
992
993 if (w > len-1)
994 error(1, 0, "image partition table overflow?");
995
996 image_pt += w;
997
998 base += parts[i].size;
999 }
1000 }
1001
1002 /** Generates and writes the image MD5 checksum */
1003 static void put_md5(uint8_t *md5, uint8_t *buffer, unsigned int len) {
1004 MD5_CTX ctx;
1005
1006 MD5_Init(&ctx);
1007 MD5_Update(&ctx, md5_salt, (unsigned int)sizeof(md5_salt));
1008 MD5_Update(&ctx, buffer, len);
1009 MD5_Final(md5, &ctx);
1010 }
1011
1012
1013 /**
1014 Generates the firmware image in factory format
1015
1016 Image format:
1017
1018 Bytes (hex) Usage
1019 ----------- -----
1020 0000-0003 Image size (4 bytes, big endian)
1021 0004-0013 MD5 hash (hash of a 16 byte salt and the image data starting with byte 0x14)
1022 0014-0017 Vendor information length (without padding) (4 bytes, big endian)
1023 0018-1013 Vendor information (4092 bytes, padded with 0xff; there seem to be older
1024 (VxWorks-based) TP-LINK devices which use a smaller vendor information block)
1025 1014-1813 Image partition table (2048 bytes, padded with 0xff)
1026 1814-xxxx Firmware partitions
1027 */
1028 static void * generate_factory_image(const struct device_info *info, const struct image_partition_entry *parts, size_t *len) {
1029 *len = 0x1814;
1030
1031 size_t i;
1032 for (i = 0; parts[i].name; i++)
1033 *len += parts[i].size;
1034
1035 uint8_t *image = malloc(*len);
1036 if (!image)
1037 error(1, errno, "malloc");
1038
1039 memset(image, 0xff, *len);
1040 put32(image, *len);
1041
1042 if (info->vendor) {
1043 size_t vendor_len = strlen(info->vendor);
1044 put32(image+0x14, vendor_len);
1045 memcpy(image+0x18, info->vendor, vendor_len);
1046 }
1047
1048 put_partitions(image + 0x1014, info->partitions, parts);
1049 put_md5(image+0x04, image+0x14, *len-0x14);
1050
1051 return image;
1052 }
1053
1054 /**
1055 Generates the firmware image in sysupgrade format
1056
1057 This makes some assumptions about the provided flash and image partition tables and
1058 should be generalized when TP-LINK starts building its safeloader into hardware with
1059 different flash layouts.
1060 */
1061 static void * generate_sysupgrade_image(const struct device_info *info, const struct image_partition_entry *image_parts, size_t *len) {
1062 size_t i, j;
1063 size_t flash_first_partition_index = 0;
1064 size_t flash_last_partition_index = 0;
1065 const struct flash_partition_entry *flash_first_partition = NULL;
1066 const struct flash_partition_entry *flash_last_partition = NULL;
1067 const struct image_partition_entry *image_last_partition = NULL;
1068
1069 /** Find first and last partitions */
1070 for (i = 0; info->partitions[i].name; i++) {
1071 if (!strcmp(info->partitions[i].name, info->first_sysupgrade_partition)) {
1072 flash_first_partition = &info->partitions[i];
1073 flash_first_partition_index = i;
1074 } else if (!strcmp(info->partitions[i].name, info->last_sysupgrade_partition)) {
1075 flash_last_partition = &info->partitions[i];
1076 flash_last_partition_index = i;
1077 }
1078 }
1079
1080 assert(flash_first_partition && flash_last_partition);
1081 assert(flash_first_partition_index < flash_last_partition_index);
1082
1083 /** Find last partition from image to calculate needed size */
1084 for (i = 0; image_parts[i].name; i++) {
1085 if (!strcmp(image_parts[i].name, info->last_sysupgrade_partition)) {
1086 image_last_partition = &image_parts[i];
1087 break;
1088 }
1089 }
1090
1091 assert(image_last_partition);
1092
1093 *len = flash_last_partition->base - flash_first_partition->base + image_last_partition->size;
1094
1095 uint8_t *image = malloc(*len);
1096 if (!image)
1097 error(1, errno, "malloc");
1098
1099 memset(image, 0xff, *len);
1100
1101 for (i = flash_first_partition_index; i <= flash_last_partition_index; i++) {
1102 for (j = 0; image_parts[j].name; j++) {
1103 if (!strcmp(info->partitions[i].name, image_parts[j].name)) {
1104 if (image_parts[j].size > info->partitions[i].size)
1105 error(1, 0, "%s partition too big (more than %u bytes)", info->partitions[i].name, (unsigned)info->partitions[i].size);
1106 memcpy(image + info->partitions[i].base - flash_first_partition->base, image_parts[j].data, image_parts[j].size);
1107 break;
1108 }
1109
1110 assert(image_parts[j].name);
1111 }
1112 }
1113
1114 return image;
1115 }
1116
1117 /** Generates an image according to a given layout and writes it to a file */
1118 static void build_image(const char *output,
1119 const char *kernel_image,
1120 const char *rootfs_image,
1121 uint32_t rev,
1122 bool add_jffs2_eof,
1123 bool sysupgrade,
1124 const struct device_info *info) {
1125
1126 struct image_partition_entry parts[7] = {};
1127
1128 parts[0] = make_partition_table(info->partitions);
1129 if (info->soft_ver)
1130 parts[1] = make_soft_version_from_string(info->soft_ver);
1131 else
1132 parts[1] = make_soft_version(rev);
1133
1134 parts[2] = make_support_list(info);
1135 parts[3] = read_file("os-image", kernel_image, false);
1136 parts[4] = read_file("file-system", rootfs_image, add_jffs2_eof);
1137
1138 if (strcasecmp(info->id, "ARCHER-C25-V1") == 0) {
1139 const char mdat[11] = {0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00};
1140 parts[5] = put_data("extra-para", mdat, 11);
1141 }
1142
1143 size_t len;
1144 void *image;
1145 if (sysupgrade)
1146 image = generate_sysupgrade_image(info, parts, &len);
1147 else
1148 image = generate_factory_image(info, parts, &len);
1149
1150 FILE *file = fopen(output, "wb");
1151 if (!file)
1152 error(1, errno, "unable to open output file");
1153
1154 if (fwrite(image, len, 1, file) != 1)
1155 error(1, 0, "unable to write output file");
1156
1157 fclose(file);
1158
1159 free(image);
1160
1161 size_t i;
1162 for (i = 0; parts[i].name; i++)
1163 free_image_partition(parts[i]);
1164 }
1165
1166 /** Usage output */
1167 static void usage(const char *argv0) {
1168 fprintf(stderr,
1169 "Usage: %s [OPTIONS...]\n"
1170 "\n"
1171 "Options:\n"
1172 " -B <board> create image for the board specified with <board>\n"
1173 " -k <file> read kernel image from the file <file>\n"
1174 " -r <file> read rootfs image from the file <file>\n"
1175 " -o <file> write output to the file <file>\n"
1176 " -V <rev> sets the revision number to <rev>\n"
1177 " -j add jffs2 end-of-filesystem markers\n"
1178 " -S create sysupgrade instead of factory image\n"
1179 " -h show this help\n",
1180 argv0
1181 );
1182 };
1183
1184
1185 static const struct device_info *find_board(const char *id)
1186 {
1187 struct device_info *board = NULL;
1188
1189 for (board = boards; board->id != NULL; board++)
1190 if (strcasecmp(id, board->id) == 0)
1191 return board;
1192
1193 return NULL;
1194 }
1195
1196 int main(int argc, char *argv[]) {
1197 const char *board = NULL, *kernel_image = NULL, *rootfs_image = NULL, *output = NULL;
1198 bool add_jffs2_eof = false, sysupgrade = false;
1199 unsigned rev = 0;
1200 const struct device_info *info;
1201 set_source_date_epoch();
1202
1203 while (true) {
1204 int c;
1205
1206 c = getopt(argc, argv, "B:k:r:o:V:jSh");
1207 if (c == -1)
1208 break;
1209
1210 switch (c) {
1211 case 'B':
1212 board = optarg;
1213 break;
1214
1215 case 'k':
1216 kernel_image = optarg;
1217 break;
1218
1219 case 'r':
1220 rootfs_image = optarg;
1221 break;
1222
1223 case 'o':
1224 output = optarg;
1225 break;
1226
1227 case 'V':
1228 sscanf(optarg, "r%u", &rev);
1229 break;
1230
1231 case 'j':
1232 add_jffs2_eof = true;
1233 break;
1234
1235 case 'S':
1236 sysupgrade = true;
1237 break;
1238
1239 case 'h':
1240 usage(argv[0]);
1241 return 0;
1242
1243 default:
1244 usage(argv[0]);
1245 return 1;
1246 }
1247 }
1248
1249 if (!board)
1250 error(1, 0, "no board has been specified");
1251 if (!kernel_image)
1252 error(1, 0, "no kernel image has been specified");
1253 if (!rootfs_image)
1254 error(1, 0, "no rootfs image has been specified");
1255 if (!output)
1256 error(1, 0, "no output filename has been specified");
1257
1258 info = find_board(board);
1259
1260 if (info == NULL)
1261 error(1, 0, "unsupported board %s", board);
1262
1263 build_image(output, kernel_image, rootfs_image, rev, add_jffs2_eof, sysupgrade, info);
1264
1265 return 0;
1266 }