Home | History | Annotate | Line # | Download | only in udf
udf_allocation.c revision 1.26
      1  1.26  reinoud /* $NetBSD: udf_allocation.c,v 1.26 2009/06/24 17:09:13 reinoud Exp $ */
      2   1.1  reinoud 
      3   1.1  reinoud /*
      4   1.1  reinoud  * Copyright (c) 2006, 2008 Reinoud Zandijk
      5   1.1  reinoud  * All rights reserved.
      6   1.1  reinoud  *
      7   1.1  reinoud  * Redistribution and use in source and binary forms, with or without
      8   1.1  reinoud  * modification, are permitted provided that the following conditions
      9   1.1  reinoud  * are met:
     10   1.1  reinoud  * 1. Redistributions of source code must retain the above copyright
     11   1.1  reinoud  *    notice, this list of conditions and the following disclaimer.
     12   1.1  reinoud  * 2. Redistributions in binary form must reproduce the above copyright
     13   1.1  reinoud  *    notice, this list of conditions and the following disclaimer in the
     14   1.1  reinoud  *    documentation and/or other materials provided with the distribution.
     15   1.1  reinoud  *
     16   1.1  reinoud  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     17   1.1  reinoud  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     18   1.1  reinoud  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     19   1.1  reinoud  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     20   1.1  reinoud  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     21   1.1  reinoud  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     22   1.1  reinoud  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     23   1.1  reinoud  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     24   1.1  reinoud  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     25   1.1  reinoud  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26   1.1  reinoud  *
     27   1.1  reinoud  */
     28   1.1  reinoud 
     29   1.1  reinoud #include <sys/cdefs.h>
     30   1.1  reinoud #ifndef lint
     31  1.26  reinoud __KERNEL_RCSID(0, "$NetBSD: udf_allocation.c,v 1.26 2009/06/24 17:09:13 reinoud Exp $");
     32   1.1  reinoud #endif /* not lint */
     33   1.1  reinoud 
     34   1.1  reinoud 
     35   1.1  reinoud #if defined(_KERNEL_OPT)
     36   1.1  reinoud #include "opt_compat_netbsd.h"
     37   1.1  reinoud #endif
     38   1.1  reinoud 
     39   1.1  reinoud /* TODO strip */
     40   1.1  reinoud #include <sys/param.h>
     41   1.1  reinoud #include <sys/systm.h>
     42   1.1  reinoud #include <sys/sysctl.h>
     43   1.1  reinoud #include <sys/namei.h>
     44   1.1  reinoud #include <sys/proc.h>
     45   1.1  reinoud #include <sys/kernel.h>
     46   1.1  reinoud #include <sys/vnode.h>
     47   1.1  reinoud #include <miscfs/genfs/genfs_node.h>
     48   1.1  reinoud #include <sys/mount.h>
     49   1.1  reinoud #include <sys/buf.h>
     50   1.1  reinoud #include <sys/file.h>
     51   1.1  reinoud #include <sys/device.h>
     52   1.1  reinoud #include <sys/disklabel.h>
     53   1.1  reinoud #include <sys/ioctl.h>
     54   1.1  reinoud #include <sys/malloc.h>
     55   1.1  reinoud #include <sys/dirent.h>
     56   1.1  reinoud #include <sys/stat.h>
     57   1.1  reinoud #include <sys/conf.h>
     58   1.1  reinoud #include <sys/kauth.h>
     59   1.1  reinoud #include <sys/kthread.h>
     60   1.1  reinoud #include <dev/clock_subr.h>
     61   1.1  reinoud 
     62   1.1  reinoud #include <fs/udf/ecma167-udf.h>
     63   1.1  reinoud #include <fs/udf/udf_mount.h>
     64   1.1  reinoud 
     65   1.1  reinoud #include "udf.h"
     66   1.1  reinoud #include "udf_subr.h"
     67   1.1  reinoud #include "udf_bswap.h"
     68   1.1  reinoud 
     69   1.1  reinoud 
     70   1.1  reinoud #define VTOI(vnode) ((struct udf_node *) vnode->v_data)
     71   1.1  reinoud 
     72   1.1  reinoud static void udf_record_allocation_in_node(struct udf_mount *ump,
     73   1.1  reinoud 	struct buf *buf, uint16_t vpart_num, uint64_t *mapping,
     74   1.1  reinoud 	struct long_ad *node_ad_cpy);
     75   1.1  reinoud 
     76   1.1  reinoud /*
     77   1.1  reinoud  * IDEA/BUSY: Each udf_node gets its own extentwalker state for all operations;
     78   1.1  reinoud  * this will hopefully/likely reduce O(nlog(n)) to O(1) for most functionality
     79   1.1  reinoud  * since actions are most likely sequencial and thus seeking doesn't need
     80   1.1  reinoud  * searching for the same or adjacent position again.
     81   1.1  reinoud  */
     82   1.1  reinoud 
     83   1.1  reinoud /* --------------------------------------------------------------------- */
     84  1.11  reinoud 
     85  1.12  reinoud #if 0
     86   1.1  reinoud #if 1
     87   1.1  reinoud static void
     88   1.1  reinoud udf_node_dump(struct udf_node *udf_node) {
     89   1.1  reinoud 	struct file_entry    *fe;
     90   1.1  reinoud 	struct extfile_entry *efe;
     91   1.1  reinoud 	struct icb_tag *icbtag;
     92  1.11  reinoud 	struct long_ad s_ad;
     93   1.1  reinoud 	uint64_t inflen;
     94  1.11  reinoud 	uint32_t icbflags, addr_type;
     95   1.1  reinoud 	uint32_t len, lb_num;
     96  1.11  reinoud 	uint32_t flags;
     97   1.1  reinoud 	int part_num;
     98  1.11  reinoud 	int lb_size, eof, slot;
     99   1.1  reinoud 
    100  1.10  reinoud 	if ((udf_verbose & UDF_DEBUG_NODEDUMP) == 0)
    101   1.1  reinoud 		return;
    102   1.1  reinoud 
    103   1.1  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
    104   1.1  reinoud 
    105   1.1  reinoud 	fe  = udf_node->fe;
    106   1.1  reinoud 	efe = udf_node->efe;
    107   1.1  reinoud 	if (fe) {
    108   1.1  reinoud 		icbtag = &fe->icbtag;
    109   1.1  reinoud 		inflen = udf_rw64(fe->inf_len);
    110   1.1  reinoud 	} else {
    111   1.1  reinoud 		icbtag = &efe->icbtag;
    112   1.1  reinoud 		inflen = udf_rw64(efe->inf_len);
    113   1.1  reinoud 	}
    114   1.1  reinoud 
    115   1.1  reinoud 	icbflags   = udf_rw16(icbtag->flags);
    116   1.1  reinoud 	addr_type  = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
    117   1.1  reinoud 
    118  1.11  reinoud 	printf("udf_node_dump %p :\n", udf_node);
    119   1.1  reinoud 
    120   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
    121  1.11  reinoud 		printf("\tIntern alloc, len = %"PRIu64"\n", inflen);
    122   1.1  reinoud 		return;
    123   1.1  reinoud 	}
    124   1.1  reinoud 
    125  1.11  reinoud 	printf("\tInflen  = %"PRIu64"\n", inflen);
    126  1.11  reinoud 	printf("\t\t");
    127   1.1  reinoud 
    128  1.11  reinoud 	slot = 0;
    129  1.11  reinoud 	for (;;) {
    130  1.11  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
    131  1.11  reinoud 		if (eof)
    132  1.11  reinoud 			break;
    133  1.11  reinoud 		part_num = udf_rw16(s_ad.loc.part_num);
    134  1.11  reinoud 		lb_num = udf_rw32(s_ad.loc.lb_num);
    135  1.11  reinoud 		len   = udf_rw32(s_ad.len);
    136  1.11  reinoud 		flags = UDF_EXT_FLAGS(len);
    137  1.11  reinoud 		len   = UDF_EXT_LEN(len);
    138   1.1  reinoud 
    139   1.1  reinoud 		printf("[");
    140   1.1  reinoud 		if (part_num >= 0)
    141   1.1  reinoud 			printf("part %d, ", part_num);
    142   1.1  reinoud 		printf("lb_num %d, len %d", lb_num, len);
    143   1.1  reinoud 		if (flags)
    144  1.10  reinoud 			printf(", flags %d", flags>>30);
    145   1.1  reinoud 		printf("] ");
    146  1.11  reinoud 
    147  1.11  reinoud 		if (flags == UDF_EXT_REDIRECT) {
    148  1.11  reinoud 			printf("\n\textent END\n\tallocation extent\n\t\t");
    149  1.11  reinoud 		}
    150  1.11  reinoud 
    151  1.11  reinoud 		slot++;
    152   1.1  reinoud 	}
    153  1.11  reinoud 	printf("\n\tl_ad END\n\n");
    154   1.1  reinoud }
    155   1.1  reinoud #else
    156   1.1  reinoud #define udf_node_dump(a)
    157   1.1  reinoud #endif
    158   1.1  reinoud 
    159   1.9  reinoud 
    160   1.9  reinoud static void
    161   1.9  reinoud udf_assert_allocated(struct udf_mount *ump, uint16_t vpart_num,
    162   1.9  reinoud 	uint32_t lb_num, uint32_t num_lb)
    163   1.9  reinoud {
    164   1.9  reinoud 	struct udf_bitmap *bitmap;
    165   1.9  reinoud 	struct part_desc *pdesc;
    166   1.9  reinoud 	uint32_t ptov;
    167   1.9  reinoud 	uint32_t bitval;
    168   1.9  reinoud 	uint8_t *bpos;
    169   1.9  reinoud 	int bit;
    170   1.9  reinoud 	int phys_part;
    171   1.9  reinoud 	int ok;
    172   1.9  reinoud 
    173  1.10  reinoud 	DPRINTF(PARANOIA, ("udf_assert_allocated: check virt lbnum %d "
    174   1.9  reinoud 			  "part %d + %d sect\n", lb_num, vpart_num, num_lb));
    175   1.9  reinoud 
    176   1.9  reinoud 	/* get partition backing up this vpart_num */
    177   1.9  reinoud 	pdesc = ump->partitions[ump->vtop[vpart_num]];
    178   1.9  reinoud 
    179   1.9  reinoud 	switch (ump->vtop_tp[vpart_num]) {
    180   1.9  reinoud 	case UDF_VTOP_TYPE_PHYS :
    181   1.9  reinoud 	case UDF_VTOP_TYPE_SPARABLE :
    182   1.9  reinoud 		/* free space to freed or unallocated space bitmap */
    183   1.9  reinoud 		ptov      = udf_rw32(pdesc->start_loc);
    184   1.9  reinoud 		phys_part = ump->vtop[vpart_num];
    185   1.9  reinoud 
    186   1.9  reinoud 		/* use unallocated bitmap */
    187   1.9  reinoud 		bitmap = &ump->part_unalloc_bits[phys_part];
    188   1.9  reinoud 
    189   1.9  reinoud 		/* if no bitmaps are defined, bail out */
    190   1.9  reinoud 		if (bitmap->bits == NULL)
    191   1.9  reinoud 			break;
    192   1.9  reinoud 
    193   1.9  reinoud 		/* check bits */
    194   1.9  reinoud 		KASSERT(bitmap->bits);
    195   1.9  reinoud 		ok = 1;
    196   1.9  reinoud 		bpos = bitmap->bits + lb_num/8;
    197   1.9  reinoud 		bit  = lb_num % 8;
    198   1.9  reinoud 		while (num_lb > 0) {
    199   1.9  reinoud 			bitval = (1 << bit);
    200   1.9  reinoud 			DPRINTF(PARANOIA, ("XXX : check %d, %p, bit %d\n",
    201   1.9  reinoud 				lb_num, bpos, bit));
    202   1.9  reinoud 			KASSERT(bitmap->bits + lb_num/8 == bpos);
    203   1.9  reinoud 			if (*bpos & bitval) {
    204   1.9  reinoud 				printf("\tlb_num %d is NOT marked busy\n",
    205   1.9  reinoud 					lb_num);
    206   1.9  reinoud 				ok = 0;
    207   1.9  reinoud 			}
    208   1.9  reinoud 			lb_num++; num_lb--;
    209   1.9  reinoud 			bit = (bit + 1) % 8;
    210   1.9  reinoud 			if (bit == 0)
    211   1.9  reinoud 				bpos++;
    212   1.9  reinoud 		}
    213   1.9  reinoud 		if (!ok) {
    214   1.9  reinoud 			/* KASSERT(0); */
    215   1.9  reinoud 		}
    216   1.9  reinoud 
    217   1.9  reinoud 		break;
    218   1.9  reinoud 	case UDF_VTOP_TYPE_VIRT :
    219   1.9  reinoud 		/* TODO check space */
    220   1.9  reinoud 		KASSERT(num_lb == 1);
    221   1.9  reinoud 		break;
    222   1.9  reinoud 	case UDF_VTOP_TYPE_META :
    223   1.9  reinoud 		/* TODO check space in the metadata bitmap */
    224   1.9  reinoud 	default:
    225   1.9  reinoud 		/* not implemented */
    226   1.9  reinoud 		break;
    227   1.9  reinoud 	}
    228   1.9  reinoud }
    229   1.9  reinoud 
    230   1.9  reinoud 
    231   1.1  reinoud static void
    232   1.1  reinoud udf_node_sanity_check(struct udf_node *udf_node,
    233  1.17  reinoud 		uint64_t *cnt_inflen, uint64_t *cnt_logblksrec)
    234  1.17  reinoud {
    235  1.17  reinoud 	union dscrptr *dscr;
    236   1.1  reinoud 	struct file_entry    *fe;
    237   1.1  reinoud 	struct extfile_entry *efe;
    238   1.1  reinoud 	struct icb_tag *icbtag;
    239  1.11  reinoud 	struct long_ad  s_ad;
    240   1.1  reinoud 	uint64_t inflen, logblksrec;
    241  1.11  reinoud 	uint32_t icbflags, addr_type;
    242  1.11  reinoud 	uint32_t len, lb_num, l_ea, l_ad, max_l_ad;
    243   1.9  reinoud 	uint16_t part_num;
    244  1.17  reinoud 	uint8_t *data_pos;
    245  1.11  reinoud 	int dscr_size, lb_size, flags, whole_lb;
    246  1.17  reinoud 	int i, slot, eof;
    247   1.1  reinoud 
    248   1.9  reinoud //	KASSERT(mutex_owned(&udf_node->ump->allocate_mutex));
    249   1.1  reinoud 
    250  1.10  reinoud 	if (1)
    251  1.10  reinoud 		udf_node_dump(udf_node);
    252  1.10  reinoud 
    253   1.1  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
    254   1.1  reinoud 
    255   1.1  reinoud 	fe  = udf_node->fe;
    256   1.1  reinoud 	efe = udf_node->efe;
    257   1.1  reinoud 	if (fe) {
    258  1.17  reinoud 		dscr       = (union dscrptr *) fe;
    259  1.17  reinoud 		icbtag     = &fe->icbtag;
    260  1.17  reinoud 		inflen     = udf_rw64(fe->inf_len);
    261  1.11  reinoud 		dscr_size  = sizeof(struct file_entry) -1;
    262   1.1  reinoud 		logblksrec = udf_rw64(fe->logblks_rec);
    263  1.11  reinoud 		l_ad       = udf_rw32(fe->l_ad);
    264   1.1  reinoud 		l_ea       = udf_rw32(fe->l_ea);
    265   1.1  reinoud 	} else {
    266  1.17  reinoud 		dscr       = (union dscrptr *) efe;
    267  1.17  reinoud 		icbtag     = &efe->icbtag;
    268  1.17  reinoud 		inflen     = udf_rw64(efe->inf_len);
    269  1.11  reinoud 		dscr_size  = sizeof(struct extfile_entry) -1;
    270   1.1  reinoud 		logblksrec = udf_rw64(efe->logblks_rec);
    271  1.11  reinoud 		l_ad       = udf_rw32(efe->l_ad);
    272   1.1  reinoud 		l_ea       = udf_rw32(efe->l_ea);
    273   1.1  reinoud 	}
    274  1.17  reinoud 	data_pos  = (uint8_t *) dscr + dscr_size + l_ea;
    275  1.11  reinoud 	max_l_ad   = lb_size - dscr_size - l_ea;
    276   1.1  reinoud 	icbflags   = udf_rw16(icbtag->flags);
    277   1.1  reinoud 	addr_type  = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
    278   1.1  reinoud 
    279  1.17  reinoud 	/* check if tail is zero */
    280  1.17  reinoud 	DPRINTF(PARANOIA, ("Sanity check blank tail\n"));
    281  1.17  reinoud 	for (i = l_ad; i < max_l_ad; i++) {
    282  1.17  reinoud 		if (data_pos[i] != 0)
    283  1.17  reinoud 			printf( "sanity_check: violation: node byte %d "
    284  1.17  reinoud 				"has value %d\n", i, data_pos[i]);
    285  1.17  reinoud 	}
    286  1.17  reinoud 
    287   1.1  reinoud 	/* reset counters */
    288   1.1  reinoud 	*cnt_inflen     = 0;
    289   1.1  reinoud 	*cnt_logblksrec = 0;
    290   1.1  reinoud 
    291   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
    292   1.1  reinoud 		KASSERT(l_ad <= max_l_ad);
    293   1.1  reinoud 		KASSERT(l_ad == inflen);
    294   1.1  reinoud 		*cnt_inflen = inflen;
    295   1.1  reinoud 		return;
    296   1.1  reinoud 	}
    297   1.1  reinoud 
    298   1.1  reinoud 	/* start counting */
    299   1.1  reinoud 	whole_lb = 1;
    300  1.11  reinoud 	slot = 0;
    301  1.11  reinoud 	for (;;) {
    302  1.11  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
    303  1.11  reinoud 		if (eof)
    304  1.11  reinoud 			break;
    305   1.1  reinoud 		KASSERT(whole_lb == 1);
    306  1.11  reinoud 
    307  1.11  reinoud 		part_num = udf_rw16(s_ad.loc.part_num);
    308  1.11  reinoud 		lb_num = udf_rw32(s_ad.loc.lb_num);
    309  1.11  reinoud 		len   = udf_rw32(s_ad.len);
    310  1.11  reinoud 		flags = UDF_EXT_FLAGS(len);
    311  1.11  reinoud 		len   = UDF_EXT_LEN(len);
    312  1.11  reinoud 
    313   1.6  reinoud 		if (flags != UDF_EXT_REDIRECT) {
    314   1.6  reinoud 			*cnt_inflen += len;
    315   1.6  reinoud 			if (flags == UDF_EXT_ALLOCATED) {
    316   1.6  reinoud 				*cnt_logblksrec += (len + lb_size -1) / lb_size;
    317   1.6  reinoud 			}
    318   1.6  reinoud 		} else {
    319   1.6  reinoud 			KASSERT(len == lb_size);
    320   1.1  reinoud 		}
    321   1.9  reinoud 		/* check allocation */
    322   1.9  reinoud 		if (flags == UDF_EXT_ALLOCATED)
    323   1.9  reinoud 			udf_assert_allocated(udf_node->ump, part_num, lb_num,
    324   1.9  reinoud 				(len + lb_size - 1) / lb_size);
    325   1.8  reinoud 
    326   1.8  reinoud 		/* check whole lb */
    327   1.1  reinoud 		whole_lb = ((len % lb_size) == 0);
    328  1.11  reinoud 
    329  1.11  reinoud 		slot++;
    330   1.1  reinoud 	}
    331   1.1  reinoud 	/* rest should be zero (ad_off > l_ad < max_l_ad - adlen) */
    332   1.1  reinoud 
    333   1.1  reinoud 	KASSERT(*cnt_inflen == inflen);
    334   1.1  reinoud 	KASSERT(*cnt_logblksrec == logblksrec);
    335   1.1  reinoud 
    336   1.9  reinoud //	KASSERT(mutex_owned(&udf_node->ump->allocate_mutex));
    337   1.1  reinoud }
    338   1.1  reinoud #else
    339  1.12  reinoud static void
    340  1.12  reinoud udf_node_sanity_check(struct udf_node *udf_node,
    341  1.12  reinoud 		uint64_t *cnt_inflen, uint64_t *cnt_logblksrec) {
    342  1.12  reinoud 	struct file_entry    *fe;
    343  1.12  reinoud 	struct extfile_entry *efe;
    344  1.12  reinoud 	struct icb_tag *icbtag;
    345  1.12  reinoud 	uint64_t inflen, logblksrec;
    346  1.12  reinoud 	int dscr_size, lb_size;
    347  1.12  reinoud 
    348  1.12  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
    349  1.12  reinoud 
    350  1.12  reinoud 	fe  = udf_node->fe;
    351  1.12  reinoud 	efe = udf_node->efe;
    352  1.12  reinoud 	if (fe) {
    353  1.12  reinoud 		icbtag = &fe->icbtag;
    354  1.12  reinoud 		inflen = udf_rw64(fe->inf_len);
    355  1.12  reinoud 		dscr_size  = sizeof(struct file_entry) -1;
    356  1.12  reinoud 		logblksrec = udf_rw64(fe->logblks_rec);
    357  1.12  reinoud 	} else {
    358  1.12  reinoud 		icbtag = &efe->icbtag;
    359  1.12  reinoud 		inflen = udf_rw64(efe->inf_len);
    360  1.12  reinoud 		dscr_size  = sizeof(struct extfile_entry) -1;
    361  1.12  reinoud 		logblksrec = udf_rw64(efe->logblks_rec);
    362  1.12  reinoud 	}
    363  1.12  reinoud 	*cnt_logblksrec = logblksrec;
    364  1.12  reinoud 	*cnt_inflen     = inflen;
    365  1.12  reinoud }
    366   1.1  reinoud #endif
    367   1.1  reinoud 
    368   1.1  reinoud /* --------------------------------------------------------------------- */
    369   1.1  reinoud 
    370  1.26  reinoud void
    371  1.26  reinoud udf_calc_freespace(struct udf_mount *ump, uint64_t *sizeblks, uint64_t *freeblks)
    372  1.26  reinoud {
    373  1.26  reinoud 	struct logvol_int_desc *lvid;
    374  1.26  reinoud 	uint32_t *pos1, *pos2;
    375  1.26  reinoud 	int vpart, num_vpart;
    376  1.26  reinoud 
    377  1.26  reinoud 	lvid = ump->logvol_integrity;
    378  1.26  reinoud 	*freeblks = *sizeblks = 0;
    379  1.26  reinoud 
    380  1.26  reinoud 	/*
    381  1.26  reinoud 	 * Sequentials media report free space directly (CD/DVD/BD-R), for the
    382  1.26  reinoud 	 * other media we need the logical volume integrity.
    383  1.26  reinoud 	 *
    384  1.26  reinoud 	 * We sum all free space up here regardless of type.
    385  1.26  reinoud 	 */
    386  1.26  reinoud 
    387  1.26  reinoud 	KASSERT(lvid);
    388  1.26  reinoud 	num_vpart = udf_rw32(lvid->num_part);
    389  1.26  reinoud 
    390  1.26  reinoud 	if (ump->discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) {
    391  1.26  reinoud 		/* use track info directly summing if there are 2 open */
    392  1.26  reinoud 		/* XXX assumption at most two tracks open */
    393  1.26  reinoud 		*freeblks = ump->data_track.free_blocks;
    394  1.26  reinoud 		if (ump->data_track.tracknr != ump->metadata_track.tracknr)
    395  1.26  reinoud 			*freeblks += ump->metadata_track.free_blocks;
    396  1.26  reinoud 		*sizeblks = ump->discinfo.last_possible_lba;
    397  1.26  reinoud 	} else {
    398  1.26  reinoud 		/* free and used space for mountpoint based on logvol integrity */
    399  1.26  reinoud 		for (vpart = 0; vpart < num_vpart; vpart++) {
    400  1.26  reinoud 			pos1 = &lvid->tables[0] + vpart;
    401  1.26  reinoud 			pos2 = &lvid->tables[0] + num_vpart + vpart;
    402  1.26  reinoud 			if (udf_rw32(*pos1) != (uint32_t) -1) {
    403  1.26  reinoud 				*freeblks += udf_rw32(*pos1);
    404  1.26  reinoud 				*sizeblks += udf_rw32(*pos2);
    405  1.26  reinoud 			}
    406  1.26  reinoud 		}
    407  1.26  reinoud 	}
    408  1.26  reinoud 	/* adjust for accounted uncommitted blocks */
    409  1.26  reinoud 	for (vpart = 0; vpart < num_vpart; vpart++)
    410  1.26  reinoud 		*freeblks -= ump->uncommitted_lbs[vpart];
    411  1.26  reinoud 
    412  1.26  reinoud 	if (*freeblks > UDF_DISC_SLACK) {
    413  1.26  reinoud 		*freeblks -= UDF_DISC_SLACK;
    414  1.26  reinoud 	} else {
    415  1.26  reinoud 		*freeblks = 0;
    416  1.26  reinoud 	}
    417  1.26  reinoud }
    418  1.26  reinoud 
    419  1.26  reinoud 
    420  1.26  reinoud static void
    421  1.26  reinoud udf_calc_vpart_freespace(struct udf_mount *ump, uint16_t vpart_num, uint64_t *freeblks)
    422  1.26  reinoud {
    423  1.26  reinoud 	struct logvol_int_desc *lvid;
    424  1.26  reinoud 	uint32_t *pos1;
    425  1.26  reinoud 
    426  1.26  reinoud 	lvid = ump->logvol_integrity;
    427  1.26  reinoud 	*freeblks = 0;
    428  1.26  reinoud 
    429  1.26  reinoud 	/*
    430  1.26  reinoud 	 * Sequentials media report free space directly (CD/DVD/BD-R), for the
    431  1.26  reinoud 	 * other media we need the logical volume integrity.
    432  1.26  reinoud 	 *
    433  1.26  reinoud 	 * We sum all free space up here regardless of type.
    434  1.26  reinoud 	 */
    435  1.26  reinoud 
    436  1.26  reinoud 	KASSERT(lvid);
    437  1.26  reinoud 	if (ump->discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) {
    438  1.26  reinoud 		/* XXX assumption at most two tracks open */
    439  1.26  reinoud 		if (vpart_num == ump->data_part) {
    440  1.26  reinoud 			*freeblks = ump->data_track.free_blocks;
    441  1.26  reinoud 		} else {
    442  1.26  reinoud 			*freeblks = ump->metadata_track.free_blocks;
    443  1.26  reinoud 		}
    444  1.26  reinoud 	} else {
    445  1.26  reinoud 		/* free and used space for mountpoint based on logvol integrity */
    446  1.26  reinoud 		pos1 = &lvid->tables[0] + vpart_num;
    447  1.26  reinoud 		if (udf_rw32(*pos1) != (uint32_t) -1)
    448  1.26  reinoud 			*freeblks += udf_rw32(*pos1);
    449  1.26  reinoud 	}
    450  1.26  reinoud 
    451  1.26  reinoud 	/* adjust for accounted uncommitted blocks */
    452  1.26  reinoud 	*freeblks -= ump->uncommitted_lbs[vpart_num];
    453  1.26  reinoud }
    454  1.26  reinoud 
    455  1.26  reinoud /* --------------------------------------------------------------------- */
    456  1.26  reinoud 
    457   1.1  reinoud int
    458   1.1  reinoud udf_translate_vtop(struct udf_mount *ump, struct long_ad *icb_loc,
    459   1.1  reinoud 		   uint32_t *lb_numres, uint32_t *extres)
    460   1.1  reinoud {
    461   1.1  reinoud 	struct part_desc       *pdesc;
    462   1.1  reinoud 	struct spare_map_entry *sme;
    463   1.1  reinoud 	struct long_ad s_icb_loc;
    464   1.1  reinoud 	uint64_t foffset, end_foffset;
    465   1.1  reinoud 	uint32_t lb_size, len;
    466   1.1  reinoud 	uint32_t lb_num, lb_rel, lb_packet;
    467   1.1  reinoud 	uint32_t udf_rw32_lbmap, ext_offset;
    468   1.1  reinoud 	uint16_t vpart;
    469   1.1  reinoud 	int rel, part, error, eof, slot, flags;
    470   1.1  reinoud 
    471   1.1  reinoud 	assert(ump && icb_loc && lb_numres);
    472   1.1  reinoud 
    473   1.1  reinoud 	vpart  = udf_rw16(icb_loc->loc.part_num);
    474   1.1  reinoud 	lb_num = udf_rw32(icb_loc->loc.lb_num);
    475   1.1  reinoud 	if (vpart > UDF_VTOP_RAWPART)
    476   1.1  reinoud 		return EINVAL;
    477   1.1  reinoud 
    478   1.1  reinoud translate_again:
    479   1.1  reinoud 	part = ump->vtop[vpart];
    480   1.1  reinoud 	pdesc = ump->partitions[part];
    481   1.1  reinoud 
    482   1.1  reinoud 	switch (ump->vtop_tp[vpart]) {
    483   1.1  reinoud 	case UDF_VTOP_TYPE_RAW :
    484   1.1  reinoud 		/* 1:1 to the end of the device */
    485   1.1  reinoud 		*lb_numres = lb_num;
    486   1.1  reinoud 		*extres = INT_MAX;
    487   1.1  reinoud 		return 0;
    488   1.1  reinoud 	case UDF_VTOP_TYPE_PHYS :
    489   1.1  reinoud 		/* transform into its disc logical block */
    490   1.1  reinoud 		if (lb_num > udf_rw32(pdesc->part_len))
    491   1.1  reinoud 			return EINVAL;
    492   1.1  reinoud 		*lb_numres = lb_num + udf_rw32(pdesc->start_loc);
    493   1.1  reinoud 
    494   1.1  reinoud 		/* extent from here to the end of the partition */
    495   1.1  reinoud 		*extres = udf_rw32(pdesc->part_len) - lb_num;
    496   1.1  reinoud 		return 0;
    497   1.1  reinoud 	case UDF_VTOP_TYPE_VIRT :
    498   1.1  reinoud 		/* only maps one logical block, lookup in VAT */
    499   1.1  reinoud 		if (lb_num >= ump->vat_entries)		/* XXX > or >= ? */
    500   1.1  reinoud 			return EINVAL;
    501   1.1  reinoud 
    502   1.1  reinoud 		/* lookup in virtual allocation table file */
    503   1.1  reinoud 		mutex_enter(&ump->allocate_mutex);
    504   1.1  reinoud 		error = udf_vat_read(ump->vat_node,
    505   1.1  reinoud 				(uint8_t *) &udf_rw32_lbmap, 4,
    506   1.1  reinoud 				ump->vat_offset + lb_num * 4);
    507   1.1  reinoud 		mutex_exit(&ump->allocate_mutex);
    508   1.1  reinoud 
    509   1.1  reinoud 		if (error)
    510   1.1  reinoud 			return error;
    511   1.1  reinoud 
    512   1.1  reinoud 		lb_num = udf_rw32(udf_rw32_lbmap);
    513   1.1  reinoud 
    514   1.1  reinoud 		/* transform into its disc logical block */
    515   1.1  reinoud 		if (lb_num > udf_rw32(pdesc->part_len))
    516   1.1  reinoud 			return EINVAL;
    517   1.1  reinoud 		*lb_numres = lb_num + udf_rw32(pdesc->start_loc);
    518   1.1  reinoud 
    519   1.1  reinoud 		/* just one logical block */
    520   1.1  reinoud 		*extres = 1;
    521   1.1  reinoud 		return 0;
    522   1.1  reinoud 	case UDF_VTOP_TYPE_SPARABLE :
    523   1.1  reinoud 		/* check if the packet containing the lb_num is remapped */
    524   1.1  reinoud 		lb_packet = lb_num / ump->sparable_packet_size;
    525   1.1  reinoud 		lb_rel    = lb_num % ump->sparable_packet_size;
    526   1.1  reinoud 
    527   1.1  reinoud 		for (rel = 0; rel < udf_rw16(ump->sparing_table->rt_l); rel++) {
    528   1.1  reinoud 			sme = &ump->sparing_table->entries[rel];
    529   1.1  reinoud 			if (lb_packet == udf_rw32(sme->org)) {
    530   1.1  reinoud 				/* NOTE maps to absolute disc logical block! */
    531   1.1  reinoud 				*lb_numres = udf_rw32(sme->map) + lb_rel;
    532   1.1  reinoud 				*extres    = ump->sparable_packet_size - lb_rel;
    533   1.1  reinoud 				return 0;
    534   1.1  reinoud 			}
    535   1.1  reinoud 		}
    536   1.1  reinoud 
    537   1.1  reinoud 		/* transform into its disc logical block */
    538   1.1  reinoud 		if (lb_num > udf_rw32(pdesc->part_len))
    539   1.1  reinoud 			return EINVAL;
    540   1.1  reinoud 		*lb_numres = lb_num + udf_rw32(pdesc->start_loc);
    541   1.1  reinoud 
    542   1.1  reinoud 		/* rest of block */
    543   1.1  reinoud 		*extres = ump->sparable_packet_size - lb_rel;
    544   1.1  reinoud 		return 0;
    545   1.1  reinoud 	case UDF_VTOP_TYPE_META :
    546   1.1  reinoud 		/* we have to look into the file's allocation descriptors */
    547   1.1  reinoud 
    548   1.1  reinoud 		/* use metadatafile allocation mutex */
    549   1.1  reinoud 		lb_size = udf_rw32(ump->logical_vol->lb_size);
    550   1.1  reinoud 
    551   1.1  reinoud 		UDF_LOCK_NODE(ump->metadata_node, 0);
    552   1.1  reinoud 
    553   1.1  reinoud 		/* get first overlapping extent */
    554   1.1  reinoud 		foffset = 0;
    555   1.1  reinoud 		slot    = 0;
    556   1.1  reinoud 		for (;;) {
    557   1.1  reinoud 			udf_get_adslot(ump->metadata_node,
    558   1.1  reinoud 				slot, &s_icb_loc, &eof);
    559   1.5  reinoud 			DPRINTF(ADWLK, ("slot %d, eof = %d, flags = %d, "
    560   1.5  reinoud 				"len = %d, lb_num = %d, part = %d\n",
    561   1.5  reinoud 				slot, eof,
    562   1.5  reinoud 				UDF_EXT_FLAGS(udf_rw32(s_icb_loc.len)),
    563   1.5  reinoud 				UDF_EXT_LEN(udf_rw32(s_icb_loc.len)),
    564   1.5  reinoud 				udf_rw32(s_icb_loc.loc.lb_num),
    565   1.5  reinoud 				udf_rw16(s_icb_loc.loc.part_num)));
    566   1.1  reinoud 			if (eof) {
    567   1.1  reinoud 				DPRINTF(TRANSLATE,
    568   1.1  reinoud 					("Meta partition translation "
    569   1.1  reinoud 					 "failed: can't seek location\n"));
    570   1.1  reinoud 				UDF_UNLOCK_NODE(ump->metadata_node, 0);
    571   1.1  reinoud 				return EINVAL;
    572   1.1  reinoud 			}
    573   1.1  reinoud 			len   = udf_rw32(s_icb_loc.len);
    574   1.1  reinoud 			flags = UDF_EXT_FLAGS(len);
    575   1.1  reinoud 			len   = UDF_EXT_LEN(len);
    576   1.1  reinoud 
    577   1.5  reinoud 			if (flags == UDF_EXT_REDIRECT) {
    578   1.5  reinoud 				slot++;
    579   1.5  reinoud 				continue;
    580   1.5  reinoud 			}
    581   1.5  reinoud 
    582   1.1  reinoud 			end_foffset = foffset + len;
    583   1.1  reinoud 
    584   1.1  reinoud 			if (end_foffset > lb_num * lb_size)
    585   1.1  reinoud 				break;	/* found */
    586   1.5  reinoud 			foffset = end_foffset;
    587   1.1  reinoud 			slot++;
    588   1.1  reinoud 		}
    589   1.1  reinoud 		/* found overlapping slot */
    590   1.1  reinoud 		ext_offset = lb_num * lb_size - foffset;
    591   1.1  reinoud 
    592   1.1  reinoud 		/* process extent offset */
    593   1.1  reinoud 		lb_num   = udf_rw32(s_icb_loc.loc.lb_num);
    594   1.1  reinoud 		vpart    = udf_rw16(s_icb_loc.loc.part_num);
    595   1.1  reinoud 		lb_num  += (ext_offset + lb_size -1) / lb_size;
    596   1.1  reinoud 		ext_offset = 0;
    597   1.1  reinoud 
    598   1.1  reinoud 		UDF_UNLOCK_NODE(ump->metadata_node, 0);
    599   1.1  reinoud 		if (flags != UDF_EXT_ALLOCATED) {
    600   1.1  reinoud 			DPRINTF(TRANSLATE, ("Metadata partition translation "
    601   1.1  reinoud 					    "failed: not allocated\n"));
    602   1.1  reinoud 			return EINVAL;
    603   1.1  reinoud 		}
    604   1.1  reinoud 
    605   1.1  reinoud 		/*
    606   1.1  reinoud 		 * vpart and lb_num are updated, translate again since we
    607   1.1  reinoud 		 * might be mapped on sparable media
    608   1.1  reinoud 		 */
    609   1.1  reinoud 		goto translate_again;
    610   1.1  reinoud 	default:
    611   1.1  reinoud 		printf("UDF vtop translation scheme %d unimplemented yet\n",
    612   1.1  reinoud 			ump->vtop_tp[vpart]);
    613   1.1  reinoud 	}
    614   1.1  reinoud 
    615   1.1  reinoud 	return EINVAL;
    616   1.1  reinoud }
    617   1.1  reinoud 
    618  1.16  reinoud 
    619  1.16  reinoud /* XXX  provisional primitive braindead version */
    620  1.16  reinoud /* TODO use ext_res */
    621  1.16  reinoud void
    622  1.16  reinoud udf_translate_vtop_list(struct udf_mount *ump, uint32_t sectors,
    623  1.16  reinoud 	uint16_t vpart_num, uint64_t *lmapping, uint64_t *pmapping)
    624  1.16  reinoud {
    625  1.16  reinoud 	struct long_ad loc;
    626  1.16  reinoud 	uint32_t lb_numres, ext_res;
    627  1.16  reinoud 	int sector;
    628  1.16  reinoud 
    629  1.16  reinoud 	for (sector = 0; sector < sectors; sector++) {
    630  1.16  reinoud 		memset(&loc, 0, sizeof(struct long_ad));
    631  1.16  reinoud 		loc.loc.part_num = udf_rw16(vpart_num);
    632  1.16  reinoud 		loc.loc.lb_num   = udf_rw32(*lmapping);
    633  1.16  reinoud 		udf_translate_vtop(ump, &loc, &lb_numres, &ext_res);
    634  1.16  reinoud 		*pmapping = lb_numres;
    635  1.16  reinoud 		lmapping++; pmapping++;
    636  1.16  reinoud 	}
    637  1.16  reinoud }
    638  1.16  reinoud 
    639  1.16  reinoud 
    640   1.1  reinoud /* --------------------------------------------------------------------- */
    641   1.1  reinoud 
    642   1.1  reinoud /*
    643   1.1  reinoud  * Translate an extent (in logical_blocks) into logical block numbers; used
    644   1.1  reinoud  * for read and write operations. DOESNT't check extents.
    645   1.1  reinoud  */
    646   1.1  reinoud 
    647   1.1  reinoud int
    648   1.1  reinoud udf_translate_file_extent(struct udf_node *udf_node,
    649   1.1  reinoud 		          uint32_t from, uint32_t num_lb,
    650   1.1  reinoud 			  uint64_t *map)
    651   1.1  reinoud {
    652   1.1  reinoud 	struct udf_mount *ump;
    653   1.1  reinoud 	struct icb_tag *icbtag;
    654   1.1  reinoud 	struct long_ad t_ad, s_ad;
    655   1.1  reinoud 	uint64_t transsec;
    656   1.1  reinoud 	uint64_t foffset, end_foffset;
    657   1.1  reinoud 	uint32_t transsec32;
    658   1.1  reinoud 	uint32_t lb_size;
    659   1.1  reinoud 	uint32_t ext_offset;
    660   1.1  reinoud 	uint32_t lb_num, len;
    661   1.1  reinoud 	uint32_t overlap, translen;
    662   1.1  reinoud 	uint16_t vpart_num;
    663   1.1  reinoud 	int eof, error, flags;
    664   1.1  reinoud 	int slot, addr_type, icbflags;
    665   1.1  reinoud 
    666   1.1  reinoud 	if (!udf_node)
    667   1.1  reinoud 		return ENOENT;
    668   1.1  reinoud 
    669   1.1  reinoud 	KASSERT(num_lb > 0);
    670   1.1  reinoud 
    671   1.1  reinoud 	UDF_LOCK_NODE(udf_node, 0);
    672   1.1  reinoud 
    673   1.1  reinoud 	/* initialise derivative vars */
    674   1.1  reinoud 	ump = udf_node->ump;
    675   1.1  reinoud 	lb_size = udf_rw32(ump->logical_vol->lb_size);
    676   1.1  reinoud 
    677   1.1  reinoud 	if (udf_node->fe) {
    678   1.1  reinoud 		icbtag = &udf_node->fe->icbtag;
    679   1.1  reinoud 	} else {
    680   1.1  reinoud 		icbtag = &udf_node->efe->icbtag;
    681   1.1  reinoud 	}
    682   1.1  reinoud 	icbflags  = udf_rw16(icbtag->flags);
    683   1.1  reinoud 	addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
    684   1.1  reinoud 
    685   1.1  reinoud 	/* do the work */
    686   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
    687   1.1  reinoud 		*map = UDF_TRANS_INTERN;
    688   1.1  reinoud 		UDF_UNLOCK_NODE(udf_node, 0);
    689   1.1  reinoud 		return 0;
    690   1.1  reinoud 	}
    691   1.1  reinoud 
    692   1.1  reinoud 	/* find first overlapping extent */
    693   1.1  reinoud 	foffset = 0;
    694   1.1  reinoud 	slot    = 0;
    695   1.1  reinoud 	for (;;) {
    696   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
    697   1.4  reinoud 		DPRINTF(ADWLK, ("slot %d, eof = %d, flags = %d, len = %d, "
    698   1.4  reinoud 			"lb_num = %d, part = %d\n", slot, eof,
    699   1.4  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)),
    700   1.4  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
    701   1.4  reinoud 			udf_rw32(s_ad.loc.lb_num),
    702   1.4  reinoud 			udf_rw16(s_ad.loc.part_num)));
    703   1.1  reinoud 		if (eof) {
    704   1.1  reinoud 			DPRINTF(TRANSLATE,
    705   1.1  reinoud 				("Translate file extent "
    706   1.1  reinoud 				 "failed: can't seek location\n"));
    707   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
    708   1.1  reinoud 			return EINVAL;
    709   1.1  reinoud 		}
    710   1.1  reinoud 		len    = udf_rw32(s_ad.len);
    711   1.1  reinoud 		flags  = UDF_EXT_FLAGS(len);
    712   1.1  reinoud 		len    = UDF_EXT_LEN(len);
    713   1.1  reinoud 		lb_num = udf_rw32(s_ad.loc.lb_num);
    714   1.1  reinoud 
    715   1.1  reinoud 		if (flags == UDF_EXT_REDIRECT) {
    716   1.1  reinoud 			slot++;
    717   1.1  reinoud 			continue;
    718   1.1  reinoud 		}
    719   1.1  reinoud 
    720   1.1  reinoud 		end_foffset = foffset + len;
    721   1.1  reinoud 
    722   1.1  reinoud 		if (end_foffset > from * lb_size)
    723   1.1  reinoud 			break;	/* found */
    724   1.1  reinoud 		foffset = end_foffset;
    725   1.1  reinoud 		slot++;
    726   1.1  reinoud 	}
    727   1.1  reinoud 	/* found overlapping slot */
    728   1.1  reinoud 	ext_offset = from * lb_size - foffset;
    729   1.1  reinoud 
    730   1.1  reinoud 	for (;;) {
    731   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
    732   1.4  reinoud 		DPRINTF(ADWLK, ("slot %d, eof = %d, flags = %d, len = %d, "
    733   1.4  reinoud 			"lb_num = %d, part = %d\n", slot, eof,
    734   1.4  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)),
    735   1.4  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
    736   1.4  reinoud 			udf_rw32(s_ad.loc.lb_num),
    737   1.4  reinoud 			udf_rw16(s_ad.loc.part_num)));
    738   1.1  reinoud 		if (eof) {
    739   1.1  reinoud 			DPRINTF(TRANSLATE,
    740   1.1  reinoud 				("Translate file extent "
    741   1.1  reinoud 				 "failed: past eof\n"));
    742   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
    743   1.1  reinoud 			return EINVAL;
    744   1.1  reinoud 		}
    745   1.1  reinoud 
    746   1.1  reinoud 		len    = udf_rw32(s_ad.len);
    747   1.1  reinoud 		flags  = UDF_EXT_FLAGS(len);
    748   1.1  reinoud 		len    = UDF_EXT_LEN(len);
    749   1.1  reinoud 
    750   1.1  reinoud 		lb_num    = udf_rw32(s_ad.loc.lb_num);
    751   1.1  reinoud 		vpart_num = udf_rw16(s_ad.loc.part_num);
    752   1.1  reinoud 
    753   1.1  reinoud 		end_foffset = foffset + len;
    754   1.1  reinoud 
    755   1.1  reinoud 		/* process extent, don't forget to advance on ext_offset! */
    756   1.1  reinoud 		lb_num  += (ext_offset + lb_size -1) / lb_size;
    757   1.1  reinoud 		overlap  = (len - ext_offset + lb_size -1) / lb_size;
    758   1.1  reinoud 		ext_offset = 0;
    759   1.1  reinoud 
    760   1.1  reinoud 		/*
    761   1.1  reinoud 		 * note that the while(){} is nessisary for the extent that
    762   1.1  reinoud 		 * the udf_translate_vtop() returns doens't have to span the
    763   1.1  reinoud 		 * whole extent.
    764   1.1  reinoud 		 */
    765   1.1  reinoud 
    766   1.1  reinoud 		overlap = MIN(overlap, num_lb);
    767   1.4  reinoud 		while (overlap && (flags != UDF_EXT_REDIRECT)) {
    768   1.1  reinoud 			switch (flags) {
    769   1.1  reinoud 			case UDF_EXT_FREE :
    770   1.1  reinoud 			case UDF_EXT_ALLOCATED_BUT_NOT_USED :
    771   1.1  reinoud 				transsec = UDF_TRANS_ZERO;
    772   1.1  reinoud 				translen = overlap;
    773   1.1  reinoud 				while (overlap && num_lb && translen) {
    774   1.1  reinoud 					*map++ = transsec;
    775   1.1  reinoud 					lb_num++;
    776   1.1  reinoud 					overlap--; num_lb--; translen--;
    777   1.1  reinoud 				}
    778   1.1  reinoud 				break;
    779   1.1  reinoud 			case UDF_EXT_ALLOCATED :
    780   1.1  reinoud 				t_ad.loc.lb_num   = udf_rw32(lb_num);
    781   1.1  reinoud 				t_ad.loc.part_num = udf_rw16(vpart_num);
    782   1.1  reinoud 				error = udf_translate_vtop(ump,
    783   1.1  reinoud 						&t_ad, &transsec32, &translen);
    784   1.1  reinoud 				transsec = transsec32;
    785   1.1  reinoud 				if (error) {
    786   1.1  reinoud 					UDF_UNLOCK_NODE(udf_node, 0);
    787   1.1  reinoud 					return error;
    788   1.1  reinoud 				}
    789   1.1  reinoud 				while (overlap && num_lb && translen) {
    790   1.1  reinoud 					*map++ = transsec;
    791   1.1  reinoud 					lb_num++; transsec++;
    792   1.1  reinoud 					overlap--; num_lb--; translen--;
    793   1.1  reinoud 				}
    794   1.1  reinoud 				break;
    795   1.4  reinoud 			default:
    796   1.4  reinoud 				DPRINTF(TRANSLATE,
    797   1.4  reinoud 					("Translate file extent "
    798   1.4  reinoud 					 "failed: bad flags %x\n", flags));
    799   1.4  reinoud 				UDF_UNLOCK_NODE(udf_node, 0);
    800   1.4  reinoud 				return EINVAL;
    801   1.1  reinoud 			}
    802   1.1  reinoud 		}
    803   1.1  reinoud 		if (num_lb == 0)
    804   1.1  reinoud 			break;
    805   1.1  reinoud 
    806   1.1  reinoud 		if (flags != UDF_EXT_REDIRECT)
    807   1.1  reinoud 			foffset = end_foffset;
    808   1.1  reinoud 		slot++;
    809   1.1  reinoud 	}
    810   1.1  reinoud 	UDF_UNLOCK_NODE(udf_node, 0);
    811   1.1  reinoud 
    812   1.1  reinoud 	return 0;
    813   1.1  reinoud }
    814   1.1  reinoud 
    815   1.1  reinoud /* --------------------------------------------------------------------- */
    816   1.1  reinoud 
    817   1.1  reinoud static int
    818   1.1  reinoud udf_search_free_vatloc(struct udf_mount *ump, uint32_t *lbnumres)
    819   1.1  reinoud {
    820   1.1  reinoud 	uint32_t lb_size, lb_num, lb_map, udf_rw32_lbmap;
    821   1.1  reinoud 	uint8_t *blob;
    822   1.1  reinoud 	int entry, chunk, found, error;
    823   1.1  reinoud 
    824   1.1  reinoud 	KASSERT(ump);
    825   1.1  reinoud 	KASSERT(ump->logical_vol);
    826   1.1  reinoud 
    827   1.1  reinoud 	lb_size = udf_rw32(ump->logical_vol->lb_size);
    828   1.1  reinoud 	blob = malloc(lb_size, M_UDFTEMP, M_WAITOK);
    829   1.1  reinoud 
    830   1.1  reinoud 	/* TODO static allocation of search chunk */
    831   1.1  reinoud 
    832   1.1  reinoud 	lb_num = MIN(ump->vat_entries, ump->vat_last_free_lb);
    833   1.1  reinoud 	found  = 0;
    834   1.1  reinoud 	error  = 0;
    835   1.1  reinoud 	entry  = 0;
    836   1.1  reinoud 	do {
    837   1.1  reinoud 		chunk = MIN(lb_size, (ump->vat_entries - lb_num) * 4);
    838   1.1  reinoud 		if (chunk <= 0)
    839   1.1  reinoud 			break;
    840   1.1  reinoud 		/* load in chunk */
    841   1.1  reinoud 		error = udf_vat_read(ump->vat_node, blob, chunk,
    842   1.1  reinoud 				ump->vat_offset + lb_num * 4);
    843   1.1  reinoud 
    844   1.1  reinoud 		if (error)
    845   1.1  reinoud 			break;
    846   1.1  reinoud 
    847   1.1  reinoud 		/* search this chunk */
    848   1.1  reinoud 		for (entry=0; entry < chunk /4; entry++, lb_num++) {
    849   1.1  reinoud 			udf_rw32_lbmap = *((uint32_t *) (blob + entry * 4));
    850   1.1  reinoud 			lb_map = udf_rw32(udf_rw32_lbmap);
    851   1.1  reinoud 			if (lb_map == 0xffffffff) {
    852   1.1  reinoud 				found = 1;
    853   1.1  reinoud 				break;
    854   1.1  reinoud 			}
    855   1.1  reinoud 		}
    856   1.1  reinoud 	} while (!found);
    857   1.1  reinoud 	if (error) {
    858   1.1  reinoud 		printf("udf_search_free_vatloc: error reading in vat chunk "
    859   1.1  reinoud 			"(lb %d, size %d)\n", lb_num, chunk);
    860   1.1  reinoud 	}
    861   1.1  reinoud 
    862   1.1  reinoud 	if (!found) {
    863   1.1  reinoud 		/* extend VAT */
    864   1.1  reinoud 		DPRINTF(WRITE, ("udf_search_free_vatloc: extending\n"));
    865   1.1  reinoud 		lb_num = ump->vat_entries;
    866   1.1  reinoud 		ump->vat_entries++;
    867   1.1  reinoud 	}
    868   1.1  reinoud 
    869   1.1  reinoud 	/* mark entry with initialiser just in case */
    870   1.1  reinoud 	lb_map = udf_rw32(0xfffffffe);
    871   1.1  reinoud 	udf_vat_write(ump->vat_node, (uint8_t *) &lb_map, 4,
    872   1.1  reinoud 		ump->vat_offset + lb_num *4);
    873   1.1  reinoud 	ump->vat_last_free_lb = lb_num;
    874   1.1  reinoud 
    875   1.1  reinoud 	free(blob, M_UDFTEMP);
    876   1.1  reinoud 	*lbnumres = lb_num;
    877   1.1  reinoud 	return 0;
    878   1.1  reinoud }
    879   1.1  reinoud 
    880   1.1  reinoud 
    881   1.1  reinoud static void
    882   1.1  reinoud udf_bitmap_allocate(struct udf_bitmap *bitmap, int ismetadata,
    883  1.16  reinoud 	uint32_t *num_lb, uint64_t *lmappos)
    884   1.1  reinoud {
    885   1.1  reinoud 	uint32_t offset, lb_num, bit;
    886   1.1  reinoud 	int32_t  diff;
    887   1.1  reinoud 	uint8_t *bpos;
    888   1.1  reinoud 	int pass;
    889   1.1  reinoud 
    890   1.1  reinoud 	if (!ismetadata) {
    891   1.1  reinoud 		/* heuristic to keep the two pointers not too close */
    892   1.1  reinoud 		diff = bitmap->data_pos - bitmap->metadata_pos;
    893   1.1  reinoud 		if ((diff >= 0) && (diff < 1024))
    894   1.1  reinoud 			bitmap->data_pos = bitmap->metadata_pos + 1024;
    895   1.1  reinoud 	}
    896   1.1  reinoud 	offset = ismetadata ? bitmap->metadata_pos : bitmap->data_pos;
    897   1.1  reinoud 	offset &= ~7;
    898   1.1  reinoud 	for (pass = 0; pass < 2; pass++) {
    899   1.1  reinoud 		if (offset >= bitmap->max_offset)
    900   1.1  reinoud 			offset = 0;
    901   1.1  reinoud 
    902   1.1  reinoud 		while (offset < bitmap->max_offset) {
    903   1.1  reinoud 			if (*num_lb == 0)
    904   1.1  reinoud 				break;
    905   1.1  reinoud 
    906   1.1  reinoud 			/* use first bit not set */
    907   1.1  reinoud 			bpos  = bitmap->bits + offset/8;
    908   1.9  reinoud 			bit = ffs(*bpos);	/* returns 0 or 1..8 */
    909   1.1  reinoud 			if (bit == 0) {
    910   1.1  reinoud 				offset += 8;
    911   1.1  reinoud 				continue;
    912   1.1  reinoud 			}
    913  1.20  reinoud 
    914  1.20  reinoud 			/* check for ffs overshoot */
    915  1.20  reinoud 			if (offset + bit-1 >= bitmap->max_offset) {
    916  1.20  reinoud 				offset = bitmap->max_offset;
    917  1.20  reinoud 				break;
    918  1.20  reinoud 			}
    919  1.20  reinoud 
    920   1.9  reinoud 			DPRINTF(PARANOIA, ("XXX : allocate %d, %p, bit %d\n",
    921   1.9  reinoud 				offset + bit -1, bpos, bit-1));
    922   1.1  reinoud 			*bpos &= ~(1 << (bit-1));
    923   1.1  reinoud 			lb_num = offset + bit-1;
    924   1.1  reinoud 			*lmappos++ = lb_num;
    925   1.1  reinoud 			*num_lb = *num_lb - 1;
    926   1.1  reinoud 			// offset = (offset & ~7);
    927   1.1  reinoud 		}
    928   1.1  reinoud 	}
    929   1.1  reinoud 
    930   1.1  reinoud 	if (ismetadata) {
    931   1.1  reinoud 		bitmap->metadata_pos = offset;
    932   1.1  reinoud 	} else {
    933   1.1  reinoud 		bitmap->data_pos = offset;
    934   1.1  reinoud 	}
    935   1.1  reinoud }
    936   1.1  reinoud 
    937   1.1  reinoud 
    938   1.1  reinoud static void
    939   1.1  reinoud udf_bitmap_free(struct udf_bitmap *bitmap, uint32_t lb_num, uint32_t num_lb)
    940   1.1  reinoud {
    941   1.1  reinoud 	uint32_t offset;
    942   1.1  reinoud 	uint32_t bit, bitval;
    943   1.1  reinoud 	uint8_t *bpos;
    944   1.1  reinoud 
    945   1.1  reinoud 	offset = lb_num;
    946   1.1  reinoud 
    947   1.1  reinoud 	/* starter bits */
    948   1.1  reinoud 	bpos = bitmap->bits + offset/8;
    949   1.1  reinoud 	bit = offset % 8;
    950   1.1  reinoud 	while ((bit != 0) && (num_lb > 0)) {
    951   1.1  reinoud 		bitval = (1 << bit);
    952   1.1  reinoud 		KASSERT((*bpos & bitval) == 0);
    953   1.9  reinoud 		DPRINTF(PARANOIA, ("XXX : free %d, %p, %d\n",
    954   1.9  reinoud 			offset, bpos, bit));
    955   1.1  reinoud 		*bpos |= bitval;
    956   1.1  reinoud 		offset++; num_lb--;
    957   1.1  reinoud 		bit = (bit + 1) % 8;
    958   1.1  reinoud 	}
    959   1.1  reinoud 	if (num_lb == 0)
    960   1.1  reinoud 		return;
    961   1.1  reinoud 
    962   1.1  reinoud 	/* whole bytes */
    963   1.1  reinoud 	KASSERT(bit == 0);
    964   1.1  reinoud 	bpos = bitmap->bits + offset / 8;
    965   1.1  reinoud 	while (num_lb >= 8) {
    966   1.1  reinoud 		KASSERT((*bpos == 0));
    967   1.9  reinoud 		DPRINTF(PARANOIA, ("XXX : free %d + 8, %p\n", offset, bpos));
    968   1.1  reinoud 		*bpos = 255;
    969   1.1  reinoud 		offset += 8; num_lb -= 8;
    970   1.1  reinoud 		bpos++;
    971   1.1  reinoud 	}
    972   1.1  reinoud 
    973   1.1  reinoud 	/* stop bits */
    974   1.1  reinoud 	KASSERT(num_lb < 8);
    975   1.1  reinoud 	bit = 0;
    976   1.1  reinoud 	while (num_lb > 0) {
    977   1.1  reinoud 		bitval = (1 << bit);
    978   1.1  reinoud 		KASSERT((*bpos & bitval) == 0);
    979   1.9  reinoud 		DPRINTF(PARANOIA, ("XXX : free %d, %p, %d\n",
    980   1.9  reinoud 			offset, bpos, bit));
    981   1.1  reinoud 		*bpos |= bitval;
    982   1.1  reinoud 		offset++; num_lb--;
    983   1.1  reinoud 		bit = (bit + 1) % 8;
    984   1.1  reinoud 	}
    985   1.1  reinoud }
    986   1.1  reinoud 
    987  1.26  reinoud /* --------------------------------------------------------------------- */
    988  1.26  reinoud 
    989  1.26  reinoud /*
    990  1.26  reinoud  * We check for overall disc space with a margin to prevent critical
    991  1.26  reinoud  * conditions.  If disc space is low we try to force a sync() to improve our
    992  1.26  reinoud  * estimates.  When confronted with meta-data partition size shortage we know
    993  1.26  reinoud  * we have to check if it can be extended and we need to extend it when
    994  1.26  reinoud  * needed.
    995  1.26  reinoud  *
    996  1.26  reinoud  * A 2nd strategy we could use when disc space is getting low on a disc
    997  1.26  reinoud  * formatted with a meta-data partition is to see if there are sparse areas in
    998  1.26  reinoud  * the meta-data partition and free blocks there for extra data.
    999  1.26  reinoud  */
   1000  1.26  reinoud 
   1001  1.26  reinoud void
   1002  1.26  reinoud udf_do_reserve_space(struct udf_mount *ump, struct udf_node *udf_node,
   1003  1.26  reinoud 	uint16_t vpart_num, uint32_t num_lb)
   1004  1.26  reinoud {
   1005  1.26  reinoud 	ump->uncommitted_lbs[vpart_num] += num_lb;
   1006  1.26  reinoud 	if (udf_node)
   1007  1.26  reinoud 		udf_node->uncommitted_lbs += num_lb;
   1008  1.26  reinoud }
   1009  1.26  reinoud 
   1010  1.26  reinoud 
   1011  1.26  reinoud void
   1012  1.26  reinoud udf_do_unreserve_space(struct udf_mount *ump, struct udf_node *udf_node,
   1013  1.26  reinoud 	uint16_t vpart_num, uint32_t num_lb)
   1014  1.26  reinoud {
   1015  1.26  reinoud 	ump->uncommitted_lbs[vpart_num] -= num_lb;
   1016  1.26  reinoud 	if (ump->uncommitted_lbs[vpart_num] < 0) {
   1017  1.26  reinoud 		DPRINTF(RESERVE, ("UDF: underflow on partition reservation, "
   1018  1.26  reinoud 			"part %d: %d\n", vpart_num,
   1019  1.26  reinoud 			ump->uncommitted_lbs[vpart_num]));
   1020  1.26  reinoud 		ump->uncommitted_lbs[vpart_num] = 0;
   1021  1.26  reinoud 	}
   1022  1.26  reinoud 	if (udf_node) {
   1023  1.26  reinoud 		udf_node->uncommitted_lbs -= num_lb;
   1024  1.26  reinoud 		if (udf_node->uncommitted_lbs < 0) {
   1025  1.26  reinoud 			DPRINTF(RESERVE, ("UDF: underflow of node "
   1026  1.26  reinoud 				"reservation : %d\n",
   1027  1.26  reinoud 				udf_node->uncommitted_lbs));
   1028  1.26  reinoud 			udf_node->uncommitted_lbs = 0;
   1029  1.26  reinoud 		}
   1030  1.26  reinoud 	}
   1031  1.26  reinoud }
   1032  1.26  reinoud 
   1033  1.26  reinoud 
   1034  1.26  reinoud int
   1035  1.26  reinoud udf_reserve_space(struct udf_mount *ump, struct udf_node *udf_node,
   1036  1.26  reinoud 	int udf_c_type, uint16_t vpart_num, uint32_t num_lb, int can_fail)
   1037  1.26  reinoud {
   1038  1.26  reinoud 	uint64_t freeblks;
   1039  1.26  reinoud 	uint64_t slack;
   1040  1.26  reinoud 	int i, error;
   1041  1.26  reinoud 
   1042  1.26  reinoud 	slack = 0;
   1043  1.26  reinoud 	if (can_fail)
   1044  1.26  reinoud 		slack = UDF_DISC_SLACK;
   1045  1.26  reinoud 
   1046  1.26  reinoud 	error = 0;
   1047  1.26  reinoud 	mutex_enter(&ump->allocate_mutex);
   1048  1.26  reinoud 
   1049  1.26  reinoud 	/* check if there is enough space available */
   1050  1.26  reinoud 	for (i = 0; i < 16; i++) {	/* XXX arbitrary number */
   1051  1.26  reinoud 		udf_calc_vpart_freespace(ump, vpart_num, &freeblks);
   1052  1.26  reinoud 		if (num_lb + slack < freeblks)
   1053  1.26  reinoud 			break;
   1054  1.26  reinoud 		/* issue SYNC */
   1055  1.26  reinoud 		DPRINTF(RESERVE, ("udf_reserve_space: issuing sync\n"));
   1056  1.26  reinoud 		mutex_exit(&ump->allocate_mutex);
   1057  1.26  reinoud 		udf_do_sync(ump, FSCRED, 0);
   1058  1.26  reinoud 		mutex_enter(&mntvnode_lock);
   1059  1.26  reinoud 		/* 1/4 second wait */
   1060  1.26  reinoud 		cv_timedwait(&ump->dirtynodes_cv, &mntvnode_lock,
   1061  1.26  reinoud 			hz/4);
   1062  1.26  reinoud 		mutex_exit(&mntvnode_lock);
   1063  1.26  reinoud 		mutex_enter(&ump->allocate_mutex);
   1064  1.26  reinoud 	}
   1065  1.26  reinoud 
   1066  1.26  reinoud 	/* check if there is enough space available now */
   1067  1.26  reinoud 	udf_calc_vpart_freespace(ump, vpart_num, &freeblks);
   1068  1.26  reinoud 	if (num_lb + slack >= freeblks) {
   1069  1.26  reinoud 		DPRINTF(RESERVE, ("udf_reserve_space: try to juggle partitions\n"));
   1070  1.26  reinoud 		/* TODO juggle with data and metadata partitions if possible */
   1071  1.26  reinoud 	}
   1072  1.26  reinoud 
   1073  1.26  reinoud 	/* check if there is enough space available now */
   1074  1.26  reinoud 	udf_calc_vpart_freespace(ump, vpart_num, &freeblks);
   1075  1.26  reinoud 	if (num_lb + slack <= freeblks) {
   1076  1.26  reinoud 		udf_do_reserve_space(ump, udf_node, vpart_num, num_lb);
   1077  1.26  reinoud 	} else {
   1078  1.26  reinoud 		DPRINTF(RESERVE, ("udf_reserve_space: out of disc space\n"));
   1079  1.26  reinoud 		error = ENOSPC;
   1080  1.26  reinoud 	}
   1081  1.26  reinoud 
   1082  1.26  reinoud 	mutex_exit(&ump->allocate_mutex);
   1083  1.26  reinoud 	return error;
   1084  1.26  reinoud }
   1085  1.26  reinoud 
   1086  1.26  reinoud 
   1087  1.26  reinoud void
   1088  1.26  reinoud udf_cleanup_reservation(struct udf_node *udf_node)
   1089  1.26  reinoud {
   1090  1.26  reinoud 	struct udf_mount *ump = udf_node->ump;
   1091  1.26  reinoud 	int vpart_num;
   1092  1.26  reinoud 
   1093  1.26  reinoud 	mutex_enter(&ump->allocate_mutex);
   1094  1.26  reinoud 
   1095  1.26  reinoud 	/* compensate for overlapping blocks */
   1096  1.26  reinoud 	DPRINTF(RESERVE, ("UDF: overlapped %d blocks in count\n", udf_node->uncommitted_lbs));
   1097  1.26  reinoud 
   1098  1.26  reinoud 	vpart_num = udf_get_record_vpart(ump, udf_get_c_type(udf_node));
   1099  1.26  reinoud 	udf_do_unreserve_space(ump, udf_node, vpart_num, udf_node->uncommitted_lbs);
   1100  1.26  reinoud 
   1101  1.26  reinoud 	DPRINTF(RESERVE, ("\ttotal now %d\n", ump->uncommitted_lbs[vpart_num]));
   1102  1.26  reinoud 
   1103  1.26  reinoud 	/* sanity */
   1104  1.26  reinoud 	if (ump->uncommitted_lbs[vpart_num] < 0)
   1105  1.26  reinoud 		ump->uncommitted_lbs[vpart_num] = 0;
   1106  1.26  reinoud 
   1107  1.26  reinoud 	mutex_exit(&ump->allocate_mutex);
   1108  1.26  reinoud }
   1109  1.26  reinoud 
   1110  1.26  reinoud /* --------------------------------------------------------------------- */
   1111  1.26  reinoud 
   1112  1.26  reinoud /*
   1113  1.26  reinoud  * Allocate an extent of given length on given virt. partition. It doesn't
   1114  1.26  reinoud  * have to be one stretch.
   1115  1.26  reinoud  */
   1116   1.1  reinoud 
   1117  1.26  reinoud int
   1118  1.26  reinoud udf_allocate_space(struct udf_mount *ump, struct udf_node *udf_node,
   1119  1.26  reinoud 	int udf_c_type, uint16_t vpart_num, uint32_t num_lb, uint64_t *lmapping)
   1120   1.1  reinoud {
   1121   1.1  reinoud 	struct mmc_trackinfo *alloc_track, *other_track;
   1122   1.1  reinoud 	struct udf_bitmap *bitmap;
   1123   1.1  reinoud 	struct part_desc *pdesc;
   1124   1.1  reinoud 	struct logvol_int_desc *lvid;
   1125  1.16  reinoud 	uint64_t *lmappos;
   1126   1.1  reinoud 	uint32_t ptov, lb_num, *freepos, free_lbs;
   1127   1.1  reinoud 	int lb_size, alloc_num_lb;
   1128  1.17  reinoud 	int alloc_type, error;
   1129  1.17  reinoud 	int is_node;
   1130   1.1  reinoud 
   1131  1.17  reinoud 	DPRINTF(CALL, ("udf_allocate_space(ctype %d, vpart %d, num_lb %d\n",
   1132  1.17  reinoud 		udf_c_type, vpart_num, num_lb));
   1133   1.1  reinoud 	mutex_enter(&ump->allocate_mutex);
   1134  1.16  reinoud 
   1135   1.1  reinoud 	lb_size = udf_rw32(ump->logical_vol->lb_size);
   1136   1.1  reinoud 	KASSERT(lb_size == ump->discinfo.sector_size);
   1137   1.1  reinoud 
   1138  1.17  reinoud 	alloc_type =  ump->vtop_alloc[vpart_num];
   1139  1.17  reinoud 	is_node    = (udf_c_type == UDF_C_NODE);
   1140   1.1  reinoud 
   1141   1.1  reinoud 	lmappos = lmapping;
   1142  1.16  reinoud 	error = 0;
   1143   1.1  reinoud 	switch (alloc_type) {
   1144   1.1  reinoud 	case UDF_ALLOC_VAT :
   1145   1.1  reinoud 		/* search empty slot in VAT file */
   1146   1.1  reinoud 		KASSERT(num_lb == 1);
   1147   1.1  reinoud 		error = udf_search_free_vatloc(ump, &lb_num);
   1148  1.26  reinoud 		if (!error) {
   1149   1.1  reinoud 			*lmappos = lb_num;
   1150  1.26  reinoud 
   1151  1.26  reinoud 			/* reserve on the backing sequential partition since
   1152  1.26  reinoud 			 * that partition is credited back later */
   1153  1.26  reinoud 			udf_do_reserve_space(ump, udf_node,
   1154  1.26  reinoud 				ump->vtop[vpart_num], num_lb);
   1155  1.26  reinoud 		}
   1156   1.1  reinoud 		break;
   1157   1.1  reinoud 	case UDF_ALLOC_SEQUENTIAL :
   1158   1.1  reinoud 		/* sequential allocation on recordable media */
   1159  1.17  reinoud 		/* get partition backing up this vpart_num_num */
   1160  1.17  reinoud 		pdesc = ump->partitions[ump->vtop[vpart_num]];
   1161  1.16  reinoud 
   1162   1.1  reinoud 		/* calculate offset from physical base partition */
   1163   1.1  reinoud 		ptov  = udf_rw32(pdesc->start_loc);
   1164   1.1  reinoud 
   1165  1.17  reinoud 		/* get our track descriptors */
   1166  1.17  reinoud 		if (vpart_num == ump->node_part) {
   1167  1.17  reinoud 			alloc_track = &ump->metadata_track;
   1168  1.17  reinoud 			other_track = &ump->data_track;
   1169  1.17  reinoud 		} else {
   1170  1.17  reinoud 			alloc_track = &ump->data_track;
   1171  1.17  reinoud 			other_track = &ump->metadata_track;
   1172  1.17  reinoud 		}
   1173  1.17  reinoud 
   1174  1.17  reinoud 		/* allocate */
   1175   1.1  reinoud 		for (lb_num = 0; lb_num < num_lb; lb_num++) {
   1176   1.1  reinoud 			*lmappos++ = alloc_track->next_writable - ptov;
   1177   1.1  reinoud 			alloc_track->next_writable++;
   1178   1.1  reinoud 			alloc_track->free_blocks--;
   1179   1.1  reinoud 		}
   1180  1.17  reinoud 
   1181  1.17  reinoud 		/* keep other track up-to-date */
   1182   1.1  reinoud 		if (alloc_track->tracknr == other_track->tracknr)
   1183   1.1  reinoud 			memcpy(other_track, alloc_track,
   1184   1.1  reinoud 				sizeof(struct mmc_trackinfo));
   1185   1.1  reinoud 		break;
   1186   1.1  reinoud 	case UDF_ALLOC_SPACEMAP :
   1187  1.16  reinoud 		/* try to allocate on unallocated bits */
   1188   1.1  reinoud 		alloc_num_lb = num_lb;
   1189  1.17  reinoud 		bitmap = &ump->part_unalloc_bits[vpart_num];
   1190  1.17  reinoud 		udf_bitmap_allocate(bitmap, is_node, &alloc_num_lb, lmappos);
   1191   1.1  reinoud 		ump->lvclose |= UDF_WRITE_PART_BITMAPS;
   1192  1.16  reinoud 
   1193  1.16  reinoud 		/* have we allocated all? */
   1194   1.1  reinoud 		if (alloc_num_lb) {
   1195   1.1  reinoud 			/* TODO convert freed to unalloc and try again */
   1196   1.1  reinoud 			/* free allocated piece for now */
   1197   1.1  reinoud 			lmappos = lmapping;
   1198   1.1  reinoud 			for (lb_num=0; lb_num < num_lb-alloc_num_lb; lb_num++) {
   1199   1.1  reinoud 				udf_bitmap_free(bitmap, *lmappos++, 1);
   1200   1.1  reinoud 			}
   1201   1.1  reinoud 			error = ENOSPC;
   1202   1.1  reinoud 		}
   1203   1.1  reinoud 		if (!error) {
   1204   1.1  reinoud 			/* adjust freecount */
   1205   1.1  reinoud 			lvid = ump->logvol_integrity;
   1206  1.17  reinoud 			freepos = &lvid->tables[0] + vpart_num;
   1207   1.1  reinoud 			free_lbs = udf_rw32(*freepos);
   1208   1.1  reinoud 			*freepos = udf_rw32(free_lbs - num_lb);
   1209   1.1  reinoud 		}
   1210   1.1  reinoud 		break;
   1211  1.15  reinoud 	case UDF_ALLOC_METABITMAP :		/* UDF 2.50, 2.60 BluRay-RE */
   1212  1.16  reinoud 		/* allocate on metadata unallocated bits */
   1213  1.16  reinoud 		alloc_num_lb = num_lb;
   1214  1.16  reinoud 		bitmap = &ump->metadata_unalloc_bits;
   1215  1.17  reinoud 		udf_bitmap_allocate(bitmap, is_node, &alloc_num_lb, lmappos);
   1216  1.16  reinoud 		ump->lvclose |= UDF_WRITE_PART_BITMAPS;
   1217  1.16  reinoud 
   1218  1.16  reinoud 		/* have we allocated all? */
   1219  1.17  reinoud 		if (alloc_num_lb) {
   1220  1.16  reinoud 			/* YIKES! TODO we need to extend the metadata partition */
   1221  1.16  reinoud 			/* free allocated piece for now */
   1222  1.16  reinoud 			lmappos = lmapping;
   1223  1.16  reinoud 			for (lb_num=0; lb_num < num_lb-alloc_num_lb; lb_num++) {
   1224  1.16  reinoud 				udf_bitmap_free(bitmap, *lmappos++, 1);
   1225  1.16  reinoud 			}
   1226  1.16  reinoud 			error = ENOSPC;
   1227  1.16  reinoud 		}
   1228  1.16  reinoud 		if (!error) {
   1229  1.16  reinoud 			/* adjust freecount */
   1230  1.16  reinoud 			lvid = ump->logvol_integrity;
   1231  1.17  reinoud 			freepos = &lvid->tables[0] + vpart_num;
   1232  1.16  reinoud 			free_lbs = udf_rw32(*freepos);
   1233  1.16  reinoud 			*freepos = udf_rw32(free_lbs - num_lb);
   1234  1.16  reinoud 		}
   1235  1.16  reinoud 		break;
   1236  1.15  reinoud 	case UDF_ALLOC_METASEQUENTIAL :		/* UDF 2.60       BluRay-R  */
   1237  1.15  reinoud 	case UDF_ALLOC_RELAXEDSEQUENTIAL :	/* UDF 2.50/~meta BluRay-R  */
   1238   1.1  reinoud 		printf("ALERT: udf_allocate_space : allocation %d "
   1239   1.1  reinoud 				"not implemented yet!\n", alloc_type);
   1240   1.1  reinoud 		/* TODO implement, doesn't have to be contiguous */
   1241   1.1  reinoud 		error = ENOSPC;
   1242   1.1  reinoud 		break;
   1243   1.1  reinoud 	}
   1244   1.1  reinoud 
   1245  1.26  reinoud 	if (!error) {
   1246  1.26  reinoud 		/* credit our partition since we have committed the space */
   1247  1.26  reinoud 		udf_do_unreserve_space(ump, udf_node, vpart_num, num_lb);
   1248  1.26  reinoud 	}
   1249  1.26  reinoud 
   1250   1.1  reinoud #ifdef DEBUG
   1251   1.1  reinoud 	if (udf_verbose & UDF_DEBUG_ALLOC) {
   1252   1.1  reinoud 		lmappos = lmapping;
   1253  1.16  reinoud 		printf("udf_allocate_space, allocated logical lba :\n");
   1254   1.1  reinoud 		for (lb_num = 0; lb_num < num_lb; lb_num++) {
   1255  1.25  reinoud 			printf("%s %"PRIu64, (lb_num > 0)?",":"",
   1256  1.16  reinoud 				*lmappos++);
   1257   1.1  reinoud 		}
   1258  1.16  reinoud 		printf("\n");
   1259   1.1  reinoud 	}
   1260   1.1  reinoud #endif
   1261   1.1  reinoud 	mutex_exit(&ump->allocate_mutex);
   1262   1.1  reinoud 
   1263   1.1  reinoud 	return error;
   1264   1.1  reinoud }
   1265   1.1  reinoud 
   1266   1.1  reinoud /* --------------------------------------------------------------------- */
   1267   1.1  reinoud 
   1268   1.1  reinoud void
   1269   1.1  reinoud udf_free_allocated_space(struct udf_mount *ump, uint32_t lb_num,
   1270   1.1  reinoud 	uint16_t vpart_num, uint32_t num_lb)
   1271   1.1  reinoud {
   1272   1.1  reinoud 	struct udf_bitmap *bitmap;
   1273   1.1  reinoud 	struct part_desc *pdesc;
   1274   1.1  reinoud 	struct logvol_int_desc *lvid;
   1275   1.1  reinoud 	uint32_t ptov, lb_map, udf_rw32_lbmap;
   1276   1.1  reinoud 	uint32_t *freepos, free_lbs;
   1277   1.1  reinoud 	int phys_part;
   1278   1.1  reinoud 	int error;
   1279   1.1  reinoud 
   1280   1.1  reinoud 	DPRINTF(ALLOC, ("udf_free_allocated_space: freeing virt lbnum %d "
   1281   1.1  reinoud 			  "part %d + %d sect\n", lb_num, vpart_num, num_lb));
   1282   1.1  reinoud 
   1283  1.10  reinoud 	/* no use freeing zero length */
   1284  1.10  reinoud 	if (num_lb == 0)
   1285  1.10  reinoud 		return;
   1286  1.10  reinoud 
   1287   1.1  reinoud 	mutex_enter(&ump->allocate_mutex);
   1288   1.1  reinoud 
   1289   1.1  reinoud 	/* get partition backing up this vpart_num */
   1290   1.1  reinoud 	pdesc = ump->partitions[ump->vtop[vpart_num]];
   1291   1.1  reinoud 
   1292   1.1  reinoud 	switch (ump->vtop_tp[vpart_num]) {
   1293   1.1  reinoud 	case UDF_VTOP_TYPE_PHYS :
   1294   1.1  reinoud 	case UDF_VTOP_TYPE_SPARABLE :
   1295   1.1  reinoud 		/* free space to freed or unallocated space bitmap */
   1296   1.1  reinoud 		ptov      = udf_rw32(pdesc->start_loc);
   1297   1.1  reinoud 		phys_part = ump->vtop[vpart_num];
   1298   1.1  reinoud 
   1299   1.1  reinoud 		/* first try freed space bitmap */
   1300   1.1  reinoud 		bitmap    = &ump->part_freed_bits[phys_part];
   1301   1.1  reinoud 
   1302   1.1  reinoud 		/* if not defined, use unallocated bitmap */
   1303   1.1  reinoud 		if (bitmap->bits == NULL)
   1304   1.1  reinoud 			bitmap = &ump->part_unalloc_bits[phys_part];
   1305   1.1  reinoud 
   1306  1.17  reinoud 		/* if no bitmaps are defined, bail out; XXX OK? */
   1307   1.1  reinoud 		if (bitmap->bits == NULL)
   1308   1.1  reinoud 			break;
   1309   1.1  reinoud 
   1310   1.1  reinoud 		/* free bits if its defined */
   1311   1.1  reinoud 		KASSERT(bitmap->bits);
   1312   1.1  reinoud 		ump->lvclose |= UDF_WRITE_PART_BITMAPS;
   1313   1.1  reinoud 		udf_bitmap_free(bitmap, lb_num, num_lb);
   1314   1.1  reinoud 
   1315   1.1  reinoud 		/* adjust freecount */
   1316   1.1  reinoud 		lvid = ump->logvol_integrity;
   1317   1.1  reinoud 		freepos = &lvid->tables[0] + vpart_num;
   1318   1.1  reinoud 		free_lbs = udf_rw32(*freepos);
   1319   1.1  reinoud 		*freepos = udf_rw32(free_lbs + num_lb);
   1320   1.1  reinoud 		break;
   1321   1.1  reinoud 	case UDF_VTOP_TYPE_VIRT :
   1322   1.1  reinoud 		/* free this VAT entry */
   1323   1.1  reinoud 		KASSERT(num_lb == 1);
   1324   1.1  reinoud 
   1325   1.1  reinoud 		lb_map = 0xffffffff;
   1326   1.1  reinoud 		udf_rw32_lbmap = udf_rw32(lb_map);
   1327   1.1  reinoud 		error = udf_vat_write(ump->vat_node,
   1328   1.1  reinoud 			(uint8_t *) &udf_rw32_lbmap, 4,
   1329   1.1  reinoud 			ump->vat_offset + lb_num * 4);
   1330   1.1  reinoud 		KASSERT(error == 0);
   1331   1.1  reinoud 		ump->vat_last_free_lb = MIN(ump->vat_last_free_lb, lb_num);
   1332   1.1  reinoud 		break;
   1333   1.1  reinoud 	case UDF_VTOP_TYPE_META :
   1334   1.1  reinoud 		/* free space in the metadata bitmap */
   1335  1.17  reinoud 		bitmap = &ump->metadata_unalloc_bits;
   1336  1.17  reinoud 		KASSERT(bitmap->bits);
   1337  1.17  reinoud 
   1338  1.17  reinoud 		ump->lvclose |= UDF_WRITE_PART_BITMAPS;
   1339  1.17  reinoud 		udf_bitmap_free(bitmap, lb_num, num_lb);
   1340  1.17  reinoud 
   1341  1.17  reinoud 		/* adjust freecount */
   1342  1.17  reinoud 		lvid = ump->logvol_integrity;
   1343  1.17  reinoud 		freepos = &lvid->tables[0] + vpart_num;
   1344  1.17  reinoud 		free_lbs = udf_rw32(*freepos);
   1345  1.17  reinoud 		*freepos = udf_rw32(free_lbs + num_lb);
   1346  1.17  reinoud 		break;
   1347   1.1  reinoud 	default:
   1348   1.1  reinoud 		printf("ALERT: udf_free_allocated_space : allocation %d "
   1349   1.1  reinoud 			"not implemented yet!\n", ump->vtop_tp[vpart_num]);
   1350   1.1  reinoud 		break;
   1351   1.1  reinoud 	}
   1352   1.1  reinoud 
   1353   1.1  reinoud 	mutex_exit(&ump->allocate_mutex);
   1354   1.1  reinoud }
   1355   1.1  reinoud 
   1356   1.1  reinoud /* --------------------------------------------------------------------- */
   1357   1.1  reinoud 
   1358   1.1  reinoud /*
   1359   1.1  reinoud  * Allocate a buf on disc for direct write out. The space doesn't have to be
   1360   1.1  reinoud  * contiguous as the caller takes care of this.
   1361   1.1  reinoud  */
   1362   1.1  reinoud 
   1363   1.1  reinoud void
   1364   1.1  reinoud udf_late_allocate_buf(struct udf_mount *ump, struct buf *buf,
   1365  1.17  reinoud 	uint64_t *lmapping, struct long_ad *node_ad_cpy, uint16_t *vpart_nump)
   1366   1.1  reinoud {
   1367   1.1  reinoud 	struct udf_node  *udf_node = VTOI(buf->b_vp);
   1368   1.1  reinoud 	int lb_size, blks, udf_c_type;
   1369  1.17  reinoud 	int vpart_num, num_lb;
   1370   1.1  reinoud 	int error, s;
   1371   1.1  reinoud 
   1372   1.1  reinoud 	/*
   1373   1.1  reinoud 	 * for each sector in the buf, allocate a sector on disc and record
   1374   1.1  reinoud 	 * its position in the provided mapping array.
   1375   1.1  reinoud 	 *
   1376   1.1  reinoud 	 * If its userdata or FIDs, record its location in its node.
   1377   1.1  reinoud 	 */
   1378   1.1  reinoud 
   1379   1.1  reinoud 	lb_size    = udf_rw32(ump->logical_vol->lb_size);
   1380   1.1  reinoud 	num_lb     = (buf->b_bcount + lb_size -1) / lb_size;
   1381   1.1  reinoud 	blks       = lb_size / DEV_BSIZE;
   1382   1.1  reinoud 	udf_c_type = buf->b_udf_c_type;
   1383   1.1  reinoud 
   1384   1.1  reinoud 	KASSERT(lb_size == ump->discinfo.sector_size);
   1385   1.1  reinoud 
   1386  1.17  reinoud 	/* select partition to record the buffer on */
   1387  1.26  reinoud 	vpart_num = *vpart_nump = udf_get_record_vpart(ump, udf_c_type);
   1388   1.1  reinoud 
   1389   1.1  reinoud 	if (udf_c_type == UDF_C_NODE) {
   1390   1.1  reinoud 		/* if not VAT, its allready allocated */
   1391  1.17  reinoud 		if (ump->vtop_alloc[ump->node_part] != UDF_ALLOC_VAT)
   1392   1.1  reinoud 			return;
   1393   1.1  reinoud 
   1394  1.17  reinoud 		/* allocate on its backing sequential partition */
   1395  1.17  reinoud 		vpart_num = ump->data_part;
   1396   1.1  reinoud 	}
   1397   1.1  reinoud 
   1398  1.26  reinoud 	/* XXX can this still happen? */
   1399  1.17  reinoud 	/* do allocation on the selected partition */
   1400  1.26  reinoud 	error = udf_allocate_space(ump, udf_node, udf_c_type,
   1401  1.17  reinoud 			vpart_num, num_lb, lmapping);
   1402   1.1  reinoud 	if (error) {
   1403  1.26  reinoud 		/*
   1404  1.26  reinoud 		 * ARGH! we haven't done our accounting right! it should
   1405  1.26  reinoud 		 * allways succeed.
   1406  1.26  reinoud 		 */
   1407   1.1  reinoud 		panic("UDF disc allocation accounting gone wrong");
   1408   1.1  reinoud 	}
   1409   1.1  reinoud 
   1410   1.1  reinoud 	/* If its userdata or FIDs, record its allocation in its node. */
   1411  1.16  reinoud 	if ((udf_c_type == UDF_C_USERDATA) ||
   1412  1.16  reinoud 	    (udf_c_type == UDF_C_FIDS) ||
   1413  1.16  reinoud 	    (udf_c_type == UDF_C_METADATA_SBM))
   1414  1.16  reinoud 	{
   1415  1.17  reinoud 		udf_record_allocation_in_node(ump, buf, vpart_num, lmapping,
   1416   1.1  reinoud 			node_ad_cpy);
   1417   1.1  reinoud 		/* decrement our outstanding bufs counter */
   1418   1.1  reinoud 		s = splbio();
   1419   1.1  reinoud 			udf_node->outstanding_bufs--;
   1420   1.1  reinoud 		splx(s);
   1421   1.1  reinoud 	}
   1422   1.1  reinoud }
   1423   1.1  reinoud 
   1424   1.1  reinoud /* --------------------------------------------------------------------- */
   1425   1.1  reinoud 
   1426   1.1  reinoud /*
   1427   1.1  reinoud  * Try to merge a1 with the new piece a2. udf_ads_merge returns error when not
   1428   1.1  reinoud  * possible (anymore); a2 returns the rest piece.
   1429   1.1  reinoud  */
   1430   1.1  reinoud 
   1431   1.1  reinoud static int
   1432   1.1  reinoud udf_ads_merge(uint32_t lb_size, struct long_ad *a1, struct long_ad *a2)
   1433   1.1  reinoud {
   1434   1.1  reinoud 	uint32_t max_len, merge_len;
   1435   1.1  reinoud 	uint32_t a1_len, a2_len;
   1436   1.1  reinoud 	uint32_t a1_flags, a2_flags;
   1437   1.1  reinoud 	uint32_t a1_lbnum, a2_lbnum;
   1438   1.1  reinoud 	uint16_t a1_part, a2_part;
   1439   1.1  reinoud 
   1440   1.1  reinoud 	max_len = ((UDF_EXT_MAXLEN / lb_size) * lb_size);
   1441   1.1  reinoud 
   1442   1.1  reinoud 	a1_flags = UDF_EXT_FLAGS(udf_rw32(a1->len));
   1443   1.1  reinoud 	a1_len   = UDF_EXT_LEN(udf_rw32(a1->len));
   1444   1.1  reinoud 	a1_lbnum = udf_rw32(a1->loc.lb_num);
   1445   1.1  reinoud 	a1_part  = udf_rw16(a1->loc.part_num);
   1446   1.1  reinoud 
   1447   1.1  reinoud 	a2_flags = UDF_EXT_FLAGS(udf_rw32(a2->len));
   1448   1.1  reinoud 	a2_len   = UDF_EXT_LEN(udf_rw32(a2->len));
   1449   1.1  reinoud 	a2_lbnum = udf_rw32(a2->loc.lb_num);
   1450   1.1  reinoud 	a2_part  = udf_rw16(a2->loc.part_num);
   1451   1.1  reinoud 
   1452   1.1  reinoud 	/* defines same space */
   1453   1.1  reinoud 	if (a1_flags != a2_flags)
   1454   1.1  reinoud 		return 1;
   1455   1.1  reinoud 
   1456   1.1  reinoud 	if (a1_flags != UDF_EXT_FREE) {
   1457   1.1  reinoud 		/* the same partition */
   1458   1.1  reinoud 		if (a1_part != a2_part)
   1459   1.1  reinoud 			return 1;
   1460   1.1  reinoud 
   1461   1.1  reinoud 		/* a2 is successor of a1 */
   1462   1.1  reinoud 		if (a1_lbnum * lb_size + a1_len != a2_lbnum * lb_size)
   1463   1.1  reinoud 			return 1;
   1464   1.1  reinoud 	}
   1465   1.9  reinoud 
   1466   1.1  reinoud 	/* merge as most from a2 if possible */
   1467   1.1  reinoud 	merge_len = MIN(a2_len, max_len - a1_len);
   1468   1.1  reinoud 	a1_len   += merge_len;
   1469   1.1  reinoud 	a2_len   -= merge_len;
   1470   1.1  reinoud 	a2_lbnum += merge_len/lb_size;
   1471   1.1  reinoud 
   1472   1.1  reinoud 	a1->len = udf_rw32(a1_len | a1_flags);
   1473   1.1  reinoud 	a2->len = udf_rw32(a2_len | a2_flags);
   1474   1.1  reinoud 	a2->loc.lb_num = udf_rw32(a2_lbnum);
   1475   1.1  reinoud 
   1476   1.1  reinoud 	if (a2_len > 0)
   1477   1.1  reinoud 		return 1;
   1478   1.1  reinoud 
   1479   1.1  reinoud 	/* there is space over to merge */
   1480   1.1  reinoud 	return 0;
   1481   1.1  reinoud }
   1482   1.1  reinoud 
   1483   1.1  reinoud /* --------------------------------------------------------------------- */
   1484   1.1  reinoud 
   1485   1.1  reinoud static void
   1486   1.1  reinoud udf_wipe_adslots(struct udf_node *udf_node)
   1487   1.1  reinoud {
   1488   1.1  reinoud 	struct file_entry      *fe;
   1489   1.1  reinoud 	struct extfile_entry   *efe;
   1490   1.1  reinoud 	struct alloc_ext_entry *ext;
   1491   1.1  reinoud 	uint64_t inflen, objsize;
   1492   1.1  reinoud 	uint32_t lb_size, dscr_size, l_ea, l_ad, max_l_ad, crclen;
   1493   1.1  reinoud 	uint8_t *data_pos;
   1494   1.1  reinoud 	int extnr;
   1495   1.1  reinoud 
   1496   1.1  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
   1497   1.1  reinoud 
   1498   1.1  reinoud 	fe  = udf_node->fe;
   1499   1.1  reinoud 	efe = udf_node->efe;
   1500   1.1  reinoud 	if (fe) {
   1501   1.1  reinoud 		inflen  = udf_rw64(fe->inf_len);
   1502   1.1  reinoud 		objsize = inflen;
   1503   1.1  reinoud 		dscr_size  = sizeof(struct file_entry) -1;
   1504   1.1  reinoud 		l_ea       = udf_rw32(fe->l_ea);
   1505   1.1  reinoud 		l_ad       = udf_rw32(fe->l_ad);
   1506   1.1  reinoud 		data_pos = (uint8_t *) fe + dscr_size + l_ea;
   1507   1.1  reinoud 	} else {
   1508   1.1  reinoud 		inflen  = udf_rw64(efe->inf_len);
   1509   1.1  reinoud 		objsize = udf_rw64(efe->obj_size);
   1510   1.1  reinoud 		dscr_size  = sizeof(struct extfile_entry) -1;
   1511   1.1  reinoud 		l_ea       = udf_rw32(efe->l_ea);
   1512   1.1  reinoud 		l_ad       = udf_rw32(efe->l_ad);
   1513   1.1  reinoud 		data_pos = (uint8_t *) efe + dscr_size + l_ea;
   1514   1.1  reinoud 	}
   1515   1.1  reinoud 	max_l_ad = lb_size - dscr_size - l_ea;
   1516   1.1  reinoud 
   1517   1.1  reinoud 	/* wipe fe/efe */
   1518   1.1  reinoud 	memset(data_pos, 0, max_l_ad);
   1519   1.1  reinoud 	crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea;
   1520   1.1  reinoud 	if (fe) {
   1521   1.1  reinoud 		fe->l_ad         = udf_rw32(0);
   1522   1.1  reinoud 		fe->logblks_rec  = udf_rw64(0);
   1523  1.22  reinoud 		fe->tag.desc_crc_len = udf_rw16(crclen);
   1524   1.1  reinoud 	} else {
   1525   1.1  reinoud 		efe->l_ad        = udf_rw32(0);
   1526   1.1  reinoud 		efe->logblks_rec = udf_rw64(0);
   1527  1.22  reinoud 		efe->tag.desc_crc_len = udf_rw16(crclen);
   1528   1.1  reinoud 	}
   1529   1.1  reinoud 
   1530   1.1  reinoud 	/* wipe all allocation extent entries */
   1531   1.1  reinoud 	for (extnr = 0; extnr < udf_node->num_extensions; extnr++) {
   1532   1.1  reinoud 		ext = udf_node->ext[extnr];
   1533   1.1  reinoud 		dscr_size  = sizeof(struct alloc_ext_entry) -1;
   1534  1.11  reinoud 		data_pos = (uint8_t *) ext->data;
   1535   1.1  reinoud 		max_l_ad = lb_size - dscr_size;
   1536   1.1  reinoud 		memset(data_pos, 0, max_l_ad);
   1537   1.1  reinoud 		ext->l_ad = udf_rw32(0);
   1538   1.1  reinoud 
   1539   1.1  reinoud 		crclen = dscr_size - UDF_DESC_TAG_LENGTH;
   1540  1.22  reinoud 		ext->tag.desc_crc_len = udf_rw16(crclen);
   1541   1.1  reinoud 	}
   1542  1.11  reinoud 	udf_node->i_flags |= IN_NODE_REBUILD;
   1543   1.1  reinoud }
   1544   1.1  reinoud 
   1545   1.1  reinoud /* --------------------------------------------------------------------- */
   1546   1.1  reinoud 
   1547   1.1  reinoud void
   1548   1.1  reinoud udf_get_adslot(struct udf_node *udf_node, int slot, struct long_ad *icb,
   1549   1.1  reinoud 	int *eof) {
   1550   1.1  reinoud 	struct file_entry      *fe;
   1551   1.1  reinoud 	struct extfile_entry   *efe;
   1552   1.1  reinoud 	struct alloc_ext_entry *ext;
   1553   1.1  reinoud 	struct icb_tag *icbtag;
   1554   1.1  reinoud 	struct short_ad *short_ad;
   1555  1.11  reinoud 	struct long_ad *long_ad, l_icb;
   1556   1.1  reinoud 	uint32_t offset;
   1557  1.11  reinoud 	uint32_t lb_size, dscr_size, l_ea, l_ad, flags;
   1558   1.1  reinoud 	uint8_t *data_pos;
   1559   1.1  reinoud 	int icbflags, addr_type, adlen, extnr;
   1560   1.1  reinoud 
   1561   1.1  reinoud 	/* determine what descriptor we are in */
   1562   1.1  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
   1563   1.1  reinoud 
   1564   1.1  reinoud 	fe  = udf_node->fe;
   1565   1.1  reinoud 	efe = udf_node->efe;
   1566   1.1  reinoud 	if (fe) {
   1567   1.1  reinoud 		icbtag  = &fe->icbtag;
   1568   1.1  reinoud 		dscr_size  = sizeof(struct file_entry) -1;
   1569   1.1  reinoud 		l_ea       = udf_rw32(fe->l_ea);
   1570   1.1  reinoud 		l_ad       = udf_rw32(fe->l_ad);
   1571   1.1  reinoud 		data_pos = (uint8_t *) fe + dscr_size + l_ea;
   1572   1.1  reinoud 	} else {
   1573   1.1  reinoud 		icbtag  = &efe->icbtag;
   1574   1.1  reinoud 		dscr_size  = sizeof(struct extfile_entry) -1;
   1575   1.1  reinoud 		l_ea       = udf_rw32(efe->l_ea);
   1576   1.1  reinoud 		l_ad       = udf_rw32(efe->l_ad);
   1577   1.1  reinoud 		data_pos = (uint8_t *) efe + dscr_size + l_ea;
   1578   1.1  reinoud 	}
   1579   1.1  reinoud 
   1580   1.1  reinoud 	icbflags  = udf_rw16(icbtag->flags);
   1581   1.1  reinoud 	addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   1582   1.1  reinoud 
   1583   1.1  reinoud 	/* just in case we're called on an intern, its EOF */
   1584   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
   1585   1.1  reinoud 		memset(icb, 0, sizeof(struct long_ad));
   1586   1.1  reinoud 		*eof = 1;
   1587   1.1  reinoud 		return;
   1588   1.1  reinoud 	}
   1589   1.1  reinoud 
   1590   1.1  reinoud 	adlen = 0;
   1591   1.1  reinoud 	if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1592   1.1  reinoud 		adlen = sizeof(struct short_ad);
   1593   1.1  reinoud 	} else if (addr_type == UDF_ICB_LONG_ALLOC) {
   1594   1.1  reinoud 		adlen = sizeof(struct long_ad);
   1595   1.1  reinoud 	}
   1596   1.1  reinoud 
   1597   1.1  reinoud 	/* if offset too big, we go to the allocation extensions */
   1598   1.1  reinoud 	offset = slot * adlen;
   1599   1.3  reinoud 	extnr  = -1;
   1600  1.11  reinoud 	while (offset >= l_ad) {
   1601  1.11  reinoud 		/* check if our last entry is a redirect */
   1602  1.11  reinoud 		if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1603  1.11  reinoud 			short_ad = (struct short_ad *) (data_pos + l_ad-adlen);
   1604  1.11  reinoud 			l_icb.len          = short_ad->len;
   1605  1.11  reinoud 			l_icb.loc.part_num = udf_node->loc.loc.part_num;
   1606  1.11  reinoud 			l_icb.loc.lb_num   = short_ad->lb_num;
   1607  1.11  reinoud 		} else {
   1608  1.11  reinoud 			KASSERT(addr_type == UDF_ICB_LONG_ALLOC);
   1609  1.11  reinoud 			long_ad = (struct long_ad *) (data_pos + l_ad-adlen);
   1610  1.11  reinoud 			l_icb = *long_ad;
   1611  1.11  reinoud 		}
   1612  1.11  reinoud 		flags = UDF_EXT_FLAGS(udf_rw32(l_icb.len));
   1613  1.11  reinoud 		if (flags != UDF_EXT_REDIRECT) {
   1614  1.11  reinoud 			l_ad = 0;	/* force EOF */
   1615  1.11  reinoud 			break;
   1616  1.11  reinoud 		}
   1617  1.11  reinoud 
   1618  1.11  reinoud 		/* advance to next extent */
   1619   1.3  reinoud 		extnr++;
   1620  1.11  reinoud 		if (extnr >= udf_node->num_extensions) {
   1621  1.11  reinoud 			l_ad = 0;	/* force EOF */
   1622  1.11  reinoud 			break;
   1623  1.11  reinoud 		}
   1624  1.11  reinoud 		offset = offset - l_ad;
   1625   1.1  reinoud 		ext  = udf_node->ext[extnr];
   1626   1.1  reinoud 		dscr_size  = sizeof(struct alloc_ext_entry) -1;
   1627   1.1  reinoud 		l_ad = udf_rw32(ext->l_ad);
   1628   1.3  reinoud 		data_pos = (uint8_t *) ext + dscr_size;
   1629   1.1  reinoud 	}
   1630   1.1  reinoud 
   1631  1.11  reinoud 	/* XXX l_ad == 0 should be enough to check */
   1632   1.1  reinoud 	*eof = (offset >= l_ad) || (l_ad == 0);
   1633   1.1  reinoud 	if (*eof) {
   1634  1.11  reinoud 		DPRINTF(PARANOIDADWLK, ("returning EOF, extnr %d, offset %d, "
   1635  1.11  reinoud 			"l_ad %d\n", extnr, offset, l_ad));
   1636   1.1  reinoud 		memset(icb, 0, sizeof(struct long_ad));
   1637   1.1  reinoud 		return;
   1638   1.1  reinoud 	}
   1639   1.1  reinoud 
   1640   1.1  reinoud 	/* get the element */
   1641   1.1  reinoud 	if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1642   1.1  reinoud 		short_ad = (struct short_ad *) (data_pos + offset);
   1643   1.1  reinoud 		icb->len          = short_ad->len;
   1644   1.5  reinoud 		icb->loc.part_num = udf_node->loc.loc.part_num;
   1645   1.1  reinoud 		icb->loc.lb_num   = short_ad->lb_num;
   1646   1.1  reinoud 	} else if (addr_type == UDF_ICB_LONG_ALLOC) {
   1647   1.1  reinoud 		long_ad = (struct long_ad *) (data_pos + offset);
   1648   1.1  reinoud 		*icb = *long_ad;
   1649   1.1  reinoud 	}
   1650  1.11  reinoud 	DPRINTF(PARANOIDADWLK, ("returning element : v %d, lb %d, len %d, "
   1651  1.11  reinoud 		"flags %d\n", icb->loc.part_num, icb->loc.lb_num,
   1652  1.11  reinoud 		UDF_EXT_LEN(icb->len), UDF_EXT_FLAGS(icb->len)));
   1653   1.1  reinoud }
   1654   1.1  reinoud 
   1655   1.1  reinoud /* --------------------------------------------------------------------- */
   1656   1.1  reinoud 
   1657   1.1  reinoud int
   1658  1.11  reinoud udf_append_adslot(struct udf_node *udf_node, int *slot, struct long_ad *icb) {
   1659  1.11  reinoud 	struct udf_mount *ump = udf_node->ump;
   1660  1.13  reinoud 	union dscrptr          *dscr, *extdscr;
   1661   1.1  reinoud 	struct file_entry      *fe;
   1662   1.1  reinoud 	struct extfile_entry   *efe;
   1663   1.1  reinoud 	struct alloc_ext_entry *ext;
   1664   1.1  reinoud 	struct icb_tag *icbtag;
   1665   1.1  reinoud 	struct short_ad *short_ad;
   1666  1.11  reinoud 	struct long_ad *long_ad, o_icb, l_icb;
   1667   1.1  reinoud 	uint64_t logblks_rec, *logblks_rec_p;
   1668  1.16  reinoud 	uint64_t lmapping;
   1669  1.11  reinoud 	uint32_t offset, rest, len, lb_num;
   1670   1.1  reinoud 	uint32_t lb_size, dscr_size, l_ea, l_ad, *l_ad_p, max_l_ad, crclen;
   1671  1.11  reinoud 	uint32_t flags;
   1672  1.11  reinoud 	uint16_t vpart_num;
   1673   1.1  reinoud 	uint8_t *data_pos;
   1674   1.1  reinoud 	int icbflags, addr_type, adlen, extnr;
   1675  1.11  reinoud 	int error;
   1676   1.1  reinoud 
   1677  1.11  reinoud 	lb_size = udf_rw32(ump->logical_vol->lb_size);
   1678  1.17  reinoud 	vpart_num = udf_rw16(udf_node->loc.loc.part_num);
   1679   1.1  reinoud 
   1680  1.17  reinoud 	/* determine what descriptor we are in */
   1681   1.1  reinoud 	fe  = udf_node->fe;
   1682   1.1  reinoud 	efe = udf_node->efe;
   1683   1.1  reinoud 	if (fe) {
   1684   1.1  reinoud 		icbtag  = &fe->icbtag;
   1685   1.1  reinoud 		dscr      = (union dscrptr *) fe;
   1686   1.1  reinoud 		dscr_size = sizeof(struct file_entry) -1;
   1687   1.1  reinoud 
   1688   1.1  reinoud 		l_ea      = udf_rw32(fe->l_ea);
   1689   1.1  reinoud 		l_ad_p    = &fe->l_ad;
   1690   1.1  reinoud 		logblks_rec_p = &fe->logblks_rec;
   1691   1.1  reinoud 	} else {
   1692   1.1  reinoud 		icbtag    = &efe->icbtag;
   1693   1.1  reinoud 		dscr      = (union dscrptr *) efe;
   1694   1.1  reinoud 		dscr_size = sizeof(struct extfile_entry) -1;
   1695   1.1  reinoud 
   1696   1.1  reinoud 		l_ea      = udf_rw32(efe->l_ea);
   1697   1.1  reinoud 		l_ad_p    = &efe->l_ad;
   1698   1.1  reinoud 		logblks_rec_p = &efe->logblks_rec;
   1699   1.1  reinoud 	}
   1700   1.1  reinoud 	data_pos  = (uint8_t *) dscr + dscr_size + l_ea;
   1701   1.1  reinoud 	max_l_ad = lb_size - dscr_size - l_ea;
   1702   1.1  reinoud 
   1703   1.1  reinoud 	icbflags  = udf_rw16(icbtag->flags);
   1704   1.1  reinoud 	addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   1705   1.1  reinoud 
   1706   1.1  reinoud 	/* just in case we're called on an intern, its EOF */
   1707   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
   1708   1.1  reinoud 		panic("udf_append_adslot on UDF_ICB_INTERN_ALLOC\n");
   1709   1.1  reinoud 	}
   1710   1.1  reinoud 
   1711   1.1  reinoud 	adlen = 0;
   1712   1.1  reinoud 	if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1713   1.1  reinoud 		adlen = sizeof(struct short_ad);
   1714   1.1  reinoud 	} else if (addr_type == UDF_ICB_LONG_ALLOC) {
   1715   1.1  reinoud 		adlen = sizeof(struct long_ad);
   1716   1.1  reinoud 	}
   1717   1.1  reinoud 
   1718  1.17  reinoud 	/* clean up given long_ad since it can be a synthesized one */
   1719  1.11  reinoud 	flags = UDF_EXT_FLAGS(udf_rw32(icb->len));
   1720  1.11  reinoud 	if (flags == UDF_EXT_FREE) {
   1721   1.9  reinoud 		icb->loc.part_num = udf_rw16(0);
   1722   1.9  reinoud 		icb->loc.lb_num   = udf_rw32(0);
   1723   1.9  reinoud 	}
   1724   1.9  reinoud 
   1725   1.1  reinoud 	/* if offset too big, we go to the allocation extensions */
   1726  1.11  reinoud 	l_ad   = udf_rw32(*l_ad_p);
   1727  1.11  reinoud 	offset = (*slot) * adlen;
   1728  1.11  reinoud 	extnr  = -1;
   1729  1.11  reinoud 	while (offset >= l_ad) {
   1730  1.11  reinoud 		/* check if our last entry is a redirect */
   1731  1.11  reinoud 		if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1732  1.11  reinoud 			short_ad = (struct short_ad *) (data_pos + l_ad-adlen);
   1733  1.11  reinoud 			l_icb.len          = short_ad->len;
   1734  1.11  reinoud 			l_icb.loc.part_num = udf_node->loc.loc.part_num;
   1735  1.11  reinoud 			l_icb.loc.lb_num   = short_ad->lb_num;
   1736  1.11  reinoud 		} else {
   1737  1.11  reinoud 			KASSERT(addr_type == UDF_ICB_LONG_ALLOC);
   1738  1.11  reinoud 			long_ad = (struct long_ad *) (data_pos + l_ad-adlen);
   1739  1.11  reinoud 			l_icb = *long_ad;
   1740  1.11  reinoud 		}
   1741  1.11  reinoud 		flags = UDF_EXT_FLAGS(udf_rw32(l_icb.len));
   1742  1.11  reinoud 		if (flags != UDF_EXT_REDIRECT) {
   1743  1.11  reinoud 			/* only one past the last one is adressable */
   1744  1.11  reinoud 			break;
   1745  1.11  reinoud 		}
   1746  1.11  reinoud 
   1747  1.11  reinoud 		/* advance to next extent */
   1748  1.11  reinoud 		extnr++;
   1749  1.11  reinoud 		KASSERT(extnr < udf_node->num_extensions);
   1750  1.11  reinoud 		offset = offset - l_ad;
   1751  1.11  reinoud 
   1752   1.1  reinoud 		ext  = udf_node->ext[extnr];
   1753   1.1  reinoud 		dscr = (union dscrptr *) ext;
   1754   1.1  reinoud 		dscr_size  = sizeof(struct alloc_ext_entry) -1;
   1755  1.11  reinoud 		max_l_ad = lb_size - dscr_size;
   1756   1.1  reinoud 		l_ad_p = &ext->l_ad;
   1757  1.11  reinoud 		l_ad   = udf_rw32(*l_ad_p);
   1758  1.11  reinoud 		data_pos = (uint8_t *) ext + dscr_size;
   1759  1.11  reinoud 	}
   1760  1.11  reinoud 	DPRINTF(PARANOIDADWLK, ("append, ext %d, offset %d, l_ad %d\n",
   1761  1.11  reinoud 		extnr, offset, udf_rw32(*l_ad_p)));
   1762  1.11  reinoud 	KASSERT(l_ad == udf_rw32(*l_ad_p));
   1763   1.1  reinoud 
   1764   1.1  reinoud 	/* offset is offset within the current (E)FE/AED */
   1765   1.1  reinoud 	l_ad   = udf_rw32(*l_ad_p);
   1766  1.22  reinoud 	crclen = udf_rw16(dscr->tag.desc_crc_len);
   1767   1.1  reinoud 	logblks_rec = udf_rw64(*logblks_rec_p);
   1768   1.1  reinoud 
   1769   1.1  reinoud 	/* overwriting old piece? */
   1770   1.1  reinoud 	if (offset < l_ad) {
   1771   1.1  reinoud 		/* overwrite entry; compensate for the old element */
   1772   1.1  reinoud 		if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1773   1.1  reinoud 			short_ad = (struct short_ad *) (data_pos + offset);
   1774   1.1  reinoud 			o_icb.len          = short_ad->len;
   1775   1.1  reinoud 			o_icb.loc.part_num = udf_rw16(0);	/* ignore */
   1776   1.1  reinoud 			o_icb.loc.lb_num   = short_ad->lb_num;
   1777   1.1  reinoud 		} else if (addr_type == UDF_ICB_LONG_ALLOC) {
   1778   1.1  reinoud 			long_ad = (struct long_ad *) (data_pos + offset);
   1779   1.1  reinoud 			o_icb = *long_ad;
   1780   1.1  reinoud 		} else {
   1781   1.1  reinoud 			panic("Invalid address type in udf_append_adslot\n");
   1782   1.1  reinoud 		}
   1783   1.1  reinoud 
   1784   1.1  reinoud 		len = udf_rw32(o_icb.len);
   1785   1.1  reinoud 		if (UDF_EXT_FLAGS(len) == UDF_EXT_ALLOCATED) {
   1786   1.1  reinoud 			/* adjust counts */
   1787   1.1  reinoud 			len = UDF_EXT_LEN(len);
   1788   1.1  reinoud 			logblks_rec -= (len + lb_size -1) / lb_size;
   1789   1.1  reinoud 		}
   1790   1.1  reinoud 	}
   1791   1.1  reinoud 
   1792  1.11  reinoud 	/* check if we're not appending a redirection */
   1793  1.11  reinoud 	flags = UDF_EXT_FLAGS(udf_rw32(icb->len));
   1794  1.11  reinoud 	KASSERT(flags != UDF_EXT_REDIRECT);
   1795  1.11  reinoud 
   1796  1.11  reinoud 	/* round down available space */
   1797  1.11  reinoud 	rest = adlen * ((max_l_ad - offset) / adlen);
   1798   1.1  reinoud 	if (rest <= adlen) {
   1799  1.11  reinoud 		/* have to append aed, see if we already have a spare one */
   1800  1.11  reinoud 		extnr++;
   1801  1.11  reinoud 		ext = udf_node->ext[extnr];
   1802  1.11  reinoud 		l_icb = udf_node->ext_loc[extnr];
   1803  1.11  reinoud 		if (ext == NULL) {
   1804  1.11  reinoud 			DPRINTF(ALLOC,("adding allocation extent %d\n", extnr));
   1805  1.17  reinoud 
   1806  1.26  reinoud 			error = udf_reserve_space(ump, NULL, UDF_C_NODE,
   1807  1.26  reinoud 					vpart_num, 1, /* can fail */ false);
   1808  1.26  reinoud 			if (error) {
   1809  1.26  reinoud 				printf("UDF: couldn't reserve space for AED!\n");
   1810  1.26  reinoud 				return error;
   1811  1.26  reinoud 			}
   1812  1.26  reinoud 			error = udf_allocate_space(ump, NULL, UDF_C_NODE,
   1813  1.26  reinoud 					vpart_num, 1, &lmapping);
   1814  1.11  reinoud 			lb_num = lmapping;
   1815  1.11  reinoud 			if (error)
   1816  1.26  reinoud 				panic("UDF: couldn't allocate AED!\n");
   1817  1.11  reinoud 
   1818  1.11  reinoud 			/* initialise pointer to location */
   1819  1.11  reinoud 			memset(&l_icb, 0, sizeof(struct long_ad));
   1820  1.11  reinoud 			l_icb.len = udf_rw32(lb_size | UDF_EXT_REDIRECT);
   1821  1.11  reinoud 			l_icb.loc.lb_num   = udf_rw32(lb_num);
   1822  1.11  reinoud 			l_icb.loc.part_num = udf_rw16(vpart_num);
   1823  1.11  reinoud 
   1824  1.11  reinoud 			/* create new aed descriptor */
   1825  1.13  reinoud 			udf_create_logvol_dscr(ump, udf_node, &l_icb, &extdscr);
   1826  1.13  reinoud 			ext = &extdscr->aee;
   1827  1.11  reinoud 
   1828  1.11  reinoud 			udf_inittag(ump, &ext->tag, TAGID_ALLOCEXTENT, lb_num);
   1829  1.11  reinoud 			dscr_size  = sizeof(struct alloc_ext_entry) -1;
   1830  1.11  reinoud 			max_l_ad = lb_size - dscr_size;
   1831  1.11  reinoud 			memset(ext->data, 0, max_l_ad);
   1832  1.11  reinoud 			ext->l_ad = udf_rw32(0);
   1833  1.11  reinoud 			ext->tag.desc_crc_len =
   1834  1.22  reinoud 				udf_rw16(dscr_size - UDF_DESC_TAG_LENGTH);
   1835  1.11  reinoud 
   1836  1.11  reinoud 			/* declare aed */
   1837  1.11  reinoud 			udf_node->num_extensions++;
   1838  1.11  reinoud 			udf_node->ext_loc[extnr] = l_icb;
   1839  1.11  reinoud 			udf_node->ext[extnr] = ext;
   1840  1.11  reinoud 		}
   1841  1.11  reinoud 		/* add redirect and adjust l_ad and crclen for old descr */
   1842  1.11  reinoud 		if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1843  1.11  reinoud 			short_ad = (struct short_ad *) (data_pos + offset);
   1844  1.11  reinoud 			short_ad->len    = l_icb.len;
   1845  1.11  reinoud 			short_ad->lb_num = l_icb.loc.lb_num;
   1846  1.11  reinoud 		} else if (addr_type == UDF_ICB_LONG_ALLOC) {
   1847  1.11  reinoud 			long_ad = (struct long_ad *) (data_pos + offset);
   1848  1.11  reinoud 			*long_ad = l_icb;
   1849  1.11  reinoud 		}
   1850  1.11  reinoud 		l_ad   += adlen;
   1851  1.11  reinoud 		crclen += adlen;
   1852  1.22  reinoud 		dscr->tag.desc_crc_len = udf_rw16(crclen);
   1853  1.11  reinoud 		*l_ad_p = udf_rw32(l_ad);
   1854  1.11  reinoud 
   1855  1.11  reinoud 		/* advance to the new extension */
   1856  1.11  reinoud 		KASSERT(ext != NULL);
   1857  1.11  reinoud 		dscr = (union dscrptr *) ext;
   1858  1.11  reinoud 		dscr_size  = sizeof(struct alloc_ext_entry) -1;
   1859  1.11  reinoud 		max_l_ad = lb_size - dscr_size;
   1860  1.11  reinoud 		data_pos = (uint8_t *) dscr + dscr_size;
   1861  1.11  reinoud 
   1862  1.11  reinoud 		l_ad_p = &ext->l_ad;
   1863  1.11  reinoud 		l_ad   = udf_rw32(*l_ad_p);
   1864  1.22  reinoud 		crclen = udf_rw16(dscr->tag.desc_crc_len);
   1865  1.11  reinoud 		offset = 0;
   1866  1.11  reinoud 
   1867  1.11  reinoud 		/* adjust callees slot count for link insert */
   1868  1.11  reinoud 		*slot += 1;
   1869   1.1  reinoud 	}
   1870   1.1  reinoud 
   1871   1.1  reinoud 	/* write out the element */
   1872  1.11  reinoud 	DPRINTF(PARANOIDADWLK, ("adding element : %p : v %d, lb %d, "
   1873  1.11  reinoud 			"len %d, flags %d\n", data_pos + offset,
   1874  1.11  reinoud 			icb->loc.part_num, icb->loc.lb_num,
   1875  1.11  reinoud 			UDF_EXT_LEN(icb->len), UDF_EXT_FLAGS(icb->len)));
   1876   1.1  reinoud 	if (addr_type == UDF_ICB_SHORT_ALLOC) {
   1877   1.1  reinoud 		short_ad = (struct short_ad *) (data_pos + offset);
   1878   1.1  reinoud 		short_ad->len    = icb->len;
   1879   1.1  reinoud 		short_ad->lb_num = icb->loc.lb_num;
   1880   1.1  reinoud 	} else if (addr_type == UDF_ICB_LONG_ALLOC) {
   1881   1.1  reinoud 		long_ad = (struct long_ad *) (data_pos + offset);
   1882   1.1  reinoud 		*long_ad = *icb;
   1883   1.1  reinoud 	}
   1884   1.1  reinoud 
   1885   1.1  reinoud 	/* adjust logblks recorded count */
   1886  1.23  reinoud 	len = udf_rw32(icb->len);
   1887  1.23  reinoud 	flags = UDF_EXT_FLAGS(len);
   1888  1.11  reinoud 	if (flags == UDF_EXT_ALLOCATED)
   1889  1.23  reinoud 		logblks_rec += (UDF_EXT_LEN(len) + lb_size -1) / lb_size;
   1890   1.1  reinoud 	*logblks_rec_p = udf_rw64(logblks_rec);
   1891   1.1  reinoud 
   1892   1.1  reinoud 	/* adjust l_ad and crclen when needed */
   1893   1.1  reinoud 	if (offset >= l_ad) {
   1894   1.1  reinoud 		l_ad   += adlen;
   1895   1.1  reinoud 		crclen += adlen;
   1896  1.22  reinoud 		dscr->tag.desc_crc_len = udf_rw16(crclen);
   1897   1.1  reinoud 		*l_ad_p = udf_rw32(l_ad);
   1898   1.1  reinoud 	}
   1899   1.1  reinoud 
   1900   1.1  reinoud 	return 0;
   1901   1.1  reinoud }
   1902   1.1  reinoud 
   1903   1.1  reinoud /* --------------------------------------------------------------------- */
   1904   1.1  reinoud 
   1905  1.11  reinoud static void
   1906  1.11  reinoud udf_count_alloc_exts(struct udf_node *udf_node)
   1907  1.11  reinoud {
   1908  1.11  reinoud 	struct long_ad s_ad;
   1909  1.11  reinoud 	uint32_t lb_num, len, flags;
   1910  1.11  reinoud 	uint16_t vpart_num;
   1911  1.11  reinoud 	int slot, eof;
   1912  1.11  reinoud 	int num_extents, extnr;
   1913  1.11  reinoud 	int lb_size;
   1914  1.11  reinoud 
   1915  1.11  reinoud 	if (udf_node->num_extensions == 0)
   1916  1.11  reinoud 		return;
   1917  1.11  reinoud 
   1918  1.11  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
   1919  1.11  reinoud 	/* count number of allocation extents in use */
   1920  1.11  reinoud 	num_extents = 0;
   1921  1.11  reinoud 	slot = 0;
   1922  1.11  reinoud 	for (;;) {
   1923  1.11  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
   1924  1.11  reinoud 		if (eof)
   1925  1.11  reinoud 			break;
   1926  1.11  reinoud 		len   = udf_rw32(s_ad.len);
   1927  1.11  reinoud 		flags = UDF_EXT_FLAGS(len);
   1928  1.11  reinoud 
   1929  1.11  reinoud 		if (flags == UDF_EXT_REDIRECT)
   1930  1.11  reinoud 			num_extents++;
   1931  1.11  reinoud 
   1932  1.11  reinoud 		slot++;
   1933  1.11  reinoud 	}
   1934  1.11  reinoud 
   1935  1.11  reinoud 	DPRINTF(ALLOC, ("udf_count_alloc_ext counted %d live extents\n",
   1936  1.11  reinoud 		num_extents));
   1937  1.11  reinoud 
   1938  1.11  reinoud 	/* XXX choice: we could delay freeing them on node writeout */
   1939  1.11  reinoud 	/* free excess entries */
   1940  1.11  reinoud 	extnr = num_extents;
   1941  1.11  reinoud 	for (;extnr < udf_node->num_extensions; extnr++) {
   1942  1.11  reinoud 		DPRINTF(ALLOC, ("freeing alloc ext %d\n", extnr));
   1943  1.11  reinoud 		/* free dscriptor */
   1944  1.11  reinoud 		s_ad = udf_node->ext_loc[extnr];
   1945  1.11  reinoud 		udf_free_logvol_dscr(udf_node->ump, &s_ad,
   1946  1.11  reinoud 			udf_node->ext[extnr]);
   1947  1.11  reinoud 		udf_node->ext[extnr] = NULL;
   1948  1.11  reinoud 
   1949  1.11  reinoud 		/* free disc space */
   1950  1.11  reinoud 		lb_num    = udf_rw32(s_ad.loc.lb_num);
   1951  1.11  reinoud 		vpart_num = udf_rw16(s_ad.loc.part_num);
   1952  1.11  reinoud 		udf_free_allocated_space(udf_node->ump, lb_num, vpart_num, 1);
   1953  1.11  reinoud 
   1954  1.11  reinoud 		memset(&udf_node->ext_loc[extnr], 0, sizeof(struct long_ad));
   1955  1.11  reinoud 	}
   1956  1.11  reinoud 
   1957  1.11  reinoud 	/* set our new number of allocation extents */
   1958  1.11  reinoud 	udf_node->num_extensions = num_extents;
   1959  1.11  reinoud }
   1960  1.11  reinoud 
   1961  1.11  reinoud 
   1962  1.11  reinoud /* --------------------------------------------------------------------- */
   1963  1.11  reinoud 
   1964   1.1  reinoud /*
   1965   1.1  reinoud  * Adjust the node's allocation descriptors to reflect the new mapping; do
   1966   1.1  reinoud  * take note that we might glue to existing allocation descriptors.
   1967   1.1  reinoud  *
   1968   1.1  reinoud  * XXX Note there can only be one allocation being recorded/mount; maybe
   1969   1.1  reinoud  * explicit allocation in shedule thread?
   1970   1.1  reinoud  */
   1971   1.1  reinoud 
   1972   1.1  reinoud static void
   1973   1.1  reinoud udf_record_allocation_in_node(struct udf_mount *ump, struct buf *buf,
   1974   1.1  reinoud 	uint16_t vpart_num, uint64_t *mapping, struct long_ad *node_ad_cpy)
   1975   1.1  reinoud {
   1976   1.1  reinoud 	struct vnode    *vp = buf->b_vp;
   1977   1.1  reinoud 	struct udf_node *udf_node = VTOI(vp);
   1978   1.1  reinoud 	struct file_entry      *fe;
   1979   1.1  reinoud 	struct extfile_entry   *efe;
   1980   1.1  reinoud 	struct icb_tag  *icbtag;
   1981   1.1  reinoud 	struct long_ad   s_ad, c_ad;
   1982   1.1  reinoud 	uint64_t inflen, from, till;
   1983   1.1  reinoud 	uint64_t foffset, end_foffset, restart_foffset;
   1984   1.1  reinoud 	uint64_t orig_inflen, orig_lbrec, new_inflen, new_lbrec;
   1985   1.1  reinoud 	uint32_t num_lb, len, flags, lb_num;
   1986   1.1  reinoud 	uint32_t run_start;
   1987   1.9  reinoud 	uint32_t slot_offset, replace_len, replace;
   1988   1.1  reinoud 	int addr_type, icbflags;
   1989  1.16  reinoud //	int udf_c_type = buf->b_udf_c_type;
   1990   1.1  reinoud 	int lb_size, run_length, eof;
   1991   1.1  reinoud 	int slot, cpy_slot, cpy_slots, restart_slot;
   1992   1.1  reinoud 	int error;
   1993   1.1  reinoud 
   1994   1.1  reinoud 	DPRINTF(ALLOC, ("udf_record_allocation_in_node\n"));
   1995   1.1  reinoud 
   1996  1.16  reinoud #if 0
   1997  1.16  reinoud 	/* XXX disable sanity check for now */
   1998   1.1  reinoud 	/* sanity check ... should be panic ? */
   1999   1.1  reinoud 	if ((udf_c_type != UDF_C_USERDATA) && (udf_c_type != UDF_C_FIDS))
   2000   1.1  reinoud 		return;
   2001  1.16  reinoud #endif
   2002   1.1  reinoud 
   2003   1.1  reinoud 	lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size);
   2004   1.1  reinoud 
   2005   1.1  reinoud 	/* do the job */
   2006   1.1  reinoud 	UDF_LOCK_NODE(udf_node, 0);	/* XXX can deadlock ? */
   2007   1.9  reinoud 	udf_node_sanity_check(udf_node, &orig_inflen, &orig_lbrec);
   2008   1.1  reinoud 
   2009   1.1  reinoud 	fe  = udf_node->fe;
   2010   1.1  reinoud 	efe = udf_node->efe;
   2011   1.1  reinoud 	if (fe) {
   2012   1.1  reinoud 		icbtag = &fe->icbtag;
   2013   1.1  reinoud 		inflen = udf_rw64(fe->inf_len);
   2014   1.1  reinoud 	} else {
   2015   1.1  reinoud 		icbtag = &efe->icbtag;
   2016   1.1  reinoud 		inflen = udf_rw64(efe->inf_len);
   2017   1.1  reinoud 	}
   2018   1.1  reinoud 
   2019   1.1  reinoud 	/* do check if `till' is not past file information length */
   2020   1.1  reinoud 	from = buf->b_lblkno * lb_size;
   2021   1.1  reinoud 	till = MIN(inflen, from + buf->b_resid);
   2022   1.1  reinoud 
   2023   1.1  reinoud 	num_lb = (till - from + lb_size -1) / lb_size;
   2024   1.1  reinoud 
   2025   1.9  reinoud 	DPRINTF(ALLOC, ("record allocation from %"PRIu64" + %d\n", from, buf->b_bcount));
   2026   1.1  reinoud 
   2027   1.1  reinoud 	icbflags  = udf_rw16(icbtag->flags);
   2028   1.1  reinoud 	addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   2029   1.1  reinoud 
   2030   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
   2031   1.1  reinoud 		/* nothing to do */
   2032   1.1  reinoud 		/* XXX clean up rest of node? just in case? */
   2033   1.1  reinoud 		UDF_UNLOCK_NODE(udf_node, 0);
   2034   1.1  reinoud 		return;
   2035   1.1  reinoud 	}
   2036   1.1  reinoud 
   2037   1.1  reinoud 	slot     = 0;
   2038   1.1  reinoud 	cpy_slot = 0;
   2039   1.1  reinoud 	foffset  = 0;
   2040   1.1  reinoud 
   2041   1.1  reinoud 	/* 1) copy till first overlap piece to the rewrite buffer */
   2042   1.1  reinoud 	for (;;) {
   2043   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
   2044   1.1  reinoud 		if (eof) {
   2045   1.1  reinoud 			DPRINTF(WRITE,
   2046   1.1  reinoud 				("Record allocation in node "
   2047   1.1  reinoud 				 "failed: encountered EOF\n"));
   2048   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
   2049   1.1  reinoud 			buf->b_error = EINVAL;
   2050   1.1  reinoud 			return;
   2051   1.1  reinoud 		}
   2052   1.1  reinoud 		len   = udf_rw32(s_ad.len);
   2053   1.1  reinoud 		flags = UDF_EXT_FLAGS(len);
   2054   1.1  reinoud 		len   = UDF_EXT_LEN(len);
   2055   1.1  reinoud 
   2056   1.1  reinoud 		if (flags == UDF_EXT_REDIRECT) {
   2057   1.1  reinoud 			slot++;
   2058   1.1  reinoud 			continue;
   2059   1.1  reinoud 		}
   2060   1.1  reinoud 
   2061   1.1  reinoud 		end_foffset = foffset + len;
   2062   1.1  reinoud 		if (end_foffset > from)
   2063   1.1  reinoud 			break;	/* found */
   2064   1.1  reinoud 
   2065   1.1  reinoud 		node_ad_cpy[cpy_slot++] = s_ad;
   2066   1.1  reinoud 
   2067   1.1  reinoud 		DPRINTF(ALLOC, ("\t1: vp %d, lb %d, len %d, flags %d "
   2068   1.1  reinoud 			"-> stack\n",
   2069   1.1  reinoud 			udf_rw16(s_ad.loc.part_num),
   2070   1.1  reinoud 			udf_rw32(s_ad.loc.lb_num),
   2071   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2072   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2073   1.1  reinoud 
   2074   1.1  reinoud 		foffset = end_foffset;
   2075   1.1  reinoud 		slot++;
   2076   1.1  reinoud 	}
   2077   1.1  reinoud 	restart_slot    = slot;
   2078   1.1  reinoud 	restart_foffset = foffset;
   2079   1.1  reinoud 
   2080   1.1  reinoud 	/* 2) trunc overlapping slot at overlap and copy it */
   2081   1.1  reinoud 	slot_offset = from - foffset;
   2082   1.1  reinoud 	if (slot_offset > 0) {
   2083   1.1  reinoud 		DPRINTF(ALLOC, ("\tslot_offset = %d, flags = %d (%d)\n",
   2084   1.1  reinoud 				slot_offset, flags >> 30, flags));
   2085   1.1  reinoud 
   2086   1.1  reinoud 		s_ad.len = udf_rw32(slot_offset | flags);
   2087   1.1  reinoud 		node_ad_cpy[cpy_slot++] = s_ad;
   2088   1.1  reinoud 
   2089   1.1  reinoud 		DPRINTF(ALLOC, ("\t2: vp %d, lb %d, len %d, flags %d "
   2090   1.1  reinoud 			"-> stack\n",
   2091   1.1  reinoud 			udf_rw16(s_ad.loc.part_num),
   2092   1.1  reinoud 			udf_rw32(s_ad.loc.lb_num),
   2093   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2094   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2095   1.1  reinoud 	}
   2096   1.1  reinoud 	foffset += slot_offset;
   2097   1.1  reinoud 
   2098   1.1  reinoud 	/* 3) insert new mappings */
   2099   1.1  reinoud 	memset(&s_ad, 0, sizeof(struct long_ad));
   2100   1.1  reinoud 	lb_num = 0;
   2101   1.1  reinoud 	for (lb_num = 0; lb_num < num_lb; lb_num++) {
   2102   1.1  reinoud 		run_start  = mapping[lb_num];
   2103   1.1  reinoud 		run_length = 1;
   2104   1.1  reinoud 		while (lb_num < num_lb-1) {
   2105   1.1  reinoud 			if (mapping[lb_num+1] != mapping[lb_num]+1)
   2106   1.1  reinoud 				if (mapping[lb_num+1] != mapping[lb_num])
   2107   1.1  reinoud 					break;
   2108   1.1  reinoud 			run_length++;
   2109   1.1  reinoud 			lb_num++;
   2110   1.1  reinoud 		}
   2111   1.1  reinoud 		/* insert slot for this mapping */
   2112   1.1  reinoud 		len = run_length * lb_size;
   2113   1.1  reinoud 
   2114   1.1  reinoud 		/* bounds checking */
   2115   1.1  reinoud 		if (foffset + len > till)
   2116   1.1  reinoud 			len = till - foffset;
   2117   1.1  reinoud 		KASSERT(foffset + len <= inflen);
   2118   1.1  reinoud 
   2119   1.1  reinoud 		s_ad.len = udf_rw32(len | UDF_EXT_ALLOCATED);
   2120   1.1  reinoud 		s_ad.loc.part_num = udf_rw16(vpart_num);
   2121   1.1  reinoud 		s_ad.loc.lb_num   = udf_rw32(run_start);
   2122   1.1  reinoud 
   2123   1.1  reinoud 		foffset += len;
   2124   1.1  reinoud 
   2125   1.1  reinoud 		/* paranoia */
   2126   1.1  reinoud 		if (len == 0) {
   2127   1.1  reinoud 			DPRINTF(WRITE,
   2128   1.1  reinoud 				("Record allocation in node "
   2129   1.1  reinoud 				 "failed: insert failed\n"));
   2130   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
   2131   1.1  reinoud 			buf->b_error = EINVAL;
   2132   1.1  reinoud 			return;
   2133   1.1  reinoud 		}
   2134   1.1  reinoud 		node_ad_cpy[cpy_slot++] = s_ad;
   2135   1.1  reinoud 
   2136   1.1  reinoud 		DPRINTF(ALLOC, ("\t3: insert new mapping vp %d lb %d, len %d, "
   2137   1.1  reinoud 				"flags %d -> stack\n",
   2138   1.1  reinoud 			udf_rw16(s_ad.loc.part_num), udf_rw32(s_ad.loc.lb_num),
   2139   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2140   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2141   1.1  reinoud 	}
   2142   1.1  reinoud 
   2143   1.1  reinoud 	/* 4) pop replaced length */
   2144   1.9  reinoud 	slot    = restart_slot;
   2145   1.1  reinoud 	foffset = restart_foffset;
   2146   1.1  reinoud 
   2147   1.9  reinoud 	replace_len = till - foffset;	/* total amount of bytes to pop */
   2148   1.8  reinoud 	slot_offset = from - foffset;	/* offset in first encounted slot */
   2149   1.9  reinoud 	KASSERT((slot_offset % lb_size) == 0);
   2150   1.9  reinoud 
   2151   1.1  reinoud 	for (;;) {
   2152   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
   2153   1.1  reinoud 		if (eof)
   2154   1.1  reinoud 			break;
   2155   1.1  reinoud 
   2156   1.1  reinoud 		len    = udf_rw32(s_ad.len);
   2157   1.1  reinoud 		flags  = UDF_EXT_FLAGS(len);
   2158   1.1  reinoud 		len    = UDF_EXT_LEN(len);
   2159   1.1  reinoud 		lb_num = udf_rw32(s_ad.loc.lb_num);
   2160   1.1  reinoud 
   2161   1.1  reinoud 		if (flags == UDF_EXT_REDIRECT) {
   2162   1.1  reinoud 			slot++;
   2163   1.1  reinoud 			continue;
   2164   1.1  reinoud 		}
   2165   1.1  reinoud 
   2166   1.8  reinoud 		DPRINTF(ALLOC, ("\t4i: got slot %d, slot_offset %d, "
   2167   1.9  reinoud 				"replace_len %d, "
   2168   1.8  reinoud 				"vp %d, lb %d, len %d, flags %d\n",
   2169   1.9  reinoud 			slot, slot_offset, replace_len,
   2170   1.8  reinoud 			udf_rw16(s_ad.loc.part_num),
   2171   1.1  reinoud 			udf_rw32(s_ad.loc.lb_num),
   2172   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2173   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2174   1.1  reinoud 
   2175   1.9  reinoud 		/* adjust for slot offset */
   2176   1.9  reinoud 		if (slot_offset) {
   2177   1.9  reinoud 			DPRINTF(ALLOC, ("\t4s: skipping %d\n", slot_offset));
   2178   1.9  reinoud 			lb_num += slot_offset / lb_size;
   2179   1.9  reinoud 			len    -= slot_offset;
   2180   1.9  reinoud 			foffset += slot_offset;
   2181   1.9  reinoud 			replace_len -= slot_offset;
   2182   1.9  reinoud 
   2183   1.9  reinoud 			/* mark adjusted */
   2184   1.9  reinoud 			slot_offset = 0;
   2185   1.9  reinoud 		}
   2186   1.9  reinoud 
   2187   1.9  reinoud 		/* advance for (the rest of) this slot */
   2188   1.9  reinoud 		replace = MIN(len, replace_len);
   2189   1.9  reinoud 		DPRINTF(ALLOC, ("\t4d: replacing %d\n", replace));
   2190   1.9  reinoud 
   2191   1.9  reinoud 		/* advance for this slot */
   2192   1.9  reinoud 		if (replace) {
   2193  1.10  reinoud 			/* note: dont round DOWN on num_lb since we then
   2194  1.10  reinoud 			 * forget the last partial one */
   2195   1.9  reinoud 			num_lb = (replace + lb_size - 1) / lb_size;
   2196   1.9  reinoud 			if (flags != UDF_EXT_FREE) {
   2197   1.9  reinoud 				udf_free_allocated_space(ump, lb_num,
   2198   1.9  reinoud 					udf_rw16(s_ad.loc.part_num), num_lb);
   2199   1.9  reinoud 			}
   2200   1.9  reinoud 			lb_num      += num_lb;
   2201   1.9  reinoud 			len         -= replace;
   2202   1.9  reinoud 			foffset     += replace;
   2203   1.9  reinoud 			replace_len -= replace;
   2204   1.9  reinoud 		}
   2205   1.1  reinoud 
   2206   1.9  reinoud 		/* do we have a slot tail ? */
   2207   1.1  reinoud 		if (len) {
   2208   1.9  reinoud 			KASSERT(foffset % lb_size == 0);
   2209   1.1  reinoud 
   2210   1.1  reinoud 			/* we arrived at our point, push remainder */
   2211   1.1  reinoud 			s_ad.len        = udf_rw32(len | flags);
   2212   1.1  reinoud 			s_ad.loc.lb_num = udf_rw32(lb_num);
   2213   1.9  reinoud 			if (flags == UDF_EXT_FREE)
   2214   1.9  reinoud 				s_ad.loc.lb_num = udf_rw32(0);
   2215   1.1  reinoud 			node_ad_cpy[cpy_slot++] = s_ad;
   2216   1.1  reinoud 			foffset += len;
   2217   1.1  reinoud 			slot++;
   2218   1.1  reinoud 
   2219   1.1  reinoud 			DPRINTF(ALLOC, ("\t4: vp %d, lb %d, len %d, flags %d "
   2220   1.1  reinoud 				"-> stack\n",
   2221   1.1  reinoud 				udf_rw16(s_ad.loc.part_num),
   2222   1.1  reinoud 				udf_rw32(s_ad.loc.lb_num),
   2223   1.1  reinoud 				UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2224   1.1  reinoud 				UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2225   1.1  reinoud 			break;
   2226   1.1  reinoud 		}
   2227   1.9  reinoud 
   2228   1.1  reinoud 		slot++;
   2229   1.1  reinoud 	}
   2230   1.1  reinoud 
   2231   1.1  reinoud 	/* 5) copy remainder */
   2232   1.1  reinoud 	for (;;) {
   2233   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
   2234   1.1  reinoud 		if (eof)
   2235   1.1  reinoud 			break;
   2236   1.1  reinoud 
   2237   1.1  reinoud 		len   = udf_rw32(s_ad.len);
   2238   1.1  reinoud 		flags = UDF_EXT_FLAGS(len);
   2239   1.1  reinoud 		len   = UDF_EXT_LEN(len);
   2240   1.1  reinoud 
   2241   1.1  reinoud 		if (flags == UDF_EXT_REDIRECT) {
   2242   1.1  reinoud 			slot++;
   2243   1.1  reinoud 			continue;
   2244   1.1  reinoud 		}
   2245   1.1  reinoud 
   2246   1.1  reinoud 		node_ad_cpy[cpy_slot++] = s_ad;
   2247   1.1  reinoud 
   2248   1.1  reinoud 		DPRINTF(ALLOC, ("\t5: insert new mapping "
   2249   1.1  reinoud 			"vp %d lb %d, len %d, flags %d "
   2250   1.1  reinoud 			"-> stack\n",
   2251   1.1  reinoud 		udf_rw16(s_ad.loc.part_num),
   2252   1.1  reinoud 		udf_rw32(s_ad.loc.lb_num),
   2253   1.1  reinoud 		UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2254   1.1  reinoud 		UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2255   1.1  reinoud 
   2256   1.1  reinoud 		slot++;
   2257   1.1  reinoud 	}
   2258   1.1  reinoud 
   2259   1.1  reinoud 	/* 6) reset node descriptors */
   2260   1.1  reinoud 	udf_wipe_adslots(udf_node);
   2261   1.1  reinoud 
   2262   1.1  reinoud 	/* 7) copy back extents; merge when possible. Recounting on the fly */
   2263   1.1  reinoud 	cpy_slots = cpy_slot;
   2264   1.1  reinoud 
   2265   1.1  reinoud 	c_ad = node_ad_cpy[0];
   2266   1.1  reinoud 	slot = 0;
   2267   1.1  reinoud 	DPRINTF(ALLOC, ("\t7s: stack -> got mapping vp %d "
   2268   1.1  reinoud 		"lb %d, len %d, flags %d\n",
   2269   1.1  reinoud 	udf_rw16(c_ad.loc.part_num),
   2270   1.1  reinoud 	udf_rw32(c_ad.loc.lb_num),
   2271   1.1  reinoud 	UDF_EXT_LEN(udf_rw32(c_ad.len)),
   2272   1.1  reinoud 	UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30));
   2273   1.1  reinoud 
   2274   1.1  reinoud 	for (cpy_slot = 1; cpy_slot < cpy_slots; cpy_slot++) {
   2275   1.1  reinoud 		s_ad = node_ad_cpy[cpy_slot];
   2276   1.1  reinoud 
   2277   1.1  reinoud 		DPRINTF(ALLOC, ("\t7i: stack -> got mapping vp %d "
   2278   1.1  reinoud 			"lb %d, len %d, flags %d\n",
   2279   1.1  reinoud 		udf_rw16(s_ad.loc.part_num),
   2280   1.1  reinoud 		udf_rw32(s_ad.loc.lb_num),
   2281   1.1  reinoud 		UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2282   1.1  reinoud 		UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2283   1.1  reinoud 
   2284   1.1  reinoud 		/* see if we can merge */
   2285   1.1  reinoud 		if (udf_ads_merge(lb_size, &c_ad, &s_ad)) {
   2286   1.1  reinoud 			/* not mergable (anymore) */
   2287   1.1  reinoud 			DPRINTF(ALLOC, ("\t7: appending vp %d lb %d, "
   2288   1.1  reinoud 				"len %d, flags %d\n",
   2289   1.1  reinoud 			udf_rw16(c_ad.loc.part_num),
   2290   1.1  reinoud 			udf_rw32(c_ad.loc.lb_num),
   2291   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(c_ad.len)),
   2292   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30));
   2293   1.1  reinoud 
   2294  1.11  reinoud 			error = udf_append_adslot(udf_node, &slot, &c_ad);
   2295   1.1  reinoud 			if (error) {
   2296   1.1  reinoud 				buf->b_error = error;
   2297   1.1  reinoud 				goto out;
   2298   1.1  reinoud 			}
   2299   1.1  reinoud 			c_ad = s_ad;
   2300   1.1  reinoud 			slot++;
   2301   1.1  reinoud 		}
   2302   1.1  reinoud 	}
   2303   1.1  reinoud 
   2304   1.1  reinoud 	/* 8) push rest slot (if any) */
   2305   1.1  reinoud 	if (UDF_EXT_LEN(c_ad.len) > 0) {
   2306   1.1  reinoud 		DPRINTF(ALLOC, ("\t8: last append vp %d lb %d, "
   2307   1.1  reinoud 				"len %d, flags %d\n",
   2308   1.1  reinoud 		udf_rw16(c_ad.loc.part_num),
   2309   1.1  reinoud 		udf_rw32(c_ad.loc.lb_num),
   2310   1.1  reinoud 		UDF_EXT_LEN(udf_rw32(c_ad.len)),
   2311   1.1  reinoud 		UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30));
   2312   1.1  reinoud 
   2313  1.11  reinoud 		error = udf_append_adslot(udf_node, &slot, &c_ad);
   2314   1.1  reinoud 		if (error) {
   2315   1.1  reinoud 			buf->b_error = error;
   2316   1.1  reinoud 			goto out;
   2317   1.1  reinoud 		}
   2318   1.1  reinoud 	}
   2319   1.1  reinoud 
   2320   1.1  reinoud out:
   2321  1.11  reinoud 	udf_count_alloc_exts(udf_node);
   2322  1.11  reinoud 
   2323   1.1  reinoud 	/* the node's descriptors should now be sane */
   2324   1.9  reinoud 	udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec);
   2325   1.1  reinoud 	UDF_UNLOCK_NODE(udf_node, 0);
   2326   1.1  reinoud 
   2327   1.1  reinoud 	KASSERT(orig_inflen == new_inflen);
   2328   1.1  reinoud 	KASSERT(new_lbrec >= orig_lbrec);
   2329   1.1  reinoud 
   2330   1.1  reinoud 	return;
   2331   1.1  reinoud }
   2332   1.1  reinoud 
   2333   1.1  reinoud /* --------------------------------------------------------------------- */
   2334   1.1  reinoud 
   2335   1.1  reinoud int
   2336   1.1  reinoud udf_grow_node(struct udf_node *udf_node, uint64_t new_size)
   2337   1.1  reinoud {
   2338   1.1  reinoud 	union dscrptr *dscr;
   2339   1.1  reinoud 	struct vnode *vp = udf_node->vnode;
   2340   1.1  reinoud 	struct udf_mount *ump = udf_node->ump;
   2341   1.1  reinoud 	struct file_entry    *fe;
   2342   1.1  reinoud 	struct extfile_entry *efe;
   2343   1.1  reinoud 	struct icb_tag  *icbtag;
   2344   1.1  reinoud 	struct long_ad c_ad, s_ad;
   2345   1.1  reinoud 	uint64_t size_diff, old_size, inflen, objsize, chunk, append_len;
   2346   1.1  reinoud 	uint64_t foffset, end_foffset;
   2347   1.1  reinoud 	uint64_t orig_inflen, orig_lbrec, new_inflen, new_lbrec;
   2348   1.1  reinoud 	uint32_t lb_size, dscr_size, crclen, lastblock_grow;
   2349  1.17  reinoud 	uint32_t icbflags, len, flags, max_len;
   2350   1.1  reinoud 	uint32_t max_l_ad, l_ad, l_ea;
   2351  1.17  reinoud 	uint16_t my_part, dst_part;
   2352   1.1  reinoud 	uint8_t *data_pos, *evacuated_data;
   2353  1.17  reinoud 	int addr_type;
   2354   1.1  reinoud 	int slot, cpy_slot;
   2355  1.26  reinoud 	int eof, error;
   2356   1.1  reinoud 
   2357   1.1  reinoud 	DPRINTF(ALLOC, ("udf_grow_node\n"));
   2358   1.9  reinoud 
   2359   1.9  reinoud 	UDF_LOCK_NODE(udf_node, 0);
   2360   1.1  reinoud 	udf_node_sanity_check(udf_node, &orig_inflen, &orig_lbrec);
   2361   1.1  reinoud 
   2362   1.1  reinoud 	lb_size = udf_rw32(ump->logical_vol->lb_size);
   2363   1.1  reinoud 	max_len = ((UDF_EXT_MAXLEN / lb_size) * lb_size);
   2364   1.1  reinoud 
   2365   1.1  reinoud 	fe  = udf_node->fe;
   2366   1.1  reinoud 	efe = udf_node->efe;
   2367   1.1  reinoud 	if (fe) {
   2368   1.1  reinoud 		dscr       = (union dscrptr *) fe;
   2369   1.1  reinoud 		icbtag  = &fe->icbtag;
   2370   1.1  reinoud 		inflen  = udf_rw64(fe->inf_len);
   2371   1.1  reinoud 		objsize = inflen;
   2372   1.1  reinoud 		dscr_size  = sizeof(struct file_entry) -1;
   2373   1.1  reinoud 		l_ea       = udf_rw32(fe->l_ea);
   2374   1.1  reinoud 		l_ad       = udf_rw32(fe->l_ad);
   2375   1.1  reinoud 	} else {
   2376   1.1  reinoud 		dscr       = (union dscrptr *) efe;
   2377   1.1  reinoud 		icbtag  = &efe->icbtag;
   2378   1.1  reinoud 		inflen  = udf_rw64(efe->inf_len);
   2379   1.1  reinoud 		objsize = udf_rw64(efe->obj_size);
   2380   1.1  reinoud 		dscr_size  = sizeof(struct extfile_entry) -1;
   2381   1.1  reinoud 		l_ea       = udf_rw32(efe->l_ea);
   2382   1.1  reinoud 		l_ad       = udf_rw32(efe->l_ad);
   2383   1.1  reinoud 	}
   2384   1.1  reinoud 	data_pos  = (uint8_t *) dscr + dscr_size + l_ea;
   2385   1.1  reinoud 	max_l_ad = lb_size - dscr_size - l_ea;
   2386   1.1  reinoud 
   2387   1.1  reinoud 	icbflags   = udf_rw16(icbtag->flags);
   2388   1.1  reinoud 	addr_type  = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   2389   1.1  reinoud 
   2390   1.1  reinoud 	old_size  = inflen;
   2391   1.1  reinoud 	size_diff = new_size - old_size;
   2392   1.1  reinoud 
   2393   1.1  reinoud 	DPRINTF(ALLOC, ("\tfrom %"PRIu64" to %"PRIu64"\n", old_size, new_size));
   2394   1.1  reinoud 
   2395   1.1  reinoud 	evacuated_data = NULL;
   2396   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
   2397   1.1  reinoud 		if (l_ad + size_diff <= max_l_ad) {
   2398   1.1  reinoud 			/* only reflect size change directly in the node */
   2399   1.1  reinoud 			inflen  += size_diff;
   2400   1.1  reinoud 			objsize += size_diff;
   2401   1.1  reinoud 			l_ad    += size_diff;
   2402   1.1  reinoud 			crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea + l_ad;
   2403   1.1  reinoud 			if (fe) {
   2404   1.1  reinoud 				fe->inf_len   = udf_rw64(inflen);
   2405   1.1  reinoud 				fe->l_ad      = udf_rw32(l_ad);
   2406  1.22  reinoud 				fe->tag.desc_crc_len = udf_rw16(crclen);
   2407   1.1  reinoud 			} else {
   2408   1.1  reinoud 				efe->inf_len  = udf_rw64(inflen);
   2409   1.1  reinoud 				efe->obj_size = udf_rw64(objsize);
   2410   1.1  reinoud 				efe->l_ad     = udf_rw32(l_ad);
   2411  1.22  reinoud 				efe->tag.desc_crc_len = udf_rw16(crclen);
   2412   1.1  reinoud 			}
   2413   1.1  reinoud 			error = 0;
   2414   1.1  reinoud 
   2415   1.1  reinoud 			/* set new size for uvm */
   2416   1.1  reinoud 			uvm_vnp_setsize(vp, old_size);
   2417   1.1  reinoud 			uvm_vnp_setwritesize(vp, new_size);
   2418   1.1  reinoud 
   2419   1.1  reinoud #if 0
   2420   1.1  reinoud 			/* zero append space in buffer */
   2421   1.1  reinoud 			uvm_vnp_zerorange(vp, old_size, new_size - old_size);
   2422   1.1  reinoud #endif
   2423   1.1  reinoud 
   2424   1.9  reinoud 			udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec);
   2425   1.9  reinoud 
   2426   1.1  reinoud 			/* unlock */
   2427   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
   2428   1.1  reinoud 
   2429   1.1  reinoud 			KASSERT(new_inflen == orig_inflen + size_diff);
   2430   1.1  reinoud 			KASSERT(new_lbrec == orig_lbrec);
   2431   1.1  reinoud 			KASSERT(new_lbrec == 0);
   2432   1.1  reinoud 			return 0;
   2433   1.1  reinoud 		}
   2434   1.1  reinoud 
   2435   1.1  reinoud 		DPRINTF(ALLOC, ("\tCONVERT from internal\n"));
   2436   1.1  reinoud 
   2437   1.1  reinoud 		if (old_size > 0) {
   2438   1.1  reinoud 			/* allocate some space and copy in the stuff to keep */
   2439   1.1  reinoud 			evacuated_data = malloc(lb_size, M_UDFTEMP, M_WAITOK);
   2440   1.1  reinoud 			memset(evacuated_data, 0, lb_size);
   2441   1.1  reinoud 
   2442   1.1  reinoud 			/* node is locked, so safe to exit mutex */
   2443   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
   2444   1.1  reinoud 
   2445   1.1  reinoud 			/* read in using the `normal' vn_rdwr() */
   2446   1.1  reinoud 			error = vn_rdwr(UIO_READ, udf_node->vnode,
   2447   1.1  reinoud 					evacuated_data, old_size, 0,
   2448   1.1  reinoud 					UIO_SYSSPACE, IO_ALTSEMANTICS | IO_NODELOCKED,
   2449   1.1  reinoud 					FSCRED, NULL, NULL);
   2450   1.1  reinoud 
   2451   1.1  reinoud 			/* enter again */
   2452   1.1  reinoud 			UDF_LOCK_NODE(udf_node, 0);
   2453   1.1  reinoud 		}
   2454   1.1  reinoud 
   2455  1.14  reinoud 		/* convert to a normal alloc and select type */
   2456  1.17  reinoud 		my_part  = udf_rw16(udf_node->loc.loc.part_num);
   2457  1.26  reinoud 		dst_part = udf_get_record_vpart(ump, udf_get_c_type(udf_node));
   2458  1.17  reinoud 		addr_type = UDF_ICB_SHORT_ALLOC;
   2459  1.17  reinoud 		if (dst_part != my_part)
   2460  1.17  reinoud 			addr_type = UDF_ICB_LONG_ALLOC;
   2461  1.14  reinoud 
   2462   1.1  reinoud 		icbflags &= ~UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   2463  1.17  reinoud 		icbflags |= addr_type;
   2464   1.1  reinoud 		icbtag->flags = udf_rw16(icbflags);
   2465   1.1  reinoud 
   2466   1.1  reinoud 		/* wipe old descriptor space */
   2467   1.1  reinoud 		udf_wipe_adslots(udf_node);
   2468   1.1  reinoud 
   2469   1.1  reinoud 		memset(&c_ad, 0, sizeof(struct long_ad));
   2470   1.1  reinoud 		c_ad.len          = udf_rw32(old_size | UDF_EXT_FREE);
   2471   1.1  reinoud 		c_ad.loc.part_num = udf_rw16(0); /* not relevant */
   2472   1.1  reinoud 		c_ad.loc.lb_num   = udf_rw32(0); /* not relevant */
   2473   1.1  reinoud 
   2474   1.1  reinoud 		slot = 0;
   2475   1.1  reinoud 	} else {
   2476   1.1  reinoud 		/* goto the last entry (if any) */
   2477   1.1  reinoud 		slot     = 0;
   2478   1.1  reinoud 		cpy_slot = 0;
   2479   1.1  reinoud 		foffset  = 0;
   2480   1.1  reinoud 		memset(&c_ad, 0, sizeof(struct long_ad));
   2481   1.1  reinoud 		for (;;) {
   2482   1.1  reinoud 			udf_get_adslot(udf_node, slot, &c_ad, &eof);
   2483   1.1  reinoud 			if (eof)
   2484   1.1  reinoud 				break;
   2485   1.1  reinoud 
   2486   1.1  reinoud 			len   = udf_rw32(c_ad.len);
   2487   1.1  reinoud 			flags = UDF_EXT_FLAGS(len);
   2488   1.1  reinoud 			len   = UDF_EXT_LEN(len);
   2489   1.1  reinoud 
   2490   1.1  reinoud 			end_foffset = foffset + len;
   2491   1.1  reinoud 			if (flags != UDF_EXT_REDIRECT)
   2492   1.1  reinoud 				foffset = end_foffset;
   2493   1.1  reinoud 
   2494   1.1  reinoud 			slot++;
   2495   1.1  reinoud 		}
   2496   1.1  reinoud 		/* at end of adslots */
   2497   1.1  reinoud 
   2498   1.1  reinoud 		/* special case if the old size was zero, then there is no last slot */
   2499   1.1  reinoud 		if (old_size == 0) {
   2500   1.1  reinoud 			c_ad.len          = udf_rw32(0 | UDF_EXT_FREE);
   2501   1.1  reinoud 			c_ad.loc.part_num = udf_rw16(0); /* not relevant */
   2502   1.1  reinoud 			c_ad.loc.lb_num   = udf_rw32(0); /* not relevant */
   2503   1.1  reinoud 		} else {
   2504   1.1  reinoud 			/* refetch last slot */
   2505   1.1  reinoud 			slot--;
   2506   1.1  reinoud 			udf_get_adslot(udf_node, slot, &c_ad, &eof);
   2507   1.1  reinoud 		}
   2508   1.1  reinoud 	}
   2509   1.1  reinoud 
   2510   1.1  reinoud 	/*
   2511   1.1  reinoud 	 * If the length of the last slot is not a multiple of lb_size, adjust
   2512   1.1  reinoud 	 * length so that it is; don't forget to adjust `append_len'! relevant for
   2513   1.1  reinoud 	 * extending existing files
   2514   1.1  reinoud 	 */
   2515   1.1  reinoud 	len   = udf_rw32(c_ad.len);
   2516   1.1  reinoud 	flags = UDF_EXT_FLAGS(len);
   2517   1.1  reinoud 	len   = UDF_EXT_LEN(len);
   2518   1.1  reinoud 
   2519   1.1  reinoud 	lastblock_grow = 0;
   2520   1.1  reinoud 	if (len % lb_size > 0) {
   2521   1.1  reinoud 		lastblock_grow = lb_size - (len % lb_size);
   2522   1.1  reinoud 		lastblock_grow = MIN(size_diff, lastblock_grow);
   2523   1.1  reinoud 		len += lastblock_grow;
   2524   1.1  reinoud 		c_ad.len = udf_rw32(len | flags);
   2525   1.1  reinoud 
   2526   1.1  reinoud 		/* TODO zero appened space in buffer! */
   2527   1.1  reinoud 		/* using uvm_vnp_zerorange(vp, old_size, new_size - old_size); ? */
   2528   1.1  reinoud 	}
   2529   1.1  reinoud 	memset(&s_ad, 0, sizeof(struct long_ad));
   2530   1.1  reinoud 
   2531   1.1  reinoud 	/* size_diff can be bigger than allowed, so grow in chunks */
   2532   1.1  reinoud 	append_len = size_diff - lastblock_grow;
   2533   1.1  reinoud 	while (append_len > 0) {
   2534   1.1  reinoud 		chunk = MIN(append_len, max_len);
   2535   1.1  reinoud 		s_ad.len = udf_rw32(chunk | UDF_EXT_FREE);
   2536   1.1  reinoud 		s_ad.loc.part_num = udf_rw16(0);
   2537   1.1  reinoud 		s_ad.loc.lb_num   = udf_rw32(0);
   2538   1.1  reinoud 
   2539   1.1  reinoud 		if (udf_ads_merge(lb_size, &c_ad, &s_ad)) {
   2540   1.1  reinoud 			/* not mergable (anymore) */
   2541  1.11  reinoud 			error = udf_append_adslot(udf_node, &slot, &c_ad);
   2542   1.1  reinoud 			if (error)
   2543   1.1  reinoud 				goto errorout;
   2544   1.1  reinoud 			slot++;
   2545   1.1  reinoud 			c_ad = s_ad;
   2546   1.1  reinoud 			memset(&s_ad, 0, sizeof(struct long_ad));
   2547   1.1  reinoud 		}
   2548   1.1  reinoud 		append_len -= chunk;
   2549   1.1  reinoud 	}
   2550   1.1  reinoud 
   2551   1.1  reinoud 	/* if there is a rest piece in the accumulator, append it */
   2552   1.8  reinoud 	if (UDF_EXT_LEN(udf_rw32(c_ad.len)) > 0) {
   2553  1.11  reinoud 		error = udf_append_adslot(udf_node, &slot, &c_ad);
   2554   1.1  reinoud 		if (error)
   2555   1.1  reinoud 			goto errorout;
   2556   1.1  reinoud 		slot++;
   2557   1.1  reinoud 	}
   2558   1.1  reinoud 
   2559   1.1  reinoud 	/* if there is a rest piece that didn't fit, append it */
   2560   1.8  reinoud 	if (UDF_EXT_LEN(udf_rw32(s_ad.len)) > 0) {
   2561  1.11  reinoud 		error = udf_append_adslot(udf_node, &slot, &s_ad);
   2562   1.1  reinoud 		if (error)
   2563   1.1  reinoud 			goto errorout;
   2564   1.1  reinoud 		slot++;
   2565   1.1  reinoud 	}
   2566   1.1  reinoud 
   2567   1.1  reinoud 	inflen  += size_diff;
   2568   1.1  reinoud 	objsize += size_diff;
   2569   1.1  reinoud 	if (fe) {
   2570   1.1  reinoud 		fe->inf_len   = udf_rw64(inflen);
   2571   1.1  reinoud 	} else {
   2572   1.1  reinoud 		efe->inf_len  = udf_rw64(inflen);
   2573   1.1  reinoud 		efe->obj_size = udf_rw64(objsize);
   2574   1.1  reinoud 	}
   2575   1.1  reinoud 	error = 0;
   2576   1.1  reinoud 
   2577   1.1  reinoud 	if (evacuated_data) {
   2578   1.1  reinoud 		/* set new write size for uvm */
   2579   1.1  reinoud 		uvm_vnp_setwritesize(vp, old_size);
   2580   1.1  reinoud 
   2581   1.1  reinoud 		/* write out evacuated data */
   2582   1.1  reinoud 		error = vn_rdwr(UIO_WRITE, udf_node->vnode,
   2583   1.1  reinoud 				evacuated_data, old_size, 0,
   2584   1.1  reinoud 				UIO_SYSSPACE, IO_ALTSEMANTICS | IO_NODELOCKED,
   2585   1.1  reinoud 				FSCRED, NULL, NULL);
   2586   1.1  reinoud 		uvm_vnp_setsize(vp, old_size);
   2587   1.1  reinoud 	}
   2588   1.1  reinoud 
   2589   1.1  reinoud errorout:
   2590   1.1  reinoud 	if (evacuated_data)
   2591   1.1  reinoud 		free(evacuated_data, M_UDFTEMP);
   2592   1.9  reinoud 
   2593  1.11  reinoud 	udf_count_alloc_exts(udf_node);
   2594  1.11  reinoud 
   2595   1.9  reinoud 	udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec);
   2596   1.1  reinoud 	UDF_UNLOCK_NODE(udf_node, 0);
   2597   1.1  reinoud 
   2598   1.1  reinoud 	KASSERT(new_inflen == orig_inflen + size_diff);
   2599   1.1  reinoud 	KASSERT(new_lbrec == orig_lbrec);
   2600   1.1  reinoud 
   2601   1.1  reinoud 	return error;
   2602   1.1  reinoud }
   2603   1.1  reinoud 
   2604   1.1  reinoud /* --------------------------------------------------------------------- */
   2605   1.1  reinoud 
   2606   1.1  reinoud int
   2607   1.1  reinoud udf_shrink_node(struct udf_node *udf_node, uint64_t new_size)
   2608   1.1  reinoud {
   2609   1.1  reinoud 	struct vnode *vp = udf_node->vnode;
   2610   1.1  reinoud 	struct udf_mount *ump = udf_node->ump;
   2611   1.1  reinoud 	struct file_entry    *fe;
   2612   1.1  reinoud 	struct extfile_entry *efe;
   2613   1.1  reinoud 	struct icb_tag  *icbtag;
   2614   1.1  reinoud 	struct long_ad c_ad, s_ad, *node_ad_cpy;
   2615   1.1  reinoud 	uint64_t size_diff, old_size, inflen, objsize;
   2616   1.1  reinoud 	uint64_t foffset, end_foffset;
   2617   1.1  reinoud 	uint64_t orig_inflen, orig_lbrec, new_inflen, new_lbrec;
   2618   1.1  reinoud 	uint32_t lb_size, dscr_size, crclen;
   2619   1.1  reinoud 	uint32_t slot_offset;
   2620   1.1  reinoud 	uint32_t len, flags, max_len;
   2621   1.1  reinoud 	uint32_t num_lb, lb_num;
   2622   1.1  reinoud 	uint32_t max_l_ad, l_ad, l_ea;
   2623   1.1  reinoud 	uint16_t vpart_num;
   2624   1.1  reinoud 	uint8_t *data_pos;
   2625   1.1  reinoud 	int icbflags, addr_type;
   2626   1.1  reinoud 	int slot, cpy_slot, cpy_slots;
   2627   1.1  reinoud 	int eof, error;
   2628   1.1  reinoud 
   2629   1.1  reinoud 	DPRINTF(ALLOC, ("udf_shrink_node\n"));
   2630   1.9  reinoud 
   2631   1.9  reinoud 	UDF_LOCK_NODE(udf_node, 0);
   2632   1.1  reinoud 	udf_node_sanity_check(udf_node, &orig_inflen, &orig_lbrec);
   2633   1.1  reinoud 
   2634   1.1  reinoud 	lb_size = udf_rw32(ump->logical_vol->lb_size);
   2635   1.1  reinoud 	max_len = ((UDF_EXT_MAXLEN / lb_size) * lb_size);
   2636   1.1  reinoud 
   2637   1.1  reinoud 	/* do the work */
   2638   1.1  reinoud 	fe  = udf_node->fe;
   2639   1.1  reinoud 	efe = udf_node->efe;
   2640   1.1  reinoud 	if (fe) {
   2641   1.1  reinoud 		icbtag  = &fe->icbtag;
   2642   1.1  reinoud 		inflen  = udf_rw64(fe->inf_len);
   2643   1.1  reinoud 		objsize = inflen;
   2644   1.1  reinoud 		dscr_size  = sizeof(struct file_entry) -1;
   2645   1.1  reinoud 		l_ea       = udf_rw32(fe->l_ea);
   2646   1.1  reinoud 		l_ad       = udf_rw32(fe->l_ad);
   2647   1.1  reinoud 		data_pos = (uint8_t *) fe + dscr_size + l_ea;
   2648   1.1  reinoud 	} else {
   2649   1.1  reinoud 		icbtag  = &efe->icbtag;
   2650   1.1  reinoud 		inflen  = udf_rw64(efe->inf_len);
   2651   1.1  reinoud 		objsize = udf_rw64(efe->obj_size);
   2652   1.1  reinoud 		dscr_size  = sizeof(struct extfile_entry) -1;
   2653   1.1  reinoud 		l_ea       = udf_rw32(efe->l_ea);
   2654   1.1  reinoud 		l_ad       = udf_rw32(efe->l_ad);
   2655   1.1  reinoud 		data_pos = (uint8_t *) efe + dscr_size + l_ea;
   2656   1.1  reinoud 	}
   2657   1.1  reinoud 	max_l_ad = lb_size - dscr_size - l_ea;
   2658   1.1  reinoud 
   2659   1.1  reinoud 	icbflags   = udf_rw16(icbtag->flags);
   2660   1.1  reinoud 	addr_type  = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   2661   1.1  reinoud 
   2662   1.1  reinoud 	old_size  = inflen;
   2663   1.1  reinoud 	size_diff = old_size - new_size;
   2664   1.1  reinoud 
   2665   1.1  reinoud 	DPRINTF(ALLOC, ("\tfrom %"PRIu64" to %"PRIu64"\n", old_size, new_size));
   2666   1.1  reinoud 
   2667   1.1  reinoud 	/* shrink the node to its new size */
   2668   1.1  reinoud 	if (addr_type == UDF_ICB_INTERN_ALLOC) {
   2669   1.1  reinoud 		/* only reflect size change directly in the node */
   2670   1.1  reinoud 		KASSERT(new_size <= max_l_ad);
   2671   1.1  reinoud 		inflen  -= size_diff;
   2672   1.1  reinoud 		objsize -= size_diff;
   2673   1.1  reinoud 		l_ad    -= size_diff;
   2674   1.1  reinoud 		crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea + l_ad;
   2675   1.1  reinoud 		if (fe) {
   2676   1.1  reinoud 			fe->inf_len   = udf_rw64(inflen);
   2677   1.1  reinoud 			fe->l_ad      = udf_rw32(l_ad);
   2678  1.22  reinoud 			fe->tag.desc_crc_len = udf_rw16(crclen);
   2679   1.1  reinoud 		} else {
   2680   1.1  reinoud 			efe->inf_len  = udf_rw64(inflen);
   2681   1.1  reinoud 			efe->obj_size = udf_rw64(objsize);
   2682   1.1  reinoud 			efe->l_ad     = udf_rw32(l_ad);
   2683  1.22  reinoud 			efe->tag.desc_crc_len = udf_rw16(crclen);
   2684   1.1  reinoud 		}
   2685   1.1  reinoud 		error = 0;
   2686   1.7  reinoud 
   2687   1.7  reinoud 		/* clear the space in the descriptor */
   2688   1.7  reinoud 		KASSERT(old_size > new_size);
   2689   1.7  reinoud 		memset(data_pos + new_size, 0, old_size - new_size);
   2690   1.7  reinoud 
   2691   1.1  reinoud 		/* TODO zero appened space in buffer! */
   2692   1.1  reinoud 		/* using uvm_vnp_zerorange(vp, old_size, old_size - new_size); ? */
   2693   1.1  reinoud 
   2694   1.1  reinoud 		/* set new size for uvm */
   2695   1.1  reinoud 		uvm_vnp_setsize(vp, new_size);
   2696   1.9  reinoud 
   2697   1.9  reinoud 		udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec);
   2698   1.1  reinoud 		UDF_UNLOCK_NODE(udf_node, 0);
   2699   1.1  reinoud 
   2700   1.1  reinoud 		KASSERT(new_inflen == orig_inflen - size_diff);
   2701   1.1  reinoud 		KASSERT(new_lbrec == orig_lbrec);
   2702   1.1  reinoud 		KASSERT(new_lbrec == 0);
   2703   1.1  reinoud 
   2704   1.1  reinoud 		return 0;
   2705   1.1  reinoud 	}
   2706   1.1  reinoud 
   2707   1.1  reinoud 	/* setup node cleanup extents copy space */
   2708   1.1  reinoud 	node_ad_cpy = malloc(lb_size * UDF_MAX_ALLOC_EXTENTS,
   2709   1.1  reinoud 		M_UDFMNT, M_WAITOK);
   2710   1.1  reinoud 	memset(node_ad_cpy, 0, lb_size * UDF_MAX_ALLOC_EXTENTS);
   2711   1.1  reinoud 
   2712   1.1  reinoud 	/*
   2713   1.1  reinoud 	 * Shrink the node by releasing the allocations and truncate the last
   2714   1.1  reinoud 	 * allocation to the new size. If the new size fits into the
   2715   1.1  reinoud 	 * allocation descriptor itself, transform it into an
   2716   1.1  reinoud 	 * UDF_ICB_INTERN_ALLOC.
   2717   1.1  reinoud 	 */
   2718   1.1  reinoud 	slot     = 0;
   2719   1.1  reinoud 	cpy_slot = 0;
   2720   1.1  reinoud 	foffset  = 0;
   2721   1.1  reinoud 
   2722   1.1  reinoud 	/* 1) copy till first overlap piece to the rewrite buffer */
   2723   1.1  reinoud 	for (;;) {
   2724   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
   2725   1.1  reinoud 		if (eof) {
   2726   1.1  reinoud 			DPRINTF(WRITE,
   2727   1.1  reinoud 				("Shrink node failed: "
   2728   1.1  reinoud 				 "encountered EOF\n"));
   2729   1.1  reinoud 			error = EINVAL;
   2730   1.1  reinoud 			goto errorout; /* panic? */
   2731   1.1  reinoud 		}
   2732   1.1  reinoud 		len   = udf_rw32(s_ad.len);
   2733   1.1  reinoud 		flags = UDF_EXT_FLAGS(len);
   2734   1.1  reinoud 		len   = UDF_EXT_LEN(len);
   2735   1.1  reinoud 
   2736   1.1  reinoud 		if (flags == UDF_EXT_REDIRECT) {
   2737   1.1  reinoud 			slot++;
   2738   1.1  reinoud 			continue;
   2739   1.1  reinoud 		}
   2740   1.1  reinoud 
   2741   1.1  reinoud 		end_foffset = foffset + len;
   2742   1.1  reinoud 		if (end_foffset > new_size)
   2743   1.1  reinoud 			break;	/* found */
   2744   1.1  reinoud 
   2745   1.1  reinoud 		node_ad_cpy[cpy_slot++] = s_ad;
   2746   1.1  reinoud 
   2747   1.1  reinoud 		DPRINTF(ALLOC, ("\t1: vp %d, lb %d, len %d, flags %d "
   2748   1.1  reinoud 			"-> stack\n",
   2749   1.1  reinoud 			udf_rw16(s_ad.loc.part_num),
   2750   1.1  reinoud 			udf_rw32(s_ad.loc.lb_num),
   2751   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2752   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2753   1.1  reinoud 
   2754   1.1  reinoud 		foffset = end_foffset;
   2755   1.1  reinoud 		slot++;
   2756   1.1  reinoud 	}
   2757   1.1  reinoud 	slot_offset = new_size - foffset;
   2758   1.1  reinoud 
   2759   1.1  reinoud 	/* 2) trunc overlapping slot at overlap and copy it */
   2760   1.1  reinoud 	if (slot_offset > 0) {
   2761   1.1  reinoud 		lb_num    = udf_rw32(s_ad.loc.lb_num);
   2762   1.1  reinoud 		vpart_num = udf_rw16(s_ad.loc.part_num);
   2763   1.1  reinoud 
   2764   1.1  reinoud 		if (flags == UDF_EXT_ALLOCATED) {
   2765  1.10  reinoud 			/* note: round DOWN on num_lb */
   2766   1.1  reinoud 			lb_num += (slot_offset + lb_size -1) / lb_size;
   2767  1.10  reinoud 			num_lb  = (len - slot_offset) / lb_size;
   2768   1.1  reinoud 
   2769   1.1  reinoud 			udf_free_allocated_space(ump, lb_num, vpart_num, num_lb);
   2770   1.1  reinoud 		}
   2771   1.1  reinoud 
   2772   1.1  reinoud 		s_ad.len = udf_rw32(slot_offset | flags);
   2773   1.1  reinoud 		node_ad_cpy[cpy_slot++] = s_ad;
   2774   1.1  reinoud 		slot++;
   2775   1.1  reinoud 
   2776   1.1  reinoud 		DPRINTF(ALLOC, ("\t2: vp %d, lb %d, len %d, flags %d "
   2777   1.1  reinoud 			"-> stack\n",
   2778   1.1  reinoud 			udf_rw16(s_ad.loc.part_num),
   2779   1.1  reinoud 			udf_rw32(s_ad.loc.lb_num),
   2780   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2781   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2782   1.1  reinoud 	}
   2783   1.1  reinoud 
   2784   1.1  reinoud 	/* 3) delete remainder */
   2785   1.1  reinoud 	for (;;) {
   2786   1.1  reinoud 		udf_get_adslot(udf_node, slot, &s_ad, &eof);
   2787   1.1  reinoud 		if (eof)
   2788   1.1  reinoud 			break;
   2789   1.1  reinoud 
   2790   1.1  reinoud 		len       = udf_rw32(s_ad.len);
   2791   1.1  reinoud 		flags     = UDF_EXT_FLAGS(len);
   2792   1.1  reinoud 		len       = UDF_EXT_LEN(len);
   2793   1.1  reinoud 
   2794   1.1  reinoud 		if (flags == UDF_EXT_REDIRECT) {
   2795   1.1  reinoud 			slot++;
   2796   1.1  reinoud 			continue;
   2797   1.1  reinoud 		}
   2798   1.1  reinoud 
   2799   1.1  reinoud 		DPRINTF(ALLOC, ("\t3: delete remainder "
   2800   1.1  reinoud 			"vp %d lb %d, len %d, flags %d\n",
   2801   1.1  reinoud 		udf_rw16(s_ad.loc.part_num),
   2802   1.1  reinoud 		udf_rw32(s_ad.loc.lb_num),
   2803   1.1  reinoud 		UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2804   1.1  reinoud 		UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2805   1.1  reinoud 
   2806   1.1  reinoud 		if (flags == UDF_EXT_ALLOCATED) {
   2807   1.1  reinoud 			lb_num    = udf_rw32(s_ad.loc.lb_num);
   2808   1.1  reinoud 			vpart_num = udf_rw16(s_ad.loc.part_num);
   2809   1.1  reinoud 			num_lb    = (len + lb_size - 1) / lb_size;
   2810   1.1  reinoud 
   2811   1.1  reinoud 			udf_free_allocated_space(ump, lb_num, vpart_num,
   2812   1.1  reinoud 				num_lb);
   2813   1.1  reinoud 		}
   2814   1.1  reinoud 
   2815   1.1  reinoud 		slot++;
   2816   1.1  reinoud 	}
   2817   1.1  reinoud 
   2818   1.1  reinoud 	/* 4) if it will fit into the descriptor then convert */
   2819   1.1  reinoud 	if (new_size < max_l_ad) {
   2820   1.1  reinoud 		/*
   2821   1.1  reinoud 		 * resque/evacuate old piece by reading it in, and convert it
   2822   1.1  reinoud 		 * to internal alloc.
   2823   1.1  reinoud 		 */
   2824   1.1  reinoud 		if (new_size == 0) {
   2825   1.1  reinoud 			/* XXX/TODO only for zero sizing now */
   2826   1.1  reinoud 			udf_wipe_adslots(udf_node);
   2827   1.1  reinoud 
   2828   1.1  reinoud 			icbflags &= ~UDF_ICB_TAG_FLAGS_ALLOC_MASK;
   2829   1.1  reinoud 			icbflags |=  UDF_ICB_INTERN_ALLOC;
   2830   1.1  reinoud 			icbtag->flags = udf_rw16(icbflags);
   2831   1.1  reinoud 
   2832   1.1  reinoud 			inflen  -= size_diff;	KASSERT(inflen == 0);
   2833   1.1  reinoud 			objsize -= size_diff;
   2834   1.1  reinoud 			l_ad     = new_size;
   2835   1.1  reinoud 			crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea + l_ad;
   2836   1.1  reinoud 			if (fe) {
   2837   1.1  reinoud 				fe->inf_len   = udf_rw64(inflen);
   2838   1.1  reinoud 				fe->l_ad      = udf_rw32(l_ad);
   2839  1.22  reinoud 				fe->tag.desc_crc_len = udf_rw16(crclen);
   2840   1.1  reinoud 			} else {
   2841   1.1  reinoud 				efe->inf_len  = udf_rw64(inflen);
   2842   1.1  reinoud 				efe->obj_size = udf_rw64(objsize);
   2843   1.1  reinoud 				efe->l_ad     = udf_rw32(l_ad);
   2844  1.22  reinoud 				efe->tag.desc_crc_len = udf_rw16(crclen);
   2845   1.1  reinoud 			}
   2846   1.1  reinoud 			/* eventually copy in evacuated piece */
   2847   1.1  reinoud 			/* set new size for uvm */
   2848   1.1  reinoud 			uvm_vnp_setsize(vp, new_size);
   2849   1.1  reinoud 
   2850   1.1  reinoud 			free(node_ad_cpy, M_UDFMNT);
   2851   1.9  reinoud 			udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec);
   2852   1.9  reinoud 
   2853   1.1  reinoud 			UDF_UNLOCK_NODE(udf_node, 0);
   2854   1.1  reinoud 
   2855   1.1  reinoud 			KASSERT(new_inflen == orig_inflen - size_diff);
   2856   1.1  reinoud 			KASSERT(new_inflen == 0);
   2857   1.1  reinoud 			KASSERT(new_lbrec == 0);
   2858   1.1  reinoud 
   2859   1.1  reinoud 			return 0;
   2860   1.1  reinoud 		}
   2861   1.1  reinoud 
   2862   1.1  reinoud 		printf("UDF_SHRINK_NODE: could convert to internal alloc!\n");
   2863   1.1  reinoud 	}
   2864   1.1  reinoud 
   2865   1.1  reinoud 	/* 5) reset node descriptors */
   2866   1.1  reinoud 	udf_wipe_adslots(udf_node);
   2867   1.1  reinoud 
   2868   1.1  reinoud 	/* 6) copy back extents; merge when possible. Recounting on the fly */
   2869   1.1  reinoud 	cpy_slots = cpy_slot;
   2870   1.1  reinoud 
   2871   1.1  reinoud 	c_ad = node_ad_cpy[0];
   2872   1.1  reinoud 	slot = 0;
   2873   1.1  reinoud 	for (cpy_slot = 1; cpy_slot < cpy_slots; cpy_slot++) {
   2874   1.1  reinoud 		s_ad = node_ad_cpy[cpy_slot];
   2875   1.1  reinoud 
   2876   1.1  reinoud 		DPRINTF(ALLOC, ("\t6: stack -> got mapping vp %d "
   2877   1.1  reinoud 			"lb %d, len %d, flags %d\n",
   2878   1.1  reinoud 		udf_rw16(s_ad.loc.part_num),
   2879   1.1  reinoud 		udf_rw32(s_ad.loc.lb_num),
   2880   1.1  reinoud 		UDF_EXT_LEN(udf_rw32(s_ad.len)),
   2881   1.1  reinoud 		UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30));
   2882   1.1  reinoud 
   2883   1.1  reinoud 		/* see if we can merge */
   2884   1.1  reinoud 		if (udf_ads_merge(lb_size, &c_ad, &s_ad)) {
   2885   1.1  reinoud 			/* not mergable (anymore) */
   2886   1.1  reinoud 			DPRINTF(ALLOC, ("\t6: appending vp %d lb %d, "
   2887   1.1  reinoud 				"len %d, flags %d\n",
   2888   1.1  reinoud 			udf_rw16(c_ad.loc.part_num),
   2889   1.1  reinoud 			udf_rw32(c_ad.loc.lb_num),
   2890   1.1  reinoud 			UDF_EXT_LEN(udf_rw32(c_ad.len)),
   2891   1.1  reinoud 			UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30));
   2892   1.1  reinoud 
   2893  1.11  reinoud 			error = udf_append_adslot(udf_node, &slot, &c_ad);
   2894   1.1  reinoud 			if (error)
   2895   1.1  reinoud 				goto errorout; /* panic? */
   2896   1.1  reinoud 			c_ad = s_ad;
   2897   1.1  reinoud 			slot++;
   2898   1.1  reinoud 		}
   2899   1.1  reinoud 	}
   2900   1.1  reinoud 
   2901   1.1  reinoud 	/* 7) push rest slot (if any) */
   2902   1.1  reinoud 	if (UDF_EXT_LEN(c_ad.len) > 0) {
   2903   1.1  reinoud 		DPRINTF(ALLOC, ("\t7: last append vp %d lb %d, "
   2904   1.1  reinoud 				"len %d, flags %d\n",
   2905   1.1  reinoud 		udf_rw16(c_ad.loc.part_num),
   2906   1.1  reinoud 		udf_rw32(c_ad.loc.lb_num),
   2907   1.1  reinoud 		UDF_EXT_LEN(udf_rw32(c_ad.len)),
   2908   1.1  reinoud 		UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30));
   2909   1.1  reinoud 
   2910  1.11  reinoud 		error = udf_append_adslot(udf_node, &slot, &c_ad);
   2911   1.1  reinoud 		if (error)
   2912   1.1  reinoud 			goto errorout; /* panic? */
   2913   1.1  reinoud 		;
   2914   1.1  reinoud 	}
   2915   1.1  reinoud 
   2916   1.1  reinoud 	inflen  -= size_diff;
   2917   1.1  reinoud 	objsize -= size_diff;
   2918   1.1  reinoud 	if (fe) {
   2919   1.1  reinoud 		fe->inf_len   = udf_rw64(inflen);
   2920   1.1  reinoud 	} else {
   2921   1.1  reinoud 		efe->inf_len  = udf_rw64(inflen);
   2922   1.1  reinoud 		efe->obj_size = udf_rw64(objsize);
   2923   1.1  reinoud 	}
   2924   1.1  reinoud 	error = 0;
   2925   1.1  reinoud 
   2926   1.1  reinoud 	/* set new size for uvm */
   2927   1.1  reinoud 	uvm_vnp_setsize(vp, new_size);
   2928   1.1  reinoud 
   2929   1.1  reinoud errorout:
   2930   1.1  reinoud 	free(node_ad_cpy, M_UDFMNT);
   2931   1.9  reinoud 
   2932  1.11  reinoud 	udf_count_alloc_exts(udf_node);
   2933  1.11  reinoud 
   2934   1.9  reinoud 	udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec);
   2935   1.1  reinoud 	UDF_UNLOCK_NODE(udf_node, 0);
   2936   1.1  reinoud 
   2937   1.1  reinoud 	KASSERT(new_inflen == orig_inflen - size_diff);
   2938   1.1  reinoud 
   2939   1.1  reinoud 	return error;
   2940   1.1  reinoud }
   2941   1.1  reinoud 
   2942