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