| 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 source tree.
|
|---|
| 9 | */
|
|---|
| 10 | #include <sys/types.h>
|
|---|
| 11 | #include <sys/stat.h>
|
|---|
| 12 | #include <fcntl.h>
|
|---|
| 13 | #include <stdlib.h>
|
|---|
| 14 | #include <string.h>
|
|---|
| 15 | #include <stdio.h>
|
|---|
| 16 | #include <unistd.h>
|
|---|
| 17 | #include <ctype.h>
|
|---|
| 18 |
|
|---|
| 19 | #undef ARRAY_SIZE
|
|---|
| 20 | #define ARRAY_SIZE(x) ((unsigned)(sizeof(x) / sizeof((x)[0])))
|
|---|
| 21 |
|
|---|
| 22 | #include "../include/autoconf.h"
|
|---|
| 23 | #include "../include/applet_metadata.h"
|
|---|
| 24 |
|
|---|
| 25 | struct bb_applet {
|
|---|
| 26 | const char *name;
|
|---|
| 27 | const char *main;
|
|---|
| 28 | enum bb_install_loc_t install_loc;
|
|---|
| 29 | enum bb_suid_t need_suid;
|
|---|
| 30 | /* true if instead of fork(); exec("applet"); waitpid();
|
|---|
| 31 | * one can do fork(); exit(applet_main(argc,argv)); waitpid(); */
|
|---|
| 32 | unsigned char noexec;
|
|---|
| 33 | /* Even nicer */
|
|---|
| 34 | /* true if instead of fork(); exec("applet"); waitpid();
|
|---|
| 35 | * one can simply call applet_main(argc,argv); */
|
|---|
| 36 | unsigned char nofork;
|
|---|
| 37 | };
|
|---|
| 38 |
|
|---|
| 39 | /* Define struct bb_applet applets[] */
|
|---|
| 40 | #include "../include/applets.h"
|
|---|
| 41 |
|
|---|
| 42 | enum { NUM_APPLETS = ARRAY_SIZE(applets) };
|
|---|
| 43 |
|
|---|
| 44 | static int cmp_name(const void *a, const void *b)
|
|---|
| 45 | {
|
|---|
| 46 | const struct bb_applet *aa = a;
|
|---|
| 47 | const struct bb_applet *bb = b;
|
|---|
| 48 | return strcmp(aa->name, bb->name);
|
|---|
| 49 | }
|
|---|
| 50 |
|
|---|
| 51 | static int str_isalnum_(const char *s)
|
|---|
| 52 | {
|
|---|
| 53 | while (*s) {
|
|---|
| 54 | if (!isalnum(*s) && *s != '_')
|
|---|
| 55 | return 0;
|
|---|
| 56 | s++;
|
|---|
| 57 | }
|
|---|
| 58 | return 1;
|
|---|
| 59 | }
|
|---|
| 60 |
|
|---|
| 61 | int main(int argc, char **argv)
|
|---|
| 62 | {
|
|---|
| 63 | int i, j;
|
|---|
| 64 |
|
|---|
| 65 | // In find_applet_by_name(), before linear search, narrow it down
|
|---|
| 66 | // by looking at N "equidistant" names. With ~350 applets:
|
|---|
| 67 | // KNOWN_APPNAME_OFFSETS cycles
|
|---|
| 68 | // 0 9057
|
|---|
| 69 | // 2 4604 + ~100 bytes of code
|
|---|
| 70 | // 4 2407 + 4 bytes
|
|---|
| 71 | // 8 1342 + 8 bytes
|
|---|
| 72 | // 16 908 + 16 bytes
|
|---|
| 73 | // 32 884 + 32 bytes
|
|---|
| 74 | // With 8, int16_t applet_nameofs[] table has 7 elements.
|
|---|
| 75 | int KNOWN_APPNAME_OFFSETS = 8;
|
|---|
| 76 | // With 128 applets we do two linear searches, with 1..7 strcmp's in the first one
|
|---|
| 77 | // and 1..16 strcmp's in the second. With 256 apps, second search does 1..32 strcmp's.
|
|---|
| 78 | if (NUM_APPLETS < 128)
|
|---|
| 79 | KNOWN_APPNAME_OFFSETS = 4;
|
|---|
| 80 | if (NUM_APPLETS < 32)
|
|---|
| 81 | KNOWN_APPNAME_OFFSETS = 0;
|
|---|
| 82 |
|
|---|
| 83 | qsort(applets, NUM_APPLETS, sizeof(applets[0]), cmp_name);
|
|---|
| 84 |
|
|---|
| 85 | if (!argv[1])
|
|---|
| 86 | return 1;
|
|---|
| 87 | i = open(argv[1], O_WRONLY | O_TRUNC | O_CREAT, 0666);
|
|---|
| 88 | if (i < 0)
|
|---|
| 89 | return 1;
|
|---|
| 90 | dup2(i, 1);
|
|---|
| 91 |
|
|---|
| 92 | /* Keep in sync with include/busybox.h! */
|
|---|
| 93 |
|
|---|
| 94 | printf("/* This is a generated file, don't edit */\n\n");
|
|---|
| 95 |
|
|---|
| 96 | printf("#define NUM_APPLETS %u\n", NUM_APPLETS);
|
|---|
| 97 | if (NUM_APPLETS == 1) {
|
|---|
| 98 | printf("#define SINGLE_APPLET_STR \"%s\"\n", applets[0].name);
|
|---|
| 99 | printf("#define SINGLE_APPLET_MAIN %s_main\n", applets[0].main);
|
|---|
| 100 | }
|
|---|
| 101 |
|
|---|
| 102 | printf("#define KNOWN_APPNAME_OFFSETS %u\n\n", KNOWN_APPNAME_OFFSETS);
|
|---|
| 103 | if (KNOWN_APPNAME_OFFSETS > 0) {
|
|---|
| 104 | int ofs, offset[KNOWN_APPNAME_OFFSETS], index[KNOWN_APPNAME_OFFSETS];
|
|---|
| 105 | for (i = 0; i < KNOWN_APPNAME_OFFSETS; i++)
|
|---|
| 106 | index[i] = i * NUM_APPLETS / KNOWN_APPNAME_OFFSETS;
|
|---|
| 107 | ofs = 0;
|
|---|
| 108 | for (i = 0; i < NUM_APPLETS; i++) {
|
|---|
| 109 | for (j = 0; j < KNOWN_APPNAME_OFFSETS; j++)
|
|---|
| 110 | if (i == index[j])
|
|---|
| 111 | offset[j] = ofs;
|
|---|
| 112 | ofs += strlen(applets[i].name) + 1;
|
|---|
| 113 | }
|
|---|
| 114 | /* If the list of names is too long refuse to proceed */
|
|---|
| 115 | if (ofs > 0xffff)
|
|---|
| 116 | return 1;
|
|---|
| 117 | printf("const uint16_t applet_nameofs[] ALIGN2 = {\n");
|
|---|
| 118 | for (i = 1; i < KNOWN_APPNAME_OFFSETS; i++)
|
|---|
| 119 | printf("%d,\n", offset[i]);
|
|---|
| 120 | printf("};\n\n");
|
|---|
| 121 | }
|
|---|
| 122 |
|
|---|
| 123 | //printf("#ifndef SKIP_definitions\n");
|
|---|
| 124 | printf("const char applet_names[] ALIGN1 = \"\"\n");
|
|---|
| 125 | for (i = 0; i < NUM_APPLETS; i++) {
|
|---|
| 126 | printf("\"%s\" \"\\0\"\n", applets[i].name);
|
|---|
| 127 | // if (MAX_APPLET_NAME_LEN < strlen(applets[i].name))
|
|---|
| 128 | // MAX_APPLET_NAME_LEN = strlen(applets[i].name);
|
|---|
| 129 | }
|
|---|
| 130 | printf(";\n\n");
|
|---|
| 131 |
|
|---|
| 132 | for (i = 0; i < NUM_APPLETS; i++) {
|
|---|
| 133 | if (str_isalnum_(applets[i].name))
|
|---|
| 134 | printf("#define APPLET_NO_%s %d\n", applets[i].name, i);
|
|---|
| 135 | }
|
|---|
| 136 | printf("\n");
|
|---|
| 137 |
|
|---|
| 138 | printf("#ifndef SKIP_applet_main\n");
|
|---|
| 139 | printf("int (*const applet_main[])(int argc, char **argv) = {\n");
|
|---|
| 140 | for (i = 0; i < NUM_APPLETS; i++) {
|
|---|
| 141 | printf("%s_main,\n", applets[i].main);
|
|---|
| 142 | }
|
|---|
| 143 | printf("};\n");
|
|---|
| 144 | printf("#endif\n\n");
|
|---|
| 145 |
|
|---|
| 146 | #if ENABLE_FEATURE_PREFER_APPLETS
|
|---|
| 147 | printf("const uint8_t applet_flags[] ALIGN1 = {\n");
|
|---|
| 148 | i = 0;
|
|---|
| 149 | while (i < NUM_APPLETS) {
|
|---|
| 150 | int v = applets[i].nofork + (applets[i].noexec << 1);
|
|---|
| 151 | if (++i < NUM_APPLETS)
|
|---|
| 152 | v |= (applets[i].nofork + (applets[i].noexec << 1)) << 2;
|
|---|
| 153 | if (++i < NUM_APPLETS)
|
|---|
| 154 | v |= (applets[i].nofork + (applets[i].noexec << 1)) << 4;
|
|---|
| 155 | if (++i < NUM_APPLETS)
|
|---|
| 156 | v |= (applets[i].nofork + (applets[i].noexec << 1)) << 6;
|
|---|
| 157 | printf("0x%02x,\n", v);
|
|---|
| 158 | i++;
|
|---|
| 159 | }
|
|---|
| 160 | printf("};\n\n");
|
|---|
| 161 | #endif
|
|---|
| 162 |
|
|---|
| 163 | #if ENABLE_FEATURE_SUID
|
|---|
| 164 | printf("const uint8_t applet_suid[] ALIGN1 = {\n");
|
|---|
| 165 | i = 0;
|
|---|
| 166 | while (i < NUM_APPLETS) {
|
|---|
| 167 | int v = applets[i].need_suid; /* 2 bits */
|
|---|
| 168 | if (++i < NUM_APPLETS)
|
|---|
| 169 | v |= applets[i].need_suid << 2;
|
|---|
| 170 | if (++i < NUM_APPLETS)
|
|---|
| 171 | v |= applets[i].need_suid << 4;
|
|---|
| 172 | if (++i < NUM_APPLETS)
|
|---|
| 173 | v |= applets[i].need_suid << 6;
|
|---|
| 174 | printf("0x%02x,\n", v);
|
|---|
| 175 | i++;
|
|---|
| 176 | }
|
|---|
| 177 | printf("};\n\n");
|
|---|
| 178 | #endif
|
|---|
| 179 |
|
|---|
| 180 | #if ENABLE_FEATURE_INSTALLER
|
|---|
| 181 | printf("const uint8_t applet_install_loc[] ALIGN1 = {\n");
|
|---|
| 182 | i = 0;
|
|---|
| 183 | while (i < NUM_APPLETS) {
|
|---|
| 184 | int v = applets[i].install_loc; /* 3 bits */
|
|---|
| 185 | if (++i < NUM_APPLETS)
|
|---|
| 186 | v |= applets[i].install_loc << 4; /* 3 bits */
|
|---|
| 187 | printf("0x%02x,\n", v);
|
|---|
| 188 | i++;
|
|---|
| 189 | }
|
|---|
| 190 | printf("};\n");
|
|---|
| 191 | #endif
|
|---|
| 192 | //printf("#endif /* SKIP_definitions */\n");
|
|---|
| 193 | // printf("\n");
|
|---|
| 194 | // printf("#define MAX_APPLET_NAME_LEN %u\n", MAX_APPLET_NAME_LEN);
|
|---|
| 195 |
|
|---|
| 196 | if (argv[2]) {
|
|---|
| 197 | char line_old[80];
|
|---|
| 198 | char line_new[80];
|
|---|
| 199 | FILE *fp;
|
|---|
| 200 |
|
|---|
| 201 | line_old[0] = 0;
|
|---|
| 202 | fp = fopen(argv[2], "r");
|
|---|
| 203 | if (fp) {
|
|---|
| 204 | fgets(line_old, sizeof(line_old), fp);
|
|---|
| 205 | fclose(fp);
|
|---|
| 206 | }
|
|---|
| 207 | sprintf(line_new, "#define NUM_APPLETS %u\n", NUM_APPLETS);
|
|---|
| 208 | if (strcmp(line_old, line_new) != 0) {
|
|---|
| 209 | fp = fopen(argv[2], "w");
|
|---|
| 210 | if (!fp)
|
|---|
| 211 | return 1;
|
|---|
| 212 | fputs(line_new, fp);
|
|---|
| 213 | }
|
|---|
| 214 | }
|
|---|
| 215 |
|
|---|
| 216 | return 0;
|
|---|
| 217 | }
|
|---|
