/*
 * linux/fs/befs/datastream.c
 *
 * Copyright (C) 2001 Will Dyson <will_dyson@pobox.com>
 *
 * Based on portions of file.c by Makoto Kato <m_kato@ga2.so-net.ne.jp>
 *
 * Many thanks to Dominic Giampaolo, author of "Practical File System
 * Design with the Be File System", for such a helpful book.
 *
 */

#include <linux/kernel.h>
#include <linux/buffer_head.h>
#include <linux/string.h>

#include "befs.h"
#include "datastream.h"
#include "io.h"

const befs_inode_addr BAD_IADDR = { 0, 0, 0 };

static int befs_find_brun_direct(struct super_block *sb,
                                 const befs_data_stream *data,
                                 befs_blocknr_t blockno, befs_block_run *run);

static int befs_find_brun_indirect(struct super_block *sb,
                                   const befs_data_stream *data,
                                   befs_blocknr_t blockno,
                                   befs_block_run *run);

static int befs_find_brun_dblindirect(struct super_block *sb,
                                      const befs_data_stream *data,
                                      befs_blocknr_t blockno,
                                      befs_block_run *run);

/**
 * befs_read_datastream - get buffer_head containing data, starting from pos.
 * @sb: Filesystem superblock
 * @ds: datastream to find data with
 * @pos: start of data
 * @off: offset of data in buffer_head->b_data
 *
 * Returns pointer to buffer_head containing data starting with offset @off,
 * if you don't need to know offset just set @off = NULL.
 */
struct buffer_head *
befs_read_datastream(struct super_block *sb, const befs_data_stream *ds,
                     befs_off_t pos, uint *off)
{
        struct buffer_head *bh;
        befs_block_run run;
        befs_blocknr_t block;   /* block coresponding to pos */

        befs_debug(sb, "---> %s %llu", __func__, pos);
        block = pos >> BEFS_SB(sb)->block_shift;
        if (off)
                *off = pos - (block << BEFS_SB(sb)->block_shift);

        if (befs_fblock2brun(sb, ds, block, &run) != BEFS_OK) {
                befs_error(sb, "BeFS: Error finding disk addr of block %lu",
                           (unsigned long)block);
                befs_debug(sb, "<--- %s ERROR", __func__);
                return NULL;
        }
        bh = befs_bread_iaddr(sb, run);
        if (!bh) {
                befs_error(sb, "BeFS: Error reading block %lu from datastream",
                           (unsigned long)block);
                return NULL;
        }

        befs_debug(sb, "<--- %s read data, starting at %llu", __func__, pos);

        return bh;
}

/**
 * befs_fblock2brun - give back block run for fblock
 * @sb: the superblock
 * @data: datastream to read from
 * @fblock: the blocknumber with the file position to find
 * @run: The found run is passed back through this pointer
 *
 * Takes a file position and gives back a brun who's starting block
 * is block number fblock of the file.
 * 
 * Returns BEFS_OK or BEFS_ERR.
 * 
 * Calls specialized functions for each of the three possible
 * datastream regions.
 *
 * 2001-11-15 Will Dyson
 */
int
befs_fblock2brun(struct super_block *sb, const befs_data_stream *data,
                 befs_blocknr_t fblock, befs_block_run *run)
{
        int err;
        befs_off_t pos = fblock << BEFS_SB(sb)->block_shift;

        if (pos < data->max_direct_range) {
                err = befs_find_brun_direct(sb, data, fblock, run);

        } else if (pos < data->max_indirect_range) {
                err = befs_find_brun_indirect(sb, data, fblock, run);

        } else if (pos < data->max_double_indirect_range) {
                err = befs_find_brun_dblindirect(sb, data, fblock, run);

        } else {
                befs_error(sb,
                           "befs_fblock2brun() was asked to find block %lu, "
                           "which is not mapped by the datastream\n",
                           (unsigned long)fblock);
                err = BEFS_ERR;
        }
        return err;
}

/**
 * befs_read_lsmylink - read long symlink from datastream.
 * @sb: Filesystem superblock 
 * @ds: Datastream to read from
 * @buff: Buffer in which to place long symlink data
 * @len: Length of the long symlink in bytes
 *
 * Returns the number of bytes read
 */
