diff --git a/src/utils/utils_refcnt.c b/src/utils/utils_refcnt.c new file mode 100644 index 0000000..045e952 --- /dev/null +++ b/src/utils/utils_refcnt.c @@ -0,0 +1,53 @@ +/* + * Copyright(c) 2019 Intel Corporation + * SPDX-License-Identifier: BSD-3-Clause-Clear + */ + +#include "../utils/utils_refcnt.h" + +void ocf_refcnt_dec(struct ocf_refcnt *rc) +{ + int val = env_atomic_dec_return(&rc->counter); + ENV_BUG_ON(val < 0); + + if (!val && env_atomic_cmpxchg(&rc->callback, 1, 0)) + rc->cb(rc->priv); +} + +bool ocf_refcnt_inc(struct ocf_refcnt *rc) +{ + if (!env_atomic_read(&rc->freeze)) { + env_atomic_inc(&rc->counter); + if (!env_atomic_read(&rc->freeze)) + return true; + else + ocf_refcnt_dec(rc); + } + + return false; +} + + +void ocf_refcnt_freeze(struct ocf_refcnt *rc) +{ + env_atomic_inc(&rc->freeze); +} + +void ocf_refcnt_register_zero_cb(struct ocf_refcnt *rc, ocf_refcnt_cb_t cb, + void *priv) +{ + ENV_BUG_ON(!env_atomic_read(&rc->freeze)); + ENV_BUG_ON(env_atomic_read(&rc->callback)); + + env_atomic_inc(&rc->counter); + rc->cb = cb; + rc->priv = priv; + env_atomic_set(&rc->callback, 1); + ocf_refcnt_dec(rc); +} + +void ocf_refcnt_unfreeze(struct ocf_refcnt *rc) +{ + int val = env_atomic_dec_return(&rc->freeze); + ENV_BUG_ON(val < 0); +} diff --git a/src/utils/utils_refcnt.h b/src/utils/utils_refcnt.h new file mode 100644 index 0000000..5df2714 --- /dev/null +++ b/src/utils/utils_refcnt.h @@ -0,0 +1,43 @@ +/* + * Copyright(c) 2019 Intel Corporation + * SPDX-License-Identifier: BSD-3-Clause-Clear + */ + +#ifndef __OCF_REFCNT_H__ +#define __OCF_REFCNT_H__ + +#include "ocf_env.h" + +typedef void (*ocf_refcnt_cb_t)(void *priv); + +struct ocf_refcnt +{ + env_atomic counter; + env_atomic freeze; + env_atomic callback; + ocf_refcnt_cb_t cb; + void *priv; +}; + +/* Try to increment counter. Returns true if successfull, false if freezed */ +bool ocf_refcnt_inc(struct ocf_refcnt *rc); + +/* Decrement reference counter */ +void ocf_refcnt_dec(struct ocf_refcnt *rc); + +/* Disallow incrementing of underlying counter - attempts to increment counter + * will be failing until ocf_refcnt_unfreeze is calleed. + * It's ok to call freeze multiple times, in which case counter is freezed + * until all freeze calls are offset by a corresponding unfreeze.*/ +void ocf_refcnt_freeze(struct ocf_refcnt *rc); + +/* Cancel the effect of single ocf_refcnt_freeze call */ +void ocf_refcnt_unfreeze(struct ocf_refcnt *rc); + +/* Register callback to be called when reference counter drops to 0. + * Must be called after counter is freezed. + * Cannot be called until previously regsitered callback had fired. */ +void ocf_refcnt_register_zero_cb(struct ocf_refcnt *rc, ocf_refcnt_cb_t cb, + void *priv); + +#endif // __OCF_REFCNT_H__