// SPDX-License-Identifier: GPL-2.0-only
/*
 *  linux/fs/adfs/map.c
 *
 *  Copyright (C) 1997-2002 Russell King
 */
#include <linux/slab.h>
#include <linux/statfs.h>
#include <linux/unaligned.h>
#include "adfs.h"

/*
 * The ADFS map is basically a set of sectors.  Each sector is called a
 * zone which contains a bitstream made up of variable sized fragments.
 * Each bit refers to a set of bytes in the filesystem, defined by
 * log2bpmb.  This may be larger or smaller than the sector size, but
 * the overall size it describes will always be a round number of
 * sectors.  A fragment id is always idlen bits long.
 *
 *  < idlen > <       n        > <1>
 * +---------+-------//---------+---+
 * | frag id |  0000....000000  | 1 |
 * +---------+-------//---------+---+
 *
 * The physical disk space used by a fragment is taken from the start of
 * the fragment id up to and including the '1' bit - ie, idlen + n + 1
 * bits.
 *
 * A fragment id can be repeated multiple times in the whole map for
 * large or fragmented files.  The first map zone a fragment starts in
 * is given by fragment id / ids_per_zone - this allows objects to start
 * from any zone on the disk.
 *
 * Free space is described by a linked list of fragments.  Each free
 * fragment describes free space in the same way as the other fragments,
 * however, the frag id specifies an offset (in map bits) from the end
 * of this fragment to the start of the next free fragment.
 *
 * Objects stored on the disk are allocated object ids (we use these as
 * our inode numbers.)  Object ids contain a fragment id and an optional
 * offset.  This allows a directory fragment to contain small files
 * associated with that directory.
 */

/*
 * For the future...
 */
static DEFINE_RWLOCK(adfs_map_lock);

/*
 * This is fun.  We need to load up to 19 bits from the map at an
 * arbitrary bit alignment.  (We're limited to 19 bits by F+ version 2).
 */
#define GET_FRAG_ID(_map,_start,_idmask)                                \
        ({                                                              \
                unsigned char *_m = _map + (_start >> 3);               \
                u32 _frag = get_unaligned_le32(_m);                     \
                _frag >>= (_start & 7);                                 \
                _frag & _idmask;                                        \
        })

/*
 * return the map bit offset of the fragment frag_id in the zone dm.
 * Note that the loop is optimised for best asm code - look at the
 * output of:
 *  gcc -D__KERNEL__ -O2 -I../../include -o - -S map.c
 */
static int lookup_zone(const struct adfs_discmap *dm, const unsigned int idlen,
                       const u32 frag_id, unsigned int *offset)
{
        const unsigned int endbit = dm->dm_endbit;
        const u32 idmask = (1 << idlen) - 1;
        unsigned char *map = dm->dm_bh->b_data;
        unsigned int start = dm->dm_startbit;
        unsigned int freelink, fragend;
        u32 frag;

        frag = GET_FRAG_ID(map, 8, idmask & 0x7fff);
        freelink = frag ? 8 + frag : 0;

        do {
                frag = GET_FRAG_ID(map, start, idmask);

                fragend = find_next_bit_le(map, endbit, start + idlen);
                if (fragend >= endbit)
                        goto error;

                if (start == freelink) {
                        freelink += frag & 0x7fff;
                } else if (frag == frag_id) {
                        unsigned int length = fragend + 1 - start;

                        if (*offset < length)
                                return start + *offset;
                        *offset -= length;
                }

                start = fragend + 1;
        } while (start < endbit);
        return -1;

error:
        printk(KERN_ERR "adfs: oversized fragment 0x%x at 0x%x-0x%x\n",
                frag, start, fragend);
        return -1;
}

/*
 * Scan the free space map, for this zone, calculating the total
 * number of map bits in each free space fragment.
 *
 * Note: idmask is limited to 15 bits [3.2]
 */
