HAMMER 45/Many: Stabilization pass, undo sequencing.

[dragonfly.git] / sys / vfs / hammer / hammer_ondisk.c

/*
 * Copyright (c) 2007-2008 The DragonFly Project.  All rights reserved.
 * 
 * This code is derived from software contributed to The DragonFly Project
 * by Matthew Dillon <dillon@backplane.com>
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 3. Neither the name of The DragonFly Project nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific, prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * 
 * $DragonFly: src/sys/vfs/hammer/hammer_ondisk.c,v 1.45 2008/05/15 03:36:40 dillon Exp $
 */
/*
 * Manage HAMMER's on-disk structures.  These routines are primarily
 * responsible for interfacing with the kernel's I/O subsystem and for
 * managing in-memory structures.
 */

#include "hammer.h"
#include <sys/fcntl.h>
#include <sys/nlookup.h>
#include <sys/buf.h>
#include <sys/buf2.h>

static void hammer_free_volume(hammer_volume_t volume);
static int hammer_load_volume(hammer_volume_t volume);
static int hammer_load_buffer(hammer_buffer_t buffer, int isnew);
static int hammer_load_node(hammer_node_t node, int isnew);

/*
 * Red-Black tree support for various structures
 */
static int
hammer_ino_rb_compare(hammer_inode_t ip1, hammer_inode_t ip2)
{
        if (ip1->obj_id < ip2->obj_id)
                return(-1);
        if (ip1->obj_id > ip2->obj_id)
                return(1);
        if (ip1->obj_asof < ip2->obj_asof)
                return(-1);
        if (ip1->obj_asof > ip2->obj_asof)
                return(1);
        return(0);
}

static int
hammer_inode_info_cmp(hammer_inode_info_t info, hammer_inode_t ip)
{
        if (info->obj_id < ip->obj_id)
                return(-1);
        if (info->obj_id > ip->obj_id)
                return(1);
        if (info->obj_asof < ip->obj_asof)
                return(-1);
        if (info->obj_asof > ip->obj_asof)
                return(1);
        return(0);
}

static int
hammer_vol_rb_compare(hammer_volume_t vol1, hammer_volume_t vol2)
{
        if (vol1->vol_no < vol2->vol_no)
                return(-1);
        if (vol1->vol_no > vol2->vol_no)
                return(1);
        return(0);
}

static int
hammer_buf_rb_compare(hammer_buffer_t buf1, hammer_buffer_t buf2)
{
        if (buf1->zone2_offset < buf2->zone2_offset)
                return(-1);
        if (buf1->zone2_offset > buf2->zone2_offset)
                return(1);
        return(0);
}

static int
hammer_nod_rb_compare(hammer_node_t node1, hammer_node_t node2)
{
        if (node1->node_offset < node2->node_offset)
                return(-1);
        if (node1->node_offset > node2->node_offset)
                return(1);
        return(0);
}

/*
 * Note: The lookup function for hammer_ino_rb_tree winds up being named
 * hammer_ino_rb_tree_RB_LOOKUP_INFO(root, info).  The other lookup
 * functions are normal, e.g. hammer_buf_rb_tree_RB_LOOKUP(root, zone2_offset).
 */
RB_GENERATE(hammer_ino_rb_tree, hammer_inode, rb_node, hammer_ino_rb_compare);
RB_GENERATE_XLOOKUP(hammer_ino_rb_tree, INFO, hammer_inode, rb_node,
                hammer_inode_info_cmp, hammer_inode_info_t);
RB_GENERATE2(hammer_vol_rb_tree, hammer_volume, rb_node,
             hammer_vol_rb_compare, int32_t, vol_no);
RB_GENERATE2(hammer_buf_rb_tree, hammer_buffer, rb_node,
             hammer_buf_rb_compare, hammer_off_t, zone2_offset);
RB_GENERATE2(hammer_nod_rb_tree, hammer_node, rb_node,
             hammer_nod_rb_compare, hammer_off_t, node_offset);

/************************************************************************
 *                              VOLUMES                                 *
 ************************************************************************
 *
 * Load a HAMMER volume by name.  Returns 0 on success or a positive error
 * code on failure.  Volumes must be loaded at mount time, get_volume() will
 * not load a new volume.
 *
 * Calls made to hammer_load_volume() or single-threaded
 */
