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