mirror of
https://github.com/sheumann/hush.git
synced 2024-08-20 04:28:57 +00:00
218 lines
4.7 KiB
C
218 lines
4.7 KiB
C
/* vi: set sw=4 ts=4: */
|
|
#include "busybox.h"
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#undef APPLET
|
|
#undef APPLET_NOUSAGE
|
|
#undef PROTOTYPES
|
|
#include "applets.h"
|
|
|
|
#define bb_need_full_version
|
|
#define BB_DECLARE_EXTERN
|
|
#include "messages.c"
|
|
|
|
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",
|
|
};
|
|
|
|
/* 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 {
|
|
error_msg("%s: %s\n", proc, strerror(errno));
|
|
return NULL;
|
|
}
|
|
return strdup(path);
|
|
}
|
|
|
|
/* create (sym)links for each applet */
|
|
static void install_links(const char *busybox, int use_symbolic_links)
|
|
{
|
|
__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);
|
|
|
|
if (rc) {
|
|
error_msg("%s: %s\n", command, strerror(errno));
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif /* BB_FEATURE_INSTALLER */
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
struct BB_applet search_applet, *applet;
|
|
const char *s;
|
|
applet_name = "busybox";
|
|
|
|
#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;
|
|
|
|
/* to use symlinks, or not to use symlinks... */
|
|
if (argc > 2) {
|
|
if ((strcmp(argv[2], "-s") == 0)) {
|
|
use_symbolic_links = 1;
|
|
}
|
|
}
|
|
|
|
/* link */
|
|
busybox = busybox_fullpath();
|
|
if (busybox) {
|
|
install_links(busybox, use_symbolic_links);
|
|
free(busybox);
|
|
} else {
|
|
rc = 1;
|
|
}
|
|
return rc;
|
|
}
|
|
#endif /* BB_FEATURE_INSTALLER */
|
|
|
|
for (s = applet_name = argv[0]; *s != '\0';) {
|
|
if (*s++ == '/')
|
|
applet_name = s;
|
|
}
|
|
|
|
#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)));
|
|
}
|
|
#endif
|
|
|
|
/* Do a binary search to find the applet entry given the name. */
|
|
search_applet.name = applet_name;
|
|
applet = bsearch(&search_applet, applets, NUM_APPLETS,
|
|
sizeof(struct BB_applet), applet_name_compare);
|
|
if (applet != NULL) {
|
|
if (applet->usage && argv[1] && strcmp(argv[1], "--help") == 0)
|
|
usage(applet->usage);
|
|
exit((*(applet->main)) (argc, argv));
|
|
}
|
|
|
|
return(busybox_main(argc, argv));
|
|
}
|
|
|
|
|
|
int busybox_main(int argc, char **argv)
|
|
{
|
|
int col = 0;
|
|
int ps_index;
|
|
char *index, *index2;
|
|
|
|
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;
|
|
|
|
/* We do not want the word "busybox" to show up in ps, so we move
|
|
* everything in argv around to fake ps into showing what we want it to
|
|
* show. Since we are only shrinking the string, we don't need to move
|
|
* __environ or any of that tedious stuff... */
|
|
ps_index = 0;
|
|
index=*argv;
|
|
index2=argv[argc];
|
|
index2+=strlen(argv[argc]);
|
|
while(ps_index < argc) {
|
|
argv[ps_index]=index;
|
|
memmove(index, argv[ps_index+1], strlen(argv[ps_index+1])+1);
|
|
index+=(strlen(index));
|
|
*index='\0';
|
|
index++;
|
|
ps_index++;
|
|
}
|
|
while(index<=index2)
|
|
*index++='\0';
|
|
argv[ps_index]=NULL;
|
|
|
|
return (main(argc, argv));
|
|
}
|
|
|
|
/*
|
|
Local Variables:
|
|
c-file-style: "linux"
|
|
c-basic-offset: 4
|
|
tab-width: 4
|
|
End:
|
|
*/
|