int
hammer_install_volume(struct hammer_mount *hmp, const char *volname)
{
        struct mount *mp;
        hammer_volume_t volume;
        struct hammer_volume_ondisk *ondisk;
        struct nlookupdata nd;
        struct buf *bp = NULL;
        int error;
        int ronly;
        int setmp = 0;

        mp = hmp->mp;
        ronly = ((mp->mnt_flag & MNT_RDONLY) ? 1 : 0);

        /*
         * Allocate a volume structure
         */
        ++hammer_count_volumes;
        volume = kmalloc(sizeof(*volume), M_HAMMER, M_WAITOK|M_ZERO);
        volume->vol_name = kstrdup(volname, M_HAMMER);
        hammer_io_init(&volume->io, hmp, HAMMER_STRUCTURE_VOLUME);
        volume->io.offset = 0LL;

        /*
         * Get the device vnode
         */
        error = nlookup_init(&nd, volume->vol_name, UIO_SYSSPACE, NLC_FOLLOW);
        if (error == 0)
                error = nlookup(&nd);
        if (error == 0)
                error = cache_vref(&nd.nl_nch, nd.nl_cred, &volume->devvp);
        nlookup_done(&nd);
        if (error == 0) {
                if (vn_isdisk(volume->devvp, &error)) {
                        error = vfs_mountedon(volume->devvp);
                }
        }
        if (error == 0 &&
            count_udev(volume->devvp->v_umajor, volume->devvp->v_uminor) > 0) {
                error = EBUSY;
        }
        if (error == 0) {
                vn_lock(volume->devvp, LK_EXCLUSIVE | LK_RETRY);
                error = vinvalbuf(volume->devvp, V_SAVE, 0, 0);
                if (error == 0) {
                        error = VOP_OPEN(volume->devvp, 
                                         (ronly ? FREAD : FREAD|FWRITE),
                                         FSCRED, NULL);
                }
                vn_unlock(volume->devvp);
        }
        if (error) {
                hammer_free_volume(volume);
                return(error);
        }
        volume->devvp->v_rdev->si_mountpoint = mp;
        setmp = 1;

        /*
         * Extract the volume number from the volume header and do various
         * sanity checks.
         */
        error = bread(volume->devvp, 0LL, HAMMER_BUFSIZE, &bp);
        if (error)
                goto late_failure;
        ondisk = (void *)bp->b_data;
        if (ondisk->vol_signature != HAMMER_FSBUF_VOLUME) {
                kprintf("hammer_mount: volume %s has an invalid header\n",
                        volume->vol_name);
                error = EFTYPE;
                goto late_failure;
        }
        volume->vol_no = ondisk->vol_no;
        volume->buffer_base = ondisk->vol_buf_beg;
        volume->vol_flags = ondisk->vol_flags;
        volume->nblocks = ondisk->vol_nblocks; 
        volume->maxbuf_off = HAMMER_ENCODE_RAW_BUFFER(volume->vol_no,
                                    ondisk->vol_buf_end - ondisk->vol_buf_beg);
        RB_INIT(&volume->rb_bufs_root);

        if (RB_EMPTY(&hmp->rb_vols_root)) {
                hmp->fsid = ondisk->vol_fsid;
        } else if (bcmp(&hmp->fsid, &ondisk->vol_fsid, sizeof(uuid_t))) {
                kprintf("hammer_mount: volume %s's fsid does not match "
                        "other volumes\n", volume->vol_name);
                error = EFTYPE;
                goto late_failure;
        }

        /*
         * Insert the volume structure into the red-black tree.
         */
        if (RB_INSERT(hammer_vol_rb_tree, &hmp->rb_vols_root, volume)) {
                kprintf("hammer_mount: volume %s has a duplicate vol_no %d\n",
                        volume->vol_name, volume->vol_no);
                error = EEXIST;
        }

        /*
         * Set the root volume .  HAMMER special cases rootvol the structure.
         * We do not hold a ref because this would prevent related I/O
         * from being flushed.
         */
        if (error == 0 && ondisk->vol_rootvol == ondisk->vol_no) {
                hmp->rootvol = volume;
                if (bp) {
                        brelse(bp);
                        bp = NULL;
                }
                hmp->fsid_udev = dev2udev(vn_todev(volume->devvp));
                hmp->mp->mnt_stat.f_blocks += ondisk->vol0_stat_bigblocks *
                        (HAMMER_LARGEBLOCK_SIZE / HAMMER_BUFSIZE);
        }
late_failure:
        if (bp)
                brelse(bp);
        if (error) {
                /*vinvalbuf(volume->devvp, V_SAVE, 0, 0);*/
                if (setmp)
                        volume->devvp->v_rdev->si_mountpoint = NULL;
                VOP_CLOSE(volume->devvp, ronly ? FREAD : FREAD|FWRITE);
                hammer_free_volume(volume);
        }
        return (error);
}

/*
 * Unload and free a HAMMER volume.  Must return >= 0 to continue scan
 * so returns -1 on failure.
 */
int
hammer_unload_volume(hammer_volume_t volume, void *data __unused)
{
        struct hammer_mount *hmp = volume->io.hmp;
        int ronly = ((hmp->mp->mnt_flag & MNT_RDONLY) ? 1 : 0);

        /*
         * Clean up the root volume pointer, which is held unlocked in hmp.
         */
        if (hmp->rootvol == volume)
                hmp->rootvol = NULL;

        /*
         * Unload buffers.
         */
        RB_SCAN(hammer_buf_rb_tree, &volume->rb_bufs_root, NULL,
                        hammer_unload_buffer, NULL);

        /*
         * Release our buffer and flush anything left in the buffer cache.
         */
        volume->io.waitdep = 1;
        hammer_io_release(&volume->io, 1);

        /*
         * There should be no references on the volume, no clusters, and
         * no super-clusters.
         */
        KKASSERT(volume->io.lock.refs == 0);
        KKASSERT(RB_EMPTY(&volume->rb_bufs_root));

        volume->ondisk = NULL;
        if (volume->devvp) {
                if (volume->devvp->v_rdev &&
                    volume->devvp->v_rdev->si_mountpoint == hmp->mp
                ) {
                        volume->devvp->v_rdev->si_mountpoint = NULL;
                }
                if (ronly) {
                        vinvalbuf(volume->devvp, 0, 0, 0);
                        VOP_CLOSE(volume->devvp, FREAD);
                } else {
                        vinvalbuf(volume->devvp, V_SAVE, 0, 0);
                        VOP_CLOSE(volume->devvp, FREAD|FWRITE);
                }
        }

        /*
         * Destroy the structure
         */
        RB_REMOVE(hammer_vol_rb_tree, &hmp->rb_vols_root, volume);
        hammer_free_volume(volume);
        return(0);
}

static
void
hammer_free_volume(hammer_volume_t volume)
{
        if (volume->vol_name) {
                kfree(volume->vol_name, M_HAMMER);
                volume->vol_name = NULL;
        }
        if (volume->devvp) {
                vrele(volume->devvp);
                volume->devvp = NULL;
        }
        --hammer_count_volumes;
        kfree(volume, M_HAMMER);
}

