snapinfo.c revision 167626 
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 167626 2007-03-16 03:23:32Z 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		path = *argv;
85
86		if (strrchr(path, '/') == NULL ||	/* is absolute path */
87		    (stat(path, &st) == -1) || 		/* is it stat'able */
88		    !(st.st_mode & S_IFDIR))  		/* is it a directory */
89			usage();
90	}
91
92	fscount = getmntinfo(&mntbuf, MNT_WAIT);
93	for (n = 0; n < fscount; n++) {
94		if (!strncmp(mntbuf[n].f_fstypename, "ufs", 3)) {
95			if (all || strcmp(path, mntbuf[n].f_mntonname) == 0) {
96				find_snapshot(&mntbuf[n]);
97				done++;
98			}
99		}
100	}
101
102	if (done == 0)
103		usage();
104
105	return (0);
106}
107
108int
109find_snapshot(struct statfs *sfs)
110{
111	struct uufsd disk;
112	int j, snapcount = 0;
113
114	if (ufs_disk_fillout(&disk, sfs->f_mntfromname) == -1)
115		perror("ufs_disk_fillout");
116
117	if (verbose)
118		printf("%s mounted on %s\n", disk.d_name, disk.d_fs.fs_fsmnt);
119
120	for (j = 0; j < FSMAXSNAP; j++) {
121		if (disk.d_fs.fs_snapinum[j]) {
122			inode = disk.d_fs.fs_snapinum[j];
123			find_inum(sfs->f_mntonname);
124			snapcount++;
125		}
126	}
127
128	if (!snapcount && verbose)
129		printf("\tno snapshots found\n");
130
131	return 0;
132}
133
134int
135compare_function(const char *path, const struct stat *st, int flags,
136struct FTW * ftwv)
137{
138
139	if (flags == FTW_F && st->st_ino == inode) {
140		if (verbose)
141			printf("\tsnapshot ");
142		printf("%s", path);
143		if (verbose)
144			printf(" (inode %d)", st->st_ino);
145		printf("\n");
146		if (!cont_search)
147			return (EEXIST);
148	}
149
150	return (0);
151}
152
153void
154find_inum(char *path)
155{
156	int ret;
157
158	ret = nftw(path, compare_function, 1, FTW_PHYS|FTW_MOUNT);
159	if (ret != EEXIST && ret != 0) {
160		perror("ftw");
161		exit(ret);
162	}
163}
164
165void
166usage(void)
167{
168
169	printf("usage: snapinfo [-v] -a\n");
170	printf("       snapinfo [-v] mountpoint\n");
171	exit(1);
172}
173