73 lines
1.7 KiB
C
73 lines
1.7 KiB
C
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
int main(int, char **);
|
|
|
|
static int toolbox_main(int argc, char **argv)
|
|
{
|
|
// "toolbox foo ..." is equivalent to "foo ..."
|
|
if (argc > 1) {
|
|
return main(argc - 1, argv + 1);
|
|
} else {
|
|
printf("Toolbox!\n");
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
#define TOOL(name) int name##_main(int, char**);
|
|
#include "tools.h"
|
|
#undef TOOL
|
|
|
|
static struct
|
|
{
|
|
const char *name;
|
|
int (*func)(int, char**);
|
|
} tools[] = {
|
|
{ "toolbox", toolbox_main },
|
|
#define TOOL(name) { #name, name##_main },
|
|
#include "tools.h"
|
|
#undef TOOL
|
|
{ 0, 0 },
|
|
};
|
|
|
|
static void SIGPIPE_handler(int signal) {
|
|
// Those desktop Linux tools that catch SIGPIPE seem to agree that it's
|
|
// a successful way to exit, not a failure. (Which makes sense --- we were
|
|
// told to stop by a reader, rather than failing to continue ourselves.)
|
|
_exit(0);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int i;
|
|
char *name = argv[0];
|
|
|
|
// Let's assume that none of this code handles broken pipes. At least ls,
|
|
// ps, and top were broken (though I'd previously added this fix locally
|
|
// to top). We exit rather than use SIG_IGN because tools like top will
|
|
// just keep on writing to nowhere forever if we don't stop them.
|
|
signal(SIGPIPE, SIGPIPE_handler);
|
|
|
|
if((argc > 1) && (argv[1][0] == '@')) {
|
|
name = argv[1] + 1;
|
|
argc--;
|
|
argv++;
|
|
} else {
|
|
char *cmd = strrchr(argv[0], '/');
|
|
if (cmd)
|
|
name = cmd + 1;
|
|
}
|
|
|
|
for(i = 0; tools[i].name; i++){
|
|
if(!strcmp(tools[i].name, name)){
|
|
return tools[i].func(argc, argv);
|
|
}
|
|
}
|
|
|
|
printf("%s: no such tool\n", argv[0]);
|
|
return -1;
|
|
}
|