/*
 * Get a HAMMER volume.  The volume must already exist.
 */
hammer_volume_t
hammer_get_volume(struct hammer_mount *hmp, int32_t vol_no, int *errorp)
{
        struct hammer_volume *volume;

        /*
         * Locate the volume structure
         */
        volume = RB_LOOKUP(hammer_vol_rb_tree, &hmp->rb_vols_root, vol_no);
        if (volume == NULL) {
                *errorp = ENOENT;
                return(NULL);
        }
        hammer_ref(&volume->io.lock);

        /*
         * Deal with on-disk info
         */
        if (volume->ondisk == NULL || volume->io.loading) {
                *errorp = hammer_load_volume(volume);
                if (*errorp) {
                        hammer_rel_volume(volume, 1);
                        volume = NULL;
                }
        } else {
                *errorp = 0;
        }
        return(volume);
}

int
hammer_ref_volume(hammer_volume_t volume)
{
        int error;

        hammer_ref(&volume->io.lock);

        /*
         * Deal with on-disk info
         */
        if (volume->ondisk == NULL || volume->io.loading) {
                error = hammer_load_volume(volume);
                if (error)
                        hammer_rel_volume(volume, 1);
        } else {
                error = 0;
        }
        return (error);
}

hammer_volume_t
hammer_get_root_volume(struct hammer_mount *hmp, int *errorp)
{
        hammer_volume_t volume;

        volume = hmp->rootvol;
        KKASSERT(volume != NULL);
        hammer_ref(&volume->io.lock);

        /*
         * Deal with on-disk info
         */
        if (volume->ondisk == NULL || volume->io.loading) {
                *errorp = hammer_load_volume(volume);
                if (*errorp) {
                        hammer_rel_volume(volume, 1);
                        volume = NULL;
                }
        } else {
                *errorp = 0;
        }
        return (volume);
}

/*
 * Load a volume's on-disk information.  The volume must be referenced and
 * not locked.  We temporarily acquire an exclusive lock to interlock
 * against releases or multiple get's.
 */
static int
hammer_load_volume(hammer_volume_t volume)
{
        int error;

        ++volume->io.loading;
        hammer_lock_ex(&volume->io.lock);

        if (volume->ondisk == NULL) {
                error = hammer_io_read(volume->devvp, &volume->io);
                if (error == 0)
                        volume->ondisk = (void *)volume->io.bp->b_data;
        } else {
                error = 0;
        }
        --volume->io.loading;
        hammer_unlock(&volume->io.lock);
        return(error);
}

/*
 * Release a volume.  Call hammer_io_release on the last reference.  We have
 * to acquire an exclusive lock to interlock against volume->ondisk tests
 * in hammer_load_volume(), and hammer_io_release() also expects an exclusive
 * lock to be held.
 *
 * Volumes are not unloaded from memory during normal operation.
 */
void
hammer_rel_volume(hammer_volume_t volume, int flush)
{
        crit_enter();
        if (volume->io.lock.refs == 1) {
                ++volume->io.loading;
                hammer_lock_ex(&volume->io.lock);
                if (volume->io.lock.refs == 1) {
                        volume->ondisk = NULL;
                        hammer_io_release(&volume->io, flush);
                }
                --volume->io.loading;
                hammer_unlock(&volume->io.lock);
        }
        hammer_unref(&volume->io.lock);
        crit_exit();
}

/************************************************************************
 *                              BUFFERS                                 *
 ************************************************************************
 *
 * Manage buffers.  Currently all blockmap-backed zones are translated
 * to zone-2 buffer offsets.
 */
