1 #include <signal.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4 #include <string.h>
5 #include <unistd.h>
6
7 int main(int, char **);
8
toolbox_main(int argc,char ** argv)9 static int toolbox_main(int argc, char **argv)
10 {
11 // "toolbox foo ..." is equivalent to "foo ..."
12 if (argc > 1) {
13 return main(argc - 1, argv + 1);
14 } else {
15 printf("Toolbox!\n");
16 return 0;
17 }
18 }
19
20 #define TOOL(name) int name##_main(int, char**);
21 #include "tools.h"
22 #undef TOOL
23
24 static struct
25 {
26 const char *name;
27 int (*func)(int, char**);
28 } tools[] = {
29 { "toolbox", toolbox_main },
30 #define TOOL(name) { #name, name##_main },
31 #include "tools.h"
32 #undef TOOL
33 { 0, 0 },
34 };
35
SIGPIPE_handler(int signal)36 static void SIGPIPE_handler(int signal) {
37 // Those desktop Linux tools that catch SIGPIPE seem to agree that it's
38 // a successful way to exit, not a failure. (Which makes sense --- we were
39 // told to stop by a reader, rather than failing to continue ourselves.)
40 _exit(0);
41 }
42
main(int argc,char ** argv)43 int main(int argc, char **argv)
44 {
45 int i;
46 char *name = argv[0];
47
48 // Let's assume that none of this code handles broken pipes. At least ls,
49 // ps, and top were broken (though I'd previously added this fix locally
50 // to top). We exit rather than use SIG_IGN because tools like top will
51 // just keep on writing to nowhere forever if we don't stop them.
52 signal(SIGPIPE, SIGPIPE_handler);
53
54 if((argc > 1) && (argv[1][0] == '@')) {
55 name = argv[1] + 1;
56 argc--;
57 argv++;
58 } else {
59 char *cmd = strrchr(argv[0], '/');
60 if (cmd)
61 name = cmd + 1;
62 }
63
64 for(i = 0; tools[i].name; i++){
65 if(!strcmp(tools[i].name, name)){
66 return tools[i].func(argc, argv);
67 }
68 }
69
70 printf("%s: no such tool\n", argv[0]);
71 return -1;
72 }
73