1/*	$NetBSD: fsutil.c,v 1.15 2006/06/05 16:52:05 christos 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. Neither the name of the University nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32#include <sys/cdefs.h>
33#ifndef lint
34__RCSID("$NetBSD: fsutil.c,v 1.15 2006/06/05 16:52:05 christos Exp $");
35#endif /* not lint */
36__FBSDID("$FreeBSD$");
37
38#include <sys/param.h>
39#include <sys/stat.h>
40#include <sys/mount.h>
41
42#include <err.h>
43#include <errno.h>
44#include <fstab.h>
45#include <paths.h>
46#include <stdarg.h>
47#include <stdio.h>
48#include <stdlib.h>
49#include <string.h>
50
51#include "fsutil.h"
52
53static const char *dev = NULL;
54static int preen = 0;
55
56static void vmsg(int, const char *, va_list) __printflike(2, 0);
57
58void
59setcdevname(const char *cd, int pr)
60{
61	dev = cd;
62	preen = pr;
63}
64
65const char *
66cdevname(void)
67{
68	return dev;
69}
70
71static void
72vmsg(int fatal, const char *fmt, va_list ap)
73{
74	if (!fatal && preen)
75		(void) printf("%s: ", dev);
76
77	(void) vprintf(fmt, ap);
78
79	if (fatal && preen)
80		(void) printf("\n");
81
82	if (fatal && preen) {
83		(void) printf(
84		    "%s: UNEXPECTED INCONSISTENCY; RUN %s MANUALLY.\n",
85		    dev, getprogname());
86		exit(8);
87	}
88}
89
90/*VARARGS*/
91void
92pfatal(const char *fmt, ...)
93{
94	va_list ap;
95
96	va_start(ap, fmt);
97	vmsg(1, fmt, ap);
98	va_end(ap);
99}
100
101/*VARARGS*/
102void
103pwarn(const char *fmt, ...)
104{
105	va_list ap;
106
107	va_start(ap, fmt);
108	vmsg(0, fmt, ap);
109	va_end(ap);
110}
111
112void
113perr(const char *fmt, ...)
114{
115	va_list ap;
116
117	va_start(ap, fmt);
118	vmsg(1, fmt, ap);
119	va_end(ap);
120}
121
122void
123panic(const char *fmt, ...)
124{
125	va_list ap;
126
127	va_start(ap, fmt);
128	vmsg(1, fmt, ap);
129	va_end(ap);
130	exit(8);
131}
132
133const char *
134devcheck(const char *origname)
135{
136	struct stat stslash, stchar;
137
138	if (stat("/", &stslash) < 0) {
139		perr("Can't stat `/'");
140		return (origname);
141	}
142	if (stat(origname, &stchar) < 0) {
143		perr("Can't stat %s\n", origname);
144		return (origname);
145	}
146	if (!S_ISCHR(stchar.st_mode)) {
147		perr("%s is not a char device\n", origname);
148	}
149	return (origname);
150}
151
152/*
153 * Get the mount point information for name.
154 */
155struct statfs *
156getmntpt(const char *name)
157{
158	struct stat devstat, mntdevstat;
159	char device[sizeof(_PATH_DEV) - 1 + MNAMELEN];
160	char *dev_name;
161	struct statfs *mntbuf, *statfsp;
162	int i, mntsize, isdev;
163
164	if (stat(name, &devstat) != 0)
165		return (NULL);
166	if (S_ISCHR(devstat.st_mode) || S_ISBLK(devstat.st_mode))
167		isdev = 1;
168	else
169		isdev = 0;
170	mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
171	for (i = 0; i < mntsize; i++) {
172		statfsp = &mntbuf[i];
173		dev_name = statfsp->f_mntfromname;
174		if (*dev_name != '/') {
175			strcpy(device, _PATH_DEV);
176			strcat(device, dev_name);
177			strcpy(statfsp->f_mntfromname, device);
178		}
179		if (isdev == 0) {
180			if (strcmp(name, statfsp->f_mntonname))
181				continue;
182			return (statfsp);
183		}
184		if (stat(dev_name, &mntdevstat) == 0 &&
185		    mntdevstat.st_rdev == devstat.st_rdev)
186			return (statfsp);
187	}
188	statfsp = NULL;
189	return (statfsp);
190}
191
192
193void *
194emalloc(size_t s)
195{
196	void *p;
197
198	p = malloc(s);
199	if (p == NULL)
200		err(1, "malloc failed");
201	return (p);
202}
203
204
205void *
206erealloc(void *p, size_t s)
207{
208	void *q;
209
210	q = realloc(p, s);
211	if (q == NULL)
212		err(1, "realloc failed");
213	return (q);
214}
215
216
217char *
218estrdup(const char *s)
219{
220	char *p;
221
222	p = strdup(s);
223	if (p == NULL)
224		err(1, "strdup failed");
225	return (p);
226}
227