hammer_buffer_t
hammer_get_buffer(hammer_mount_t hmp, hammer_off_t buf_offset,
                  int isnew, int *errorp)
{
        hammer_buffer_t buffer;
        hammer_volume_t volume;
        hammer_off_t    zoneX_offset;
        hammer_io_type_t iotype;
        int vol_no;
        int zone;

        zoneX_offset = buf_offset;
        zone = HAMMER_ZONE_DECODE(buf_offset);

        /*
         * What is the buffer class?
         */
        switch(zone) {
        case HAMMER_ZONE_LARGE_DATA_INDEX:
        case HAMMER_ZONE_SMALL_DATA_INDEX:
                iotype = HAMMER_STRUCTURE_DATA_BUFFER;
                break;
        case HAMMER_ZONE_UNDO_INDEX:
                iotype = HAMMER_STRUCTURE_UNDO_BUFFER;
                break;
        default:
                iotype = HAMMER_STRUCTURE_META_BUFFER;
                break;
        }

        /*
         * Handle blockmap offset translations
         */
        if (zone >= HAMMER_ZONE_BTREE_INDEX) {
                buf_offset = hammer_blockmap_lookup(hmp, buf_offset, errorp);
                KKASSERT(*errorp == 0);
        } else if (zone == HAMMER_ZONE_UNDO_INDEX) {
                buf_offset = hammer_undo_lookup(hmp, buf_offset, errorp);
                KKASSERT(*errorp == 0);
        }

        /*
         * Locate the buffer given its zone-2 offset.
         */
        buf_offset &= ~HAMMER_BUFMASK64;
        KKASSERT((buf_offset & HAMMER_ZONE_RAW_BUFFER) ==
                 HAMMER_ZONE_RAW_BUFFER);
        vol_no = HAMMER_VOL_DECODE(buf_offset);
        volume = hammer_get_volume(hmp, vol_no, errorp);
        if (volume == NULL)
                return(NULL);

        /*
         * NOTE: buf_offset and maxbuf_off are both full offset
         * specifications.
         */
        KKASSERT(buf_offset < volume->maxbuf_off);

        /*
         * Locate and lock the buffer structure, creating one if necessary.
         */
again:
        buffer = RB_LOOKUP(hammer_buf_rb_tree, &volume->rb_bufs_root,
                           buf_offset);
        if (buffer == NULL) {
                ++hammer_count_buffers;
                buffer = kmalloc(sizeof(*buffer), M_HAMMER, M_WAITOK|M_ZERO);
                buffer->zone2_offset = buf_offset;
                buffer->volume = volume;

                hammer_io_init(&buffer->io, hmp, iotype);
                buffer->io.offset = volume->ondisk->vol_buf_beg +
                                    (buf_offset & HAMMER_OFF_SHORT_MASK);
                TAILQ_INIT(&buffer->clist);
                hammer_ref(&buffer->io.lock);

                /*
                 * Insert the buffer into the RB tree and handle late
                 * collisions.
                 */
                if (RB_INSERT(hammer_buf_rb_tree, &volume->rb_bufs_root, buffer)) {
                        hammer_unref(&buffer->io.lock);
                        --hammer_count_buffers;
                        kfree(buffer, M_HAMMER);
                        goto again;
                }
                hammer_ref(&volume->io.lock);
        } else {
                hammer_ref(&buffer->io.lock);

                /*
                 * The buffer is no longer loose if it has a ref.
                 */
                if (buffer->io.mod_list == &hmp->lose_list) {
                        TAILQ_REMOVE(buffer->io.mod_list, &buffer->io,
                                     mod_entry);
                        buffer->io.mod_list = NULL;
                }
                if (buffer->io.lock.refs == 1)
                        hammer_io_reinit(&buffer->io, iotype);
                else
                        KKASSERT(buffer->io.type == iotype);
        }

        /*
         * Cache the blockmap translation
         */
        if ((zoneX_offset & HAMMER_ZONE_RAW_BUFFER) != HAMMER_ZONE_RAW_BUFFER)
                buffer->zoneX_offset = zoneX_offset;

        /*
         * Deal with on-disk info
         */
        if (buffer->ondisk == NULL || buffer->io.loading) {
                *errorp = hammer_load_buffer(buffer, isnew);
                if (*errorp) {
                        hammer_rel_buffer(buffer, 1);
                        buffer = NULL;
                }
        } else {
                *errorp = 0;
        }
        hammer_rel_volume(volume, 0);
        return(buffer);
}

static int
hammer_load_buffer(hammer_buffer_t buffer, int isnew)
{
        hammer_volume_t volume;
        int error;

        /*
         * Load the buffer's on-disk info
         */
        volume = buffer->volume;
        ++buffer->io.loading;
        hammer_lock_ex(&buffer->io.lock);

        if (buffer->ondisk == NULL) {
                if (isnew) {
                        error = hammer_io_new(volume->devvp, &buffer->io);
                } else {
                        error = hammer_io_read(volume->devvp, &buffer->io);
                }
                if (error == 0)
                        buffer->ondisk = (void *)buffer->io.bp->b_data;
        } else if (isnew) {
                error = hammer_io_new(volume->devvp, &buffer->io);
        } else {
                error = 0;
        }
        --buffer->io.loading;
        hammer_unlock(&buffer->io.lock);
        return (error);
}

/*
 * NOTE: Called from RB_SCAN, must return >= 0 for scan to continue.
 */
int
hammer_unload_buffer(hammer_buffer_t buffer, void *data __unused)
{
        hammer_ref(&buffer->io.lock);
        hammer_flush_buffer_nodes(buffer);
        KKASSERT(buffer->io.lock.refs == 1);
        hammer_rel_buffer(buffer, 2);
        return(0);
}

/*
 * Reference a buffer that is either already referenced or via a specially
 * handled pointer (aka cursor->buffer).
 */
int
hammer_ref_buffer(hammer_buffer_t buffer)
{
        int error;

        hammer_ref(&buffer->io.lock);

        /*
         * No longer loose
         */
        if (buffer->io.mod_list == &buffer->io.hmp->lose_list) {
                TAILQ_REMOVE(buffer->io.mod_list, &buffer->io, mod_entry);
                buffer->io.mod_list = NULL;
        }

        if (buffer->ondisk == NULL || buffer->io.loading) {
                error = hammer_load_buffer(buffer, 0);
                if (error) {
                        hammer_rel_buffer(buffer, 1);
                        /*
                         * NOTE: buffer pointer can become stale after
                         * the above release.
                         */
                }
        } else {
                error = 0;
        }
        return(error);
}

/*
 * Release a buffer.  We have to deal with several places where
 * another thread can ref the buffer.
 *
 * Only destroy the structure itself if the related buffer cache buffer
 * was disassociated from it.  This ties the management of the structure
 * to the buffer cache subsystem.  buffer->ondisk determines whether the
 * embedded io is referenced or not.
 */
