1#include <signal.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <string.h>
5#include <unistd.h>
6
7int main(int, char **);
8
9static 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
24static 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
36static 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
43int 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