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