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