Home | History | Annotate | Line # | Download | only in v7fs
v7fs_datablock.c revision 1.5
      1  1.5  apb /*	$NetBSD: v7fs_datablock.c,v 1.5 2011/08/14 09:02:07 apb Exp $	*/
      2  1.1  uch 
      3  1.1  uch /*-
      4  1.1  uch  * Copyright (c) 2011 The NetBSD Foundation, Inc.
      5  1.1  uch  * All rights reserved.
      6  1.1  uch  *
      7  1.1  uch  * This code is derived from software contributed to The NetBSD Foundation
      8  1.1  uch  * by UCHIYAMA Yasushi.
      9  1.1  uch  *
     10  1.1  uch  * Redistribution and use in source and binary forms, with or without
     11  1.1  uch  * modification, are permitted provided that the following conditions
     12  1.1  uch  * are met:
     13  1.1  uch  * 1. Redistributions of source code must retain the above copyright
     14  1.1  uch  *    notice, this list of conditions and the following disclaimer.
     15  1.1  uch  * 2. Redistributions in binary form must reproduce the above copyright
     16  1.1  uch  *    notice, this list of conditions and the following disclaimer in the
     17  1.1  uch  *    documentation and/or other materials provided with the distribution.
     18  1.1  uch  *
     19  1.1  uch  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  1.1  uch  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  1.1  uch  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  1.1  uch  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  1.1  uch  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  1.1  uch  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  1.1  uch  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  1.1  uch  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  1.1  uch  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  1.1  uch  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  1.1  uch  * POSSIBILITY OF SUCH DAMAGE.
     30  1.1  uch  */
     31  1.1  uch 
     32  1.4  apb #if HAVE_NBTOOL_CONFIG_H
     33  1.4  apb #include "nbtool_config.h"
     34  1.4  apb #endif
     35  1.4  apb 
     36  1.1  uch #include <sys/cdefs.h>
     37  1.5  apb __KERNEL_RCSID(0, "$NetBSD: v7fs_datablock.c,v 1.5 2011/08/14 09:02:07 apb Exp $");
     38  1.1  uch #if defined _KERNEL_OPT
     39  1.1  uch #include "opt_v7fs.h"
     40  1.1  uch #endif
     41  1.1  uch 
     42  1.1  uch #include <sys/types.h>
     43  1.1  uch #ifdef _KERNEL
     44  1.1  uch #include <sys/systm.h>
     45  1.1  uch #include <sys/param.h>
     46  1.1  uch #else
     47  1.1  uch #include <stdio.h>
     48  1.1  uch #include <string.h>
     49  1.1  uch #include <errno.h>
     50  1.1  uch #endif
     51  1.1  uch 
     52  1.1  uch #include "v7fs.h"
     53  1.1  uch #include "v7fs_impl.h"
     54  1.1  uch #include "v7fs_endian.h"
     55  1.1  uch #include "v7fs_inode.h"
     56  1.1  uch #include "v7fs_datablock.h"
     57  1.1  uch #include "v7fs_superblock.h"
     58  1.1  uch 
     59  1.1  uch #ifdef V7FS_DATABLOCK_DEBUG
     60  1.1  uch #define	DPRINTF(fmt, args...)	printf("%s: " fmt, __func__, ##args)
     61  1.1  uch #else
     62  1.1  uch #define	DPRINTF(fmt, args...)	((void)0)
     63  1.1  uch #endif
     64  1.1  uch 
     65  1.1  uch static int v7fs_datablock_deallocate(struct v7fs_self *, v7fs_daddr_t);
     66  1.5  apb static int v7fs_loop1(struct v7fs_self *, v7fs_daddr_t, size_t *,
     67  1.1  uch     int (*)(struct v7fs_self *, void *, v7fs_daddr_t, size_t), void *);
     68  1.5  apb static int v7fs_loop2(struct v7fs_self *, v7fs_daddr_t, size_t *,
     69  1.1  uch     int (*)(struct v7fs_self *, void *, v7fs_daddr_t, size_t), void *);
     70  1.5  apb static v7fs_daddr_t v7fs_link(struct v7fs_self *, v7fs_daddr_t, int);
     71  1.5  apb static v7fs_daddr_t v7fs_add_leaf(struct v7fs_self *, v7fs_daddr_t, int);
     72  1.5  apb static v7fs_daddr_t v7fs_unlink(struct v7fs_self *, v7fs_daddr_t, int);
     73  1.5  apb static v7fs_daddr_t v7fs_remove_leaf(struct v7fs_self *, v7fs_daddr_t, int);
     74  1.5  apb static v7fs_daddr_t v7fs_remove_self(struct v7fs_self *, v7fs_daddr_t);
     75  1.1  uch 
     76  1.1  uch #ifdef V7FS_DATABLOCK_DEBUG
     77  1.1  uch void daddr_map_dump(const struct v7fs_daddr_map *);
     78  1.1  uch #else
     79  1.1  uch #define	daddr_map_dump(x)	((void)0)
     80  1.1  uch #endif
     81  1.1  uch 
     82  1.1  uch bool
     83  1.1  uch datablock_number_sanity(const struct v7fs_self *fs, v7fs_daddr_t blk)
     84  1.1  uch {
     85  1.1  uch 	const struct v7fs_superblock *sb = &fs->superblock;
     86  1.1  uch 	bool ok = (blk >= sb->datablock_start_sector) &&
     87  1.1  uch 	    (blk < sb->volume_size);
     88  1.1  uch 
     89  1.1  uch #ifdef V7FS_DATABLOCK_DEBUG
     90  1.1  uch 	if (!ok) {
     91  1.1  uch 		DPRINTF("Bad data block #%d\n", blk);
     92  1.1  uch 	}
     93  1.1  uch #endif
     94  1.1  uch 
     95  1.1  uch 	return ok;
     96  1.1  uch }
     97  1.1  uch 
     98  1.1  uch int
     99  1.1  uch v7fs_datablock_allocate(struct v7fs_self *fs, v7fs_daddr_t *block_number)
    100  1.1  uch {
    101  1.1  uch 	struct v7fs_superblock *sb = &fs->superblock;
    102  1.1  uch 	v7fs_daddr_t blk;
    103  1.1  uch 	int error = 0;
    104  1.1  uch 
    105  1.1  uch 	*block_number = 0;
    106  1.1  uch 	SUPERB_LOCK(fs);
    107  1.1  uch 	do {
    108  1.1  uch 		if (!sb->total_freeblock) {
    109  1.1  uch 			DPRINTF("free block exhausted!!!\n");
    110  1.1  uch 			SUPERB_UNLOCK(fs);
    111  1.1  uch 			return ENOSPC;
    112  1.1  uch 		}
    113  1.1  uch 
    114  1.1  uch 		/* Get free block from superblock cache. */
    115  1.1  uch 		blk = sb->freeblock[--sb->nfreeblock];
    116  1.1  uch 		sb->total_freeblock--;
    117  1.1  uch 		sb->modified = 1;
    118  1.1  uch 
    119  1.1  uch 		/* If nfreeblock is zero, it block is next freeblock link. */
    120  1.1  uch 		if (sb->nfreeblock == 0) {
    121  1.1  uch 			if ((error = v7fs_freeblock_update(fs, blk))) {
    122  1.1  uch 				DPRINTF("no freeblock!!!\n");
    123  1.1  uch 				SUPERB_UNLOCK(fs);
    124  1.1  uch 				return error;
    125  1.1  uch 			}
    126  1.1  uch 			/* This freeblock link is no longer required. */
    127  1.1  uch 			/* use as data block. */
    128  1.1  uch 		}
    129  1.1  uch 	} while (!datablock_number_sanity(fs, blk)); /* skip bogus block. */
    130  1.1  uch 	SUPERB_UNLOCK(fs);
    131  1.1  uch 
    132  1.1  uch 	DPRINTF("Get freeblock %d\n", blk);
    133  1.1  uch 	/* Zero clear datablock. */
    134  1.1  uch 	void *buf;
    135  1.1  uch 	if (!(buf = scratch_read(fs, blk)))
    136  1.1  uch 		return EIO;
    137  1.1  uch 	memset(buf, 0, V7FS_BSIZE);
    138  1.1  uch 	if (!fs->io.write(fs->io.cookie, buf, blk))
    139  1.1  uch 		error = EIO;
    140  1.1  uch 	scratch_free(fs, buf);
    141  1.1  uch 
    142  1.1  uch 	if (error == 0)
    143  1.1  uch 		*block_number = blk;
    144  1.1  uch 
    145  1.1  uch 	return error;
    146  1.1  uch }
    147  1.1  uch 
    148  1.1  uch static int
    149  1.1  uch v7fs_datablock_deallocate(struct v7fs_self *fs, v7fs_daddr_t blk)
    150  1.1  uch {
    151  1.1  uch 	struct v7fs_superblock *sb = &fs->superblock;
    152  1.1  uch 	void *buf;
    153  1.1  uch 	int error = 0;
    154  1.1  uch 
    155  1.1  uch 	if (!datablock_number_sanity(fs, blk))
    156  1.1  uch 		return EIO;
    157  1.1  uch 
    158  1.1  uch 	/* Add to in-core freelist. */
    159  1.1  uch 	SUPERB_LOCK(fs);
    160  1.1  uch 	if (sb->nfreeblock < V7FS_MAX_FREEBLOCK) {
    161  1.1  uch 		sb->freeblock[sb->nfreeblock++] = blk;
    162  1.1  uch 		sb->total_freeblock++;
    163  1.1  uch 		sb->modified = 1;
    164  1.1  uch 		DPRINTF("n_freeblock=%d\n", sb->total_freeblock);
    165  1.1  uch 		SUPERB_UNLOCK(fs);
    166  1.1  uch 		return 0;
    167  1.1  uch 	}
    168  1.1  uch 
    169  1.1  uch 	/* No space to push. */
    170  1.1  uch 	/* Make this block to freeblock list.and current cache moved to this. */
    171  1.1  uch 	if (!(buf = scratch_read(fs, blk))) {
    172  1.1  uch 		SUPERB_UNLOCK(fs);
    173  1.1  uch 		return EIO;	/* Fatal */
    174  1.1  uch 	}
    175  1.1  uch 
    176  1.1  uch 	struct v7fs_freeblock *fb = (struct v7fs_freeblock *)buf;
    177  1.1  uch 	fb->nfreeblock = V7FS_MAX_FREEBLOCK;
    178  1.1  uch 	int i;
    179  1.1  uch 	for (i = 0; i < V7FS_MAX_FREEBLOCK; i++)
    180  1.1  uch 		fb->freeblock[i] = V7FS_VAL32(fs, sb->freeblock[i]);
    181  1.1  uch 
    182  1.1  uch 	if (!fs->io.write(fs->io.cookie, (uint8_t *)fb, blk)) {
    183  1.1  uch 		error =  EIO;	/* Fatal */
    184  1.1  uch 	} else {
    185  1.1  uch 		/* Link. on next allocate, this block is used as datablock, */
    186  1.1  uch 		/* and swap outed freeblock list is restored. */
    187  1.1  uch 		sb->freeblock[0] = blk;
    188  1.1  uch 		sb->nfreeblock = 1;
    189  1.1  uch 		sb->total_freeblock++;
    190  1.1  uch 		sb->modified = 1;
    191  1.1  uch 		DPRINTF("n_freeblock=%d\n", sb->total_freeblock);
    192  1.1  uch 	}
    193  1.1  uch 	SUPERB_UNLOCK(fs);
    194  1.1  uch 	scratch_free(fs, buf);
    195  1.1  uch 
    196  1.1  uch 	return error;
    197  1.1  uch }
    198  1.1  uch 
    199  1.3  uch int
    200  1.1  uch v7fs_datablock_addr(size_t sz, struct v7fs_daddr_map *map)
    201  1.1  uch {
    202  1.1  uch #define	NIDX		V7FS_DADDR_PER_BLOCK
    203  1.1  uch #define	DIRECT_SZ	(V7FS_NADDR_DIRECT * V7FS_BSIZE)
    204  1.1  uch #define	IDX1_SZ		(NIDX * V7FS_BSIZE)
    205  1.1  uch #define	IDX2_SZ		(NIDX * NIDX * V7FS_BSIZE)
    206  1.1  uch #define	ROUND(x, a)	((((x) + ((a) - 1)) & ~((a) - 1)))
    207  1.1  uch 	if (!sz) {
    208  1.1  uch 		map->level = 0;
    209  1.1  uch 		map->index[0] = 0;
    210  1.1  uch 		return 0;
    211  1.1  uch 	}
    212  1.1  uch 
    213  1.1  uch 	sz = V7FS_ROUND_BSIZE(sz);
    214  1.1  uch 
    215  1.1  uch 	/* Direct */
    216  1.1  uch 	if (sz <= DIRECT_SZ) {
    217  1.1  uch 		map->level = 0;
    218  1.1  uch 		map->index[0] = (sz >> V7FS_BSHIFT) - 1;
    219  1.1  uch 		return 0;
    220  1.1  uch 	}
    221  1.1  uch 	/* Index 1 */
    222  1.1  uch 	sz -= DIRECT_SZ;
    223  1.1  uch 
    224  1.1  uch 	if (sz <= IDX1_SZ) {
    225  1.1  uch 		map->level = 1;
    226  1.1  uch 		map->index[0] = (sz >> V7FS_BSHIFT) - 1;
    227  1.1  uch 		return 0;
    228  1.1  uch 	}
    229  1.1  uch 	sz -= IDX1_SZ;
    230  1.1  uch 
    231  1.1  uch 	/* Index 2 */
    232  1.1  uch 	if (sz <= IDX2_SZ) {
    233  1.1  uch 		map->level = 2;
    234  1.1  uch 		map->index[0] = ROUND(sz, IDX1_SZ) / IDX1_SZ - 1;
    235  1.1  uch 		map->index[1] = ((sz - (map->index[0] * IDX1_SZ)) >>
    236  1.1  uch 		    V7FS_BSHIFT) - 1;
    237  1.1  uch 		return 0;
    238  1.1  uch 	}
    239  1.1  uch 	sz -= IDX2_SZ;
    240  1.1  uch 
    241  1.1  uch 	/* Index 3 */
    242  1.1  uch 	map->level = 3;
    243  1.1  uch 	map->index[0] = ROUND(sz, IDX2_SZ) / IDX2_SZ - 1;
    244  1.1  uch 	sz -= map->index[0] * IDX2_SZ;
    245  1.1  uch 	map->index[1] = ROUND(sz, IDX1_SZ) / IDX1_SZ - 1;
    246  1.1  uch 	sz -= map->index[1] * IDX1_SZ;
    247  1.1  uch 	map->index[2] = (sz >> V7FS_BSHIFT) - 1;
    248  1.1  uch 
    249  1.1  uch 	return map->index[2] >= NIDX ? ENOSPC : 0;
    250  1.1  uch }
    251  1.1  uch 
    252  1.1  uch int
    253  1.1  uch v7fs_datablock_foreach(struct v7fs_self *fs, struct v7fs_inode *p,
    254  1.1  uch     int (*func)(struct v7fs_self *, void *, v7fs_daddr_t, size_t), void *ctx)
    255  1.1  uch {
    256  1.1  uch 	size_t i;
    257  1.1  uch 	v7fs_daddr_t blk, blk2;
    258  1.1  uch 	size_t filesize;
    259  1.1  uch 	bool last;
    260  1.1  uch 	int ret;
    261  1.1  uch 
    262  1.1  uch 	if (!(filesize = v7fs_inode_filesize(p)))
    263  1.2  uch 		return V7FS_ITERATOR_END;
    264  1.1  uch #ifdef V7FS_DATABLOCK_DEBUG
    265  1.1  uch 	size_t sz = filesize;
    266  1.1  uch #endif
    267  1.1  uch 
    268  1.1  uch 	/* Direct */
    269  1.1  uch 	for (i = 0; i < V7FS_NADDR_DIRECT; i++, filesize -= V7FS_BSIZE) {
    270  1.1  uch 		blk = p->addr[i];
    271  1.1  uch 		if (!datablock_number_sanity(fs, blk)) {
    272  1.1  uch 			DPRINTF("inode#%d direct=%zu filesize=%zu\n",
    273  1.1  uch 			    p->inode_number, i, sz);
    274  1.1  uch 			return EIO;
    275  1.1  uch 		}
    276  1.1  uch 
    277  1.1  uch 		last = filesize <= V7FS_BSIZE;
    278  1.1  uch 		if ((ret = func(fs, ctx, blk, last ? filesize : V7FS_BSIZE)))
    279  1.1  uch 			return ret;
    280  1.1  uch 		if (last)
    281  1.1  uch 			return V7FS_ITERATOR_END;
    282  1.1  uch 	}
    283  1.1  uch 
    284  1.1  uch 	/* Index 1 */
    285  1.1  uch 	blk = p->addr[V7FS_NADDR_INDEX1];
    286  1.1  uch 	if (!datablock_number_sanity(fs, blk))
    287  1.1  uch 		return EIO;
    288  1.1  uch 
    289  1.5  apb 	if ((ret = v7fs_loop1(fs, blk, &filesize, func, ctx)))
    290  1.1  uch 		return ret;
    291  1.1  uch 
    292  1.1  uch 	/* Index 2 */
    293  1.1  uch 	blk = p->addr[V7FS_NADDR_INDEX2];
    294  1.1  uch 	if (!datablock_number_sanity(fs, blk))
    295  1.1  uch 		return EIO;
    296  1.1  uch 
    297  1.5  apb 	if ((ret = v7fs_loop2(fs, blk, &filesize, func, ctx)))
    298  1.1  uch 		return ret;
    299  1.1  uch 
    300  1.1  uch 	/* Index 3 */
    301  1.1  uch 	blk = p->addr[V7FS_NADDR_INDEX3];
    302  1.1  uch 	if (!datablock_number_sanity(fs, blk))
    303  1.1  uch 		return EIO;
    304  1.1  uch 
    305  1.1  uch 	for (i = 0; i < V7FS_DADDR_PER_BLOCK; i++) {
    306  1.5  apb 		blk2 = v7fs_link(fs, blk, i);
    307  1.1  uch 		if (!datablock_number_sanity(fs, blk))
    308  1.1  uch 			return EIO;
    309  1.1  uch 
    310  1.5  apb 		if ((ret = v7fs_loop2(fs, blk2, &filesize, func, ctx)))
    311  1.1  uch 			return ret;
    312  1.1  uch 	}
    313  1.1  uch 
    314  1.1  uch 	return EFBIG;
    315  1.1  uch }
    316  1.1  uch 
    317  1.1  uch static int
    318  1.5  apb v7fs_loop2(struct v7fs_self *fs, v7fs_daddr_t listblk, size_t *filesize,
    319  1.1  uch     int (*func)(struct v7fs_self *, void *, v7fs_daddr_t, size_t), void *ctx)
    320  1.1  uch {
    321  1.1  uch 	v7fs_daddr_t blk;
    322  1.1  uch 	int ret;
    323  1.1  uch 	size_t j;
    324  1.1  uch 
    325  1.1  uch 	for (j = 0; j < V7FS_DADDR_PER_BLOCK; j++) {
    326  1.5  apb 		blk = v7fs_link(fs, listblk, j);
    327  1.1  uch 		if (!datablock_number_sanity(fs, blk))
    328  1.1  uch 			return EIO;
    329  1.5  apb 		if ((ret = v7fs_loop1(fs, blk, filesize, func, ctx)))
    330  1.1  uch 			return ret;
    331  1.1  uch 	}
    332  1.1  uch 
    333  1.1  uch 	return 0;
    334  1.1  uch }
    335  1.1  uch 
    336  1.1  uch static int
    337  1.5  apb v7fs_loop1(struct v7fs_self *fs, v7fs_daddr_t listblk, size_t *filesize,
    338  1.1  uch     int (*func)(struct v7fs_self *, void *, v7fs_daddr_t, size_t), void *ctx)
    339  1.1  uch {
    340  1.1  uch 	v7fs_daddr_t blk;
    341  1.1  uch 	bool last;
    342  1.1  uch 	int ret;
    343  1.1  uch 	size_t k;
    344  1.1  uch 
    345  1.1  uch 	for (k = 0; k < V7FS_DADDR_PER_BLOCK; k++, *filesize -= V7FS_BSIZE) {
    346  1.5  apb 		blk = v7fs_link(fs, listblk, k);
    347  1.1  uch 		if (!datablock_number_sanity(fs, blk))
    348  1.1  uch 			return EIO;
    349  1.1  uch 		last = *filesize <= V7FS_BSIZE;
    350  1.1  uch 		if ((ret = func(fs, ctx, blk, last ? *filesize : V7FS_BSIZE)))
    351  1.1  uch 			return ret;
    352  1.1  uch 		if (last)
    353  1.1  uch 			return V7FS_ITERATOR_END;
    354  1.1  uch 	}
    355  1.1  uch 
    356  1.1  uch 	return 0;
    357  1.1  uch }
    358  1.1  uch 
    359  1.1  uch v7fs_daddr_t
    360  1.1  uch v7fs_datablock_last(struct v7fs_self *fs, struct v7fs_inode *inode,
    361  1.1  uch     v7fs_off_t ofs)
    362  1.1  uch {
    363  1.1  uch 	struct v7fs_daddr_map map;
    364  1.1  uch 	v7fs_daddr_t blk = 0;
    365  1.1  uch 	v7fs_daddr_t *addr = inode->addr;
    366  1.1  uch 
    367  1.1  uch 	/* Inquire last data block location. */
    368  1.1  uch 	if (v7fs_datablock_addr(ofs, &map) != 0)
    369  1.1  uch 		return 0;
    370  1.1  uch 
    371  1.1  uch 	switch (map.level)
    372  1.1  uch 	{
    373  1.1  uch 	case 0: /*Direct */
    374  1.1  uch 		blk = inode->addr[map.index[0]];
    375  1.1  uch 		break;
    376  1.1  uch 	case 1: /*Index1 */
    377  1.5  apb 		blk = v7fs_link(fs, addr[V7FS_NADDR_INDEX1], map.index[0]);
    378  1.1  uch 		break;
    379  1.1  uch 	case 2: /*Index2 */
    380  1.5  apb 		blk = v7fs_link(fs, v7fs_link(fs,
    381  1.5  apb 		    addr[V7FS_NADDR_INDEX2], map.index[0]), map.index[1]);
    382  1.1  uch 		break;
    383  1.1  uch 	case 3: /*Index3 */
    384  1.5  apb 		blk = v7fs_link(fs, v7fs_link(fs, v7fs_link(fs,
    385  1.5  apb 		    addr[V7FS_NADDR_INDEX3], map.index[0]), map.index[1]),
    386  1.5  apb 		    map.index[2]);
    387  1.1  uch 		break;
    388  1.1  uch 	}
    389  1.1  uch 
    390  1.1  uch 	return blk;
    391  1.1  uch }
    392  1.1  uch 
    393  1.1  uch int
    394  1.1  uch v7fs_datablock_expand(struct v7fs_self *fs, struct v7fs_inode *inode, size_t sz)
    395  1.1  uch {
    396  1.1  uch 	size_t old_filesize = inode->filesize;
    397  1.1  uch 	size_t new_filesize = old_filesize + sz;
    398  1.1  uch 	struct v7fs_daddr_map oldmap, newmap;
    399  1.1  uch 	v7fs_daddr_t blk, idxblk;
    400  1.1  uch 	int error;
    401  1.1  uch 	v7fs_daddr_t old_nblk = V7FS_ROUND_BSIZE(old_filesize) >> V7FS_BSHIFT;
    402  1.1  uch 	v7fs_daddr_t new_nblk = V7FS_ROUND_BSIZE(new_filesize) >> V7FS_BSHIFT;
    403  1.1  uch 
    404  1.1  uch 	if (old_nblk == new_nblk) {
    405  1.1  uch 		inode->filesize += sz;
    406  1.1  uch 		v7fs_inode_writeback(fs, inode);
    407  1.1  uch 		return 0; /* no need to expand. */
    408  1.1  uch 	}
    409  1.1  uch 	struct v7fs_inode backup = *inode;
    410  1.1  uch 	v7fs_daddr_t required_blk = new_nblk - old_nblk;
    411  1.1  uch 
    412  1.1  uch 	DPRINTF("%zu->%zu, required block=%d\n", old_filesize, new_filesize,
    413  1.1  uch 	    required_blk);
    414  1.1  uch 
    415  1.1  uch 	v7fs_datablock_addr(old_filesize, &oldmap);
    416  1.1  uch 	v7fs_daddr_t i;
    417  1.1  uch 	for (i = 0; i < required_blk; i++) {
    418  1.1  uch 		v7fs_datablock_addr(old_filesize + (i+1) * V7FS_BSIZE, &newmap);
    419  1.1  uch 		daddr_map_dump(&oldmap);
    420  1.1  uch 		daddr_map_dump(&newmap);
    421  1.1  uch 
    422  1.1  uch 		if (oldmap.level != newmap.level) {
    423  1.1  uch 			/* Allocate index area */
    424  1.1  uch 			if ((error = v7fs_datablock_allocate(fs, &idxblk)))
    425  1.1  uch 				return error;
    426  1.1  uch 
    427  1.1  uch 			switch (newmap.level) {
    428  1.1  uch 			case 1:
    429  1.1  uch 				DPRINTF("0->1\n");
    430  1.1  uch 				inode->addr[V7FS_NADDR_INDEX1] = idxblk;
    431  1.5  apb 				blk = v7fs_add_leaf(fs, idxblk, 0);
    432  1.1  uch 				break;
    433  1.1  uch 			case 2:
    434  1.1  uch 				DPRINTF("1->2\n");
    435  1.1  uch 				inode->addr[V7FS_NADDR_INDEX2] = idxblk;
    436  1.5  apb 				blk = v7fs_add_leaf(fs, v7fs_add_leaf(fs,
    437  1.5  apb 				    idxblk, 0), 0);
    438  1.1  uch 				break;
    439  1.1  uch 			case 3:
    440  1.1  uch 				DPRINTF("2->3\n");
    441  1.1  uch 				inode->addr[V7FS_NADDR_INDEX3] = idxblk;
    442  1.5  apb 				blk = v7fs_add_leaf(fs, v7fs_add_leaf(fs,
    443  1.5  apb 				    v7fs_add_leaf(fs, idxblk, 0), 0), 0);
    444  1.1  uch 				break;
    445  1.1  uch 			}
    446  1.1  uch 		} else {
    447  1.1  uch 			switch (newmap.level) {
    448  1.1  uch 			case 0:
    449  1.1  uch 				if ((error = v7fs_datablock_allocate(fs, &blk)))
    450  1.1  uch 					return error;
    451  1.1  uch 				inode->addr[newmap.index[0]] = blk;
    452  1.1  uch 				DPRINTF("direct index %d = blk%d\n",
    453  1.1  uch 				    newmap.index[0], blk);
    454  1.1  uch 				break;
    455  1.1  uch 			case 1:
    456  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX1];
    457  1.5  apb 				blk = v7fs_add_leaf(fs, idxblk,
    458  1.5  apb 				    newmap.index[0]);
    459  1.1  uch 				break;
    460  1.1  uch 			case 2:
    461  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX2];
    462  1.5  apb 				if (oldmap.index[0] != newmap.index[0]) {
    463  1.5  apb 					v7fs_add_leaf(fs, idxblk,
    464  1.5  apb 					    newmap.index[0]);
    465  1.5  apb 				}
    466  1.5  apb 				blk = v7fs_add_leaf(fs, v7fs_link(fs,idxblk,
    467  1.1  uch 				    newmap.index[0]), newmap.index[1]);
    468  1.1  uch 				break;
    469  1.1  uch 			case 3:
    470  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX3];
    471  1.1  uch 
    472  1.5  apb 				if (oldmap.index[0] != newmap.index[0]) {
    473  1.5  apb 					v7fs_add_leaf(fs, idxblk,
    474  1.5  apb 					    newmap.index[0]);
    475  1.5  apb 				}
    476  1.1  uch 
    477  1.5  apb 				if (oldmap.index[1] != newmap.index[1]) {
    478  1.5  apb 					v7fs_add_leaf(fs, v7fs_link(fs, idxblk,
    479  1.1  uch 					    newmap.index[0]), newmap.index[1]);
    480  1.5  apb 				}
    481  1.5  apb 				blk = v7fs_add_leaf(fs, v7fs_link(fs,
    482  1.5  apb 				    v7fs_link(fs, idxblk, newmap.index[0]),
    483  1.5  apb 				    newmap.index[1]), newmap.index[2]);
    484  1.1  uch 				break;
    485  1.1  uch 			}
    486  1.1  uch 		}
    487  1.1  uch 		if (!blk) {
    488  1.1  uch 			*inode = backup; /* structure copy; */
    489  1.1  uch 			return ENOSPC;
    490  1.1  uch 		}
    491  1.1  uch 		oldmap = newmap;
    492  1.1  uch 	}
    493  1.1  uch 	inode->filesize += sz;
    494  1.1  uch 	v7fs_inode_writeback(fs, inode);
    495  1.1  uch 
    496  1.1  uch 	return 0;
    497  1.1  uch }
    498  1.1  uch 
    499  1.1  uch static v7fs_daddr_t
    500  1.5  apb v7fs_link(struct v7fs_self *fs, v7fs_daddr_t listblk, int n)
    501  1.1  uch {
    502  1.1  uch 	v7fs_daddr_t *list;
    503  1.1  uch 	v7fs_daddr_t blk;
    504  1.1  uch 	void *buf;
    505  1.1  uch 
    506  1.1  uch 	if (!datablock_number_sanity(fs, listblk))
    507  1.1  uch 		return 0;
    508  1.1  uch 	if (!(buf = scratch_read(fs, listblk)))
    509  1.1  uch 		return 0;
    510  1.1  uch 	list = (v7fs_daddr_t *)buf;
    511  1.1  uch 	blk = V7FS_VAL32(fs, list[n]);
    512  1.1  uch 	scratch_free(fs, buf);
    513  1.1  uch 
    514  1.1  uch 	if (!datablock_number_sanity(fs, blk))
    515  1.1  uch 		return 0;
    516  1.1  uch 
    517  1.1  uch 	return blk;
    518  1.1  uch }
    519  1.1  uch 
    520  1.1  uch static v7fs_daddr_t
    521  1.5  apb v7fs_add_leaf(struct v7fs_self *fs, v7fs_daddr_t up, int idx)
    522  1.1  uch {
    523  1.1  uch 	v7fs_daddr_t newblk;
    524  1.1  uch 	v7fs_daddr_t *daddr_list;
    525  1.1  uch 	int error = 0;
    526  1.1  uch 	void *buf;
    527  1.1  uch 
    528  1.1  uch 	if (!up)
    529  1.1  uch 		return 0;
    530  1.1  uch 	if (!datablock_number_sanity(fs, up))
    531  1.1  uch 		return 0;
    532  1.1  uch 
    533  1.1  uch 	if ((error = v7fs_datablock_allocate(fs, &newblk)))
    534  1.1  uch 		return 0;
    535  1.1  uch 	if (!(buf = scratch_read(fs, up)))
    536  1.1  uch 		return 0;
    537  1.1  uch 	daddr_list = (v7fs_daddr_t *)buf;
    538  1.1  uch 	daddr_list[idx] = V7FS_VAL32(fs, newblk);
    539  1.1  uch 	if (!fs->io.write(fs->io.cookie, buf, up))
    540  1.1  uch 		newblk = 0;
    541  1.1  uch 	scratch_free(fs, buf);
    542  1.1  uch 
    543  1.1  uch 	return newblk;
    544  1.1  uch }
    545  1.1  uch 
    546  1.1  uch int
    547  1.1  uch v7fs_datablock_contract(struct v7fs_self *fs, struct v7fs_inode *inode,
    548  1.1  uch     size_t sz)
    549  1.1  uch {
    550  1.1  uch 	size_t old_filesize = inode->filesize;
    551  1.1  uch 	size_t new_filesize = old_filesize - sz;
    552  1.1  uch 	struct v7fs_daddr_map oldmap, newmap;
    553  1.1  uch 	v7fs_daddr_t blk, idxblk;
    554  1.1  uch 	int error = 0;
    555  1.1  uch 	v7fs_daddr_t old_nblk = V7FS_ROUND_BSIZE(old_filesize) >> V7FS_BSHIFT;
    556  1.1  uch 	v7fs_daddr_t new_nblk = V7FS_ROUND_BSIZE(new_filesize) >> V7FS_BSHIFT;
    557  1.1  uch 
    558  1.1  uch 	if (old_nblk == new_nblk) {
    559  1.1  uch 		inode->filesize -= sz;
    560  1.1  uch 		v7fs_inode_writeback(fs, inode);
    561  1.1  uch 		return 0; /* no need to contract; */
    562  1.1  uch 	}
    563  1.1  uch 	v7fs_daddr_t erase_blk = old_nblk - new_nblk;
    564  1.1  uch 
    565  1.1  uch 	DPRINTF("%zu->%zu # of erased block=%d\n", old_filesize, new_filesize,
    566  1.1  uch 	    erase_blk);
    567  1.1  uch 
    568  1.1  uch 	v7fs_datablock_addr(old_filesize, &oldmap);
    569  1.1  uch 	v7fs_daddr_t i;
    570  1.1  uch 	for (i = 0; i < erase_blk; i++) {
    571  1.1  uch 		v7fs_datablock_addr(old_filesize - (i+1) * V7FS_BSIZE, &newmap);
    572  1.1  uch 
    573  1.1  uch 		if (oldmap.level != newmap.level) {
    574  1.1  uch 			switch (newmap.level) {
    575  1.1  uch 			case 0: /*1->0 */
    576  1.1  uch 				DPRINTF("1->0\n");
    577  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX1];
    578  1.1  uch 				inode->addr[V7FS_NADDR_INDEX1] = 0;
    579  1.1  uch 				error = v7fs_datablock_deallocate(fs,
    580  1.5  apb 				    v7fs_remove_self(fs, idxblk));
    581  1.1  uch 				break;
    582  1.1  uch 			case 1: /*2->1 */
    583  1.1  uch 				DPRINTF("2->1\n");
    584  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX2];
    585  1.1  uch 				inode->addr[V7FS_NADDR_INDEX2] = 0;
    586  1.1  uch 				error = v7fs_datablock_deallocate(fs,
    587  1.5  apb 				    v7fs_remove_self(fs, v7fs_remove_self(fs,
    588  1.5  apb 				    idxblk)));
    589  1.1  uch 				break;
    590  1.1  uch 			case 2:/*3->2 */
    591  1.1  uch 				DPRINTF("3->2\n");
    592  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX3];
    593  1.1  uch 				inode->addr[V7FS_NADDR_INDEX3] = 0;
    594  1.1  uch 				error = v7fs_datablock_deallocate(fs,
    595  1.5  apb 				    v7fs_remove_self(fs, v7fs_remove_self(fs,
    596  1.5  apb 					v7fs_remove_self(fs, idxblk))));
    597  1.1  uch 				break;
    598  1.1  uch 			}
    599  1.1  uch 		} else {
    600  1.1  uch 			switch (newmap.level) {
    601  1.1  uch 			case 0:
    602  1.1  uch 				DPRINTF("[0] %d\n", oldmap.index[0]);
    603  1.1  uch 				blk = inode->addr[oldmap.index[0]];
    604  1.1  uch 				error = v7fs_datablock_deallocate(fs, blk);
    605  1.1  uch 				break;
    606  1.1  uch 			case 1:
    607  1.1  uch 				DPRINTF("[1] %d\n", oldmap.index[0]);
    608  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX1];
    609  1.5  apb 				v7fs_remove_leaf(fs, idxblk, oldmap.index[0]);
    610  1.1  uch 
    611  1.1  uch 				break;
    612  1.1  uch 			case 2:
    613  1.1  uch 				DPRINTF("[2] %d %d\n", oldmap.index[0],
    614  1.1  uch 				    oldmap.index[1]);
    615  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX2];
    616  1.5  apb 				v7fs_remove_leaf(fs, v7fs_link(fs, idxblk,
    617  1.1  uch 				    oldmap.index[0]), oldmap.index[1]);
    618  1.1  uch 				if (oldmap.index[0] != newmap.index[0]) {
    619  1.5  apb 					v7fs_remove_leaf(fs, idxblk,
    620  1.1  uch 					    oldmap.index[0]);
    621  1.1  uch 				}
    622  1.1  uch 				break;
    623  1.1  uch 			case 3:
    624  1.1  uch 				DPRINTF("[2] %d %d %d\n", oldmap.index[0],
    625  1.1  uch 				    oldmap.index[1], oldmap.index[2]);
    626  1.1  uch 				idxblk = inode->addr[V7FS_NADDR_INDEX3];
    627  1.5  apb 				v7fs_remove_leaf(fs, v7fs_link(fs,
    628  1.5  apb 				    v7fs_link(fs, idxblk, oldmap.index[0]),
    629  1.5  apb 				    oldmap.index[1]), oldmap.index[2]);
    630  1.1  uch 
    631  1.1  uch 				if (oldmap.index[1] != newmap.index[1])	{
    632  1.5  apb 					v7fs_remove_leaf(fs, v7fs_link(fs,
    633  1.5  apb 					    idxblk, oldmap.index[0]),
    634  1.5  apb 					    oldmap.index[1]);
    635  1.1  uch 				}
    636  1.1  uch 				if (oldmap.index[0] != newmap.index[0]) {
    637  1.5  apb 					v7fs_remove_leaf(fs, idxblk,
    638  1.1  uch 					    oldmap.index[0]);
    639  1.1  uch 				}
    640  1.1  uch 				break;
    641  1.1  uch 			}
    642  1.1  uch 		}
    643  1.1  uch 		oldmap = newmap;
    644  1.1  uch 	}
    645  1.1  uch 	inode->filesize -= sz;
    646  1.1  uch 	v7fs_inode_writeback(fs, inode);
    647  1.1  uch 
    648  1.1  uch 	return error;
    649  1.1  uch }
    650  1.1  uch 
    651  1.1  uch static v7fs_daddr_t
    652  1.5  apb v7fs_unlink(struct v7fs_self *fs, v7fs_daddr_t idxblk, int n)
    653  1.1  uch {
    654  1.1  uch 	v7fs_daddr_t *daddr_list;
    655  1.1  uch 	v7fs_daddr_t blk;
    656  1.1  uch 	void *buf;
    657  1.1  uch 
    658  1.1  uch 	if (!(buf = scratch_read(fs, idxblk)))
    659  1.1  uch 		return 0;
    660  1.1  uch 	daddr_list = (v7fs_daddr_t *)buf;
    661  1.1  uch 	blk = V7FS_VAL32(fs, daddr_list[n]);
    662  1.1  uch 	daddr_list[n] = 0;
    663  1.1  uch 	fs->io.write(fs->io.cookie, buf, idxblk);
    664  1.1  uch 	scratch_free(fs, buf);
    665  1.1  uch 
    666  1.1  uch 	return blk; /* unlinked block. */
    667  1.1  uch }
    668  1.1  uch 
    669  1.1  uch static v7fs_daddr_t
    670  1.5  apb v7fs_remove_self(struct v7fs_self *fs, v7fs_daddr_t up)
    671  1.1  uch {
    672  1.1  uch 	v7fs_daddr_t down;
    673  1.1  uch 
    674  1.1  uch 	if (!datablock_number_sanity(fs, up))
    675  1.1  uch 		return 0;
    676  1.1  uch 
    677  1.1  uch 	/* At 1st, remove from datablock list. */
    678  1.5  apb 	down = v7fs_unlink(fs, up, 0);
    679  1.1  uch 
    680  1.1  uch 	/* link self to freelist. */
    681  1.1  uch 	v7fs_datablock_deallocate(fs, up);
    682  1.1  uch 
    683  1.1  uch 	return down;
    684  1.1  uch }
    685  1.1  uch 
    686  1.1  uch static v7fs_daddr_t
    687  1.5  apb v7fs_remove_leaf(struct v7fs_self *fs, v7fs_daddr_t up, int n)
    688  1.1  uch {
    689  1.1  uch 	v7fs_daddr_t down;
    690  1.1  uch 
    691  1.1  uch 	if (!datablock_number_sanity(fs, up))
    692  1.1  uch 		return 0;
    693  1.1  uch 
    694  1.1  uch 	/* At 1st, remove from datablock list. */
    695  1.5  apb 	down = v7fs_unlink(fs, up, n);
    696  1.1  uch 
    697  1.1  uch 	/* link leaf to freelist. */
    698  1.1  uch 	v7fs_datablock_deallocate(fs, down);
    699  1.1  uch 
    700  1.1  uch 	return down;
    701  1.1  uch }
    702  1.1  uch 
    703  1.1  uch int
    704  1.1  uch v7fs_datablock_size_change(struct v7fs_self *fs, size_t newsz,
    705  1.1  uch     struct v7fs_inode *inode)
    706  1.1  uch {
    707  1.1  uch 	ssize_t diff = newsz - v7fs_inode_filesize(inode);
    708  1.1  uch 	int error = 0;
    709  1.1  uch 
    710  1.1  uch 	if (diff > 0)
    711  1.1  uch 		error = v7fs_datablock_expand(fs, inode, diff);
    712  1.1  uch 	else if (diff < 0)
    713  1.1  uch 		error = v7fs_datablock_contract(fs, inode, -diff);
    714  1.1  uch 
    715  1.1  uch 	return error;
    716  1.1  uch }
    717  1.1  uch 
    718  1.1  uch #ifdef V7FS_DATABLOCK_DEBUG
    719  1.1  uch void
    720  1.1  uch daddr_map_dump(const struct v7fs_daddr_map *map)
    721  1.1  uch {
    722  1.1  uch 
    723  1.1  uch 	DPRINTF("level %d ", map->level);
    724  1.1  uch 	int m, n = !map->level ? 1 : map->level;
    725  1.1  uch 	for (m = 0; m < n; m++)
    726  1.1  uch 		printf("[%d]", map->index[m]);
    727  1.1  uch 	printf("\n");
    728  1.1  uch }
    729  1.1  uch #endif
    730