size_t
befs_read_lsymlink(struct super_block *sb, const befs_data_stream *ds,
                   void *buff, befs_off_t len)
{
        befs_off_t bytes_read = 0;      /* bytes readed */
        u16 plen;
        struct buffer_head *bh;

        befs_debug(sb, "---> %s length: %llu", __func__, len);

        while (bytes_read < len) {
                bh = befs_read_datastream(sb, ds, bytes_read, NULL);
                if (!bh) {
                        befs_error(sb, "BeFS: Error reading datastream block "
                                   "starting from %llu", bytes_read);
                        befs_debug(sb, "<--- %s ERROR", __func__);
                        return bytes_read;

                }
                plen = ((bytes_read + BEFS_SB(sb)->block_size) < len) ?
                    BEFS_SB(sb)->block_size : len - bytes_read;
                memcpy(buff + bytes_read, bh->b_data, plen);
                brelse(bh);
                bytes_read += plen;
        }

        befs_debug(sb, "<--- %s read %u bytes", __func__, (unsigned int)
                   bytes_read);
        return bytes_read;
}

/**
 * befs_count_blocks - blocks used by a file
 * @sb: Filesystem superblock
 * @ds: Datastream of the file
 *
 * Counts the number of fs blocks that the file represented by
 * inode occupies on the filesystem, counting both regular file
 * data and filesystem metadata (and eventually attribute data
 * when we support attributes)
*/

befs_blocknr_t
befs_count_blocks(struct super_block *sb, const befs_data_stream *ds)
{
        befs_blocknr_t blocks;
        befs_blocknr_t datablocks;      /* File data blocks */
        befs_blocknr_t metablocks;      /* FS metadata blocks */
        struct befs_sb_info *befs_sb = BEFS_SB(sb);

        befs_debug(sb, "---> %s", __func__);

        datablocks = ds->size >> befs_sb->block_shift;
        if (ds->size & (befs_sb->block_size - 1))
                datablocks += 1;

        metablocks = 1;         /* Start with 1 block for inode */

        /* Size of indirect block */
        if (ds->size > ds->max_direct_range)
                metablocks += ds->indirect.len;

        /*
         * Double indir block, plus all the indirect blocks it maps.
         * In the double-indirect range, all block runs of data are
         * BEFS_DBLINDIR_BRUN_LEN blocks long. Therefore, we know
         * how many data block runs are in the double-indirect region,
         * and from that we know how many indirect blocks it takes to
         * map them. We assume that the indirect blocks are also
         * BEFS_DBLINDIR_BRUN_LEN blocks long.
         */
        if (ds->size > ds->max_indirect_range && ds->max_indirect_range != 0) {
                uint dbl_bytes;
                uint dbl_bruns;
                uint indirblocks;

                dbl_bytes =
                    ds->max_double_indirect_range - ds->max_indirect_range;
                dbl_bruns =
                    dbl_bytes / (befs_sb->block_size * BEFS_DBLINDIR_BRUN_LEN);
                indirblocks = dbl_bruns / befs_iaddrs_per_block(sb);

                metablocks += ds->double_indirect.len;
                metablocks += indirblocks;
        }

        blocks = datablocks + metablocks;
        befs_debug(sb, "<--- %s %u blocks", __func__, (unsigned int)blocks);

        return blocks;
}

/**
 * befs_find_brun_direct - find a direct block run in the datastream
 * @sb: the superblock
 * @data: the datastream
 * @blockno: the blocknumber to find
 * @run: The found run is passed back through this pointer
 *
 * Finds the block run that starts at file block number blockno
 * in the file represented by the datastream data, if that
 * blockno is in the direct region of the datastream.
 *
 * Return value is BEFS_OK if the blockrun is found, BEFS_ERR
 * otherwise.
 *
 * Algorithm:
 * Linear search. Checks each element of array[] to see if it
 * contains the blockno-th filesystem block. This is necessary
 * because the block runs map variable amounts of data. Simply
 * keeps a count of the number of blocks searched so far (sum),
 * incrementing this by the length of each block run as we come
 * across it. Adds sum to *count before returning (this is so
 * you can search multiple arrays that are logicaly one array,
 * as in the indirect region code).
 *
 * When/if blockno is found, if blockno is inside of a block
 * run as stored on disk, we offset the start and length members
 * of the block run, so that blockno is the start and len is
 * still valid (the run ends in the same place).
 */
static int
befs_find_brun_direct(struct super_block *sb, const befs_data_stream *data,
                      befs_blocknr_t blockno, befs_block_run *run)
{
        int i;
        const befs_block_run *array = data->direct;
        befs_blocknr_t sum;

        befs_debug(sb, "---> %s, find %lu", __func__, (unsigned long)blockno);

        for (i = 0, sum = 0; i < BEFS_NUM_DIRECT_BLOCKS;
             sum += array[i].len, i++) {
                if (blockno >= sum && blockno < sum + (array[i].len)) {
                        int offset = blockno - sum;

                        run->allocation_group = array[i].allocation_group;
                        run->start = array[i].start + offset;
                        run->len = array[i].len - offset;

                        befs_debug(sb, "---> %s, "
                                   "found %lu at direct[%d]", __func__,
                                   (unsigned long)blockno, i);
                        return BEFS_OK;
                }
        }

        befs_error(sb, "%s failed to find file block %lu", __func__,
                   (unsigned long)blockno);
        befs_debug(sb, "---> %s ERROR", __func__);
        return BEFS_ERR;
}

