/* SPDX-License-Identifier: GPL-2.0 */
#ifndef KUBLK_UTILS_H
#define KUBLK_UTILS_H

#ifndef min
#define min(a, b) ((a) < (b) ? (a) : (b))
#endif

#define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))

#ifndef offsetof
#define offsetof(TYPE, MEMBER)  ((size_t)&((TYPE *)0)->MEMBER)
#endif

#ifndef container_of
#define container_of(ptr, type, member) ({                              \
        unsigned long __mptr = (unsigned long)(ptr);                    \
        ((type *)(__mptr - offsetof(type, member))); })
#endif

#define round_up(val, rnd) \
        (((val) + ((rnd) - 1)) & ~((rnd) - 1))

/* small sized & per-thread allocator */
struct allocator {
        unsigned int size;
        cpu_set_t *set;
};

static inline int allocator_init(struct allocator *a, unsigned size)
{
        a->set = CPU_ALLOC(size);
        a->size = size;

        if (a->set)
                return 0;
        return -ENOMEM;
}

static inline void allocator_deinit(struct allocator *a)
{
        CPU_FREE(a->set);
        a->set = NULL;
        a->size = 0;
}

static inline int allocator_get(struct allocator *a)
{
        int i;

        for (i = 0; i < a->size; i += 1) {
                size_t set_size = CPU_ALLOC_SIZE(a->size);

                if (!CPU_ISSET_S(i, set_size, a->set)) {
                        CPU_SET_S(i, set_size, a->set);
                        return i;
                }
        }

        return -1;
}

static inline void allocator_put(struct allocator *a, int i)
{
        size_t set_size = CPU_ALLOC_SIZE(a->size);

        if (i >= 0 && i < a->size)
                CPU_CLR_S(i, set_size, a->set);
}

static inline int allocator_get_val(struct allocator *a, int i)
{
        size_t set_size = CPU_ALLOC_SIZE(a->size);

        return CPU_ISSET_S(i, set_size, a->set);
}

static inline unsigned int ilog2(unsigned int x)
{
        if (x == 0)
                return 0;
        return (sizeof(x) * 8 - 1) - __builtin_clz(x);
}

#define UBLK_DBG_DEV            (1U << 0)
#define UBLK_DBG_THREAD         (1U << 1)
#define UBLK_DBG_IO_CMD         (1U << 2)
#define UBLK_DBG_IO             (1U << 3)
#define UBLK_DBG_CTRL_CMD       (1U << 4)
#define UBLK_LOG                (1U << 5)

extern unsigned int ublk_dbg_mask;

static inline void ublk_err(const char *fmt, ...)
{
        va_list ap;

        va_start(ap, fmt);
        vfprintf(stderr, fmt, ap);
        va_end(ap);
}

static inline void ublk_log(const char *fmt, ...)
{
        if (ublk_dbg_mask & UBLK_LOG) {
                va_list ap;

                va_start(ap, fmt);
                vfprintf(stdout, fmt, ap);
                va_end(ap);
        }
}

static inline void ublk_dbg(int level, const char *fmt, ...)
{
        if (level & ublk_dbg_mask) {
                va_list ap;

                va_start(ap, fmt);
                vfprintf(stdout, fmt, ap);
                va_end(ap);
        }
}

#define ublk_assert(x)  do { \
        if (!(x)) {     \
                ublk_err("%s %d: assert!\n", __func__, __LINE__); \
                assert(x);      \
        }       \
} while (0)

#endif