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 * const btrfs_cmd_group_usage
[] = {
26 "btrfs [--help] [--version] <group> [<group>...] <command> [<args>]",
30 static const char btrfs_cmd_group_info
[] =
31 "Use --help as an argument for information on a specific group or command.";
33 char argv0_buf
[ARGV0_BUF_SIZE
] = "btrfs";
35 static inline const char *skip_prefix(const char *str
, const char *prefix
)
37 size_t len
= strlen(prefix
);
38 return strncmp(str
, prefix
, len
) ? NULL
: str
+ len
;
41 int prefixcmp(const char *str
, const char *prefix
)
43 for (; ; str
++, prefix
++)
46 else if (*str
!= *prefix
)
47 return (unsigned char)*prefix
- (unsigned char)*str
;
50 static int parse_one_token(const char *arg
, const struct cmd_group
*grp
,
51 const struct cmd_struct
**cmd_ret
)
53 const struct cmd_struct
*cmd
= grp
->commands
;
54 const struct cmd_struct
*abbrev_cmd
= NULL
, *ambiguous_cmd
= NULL
;
56 for (; cmd
->token
; cmd
++) {
59 rest
= skip_prefix(arg
, cmd
->token
);
61 if (!prefixcmp(cmd
->token
, arg
)) {
64 * If this is abbreviated, it is
65 * ambiguous. So when there is no
66 * exact match later, we need to
69 ambiguous_cmd
= abbrev_cmd
;
86 *cmd_ret
= abbrev_cmd
;
93 static const struct cmd_struct
*
94 parse_command_token(const char *arg
, const struct cmd_group
*grp
)
96 const struct cmd_struct
*cmd
;
98 switch(parse_one_token(arg
, grp
, &cmd
)) {
100 help_unknown_token(arg
, grp
);
102 help_ambiguous_token(arg
, grp
);
108 void handle_help_options_next_level(const struct cmd_struct
*cmd
,
109 int argc
, char **argv
)
114 if (!strcmp(argv
[1], "--help")) {
118 help_command_group(cmd
->next
, argc
, argv
);
120 usage_command(cmd
, 1, 0);
127 static void fixup_argv0(char **argv
, const char *token
)
129 int len
= strlen(argv0_buf
);
131 snprintf(argv0_buf
+ len
, sizeof(argv0_buf
) - len
, " %s", token
);
135 int handle_command_group(const struct cmd_group
*grp
, int argc
,
139 const struct cmd_struct
*cmd
;
144 usage_command_group(grp
, 0, 0);
148 cmd
= parse_command_token(argv
[0], grp
);
150 handle_help_options_next_level(cmd
, argc
, argv
);
152 fixup_argv0(argv
, cmd
->token
);
153 return cmd
->fn(argc
, argv
);
156 int check_argc_exact(int nargs
, int expected
)
158 if (nargs
< expected
)
159 fprintf(stderr
, "%s: too few arguments\n", argv0_buf
);
160 if (nargs
> expected
)
161 fprintf(stderr
, "%s: too many arguments\n", argv0_buf
);
163 return nargs
!= expected
;
166 int check_argc_min(int nargs
, int expected
)
168 if (nargs
< expected
) {
169 fprintf(stderr
, "%s: too few arguments\n", argv0_buf
);
176 int check_argc_max(int nargs
, int expected
)
178 if (nargs
> expected
) {
179 fprintf(stderr
, "%s: too many arguments\n", argv0_buf
);
186 const struct cmd_group btrfs_cmd_group
;
188 static const char * const cmd_help_usage
[] = {
189 "btrfs help [--full]",
190 "Dislay help information",
192 "--full display detailed help on every command",
196 static int cmd_help(int argc
, char **argv
)
198 help_command_group(&btrfs_cmd_group
, argc
, argv
);
202 static const char * const cmd_version_usage
[] = {
204 "Display btrfs-progs version",
208 static int cmd_version(int argc
, char **argv
)
210 printf("%s\n", BTRFS_BUILD_VERSION
);
214 static int handle_options(int *argc
, char ***argv
)
216 char **orig_argv
= *argv
;
219 const char *arg
= (*argv
)[0];
223 if (!strcmp(arg
, "--help")) {
225 } else if (!strcmp(arg
, "--version")) {
228 fprintf(stderr
, "Unknown option: %s\n", arg
);
229 fprintf(stderr
, "usage: %s\n",
230 btrfs_cmd_group
.usagestr
[0]);
238 return (*argv
) - orig_argv
;
241 const struct cmd_group btrfs_cmd_group
= {
242 btrfs_cmd_group_usage
, btrfs_cmd_group_info
, {
243 { "subvolume", cmd_subvolume
, NULL
, &subvolume_cmd_group
, 0 },
244 { "filesystem", cmd_filesystem
, NULL
, &filesystem_cmd_group
, 0 },
245 { "balance", cmd_balance
, NULL
, &balance_cmd_group
, 0 },
246 { "device", cmd_device
, NULL
, &device_cmd_group
, 0 },
247 { "scrub", cmd_scrub
, NULL
, &scrub_cmd_group
, 0 },
248 { "inspect-internal", cmd_inspect
, NULL
, &inspect_cmd_group
, 0 },
249 { "help", cmd_help
, cmd_help_usage
, NULL
, 0 },
250 { "version", cmd_version
, cmd_version_usage
, NULL
, 0 },
255 int main(int argc
, char **argv
)
257 const struct cmd_struct
*cmd
;
261 handle_options(&argc
, &argv
);
263 if (!prefixcmp(argv
[0], "--"))
266 usage_command_group(&btrfs_cmd_group
, 0, 0);
270 cmd
= parse_command_token(argv
[0], &btrfs_cmd_group
);
272 handle_help_options_next_level(cmd
, argc
, argv
);
274 fixup_argv0(argv
, cmd
->token
);
275 exit(cmd
->fn(argc
, argv
));