X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=applets%2Fbusybox.c;h=833e8d37292ccb8fa603aa43d311182692a256b8;hb=f51d2069ff3a38be49a86212e1d62f2fd89c5943;hp=b65b13e65598575cdb93aa298decc2988a7222a9;hpb=61677feff7f549a48267c2c0c50a7420de6e2599;p=oweals%2Fbusybox.git

diff --git a/applets/busybox.c b/applets/busybox.c
index b65b13e65..833e8d372 100644
--- a/applets/busybox.c
+++ b/applets/busybox.c
@@ -1,405 +1,151 @@
 /* vi: set sw=4 ts=4: */
-#include "internal.h"
 #include <stdio.h>
 #include <string.h>
+#include <unistd.h>
 #include <errno.h>
-
-static int been_there_done_that = 0;
-
-/* It has been alledged that doing such things can
- * help reduce binary size when staticly linking,
- * of course with glibc, this is unlikely as long
- * as we use things like printf -- perhaps a printf
- * replacement may be in order 
- */
-#if 0
-void exit(int status) __attribute__ ((noreturn));
-void exit(int status)
-{
-	_exit(status);
-};
-void abort(void) __attribute__ ((__noreturn__));
-void abort(void)
-{
-	_exit(0);
-};
-int atexit(void (*__func) (void))
-{
-	_exit(0);
-};
-void *__libc_stack_end;
+#include <stdlib.h>
+#include "busybox.h"
+#if ENABLE_LOCALE_SUPPORT
+#include <locale.h>
+#else
+#define setlocale(x,y)
 #endif
 
