btrfs-progs: catch memory allocation failure from alloc_tree_backref
[platform/upstream/btrfs-progs.git] / btrfs-zero-log.c
1 /*
2  * Copyright (C) 2007 Oracle.  All rights reserved.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public
6  * License v2 as published by the Free Software Foundation.
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public
14  * License along with this program; if not, write to the
15  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16  * Boston, MA 021110-1307, USA.
17  */
18
19 #include "kerncompat.h"
20
21 #include <stdio.h>
22 #include <unistd.h>
23 #include "ctree.h"
24 #include "disk-io.h"
25 #include "transaction.h"
26 #include "utils.h"
27
28 __attribute__((noreturn)) static void print_usage(void)
29 {
30         fprintf(stderr, "usage: btrfs-zero-log dev\n");
31         exit(1);
32 }
33
34 int main(int ac, char **av)
35 {
36         struct btrfs_root *root;
37         struct btrfs_trans_handle *trans;
38         struct btrfs_super_block *sb;
39         int ret;
40
41         set_argv0(av);
42         if (check_argc_exact(ac, 2))
43                 print_usage();
44
45         radix_tree_init();
46
47         printf("WARNING: this utility is deprecated, please use 'btrfs rescue zero-log'\n\n");
48
49         if ((ret = check_mounted(av[1])) < 0) {
50                 fprintf(stderr, "ERROR: could not check mount status: %s\n", strerror(-ret));
51                 goto out;
52         } else if (ret) {
53                 fprintf(stderr, "ERROR: %s is currently mounted\n", av[1]);
54                 ret = -EBUSY;
55                 goto out;
56         }
57
58         root = open_ctree(av[1], 0, OPEN_CTREE_WRITES | OPEN_CTREE_PARTIAL);
59         if (!root) {
60                 fprintf(stderr, "ERROR: cannot open ctree\n");
61                 return 1;
62         }
63
64         sb = root->fs_info->super_copy;
65         printf("Clearing log on %s, previous log_root %llu, level %u\n",
66                         av[1],
67                         (unsigned long long)btrfs_super_log_root(sb),
68                         (unsigned)btrfs_super_log_root_level(sb));
69         trans = btrfs_start_transaction(root, 1);
70         btrfs_set_super_log_root(root->fs_info->super_copy, 0);
71         btrfs_set_super_log_root_level(root->fs_info->super_copy, 0);
72         btrfs_commit_transaction(trans, root);
73         close_ctree(root);
74 out:
75         return !!ret;
76 }