fsutil.c revision 93102 
1/*	$NetBSD: fsutil.c,v 1.7 1998/07/30 17:41:03 thorpej Exp $	*/
2
3/*
4 * Copyright (c) 1990, 1993
5 *	The Regents of the University of California.  All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 *    must display the following acknowledgement:
17 *	This product includes software developed by the University of
18 *	California, Berkeley and its contributors.
19 * 4. Neither the name of the University nor the names of its contributors
20 *    may be used to endorse or promote products derived from this software
21 *    without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36#include <sys/cdefs.h>
37#ifndef lint
38__RCSID("$NetBSD: fsutil.c,v 1.7 1998/07/30 17:41:03 thorpej Exp $");
39#endif /* not lint */
40__FBSDID("$FreeBSD: head/sbin/fsck/fsutil.c 93102 2002-03-24 15:06:48Z markm $");
41
42#include <sys/param.h>
43#include <sys/stat.h>
44#include <sys/mount.h>
45
46#include <err.h>
47#include <errno.h>
48#include <fstab.h>
49#include <paths.h>
50#include <stdarg.h>
51#include <stdio.h>
52#include <stdlib.h>
53#include <string.h>
54
55#include "fsutil.h"
56
57static const char *dev = NULL;
58static int hot = 0;
59static int preen = 0;
60
61static void vmsg(int, const char *, va_list) __printflike(2, 0);
62
63void
64setcdevname(const char *cd, int pr)
65{
66	dev = cd;
67	preen = pr;
68}
69
70const char *
71cdevname(void)
72{
73	return dev;
74}
75
76int
77hotroot(void)
78{
79	return hot;
80}
81
82/*VARARGS*/
83void
84errexit(const char *fmt, ...)
85{
86	va_list ap;
87
88	va_start(ap, fmt);
89	(void) vfprintf(stderr, fmt, ap);
90	va_end(ap);
91	exit(8);
92}
93
94static void
95vmsg(int fatal, const char *fmt, va_list ap)
96{
97	if (!fatal && preen)
98		(void) printf("%s: ", dev);
99
100	(void) vprintf(fmt, ap);
101
102	if (fatal && preen)
103		(void) printf("\n");
104
105	if (fatal && preen) {
106		(void) printf(
107		    "%s: UNEXPECTED INCONSISTENCY; RUN %s MANUALLY.\n",
108		    dev, getprogname());
109		exit(8);
110	}
111}
112
113/*VARARGS*/
114void
115pfatal(const char *fmt, ...)
116{
117	va_list ap;
118
119	va_start(ap, fmt);
120	vmsg(1, fmt, ap);
121	va_end(ap);
122}
123
124/*VARARGS*/
125void
126pwarn(const char *fmt, ...)
127{
128	va_list ap;
129
130	va_start(ap, fmt);
131	vmsg(0, fmt, ap);
132	va_end(ap);
133}
134
135void
136perror(const char *s)
137{
138	pfatal("%s (%s)", s, strerror(errno));
139}
140
141void
142panic(const char *fmt, ...)
143{
144	va_list ap;
145
146	va_start(ap, fmt);
147	vmsg(1, fmt, ap);
148	va_end(ap);
149	exit(8);
150}
151
152const char *
153unrawname(const char *name)
154{
155	static char unrawbuf[32];
156	const char *dp;
157	struct stat stb;
158
159	if ((dp = strrchr(name, '/')) == 0)
160		return (name);
161	if (stat(name, &stb) < 0)
162		return (name);
163	if (!S_ISCHR(stb.st_mode))
164		return (name);
165	if (dp[1] != 'r')
166		return (name);
167	(void)snprintf(unrawbuf, 32, "%.*s/%s", (int)(dp - name), name, dp + 2);
168	return (unrawbuf);
169}
170
171const char *
172rawname(const char *name)
173{
174	static char rawbuf[32];
175	const char *dp;
176
177	if ((dp = strrchr(name, '/')) == 0)
178		return (0);
179	(void)snprintf(rawbuf, 32, "%.*s/r%s", (int)(dp - name), name, dp + 1);
180	return (rawbuf);
181}
182
183const char *
184devcheck(const char *origname)
185{
186	struct stat stslash, stchar;
187
188	if (stat("/", &stslash) < 0) {
189		perror("/");
190		printf("Can't stat root\n");
191		return (origname);
192	}
193	if (stat(origname, &stchar) < 0) {
194		perror(origname);
195		printf("Can't stat %s\n", origname);
196		return (origname);
197	}
198	if (!S_ISCHR(stchar.st_mode)) {
199		perror(origname);
200		printf("%s is not a char device\n", origname);
201	}
202	return (origname);
203}
204
205/*
206 * Get the mount point information for name.
207 */
208struct statfs *
209getmntpt(const char *name)
210{
211	struct stat devstat, mntdevstat;
212	char device[sizeof(_PATH_DEV) - 1 + MNAMELEN];
213	char *devname;
214	struct statfs *mntbuf, *statfsp;
215	int i, mntsize, isdev;
216
217	if (stat(name, &devstat) != 0)
218		return (NULL);
219	if (S_ISCHR(devstat.st_mode) || S_ISBLK(devstat.st_mode))
220		isdev = 1;
221	else
222		isdev = 0;
223	mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
224	for (i = 0; i < mntsize; i++) {
225		statfsp = &mntbuf[i];
226		devname = statfsp->f_mntfromname;
227		if (*devname != '/') {
228			strcpy(device, _PATH_DEV);
229			strcat(device, devname);
230			strcpy(statfsp->f_mntfromname, device);
231		}
232		if (isdev == 0) {
233			if (strcmp(name, statfsp->f_mntonname))
234				continue;
235			return (statfsp);
236		}
237		if (stat(devname, &mntdevstat) == 0 &&
238		    mntdevstat.st_rdev == devstat.st_rdev)
239			return (statfsp);
240	}
241	statfsp = NULL;
242	return (statfsp);
243}
244
245
246void *
247emalloc(size_t s)
248{
249	void *p;
250
251	p = malloc(s);
252	if (p == NULL)
253		err(1, "malloc failed");
254	return (p);
255}
256
257
258void *
259erealloc(void *p, size_t s)
260{
261	void *q;
262
263	q = realloc(p, s);
264	if (q == NULL)
265		err(1, "realloc failed");
266	return (q);
267}
268
269
270char *
271estrdup(const char *s)
272{
273	char *p;
274
275	p = strdup(s);
276	if (p == NULL)
277		err(1, "strdup failed");
278	return (p);
279}
280