void
hammer_rel_buffer(hammer_buffer_t buffer, int flush)
{
        hammer_volume_t volume;
        int freeme = 0;

        crit_enter();
        if (buffer->io.lock.refs == 1) {
                ++buffer->io.loading;   /* force interlock check */
                hammer_lock_ex(&buffer->io.lock);
                if (buffer->io.lock.refs == 1) {
                        hammer_io_release(&buffer->io, flush);
                        hammer_flush_buffer_nodes(buffer);
                        KKASSERT(TAILQ_EMPTY(&buffer->clist));

                        if (buffer->io.bp == NULL &&
                            buffer->io.lock.refs == 1) {
                                /*
                                 * Final cleanup
                                 */
                                volume = buffer->volume;
                                RB_REMOVE(hammer_buf_rb_tree,
                                          &volume->rb_bufs_root, buffer);
                                buffer->volume = NULL; /* sanity */
                                hammer_rel_volume(volume, 0);
                                freeme = 1;
                        }
                }
                --buffer->io.loading;
                hammer_unlock(&buffer->io.lock);
        }
        hammer_unref(&buffer->io.lock);
        crit_exit();
        if (freeme) {
                KKASSERT(buffer->io.mod_list == NULL);
                --hammer_count_buffers;
                kfree(buffer, M_HAMMER);
        }
}

/*
 * Remove the zoneX translation cache for a buffer given its zone-2 offset.
 */
void
hammer_uncache_buffer(hammer_mount_t hmp, hammer_off_t buf_offset)
{
        hammer_volume_t volume;
        hammer_buffer_t buffer;
        int vol_no;
        int error;

        buf_offset &= ~HAMMER_BUFMASK64;
        KKASSERT((buf_offset & HAMMER_ZONE_RAW_BUFFER) ==
                 HAMMER_ZONE_RAW_BUFFER);
        vol_no = HAMMER_VOL_DECODE(buf_offset);
        volume = hammer_get_volume(hmp, vol_no, &error);
        KKASSERT(volume != 0);
        KKASSERT(buf_offset < volume->maxbuf_off);

        buffer = RB_LOOKUP(hammer_buf_rb_tree, &volume->rb_bufs_root,
                           buf_offset);
        if (buffer)
                buffer->zoneX_offset = 0;
        hammer_rel_volume(volume, 0);
}

/*
 * Access the filesystem buffer containing the specified hammer offset.
 * buf_offset is a conglomeration of the volume number and vol_buf_beg
 * relative buffer offset.  It must also have bit 55 set to be valid.
 * (see hammer_off_t in hammer_disk.h).
 *
 * Any prior buffer in *bufferp will be released and replaced by the
 * requested buffer.
 */
void *
hammer_bread(hammer_mount_t hmp, hammer_off_t buf_offset, int *errorp, 
             struct hammer_buffer **bufferp)
{
        hammer_buffer_t buffer;
        int32_t xoff = (int32_t)buf_offset & HAMMER_BUFMASK;

        buf_offset &= ~HAMMER_BUFMASK64;
        KKASSERT((buf_offset & HAMMER_OFF_ZONE_MASK) != 0);

        buffer = *bufferp;
        if (buffer == NULL || (buffer->zone2_offset != buf_offset &&
                               buffer->zoneX_offset != buf_offset)) {
                if (buffer)
                        hammer_rel_buffer(buffer, 0);
                buffer = hammer_get_buffer(hmp, buf_offset, 0, errorp);
                *bufferp = buffer;
        } else {
                *errorp = 0;
        }

        /*
         * Return a pointer to the buffer data.
         */
        if (buffer == NULL)
                return(NULL);
        else
                return((char *)buffer->ondisk + xoff);
}

/*
 * Access the filesystem buffer containing the specified hammer offset.
 * No disk read operation occurs.  The result buffer may contain garbage.
 *
 * Any prior buffer in *bufferp will be released and replaced by the
 * requested buffer.
 *
 * This function marks the buffer dirty but does not increment its
 * modify_refs count.
 */
void *
hammer_bnew(hammer_mount_t hmp, hammer_off_t buf_offset, int *errorp, 
             struct hammer_buffer **bufferp)
{
        hammer_buffer_t buffer;
        int32_t xoff = (int32_t)buf_offset & HAMMER_BUFMASK;

        buf_offset &= ~HAMMER_BUFMASK64;

        buffer = *bufferp;
        if (buffer == NULL || (buffer->zone2_offset != buf_offset &&
                               buffer->zoneX_offset != buf_offset)) {
                if (buffer)
                        hammer_rel_buffer(buffer, 0);
                buffer = hammer_get_buffer(hmp, buf_offset, 1, errorp);
                *bufferp = buffer;
        } else {
                *errorp = 0;
        }

        /*
         * Return a pointer to the buffer data.
         */
        if (buffer == NULL)
                return(NULL);
        else
                return((char *)buffer->ondisk + xoff);
}

/************************************************************************
 *                              NODES                                   *
 ************************************************************************
 *
 * Manage B-Tree nodes.  B-Tree nodes represent the primary indexing
 * method used by the HAMMER filesystem.
 *
 * Unlike other HAMMER structures, a hammer_node can be PASSIVELY
 * associated with its buffer, and will only referenced the buffer while
 * the node itself is referenced.
 *
 * A hammer_node can also be passively associated with other HAMMER
 * structures, such as inodes, while retaining 0 references.  These
 * associations can be cleared backwards using a pointer-to-pointer in
 * the hammer_node.
 *
 * This allows the HAMMER implementation to cache hammer_nodes long-term
 * and short-cut a great deal of the infrastructure's complexity.  In
 * most cases a cached node can be reacquired without having to dip into
 * either the buffer or cluster management code.
 *
 * The caller must pass a referenced cluster on call and will retain
 * ownership of the reference on return.  The node will acquire its own
 * additional references, if necessary.
 */
