contrib/bsdinstaller-1.1.6/src/lib/libinstaller/mount.c

   1 /*
   2  * Copyright (c)2004 The DragonFly Project.  All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  *   Redistributions of source code must retain the above copyright
   9  *   notice, this list of conditions and the following disclaimer.
  10  *
  11  *   Redistributions in binary form must reproduce the above copyright
  12  *   notice, this list of conditions and the following disclaimer in
  13  *   the documentation and/or other materials provided with the
  14  *   distribution.
  15  *
  16  *   Neither the name of the DragonFly Project nor the names of its
  17  *   contributors may be used to endorse or promote products derived
  18  *   from this software without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  23  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  24  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  25  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  26  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  27  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  29  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  31  * OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 /*
  35  * mount.c
  36  * $Id: mount.c,v 1.4 2005/02/10 03:33:49 cpressey Exp $
  37  */
  38
  39 #include <sys/stat.h>
  40 #include <sys/param.h>
  41 #include <sys/ucred.h>
  42 #include <sys/mount.h>
  43
  44 #include <stdarg.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <string.h>
  48 #include <unistd.h>
  49
  50 #include "aura/fspred.h"
  51
  52 #include "commands.h"
  53 #include "functions.h"
  54
  55 #if (__NetBSD_Version__ >= 200040000)
  56 #define STATFS statvfs
  57 #else
  58 #define STATFS statfs
  59 #endif
  60
  61 static void     unmount_all_under_r(struct i_fn_args *, struct commands *,
  62                                     const char *, struct STATFS *, int);
  63
  64 static void
  65 unmount_all_under_r(struct i_fn_args *a, struct commands *cmds,
  66                     const char *mtpt, struct STATFS *mt_array, int count)
  67 {
  68         struct STATFS *mt_ptr;
  69         int k = count;
  70         int unmount_me = 0;
  71
  72         for (mt_ptr = mt_array; k > 0; mt_ptr++, k--) {
  73                 if (strcmp(mt_ptr->f_mntonname, mtpt) == 0)
  74                         unmount_me = 1;
  75
  76                 if (strncmp(mt_ptr->f_mntonname, mtpt, strlen(mtpt)) == 0 &&
  77                     strlen(mtpt) < strlen(mt_ptr->f_mntonname)) {
  78                         unmount_all_under_r(a, cmds,
  79                             mt_ptr->f_mntonname, mt_array, count);
  80                 }
  81         }
  82
  83         if (unmount_me) {
  84                 command_add(cmds, "%s%s %s",
  85                     a->os_root, cmd_name(a, "UMOUNT"), mtpt);
  86         }
  87 }
  88
  89 /*
  90  * Unmount all mountpoints under a given mountpoint.  Recursively unmounts
  91  * dependent mountpoints, so that unmount_all'ing /mnt will first unmount
  92  * /mnt/usr/local, then /mnt/usr, then /mnt itself.
  93  */
  94 void
  95 unmount_all_under(struct i_fn_args *a, struct commands *cmds, const char *fmt, ...)
  96 {
  97         struct STATFS *mt_array;
  98         int count;
  99         char *mtpt;
 100         va_list args;
 101
 102         va_start(args, fmt);
 103         vasprintf(&mtpt, fmt, args);
 104         va_end(args);
 105
 106         count = getmntinfo(&mt_array, MNT_WAIT);
 107         unmount_all_under_r(a, cmds, mtpt, mt_array, count);
 108
 109         free(mtpt);
 110 }