63 lines
1.2 KiB
C
63 lines
1.2 KiB
C
#include <id/id_alloc.h>
|
|
#include <libk/bm.h>
|
|
#include <libk/std.h>
|
|
#include <mm/malloc.h>
|
|
#include <status.h>
|
|
#include <sync/spin_lock.h>
|
|
#include <sys/debug.h>
|
|
|
|
bool id_alloc_init(struct id_alloc* ida, size_t nbits) {
|
|
size_t buffer_size = (nbits + 7) / 8;
|
|
|
|
uint8_t* buffer = malloc(buffer_size);
|
|
|
|
if (buffer == NULL)
|
|
return false;
|
|
|
|
bm_init(&ida->bm, buffer, nbits);
|
|
ida->lock = SPIN_LOCK_INIT;
|
|
|
|
return true;
|
|
}
|
|
|
|
void id_alloc_fini(struct id_alloc* ida) {
|
|
free(ida->bm.base);
|
|
ida->bm.base = NULL;
|
|
ida->bm.nbits = 0;
|
|
}
|
|
|
|
int id_alloc(struct id_alloc* ida) {
|
|
uint64_t fid;
|
|
|
|
spin_lock(&ida->lock, &fid);
|
|
|
|
size_t start = ida->next_id;
|
|
size_t current = start;
|
|
do {
|
|
if (!bm_test(&ida->bm, current)) {
|
|
bm_set(&ida->bm, current);
|
|
|
|
ida->next_id = (current + 1) % ida->bm.nbits;
|
|
|
|
spin_unlock(&ida->lock, fid);
|
|
return (int)current;
|
|
}
|
|
|
|
current = (current + 1) % ida->bm.nbits;
|
|
} while (current != start);
|
|
|
|
spin_unlock(&ida->lock, fid);
|
|
return -ST_OOM_ERROR;
|
|
}
|
|
|
|
void id_free(struct id_alloc* ida, int id) {
|
|
uint64_t fid;
|
|
|
|
if (id < 0)
|
|
return;
|
|
|
|
spin_lock(&ida->lock, &fid);
|
|
bm_clear(&ida->bm, (size_t)id);
|
|
spin_unlock(&ida->lock, fid);
|
|
}
|