#include <config.h>
#include <stddef.h>
#include <isc/quota.h>
#include <isc/util.h>
isc_result_t
isc_quota_init(isc_quota_t *quota, int max) {
quota->max = max;
quota->used = 0;
return (isc_mutex_init("a->lock));
}
void
isc_quota_destroy(isc_quota_t *quota) {
INSIST(quota->used == 0);
quota->max = -1;
quota->used = -1;
DESTROYLOCK("a->lock);
}
isc_result_t
isc_quota_reserve(isc_quota_t *quota) {
isc_result_t result;
LOCK("a->lock);
if (quota->used < quota->max) {
quota->used++;
result = ISC_R_SUCCESS;
} else {
result = ISC_R_QUOTA;
}
UNLOCK("a->lock);
return (result);
}
void
isc_quota_release(isc_quota_t *quota) {
LOCK("a->lock);
INSIST(quota->used > 0);
quota->used--;
UNLOCK("a->lock);
}
isc_result_t
isc_quota_attach(isc_quota_t *quota, isc_quota_t **p)
{
isc_result_t result;
INSIST(p != NULL && *p == NULL);
result = isc_quota_reserve(quota);
if (result != ISC_R_SUCCESS)
return (result);
*p = quota;
return (ISC_R_SUCCESS);
}
void
isc_quota_detach(isc_quota_t **p)
{
INSIST(p != NULL && *p != NULL);
isc_quota_release(*p);
*p = NULL;
}