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