backfire: firmware-utils/mkdniimg: don't use NA region by default (backport of r24981)
[openwrt-10.03/.git] / tools / firmware-utils / src / mkdniimg.c
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 #define DNI_HDR_LEN     128
22
23 /*
24  * Globals
25  */
26 static char *ifname;
27 static char *progname;
28 static char *ofname;
29 static char *version = "1.00.00";
30 static char *region = "";
31
32 static char *board_id;
33 /*
34  * Message macros
35  */
36 #define ERR(fmt, ...) do { \
37         fflush(0); \
38         fprintf(stderr, "[%s] *** error: " fmt "\n", \
39                         progname, ## __VA_ARGS__ ); \
40 } while (0)
41
42 #define ERRS(fmt, ...) do { \
43         int save = errno; \
44         fflush(0); \
45         fprintf(stderr, "[%s] *** error: " fmt "\n", \
46                         progname, ## __VA_ARGS__, strerror(save)); \
47 } while (0)
48
49 void usage(int status)
50 {
51         FILE *stream = (status != EXIT_SUCCESS) ? stderr : stdout;
52         struct board_info *board;
53
54         fprintf(stream, "Usage: %s [OPTIONS...]\n", progname);
55         fprintf(stream,
56 "\n"
57 "Options:\n"
58 "  -B <board>      create image for the board specified with <board>\n"
59 "  -i <file>       read input from the file <file>\n"
60 "  -o <file>       write output to the file <file>\n"
61 "  -v <version>    set image version to <version>\n"
62 "  -r <region>     set image region to <region>\n"
63 "  -h              show this screen\n"
64         );
65
66         exit(status);
67 }
68
69 int main(int argc, char *argv[])
70 {
71         int res = EXIT_FAILURE;
72         int buflen;
73         int err;
74         struct stat st;
75         char *buf;
76         int i;
77         uint8_t csum;
78
79         FILE *outfile, *infile;
80
81         progname = basename(argv[0]);
82
83         while ( 1 ) {
84                 int c;
85
86                 c = getopt(argc, argv, "B:i:o:v:r:h");
87                 if (c == -1)
88                         break;
89
90                 switch (c) {
91                 case 'B':
92                         board_id = optarg;
93                         break;
94                 case 'i':
95                         ifname = optarg;
96                         break;
97                 case 'o':
98                         ofname = optarg;
99                         break;
100                 case 'v':
101                         version = optarg;
102                         break;
103                 case 'r':
104                         region = optarg;
105                         break;
106                 case 'h':
107                         usage(EXIT_SUCCESS);
108                         break;
109                 default:
110                         usage(EXIT_FAILURE);
111                         break;
112                 }
113         }
114
115         if (board_id == NULL) {
116                 ERR("no board specified");
117                 goto err;
118         }
119
120         if (ifname == NULL) {
121                 ERR("no input file specified");
122                 goto err;
123         }
124
125         if (ofname == NULL) {
126                 ERR("no output file specified");
127                 goto err;
128         }
129
130         err = stat(ifname, &st);
131         if (err){
132                 ERRS("stat failed on %s", ifname);
133                 goto err;
134         }
135
136         buflen = st.st_size + DNI_HDR_LEN + 1;
137         buf = malloc(buflen);
138         if (!buf) {
139                 ERR("no memory for buffer\n");
140                 goto err;
141         }
142
143         memset(buf, 0, DNI_HDR_LEN);
144         snprintf(buf, DNI_HDR_LEN, "device:%s\nversion:V%s\nregion:%s\n",
145                  board_id, version, region);
146
147         infile = fopen(ifname, "r");
148         if (infile == NULL) {
149                 ERRS("could not open \"%s\" for reading", ifname);
150                 goto err_free;
151         }
152
153         errno = 0;
154         fread(buf +  DNI_HDR_LEN, st.st_size, 1, infile);
155         if (errno != 0) {
156                 ERRS("unable to read from file %s", ifname);
157                 goto err_close_in;
158         }
159
160         csum = 0;
161         for (i = 0; i < (st.st_size + DNI_HDR_LEN); i++)
162                 csum += buf[i];
163
164         csum = 0xff - csum;
165         buf[st.st_size + DNI_HDR_LEN] = csum;
166
167         outfile = fopen(ofname, "w");
168         if (outfile == NULL) {
169                 ERRS("could not open \"%s\" for writing", ofname);
170                 goto err_close_in;
171         }
172
173         errno = 0;
174         fwrite(buf, buflen, 1, outfile);
175         if (errno) {
176                 ERRS("unable to write to file %s", ofname);
177                 goto err_close_out;
178         }
179
180         res = EXIT_SUCCESS;
181
182  out_flush:
183         fflush(outfile);
184
185  err_close_out:
186         fclose(outfile);
187         if (res != EXIT_SUCCESS) {
188                 unlink(ofname);
189         }
190
191  err_close_in:
192         fclose(infile);
193
194  err_free:
195         free(buf);
196
197  err:
198         return res;
199 }