Contents of /tags/mkinitrd-6_1_11/busybox/applets/applet_tables.c
Parent Directory | Revision Log
Revision 928 -
(show annotations)
(download)
Wed Oct 28 13:31:19 2009 UTC (14 years, 11 months ago) by niro
File MIME type: text/plain
File size: 3077 byte(s)
Wed Oct 28 13:31:19 2009 UTC (14 years, 11 months ago) by niro
File MIME type: text/plain
File size: 3077 byte(s)
tagged 'mkinitrd-6_1_11'
1 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * Applet table generator. |
4 | * Runs on host and produces include/applet_tables.h |
5 | * |
6 | * Copyright (C) 2007 Denys Vlasenko <vda.linux@googlemail.com> |
7 | * |
8 | * Licensed under GPLv2, see file License in this tarball for details. |
9 | */ |
10 | |
11 | #include <stdlib.h> |
12 | #include <string.h> |
13 | #include <stdio.h> |
14 | |
15 | #include "../include/autoconf.h" |
16 | #include "../include/busybox.h" |
17 | |
18 | struct bb_applet { |
19 | const char *name; |
20 | const char *main; |
21 | enum bb_install_loc_t install_loc; |
22 | enum bb_suid_t need_suid; |
23 | /* true if instead of fork(); exec("applet"); waitpid(); |
24 | * one can do fork(); exit(applet_main(argc,argv)); waitpid(); */ |
25 | unsigned char noexec; |
26 | /* Even nicer */ |
27 | /* true if instead of fork(); exec("applet"); waitpid(); |
28 | * one can simply call applet_main(argc,argv); */ |
29 | unsigned char nofork; |
30 | }; |
31 | |
32 | /* Define struct bb_applet applets[] */ |
33 | #include "../include/applets.h" |
34 | |
35 | enum { NUM_APPLETS = ARRAY_SIZE(applets) }; |
36 | |
37 | static int offset[NUM_APPLETS]; |
38 | |
39 | static int cmp_name(const void *a, const void *b) |
40 | { |
41 | const struct bb_applet *aa = a; |
42 | const struct bb_applet *bb = b; |
43 | return strcmp(aa->name, bb->name); |
44 | } |
45 | |
46 | int main(int argc, char **argv) |
47 | { |
48 | int i; |
49 | int ofs; |
50 | unsigned MAX_APPLET_NAME_LEN = 1; |
51 | |
52 | qsort(applets, NUM_APPLETS, sizeof(applets[0]), cmp_name); |
53 | |
54 | ofs = 0; |
55 | for (i = 0; i < NUM_APPLETS; i++) { |
56 | offset[i] = ofs; |
57 | ofs += strlen(applets[i].name) + 1; |
58 | } |
59 | /* We reuse 4 high-order bits of offset array for other purposes, |
60 | * so if they are indeed needed, refuse to proceed */ |
61 | if (ofs > 0xfff) |
62 | return 1; |
63 | if (!argv[1]) |
64 | return 1; |
65 | |
66 | i = open(argv[1], O_WRONLY | O_TRUNC | O_CREAT, 0666); |
67 | if (i < 0) |
68 | return 1; |
69 | dup2(i, 1); |
70 | |
71 | /* Keep in sync with include/busybox.h! */ |
72 | |
73 | puts("/* This is a generated file, don't edit */\n"); |
74 | |
75 | printf("#define NUM_APPLETS %u\n", NUM_APPLETS); |
76 | if (NUM_APPLETS == 1) { |
77 | printf("#define SINGLE_APPLET_STR \"%s\"\n", applets[0].name); |
78 | printf("#define SINGLE_APPLET_MAIN %s_main\n", applets[0].name); |
79 | } |
80 | |
81 | puts("\nconst char applet_names[] ALIGN1 = \"\""); |
82 | for (i = 0; i < NUM_APPLETS; i++) { |
83 | printf("\"%s\" \"\\0\"\n", applets[i].name); |
84 | if (MAX_APPLET_NAME_LEN < strlen(applets[i].name)) |
85 | MAX_APPLET_NAME_LEN = strlen(applets[i].name); |
86 | } |
87 | puts(";"); |
88 | |
89 | puts("\nint (*const applet_main[])(int argc, char **argv) = {"); |
90 | for (i = 0; i < NUM_APPLETS; i++) { |
91 | printf("%s_main,\n", applets[i].main); |
92 | } |
93 | puts("};"); |
94 | |
95 | puts("const uint16_t applet_nameofs[] ALIGN2 = {"); |
96 | for (i = 0; i < NUM_APPLETS; i++) { |
97 | printf("0x%04x,\n", |
98 | offset[i] |
99 | #if ENABLE_FEATURE_PREFER_APPLETS |
100 | + (applets[i].nofork << 12) |
101 | + (applets[i].noexec << 13) |
102 | #endif |
103 | #if ENABLE_FEATURE_SUID |
104 | + (applets[i].need_suid << 14) /* 2 bits */ |
105 | #endif |
106 | ); |
107 | } |
108 | puts("};"); |
109 | |
110 | #if ENABLE_FEATURE_INSTALLER |
111 | puts("const uint8_t applet_install_loc[] ALIGN1 = {"); |
112 | i = 0; |
113 | while (i < NUM_APPLETS) { |
114 | int v = applets[i].install_loc; /* 3 bits */ |
115 | if (++i < NUM_APPLETS) |
116 | v |= applets[i].install_loc << 4; /* 3 bits */ |
117 | printf("0x%02x,\n", v); |
118 | i++; |
119 | } |
120 | puts("};\n"); |
121 | #endif |
122 | |
123 | printf("#define MAX_APPLET_NAME_LEN %u\n", MAX_APPLET_NAME_LEN); |
124 | |
125 | return 0; |
126 | } |