1 /*
2 * kmod - one tool to rule them all
3 *
4 * Copyright (C) 2011-2013 ProFUSION embedded systems
5 *
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include <errno.h>
21 #include <getopt.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25
26 #include <shared/util.h>
27
28 #include <libkmod/libkmod.h>
29
30 #include "kmod.h"
31
32 static const char options_s[] = "+hV";
33 static const struct option options[] = {
34 { "help", no_argument, NULL, 'h' },
35 { "version", no_argument, NULL, 'V' },
36 {}
37 };
38
39 static const struct kmod_cmd kmod_cmd_help;
40
41 static const struct kmod_cmd *kmod_cmds[] = {
42 &kmod_cmd_help,
43 &kmod_cmd_list,
44 &kmod_cmd_static_nodes,
45
46 #ifdef ENABLE_EXPERIMENTAL
47 &kmod_cmd_insert,
48 &kmod_cmd_remove,
49 #endif
50 };
51
52 static const struct kmod_cmd *kmod_compat_cmds[] = {
53 &kmod_cmd_compat_lsmod,
54 &kmod_cmd_compat_rmmod,
55 &kmod_cmd_compat_insmod,
56 &kmod_cmd_compat_modinfo,
57 &kmod_cmd_compat_modprobe,
58 &kmod_cmd_compat_depmod,
59 };
60
kmod_help(int argc,char * argv[])61 static int kmod_help(int argc, char *argv[])
62 {
63 size_t i;
64
65 printf("kmod - Manage kernel modules: list, load, unload, etc\n"
66 "Usage:\n"
67 "\t%s [options] command [command_options]\n\n"
68 "Options:\n"
69 "\t-V, --version show version\n"
70 "\t-h, --help show this help\n\n"
71 "Commands:\n", basename(argv[0]));
72
73 for (i = 0; i < ARRAY_SIZE(kmod_cmds); i++) {
74 if (kmod_cmds[i]->help != NULL) {
75 printf(" %-12s %s\n", kmod_cmds[i]->name,
76 kmod_cmds[i]->help);
77 }
78 }
79
80 puts("\nkmod also handles gracefully if called from following symlinks:");
81
82 for (i = 0; i < ARRAY_SIZE(kmod_compat_cmds); i++) {
83 if (kmod_compat_cmds[i]->help != NULL) {
84 printf(" %-12s %s\n", kmod_compat_cmds[i]->name,
85 kmod_compat_cmds[i]->help);
86 }
87 }
88
89 return EXIT_SUCCESS;
90 }
91
92 static const struct kmod_cmd kmod_cmd_help = {
93 .name = "help",
94 .cmd = kmod_help,
95 .help = "Show help message",
96 };
97
handle_kmod_commands(int argc,char * argv[])98 static int handle_kmod_commands(int argc, char *argv[])
99 {
100 const char *cmd;
101 int err = 0;
102 size_t i;
103
104 for (;;) {
105 int c;
106
107 c = getopt_long(argc, argv, options_s, options, NULL);
108 if (c == -1)
109 break;
110
111 switch (c) {
112 case 'h':
113 kmod_help(argc, argv);
114 return EXIT_SUCCESS;
115 case 'V':
116 puts(PACKAGE " version " VERSION);
117 puts(KMOD_FEATURES);
118 return EXIT_SUCCESS;
119 case '?':
120 return EXIT_FAILURE;
121 default:
122 fprintf(stderr, "Error: unexpected getopt_long() value '%c'.\n", c);
123 return EXIT_FAILURE;
124 }
125 }
126
127 if (optind >= argc) {
128 fputs("missing command\n", stderr);
129 goto fail;
130 }
131
132 cmd = argv[optind];
133
134 for (i = 0, err = -EINVAL; i < ARRAY_SIZE(kmod_cmds); i++) {
135 if (streq(kmod_cmds[i]->name, cmd)) {
136 err = kmod_cmds[i]->cmd(--argc, ++argv);
137 break;
138 }
139 }
140
141 if (err < 0) {
142 fprintf(stderr, "invalid command '%s'\n", cmd);
143 goto fail;
144 }
145
146 return err;
147
148 fail:
149 kmod_help(argc, argv);
150 return EXIT_FAILURE;
151 }
152
153
handle_kmod_compat_commands(int argc,char * argv[])154 static int handle_kmod_compat_commands(int argc, char *argv[])
155 {
156 const char *cmd;
157 size_t i;
158
159 cmd = basename(argv[0]);
160
161 for (i = 0; i < ARRAY_SIZE(kmod_compat_cmds); i++) {
162 if (streq(kmod_compat_cmds[i]->name, cmd))
163 return kmod_compat_cmds[i]->cmd(argc, argv);
164 }
165
166 return -ENOENT;
167 }
168
main(int argc,char * argv[])169 int main(int argc, char *argv[])
170 {
171 int err;
172
173 if (streq(program_invocation_short_name, "kmod"))
174 err = handle_kmod_commands(argc, argv);
175 else
176 err = handle_kmod_compat_commands(argc, argv);
177
178 return err;
179 }
180