hammer_node_t
hammer_get_node(hammer_mount_t hmp, hammer_off_t node_offset,
                int isnew, int *errorp)
{
        hammer_node_t node;

        KKASSERT((node_offset & HAMMER_OFF_ZONE_MASK) == HAMMER_ZONE_BTREE);

        /*
         * Locate the structure, allocating one if necessary.
         */
again:
        node = RB_LOOKUP(hammer_nod_rb_tree, &hmp->rb_nods_root, node_offset);
        if (node == NULL) {
                ++hammer_count_nodes;
                node = kmalloc(sizeof(*node), M_HAMMER, M_WAITOK|M_ZERO);
                node->node_offset = node_offset;
                node->hmp = hmp;
                if (RB_INSERT(hammer_nod_rb_tree, &hmp->rb_nods_root, node)) {
                        --hammer_count_nodes;
                        kfree(node, M_HAMMER);
                        goto again;
                }
        }
        hammer_ref(&node->lock);
        if (node->ondisk)
                *errorp = 0;
        else
                *errorp = hammer_load_node(node, isnew);
        if (*errorp) {
                hammer_rel_node(node);
                node = NULL;
        }
        return(node);
}

/*
 * Reference an already-referenced node.
 */
void
hammer_ref_node(hammer_node_t node)
{
        KKASSERT(node->lock.refs > 0 && node->ondisk != NULL);
        hammer_ref(&node->lock);
}

/*
 * Load a node's on-disk data reference.
 */
static int
hammer_load_node(hammer_node_t node, int isnew)
{
        hammer_buffer_t buffer;
        int error;

        error = 0;
        ++node->loading;
        hammer_lock_ex(&node->lock);
        if (node->ondisk == NULL) {
                /*
                 * This is a little confusing but the jist is that
                 * node->buffer determines whether the node is on
                 * the buffer's clist and node->ondisk determines
                 * whether the buffer is referenced.
                 *
                 * We could be racing a buffer release, in which case
                 * node->buffer may become NULL while we are blocked
                 * referencing the buffer.
                 */
                if ((buffer = node->buffer) != NULL) {
                        error = hammer_ref_buffer(buffer);
                        if (error == 0 && node->buffer == NULL) {
                                TAILQ_INSERT_TAIL(&buffer->clist,
                                                  node, entry);
                                node->buffer = buffer;
                        }
                } else {
                        buffer = hammer_get_buffer(node->hmp,
                                                   node->node_offset, 0,
                                                   &error);
                        if (buffer) {
                                KKASSERT(error == 0);
                                TAILQ_INSERT_TAIL(&buffer->clist,
                                                  node, entry);
                                node->buffer = buffer;
                        }
                }
                if (error == 0) {
                        node->ondisk = (void *)((char *)buffer->ondisk +
                               (node->node_offset & HAMMER_BUFMASK));
                        if (isnew == 0 &&
                            hammer_crc_test_btree(node->ondisk) == 0) {
                                Debugger("CRC FAILED: B-TREE NODE");
                        }
                }
        }
        --node->loading;
        hammer_unlock(&node->lock);
        return (error);
}

/*
 * Safely reference a node, interlock against flushes via the IO subsystem.
 */
hammer_node_t
hammer_ref_node_safe(struct hammer_mount *hmp, struct hammer_node **cache,
                     int *errorp)
{
        hammer_node_t node;

        node = *cache;
        if (node != NULL) {
                hammer_ref(&node->lock);
                if (node->ondisk)
                        *errorp = 0;
                else
                        *errorp = hammer_load_node(node, 0);
                if (*errorp) {
                        hammer_rel_node(node);
                        node = NULL;
                }
        } else {
                *errorp = ENOENT;
        }
        return(node);
}

/*
 * Release a hammer_node.  On the last release the node dereferences
 * its underlying buffer and may or may not be destroyed.
 */
void
hammer_rel_node(hammer_node_t node)
{
        hammer_buffer_t buffer;

        /*
         * If this isn't the last ref just decrement the ref count and
         * return.
         */
        if (node->lock.refs > 1) {
                hammer_unref(&node->lock);
                return;
        }

        /*
         * If there is no ondisk info or no buffer the node failed to load,
         * remove the last reference and destroy the node.
         */
        if (node->ondisk == NULL) {
                hammer_unref(&node->lock);
                hammer_flush_node(node);
                /* node is stale now */
                return;
        }

        /*
         * Do final cleanups and then either destroy the node and leave it
         * passively cached.  The buffer reference is removed regardless.
         */
        buffer = node->buffer;
        node->ondisk = NULL;

        if ((node->flags & HAMMER_NODE_FLUSH) == 0) {
                hammer_unref(&node->lock);
                hammer_rel_buffer(buffer, 0);
                return;
        }

        /*
         * Destroy the node.
         */
        hammer_unref(&node->lock);
        hammer_flush_node(node);
        /* node is stale */
        hammer_rel_buffer(buffer, 0);
}

/*
 * Free space on-media associated with a B-Tree node.
 */
void
hammer_delete_node(hammer_transaction_t trans, hammer_node_t node)
{
        KKASSERT((node->flags & HAMMER_NODE_DELETED) == 0);
        node->flags |= HAMMER_NODE_DELETED;
        hammer_blockmap_free(trans, node->node_offset, sizeof(*node->ondisk));
}

/*
 * Passively cache a referenced hammer_node in *cache.  The caller may
 * release the node on return.
 */
void
hammer_cache_node(hammer_node_t node, struct hammer_node **cache)
{
        hammer_node_t old;

        /*
         * If the node is being deleted, don't cache it!
         */
        if (node->flags & HAMMER_NODE_DELETED)
                return;

        /*
         * Cache the node.  If we previously cached a different node we
         * have to give HAMMER a chance to destroy it.
         */
again:
        if (node->cache1 != cache) {
                if (node->cache2 != cache) {
                        if ((old = *cache) != NULL) {
                                KKASSERT(node->lock.refs != 0);
                                hammer_uncache_node(cache);
                                goto again;
                        }
                        if (node->cache2)
                                *node->cache2 = NULL;
                        node->cache2 = node->cache1;
                        node->cache1 = cache;
                        *cache = node;
                } else {
                        struct hammer_node **tmp;
                        tmp = node->cache1;
                        node->cache1 = node->cache2;
                        node->cache2 = tmp;
                }
        }
}

