2 * This program is free software; you can redistribute it and/or
3 * modify it under the terms of the GNU General Public
4 * License v2 as published by the Free Software Foundation.
6 * This program is distributed in the hope that it will be useful,
7 * but WITHOUT ANY WARRANTY; without even the implied warranty of
8 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
9 * General Public License for more details.
11 * You should have received a copy of the GNU General Public
12 * License along with this program; if not, write to the
13 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
14 * Boston, MA 021110-1307, USA.
25 static const char btrfs_cmd_group_usage[] =
26 "btrfs [--help] [--version] <group> [<group>...] <command> [<args>]";
28 static const char btrfs_cmd_group_info[] =
29 "Use --help as an argument for information on a specific group or command.";
31 char argv0_buf[ARGV0_BUF_SIZE] = "btrfs";
33 static inline const char *skip_prefix(const char *str, const char *prefix)
35 size_t len = strlen(prefix);
36 return strncmp(str, prefix, len) ? NULL : str + len;
39 int prefixcmp(const char *str, const char *prefix)
41 for (; ; str++, prefix++)
44 else if (*str != *prefix)
45 return (unsigned char)*prefix - (unsigned char)*str;
48 static int parse_one_token(const char *arg, const struct cmd_group *grp,
49 const struct cmd_struct **cmd_ret)
51 const struct cmd_struct *cmd = grp->commands;
52 const struct cmd_struct *abbrev_cmd = NULL, *ambiguous_cmd = NULL;
54 for (; cmd->token; cmd++) {
57 rest = skip_prefix(arg, cmd->token);
59 if (!prefixcmp(cmd->token, arg)) {
62 * If this is abbreviated, it is
63 * ambiguous. So when there is no
64 * exact match later, we need to
67 ambiguous_cmd = abbrev_cmd;
84 *cmd_ret = abbrev_cmd;
91 static const struct cmd_struct *
92 parse_command_token(const char *arg, const struct cmd_group *grp)
94 const struct cmd_struct *cmd;
96 switch(parse_one_token(arg, grp, &cmd)) {
98 help_unknown_token(arg, grp);
100 help_ambiguous_token(arg, grp);
106 void handle_help_options_next_level(const struct cmd_struct *cmd,
107 int argc, char **argv)
112 if (!strcmp(argv[1], "--help")) {
116 help_command_group(cmd->next, argc, argv);
118 usage_command(cmd, 1, 0);
125 static void fixup_argv0(char **argv, const char *token)
127 int len = strlen(argv0_buf);
129 snprintf(argv0_buf + len, sizeof(argv0_buf) - len, " %s", token);
133 int handle_command_group(const struct cmd_group *grp, int argc,
137 const struct cmd_struct *cmd;
142 usage_command_group(grp, 0, 0);
146 cmd = parse_command_token(argv[0], grp);
148 handle_help_options_next_level(cmd, argc, argv);
150 fixup_argv0(argv, cmd->token);
151 return cmd->fn(argc, argv);
154 int check_argc_exact(int nargs, int expected)
156 if (nargs < expected)
157 fprintf(stderr, "%s: too few arguments\n", argv0_buf);
158 if (nargs > expected)
159 fprintf(stderr, "%s: too many arguments\n", argv0_buf);
161 return nargs != expected;
164 int check_argc_min(int nargs, int expected)
166 if (nargs < expected) {
167 fprintf(stderr, "%s: too few arguments\n", argv0_buf);
174 int check_argc_max(int nargs, int expected)
176 if (nargs > expected) {
177 fprintf(stderr, "%s: too many arguments\n", argv0_buf);
184 const struct cmd_group btrfs_cmd_group;
186 static const char * const cmd_help_usage[] = {
187 "btrfs help [--full]",
188 "Dislay help information",
190 "--full display detailed help on every command",
194 static int cmd_help(int argc, char **argv)
196 help_command_group(&btrfs_cmd_group, argc, argv);
200 static const char * const cmd_version_usage[] = {
202 "Display btrfs-progs version",
206 static int cmd_version(int argc, char **argv)
208 printf("%s\n", BTRFS_BUILD_VERSION);
212 static int handle_options(int *argc, char ***argv)
214 char **orig_argv = *argv;
217 const char *arg = (*argv)[0];
221 if (!strcmp(arg, "--help")) {
223 } else if (!strcmp(arg, "--version")) {
226 fprintf(stderr, "Unknown option: %s\n", arg);
227 fprintf(stderr, "usage: %s\n",
228 btrfs_cmd_group.usagestr);
236 return (*argv) - orig_argv;
239 const struct cmd_group btrfs_cmd_group = {
240 btrfs_cmd_group_usage, btrfs_cmd_group_info, {
241 { "subvolume", cmd_subvolume, NULL, &subvolume_cmd_group, 0 },
242 { "filesystem", cmd_filesystem, NULL, &filesystem_cmd_group, 0 },
243 { "device", cmd_device, NULL, &device_cmd_group, 0 },
244 { "scrub", cmd_scrub, NULL, &scrub_cmd_group, 0 },
245 { "inspect-internal", cmd_inspect, NULL, &inspect_cmd_group, 0 },
246 { "help", cmd_help, cmd_help_usage, NULL, 0 },
247 { "version", cmd_version, cmd_version_usage, NULL, 0 },
252 int main(int argc, char **argv)
254 const struct cmd_struct *cmd;
258 handle_options(&argc, &argv);
260 if (!prefixcmp(argv[0], "--"))
263 usage_command_group(&btrfs_cmd_group, 0, 0);
267 cmd = parse_command_token(argv[0], &btrfs_cmd_group);
269 handle_help_options_next_level(cmd, argc, argv);
271 fixup_argv0(argv, cmd->token);
272 exit(cmd->fn(argc, argv));