-#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"
-static int been_there_done_that = 0;
+#undef APPLET
+#undef APPLET_NOUSAGE
+#undef PROTOTYPES
+#include "applets.h"
-static const struct Applet applets[] = {
+#define bb_need_full_version
+#define BB_DECLARE_EXTERN
+#include "messages.c"
-#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
- {0}
+static int been_there_done_that = 0;
+const char *applet_name;
+
+#ifdef BB_FEATURE_INSTALLER
+/*
+ * directory table
+ * this should be consistent w/ the enum, busybox.h::Location,
+ * or else...
+ */
+static char* install_dir[] = {
+ "/",
+ "/bin",
+ "/sbin",
+ "/usr/bin",
+ "/usr/sbin",
};
-int main(int argc, char **argv)
+/* abstract link() */
+typedef int (*__link_f)(const char *, const char *);
+
+/*
+ * Where in the filesystem is this busybox?
+ * [return]
+ * malloc'd string w/ full pathname of busybox's location
+ * NULL on failure
+ */
+static char *busybox_fullpath()
+{
+ pid_t pid;
+ char path[256];
+ char proc[256];
+ int len;
+
+ pid = getpid();
+ sprintf(proc, "/proc/%d/exe", pid);
+ len = readlink(proc, path, 256);
+ if (len != -1) {
+ path[len] = 0;
+ } else {
+ perror_msg("%s", proc);
+ return NULL;
+ }
+ return strdup(path);
+}
+
+/* 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;
+
+ char command[256];
+ int i;
+ int rc;
+
+ if (use_symbolic_links)
+ Link = symlink;
+
+ for (i = 0; applets[i].name != NULL; i++) {
+ sprintf ( command, "%s/%s",
+ install_dir[applets[i].location], applets[i].name);
+ rc = Link(busybox, command);
- while (*s != '\0') {
- if (*s++ == '/')
- name = s;
- }
+ if (rc) {
+ perror_msg("%s", command);
+ }
+ }
+}
+
+#endif /* BB_FEATURE_INSTALLER */
+
+int main(int argc, char **argv)
+{
+ const char *s;
- while (a->name != 0) {
- if (strcmp(name, a->name) == 0) {
- int status;
+ for (s = applet_name = argv[0]; *s != '\0';) {
+ if (*s++ == '/')
+ applet_name = s;
+ }
- status = ((*(a->main)) (argc, argv));
- if (status < 0) {
- fprintf(stderr, "%s: %s\n", a->name, strerror(errno));
- }
- fprintf(stderr, "\n");
- exit(status);
+#ifdef BB_SH
+ /* Add in a special case hack -- whenever **argv == '-'
+ * (i.e. '-su' or '-sh') always invoke the shell */
+ if (**argv == '-' && *(*argv+1)!= '-') {
+ exit(((*(shell_main)) (argc, argv)));
}
- a++;
- }
- exit (busybox_main(argc, argv));
+#endif
+
+ run_applet_by_name(applet_name, argc, argv);
+ error_msg_and_die("applet not found");
}
int busybox_main(int argc, char **argv)
{
- int col=0;
- argc--;
- argv++;
+ int col = 0, len, i;
+
+#ifdef BB_FEATURE_INSTALLER
+ /*
+ * 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 (argc > 1 && (strcmp(argv[1], "--install") == 0)) {
+ 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 = busybox_fullpath();
+ if (busybox) {
+ install_links(busybox, use_symbolic_links);
+ free(busybox);
+ } else {
+ rc = 1;
+ }
+ return rc;
}
- fprintf(stderr, "\n\n");
- exit(-1);
- } else {
+#endif /* BB_FEATURE_INSTALLER */
+
+ argc--;
+
/* If we've already been here once, exit now */
+ if (been_there_done_that == 1 || argc < 1) {
+ const struct BB_applet *a = applets;
+
+ fprintf(stderr, "%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", full_version);
+
+ while (a->name != 0) {
+ col +=
+ fprintf(stderr, "%s%s", ((col == 0) ? "\t" : ", "),
+ (a++)->name);
+ if (col > 60 && a->name != 0) {
+ fprintf(stderr, ",\n");
+ col = 0;
+ }
+ }
+ fprintf(stderr, "\n\n");
+ exit(-1);
+ }
+
+ /* Flag that we've been here already */
been_there_done_that = 1;
+
+ /* Move the command line down a notch */
+ len = argv[argc] + strlen(argv[argc]) - argv[1];
+ memmove(argv[0], argv[1], len);
+ memset(argv[0] + len, 0, argv[1] - argv[0]);
+
+ /* Fix up the argv pointers */
+ len = argv[1] - argv[0];
+ memmove(argv, argv + 1, sizeof(char *) * (argc + 1));
+ for (i = 0; i < argc; i++)
+ argv[i] -= len;
+
return (main(argc, argv));
- }
}
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/
projects / oweals / busybox.git / blobdiff