void
hammer_uncache_node(struct hammer_node **cache)
{
        hammer_node_t node;

        if ((node = *cache) != NULL) {
                *cache = NULL;
                if (node->cache1 == cache) {
                        node->cache1 = node->cache2;
                        node->cache2 = NULL;
                } else if (node->cache2 == cache) {
                        node->cache2 = NULL;
                } else {
                        panic("hammer_uncache_node: missing cache linkage");
                }
                if (node->cache1 == NULL && node->cache2 == NULL)
                        hammer_flush_node(node);
        }
}

/*
 * Remove a node's cache references and destroy the node if it has no
 * other references or backing store.
 */
void
hammer_flush_node(hammer_node_t node)
{
        hammer_buffer_t buffer;

        if (node->cache1)
                *node->cache1 = NULL;
        if (node->cache2)
                *node->cache2 = NULL;
        if (node->lock.refs == 0 && node->ondisk == NULL) {
                RB_REMOVE(hammer_nod_rb_tree, &node->hmp->rb_nods_root, node);
                if ((buffer = node->buffer) != NULL) {
                        node->buffer = NULL;
                        TAILQ_REMOVE(&buffer->clist, node, entry);
                        /* buffer is unreferenced because ondisk is NULL */
                }
                --hammer_count_nodes;
                kfree(node, M_HAMMER);
        }
}

/*
 * Flush passively cached B-Tree nodes associated with this buffer.
 * This is only called when the buffer is about to be destroyed, so
 * none of the nodes should have any references.  The buffer is locked.
 *
 * We may be interlocked with the buffer.
 */
void
hammer_flush_buffer_nodes(hammer_buffer_t buffer)
{
        hammer_node_t node;

        while ((node = TAILQ_FIRST(&buffer->clist)) != NULL) {
                KKASSERT(node->ondisk == NULL);

                if (node->lock.refs == 0) {
                        hammer_ref(&node->lock);
                        node->flags |= HAMMER_NODE_FLUSH;
                        hammer_rel_node(node);
                } else {
                        KKASSERT(node->loading != 0);
                        KKASSERT(node->buffer != NULL);
                        buffer = node->buffer;
                        node->buffer = NULL;
                        TAILQ_REMOVE(&buffer->clist, node, entry);
                        /* buffer is unreferenced because ondisk is NULL */
                }
        }
}


/************************************************************************
 *                              ALLOCATORS                              *
 ************************************************************************/

/*
 * Allocate a B-Tree node.
 */
hammer_node_t
hammer_alloc_btree(hammer_transaction_t trans, int *errorp)
{
        hammer_buffer_t buffer = NULL;
        hammer_node_t node = NULL;
        hammer_off_t node_offset;

        node_offset = hammer_blockmap_alloc(trans, HAMMER_ZONE_BTREE_INDEX,
                                            sizeof(struct hammer_node_ondisk),
                                            errorp);
        if (*errorp == 0) {
                node = hammer_get_node(trans->hmp, node_offset, 1, errorp);
                hammer_modify_node_noundo(trans, node);
                bzero(node->ondisk, sizeof(*node->ondisk));
                hammer_modify_node_done(node);
        }
        if (buffer)
                hammer_rel_buffer(buffer, 0);
        return(node);
}

#if 0

/*
 * The returned buffers are already appropriately marked as being modified.
 * If the caller marks them again unnecessary undo records may be generated.
 *
 * In-band data is indicated by data_bufferp == NULL.  Pass a data_len of 0
 * for zero-fill (caller modifies data_len afterwords).
 *
 * If the caller is responsible for calling hammer_modify_*() prior to making
 * any additional modifications to either the returned record buffer or the
 * returned data buffer.
 */
void *
hammer_alloc_record(hammer_transaction_t trans, 
                    hammer_off_t *rec_offp, u_int16_t rec_type, 
                    struct hammer_buffer **rec_bufferp,
                    int32_t data_len, void **datap,
                    hammer_off_t *data_offp,
                    struct hammer_buffer **data_bufferp, int *errorp)
{
        hammer_record_ondisk_t rec;
        hammer_off_t rec_offset;
        hammer_off_t data_offset;
        int32_t reclen;

        if (datap)
                *datap = NULL;

        /*
         * Allocate the record
         */
        rec_offset = hammer_blockmap_alloc(trans, HAMMER_ZONE_RECORD_INDEX,
                                           HAMMER_RECORD_SIZE, errorp);
        if (*errorp)
                return(NULL);
        if (data_offp)
                *data_offp = 0;

        /*
         * Allocate data
         */
        if (data_len) {
                if (data_bufferp == NULL) {
                        switch(rec_type) {
                        case HAMMER_RECTYPE_DATA:
                                reclen = offsetof(struct hammer_data_record,
                                                  data[0]);
                                break;
                        case HAMMER_RECTYPE_DIRENTRY:
                                reclen = offsetof(struct hammer_entry_record,
                                                  name[0]);
                                break;
                        default:
                                panic("hammer_alloc_record: illegal "
                                      "in-band data");
                                /* NOT REACHED */
                                reclen = 0;
                                break;
                        }
                        KKASSERT(reclen + data_len <= HAMMER_RECORD_SIZE);
                        data_offset = rec_offset + reclen;
                } else if (data_len < HAMMER_BUFSIZE) {
                        data_offset = hammer_blockmap_alloc(trans,
                                                HAMMER_ZONE_SMALL_DATA_INDEX,
                                                data_len, errorp);
                        *data_offp = data_offset;
                } else {
                        data_offset = hammer_blockmap_alloc(trans,
                                                HAMMER_ZONE_LARGE_DATA_INDEX,
                                                data_len, errorp);
                        *data_offp = data_offset;
                }
        } else {
                data_offset = 0;
        }
        if (*errorp) {
                hammer_blockmap_free(trans, rec_offset, HAMMER_RECORD_SIZE);
                return(NULL);
        }

