1 | /* |
---|
2 | * Copyright (C) 2004 Manuel Novoa III <mjn3@codepoet.org> |
---|
3 | * |
---|
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. |
---|
8 | * |
---|
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. |
---|
13 | * |
---|
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 |
---|
17 | */ |
---|
18 | |
---|
19 | /* July 29, 2004 |
---|
20 | * |
---|
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 |
---|
23 | * the job for me. |
---|
24 | * |
---|
25 | * Extensions: |
---|
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. |
---|
29 | */ |
---|
30 | |
---|
31 | /* January 12, 2005 |
---|
32 | * |
---|
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. |
---|
36 | * |
---|
37 | * Extensions: |
---|
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 |
---|
40 | */ |
---|
41 | |
---|
42 | #include <stdio.h> |
---|
43 | #include <stdlib.h> |
---|
44 | #include <string.h> |
---|
45 | #include <time.h> |
---|
46 | #include <unistd.h> |
---|
47 | #include <sys/stat.h> |
---|
48 | |
---|
49 | /**********************************************************************/ |
---|
50 | |
---|
51 | #define CODE_ID "U2ND" /* from code_pattern.h */ |
---|
52 | #define CODE_PATTERN "W54S" /* from code_pattern.h */ |
---|
53 | #define PBOT_PATTERN "PBOT" |
---|
54 | |
---|
55 | #define CYBERTAN_VERSION "v3.37.2" /* from cyutils.h */ |
---|
56 | |
---|
57 | /* WRT54G v2.2 and WRT54GS v1.1 "flags" (from 3.37.32 firmware cyutils.h) */ |
---|
58 | #define SUPPORT_4712_CHIP 0x0001 |
---|
59 | #define SUPPORT_INTEL_FLASH 0x0002 |
---|
60 | #define SUPPORT_5325E_SWITCH 0x0004 |
---|
61 | |
---|
62 | struct code_header { /* from cyutils.h */ |
---|
63 | char magic[4]; |
---|
64 | char res1[4]; /* for extra magic */ |
---|
65 | char fwdate[3]; |
---|
66 | char fwvern[3]; |
---|
67 | char id[4]; /* U2ND */ |
---|
68 | char hw_ver; /* 0: for 4702, 1: for 4712 -- new in 2.04.3 */ |
---|
69 | char unused; |
---|
70 | unsigned char flags[2]; /* SUPPORT_ flags new for 3.37.2 (WRT54G v2.2 and WRT54GS v1.1) */ |
---|
71 | unsigned char res2[10]; |
---|
72 | } ; |
---|
73 | |
---|
74 | /**********************************************************************/ |
---|
75 | |
---|
76 | void usage(void) __attribute__ (( __noreturn__ )); |
---|
77 | |
---|
78 | void usage(void) |
---|
79 | { |
---|
80 | fprintf(stderr, "Usage: addpattern [-i trxfile] [-o binfile] [-p pattern] [-g] [-b] [-v v#.#.#] [-{0|1|2|4}]\n"); |
---|
81 | exit(EXIT_FAILURE); |
---|
82 | } |
---|
83 | |
---|
84 | int main(int argc, char **argv) |
---|
85 | { |
---|
86 | char buf[1024]; /* keep this at 1k or adjust garbage calc below */ |
---|
87 | struct code_header *hdr; |
---|
88 | FILE *in = stdin; |
---|
89 | FILE *out = stdout; |
---|
90 | char *ifn = NULL; |
---|
91 | char *ofn = NULL; |
---|
92 | char *pattern = CODE_PATTERN; |
---|
93 | char *pbotpat = PBOT_PATTERN; |
---|
94 | char *version = CYBERTAN_VERSION; |
---|
95 | int gflag = 0; |
---|
96 | int pbotflag = 0; |
---|
97 | int c; |
---|
98 | int v0, v1, v2; |
---|
99 | size_t off, n; |
---|
100 | time_t t; |
---|
101 | struct tm *ptm; |
---|
102 | |
---|
103 | fprintf(stderr, "mjn3's addpattern replacement - v0.81\n"); |
---|
104 | |
---|
105 | hdr = (struct code_header *) buf; |
---|
106 | memset(hdr, 0, sizeof(struct code_header)); |
---|
107 | |
---|
108 | while ((c = getopt(argc, argv, "i:o:p:gbv:0124")) != -1) { |
---|
109 | switch (c) { |
---|
110 | case 'i': |
---|
111 | ifn = optarg; |
---|
112 | break; |
---|
113 | case 'o': |
---|
114 | ofn = optarg; |
---|
115 | break; |
---|
116 | case 'p': |
---|
117 | pattern = optarg; |
---|
118 | break; |
---|
119 | case 'g': |
---|
120 | gflag = 1; |
---|
121 | break; |
---|
122 | case 'b': |
---|
123 | pbotflag = 1; |
---|
124 | break; |
---|
125 | case 'v': /* extension to allow setting version */ |
---|
126 | version = optarg; |
---|
127 | break; |
---|
128 | case '0': |
---|
129 | hdr->hw_ver = 0; |
---|
130 | break; |
---|
131 | case '1': |
---|
132 | hdr->hw_ver = 1; |
---|
133 | break; |
---|
134 | case '2': /* new 54G v2.2 and 54GS v1.1 flags */ |
---|
135 | hdr->hw_ver = 1; |
---|
136 | hdr->flags[0] |= SUPPORT_4712_CHIP; |
---|
137 | hdr->flags[0] |= SUPPORT_INTEL_FLASH; |
---|
138 | hdr->flags[0] |= SUPPORT_5325E_SWITCH; |
---|
139 | break; |
---|
140 | case '4': |
---|
141 | /* V4 firmware sets the flags to 0x1f */ |
---|
142 | hdr->hw_ver = 0; |
---|
143 | hdr->flags[0] = 0x1f; |
---|
144 | break; |
---|
145 | |
---|
146 | default: |
---|
147 | usage(); |
---|
148 | } |
---|
149 | } |
---|
150 | |
---|
151 | if (optind != argc) { |
---|
152 | fprintf(stderr, "illegal arg \"%s\"\n", argv[optind]); |
---|
153 | usage(); |
---|
154 | } |
---|
155 | |
---|
156 | if (strlen(pattern) != 4) { |
---|
157 | fprintf(stderr, "illegal pattern \"%s\": length != 4\n", pattern); |
---|
158 | usage(); |
---|
159 | } |
---|
160 | |
---|
161 | if (ifn && !(in = fopen(ifn, "r"))) { |
---|
162 | fprintf(stderr, "can not open \"%s\" for reading\n", ifn); |
---|
163 | usage(); |
---|
164 | } |
---|
165 | |
---|
166 | if (ofn && !(out = fopen(ofn, "w"))) { |
---|
167 | fprintf(stderr, "can not open \"%s\" for writing\n", ofn); |
---|
168 | usage(); |
---|
169 | } |
---|
170 | |
---|
171 | if (time(&t) == (time_t)(-1)) { |
---|
172 | fprintf(stderr, "time call failed\n"); |
---|
173 | return EXIT_FAILURE; |
---|
174 | } |
---|
175 | |
---|
176 | ptm = localtime(&t); |
---|
177 | |
---|
178 | if (3 != sscanf(version, "v%d.%d.%d", &v0, &v1, &v2)) { |
---|
179 | fprintf(stderr, "bad version string \"%s\"\n", version); |
---|
180 | return EXIT_FAILURE; |
---|
181 | } |
---|
182 | |
---|
183 | memcpy(&hdr->magic, pattern, 4); |
---|
184 | if (pbotflag) |
---|
185 | memcpy(&hdr->res1, pbotpat, 4); |
---|
186 | hdr->fwdate[0] = ptm->tm_year % 100; |
---|
187 | hdr->fwdate[1] = ptm->tm_mon + 1; |
---|
188 | hdr->fwdate[2] = ptm->tm_mday; |
---|
189 | hdr->fwvern[0] = v0; |
---|
190 | hdr->fwvern[1] = v1; |
---|
191 | hdr->fwvern[2] = v2; |
---|
192 | memcpy(&hdr->id, CODE_ID, strlen(CODE_ID)); |
---|
193 | |
---|
194 | off = sizeof(struct code_header); |
---|
195 | |
---|
196 | fprintf(stderr, "writing firmware v%d.%d.%d on %d/%d/%d (y/m/d)\n", |
---|
197 | v0, v1, v2, |
---|
198 | hdr->fwdate[0], hdr->fwdate[1], hdr->fwdate[2]); |
---|
199 | |
---|
200 | |
---|
201 | while ((n = fread(buf + off, 1, sizeof(buf)-off, in) + off) > 0) { |
---|
202 | off = 0; |
---|
203 | if (n < sizeof(buf)) { |
---|
204 | if (ferror(in)) { |
---|
205 | FREAD_ERROR: |
---|
206 | fprintf(stderr, "fread error\n"); |
---|
207 | return EXIT_FAILURE; |
---|
208 | } |
---|
209 | if (gflag) { |
---|
210 | gflag = sizeof(buf) - n; |
---|
211 | memset(buf + n, 0xff, gflag); |
---|
212 | fprintf(stderr, "adding %d bytes of garbage\n", gflag); |
---|
213 | n = sizeof(buf); |
---|
214 | } |
---|
215 | } |
---|
216 | if (!fwrite(buf, n, 1, out)) { |
---|
217 | FWRITE_ERROR: |
---|
218 | fprintf(stderr, "fwrite error\n"); |
---|
219 | return EXIT_FAILURE; |
---|
220 | } |
---|
221 | } |
---|
222 | |
---|
223 | if (ferror(in)) { |
---|
224 | goto FREAD_ERROR; |
---|
225 | } |
---|
226 | |
---|
227 | if (fflush(out)) { |
---|
228 | goto FWRITE_ERROR; |
---|
229 | } |
---|
230 | |
---|
231 | fclose(in); |
---|
232 | fclose(out); |
---|
233 | |
---|
234 | return EXIT_SUCCESS; |
---|
235 | } |
---|