usr.sbin/fstyp/fstyp.c

   1 /*-
   2  * Copyright (c) 2016 The DragonFly Project
   3  * Copyright (c) 2014 The FreeBSD Foundation
   4  * All rights reserved.
   5  *
   6  * This software was developed by Edward Tomasz Napierala under sponsorship
   7  * from the FreeBSD Foundation.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28  * SUCH DAMAGE.
  29  *
  30  */
  31
  32 #include <sys/param.h>
  33 #include <sys/diskslice.h>
  34 #include <sys/ioctl.h>
  35 #include <sys/stat.h>
  36 #include <err.h>
  37 #include <errno.h>
  38 #include <iconv.h>
  39 #include <locale.h>
  40 #include <stdbool.h>
  41 #include <stddef.h>
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44 #include <string.h>
  45 #include <unistd.h>
  46 #include <vis.h>
  47
  48 #include "fstyp.h"
  49
  50 #define LABEL_LEN       512
  51
  52 bool show_label = false;
  53
  54 typedef int (*fstyp_function)(FILE *, char *, size_t, const char *);
  55 typedef int (*fsvtyp_function)(const char *, char *, size_t);
  56
  57 static struct {
  58         const char      *name;
  59         fstyp_function  function;
  60         bool            unmountable;
  61         const char      *precache_encoding;
  62 } fstypes[] = {
  63         { "apfs", &fstyp_apfs, true, NULL },
  64         { "befs", &fstyp_befs, false, NULL },
  65         { "cd9660", &fstyp_cd9660, false, NULL },
  66         { "exfat", &fstyp_exfat, false, EXFAT_ENC },
  67         { "ext2fs", &fstyp_ext2fs, false, NULL },
  68         { "hfs+", &fstyp_hfsp, false, NULL },
  69         { "msdosfs", &fstyp_msdosfs, false, NULL },
  70         { "ntfs", &fstyp_ntfs, false, NTFS_ENC },
  71         { "ufs", &fstyp_ufs, false, NULL },
  72         { "hammer", &fstyp_hammer, false, NULL },
  73         { "hammer2", &fstyp_hammer2, false, NULL },
  74         { NULL, NULL, NULL, NULL }
  75 };
  76
  77 static struct {
  78         const char      *name;
  79         fsvtyp_function function;
  80         bool            unmountable;
  81         const char      *precache_encoding;
  82 } fsvtypes[] = {
  83         { "hammer", &fsvtyp_hammer, false, NULL }, /* Must be before partial */
  84         { "hammer(partial)", &fsvtyp_hammer_partial, true, NULL },
  85         { "hammer2", &fsvtyp_hammer2, false, NULL }, /* Must be before partial */
  86         { "hammer2(partial)", &fsvtyp_hammer2_partial, true, NULL },
  87         { NULL, NULL, NULL, NULL }
  88 };
  89
  90 void *
  91 read_buf(FILE *fp, off_t off, size_t len)
  92 {
  93         int error;
  94         size_t nread;
  95         void *buf;
  96
  97         error = fseek(fp, off, SEEK_SET);
  98         if (error != 0) {
  99                 warn("cannot seek to %jd", (uintmax_t)off);
 100                 return (NULL);
 101         }
 102
 103         buf = malloc(len);
 104         if (buf == NULL) {
 105                 warn("cannot malloc %zd bytes of memory", len);
 106                 return (NULL);
 107         }
 108
 109         nread = fread(buf, len, 1, fp);
 110         if (nread != 1) {
 111                 free(buf);
 112                 if (feof(fp) == 0)
 113                         warn("fread");
 114                 return (NULL);
 115         }
 116
 117         return (buf);
 118 }
 119
 120 char *
 121 checked_strdup(const char *s)
 122 {
 123         char *c;
 124
 125         c = strdup(s);
 126         if (c == NULL)
 127                 err(1, "strdup");
 128         return (c);
 129 }
 130
 131 void
 132 rtrim(char *label, size_t size)
 133 {
 134         ptrdiff_t i;
 135
 136         for (i = size - 1; i >= 0; i--) {
 137                 if (label[i] == '\0')
 138                         continue;
 139                 else if (label[i] == ' ')
 140                         label[i] = '\0';
 141                 else
 142                         break;
 143         }
 144 }
 145
 146 static void
 147 usage(void)
 148 {
 149
 150         fprintf(stderr, "usage: fstyp [-l] [-s] [-u] special\n");
 151         exit(1);
 152 }
 153
 154 static void
 155 type_check(const char *path, FILE *fp)
 156 {
 157         int error, fd;
 158         struct stat sb;
 159         struct partinfo pinfo;
 160
 161         fd = fileno(fp);
 162
 163         error = fstat(fd, &sb);
 164         if (error != 0)
 165                 err(1, "%s: fstat", path);
 166
 167         if (S_ISREG(sb.st_mode))
 168                 return;
 169
 170         error = ioctl(fd, DIOCGPART, &pinfo);
 171         if (error != 0)
 172                 errx(1, "%s: not a disk", path);
 173 }
 174
 175 int
 176 main(int argc, char **argv)
 177 {
 178         int ch, error, i, nbytes;
 179         bool ignore_type = false, show_unmountable = false;
 180         char label[LABEL_LEN + 1], strvised[LABEL_LEN * 4 + 1];
 181         char fdpath[MAXPATHLEN];
 182         char *p;
 183         const char *path;
 184         const char *name = NULL;
 185         FILE *fp;
 186         fstyp_function fstyp_f;
 187         fsvtyp_function fsvtyp_f;
 188
 189         while ((ch = getopt(argc, argv, "lsu")) != -1) {
 190                 switch (ch) {
 191                 case 'l':
 192                         show_label = true;
 193                         break;
 194                 case 's':
 195                         ignore_type = true;
 196                         break;
 197                 case 'u':
 198                         show_unmountable = true;
 199                         break;
 200                 default:
 201                         usage();
 202                 }
 203         }
 204
 205         argc -= optind;
 206         argv += optind;
 207         if (argc != 1)
 208                 usage();
 209
 210         path = argv[0];
 211
 212         if (setlocale(LC_CTYPE, "") == NULL)
 213                 err(1, "setlocale");
 214
 215         /*
 216          * DragonFly: Filesystems may have syntax to decorate path.
 217          * Make a wild guess.
 218          */
 219         strlcpy(fdpath, path, sizeof(fdpath));
 220         p = strchr(fdpath, '@');
 221         if (p)
 222                 *p = '\0';
 223
 224         fp = fopen(fdpath, "r");
 225         if (fp == NULL) {
 226                 if (strcmp(path, fdpath))
 227                         fp = fopen(path, "r");
 228                 if (fp == NULL)
 229                         goto fsvtyp; /* DragonFly */
 230                 else
 231                         strlcpy(fdpath, path, sizeof(fdpath));
 232         }
 233
 234         if (ignore_type == false)
 235                 type_check(fdpath, fp);
 236
 237         memset(label, '\0', sizeof(label));
 238
 239         for (i = 0;; i++) {
 240                 if (show_unmountable == false && fstypes[i].unmountable == true)
 241                         continue;
 242                 fstyp_f = fstypes[i].function;
 243                 if (fstyp_f == NULL)
 244                         break;
 245
 246                 error = fstyp_f(fp, label, sizeof(label), path);
 247                 if (error == 0) {
 248                         name = fstypes[i].name;
 249                         goto done;
 250                 }
 251         }
 252 fsvtyp:
 253         for (i = 0;; i++) {
 254                 if (show_unmountable == false && fsvtypes[i].unmountable == true)
 255                         continue;
 256                 fsvtyp_f = fsvtypes[i].function;
 257                 if (fsvtyp_f == NULL)
 258                         break;
 259
 260                 error = fsvtyp_f(path, label, sizeof(label));
 261                 if (error == 0) {
 262                         name = fsvtypes[i].name;
 263                         goto done;
 264                 }
 265         }
 266
 267         warnx("%s: filesystem not recognized", path);
 268         return (1);
 269 done:
 270         if (show_label && label[0] != '\0') {
 271                 /*
 272                  * XXX: I'd prefer VIS_HTTPSTYLE, but it unconditionally
 273                  *      encodes spaces.
 274                  */
 275                 nbytes = strsnvis(strvised, sizeof(strvised), label,
 276                     VIS_GLOB | VIS_NL, "\"'$");
 277                 if (nbytes == -1)
 278                         err(1, "strsnvis");
 279
 280                 printf("%s %s\n", name, strvised);
 281         } else {
 282                 printf("%s\n", name);
 283         }
 284
 285         return (0);
 286 }