-static const struct Applet applets[] = {
+const char *bb_applet_name;
 
-#ifdef BB_BASENAME
-	{"basename", basename_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_BUSYBOX
-	{"busybox", busybox_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_BLOCK_DEVICE
-	{"block_device", block_device_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_CAT
-	{"cat", cat_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_CHMOD_CHOWN_CHGRP
-	{"chmod", chmod_chown_chgrp_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_CHMOD_CHOWN_CHGRP
-	{"chown", chmod_chown_chgrp_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_CHMOD_CHOWN_CHGRP
-	{"chgrp", chmod_chown_chgrp_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_CHROOT
-	{"chroot", chroot_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_CLEAR
-	{"clear", clear_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_CHVT
-	{"chvt", chvt_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_CP_MV
-	{"cp", cp_mv_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_CP_MV
-	{"mv", cp_mv_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_DATE
-	{"date", date_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_DD
-	{"dd", dd_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_DF
-	{"df", df_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_DIRNAME
-	{"dirname", dirname_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_DMESG
-	{"dmesg", dmesg_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_DU
-	{"du", du_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_DUTMP
-	{"dutmp", dutmp_main, _BB_DIR_USR_SBIN},
-#endif
-#ifdef BB_ECHO
-	{"echo", echo_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_FBSET
-	{"fbset", fbset_main, _BB_DIR_USR_SBIN},
-#endif
-#ifdef BB_FDFLUSH
-	{"fdflush", fdflush_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_FIND
-	{"find", find_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_FREE
-	{"free", free_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_FREERAMDISK
-	{"freeramdisk", freeramdisk_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_DEALLOCVT
-	{"deallocvt", deallocvt_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_FSCK_MINIX
-	{"fsck.minix", fsck_minix_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_MKFS_MINIX
-	{"mkfs.minix", mkfs_minix_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_GREP
-	{"grep", grep_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_HALT
-	{"halt", halt_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_HEAD
-	{"head", head_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_HOSTID
-	{"hostid", hostid_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_HOSTNAME
-	{"hostname", hostname_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_INIT
-	{"init", init_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_INSMOD
-	{"insmod", insmod_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_FEATURE_LINUXRC		//
-	{"linuxrc", init_main, _BB_DIR_ROOT},
-#endif
-#ifdef BB_KILL
-	{"kill", kill_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_KILLALL
-	{"killall", kill_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_LENGTH
-	{"length", length_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_LN
-	{"ln", ln_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_LOADACM
-	{"loadacm", loadacm_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_LOADFONT
-	{"loadfont", loadfont_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_LOADKMAP
-	{"loadkmap", loadkmap_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_LS
-	{"ls", ls_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_LSMOD
-	{"lsmod", lsmod_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_MAKEDEVS
-	{"makedevs", makedevs_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_MATH
-	{"math", math_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_MKDIR
-	{"mkdir", mkdir_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_MKFIFO
-	{"mkfifo", mkfifo_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_MKNOD
-	{"mknod", mknod_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_MKSWAP
-	{"mkswap", mkswap_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_MNC
-	{"mnc", mnc_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_MORE
-	{"more", more_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_MOUNT
-	{"mount", mount_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_MT
-	{"mt", mt_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_NSLOOKUP
-	{"nslookup", nslookup_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_PING
-	{"ping", ping_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_POWEROFF
-	{"poweroff", poweroff_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_PRINTF
-	{"printf", printf_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_PS
-	{"ps", ps_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_PWD
-	{"pwd", pwd_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_REBOOT
-	{"reboot", reboot_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_RM
-	{"rm", rm_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_RMDIR
-	{"rmdir", rmdir_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_RMMOD
-	{"rmmod", rmmod_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_SED
-	{"sed", sed_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_SH
-	{"sh", shell_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_SFDISK
-	{"sfdisk", sfdisk_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_SLEEP
-	{"sleep", sleep_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_SORT
-	{"sort", sort_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_SYNC
-	{"sync", sync_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_SYSLOGD
-	{"syslogd", syslogd_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_LOGGER
-	{"logger", logger_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_LOGNAME
-	{"logname", logname_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_SWAPONOFF
-	{"swapon", swap_on_off_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_SWAPONOFF
-	{"swapoff", swap_on_off_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_TAIL
-	{"tail", tail_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_TAR
-	{"tar", tar_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_TELNET
-	{"telnet", telnet_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_TEST
-	{"[", test_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_TEST
-	{"test", test_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_TEE
-	{"tee", tee_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_TOUCH
-	{"touch", touch_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_TR
-	{"tr", tr_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_TRUE_FALSE
-	{"true", true_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_TRUE_FALSE
-	{"false", false_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_TTY
-	{"tty", tty_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_UMOUNT
-	{"umount", umount_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_UNAME
-	{"uname", uname_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_UPTIME
-	{"uptime", uptime_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_UNIQ
-	{"uniq", uniq_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_UPDATE
-	{"update", update_main, _BB_DIR_SBIN},
-#endif
-#ifdef BB_USLEEP
-	{"usleep", usleep_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_WC
-	{"wc", wc_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_WHOAMI
-	{"whoami", whoami_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_YES
-	{"yes", yes_main, _BB_DIR_USR_BIN},
-#endif
-#ifdef BB_GUNZIP
-	{"zcat", gunzip_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_GUNZIP
-	{"gunzip", gunzip_main, _BB_DIR_BIN},
-#endif
-#ifdef BB_GZIP
-	{"gzip", gzip_main, _BB_DIR_BIN},
-#endif
-	{0}
+#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";
+
+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
 };
 
+/* abstract link() */
+typedef int (*__link_f)(const char *, const char *);
 
-
-int main(int argc, char **argv)
+/* create (sym)links for each applet */
+static void install_links(const char *busybox, int use_symbolic_links)
 {
-	char				*s;
-	char				*name;
-	const struct Applet	*a		= applets;
-
-	for (s = name = argv[0]; *s != '\0';) {
-		if (*s++ == '/')
-			name = s;
-	}
+	__link_f Link = link;
 
-	*argv = name;
+	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);
 		}
-		a++;
+		free(fpc);
 	}
-	exit(busybox_main(argc, argv));
 }
 
+#else
+#define install_links(x,y)
+#endif /* CONFIG_FEATURE_INSTALLER */
 
-int busybox_main(int argc, char **argv)
+int main(int argc, char **argv)
 {
-	int col = 0;
+	const char *s;
 
-	argc--;
-	argv++;
+	bb_applet_name=argv[0];
+	if (*bb_applet_name == '-') bb_applet_name++;
+	for (s = bb_applet_name; *s ;)
+		if (*(s++) == '/') bb_applet_name = s;
 
-	if (been_there_done_that == 1 || argc < 1) {
-		const struct Applet *a = applets;
+	/* Set locale for everybody except `init' */
+	if(ENABLE_LOCALE_SUPPORT && (!ENABLE_INIT || getpid()==1))
+		setlocale(LC_ALL, "");
 
-		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,
-				"\tBusyBox is a multi-call binary that combines many common Unix utilities into a\n"
-				"\tsingle executable.  Most people will create a link to busybox for each function\n"
-				"\tthey wish to use, and BusyBox will act like whatever it was invoked as.\n");
-		fprintf(stderr, "\nCurrently defined functions:\n");
+	run_applet_by_name(bb_applet_name, argc, argv);
+	bb_error_msg_and_die("applet not found");
+}
 
-		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;
+int busybox_main(int argc, char **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;
+
+		/* to use symlinks, or not to use symlinks... */
+		if (argc > 2) {
+			if ((strcmp(argv[2], "-s") == 0)) {
+				use_symbolic_links = 1;
 			}
 		}
-		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));
+
+		/* link */
+		busybox = xreadlink("/proc/self/exe");
+		if (busybox) {
+			install_links(busybox, use_symbolic_links);
+			free(busybox);
+		} else {
+			rc = 1;
+		}
+		return rc;
 	}
+
+	/* 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");
 }
 
 /*