1ca987d46SWarner Losh /*-
2ca987d46SWarner Losh * Copyright (c) 2013 Hans Petter Selasky. All rights reserved.
3ca987d46SWarner Losh *
4ca987d46SWarner Losh * Redistribution and use in source and binary forms, with or without
5ca987d46SWarner Losh * modification, are permitted provided that the following conditions
6ca987d46SWarner Losh * are met:
7ca987d46SWarner Losh * 1. Redistributions of source code must retain the above copyright
8ca987d46SWarner Losh * notice, this list of conditions and the following disclaimer.
9ca987d46SWarner Losh * 2. Redistributions in binary form must reproduce the above copyright
10ca987d46SWarner Losh * notice, this list of conditions and the following disclaimer in the
11ca987d46SWarner Losh * documentation and/or other materials provided with the distribution.
12ca987d46SWarner Losh *
13ca987d46SWarner Losh * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
14ca987d46SWarner Losh * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15ca987d46SWarner Losh * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
16ca987d46SWarner Losh * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
17ca987d46SWarner Losh * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
18ca987d46SWarner Losh * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
19ca987d46SWarner Losh * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
20ca987d46SWarner Losh * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
21ca987d46SWarner Losh * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
22ca987d46SWarner Losh * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
23ca987d46SWarner Losh * SUCH DAMAGE.
24ca987d46SWarner Losh */
25ca987d46SWarner Losh
26ca987d46SWarner Losh /*
27ca987d46SWarner Losh * This utility sorts sysinit structure entries in binary format and
28ca987d46SWarner Losh * prints out the result in C-format.
29ca987d46SWarner Losh */
30ca987d46SWarner Losh
31ca987d46SWarner Losh #include <stdio.h>
32ca987d46SWarner Losh #include <stdint.h>
33ca987d46SWarner Losh #include <stdlib.h>
34ca987d46SWarner Losh #include <err.h>
35ca987d46SWarner Losh #include <unistd.h>
36ca987d46SWarner Losh #include <fcntl.h>
37ca987d46SWarner Losh #include <string.h>
38ca987d46SWarner Losh #include <sysexits.h>
39ca987d46SWarner Losh #include "sysinit.h"
40ca987d46SWarner Losh
41ca987d46SWarner Losh static int opt_R;
42ca987d46SWarner Losh static const char *input_f;
43ca987d46SWarner Losh static const char *output_f;
44ca987d46SWarner Losh static const char *struct_name;
45ca987d46SWarner Losh static const char *keyword;
46ca987d46SWarner Losh static struct sysinit_data **start;
47ca987d46SWarner Losh static struct sysinit_data **stop;
48ca987d46SWarner Losh
49ca987d46SWarner Losh static int input_file = -1;
50ca987d46SWarner Losh static int output_file = -1;
51ca987d46SWarner Losh
52ca987d46SWarner Losh static uint8_t *input_ptr;
53ca987d46SWarner Losh static uint32_t input_len;
54ca987d46SWarner Losh
55ca987d46SWarner Losh static uint32_t endian32;
56ca987d46SWarner Losh
57ca987d46SWarner Losh static char scratch_buf[4096];
58ca987d46SWarner Losh
59ca987d46SWarner Losh static int success;
60ca987d46SWarner Losh
61ca987d46SWarner Losh static void do_sysinit(void);
62ca987d46SWarner Losh
63ca987d46SWarner Losh /* the following function converts the numbers into host endian format */
64ca987d46SWarner Losh
65ca987d46SWarner Losh static uint32_t
read32(uint32_t val)66ca987d46SWarner Losh read32(uint32_t val)
67ca987d46SWarner Losh {
68ca987d46SWarner Losh uint32_t temp;
69ca987d46SWarner Losh uint32_t endian;
70ca987d46SWarner Losh
71ca987d46SWarner Losh endian = endian32;
72ca987d46SWarner Losh temp = 0;
73ca987d46SWarner Losh
74ca987d46SWarner Losh while (val) {
75ca987d46SWarner Losh temp |= (val & 0xF) << ((endian & 0xF) * 4);
76ca987d46SWarner Losh endian >>= 4;
77ca987d46SWarner Losh val >>= 4;
78ca987d46SWarner Losh }
79ca987d46SWarner Losh return (temp);
80ca987d46SWarner Losh }
81ca987d46SWarner Losh
82ca987d46SWarner Losh static void
do_write(int fd,const char * buf)83ca987d46SWarner Losh do_write(int fd, const char *buf)
84ca987d46SWarner Losh {
85ca987d46SWarner Losh int len = strlen(buf);
86ca987d46SWarner Losh
87ca987d46SWarner Losh if (write(fd, buf, len) != len)
88ca987d46SWarner Losh err(EX_SOFTWARE, "Could not write to output file");
89ca987d46SWarner Losh }
90ca987d46SWarner Losh
91ca987d46SWarner Losh static void *
do_malloc(int size)92ca987d46SWarner Losh do_malloc(int size)
93ca987d46SWarner Losh {
94ca987d46SWarner Losh void *ptr;
95ca987d46SWarner Losh
96ca987d46SWarner Losh ptr = malloc(size);
97ca987d46SWarner Losh if (ptr == NULL)
98ca987d46SWarner Losh errx(EX_SOFTWARE, "Could not allocate memory");
99ca987d46SWarner Losh return (ptr);
100ca987d46SWarner Losh }
101ca987d46SWarner Losh
102ca987d46SWarner Losh static void
usage(void)103ca987d46SWarner Losh usage(void)
104ca987d46SWarner Losh {
105ca987d46SWarner Losh errx(EX_USAGE, "sysinit -i sysinit.bin -o sysinit_data.c \\\n"
106ca987d46SWarner Losh "\t" "-k sysinit -s sysinit_data [ -R (reverse)]");
107ca987d46SWarner Losh }
108ca987d46SWarner Losh
109ca987d46SWarner Losh static void
cleanup(void)110ca987d46SWarner Losh cleanup(void)
111ca987d46SWarner Losh {
112ca987d46SWarner Losh if (output_file >= 0)
113ca987d46SWarner Losh close(output_file);
114ca987d46SWarner Losh if (input_file >= 0)
115ca987d46SWarner Losh close(input_file);
116ca987d46SWarner Losh if (success == 0) {
117ca987d46SWarner Losh if (output_f)
118ca987d46SWarner Losh unlink(output_f);
119ca987d46SWarner Losh }
120ca987d46SWarner Losh }
121ca987d46SWarner Losh
122ca987d46SWarner Losh static int
compare(const void * _pa,const void * _pb)123ca987d46SWarner Losh compare(const void *_pa, const void *_pb)
124ca987d46SWarner Losh {
125ca987d46SWarner Losh const struct sysinit_data * const *pa = _pa;
126ca987d46SWarner Losh const struct sysinit_data * const *pb = _pb;
127ca987d46SWarner Losh
128ca987d46SWarner Losh if ((*pa)->dw_msb_value > (*pb)->dw_msb_value)
129ca987d46SWarner Losh return (1);
130ca987d46SWarner Losh
131ca987d46SWarner Losh if ((*pa)->dw_msb_value < (*pb)->dw_msb_value)
132ca987d46SWarner Losh return (-1);
133ca987d46SWarner Losh
134ca987d46SWarner Losh if ((*pa)->dw_lsb_value > (*pb)->dw_lsb_value)
135ca987d46SWarner Losh return (1);
136ca987d46SWarner Losh
137ca987d46SWarner Losh if ((*pa)->dw_lsb_value < (*pb)->dw_lsb_value)
138ca987d46SWarner Losh return (-1);
139ca987d46SWarner Losh
140ca987d46SWarner Losh return (0); /* equal */
141ca987d46SWarner Losh }
142ca987d46SWarner Losh
143ca987d46SWarner Losh static int
compare_R(const void * _pa,const void * _pb)144ca987d46SWarner Losh compare_R(const void *_pa, const void *_pb)
145ca987d46SWarner Losh {
146ca987d46SWarner Losh const struct sysinit_data * const *pa = _pa;
147ca987d46SWarner Losh const struct sysinit_data * const *pb = _pb;
148ca987d46SWarner Losh
149ca987d46SWarner Losh if ((*pa)->dw_msb_value > (*pb)->dw_msb_value)
150ca987d46SWarner Losh return (-1);
151ca987d46SWarner Losh
152ca987d46SWarner Losh if ((*pa)->dw_msb_value < (*pb)->dw_msb_value)
153ca987d46SWarner Losh return (1);
154ca987d46SWarner Losh
155ca987d46SWarner Losh if ((*pa)->dw_lsb_value > (*pb)->dw_lsb_value)
156ca987d46SWarner Losh return (-1);
157ca987d46SWarner Losh
158ca987d46SWarner Losh if ((*pa)->dw_lsb_value < (*pb)->dw_lsb_value)
159ca987d46SWarner Losh return (1);
160ca987d46SWarner Losh
161ca987d46SWarner Losh return (0); /* equal */
162ca987d46SWarner Losh }
163ca987d46SWarner Losh
164ca987d46SWarner Losh int
main(int argc,char ** argv)165ca987d46SWarner Losh main(int argc, char **argv)
166ca987d46SWarner Losh {
167ca987d46SWarner Losh struct sysinit_data **sipp;
168ca987d46SWarner Losh int c;
169ca987d46SWarner Losh int entries;
170ca987d46SWarner Losh off_t off;
171ca987d46SWarner Losh
172ca987d46SWarner Losh while ((c = getopt(argc, argv, "k:s:i:o:Rh")) != -1) {
173ca987d46SWarner Losh switch (c) {
174ca987d46SWarner Losh case 'i':
175ca987d46SWarner Losh input_f = optarg;
176ca987d46SWarner Losh break;
177ca987d46SWarner Losh case 'o':
178ca987d46SWarner Losh output_f = optarg;
179ca987d46SWarner Losh break;
180ca987d46SWarner Losh case 'R':
181ca987d46SWarner Losh opt_R = 1;
182ca987d46SWarner Losh break;
183ca987d46SWarner Losh case 'k':
184ca987d46SWarner Losh keyword = optarg;
185ca987d46SWarner Losh break;
186ca987d46SWarner Losh case 's':
187ca987d46SWarner Losh struct_name = optarg;
188ca987d46SWarner Losh break;
189ca987d46SWarner Losh default:
190ca987d46SWarner Losh usage();
191ca987d46SWarner Losh }
192ca987d46SWarner Losh }
193ca987d46SWarner Losh
194ca987d46SWarner Losh if (input_f == NULL || output_f == NULL ||
195ca987d46SWarner Losh struct_name == NULL || keyword == NULL)
196ca987d46SWarner Losh usage();
197ca987d46SWarner Losh
198ca987d46SWarner Losh atexit(&cleanup);
199ca987d46SWarner Losh
200ca987d46SWarner Losh cleanup();
201ca987d46SWarner Losh
202ca987d46SWarner Losh input_file = open(input_f, O_RDONLY);
203ca987d46SWarner Losh if (input_file < 0)
204ca987d46SWarner Losh err(EX_SOFTWARE, "Could not open input file: %s", input_f);
205ca987d46SWarner Losh
206ca987d46SWarner Losh output_file = open(output_f, O_TRUNC | O_CREAT | O_RDWR, 0600);
207ca987d46SWarner Losh if (output_file < 0)
208ca987d46SWarner Losh err(EX_SOFTWARE, "Could not open output file: %s", output_f);
209ca987d46SWarner Losh
210ca987d46SWarner Losh off = lseek(input_file, 0, SEEK_END);
211ca987d46SWarner Losh
212ca987d46SWarner Losh input_ptr = do_malloc(off);
213ca987d46SWarner Losh input_len = off;
214ca987d46SWarner Losh
215ca987d46SWarner Losh if (input_len % (uint32_t)sizeof(struct sysinit_data)) {
216ca987d46SWarner Losh errx(EX_SOFTWARE, "Input file size is not divisible by %u",
217ca987d46SWarner Losh (unsigned int)sizeof(struct sysinit_data));
218ca987d46SWarner Losh }
219ca987d46SWarner Losh off = lseek(input_file, 0, SEEK_SET);
220ca987d46SWarner Losh if (off < 0)
221ca987d46SWarner Losh err(EX_SOFTWARE, "Could not seek to start of input file");
222ca987d46SWarner Losh
223ca987d46SWarner Losh if (read(input_file, input_ptr, input_len) != input_len)
224ca987d46SWarner Losh err(EX_SOFTWARE, "Could not read input file");
225ca987d46SWarner Losh
226ca987d46SWarner Losh entries = input_len / (uint32_t)sizeof(struct sysinit_data);
227ca987d46SWarner Losh
228ca987d46SWarner Losh start = do_malloc(sizeof(void *) * entries);
229ca987d46SWarner Losh stop = start + entries;
230ca987d46SWarner Losh
231ca987d46SWarner Losh for (c = 0; c != entries; c++)
232ca987d46SWarner Losh start[c] = &((struct sysinit_data *)input_ptr)[c];
233ca987d46SWarner Losh
234ca987d46SWarner Losh if (start != stop)
235ca987d46SWarner Losh endian32 = (*start)->dw_endian32;
236ca987d46SWarner Losh
237ca987d46SWarner Losh /* switch all fields to host endian order */
238ca987d46SWarner Losh for (sipp = start; sipp < stop; sipp++) {
239ca987d46SWarner Losh (*sipp)->dw_lsb_value = read32((*sipp)->dw_lsb_value);
240ca987d46SWarner Losh (*sipp)->dw_msb_value = read32((*sipp)->dw_msb_value);
241ca987d46SWarner Losh (*sipp)->dw_file_line = read32((*sipp)->dw_file_line);
242ca987d46SWarner Losh }
243ca987d46SWarner Losh
244ca987d46SWarner Losh if (opt_R == 0) {
245ca987d46SWarner Losh /* sort entries, rising numerical order */
246ca987d46SWarner Losh qsort(start, entries, sizeof(void *), &compare);
247ca987d46SWarner Losh } else {
248ca987d46SWarner Losh /* sort entries, falling numerical order */
249ca987d46SWarner Losh qsort(start, entries, sizeof(void *), &compare_R);
250ca987d46SWarner Losh }
251ca987d46SWarner Losh
252ca987d46SWarner Losh /* safe all strings */
253ca987d46SWarner Losh for (sipp = start; sipp < stop; sipp++) {
254ca987d46SWarner Losh (*sipp)->b_keyword_name[sizeof((*sipp)->b_keyword_name) - 1] = 0;
255ca987d46SWarner Losh (*sipp)->b_global_type[sizeof((*sipp)->b_global_type) - 1] = 0;
256ca987d46SWarner Losh (*sipp)->b_global_name[sizeof((*sipp)->b_global_name) - 1] = 0;
257ca987d46SWarner Losh (*sipp)->b_file_name[sizeof((*sipp)->b_file_name) - 1] = 0;
258ca987d46SWarner Losh (*sipp)->b_debug_info[sizeof((*sipp)->b_debug_info) - 1] = 0;
259ca987d46SWarner Losh }
260ca987d46SWarner Losh
261ca987d46SWarner Losh if (strcmp(keyword, "sysinit") == 0)
262ca987d46SWarner Losh do_sysinit();
263ca987d46SWarner Losh else if (strcmp(keyword, "sysuninit") == 0)
264ca987d46SWarner Losh do_sysinit();
265ca987d46SWarner Losh else
266ca987d46SWarner Losh errx(EX_USAGE, "Unknown keyword '%s'", keyword);
267ca987d46SWarner Losh
268ca987d46SWarner Losh success = 1;
269ca987d46SWarner Losh
270ca987d46SWarner Losh return (0);
271ca987d46SWarner Losh }
272ca987d46SWarner Losh
273ca987d46SWarner Losh static void
do_sysinit(void)274ca987d46SWarner Losh do_sysinit(void)
275ca987d46SWarner Losh {
276ca987d46SWarner Losh struct sysinit_data **sipp;
277ca987d46SWarner Losh int c;
278ca987d46SWarner Losh
279ca987d46SWarner Losh snprintf(scratch_buf, sizeof(scratch_buf),
280ca987d46SWarner Losh "/*\n"
281ca987d46SWarner Losh " * This file was automatically generated.\n"
282ca987d46SWarner Losh " * Please do not edit.\n"
283ca987d46SWarner Losh " */\n\n");
284ca987d46SWarner Losh
285ca987d46SWarner Losh /* write out externals */
286ca987d46SWarner Losh for (c = 0, sipp = start; sipp < stop; c++, sipp++) {
287ca987d46SWarner Losh if (strcmp((const char *)(*sipp)->b_keyword_name, keyword))
288ca987d46SWarner Losh continue;
289ca987d46SWarner Losh if ((*sipp)->dw_msb_value == 0)
290ca987d46SWarner Losh continue;
291ca987d46SWarner Losh
292ca987d46SWarner Losh snprintf(scratch_buf, sizeof(scratch_buf),
293ca987d46SWarner Losh "/* #%04u: %s entry at %s:%u */\n",
294ca987d46SWarner Losh c, (*sipp)->b_debug_info, (*sipp)->b_file_name,
295ca987d46SWarner Losh (unsigned int)(*sipp)->dw_file_line);
296ca987d46SWarner Losh
297ca987d46SWarner Losh do_write(output_file, scratch_buf);
298ca987d46SWarner Losh
299ca987d46SWarner Losh snprintf(scratch_buf, sizeof(scratch_buf),
300ca987d46SWarner Losh "extern %s %s;\n\n", (*sipp)->b_global_type,
301ca987d46SWarner Losh (*sipp)->b_global_name);
302ca987d46SWarner Losh
303ca987d46SWarner Losh do_write(output_file, scratch_buf);
304ca987d46SWarner Losh }
305ca987d46SWarner Losh
306ca987d46SWarner Losh snprintf(scratch_buf, sizeof(scratch_buf),
307ca987d46SWarner Losh "const void *%s[] = {\n", struct_name);
308ca987d46SWarner Losh
309ca987d46SWarner Losh do_write(output_file, scratch_buf);
310ca987d46SWarner Losh
311ca987d46SWarner Losh /* write out actual table */
312ca987d46SWarner Losh for (c = 0, sipp = start; sipp < stop; c++, sipp++) {
313ca987d46SWarner Losh if (strcmp((const char *)(*sipp)->b_keyword_name, keyword))
314ca987d46SWarner Losh continue;
315ca987d46SWarner Losh if ((*sipp)->dw_msb_value == 0)
316ca987d46SWarner Losh continue;
317ca987d46SWarner Losh
318ca987d46SWarner Losh snprintf(scratch_buf, sizeof(scratch_buf),
319ca987d46SWarner Losh "\t&%s, /* #%04u */\n",
320ca987d46SWarner Losh (*sipp)->b_global_name, (unsigned int)c);
321ca987d46SWarner Losh
322ca987d46SWarner Losh do_write(output_file, scratch_buf);
323ca987d46SWarner Losh }
324ca987d46SWarner Losh
325ca987d46SWarner Losh snprintf(scratch_buf, sizeof(scratch_buf),
326ca987d46SWarner Losh "\t(const void *)0\n"
327ca987d46SWarner Losh "};\n");
328ca987d46SWarner Losh
329ca987d46SWarner Losh do_write(output_file, scratch_buf);
330ca987d46SWarner Losh }
331