Fix segfault on `mount -t nfs' reported by Gratien D'haese.
[oweals/busybox.git] / mount.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini mount implementation for busybox
4  *
5  * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20  *
21  * 3/21/1999    Charles P. Wright <cpwright@cpwright.com>
22  *              searches through fstab when -a is passed
23  *              will try mounting stuff with all fses when passed -t auto
24  *
25  * 1999-04-17   Dave Cinege...Rewrote -t auto. Fixed ro mtab.
26  *
27  * 1999-10-07   Erik Andersen <andersen@lineo.com>, <andersee@debian.org>.
28  *              Rewrite of a lot of code. Removed mtab usage (I plan on
29  *              putting it back as a compile-time option some time), 
30  *              major adjustments to option parsing, and some serious 
31  *              dieting all around.
32  *
33  * 1999-11-06   mtab suppport is back - andersee
34  *
35  * 2000-01-12   Ben Collins <bcollins@debian.org>, Borrowed utils-linux's
36  *              mount to add loop support.
37  *
38  * 2000-04-30   Dave Cinege <dcinege@psychosis.com>
39  *              Rewrote fstab while loop and lower mount section. Can now do
40  *              single mounts from fstab. Can override fstab options for single
41  *              mount. Common mount_one call for single mounts and 'all'. Fixed
42  *              mtab updating and stale entries. Removed 'remount' default. 
43  *      
44  */
45
46 #include <stdlib.h>
47 #include <unistd.h>
48 #include <errno.h>
49 #include <string.h>
50 #include <stdio.h>
51 #include <mntent.h>
52 #include <ctype.h>
53 #if defined BB_FEATURE_USE_DEVPS_PATCH
54 #include <linux/devmtab.h> /* For Erik's nifty devmtab device driver */
55 #endif
56 #include "busybox.h"
57
58 enum {
59         MS_MGC_VAL = 0xc0ed0000, /* Magic number indicatng "new" flags */
60         MS_RDONLY = 1,      /* Mount read-only */
61         MS_NOSUID = 2,      /* Ignore suid and sgid bits */
62         MS_NODEV = 4,      /* Disallow access to device special files */
63         MS_NOEXEC = 8,      /* Disallow program execution */
64         MS_SYNCHRONOUS = 16,      /* Writes are synced at once */
65         MS_REMOUNT = 32,      /* Alter flags of a mounted FS */
66         MS_MANDLOCK = 64,      /* Allow mandatory locks on an FS */
67         S_QUOTA = 128,     /* Quota initialized for file/directory/symlink */
68         S_APPEND = 256,     /* Append-only file */
69         S_IMMUTABLE = 512,     /* Immutable file */
70         MS_NOATIME = 1024,    /* Do not update access times. */
71         MS_NODIRATIME = 2048,    /* Do not update directory access times */
72 };
73
74
75 #if defined BB_FEATURE_MOUNT_LOOP
76 #include <fcntl.h>
77 #include <sys/ioctl.h>
78 static int use_loop = FALSE;
79 #endif
80
81 extern int mount (__const char *__special_file, __const char *__dir,
82                         __const char *__fstype, unsigned long int __rwflag,
83                         __const void *__data);
84 extern int umount (__const char *__special_file);
85 extern int umount2 (__const char *__special_file, int __flags);
86
87 extern int sysfs( int option, unsigned int fs_index, char * buf);
88
89 extern const char mtab_file[];  /* Defined in utility.c */
90
91 struct mount_options {
92         const char *name;
93         unsigned long and;
94         unsigned long or;
95 };
96
97 static const struct mount_options mount_options[] = {
98         {"async", ~MS_SYNCHRONOUS, 0},
99         {"atime", ~0, ~MS_NOATIME},
100         {"defaults", ~0, 0},
101         {"dev", ~MS_NODEV, 0},
102         {"diratime", ~0, ~MS_NODIRATIME},
103         {"exec", ~MS_NOEXEC, 0},
104         {"noatime", ~0, MS_NOATIME},
105         {"nodev", ~0, MS_NODEV},
106         {"nodiratime", ~0, MS_NODIRATIME},
107         {"noexec", ~0, MS_NOEXEC},
108         {"nosuid", ~0, MS_NOSUID},
109         {"remount", ~0, MS_REMOUNT},
110         {"ro", ~0, MS_RDONLY},
111         {"rw", ~MS_RDONLY, 0},
112         {"suid", ~MS_NOSUID, 0},
113         {"sync", ~0, MS_SYNCHRONOUS},
114         {0, 0, 0}
115 };
116
117 static int
118 do_mount(char *specialfile, char *dir, char *filesystemtype,
119                  long flags, void *string_flags, int useMtab, int fakeIt,
120                  char *mtab_opts)
121 {
122         int status = 0;
123 #if defined BB_FEATURE_MOUNT_LOOP
124         char *lofile = NULL;
125 #endif
126
127         if (fakeIt == FALSE)
128         {
129 #if defined BB_FEATURE_MOUNT_LOOP
130                 if (use_loop==TRUE) {
131                         int loro = flags & MS_RDONLY;
132                         
133                         lofile = specialfile;
134
135                         specialfile = find_unused_loop_device();
136                         if (specialfile == NULL) {
137                                 error_msg_and_die("Could not find a spare loop device");
138                         }
139                         if (set_loop(specialfile, lofile, 0, &loro)) {
140                                 error_msg_and_die("Could not setup loop device");
141                         }
142                         if (!(flags & MS_RDONLY) && loro) {     /* loop is ro, but wanted rw */
143                                 error_msg("WARNING: loop device is read-only");
144                                 flags &= ~MS_RDONLY;
145                         }
146                 }
147 #endif
148                 status = mount(specialfile, dir, filesystemtype, flags, string_flags);
149                 if (errno == EROFS) {
150                         error_msg("%s is write-protected, mounting read-only", specialfile);
151                         status = mount(specialfile, dir, filesystemtype, flags |= MS_RDONLY, string_flags);
152                 }
153         }
154
155
156         /* If the mount was sucessful, do anything needed, then return TRUE */
157         if (status == 0 || fakeIt==TRUE) {
158
159 #if defined BB_FEATURE_MTAB_SUPPORT
160                 if (useMtab == TRUE) {
161                         erase_mtab(specialfile);        // Clean any stale entries
162                         write_mtab(specialfile, dir, filesystemtype, flags, mtab_opts);
163                 }
164 #endif
165                 return (TRUE);
166         }
167
168         /* Bummer.  mount failed.  Clean up */
169 #if defined BB_FEATURE_MOUNT_LOOP
170         if (lofile != NULL) {
171                 del_loop(specialfile);
172         }
173 #endif
174
175         if (errno == EPERM) {
176                 error_msg_and_die("permission denied. Are you root?");
177         }
178
179         return (FALSE);
180 }
181
182
183
184 /* Seperate standard mount options from the nonstandard string options */
185 static void
186 parse_mount_options(char *options, int *flags, char *strflags)
187 {
188         while (options) {
189                 int gotone = FALSE;
190                 char *comma = strchr(options, ',');
191                 const struct mount_options *f = mount_options;
192
193                 if (comma)
194                         *comma = '\0';
195
196                 while (f->name != 0) {
197                         if (strcasecmp(f->name, options) == 0) {
198
199                                 *flags &= f->and;
200                                 *flags |= f->or;
201                                 gotone = TRUE;
202                                 break;
203                         }
204                         f++;
205                 }
206 #if defined BB_FEATURE_MOUNT_LOOP
207                 if (gotone == FALSE && !strcasecmp("loop", options)) {  /* loop device support */
208                         use_loop = TRUE;
209                         gotone = TRUE;
210                 }
211 #endif
212                 if (*strflags && strflags != '\0' && gotone == FALSE) {
213                         char *temp = strflags;
214
215                         temp += strlen(strflags);
216                         *temp++ = ',';
217                         *temp++ = '\0';
218                 }
219                 if (gotone == FALSE)
220                         strcat(strflags, options);
221                 if (comma) {
222                         *comma = ',';
223                         options = ++comma;
224                 } else {
225                         break;
226                 }
227         }
228 }
229
230 extern int
231 mount_one(char *blockDevice, char *directory, char *filesystemType,
232                   unsigned long flags, char *string_flags, int useMtab, int fakeIt,
233                   char *mtab_opts, int whineOnErrors)
234 {
235         int status = 0;
236
237         if (strcmp(filesystemType, "auto") == 0) {
238                 static const char *noauto_array[] = { "tmpfs", "shm", "proc", "ramfs", "devpts", "devfs", 0 };
239                 const char **noauto_fstype;
240                 const int num_of_filesystems = sysfs(3, 0, 0);
241                 char buf[255];
242                 int i=0;
243
244                 filesystemType=buf;
245
246                 while(i < num_of_filesystems) {
247                         sysfs(2, i++, filesystemType);
248                         for (noauto_fstype = noauto_array; *noauto_fstype; noauto_fstype++) {
249                                 if (!strcmp(filesystemType, *noauto_fstype)) {
250                                         break;
251                                 }
252                         }
253                         if (!*noauto_fstype) {
254                                 status = do_mount(blockDevice, directory, filesystemType,
255                                         flags | MS_MGC_VAL, string_flags,
256                                         useMtab, fakeIt, mtab_opts);
257                                 if (status == TRUE)
258                                         break;
259                         }
260                 }
261         } else {
262                 status = do_mount(blockDevice, directory, filesystemType,
263                                 flags | MS_MGC_VAL, string_flags, useMtab,
264                                 fakeIt, mtab_opts);
265         }
266
267         if (status == FALSE) {
268                 if (whineOnErrors == TRUE) {
269                         perror_msg("Mounting %s on %s failed", blockDevice, directory);
270                 }
271                 return (FALSE);
272         }
273         return (TRUE);
274 }
275
276 extern int mount_main(int argc, char **argv)
277 {
278         char string_flags_buf[1024] = "";
279         char *string_flags = string_flags_buf;
280         char *extra_opts = string_flags_buf;
281         int flags = 0;
282         char *filesystemType = "auto";
283         char *device = NULL;
284         char *directory = NULL;
285         int all = FALSE;
286         int fakeIt = FALSE;
287         int useMtab = TRUE;
288         int i;
289         int rc = EXIT_FAILURE;
290         int fstabmount = FALSE; 
291
292 #if defined BB_FEATURE_USE_DEVPS_PATCH
293         if (argc == 1) {
294                 int fd, i, numfilesystems;
295                 char device[] = "/dev/mtab";
296                 struct k_mntent *mntentlist;
297
298                 /* open device */ 
299                 fd = open(device, O_RDONLY);
300                 if (fd < 0)
301                         perror_msg_and_die("open failed for `%s'", device);
302
303                 /* How many mounted filesystems?  We need to know to 
304                  * allocate enough space for later... */
305                 numfilesystems = ioctl (fd, DEVMTAB_COUNT_MOUNTS);
306                 if (numfilesystems<0)
307                         perror_msg_and_die( "\nDEVMTAB_COUNT_MOUNTS");
308                 mntentlist = (struct k_mntent *) xcalloc ( numfilesystems, sizeof(struct k_mntent));
309                 
310                 /* Grab the list of mounted filesystems */
311                 if (ioctl (fd, DEVMTAB_GET_MOUNTS, mntentlist)<0)
312                         perror_msg_and_die( "\nDEVMTAB_GET_MOUNTS");
313
314                 for( i = 0 ; i < numfilesystems ; i++) {
315                         printf( "%s %s %s %s %d %d\n", mntentlist[i].mnt_fsname,
316                                         mntentlist[i].mnt_dir, mntentlist[i].mnt_type, 
317                                         mntentlist[i].mnt_opts, mntentlist[i].mnt_freq, 
318                                         mntentlist[i].mnt_passno);
319                 }
320 #ifdef BB_FEATURE_CLEAN_UP
321                 /* Don't bother to close files or free memory.  Exit 
322                  * does that automagically, so we can save a few bytes */
323                 free( mntentlist);
324                 close(fd);
325 #endif
326                 return EXIT_SUCCESS;
327         }
328 #else
329         if (argc == 1) {
330                 FILE *mountTable = setmntent(mtab_file, "r");
331
332                 if (mountTable) {
333                         struct mntent *m;
334
335                         while ((m = getmntent(mountTable)) != 0) {
336                                 char *blockDevice = m->mnt_fsname;
337                                 if (strcmp(blockDevice, "/dev/root") == 0) {
338                                         find_real_root_device_name( blockDevice);
339                                 }
340                                 printf("%s on %s type %s (%s)\n", blockDevice, m->mnt_dir,
341                                            m->mnt_type, m->mnt_opts);
342                         }
343                         endmntent(mountTable);
344                 } else {
345                         perror_msg_and_die("%s", mtab_file);
346                 }
347                 return EXIT_SUCCESS;
348         }
349 #endif
350
351         /* Parse options */
352         i = --argc;
353         argv++;
354         while (i > 0 && **argv) {
355                 if (**argv == '-') {
356                         char *opt = *argv;
357
358                         while (i > 0 && *++opt)
359                                 switch (*opt) {
360                                 case 'o':
361                                         if (--i == 0) {
362                                                 goto goodbye;
363                                         }
364                                         parse_mount_options(*(++argv), &flags, string_flags);
365                                         break;
366                                 case 'r':
367                                         flags |= MS_RDONLY;
368                                         break;
369                                 case 't':
370                                         if (--i == 0) {
371                                                 goto goodbye;
372                                         }
373                                         filesystemType = *(++argv);
374                                         break;
375                                 case 'w':
376                                         flags &= ~MS_RDONLY;
377                                         break;
378                                 case 'a':
379                                         all = TRUE;
380                                         break;
381                                 case 'f':
382                                         fakeIt = TRUE;
383                                         break;
384 #ifdef BB_FEATURE_MTAB_SUPPORT
385                                 case 'n':
386                                         useMtab = FALSE;
387                                         break;
388 #endif
389                                 case 'v':
390                                         break; /* ignore -v */
391                                 case 'h':
392                                 case '-':
393                                         goto goodbye;
394                                 }
395                 } else {
396                         if (device == NULL)
397                                 device = *argv;
398                         else if (directory == NULL)
399                                 directory = *argv;
400                         else {
401                                 goto goodbye;
402                         }
403                 }
404                 i--;
405                 argv++;
406         }
407
408         if (device == NULL && directory == NULL)
409                 goto goodbye;
410
411         if (all == TRUE || directory == NULL) {
412                 struct mntent *m;
413                 FILE *f = setmntent("/etc/fstab", "r");
414                 fstabmount = TRUE;
415
416                 if (f == NULL)
417                         perror_msg_and_die( "\nCannot read /etc/fstab");
418
419                 while ((m = getmntent(f)) != NULL) {
420                         if (all == FALSE && directory == NULL && (
421                                 (strcmp(device, m->mnt_fsname) != 0) &&
422                                 (strcmp(device, m->mnt_dir) != 0) ) ) {
423                                 continue;
424                         }
425                         
426                         if (all == TRUE && (                            // If we're mounting 'all'
427                                 (strstr(m->mnt_opts, "noauto")) ||      // and the file system isn't noauto,
428                                 (strstr(m->mnt_type, "swap")) ||        // and isn't swap or nfs, then mount it
429                                 (strstr(m->mnt_type, "nfs")) ) ) {
430                                 continue;
431                         }
432                         
433                         if (all == TRUE || flags == 0) {        // Allow single mount to override fstab flags
434                                 flags = 0;
435                                 *string_flags = '\0';
436                                 parse_mount_options(m->mnt_opts, &flags, string_flags);
437                         }
438                         
439                         device = strdup(m->mnt_fsname);
440                         directory = strdup(m->mnt_dir);
441                         filesystemType = strdup(m->mnt_type);
442 singlemount:                    
443                         string_flags = strdup(string_flags);
444                         rc = EXIT_SUCCESS;
445 #ifdef BB_NFSMOUNT
446                         if (strchr(device, ':') != NULL)
447                                 filesystemType = "nfs";
448                         if (strcmp(filesystemType, "nfs") == 0) {
449                                 if (nfsmount (device, directory, &flags, &extra_opts,
450                                                         &string_flags, 1)) {
451                                         perror_msg("nfsmount failed");
452                                         rc = EXIT_FAILURE;
453                                 }
454                         }
455 #endif
456                         if (!mount_one(device, directory, filesystemType, flags,
457                                         string_flags, useMtab, fakeIt, extra_opts, TRUE))
458                                 rc = EXIT_FAILURE;
459                                 
460                         if (all == FALSE)
461                                 break;
462                 }
463                 if (fstabmount == TRUE)
464                         endmntent(f);
465                         
466                 if (all == FALSE && fstabmount == TRUE && directory == NULL)
467                         fprintf(stderr, "Can't find %s in /etc/fstab\n", device);
468         
469                 return rc;
470         }
471         
472         goto singlemount;
473         
474 goodbye:
475         show_usage();
476 }