balloc.h

#ifndef _HK_BALLOC_H_
#define _HK_BALLOC_H_

#include "hunter.h"

enum hk_ind_upt_type {
    VALIDATE_BLK = 0,
    INVALIDATE_BLK,
    PREP_LAYOUT_REMOVE,
    PREP_LAYOUT_APPEND,
    PREP_LAYOUT_GAP,
    FREE_LAYOUT
};

struct hk_indicator {
    u64 valid_blks;   //! Valid blks => hdr is valid
    u64 invalid_blks; //! Invalid blks => hdr is valid and in the boundary of atomic counter
    u64 free_blks;    //! Free blks => behind the boundary of atomic counter
    u64 prep_blks;    //! Preparing blks => Prepared Blks
    u64 total_blks;
};

enum hk_layout_type {
    LAYOUT_APPEND = 0,
    LAYOUT_GAP
};

struct hk_layout_info {
    struct mutex layout_lock;
    u64 atomic_counter;
    u32 cpuid;
    u64 layout_start;
    u64 layout_end;
    u64 layout_blks;

    u64 num_gaps_indram;
    struct rb_root_cached gaps_tree;

    // Statistics
    struct hk_indicator ind;
};

static inline void hk_dump_layout_info(struct hk_layout_info *layout)
{
    struct hk_indicator *ind = &layout->ind;
    hk_info("layout: %d===>\n", layout->cpuid);
    hk_info("-----------------------------------\n");
    hk_info("tail: 0x%llx\n", layout->atomic_counter);
    hk_info("valid_blks: %llu, invalid_blks: %llu, free_blks: %llu, prep_blks: %llu, total: %llu\n",
            ind->valid_blks, ind->invalid_blks, ind->free_blks, ind->prep_blks, ind->total_blks);
}

struct hk_layout_prep {
    int cpuid;
    u64 target_addr;
    u64 blks_prepared;
    bool is_overflow;
};

struct hk_layout_preps {
    bool is_enough_space;
    u32 num_layout;
    u32 idx;
    struct hk_layout_prep preps[HK_MAX_LAYOUTS];
};

#define traverse_layout_blks(addr, layout)         for (addr = layout->layout_start; addr < layout->layout_start + layout->atomic_counter; addr += HK_PBLK_SZ)
#define traverse_layout_blks_reverse(addr, layout) for (addr = layout->layout_start + layout->atomic_counter - HK_PBLK_SZ; addr >= layout->layout_start; addr -= HK_PBLK_SZ)
#define GET_LAST_BLK_FROM_LAYOUT(layout)           (layout->atomic_counter + layout->layout_start - HK_PBLK_SZ)
#endif /* _HK_BALLOC_H */