snapinfo.c revision 167633
1/*-
2 * Copyright (c) 2005 Mark Santcroos <marks@freebsd.org>
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 *    notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 *    notice, this list of conditions and the following disclaimer in the
11 *    documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
14 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 *
24 * $FreeBSD: head/usr.sbin/snapinfo/snapinfo.c 167633 2007-03-16 12:36:54Z pjd $
25 *
26 */
27
28#include <sys/param.h>
29#include <sys/mount.h>
30
31#include <ufs/ufs/dinode.h>
32#include <ufs/ffs/fs.h>
33
34#include <errno.h>
35#include <ftw.h>
36#include <libufs.h>
37#include <stdio.h>
38#include <stdlib.h>
39#include <string.h>
40#include <unistd.h>
41
42void	find_inum(char *path);
43void	usage(void);
44int	compare_function(const char *, const struct stat *, int, struct FTW *);
45int	find_snapshot(struct statfs *sfs);
46
47int verbose;
48int cont_search;
49uint32_t inode;
50
51int
52main(int argc, char **argv)
53{
54	char *path;
55	struct stat st;
56	struct statfs *mntbuf;
57	int all = 0, ch, done = 0, fscount, n;
58
59	while ((ch = getopt(argc, argv, "adv")) != -1) {
60		switch (ch) {
61		case 'a':
62			all++;
63			break;
64		case 'd':
65			/* continue to search when matching inode is found
66			 * this feature is not documented */
67			cont_search++;
68			break;
69		case 'v':
70			verbose++;
71			break;
72		default:
73			usage();
74		}
75	}
76
77	argc -= optind;
78	argv += optind;
79
80	if ((all == 0 && argc != 1) || (all == 1 && argc > 0))
81		usage();
82
83	if (!all) {
84		char resolved[PATH_MAX];
85
86		path = *argv;
87		/*
88		 * mount(8) use realpath(3) before mounting file system,
89		 * so let's do the same with the given path.
90		 */
91		if (realpath(path, resolved) == NULL ||	/* can create full path */
92		    stat(resolved, &st) == -1 ||	/* is it stat'able */
93		    !S_ISDIR(st.st_mode)) {		/* is it a directory */
94			usage();
95		}
96		path = resolved;
97	}
98
99	fscount = getmntinfo(&mntbuf, MNT_WAIT);
100	for (n = 0; n < fscount; n++) {
101		if (!strncmp(mntbuf[n].f_fstypename, "ufs", 3)) {
102			if (all || strcmp(path, mntbuf[n].f_mntonname) == 0) {
103				find_snapshot(&mntbuf[n]);
104				done++;
105			}
106		}
107	}
108
109	if (done == 0)
110		usage();
111
112	return (0);
113}
114
115int
116find_snapshot(struct statfs *sfs)
117{
118	struct uufsd disk;
119	int j, snapcount = 0;
120
121	if (ufs_disk_fillout(&disk, sfs->f_mntfromname) == -1)
122		perror("ufs_disk_fillout");
123
124	if (verbose)
125		printf("%s mounted on %s\n", disk.d_name, disk.d_fs.fs_fsmnt);
126
127	for (j = 0; j < FSMAXSNAP; j++) {
128		if (disk.d_fs.fs_snapinum[j]) {
129			inode = disk.d_fs.fs_snapinum[j];
130			find_inum(sfs->f_mntonname);
131			snapcount++;
132		}
133	}
134
135	if (!snapcount && verbose)
136		printf("\tno snapshots found\n");
137
138	return 0;
139}
140
141int
142compare_function(const char *path, const struct stat *st, int flags,
143struct FTW * ftwv)
144{
145
146	if (flags == FTW_F && st->st_ino == inode) {
147		if (verbose)
148			printf("\tsnapshot ");
149		printf("%s", path);
150		if (verbose)
151			printf(" (inode %d)", st->st_ino);
152		printf("\n");
153		if (!cont_search)
154			return (EEXIST);
155	}
156
157	return (0);
158}
159
160void
161find_inum(char *path)
162{
163	int ret;
164
165	ret = nftw(path, compare_function, 1, FTW_PHYS|FTW_MOUNT);
166	if (ret != EEXIST && ret != 0) {
167		perror("ftw");
168		exit(ret);
169	}
170}
171
172void
173usage(void)
174{
175
176	printf("usage: snapinfo [-v] -a\n");
177	printf("       snapinfo [-v] mountpoint\n");
178	exit(1);
179}
180