Rework option handling to save space.
[oweals/busybox.git] / util-linux / 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 <limits.h>
47 #include <stdlib.h>
48 #include <unistd.h>
49 #include <errno.h>
50 #include <string.h>
51 #include <stdio.h>
52 #include <mntent.h>
53 #include <ctype.h>
54 #include "busybox.h"
55 #if defined BB_FEATURE_USE_DEVPS_PATCH
56 #       include <linux/devmtab.h> /* For Erik's nifty devmtab device driver */
57 #endif
58
59 enum {
60         MS_MGC_VAL = 0xc0ed0000, /* Magic number indicatng "new" flags */
61         MS_RDONLY = 1,      /* Mount read-only */
62         MS_NOSUID = 2,      /* Ignore suid and sgid bits */
63         MS_NODEV = 4,      /* Disallow access to device special files */
64         MS_NOEXEC = 8,      /* Disallow program execution */
65         MS_SYNCHRONOUS = 16,      /* Writes are synced at once */
66         MS_REMOUNT = 32,      /* Alter flags of a mounted FS */
67         MS_MANDLOCK = 64,      /* Allow mandatory locks on an FS */
68         S_QUOTA = 128,     /* Quota initialized for file/directory/symlink */
69         S_APPEND = 256,     /* Append-only file */
70         S_IMMUTABLE = 512,     /* Immutable file */
71         MS_NOATIME = 1024,    /* Do not update access times. */
72         MS_NODIRATIME = 2048,    /* Do not update directory access times */
73         MS_BIND = 4096,    /* Use the new linux 2.4.x "mount --bind" feature */
74 };
75
76
77 #if defined BB_FEATURE_MOUNT_LOOP
78 #include <fcntl.h>
79 #include <sys/ioctl.h>
80 static int use_loop = FALSE;
81 #endif
82
83 extern int mount (__const char *__special_file, __const char *__dir,
84                         __const char *__fstype, unsigned long int __rwflag,
85                         __const void *__data);
86 extern int umount (__const char *__special_file);
87 extern int umount2 (__const char *__special_file, int __flags);
88
89 extern int sysfs( int option, unsigned int fs_index, char * buf);
90
91 extern const char mtab_file[];  /* Defined in utility.c */
92
93 struct mount_options {
94         const char *name;
95         unsigned long and;
96         unsigned long or;
97 };
98
99 static const struct mount_options mount_options[] = {
100         {"async", ~MS_SYNCHRONOUS, 0},
101         {"atime", ~0, ~MS_NOATIME},
102         {"defaults", ~0, 0},
103         {"dev", ~MS_NODEV, 0},
104         {"diratime", ~0, ~MS_NODIRATIME},
105         {"exec", ~MS_NOEXEC, 0},
106         {"noatime", ~0, MS_NOATIME},
107         {"nodev", ~0, MS_NODEV},
108         {"nodiratime", ~0, MS_NODIRATIME},
109         {"noexec", ~0, MS_NOEXEC},
110         {"nosuid", ~0, MS_NOSUID},
111         {"remount", ~0, MS_REMOUNT},
112         {"ro", ~0, MS_RDONLY},
113         {"rw", ~MS_RDONLY, 0},
114         {"suid", ~MS_NOSUID, 0},
115         {"sync", ~0, MS_SYNCHRONOUS},
116         {"bind", ~0, MS_BIND},
117         {0, 0, 0}
118 };
119
120 static int
121 do_mount(char *specialfile, char *dir, char *filesystemtype,
122                  long flags, void *string_flags, int useMtab, int fakeIt,
123                  char *mtab_opts, int mount_all)
124 {
125         int status = 0;
126 #if defined BB_FEATURE_MOUNT_LOOP
127         char *lofile = NULL;
128 #endif
129
130         if (fakeIt == FALSE)
131         {
132 #if defined BB_FEATURE_MOUNT_LOOP
133                 if (use_loop==TRUE) {
134                         int loro = flags & MS_RDONLY;
135                         
136                         lofile = specialfile;
137
138                         specialfile = find_unused_loop_device();
139                         if (specialfile == NULL) {
140                                 error_msg_and_die("Could not find a spare loop device");
141                         }
142                         if (set_loop(specialfile, lofile, 0, &loro)) {
143                                 error_msg_and_die("Could not setup loop device");
144                         }
145                         if (!(flags & MS_RDONLY) && loro) {     /* loop is ro, but wanted rw */
146                                 error_msg("WARNING: loop device is read-only");
147                                 flags |= MS_RDONLY;
148                         }
149                 }
150 #endif
151                 status = mount(specialfile, dir, filesystemtype, flags, string_flags);
152                 if (status < 0 && errno == EROFS) {
153                         error_msg("%s is write-protected, mounting read-only", specialfile);
154                         status = mount(specialfile, dir, filesystemtype, flags |= MS_RDONLY, string_flags);
155                 }
156                 /* Don't whine about already mounted filesystems when mounting all. */
157                 if (status < 0 && errno == EBUSY && mount_all)
158                         return TRUE;
159         }
160
161
162         /* If the mount was sucessful, do anything needed, then return TRUE */
163         if (status == 0 || fakeIt==TRUE) {
164
165 #if defined BB_FEATURE_MTAB_SUPPORT
166                 if (useMtab == TRUE) {
167                         erase_mtab(specialfile);        // Clean any stale entries
168                         write_mtab(specialfile, dir, filesystemtype, flags, mtab_opts);
169                 }
170 #endif
171                 return (TRUE);
172         }
173
174         /* Bummer.  mount failed.  Clean up */
175 #if defined BB_FEATURE_MOUNT_LOOP
176         if (lofile != NULL) {
177                 del_loop(specialfile);
178         }
179 #endif
180
181         if (errno == EPERM) {
182                 error_msg_and_die("permission denied. Are you root?");
183         }
184
185         return (FALSE);
186 }
187
188
189
190 /* Seperate standard mount options from the nonstandard string options */
191 static void
192 parse_mount_options(char *options, int *flags, char *strflags)
193 {
194         while (options) {
195                 int gotone = FALSE;
196                 char *comma = strchr(options, ',');
197                 const struct mount_options *f = mount_options;
198
199                 if (comma)
200                         *comma = '\0';
201
202                 while (f->name != 0) {
203                         if (strcasecmp(f->name, options) == 0) {
204
205                                 *flags &= f->and;
206                                 *flags |= f->or;
207                                 gotone = TRUE;
208                                 break;
209                         }
210                         f++;
211                 }
212 #if defined BB_FEATURE_MOUNT_LOOP
213                 if (gotone == FALSE && !strcasecmp("loop", options)) {  /* loop device support */
214                         use_loop = TRUE;
215                         gotone = TRUE;
216                 }
217 #endif
218                 if (*strflags && strflags != '\0' && gotone == FALSE) {
219                         char *temp = strflags;
220
221                         temp += strlen(strflags);
222                         *temp++ = ',';
223                         *temp++ = '\0';
224                 }
225                 if (gotone == FALSE)
226                         strcat(strflags, options);
227                 if (comma) {
228                         *comma = ',';
229                         options = ++comma;
230                 } else {
231                         break;
232                 }
233         }
234 }
235
236 static int
237 mount_one(char *blockDevice, char *directory, char *filesystemType,
238                   unsigned long flags, char *string_flags, int useMtab, int fakeIt,
239                   char *mtab_opts, int whineOnErrors, int mount_all)
240 {
241         int status = 0;
242
243         if (strcmp(filesystemType, "auto") == 0) {
244                 static const char *noauto_array[] = { "tmpfs", "shm", "proc", "ramfs", "devpts", "devfs", 0 };
245                 const char **noauto_fstype;
246                 const int num_of_filesystems = sysfs(3, 0, 0);
247                 char buf[255];
248                 int i=0;
249
250                 filesystemType=buf;
251
252                 while(i < num_of_filesystems) {
253                         sysfs(2, i++, filesystemType);
254                         for (noauto_fstype = noauto_array; *noauto_fstype; noauto_fstype++) {
255                                 if (!strcmp(filesystemType, *noauto_fstype)) {
256                                         break;
257                                 }
258                         }
259                         if (!*noauto_fstype) {
260                                 status = do_mount(blockDevice, directory, filesystemType,
261                                         flags | MS_MGC_VAL, string_flags,
262                                         useMtab, fakeIt, mtab_opts, mount_all);
263                                 if (status == TRUE)
264                                         break;
265                         }
266                 }
267         } else {
268                 status = do_mount(blockDevice, directory, filesystemType,
269                                 flags | MS_MGC_VAL, string_flags, useMtab,
270                                 fakeIt, mtab_opts, mount_all);
271         }
272
273         if (status == FALSE) {
274                 if (whineOnErrors == TRUE) {
275                         perror_msg("Mounting %s on %s failed", blockDevice, directory);
276                 }
277                 return (FALSE);
278         }
279         return (TRUE);
280 }
281
282 void show_mounts()
283 {
284 #if defined BB_FEATURE_USE_DEVPS_PATCH
285         int fd, i, numfilesystems;
286         char device[] = "/dev/mtab";
287         struct k_mntent *mntentlist;
288
289         /* open device */ 
290         fd = open(device, O_RDONLY);
291         if (fd < 0)
292                 perror_msg_and_die("open failed for `%s'", device);
293
294         /* How many mounted filesystems?  We need to know to 
295          * allocate enough space for later... */
296         numfilesystems = ioctl (fd, DEVMTAB_COUNT_MOUNTS);
297         if (numfilesystems<0)
298                 perror_msg_and_die( "\nDEVMTAB_COUNT_MOUNTS");
299         mntentlist = (struct k_mntent *) xcalloc ( numfilesystems, sizeof(struct k_mntent));
300                 
301         /* Grab the list of mounted filesystems */
302         if (ioctl (fd, DEVMTAB_GET_MOUNTS, mntentlist)<0)
303                 perror_msg_and_die( "\nDEVMTAB_GET_MOUNTS");
304
305         for( i = 0 ; i < numfilesystems ; i++) {
306                 printf( "%s %s %s %s %d %d\n", mntentlist[i].mnt_fsname,
307                                 mntentlist[i].mnt_dir, mntentlist[i].mnt_type, 
308                                 mntentlist[i].mnt_opts, mntentlist[i].mnt_freq, 
309                                 mntentlist[i].mnt_passno);
310         }
311 #ifdef BB_FEATURE_CLEAN_UP
312         /* Don't bother to close files or free memory.  Exit 
313          * does that automagically, so we can save a few bytes */
314         free( mntentlist);
315         close(fd);
316 #endif
317         exit(EXIT_SUCCESS);
318 #else
319         FILE *mountTable = setmntent(mtab_file, "r");
320
321         if (mountTable) {
322                 struct mntent *m;
323
324                 while ((m = getmntent(mountTable)) != 0) {
325                         char *blockDevice = m->mnt_fsname;
326                         if (strcmp(blockDevice, "/dev/root") == 0) {
327                                 blockDevice = find_real_root_device_name(blockDevice);
328                         }
329                         printf("%s on %s type %s (%s)\n", blockDevice, m->mnt_dir,
330                                    m->mnt_type, m->mnt_opts);
331 #ifdef BB_FEATURE_CLEAN_UP
332                         if(blockDevice != m->mnt_fsname)
333                                 free(blockDevice);
334 #endif
335                 }
336                 endmntent(mountTable);
337         } else {
338                 perror_msg_and_die("%s", mtab_file);
339         }
340         exit(EXIT_SUCCESS);
341 #endif
342 }
343
344 extern int mount_main(int argc, char **argv)
345 {
346         struct stat statbuf;
347         char string_flags_buf[1024] = "";
348         char *string_flags = string_flags_buf;
349         char *extra_opts = string_flags_buf;
350         int flags = 0;
351         char *filesystemType = "auto";
352         char *device = xmalloc(PATH_MAX);
353         char *directory = xmalloc(PATH_MAX);
354         int all = FALSE;
355         int fakeIt = FALSE;
356         int useMtab = TRUE;
357         int rc = EXIT_FAILURE;
358         int fstabmount = FALSE; 
359         int opt;
360
361         /* Parse options */
362         while ((opt = getopt(argc, argv, "o:rt:wafnv")) > 0) {
363                 switch (opt) {
364                 case 'o':
365                         parse_mount_options(optarg, &flags, string_flags);
366                         break;
367                 case 'r':
368                         flags |= MS_RDONLY;
369                         break;
370                 case 't':
371                         filesystemType = optarg;
372                         break;
373                 case 'w':
374                         flags &= ~MS_RDONLY;
375                         break;
376                 case 'a':
377                         all = TRUE;
378                         break;
379                 case 'f':
380                         fakeIt = TRUE;
381                         break;
382 #ifdef BB_FEATURE_MTAB_SUPPORT
383                 case 'n':
384                         useMtab = FALSE;
385                         break;
386 #endif
387                 case 'v':
388                         break; /* ignore -v */
389                 }
390         }
391
392         if (!all && optind == argc)
393                 show_mounts();
394
395         if (optind < argc) {
396                 /* if device is a filename get its real path */
397                 if (stat(argv[optind], &statbuf) == 0) {
398                         realpath(argv[optind], device);
399                 } else {
400                         safe_strncpy(device, argv[optind], PATH_MAX);
401                 }
402         }
403
404         if (optind + 1 < argc) {
405                 if (realpath(argv[optind + 1], directory) == NULL) {
406                         perror_msg_and_die("%s", directory);
407                 }
408         }
409         
410         if (all == TRUE || optind + 1 == argc) {
411                 struct mntent *m = NULL;
412                 FILE *f = setmntent("/etc/fstab", "r");
413                 fstabmount = TRUE;
414
415                 if (f == NULL)
416                         perror_msg_and_die( "\nCannot read /etc/fstab");
417
418                 while ((m = getmntent(f)) != NULL) {
419                         if (all == FALSE && optind + 1 == argc && (
420                                 (strcmp(device, m->mnt_fsname) != 0) &&
421                                 (strcmp(device, m->mnt_dir) != 0) ) ) {
422                                 continue;
423                         }
424                         
425                         if (all == TRUE && (                            // If we're mounting 'all'
426                                 (strstr(m->mnt_opts, "noauto")) ||      // and the file system isn't noauto,
427                                 (strstr(m->mnt_type, "swap")) ||        // and isn't swap or nfs, then mount it
428                                 (strstr(m->mnt_type, "nfs")) ) ) {
429                                 continue;
430                         }
431                         
432                         if (all == TRUE || flags == 0) {        // Allow single mount to override fstab flags
433                                 flags = 0;
434                                 *string_flags = '\0';
435                                 parse_mount_options(m->mnt_opts, &flags, string_flags);
436                         }
437                         
438                         strcpy(device, m->mnt_fsname);
439                         strcpy(directory, m->mnt_dir);
440                         filesystemType = strdup(m->mnt_type);
441 singlemount:                    
442                         string_flags = strdup(string_flags);
443                         rc = EXIT_SUCCESS;
444 #ifdef BB_NFSMOUNT
445                         if (strchr(device, ':') != NULL)
446                                 filesystemType = "nfs";
447                         if (strcmp(filesystemType, "nfs") == 0) {
448                                 if (nfsmount (device, directory, &flags, &extra_opts,
449                                                         &string_flags, 1)) {
450                                         perror_msg("nfsmount failed");
451                                         rc = EXIT_FAILURE;
452                                 }
453                         }
454 #endif
455                         if (!mount_one(device, directory, filesystemType, flags,
456                                         string_flags, useMtab, fakeIt, extra_opts, TRUE, all))
457                                 rc = EXIT_FAILURE;
458                                 
459                         if (all == FALSE)
460                                 break;
461                 }
462                 if (fstabmount == TRUE)
463                         endmntent(f);
464                         
465                 if (all == FALSE && fstabmount == TRUE && m == NULL)
466                         fprintf(stderr, "Can't find %s in /etc/fstab\n", device);
467         
468                 return rc;
469         }
470         
471         goto singlemount;
472 }