2 * Copyright (C) 2004 Manuel Novoa III <mjn3@codepoet.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 * This is a hacked replacement for the 'addpattern' utility used to
22 * create wrt54g .bin firmware files. It isn't pretty, but it does
26 * -v allows setting the version string on the command line.
27 * -{0|1} sets the (currently ignored) hw_ver flag in the header
28 * to 0 or 1 respectively.
33 * Modified by rodent at rodent dot za dot net
34 * Support added for the new WRT54G v2.2 and WRT54GS v1.1 "flags"
35 * Without the flags set to 0x7, the above units will refuse to flash.
38 * -{0|1|2} sets {0|1} sets hw_ver flag to 0/1. {2} sets hw_ver to 1
39 * and adds the new hardware "flags" for the v2.2/v1.1 units
44 * Modified by juan.i.gonzalez at subdown dot net
45 * Support added for the AG241v2 and similar
48 * -r #.# adds revision hardware flags. AG241v2 and similar.
50 * AG241V2 firmware sets the hw_ver to 0x44.
54 * Convert 2.0 to 20 to be an integer, and add 0x30 to skip special ASCII
55 * #define HW_Version ((HW_REV * 10) + 0x30) -> from cyutils.h
65 /**********************************************************************/
67 #define CODE_ID "U2ND" /* from code_pattern.h */
68 #define CODE_PATTERN "W54S" /* from code_pattern.h */
69 #define PBOT_PATTERN "PBOT"
71 #define CYBERTAN_VERSION "v3.37.2" /* from cyutils.h */
73 /* WRT54G v2.2 and WRT54GS v1.1 "flags" (from 3.37.32 firmware cyutils.h) */
74 #define SUPPORT_4712_CHIP 0x0001
75 #define SUPPORT_INTEL_FLASH 0x0002
76 #define SUPPORT_5325E_SWITCH 0x0004
77 /* (from 3.00.24 firmware cyutils.h) */
78 #define SUPPORT_4704_CHIP 0x0008
79 #define SUPPORT_5352E_CHIP 0x0010
80 /* (from WD My Net Wi-Fi Range Extender's cyutils.s) */
81 #define SUPPORT_4703_CHIP 0x0020
83 struct code_header
{ /* from cyutils.h */
87 char id
[4]; /* U2ND */
88 char hw_ver
; /* 0: for 4702, 1: for 4712 -- new in 2.04.3 */
90 unsigned char sn
; // Serial Number
91 unsigned char flags
[2]; /* SUPPORT_ flags new for 3.37.2 (WRT54G v2.2 and WRT54GS v1.1) */
92 unsigned char stable
[2]; // The image is stable (for dual image)
93 unsigned char try1
[2]; // Try to boot image first time (for dual image)
94 unsigned char try2
[2]; // Try to boot image second time (for dual image)
95 unsigned char try3
[2]; // Try to boot image third time (for dual_image)
96 unsigned char res3
[2];
107 struct board_info boards
[] = {
113 .flags
= {0x3f, 0x00},
117 .pattern
= "WDHNSTFH",
120 .flags
= {0x3f, 0x00},
122 /* Terminating entry */
127 /**********************************************************************/
129 void usage(void) __attribute__ (( __noreturn__
));
133 fprintf(stderr
, "Usage: addpattern [-i trxfile] [-o binfile] [-B board_id] [-p pattern] [-s serial] [-g] [-b] [-v v#.#.#] [-r #.#] [-{0|1|2|4|5}] -h\n");
137 struct board_info
*find_board(char *id
)
139 struct board_info
*board
;
141 for (board
= boards
; board
->id
!= NULL
; board
++)
142 if (strcasecmp(id
, board
->id
) == 0)
148 int main(int argc
, char **argv
)
150 char buf
[1024]; /* keep this at 1k or adjust garbage calc below */
151 struct code_header
*hdr
;
156 char *pattern
= CODE_PATTERN
;
157 char *pbotpat
= PBOT_PATTERN
;
158 char *version
= CYBERTAN_VERSION
;
159 char *board_id
= NULL
;
160 struct board_info
*board
= NULL
;
169 fprintf(stderr
, "mjn3's addpattern replacement - v0.81\n");
171 hdr
= (struct code_header
*) buf
;
172 memset(hdr
, 0, sizeof(struct code_header
));
174 while ((c
= getopt(argc
, argv
, "i:o:p:s:gbv:01245hr:B:")) != -1) {
186 hdr
->sn
= (unsigned char) atoi (optarg
);
194 case 'v': /* extension to allow setting version */
203 case '2': /* new 54G v2.2 and 54GS v1.1 flags */
205 hdr
->flags
[0] |= SUPPORT_4712_CHIP
;
206 hdr
->flags
[0] |= SUPPORT_INTEL_FLASH
;
207 hdr
->flags
[0] |= SUPPORT_5325E_SWITCH
;
210 /* V4 firmware sets the flags to 0x1f */
212 hdr
->flags
[0] = 0x1f;
215 /* V5 is appended to trxV2 image */
216 hdr
->stable
[0] = 0x73; // force image to be stable
217 hdr
->stable
[1] = 0x00;
218 hdr
->try1
[0] = 0x74; // force try1 to be set
220 hdr
->try2
[0] = hdr
->try2
[1] = 0xFF;
221 hdr
->try3
[0] = hdr
->try3
[1] = 0xFF;
224 hdr
->hw_ver
= (char)(atof(optarg
)*10)+0x30;
236 if (optind
!= argc
|| optind
== 1) {
237 fprintf(stderr
, "illegal arg \"%s\"\n", argv
[optind
]);
242 board
= find_board(board_id
);
244 fprintf(stderr
, "unknown board \"%s\"\n", board_id
);
247 pattern
= board
->pattern
;
248 hdr
->hw_ver
= board
->hw_ver
;
250 hdr
->flags
[0] = board
->flags
[0];
251 hdr
->flags
[1] = board
->flags
[1];
254 if (strlen(pattern
) > 8) {
255 fprintf(stderr
, "illegal pattern \"%s\"\n", pattern
);
259 if (ifn
&& !(in
= fopen(ifn
, "r"))) {
260 fprintf(stderr
, "can not open \"%s\" for reading\n", ifn
);
264 if (ofn
&& !(out
= fopen(ofn
, "w"))) {
265 fprintf(stderr
, "can not open \"%s\" for writing\n", ofn
);
269 if (time(&t
) == (time_t)(-1)) {
270 fprintf(stderr
, "time call failed\n");
276 if (3 != sscanf(version
, "v%d.%d.%d", &v0
, &v1
, &v2
)) {
277 fprintf(stderr
, "bad version string \"%s\"\n", version
);
281 memcpy(hdr
->magic
, pattern
, strlen(pattern
));
283 memcpy(&hdr
->magic
[4], pbotpat
, 4);
284 hdr
->fwdate
[0] = ptm
->tm_year
% 100;
285 hdr
->fwdate
[1] = ptm
->tm_mon
+ 1;
286 hdr
->fwdate
[2] = ptm
->tm_mday
;
290 memcpy(hdr
->id
, CODE_ID
, strlen(CODE_ID
));
292 off
= sizeof(struct code_header
);
294 fprintf(stderr
, "writing firmware v%d.%d.%d on %d/%d/%d (y/m/d)\n",
296 hdr
->fwdate
[0], hdr
->fwdate
[1], hdr
->fwdate
[2]);
299 while ((n
= fread(buf
+ off
, 1, sizeof(buf
)-off
, in
) + off
) > 0) {
301 if (n
< sizeof(buf
)) {
304 fprintf(stderr
, "fread error\n");
308 gflag
= sizeof(buf
) - n
;
309 memset(buf
+ n
, 0xff, gflag
);
310 fprintf(stderr
, "adding %d bytes of garbage\n", gflag
);
314 if (!fwrite(buf
, n
, 1, out
)) {
316 fprintf(stderr
, "fwrite error\n");