Btrfs-progs: fix closing of opendir()
[platform/upstream/btrfs-progs.git] / cmds-inspect.c
1 /*
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.
5  *
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.
10  *
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.
15  */
16
17 #include <stdio.h>
18 #include <stdlib.h>
19 #include <unistd.h>
20 #include <stdint.h>
21 #include <sys/ioctl.h>
22 #include <errno.h>
23
24 #include "kerncompat.h"
25 #include "ioctl.h"
26 #include "utils.h"
27 #include "ctree.h"
28 #include "send-utils.h"
29
30 #include "commands.h"
31 #include "btrfs-list.h"
32
33 static const char * const inspect_cmd_group_usage[] = {
34         "btrfs inspect-internal <command> <args>",
35         NULL
36 };
37
38 static int __ino_to_path_fd(u64 inum, int fd, int verbose, const char *prepend)
39 {
40         int ret;
41         int i;
42         struct btrfs_ioctl_ino_path_args ipa;
43         struct btrfs_data_container *fspath;
44
45         fspath = malloc(4096);
46         if (!fspath)
47                 return 1;
48
49         memset(fspath, 0, sizeof(*fspath));
50         ipa.inum = inum;
51         ipa.size = 4096;
52         ipa.fspath = (uintptr_t)fspath;
53
54         ret = ioctl(fd, BTRFS_IOC_INO_PATHS, &ipa);
55         if (ret) {
56                 printf("ioctl ret=%d, error: %s\n", ret, strerror(errno));
57                 goto out;
58         }
59
60         if (verbose)
61                 printf("ioctl ret=%d, bytes_left=%lu, bytes_missing=%lu, "
62                         "cnt=%d, missed=%d\n", ret,
63                         (unsigned long)fspath->bytes_left,
64                         (unsigned long)fspath->bytes_missing,
65                         fspath->elem_cnt, fspath->elem_missed);
66
67         for (i = 0; i < fspath->elem_cnt; ++i) {
68                 u64 ptr;
69                 char *str;
70                 ptr = (u64)(unsigned long)fspath->val;
71                 ptr += fspath->val[i];
72                 str = (char *)(unsigned long)ptr;
73                 if (prepend)
74                         printf("%s/%s\n", prepend, str);
75                 else
76                         printf("%s\n", str);
77         }
78
79 out:
80         free(fspath);
81         return ret;
82 }
83
84 static const char * const cmd_inode_resolve_usage[] = {
85         "btrfs inspect-internal inode-resolve [-v] <inode> <path>",
86         "Get file system paths for the given inode",
87         NULL
88 };
89
90 static int cmd_inode_resolve(int argc, char **argv)
91 {
92         int fd;
93         int verbose = 0;
94         int ret;
95         DIR *dirstream = NULL;
96
97         optind = 1;
98         while (1) {
99                 int c = getopt(argc, argv, "v");
100                 if (c < 0)
101                         break;
102
103                 switch (c) {
104                 case 'v':
105                         verbose = 1;
106                         break;
107                 default:
108                         usage(cmd_inode_resolve_usage);
109                 }
110         }
111
112         if (check_argc_exact(argc - optind, 2))
113                 usage(cmd_inode_resolve_usage);
114
115         fd = open_file_or_dir(argv[optind+1], &dirstream);
116         if (fd < 0) {
117                 fprintf(stderr, "ERROR: can't access '%s'\n", argv[optind+1]);
118                 return 12;
119         }
120
121         ret = __ino_to_path_fd(atoll(argv[optind]), fd, verbose,
122                                argv[optind+1]);
123         close_file_or_dir(fd, dirstream);
124         return ret;
125
126 }
127
128 static const char * const cmd_logical_resolve_usage[] = {
129         "btrfs inspect-internal logical-resolve [-Pv] [-s bufsize] <logical> <path>",
130         "Get file system paths for the given logical address",
131         "-P          skip the path resolving and print the inodes instead",
132         "-v          verbose mode",
133         "-s bufsize  set inode container's size. This is used to increase inode",
134         "            container's size in case it is not enough to read all the ",
135         "            resolved results. The max value one can set is 64k",
136         NULL
137 };
138
139 static int cmd_logical_resolve(int argc, char **argv)
140 {
141         int ret;
142         int fd;
143         int i;
144         int verbose = 0;
145         int getpath = 1;
146         int bytes_left;
147         struct btrfs_ioctl_logical_ino_args loi;
148         struct btrfs_data_container *inodes;
149         u64 size = 4096;
150         char full_path[4096];
151         char *path_ptr;
152         DIR *dirstream = NULL;
153
154         optind = 1;
155         while (1) {
156                 int c = getopt(argc, argv, "Pvs:");
157                 if (c < 0)
158                         break;
159
160                 switch (c) {
161                 case 'P':
162                         getpath = 0;
163                         break;
164                 case 'v':
165                         verbose = 1;
166                         break;
167                 case 's':
168                         size = atoll(optarg);
169                         break;
170                 default:
171                         usage(cmd_logical_resolve_usage);
172                 }
173         }
174
175         if (check_argc_exact(argc - optind, 2))
176                 usage(cmd_logical_resolve_usage);
177
178         size = min(size, (u64)64 * 1024);
179         inodes = malloc(size);
180         if (!inodes)
181                 return 1;
182
183         memset(inodes, 0, sizeof(*inodes));
184         loi.logical = atoll(argv[optind]);
185         loi.size = size;
186         loi.inodes = (uintptr_t)inodes;
187
188         fd = open_file_or_dir(argv[optind+1], &dirstream);
189         if (fd < 0) {
190                 fprintf(stderr, "ERROR: can't access '%s'\n", argv[optind+1]);
191                 ret = 12;
192                 goto out;
193         }
194
195         ret = ioctl(fd, BTRFS_IOC_LOGICAL_INO, &loi);
196         if (ret) {
197                 printf("ioctl ret=%d, error: %s\n", ret, strerror(errno));
198                 goto out;
199         }
200
201         if (verbose)
202                 printf("ioctl ret=%d, total_size=%llu, bytes_left=%lu, "
203                         "bytes_missing=%lu, cnt=%d, missed=%d\n",
204                         ret, size,
205                         (unsigned long)inodes->bytes_left,
206                         (unsigned long)inodes->bytes_missing,
207                         inodes->elem_cnt, inodes->elem_missed);
208
209         bytes_left = sizeof(full_path);
210         ret = snprintf(full_path, bytes_left, "%s/", argv[optind+1]);
211         path_ptr = full_path + ret;
212         bytes_left -= ret + 1;
213         BUG_ON(bytes_left < 0);
214
215         for (i = 0; i < inodes->elem_cnt; i += 3) {
216                 u64 inum = inodes->val[i];
217                 u64 offset = inodes->val[i+1];
218                 u64 root = inodes->val[i+2];
219                 int path_fd;
220                 char *name;
221                 DIR *dirs = NULL;
222
223                 if (getpath) {
224                         name = btrfs_list_path_for_root(fd, root);
225                         if (IS_ERR(name)) {
226                                 ret = PTR_ERR(name);
227                                 goto out;
228                         }
229                         if (!name) {
230                                 path_ptr[-1] = '\0';
231                                 path_fd = fd;
232                         } else {
233                                 path_ptr[-1] = '/';
234                                 ret = snprintf(path_ptr, bytes_left, "%s",
235                                                 name);
236                                 BUG_ON(ret >= bytes_left);
237                                 free(name);
238                                 path_fd = open_file_or_dir(full_path, &dirs);
239                                 if (path_fd < 0) {
240                                         fprintf(stderr, "ERROR: can't access "
241                                                 "'%s'\n", full_path);
242                                         goto out;
243                                 }
244                         }
245                         __ino_to_path_fd(inum, path_fd, verbose, full_path);
246                         if (path_fd != fd)
247                                 close_file_or_dir(path_fd, dirs);
248                 } else {
249                         printf("inode %llu offset %llu root %llu\n", inum,
250                                 offset, root);
251                 }
252         }
253
254 out:
255         close_file_or_dir(fd, dirstream);
256         free(inodes);
257         return ret;
258 }
259
260 static const char * const cmd_subvolid_resolve_usage[] = {
261         "btrfs inspect-internal subvolid-resolve <subvolid> <path>",
262         "Get file system paths for the given subvolume ID.",
263         NULL
264 };
265
266 static int cmd_subvolid_resolve(int argc, char **argv)
267 {
268         int ret;
269         int fd = -1;
270         u64 subvol_id;
271         char path[BTRFS_PATH_NAME_MAX + 1];
272         DIR *dirstream = NULL;
273
274         if (check_argc_exact(argc, 3))
275                 usage(cmd_subvolid_resolve_usage);
276
277         fd = open_file_or_dir(argv[2], &dirstream);
278         if (fd < 0) {
279                 fprintf(stderr, "ERROR: can't access '%s'\n", argv[2]);
280                 ret = -ENOENT;
281                 goto out;
282         }
283
284         subvol_id = atoll(argv[1]);
285         ret = btrfs_subvolid_resolve(fd, path, sizeof(path), subvol_id);
286
287         if (ret) {
288                 fprintf(stderr,
289                         "%s: btrfs_subvolid_resolve(subvol_id %llu) failed with ret=%d\n",
290                         argv[0], (unsigned long long)subvol_id, ret);
291                 goto out;
292         }
293
294         path[BTRFS_PATH_NAME_MAX] = '\0';
295         printf("%s\n", path);
296
297 out:
298         close_file_or_dir(fd, dirstream);
299         return ret ? 1 : 0;
300 }
301
302 const struct cmd_group inspect_cmd_group = {
303         inspect_cmd_group_usage, NULL, {
304                 { "inode-resolve", cmd_inode_resolve, cmd_inode_resolve_usage,
305                         NULL, 0 },
306                 { "logical-resolve", cmd_logical_resolve,
307                         cmd_logical_resolve_usage, NULL, 0 },
308                 { "subvolid-resolve", cmd_subvolid_resolve,
309                         cmd_subvolid_resolve_usage, NULL, 0 },
310                 { 0, 0, 0, 0, 0 }
311         }
312 };
313
314 int cmd_inspect(int argc, char **argv)
315 {
316         return handle_command_group(&inspect_cmd_group, argc, argv);
317 }