usr.sbin/fstyp/ntfs.c

   1 /*-
   2  * Copyright (c) 2016 The DragonFly Project
   3  * Copyright (c) 2005 Takanori Watanabe
   4  * Copyright (c) 2014 The FreeBSD Foundation
   5  * All rights reserved.
   6  *
   7  * This software was developed by Edward Tomasz Napierala under sponsorship
   8  * from the FreeBSD Foundation.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  */
  31
  32 #include <err.h>
  33 #include <iconv.h>
  34 #include <stdint.h>
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <string.h>
  38
  39 #include "fstyp.h"
  40
  41 #define NTFS_A_VOLUMENAME       0x60
  42 #define NTFS_FILEMAGIC          ((uint32_t)(0x454C4946))
  43 #define NTFS_VOLUMEINO          3
  44
  45 struct ntfs_attr {
  46         uint32_t        a_type;
  47         uint32_t        reclen;
  48         uint8_t         a_flag;
  49         uint8_t         a_namelen;
  50         uint8_t         a_nameoff;
  51         uint8_t         reserved1;
  52         uint8_t         a_compression;
  53         uint8_t         reserved2;
  54         uint16_t        a_index;
  55         uint16_t        a_datalen;
  56         uint16_t        reserved3;
  57         uint16_t        a_dataoff;
  58         uint16_t        a_indexed;
  59 } __packed;
  60
  61 struct ntfs_filerec {
  62         uint32_t        fr_hdrmagic;
  63         uint16_t        fr_hdrfoff;
  64         uint16_t        fr_hdrfnum;
  65         uint8_t         reserved[8];
  66         uint16_t        fr_seqnum;
  67         uint16_t        fr_nlink;
  68         uint16_t        fr_attroff;
  69         uint16_t        fr_flags;
  70         uint32_t        fr_size;
  71         uint32_t        fr_allocated;
  72         uint64_t        fr_mainrec;
  73         uint16_t        fr_attrnum;
  74 } __packed;
  75
  76 struct ntfs_bootfile {
  77         uint8_t         reserved1[3];
  78         uint8_t         bf_sysid[8];
  79         uint16_t        bf_bps;
  80         uint8_t         bf_spc;
  81         uint8_t         reserved2[7];
  82         uint8_t         bf_media;
  83         uint8_t         reserved3[2];
  84         uint16_t        bf_spt;
  85         uint16_t        bf_heads;
  86         uint8_t         reserver4[12];
  87         uint64_t        bf_spv;
  88         uint64_t        bf_mftcn;
  89         uint64_t        bf_mftmirrcn;
  90         int8_t          bf_mftrecsz;
  91         uint32_t        bf_ibsz;
  92         uint32_t        bf_volsn;
  93 } __packed;
  94
  95 static void
  96 convert_label(const void *label /* LE */, size_t labellen, char *label_out,
  97     size_t label_sz)
  98 {
  99         char *label_out_orig;
 100         iconv_t cd;
 101         size_t rc;
 102
 103         /* dstname="" means convert to the current locale. */
 104         cd = iconv_open("", NTFS_ENC);
 105         if (cd == (iconv_t)-1) {
 106                 warn("ntfs: Could not open iconv");
 107                 return;
 108         }
 109
 110         label_out_orig = label_out;
 111
 112         rc = iconv(cd, __DECONST(char **, &label), &labellen, &label_out,
 113             &label_sz);
 114         if (rc == (size_t)-1) {
 115                 warn("ntfs: iconv()");
 116                 *label_out_orig = '\0';
 117         } else {
 118                 /* NUL-terminate result (iconv advances label_out). */
 119                 if (label_sz == 0)
 120                         label_out--;
 121                 *label_out = '\0';
 122         }
 123
 124         iconv_close(cd);
 125 }
 126
 127 int
 128 fstyp_ntfs(FILE *fp, char *label, size_t size, const char *devpath)
 129 {
 130         struct ntfs_bootfile *bf;
 131         struct ntfs_filerec *fr;
 132         struct ntfs_attr *atr;
 133         off_t voloff;
 134         char *filerecp, *ap;
 135         int8_t mftrecsz;
 136         int recsize;
 137
 138         filerecp = NULL;
 139
 140         bf = (struct ntfs_bootfile *)read_buf(fp, 0, 512);
 141         if (bf == NULL || strncmp(bf->bf_sysid, "NTFS    ", 8) != 0)
 142                 goto fail;
 143         if (!show_label)
 144                 goto ok;
 145
 146         mftrecsz = bf->bf_mftrecsz;
 147         recsize = (mftrecsz > 0) ? (mftrecsz * bf->bf_bps * bf->bf_spc) : (1 << -mftrecsz);
 148
 149         voloff = bf->bf_mftcn * bf->bf_spc * bf->bf_bps +
 150             recsize * NTFS_VOLUMEINO;
 151
 152         filerecp = read_buf(fp, voloff, recsize);
 153         if (filerecp == NULL)
 154                 goto fail;
 155         fr = (struct ntfs_filerec *)filerecp;
 156
 157         if (fr->fr_hdrmagic != NTFS_FILEMAGIC)
 158                 goto fail;
 159
 160         for (ap = filerecp + fr->fr_attroff;
 161             atr = (struct ntfs_attr *)ap, (int)atr->a_type != -1;
 162             ap += atr->reclen) {
 163                 if (atr->a_type != NTFS_A_VOLUMENAME)
 164                         continue;
 165
 166                 convert_label(ap + atr->a_dataoff,
 167                     atr->a_datalen, label, size);
 168                 break;
 169         }
 170
 171 ok:
 172         free(bf);
 173         free(filerecp);
 174
 175         return (0);
 176
 177 fail:
 178         free(bf);
 179         free(filerecp);
 180
 181         return (1);
 182 }