btrfs-progs: check: change the way lowmem mode traverses metadata

[platform/upstream/btrfs-progs.git] / cmds-rescue.c

diff --git a/cmds-rescue.c b/cmds-rescue.c
index 9895425..d1bec02 100644 (file)
--- a/cmds-rescue.c
+++ b/cmds-rescue.c
@@ -24,6 +24,7 @@
 #include "disk-io.h"
 #include "commands.h"
 #include "utils.h"
+#include "help.h"
 
 static const char * const rescue_cmd_group_usage[] = {
        "btrfs rescue <command> [options] <path>",
@@ -67,8 +68,7 @@ static int cmd_rescue_chunk_recover(int argc, char *argv[])
                }
        }
 
-       argc = argc - optind;
-       if (check_argc_exact(argc, 1))
+       if (check_argc_exact(argc - optind, 1))
                usage(cmd_rescue_chunk_recover_usage);
 
        file = argv[optind];
@@ -133,8 +133,7 @@ static int cmd_rescue_super_recover(int argc, char **argv)
                        usage(cmd_rescue_super_recover_usage);
                }
        }
-       argc = argc - optind;
-       if (check_argc_exact(argc, 1))
+       if (check_argc_exact(argc - optind, 1))
                usage(cmd_rescue_super_recover_usage);
 
        dname = argv[optind];
@@ -165,6 +164,8 @@ static int cmd_rescue_zero_log(int argc, char **argv)
        char *devname;
        int ret;
 
+       clean_args_no_options(argc, argv, cmd_rescue_zero_log_usage);
+
        if (check_argc_exact(argc, 2))
                usage(cmd_rescue_zero_log_usage);
 
@@ -190,6 +191,7 @@ static int cmd_rescue_zero_log(int argc, char **argv)
                        (unsigned long long)btrfs_super_log_root(sb),
                        (unsigned)btrfs_super_log_root_level(sb));
        trans = btrfs_start_transaction(root, 1);
+       BUG_ON(IS_ERR(trans));
        btrfs_set_super_log_root(sb, 0);
        btrfs_set_super_log_root_level(sb, 0);
        btrfs_commit_transaction(trans, root);