103 lines · c
1// SPDX-License-Identifier: GPL-2.02/*3 * Copyright IBM Corp. 2007, 20124 * Author(s): Peter Oberparleiter <peter.oberparleiter@de.ibm.com>5 */6 7#include <linux/vmalloc.h>8#include <linux/bitmap.h>9#include <linux/bitops.h>10#include "idset.h"11#include "css.h"12 13struct idset {14 int num_ssid;15 int num_id;16 unsigned long bitmap[];17};18 19static inline unsigned long idset_bitmap_size(int num_ssid, int num_id)20{21 return bitmap_size(size_mul(num_ssid, num_id));22}23 24static struct idset *idset_new(int num_ssid, int num_id)25{26 struct idset *set;27 28 set = vmalloc(sizeof(struct idset) +29 idset_bitmap_size(num_ssid, num_id));30 if (set) {31 set->num_ssid = num_ssid;32 set->num_id = num_id;33 memset(set->bitmap, 0, idset_bitmap_size(num_ssid, num_id));34 }35 return set;36}37 38void idset_free(struct idset *set)39{40 vfree(set);41}42 43void idset_fill(struct idset *set)44{45 memset(set->bitmap, 0xff,46 idset_bitmap_size(set->num_ssid, set->num_id));47}48 49static inline void idset_add(struct idset *set, int ssid, int id)50{51 set_bit(ssid * set->num_id + id, set->bitmap);52}53 54static inline void idset_del(struct idset *set, int ssid, int id)55{56 clear_bit(ssid * set->num_id + id, set->bitmap);57}58 59static inline int idset_contains(struct idset *set, int ssid, int id)60{61 return test_bit(ssid * set->num_id + id, set->bitmap);62}63 64struct idset *idset_sch_new(void)65{66 return idset_new(max_ssid + 1, __MAX_SUBCHANNEL + 1);67}68 69void idset_sch_add(struct idset *set, struct subchannel_id schid)70{71 idset_add(set, schid.ssid, schid.sch_no);72}73 74void idset_sch_del(struct idset *set, struct subchannel_id schid)75{76 idset_del(set, schid.ssid, schid.sch_no);77}78 79/* Clear ids starting from @schid up to end of subchannel set. */80void idset_sch_del_subseq(struct idset *set, struct subchannel_id schid)81{82 int pos = schid.ssid * set->num_id + schid.sch_no;83 84 bitmap_clear(set->bitmap, pos, set->num_id - schid.sch_no);85}86 87int idset_sch_contains(struct idset *set, struct subchannel_id schid)88{89 return idset_contains(set, schid.ssid, schid.sch_no);90}91 92int idset_is_empty(struct idset *set)93{94 return bitmap_empty(set->bitmap, set->num_ssid * set->num_id);95}96 97void idset_add_set(struct idset *to, struct idset *from)98{99 int len = min(to->num_ssid * to->num_id, from->num_ssid * from->num_id);100 101 bitmap_or(to->bitmap, to->bitmap, from->bitmap, len);102}103