mailbox: mediatek: Add check for possible failure of kzalloc
[sfrench/cifs-2.6.git] / drivers / mailbox / mtk-cmdq-mailbox.c
1 // SPDX-License-Identifier: GPL-2.0
2 //
3 // Copyright (c) 2018 MediaTek Inc.
4
5 #include <linux/bitops.h>
6 #include <linux/clk.h>
7 #include <linux/clk-provider.h>
8 #include <linux/dma-mapping.h>
9 #include <linux/errno.h>
10 #include <linux/interrupt.h>
11 #include <linux/iopoll.h>
12 #include <linux/kernel.h>
13 #include <linux/module.h>
14 #include <linux/platform_device.h>
15 #include <linux/mailbox_controller.h>
16 #include <linux/mailbox/mtk-cmdq-mailbox.h>
17 #include <linux/of_device.h>
18
19 #define CMDQ_OP_CODE_MASK               (0xff << CMDQ_OP_CODE_SHIFT)
20 #define CMDQ_IRQ_MASK                   0xffff
21 #define CMDQ_NUM_CMD(t)                 (t->cmd_buf_size / CMDQ_INST_SIZE)
22
23 #define CMDQ_CURR_IRQ_STATUS            0x10
24 #define CMDQ_THR_SLOT_CYCLES            0x30
25 #define CMDQ_THR_BASE                   0x100
26 #define CMDQ_THR_SIZE                   0x80
27 #define CMDQ_THR_WARM_RESET             0x00
28 #define CMDQ_THR_ENABLE_TASK            0x04
29 #define CMDQ_THR_SUSPEND_TASK           0x08
30 #define CMDQ_THR_CURR_STATUS            0x0c
31 #define CMDQ_THR_IRQ_STATUS             0x10
32 #define CMDQ_THR_IRQ_ENABLE             0x14
33 #define CMDQ_THR_CURR_ADDR              0x20
34 #define CMDQ_THR_END_ADDR               0x24
35 #define CMDQ_THR_WAIT_TOKEN             0x30
36 #define CMDQ_THR_PRIORITY               0x40
37
38 #define CMDQ_THR_ACTIVE_SLOT_CYCLES     0x3200
39 #define CMDQ_THR_ENABLED                0x1
40 #define CMDQ_THR_DISABLED               0x0
41 #define CMDQ_THR_SUSPEND                0x1
42 #define CMDQ_THR_RESUME                 0x0
43 #define CMDQ_THR_STATUS_SUSPENDED       BIT(1)
44 #define CMDQ_THR_DO_WARM_RESET          BIT(0)
45 #define CMDQ_THR_IRQ_DONE               0x1
46 #define CMDQ_THR_IRQ_ERROR              0x12
47 #define CMDQ_THR_IRQ_EN                 (CMDQ_THR_IRQ_ERROR | CMDQ_THR_IRQ_DONE)
48 #define CMDQ_THR_IS_WAITING             BIT(31)
49
50 #define CMDQ_JUMP_BY_OFFSET             0x10000000
51 #define CMDQ_JUMP_BY_PA                 0x10000001
52
53 struct cmdq_thread {
54         struct mbox_chan        *chan;
55         void __iomem            *base;
56         struct list_head        task_busy_list;
57         u32                     priority;
58         bool                    atomic_exec;
59 };
60
61 struct cmdq_task {
62         struct cmdq             *cmdq;
63         struct list_head        list_entry;
64         dma_addr_t              pa_base;
65         struct cmdq_thread      *thread;
66         struct cmdq_pkt         *pkt; /* the packet sent from mailbox client */
67 };
68
69 struct cmdq {
70         struct mbox_controller  mbox;
71         void __iomem            *base;
72         u32                     irq;
73         u32                     thread_nr;
74         struct cmdq_thread      *thread;
75         struct clk              *clock;
76         bool                    suspended;
77 };
78
79 static int cmdq_thread_suspend(struct cmdq *cmdq, struct cmdq_thread *thread)
80 {
81         u32 status;
82
83         writel(CMDQ_THR_SUSPEND, thread->base + CMDQ_THR_SUSPEND_TASK);
84
85         /* If already disabled, treat as suspended successful. */
86         if (!(readl(thread->base + CMDQ_THR_ENABLE_TASK) & CMDQ_THR_ENABLED))
87                 return 0;
88
89         if (readl_poll_timeout_atomic(thread->base + CMDQ_THR_CURR_STATUS,
90                         status, status & CMDQ_THR_STATUS_SUSPENDED, 0, 10)) {
91                 dev_err(cmdq->mbox.dev, "suspend GCE thread 0x%x failed\n",
92                         (u32)(thread->base - cmdq->base));
93                 return -EFAULT;
94         }
95
96         return 0;
97 }
98
99 static void cmdq_thread_resume(struct cmdq_thread *thread)
100 {
101         writel(CMDQ_THR_RESUME, thread->base + CMDQ_THR_SUSPEND_TASK);
102 }
103
104 static void cmdq_init(struct cmdq *cmdq)
105 {
106         WARN_ON(clk_enable(cmdq->clock) < 0);
107         writel(CMDQ_THR_ACTIVE_SLOT_CYCLES, cmdq->base + CMDQ_THR_SLOT_CYCLES);
108         clk_disable(cmdq->clock);
109 }
110
111 static int cmdq_thread_reset(struct cmdq *cmdq, struct cmdq_thread *thread)
112 {
113         u32 warm_reset;
114
115         writel(CMDQ_THR_DO_WARM_RESET, thread->base + CMDQ_THR_WARM_RESET);
116         if (readl_poll_timeout_atomic(thread->base + CMDQ_THR_WARM_RESET,
117                         warm_reset, !(warm_reset & CMDQ_THR_DO_WARM_RESET),
118                         0, 10)) {
119                 dev_err(cmdq->mbox.dev, "reset GCE thread 0x%x failed\n",
120                         (u32)(thread->base - cmdq->base));
121                 return -EFAULT;
122         }
123
124         return 0;
125 }
126
127 static void cmdq_thread_disable(struct cmdq *cmdq, struct cmdq_thread *thread)
128 {
129         cmdq_thread_reset(cmdq, thread);
130         writel(CMDQ_THR_DISABLED, thread->base + CMDQ_THR_ENABLE_TASK);
131 }
132
133 /* notify GCE to re-fetch commands by setting GCE thread PC */
134 static void cmdq_thread_invalidate_fetched_data(struct cmdq_thread *thread)
135 {
136         writel(readl(thread->base + CMDQ_THR_CURR_ADDR),
137                thread->base + CMDQ_THR_CURR_ADDR);
138 }
139
140 static void cmdq_task_insert_into_thread(struct cmdq_task *task)
141 {
142         struct device *dev = task->cmdq->mbox.dev;
143         struct cmdq_thread *thread = task->thread;
144         struct cmdq_task *prev_task = list_last_entry(
145                         &thread->task_busy_list, typeof(*task), list_entry);
146         u64 *prev_task_base = prev_task->pkt->va_base;
147
148         /* let previous task jump to this task */
149         dma_sync_single_for_cpu(dev, prev_task->pa_base,
150                                 prev_task->pkt->cmd_buf_size, DMA_TO_DEVICE);
151         prev_task_base[CMDQ_NUM_CMD(prev_task->pkt) - 1] =
152                 (u64)CMDQ_JUMP_BY_PA << 32 | task->pa_base;
153         dma_sync_single_for_device(dev, prev_task->pa_base,
154                                    prev_task->pkt->cmd_buf_size, DMA_TO_DEVICE);
155
156         cmdq_thread_invalidate_fetched_data(thread);
157 }
158
159 static bool cmdq_command_is_wfe(u64 cmd)
160 {
161         u64 wfe_option = CMDQ_WFE_UPDATE | CMDQ_WFE_WAIT | CMDQ_WFE_WAIT_VALUE;
162         u64 wfe_op = (u64)(CMDQ_CODE_WFE << CMDQ_OP_CODE_SHIFT) << 32;
163         u64 wfe_mask = (u64)CMDQ_OP_CODE_MASK << 32 | 0xffffffff;
164
165         return ((cmd & wfe_mask) == (wfe_op | wfe_option));
166 }
167
168 /* we assume tasks in the same display GCE thread are waiting the same event. */
169 static void cmdq_task_remove_wfe(struct cmdq_task *task)
170 {
171         struct device *dev = task->cmdq->mbox.dev;
172         u64 *base = task->pkt->va_base;
173         int i;
174
175         dma_sync_single_for_cpu(dev, task->pa_base, task->pkt->cmd_buf_size,
176                                 DMA_TO_DEVICE);
177         for (i = 0; i < CMDQ_NUM_CMD(task->pkt); i++)
178                 if (cmdq_command_is_wfe(base[i]))
179                         base[i] = (u64)CMDQ_JUMP_BY_OFFSET << 32 |
180                                   CMDQ_JUMP_PASS;
181         dma_sync_single_for_device(dev, task->pa_base, task->pkt->cmd_buf_size,
182                                    DMA_TO_DEVICE);
183 }
184
185 static bool cmdq_thread_is_in_wfe(struct cmdq_thread *thread)
186 {
187         return readl(thread->base + CMDQ_THR_WAIT_TOKEN) & CMDQ_THR_IS_WAITING;
188 }
189
190 static void cmdq_thread_wait_end(struct cmdq_thread *thread,
191                                  unsigned long end_pa)
192 {
193         struct device *dev = thread->chan->mbox->dev;
194         unsigned long curr_pa;
195
196         if (readl_poll_timeout_atomic(thread->base + CMDQ_THR_CURR_ADDR,
197                         curr_pa, curr_pa == end_pa, 1, 20))
198                 dev_err(dev, "GCE thread cannot run to end.\n");
199 }
200
201 static void cmdq_task_exec_done(struct cmdq_task *task, enum cmdq_cb_status sta)
202 {
203         struct cmdq_task_cb *cb = &task->pkt->async_cb;
204         struct cmdq_cb_data data;
205
206         WARN_ON(cb->cb == (cmdq_async_flush_cb)NULL);
207         data.sta = sta;
208         data.data = cb->data;
209         cb->cb(data);
210
211         list_del(&task->list_entry);
212 }
213
214 static void cmdq_task_handle_error(struct cmdq_task *task)
215 {
216         struct cmdq_thread *thread = task->thread;
217         struct cmdq_task *next_task;
218
219         dev_err(task->cmdq->mbox.dev, "task 0x%p error\n", task);
220         WARN_ON(cmdq_thread_suspend(task->cmdq, thread) < 0);
221         next_task = list_first_entry_or_null(&thread->task_busy_list,
222                         struct cmdq_task, list_entry);
223         if (next_task)
224                 writel(next_task->pa_base, thread->base + CMDQ_THR_CURR_ADDR);
225         cmdq_thread_resume(thread);
226 }
227
228 static void cmdq_thread_irq_handler(struct cmdq *cmdq,
229                                     struct cmdq_thread *thread)
230 {
231         struct cmdq_task *task, *tmp, *curr_task = NULL;
232         u32 curr_pa, irq_flag, task_end_pa;
233         bool err;
234
235         irq_flag = readl(thread->base + CMDQ_THR_IRQ_STATUS);
236         writel(~irq_flag, thread->base + CMDQ_THR_IRQ_STATUS);
237
238         /*
239          * When ISR call this function, another CPU core could run
240          * "release task" right before we acquire the spin lock, and thus
241          * reset / disable this GCE thread, so we need to check the enable
242          * bit of this GCE thread.
243          */
244         if (!(readl(thread->base + CMDQ_THR_ENABLE_TASK) & CMDQ_THR_ENABLED))
245                 return;
246
247         if (irq_flag & CMDQ_THR_IRQ_ERROR)
248                 err = true;
249         else if (irq_flag & CMDQ_THR_IRQ_DONE)
250                 err = false;
251         else
252                 return;
253
254         curr_pa = readl(thread->base + CMDQ_THR_CURR_ADDR);
255
256         list_for_each_entry_safe(task, tmp, &thread->task_busy_list,
257                                  list_entry) {
258                 task_end_pa = task->pa_base + task->pkt->cmd_buf_size;
259                 if (curr_pa >= task->pa_base && curr_pa < task_end_pa)
260                         curr_task = task;
261
262                 if (!curr_task || curr_pa == task_end_pa - CMDQ_INST_SIZE) {
263                         cmdq_task_exec_done(task, CMDQ_CB_NORMAL);
264                         kfree(task);
265                 } else if (err) {
266                         cmdq_task_exec_done(task, CMDQ_CB_ERROR);
267                         cmdq_task_handle_error(curr_task);
268                         kfree(task);
269                 }
270
271                 if (curr_task)
272                         break;
273         }
274
275         if (list_empty(&thread->task_busy_list)) {
276                 cmdq_thread_disable(cmdq, thread);
277                 clk_disable(cmdq->clock);
278         }
279 }
280
281 static irqreturn_t cmdq_irq_handler(int irq, void *dev)
282 {
283         struct cmdq *cmdq = dev;
284         unsigned long irq_status, flags = 0L;
285         int bit;
286
287         irq_status = readl(cmdq->base + CMDQ_CURR_IRQ_STATUS) & CMDQ_IRQ_MASK;
288         if (!(irq_status ^ CMDQ_IRQ_MASK))
289                 return IRQ_NONE;
290
291         for_each_clear_bit(bit, &irq_status, fls(CMDQ_IRQ_MASK)) {
292                 struct cmdq_thread *thread = &cmdq->thread[bit];
293
294                 spin_lock_irqsave(&thread->chan->lock, flags);
295                 cmdq_thread_irq_handler(cmdq, thread);
296                 spin_unlock_irqrestore(&thread->chan->lock, flags);
297         }
298
299         return IRQ_HANDLED;
300 }
301
302 static int cmdq_suspend(struct device *dev)
303 {
304         struct cmdq *cmdq = dev_get_drvdata(dev);
305         struct cmdq_thread *thread;
306         int i;
307         bool task_running = false;
308
309         cmdq->suspended = true;
310
311         for (i = 0; i < cmdq->thread_nr; i++) {
312                 thread = &cmdq->thread[i];
313                 if (!list_empty(&thread->task_busy_list)) {
314                         task_running = true;
315                         break;
316                 }
317         }
318
319         if (task_running)
320                 dev_warn(dev, "exist running task(s) in suspend\n");
321
322         clk_unprepare(cmdq->clock);
323
324         return 0;
325 }
326
327 static int cmdq_resume(struct device *dev)
328 {
329         struct cmdq *cmdq = dev_get_drvdata(dev);
330
331         WARN_ON(clk_prepare(cmdq->clock) < 0);
332         cmdq->suspended = false;
333         return 0;
334 }
335
336 static int cmdq_remove(struct platform_device *pdev)
337 {
338         struct cmdq *cmdq = platform_get_drvdata(pdev);
339
340         mbox_controller_unregister(&cmdq->mbox);
341         clk_unprepare(cmdq->clock);
342
343         if (cmdq->mbox.chans)
344                 devm_kfree(&pdev->dev, cmdq->mbox.chans);
345
346         if (cmdq->thread)
347                 devm_kfree(&pdev->dev, cmdq->thread);
348
349         devm_kfree(&pdev->dev, cmdq);
350
351         return 0;
352 }
353
354 static int cmdq_mbox_send_data(struct mbox_chan *chan, void *data)
355 {
356         struct cmdq_pkt *pkt = (struct cmdq_pkt *)data;
357         struct cmdq_thread *thread = (struct cmdq_thread *)chan->con_priv;
358         struct cmdq *cmdq = dev_get_drvdata(chan->mbox->dev);
359         struct cmdq_task *task;
360         unsigned long curr_pa, end_pa;
361
362         /* Client should not flush new tasks if suspended. */
363         WARN_ON(cmdq->suspended);
364
365         task = kzalloc(sizeof(*task), GFP_ATOMIC);
366         if (!task)
367                 return -ENOMEM;
368
369         task->cmdq = cmdq;
370         INIT_LIST_HEAD(&task->list_entry);
371         task->pa_base = pkt->pa_base;
372         task->thread = thread;
373         task->pkt = pkt;
374
375         if (list_empty(&thread->task_busy_list)) {
376                 WARN_ON(clk_enable(cmdq->clock) < 0);
377                 WARN_ON(cmdq_thread_reset(cmdq, thread) < 0);
378
379                 writel(task->pa_base, thread->base + CMDQ_THR_CURR_ADDR);
380                 writel(task->pa_base + pkt->cmd_buf_size,
381                        thread->base + CMDQ_THR_END_ADDR);
382                 writel(thread->priority, thread->base + CMDQ_THR_PRIORITY);
383                 writel(CMDQ_THR_IRQ_EN, thread->base + CMDQ_THR_IRQ_ENABLE);
384                 writel(CMDQ_THR_ENABLED, thread->base + CMDQ_THR_ENABLE_TASK);
385         } else {
386                 WARN_ON(cmdq_thread_suspend(cmdq, thread) < 0);
387                 curr_pa = readl(thread->base + CMDQ_THR_CURR_ADDR);
388                 end_pa = readl(thread->base + CMDQ_THR_END_ADDR);
389
390                 /*
391                  * Atomic execution should remove the following wfe, i.e. only
392                  * wait event at first task, and prevent to pause when running.
393                  */
394                 if (thread->atomic_exec) {
395                         /* GCE is executing if command is not WFE */
396                         if (!cmdq_thread_is_in_wfe(thread)) {
397                                 cmdq_thread_resume(thread);
398                                 cmdq_thread_wait_end(thread, end_pa);
399                                 WARN_ON(cmdq_thread_suspend(cmdq, thread) < 0);
400                                 /* set to this task directly */
401                                 writel(task->pa_base,
402                                        thread->base + CMDQ_THR_CURR_ADDR);
403                         } else {
404                                 cmdq_task_insert_into_thread(task);
405                                 cmdq_task_remove_wfe(task);
406                                 smp_mb(); /* modify jump before enable thread */
407                         }
408                 } else {
409                         /* check boundary */
410                         if (curr_pa == end_pa - CMDQ_INST_SIZE ||
411                             curr_pa == end_pa) {
412                                 /* set to this task directly */
413                                 writel(task->pa_base,
414                                        thread->base + CMDQ_THR_CURR_ADDR);
415                         } else {
416                                 cmdq_task_insert_into_thread(task);
417                                 smp_mb(); /* modify jump before enable thread */
418                         }
419                 }
420                 writel(task->pa_base + pkt->cmd_buf_size,
421                        thread->base + CMDQ_THR_END_ADDR);
422                 cmdq_thread_resume(thread);
423         }
424         list_move_tail(&task->list_entry, &thread->task_busy_list);
425
426         return 0;
427 }
428
429 static int cmdq_mbox_startup(struct mbox_chan *chan)
430 {
431         return 0;
432 }
433
434 static void cmdq_mbox_shutdown(struct mbox_chan *chan)
435 {
436 }
437
438 static const struct mbox_chan_ops cmdq_mbox_chan_ops = {
439         .send_data = cmdq_mbox_send_data,
440         .startup = cmdq_mbox_startup,
441         .shutdown = cmdq_mbox_shutdown,
442 };
443
444 static struct mbox_chan *cmdq_xlate(struct mbox_controller *mbox,
445                 const struct of_phandle_args *sp)
446 {
447         int ind = sp->args[0];
448         struct cmdq_thread *thread;
449
450         if (ind >= mbox->num_chans)
451                 return ERR_PTR(-EINVAL);
452
453         thread = (struct cmdq_thread *)mbox->chans[ind].con_priv;
454         thread->priority = sp->args[1];
455         thread->atomic_exec = (sp->args[2] != 0);
456         thread->chan = &mbox->chans[ind];
457
458         return &mbox->chans[ind];
459 }
460
461 static int cmdq_probe(struct platform_device *pdev)
462 {
463         struct device *dev = &pdev->dev;
464         struct resource *res;
465         struct cmdq *cmdq;
466         int err, i;
467
468         cmdq = devm_kzalloc(dev, sizeof(*cmdq), GFP_KERNEL);
469         if (!cmdq)
470                 return -ENOMEM;
471
472         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
473         cmdq->base = devm_ioremap_resource(dev, res);
474         if (IS_ERR(cmdq->base)) {
475                 dev_err(dev, "failed to ioremap gce\n");
476                 return PTR_ERR(cmdq->base);
477         }
478
479         cmdq->irq = platform_get_irq(pdev, 0);
480         if (!cmdq->irq) {
481                 dev_err(dev, "failed to get irq\n");
482                 return -EINVAL;
483         }
484         err = devm_request_irq(dev, cmdq->irq, cmdq_irq_handler, IRQF_SHARED,
485                                "mtk_cmdq", cmdq);
486         if (err < 0) {
487                 dev_err(dev, "failed to register ISR (%d)\n", err);
488                 return err;
489         }
490
491         dev_dbg(dev, "cmdq device: addr:0x%p, va:0x%p, irq:%d\n",
492                 dev, cmdq->base, cmdq->irq);
493
494         cmdq->clock = devm_clk_get(dev, "gce");
495         if (IS_ERR(cmdq->clock)) {
496                 dev_err(dev, "failed to get gce clk\n");
497                 return PTR_ERR(cmdq->clock);
498         }
499
500         cmdq->thread_nr = (u32)(unsigned long)of_device_get_match_data(dev);
501         cmdq->mbox.dev = dev;
502         cmdq->mbox.chans = devm_kcalloc(dev, cmdq->thread_nr,
503                                         sizeof(*cmdq->mbox.chans), GFP_KERNEL);
504         if (!cmdq->mbox.chans)
505                 return -ENOMEM;
506
507         cmdq->mbox.num_chans = cmdq->thread_nr;
508         cmdq->mbox.ops = &cmdq_mbox_chan_ops;
509         cmdq->mbox.of_xlate = cmdq_xlate;
510
511         /* make use of TXDONE_BY_ACK */
512         cmdq->mbox.txdone_irq = false;
513         cmdq->mbox.txdone_poll = false;
514
515         cmdq->thread = devm_kcalloc(dev, cmdq->thread_nr,
516                                         sizeof(*cmdq->thread), GFP_KERNEL);
517         if (!cmdq->thread)
518                 return -ENOMEM;
519
520         for (i = 0; i < cmdq->thread_nr; i++) {
521                 cmdq->thread[i].base = cmdq->base + CMDQ_THR_BASE +
522                                 CMDQ_THR_SIZE * i;
523                 INIT_LIST_HEAD(&cmdq->thread[i].task_busy_list);
524                 cmdq->mbox.chans[i].con_priv = (void *)&cmdq->thread[i];
525         }
526
527         err = mbox_controller_register(&cmdq->mbox);
528         if (err < 0) {
529                 dev_err(dev, "failed to register mailbox: %d\n", err);
530                 return err;
531         }
532
533         platform_set_drvdata(pdev, cmdq);
534         WARN_ON(clk_prepare(cmdq->clock) < 0);
535
536         cmdq_init(cmdq);
537
538         return 0;
539 }
540
541 static const struct dev_pm_ops cmdq_pm_ops = {
542         .suspend = cmdq_suspend,
543         .resume = cmdq_resume,
544 };
545
546 static const struct of_device_id cmdq_of_ids[] = {
547         {.compatible = "mediatek,mt8173-gce", .data = (void *)16},
548         {}
549 };
550
551 static struct platform_driver cmdq_drv = {
552         .probe = cmdq_probe,
553         .remove = cmdq_remove,
554         .driver = {
555                 .name = "mtk_cmdq",
556                 .pm = &cmdq_pm_ops,
557                 .of_match_table = cmdq_of_ids,
558         }
559 };
560
561 static int __init cmdq_drv_init(void)
562 {
563         return platform_driver_register(&cmdq_drv);
564 }
565
566 static void __exit cmdq_drv_exit(void)
567 {
568         platform_driver_unregister(&cmdq_drv);
569 }
570
571 subsys_initcall(cmdq_drv_init);
572 module_exit(cmdq_drv_exit);
573
574 MODULE_LICENSE("GPL v2");