xref: /freebsd/sys/geom/label/g_label_ntfs.c (revision a521f2116473fbd8c09db395518f060a27d02334)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2005 Takanori Watanabe
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28 
29 #include <sys/cdefs.h>
30 __FBSDID("$FreeBSD$");
31 
32 #include <sys/param.h>
33 #include <sys/systm.h>
34 #include <sys/kernel.h>
35 #include <sys/malloc.h>
36 
37 #include <geom/geom.h>
38 #include <geom/label/g_label.h>
39 
40 #define	NTFS_A_VOLUMENAME	0x60
41 #define	NTFS_FILEMAGIC		((uint32_t)(0x454C4946))
42 #define	NTFS_VOLUMEINO		3
43 
44 struct ntfs_attr {
45 	uint32_t	a_type;
46 	uint32_t	reclen;
47 	uint8_t		a_flag;
48 	uint8_t		a_namelen;
49 	uint8_t		a_nameoff;
50 	uint8_t		reserved1;
51 	uint8_t		a_compression;
52 	uint8_t		reserved2;
53 	uint16_t	a_index;
54 	uint16_t	a_datalen;
55 	uint16_t	reserved3;
56 	uint16_t	a_dataoff;
57 	uint16_t	a_indexed;
58 } __packed;
59 
60 struct ntfs_filerec {
61 	uint32_t	fr_hdrmagic;
62 	uint16_t	fr_hdrfoff;
63 	uint16_t	fr_hdrfnum;
64 	uint8_t		reserved[8];
65 	uint16_t	fr_seqnum;
66 	uint16_t	fr_nlink;
67 	uint16_t	fr_attroff;
68 	uint16_t	fr_flags;
69 	uint32_t	fr_size;
70 	uint32_t	fr_allocated;
71 	uint64_t	fr_mainrec;
72 	uint16_t	fr_attrnum;
73 } __packed;
74 
75 struct ntfs_bootfile {
76 	uint8_t		reserved1[3];
77 	uint8_t		bf_sysid[8];
78 	uint16_t	bf_bps;
79 	uint8_t		bf_spc;
80 	uint8_t		reserved2[7];
81 	uint8_t		bf_media;
82 	uint8_t		reserved3[2];
83 	uint16_t	bf_spt;
84 	uint16_t	bf_heads;
85 	uint8_t		reserver4[12];
86 	uint64_t	bf_spv;
87 	uint64_t	bf_mftcn;
88 	uint64_t	bf_mftmirrcn;
89 	int8_t		bf_mftrecsz;
90 	uint32_t	bf_ibsz;
91 	uint32_t	bf_volsn;
92 } __packed;
93 
94 static void
95 g_label_ntfs_taste(struct g_consumer *cp, char *label, size_t size)
96 {
97 	struct g_provider *pp;
98 	struct ntfs_bootfile *bf;
99 	struct ntfs_filerec *fr;
100 	struct ntfs_attr *atr;
101 	off_t voloff;
102 	char *filerecp, *ap;
103 	int8_t mftrecsz;
104 	char vnchar;
105 	int recsize, j;
106 
107 	g_topology_assert_not();
108 
109 	label[0] = '\0';
110 	pp = cp->provider;
111 	filerecp = NULL;
112 
113 	bf = (struct ntfs_bootfile *)g_read_data(cp, 0, pp->sectorsize, NULL);
114 	if (bf == NULL || strncmp(bf->bf_sysid, "NTFS    ", 8) != 0)
115 		goto done;
116 
117 	mftrecsz = bf->bf_mftrecsz;
118 	recsize = (mftrecsz > 0) ? (mftrecsz * bf->bf_bps * bf->bf_spc) : (1 << -mftrecsz);
119 	if (recsize == 0 || recsize % pp->sectorsize != 0)
120 		goto done;
121 
122 	voloff = bf->bf_mftcn * bf->bf_spc * bf->bf_bps +
123 	    recsize * NTFS_VOLUMEINO;
124 	if (voloff % pp->sectorsize != 0)
125 		goto done;
126 
127 	filerecp = g_read_data(cp, voloff, recsize, NULL);
128 	if (filerecp == NULL)
129 		goto done;
130 	fr = (struct ntfs_filerec *)filerecp;
131 
132 	if (fr->fr_hdrmagic != NTFS_FILEMAGIC)
133 		goto done;
134 
135 	for (ap = filerecp + fr->fr_attroff;
136 	    atr = (struct ntfs_attr *)ap, atr->a_type != -1;
137 	    ap += atr->reclen) {
138 		if (atr->a_type == NTFS_A_VOLUMENAME) {
139 			if(atr->a_datalen >= size *2){
140 				label[0] = 0;
141 				goto done;
142 			}
143 			/*
144 			 *UNICODE to ASCII.
145 			 * Should we need to use iconv(9)?
146 			 */
147 			for (j = 0; j < atr->a_datalen; j++) {
148 				vnchar = *(ap + atr->a_dataoff + j);
149 				if (j & 1) {
150 					if (vnchar) {
151 						label[0] = 0;
152 						goto done;
153 					}
154 				} else {
155 					label[j / 2] = vnchar;
156 				}
157 			}
158 			label[j / 2] = 0;
159 			break;
160 		}
161 	}
162 done:
163 	if (bf != NULL)
164 		g_free(bf);
165 	if (filerecp != NULL)
166 		g_free(filerecp);
167 }
168 
169 struct g_label_desc g_label_ntfs = {
170 	.ld_taste = g_label_ntfs_taste,
171 	.ld_dirprefix = "ntfs/",
172 	.ld_enabled = 1
173 };
174 
175 G_LABEL_INIT(ntfs, g_label_ntfs, "Create device nodes for NTFS volumes");
176