/*      $NetBSD: buf.c,v 1.28 2023/03/14 10:36:06 kre Exp $     */

/*
* Copyright (c) 2001 Wasabi Systems, Inc.
* All rights reserved.
*
* Written by Luke Mewburn for Wasabi Systems, Inc.
*
* 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. All advertising materials mentioning features or use of this software
*    must display the following acknowledgement:
*      This product includes software developed for the NetBSD Project by
*      Wasabi Systems, Inc.
* 4. The name of Wasabi Systems, Inc. may not be used to endorse
*    or promote products derived from this software without specific prior
*    written permission.
*
* THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``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 WASABI SYSTEMS, INC
* 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.
*/

#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>
#if defined(__RCSID) && !defined(__lint)
__RCSID("$NetBSD: buf.c,v 1.28 2023/03/14 10:36:06 kre Exp $");
#endif  /* !__lint */

#include <sys/param.h>
#include <sys/time.h>

#include <assert.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <util.h>

#include "makefs.h"
#include "buf.h"

TAILQ_HEAD(buftailhead,buf) buftail;

int
bread(struct vnode *vp, daddr_t blkno, int size, int u2 __unused,
       struct buf **bpp)
{
       off_t   offset;
       ssize_t rv;
       fsinfo_t *fs = vp->fs;
       int saved_errno;

       assert (bpp != NULL);

       if (debug & DEBUG_BUF_BREAD)
               printf("%s: blkno %jd size %d\n", __func__,
                   (intmax_t)blkno, size);
       *bpp = getblk(vp, blkno, size, 0, 0);
       offset = (*bpp)->b_blkno * fs->sectorsize + fs->offset;
       if (debug & DEBUG_BUF_BREAD)
               printf("%s: blkno %jd offset %jd bcount %ld\n", __func__,
                   (intmax_t)(*bpp)->b_blkno, (intmax_t) offset,
                   (*bpp)->b_bcount);
       if (lseek((*bpp)->b_fs->fd, offset, SEEK_SET) == -1) {
               saved_errno = errno;
               warn("%s: lseek %jd (%jd)", __func__,
                   (intmax_t)(*bpp)->b_blkno, (intmax_t)offset);
               goto out;
       }
       rv = read((*bpp)->b_fs->fd, (*bpp)->b_data, (size_t)(*bpp)->b_bcount);
       saved_errno = errno;
       if (debug & DEBUG_BUF_BREAD)
               printf("%s: read %ld (%jd) returned %zd\n", __func__,
                   (*bpp)->b_bcount, (intmax_t)offset, rv);
       if (rv == -1) {                         /* read error */
               warn("%s: read %ld (%jd) returned %zd", __func__,
                   (*bpp)->b_bcount, (intmax_t)offset, rv);
               goto out;
       }
       if (rv != (*bpp)->b_bcount) {   /* short read */
               saved_errno = ENOSPC;
               warn("%s: read %ld (%jd) returned %zd", __func__,
                   (*bpp)->b_bcount, (intmax_t)offset, rv);
               goto out;
       }
       return 0;
out:
       brelse(*bpp, 0);
       *bpp = NULL;
#if 1
       __USE(saved_errno);
       exit(EXIT_FAILURE);
#else
       return saved_errno;
#endif
}

void
brelse(struct buf *bp, int u1 __unused)
{

       assert (bp != NULL);
       assert (bp->b_data != NULL);

       if (bp->b_lblkno < 0) {
               /*
                * XXX  don't remove any buffers with negative logical block
                *      numbers (lblkno), so that we retain the mapping
                *      of negative lblkno -> real blkno that ffs_balloc()
                *      sets up.
                *
                *      if we instead released these buffers, and implemented
                *      ufs_strategy() (and ufs_bmaparray()) and called those
                *      from bread() and bwrite() to convert the lblkno to
                *      a real blkno, we'd add a lot more code & complexity
                *      and reading off disk, for little gain, because this
                *      simple hack works for our purpose.
                */
               bp->b_bcount = 0;
               return;
       }

       TAILQ_REMOVE(&buftail, bp, b_tailq);
       free(bp->b_data);
       free(bp);
}

int
bwrite(struct buf *bp)
{
       off_t   offset;
       ssize_t rv;
       size_t  bytes;
       int     e;
       fsinfo_t *fs = bp->b_fs;

       assert (bp != NULL);
       offset = bp->b_blkno * fs->sectorsize + fs->offset;
       bytes  = (size_t)bp->b_bcount;
       if (debug & DEBUG_BUF_BWRITE)
               printf("%s: blkno %jd offset %jd bcount %zu\n", __func__,
                   (intmax_t)bp->b_blkno, (intmax_t) offset, bytes);
       if (lseek(bp->b_fs->fd, offset, SEEK_SET) == -1)
               return errno;
       rv = write(bp->b_fs->fd, bp->b_data, bytes);
       e = errno;
       if (debug & DEBUG_BUF_BWRITE)
               printf("%s: write %ld (offset %jd) returned %jd\n", __func__,
                   bp->b_bcount, (intmax_t)offset, (intmax_t)rv);
       brelse(bp, 0);
       if (rv == (ssize_t)bytes)
               return 0;
       if (rv == -1)           /* write error */
               return e;
       return EAGAIN;
}

void
bcleanup(void)
{
       struct buf *bp;

       /*
        * XXX  this really shouldn't be necessary, but i'm curious to
        *      know why there's still some buffers lying around that
        *      aren't brelse()d
        */

       if (TAILQ_EMPTY(&buftail))
               return;

       printf("%s: unflushed buffers:\n", __func__);
       TAILQ_FOREACH(bp, &buftail, b_tailq) {
               printf("\tlblkno %10jd blkno %10jd count %6ld  bufsize %6ld\n",
                   (intmax_t)bp->b_lblkno, (intmax_t)bp->b_blkno,
                   bp->b_bcount, bp->b_bufsize);
       }
       printf("%s: done\n", __func__);
}

struct buf *
getblk(struct vnode *vp, daddr_t blkno, int size, int u1 __unused,
   int u2 __unused)
{
       static int buftailinitted;
       struct buf *bp;
       void *n;

       if (debug & DEBUG_BUF_GETBLK)
               printf("%s: blkno %jd size %d\n", __func__,
                   (intmax_t)blkno, size);

       bp = NULL;
       if (!buftailinitted) {
               if (debug & DEBUG_BUF_GETBLK)
                       printf("%s: initialising tailq\n", __func__);
               TAILQ_INIT(&buftail);
               buftailinitted = 1;
       } else {
               TAILQ_FOREACH(bp, &buftail, b_tailq) {
                       if (bp->b_lblkno != blkno)
                               continue;
                       break;
               }
       }
       if (bp == NULL) {
               bp = ecalloc(1, sizeof(*bp));
               bp->b_bufsize = 0;
               bp->b_blkno = bp->b_lblkno = blkno;
               bp->b_fs = vp->fs;
               bp->b_data = NULL;
               TAILQ_INSERT_HEAD(&buftail, bp, b_tailq);
       }
       bp->b_bcount = size;
       if (bp->b_data == NULL || bp->b_bcount > bp->b_bufsize) {
               n = erealloc(bp->b_data, (size_t)size);
               memset(n, 0, (size_t)size);
               bp->b_data = n;
               bp->b_bufsize = size;
       }

       return bp;
}