/*
 * Copyright 2009 Colin Günther, coling@gmx.de
 * Copyright 2018, Haiku, Inc.
 * All rights reserved. Distributed under the terms of the MIT license.
 */
 
 
/*! Implementation of a number allocator.*/
 
 
extern "C" {
#include "unit.h"
}
 
#include <compat/sys/mutex.h>
 
#include <stdlib.h>
#include <util/RadixBitmap.h>
 
 
#define ID_STORE_FULL -1
 
 
extern struct mtx gIdStoreLock;
 
 
struct unrhdr*
new_unrhdr(int low, int high, struct mtx* mutex)
{
	struct unrhdr* idStore;
	uint32 maxIdCount = high - low + 1;
 
	KASSERT(low <= high,
		("ID-Store: use error: %s(%u, %u)", __func__, low, high));
 
	idStore = (unrhdr*)malloc(sizeof *idStore);
	if (idStore == NULL)
		return NULL;
 
	idStore->idBuffer = radix_bitmap_create(maxIdCount);
	if (idStore->idBuffer == NULL) {
		free(idStore);
		return NULL;
	}
 
	if (mutex)
		idStore->storeMutex = mutex;
	else
		idStore->storeMutex = &gIdStoreLock;
 
	idStore->idBias = low;
 
	return idStore;
}
 
 
void
delete_unrhdr(struct unrhdr* idStore)
{
	KASSERT(idStore != NULL,
		("ID-Store: %s: NULL pointer as argument.", __func__));
 
	mtx_lock(idStore->storeMutex);
 
	KASSERT(idStore->idBuffer->root_size == 0,
		("ID-Store: %s: some ids are still in use..", __func__));
 
	radix_bitmap_destroy(idStore->idBuffer);
	mtx_unlock(idStore->storeMutex);
 
	free(idStore);
	idStore = NULL;
}
 
 
int
alloc_unr(struct unrhdr* idStore)
{
	int id;
 
	KASSERT(idStore != NULL,
		("ID-Store: %s: NULL pointer as argument.", __func__));
 
	mtx_lock(idStore->storeMutex);
 
	radix_slot_t slotIndex;
	id = ID_STORE_FULL;
 
	slotIndex = radix_bitmap_alloc(idStore->idBuffer, 1);
	if (slotIndex != RADIX_SLOT_NONE)
		id = slotIndex + idStore->idBias;
 
	mtx_unlock(idStore->storeMutex);
 
	return id;
}
 
 
void
free_unr(struct unrhdr* idStore, u_int identity)
{
	KASSERT(idStore != NULL,
		("ID-Store: %s: NULL pointer as argument.", __func__));
 
	mtx_lock(idStore->storeMutex);
 
	uint32 slotIndex = (int32)identity - idStore->idBias;
	KASSERT(slotIndex >= 0, ("ID-Store: %s(%p, %u): second "
		"parameter is not in interval.", __func__, idStore, identity));
 
	radix_bitmap_dealloc(idStore->idBuffer, slotIndex, 1);
 
	mtx_unlock(idStore->storeMutex);
}