if (!context)
return NULL;
- atomic_inc(&context->ref_count);
+ kref_init(&context->refcount);
return context;
}
void dc_retain_state(struct dc_state *context)
{
- ASSERT(atomic_read(&context->ref_count) > 0);
- atomic_inc(&context->ref_count);
+ kref_get(&context->refcount);
}
-void dc_release_state(struct dc_state *context)
+static void dc_state_free(struct kref *kref)
{
- ASSERT(atomic_read(&context->ref_count) > 0);
- atomic_dec(&context->ref_count);
+ struct dc_state *context = container_of(kref, struct dc_state, refcount);
+ dc_resource_state_destruct(context);
+ kfree(context);
+}
- if (atomic_read(&context->ref_count) == 0) {
- dc_resource_state_destruct(context);
- kfree(context);
- }
+void dc_release_state(struct dc_state *context)
+{
+ kref_put(&context->refcount, dc_state_free);
}
static bool is_surface_in_context(
struct dc *dc,
enum dc_acpi_cm_power_state power_state)
{
- atomic_t ref_count;
+ struct kref refcount;
switch (power_state) {
case DC_ACPI_CM_POWER_STATE_D0:
*/
/* Preserve refcount */
- ref_count = dc->current_state->ref_count;
+ refcount = dc->current_state->refcount;
dc_resource_state_destruct(dc->current_state);
memset(dc->current_state, 0,
sizeof(*dc->current_state));
- dc->current_state->ref_count = ref_count;
+ dc->current_state->refcount = refcount;
break;
}
struct dc_state *dst_ctx)
{
int i, j;
- atomic_t ref_count = dst_ctx->ref_count;
+ struct kref refcount = dst_ctx->refcount;
*dst_ctx = *src_ctx;
}
/* context refcount should not be overridden */
- dst_ctx->ref_count = ref_count;
+ dst_ctx->refcount = refcount;
}