tools: firmware-utils: fix compiler warnings
[openwrt/openwrt.git] / tools / firmware-utils / src / asustrx.c
1 /*
2 * asustrx
3 *
4 * Copyright (C) 2015 Rafał Miłecki <zajec5@gmail.com>
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the Free
8 * Software Foundation; either version 2 of the License, or (at your option)
9 * any later version.
10 */
11
12 #include <byteswap.h>
13 #include <errno.h>
14 #include <stdint.h>
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <string.h>
18 #include <unistd.h>
19
20 #if __BYTE_ORDER == __BIG_ENDIAN
21 #define cpu_to_le32(x) bswap_32(x)
22 #define le32_to_cpu(x) bswap_32(x)
23 #elif __BYTE_ORDER == __LITTLE_ENDIAN
24 #define cpu_to_le32(x) (x)
25 #define le32_to_cpu(x) (x)
26 #else
27 #error "Unsupported endianness"
28 #endif
29
30 #define TRX_MAGIC 0x30524448
31 #define TRX_FLAGS_OFFSET 12
32
33 struct trx_header {
34 uint32_t magic;
35 uint32_t length;
36 uint32_t crc32;
37 uint16_t flags;
38 uint16_t version;
39 uint32_t offset[3];
40 };
41
42 struct asustrx_tail {
43 uint8_t version[4];
44 char productid[12];
45 uint8_t unused[48];
46 };
47
48 char *in_path = NULL;
49 char *out_path = NULL;
50 char *productid = NULL;
51 uint8_t version[4] = { };
52
53 static const uint32_t crc32_tbl[] = {
54 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba,
55 0x076dc419, 0x706af48f, 0xe963a535, 0x9e6495a3,
56 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
57 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91,
58 0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
59 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
60 0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec,
61 0x14015c4f, 0x63066cd9, 0xfa0f3d63, 0x8d080df5,
62 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
63 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
64 0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940,
65 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
66 0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116,
67 0x21b4f4b5, 0x56b3c423, 0xcfba9599, 0xb8bda50f,
68 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
69 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d,
70 0x76dc4190, 0x01db7106, 0x98d220bc, 0xefd5102a,
71 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
72 0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818,
73 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
74 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
75 0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457,
76 0x65b0d9c6, 0x12b7e950, 0x8bbeb8ea, 0xfcb9887c,
77 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
78 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
79 0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb,
80 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
81 0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9,
82 0x5005713c, 0x270241aa, 0xbe0b1010, 0xc90c2086,
83 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
84 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4,
85 0x59b33d17, 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad,
86 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
87 0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683,
88 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
89 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
90 0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe,
91 0xf762575d, 0x806567cb, 0x196c3671, 0x6e6b06e7,
92 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
93 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
94 0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252,
95 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
96 0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60,
97 0xdf60efc3, 0xa867df55, 0x316e8eef, 0x4669be79,
98 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
99 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f,
100 0xc5ba3bbe, 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04,
101 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
102 0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a,
103 0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
104 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
105 0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21,
106 0x86d3d2d4, 0xf1d4e242, 0x68ddb3f8, 0x1fda836e,
107 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
108 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
109 0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45,
110 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
111 0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db,
112 0xaed16a4a, 0xd9d65adc, 0x40df0b66, 0x37d83bf0,
113 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
114 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6,
115 0xbad03605, 0xcdd70693, 0x54de5729, 0x23d967bf,
116 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
117 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d,
118 };
119
120 static void parse_options(int argc, char **argv) {
121 int c;
122
123 while ((c = getopt(argc, argv, "i:o:p:v:")) != -1) {
124 switch (c) {
125 case 'i':
126 in_path = optarg;
127 break;
128 case 'o':
129 out_path = optarg;
130 break;
131 case 'p':
132 productid = optarg;
133 break;
134 case 'v':
135 if (sscanf(optarg, "%hu.%hu.%hu.%hu", &version[0], &version[1], &version[2], &version[3]) != 4)
136 fprintf(stderr, "Version %s doesn't match suppored 4-digits format\n", optarg);
137 break;
138 }
139 }
140 }
141
142 static void usage() {
143 printf("Usage:\n");
144 printf("\t-i file\t\t\t\tinput TRX file\n");
145 printf("\t-o file\t\t\t\toutput Asus TRX file\n");
146 printf("\t-p productid\t\t\tproduct (device) ID\n");
147 printf("\t-v version\t\t\tfirmware version formatted with 4 digits like: 1.2.3.4\n");
148 }
149
150 int main(int argc, char **argv) {
151 struct trx_header hdr;
152 struct asustrx_tail tail = { };
153 FILE *in, *out;
154 uint8_t buf[1024];
155 size_t bytes;
156 size_t length = 0;
157 uint32_t crc32 = 0xffffffff;
158 int i;
159 int err = 0;
160
161 /* Parse & validate arguments */
162 parse_options(argc, argv);
163 if (!in_path || !out_path || !productid) {
164 usage();
165 err = -EINVAL;
166 goto err;
167 }
168
169 /* Fill Asus tail */
170 tail.version[0] = version[0];
171 tail.version[1] = version[1];
172 tail.version[2] = version[2];
173 tail.version[3] = version[3];
174 strncpy(tail.productid, productid, sizeof(tail.productid));
175
176 /* Open files */
177 in = fopen(in_path, "r");
178 if (!in) {
179 fprintf(stderr, "Couldn't open %s\n", in_path);
180 err = -EIO;
181 goto err;
182 }
183 out = fopen(out_path, "w+");
184 if (!out) {
185 fprintf(stderr, "Couldn't open %s\n", out_path);
186 err = -EIO;
187 goto err;
188 }
189
190 /* Check is there is empty place for Asus tail */
191 bytes = sizeof(struct asustrx_tail);
192 fseek(in, -bytes, SEEK_END);
193 if (fread(buf, 1, bytes, in) != bytes) {
194 fprintf(stderr, "Couldn't read %zu B from %s\n", bytes, in_path);
195 err = -EIO;
196 goto err;
197 }
198 for (i = 0; i < bytes; i++) {
199 if (buf[i]) {
200 fprintf(stderr, "Input TRX doesn't have last 64 B empty %s\n", out_path);
201 err = -ENOSPC;
202 goto err;
203 }
204 }
205
206 /* Copy whole TRX */
207 rewind(in);
208 while ((bytes = fread(buf, 1, sizeof(buf), in)) > 0) {
209 if (fwrite(buf, 1, bytes, out) != bytes) {
210 fprintf(stderr, "Couldn't write %zu B to %s\n", bytes, out_path);
211 err = -EIO;
212 goto err;
213 }
214 }
215
216 /* Overwrite last 64 B with Asus tail */
217 bytes = sizeof(tail);
218 fseek(out, -bytes, SEEK_CUR);
219 if (fwrite(&tail, 1, bytes, out) != bytes) {
220 fprintf(stderr, "Couldn't write %zu B to %s\n", bytes, out_path);
221 err = -EIO;
222 goto err;
223 }
224
225 /* Calculate crc32 */
226 fseek(out, TRX_FLAGS_OFFSET, SEEK_SET);
227 length = TRX_FLAGS_OFFSET;
228 while ((bytes = fread(buf, 1, sizeof(buf), out )) > 0) {
229 length += bytes;
230 for (i = 0; i < bytes; i++)
231 crc32 = crc32_tbl[(crc32 ^ buf[i]) & 0xff] ^ (crc32 >> 8);
232 }
233
234 /* Update header */
235 bytes = sizeof(hdr);
236 rewind(out);
237 if (fread(&hdr, 1, sizeof(hdr), out) != bytes) {
238 fprintf(stderr, "Couldn't read %zu B from %s\n", bytes, in_path);
239 err = -EIO;
240 goto err;
241 }
242 hdr.crc32 = cpu_to_le32(crc32);
243 rewind(out);
244 if (fwrite(&hdr, 1, bytes, out) != bytes) {
245 fprintf(stderr, "Couldn't write %zu B to %s\n", bytes, out_path);
246 err = -EIO;
247 goto err;
248 }
249
250 err:
251 if (out)
252 fclose(out);
253 if (in)
254 fclose(in);
255 return err;
256 }