/**
 * befs_find_brun_indirect - find a block run in the datastream
 * @sb: the superblock
 * @data: the datastream
 * @blockno: the blocknumber to find
 * @run: The found run is passed back through this pointer
 *
 * Finds the block run that starts at file block number blockno
 * in the file represented by the datastream data, if that
 * blockno is in the indirect region of the datastream.
 *
 * Return value is BEFS_OK if the blockrun is found, BEFS_ERR
 * otherwise.
 *
 * Algorithm:
 * For each block in the indirect run of the datastream, read
 * it in and search through it for search_blk.
 *
 * XXX:
 * Really should check to make sure blockno is inside indirect
 * region.
 */
static int
befs_find_brun_indirect(struct super_block *sb,
                        const befs_data_stream *data,
                        befs_blocknr_t blockno,
                        befs_block_run *run)
{
        int i, j;
        befs_blocknr_t sum = 0;
        befs_blocknr_t indir_start_blk;
        befs_blocknr_t search_blk;
        struct buffer_head *indirblock;
        befs_disk_block_run *array;

        befs_block_run indirect = data->indirect;
        befs_blocknr_t indirblockno = iaddr2blockno(sb, &indirect);
        int arraylen = befs_iaddrs_per_block(sb);

        befs_debug(sb, "---> %s, find %lu", __func__, (unsigned long)blockno);

        indir_start_blk = data->max_direct_range >> BEFS_SB(sb)->block_shift;
        search_blk = blockno - indir_start_blk;

        /* Examine blocks of the indirect run one at a time */
        for (i = 0; i < indirect.len; i++) {
                indirblock = sb_bread(sb, indirblockno + i);
                if (indirblock == NULL) {
                        befs_error(sb, "---> %s failed to read "
                                   "disk block %lu from the indirect brun",
                                   __func__, (unsigned long)indirblockno + i);
                        befs_debug(sb, "<--- %s ERROR", __func__);
                        return BEFS_ERR;
                }

                array = (befs_disk_block_run *) indirblock->b_data;

                for (j = 0; j < arraylen; ++j) {
                        int len = fs16_to_cpu(sb, array[j].len);

                        if (search_blk >= sum && search_blk < sum + len) {
                                int offset = search_blk - sum;
                                run->allocation_group =
                                    fs32_to_cpu(sb, array[j].allocation_group);
                                run->start =
                                    fs16_to_cpu(sb, array[j].start) + offset;
                                run->len =
                                    fs16_to_cpu(sb, array[j].len) - offset;

                                brelse(indirblock);
                                befs_debug(sb,
                                           "<--- %s found file block "
                                           "%lu at indirect[%d]", __func__,
                                           (unsigned long)blockno,
                                           j + (i * arraylen));
                                return BEFS_OK;
                        }
                        sum += len;
                }

                brelse(indirblock);
        }

        /* Only fallthrough is an error */
        befs_error(sb, "BeFS: %s failed to find "
                   "file block %lu", __func__, (unsigned long)blockno);

        befs_debug(sb, "<--- %s ERROR", __func__);
        return BEFS_ERR;
}

/**
 * befs_find_brun_dblindirect - find a block run in the datastream
 * @sb: the superblock
 * @data: the datastream
 * @blockno: the blocknumber to find
 * @run: The found run is passed back through this pointer
 *
 * Finds the block run that starts at file block number blockno
 * in the file represented by the datastream data, if that
 * blockno is in the double-indirect region of the datastream.
 *
 * Return value is BEFS_OK if the blockrun is found, BEFS_ERR
 * otherwise.
 *
 * Algorithm:
 * The block runs in the double-indirect region are different.
 * They are always allocated 4 fs blocks at a time, so each
 * block run maps a constant amount of file data. This means
 * that we can directly calculate how many block runs into the
 * double-indirect region we need to go to get to the one that
 * maps a particular filesystem block.
 *
 * We do this in two stages. First we calculate which of the
 * inode addresses in the double-indirect block will point us
 * to the indirect block that contains the mapping for the data,
 * then we calculate which of the inode addresses in that
 * indirect block maps the data block we are after.
 *
 * Oh, and once we've done that, we actually read in the blocks
 * that contain the inode addresses we calculated above. Even
 * though the double-indirect run may be several blocks long,
 * we can calculate which of those blocks will contain the index
 * we are after and only read that one. We then follow it to
 * the indirect block and perform a similar process to find
 * the actual block run that maps the data block we are interested
 * in.
 *
 * Then we offset the run as in befs_find_brun_array() and we are
 * done.
 */
