1 // SPDX-License-Identifier: GPL-2.0 OR MIT
2 /* Copyright 2018-2019 Qiang Yu <yuq825@gmail.com> */
3 
4 #include <linux/slab.h>
5 
6 #include "lima_device.h"
7 #include "lima_ctx.h"
8 
lima_ctx_create(struct lima_device * dev,struct lima_ctx_mgr * mgr,u32 * id)9 int lima_ctx_create(struct lima_device *dev, struct lima_ctx_mgr *mgr, u32 *id)
10 {
11 	struct lima_ctx *ctx;
12 	int i, err;
13 
14 	ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
15 	if (!ctx)
16 		return -ENOMEM;
17 	ctx->dev = dev;
18 	kref_init(&ctx->refcnt);
19 
20 	for (i = 0; i < lima_pipe_num; i++) {
21 		err = lima_sched_context_init(dev->pipe + i, ctx->context + i, &ctx->guilty);
22 		if (err)
23 			goto err_out0;
24 	}
25 
26 	err = xa_alloc(&mgr->handles, id, ctx, xa_limit_32b, GFP_KERNEL);
27 	if (err < 0)
28 		goto err_out0;
29 
30 	ctx->pid = task_pid_nr(current);
31 	get_task_comm(ctx->pname, current);
32 
33 	return 0;
34 
35 err_out0:
36 	for (i--; i >= 0; i--)
37 		lima_sched_context_fini(dev->pipe + i, ctx->context + i);
38 	kfree(ctx);
39 	return err;
40 }
41 
lima_ctx_do_release(struct kref * ref)42 static void lima_ctx_do_release(struct kref *ref)
43 {
44 	struct lima_ctx *ctx = container_of(ref, struct lima_ctx, refcnt);
45 	int i;
46 
47 	for (i = 0; i < lima_pipe_num; i++)
48 		lima_sched_context_fini(ctx->dev->pipe + i, ctx->context + i);
49 	kfree(ctx);
50 }
51 
lima_ctx_free(struct lima_ctx_mgr * mgr,u32 id)52 int lima_ctx_free(struct lima_ctx_mgr *mgr, u32 id)
53 {
54 	struct lima_ctx *ctx;
55 	int ret = 0;
56 
57 	mutex_lock(&mgr->lock);
58 	ctx = xa_erase(&mgr->handles, id);
59 	if (ctx)
60 		kref_put(&ctx->refcnt, lima_ctx_do_release);
61 	else
62 		ret = -EINVAL;
63 	mutex_unlock(&mgr->lock);
64 	return ret;
65 }
66 
lima_ctx_get(struct lima_ctx_mgr * mgr,u32 id)67 struct lima_ctx *lima_ctx_get(struct lima_ctx_mgr *mgr, u32 id)
68 {
69 	struct lima_ctx *ctx;
70 
71 	mutex_lock(&mgr->lock);
72 	ctx = xa_load(&mgr->handles, id);
73 	if (ctx)
74 		kref_get(&ctx->refcnt);
75 	mutex_unlock(&mgr->lock);
76 	return ctx;
77 }
78 
lima_ctx_put(struct lima_ctx * ctx)79 void lima_ctx_put(struct lima_ctx *ctx)
80 {
81 	kref_put(&ctx->refcnt, lima_ctx_do_release);
82 }
83 
lima_ctx_mgr_init(struct lima_ctx_mgr * mgr)84 void lima_ctx_mgr_init(struct lima_ctx_mgr *mgr)
85 {
86 	mutex_init(&mgr->lock);
87 	xa_init_flags(&mgr->handles, XA_FLAGS_ALLOC);
88 }
89 
lima_ctx_mgr_fini(struct lima_ctx_mgr * mgr)90 void lima_ctx_mgr_fini(struct lima_ctx_mgr *mgr)
91 {
92 	struct lima_ctx *ctx;
93 	unsigned long id;
94 
95 	xa_for_each(&mgr->handles, id, ctx) {
96 		kref_put(&ctx->refcnt, lima_ctx_do_release);
97 	}
98 
99 	xa_destroy(&mgr->handles);
100 	mutex_destroy(&mgr->lock);
101 }
102