static unsigned int
scan_free_map(struct adfs_sb_info *asb, struct adfs_discmap *dm)
{
        const unsigned int endbit = dm->dm_endbit;
        const unsigned int idlen  = asb->s_idlen;
        const unsigned int frag_idlen = idlen <= 15 ? idlen : 15;
        const u32 idmask = (1 << frag_idlen) - 1;
        unsigned char *map = dm->dm_bh->b_data;
        unsigned int start = 8, fragend;
        u32 frag;
        unsigned long total = 0;

        /*
         * get fragment id
         */
        frag = GET_FRAG_ID(map, start, idmask);

        /*
         * If the freelink is null, then no free fragments
         * exist in this zone.
         */
        if (frag == 0)
                return 0;

        do {
                start += frag;

                frag = GET_FRAG_ID(map, start, idmask);

                fragend = find_next_bit_le(map, endbit, start + idlen);
                if (fragend >= endbit)
                        goto error;

                total += fragend + 1 - start;
        } while (frag >= idlen + 1);

        if (frag != 0)
                printk(KERN_ERR "adfs: undersized free fragment\n");

        return total;
error:
        printk(KERN_ERR "adfs: oversized free fragment\n");
        return 0;
}

static int scan_map(struct adfs_sb_info *asb, unsigned int zone,
                    const u32 frag_id, unsigned int mapoff)
{
        const unsigned int idlen = asb->s_idlen;
        struct adfs_discmap *dm, *dm_end;
        int result;

        dm      = asb->s_map + zone;
        zone    = asb->s_map_size;
        dm_end  = asb->s_map + zone;

        do {
                result = lookup_zone(dm, idlen, frag_id, &mapoff);

                if (result != -1)
                        goto found;

                dm ++;
                if (dm == dm_end)
                        dm = asb->s_map;
        } while (--zone > 0);

        return -1;
found:
        result -= dm->dm_startbit;
        result += dm->dm_startblk;

        return result;
}

/*
 * calculate the amount of free blocks in the map.
 *
 *              n=1
 *  total_free = E(free_in_zone_n)
 *              nzones
 */
void adfs_map_statfs(struct super_block *sb, struct kstatfs *buf)
{
        struct adfs_sb_info *asb = ADFS_SB(sb);
        struct adfs_discrecord *dr = adfs_map_discrecord(asb->s_map);
        struct adfs_discmap *dm;
        unsigned int total = 0;
        unsigned int zone;

        dm   = asb->s_map;
        zone = asb->s_map_size;

        do {
                total += scan_free_map(asb, dm++);
        } while (--zone > 0);

        buf->f_blocks  = adfs_disc_size(dr) >> sb->s_blocksize_bits;
        buf->f_files   = asb->s_ids_per_zone * asb->s_map_size;
        buf->f_bavail  =
        buf->f_bfree   = signed_asl(total, asb->s_map2blk);
}

int adfs_map_lookup(struct super_block *sb, u32 frag_id, unsigned int offset)
{
        struct adfs_sb_info *asb = ADFS_SB(sb);
        unsigned int zone, mapoff;
        int result;

        /*
         * map & root fragment is special - it starts in the center of the
         * disk.  The other fragments start at zone (frag / ids_per_zone)
         */
        if (frag_id == ADFS_ROOT_FRAG)
                zone = asb->s_map_size >> 1;
        else
                zone = frag_id / asb->s_ids_per_zone;

        if (zone >= asb->s_map_size)
                goto bad_fragment;

        /* Convert sector offset to map offset */
        mapoff = signed_asl(offset, -asb->s_map2blk);

        read_lock(&adfs_map_lock);
        result = scan_map(asb, zone, frag_id, mapoff);
        read_unlock(&adfs_map_lock);

        if (result > 0) {
                unsigned int secoff;

                /* Calculate sector offset into map block */
                secoff = offset - signed_asl(mapoff, asb->s_map2blk);
                return secoff + signed_asl(result, asb->s_map2blk);
        }

        adfs_error(sb, "fragment 0x%04x at offset %d not found in map",
                   frag_id, offset);
        return 0;

bad_fragment:
        adfs_error(sb, "invalid fragment 0x%04x (zone = %d, max = %d)",
                   frag_id, zone, asb->s_map_size);
        return 0;
}

static unsigned char adfs_calczonecheck(struct super_block *sb, unsigned char *map)
{
        unsigned int v0, v1, v2, v3;
        int i;

        v0 = v1 = v2 = v3 = 0;
        for (i = sb->s_blocksize - 4; i; i -= 4) {
                v0 += map[i]     + (v3 >> 8);
                v3 &= 0xff;
                v1 += map[i + 1] + (v0 >> 8);
                v0 &= 0xff;
                v2 += map[i + 2] + (v1 >> 8);
                v1 &= 0xff;
                v3 += map[i + 3] + (v2 >> 8);
                v2 &= 0xff;
        }
        v0 +=           v3 >> 8;
        v1 += map[1] + (v0 >> 8);
        v2 += map[2] + (v1 >> 8);
        v3 += map[3] + (v2 >> 8);

        return v0 ^ v1 ^ v2 ^ v3;
}

