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