btrfs-progs: fsck-tests: Introduce test case with keyed data backref with the extent...
[platform/upstream/btrfs-progs.git] / cmds-rescue.c
index cfbc198..c40088a 100644 (file)
 #include "kerncompat.h"
 
 #include <getopt.h>
+#include "ctree.h"
+#include "volumes.h"
+#include "transaction.h"
+#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>",
@@ -28,8 +33,9 @@ static const char * const rescue_cmd_group_usage[] = {
 };
 
 int btrfs_recover_chunk_tree(char *path, int verbose, int yes);
+int btrfs_recover_superblocks(char *path, int verbose, int yes);
 
-const char * const cmd_chunk_recover_usage[] = {
+static const char * const cmd_rescue_chunk_recover_usage[] = {
        "btrfs rescue chunk-recover [options] <device>",
        "Recover the chunk tree by scanning the devices one by one.",
        "",
@@ -39,7 +45,7 @@ const char * const cmd_chunk_recover_usage[] = {
        NULL
 };
 
-int cmd_chunk_recover(int argc, char *argv[])
+static int cmd_rescue_chunk_recover(int argc, char *argv[])
 {
        int ret = 0;
        char *file;
@@ -59,38 +65,202 @@ int cmd_chunk_recover(int argc, char *argv[])
                        break;
                case 'h':
                default:
-                       usage(cmd_chunk_recover_usage);
+                       usage(cmd_rescue_chunk_recover_usage);
                }
        }
 
-       argc = argc - optind;
-       if (argc == 0)
-               usage(cmd_chunk_recover_usage);
+       if (check_argc_exact(argc - optind, 1))
+               usage(cmd_rescue_chunk_recover_usage);
 
        file = argv[optind];
 
        ret = check_mounted(file);
-       if (ret) {
-               fprintf(stderr, "the device is busy\n");
-               return ret;
+       if (ret < 0) {
+               error("could not check mount status: %s", strerror(-ret));
+               return 1;
+       } else if (ret) {
+               error("the device is busy");
+               return 1;
        }
 
        ret = btrfs_recover_chunk_tree(file, verbose, yes);
        if (!ret) {
-               fprintf(stdout, "Recover the chunk tree successfully.\n");
+               fprintf(stdout, "Chunk tree recovered successfully\n");
        } else if (ret > 0) {
                ret = 0;
-               fprintf(stdout, "Abort to rebuild the on-disk chunk tree.\n");
+               fprintf(stdout, "Chunk tree recovery aborted\n");
        } else {
-               fprintf(stdout, "Fail to recover the chunk tree.\n");
+               fprintf(stdout, "Chunk tree recovery failed\n");
        }
        return ret;
 }
 
+static const char * const cmd_rescue_super_recover_usage[] = {
+       "btrfs rescue super-recover [options] <device>",
+       "Recover bad superblocks from good copies",
+       "",
+       "-y     Assume an answer of `yes' to all questions",
+       "-v     Verbose mode",
+       NULL
+};
+
+/*
+ * return codes:
+ *   0 : All superblocks are valid, no need to recover
+ *   1 : Usage or syntax error
+ *   2 : Recover all bad superblocks successfully
+ *   3 : Fail to Recover bad supeblocks
+ *   4 : Abort to recover bad superblocks
+ */
+static int cmd_rescue_super_recover(int argc, char **argv)
+{
+       int ret;
+       int verbose = 0;
+       int yes = 0;
+       char *dname;
+
+       while (1) {
+               int c = getopt(argc, argv, "vy");
+               if (c < 0)
+                       break;
+               switch (c) {
+               case 'v':
+                       verbose = 1;
+                       break;
+               case 'y':
+                       yes = 1;
+                       break;
+               default:
+                       usage(cmd_rescue_super_recover_usage);
+               }
+       }
+       if (check_argc_exact(argc - optind, 1))
+               usage(cmd_rescue_super_recover_usage);
+
+       dname = argv[optind];
+       ret = check_mounted(dname);
+       if (ret < 0) {
+               error("could not check mount status: %s", strerror(-ret));
+               return 1;
+       } else if (ret) {
+               error("the device is busy");
+               return 1;
+       }
+       ret = btrfs_recover_superblocks(dname, verbose, yes);
+       return ret;
+}
+
+static const char * const cmd_rescue_zero_log_usage[] = {
+       "btrfs rescue zero-log <device>",
+       "Clear the tree log. Usable if it's corrupted and prevents mount.",
+       "",
+       NULL
+};
+
+static int cmd_rescue_zero_log(int argc, char **argv)
+{
+       struct btrfs_root *root;
+       struct btrfs_trans_handle *trans;
+       struct btrfs_super_block *sb;
+       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);
+
+       devname = argv[optind];
+       ret = check_mounted(devname);
+       if (ret < 0) {
+               error("could not check mount status: %s", strerror(-ret));
+               goto out;
+       } else if (ret) {
+               error("%s is currently mounted", devname);
+               ret = -EBUSY;
+               goto out;
+       }
+
+       root = open_ctree(devname, 0, OPEN_CTREE_WRITES | OPEN_CTREE_PARTIAL);
+       if (!root) {
+               error("could not open ctree");
+               return 1;
+       }
+
+       sb = root->fs_info->super_copy;
+       printf("Clearing log on %s, previous log_root %llu, level %u\n",
+                       devname,
+                       (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);
+       close_ctree(root);
+
+out:
+       return !!ret;
+}
+
+static const char * const cmd_rescue_fix_device_size_usage[] = {
+       "btrfs rescue fix-device-size <device>",
+       "Re-align device and super block sizes. Usable if newer kernel refuse to mount it due to mismatch super size",
+       "",
+       NULL
+};
+
+static int cmd_rescue_fix_device_size(int argc, char **argv)
+{
+       struct btrfs_fs_info *fs_info;
+       char *devname;
+       int ret;
+
+       clean_args_no_options(argc, argv, cmd_rescue_fix_device_size_usage);
+
+       if (check_argc_exact(argc, 2))
+               usage(cmd_rescue_fix_device_size_usage);
+
+       devname = argv[optind];
+       ret = check_mounted(devname);
+       if (ret < 0) {
+               error("could not check mount status: %s", strerror(-ret));
+               goto out;
+       } else if (ret) {
+               error("%s is currently mounted", devname);
+               ret = -EBUSY;
+               goto out;
+       }
+
+       fs_info = open_ctree_fs_info(devname, 0, 0, 0, OPEN_CTREE_WRITES |
+                                    OPEN_CTREE_PARTIAL);
+       if (!fs_info) {
+               error("could not open btrfs");
+               ret = -EIO;
+               goto out;
+       }
+
+       ret = btrfs_fix_device_and_super_size(fs_info);
+       if (ret > 0)
+               ret = 0;
+       close_ctree(fs_info->tree_root);
+out:
+       return !!ret;
+}
+
+static const char rescue_cmd_group_info[] =
+"toolbox for specific rescue operations";
+
 const struct cmd_group rescue_cmd_group = {
-       rescue_cmd_group_usage, NULL, {
-               { "chunk-recover", cmd_chunk_recover, cmd_chunk_recover_usage, NULL, 0},
-               { 0, 0, 0, 0, 0 }
+       rescue_cmd_group_usage, rescue_cmd_group_info, {
+               { "chunk-recover", cmd_rescue_chunk_recover,
+                       cmd_rescue_chunk_recover_usage, NULL, 0},
+               { "super-recover", cmd_rescue_super_recover,
+                       cmd_rescue_super_recover_usage, NULL, 0},
+               { "zero-log", cmd_rescue_zero_log, cmd_rescue_zero_log_usage, NULL, 0},
+               { "fix-device-size", cmd_rescue_fix_device_size,
+                       cmd_rescue_fix_device_size_usage, NULL, 0},
+               NULL_CMD_STRUCT
        }
 };