static int
befs_find_brun_dblindirect(struct super_block *sb,
                           const befs_data_stream *data,
                           befs_blocknr_t blockno,
                           befs_block_run *run)
{
        int dblindir_indx;
        int indir_indx;
        int offset;
        int dbl_which_block;
        int which_block;
        int dbl_block_indx;
        int block_indx;
        off_t dblindir_leftover;
        befs_blocknr_t blockno_at_run_start;
        struct buffer_head *dbl_indir_block;
        struct buffer_head *indir_block;
        befs_block_run indir_run;
        befs_disk_inode_addr *iaddr_array;

        befs_blocknr_t indir_start_blk =
            data->max_indirect_range >> BEFS_SB(sb)->block_shift;

        off_t dbl_indir_off = blockno - indir_start_blk;

        /* number of data blocks mapped by each of the iaddrs in
         * the indirect block pointed to by the double indirect block
         */
        size_t iblklen = BEFS_DBLINDIR_BRUN_LEN;

        /* number of data blocks mapped by each of the iaddrs in
         * the double indirect block
         */
        size_t diblklen = iblklen * befs_iaddrs_per_block(sb)
            * BEFS_DBLINDIR_BRUN_LEN;

        befs_debug(sb, "---> %s find %lu", __func__, (unsigned long)blockno);

        /* First, discover which of the double_indir->indir blocks
         * contains pos. Then figure out how much of pos that
         * accounted for. Then discover which of the iaddrs in
         * the indirect block contains pos.
         */

        dblindir_indx = dbl_indir_off / diblklen;
        dblindir_leftover = dbl_indir_off % diblklen;
        indir_indx = dblindir_leftover / diblklen;

        /* Read double indirect block */
        dbl_which_block = dblindir_indx / befs_iaddrs_per_block(sb);
        if (dbl_which_block > data->double_indirect.len) {
                befs_error(sb, "The double-indirect index calculated by "
                           "%s, %d, is outside the range "
                           "of the double-indirect block", __func__,
                           dblindir_indx);
                return BEFS_ERR;
        }

        dbl_indir_block =
            sb_bread(sb, iaddr2blockno(sb, &data->double_indirect) +
                                        dbl_which_block);
        if (dbl_indir_block == NULL) {
                befs_error(sb, "%s couldn't read the "
                           "double-indirect block at blockno %lu", __func__,
                           (unsigned long)
                           iaddr2blockno(sb, &data->double_indirect) +
                           dbl_which_block);
                return BEFS_ERR;
        }

        dbl_block_indx =
            dblindir_indx - (dbl_which_block * befs_iaddrs_per_block(sb));
        iaddr_array = (befs_disk_inode_addr *) dbl_indir_block->b_data;
        indir_run = fsrun_to_cpu(sb, iaddr_array[dbl_block_indx]);
        brelse(dbl_indir_block);

        /* Read indirect block */
        which_block = indir_indx / befs_iaddrs_per_block(sb);
        if (which_block > indir_run.len) {
                befs_error(sb, "The indirect index calculated by "
                           "%s, %d, is outside the range "
                           "of the indirect block", __func__, indir_indx);
                return BEFS_ERR;
        }

        indir_block =
            sb_bread(sb, iaddr2blockno(sb, &indir_run) + which_block);
        if (indir_block == NULL) {
                befs_error(sb, "%s couldn't read the indirect block "
                           "at blockno %lu", __func__, (unsigned long)
                           iaddr2blockno(sb, &indir_run) + which_block);
                return BEFS_ERR;
        }

        block_indx = indir_indx - (which_block * befs_iaddrs_per_block(sb));
        iaddr_array = (befs_disk_inode_addr *) indir_block->b_data;
        *run = fsrun_to_cpu(sb, iaddr_array[block_indx]);
        brelse(indir_block);

        blockno_at_run_start = indir_start_blk;
        blockno_at_run_start += diblklen * dblindir_indx;
        blockno_at_run_start += iblklen * indir_indx;
        offset = blockno - blockno_at_run_start;

        run->start += offset;
        run->len -= offset;

        befs_debug(sb, "Found file block %lu in double_indirect[%d][%d],"
                   " double_indirect_leftover = %lu", (unsigned long)
                   blockno, dblindir_indx, indir_indx, dblindir_leftover);

        return BEFS_OK;
}