1 // SPDX-License-Identifier: GPL-2.0 OR MIT
4 #include <linux/module.h>
5 #include <linux/of_platform.h>
6 #include <linux/uaccess.h>
7 #include <linux/slab.h>
8 #include <drm/drm_ioctl.h>
9 #include <drm/drm_drv.h>
10 #include <drm/drm_prime.h>
11 #include <drm/lima_drm.h>
15 #include "lima_gem_prime.h"
18 int lima_sched_timeout_ms;
20 MODULE_PARM_DESC(sched_timeout_ms, "task run timeout in ms");
21 module_param_named(sched_timeout_ms, lima_sched_timeout_ms, int, 0444);
23 static int lima_ioctl_get_param(struct drm_device *dev, void *data, struct drm_file *file)
25 struct drm_lima_get_param *args = data;
26 struct lima_device *ldev = to_lima_dev(dev);
31 switch (args->param) {
32 case DRM_LIMA_PARAM_GPU_ID:
34 case lima_gpu_mali400:
35 args->value = DRM_LIMA_PARAM_GPU_ID_MALI400;
37 case lima_gpu_mali450:
38 args->value = DRM_LIMA_PARAM_GPU_ID_MALI450;
41 args->value = DRM_LIMA_PARAM_GPU_ID_UNKNOWN;
46 case DRM_LIMA_PARAM_NUM_PP:
47 args->value = ldev->pipe[lima_pipe_pp].num_processor;
50 case DRM_LIMA_PARAM_GP_VERSION:
51 args->value = ldev->gp_version;
54 case DRM_LIMA_PARAM_PP_VERSION:
55 args->value = ldev->pp_version;
65 static int lima_ioctl_gem_create(struct drm_device *dev, void *data, struct drm_file *file)
67 struct drm_lima_gem_create *args = data;
78 return lima_gem_create_handle(dev, file, args->size, args->flags, &args->handle);
81 static int lima_ioctl_gem_info(struct drm_device *dev, void *data, struct drm_file *file)
83 struct drm_lima_gem_info *args = data;
85 return lima_gem_get_info(file, args->handle, &args->va, &args->offset);
88 static int lima_ioctl_gem_submit(struct drm_device *dev, void *data, struct drm_file *file)
90 struct drm_lima_gem_submit *args = data;
91 struct lima_device *ldev = to_lima_dev(dev);
92 struct lima_drm_priv *priv = file->driver_priv;
93 struct drm_lima_gem_submit_bo *bos;
94 struct lima_sched_pipe *pipe;
95 struct lima_sched_task *task;
97 struct lima_submit submit = {0};
101 if (args->pipe >= lima_pipe_num || args->nr_bos == 0)
104 if (args->flags & ~(LIMA_SUBMIT_FLAG_EXPLICIT_FENCE))
107 pipe = ldev->pipe + args->pipe;
108 if (args->frame_size != pipe->frame_size)
111 bos = kvcalloc(args->nr_bos, sizeof(*submit.bos) + sizeof(*submit.lbos), GFP_KERNEL);
115 size = args->nr_bos * sizeof(*submit.bos);
116 if (copy_from_user(bos, u64_to_user_ptr(args->bos), size)) {
121 task = kmem_cache_zalloc(pipe->task_slab, GFP_KERNEL);
127 task->frame = task + 1;
128 if (copy_from_user(task->frame, u64_to_user_ptr(args->frame), args->frame_size)) {
133 err = pipe->task_validate(pipe, task);
137 ctx = lima_ctx_get(&priv->ctx_mgr, args->ctx);
143 submit.pipe = args->pipe;
145 submit.lbos = (void *)bos + size;
146 submit.nr_bos = args->nr_bos;
149 submit.flags = args->flags;
150 submit.in_sync[0] = args->in_sync[0];
151 submit.in_sync[1] = args->in_sync[1];
152 submit.out_sync = args->out_sync;
154 err = lima_gem_submit(file, &submit);
159 kmem_cache_free(pipe->task_slab, task);
165 static int lima_ioctl_gem_wait(struct drm_device *dev, void *data, struct drm_file *file)
167 struct drm_lima_gem_wait *args = data;
169 if (args->op & ~(LIMA_GEM_WAIT_READ|LIMA_GEM_WAIT_WRITE))
172 return lima_gem_wait(file, args->handle, args->op, args->timeout_ns);
175 static int lima_ioctl_ctx_create(struct drm_device *dev, void *data, struct drm_file *file)
177 struct drm_lima_ctx_create *args = data;
178 struct lima_drm_priv *priv = file->driver_priv;
179 struct lima_device *ldev = to_lima_dev(dev);
184 return lima_ctx_create(ldev, &priv->ctx_mgr, &args->id);
187 static int lima_ioctl_ctx_free(struct drm_device *dev, void *data, struct drm_file *file)
189 struct drm_lima_ctx_create *args = data;
190 struct lima_drm_priv *priv = file->driver_priv;
195 return lima_ctx_free(&priv->ctx_mgr, args->id);
198 static int lima_drm_driver_open(struct drm_device *dev, struct drm_file *file)
201 struct lima_drm_priv *priv;
202 struct lima_device *ldev = to_lima_dev(dev);
204 priv = kzalloc(sizeof(*priv), GFP_KERNEL);
208 priv->vm = lima_vm_create(ldev);
214 lima_ctx_mgr_init(&priv->ctx_mgr);
216 file->driver_priv = priv;
224 static void lima_drm_driver_postclose(struct drm_device *dev, struct drm_file *file)
226 struct lima_drm_priv *priv = file->driver_priv;
228 lima_ctx_mgr_fini(&priv->ctx_mgr);
229 lima_vm_put(priv->vm);
233 static const struct drm_ioctl_desc lima_drm_driver_ioctls[] = {
234 DRM_IOCTL_DEF_DRV(LIMA_GET_PARAM, lima_ioctl_get_param, DRM_AUTH|DRM_RENDER_ALLOW),
235 DRM_IOCTL_DEF_DRV(LIMA_GEM_CREATE, lima_ioctl_gem_create, DRM_AUTH|DRM_RENDER_ALLOW),
236 DRM_IOCTL_DEF_DRV(LIMA_GEM_INFO, lima_ioctl_gem_info, DRM_AUTH|DRM_RENDER_ALLOW),
237 DRM_IOCTL_DEF_DRV(LIMA_GEM_SUBMIT, lima_ioctl_gem_submit, DRM_AUTH|DRM_RENDER_ALLOW),
238 DRM_IOCTL_DEF_DRV(LIMA_GEM_WAIT, lima_ioctl_gem_wait, DRM_AUTH|DRM_RENDER_ALLOW),
239 DRM_IOCTL_DEF_DRV(LIMA_CTX_CREATE, lima_ioctl_ctx_create, DRM_AUTH|DRM_RENDER_ALLOW),
240 DRM_IOCTL_DEF_DRV(LIMA_CTX_FREE, lima_ioctl_ctx_free, DRM_AUTH|DRM_RENDER_ALLOW),
243 static const struct file_operations lima_drm_driver_fops = {
244 .owner = THIS_MODULE,
246 .release = drm_release,
247 .unlocked_ioctl = drm_ioctl,
249 .compat_ioctl = drm_compat_ioctl,
251 .mmap = lima_gem_mmap,
254 static struct drm_driver lima_drm_driver = {
255 .driver_features = DRIVER_RENDER | DRIVER_GEM | DRIVER_PRIME | DRIVER_SYNCOBJ,
256 .open = lima_drm_driver_open,
257 .postclose = lima_drm_driver_postclose,
258 .ioctls = lima_drm_driver_ioctls,
259 .num_ioctls = ARRAY_SIZE(lima_drm_driver_ioctls),
260 .fops = &lima_drm_driver_fops,
261 .gem_free_object_unlocked = lima_gem_free_object,
262 .gem_open_object = lima_gem_object_open,
263 .gem_close_object = lima_gem_object_close,
264 .gem_vm_ops = &lima_gem_vm_ops,
272 .prime_fd_to_handle = drm_gem_prime_fd_to_handle,
273 .gem_prime_import_sg_table = lima_gem_prime_import_sg_table,
274 .prime_handle_to_fd = drm_gem_prime_handle_to_fd,
275 .gem_prime_get_sg_table = lima_gem_prime_get_sg_table,
276 .gem_prime_mmap = lima_gem_prime_mmap,
279 static int lima_pdev_probe(struct platform_device *pdev)
281 struct lima_device *ldev;
282 struct drm_device *ddev;
285 err = lima_sched_slab_init();
289 ldev = devm_kzalloc(&pdev->dev, sizeof(*ldev), GFP_KERNEL);
296 ldev->dev = &pdev->dev;
297 ldev->id = (enum lima_gpu_id)of_device_get_match_data(&pdev->dev);
299 platform_set_drvdata(pdev, ldev);
301 /* Allocate and initialize the DRM device. */
302 ddev = drm_dev_alloc(&lima_drm_driver, &pdev->dev);
304 return PTR_ERR(ddev);
306 ddev->dev_private = ldev;
309 err = lima_device_init(ldev);
311 dev_err(&pdev->dev, "Fatal error during GPU init\n");
316 * Register the DRM device with the core and the connectors with
319 err = drm_dev_register(ddev, 0);
326 lima_device_fini(ldev);
330 lima_sched_slab_fini();
334 static int lima_pdev_remove(struct platform_device *pdev)
336 struct lima_device *ldev = platform_get_drvdata(pdev);
337 struct drm_device *ddev = ldev->ddev;
339 drm_dev_unregister(ddev);
340 lima_device_fini(ldev);
342 lima_sched_slab_fini();
346 static const struct of_device_id dt_match[] = {
347 { .compatible = "arm,mali-400", .data = (void *)lima_gpu_mali400 },
348 { .compatible = "arm,mali-450", .data = (void *)lima_gpu_mali450 },
351 MODULE_DEVICE_TABLE(of, dt_match);
353 static struct platform_driver lima_platform_driver = {
354 .probe = lima_pdev_probe,
355 .remove = lima_pdev_remove,
358 .of_match_table = dt_match,
362 static int __init lima_init(void)
364 return platform_driver_register(&lima_platform_driver);
366 module_init(lima_init);
368 static void __exit lima_exit(void)
370 platform_driver_unregister(&lima_platform_driver);
372 module_exit(lima_exit);
374 MODULE_AUTHOR("Lima Project Developers");
375 MODULE_DESCRIPTION("Lima DRM Driver");
376 MODULE_LICENSE("GPL v2");