124 lines
2.4 KiB
C
124 lines
2.4 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
|
|
/*! \file */
|
|
|
|
#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;
|
|
quota->soft = 0;
|
|
return (isc_mutex_init("a->lock));
|
|
}
|
|
|
|
void
|
|
isc_quota_destroy(isc_quota_t *quota) {
|
|
INSIST(quota->used == 0);
|
|
quota->max = 0;
|
|
quota->used = 0;
|
|
quota->soft = 0;
|
|
DESTROYLOCK("a->lock);
|
|
}
|
|
|
|
void
|
|
isc_quota_soft(isc_quota_t *quota, int soft) {
|
|
LOCK("a->lock);
|
|
quota->soft = soft;
|
|
UNLOCK("a->lock);
|
|
}
|
|
|
|
void
|
|
isc_quota_max(isc_quota_t *quota, int max) {
|
|
LOCK("a->lock);
|
|
quota->max = max;
|
|
UNLOCK("a->lock);
|
|
}
|
|
|
|
isc_result_t
|
|
isc_quota_reserve(isc_quota_t *quota) {
|
|
isc_result_t result;
|
|
LOCK("a->lock);
|
|
if (quota->max == 0 || quota->used < quota->max) {
|
|
if (quota->soft == 0 || quota->used < quota->soft)
|
|
result = ISC_R_SUCCESS;
|
|
else
|
|
result = ISC_R_SOFTQUOTA;
|
|
quota->used++;
|
|
} 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);
|
|
}
|
|
|
|
static isc_result_t
|
|
doattach(isc_quota_t *quota, isc_quota_t **p, bool force) {
|
|
isc_result_t result;
|
|
REQUIRE(p != NULL && *p == NULL);
|
|
|
|
result = isc_quota_reserve(quota);
|
|
if (result == ISC_R_SUCCESS || result == ISC_R_SOFTQUOTA) {
|
|
*p = quota;
|
|
} else if (result == ISC_R_QUOTA && force) {
|
|
/* attach anyway */
|
|
LOCK("a->lock);
|
|
quota->used++;
|
|
UNLOCK("a->lock);
|
|
|
|
*p = quota;
|
|
result = ISC_R_SUCCESS;
|
|
}
|
|
|
|
return (result);
|
|
}
|
|
|
|
isc_result_t
|
|
isc_quota_attach(isc_quota_t *quota, isc_quota_t **p) {
|
|
return (doattach(quota, p, false));
|
|
}
|
|
|
|
isc_result_t
|
|
isc_quota_force(isc_quota_t *quota, isc_quota_t **p) {
|
|
return (doattach(quota, p, true));
|
|
}
|
|
|
|
void
|
|
isc_quota_detach(isc_quota_t **p) {
|
|
INSIST(p != NULL && *p != NULL);
|
|
isc_quota_release(*p);
|
|
*p = NULL;
|
|
}
|
|
|
|
unsigned int
|
|
isc_quota_getused(isc_quota_t *quota) {
|
|
int used;
|
|
|
|
LOCK("a->lock);
|
|
used = quota->used;
|
|
UNLOCK("a->lock);
|
|
return (used);
|
|
}
|