static int adfs_checkmap(struct super_block *sb, struct adfs_discmap *dm)
{
        unsigned char crosscheck = 0, zonecheck = 1;
        int i;

        for (i = 0; i < ADFS_SB(sb)->s_map_size; i++) {
                unsigned char *map;

                map = dm[i].dm_bh->b_data;

                if (adfs_calczonecheck(sb, map) != map[0]) {
                        adfs_error(sb, "zone %d fails zonecheck", i);
                        zonecheck = 0;
                }
                crosscheck ^= map[3];
        }
        if (crosscheck != 0xff)
                adfs_error(sb, "crosscheck != 0xff");
        return crosscheck == 0xff && zonecheck;
}

/*
 * Layout the map - the first zone contains a copy of the disc record,
 * and the last zone must be limited to the size of the filesystem.
 */
static void adfs_map_layout(struct adfs_discmap *dm, unsigned int nzones,
                            struct adfs_discrecord *dr)
{
        unsigned int zone, zone_size;
        u64 size;

        zone_size = (8 << dr->log2secsize) - le16_to_cpu(dr->zone_spare);

        dm[0].dm_bh       = NULL;
        dm[0].dm_startblk = 0;
        dm[0].dm_startbit = 32 + ADFS_DR_SIZE_BITS;
        dm[0].dm_endbit   = 32 + zone_size;

        for (zone = 1; zone < nzones; zone++) {
                dm[zone].dm_bh       = NULL;
                dm[zone].dm_startblk = zone * zone_size - ADFS_DR_SIZE_BITS;
                dm[zone].dm_startbit = 32;
                dm[zone].dm_endbit   = 32 + zone_size;
        }

        size = adfs_disc_size(dr) >> dr->log2bpmb;
        size -= (nzones - 1) * zone_size - ADFS_DR_SIZE_BITS;
        dm[nzones - 1].dm_endbit = 32 + size;
}

static int adfs_map_read(struct adfs_discmap *dm, struct super_block *sb,
                         unsigned int map_addr, unsigned int nzones)
{
        unsigned int zone;

        for (zone = 0; zone < nzones; zone++) {
                dm[zone].dm_bh = sb_bread(sb, map_addr + zone);
                if (!dm[zone].dm_bh)
                        return -EIO;
        }

        return 0;
}

static void adfs_map_relse(struct adfs_discmap *dm, unsigned int nzones)
{
        unsigned int zone;

        for (zone = 0; zone < nzones; zone++)
                brelse(dm[zone].dm_bh);
}

struct adfs_discmap *adfs_read_map(struct super_block *sb, struct adfs_discrecord *dr)
{
        struct adfs_sb_info *asb = ADFS_SB(sb);
        struct adfs_discmap *dm;
        unsigned int map_addr, zone_size, nzones;
        int ret;

        nzones    = dr->nzones | dr->nzones_high << 8;
        zone_size = (8 << dr->log2secsize) - le16_to_cpu(dr->zone_spare);

        asb->s_idlen = dr->idlen;
        asb->s_map_size = nzones;
        asb->s_map2blk = dr->log2bpmb - dr->log2secsize;
        asb->s_log2sharesize = dr->log2sharesize;
        asb->s_ids_per_zone = zone_size / (asb->s_idlen + 1);

        map_addr = (nzones >> 1) * zone_size -
                     ((nzones > 1) ? ADFS_DR_SIZE_BITS : 0);
        map_addr = signed_asl(map_addr, asb->s_map2blk);

        dm = kmalloc_objs(*dm, nzones);
        if (dm == NULL) {
                adfs_error(sb, "not enough memory");
                return ERR_PTR(-ENOMEM);
        }

        adfs_map_layout(dm, nzones, dr);

        ret = adfs_map_read(dm, sb, map_addr, nzones);
        if (ret) {
                adfs_error(sb, "unable to read map");
                goto error_free;
        }

        if (adfs_checkmap(sb, dm))
                return dm;

        adfs_error(sb, "map corrupted");

error_free:
        adfs_map_relse(dm, nzones);
        kfree(dm);
        return ERR_PTR(-EIO);
}

void adfs_free_map(struct super_block *sb)
{
        struct adfs_sb_info *asb = ADFS_SB(sb);

        adfs_map_relse(asb->s_map, asb->s_map_size);
        kfree(asb->s_map);
}