1 // SPDX-License-Identifier: GPL-2.0-only
3 * Copyright (C) 2009-2011 Gabor Juhos <juhosg@openwrt.org>
11 #include <getopt.h> /* for getopt() */
12 #include <netinet/in.h>
14 #include "buffalo-lib.h"
16 #define ERR(fmt, ...) do { \
18 fprintf(stderr, "[%s] *** error: " fmt "\n", \
19 progname, ## __VA_ARGS__ ); \
22 static char *region_table
[] = {
23 "JP", "US", "EU", "AP", "TW", "KR"
26 #define MAX_INPUT_FILES 2
28 static char *progname
;
29 static char *ifname
[MAX_INPUT_FILES
];
30 static ssize_t fsize
[MAX_INPUT_FILES
];
35 static char *language
;
37 static char *platform
;
40 static char *minor
= "1.01";
42 static uint32_t base1
;
43 static uint32_t base2
;
44 static char *region_code
;
45 static uint32_t region_mask
;
46 static int num_regions
;
49 void usage(int status
)
51 FILE *stream
= (status
!= EXIT_SUCCESS
) ? stderr
: stdout
;
53 fprintf(stream
, "Usage: %s [OPTIONS...]\n", progname
);
57 " -a <platform> set platform to <platform>\n"
58 " -b <brand> set brand to <brand>\n"
61 " -f <flag> set flag to <flag>\n"
62 " -i <file> read input from the file <file>\n"
63 " -I <file> read input from the file <file> for DHP series\n"
64 " -l <language> set language to <language>\n"
65 " -m <version> set minor version to <version>\n"
66 " -o <file> write output to the file <file>\n"
67 " -p <product> set product to <product>\n"
68 " -r <region> set image region to <region>\n"
69 " valid regions: JP, US, EU, AP, TW, KR, M_\n"
70 " -s skip CRC calculation\n"
71 " -v <version> set major version to <version>\n"
72 " -w <version> set harwdware version to <version>\n"
73 " -h show this screen\n"
79 static int check_params(void)
82 #define CHECKSTR(_var, _name, _len) do { \
83 if ((_var) == NULL) { \
84 ERR("no %s specified", (_name)); \
88 strlen((_var)) > ((_len) - 1)) { \
89 ERR("%s is too long", (_name)); \
95 ERR("no input files specified");
97 CHECKSTR(ofname
, "output file", 0);
98 CHECKSTR(brand
, "brand", TAG_BRAND_LEN
);
99 CHECKSTR(product
, "product", TAG_PRODUCT_LEN
);
100 CHECKSTR(platform
, "platform", TAG_PLATFORM_LEN
);
101 CHECKSTR(major
, "major version", TAG_VERSION_LEN
);
102 CHECKSTR(minor
, "minor version", TAG_VERSION_LEN
);
103 CHECKSTR(language
, "language", TAG_LANGUAGE_LEN
);
106 CHECKSTR(hwver
, "hardware version", 2);
108 if (num_regions
== 0) {
109 ERR("no region code specified");
118 static int process_region(char *reg
)
122 if (strlen(reg
) != 2) {
123 ERR("invalid region code '%s'", reg
);
127 if (strcmp(reg
, "M_") == 0) {
134 for (i
= 0; i
< ARRAY_SIZE(region_table
); i
++)
135 if (strcmp(reg
, region_table
[i
]) == 0) {
137 region_mask
|= 1 << i
;
142 ERR("unknown region code '%s'", reg
);
146 static int process_ifname(char *name
)
148 if (num_files
>= ARRAY_SIZE(ifname
)) {
149 ERR("too many input files specified");
153 ifname
[num_files
++] = name
;
157 static void fixup_tag(unsigned char *buf
, ssize_t buflen
)
159 struct buffalo_tag
*tag
= (struct buffalo_tag
*) buf
;
161 memset(tag
, '\0', sizeof(*tag
));
163 memcpy(tag
->brand
, brand
, strlen(brand
));
164 memcpy(tag
->product
, product
, strlen(product
));
165 memcpy(tag
->platform
, platform
, strlen(platform
));
166 memcpy(tag
->ver_major
, major
, strlen(major
));
167 memcpy(tag
->ver_minor
, minor
, strlen(minor
));
168 memcpy(tag
->language
, language
, strlen(language
));
170 if (num_regions
> 1) {
171 tag
->region_code
[0] = 'M';
172 tag
->region_code
[1] = '_';
173 tag
->region_mask
= htonl(region_mask
);
175 memcpy(tag
->region_code
, region_code
, 2);
178 tag
->len
= htonl(buflen
);
179 tag
->data_len
= htonl(fsize
[0]);
180 tag
->base1
= htonl(base1
);
181 tag
->base2
= htonl(base2
);
185 memcpy(tag
->hwv
, "hwv", 3);
186 memcpy(tag
->hwv_val
, hwver
, strlen(hwver
));
190 tag
->crc
= htonl(buffalo_crc(buf
, buflen
));
193 static void fixup_tag2(unsigned char *buf
, ssize_t buflen
)
195 struct buffalo_tag2
*tag
= (struct buffalo_tag2
*) buf
;
197 memset(tag
, '\0', sizeof(*tag
));
199 memcpy(tag
->brand
, brand
, strlen(brand
));
200 memcpy(tag
->product
, product
, strlen(product
));
201 memcpy(tag
->platform
, platform
, strlen(platform
));
202 memcpy(tag
->ver_major
, major
, strlen(major
));
203 memcpy(tag
->ver_minor
, minor
, strlen(minor
));
204 memcpy(tag
->language
, language
, strlen(language
));
206 if (num_regions
> 1) {
207 tag
->region_code
[0] = 'M';
208 tag
->region_code
[1] = '_';
209 tag
->region_mask
= htonl(region_mask
);
211 memcpy(tag
->region_code
, region_code
, 2);
214 tag
->total_len
= htonl(buflen
);
215 tag
->len1
= htonl(fsize
[0]);
216 tag
->len2
= htonl(fsize
[1]);
220 memcpy(tag
->hwv
, "hwv", 3);
221 memcpy(tag
->hwv_val
, hwver
, strlen(hwver
));
225 tag
->crc
= htonl(buffalo_crc(buf
, buflen
));
228 static void fixup_tag3(unsigned char *buf
, ssize_t totlen
)
230 struct buffalo_tag3
*tag
= (struct buffalo_tag3
*) buf
;
232 memset(tag
, '\0', sizeof(*tag
));
234 memcpy(tag
->brand
, brand
, strlen(brand
));
235 memcpy(tag
->product
, product
, strlen(product
));
236 memcpy(tag
->platform
, platform
, strlen(platform
));
237 memcpy(tag
->ver_major
, major
, strlen(major
));
238 memcpy(tag
->ver_minor
, minor
, strlen(minor
));
239 memcpy(tag
->language
, language
, strlen(language
));
241 if (num_regions
> 1) {
242 tag
->region_code
[0] = 'M';
243 tag
->region_code
[1] = '_';
244 tag
->region_mask
= htonl(region_mask
);
246 memcpy(tag
->region_code
, region_code
, 2);
249 tag
->total_len
= htonl(totlen
);
250 tag
->len1
= htonl(fsize
[0]);
251 tag
->base2
= htonl(base2
);
254 memcpy(tag
->hwv
, "hwv", 3);
255 memcpy(tag
->hwv_val
, hwver
, strlen(hwver
));
259 static int tag_file(void)
270 hdrlen
= sizeof(struct buffalo_tag3
);
271 else if (num_files
== 1)
272 hdrlen
= sizeof(struct buffalo_tag
);
274 hdrlen
= sizeof(struct buffalo_tag2
);
278 for (i
= 0; i
< num_files
; i
++) {
279 fsize
[i
] = get_file_size(ifname
[i
]);
281 ERR("unable to get size of '%s'", ifname
[i
]);
287 buf
= malloc(buflen
);
289 ERR("no memory for buffer\n");
294 for (i
= 0; i
< num_files
; i
++) {
295 err
= read_file_to_buf(ifname
[i
], buf
+ offset
, fsize
[i
]);
297 ERR("unable to read from file '%s'", ifname
[i
]);
305 fixup_tag3(buf
, fsize
[0] + 200);
306 else if (num_files
== 1)
307 fixup_tag(buf
, buflen
);
309 fixup_tag2(buf
, buflen
);
311 err
= write_buf_to_file(ofname
, buf
, buflen
);
313 ERR("unable to write to file '%s'", ofname
);
325 int main(int argc
, char *argv
[])
327 int res
= EXIT_FAILURE
;
330 progname
= basename(argv
[0]);
335 c
= getopt(argc
, argv
, "a:b:c:d:f:hi:l:m:o:p:r:sv:w:I:");
347 base1
= strtoul(optarg
, NULL
, 16);
350 base2
= strtoul(optarg
, NULL
, 16);
353 flag
= strtoul(optarg
, NULL
, 2);
359 err
= process_ifname(optarg
);
376 err
= process_region(optarg
);
398 err
= check_params();