main.c

   1 /* vi: set ts=4 :*/
   2 /* Toybox infrastructure.
   3  *
   4  * Copyright 2006 Rob Landley <rob@landley.net>
   5  *
   6  * Licensed under GPL version 2, see file LICENSE in this tarball for details.
   7  */
   8
   9 #include "toys.h"
  10
  11 // The monster fun applet list.
  12
  13 struct toy_list toy_list[] = {
  14         // This one is out of order on purpose.
  15         {"toybox", toybox_main, 0},
  16         // The rest of these are alphabetical, for binary search.
  17         {"cd", cd_main, TOYFLAG_NOFORK},
  18         {"df", df_main, TOYFLAG_USR|TOYFLAG_SBIN},
  19         {"exit", exit_main, TOYFLAG_NOFORK},
  20         {"toysh", toysh_main, TOYFLAG_BIN}
  21 };
  22
  23 #define TOY_LIST_LEN (sizeof(toy_list)/sizeof(struct toy_list))
  24
  25 // global context for this applet.
  26
  27 struct toy_context toys;
  28
  29 struct toy_list *toy_find(char *name)
  30 {
  31         int top, bottom, middle;
  32
  33         // If the name starts with "toybox", accept that as a match.  Otherwise
  34         // skip the first entry, which is out of order.
  35
  36         if (!strncmp(name,"toybox",6)) return toy_list;
  37         bottom = 1;
  38
  39         // Binary search to find this applet.
  40
  41         top = TOY_LIST_LEN-1;
  42         for (;;) {
  43                 int result;
  44
  45                 middle = (top+bottom)/2;
  46                 if (middle<bottom || middle>top) return NULL;
  47                 result = strcmp(name,toy_list[middle].name);
  48                 if (!result) return toy_list+middle;
  49                 if (result<0) top=--middle;
  50                 else bottom = ++middle;
  51         }
  52 }
  53
  54 // Run a toy.
  55 void toy_exec(char *argv[])
  56 {
  57         struct toy_list *which;
  58
  59         which = toy_find(argv[0]);
  60         if (!which) return;
  61
  62         // Free old toys contents here?
  63
  64         toys.which = which;
  65         toys.argv = argv;
  66         for (toys.argc = 0; argv[toys.argc]; toys.argc++);
  67         toys.exitval = 1;
  68
  69         exit(toys.which->toy_main());
  70 }
  71
  72 int toybox_main(void)
  73 {
  74         static char *toy_paths[]={"usr/","bin/","sbin/",0};
  75         int i, len = 0;
  76
  77         if (toys.argv[1]) {
  78                 if (toys.argv[1][0]!='-') {
  79                         toy_exec(toys.argv+1);
  80                         error_exit("No behavior for %s\n",toys.argv[1]);
  81                 }
  82         }
  83
  84         // Output list of applets.
  85         for (i=1; i<TOY_LIST_LEN; i++) {
  86                 int fl = toy_list[i].flags;
  87                 if (fl & TOYMASK_LOCATION) {
  88                         if (toys.argv[1]) {
  89                                 int j;
  90                                 for (j=0; toy_paths[j]; j++)
  91                                         if (fl & (1<<j)) len += printf("%s", toy_paths[j]);
  92                         }
  93                         len += printf("%s ",toy_list[i].name);
  94                         if (len>65) {
  95                                 putchar('\n');
  96                                 len=0;
  97                         }
  98                 }
  99         }
 100         putchar('\n');
 101         return 0;
 102 }
 103
 104 int main(int argc, char *argv[])
 105 {
 106         char *name;
 107
 108         // Figure out which applet to call.
 109         name = rindex(argv[0], '/');
 110         if (!name) name=argv[0];
 111         else name++;
 112         argv[0] = name;
 113
 114         toys.argv = argv-1;
 115         return toybox_main();
 116 }