1 /* vi: set sw=4 ts=4: */
3 * Applet table generator.
4 * Runs on host and produces include/applet_tables.h
6 * Copyright (C) 2007 Denys Vlasenko <vda.linux@googlemail.com>
8 * Licensed under GPLv2, see file LICENSE in this source tree.
10 #include <sys/types.h>
21 #define ARRAY_SIZE(x) ((unsigned)(sizeof(x) / sizeof((x)[0])))
23 #include "../include/autoconf.h"
24 #include "../include/applet_metadata.h"
29 enum bb_install_loc_t install_loc;
30 enum bb_suid_t need_suid;
31 /* true if instead of fork(); exec("applet"); waitpid();
32 * one can do fork(); exit(applet_main(argc,argv)); waitpid(); */
35 /* true if instead of fork(); exec("applet"); waitpid();
36 * one can simply call applet_main(argc,argv); */
40 /* Define struct bb_applet applets[] */
41 #include "../include/applets.h"
43 enum { NUM_APPLETS = ARRAY_SIZE(applets) };
45 static int cmp_name(const void *a, const void *b)
47 const struct bb_applet *aa = a;
48 const struct bb_applet *bb = b;
49 return strcmp(aa->name, bb->name);
52 static int str_isalnum_(const char *s)
55 if (!isalnum(*s) && *s != '_')
62 int main(int argc, char **argv)
65 char tmp1[PATH_MAX], tmp2[PATH_MAX];
67 // In find_applet_by_name(), before linear search, narrow it down
68 // by looking at N "equidistant" names. With ~350 applets:
69 // KNOWN_APPNAME_OFFSETS cycles
71 // 2 4604 + ~100 bytes of code
76 // With 8, int16_t applet_nameofs[] table has 7 elements.
77 int KNOWN_APPNAME_OFFSETS = 8;
78 // With 128 applets we do two linear searches, with 1..7 strcmp's in the first one
79 // and 1..16 strcmp's in the second. With 256 apps, second search does 1..32 strcmp's.
80 if (NUM_APPLETS < 128)
81 KNOWN_APPNAME_OFFSETS = 4;
83 KNOWN_APPNAME_OFFSETS = 0;
85 qsort(applets, NUM_APPLETS, sizeof(applets[0]), cmp_name);
89 snprintf(tmp1, PATH_MAX, "%s.%u.new", argv[1], (int) getpid());
90 i = open(tmp1, O_WRONLY | O_TRUNC | O_CREAT, 0666);
95 /* Keep in sync with include/busybox.h! */
97 printf("/* This is a generated file, don't edit */\n\n");
99 printf("#define NUM_APPLETS %u\n", NUM_APPLETS);
100 if (NUM_APPLETS == 1) {
101 printf("#define SINGLE_APPLET_STR \"%s\"\n", applets[0].name);
102 printf("#define SINGLE_APPLET_MAIN %s_main\n", applets[0].main);
105 printf("#define KNOWN_APPNAME_OFFSETS %u\n\n", KNOWN_APPNAME_OFFSETS);
106 if (KNOWN_APPNAME_OFFSETS > 0) {
107 int ofs, offset[KNOWN_APPNAME_OFFSETS], index[KNOWN_APPNAME_OFFSETS];
108 for (i = 0; i < KNOWN_APPNAME_OFFSETS; i++)
109 index[i] = i * NUM_APPLETS / KNOWN_APPNAME_OFFSETS;
111 for (i = 0; i < NUM_APPLETS; i++) {
112 for (j = 0; j < KNOWN_APPNAME_OFFSETS; j++)
115 ofs += strlen(applets[i].name) + 1;
117 /* If the list of names is too long refuse to proceed */
120 printf("const uint16_t applet_nameofs[] ALIGN2 = {\n");
121 for (i = 1; i < KNOWN_APPNAME_OFFSETS; i++)
122 printf("%d,\n", offset[i]);
126 //printf("#ifndef SKIP_definitions\n");
127 printf("const char applet_names[] ALIGN1 = \"\"\n");
128 for (i = 0; i < NUM_APPLETS; i++) {
129 printf("\"%s\" \"\\0\"\n", applets[i].name);
130 // if (MAX_APPLET_NAME_LEN < strlen(applets[i].name))
131 // MAX_APPLET_NAME_LEN = strlen(applets[i].name);
135 for (i = 0; i < NUM_APPLETS; i++) {
136 if (str_isalnum_(applets[i].name))
137 printf("#define APPLET_NO_%s %d\n", applets[i].name, i);
141 printf("#ifndef SKIP_applet_main\n");
142 printf("int (*const applet_main[])(int argc, char **argv) = {\n");
143 for (i = 0; i < NUM_APPLETS; i++) {
144 printf("%s_main,\n", applets[i].main);
147 printf("#endif\n\n");
149 #if ENABLE_FEATURE_PREFER_APPLETS \
150 || ENABLE_FEATURE_SH_STANDALONE \
151 || ENABLE_FEATURE_SH_NOFORK
152 printf("const uint8_t applet_flags[] ALIGN1 = {\n");
154 while (i < NUM_APPLETS) {
155 int v = applets[i].nofork + (applets[i].noexec << 1);
156 if (++i < NUM_APPLETS)
157 v |= (applets[i].nofork + (applets[i].noexec << 1)) << 2;
158 if (++i < NUM_APPLETS)
159 v |= (applets[i].nofork + (applets[i].noexec << 1)) << 4;
160 if (++i < NUM_APPLETS)
161 v |= (applets[i].nofork + (applets[i].noexec << 1)) << 6;
162 printf("0x%02x,\n", v);
168 #if ENABLE_FEATURE_SUID
169 printf("const uint8_t applet_suid[] ALIGN1 = {\n");
171 while (i < NUM_APPLETS) {
172 int v = applets[i].need_suid; /* 2 bits */
173 if (++i < NUM_APPLETS)
174 v |= applets[i].need_suid << 2;
175 if (++i < NUM_APPLETS)
176 v |= applets[i].need_suid << 4;
177 if (++i < NUM_APPLETS)
178 v |= applets[i].need_suid << 6;
179 printf("0x%02x,\n", v);
185 #if ENABLE_FEATURE_INSTALLER
186 printf("const uint8_t applet_install_loc[] ALIGN1 = {\n");
188 while (i < NUM_APPLETS) {
189 int v = applets[i].install_loc; /* 3 bits */
190 if (++i < NUM_APPLETS)
191 v |= applets[i].install_loc << 4; /* 3 bits */
192 printf("0x%02x,\n", v);
197 //printf("#endif /* SKIP_definitions */\n");
200 // printf("#define MAX_APPLET_NAME_LEN %u\n", MAX_APPLET_NAME_LEN);
205 // char line_old[80];
207 sprintf(line_new, "#define NUM_APPLETS %u\n", NUM_APPLETS);
209 // fp = fopen(argv[2], "r");
211 // fgets(line_old, sizeof(line_old), fp);
214 // if (strcmp(line_old, line_new) != 0) {
215 snprintf(tmp2, PATH_MAX, "%s.%u.new", argv[2], (int) getpid());
216 fp = fopen(tmp2, "w");
227 if (rename(tmp1, argv[1]))
229 if (rename(tmp2, argv[2]))