        /*
         * Basic return values.
         *
         * Note that because this is a 'new' buffer, there is no need to
         * generate UNDO records for it.
         */
        *rec_offp = rec_offset;
        rec = hammer_bread(trans->hmp, rec_offset, errorp, rec_bufferp);
        hammer_modify_buffer(trans, *rec_bufferp, NULL, 0);
        bzero(rec, sizeof(*rec));
        KKASSERT(*errorp == 0);
        rec->base.data_off = data_offset;
        rec->base.data_len = data_len;
        hammer_modify_buffer_done(*rec_bufferp);

        if (data_bufferp) {
                if (data_len) {
                        *datap = hammer_bread(trans->hmp, data_offset, errorp,
                                              data_bufferp);
                        KKASSERT(*errorp == 0);
                } else {
                        *datap = NULL;
                }
        } else if (data_len) {
                KKASSERT(data_offset + data_len - rec_offset <=
                         HAMMER_RECORD_SIZE); 
                if (datap) {
                        *datap = (void *)((char *)rec +
                                          (int32_t)(data_offset - rec_offset));
                }
        } else {
                KKASSERT(datap == NULL);
        }
        KKASSERT(*errorp == 0);
        return(rec);
}

#endif

/*
 * Allocate data.  If the address of a data buffer is supplied then
 * any prior non-NULL *data_bufferp will be released and *data_bufferp
 * will be set to the related buffer.  The caller must release it when
 * finally done.  The initial *data_bufferp should be set to NULL by
 * the caller.
 *
 * The caller is responsible for making hammer_modify*() calls on the
 * *data_bufferp.
 */
void *
hammer_alloc_data(hammer_transaction_t trans, int32_t data_len, 
                  hammer_off_t *data_offsetp,
                  struct hammer_buffer **data_bufferp, int *errorp)
{
        void *data;

        /*
         * Allocate data
         */
        if (data_len) {
                if (data_len < HAMMER_BUFSIZE) {
                        *data_offsetp = hammer_blockmap_alloc(trans,
                                                HAMMER_ZONE_SMALL_DATA_INDEX,
                                                data_len, errorp);
                } else {
                        *data_offsetp = hammer_blockmap_alloc(trans,
                                                HAMMER_ZONE_LARGE_DATA_INDEX,
                                                data_len, errorp);
                }
        } else {
                *data_offsetp = 0;
        }
        if (*errorp == 0 && data_bufferp) {
                if (data_len) {
                        data = hammer_bread(trans->hmp, *data_offsetp, errorp,
                                            data_bufferp);
                        KKASSERT(*errorp == 0);
                } else {
                        data = NULL;
                }
        } else {
                data = NULL;
        }
        KKASSERT(*errorp == 0);
        return(data);
}

/*
 * Sync dirty buffers to the media and clean-up any loose ends.
 */
static int hammer_sync_scan1(struct mount *mp, struct vnode *vp, void *data);
static int hammer_sync_scan2(struct mount *mp, struct vnode *vp, void *data);

int
hammer_queue_inodes_flusher(hammer_mount_t hmp, int waitfor)
{
        struct hammer_sync_info info;

        info.error = 0;
        info.waitfor = waitfor;
        if (waitfor == MNT_WAIT) {
                vmntvnodescan(hmp->mp, VMSC_GETVP,
                              hammer_sync_scan1, hammer_sync_scan2, &info);
        } else {
                vmntvnodescan(hmp->mp, VMSC_GETVP|VMSC_NOWAIT,
                              hammer_sync_scan1, hammer_sync_scan2, &info);
        }
        return(info.error);
}

int
hammer_sync_hmp(hammer_mount_t hmp, int waitfor)
{
        struct hammer_sync_info info;

        info.error = 0;
        info.waitfor = waitfor;

        vmntvnodescan(hmp->mp, VMSC_GETVP|VMSC_NOWAIT,
                      hammer_sync_scan1, hammer_sync_scan2, &info);
        if (waitfor == MNT_WAIT)
                hammer_flusher_sync(hmp);
        else
                hammer_flusher_async(hmp);

        return(info.error);
}

static int
hammer_sync_scan1(struct mount *mp, struct vnode *vp, void *data)
{
        struct hammer_inode *ip;

        ip = VTOI(vp);
        if (vp->v_type == VNON || ip == NULL ||
            ((ip->flags & HAMMER_INODE_MODMASK) == 0 &&
             RB_EMPTY(&vp->v_rbdirty_tree))) {
                return(-1);
        }
        return(0);
}

static int
hammer_sync_scan2(struct mount *mp, struct vnode *vp, void *data)
{
        struct hammer_sync_info *info = data;
        struct hammer_inode *ip;
        int error;

        ip = VTOI(vp);
        if (vp->v_type == VNON || vp->v_type == VBAD ||
            ((ip->flags & HAMMER_INODE_MODMASK) == 0 &&
             RB_EMPTY(&vp->v_rbdirty_tree))) {
                return(0);
        }
        error = VOP_FSYNC(vp, info->waitfor);
        if (error)
                info->error = error;
        return(0);
}