-#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_more_main},
-#endif
-#ifdef BB_CHMOD //bin
- {"chmod", chmod_main},
-#endif
-#ifdef BB_CHOWN //bin
- {"chown", chown_main},
- {"chgrp", chown_main},
-#endif
-#ifdef BB_CHROOT //sbin
- {"chroot", chroot_main},
-#endif
-#ifdef BB_CLEAR //usr/bin
- {"clear", clear_main},
-#endif
-#ifdef BB_CP //bin
- {"cp", dyadic_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_DUTMP //usr/sbin
- {"dutmp", cat_more_main},
-#endif
-#ifdef BB_FALSE //bin
- {"false", false_main},
-#endif
-#ifdef BB_FDFLUSH //bin
- {"fdflush", monadic_main},
-#endif
-#ifdef BB_FIND //usr/bin
- {"find", find_main},
-#endif
-#ifdef BB_GREP //bin
- {"grep", grep_main},
-#endif
-#ifdef BB_HALT //sbin
- {"halt", halt_main},
-#endif
-#ifdef BB_INIT //sbin
- {"init", 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", dyadic_main},
-#endif
-#ifdef BB_LOADKMAP //sbin
- {"loadkmap", loadkmap_main},
-#endif
-#ifdef BB_LOSETUP //sbin
- {"losetup", losetup_main},
-#endif
-#ifdef BB_LS //bin
- {"ls", ls_main},
-#endif
-#ifdef BB_MAKEDEVS //sbin
- {"makedevs", makedevs_main},
-#endif
-#ifdef BB_MATH //usr/bin
- {"math", math_main},
-#endif
-#ifdef BB_MKDIR //bin
- {"mkdir", monadic_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", cat_more_main},
-#endif
-#ifdef BB_MOUNT //bin
- {"mount", mount_main},
-#endif
-#ifdef BB_MT //bin
- {"mt", mt_main},
-#endif
-#ifdef BB_MV //bin
- {"mv", dyadic_main},
-#endif
-#ifdef BB_PRINTF //usr/bin
- {"printf", printf_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", monadic_main},
-#endif
-#ifdef BB_SLEEP //bin
- {"sleep", sleep_main},
-#endif
-#ifdef BB_TAR //bin
- {"tar", tar_main},
-#endif
-#ifdef BB_SWAPOFF //sbin
- {"swapoff", monadic_main},
-#endif
-#ifdef BB_SWAPON //sbin
- {"swapon", monadic_main},
-#endif
-#ifdef BB_SYNC //bin
- {"sync", sync_main},
-#endif
-#ifdef BB_TOUCH //usr/bin
- {"touch", monadic_main},
-#endif
-#ifdef BB_TRUE //bin
- {"true", true_main},
-#endif
-#ifdef BB_UMOUNT //bin
- {"umount", umount_main},
-#endif
-#ifdef BB_UPDATE //sbin
- {"update", update_main},
-#endif
-#ifdef BB_ZCAT //bin
- {"zcat", zcat_main},
- {"gunzip", zcat_main},
-#endif
-#ifdef BB_GZIP //bin
- {"gzip", gzip_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;
-
- while (*s != '\0') {
- if (*s++ == '/')
- name = s;
- }
-
- while (a->name != 0) {
- if (strcmp(name, a->name) == 0) {
- int status;
-
- status = ((*(a->main)) (argc, argv));
- if (status < 0) {
- fprintf(stderr, "%s: %s\n", a->name, strerror(errno));
- }
- fprintf(stderr, "\n");
- exit(status);
+ __link_f Link = link;
+
+ char *fpc;
+ int i;
+ int rc;
+
+ if (use_symbolic_links)
+ Link = symlink;
+
+ 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++;
- }
- return (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)
{
- argc--;
- argv++;
-
- /* If we've already been here once, exit now */
- if (been_there_done_that == 1)
- return -1;
- been_there_done_that = 1;
-
- if (argc < 1) {
- const struct Applet *a = applets;
- fprintf(stderr, "BusyBox v%s (%s) multi-call binary -- GPL2\n",
- BB_VER, BB_BT);
- fprintf(stderr, "Usage: busybox [function] [arguments]...\n");
- fprintf(stderr,
- "\n\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");
-
- if (a->name != 0) {
- fprintf(stderr, "%s", a->name);
- a++;
- }
- while (a->name != 0) {
- fprintf(stderr, ", %s", a->name);
- a++;
+ /*
+ * 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;
+
+ /* to use symlinks, or not to use symlinks... */
+ if (argc > 2) {
+ if ((strcmp(argv[2], "-s") == 0)) {
+ use_symbolic_links = 1;
+ }
+ }
+
+ /* 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
- 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:
+*/
projects / oweals / busybox.git / blobdiff