xref: /openbmc/linux/drivers/mailbox/tegra-hsp.c (revision ded1ffea)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2016-2023, NVIDIA CORPORATION.  All rights reserved.
4  */
5 
6 #include <linux/delay.h>
7 #include <linux/interrupt.h>
8 #include <linux/io.h>
9 #include <linux/mailbox_controller.h>
10 #include <linux/of.h>
11 #include <linux/platform_device.h>
12 #include <linux/pm.h>
13 #include <linux/slab.h>
14 
15 #include <soc/tegra/fuse.h>
16 
17 #include <dt-bindings/mailbox/tegra186-hsp.h>
18 
19 #include "mailbox.h"
20 
21 #define HSP_INT_IE(x)		(0x100 + ((x) * 4))
22 #define HSP_INT_IV		0x300
23 #define HSP_INT_IR		0x304
24 
25 #define HSP_INT_EMPTY_SHIFT	0
26 #define HSP_INT_EMPTY_MASK	0xff
27 #define HSP_INT_FULL_SHIFT	8
28 #define HSP_INT_FULL_MASK	0xff
29 
30 #define HSP_INT_DIMENSIONING	0x380
31 #define HSP_nSM_SHIFT		0
32 #define HSP_nSS_SHIFT		4
33 #define HSP_nAS_SHIFT		8
34 #define HSP_nDB_SHIFT		12
35 #define HSP_nSI_SHIFT		16
36 #define HSP_nINT_MASK		0xf
37 
38 #define HSP_DB_TRIGGER	0x0
39 #define HSP_DB_ENABLE	0x4
40 #define HSP_DB_RAW	0x8
41 #define HSP_DB_PENDING	0xc
42 
43 #define HSP_SM_SHRD_MBOX	0x0
44 #define HSP_SM_SHRD_MBOX_FULL	BIT(31)
45 #define HSP_SM_SHRD_MBOX_FULL_INT_IE	0x04
46 #define HSP_SM_SHRD_MBOX_EMPTY_INT_IE	0x08
47 
48 #define HSP_SHRD_MBOX_TYPE1_TAG		0x40
49 #define HSP_SHRD_MBOX_TYPE1_DATA0	0x48
50 #define HSP_SHRD_MBOX_TYPE1_DATA1	0x4c
51 #define HSP_SHRD_MBOX_TYPE1_DATA2	0x50
52 #define HSP_SHRD_MBOX_TYPE1_DATA3	0x54
53 
54 #define HSP_DB_CCPLEX		1
55 #define HSP_DB_BPMP		3
56 #define HSP_DB_MAX		7
57 
58 #define HSP_MBOX_TYPE_MASK	0xff
59 
60 struct tegra_hsp_channel;
61 struct tegra_hsp;
62 
63 struct tegra_hsp_channel {
64 	struct tegra_hsp *hsp;
65 	struct mbox_chan *chan;
66 	void __iomem *regs;
67 };
68 
69 struct tegra_hsp_doorbell {
70 	struct tegra_hsp_channel channel;
71 	struct list_head list;
72 	const char *name;
73 	unsigned int master;
74 	unsigned int index;
75 };
76 
77 struct tegra_hsp_sm_ops {
78 	void (*send)(struct tegra_hsp_channel *channel, void *data);
79 	void (*recv)(struct tegra_hsp_channel *channel);
80 };
81 
82 struct tegra_hsp_mailbox {
83 	struct tegra_hsp_channel channel;
84 	const struct tegra_hsp_sm_ops *ops;
85 	unsigned int index;
86 	bool producer;
87 };
88 
89 struct tegra_hsp_db_map {
90 	const char *name;
91 	unsigned int master;
92 	unsigned int index;
93 };
94 
95 struct tegra_hsp_soc {
96 	const struct tegra_hsp_db_map *map;
97 	bool has_per_mb_ie;
98 	bool has_128_bit_mb;
99 	unsigned int reg_stride;
100 };
101 
102 struct tegra_hsp {
103 	struct device *dev;
104 	const struct tegra_hsp_soc *soc;
105 	struct mbox_controller mbox_db;
106 	struct mbox_controller mbox_sm;
107 	void __iomem *regs;
108 	unsigned int doorbell_irq;
109 	unsigned int *shared_irqs;
110 	unsigned int shared_irq;
111 	unsigned int num_sm;
112 	unsigned int num_as;
113 	unsigned int num_ss;
114 	unsigned int num_db;
115 	unsigned int num_si;
116 
117 	spinlock_t lock;
118 	struct lock_class_key lock_key;
119 
120 	struct list_head doorbells;
121 	struct tegra_hsp_mailbox *mailboxes;
122 
123 	unsigned long mask;
124 };
125 
126 static inline u32 tegra_hsp_readl(struct tegra_hsp *hsp, unsigned int offset)
127 {
128 	return readl(hsp->regs + offset);
129 }
130 
131 static inline void tegra_hsp_writel(struct tegra_hsp *hsp, u32 value,
132 				    unsigned int offset)
133 {
134 	writel(value, hsp->regs + offset);
135 }
136 
137 static inline u32 tegra_hsp_channel_readl(struct tegra_hsp_channel *channel,
138 					  unsigned int offset)
139 {
140 	return readl(channel->regs + offset);
141 }
142 
143 static inline void tegra_hsp_channel_writel(struct tegra_hsp_channel *channel,
144 					    u32 value, unsigned int offset)
145 {
146 	writel(value, channel->regs + offset);
147 }
148 
149 static bool tegra_hsp_doorbell_can_ring(struct tegra_hsp_doorbell *db)
150 {
151 	u32 value;
152 
153 	value = tegra_hsp_channel_readl(&db->channel, HSP_DB_ENABLE);
154 
155 	return (value & BIT(TEGRA_HSP_DB_MASTER_CCPLEX)) != 0;
156 }
157 
158 static struct tegra_hsp_doorbell *
159 __tegra_hsp_doorbell_get(struct tegra_hsp *hsp, unsigned int master)
160 {
161 	struct tegra_hsp_doorbell *entry;
162 
163 	list_for_each_entry(entry, &hsp->doorbells, list)
164 		if (entry->master == master)
165 			return entry;
166 
167 	return NULL;
168 }
169 
170 static struct tegra_hsp_doorbell *
171 tegra_hsp_doorbell_get(struct tegra_hsp *hsp, unsigned int master)
172 {
173 	struct tegra_hsp_doorbell *db;
174 	unsigned long flags;
175 
176 	spin_lock_irqsave(&hsp->lock, flags);
177 	db = __tegra_hsp_doorbell_get(hsp, master);
178 	spin_unlock_irqrestore(&hsp->lock, flags);
179 
180 	return db;
181 }
182 
183 static irqreturn_t tegra_hsp_doorbell_irq(int irq, void *data)
184 {
185 	struct tegra_hsp *hsp = data;
186 	struct tegra_hsp_doorbell *db;
187 	unsigned long master, value;
188 
189 	db = tegra_hsp_doorbell_get(hsp, TEGRA_HSP_DB_MASTER_CCPLEX);
190 	if (!db)
191 		return IRQ_NONE;
192 
193 	value = tegra_hsp_channel_readl(&db->channel, HSP_DB_PENDING);
194 	tegra_hsp_channel_writel(&db->channel, value, HSP_DB_PENDING);
195 
196 	spin_lock(&hsp->lock);
197 
198 	for_each_set_bit(master, &value, hsp->mbox_db.num_chans) {
199 		struct tegra_hsp_doorbell *db;
200 
201 		db = __tegra_hsp_doorbell_get(hsp, master);
202 		/*
203 		 * Depending on the bootloader chain, the CCPLEX doorbell will
204 		 * have some doorbells enabled, which means that requesting an
205 		 * interrupt will immediately fire.
206 		 *
207 		 * In that case, db->channel.chan will still be NULL here and
208 		 * cause a crash if not properly guarded.
209 		 *
210 		 * It remains to be seen if ignoring the doorbell in that case
211 		 * is the correct solution.
212 		 */
213 		if (db && db->channel.chan)
214 			mbox_chan_received_data(db->channel.chan, NULL);
215 	}
216 
217 	spin_unlock(&hsp->lock);
218 
219 	return IRQ_HANDLED;
220 }
221 
222 static irqreturn_t tegra_hsp_shared_irq(int irq, void *data)
223 {
224 	struct tegra_hsp *hsp = data;
225 	unsigned long bit, mask;
226 	u32 status;
227 
228 	status = tegra_hsp_readl(hsp, HSP_INT_IR) & hsp->mask;
229 
230 	/* process EMPTY interrupts first */
231 	mask = (status >> HSP_INT_EMPTY_SHIFT) & HSP_INT_EMPTY_MASK;
232 
233 	for_each_set_bit(bit, &mask, hsp->num_sm) {
234 		struct tegra_hsp_mailbox *mb = &hsp->mailboxes[bit];
235 
236 		if (mb->producer) {
237 			/*
238 			 * Disable EMPTY interrupts until data is sent with
239 			 * the next message. These interrupts are level-
240 			 * triggered, so if we kept them enabled they would
241 			 * constantly trigger until we next write data into
242 			 * the message.
243 			 */
244 			spin_lock(&hsp->lock);
245 
246 			hsp->mask &= ~BIT(HSP_INT_EMPTY_SHIFT + mb->index);
247 			tegra_hsp_writel(hsp, hsp->mask,
248 					 HSP_INT_IE(hsp->shared_irq));
249 
250 			spin_unlock(&hsp->lock);
251 
252 			mbox_chan_txdone(mb->channel.chan, 0);
253 		}
254 	}
255 
256 	/* process FULL interrupts */
257 	mask = (status >> HSP_INT_FULL_SHIFT) & HSP_INT_FULL_MASK;
258 
259 	for_each_set_bit(bit, &mask, hsp->num_sm) {
260 		struct tegra_hsp_mailbox *mb = &hsp->mailboxes[bit];
261 
262 		if (!mb->producer)
263 			mb->ops->recv(&mb->channel);
264 	}
265 
266 	return IRQ_HANDLED;
267 }
268 
269 static struct tegra_hsp_channel *
270 tegra_hsp_doorbell_create(struct tegra_hsp *hsp, const char *name,
271 			  unsigned int master, unsigned int index)
272 {
273 	struct tegra_hsp_doorbell *db;
274 	unsigned int offset;
275 	unsigned long flags;
276 
277 	db = devm_kzalloc(hsp->dev, sizeof(*db), GFP_KERNEL);
278 	if (!db)
279 		return ERR_PTR(-ENOMEM);
280 
281 	offset = (1 + (hsp->num_sm / 2) + hsp->num_ss + hsp->num_as) * SZ_64K;
282 	offset += index * hsp->soc->reg_stride;
283 
284 	db->channel.regs = hsp->regs + offset;
285 	db->channel.hsp = hsp;
286 
287 	db->name = devm_kstrdup_const(hsp->dev, name, GFP_KERNEL);
288 	db->master = master;
289 	db->index = index;
290 
291 	spin_lock_irqsave(&hsp->lock, flags);
292 	list_add_tail(&db->list, &hsp->doorbells);
293 	spin_unlock_irqrestore(&hsp->lock, flags);
294 
295 	return &db->channel;
296 }
297 
298 static int tegra_hsp_doorbell_send_data(struct mbox_chan *chan, void *data)
299 {
300 	struct tegra_hsp_doorbell *db = chan->con_priv;
301 
302 	tegra_hsp_channel_writel(&db->channel, 1, HSP_DB_TRIGGER);
303 
304 	return 0;
305 }
306 
307 static int tegra_hsp_doorbell_startup(struct mbox_chan *chan)
308 {
309 	struct tegra_hsp_doorbell *db = chan->con_priv;
310 	struct tegra_hsp *hsp = db->channel.hsp;
311 	struct tegra_hsp_doorbell *ccplex;
312 	unsigned long flags;
313 	u32 value;
314 
315 	if (db->master >= chan->mbox->num_chans) {
316 		dev_err(chan->mbox->dev,
317 			"invalid master ID %u for HSP channel\n",
318 			db->master);
319 		return -EINVAL;
320 	}
321 
322 	ccplex = tegra_hsp_doorbell_get(hsp, TEGRA_HSP_DB_MASTER_CCPLEX);
323 	if (!ccplex)
324 		return -ENODEV;
325 
326 	/*
327 	 * On simulation platforms the BPMP hasn't had a chance yet to mark
328 	 * the doorbell as ringable by the CCPLEX, so we want to skip extra
329 	 * checks here.
330 	 */
331 	if (tegra_is_silicon() && !tegra_hsp_doorbell_can_ring(db))
332 		return -ENODEV;
333 
334 	spin_lock_irqsave(&hsp->lock, flags);
335 
336 	value = tegra_hsp_channel_readl(&ccplex->channel, HSP_DB_ENABLE);
337 	value |= BIT(db->master);
338 	tegra_hsp_channel_writel(&ccplex->channel, value, HSP_DB_ENABLE);
339 
340 	spin_unlock_irqrestore(&hsp->lock, flags);
341 
342 	return 0;
343 }
344 
345 static void tegra_hsp_doorbell_shutdown(struct mbox_chan *chan)
346 {
347 	struct tegra_hsp_doorbell *db = chan->con_priv;
348 	struct tegra_hsp *hsp = db->channel.hsp;
349 	struct tegra_hsp_doorbell *ccplex;
350 	unsigned long flags;
351 	u32 value;
352 
353 	ccplex = tegra_hsp_doorbell_get(hsp, TEGRA_HSP_DB_MASTER_CCPLEX);
354 	if (!ccplex)
355 		return;
356 
357 	spin_lock_irqsave(&hsp->lock, flags);
358 
359 	value = tegra_hsp_channel_readl(&ccplex->channel, HSP_DB_ENABLE);
360 	value &= ~BIT(db->master);
361 	tegra_hsp_channel_writel(&ccplex->channel, value, HSP_DB_ENABLE);
362 
363 	spin_unlock_irqrestore(&hsp->lock, flags);
364 }
365 
366 static const struct mbox_chan_ops tegra_hsp_db_ops = {
367 	.send_data = tegra_hsp_doorbell_send_data,
368 	.startup = tegra_hsp_doorbell_startup,
369 	.shutdown = tegra_hsp_doorbell_shutdown,
370 };
371 
372 static void tegra_hsp_sm_send32(struct tegra_hsp_channel *channel, void *data)
373 {
374 	u32 value;
375 
376 	/* copy data and mark mailbox full */
377 	value = (u32)(unsigned long)data;
378 	value |= HSP_SM_SHRD_MBOX_FULL;
379 
380 	tegra_hsp_channel_writel(channel, value, HSP_SM_SHRD_MBOX);
381 }
382 
383 static void tegra_hsp_sm_recv32(struct tegra_hsp_channel *channel)
384 {
385 	u32 value;
386 	void *msg;
387 
388 	value = tegra_hsp_channel_readl(channel, HSP_SM_SHRD_MBOX);
389 	value &= ~HSP_SM_SHRD_MBOX_FULL;
390 	msg = (void *)(unsigned long)value;
391 	mbox_chan_received_data(channel->chan, msg);
392 
393 	/*
394 	 * Need to clear all bits here since some producers, such as TCU, depend
395 	 * on fields in the register getting cleared by the consumer.
396 	 *
397 	 * The mailbox API doesn't give the consumers a way of doing that
398 	 * explicitly, so we have to make sure we cover all possible cases.
399 	 */
400 	tegra_hsp_channel_writel(channel, 0x0, HSP_SM_SHRD_MBOX);
401 }
402 
403 static const struct tegra_hsp_sm_ops tegra_hsp_sm_32bit_ops = {
404 	.send = tegra_hsp_sm_send32,
405 	.recv = tegra_hsp_sm_recv32,
406 };
407 
408 static void tegra_hsp_sm_send128(struct tegra_hsp_channel *channel, void *data)
409 {
410 	u32 value[4];
411 
412 	memcpy(value, data, sizeof(value));
413 
414 	/* Copy data */
415 	tegra_hsp_channel_writel(channel, value[0], HSP_SHRD_MBOX_TYPE1_DATA0);
416 	tegra_hsp_channel_writel(channel, value[1], HSP_SHRD_MBOX_TYPE1_DATA1);
417 	tegra_hsp_channel_writel(channel, value[2], HSP_SHRD_MBOX_TYPE1_DATA2);
418 	tegra_hsp_channel_writel(channel, value[3], HSP_SHRD_MBOX_TYPE1_DATA3);
419 
420 	/* Update tag to mark mailbox full */
421 	tegra_hsp_channel_writel(channel, HSP_SM_SHRD_MBOX_FULL,
422 				 HSP_SHRD_MBOX_TYPE1_TAG);
423 }
424 
425 static void tegra_hsp_sm_recv128(struct tegra_hsp_channel *channel)
426 {
427 	u32 value[4];
428 	void *msg;
429 
430 	value[0] = tegra_hsp_channel_readl(channel, HSP_SHRD_MBOX_TYPE1_DATA0);
431 	value[1] = tegra_hsp_channel_readl(channel, HSP_SHRD_MBOX_TYPE1_DATA1);
432 	value[2] = tegra_hsp_channel_readl(channel, HSP_SHRD_MBOX_TYPE1_DATA2);
433 	value[3] = tegra_hsp_channel_readl(channel, HSP_SHRD_MBOX_TYPE1_DATA3);
434 
435 	msg = (void *)(unsigned long)value;
436 	mbox_chan_received_data(channel->chan, msg);
437 
438 	/*
439 	 * Clear data registers and tag.
440 	 */
441 	tegra_hsp_channel_writel(channel, 0x0, HSP_SHRD_MBOX_TYPE1_DATA0);
442 	tegra_hsp_channel_writel(channel, 0x0, HSP_SHRD_MBOX_TYPE1_DATA1);
443 	tegra_hsp_channel_writel(channel, 0x0, HSP_SHRD_MBOX_TYPE1_DATA2);
444 	tegra_hsp_channel_writel(channel, 0x0, HSP_SHRD_MBOX_TYPE1_DATA3);
445 	tegra_hsp_channel_writel(channel, 0x0, HSP_SHRD_MBOX_TYPE1_TAG);
446 }
447 
448 static const struct tegra_hsp_sm_ops tegra_hsp_sm_128bit_ops = {
449 	.send = tegra_hsp_sm_send128,
450 	.recv = tegra_hsp_sm_recv128,
451 };
452 
453 static int tegra_hsp_mailbox_send_data(struct mbox_chan *chan, void *data)
454 {
455 	struct tegra_hsp_mailbox *mb = chan->con_priv;
456 	struct tegra_hsp *hsp = mb->channel.hsp;
457 	unsigned long flags;
458 
459 	if (WARN_ON(!mb->producer))
460 		return -EPERM;
461 
462 	mb->ops->send(&mb->channel, data);
463 
464 	/* enable EMPTY interrupt for the shared mailbox */
465 	spin_lock_irqsave(&hsp->lock, flags);
466 
467 	hsp->mask |= BIT(HSP_INT_EMPTY_SHIFT + mb->index);
468 	tegra_hsp_writel(hsp, hsp->mask, HSP_INT_IE(hsp->shared_irq));
469 
470 	spin_unlock_irqrestore(&hsp->lock, flags);
471 
472 	return 0;
473 }
474 
475 static int tegra_hsp_mailbox_flush(struct mbox_chan *chan,
476 				   unsigned long timeout)
477 {
478 	struct tegra_hsp_mailbox *mb = chan->con_priv;
479 	struct tegra_hsp_channel *ch = &mb->channel;
480 	u32 value;
481 
482 	timeout = jiffies + msecs_to_jiffies(timeout);
483 
484 	while (time_before(jiffies, timeout)) {
485 		value = tegra_hsp_channel_readl(ch, HSP_SM_SHRD_MBOX);
486 		if ((value & HSP_SM_SHRD_MBOX_FULL) == 0) {
487 			mbox_chan_txdone(chan, 0);
488 
489 			/* Wait until channel is empty */
490 			if (chan->active_req != NULL)
491 				continue;
492 
493 			return 0;
494 		}
495 
496 		udelay(1);
497 	}
498 
499 	return -ETIME;
500 }
501 
502 static int tegra_hsp_mailbox_startup(struct mbox_chan *chan)
503 {
504 	struct tegra_hsp_mailbox *mb = chan->con_priv;
505 	struct tegra_hsp_channel *ch = &mb->channel;
506 	struct tegra_hsp *hsp = mb->channel.hsp;
507 	unsigned long flags;
508 
509 	chan->txdone_method = TXDONE_BY_IRQ;
510 
511 	/*
512 	 * Shared mailboxes start out as consumers by default. FULL and EMPTY
513 	 * interrupts are coalesced at the same shared interrupt.
514 	 *
515 	 * Keep EMPTY interrupts disabled at startup and only enable them when
516 	 * the mailbox is actually full. This is required because the FULL and
517 	 * EMPTY interrupts are level-triggered, so keeping EMPTY interrupts
518 	 * enabled all the time would cause an interrupt storm while mailboxes
519 	 * are idle.
520 	 */
521 
522 	spin_lock_irqsave(&hsp->lock, flags);
523 
524 	if (mb->producer)
525 		hsp->mask &= ~BIT(HSP_INT_EMPTY_SHIFT + mb->index);
526 	else
527 		hsp->mask |= BIT(HSP_INT_FULL_SHIFT + mb->index);
528 
529 	tegra_hsp_writel(hsp, hsp->mask, HSP_INT_IE(hsp->shared_irq));
530 
531 	spin_unlock_irqrestore(&hsp->lock, flags);
532 
533 	if (hsp->soc->has_per_mb_ie) {
534 		if (mb->producer)
535 			tegra_hsp_channel_writel(ch, 0x0,
536 						 HSP_SM_SHRD_MBOX_EMPTY_INT_IE);
537 		else
538 			tegra_hsp_channel_writel(ch, 0x1,
539 						 HSP_SM_SHRD_MBOX_FULL_INT_IE);
540 	}
541 
542 	return 0;
543 }
544 
545 static void tegra_hsp_mailbox_shutdown(struct mbox_chan *chan)
546 {
547 	struct tegra_hsp_mailbox *mb = chan->con_priv;
548 	struct tegra_hsp_channel *ch = &mb->channel;
549 	struct tegra_hsp *hsp = mb->channel.hsp;
550 	unsigned long flags;
551 
552 	if (hsp->soc->has_per_mb_ie) {
553 		if (mb->producer)
554 			tegra_hsp_channel_writel(ch, 0x0,
555 						 HSP_SM_SHRD_MBOX_EMPTY_INT_IE);
556 		else
557 			tegra_hsp_channel_writel(ch, 0x0,
558 						 HSP_SM_SHRD_MBOX_FULL_INT_IE);
559 	}
560 
561 	spin_lock_irqsave(&hsp->lock, flags);
562 
563 	if (mb->producer)
564 		hsp->mask &= ~BIT(HSP_INT_EMPTY_SHIFT + mb->index);
565 	else
566 		hsp->mask &= ~BIT(HSP_INT_FULL_SHIFT + mb->index);
567 
568 	tegra_hsp_writel(hsp, hsp->mask, HSP_INT_IE(hsp->shared_irq));
569 
570 	spin_unlock_irqrestore(&hsp->lock, flags);
571 }
572 
573 static const struct mbox_chan_ops tegra_hsp_sm_ops = {
574 	.send_data = tegra_hsp_mailbox_send_data,
575 	.flush = tegra_hsp_mailbox_flush,
576 	.startup = tegra_hsp_mailbox_startup,
577 	.shutdown = tegra_hsp_mailbox_shutdown,
578 };
579 
580 static struct mbox_chan *tegra_hsp_db_xlate(struct mbox_controller *mbox,
581 					    const struct of_phandle_args *args)
582 {
583 	struct tegra_hsp *hsp = container_of(mbox, struct tegra_hsp, mbox_db);
584 	unsigned int type = args->args[0], master = args->args[1];
585 	struct tegra_hsp_channel *channel = ERR_PTR(-ENODEV);
586 	struct tegra_hsp_doorbell *db;
587 	struct mbox_chan *chan;
588 	unsigned long flags;
589 	unsigned int i;
590 
591 	if (type != TEGRA_HSP_MBOX_TYPE_DB || !hsp->doorbell_irq)
592 		return ERR_PTR(-ENODEV);
593 
594 	db = tegra_hsp_doorbell_get(hsp, master);
595 	if (db)
596 		channel = &db->channel;
597 
598 	if (IS_ERR(channel))
599 		return ERR_CAST(channel);
600 
601 	spin_lock_irqsave(&hsp->lock, flags);
602 
603 	for (i = 0; i < mbox->num_chans; i++) {
604 		chan = &mbox->chans[i];
605 		if (!chan->con_priv) {
606 			channel->chan = chan;
607 			chan->con_priv = db;
608 			break;
609 		}
610 
611 		chan = NULL;
612 	}
613 
614 	spin_unlock_irqrestore(&hsp->lock, flags);
615 
616 	return chan ?: ERR_PTR(-EBUSY);
617 }
618 
619 static struct mbox_chan *tegra_hsp_sm_xlate(struct mbox_controller *mbox,
620 					    const struct of_phandle_args *args)
621 {
622 	struct tegra_hsp *hsp = container_of(mbox, struct tegra_hsp, mbox_sm);
623 	unsigned int type = args->args[0], index;
624 	struct tegra_hsp_mailbox *mb;
625 
626 	index = args->args[1] & TEGRA_HSP_SM_MASK;
627 
628 	if ((type & HSP_MBOX_TYPE_MASK) != TEGRA_HSP_MBOX_TYPE_SM ||
629 	    !hsp->shared_irqs || index >= hsp->num_sm)
630 		return ERR_PTR(-ENODEV);
631 
632 	mb = &hsp->mailboxes[index];
633 
634 	if (type & TEGRA_HSP_MBOX_TYPE_SM_128BIT) {
635 		if (!hsp->soc->has_128_bit_mb)
636 			return ERR_PTR(-ENODEV);
637 
638 		mb->ops = &tegra_hsp_sm_128bit_ops;
639 	} else {
640 		mb->ops = &tegra_hsp_sm_32bit_ops;
641 	}
642 
643 	if ((args->args[1] & TEGRA_HSP_SM_FLAG_TX) == 0)
644 		mb->producer = false;
645 	else
646 		mb->producer = true;
647 
648 	return mb->channel.chan;
649 }
650 
651 static int tegra_hsp_add_doorbells(struct tegra_hsp *hsp)
652 {
653 	const struct tegra_hsp_db_map *map = hsp->soc->map;
654 	struct tegra_hsp_channel *channel;
655 
656 	while (map->name) {
657 		channel = tegra_hsp_doorbell_create(hsp, map->name,
658 						    map->master, map->index);
659 		if (IS_ERR(channel))
660 			return PTR_ERR(channel);
661 
662 		map++;
663 	}
664 
665 	return 0;
666 }
667 
668 static int tegra_hsp_add_mailboxes(struct tegra_hsp *hsp, struct device *dev)
669 {
670 	int i;
671 
672 	hsp->mailboxes = devm_kcalloc(dev, hsp->num_sm, sizeof(*hsp->mailboxes),
673 				      GFP_KERNEL);
674 	if (!hsp->mailboxes)
675 		return -ENOMEM;
676 
677 	for (i = 0; i < hsp->num_sm; i++) {
678 		struct tegra_hsp_mailbox *mb = &hsp->mailboxes[i];
679 
680 		mb->index = i;
681 
682 		mb->channel.hsp = hsp;
683 		mb->channel.regs = hsp->regs + SZ_64K + i * SZ_32K;
684 		mb->channel.chan = &hsp->mbox_sm.chans[i];
685 		mb->channel.chan->con_priv = mb;
686 	}
687 
688 	return 0;
689 }
690 
691 static int tegra_hsp_request_shared_irq(struct tegra_hsp *hsp)
692 {
693 	unsigned int i, irq = 0;
694 	int err;
695 
696 	for (i = 0; i < hsp->num_si; i++) {
697 		irq = hsp->shared_irqs[i];
698 		if (irq <= 0)
699 			continue;
700 
701 		err = devm_request_irq(hsp->dev, irq, tegra_hsp_shared_irq, 0,
702 				       dev_name(hsp->dev), hsp);
703 		if (err < 0) {
704 			dev_err(hsp->dev, "failed to request interrupt: %d\n",
705 				err);
706 			continue;
707 		}
708 
709 		hsp->shared_irq = i;
710 
711 		/* disable all interrupts */
712 		tegra_hsp_writel(hsp, 0, HSP_INT_IE(hsp->shared_irq));
713 
714 		dev_dbg(hsp->dev, "interrupt requested: %u\n", irq);
715 
716 		break;
717 	}
718 
719 	if (i == hsp->num_si) {
720 		dev_err(hsp->dev, "failed to find available interrupt\n");
721 		return -ENOENT;
722 	}
723 
724 	return 0;
725 }
726 
727 static int tegra_hsp_probe(struct platform_device *pdev)
728 {
729 	struct tegra_hsp *hsp;
730 	unsigned int i;
731 	u32 value;
732 	int err;
733 
734 	hsp = devm_kzalloc(&pdev->dev, sizeof(*hsp), GFP_KERNEL);
735 	if (!hsp)
736 		return -ENOMEM;
737 
738 	hsp->dev = &pdev->dev;
739 	hsp->soc = of_device_get_match_data(&pdev->dev);
740 	INIT_LIST_HEAD(&hsp->doorbells);
741 	spin_lock_init(&hsp->lock);
742 
743 	hsp->regs = devm_platform_ioremap_resource(pdev, 0);
744 	if (IS_ERR(hsp->regs))
745 		return PTR_ERR(hsp->regs);
746 
747 	value = tegra_hsp_readl(hsp, HSP_INT_DIMENSIONING);
748 	hsp->num_sm = (value >> HSP_nSM_SHIFT) & HSP_nINT_MASK;
749 	hsp->num_ss = (value >> HSP_nSS_SHIFT) & HSP_nINT_MASK;
750 	hsp->num_as = (value >> HSP_nAS_SHIFT) & HSP_nINT_MASK;
751 	hsp->num_db = (value >> HSP_nDB_SHIFT) & HSP_nINT_MASK;
752 	hsp->num_si = (value >> HSP_nSI_SHIFT) & HSP_nINT_MASK;
753 
754 	err = platform_get_irq_byname_optional(pdev, "doorbell");
755 	if (err >= 0)
756 		hsp->doorbell_irq = err;
757 
758 	if (hsp->num_si > 0) {
759 		unsigned int count = 0;
760 
761 		hsp->shared_irqs = devm_kcalloc(&pdev->dev, hsp->num_si,
762 						sizeof(*hsp->shared_irqs),
763 						GFP_KERNEL);
764 		if (!hsp->shared_irqs)
765 			return -ENOMEM;
766 
767 		for (i = 0; i < hsp->num_si; i++) {
768 			char *name;
769 
770 			name = kasprintf(GFP_KERNEL, "shared%u", i);
771 			if (!name)
772 				return -ENOMEM;
773 
774 			err = platform_get_irq_byname_optional(pdev, name);
775 			if (err >= 0) {
776 				hsp->shared_irqs[i] = err;
777 				count++;
778 			}
779 
780 			kfree(name);
781 		}
782 
783 		if (count == 0) {
784 			devm_kfree(&pdev->dev, hsp->shared_irqs);
785 			hsp->shared_irqs = NULL;
786 		}
787 	}
788 
789 	/* setup the doorbell controller */
790 	hsp->mbox_db.of_xlate = tegra_hsp_db_xlate;
791 	hsp->mbox_db.num_chans = 32;
792 	hsp->mbox_db.dev = &pdev->dev;
793 	hsp->mbox_db.ops = &tegra_hsp_db_ops;
794 
795 	hsp->mbox_db.chans = devm_kcalloc(&pdev->dev, hsp->mbox_db.num_chans,
796 					  sizeof(*hsp->mbox_db.chans),
797 					  GFP_KERNEL);
798 	if (!hsp->mbox_db.chans)
799 		return -ENOMEM;
800 
801 	if (hsp->doorbell_irq) {
802 		err = tegra_hsp_add_doorbells(hsp);
803 		if (err < 0) {
804 			dev_err(&pdev->dev, "failed to add doorbells: %d\n",
805 			        err);
806 			return err;
807 		}
808 	}
809 
810 	err = devm_mbox_controller_register(&pdev->dev, &hsp->mbox_db);
811 	if (err < 0) {
812 		dev_err(&pdev->dev, "failed to register doorbell mailbox: %d\n",
813 			err);
814 		return err;
815 	}
816 
817 	/* setup the shared mailbox controller */
818 	hsp->mbox_sm.of_xlate = tegra_hsp_sm_xlate;
819 	hsp->mbox_sm.num_chans = hsp->num_sm;
820 	hsp->mbox_sm.dev = &pdev->dev;
821 	hsp->mbox_sm.ops = &tegra_hsp_sm_ops;
822 
823 	hsp->mbox_sm.chans = devm_kcalloc(&pdev->dev, hsp->mbox_sm.num_chans,
824 					  sizeof(*hsp->mbox_sm.chans),
825 					  GFP_KERNEL);
826 	if (!hsp->mbox_sm.chans)
827 		return -ENOMEM;
828 
829 	if (hsp->shared_irqs) {
830 		err = tegra_hsp_add_mailboxes(hsp, &pdev->dev);
831 		if (err < 0) {
832 			dev_err(&pdev->dev, "failed to add mailboxes: %d\n",
833 			        err);
834 			return err;
835 		}
836 	}
837 
838 	err = devm_mbox_controller_register(&pdev->dev, &hsp->mbox_sm);
839 	if (err < 0) {
840 		dev_err(&pdev->dev, "failed to register shared mailbox: %d\n",
841 			err);
842 		return err;
843 	}
844 
845 	platform_set_drvdata(pdev, hsp);
846 
847 	if (hsp->doorbell_irq) {
848 		err = devm_request_irq(&pdev->dev, hsp->doorbell_irq,
849 				       tegra_hsp_doorbell_irq, IRQF_NO_SUSPEND,
850 				       dev_name(&pdev->dev), hsp);
851 		if (err < 0) {
852 			dev_err(&pdev->dev,
853 			        "failed to request doorbell IRQ#%u: %d\n",
854 				hsp->doorbell_irq, err);
855 			return err;
856 		}
857 	}
858 
859 	if (hsp->shared_irqs) {
860 		err = tegra_hsp_request_shared_irq(hsp);
861 		if (err < 0)
862 			return err;
863 	}
864 
865 	lockdep_register_key(&hsp->lock_key);
866 	lockdep_set_class(&hsp->lock, &hsp->lock_key);
867 
868 	return 0;
869 }
870 
871 static int tegra_hsp_remove(struct platform_device *pdev)
872 {
873 	struct tegra_hsp *hsp = platform_get_drvdata(pdev);
874 
875 	lockdep_unregister_key(&hsp->lock_key);
876 
877 	return 0;
878 }
879 
880 static int __maybe_unused tegra_hsp_resume(struct device *dev)
881 {
882 	struct tegra_hsp *hsp = dev_get_drvdata(dev);
883 	unsigned int i;
884 	struct tegra_hsp_doorbell *db;
885 
886 	list_for_each_entry(db, &hsp->doorbells, list) {
887 		if (db->channel.chan)
888 			tegra_hsp_doorbell_startup(db->channel.chan);
889 	}
890 
891 	if (hsp->mailboxes) {
892 		for (i = 0; i < hsp->num_sm; i++) {
893 			struct tegra_hsp_mailbox *mb = &hsp->mailboxes[i];
894 
895 			if (mb->channel.chan->cl)
896 				tegra_hsp_mailbox_startup(mb->channel.chan);
897 		}
898 	}
899 
900 	return 0;
901 }
902 
903 static const struct dev_pm_ops tegra_hsp_pm_ops = {
904 	.resume_noirq = tegra_hsp_resume,
905 };
906 
907 static const struct tegra_hsp_db_map tegra186_hsp_db_map[] = {
908 	{ "ccplex", TEGRA_HSP_DB_MASTER_CCPLEX, HSP_DB_CCPLEX, },
909 	{ "bpmp",   TEGRA_HSP_DB_MASTER_BPMP,   HSP_DB_BPMP,   },
910 	{ /* sentinel */ }
911 };
912 
913 static const struct tegra_hsp_soc tegra186_hsp_soc = {
914 	.map = tegra186_hsp_db_map,
915 	.has_per_mb_ie = false,
916 	.has_128_bit_mb = false,
917 	.reg_stride = 0x100,
918 };
919 
920 static const struct tegra_hsp_soc tegra194_hsp_soc = {
921 	.map = tegra186_hsp_db_map,
922 	.has_per_mb_ie = true,
923 	.has_128_bit_mb = false,
924 	.reg_stride = 0x100,
925 };
926 
927 static const struct tegra_hsp_soc tegra234_hsp_soc = {
928 	.map = tegra186_hsp_db_map,
929 	.has_per_mb_ie = false,
930 	.has_128_bit_mb = true,
931 	.reg_stride = 0x100,
932 };
933 
934 static const struct tegra_hsp_soc tegra264_hsp_soc = {
935 	.map = tegra186_hsp_db_map,
936 	.has_per_mb_ie = false,
937 	.has_128_bit_mb = true,
938 	.reg_stride = 0x1000,
939 };
940 
941 static const struct of_device_id tegra_hsp_match[] = {
942 	{ .compatible = "nvidia,tegra186-hsp", .data = &tegra186_hsp_soc },
943 	{ .compatible = "nvidia,tegra194-hsp", .data = &tegra194_hsp_soc },
944 	{ .compatible = "nvidia,tegra234-hsp", .data = &tegra234_hsp_soc },
945 	{ .compatible = "nvidia,tegra264-hsp", .data = &tegra264_hsp_soc },
946 	{ }
947 };
948 
949 static struct platform_driver tegra_hsp_driver = {
950 	.driver = {
951 		.name = "tegra-hsp",
952 		.of_match_table = tegra_hsp_match,
953 		.pm = &tegra_hsp_pm_ops,
954 	},
955 	.probe = tegra_hsp_probe,
956 	.remove = tegra_hsp_remove,
957 };
958 
959 static int __init tegra_hsp_init(void)
960 {
961 	return platform_driver_register(&tegra_hsp_driver);
962 }
963 core_initcall(tegra_hsp_init);
964