1 /*- 2 * Copyright (c) 2005 Takanori Watanabe 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 #include <sys/cdefs.h> 32 __FBSDID("$FreeBSD$"); 33 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 #define G_LABEL_NTFS_DIR "ntfs" 46 47 struct ntfs_attr { 48 uint32_t a_type; 49 uint32_t reclen; 50 uint8_t a_flag; 51 uint8_t a_namelen; 52 uint8_t a_nameoff; 53 uint8_t reserved1; 54 uint8_t a_compression; 55 uint8_t reserved2; 56 uint16_t a_index; 57 uint16_t a_datalen; 58 uint16_t reserved3; 59 uint16_t a_dataoff; 60 uint16_t a_indexed; 61 } __packed; 62 63 struct ntfs_filerec { 64 uint32_t fr_hdrmagic; 65 uint16_t fr_hdrfoff; 66 uint16_t fr_hdrfnum; 67 uint8_t reserved[8]; 68 uint16_t fr_seqnum; 69 uint16_t fr_nlink; 70 uint16_t fr_attroff; 71 uint16_t fr_flags; 72 uint32_t fr_size; 73 uint32_t fr_allocated; 74 uint64_t fr_mainrec; 75 uint16_t fr_attrnum; 76 } __packed; 77 78 struct ntfs_bootfile { 79 uint8_t reserved1[3]; 80 uint8_t bf_sysid[8]; 81 uint16_t bf_bps; 82 uint8_t bf_spc; 83 uint8_t reserved2[7]; 84 uint8_t bf_media; 85 uint8_t reserved3[2]; 86 uint16_t bf_spt; 87 uint16_t bf_heads; 88 uint8_t reserver4[12]; 89 uint64_t bf_spv; 90 uint64_t bf_mftcn; 91 uint64_t bf_mftmirrcn; 92 int8_t bf_mftrecsz; 93 uint32_t bf_ibsz; 94 uint32_t bf_volsn; 95 } __packed; 96 97 int 98 fstyp_ntfs(FILE *fp, char *label, size_t size) 99 { 100 struct ntfs_bootfile *bf; 101 struct ntfs_filerec *fr; 102 struct ntfs_attr *atr; 103 off_t voloff; 104 char *filerecp, *ap; 105 int8_t mftrecsz; 106 char vnchar; 107 int recsize, j; 108 109 filerecp = NULL; 110 111 bf = (struct ntfs_bootfile *)read_buf(fp, 0, 512); 112 if (bf == NULL || strncmp(bf->bf_sysid, "NTFS ", 8) != 0) 113 return (1); 114 115 mftrecsz = bf->bf_mftrecsz; 116 recsize = (mftrecsz > 0) ? (mftrecsz * bf->bf_bps * bf->bf_spc) : (1 << -mftrecsz); 117 118 voloff = bf->bf_mftcn * bf->bf_spc * bf->bf_bps + 119 recsize * NTFS_VOLUMEINO; 120 121 filerecp = read_buf(fp, voloff, recsize); 122 if (filerecp == NULL) 123 goto fail; 124 fr = (struct ntfs_filerec *)filerecp; 125 126 if (fr->fr_hdrmagic != NTFS_FILEMAGIC) 127 goto fail; 128 129 for (ap = filerecp + fr->fr_attroff; 130 atr = (struct ntfs_attr *)ap, (int)atr->a_type != -1; 131 ap += atr->reclen) { 132 if (atr->a_type == NTFS_A_VOLUMENAME) { 133 if(atr->a_datalen >= size *2){ 134 goto fail; 135 } 136 /* 137 *UNICODE to ASCII. 138 * Should we need to use iconv(9)? 139 */ 140 for (j = 0; j < atr->a_datalen; j++) { 141 vnchar = *(ap + atr->a_dataoff + j); 142 if (j & 1) { 143 if (vnchar) { 144 goto fail; 145 } 146 } else { 147 label[j / 2] = vnchar; 148 } 149 } 150 label[j / 2] = 0; 151 break; 152 } 153 } 154 155 free(bf); 156 free(filerecp); 157 158 return (0); 159 160 fail: 161 free(bf); 162 free(filerecp); 163 164 return (1); 165 } 166