Tidy up util
[openafs.git] / src / util / fstab.c
1 /*
2  * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
3  *
4  * @APPLE_LICENSE_HEADER_START@
5  * 
6  * This file contains Original Code and/or Modifications of Original Code
7  * as defined in and that are subject to the Apple Public Source License
8  * Version 2.0 (the 'License'). You may not use this file except in
9  * compliance with the License. Please obtain a copy of the License at
10  * http://www.opensource.apple.com/apsl/ and read it before using this
11  * file.
12  * 
13  * The Original Code and all software distributed under the License are
14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
18  * Please see the License for the specific language governing rights and
19  * limitations under the License.
20  * 
21  * @APPLE_LICENSE_HEADER_END@
22  */
23 /*
24  * Copyright (c) 1980, 1988, 1993
25  *      The Regents of the University of California.  All rights reserved.
26  *
27  * Redistribution and use in source and binary forms, with or without
28  * modification, are permitted provided that the following conditions
29  * are met:
30  * 1. Redistributions of source code must retain the above copyright
31  *    notice, this list of conditions and the following disclaimer.
32  * 2. Redistributions in binary form must reproduce the above copyright
33  *    notice, this list of conditions and the following disclaimer in the
34  *    documentation and/or other materials provided with the distribution.
35  * 3. All advertising materials mentioning features or use of this software
36  *    must display the following acknowledgement:
37  *      This product includes software developed by the University of
38  *      California, Berkeley and its contributors.
39  * 4. Neither the name of the University nor the names of its contributors
40  *    may be used to endorse or promote products derived from this software
41  *    without specific prior written permission.
42  *
43  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
44  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
45  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
46  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
47  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
48  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
49  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
50  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
51  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
52  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
53  * SUCH DAMAGE.
54  */
55
56 #include <afs/param.h>
57
58 #if defined(AFS_DARWIN_ENV)
59 /*-----------------------------------------------------------------------
60  * This version of fstab.c is intended to be used on Darwin systems to
61  * replace getfsent() and family.  It has been modified so that rather
62  * than read /etc/fstab, it calls getfsstat() to get the real list of
63  * mounted volumes.
64  *-----------------------------------------------------------------------*/
65
66 #include <errno.h>
67 #include <fstab.h>
68 #include <stdio.h>
69 #include <string.h>
70 #include <stdlib.h>
71 #include <sys/param.h>
72 #include <sys/ucred.h>
73 #include <sys/mount.h>
74 #include <unistd.h>
75
76 #define STDERR_FILENO   2
77
78 static struct fstab _fs_fstab;
79 static struct statfs *_fs_buf;
80 static struct statfs *_fs_ptr;
81 static int _fs_count;
82
83 static void error(int);
84 static int fstabscan(void);
85
86 static int
87 fstabscan(void)
88 {
89     if (_fs_count <= 0)
90         return (0);
91     _fs_fstab.fs_spec = _fs_ptr->f_mntfromname;
92     _fs_fstab.fs_file = _fs_ptr->f_mntonname;
93     _fs_fstab.fs_vfstype = _fs_ptr->f_fstypename;
94     _fs_fstab.fs_mntops = _fs_ptr->f_fstypename;        // no mount options given
95     _fs_fstab.fs_type = (_fs_ptr->f_flags & MNT_RDONLY) ? FSTAB_RO : FSTAB_RW;
96     _fs_fstab.fs_freq = 0;
97     _fs_fstab.fs_passno = 0;
98
99     _fs_ptr++;
100     _fs_count--;
101     return (1);
102 }
103
104 struct fstab *
105 getfsent(void)
106 {
107     if (!_fs_buf && !setfsent() || !fstabscan())
108         return ((struct fstab *)NULL);
109     return (&_fs_fstab);
110 }
111
112 struct fstab *
113 getfsspec(const char *name)
114 {
115     if (setfsent())
116         while (fstabscan())
117             if (!strcmp(_fs_fstab.fs_spec, name))
118                 return (&_fs_fstab);
119     return ((struct fstab *)NULL);
120 }
121
122 struct fstab *
123 getfsfile(const char *name)
124 {
125     if (setfsent())
126         while (fstabscan())
127             if (!strcmp(_fs_fstab.fs_file, name))
128                 return (&_fs_fstab);
129     return ((struct fstab *)NULL);
130 }
131
132 int
133 setfsent(void)
134 {
135     long bufsize;
136
137     if (_fs_buf) {
138         free(_fs_buf);
139         _fs_buf = NULL;
140     }
141     if ((_fs_count = getfsstat(NULL, 0, MNT_WAIT)) < 0) {
142         error(errno);
143         return (0);
144     }
145     bufsize = (long)_fs_count *sizeof(struct statfs);
146     if ((_fs_buf = malloc(bufsize)) == NULL) {
147         error(errno);
148         return (0);
149     }
150     if (getfsstat(_fs_buf, bufsize, MNT_WAIT) < 0) {
151         error(errno);
152         return (0);
153     }
154     _fs_ptr = _fs_buf;
155     return (1);
156 }
157
158 void
159 endfsent(void)
160 {
161     if (_fs_buf) {
162         free(_fs_buf);
163         _fs_buf = NULL;
164     }
165     _fs_count = 0;
166 }
167
168 static void
169 error(int err)
170 {
171     char *p;
172
173     (void)write(STDERR_FILENO, "fstab: ", 7);
174     (void)write(STDERR_FILENO, _PATH_FSTAB, sizeof(_PATH_FSTAB) - 1);
175     (void)write(STDERR_FILENO, ": ", 1);
176     p = strerror(err);
177     (void)write(STDERR_FILENO, p, strlen(p));
178     (void)write(STDERR_FILENO, "\n", 1);
179 }
180 #endif /* defined(AFS_DARWIN_ENV) */