sbin/mount_msdosfs/mount_msdosfs.c

   1 /*      $NetBSD: mount_msdos.c,v 1.18 1997/09/16 12:24:18 lukem Exp $   */
   2
   3 /*
   4  * Copyright (c) 1994 Christopher G. Demetriou
   5  * 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. All advertising materials mentioning features or use of this software
  16  *    must display the following acknowledgement:
  17  *      This product includes software developed by Christopher G. Demetriou.
  18  * 4. The name of the author may not be used to endorse or promote products
  19  *    derived from this software without specific prior written permission
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  22  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  23  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  24  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  25  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  26  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  30  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  */
  32
  33 #ifndef lint
  34 static const char rcsid[] =
  35   "$FreeBSD$";
  36 #endif /* not lint */
  37
  38 #include <sys/param.h>
  39 #include <sys/mount.h>
  40 #include <sys/stat.h>
  41 #include <sys/iconv.h>
  42 #include <sys/linker.h>
  43 #include <sys/module.h>
  44
  45 #include <ctype.h>
  46 #include <err.h>
  47 #include <errno.h>
  48 #include <grp.h>
  49 #include <locale.h>
  50 #include <pwd.h>
  51 #include <stdio.h>
  52 /* must be after stdio to declare fparseln */
  53 #include <libutil.h>
  54 #include <stdlib.h>
  55 #include <string.h>
  56 #include <sysexits.h>
  57 #include <unistd.h>
  58
  59 #include "mntopts.h"
  60
  61 static gid_t    a_gid(char *);
  62 static uid_t    a_uid(char *);
  63 static mode_t   a_mask(char *);
  64 static void     usage(void) __dead2;
  65 static int      set_charset(struct iovec **iov, int *iovlen, const char *, const char *);
  66
  67 int
  68 main(int argc, char **argv)
  69 {
  70         struct iovec *iov = NULL;
  71         int iovlen = 0;
  72         struct stat sb;
  73         int c, set_gid, set_uid, set_mask, set_dirmask;
  74         char *dev, *dir, mntpath[MAXPATHLEN], *csp;
  75         char fstype[] = "msdosfs";
  76         char errmsg[255] = {0};
  77         char *cs_dos = NULL;
  78         char *cs_local = NULL;
  79         mode_t mask = 0, dirmask = 0;
  80         uid_t uid = 0;
  81         gid_t gid = 0;
  82
  83         set_gid = set_uid = set_mask = set_dirmask = 0;
  84
  85         while ((c = getopt(argc, argv, "sl9u:g:m:M:o:L:D:W:")) != -1) {
  86                 switch (c) {
  87                 case 's':
  88                         build_iovec(&iov, &iovlen, "shortnames", NULL, (size_t)-1);
  89                         break;
  90                 case 'l':
  91                         build_iovec(&iov, &iovlen, "longnames", NULL, (size_t)-1);
  92                         break;
  93                 case '9':
  94                         build_iovec_argf(&iov, &iovlen, "nowin95", "", (size_t)-1);
  95                         break;
  96                 case 'u':
  97                         uid = a_uid(optarg);
  98                         set_uid = 1;
  99                         break;
 100                 case 'g':
 101                         gid = a_gid(optarg);
 102                         set_gid = 1;
 103                         break;
 104                 case 'm':
 105                         mask = a_mask(optarg);
 106                         set_mask = 1;
 107                         break;
 108                 case 'M':
 109                         dirmask = a_mask(optarg);
 110                         set_dirmask = 1;
 111                         break;
 112                 case 'L': {
 113                         const char *quirk = NULL;
 114                         if (setlocale(LC_CTYPE, optarg) == NULL)
 115                                 err(EX_CONFIG, "%s", optarg);
 116                         csp = strchr(optarg,'.');
 117                         if (!csp)
 118                                 err(EX_CONFIG, "%s", optarg);
 119                         quirk = kiconv_quirkcs(csp + 1, KICONV_VENDOR_MICSFT);
 120                         build_iovec_argf(&iov, &iovlen, "cs_local", quirk);
 121                         cs_local = strdup(quirk);
 122                         }
 123                         break;
 124                 case 'D':
 125                         cs_dos = strdup(optarg);
 126                         build_iovec_argf(&iov, &iovlen, "cs_dos", cs_dos, (size_t)-1);
 127                         break;
 128                 case 'o': {
 129                         char *p = NULL;
 130                         char *val = strdup("");
 131                         p = strchr(optarg, '=');
 132                         if (p != NULL) {
 133                                 free(val);
 134                                 *p = '\0';
 135                                 val = p + 1;
 136                         }
 137                         build_iovec(&iov, &iovlen, optarg, val, (size_t)-1);
 138                         }
 139                         break;
 140                 case 'W':
 141                         if (strcmp(optarg, "iso22dos") == 0) {
 142                                 cs_local = strdup("ISO8859-2");
 143                                 cs_dos = strdup("CP852");
 144                         } else if (strcmp(optarg, "iso72dos") == 0) {
 145                                 cs_local = strdup("ISO8859-7");
 146                                 cs_dos = strdup("CP737");
 147                         } else if (strcmp(optarg, "koi2dos") == 0) {
 148                                 cs_local = strdup("KOI8-R");
 149                                 cs_dos = strdup("CP866");
 150                         } else if (strcmp(optarg, "koi8u2dos") == 0) {
 151                                 cs_local = strdup("KOI8-U");
 152                                 cs_dos = strdup("CP866");
 153                         } else {
 154                                 err(EX_NOINPUT, "%s", optarg);
 155                         }
 156                         build_iovec(&iov, &iovlen, "cs_local", cs_local, (size_t)-1);
 157                         build_iovec(&iov, &iovlen, "cs_dos", cs_dos, (size_t)-1);
 158                         break;
 159                 case '?':
 160                 default:
 161                         usage();
 162                         break;
 163                 }
 164         }
 165
 166         if (optind + 2 != argc)
 167                 usage();
 168
 169         if (set_mask && !set_dirmask) {
 170                 dirmask = mask;
 171                 set_dirmask = 1;
 172         }
 173         else if (set_dirmask && !set_mask) {
 174                 mask = dirmask;
 175                 set_mask = 1;
 176         }
 177
 178         dev = argv[optind];
 179         dir = argv[optind + 1];
 180
 181         if (cs_local != NULL) {
 182                 if (set_charset(&iov, &iovlen, cs_local, cs_dos) == -1)
 183                         err(EX_OSERR, "msdosfs_iconv");
 184                 build_iovec_argf(&iov, &iovlen, "kiconv", "");
 185         } else if (cs_dos != NULL) {
 186                 build_iovec_argf(&iov, &iovlen, "cs_local", "ISO8859-1");
 187                 if (set_charset(&iov, &iovlen, "ISO8859-1", cs_dos) == -1)
 188                         err(EX_OSERR, "msdosfs_iconv");
 189                 build_iovec_argf(&iov, &iovlen, "kiconv", "");
 190         }
 191
 192         /*
 193          * Resolve the mountpoint with realpath(3) and remove unnecessary
 194          * slashes from the devicename if there are any.
 195          */
 196         if (checkpath(dir, mntpath) != 0)
 197                 err(EX_USAGE, "%s", mntpath);
 198         (void)rmslashes(dev, dev);
 199
 200         if (!set_gid || !set_uid || !set_mask) {
 201                 if (stat(mntpath, &sb) == -1)
 202                         err(EX_OSERR, "stat %s", mntpath);
 203
 204                 if (!set_uid)
 205                         uid = sb.st_uid;
 206                 if (!set_gid)
 207                         gid = sb.st_gid;
 208                 if (!set_mask)
 209                         mask = dirmask =
 210                                 sb.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO);
 211         }
 212
 213         build_iovec(&iov, &iovlen, "fstype", fstype, (size_t)-1);
 214         build_iovec(&iov, &iovlen, "fspath", mntpath, (size_t)-1);
 215         build_iovec(&iov, &iovlen, "from", dev, (size_t)-1);
 216         build_iovec(&iov, &iovlen, "errmsg", errmsg, sizeof(errmsg));
 217         build_iovec_argf(&iov, &iovlen, "uid", "%d", uid);
 218         build_iovec_argf(&iov, &iovlen, "gid", "%u", gid);
 219         build_iovec_argf(&iov, &iovlen, "mask", "%u", mask);
 220         build_iovec_argf(&iov, &iovlen, "dirmask", "%u", dirmask);
 221
 222         if (nmount(iov, iovlen, 0) < 0) {
 223                 if (errmsg[0])
 224                         err(1, "%s: %s", dev, errmsg);
 225                 else
 226                         err(1, "%s", dev);
 227         }
 228
 229         exit (0);
 230 }
 231
 232 gid_t
 233 a_gid(char *s)
 234 {
 235         struct group *gr;
 236         char *gname;
 237         gid_t gid;
 238
 239         if ((gr = getgrnam(s)) != NULL)
 240                 gid = gr->gr_gid;
 241         else {
 242                 for (gname = s; *s && isdigit(*s); ++s);
 243                 if (!*s)
 244                         gid = atoi(gname);
 245                 else
 246                         errx(EX_NOUSER, "unknown group id: %s", gname);
 247         }
 248         return (gid);
 249 }
 250
 251 uid_t
 252 a_uid(char *s)
 253 {
 254         struct passwd *pw;
 255         char *uname;
 256         uid_t uid;
 257
 258         if ((pw = getpwnam(s)) != NULL)
 259                 uid = pw->pw_uid;
 260         else {
 261                 for (uname = s; *s && isdigit(*s); ++s);
 262                 if (!*s)
 263                         uid = atoi(uname);
 264                 else
 265                         errx(EX_NOUSER, "unknown user id: %s", uname);
 266         }
 267         return (uid);
 268 }
 269
 270 mode_t
 271 a_mask(char *s)
 272 {
 273         int done, rv;
 274         char *ep;
 275
 276         done = 0;
 277         rv = -1;
 278         if (*s >= '0' && *s <= '7') {
 279                 done = 1;
 280                 rv = strtol(optarg, &ep, 8);
 281         }
 282         if (!done || rv < 0 || *ep)
 283                 errx(EX_USAGE, "invalid file mode: %s", s);
 284         return (rv);
 285 }
 286
 287 void
 288 usage(void)
 289 {
 290         fprintf(stderr, "%s\n%s\n%s\n",
 291         "usage: mount_msdosfs [-9ls] [-D DOS_codepage] [-g gid] [-L locale]",
 292         "                     [-M mask] [-m mask] [-o options] [-u uid]",
 293         "                     [-W table] special node");
 294         exit(EX_USAGE);
 295 }
 296
 297 int
 298 set_charset(struct iovec **iov, int *iovlen, const char *cs_local, const char *cs_dos)
 299 {
 300         int error;
 301
 302         if (modfind("msdosfs_iconv") < 0)
 303                 if (kldload("msdosfs_iconv") < 0 || modfind("msdosfs_iconv") < 0) {
 304                         warnx("cannot find or load \"msdosfs_iconv\" kernel module");
 305                         return (-1);
 306                 }
 307
 308         build_iovec_argf(iov, iovlen, "cs_win", ENCODING_UNICODE);
 309         error = kiconv_add_xlat16_cspairs(ENCODING_UNICODE, cs_local);
 310         if (error && errno != EEXIST)
 311                 return (-1);
 312         if (cs_dos != NULL) {
 313                 error = kiconv_add_xlat16_cspairs(cs_dos, cs_local);
 314                 if (error && errno != EEXIST)
 315                         return (-1);
 316         } else {
 317                 build_iovec_argf(iov, iovlen, "cs_dos", cs_local);
 318                 error = kiconv_add_xlat16_cspair(cs_local, cs_local,
 319                                 KICONV_FROM_UPPER | KICONV_LOWER);
 320                 if (error && errno != EEXIST)
 321                         return (-1);
 322         }
 323
 324         return (0);
 325 }