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 } 183