/* SPDX-License-Identifier: GPL-2.0-only */
/* Copyright (C) 2024 Intel Corporation */

#ifndef __LIBETH_CACHE_H
#define __LIBETH_CACHE_H

#include <linux/cache.h>

/**
 * libeth_cacheline_group_assert - make sure cacheline group size is expected
 * @type: type of the structure containing the group
 * @grp: group name inside the struct
 * @sz: expected group size
 */
#if defined(CONFIG_64BIT) && SMP_CACHE_BYTES == 64
#define libeth_cacheline_group_assert(type, grp, sz)                          \
        static_assert(offsetof(type, __cacheline_group_end__##grp) -          \
                      offsetofend(type, __cacheline_group_begin__##grp) ==    \
                      (sz))
#define __libeth_cacheline_struct_assert(type, sz)                            \
        static_assert(sizeof(type) == (sz))
#else /* !CONFIG_64BIT || SMP_CACHE_BYTES != 64 */
#define libeth_cacheline_group_assert(type, grp, sz)                          \
        static_assert(offsetof(type, __cacheline_group_end__##grp) -          \
                      offsetofend(type, __cacheline_group_begin__##grp) <=    \
                      (sz))
#define __libeth_cacheline_struct_assert(type, sz)                            \
        static_assert(sizeof(type) <= (sz))
#endif /* !CONFIG_64BIT || SMP_CACHE_BYTES != 64 */

#define __libeth_cls1(sz1)      SMP_CACHE_ALIGN(sz1)
#define __libeth_cls2(sz1, sz2) (SMP_CACHE_ALIGN(sz1) + SMP_CACHE_ALIGN(sz2))
#define __libeth_cls3(sz1, sz2, sz3)                                          \
        (SMP_CACHE_ALIGN(sz1) + SMP_CACHE_ALIGN(sz2) + SMP_CACHE_ALIGN(sz3))
#define __libeth_cls(...)                                                     \
        CONCATENATE(__libeth_cls, COUNT_ARGS(__VA_ARGS__))(__VA_ARGS__)

/**
 * libeth_cacheline_struct_assert - make sure CL-based struct size is expected
 * @type: type of the struct
 * @...: from 1 to 3 CL group sizes (read-mostly, read-write, cold)
 *
 * When a struct contains several CL groups, it's difficult to predict its size
 * on different architectures. The macro instead takes sizes of all of the
 * groups the structure contains and generates the final struct size.
 */
#define libeth_cacheline_struct_assert(type, ...)                             \
        __libeth_cacheline_struct_assert(type, __libeth_cls(__VA_ARGS__));    \
        static_assert(__alignof(type) >= SMP_CACHE_BYTES)

/**
 * libeth_cacheline_set_assert - make sure CL-based struct layout is expected
 * @type: type of the struct
 * @ro: expected size of the read-mostly group
 * @rw: expected size of the read-write group
 * @c: expected size of the cold group
 *
 * Check that each group size is expected and then do final struct size check.
 */
#define libeth_cacheline_set_assert(type, ro, rw, c)                          \
        libeth_cacheline_group_assert(type, read_mostly, ro);                 \
        libeth_cacheline_group_assert(type, read_write, rw);                  \
        libeth_cacheline_group_assert(type, cold, c);                         \
        libeth_cacheline_struct_assert(type, ro, rw, c)

#endif /* __LIBETH_CACHE_H */