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