ntfs_subr.c revision 1.28 1 1.28 ad /* $NetBSD: ntfs_subr.c,v 1.28 2007/02/15 15:40:52 ad Exp $ */
2 1.1 jdolecek
3 1.1 jdolecek /*-
4 1.1 jdolecek * Copyright (c) 1998, 1999 Semen Ustimenko (semenu (at) FreeBSD.org)
5 1.1 jdolecek * All rights reserved.
6 1.1 jdolecek *
7 1.1 jdolecek * Redistribution and use in source and binary forms, with or without
8 1.1 jdolecek * modification, are permitted provided that the following conditions
9 1.1 jdolecek * are met:
10 1.1 jdolecek * 1. Redistributions of source code must retain the above copyright
11 1.1 jdolecek * notice, this list of conditions and the following disclaimer.
12 1.1 jdolecek * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 jdolecek * notice, this list of conditions and the following disclaimer in the
14 1.1 jdolecek * documentation and/or other materials provided with the distribution.
15 1.1 jdolecek *
16 1.1 jdolecek * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 1.1 jdolecek * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 1.1 jdolecek * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 1.1 jdolecek * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 1.1 jdolecek * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 1.1 jdolecek * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 1.1 jdolecek * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 1.1 jdolecek * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 1.1 jdolecek * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 1.1 jdolecek * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 1.1 jdolecek * SUCH DAMAGE.
27 1.1 jdolecek *
28 1.1 jdolecek * Id: ntfs_subr.c,v 1.4 1999/05/12 09:43:01 semenu Exp
29 1.1 jdolecek */
30 1.1 jdolecek
31 1.1 jdolecek #include <sys/cdefs.h>
32 1.28 ad __KERNEL_RCSID(0, "$NetBSD: ntfs_subr.c,v 1.28 2007/02/15 15:40:52 ad Exp $");
33 1.1 jdolecek
34 1.1 jdolecek #include <sys/param.h>
35 1.1 jdolecek #include <sys/systm.h>
36 1.1 jdolecek #include <sys/namei.h>
37 1.1 jdolecek #include <sys/proc.h>
38 1.1 jdolecek #include <sys/kernel.h>
39 1.1 jdolecek #include <sys/vnode.h>
40 1.1 jdolecek #include <sys/mount.h>
41 1.1 jdolecek #include <sys/buf.h>
42 1.1 jdolecek #include <sys/file.h>
43 1.1 jdolecek #include <sys/malloc.h>
44 1.1 jdolecek #include <sys/lock.h>
45 1.25 elad #include <sys/kauth.h>
46 1.1 jdolecek #if defined(__FreeBSD__)
47 1.1 jdolecek #include <machine/clock.h>
48 1.1 jdolecek #endif
49 1.1 jdolecek
50 1.1 jdolecek #include <miscfs/specfs/specdev.h>
51 1.1 jdolecek
52 1.1 jdolecek #include <fs/ntfs/ntfs.h>
53 1.1 jdolecek #include <fs/ntfs/ntfsmount.h>
54 1.1 jdolecek #include <fs/ntfs/ntfs_inode.h>
55 1.1 jdolecek #include <fs/ntfs/ntfs_vfsops.h>
56 1.1 jdolecek #include <fs/ntfs/ntfs_subr.h>
57 1.1 jdolecek #include <fs/ntfs/ntfs_compr.h>
58 1.1 jdolecek #include <fs/ntfs/ntfs_ihash.h>
59 1.1 jdolecek
60 1.20 christos #ifdef NTFS_DEBUG
61 1.1 jdolecek int ntfs_debug = NTFS_DEBUG;
62 1.1 jdolecek #endif
63 1.1 jdolecek
64 1.1 jdolecek MALLOC_DEFINE(M_NTFSNTVATTR, "NTFS vattr", "NTFS file attribute information");
65 1.1 jdolecek MALLOC_DEFINE(M_NTFSRDATA, "NTFS res data", "NTFS resident data");
66 1.1 jdolecek MALLOC_DEFINE(M_NTFSRUN, "NTFS vrun", "NTFS vrun storage");
67 1.1 jdolecek MALLOC_DEFINE(M_NTFSDECOMP, "NTFS decomp", "NTFS decompression temporary");
68 1.1 jdolecek
69 1.1 jdolecek /* Local struct used in ntfs_ntlookupfile() */
70 1.1 jdolecek struct ntfs_lookup_ctx {
71 1.1 jdolecek u_int32_t aoff;
72 1.1 jdolecek u_int32_t rdsize;
73 1.1 jdolecek cn_t cn;
74 1.1 jdolecek struct ntfs_lookup_ctx *prev;
75 1.1 jdolecek };
76 1.1 jdolecek
77 1.19 xtraeme static int ntfs_ntlookupattr(struct ntfsmount *, const char *, int,
78 1.19 xtraeme int *, char **);
79 1.19 xtraeme static int ntfs_findvattr(struct ntfsmount *, struct ntnode *,
80 1.19 xtraeme struct ntvattr **, struct ntvattr **, u_int32_t, const char *,
81 1.19 xtraeme size_t, cn_t);
82 1.19 xtraeme static int ntfs_uastricmp(struct ntfsmount *, const wchar *, size_t,
83 1.19 xtraeme const char *, size_t);
84 1.19 xtraeme static int ntfs_uastrcmp(struct ntfsmount *, const wchar *, size_t,
85 1.19 xtraeme const char *, size_t);
86 1.1 jdolecek
87 1.1 jdolecek /* table for mapping Unicode chars into uppercase; it's filled upon first
88 1.1 jdolecek * ntfs mount, freed upon last ntfs umount */
89 1.1 jdolecek static wchar *ntfs_toupper_tab;
90 1.1 jdolecek #define NTFS_U28(ch) ((((ch) & 0xE0) == 0) ? '_' : (ch) & 0xFF)
91 1.1 jdolecek #define NTFS_TOUPPER(ch) (ntfs_toupper_tab[(unsigned char)(ch)])
92 1.28 ad static kmutex_t ntfs_toupper_lock;
93 1.1 jdolecek static signed int ntfs_toupper_usecount;
94 1.1 jdolecek
95 1.1 jdolecek /* support macro for ntfs_ntvattrget() */
96 1.1 jdolecek #define NTFS_AALPCMP(aalp,type,name,namelen) ( \
97 1.1 jdolecek (aalp->al_type == type) && (aalp->al_namelen == namelen) && \
98 1.1 jdolecek !ntfs_uastrcmp(ntmp, aalp->al_name,aalp->al_namelen,name,namelen) )
99 1.1 jdolecek
100 1.1 jdolecek /*
101 1.14 perry *
102 1.1 jdolecek */
103 1.1 jdolecek int
104 1.1 jdolecek ntfs_ntvattrrele(vap)
105 1.1 jdolecek struct ntvattr * vap;
106 1.1 jdolecek {
107 1.20 christos dprintf(("ntfs_ntvattrrele: ino: %llu, type: 0x%x\n",
108 1.20 christos (unsigned long long)vap->va_ip->i_number, vap->va_type));
109 1.1 jdolecek
110 1.1 jdolecek ntfs_ntrele(vap->va_ip);
111 1.1 jdolecek
112 1.1 jdolecek return (0);
113 1.1 jdolecek }
114 1.1 jdolecek
115 1.1 jdolecek /*
116 1.1 jdolecek * find the attribute in the ntnode
117 1.1 jdolecek */
118 1.1 jdolecek static int
119 1.1 jdolecek ntfs_findvattr(ntmp, ip, lvapp, vapp, type, name, namelen, vcn)
120 1.1 jdolecek struct ntfsmount *ntmp;
121 1.1 jdolecek struct ntnode *ip;
122 1.1 jdolecek struct ntvattr **lvapp, **vapp;
123 1.1 jdolecek u_int32_t type;
124 1.1 jdolecek const char *name;
125 1.1 jdolecek size_t namelen;
126 1.1 jdolecek cn_t vcn;
127 1.1 jdolecek {
128 1.1 jdolecek int error;
129 1.1 jdolecek struct ntvattr *vap;
130 1.1 jdolecek
131 1.1 jdolecek if((ip->i_flag & IN_LOADED) == 0) {
132 1.20 christos dprintf(("ntfs_findvattr: node not loaded, ino: %llu\n",
133 1.20 christos (unsigned long long)ip->i_number));
134 1.1 jdolecek error = ntfs_loadntnode(ntmp,ip);
135 1.1 jdolecek if (error) {
136 1.18 christos printf("ntfs_findvattr: FAILED TO LOAD INO: %llu\n",
137 1.18 christos (unsigned long long)ip->i_number);
138 1.1 jdolecek return (error);
139 1.1 jdolecek }
140 1.1 jdolecek }
141 1.1 jdolecek
142 1.1 jdolecek *lvapp = NULL;
143 1.1 jdolecek *vapp = NULL;
144 1.1 jdolecek for (vap = ip->i_valist.lh_first; vap; vap = vap->va_list.le_next) {
145 1.20 christos ddprintf(("ntfs_findvattr: type: 0x%x, vcn: %qu - %qu\n",
146 1.20 christos vap->va_type, (long long) vap->va_vcnstart,
147 1.17 christos (long long) vap->va_vcnend));
148 1.1 jdolecek if ((vap->va_type == type) &&
149 1.1 jdolecek (vap->va_vcnstart <= vcn) && (vap->va_vcnend >= vcn) &&
150 1.1 jdolecek (vap->va_namelen == namelen) &&
151 1.1 jdolecek (strncmp(name, vap->va_name, namelen) == 0)) {
152 1.1 jdolecek *vapp = vap;
153 1.1 jdolecek ntfs_ntref(vap->va_ip);
154 1.1 jdolecek return (0);
155 1.1 jdolecek }
156 1.1 jdolecek if (vap->va_type == NTFS_A_ATTRLIST)
157 1.1 jdolecek *lvapp = vap;
158 1.1 jdolecek }
159 1.1 jdolecek
160 1.1 jdolecek return (-1);
161 1.1 jdolecek }
162 1.1 jdolecek
163 1.1 jdolecek /*
164 1.2 wiz * Search attribute specified in ntnode (load ntnode if necessary).
165 1.2 wiz * If not found but ATTR_A_ATTRLIST present, read it in and search through.
166 1.2 wiz * VOP_VGET node needed, and lookup through its ntnode (load if nessesary).
167 1.1 jdolecek *
168 1.1 jdolecek * ntnode should be locked
169 1.1 jdolecek */
170 1.1 jdolecek int
171 1.1 jdolecek ntfs_ntvattrget(
172 1.1 jdolecek struct ntfsmount * ntmp,
173 1.1 jdolecek struct ntnode * ip,
174 1.1 jdolecek u_int32_t type,
175 1.1 jdolecek const char *name,
176 1.1 jdolecek cn_t vcn,
177 1.1 jdolecek struct ntvattr ** vapp)
178 1.1 jdolecek {
179 1.1 jdolecek struct ntvattr *lvap = NULL;
180 1.1 jdolecek struct attr_attrlist *aalp;
181 1.1 jdolecek struct attr_attrlist *nextaalp;
182 1.1 jdolecek struct vnode *newvp;
183 1.1 jdolecek struct ntnode *newip;
184 1.1 jdolecek caddr_t alpool;
185 1.1 jdolecek size_t namelen, len;
186 1.1 jdolecek int error;
187 1.1 jdolecek
188 1.1 jdolecek *vapp = NULL;
189 1.1 jdolecek
190 1.1 jdolecek if (name) {
191 1.20 christos dprintf(("ntfs_ntvattrget: "
192 1.20 christos "ino: %llu, type: 0x%x, name: %s, vcn: %qu\n",
193 1.20 christos (unsigned long long)ip->i_number, type, name,
194 1.20 christos (long long)vcn));
195 1.1 jdolecek namelen = strlen(name);
196 1.1 jdolecek } else {
197 1.20 christos dprintf(("ntfs_ntvattrget: "
198 1.20 christos "ino: %llu, type: 0x%x, vcn: %qu\n",
199 1.20 christos (unsigned long long)ip->i_number, type, (long long)vcn));
200 1.1 jdolecek name = "";
201 1.1 jdolecek namelen = 0;
202 1.1 jdolecek }
203 1.1 jdolecek
204 1.1 jdolecek error = ntfs_findvattr(ntmp, ip, &lvap, vapp, type, name, namelen, vcn);
205 1.1 jdolecek if (error >= 0)
206 1.1 jdolecek return (error);
207 1.1 jdolecek
208 1.1 jdolecek if (!lvap) {
209 1.20 christos dprintf(("ntfs_ntvattrget: UNEXISTED ATTRIBUTE: "
210 1.20 christos "ino: %llu, type: 0x%x, name: %s, vcn: %qu\n",
211 1.20 christos (unsigned long long)ip->i_number, type, name,
212 1.20 christos (long long)vcn));
213 1.1 jdolecek return (ENOENT);
214 1.1 jdolecek }
215 1.1 jdolecek /* Scan $ATTRIBUTE_LIST for requested attribute */
216 1.1 jdolecek len = lvap->va_datalen;
217 1.1 jdolecek alpool = (caddr_t) malloc(len, M_TEMP, M_WAITOK);
218 1.1 jdolecek error = ntfs_readntvattr_plain(ntmp, ip, lvap, 0, len, alpool, &len,
219 1.1 jdolecek NULL);
220 1.1 jdolecek if (error)
221 1.1 jdolecek goto out;
222 1.1 jdolecek
223 1.1 jdolecek aalp = (struct attr_attrlist *) alpool;
224 1.1 jdolecek nextaalp = NULL;
225 1.1 jdolecek
226 1.1 jdolecek for(; len > 0; aalp = nextaalp) {
227 1.23 christos KASSERT(aalp != NULL);
228 1.20 christos dprintf(("ntfs_ntvattrget: "
229 1.20 christos "attrlist: ino: %d, attr: 0x%x, vcn: %qu\n",
230 1.20 christos aalp->al_inumber, aalp->al_type,
231 1.17 christos (long long) aalp->al_vcnstart));
232 1.1 jdolecek
233 1.1 jdolecek if (len > aalp->reclen) {
234 1.1 jdolecek nextaalp = NTFS_NEXTREC(aalp, struct attr_attrlist *);
235 1.1 jdolecek } else {
236 1.1 jdolecek nextaalp = NULL;
237 1.1 jdolecek }
238 1.1 jdolecek len -= aalp->reclen;
239 1.1 jdolecek
240 1.1 jdolecek if (!NTFS_AALPCMP(aalp, type, name, namelen) ||
241 1.1 jdolecek (nextaalp && (nextaalp->al_vcnstart <= vcn) &&
242 1.1 jdolecek NTFS_AALPCMP(nextaalp, type, name, namelen)))
243 1.1 jdolecek continue;
244 1.1 jdolecek
245 1.1 jdolecek dprintf(("ntfs_ntvattrget: attribute in ino: %d\n",
246 1.1 jdolecek aalp->al_inumber));
247 1.1 jdolecek
248 1.1 jdolecek /* this is not a main record, so we can't use just plain
249 1.1 jdolecek vget() */
250 1.1 jdolecek error = ntfs_vgetex(ntmp->ntm_mountp, aalp->al_inumber,
251 1.1 jdolecek NTFS_A_DATA, NULL, LK_EXCLUSIVE,
252 1.21 christos VG_EXT, &newvp);
253 1.1 jdolecek if (error) {
254 1.1 jdolecek printf("ntfs_ntvattrget: CAN'T VGET INO: %d\n",
255 1.1 jdolecek aalp->al_inumber);
256 1.1 jdolecek goto out;
257 1.1 jdolecek }
258 1.1 jdolecek newip = VTONT(newvp);
259 1.1 jdolecek /* XXX have to lock ntnode */
260 1.1 jdolecek error = ntfs_findvattr(ntmp, newip, &lvap, vapp,
261 1.1 jdolecek type, name, namelen, vcn);
262 1.1 jdolecek vput(newvp);
263 1.1 jdolecek if (error == 0)
264 1.1 jdolecek goto out;
265 1.1 jdolecek printf("ntfs_ntvattrget: ATTRLIST ERROR.\n");
266 1.1 jdolecek break;
267 1.1 jdolecek }
268 1.1 jdolecek error = ENOENT;
269 1.1 jdolecek
270 1.20 christos dprintf(("ntfs_ntvattrget: UNEXISTED ATTRIBUTE: "
271 1.20 christos "ino: %llu, type: 0x%x, name: %.*s, vcn: %qu\n",
272 1.20 christos (unsigned long long)ip->i_number, type, (int)namelen,
273 1.20 christos name, (long long)vcn));
274 1.1 jdolecek out:
275 1.1 jdolecek free(alpool, M_TEMP);
276 1.1 jdolecek return (error);
277 1.1 jdolecek }
278 1.1 jdolecek
279 1.1 jdolecek /*
280 1.1 jdolecek * Read ntnode from disk, make ntvattr list.
281 1.1 jdolecek *
282 1.1 jdolecek * ntnode should be locked
283 1.1 jdolecek */
284 1.1 jdolecek int
285 1.1 jdolecek ntfs_loadntnode(
286 1.1 jdolecek struct ntfsmount * ntmp,
287 1.1 jdolecek struct ntnode * ip)
288 1.1 jdolecek {
289 1.1 jdolecek struct filerec *mfrp;
290 1.1 jdolecek daddr_t bn;
291 1.1 jdolecek int error,off;
292 1.1 jdolecek struct attr *ap;
293 1.1 jdolecek struct ntvattr *nvap;
294 1.1 jdolecek
295 1.20 christos dprintf(("ntfs_loadntnode: loading ino: %llu\n",
296 1.20 christos (unsigned long long)ip->i_number));
297 1.1 jdolecek
298 1.1 jdolecek mfrp = (struct filerec *) malloc(ntfs_bntob(ntmp->ntm_bpmftrec),
299 1.1 jdolecek M_TEMP, M_WAITOK);
300 1.1 jdolecek
301 1.1 jdolecek if (ip->i_number < NTFS_SYSNODESNUM) {
302 1.1 jdolecek struct buf *bp;
303 1.1 jdolecek
304 1.1 jdolecek dprintf(("ntfs_loadntnode: read system node\n"));
305 1.1 jdolecek
306 1.1 jdolecek bn = ntfs_cntobn(ntmp->ntm_mftcn) +
307 1.1 jdolecek ntmp->ntm_bpmftrec * ip->i_number;
308 1.1 jdolecek
309 1.1 jdolecek error = bread(ntmp->ntm_devvp,
310 1.1 jdolecek bn, ntfs_bntob(ntmp->ntm_bpmftrec),
311 1.1 jdolecek NOCRED, &bp);
312 1.1 jdolecek if (error) {
313 1.1 jdolecek printf("ntfs_loadntnode: BREAD FAILED\n");
314 1.1 jdolecek brelse(bp);
315 1.1 jdolecek goto out;
316 1.1 jdolecek }
317 1.1 jdolecek memcpy(mfrp, bp->b_data, ntfs_bntob(ntmp->ntm_bpmftrec));
318 1.1 jdolecek bqrelse(bp);
319 1.1 jdolecek } else {
320 1.1 jdolecek struct vnode *vp;
321 1.1 jdolecek
322 1.1 jdolecek vp = ntmp->ntm_sysvn[NTFS_MFTINO];
323 1.1 jdolecek error = ntfs_readattr(ntmp, VTONT(vp), NTFS_A_DATA, NULL,
324 1.1 jdolecek ip->i_number * ntfs_bntob(ntmp->ntm_bpmftrec),
325 1.1 jdolecek ntfs_bntob(ntmp->ntm_bpmftrec), mfrp, NULL);
326 1.1 jdolecek if (error) {
327 1.1 jdolecek printf("ntfs_loadntnode: ntfs_readattr failed\n");
328 1.1 jdolecek goto out;
329 1.1 jdolecek }
330 1.1 jdolecek }
331 1.1 jdolecek
332 1.1 jdolecek /* Check if magic and fixups are correct */
333 1.1 jdolecek error = ntfs_procfixups(ntmp, NTFS_FILEMAGIC, (caddr_t)mfrp,
334 1.1 jdolecek ntfs_bntob(ntmp->ntm_bpmftrec));
335 1.1 jdolecek if (error) {
336 1.1 jdolecek printf("ntfs_loadntnode: BAD MFT RECORD %d\n",
337 1.1 jdolecek (u_int32_t) ip->i_number);
338 1.1 jdolecek goto out;
339 1.1 jdolecek }
340 1.1 jdolecek
341 1.20 christos dprintf(("ntfs_loadntnode: load attrs for ino: %llu\n",
342 1.20 christos (unsigned long long)ip->i_number));
343 1.1 jdolecek off = mfrp->fr_attroff;
344 1.1 jdolecek ap = (struct attr *) ((caddr_t)mfrp + off);
345 1.1 jdolecek
346 1.1 jdolecek LIST_INIT(&ip->i_valist);
347 1.14 perry
348 1.1 jdolecek while (ap->a_hdr.a_type != -1) {
349 1.1 jdolecek error = ntfs_attrtontvattr(ntmp, &nvap, ap);
350 1.1 jdolecek if (error)
351 1.1 jdolecek break;
352 1.1 jdolecek nvap->va_ip = ip;
353 1.1 jdolecek
354 1.1 jdolecek LIST_INSERT_HEAD(&ip->i_valist, nvap, va_list);
355 1.1 jdolecek
356 1.1 jdolecek off += ap->a_hdr.reclen;
357 1.1 jdolecek ap = (struct attr *) ((caddr_t)mfrp + off);
358 1.1 jdolecek }
359 1.1 jdolecek if (error) {
360 1.18 christos printf("ntfs_loadntnode: failed to load attr ino: %llu\n",
361 1.18 christos (unsigned long long)ip->i_number);
362 1.1 jdolecek goto out;
363 1.1 jdolecek }
364 1.1 jdolecek
365 1.1 jdolecek ip->i_mainrec = mfrp->fr_mainrec;
366 1.1 jdolecek ip->i_nlink = mfrp->fr_nlink;
367 1.1 jdolecek ip->i_frflag = mfrp->fr_flags;
368 1.1 jdolecek
369 1.1 jdolecek ip->i_flag |= IN_LOADED;
370 1.1 jdolecek
371 1.1 jdolecek out:
372 1.1 jdolecek free(mfrp, M_TEMP);
373 1.1 jdolecek return (error);
374 1.1 jdolecek }
375 1.14 perry
376 1.1 jdolecek /*
377 1.1 jdolecek * Routine locks ntnode and increase usecount, just opposite of
378 1.1 jdolecek * ntfs_ntput().
379 1.1 jdolecek */
380 1.1 jdolecek int
381 1.1 jdolecek ntfs_ntget(ip)
382 1.1 jdolecek struct ntnode *ip;
383 1.1 jdolecek {
384 1.20 christos dprintf(("ntfs_ntget: get ntnode %llu: %p, usecount: %d\n",
385 1.20 christos (unsigned long long)ip->i_number, ip, ip->i_usecount));
386 1.1 jdolecek
387 1.1 jdolecek simple_lock(&ip->i_interlock);
388 1.1 jdolecek ip->i_usecount++;
389 1.1 jdolecek lockmgr(&ip->i_lock, LK_EXCLUSIVE | LK_INTERLOCK, &ip->i_interlock);
390 1.1 jdolecek
391 1.1 jdolecek return 0;
392 1.1 jdolecek }
393 1.1 jdolecek
394 1.1 jdolecek /*
395 1.1 jdolecek * Routine search ntnode in hash, if found: lock, inc usecount and return.
396 1.1 jdolecek * If not in hash allocate structure for ntnode, prefill it, lock,
397 1.1 jdolecek * inc count and return.
398 1.1 jdolecek *
399 1.1 jdolecek * ntnode returned locked
400 1.1 jdolecek */
401 1.1 jdolecek int
402 1.1 jdolecek ntfs_ntlookup(
403 1.1 jdolecek struct ntfsmount * ntmp,
404 1.1 jdolecek ino_t ino,
405 1.1 jdolecek struct ntnode ** ipp)
406 1.1 jdolecek {
407 1.1 jdolecek struct ntnode *ip;
408 1.1 jdolecek
409 1.20 christos dprintf(("ntfs_ntlookup: looking for ntnode %llu\n",
410 1.20 christos (unsigned long long)ino));
411 1.1 jdolecek
412 1.28 ad if ((ip = ntfs_nthashlookup(ntmp->ntm_dev, ino)) != NULL) {
413 1.28 ad ntfs_ntget(ip);
414 1.28 ad dprintf(("ntfs_ntlookup: ntnode %llu: %p,"
415 1.28 ad " usecount: %d\n",
416 1.28 ad (unsigned long long)ino, ip, ip->i_usecount));
417 1.28 ad *ipp = ip;
418 1.28 ad return (0);
419 1.28 ad }
420 1.1 jdolecek
421 1.1 jdolecek MALLOC(ip, struct ntnode *, sizeof(struct ntnode),
422 1.1 jdolecek M_NTFSNTNODE, M_WAITOK);
423 1.20 christos ddprintf(("ntfs_ntlookup: allocating ntnode: %llu: %p\n",
424 1.20 christos (unsigned long long)ino, ip));
425 1.9 jrf bzero(ip, sizeof(struct ntnode));
426 1.1 jdolecek
427 1.28 ad mutex_enter(&ntfs_hashlock);
428 1.28 ad if ((ip = ntfs_nthashlookup(ntmp->ntm_dev, ino)) != NULL) {
429 1.28 ad mutex_exit(&ntfs_hashlock);
430 1.28 ad ntfs_ntget(ip);
431 1.28 ad FREE(ip, M_NTFSNTNODE);
432 1.28 ad dprintf(("ntfs_ntlookup: ntnode %llu: %p,"
433 1.28 ad " usecount: %d\n",
434 1.28 ad (unsigned long long)ino, ip, ip->i_usecount));
435 1.28 ad *ipp = ip;
436 1.28 ad return (0);
437 1.28 ad }
438 1.28 ad
439 1.1 jdolecek /* Generic initialization */
440 1.1 jdolecek ip->i_devvp = ntmp->ntm_devvp;
441 1.1 jdolecek ip->i_dev = ntmp->ntm_dev;
442 1.1 jdolecek ip->i_number = ino;
443 1.1 jdolecek ip->i_mp = ntmp;
444 1.1 jdolecek
445 1.1 jdolecek LIST_INIT(&ip->i_fnlist);
446 1.1 jdolecek
447 1.1 jdolecek /* init lock and lock the newborn ntnode */
448 1.1 jdolecek lockinit(&ip->i_lock, PINOD, "ntnode", 0, LK_EXCLUSIVE);
449 1.1 jdolecek simple_lock_init(&ip->i_interlock);
450 1.1 jdolecek ntfs_ntget(ip);
451 1.1 jdolecek
452 1.1 jdolecek ntfs_nthashins(ip);
453 1.1 jdolecek
454 1.28 ad mutex_exit(&ntfs_hashlock);
455 1.1 jdolecek
456 1.1 jdolecek *ipp = ip;
457 1.1 jdolecek
458 1.20 christos dprintf(("ntfs_ntlookup: ntnode %llu: %p, usecount: %d\n",
459 1.20 christos (unsigned long long)ino, ip, ip->i_usecount));
460 1.1 jdolecek
461 1.1 jdolecek return (0);
462 1.1 jdolecek }
463 1.1 jdolecek
464 1.1 jdolecek /*
465 1.1 jdolecek * Decrement usecount of ntnode and unlock it, if usecount reach zero,
466 1.1 jdolecek * deallocate ntnode.
467 1.1 jdolecek *
468 1.1 jdolecek * ntnode should be locked on entry, and unlocked on return.
469 1.1 jdolecek */
470 1.1 jdolecek void
471 1.1 jdolecek ntfs_ntput(ip)
472 1.1 jdolecek struct ntnode *ip;
473 1.1 jdolecek {
474 1.1 jdolecek struct ntvattr *vap;
475 1.1 jdolecek
476 1.20 christos dprintf(("ntfs_ntput: rele ntnode %llu: %p, usecount: %d\n",
477 1.20 christos (unsigned long long)ip->i_number, ip, ip->i_usecount));
478 1.1 jdolecek
479 1.1 jdolecek simple_lock(&ip->i_interlock);
480 1.1 jdolecek ip->i_usecount--;
481 1.1 jdolecek
482 1.1 jdolecek #ifdef DIAGNOSTIC
483 1.1 jdolecek if (ip->i_usecount < 0) {
484 1.18 christos panic("ntfs_ntput: ino: %llu usecount: %d ",
485 1.18 christos (unsigned long long)ip->i_number, ip->i_usecount);
486 1.1 jdolecek }
487 1.1 jdolecek #endif
488 1.1 jdolecek
489 1.1 jdolecek lockmgr(&ip->i_lock, LK_RELEASE|LK_INTERLOCK, &ip->i_interlock);
490 1.1 jdolecek
491 1.1 jdolecek if (ip->i_usecount == 0) {
492 1.20 christos dprintf(("ntfs_ntput: deallocating ntnode: %llu\n",
493 1.20 christos (unsigned long long)ip->i_number));
494 1.1 jdolecek
495 1.1 jdolecek if (ip->i_fnlist.lh_first)
496 1.1 jdolecek panic("ntfs_ntput: ntnode has fnodes");
497 1.1 jdolecek
498 1.1 jdolecek ntfs_nthashrem(ip);
499 1.1 jdolecek
500 1.1 jdolecek while (ip->i_valist.lh_first != NULL) {
501 1.1 jdolecek vap = ip->i_valist.lh_first;
502 1.1 jdolecek LIST_REMOVE(vap,va_list);
503 1.1 jdolecek ntfs_freentvattr(vap);
504 1.1 jdolecek }
505 1.1 jdolecek FREE(ip, M_NTFSNTNODE);
506 1.1 jdolecek }
507 1.1 jdolecek }
508 1.1 jdolecek
509 1.1 jdolecek /*
510 1.14 perry * increment usecount of ntnode
511 1.1 jdolecek */
512 1.1 jdolecek void
513 1.1 jdolecek ntfs_ntref(ip)
514 1.1 jdolecek struct ntnode *ip;
515 1.1 jdolecek {
516 1.1 jdolecek simple_lock(&ip->i_interlock);
517 1.1 jdolecek ip->i_usecount++;
518 1.1 jdolecek simple_unlock(&ip->i_interlock);
519 1.1 jdolecek
520 1.20 christos dprintf(("ntfs_ntref: ino %llu, usecount: %d\n",
521 1.20 christos (unsigned long long)ip->i_number, ip->i_usecount));
522 1.14 perry
523 1.1 jdolecek }
524 1.1 jdolecek
525 1.1 jdolecek /*
526 1.1 jdolecek * Decrement usecount of ntnode.
527 1.1 jdolecek */
528 1.1 jdolecek void
529 1.1 jdolecek ntfs_ntrele(ip)
530 1.1 jdolecek struct ntnode *ip;
531 1.1 jdolecek {
532 1.20 christos dprintf(("ntfs_ntrele: rele ntnode %llu: %p, usecount: %d\n",
533 1.20 christos (unsigned long long)ip->i_number, ip, ip->i_usecount));
534 1.1 jdolecek
535 1.1 jdolecek simple_lock(&ip->i_interlock);
536 1.1 jdolecek ip->i_usecount--;
537 1.1 jdolecek
538 1.1 jdolecek if (ip->i_usecount < 0)
539 1.18 christos panic("ntfs_ntrele: ino: %llu usecount: %d ",
540 1.18 christos (unsigned long long)ip->i_number, ip->i_usecount);
541 1.1 jdolecek simple_unlock(&ip->i_interlock);
542 1.1 jdolecek }
543 1.1 jdolecek
544 1.1 jdolecek /*
545 1.1 jdolecek * Deallocate all memory allocated for ntvattr
546 1.1 jdolecek */
547 1.1 jdolecek void
548 1.1 jdolecek ntfs_freentvattr(vap)
549 1.1 jdolecek struct ntvattr * vap;
550 1.1 jdolecek {
551 1.1 jdolecek if (vap->va_flag & NTFS_AF_INRUN) {
552 1.1 jdolecek if (vap->va_vruncn)
553 1.1 jdolecek free(vap->va_vruncn, M_NTFSRUN);
554 1.1 jdolecek if (vap->va_vruncl)
555 1.1 jdolecek free(vap->va_vruncl, M_NTFSRUN);
556 1.1 jdolecek } else {
557 1.1 jdolecek if (vap->va_datap)
558 1.1 jdolecek free(vap->va_datap, M_NTFSRDATA);
559 1.1 jdolecek }
560 1.1 jdolecek FREE(vap, M_NTFSNTVATTR);
561 1.1 jdolecek }
562 1.1 jdolecek
563 1.1 jdolecek /*
564 1.1 jdolecek * Convert disk image of attribute into ntvattr structure,
565 1.1 jdolecek * runs are expanded also.
566 1.1 jdolecek */
567 1.1 jdolecek int
568 1.1 jdolecek ntfs_attrtontvattr(
569 1.1 jdolecek struct ntfsmount * ntmp,
570 1.1 jdolecek struct ntvattr ** rvapp,
571 1.1 jdolecek struct attr * rap)
572 1.1 jdolecek {
573 1.1 jdolecek int error, i;
574 1.1 jdolecek struct ntvattr *vap;
575 1.1 jdolecek
576 1.1 jdolecek error = 0;
577 1.1 jdolecek *rvapp = NULL;
578 1.1 jdolecek
579 1.1 jdolecek MALLOC(vap, struct ntvattr *, sizeof(struct ntvattr),
580 1.1 jdolecek M_NTFSNTVATTR, M_WAITOK);
581 1.1 jdolecek bzero(vap, sizeof(struct ntvattr));
582 1.1 jdolecek vap->va_ip = NULL;
583 1.1 jdolecek vap->va_flag = rap->a_hdr.a_flag;
584 1.1 jdolecek vap->va_type = rap->a_hdr.a_type;
585 1.1 jdolecek vap->va_compression = rap->a_hdr.a_compression;
586 1.1 jdolecek vap->va_index = rap->a_hdr.a_index;
587 1.1 jdolecek
588 1.1 jdolecek ddprintf(("type: 0x%x, index: %d", vap->va_type, vap->va_index));
589 1.1 jdolecek
590 1.1 jdolecek vap->va_namelen = rap->a_hdr.a_namelen;
591 1.1 jdolecek if (rap->a_hdr.a_namelen) {
592 1.1 jdolecek wchar *unp = (wchar *) ((caddr_t) rap + rap->a_hdr.a_nameoff);
593 1.1 jdolecek ddprintf((", name:["));
594 1.1 jdolecek for (i = 0; i < vap->va_namelen; i++) {
595 1.1 jdolecek vap->va_name[i] = unp[i];
596 1.1 jdolecek ddprintf(("%c", vap->va_name[i]));
597 1.1 jdolecek }
598 1.1 jdolecek ddprintf(("]"));
599 1.1 jdolecek }
600 1.1 jdolecek if (vap->va_flag & NTFS_AF_INRUN) {
601 1.1 jdolecek ddprintf((", nonres."));
602 1.1 jdolecek vap->va_datalen = rap->a_nr.a_datalen;
603 1.1 jdolecek vap->va_allocated = rap->a_nr.a_allocated;
604 1.1 jdolecek vap->va_vcnstart = rap->a_nr.a_vcnstart;
605 1.1 jdolecek vap->va_vcnend = rap->a_nr.a_vcnend;
606 1.1 jdolecek vap->va_compressalg = rap->a_nr.a_compressalg;
607 1.1 jdolecek error = ntfs_runtovrun(&(vap->va_vruncn), &(vap->va_vruncl),
608 1.1 jdolecek &(vap->va_vruncnt),
609 1.24 mrg (u_int8_t *) rap + rap->a_nr.a_dataoff);
610 1.1 jdolecek } else {
611 1.1 jdolecek vap->va_compressalg = 0;
612 1.1 jdolecek ddprintf((", res."));
613 1.1 jdolecek vap->va_datalen = rap->a_r.a_datalen;
614 1.1 jdolecek vap->va_allocated = rap->a_r.a_datalen;
615 1.1 jdolecek vap->va_vcnstart = 0;
616 1.1 jdolecek vap->va_vcnend = ntfs_btocn(vap->va_allocated);
617 1.1 jdolecek vap->va_datap = (caddr_t) malloc(vap->va_datalen,
618 1.1 jdolecek M_NTFSRDATA, M_WAITOK);
619 1.1 jdolecek memcpy(vap->va_datap, (caddr_t) rap + rap->a_r.a_dataoff,
620 1.1 jdolecek rap->a_r.a_datalen);
621 1.1 jdolecek }
622 1.17 christos ddprintf((", len: %qu", (long long)vap->va_datalen));
623 1.1 jdolecek
624 1.1 jdolecek if (error)
625 1.1 jdolecek FREE(vap, M_NTFSNTVATTR);
626 1.1 jdolecek else
627 1.1 jdolecek *rvapp = vap;
628 1.1 jdolecek
629 1.1 jdolecek ddprintf(("\n"));
630 1.1 jdolecek
631 1.1 jdolecek return (error);
632 1.1 jdolecek }
633 1.1 jdolecek
634 1.1 jdolecek /*
635 1.1 jdolecek * Expand run into more utilizable and more memory eating format.
636 1.1 jdolecek */
637 1.1 jdolecek int
638 1.1 jdolecek ntfs_runtovrun(
639 1.1 jdolecek cn_t ** rcnp,
640 1.1 jdolecek cn_t ** rclp,
641 1.1 jdolecek u_long * rcntp,
642 1.1 jdolecek u_int8_t * run)
643 1.1 jdolecek {
644 1.1 jdolecek u_int32_t off;
645 1.1 jdolecek u_int32_t sz, i;
646 1.1 jdolecek cn_t *cn;
647 1.1 jdolecek cn_t *cl;
648 1.1 jdolecek u_long cnt;
649 1.1 jdolecek cn_t prev;
650 1.1 jdolecek cn_t tmp;
651 1.1 jdolecek
652 1.1 jdolecek off = 0;
653 1.1 jdolecek cnt = 0;
654 1.1 jdolecek i = 0;
655 1.1 jdolecek while (run[off]) {
656 1.1 jdolecek off += (run[off] & 0xF) + ((run[off] >> 4) & 0xF) + 1;
657 1.1 jdolecek cnt++;
658 1.1 jdolecek }
659 1.1 jdolecek cn = (cn_t *) malloc(cnt * sizeof(cn_t), M_NTFSRUN, M_WAITOK);
660 1.1 jdolecek cl = (cn_t *) malloc(cnt * sizeof(cn_t), M_NTFSRUN, M_WAITOK);
661 1.1 jdolecek
662 1.1 jdolecek off = 0;
663 1.1 jdolecek cnt = 0;
664 1.1 jdolecek prev = 0;
665 1.1 jdolecek while (run[off]) {
666 1.1 jdolecek
667 1.1 jdolecek sz = run[off++];
668 1.1 jdolecek cl[cnt] = 0;
669 1.1 jdolecek
670 1.1 jdolecek for (i = 0; i < (sz & 0xF); i++)
671 1.1 jdolecek cl[cnt] += (u_int32_t) run[off++] << (i << 3);
672 1.1 jdolecek
673 1.1 jdolecek sz >>= 4;
674 1.1 jdolecek if (run[off + sz - 1] & 0x80) {
675 1.1 jdolecek tmp = ((u_int64_t) - 1) << (sz << 3);
676 1.1 jdolecek for (i = 0; i < sz; i++)
677 1.1 jdolecek tmp |= (u_int64_t) run[off++] << (i << 3);
678 1.1 jdolecek } else {
679 1.1 jdolecek tmp = 0;
680 1.1 jdolecek for (i = 0; i < sz; i++)
681 1.1 jdolecek tmp |= (u_int64_t) run[off++] << (i << 3);
682 1.1 jdolecek }
683 1.1 jdolecek if (tmp)
684 1.1 jdolecek prev = cn[cnt] = prev + tmp;
685 1.1 jdolecek else
686 1.1 jdolecek cn[cnt] = tmp;
687 1.1 jdolecek
688 1.1 jdolecek cnt++;
689 1.1 jdolecek }
690 1.1 jdolecek *rcnp = cn;
691 1.1 jdolecek *rclp = cl;
692 1.1 jdolecek *rcntp = cnt;
693 1.1 jdolecek return (0);
694 1.1 jdolecek }
695 1.1 jdolecek
696 1.1 jdolecek /*
697 1.1 jdolecek * Compare unicode and ascii string case insens.
698 1.1 jdolecek */
699 1.1 jdolecek static int
700 1.1 jdolecek ntfs_uastricmp(ntmp, ustr, ustrlen, astr, astrlen)
701 1.1 jdolecek struct ntfsmount *ntmp;
702 1.1 jdolecek const wchar *ustr;
703 1.1 jdolecek size_t ustrlen;
704 1.1 jdolecek const char *astr;
705 1.1 jdolecek size_t astrlen;
706 1.1 jdolecek {
707 1.1 jdolecek size_t i;
708 1.11 jdolecek int res;
709 1.1 jdolecek
710 1.11 jdolecek for (i = 0; i < ustrlen && astrlen > 0; i++) {
711 1.1 jdolecek res = (*ntmp->ntm_wcmp)(NTFS_TOUPPER(ustr[i]),
712 1.11 jdolecek NTFS_TOUPPER((*ntmp->ntm_wget)(&astr, &astrlen)) );
713 1.1 jdolecek if (res)
714 1.1 jdolecek return res;
715 1.1 jdolecek }
716 1.1 jdolecek
717 1.11 jdolecek if (i == ustrlen && astrlen == 0)
718 1.1 jdolecek return 0;
719 1.1 jdolecek else if (i == ustrlen)
720 1.1 jdolecek return -1;
721 1.1 jdolecek else
722 1.1 jdolecek return 1;
723 1.1 jdolecek }
724 1.1 jdolecek
725 1.1 jdolecek /*
726 1.1 jdolecek * Compare unicode and ascii string case sens.
727 1.1 jdolecek */
728 1.1 jdolecek static int
729 1.1 jdolecek ntfs_uastrcmp(ntmp, ustr, ustrlen, astr, astrlen)
730 1.1 jdolecek struct ntfsmount *ntmp;
731 1.1 jdolecek const wchar *ustr;
732 1.1 jdolecek size_t ustrlen;
733 1.1 jdolecek const char *astr;
734 1.1 jdolecek size_t astrlen;
735 1.1 jdolecek {
736 1.11 jdolecek size_t i;
737 1.11 jdolecek int res;
738 1.1 jdolecek
739 1.11 jdolecek for (i = 0; (i < ustrlen) && astrlen > 0; i++) {
740 1.11 jdolecek res = (*ntmp->ntm_wcmp)(ustr[i],
741 1.11 jdolecek (*ntmp->ntm_wget)(&astr, &astrlen));
742 1.1 jdolecek if (res)
743 1.1 jdolecek return res;
744 1.1 jdolecek }
745 1.1 jdolecek
746 1.11 jdolecek if (i == ustrlen && astrlen == 0)
747 1.1 jdolecek return 0;
748 1.1 jdolecek else if (i == ustrlen)
749 1.1 jdolecek return -1;
750 1.1 jdolecek else
751 1.1 jdolecek return 1;
752 1.1 jdolecek }
753 1.1 jdolecek
754 1.14 perry /*
755 1.1 jdolecek * Search fnode in ntnode, if not found allocate and preinitialize.
756 1.1 jdolecek *
757 1.1 jdolecek * ntnode should be locked on entry.
758 1.1 jdolecek */
759 1.1 jdolecek int
760 1.1 jdolecek ntfs_fget(
761 1.27 christos struct ntfsmount *ntmp,
762 1.26 christos struct ntnode *ip,
763 1.26 christos int attrtype,
764 1.26 christos char *attrname,
765 1.26 christos struct fnode **fpp
766 1.26 christos )
767 1.1 jdolecek {
768 1.1 jdolecek struct fnode *fp;
769 1.1 jdolecek
770 1.20 christos dprintf(("ntfs_fget: ino: %llu, attrtype: 0x%x, attrname: %s\n",
771 1.20 christos (unsigned long long)ip->i_number, attrtype, attrname?attrname:""));
772 1.1 jdolecek *fpp = NULL;
773 1.1 jdolecek for (fp = ip->i_fnlist.lh_first; fp != NULL; fp = fp->f_fnlist.le_next){
774 1.1 jdolecek dprintf(("ntfs_fget: fnode: attrtype: %d, attrname: %s\n",
775 1.1 jdolecek fp->f_attrtype, fp->f_attrname?fp->f_attrname:""));
776 1.1 jdolecek
777 1.14 perry if ((attrtype == fp->f_attrtype) &&
778 1.1 jdolecek ((!attrname && !fp->f_attrname) ||
779 1.1 jdolecek (attrname && fp->f_attrname &&
780 1.1 jdolecek !strcmp(attrname,fp->f_attrname)))){
781 1.1 jdolecek dprintf(("ntfs_fget: found existed: %p\n",fp));
782 1.1 jdolecek *fpp = fp;
783 1.1 jdolecek }
784 1.1 jdolecek }
785 1.1 jdolecek
786 1.1 jdolecek if (*fpp)
787 1.1 jdolecek return (0);
788 1.1 jdolecek
789 1.1 jdolecek MALLOC(fp, struct fnode *, sizeof(struct fnode), M_NTFSFNODE, M_WAITOK);
790 1.1 jdolecek bzero(fp, sizeof(struct fnode));
791 1.1 jdolecek dprintf(("ntfs_fget: allocating fnode: %p\n",fp));
792 1.1 jdolecek
793 1.1 jdolecek fp->f_ip = ip;
794 1.1 jdolecek fp->f_attrname = attrname;
795 1.1 jdolecek if (fp->f_attrname) fp->f_flag |= FN_AATTRNAME;
796 1.1 jdolecek fp->f_attrtype = attrtype;
797 1.1 jdolecek
798 1.1 jdolecek ntfs_ntref(ip);
799 1.1 jdolecek
800 1.1 jdolecek LIST_INSERT_HEAD(&ip->i_fnlist, fp, f_fnlist);
801 1.1 jdolecek
802 1.1 jdolecek *fpp = fp;
803 1.1 jdolecek
804 1.1 jdolecek return (0);
805 1.1 jdolecek }
806 1.1 jdolecek
807 1.1 jdolecek /*
808 1.1 jdolecek * Deallocate fnode, remove it from ntnode's fnode list.
809 1.1 jdolecek *
810 1.1 jdolecek * ntnode should be locked.
811 1.1 jdolecek */
812 1.1 jdolecek void
813 1.1 jdolecek ntfs_frele(
814 1.1 jdolecek struct fnode *fp)
815 1.1 jdolecek {
816 1.1 jdolecek struct ntnode *ip = FTONT(fp);
817 1.1 jdolecek
818 1.20 christos dprintf(("ntfs_frele: fnode: %p for %llu: %p\n", fp,
819 1.20 christos (unsigned long long)ip->i_number, ip));
820 1.1 jdolecek
821 1.1 jdolecek dprintf(("ntfs_frele: deallocating fnode\n"));
822 1.1 jdolecek LIST_REMOVE(fp,f_fnlist);
823 1.1 jdolecek if (fp->f_flag & FN_AATTRNAME)
824 1.1 jdolecek FREE(fp->f_attrname, M_TEMP);
825 1.1 jdolecek if (fp->f_dirblbuf)
826 1.1 jdolecek FREE(fp->f_dirblbuf, M_NTFSDIR);
827 1.1 jdolecek FREE(fp, M_NTFSFNODE);
828 1.1 jdolecek ntfs_ntrele(ip);
829 1.1 jdolecek }
830 1.1 jdolecek
831 1.1 jdolecek /*
832 1.14 perry * Lookup attribute name in format: [[:$ATTR_TYPE]:$ATTR_NAME],
833 1.1 jdolecek * $ATTR_TYPE is searched in attrdefs read from $AttrDefs.
834 1.2 wiz * If $ATTR_TYPE not specified, ATTR_A_DATA assumed.
835 1.1 jdolecek */
836 1.1 jdolecek static int
837 1.1 jdolecek ntfs_ntlookupattr(
838 1.1 jdolecek struct ntfsmount * ntmp,
839 1.1 jdolecek const char * name,
840 1.1 jdolecek int namelen,
841 1.1 jdolecek int *attrtype,
842 1.1 jdolecek char **attrname)
843 1.1 jdolecek {
844 1.1 jdolecek const char *sys;
845 1.1 jdolecek size_t syslen, i;
846 1.1 jdolecek struct ntvattrdef *adp;
847 1.1 jdolecek
848 1.1 jdolecek if (namelen == 0)
849 1.1 jdolecek return (0);
850 1.1 jdolecek
851 1.1 jdolecek if (name[0] == '$') {
852 1.1 jdolecek sys = name;
853 1.1 jdolecek for (syslen = 0; syslen < namelen; syslen++) {
854 1.1 jdolecek if(sys[syslen] == ':') {
855 1.1 jdolecek name++;
856 1.1 jdolecek namelen--;
857 1.1 jdolecek break;
858 1.1 jdolecek }
859 1.1 jdolecek }
860 1.1 jdolecek name += syslen;
861 1.1 jdolecek namelen -= syslen;
862 1.1 jdolecek
863 1.1 jdolecek adp = ntmp->ntm_ad;
864 1.1 jdolecek for (i = 0; i < ntmp->ntm_adnum; i++, adp++){
865 1.14 perry if (syslen != adp->ad_namelen ||
866 1.1 jdolecek strncmp(sys, adp->ad_name, syslen) != 0)
867 1.1 jdolecek continue;
868 1.1 jdolecek
869 1.1 jdolecek *attrtype = adp->ad_type;
870 1.1 jdolecek goto out;
871 1.1 jdolecek }
872 1.1 jdolecek return (ENOENT);
873 1.1 jdolecek }
874 1.1 jdolecek
875 1.1 jdolecek out:
876 1.1 jdolecek if (namelen) {
877 1.1 jdolecek *attrname = (char *) malloc(namelen, M_TEMP, M_WAITOK);
878 1.1 jdolecek memcpy((*attrname), name, namelen);
879 1.1 jdolecek (*attrname)[namelen] = '\0';
880 1.1 jdolecek *attrtype = NTFS_A_DATA;
881 1.1 jdolecek }
882 1.1 jdolecek
883 1.1 jdolecek return (0);
884 1.1 jdolecek }
885 1.1 jdolecek
886 1.1 jdolecek /*
887 1.2 wiz * Lookup specified node for filename, matching cnp,
888 1.1 jdolecek * return fnode filled.
889 1.1 jdolecek */
890 1.1 jdolecek int
891 1.1 jdolecek ntfs_ntlookupfile(
892 1.1 jdolecek struct ntfsmount * ntmp,
893 1.1 jdolecek struct vnode * vp,
894 1.1 jdolecek struct componentname * cnp,
895 1.1 jdolecek struct vnode ** vpp)
896 1.1 jdolecek {
897 1.1 jdolecek struct fnode *fp = VTOF(vp);
898 1.1 jdolecek struct ntnode *ip = FTONT(fp);
899 1.1 jdolecek struct ntvattr *vap; /* Root attribute */
900 1.1 jdolecek cn_t cn = 0; /* VCN in current attribute */
901 1.1 jdolecek caddr_t rdbuf; /* Buffer to read directory's blocks */
902 1.1 jdolecek u_int32_t blsize;
903 1.1 jdolecek u_int32_t rdsize; /* Length of data to read from current block */
904 1.1 jdolecek struct attr_indexentry *iep;
905 1.1 jdolecek int error, res, anamelen, fnamelen;
906 1.1 jdolecek const char *fname,*aname;
907 1.1 jdolecek u_int32_t aoff;
908 1.1 jdolecek int attrtype = NTFS_A_DATA;
909 1.1 jdolecek char *attrname = NULL;
910 1.1 jdolecek struct fnode *nfp;
911 1.1 jdolecek struct vnode *nvp;
912 1.1 jdolecek enum vtype f_type;
913 1.1 jdolecek int fullscan = 0;
914 1.1 jdolecek struct ntfs_lookup_ctx *lookup_ctx = NULL, *tctx;
915 1.1 jdolecek
916 1.1 jdolecek error = ntfs_ntget(ip);
917 1.1 jdolecek if (error)
918 1.1 jdolecek return (error);
919 1.1 jdolecek
920 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, NTFS_A_INDXROOT, "$I30", 0, &vap);
921 1.1 jdolecek if (error || (vap->va_flag & NTFS_AF_INRUN))
922 1.1 jdolecek return (ENOTDIR);
923 1.1 jdolecek
924 1.1 jdolecek /*
925 1.1 jdolecek * Divide file name into: foofilefoofilefoofile[:attrspec]
926 1.1 jdolecek * Store like this: fname:fnamelen [aname:anamelen]
927 1.1 jdolecek */
928 1.1 jdolecek fname = cnp->cn_nameptr;
929 1.1 jdolecek aname = NULL;
930 1.1 jdolecek anamelen = 0;
931 1.1 jdolecek for (fnamelen = 0; fnamelen < cnp->cn_namelen; fnamelen++)
932 1.1 jdolecek if(fname[fnamelen] == ':') {
933 1.1 jdolecek aname = fname + fnamelen + 1;
934 1.1 jdolecek anamelen = cnp->cn_namelen - fnamelen - 1;
935 1.1 jdolecek dprintf(("ntfs_ntlookupfile: %s (%d), attr: %s (%d)\n",
936 1.1 jdolecek fname, fnamelen, aname, anamelen));
937 1.1 jdolecek break;
938 1.1 jdolecek }
939 1.1 jdolecek
940 1.1 jdolecek blsize = vap->va_a_iroot->ir_size;
941 1.1 jdolecek dprintf(("ntfs_ntlookupfile: blksz: %d\n", blsize));
942 1.1 jdolecek
943 1.1 jdolecek rdbuf = (caddr_t) malloc(blsize, M_TEMP, M_WAITOK);
944 1.1 jdolecek
945 1.1 jdolecek loop:
946 1.1 jdolecek rdsize = vap->va_datalen;
947 1.1 jdolecek dprintf(("ntfs_ntlookupfile: rdsz: %d\n", rdsize));
948 1.1 jdolecek
949 1.1 jdolecek error = ntfs_readattr(ntmp, ip, NTFS_A_INDXROOT, "$I30",
950 1.1 jdolecek 0, rdsize, rdbuf, NULL);
951 1.1 jdolecek if (error)
952 1.1 jdolecek goto fail;
953 1.1 jdolecek
954 1.1 jdolecek aoff = sizeof(struct attr_indexroot);
955 1.1 jdolecek
956 1.1 jdolecek do {
957 1.1 jdolecek iep = (struct attr_indexentry *) (rdbuf + aoff);
958 1.1 jdolecek
959 1.1 jdolecek for (; !(iep->ie_flag & NTFS_IEFLAG_LAST) && (rdsize > aoff);
960 1.1 jdolecek aoff += iep->reclen,
961 1.1 jdolecek iep = (struct attr_indexentry *) (rdbuf + aoff))
962 1.1 jdolecek {
963 1.1 jdolecek ddprintf(("scan: %d, %d\n",
964 1.1 jdolecek (u_int32_t) iep->ie_number,
965 1.1 jdolecek (u_int32_t) iep->ie_fnametype));
966 1.14 perry
967 1.2 wiz /* check the name - the case-insensitive check
968 1.1 jdolecek * has to come first, to break from this for loop
969 1.1 jdolecek * if needed, so we can dive correctly */
970 1.1 jdolecek res = ntfs_uastricmp(ntmp, iep->ie_fname,
971 1.1 jdolecek iep->ie_fnamelen, fname, fnamelen);
972 1.1 jdolecek if (!fullscan) {
973 1.1 jdolecek if (res > 0) break;
974 1.1 jdolecek if (res < 0) continue;
975 1.1 jdolecek }
976 1.1 jdolecek
977 1.1 jdolecek if (iep->ie_fnametype == 0 ||
978 1.1 jdolecek !(ntmp->ntm_flag & NTFS_MFLAG_CASEINS))
979 1.1 jdolecek {
980 1.1 jdolecek res = ntfs_uastrcmp(ntmp, iep->ie_fname,
981 1.1 jdolecek iep->ie_fnamelen, fname, fnamelen);
982 1.1 jdolecek if (res != 0 && !fullscan) continue;
983 1.1 jdolecek }
984 1.1 jdolecek
985 1.1 jdolecek /* if we perform full scan, the file does not match
986 1.1 jdolecek * and this is subnode, dive */
987 1.1 jdolecek if (fullscan && res != 0) {
988 1.1 jdolecek if (iep->ie_flag & NTFS_IEFLAG_SUBNODE) {
989 1.1 jdolecek MALLOC(tctx, struct ntfs_lookup_ctx *,
990 1.1 jdolecek sizeof(struct ntfs_lookup_ctx),
991 1.1 jdolecek M_TEMP, M_WAITOK);
992 1.1 jdolecek tctx->aoff = aoff + iep->reclen;
993 1.1 jdolecek tctx->rdsize = rdsize;
994 1.1 jdolecek tctx->cn = cn;
995 1.1 jdolecek tctx->prev = lookup_ctx;
996 1.1 jdolecek lookup_ctx = tctx;
997 1.1 jdolecek break;
998 1.1 jdolecek } else
999 1.1 jdolecek continue;
1000 1.1 jdolecek }
1001 1.1 jdolecek
1002 1.1 jdolecek if (aname) {
1003 1.1 jdolecek error = ntfs_ntlookupattr(ntmp,
1004 1.1 jdolecek aname, anamelen,
1005 1.1 jdolecek &attrtype, &attrname);
1006 1.1 jdolecek if (error)
1007 1.1 jdolecek goto fail;
1008 1.1 jdolecek }
1009 1.1 jdolecek
1010 1.2 wiz /* Check if we've found ourselves */
1011 1.1 jdolecek if ((iep->ie_number == ip->i_number) &&
1012 1.1 jdolecek (attrtype == fp->f_attrtype) &&
1013 1.1 jdolecek ((!attrname && !fp->f_attrname) ||
1014 1.1 jdolecek (attrname && fp->f_attrname &&
1015 1.1 jdolecek !strcmp(attrname, fp->f_attrname))))
1016 1.1 jdolecek {
1017 1.1 jdolecek VREF(vp);
1018 1.1 jdolecek *vpp = vp;
1019 1.1 jdolecek error = 0;
1020 1.1 jdolecek goto fail;
1021 1.1 jdolecek }
1022 1.1 jdolecek
1023 1.1 jdolecek /* free the buffer returned by ntfs_ntlookupattr() */
1024 1.1 jdolecek if (attrname) {
1025 1.1 jdolecek FREE(attrname, M_TEMP);
1026 1.1 jdolecek attrname = NULL;
1027 1.1 jdolecek }
1028 1.1 jdolecek
1029 1.1 jdolecek /* vget node, but don't load it */
1030 1.1 jdolecek error = ntfs_vgetex(ntmp->ntm_mountp,
1031 1.1 jdolecek iep->ie_number, attrtype, attrname,
1032 1.1 jdolecek LK_EXCLUSIVE, VG_DONTLOADIN | VG_DONTVALIDFN,
1033 1.21 christos &nvp);
1034 1.1 jdolecek if (error)
1035 1.1 jdolecek goto fail;
1036 1.1 jdolecek
1037 1.1 jdolecek nfp = VTOF(nvp);
1038 1.1 jdolecek
1039 1.1 jdolecek if (nfp->f_flag & FN_VALID) {
1040 1.1 jdolecek *vpp = nvp;
1041 1.1 jdolecek goto fail;
1042 1.1 jdolecek }
1043 1.1 jdolecek
1044 1.1 jdolecek nfp->f_fflag = iep->ie_fflag;
1045 1.1 jdolecek nfp->f_pnumber = iep->ie_fpnumber;
1046 1.1 jdolecek nfp->f_times = iep->ie_ftimes;
1047 1.1 jdolecek
1048 1.1 jdolecek if((nfp->f_fflag & NTFS_FFLAG_DIR) &&
1049 1.1 jdolecek (nfp->f_attrtype == NTFS_A_DATA) &&
1050 1.1 jdolecek (nfp->f_attrname == NULL))
1051 1.14 perry f_type = VDIR;
1052 1.1 jdolecek else
1053 1.14 perry f_type = VREG;
1054 1.1 jdolecek
1055 1.1 jdolecek nvp->v_type = f_type;
1056 1.1 jdolecek
1057 1.1 jdolecek if ((nfp->f_attrtype == NTFS_A_DATA) &&
1058 1.1 jdolecek (nfp->f_attrname == NULL))
1059 1.1 jdolecek {
1060 1.1 jdolecek /* Opening default attribute */
1061 1.1 jdolecek nfp->f_size = iep->ie_fsize;
1062 1.1 jdolecek nfp->f_allocated = iep->ie_fallocated;
1063 1.1 jdolecek nfp->f_flag |= FN_PRELOADED;
1064 1.1 jdolecek } else {
1065 1.1 jdolecek error = ntfs_filesize(ntmp, nfp,
1066 1.1 jdolecek &nfp->f_size, &nfp->f_allocated);
1067 1.1 jdolecek if (error) {
1068 1.1 jdolecek vput(nvp);
1069 1.1 jdolecek goto fail;
1070 1.1 jdolecek }
1071 1.1 jdolecek }
1072 1.1 jdolecek
1073 1.1 jdolecek nfp->f_flag &= ~FN_VALID;
1074 1.1 jdolecek *vpp = nvp;
1075 1.1 jdolecek goto fail;
1076 1.1 jdolecek }
1077 1.1 jdolecek
1078 1.1 jdolecek /* Dive if possible */
1079 1.1 jdolecek if (iep->ie_flag & NTFS_IEFLAG_SUBNODE) {
1080 1.1 jdolecek dprintf(("ntfs_ntlookupfile: diving\n"));
1081 1.1 jdolecek
1082 1.1 jdolecek cn = *(cn_t *) (rdbuf + aoff +
1083 1.1 jdolecek iep->reclen - sizeof(cn_t));
1084 1.1 jdolecek rdsize = blsize;
1085 1.1 jdolecek
1086 1.1 jdolecek error = ntfs_readattr(ntmp, ip, NTFS_A_INDX, "$I30",
1087 1.1 jdolecek ntfs_cntob(cn), rdsize, rdbuf, NULL);
1088 1.1 jdolecek if (error)
1089 1.1 jdolecek goto fail;
1090 1.1 jdolecek
1091 1.1 jdolecek error = ntfs_procfixups(ntmp, NTFS_INDXMAGIC,
1092 1.1 jdolecek rdbuf, rdsize);
1093 1.1 jdolecek if (error)
1094 1.1 jdolecek goto fail;
1095 1.1 jdolecek
1096 1.1 jdolecek aoff = (((struct attr_indexalloc *) rdbuf)->ia_hdrsize +
1097 1.1 jdolecek 0x18);
1098 1.1 jdolecek } else if (fullscan && lookup_ctx) {
1099 1.1 jdolecek cn = lookup_ctx->cn;
1100 1.1 jdolecek aoff = lookup_ctx->aoff;
1101 1.1 jdolecek rdsize = lookup_ctx->rdsize;
1102 1.1 jdolecek
1103 1.1 jdolecek error = ntfs_readattr(ntmp, ip,
1104 1.1 jdolecek (cn == 0) ? NTFS_A_INDXROOT : NTFS_A_INDX,
1105 1.1 jdolecek "$I30", ntfs_cntob(cn), rdsize, rdbuf, NULL);
1106 1.1 jdolecek if (error)
1107 1.1 jdolecek goto fail;
1108 1.14 perry
1109 1.1 jdolecek if (cn != 0) {
1110 1.1 jdolecek error = ntfs_procfixups(ntmp, NTFS_INDXMAGIC,
1111 1.1 jdolecek rdbuf, rdsize);
1112 1.1 jdolecek if (error)
1113 1.1 jdolecek goto fail;
1114 1.1 jdolecek }
1115 1.1 jdolecek
1116 1.1 jdolecek tctx = lookup_ctx;
1117 1.1 jdolecek lookup_ctx = lookup_ctx->prev;
1118 1.1 jdolecek FREE(tctx, M_TEMP);
1119 1.1 jdolecek } else {
1120 1.1 jdolecek dprintf(("ntfs_ntlookupfile: nowhere to dive :-(\n"));
1121 1.1 jdolecek error = ENOENT;
1122 1.1 jdolecek break;
1123 1.1 jdolecek }
1124 1.1 jdolecek } while (1);
1125 1.1 jdolecek
1126 1.1 jdolecek /* perform full scan if no entry was found */
1127 1.1 jdolecek if (!fullscan && error == ENOENT) {
1128 1.1 jdolecek fullscan = 1;
1129 1.1 jdolecek cn = 0; /* need zero, used by lookup_ctx */
1130 1.1 jdolecek
1131 1.1 jdolecek ddprintf(("ntfs_ntlookupfile: fullscan performed for: %.*s\n",
1132 1.1 jdolecek (int) fnamelen, fname));
1133 1.1 jdolecek goto loop;
1134 1.1 jdolecek }
1135 1.1 jdolecek
1136 1.1 jdolecek dprintf(("finish\n"));
1137 1.1 jdolecek
1138 1.1 jdolecek fail:
1139 1.1 jdolecek if (attrname)
1140 1.1 jdolecek FREE(attrname, M_TEMP);
1141 1.1 jdolecek if (lookup_ctx) {
1142 1.1 jdolecek while(lookup_ctx) {
1143 1.1 jdolecek tctx = lookup_ctx;
1144 1.1 jdolecek lookup_ctx = lookup_ctx->prev;
1145 1.1 jdolecek FREE(tctx, M_TEMP);
1146 1.1 jdolecek }
1147 1.1 jdolecek }
1148 1.1 jdolecek ntfs_ntvattrrele(vap);
1149 1.1 jdolecek ntfs_ntput(ip);
1150 1.1 jdolecek free(rdbuf, M_TEMP);
1151 1.1 jdolecek return (error);
1152 1.1 jdolecek }
1153 1.1 jdolecek
1154 1.1 jdolecek /*
1155 1.1 jdolecek * Check if name type is permitted to show.
1156 1.1 jdolecek */
1157 1.1 jdolecek int
1158 1.1 jdolecek ntfs_isnamepermitted(
1159 1.1 jdolecek struct ntfsmount * ntmp,
1160 1.1 jdolecek struct attr_indexentry * iep)
1161 1.1 jdolecek {
1162 1.1 jdolecek if (ntmp->ntm_flag & NTFS_MFLAG_ALLNAMES)
1163 1.1 jdolecek return 1;
1164 1.1 jdolecek
1165 1.1 jdolecek switch (iep->ie_fnametype) {
1166 1.1 jdolecek case 2:
1167 1.2 wiz ddprintf(("ntfs_isnamepermitted: skipped DOS name\n"));
1168 1.1 jdolecek return 0;
1169 1.1 jdolecek case 0: case 1: case 3:
1170 1.1 jdolecek return 1;
1171 1.1 jdolecek default:
1172 1.20 christos printf("ntfs_isnamepermitted: "
1173 1.20 christos "WARNING! Unknown file name type: %d\n",
1174 1.20 christos iep->ie_fnametype);
1175 1.1 jdolecek break;
1176 1.1 jdolecek }
1177 1.1 jdolecek return 0;
1178 1.1 jdolecek }
1179 1.1 jdolecek
1180 1.1 jdolecek /*
1181 1.1 jdolecek * Read ntfs dir like stream of attr_indexentry, not like btree of them.
1182 1.2 wiz * This is done by scanning $BITMAP:$I30 for busy clusters and reading them.
1183 1.2 wiz * Of course $INDEX_ROOT:$I30 is read before. Last read values are stored in
1184 1.2 wiz * fnode, so we can skip toward record number num almost immediately.
1185 1.1 jdolecek * Anyway this is rather slow routine. The problem is that we don't know
1186 1.1 jdolecek * how many records are there in $INDEX_ALLOCATION:$I30 block.
1187 1.1 jdolecek */
1188 1.1 jdolecek int
1189 1.1 jdolecek ntfs_ntreaddir(
1190 1.1 jdolecek struct ntfsmount * ntmp,
1191 1.1 jdolecek struct fnode * fp,
1192 1.1 jdolecek u_int32_t num,
1193 1.1 jdolecek struct attr_indexentry ** riepp)
1194 1.1 jdolecek {
1195 1.1 jdolecek struct ntnode *ip = FTONT(fp);
1196 1.1 jdolecek struct ntvattr *vap = NULL; /* IndexRoot attribute */
1197 1.1 jdolecek struct ntvattr *bmvap = NULL; /* BitMap attribute */
1198 1.1 jdolecek struct ntvattr *iavap = NULL; /* IndexAllocation attribute */
1199 1.1 jdolecek caddr_t rdbuf; /* Buffer to read directory's blocks */
1200 1.1 jdolecek u_char *bmp = NULL; /* Bitmap */
1201 1.1 jdolecek u_int32_t blsize; /* Index allocation size (2048) */
1202 1.1 jdolecek u_int32_t rdsize; /* Length of data to read */
1203 1.1 jdolecek u_int32_t attrnum; /* Current attribute type */
1204 1.1 jdolecek u_int32_t cpbl = 1; /* Clusters per directory block */
1205 1.1 jdolecek u_int32_t blnum;
1206 1.1 jdolecek struct attr_indexentry *iep;
1207 1.1 jdolecek int error = ENOENT;
1208 1.1 jdolecek u_int32_t aoff, cnum;
1209 1.1 jdolecek
1210 1.20 christos dprintf(("ntfs_ntreaddir: read ino: %llu, num: %d\n",
1211 1.20 christos (unsigned long long)ip->i_number, num));
1212 1.1 jdolecek error = ntfs_ntget(ip);
1213 1.1 jdolecek if (error)
1214 1.1 jdolecek return (error);
1215 1.1 jdolecek
1216 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, NTFS_A_INDXROOT, "$I30", 0, &vap);
1217 1.1 jdolecek if (error)
1218 1.1 jdolecek return (ENOTDIR);
1219 1.1 jdolecek
1220 1.1 jdolecek if (fp->f_dirblbuf == NULL) {
1221 1.1 jdolecek fp->f_dirblsz = vap->va_a_iroot->ir_size;
1222 1.1 jdolecek fp->f_dirblbuf = (caddr_t) malloc(
1223 1.13 jdolecek MAX(vap->va_datalen,fp->f_dirblsz), M_NTFSDIR, M_WAITOK);
1224 1.1 jdolecek }
1225 1.1 jdolecek
1226 1.1 jdolecek blsize = fp->f_dirblsz;
1227 1.1 jdolecek rdbuf = fp->f_dirblbuf;
1228 1.1 jdolecek
1229 1.15 christos dprintf(("ntfs_ntreaddir: rdbuf: %p, blsize: %d\n", rdbuf, blsize));
1230 1.1 jdolecek
1231 1.1 jdolecek if (vap->va_a_iroot->ir_flag & NTFS_IRFLAG_INDXALLOC) {
1232 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, NTFS_A_INDXBITMAP, "$I30",
1233 1.1 jdolecek 0, &bmvap);
1234 1.1 jdolecek if (error) {
1235 1.1 jdolecek error = ENOTDIR;
1236 1.1 jdolecek goto fail;
1237 1.1 jdolecek }
1238 1.1 jdolecek bmp = (u_char *) malloc(bmvap->va_datalen, M_TEMP, M_WAITOK);
1239 1.1 jdolecek error = ntfs_readattr(ntmp, ip, NTFS_A_INDXBITMAP, "$I30", 0,
1240 1.1 jdolecek bmvap->va_datalen, bmp, NULL);
1241 1.1 jdolecek if (error)
1242 1.1 jdolecek goto fail;
1243 1.1 jdolecek
1244 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, NTFS_A_INDX, "$I30",
1245 1.1 jdolecek 0, &iavap);
1246 1.1 jdolecek if (error) {
1247 1.1 jdolecek error = ENOTDIR;
1248 1.1 jdolecek goto fail;
1249 1.1 jdolecek }
1250 1.1 jdolecek cpbl = ntfs_btocn(blsize + ntfs_cntob(1) - 1);
1251 1.17 christos dprintf(("ntfs_ntreaddir: indexalloc: %qu, cpbl: %d\n",
1252 1.17 christos (long long)iavap->va_datalen, cpbl));
1253 1.1 jdolecek } else {
1254 1.1 jdolecek dprintf(("ntfs_ntreadidir: w/o BitMap and IndexAllocation\n"));
1255 1.1 jdolecek iavap = bmvap = NULL;
1256 1.1 jdolecek bmp = NULL;
1257 1.1 jdolecek }
1258 1.1 jdolecek
1259 1.1 jdolecek /* Try use previous values */
1260 1.1 jdolecek if ((fp->f_lastdnum < num) && (fp->f_lastdnum != 0)) {
1261 1.1 jdolecek attrnum = fp->f_lastdattr;
1262 1.1 jdolecek aoff = fp->f_lastdoff;
1263 1.1 jdolecek blnum = fp->f_lastdblnum;
1264 1.1 jdolecek cnum = fp->f_lastdnum;
1265 1.1 jdolecek } else {
1266 1.1 jdolecek attrnum = NTFS_A_INDXROOT;
1267 1.1 jdolecek aoff = sizeof(struct attr_indexroot);
1268 1.1 jdolecek blnum = 0;
1269 1.1 jdolecek cnum = 0;
1270 1.1 jdolecek }
1271 1.1 jdolecek
1272 1.1 jdolecek do {
1273 1.1 jdolecek dprintf(("ntfs_ntreaddir: scan: 0x%x, %d, %d, %d, %d\n",
1274 1.1 jdolecek attrnum, (u_int32_t) blnum, cnum, num, aoff));
1275 1.1 jdolecek rdsize = (attrnum == NTFS_A_INDXROOT) ? vap->va_datalen : blsize;
1276 1.1 jdolecek error = ntfs_readattr(ntmp, ip, attrnum, "$I30",
1277 1.1 jdolecek ntfs_cntob(blnum * cpbl), rdsize, rdbuf, NULL);
1278 1.1 jdolecek if (error)
1279 1.1 jdolecek goto fail;
1280 1.1 jdolecek
1281 1.1 jdolecek if (attrnum == NTFS_A_INDX) {
1282 1.1 jdolecek error = ntfs_procfixups(ntmp, NTFS_INDXMAGIC,
1283 1.1 jdolecek rdbuf, rdsize);
1284 1.1 jdolecek if (error)
1285 1.1 jdolecek goto fail;
1286 1.1 jdolecek }
1287 1.1 jdolecek if (aoff == 0)
1288 1.1 jdolecek aoff = (attrnum == NTFS_A_INDX) ?
1289 1.1 jdolecek (0x18 + ((struct attr_indexalloc *) rdbuf)->ia_hdrsize) :
1290 1.1 jdolecek sizeof(struct attr_indexroot);
1291 1.1 jdolecek
1292 1.1 jdolecek iep = (struct attr_indexentry *) (rdbuf + aoff);
1293 1.1 jdolecek for (; !(iep->ie_flag & NTFS_IEFLAG_LAST) && (rdsize > aoff);
1294 1.1 jdolecek aoff += iep->reclen,
1295 1.1 jdolecek iep = (struct attr_indexentry *) (rdbuf + aoff))
1296 1.1 jdolecek {
1297 1.1 jdolecek if (!ntfs_isnamepermitted(ntmp, iep)) continue;
1298 1.1 jdolecek
1299 1.1 jdolecek if (cnum >= num) {
1300 1.1 jdolecek fp->f_lastdnum = cnum;
1301 1.1 jdolecek fp->f_lastdoff = aoff;
1302 1.1 jdolecek fp->f_lastdblnum = blnum;
1303 1.1 jdolecek fp->f_lastdattr = attrnum;
1304 1.1 jdolecek
1305 1.1 jdolecek *riepp = iep;
1306 1.1 jdolecek
1307 1.1 jdolecek error = 0;
1308 1.1 jdolecek goto fail;
1309 1.1 jdolecek }
1310 1.1 jdolecek cnum++;
1311 1.1 jdolecek }
1312 1.1 jdolecek
1313 1.1 jdolecek if (iavap) {
1314 1.1 jdolecek if (attrnum == NTFS_A_INDXROOT)
1315 1.1 jdolecek blnum = 0;
1316 1.1 jdolecek else
1317 1.1 jdolecek blnum++;
1318 1.1 jdolecek
1319 1.1 jdolecek while (ntfs_cntob(blnum * cpbl) < iavap->va_datalen) {
1320 1.1 jdolecek if (bmp[blnum >> 3] & (1 << (blnum & 3)))
1321 1.1 jdolecek break;
1322 1.1 jdolecek blnum++;
1323 1.1 jdolecek }
1324 1.1 jdolecek
1325 1.1 jdolecek attrnum = NTFS_A_INDX;
1326 1.1 jdolecek aoff = 0;
1327 1.1 jdolecek if (ntfs_cntob(blnum * cpbl) >= iavap->va_datalen)
1328 1.1 jdolecek break;
1329 1.1 jdolecek dprintf(("ntfs_ntreaddir: blnum: %d\n", (u_int32_t) blnum));
1330 1.1 jdolecek }
1331 1.1 jdolecek } while (iavap);
1332 1.1 jdolecek
1333 1.1 jdolecek *riepp = NULL;
1334 1.1 jdolecek fp->f_lastdnum = 0;
1335 1.1 jdolecek
1336 1.1 jdolecek fail:
1337 1.1 jdolecek if (vap)
1338 1.1 jdolecek ntfs_ntvattrrele(vap);
1339 1.1 jdolecek if (bmvap)
1340 1.1 jdolecek ntfs_ntvattrrele(bmvap);
1341 1.1 jdolecek if (iavap)
1342 1.1 jdolecek ntfs_ntvattrrele(iavap);
1343 1.1 jdolecek if (bmp)
1344 1.1 jdolecek FREE(bmp, M_TEMP);
1345 1.1 jdolecek ntfs_ntput(ip);
1346 1.1 jdolecek return (error);
1347 1.1 jdolecek }
1348 1.1 jdolecek
1349 1.1 jdolecek /*
1350 1.1 jdolecek * Convert NTFS times that are in 100 ns units and begins from
1351 1.1 jdolecek * 1601 Jan 1 into unix times.
1352 1.1 jdolecek */
1353 1.1 jdolecek struct timespec
1354 1.1 jdolecek ntfs_nttimetounix(
1355 1.1 jdolecek u_int64_t nt)
1356 1.1 jdolecek {
1357 1.1 jdolecek struct timespec t;
1358 1.1 jdolecek
1359 1.1 jdolecek /* WindowNT times are in 100 ns and from 1601 Jan 1 */
1360 1.1 jdolecek t.tv_nsec = (nt % (1000 * 1000 * 10)) * 100;
1361 1.1 jdolecek t.tv_sec = nt / (1000 * 1000 * 10) -
1362 1.1 jdolecek 369LL * 365LL * 24LL * 60LL * 60LL -
1363 1.1 jdolecek 89LL * 1LL * 24LL * 60LL * 60LL;
1364 1.1 jdolecek return (t);
1365 1.1 jdolecek }
1366 1.1 jdolecek
1367 1.1 jdolecek /*
1368 1.1 jdolecek * Get file times from NTFS_A_NAME attribute.
1369 1.1 jdolecek */
1370 1.1 jdolecek int
1371 1.1 jdolecek ntfs_times(
1372 1.1 jdolecek struct ntfsmount * ntmp,
1373 1.1 jdolecek struct ntnode * ip,
1374 1.1 jdolecek ntfs_times_t * tm)
1375 1.1 jdolecek {
1376 1.1 jdolecek struct ntvattr *vap;
1377 1.1 jdolecek int error;
1378 1.1 jdolecek
1379 1.20 christos dprintf(("ntfs_times: ino: %llu...\n",
1380 1.20 christos (unsigned long long)ip->i_number));
1381 1.1 jdolecek
1382 1.1 jdolecek error = ntfs_ntget(ip);
1383 1.1 jdolecek if (error)
1384 1.1 jdolecek return (error);
1385 1.1 jdolecek
1386 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, NTFS_A_NAME, NULL, 0, &vap);
1387 1.1 jdolecek if (error) {
1388 1.1 jdolecek ntfs_ntput(ip);
1389 1.1 jdolecek return (error);
1390 1.1 jdolecek }
1391 1.1 jdolecek *tm = vap->va_a_name->n_times;
1392 1.1 jdolecek ntfs_ntvattrrele(vap);
1393 1.1 jdolecek ntfs_ntput(ip);
1394 1.1 jdolecek
1395 1.1 jdolecek return (0);
1396 1.1 jdolecek }
1397 1.1 jdolecek
1398 1.1 jdolecek /*
1399 1.14 perry * Get file sizes from corresponding attribute.
1400 1.14 perry *
1401 1.1 jdolecek * ntnode under fnode should be locked.
1402 1.1 jdolecek */
1403 1.1 jdolecek int
1404 1.1 jdolecek ntfs_filesize(
1405 1.1 jdolecek struct ntfsmount * ntmp,
1406 1.1 jdolecek struct fnode * fp,
1407 1.1 jdolecek u_int64_t * size,
1408 1.1 jdolecek u_int64_t * bytes)
1409 1.1 jdolecek {
1410 1.1 jdolecek struct ntvattr *vap;
1411 1.1 jdolecek struct ntnode *ip = FTONT(fp);
1412 1.1 jdolecek u_int64_t sz, bn;
1413 1.1 jdolecek int error;
1414 1.1 jdolecek
1415 1.20 christos dprintf(("ntfs_filesize: ino: %llu\n",
1416 1.20 christos (unsigned long long)ip->i_number));
1417 1.1 jdolecek
1418 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip,
1419 1.1 jdolecek fp->f_attrtype, fp->f_attrname, 0, &vap);
1420 1.1 jdolecek if (error)
1421 1.1 jdolecek return (error);
1422 1.1 jdolecek
1423 1.1 jdolecek bn = vap->va_allocated;
1424 1.1 jdolecek sz = vap->va_datalen;
1425 1.1 jdolecek
1426 1.1 jdolecek dprintf(("ntfs_filesize: %d bytes (%d bytes allocated)\n",
1427 1.1 jdolecek (u_int32_t) sz, (u_int32_t) bn));
1428 1.1 jdolecek
1429 1.1 jdolecek if (size)
1430 1.1 jdolecek *size = sz;
1431 1.1 jdolecek if (bytes)
1432 1.1 jdolecek *bytes = bn;
1433 1.1 jdolecek
1434 1.1 jdolecek ntfs_ntvattrrele(vap);
1435 1.1 jdolecek
1436 1.1 jdolecek return (0);
1437 1.1 jdolecek }
1438 1.1 jdolecek
1439 1.1 jdolecek /*
1440 1.1 jdolecek * This is one of write routine.
1441 1.1 jdolecek */
1442 1.1 jdolecek int
1443 1.1 jdolecek ntfs_writeattr_plain(
1444 1.1 jdolecek struct ntfsmount * ntmp,
1445 1.1 jdolecek struct ntnode * ip,
1446 1.14 perry u_int32_t attrnum,
1447 1.1 jdolecek char *attrname,
1448 1.1 jdolecek off_t roff,
1449 1.1 jdolecek size_t rsize,
1450 1.1 jdolecek void *rdata,
1451 1.1 jdolecek size_t * initp,
1452 1.1 jdolecek struct uio *uio)
1453 1.1 jdolecek {
1454 1.1 jdolecek size_t init;
1455 1.1 jdolecek int error = 0;
1456 1.1 jdolecek off_t off = roff, left = rsize, towrite;
1457 1.1 jdolecek caddr_t data = rdata;
1458 1.1 jdolecek struct ntvattr *vap;
1459 1.1 jdolecek *initp = 0;
1460 1.1 jdolecek
1461 1.1 jdolecek while (left) {
1462 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, attrnum, attrname,
1463 1.1 jdolecek ntfs_btocn(off), &vap);
1464 1.1 jdolecek if (error)
1465 1.1 jdolecek return (error);
1466 1.13 jdolecek towrite = MIN(left, ntfs_cntob(vap->va_vcnend + 1) - off);
1467 1.17 christos ddprintf(("ntfs_writeattr_plain: o: %qd, s: %qd (%qu - %qu)\n",
1468 1.17 christos (long long) off, (long long) towrite,
1469 1.17 christos (long long) vap->va_vcnstart,
1470 1.17 christos (long long) vap->va_vcnend));
1471 1.1 jdolecek error = ntfs_writentvattr_plain(ntmp, ip, vap,
1472 1.1 jdolecek off - ntfs_cntob(vap->va_vcnstart),
1473 1.1 jdolecek towrite, data, &init, uio);
1474 1.1 jdolecek if (error) {
1475 1.20 christos dprintf(("ntfs_writeattr_plain: "
1476 1.20 christos "ntfs_writentvattr_plain failed: o: %qd, s: %qd\n",
1477 1.20 christos (long long) off, (long long) towrite));
1478 1.17 christos dprintf(("ntfs_writeattr_plain: attrib: %qu - %qu\n",
1479 1.17 christos (long long) vap->va_vcnstart,
1480 1.17 christos (long long) vap->va_vcnend));
1481 1.1 jdolecek ntfs_ntvattrrele(vap);
1482 1.1 jdolecek break;
1483 1.1 jdolecek }
1484 1.1 jdolecek ntfs_ntvattrrele(vap);
1485 1.1 jdolecek left -= towrite;
1486 1.1 jdolecek off += towrite;
1487 1.1 jdolecek data = data + towrite;
1488 1.1 jdolecek *initp += init;
1489 1.1 jdolecek }
1490 1.1 jdolecek
1491 1.1 jdolecek return (error);
1492 1.1 jdolecek }
1493 1.1 jdolecek
1494 1.1 jdolecek /*
1495 1.1 jdolecek * This is one of write routine.
1496 1.1 jdolecek *
1497 1.1 jdolecek * ntnode should be locked.
1498 1.1 jdolecek */
1499 1.1 jdolecek int
1500 1.1 jdolecek ntfs_writentvattr_plain(
1501 1.1 jdolecek struct ntfsmount * ntmp,
1502 1.1 jdolecek struct ntnode * ip,
1503 1.1 jdolecek struct ntvattr * vap,
1504 1.1 jdolecek off_t roff,
1505 1.1 jdolecek size_t rsize,
1506 1.1 jdolecek void *rdata,
1507 1.1 jdolecek size_t * initp,
1508 1.1 jdolecek struct uio *uio)
1509 1.1 jdolecek {
1510 1.1 jdolecek int error = 0;
1511 1.17 christos off_t off;
1512 1.1 jdolecek int cnt;
1513 1.1 jdolecek cn_t ccn, ccl, cn, left, cl;
1514 1.1 jdolecek caddr_t data = rdata;
1515 1.1 jdolecek struct buf *bp;
1516 1.1 jdolecek size_t tocopy;
1517 1.1 jdolecek
1518 1.1 jdolecek *initp = 0;
1519 1.1 jdolecek
1520 1.1 jdolecek if ((vap->va_flag & NTFS_AF_INRUN) == 0) {
1521 1.4 jdolecek dprintf(("ntfs_writevattr_plain: CAN'T WRITE RES. ATTRIBUTE\n"));
1522 1.1 jdolecek return ENOTTY;
1523 1.1 jdolecek }
1524 1.1 jdolecek
1525 1.1 jdolecek ddprintf(("ntfs_writentvattr_plain: data in run: %lu chains\n",
1526 1.1 jdolecek vap->va_vruncnt));
1527 1.1 jdolecek
1528 1.1 jdolecek off = roff;
1529 1.1 jdolecek left = rsize;
1530 1.1 jdolecek ccl = 0;
1531 1.1 jdolecek ccn = 0;
1532 1.1 jdolecek cnt = 0;
1533 1.1 jdolecek for (; left && (cnt < vap->va_vruncnt); cnt++) {
1534 1.1 jdolecek ccn = vap->va_vruncn[cnt];
1535 1.1 jdolecek ccl = vap->va_vruncl[cnt];
1536 1.1 jdolecek
1537 1.20 christos ddprintf(("ntfs_writentvattr_plain: "
1538 1.20 christos "left %qu, cn: 0x%qx, cl: %qu, off: %qd\n",
1539 1.20 christos (long long) left, (long long) ccn,
1540 1.20 christos (long long) ccl, (long long) off));
1541 1.1 jdolecek
1542 1.1 jdolecek if (ntfs_cntob(ccl) < off) {
1543 1.1 jdolecek off -= ntfs_cntob(ccl);
1544 1.1 jdolecek cnt++;
1545 1.1 jdolecek continue;
1546 1.1 jdolecek }
1547 1.1 jdolecek if (!ccn && ip->i_number != NTFS_BOOTINO)
1548 1.1 jdolecek continue; /* XXX */
1549 1.1 jdolecek
1550 1.1 jdolecek ccl -= ntfs_btocn(off);
1551 1.1 jdolecek cn = ccn + ntfs_btocn(off);
1552 1.1 jdolecek off = ntfs_btocnoff(off);
1553 1.1 jdolecek
1554 1.1 jdolecek while (left && ccl) {
1555 1.12 jdolecek /*
1556 1.12 jdolecek * Always read and write single clusters at a time -
1557 1.12 jdolecek * we need to avoid requesting differently-sized
1558 1.12 jdolecek * blocks at the same disk offsets to avoid
1559 1.12 jdolecek * confusing the buffer cache.
1560 1.12 jdolecek */
1561 1.13 jdolecek tocopy = MIN(left, ntfs_cntob(1) - off);
1562 1.1 jdolecek cl = ntfs_btocl(tocopy + off);
1563 1.12 jdolecek KASSERT(cl == 1 && tocopy <= ntfs_cntob(1));
1564 1.20 christos ddprintf(("ntfs_writentvattr_plain: write: "
1565 1.17 christos "cn: 0x%qx cl: %qu, off: %qd len: %qu, left: %qu\n",
1566 1.17 christos (long long) cn, (long long) cl,
1567 1.17 christos (long long) off, (long long) tocopy,
1568 1.17 christos (long long) left));
1569 1.1 jdolecek if ((off == 0) && (tocopy == ntfs_cntob(cl)))
1570 1.1 jdolecek {
1571 1.1 jdolecek bp = getblk(ntmp->ntm_devvp, ntfs_cntobn(cn),
1572 1.1 jdolecek ntfs_cntob(cl), 0, 0);
1573 1.1 jdolecek clrbuf(bp);
1574 1.1 jdolecek } else {
1575 1.1 jdolecek error = bread(ntmp->ntm_devvp, ntfs_cntobn(cn),
1576 1.1 jdolecek ntfs_cntob(cl), NOCRED, &bp);
1577 1.1 jdolecek if (error) {
1578 1.1 jdolecek brelse(bp);
1579 1.1 jdolecek return (error);
1580 1.1 jdolecek }
1581 1.1 jdolecek }
1582 1.1 jdolecek if (uio)
1583 1.1 jdolecek uiomove(bp->b_data + off, tocopy, uio);
1584 1.1 jdolecek else
1585 1.1 jdolecek memcpy(bp->b_data + off, data, tocopy);
1586 1.1 jdolecek bawrite(bp);
1587 1.1 jdolecek data = data + tocopy;
1588 1.1 jdolecek *initp += tocopy;
1589 1.1 jdolecek off = 0;
1590 1.1 jdolecek left -= tocopy;
1591 1.1 jdolecek cn += cl;
1592 1.1 jdolecek ccl -= cl;
1593 1.1 jdolecek }
1594 1.1 jdolecek }
1595 1.1 jdolecek
1596 1.1 jdolecek if (left) {
1597 1.1 jdolecek printf("ntfs_writentvattr_plain: POSSIBLE RUN ERROR\n");
1598 1.1 jdolecek error = EINVAL;
1599 1.1 jdolecek }
1600 1.1 jdolecek
1601 1.1 jdolecek return (error);
1602 1.1 jdolecek }
1603 1.1 jdolecek
1604 1.1 jdolecek /*
1605 1.1 jdolecek * This is one of read routines.
1606 1.1 jdolecek *
1607 1.1 jdolecek * ntnode should be locked.
1608 1.1 jdolecek */
1609 1.1 jdolecek int
1610 1.1 jdolecek ntfs_readntvattr_plain(
1611 1.1 jdolecek struct ntfsmount * ntmp,
1612 1.1 jdolecek struct ntnode * ip,
1613 1.1 jdolecek struct ntvattr * vap,
1614 1.1 jdolecek off_t roff,
1615 1.1 jdolecek size_t rsize,
1616 1.1 jdolecek void *rdata,
1617 1.1 jdolecek size_t * initp,
1618 1.1 jdolecek struct uio *uio)
1619 1.1 jdolecek {
1620 1.1 jdolecek int error = 0;
1621 1.17 christos off_t off;
1622 1.1 jdolecek
1623 1.1 jdolecek *initp = 0;
1624 1.1 jdolecek if (vap->va_flag & NTFS_AF_INRUN) {
1625 1.1 jdolecek int cnt;
1626 1.1 jdolecek cn_t ccn, ccl, cn, left, cl;
1627 1.1 jdolecek caddr_t data = rdata;
1628 1.1 jdolecek struct buf *bp;
1629 1.1 jdolecek size_t tocopy;
1630 1.1 jdolecek
1631 1.1 jdolecek ddprintf(("ntfs_readntvattr_plain: data in run: %lu chains\n",
1632 1.1 jdolecek vap->va_vruncnt));
1633 1.1 jdolecek
1634 1.1 jdolecek off = roff;
1635 1.1 jdolecek left = rsize;
1636 1.1 jdolecek ccl = 0;
1637 1.1 jdolecek ccn = 0;
1638 1.1 jdolecek cnt = 0;
1639 1.1 jdolecek while (left && (cnt < vap->va_vruncnt)) {
1640 1.1 jdolecek ccn = vap->va_vruncn[cnt];
1641 1.1 jdolecek ccl = vap->va_vruncl[cnt];
1642 1.1 jdolecek
1643 1.20 christos ddprintf(("ntfs_readntvattr_plain: "
1644 1.20 christos "left %qu, cn: 0x%qx, cl: %qu, off: %qd\n",
1645 1.17 christos (long long) left, (long long) ccn,
1646 1.17 christos (long long) ccl, (long long) off));
1647 1.1 jdolecek
1648 1.1 jdolecek if (ntfs_cntob(ccl) < off) {
1649 1.1 jdolecek off -= ntfs_cntob(ccl);
1650 1.1 jdolecek cnt++;
1651 1.1 jdolecek continue;
1652 1.1 jdolecek }
1653 1.1 jdolecek if (ccn || ip->i_number == NTFS_BOOTINO) {
1654 1.1 jdolecek ccl -= ntfs_btocn(off);
1655 1.1 jdolecek cn = ccn + ntfs_btocn(off);
1656 1.1 jdolecek off = ntfs_btocnoff(off);
1657 1.1 jdolecek
1658 1.1 jdolecek while (left && ccl) {
1659 1.12 jdolecek /*
1660 1.12 jdolecek * Always read single clusters at a
1661 1.12 jdolecek * time - we need to avoid reading
1662 1.12 jdolecek * differently-sized blocks at the
1663 1.12 jdolecek * same disk offsets to avoid
1664 1.12 jdolecek * confusing the buffer cache.
1665 1.12 jdolecek */
1666 1.13 jdolecek tocopy = MIN(left,
1667 1.12 jdolecek ntfs_cntob(1) - off);
1668 1.1 jdolecek cl = ntfs_btocl(tocopy + off);
1669 1.12 jdolecek KASSERT(cl == 1 &&
1670 1.12 jdolecek tocopy <= ntfs_cntob(1));
1671 1.1 jdolecek
1672 1.20 christos ddprintf(("ntfs_readntvattr_plain: "
1673 1.20 christos "read: cn: 0x%qx cl: %qu, "
1674 1.17 christos "off: %qd len: %qu, left: %qu\n",
1675 1.17 christos (long long) cn,
1676 1.17 christos (long long) cl,
1677 1.17 christos (long long) off,
1678 1.17 christos (long long) tocopy,
1679 1.17 christos (long long) left));
1680 1.1 jdolecek error = bread(ntmp->ntm_devvp,
1681 1.1 jdolecek ntfs_cntobn(cn),
1682 1.1 jdolecek ntfs_cntob(cl),
1683 1.1 jdolecek NOCRED, &bp);
1684 1.1 jdolecek if (error) {
1685 1.1 jdolecek brelse(bp);
1686 1.1 jdolecek return (error);
1687 1.1 jdolecek }
1688 1.1 jdolecek if (uio) {
1689 1.1 jdolecek uiomove(bp->b_data + off,
1690 1.1 jdolecek tocopy, uio);
1691 1.1 jdolecek } else {
1692 1.1 jdolecek memcpy(data, bp->b_data + off,
1693 1.1 jdolecek tocopy);
1694 1.1 jdolecek }
1695 1.1 jdolecek brelse(bp);
1696 1.1 jdolecek data = data + tocopy;
1697 1.1 jdolecek *initp += tocopy;
1698 1.1 jdolecek off = 0;
1699 1.1 jdolecek left -= tocopy;
1700 1.1 jdolecek cn += cl;
1701 1.1 jdolecek ccl -= cl;
1702 1.1 jdolecek }
1703 1.1 jdolecek } else {
1704 1.13 jdolecek tocopy = MIN(left, ntfs_cntob(ccl) - off);
1705 1.1 jdolecek ddprintf(("ntfs_readntvattr_plain: "
1706 1.20 christos "hole: ccn: 0x%qx ccl: %qu, off: %qd, "
1707 1.17 christos " len: %qu, left: %qu\n",
1708 1.17 christos (long long) ccn, (long long) ccl,
1709 1.17 christos (long long) off, (long long) tocopy,
1710 1.17 christos (long long) left));
1711 1.1 jdolecek left -= tocopy;
1712 1.1 jdolecek off = 0;
1713 1.1 jdolecek if (uio) {
1714 1.16 christos char vbuf[] = "";
1715 1.1 jdolecek size_t remains = tocopy;
1716 1.1 jdolecek for(; remains; remains--)
1717 1.16 christos uiomove(vbuf, 1, uio);
1718 1.14 perry } else
1719 1.1 jdolecek bzero(data, tocopy);
1720 1.1 jdolecek data = data + tocopy;
1721 1.1 jdolecek }
1722 1.1 jdolecek cnt++;
1723 1.1 jdolecek }
1724 1.1 jdolecek if (left) {
1725 1.1 jdolecek printf("ntfs_readntvattr_plain: POSSIBLE RUN ERROR\n");
1726 1.1 jdolecek error = E2BIG;
1727 1.1 jdolecek }
1728 1.1 jdolecek } else {
1729 1.1 jdolecek ddprintf(("ntfs_readnvattr_plain: data is in mft record\n"));
1730 1.14 perry if (uio)
1731 1.1 jdolecek uiomove(vap->va_datap + roff, rsize, uio);
1732 1.1 jdolecek else
1733 1.1 jdolecek memcpy(rdata, vap->va_datap + roff, rsize);
1734 1.1 jdolecek *initp += rsize;
1735 1.1 jdolecek }
1736 1.1 jdolecek
1737 1.1 jdolecek return (error);
1738 1.1 jdolecek }
1739 1.1 jdolecek
1740 1.1 jdolecek /*
1741 1.1 jdolecek * This is one of read routines.
1742 1.1 jdolecek */
1743 1.1 jdolecek int
1744 1.1 jdolecek ntfs_readattr_plain(
1745 1.1 jdolecek struct ntfsmount * ntmp,
1746 1.1 jdolecek struct ntnode * ip,
1747 1.14 perry u_int32_t attrnum,
1748 1.16 christos const char *attrname,
1749 1.1 jdolecek off_t roff,
1750 1.1 jdolecek size_t rsize,
1751 1.1 jdolecek void *rdata,
1752 1.1 jdolecek size_t * initp,
1753 1.1 jdolecek struct uio *uio)
1754 1.1 jdolecek {
1755 1.1 jdolecek size_t init;
1756 1.1 jdolecek int error = 0;
1757 1.1 jdolecek off_t off = roff, left = rsize, toread;
1758 1.1 jdolecek caddr_t data = rdata;
1759 1.1 jdolecek struct ntvattr *vap;
1760 1.1 jdolecek *initp = 0;
1761 1.1 jdolecek
1762 1.1 jdolecek while (left) {
1763 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, attrnum, attrname,
1764 1.1 jdolecek ntfs_btocn(off), &vap);
1765 1.1 jdolecek if (error)
1766 1.1 jdolecek return (error);
1767 1.13 jdolecek toread = MIN(left, ntfs_cntob(vap->va_vcnend + 1) - off);
1768 1.17 christos ddprintf(("ntfs_readattr_plain: o: %qd, s: %qd (%qu - %qu)\n",
1769 1.17 christos (long long) off, (long long) toread,
1770 1.17 christos (long long) vap->va_vcnstart,
1771 1.17 christos (long long) vap->va_vcnend));
1772 1.1 jdolecek error = ntfs_readntvattr_plain(ntmp, ip, vap,
1773 1.1 jdolecek off - ntfs_cntob(vap->va_vcnstart),
1774 1.1 jdolecek toread, data, &init, uio);
1775 1.1 jdolecek if (error) {
1776 1.20 christos printf("ntfs_readattr_plain: "
1777 1.17 christos "ntfs_readntvattr_plain failed: o: %qd, s: %qd\n",
1778 1.17 christos (long long) off, (long long) toread);
1779 1.17 christos printf("ntfs_readattr_plain: attrib: %qu - %qu\n",
1780 1.17 christos (long long) vap->va_vcnstart,
1781 1.17 christos (long long) vap->va_vcnend);
1782 1.1 jdolecek ntfs_ntvattrrele(vap);
1783 1.1 jdolecek break;
1784 1.1 jdolecek }
1785 1.1 jdolecek ntfs_ntvattrrele(vap);
1786 1.1 jdolecek left -= toread;
1787 1.1 jdolecek off += toread;
1788 1.1 jdolecek data = data + toread;
1789 1.1 jdolecek *initp += init;
1790 1.1 jdolecek }
1791 1.1 jdolecek
1792 1.1 jdolecek return (error);
1793 1.1 jdolecek }
1794 1.1 jdolecek
1795 1.1 jdolecek /*
1796 1.1 jdolecek * This is one of read routines.
1797 1.1 jdolecek */
1798 1.1 jdolecek int
1799 1.1 jdolecek ntfs_readattr(
1800 1.1 jdolecek struct ntfsmount * ntmp,
1801 1.1 jdolecek struct ntnode * ip,
1802 1.1 jdolecek u_int32_t attrnum,
1803 1.16 christos const char *attrname,
1804 1.1 jdolecek off_t roff,
1805 1.1 jdolecek size_t rsize,
1806 1.1 jdolecek void *rdata,
1807 1.1 jdolecek struct uio *uio)
1808 1.1 jdolecek {
1809 1.1 jdolecek int error = 0;
1810 1.1 jdolecek struct ntvattr *vap;
1811 1.1 jdolecek size_t init;
1812 1.1 jdolecek
1813 1.20 christos ddprintf(("ntfs_readattr: reading %llu: 0x%x, from %qd size %qu"
1814 1.20 christos " bytes\n", (unsigned long long)ip->i_number, attrnum,
1815 1.20 christos (long long)roff, (long long)rsize));
1816 1.1 jdolecek
1817 1.1 jdolecek error = ntfs_ntvattrget(ntmp, ip, attrnum, attrname, 0, &vap);
1818 1.1 jdolecek if (error)
1819 1.1 jdolecek return (error);
1820 1.1 jdolecek
1821 1.1 jdolecek if ((roff > vap->va_datalen) ||
1822 1.1 jdolecek (roff + rsize > vap->va_datalen)) {
1823 1.17 christos printf("ntfs_readattr: offset too big: %qd (%qd) > %qu\n",
1824 1.17 christos (long long) roff, (long long) (roff + rsize),
1825 1.17 christos (long long) vap->va_datalen);
1826 1.1 jdolecek ntfs_ntvattrrele(vap);
1827 1.1 jdolecek return (E2BIG);
1828 1.1 jdolecek }
1829 1.1 jdolecek if (vap->va_compression && vap->va_compressalg) {
1830 1.1 jdolecek u_int8_t *cup;
1831 1.1 jdolecek u_int8_t *uup;
1832 1.1 jdolecek off_t off = roff, left = rsize, tocopy;
1833 1.1 jdolecek caddr_t data = rdata;
1834 1.1 jdolecek cn_t cn;
1835 1.1 jdolecek
1836 1.1 jdolecek ddprintf(("ntfs_ntreadattr: compression: %d\n",
1837 1.1 jdolecek vap->va_compressalg));
1838 1.1 jdolecek
1839 1.22 christos cup = malloc(ntfs_cntob(NTFS_COMPUNIT_CL),
1840 1.1 jdolecek M_NTFSDECOMP, M_WAITOK);
1841 1.22 christos uup = malloc(ntfs_cntob(NTFS_COMPUNIT_CL),
1842 1.1 jdolecek M_NTFSDECOMP, M_WAITOK);
1843 1.1 jdolecek
1844 1.1 jdolecek cn = (ntfs_btocn(roff)) & (~(NTFS_COMPUNIT_CL - 1));
1845 1.1 jdolecek off = roff - ntfs_cntob(cn);
1846 1.1 jdolecek
1847 1.1 jdolecek while (left) {
1848 1.1 jdolecek error = ntfs_readattr_plain(ntmp, ip, attrnum,
1849 1.1 jdolecek attrname, ntfs_cntob(cn),
1850 1.1 jdolecek ntfs_cntob(NTFS_COMPUNIT_CL),
1851 1.1 jdolecek cup, &init, NULL);
1852 1.1 jdolecek if (error)
1853 1.1 jdolecek break;
1854 1.1 jdolecek
1855 1.13 jdolecek tocopy = MIN(left, ntfs_cntob(NTFS_COMPUNIT_CL) - off);
1856 1.1 jdolecek
1857 1.1 jdolecek if (init == ntfs_cntob(NTFS_COMPUNIT_CL)) {
1858 1.1 jdolecek if (uio)
1859 1.1 jdolecek uiomove(cup + off, tocopy, uio);
1860 1.1 jdolecek else
1861 1.1 jdolecek memcpy(data, cup + off, tocopy);
1862 1.1 jdolecek } else if (init == 0) {
1863 1.1 jdolecek if (uio) {
1864 1.16 christos char vbuf[] = "";
1865 1.1 jdolecek size_t remains = tocopy;
1866 1.1 jdolecek for(; remains; remains--)
1867 1.16 christos uiomove(vbuf, 1, uio);
1868 1.1 jdolecek }
1869 1.1 jdolecek else
1870 1.1 jdolecek bzero(data, tocopy);
1871 1.1 jdolecek } else {
1872 1.1 jdolecek error = ntfs_uncompunit(ntmp, uup, cup);
1873 1.1 jdolecek if (error)
1874 1.1 jdolecek break;
1875 1.1 jdolecek if (uio)
1876 1.1 jdolecek uiomove(uup + off, tocopy, uio);
1877 1.1 jdolecek else
1878 1.1 jdolecek memcpy(data, uup + off, tocopy);
1879 1.1 jdolecek }
1880 1.1 jdolecek
1881 1.1 jdolecek left -= tocopy;
1882 1.1 jdolecek data = data + tocopy;
1883 1.1 jdolecek off += tocopy - ntfs_cntob(NTFS_COMPUNIT_CL);
1884 1.1 jdolecek cn += NTFS_COMPUNIT_CL;
1885 1.1 jdolecek }
1886 1.1 jdolecek
1887 1.1 jdolecek FREE(uup, M_NTFSDECOMP);
1888 1.1 jdolecek FREE(cup, M_NTFSDECOMP);
1889 1.1 jdolecek } else
1890 1.1 jdolecek error = ntfs_readattr_plain(ntmp, ip, attrnum, attrname,
1891 1.1 jdolecek roff, rsize, rdata, &init, uio);
1892 1.1 jdolecek ntfs_ntvattrrele(vap);
1893 1.1 jdolecek return (error);
1894 1.1 jdolecek }
1895 1.1 jdolecek
1896 1.1 jdolecek #if UNUSED_CODE
1897 1.1 jdolecek int
1898 1.1 jdolecek ntfs_parserun(
1899 1.1 jdolecek cn_t * cn,
1900 1.1 jdolecek cn_t * cl,
1901 1.1 jdolecek u_int8_t * run,
1902 1.1 jdolecek u_long len,
1903 1.1 jdolecek u_long *off)
1904 1.1 jdolecek {
1905 1.1 jdolecek u_int8_t sz;
1906 1.1 jdolecek int i;
1907 1.1 jdolecek
1908 1.1 jdolecek if (NULL == run) {
1909 1.1 jdolecek printf("ntfs_parsetun: run == NULL\n");
1910 1.1 jdolecek return (EINVAL);
1911 1.1 jdolecek }
1912 1.1 jdolecek sz = run[(*off)++];
1913 1.1 jdolecek if (0 == sz) {
1914 1.1 jdolecek printf("ntfs_parserun: trying to go out of run\n");
1915 1.1 jdolecek return (E2BIG);
1916 1.1 jdolecek }
1917 1.1 jdolecek *cl = 0;
1918 1.1 jdolecek if ((sz & 0xF) > 8 || (*off) + (sz & 0xF) > len) {
1919 1.20 christos printf("ntfs_parserun: "
1920 1.1 jdolecek "bad run: length too big: sz: 0x%02x (%ld < %ld + sz)\n",
1921 1.1 jdolecek sz, len, *off);
1922 1.1 jdolecek return (EINVAL);
1923 1.1 jdolecek }
1924 1.1 jdolecek for (i = 0; i < (sz & 0xF); i++)
1925 1.1 jdolecek *cl += (u_int32_t) run[(*off)++] << (i << 3);
1926 1.1 jdolecek
1927 1.1 jdolecek sz >>= 4;
1928 1.1 jdolecek if ((sz & 0xF) > 8 || (*off) + (sz & 0xF) > len) {
1929 1.20 christos printf("ntfs_parserun: "
1930 1.1 jdolecek "bad run: length too big: sz: 0x%02x (%ld < %ld + sz)\n",
1931 1.1 jdolecek sz, len, *off);
1932 1.1 jdolecek return (EINVAL);
1933 1.1 jdolecek }
1934 1.1 jdolecek for (i = 0; i < (sz & 0xF); i++)
1935 1.1 jdolecek *cn += (u_int32_t) run[(*off)++] << (i << 3);
1936 1.1 jdolecek
1937 1.1 jdolecek return (0);
1938 1.1 jdolecek }
1939 1.1 jdolecek #endif
1940 1.1 jdolecek
1941 1.1 jdolecek /*
1942 1.1 jdolecek * Process fixup routine on given buffer.
1943 1.1 jdolecek */
1944 1.1 jdolecek int
1945 1.1 jdolecek ntfs_procfixups(
1946 1.1 jdolecek struct ntfsmount * ntmp,
1947 1.1 jdolecek u_int32_t magic,
1948 1.16 christos caddr_t xbuf,
1949 1.1 jdolecek size_t len)
1950 1.1 jdolecek {
1951 1.16 christos struct fixuphdr *fhp = (struct fixuphdr *) xbuf;
1952 1.1 jdolecek int i;
1953 1.1 jdolecek u_int16_t fixup;
1954 1.1 jdolecek u_int16_t *fxp;
1955 1.1 jdolecek u_int16_t *cfxp;
1956 1.1 jdolecek
1957 1.1 jdolecek if (fhp->fh_magic != magic) {
1958 1.1 jdolecek printf("ntfs_procfixups: magic doesn't match: %08x != %08x\n",
1959 1.1 jdolecek fhp->fh_magic, magic);
1960 1.1 jdolecek return (EINVAL);
1961 1.1 jdolecek }
1962 1.1 jdolecek if ((fhp->fh_fnum - 1) * ntmp->ntm_bps != len) {
1963 1.20 christos printf("ntfs_procfixups: "
1964 1.14 perry "bad fixups number: %d for %ld bytes block\n",
1965 1.1 jdolecek fhp->fh_fnum, (long)len); /* XXX printf kludge */
1966 1.1 jdolecek return (EINVAL);
1967 1.1 jdolecek }
1968 1.1 jdolecek if (fhp->fh_foff >= ntmp->ntm_spc * ntmp->ntm_mftrecsz * ntmp->ntm_bps) {
1969 1.1 jdolecek printf("ntfs_procfixups: invalid offset: %x", fhp->fh_foff);
1970 1.1 jdolecek return (EINVAL);
1971 1.1 jdolecek }
1972 1.16 christos fxp = (u_int16_t *) (xbuf + fhp->fh_foff);
1973 1.16 christos cfxp = (u_int16_t *) (xbuf + ntmp->ntm_bps - 2);
1974 1.1 jdolecek fixup = *fxp++;
1975 1.1 jdolecek for (i = 1; i < fhp->fh_fnum; i++, fxp++) {
1976 1.1 jdolecek if (*cfxp != fixup) {
1977 1.1 jdolecek printf("ntfs_procfixups: fixup %d doesn't match\n", i);
1978 1.1 jdolecek return (EINVAL);
1979 1.1 jdolecek }
1980 1.1 jdolecek *cfxp = *fxp;
1981 1.10 drochner cfxp = (u_int16_t *)((caddr_t)cfxp + ntmp->ntm_bps);
1982 1.1 jdolecek }
1983 1.1 jdolecek return (0);
1984 1.1 jdolecek }
1985 1.1 jdolecek
1986 1.1 jdolecek #if UNUSED_CODE
1987 1.1 jdolecek int
1988 1.1 jdolecek ntfs_runtocn(
1989 1.14 perry cn_t * cn,
1990 1.1 jdolecek struct ntfsmount * ntmp,
1991 1.1 jdolecek u_int8_t * run,
1992 1.1 jdolecek u_long len,
1993 1.1 jdolecek cn_t vcn)
1994 1.1 jdolecek {
1995 1.1 jdolecek cn_t ccn = 0;
1996 1.1 jdolecek cn_t ccl = 0;
1997 1.1 jdolecek u_long off = 0;
1998 1.1 jdolecek int error = 0;
1999 1.1 jdolecek
2000 1.20 christos #ifdef NTFS_DEBUG
2001 1.1 jdolecek int i;
2002 1.15 christos printf("ntfs_runtocn: run: %p, %ld bytes, vcn:%ld\n",
2003 1.1 jdolecek run, len, (u_long) vcn);
2004 1.1 jdolecek printf("ntfs_runtocn: run: ");
2005 1.1 jdolecek for (i = 0; i < len; i++)
2006 1.1 jdolecek printf("0x%02x ", run[i]);
2007 1.1 jdolecek printf("\n");
2008 1.1 jdolecek #endif
2009 1.1 jdolecek
2010 1.1 jdolecek if (NULL == run) {
2011 1.1 jdolecek printf("ntfs_runtocn: run == NULL\n");
2012 1.1 jdolecek return (EINVAL);
2013 1.1 jdolecek }
2014 1.1 jdolecek do {
2015 1.1 jdolecek if (run[off] == 0) {
2016 1.1 jdolecek printf("ntfs_runtocn: vcn too big\n");
2017 1.1 jdolecek return (E2BIG);
2018 1.1 jdolecek }
2019 1.1 jdolecek vcn -= ccl;
2020 1.1 jdolecek error = ntfs_parserun(&ccn, &ccl, run, len, &off);
2021 1.1 jdolecek if (error) {
2022 1.1 jdolecek printf("ntfs_runtocn: ntfs_parserun failed\n");
2023 1.1 jdolecek return (error);
2024 1.1 jdolecek }
2025 1.1 jdolecek } while (ccl <= vcn);
2026 1.1 jdolecek *cn = ccn + vcn;
2027 1.1 jdolecek return (0);
2028 1.1 jdolecek }
2029 1.1 jdolecek #endif
2030 1.1 jdolecek
2031 1.1 jdolecek /*
2032 1.1 jdolecek * this initializes toupper table & dependant variables to be ready for
2033 1.1 jdolecek * later work
2034 1.1 jdolecek */
2035 1.1 jdolecek void
2036 1.1 jdolecek ntfs_toupper_init()
2037 1.1 jdolecek {
2038 1.1 jdolecek ntfs_toupper_tab = (wchar *) NULL;
2039 1.28 ad mutex_init(&ntfs_toupper_lock, MUTEX_DEFAULT, IPL_NONE);
2040 1.1 jdolecek ntfs_toupper_usecount = 0;
2041 1.1 jdolecek }
2042 1.1 jdolecek
2043 1.1 jdolecek /*
2044 1.1 jdolecek * if the ntfs_toupper_tab[] is filled already, just raise use count;
2045 1.1 jdolecek * otherwise read the data from the filesystem we are currently mounting
2046 1.1 jdolecek */
2047 1.1 jdolecek int
2048 1.1 jdolecek ntfs_toupper_use(mp, ntmp)
2049 1.1 jdolecek struct mount *mp;
2050 1.1 jdolecek struct ntfsmount *ntmp;
2051 1.1 jdolecek {
2052 1.1 jdolecek int error = 0;
2053 1.1 jdolecek struct vnode *vp;
2054 1.1 jdolecek
2055 1.1 jdolecek /* get exclusive access */
2056 1.28 ad mutex_enter(&ntfs_toupper_lock);
2057 1.14 perry
2058 1.1 jdolecek /* only read the translation data from a file if it hasn't been
2059 1.1 jdolecek * read already */
2060 1.1 jdolecek if (ntfs_toupper_tab)
2061 1.1 jdolecek goto out;
2062 1.1 jdolecek
2063 1.1 jdolecek /*
2064 1.1 jdolecek * Read in Unicode lowercase -> uppercase translation file.
2065 1.1 jdolecek * XXX for now, just the first 256 entries are used anyway,
2066 1.1 jdolecek * so don't bother reading more
2067 1.1 jdolecek */
2068 1.1 jdolecek MALLOC(ntfs_toupper_tab, wchar *, 256 * 256 * sizeof(wchar),
2069 1.1 jdolecek M_NTFSRDATA, M_WAITOK);
2070 1.1 jdolecek
2071 1.6 thorpej if ((error = VFS_VGET(mp, NTFS_UPCASEINO, &vp)))
2072 1.1 jdolecek goto out;
2073 1.1 jdolecek error = ntfs_readattr(ntmp, VTONT(vp), NTFS_A_DATA, NULL,
2074 1.1 jdolecek 0, 256*256*sizeof(wchar), (char *) ntfs_toupper_tab,
2075 1.1 jdolecek NULL);
2076 1.1 jdolecek vput(vp);
2077 1.1 jdolecek
2078 1.1 jdolecek out:
2079 1.1 jdolecek ntfs_toupper_usecount++;
2080 1.28 ad mutex_exit(&ntfs_toupper_lock);
2081 1.1 jdolecek return (error);
2082 1.1 jdolecek }
2083 1.1 jdolecek
2084 1.1 jdolecek /*
2085 1.1 jdolecek * lower the use count and if it reaches zero, free the memory
2086 1.1 jdolecek * tied by toupper table
2087 1.1 jdolecek */
2088 1.1 jdolecek void
2089 1.1 jdolecek ntfs_toupper_unuse()
2090 1.1 jdolecek {
2091 1.1 jdolecek /* get exclusive access */
2092 1.28 ad mutex_enter(&ntfs_toupper_lock);
2093 1.1 jdolecek
2094 1.1 jdolecek ntfs_toupper_usecount--;
2095 1.1 jdolecek if (ntfs_toupper_usecount == 0) {
2096 1.1 jdolecek FREE(ntfs_toupper_tab, M_NTFSRDATA);
2097 1.1 jdolecek ntfs_toupper_tab = NULL;
2098 1.1 jdolecek }
2099 1.1 jdolecek #ifdef DIAGNOSTIC
2100 1.1 jdolecek else if (ntfs_toupper_usecount < 0) {
2101 1.1 jdolecek panic("ntfs_toupper_unuse(): use count negative: %d",
2102 1.1 jdolecek ntfs_toupper_usecount);
2103 1.1 jdolecek }
2104 1.1 jdolecek #endif
2105 1.14 perry
2106 1.1 jdolecek /* release the lock */
2107 1.28 ad mutex_exit(&ntfs_toupper_lock);
2108 1.14 perry }
2109