ar7: add eva image generation (thanks Axel Gembe)
[openwrt/openwrt.git] / tools / firmware-utils / src / lzma2eva.c
1 /*
2 lzma2eva - convert lzma-compressed file to AVM EVA bootloader format
3 Copyright (C) 2007 Enrik Berkhan <Enrik.Berkhan@inka.de>
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
9
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18 */
19
20 #include <stdint.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <zlib.h> /* crc32 */
24
25 #define checksum_add32(csum, data) \
26 csum += ((uint8_t *)&data)[0]; \
27 csum += ((uint8_t *)&data)[1]; \
28 csum += ((uint8_t *)&data)[2]; \
29 csum += ((uint8_t *)&data)[3];
30
31 void
32 usage(void)
33 {
34 fprintf(stderr, "usage: lzma2eva <loadadddr> <entry> <lzmafile> <evafile>\n");
35 exit(1);
36 }
37
38 void
39 pexit(const char *msg)
40 {
41 perror(msg);
42 exit(1);
43 }
44
45 int
46 main(int argc, char *argv[])
47 {
48
49 const char *infile, *outfile;
50 FILE *in, *out;
51 static const uint8_t buf[4096];
52 size_t elems;
53
54 uint8_t properties;
55 uint32_t dictsize;
56 uint64_t datasize;
57
58 uint32_t magic = 0xfeed1281L;
59 uint32_t reclength = 0;
60 fpos_t reclengthpos;
61 uint32_t loadaddress = 0;
62 uint32_t type = 0x075a0201L; /* might be 7Z 2.1? */
63 uint32_t checksum = 0;
64
65 uint32_t compsize = 0;
66 fpos_t compsizepos;
67 uint32_t datasize32 = 0;
68 uint32_t datacrc32 = crc32(0, 0, 0);
69
70 uint32_t zero = 0;
71 uint32_t entry = 0;
72
73 if (argc != 5)
74 usage();
75
76 /* "parse" command line */
77 loadaddress = strtoul(argv[1], 0, 0);
78 entry = strtoul(argv[2], 0, 0);
79 infile = argv[3];
80 outfile = argv[4];
81
82 in = fopen(infile, "rb");
83 if (!in)
84 pexit("fopen");
85 out = fopen(outfile, "w+b");
86 if (!out)
87 pexit("fopen");
88
89 /* read LZMA header */
90 if (1 != fread(&properties, sizeof properties, 1, in))
91 pexit("fread");
92 if (1 != fread(&dictsize, sizeof dictsize, 1, in))
93 pexit("fread");
94 if (1 != fread(&datasize, sizeof datasize, 1, in))
95 pexit("fread");
96
97 /* write EVA header */
98 if (1 != fwrite(&magic, sizeof magic, 1, out))
99 pexit("fwrite");
100 if (fgetpos(out, &reclengthpos))
101 pexit("fgetpos");
102 if (1 != fwrite(&reclength, sizeof reclength, 1, out))
103 pexit("fwrite");
104 if (1 != fwrite(&loadaddress, sizeof loadaddress, 1, out))
105 pexit("fwrite");
106 if (1 != fwrite(&type, sizeof type, 1, out))
107 pexit("fwrite");
108
109 /* write EVA LZMA header */
110 if (fgetpos(out, &compsizepos))
111 pexit("fgetpos");
112 if (1 != fwrite(&compsize, sizeof compsize, 1, out))
113 pexit("fwrite");
114 /* XXX check length */
115 datasize32 = (uint32_t)datasize;
116 if (1 != fwrite(&datasize32, sizeof datasize32, 1, out))
117 pexit("fwrite");
118 if (1 != fwrite(&datacrc32, sizeof datacrc32, 1, out))
119 pexit("fwrite");
120
121 /* write modified LZMA header */
122 if (1 != fwrite(&properties, sizeof properties, 1, out))
123 pexit("fwrite");
124 if (1 != fwrite(&dictsize, sizeof dictsize, 1, out))
125 pexit("fwrite");
126 if (1 != fwrite(&zero, 3, 1, out))
127 pexit("fwrite");
128
129 /* copy compressed data, calculate crc32 */
130 while (0 < (elems = fread(&buf, sizeof buf[0], sizeof buf, in))) {
131 compsize += elems;
132 if (elems != fwrite(&buf, sizeof buf[0], elems, out))
133 pexit("fwrite");
134 datacrc32 = crc32(datacrc32, buf, elems);
135 }
136 if (ferror(in))
137 pexit("fread");
138 fclose(in);
139
140 /* re-write record length */
141 reclength = compsize + 24;
142 if (fsetpos(out, &reclengthpos))
143 pexit("fsetpos");
144 if (1 != fwrite(&reclength, sizeof reclength, 1, out))
145 pexit("fwrite");
146
147 /* re-write EVA LZMA header including size and data crc */
148 if (fsetpos(out, &compsizepos))
149 pexit("fsetpos");
150 if (1 != fwrite(&compsize, sizeof compsize, 1, out))
151 pexit("fwrite");
152 if (1 != fwrite(&datasize32, sizeof datasize32, 1, out))
153 pexit("fwrite");
154 if (1 != fwrite(&datacrc32, sizeof datacrc32, 1, out))
155 pexit("fwrite");
156
157 /* calculate record checksum */
158 checksum += reclength;
159 checksum += loadaddress;
160 checksum_add32(checksum, type);
161 checksum_add32(checksum, compsize);
162 checksum_add32(checksum, datasize32);
163 checksum_add32(checksum, datacrc32);
164 if (fseek(out, 0, SEEK_CUR))
165 pexit("fseek");
166 while (0 < (elems = fread(&buf, sizeof buf[0], sizeof buf, out))) {
167 size_t i;
168 for (i = 0; i < elems; ++i)
169 checksum += buf[i];
170 }
171 if (ferror(out))
172 pexit("fread");
173 if (fseek(out, 0, SEEK_CUR))
174 pexit("fseek");
175
176 checksum = ~checksum + 1;
177 if (1 != fwrite(&checksum, sizeof checksum, 1, out))
178 pexit("fwrite");
179
180 /* write entry record */
181 if (1 != fwrite(&zero, sizeof zero, 1, out))
182 pexit("fwrite");
183 if (1 != fwrite(&entry, sizeof entry, 1, out))
184 pexit("fwrite");
185
186 if (fclose(out))
187 pexit("fclose");
188
189 return 0;
190 }