linux-imx

// SPDX-License-Identifier: GPL-2.0

/*

 *    Copyright IBM Corp. 2007, 2012

 *    Author(s): Peter Oberparleiter <peter.oberparleiter@de.ibm.com>

 */

​

#include <linux/vmalloc.h>

#include <linux/bitmap.h>

#include <linux/bitops.h>

#include "idset.h"

#include "css.h"

​

struct idset {

    int num_ssid;

    int num_id;

    unsigned long bitmap[0];

};

​

static inline unsigned long bitmap_size(int num_ssid, int num_id)

{

    return BITS_TO_LONGS(num_ssid * num_id) * sizeof(unsigned long);

}

​

static struct idset *idset_new(int num_ssid, int num_id)

{

    struct idset *set;

​

    set = vmalloc(sizeof(struct idset) + bitmap_size(num_ssid, num_id));

    if (set) {

        set->num_ssid = num_ssid;

        set->num_id = num_id;

        memset(set->bitmap, 0, bitmap_size(num_ssid, num_id));

    }

    return set;

}

​

void idset_free(struct idset *set)

{

    vfree(set);

}

​

void idset_fill(struct idset *set)

{

    memset(set->bitmap, 0xff, bitmap_size(set->num_ssid, set->num_id));

}

​

static inline void idset_add(struct idset *set, int ssid, int id)

{

    set_bit(ssid * set->num_id + id, set->bitmap);

}

​

static inline void idset_del(struct idset *set, int ssid, int id)

{

    clear_bit(ssid * set->num_id + id, set->bitmap);

}

​

static inline int idset_contains(struct idset *set, int ssid, int id)

{

    return test_bit(ssid * set->num_id + id, set->bitmap);

}

​

static inline int idset_get_first(struct idset *set, int *ssid, int *id)

{