OpenWrt – Blame information for rev 2

Subversion Repositories:
Rev:
Rev Author Line No. Line
1 office 1 /*
2 * Copyright (C) 2009 Gabor Juhos <juhosg@openwrt.org>
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License version 2 as published
6 * by the Free Software Foundation.
7 *
8 */
9  
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <stdint.h>
13 #include <string.h>
14 #include <unistd.h> /* for unlink() */
15 #include <libgen.h>
16 #include <getopt.h> /* for getopt() */
17 #include <stdarg.h>
18 #include <errno.h>
19 #include <sys/stat.h>
20  
21 #include "sha1.h"
22  
23 #if (__BYTE_ORDER == __BIG_ENDIAN)
24 # define HOST_TO_BE32(x) (x)
25 # define BE32_TO_HOST(x) (x)
26 #else
27 # define HOST_TO_BE32(x) bswap_32(x)
28 # define BE32_TO_HOST(x) bswap_32(x)
29 #endif
30  
31  
32 struct planex_hdr {
33 uint8_t sha1sum[20];
34 char version[8];
35 uint8_t unk1[2];
36 uint32_t datalen;
37 } __attribute__ ((packed));
38  
39 struct board_info {
40 char *id;
41 uint32_t seed;
42 uint8_t unk[2];
43 uint32_t datalen;
44 };
45  
46 /*
47 * Globals
48 */
49 static char *ifname;
50 static char *progname;
51 static char *ofname;
52 static char *version = "1.00.00";
53  
54 static char *board_id;
55 static struct board_info *board;
56  
57 static struct board_info boards[] = {
58 {
59 .id = "MZK-W04NU",
60 .seed = 2,
61 .unk = {0x04, 0x08},
62 .datalen = 0x770000,
63 }, {
64 .id = "MZK-W300NH",
65 .seed = 4,
66 .unk = {0x00, 0x00},
67 .datalen = 0x770000,
68 }, {
69 /* terminating entry */
70 }
71 };
72  
73 /*
74 * Message macros
75 */
76 #define ERR(fmt, ...) do { \
77 fflush(0); \
78 fprintf(stderr, "[%s] *** error: " fmt "\n", \
79 progname, ## __VA_ARGS__ ); \
80 } while (0)
81  
82 #define ERRS(fmt, ...) do { \
83 int save = errno; \
84 fflush(0); \
85 fprintf(stderr, "[%s] *** error: " fmt ": %s\n", \
86 progname, ## __VA_ARGS__, strerror(save)); \
87 } while (0)
88  
89 static struct board_info *find_board(char *id)
90 {
91 struct board_info *ret;
92 struct board_info *board;
93  
94 ret = NULL;
95 for (board = boards; board->id != NULL; board++){
96 if (strcasecmp(id, board->id) == 0) {
97 ret = board;
98 break;
99 }
100 };
101  
102 return ret;
103 }
104  
105 void usage(int status)
106 {
107 FILE *stream = (status != EXIT_SUCCESS) ? stderr : stdout;
108 struct board_info *board;
109  
110 fprintf(stream, "Usage: %s [OPTIONS...]\n", progname);
111 fprintf(stream,
112 "\n"
113 "Options:\n"
114 " -B <board> create image for the board specified with <board>\n"
115 " -i <file> read input from the file <file>\n"
116 " -o <file> write output to the file <file>\n"
117 " -v <version> set image version to <version>\n"
118 " -h show this screen\n"
119 );
120  
121 exit(status);
122 }
123  
124 int main(int argc, char *argv[])
125 {
126 int res = EXIT_FAILURE;
127 int buflen;
128 int err;
129 struct stat st;
130 char *buf;
131 struct planex_hdr *hdr;
132 sha1_context ctx;
133 uint32_t seed;
134  
135 FILE *outfile, *infile;
136  
137 progname = basename(argv[0]);
138  
139 while ( 1 ) {
140 int c;
141  
142 c = getopt(argc, argv, "B:i:o:v:h");
143 if (c == -1)
144 break;
145  
146 switch (c) {
147 case 'B':
148 board_id = optarg;
149 break;
150 case 'i':
151 ifname = optarg;
152 break;
153 case 'o':
154 ofname = optarg;
155 break;
156 case 'v':
157 version = optarg;
158 break;
159 case 'h':
160 usage(EXIT_SUCCESS);
161 break;
162 default:
163 usage(EXIT_FAILURE);
164 break;
165 }
166 }
167  
168 if (board_id == NULL) {
169 ERR("no board specified");
170 goto err;
171 }
172  
173 board = find_board(board_id);
174 if (board == NULL) {
175 ERR("unknown board '%s'", board_id);
176 goto err;
177 };
178  
179 if (ifname == NULL) {
180 ERR("no input file specified");
181 goto err;
182 }
183  
184 if (ofname == NULL) {
185 ERR("no output file specified");
186 goto err;
187 }
188  
189 err = stat(ifname, &st);
190 if (err){
191 ERRS("stat failed on %s", ifname);
192 goto err;
193 }
194  
195 if (st.st_size > board->datalen) {
196 ERR("file '%s' is too big - max size: 0x%08X (exceeds %lu bytes)\n",
197 ifname, board->datalen, st.st_size - board->datalen);
198 goto err;
199 }
200  
201 buflen = board->datalen + 0x10000;
202 buf = malloc(buflen);
203 if (!buf) {
204 ERR("no memory for buffer\n");
205 goto err;
206 }
207  
208 memset(buf, 0xff, buflen);
209 hdr = (struct planex_hdr *)buf;
210  
211 hdr->datalen = HOST_TO_BE32(board->datalen);
212 hdr->unk1[0] = board->unk[0];
213 hdr->unk1[1] = board->unk[1];
214  
215 snprintf(hdr->version, sizeof(hdr->version), "%s", version);
216  
217 infile = fopen(ifname, "r");
218 if (infile == NULL) {
219 ERRS("could not open \"%s\" for reading", ifname);
220 goto err_free;
221 }
222  
223 errno = 0;
224 fread(buf + sizeof(*hdr), st.st_size, 1, infile);
225 if (errno != 0) {
226 ERRS("unable to read from file %s", ifname);
227 goto err_close_in;
228 }
229  
230 seed = HOST_TO_BE32(board->seed);
231 sha1_starts(&ctx);
232 sha1_update(&ctx, (uchar *) &seed, sizeof(seed));
233 sha1_update(&ctx, buf + sizeof(*hdr), board->datalen);
234 sha1_finish(&ctx, hdr->sha1sum);
235  
236 outfile = fopen(ofname, "w");
237 if (outfile == NULL) {
238 ERRS("could not open \"%s\" for writing", ofname);
239 goto err_close_in;
240 }
241  
242 errno = 0;
243 fwrite(buf, buflen, 1, outfile);
244 if (errno) {
245 ERRS("unable to write to file %s", ofname);
246 goto err_close_out;
247 }
248  
249 res = EXIT_SUCCESS;
250  
251 out_flush:
252 fflush(outfile);
253  
254 err_close_out:
255 fclose(outfile);
256 if (res != EXIT_SUCCESS) {
257 unlink(ofname);
258 }
259  
260 err_close_in:
261 fclose(infile);
262  
263 err_free:
264 free(buf);
265  
266 err:
267 return res;
268 }
269