re-order a little for alphabetical
[oweals/busybox.git] / applets / busybox.c
index 403b14008d4e136e0092e59705b6f51e16764bcf..833e8d37292ccb8fa603aa43d311182692a256b8 100644 (file)
-#include "internal.h"
+/* vi: set sw=4 ts=4: */
 #include <stdio.h>
 #include <string.h>
+#include <unistd.h>
 #include <errno.h>
+#include <stdlib.h>
+#include "busybox.h"
+#if ENABLE_LOCALE_SUPPORT
+#include <locale.h>
+#else
+#define setlocale(x,y)
+#endif
 
-static int been_there_done_that = 0;
+const char *bb_applet_name;
 
-static const struct Applet applets[] = {
+#ifdef CONFIG_FEATURE_INSTALLER
+/*
+ * directory table
+ *             this should be consistent w/ the enum, busybox.h::Location,
+ *             or else...
+ */
+static const char usr_bin [] ="/usr/bin";
+static const char usr_sbin[] ="/usr/sbin";
 
-#ifdef BB_BUSYBOX              //bin
-    {"busybox", busybox_main},
-#endif
-#ifdef BB_BLOCK_DEVICE         //sbin
-    {"block_device", block_device_main},
-#endif
-#ifdef BB_CAT                  //bin
-    {"cat", cat_main},
-#endif
-#ifdef BB_CHMOD_CHOWN_CHGRP    //bin
-    {"chmod", chmod_chown_chgrp_main},
-    {"chown", chmod_chown_chgrp_main},
-    {"chgrp", chmod_chown_chgrp_main},
-#endif
-#ifdef BB_CHROOT               //sbin
-    {"chroot", chroot_main},
-#endif
-#ifdef BB_CLEAR                        //usr/bin
-    {"clear", clear_main},
-#endif
-#ifdef BB_CHVT                 //usr/bin
-    {"chvt", chvt_main},
-#endif
-#ifdef BB_CP                   //bin
-    {"cp", cp_main},
-#endif
-#ifdef BB_DATE                 //bin
-    {"date", date_main},
-#endif
-#ifdef BB_DD                   //bin
-    {"dd", dd_main},
-#endif
-#ifdef BB_DF                   //bin
-    {"df", df_main},
-#endif
-#ifdef BB_DMESG                        //bin
-    {"dmesg", dmesg_main},
-#endif
-#ifdef BB_DU                   //bin
-    {"du", du_main},
-#endif
-#ifdef BB_DUTMP                        //usr/sbin
-    {"dutmp", dutmp_main},
-#endif
-#ifdef BB_FBSET                 //usr/sbin
-    {"fbset", fbset_main},
-#endif
-#ifdef BB_FDFLUSH              //bin
-    {"fdflush", fdflush_main},
-#endif
-#ifdef BB_FIND                 //usr/bin
-    {"find", find_main},
-#endif
-#ifdef BB_FREE                 //usr/bin
-    {"free", free_main},
-#endif
-#ifdef BB_DEALLOCVT                    //usr/bin
-    {"deallocvt", deallocvt_main},
-#endif
-#ifdef BB_FSCK_MINIX           //sbin
-    {"fsck.minix", fsck_minix_main},
-#endif
-#ifdef BB_MKFS_MINIX           //sbin
-    {"mkfs.minix", mkfs_minix_main},
-#endif
-#ifdef BB_GREP                 //bin
-    {"grep", grep_main},
-#endif
-#ifdef BB_HALT                 //sbin
-    {"halt", halt_main},
-#endif
-#ifdef BB_HEAD                 //bin
-    {"head", head_main},
-#endif
-#ifdef BB_HOSTNAME              //bin
-    {"hostname", hostname_main},
-#endif
-#ifdef BB_INIT                 //sbin
-    {"init", init_main},
-#endif
-#ifdef BB_INSMOD               //sbin
-    {"insmod", insmod_main},
-#endif
-#ifdef BB_FEATURE_LINUXRC      //
-    {"linuxrc", init_main},
-#endif
-#ifdef BB_KILL                 //bin
-    {"kill", kill_main},
-#endif
-#ifdef BB_LENGTH               //usr/bin
-    {"length", length_main},
-#endif
-#ifdef BB_LN                   //bin
-    {"ln", ln_main},
-#endif
-#ifdef BB_LOADFONT             //usr/bin
-    {"loadfont", loadfont_main},
-#endif
-#ifdef BB_LOADKMAP             //sbin
-    {"loadkmap", loadkmap_main},
-#endif
-#ifdef BB_LS                   //bin
-    {"ls", ls_main},
-#endif
-#ifdef BB_LSMOD                        //sbin
-    {"lsmod", lsmod_main},
-#endif
-#ifdef BB_MAKEDEVS             //sbin
-    {"makedevs", makedevs_main},
-#endif
-#ifdef BB_MATH                 //usr/bin
-    {"math", math_main},
-#endif
-#ifdef BB_MKDIR                        //bin
-    {"mkdir", mkdir_main},
-#endif
-#ifdef BB_MKFIFO               //usr/bin
-    {"mkfifo", mkfifo_main},
-#endif
-#ifdef BB_MKNOD                        //bin
-    {"mknod", mknod_main},
-#endif
-#ifdef BB_MKSWAP               //sbin
-    {"mkswap", mkswap_main},
-#endif
-#ifdef BB_MNC                  //usr/bin
-    {"mnc", mnc_main},
-#endif
-#ifdef BB_MORE                 //bin
-    {"more", more_main},
-#endif
-#ifdef BB_MOUNT                        //bin
-    {"mount", mount_main},
-#endif
-#ifdef BB_MT                   //bin
-    {"mt", mt_main},
-#endif
-#ifdef BB_MV                   //bin
-    {"mv", mv_main},
-#endif
-#ifdef BB_PING                  //bin
-    {"ping", ping_main},
-#endif
-#ifdef BB_POWEROFF              //sbin
-    {"poweroff", poweroff_main},
-#endif
-#ifdef BB_PRINTF               //usr/bin
-    {"printf", printf_main},
-#endif
-#ifdef BB_PS                   //bin
-    {"ps", ps_main},
-#endif
-#ifdef BB_PWD                  //bin
-    {"pwd", pwd_main},
-#endif
-#ifdef BB_REBOOT               //sbin
-    {"reboot", reboot_main},
-#endif
-#ifdef BB_RM                   //bin
-    {"rm", rm_main},
-#endif
-#ifdef BB_RMDIR                        //bin
-    {"rmdir", rmdir_main},
-#endif
-#ifdef BB_RMMOD                        //sbin
-    {"rmmod", rmmod_main},
-#endif
-#ifdef BB_SFDISK               //sbin
-    {"fdisk", sfdisk_main},
-    {"sfdisk", sfdisk_main},
-#endif
-#ifdef BB_SED                  //bin
-    {"sed", sed_main},
-#endif
-#ifdef BB_SLEEP                        //bin
-    {"sleep", sleep_main},
-#endif
-#ifdef BB_SORT                 //bin
-    {"sort", sort_main},
-#endif
-#ifdef BB_SYNC                 //bin
-    {"sync", sync_main},
-#endif
-#ifdef BB_SYSLOGD              //sbin
-    {"syslogd", syslogd_main},
-#endif
-#ifdef BB_LOGGER               //usr/bin
-    {"logger", logger_main},
-#endif
-#ifdef BB_SWAPONOFF            //sbin
-    {"swapon", swap_on_off_main},
-    {"swapoff", swap_on_off_main},
-#endif
-#ifdef BB_TAIL                 //usr/bin
-    {"tail", tail_main},
-#endif
-#ifdef BB_TAR                  //bin
-    {"tar", tar_main},
-#endif
-#ifdef BB_TEE                  //bin
-    {"tee", tee_main},
-#endif
-#ifdef BB_TOUCH                        //usr/bin
-    {"touch", touch_main},
-#endif
-#ifdef BB_TRUE_FALSE           //bin
-    {"true", true_main},
-    {"false", false_main},
-#endif
-#ifdef BB_UNAME                        //bin
-    {"uname",  uname_main},
-#endif
-#ifdef BB_UMOUNT               //bin
-    {"umount",  umount_main},
-#endif
-#ifdef BB_UPDATE               //sbin
-    {"update", update_main},
-#endif
-#ifdef BB_GUNZIP               //bin
-    {"zcat", gunzip_main},
-    {"gunzip", gunzip_main},
-#endif
-#ifdef BB_GZIP                 //bin
-    {"gzip", gzip_main},
-#endif
-#ifdef BB_LOADACM               //usr/bin
-    {"loadacm", loadacm_main},
-#endif    
-    {0}
+static const char* const install_dir[] = {
+       &usr_bin [8], /* "", equivalent to "/" for concat_path_file() */
+       &usr_bin [4], /* "/bin" */
+       &usr_sbin[4], /* "/sbin" */
+       usr_bin,
+       usr_sbin
 };
 
-int main(int argc, char **argv)
+/* abstract link() */
+typedef int (*__link_f)(const char *, const char *);
+
+/* create (sym)links for each applet */
+static void install_links(const char *busybox, int use_symbolic_links)
 {
-    char *s = argv[0];
-    char *name = argv[0];
-    const struct Applet *a = applets;
+       __link_f Link = link;
 
-    while (*s != '\0') {
-       if (*s++ == '/')
-           name = s;
-    }
+       char *fpc;
+       int i;
+       int rc;
 
-    while (a->name != 0) {
-       if (strcmp(name, a->name) == 0) {
-           int status;
+       if (use_symbolic_links)
+               Link = symlink;
 
-           status = ((*(a->main)) (argc, argv));
-           if (status < 0) {
-               fprintf(stderr, "%s: %s\n", a->name, strerror(errno));
-           }
-           fprintf(stderr, "\n");
-           exit(status);
+       for (i = 0; applets[i].name != NULL; i++) {
+               fpc = concat_path_file(
+                       install_dir[applets[i].location], applets[i].name);
+               rc = Link(busybox, fpc);
+               if (rc!=0 && errno!=EEXIST) {
+                       bb_perror_msg("%s", fpc);
+               }
+               free(fpc);
        }
-       a++;
-    }
-    exit (busybox_main(argc, argv));
 }
 
+#else
+#define install_links(x,y)
+#endif /* CONFIG_FEATURE_INSTALLER */
+
+int main(int argc, char **argv)
+{
+       const char *s;
+
+       bb_applet_name=argv[0];
+       if (*bb_applet_name == '-') bb_applet_name++;
+       for (s = bb_applet_name; *s ;)
+               if (*(s++) == '/') bb_applet_name = s;
+
+       /* Set locale for everybody except `init' */
+       if(ENABLE_LOCALE_SUPPORT && (!ENABLE_INIT || getpid()==1))
+               setlocale(LC_ALL, "");
+
+       run_applet_by_name(bb_applet_name, argc, argv);
+       bb_error_msg_and_die("applet not found");
+}
 
 int busybox_main(int argc, char **argv)
 {
-    int col=0;
-    argc--;
-    argv++;
+       /*
+        * This style of argument parsing doesn't scale well
+        * in the event that busybox starts wanting more --options.
+        * If someone has a cleaner approach, by all means implement it.
+        */
+       if (ENABLE_FEATURE_INSTALLER && argc > 1 && !strcmp(argv[1], "--install")) {
+               int use_symbolic_links = 0;
+               int rc = 0;
+               char *busybox;
 
-    if (been_there_done_that == 1 || argc < 1) {
-       const struct Applet *a = applets;
-       fprintf(stderr, "BusyBox v%s (%s) multi-call binary -- GPL2\n\n",
-               BB_VER, BB_BT);
-       fprintf(stderr, "Usage: busybox [function] [arguments]...\n");
-       fprintf(stderr, "   or: [function] [arguments]...\n\n");
-       fprintf(stderr,
-               "\tMost people will create a symlink to busybox for each\n"
-               "\tfunction name, and busybox will act like whatever you invoke it as.\n");
-       fprintf(stderr, "\nCurrently defined functions:\n");
+               /* to use symlinks, or not to use symlinks... */
+               if (argc > 2) {
+                       if ((strcmp(argv[2], "-s") == 0)) {
+                               use_symbolic_links = 1;
+                       }
+               }
 
-       while (a->name != 0) {
-           col+=fprintf(stderr, "%s%s", ((col==0)? "\t":", "), (a++)->name);
-           if (col>60) {
-               fprintf(stderr, ",\n");
-               col=0;
-           }
+               /* link */
+               busybox = xreadlink("/proc/self/exe");
+               if (busybox) {
+                       install_links(busybox, use_symbolic_links);
+                       free(busybox);
+               } else {
+                       rc = 1;
+               }
+               return rc;
        }
-       fprintf(stderr, "\n\n");
-       exit(-1);
-    } else {
-       /* If we've already been here once, exit now */
-       been_there_done_that = 1;
-       return (main(argc, argv));
-    }
+
+       /* Deal with --help.  (Also print help when called with no arguments) */
+       
+       if (argc==1 || !strcmp(argv[1],"--help") ) {
+               if (argc>2) {
+                       run_applet_by_name(bb_applet_name=argv[2], 2, argv);
+               } else {
+                       const struct BB_applet *a;
+                       int col, output_width;
+
+                       if (ENABLE_FEATURE_AUTOWIDTH) {
+                               /* Obtain the terminal width.  */
+                               get_terminal_width_height(0, &output_width, NULL);
+                               /* leading tab and room to wrap */
+                               output_width -= 20;
+                       } else output_width = 60;
+
+                       printf("%s\n\n"
+                              "Usage: busybox [function] [arguments]...\n"
+                              "   or: [function] [arguments]...\n\n"
+                              "\tBusyBox is a multi-call binary that combines many common Unix\n"
+                              "\tutilities into a single executable.  Most people will create a\n"
+                              "\tlink to busybox for each function they wish to use and BusyBox\n"
+                              "\twill act like whatever it was invoked as!\n"
+                              "\nCurrently defined functions:\n", bb_msg_full_version);
+
+                       col=0;
+                       for(a = applets; a->name;) {
+                               col += printf("%s%s", (col ? ", " : "\t"), (a++)->name);
+                               if (col > output_width && a->name) {
+                                       printf(",\n");
+                                       col = 0;
+                               }
+                       }
+                       printf("\n\n");
+                       exit(0);
+               }
+       } else run_applet_by_name(bb_applet_name=argv[1], argc-1, argv+1);
+       
+       bb_error_msg_and_die("applet not found");
 }
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/