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