#include <linux/export.h>
#include <linux/kernel.h>
#include <linux/mm.h>
#include <linux/slab.h>
#include <linux/vmalloc.h>

/* Allocate an array of spinlocks to be accessed by a hash. Two arguments
 * indicate the number of elements to allocate in the array. max_size
 * gives the maximum number of elements to allocate. cpu_mult gives
 * the number of locks per CPU to allocate. The size is rounded up
 * to a power of 2 to be suitable as a hash table.
 */

int __alloc_bucket_spinlocks(spinlock_t **locks, unsigned int *locks_mask,
                             size_t max_size, unsigned int cpu_mult, gfp_t gfp,
                             const char *name, struct lock_class_key *key)
{
        spinlock_t *tlocks = NULL;
        unsigned int i, size;
#if defined(CONFIG_PROVE_LOCKING)
        unsigned int nr_pcpus = 2;
#else
        unsigned int nr_pcpus = num_possible_cpus();
#endif

        if (cpu_mult) {
                nr_pcpus = min_t(unsigned int, nr_pcpus, 64UL);
                size = min_t(unsigned int, nr_pcpus * cpu_mult, max_size);
        } else {
                size = max_size;
        }

        if (sizeof(spinlock_t) != 0) {
                tlocks = kvmalloc_objs(spinlock_t, size, gfp);
                if (!tlocks)
                        return -ENOMEM;
                for (i = 0; i < size; i++) {
                        spin_lock_init(&tlocks[i]);
                        lockdep_init_map(&tlocks[i].dep_map, name, key, 0);
                }
        }

        *locks = tlocks;
        *locks_mask = size - 1;

        return 0;
}
EXPORT_SYMBOL(__alloc_bucket_spinlocks);

void free_bucket_spinlocks(spinlock_t *locks)
{
        kvfree(locks);
}
EXPORT_SYMBOL(free_bucket_spinlocks);