applets/applet_tables.c

   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
  18 #undef ARRAY_SIZE
  19 #define ARRAY_SIZE(x) ((unsigned)(sizeof(x) / sizeof((x)[0])))
  20
  21 #include "../include/autoconf.h"
  22 #include "../include/applet_metadata.h"
  23
  24 struct bb_applet {
  25         const char *name;
  26         const char *main;
  27         enum bb_install_loc_t install_loc;
  28         enum bb_suid_t need_suid;
  29         /* true if instead of fork(); exec("applet"); waitpid();
  30          * one can do fork(); exit(applet_main(argc,argv)); waitpid(); */
  31         unsigned char noexec;
  32         /* Even nicer */
  33         /* true if instead of fork(); exec("applet"); waitpid();
  34          * one can simply call applet_main(argc,argv); */
  35         unsigned char nofork;
  36 };
  37
  38 /* Define struct bb_applet applets[] */
  39 #include "../include/applets.h"
  40
  41 enum { NUM_APPLETS = ARRAY_SIZE(applets) };
  42
  43 static int offset[NUM_APPLETS];
  44
  45 static int cmp_name(const void *a, const void *b)
  46 {
  47         const struct bb_applet *aa = a;
  48         const struct bb_applet *bb = b;
  49         return strcmp(aa->name, bb->name);
  50 }
  51
  52 int main(int argc, char **argv)
  53 {
  54         int i;
  55         int ofs;
  56         unsigned MAX_APPLET_NAME_LEN = 1;
  57
  58         qsort(applets, NUM_APPLETS, sizeof(applets[0]), cmp_name);
  59
  60         ofs = 0;
  61         for (i = 0; i < NUM_APPLETS; i++) {
  62                 offset[i] = ofs;
  63                 ofs += strlen(applets[i].name) + 1;
  64         }
  65         /* We reuse 4 high-order bits of offset array for other purposes,
  66          * so if they are indeed needed, refuse to proceed */
  67         if (ofs > 0xfff)
  68                 return 1;
  69         if (!argv[1])
  70                 return 1;
  71
  72         i = open(argv[1], O_WRONLY | O_TRUNC | O_CREAT, 0666);
  73         if (i < 0)
  74                 return 1;
  75         dup2(i, 1);
  76
  77         /* Keep in sync with include/busybox.h! */
  78
  79         printf("/* This is a generated file, don't edit */\n\n");
  80
  81         printf("#define NUM_APPLETS %u\n", NUM_APPLETS);
  82         if (NUM_APPLETS == 1) {
  83                 char *dash_to_underscore, *p;
  84                 printf("#define SINGLE_APPLET_STR \"%s\"\n", applets[0].name);
  85                 /* Example: "ether-wake" -> "ether_wake" */
  86                 p = dash_to_underscore = strdup(applets[0].name);
  87                 p--;
  88                 while (*++p)
  89                         if (*p == '-')
  90                                 *p = '_';
  91                 printf("#define SINGLE_APPLET_MAIN %s_main\n", dash_to_underscore);
  92         }
  93         printf("\n");
  94
  95         //printf("#ifndef SKIP_definitions\n");
  96         printf("const char applet_names[] ALIGN1 = \"\"\n");
  97         for (i = 0; i < NUM_APPLETS; i++) {
  98                 printf("\"%s\" \"\\0\"\n", applets[i].name);
  99                 if (MAX_APPLET_NAME_LEN < strlen(applets[i].name))
 100                         MAX_APPLET_NAME_LEN = strlen(applets[i].name);
 101         }
 102         printf(";\n\n");
 103
 104         printf("#ifndef SKIP_applet_main\n");
 105         printf("int (*const applet_main[])(int argc, char **argv) = {\n");
 106         for (i = 0; i < NUM_APPLETS; i++) {
 107                 printf("%s_main,\n", applets[i].main);
 108         }
 109         printf("};\n");
 110         printf("#endif\n\n");
 111
 112         printf("const uint16_t applet_nameofs[] ALIGN2 = {\n");
 113         for (i = 0; i < NUM_APPLETS; i++) {
 114                 printf("0x%04x,\n",
 115                         offset[i]
 116 #if ENABLE_FEATURE_PREFER_APPLETS
 117                         + (applets[i].nofork << 12)
 118                         + (applets[i].noexec << 13)
 119 #endif
 120 #if ENABLE_FEATURE_SUID
 121                         + (applets[i].need_suid << 14) /* 2 bits */
 122 #endif
 123                 );
 124         }
 125         printf("};\n\n");
 126
 127 #if ENABLE_FEATURE_INSTALLER
 128         printf("const uint8_t applet_install_loc[] ALIGN1 = {\n");
 129         i = 0;
 130         while (i < NUM_APPLETS) {
 131                 int v = applets[i].install_loc; /* 3 bits */
 132                 if (++i < NUM_APPLETS)
 133                         v |= applets[i].install_loc << 4; /* 3 bits */
 134                 printf("0x%02x,\n", v);
 135                 i++;
 136         }
 137         printf("};\n");
 138 #endif
 139         //printf("#endif /* SKIP_definitions */\n");
 140         printf("\n");
 141         printf("#define MAX_APPLET_NAME_LEN %u\n", MAX_APPLET_NAME_LEN);
 142
 143         if (argv[2]) {
 144                 char line_old[80];
 145                 char line_new[80];
 146                 FILE *fp;
 147
 148                 line_old[0] = 0;
 149                 fp = fopen(argv[2], "r");
 150                 if (fp) {
 151                         fgets(line_old, sizeof(line_old), fp);
 152                         fclose(fp);
 153                 }
 154                 sprintf(line_new, "#define NUM_APPLETS %u\n", NUM_APPLETS);
 155                 if (strcmp(line_old, line_new) != 0) {
 156                         fp = fopen(argv[2], "w");
 157                         if (!fp)
 158                                 return 1;
 159                         fputs(line_new, fp);
 160                 }
 161         }
 162
 163         return 0;
 164 }