xref: /openbmc/linux/drivers/bus/mhi/host/init.c (revision a10c9ede)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2018-2020, The Linux Foundation. All rights reserved.
4  *
5  */
6 
7 #include <linux/bitfield.h>
8 #include <linux/debugfs.h>
9 #include <linux/device.h>
10 #include <linux/dma-direction.h>
11 #include <linux/dma-mapping.h>
12 #include <linux/idr.h>
13 #include <linux/interrupt.h>
14 #include <linux/list.h>
15 #include <linux/mhi.h>
16 #include <linux/mod_devicetable.h>
17 #include <linux/module.h>
18 #include <linux/slab.h>
19 #include <linux/vmalloc.h>
20 #include <linux/wait.h>
21 #include "internal.h"
22 
23 static DEFINE_IDA(mhi_controller_ida);
24 
25 const char * const mhi_ee_str[MHI_EE_MAX] = {
26 	[MHI_EE_PBL] = "PRIMARY BOOTLOADER",
27 	[MHI_EE_SBL] = "SECONDARY BOOTLOADER",
28 	[MHI_EE_AMSS] = "MISSION MODE",
29 	[MHI_EE_RDDM] = "RAMDUMP DOWNLOAD MODE",
30 	[MHI_EE_WFW] = "WLAN FIRMWARE",
31 	[MHI_EE_PTHRU] = "PASS THROUGH",
32 	[MHI_EE_EDL] = "EMERGENCY DOWNLOAD",
33 	[MHI_EE_FP] = "FLASH PROGRAMMER",
34 	[MHI_EE_DISABLE_TRANSITION] = "DISABLE",
35 	[MHI_EE_NOT_SUPPORTED] = "NOT SUPPORTED",
36 };
37 
38 const char * const dev_state_tran_str[DEV_ST_TRANSITION_MAX] = {
39 	[DEV_ST_TRANSITION_PBL] = "PBL",
40 	[DEV_ST_TRANSITION_READY] = "READY",
41 	[DEV_ST_TRANSITION_SBL] = "SBL",
42 	[DEV_ST_TRANSITION_MISSION_MODE] = "MISSION MODE",
43 	[DEV_ST_TRANSITION_FP] = "FLASH PROGRAMMER",
44 	[DEV_ST_TRANSITION_SYS_ERR] = "SYS ERROR",
45 	[DEV_ST_TRANSITION_DISABLE] = "DISABLE",
46 };
47 
48 const char * const mhi_ch_state_type_str[MHI_CH_STATE_TYPE_MAX] = {
49 	[MHI_CH_STATE_TYPE_RESET] = "RESET",
50 	[MHI_CH_STATE_TYPE_STOP] = "STOP",
51 	[MHI_CH_STATE_TYPE_START] = "START",
52 };
53 
54 static const char * const mhi_pm_state_str[] = {
55 	[MHI_PM_STATE_DISABLE] = "DISABLE",
56 	[MHI_PM_STATE_POR] = "POWER ON RESET",
57 	[MHI_PM_STATE_M0] = "M0",
58 	[MHI_PM_STATE_M2] = "M2",
59 	[MHI_PM_STATE_M3_ENTER] = "M?->M3",
60 	[MHI_PM_STATE_M3] = "M3",
61 	[MHI_PM_STATE_M3_EXIT] = "M3->M0",
62 	[MHI_PM_STATE_FW_DL_ERR] = "Firmware Download Error",
63 	[MHI_PM_STATE_SYS_ERR_DETECT] = "SYS ERROR Detect",
64 	[MHI_PM_STATE_SYS_ERR_PROCESS] = "SYS ERROR Process",
65 	[MHI_PM_STATE_SHUTDOWN_PROCESS] = "SHUTDOWN Process",
66 	[MHI_PM_STATE_LD_ERR_FATAL_DETECT] = "Linkdown or Error Fatal Detect",
67 };
68 
69 const char *to_mhi_pm_state_str(u32 state)
70 {
71 	int index;
72 
73 	if (state)
74 		index = __fls(state);
75 
76 	if (!state || index >= ARRAY_SIZE(mhi_pm_state_str))
77 		return "Invalid State";
78 
79 	return mhi_pm_state_str[index];
80 }
81 
82 static ssize_t serial_number_show(struct device *dev,
83 				  struct device_attribute *attr,
84 				  char *buf)
85 {
86 	struct mhi_device *mhi_dev = to_mhi_device(dev);
87 	struct mhi_controller *mhi_cntrl = mhi_dev->mhi_cntrl;
88 
89 	return sysfs_emit(buf, "Serial Number: %u\n",
90 			mhi_cntrl->serial_number);
91 }
92 static DEVICE_ATTR_RO(serial_number);
93 
94 static ssize_t oem_pk_hash_show(struct device *dev,
95 				struct device_attribute *attr,
96 				char *buf)
97 {
98 	struct mhi_device *mhi_dev = to_mhi_device(dev);
99 	struct mhi_controller *mhi_cntrl = mhi_dev->mhi_cntrl;
100 	int i, cnt = 0;
101 
102 	for (i = 0; i < ARRAY_SIZE(mhi_cntrl->oem_pk_hash); i++)
103 		cnt += sysfs_emit_at(buf, cnt, "OEMPKHASH[%d]: 0x%x\n",
104 				i, mhi_cntrl->oem_pk_hash[i]);
105 
106 	return cnt;
107 }
108 static DEVICE_ATTR_RO(oem_pk_hash);
109 
110 static ssize_t soc_reset_store(struct device *dev,
111 			       struct device_attribute *attr,
112 			       const char *buf,
113 			       size_t count)
114 {
115 	struct mhi_device *mhi_dev = to_mhi_device(dev);
116 	struct mhi_controller *mhi_cntrl = mhi_dev->mhi_cntrl;
117 
118 	mhi_soc_reset(mhi_cntrl);
119 	return count;
120 }
121 static DEVICE_ATTR_WO(soc_reset);
122 
123 static struct attribute *mhi_dev_attrs[] = {
124 	&dev_attr_serial_number.attr,
125 	&dev_attr_oem_pk_hash.attr,
126 	&dev_attr_soc_reset.attr,
127 	NULL,
128 };
129 ATTRIBUTE_GROUPS(mhi_dev);
130 
131 /* MHI protocol requires the transfer ring to be aligned with ring length */
132 static int mhi_alloc_aligned_ring(struct mhi_controller *mhi_cntrl,
133 				  struct mhi_ring *ring,
134 				  u64 len)
135 {
136 	ring->alloc_size = len + (len - 1);
137 	ring->pre_aligned = dma_alloc_coherent(mhi_cntrl->cntrl_dev, ring->alloc_size,
138 					       &ring->dma_handle, GFP_KERNEL);
139 	if (!ring->pre_aligned)
140 		return -ENOMEM;
141 
142 	ring->iommu_base = (ring->dma_handle + (len - 1)) & ~(len - 1);
143 	ring->base = ring->pre_aligned + (ring->iommu_base - ring->dma_handle);
144 
145 	return 0;
146 }
147 
148 void mhi_deinit_free_irq(struct mhi_controller *mhi_cntrl)
149 {
150 	int i;
151 	struct mhi_event *mhi_event = mhi_cntrl->mhi_event;
152 
153 	for (i = 0; i < mhi_cntrl->total_ev_rings; i++, mhi_event++) {
154 		if (mhi_event->offload_ev)
155 			continue;
156 
157 		free_irq(mhi_cntrl->irq[mhi_event->irq], mhi_event);
158 	}
159 
160 	free_irq(mhi_cntrl->irq[0], mhi_cntrl);
161 }
162 
163 int mhi_init_irq_setup(struct mhi_controller *mhi_cntrl)
164 {
165 	struct mhi_event *mhi_event = mhi_cntrl->mhi_event;
166 	struct device *dev = &mhi_cntrl->mhi_dev->dev;
167 	unsigned long irq_flags = IRQF_SHARED | IRQF_NO_SUSPEND;
168 	int i, ret;
169 
170 	/* if controller driver has set irq_flags, use it */
171 	if (mhi_cntrl->irq_flags)
172 		irq_flags = mhi_cntrl->irq_flags;
173 
174 	/* Setup BHI_INTVEC IRQ */
175 	ret = request_threaded_irq(mhi_cntrl->irq[0], mhi_intvec_handler,
176 				   mhi_intvec_threaded_handler,
177 				   irq_flags,
178 				   "bhi", mhi_cntrl);
179 	if (ret)
180 		return ret;
181 
182 	for (i = 0; i < mhi_cntrl->total_ev_rings; i++, mhi_event++) {
183 		if (mhi_event->offload_ev)
184 			continue;
185 
186 		if (mhi_event->irq >= mhi_cntrl->nr_irqs) {
187 			dev_err(dev, "irq %d not available for event ring\n",
188 				mhi_event->irq);
189 			ret = -EINVAL;
190 			goto error_request;
191 		}
192 
193 		ret = request_irq(mhi_cntrl->irq[mhi_event->irq],
194 				  mhi_irq_handler,
195 				  irq_flags,
196 				  "mhi", mhi_event);
197 		if (ret) {
198 			dev_err(dev, "Error requesting irq:%d for ev:%d\n",
199 				mhi_cntrl->irq[mhi_event->irq], i);
200 			goto error_request;
201 		}
202 	}
203 
204 	return 0;
205 
206 error_request:
207 	for (--i, --mhi_event; i >= 0; i--, mhi_event--) {
208 		if (mhi_event->offload_ev)
209 			continue;
210 
211 		free_irq(mhi_cntrl->irq[mhi_event->irq], mhi_event);
212 	}
213 	free_irq(mhi_cntrl->irq[0], mhi_cntrl);
214 
215 	return ret;
216 }
217 
218 void mhi_deinit_dev_ctxt(struct mhi_controller *mhi_cntrl)
219 {
220 	int i;
221 	struct mhi_ctxt *mhi_ctxt = mhi_cntrl->mhi_ctxt;
222 	struct mhi_cmd *mhi_cmd;
223 	struct mhi_event *mhi_event;
224 	struct mhi_ring *ring;
225 
226 	mhi_cmd = mhi_cntrl->mhi_cmd;
227 	for (i = 0; i < NR_OF_CMD_RINGS; i++, mhi_cmd++) {
228 		ring = &mhi_cmd->ring;
229 		dma_free_coherent(mhi_cntrl->cntrl_dev, ring->alloc_size,
230 				  ring->pre_aligned, ring->dma_handle);
231 		ring->base = NULL;
232 		ring->iommu_base = 0;
233 	}
234 
235 	dma_free_coherent(mhi_cntrl->cntrl_dev,
236 			  sizeof(*mhi_ctxt->cmd_ctxt) * NR_OF_CMD_RINGS,
237 			  mhi_ctxt->cmd_ctxt, mhi_ctxt->cmd_ctxt_addr);
238 
239 	mhi_event = mhi_cntrl->mhi_event;
240 	for (i = 0; i < mhi_cntrl->total_ev_rings; i++, mhi_event++) {
241 		if (mhi_event->offload_ev)
242 			continue;
243 
244 		ring = &mhi_event->ring;
245 		dma_free_coherent(mhi_cntrl->cntrl_dev, ring->alloc_size,
246 				  ring->pre_aligned, ring->dma_handle);
247 		ring->base = NULL;
248 		ring->iommu_base = 0;
249 	}
250 
251 	dma_free_coherent(mhi_cntrl->cntrl_dev, sizeof(*mhi_ctxt->er_ctxt) *
252 			  mhi_cntrl->total_ev_rings, mhi_ctxt->er_ctxt,
253 			  mhi_ctxt->er_ctxt_addr);
254 
255 	dma_free_coherent(mhi_cntrl->cntrl_dev, sizeof(*mhi_ctxt->chan_ctxt) *
256 			  mhi_cntrl->max_chan, mhi_ctxt->chan_ctxt,
257 			  mhi_ctxt->chan_ctxt_addr);
258 
259 	kfree(mhi_ctxt);
260 	mhi_cntrl->mhi_ctxt = NULL;
261 }
262 
263 int mhi_init_dev_ctxt(struct mhi_controller *mhi_cntrl)
264 {
265 	struct mhi_ctxt *mhi_ctxt;
266 	struct mhi_chan_ctxt *chan_ctxt;
267 	struct mhi_event_ctxt *er_ctxt;
268 	struct mhi_cmd_ctxt *cmd_ctxt;
269 	struct mhi_chan *mhi_chan;
270 	struct mhi_event *mhi_event;
271 	struct mhi_cmd *mhi_cmd;
272 	u32 tmp;
273 	int ret = -ENOMEM, i;
274 
275 	atomic_set(&mhi_cntrl->dev_wake, 0);
276 	atomic_set(&mhi_cntrl->pending_pkts, 0);
277 
278 	mhi_ctxt = kzalloc(sizeof(*mhi_ctxt), GFP_KERNEL);
279 	if (!mhi_ctxt)
280 		return -ENOMEM;
281 
282 	/* Setup channel ctxt */
283 	mhi_ctxt->chan_ctxt = dma_alloc_coherent(mhi_cntrl->cntrl_dev,
284 						 sizeof(*mhi_ctxt->chan_ctxt) *
285 						 mhi_cntrl->max_chan,
286 						 &mhi_ctxt->chan_ctxt_addr,
287 						 GFP_KERNEL);
288 	if (!mhi_ctxt->chan_ctxt)
289 		goto error_alloc_chan_ctxt;
290 
291 	mhi_chan = mhi_cntrl->mhi_chan;
292 	chan_ctxt = mhi_ctxt->chan_ctxt;
293 	for (i = 0; i < mhi_cntrl->max_chan; i++, chan_ctxt++, mhi_chan++) {
294 		/* Skip if it is an offload channel */
295 		if (mhi_chan->offload_ch)
296 			continue;
297 
298 		tmp = le32_to_cpu(chan_ctxt->chcfg);
299 		tmp &= ~CHAN_CTX_CHSTATE_MASK;
300 		tmp |= FIELD_PREP(CHAN_CTX_CHSTATE_MASK, MHI_CH_STATE_DISABLED);
301 		tmp &= ~CHAN_CTX_BRSTMODE_MASK;
302 		tmp |= FIELD_PREP(CHAN_CTX_BRSTMODE_MASK, mhi_chan->db_cfg.brstmode);
303 		tmp &= ~CHAN_CTX_POLLCFG_MASK;
304 		tmp |= FIELD_PREP(CHAN_CTX_POLLCFG_MASK, mhi_chan->db_cfg.pollcfg);
305 		chan_ctxt->chcfg = cpu_to_le32(tmp);
306 
307 		chan_ctxt->chtype = cpu_to_le32(mhi_chan->type);
308 		chan_ctxt->erindex = cpu_to_le32(mhi_chan->er_index);
309 
310 		mhi_chan->ch_state = MHI_CH_STATE_DISABLED;
311 		mhi_chan->tre_ring.db_addr = (void __iomem *)&chan_ctxt->wp;
312 	}
313 
314 	/* Setup event context */
315 	mhi_ctxt->er_ctxt = dma_alloc_coherent(mhi_cntrl->cntrl_dev,
316 					       sizeof(*mhi_ctxt->er_ctxt) *
317 					       mhi_cntrl->total_ev_rings,
318 					       &mhi_ctxt->er_ctxt_addr,
319 					       GFP_KERNEL);
320 	if (!mhi_ctxt->er_ctxt)
321 		goto error_alloc_er_ctxt;
322 
323 	er_ctxt = mhi_ctxt->er_ctxt;
324 	mhi_event = mhi_cntrl->mhi_event;
325 	for (i = 0; i < mhi_cntrl->total_ev_rings; i++, er_ctxt++,
326 		     mhi_event++) {
327 		struct mhi_ring *ring = &mhi_event->ring;
328 
329 		/* Skip if it is an offload event */
330 		if (mhi_event->offload_ev)
331 			continue;
332 
333 		tmp = le32_to_cpu(er_ctxt->intmod);
334 		tmp &= ~EV_CTX_INTMODC_MASK;
335 		tmp &= ~EV_CTX_INTMODT_MASK;
336 		tmp |= FIELD_PREP(EV_CTX_INTMODT_MASK, mhi_event->intmod);
337 		er_ctxt->intmod = cpu_to_le32(tmp);
338 
339 		er_ctxt->ertype = cpu_to_le32(MHI_ER_TYPE_VALID);
340 		er_ctxt->msivec = cpu_to_le32(mhi_event->irq);
341 		mhi_event->db_cfg.db_mode = true;
342 
343 		ring->el_size = sizeof(struct mhi_ring_element);
344 		ring->len = ring->el_size * ring->elements;
345 		ret = mhi_alloc_aligned_ring(mhi_cntrl, ring, ring->len);
346 		if (ret)
347 			goto error_alloc_er;
348 
349 		/*
350 		 * If the read pointer equals to the write pointer, then the
351 		 * ring is empty
352 		 */
353 		ring->rp = ring->wp = ring->base;
354 		er_ctxt->rbase = cpu_to_le64(ring->iommu_base);
355 		er_ctxt->rp = er_ctxt->wp = er_ctxt->rbase;
356 		er_ctxt->rlen = cpu_to_le64(ring->len);
357 		ring->ctxt_wp = &er_ctxt->wp;
358 	}
359 
360 	/* Setup cmd context */
361 	ret = -ENOMEM;
362 	mhi_ctxt->cmd_ctxt = dma_alloc_coherent(mhi_cntrl->cntrl_dev,
363 						sizeof(*mhi_ctxt->cmd_ctxt) *
364 						NR_OF_CMD_RINGS,
365 						&mhi_ctxt->cmd_ctxt_addr,
366 						GFP_KERNEL);
367 	if (!mhi_ctxt->cmd_ctxt)
368 		goto error_alloc_er;
369 
370 	mhi_cmd = mhi_cntrl->mhi_cmd;
371 	cmd_ctxt = mhi_ctxt->cmd_ctxt;
372 	for (i = 0; i < NR_OF_CMD_RINGS; i++, mhi_cmd++, cmd_ctxt++) {
373 		struct mhi_ring *ring = &mhi_cmd->ring;
374 
375 		ring->el_size = sizeof(struct mhi_ring_element);
376 		ring->elements = CMD_EL_PER_RING;
377 		ring->len = ring->el_size * ring->elements;
378 		ret = mhi_alloc_aligned_ring(mhi_cntrl, ring, ring->len);
379 		if (ret)
380 			goto error_alloc_cmd;
381 
382 		ring->rp = ring->wp = ring->base;
383 		cmd_ctxt->rbase = cpu_to_le64(ring->iommu_base);
384 		cmd_ctxt->rp = cmd_ctxt->wp = cmd_ctxt->rbase;
385 		cmd_ctxt->rlen = cpu_to_le64(ring->len);
386 		ring->ctxt_wp = &cmd_ctxt->wp;
387 	}
388 
389 	mhi_cntrl->mhi_ctxt = mhi_ctxt;
390 
391 	return 0;
392 
393 error_alloc_cmd:
394 	for (--i, --mhi_cmd; i >= 0; i--, mhi_cmd--) {
395 		struct mhi_ring *ring = &mhi_cmd->ring;
396 
397 		dma_free_coherent(mhi_cntrl->cntrl_dev, ring->alloc_size,
398 				  ring->pre_aligned, ring->dma_handle);
399 	}
400 	dma_free_coherent(mhi_cntrl->cntrl_dev,
401 			  sizeof(*mhi_ctxt->cmd_ctxt) * NR_OF_CMD_RINGS,
402 			  mhi_ctxt->cmd_ctxt, mhi_ctxt->cmd_ctxt_addr);
403 	i = mhi_cntrl->total_ev_rings;
404 	mhi_event = mhi_cntrl->mhi_event + i;
405 
406 error_alloc_er:
407 	for (--i, --mhi_event; i >= 0; i--, mhi_event--) {
408 		struct mhi_ring *ring = &mhi_event->ring;
409 
410 		if (mhi_event->offload_ev)
411 			continue;
412 
413 		dma_free_coherent(mhi_cntrl->cntrl_dev, ring->alloc_size,
414 				  ring->pre_aligned, ring->dma_handle);
415 	}
416 	dma_free_coherent(mhi_cntrl->cntrl_dev, sizeof(*mhi_ctxt->er_ctxt) *
417 			  mhi_cntrl->total_ev_rings, mhi_ctxt->er_ctxt,
418 			  mhi_ctxt->er_ctxt_addr);
419 
420 error_alloc_er_ctxt:
421 	dma_free_coherent(mhi_cntrl->cntrl_dev, sizeof(*mhi_ctxt->chan_ctxt) *
422 			  mhi_cntrl->max_chan, mhi_ctxt->chan_ctxt,
423 			  mhi_ctxt->chan_ctxt_addr);
424 
425 error_alloc_chan_ctxt:
426 	kfree(mhi_ctxt);
427 
428 	return ret;
429 }
430 
431 int mhi_init_mmio(struct mhi_controller *mhi_cntrl)
432 {
433 	u32 val;
434 	int i, ret;
435 	struct mhi_chan *mhi_chan;
436 	struct mhi_event *mhi_event;
437 	void __iomem *base = mhi_cntrl->regs;
438 	struct device *dev = &mhi_cntrl->mhi_dev->dev;
439 	struct {
440 		u32 offset;
441 		u32 val;
442 	} reg_info[] = {
443 		{
444 			CCABAP_HIGHER,
445 			upper_32_bits(mhi_cntrl->mhi_ctxt->chan_ctxt_addr),
446 		},
447 		{
448 			CCABAP_LOWER,
449 			lower_32_bits(mhi_cntrl->mhi_ctxt->chan_ctxt_addr),
450 		},
451 		{
452 			ECABAP_HIGHER,
453 			upper_32_bits(mhi_cntrl->mhi_ctxt->er_ctxt_addr),
454 		},
455 		{
456 			ECABAP_LOWER,
457 			lower_32_bits(mhi_cntrl->mhi_ctxt->er_ctxt_addr),
458 		},
459 		{
460 			CRCBAP_HIGHER,
461 			upper_32_bits(mhi_cntrl->mhi_ctxt->cmd_ctxt_addr),
462 		},
463 		{
464 			CRCBAP_LOWER,
465 			lower_32_bits(mhi_cntrl->mhi_ctxt->cmd_ctxt_addr),
466 		},
467 		{
468 			MHICTRLBASE_HIGHER,
469 			upper_32_bits(mhi_cntrl->iova_start),
470 		},
471 		{
472 			MHICTRLBASE_LOWER,
473 			lower_32_bits(mhi_cntrl->iova_start),
474 		},
475 		{
476 			MHIDATABASE_HIGHER,
477 			upper_32_bits(mhi_cntrl->iova_start),
478 		},
479 		{
480 			MHIDATABASE_LOWER,
481 			lower_32_bits(mhi_cntrl->iova_start),
482 		},
483 		{
484 			MHICTRLLIMIT_HIGHER,
485 			upper_32_bits(mhi_cntrl->iova_stop),
486 		},
487 		{
488 			MHICTRLLIMIT_LOWER,
489 			lower_32_bits(mhi_cntrl->iova_stop),
490 		},
491 		{
492 			MHIDATALIMIT_HIGHER,
493 			upper_32_bits(mhi_cntrl->iova_stop),
494 		},
495 		{
496 			MHIDATALIMIT_LOWER,
497 			lower_32_bits(mhi_cntrl->iova_stop),
498 		},
499 		{0, 0}
500 	};
501 
502 	dev_dbg(dev, "Initializing MHI registers\n");
503 
504 	/* Read channel db offset */
505 	ret = mhi_read_reg(mhi_cntrl, base, CHDBOFF, &val);
506 	if (ret) {
507 		dev_err(dev, "Unable to read CHDBOFF register\n");
508 		return -EIO;
509 	}
510 
511 	/* Setup wake db */
512 	mhi_cntrl->wake_db = base + val + (8 * MHI_DEV_WAKE_DB);
513 	mhi_cntrl->wake_set = false;
514 
515 	/* Setup channel db address for each channel in tre_ring */
516 	mhi_chan = mhi_cntrl->mhi_chan;
517 	for (i = 0; i < mhi_cntrl->max_chan; i++, val += 8, mhi_chan++)
518 		mhi_chan->tre_ring.db_addr = base + val;
519 
520 	/* Read event ring db offset */
521 	ret = mhi_read_reg(mhi_cntrl, base, ERDBOFF, &val);
522 	if (ret) {
523 		dev_err(dev, "Unable to read ERDBOFF register\n");
524 		return -EIO;
525 	}
526 
527 	/* Setup event db address for each ev_ring */
528 	mhi_event = mhi_cntrl->mhi_event;
529 	for (i = 0; i < mhi_cntrl->total_ev_rings; i++, val += 8, mhi_event++) {
530 		if (mhi_event->offload_ev)
531 			continue;
532 
533 		mhi_event->ring.db_addr = base + val;
534 	}
535 
536 	/* Setup DB register for primary CMD rings */
537 	mhi_cntrl->mhi_cmd[PRIMARY_CMD_RING].ring.db_addr = base + CRDB_LOWER;
538 
539 	/* Write to MMIO registers */
540 	for (i = 0; reg_info[i].offset; i++)
541 		mhi_write_reg(mhi_cntrl, base, reg_info[i].offset,
542 			      reg_info[i].val);
543 
544 	ret = mhi_write_reg_field(mhi_cntrl, base, MHICFG, MHICFG_NER_MASK,
545 				  mhi_cntrl->total_ev_rings);
546 	if (ret) {
547 		dev_err(dev, "Unable to write MHICFG register\n");
548 		return ret;
549 	}
550 
551 	ret = mhi_write_reg_field(mhi_cntrl, base, MHICFG, MHICFG_NHWER_MASK,
552 				  mhi_cntrl->hw_ev_rings);
553 	if (ret) {
554 		dev_err(dev, "Unable to write MHICFG register\n");
555 		return ret;
556 	}
557 
558 	return 0;
559 }
560 
561 void mhi_deinit_chan_ctxt(struct mhi_controller *mhi_cntrl,
562 			  struct mhi_chan *mhi_chan)
563 {
564 	struct mhi_ring *buf_ring;
565 	struct mhi_ring *tre_ring;
566 	struct mhi_chan_ctxt *chan_ctxt;
567 	u32 tmp;
568 
569 	buf_ring = &mhi_chan->buf_ring;
570 	tre_ring = &mhi_chan->tre_ring;
571 	chan_ctxt = &mhi_cntrl->mhi_ctxt->chan_ctxt[mhi_chan->chan];
572 
573 	if (!chan_ctxt->rbase) /* Already uninitialized */
574 		return;
575 
576 	dma_free_coherent(mhi_cntrl->cntrl_dev, tre_ring->alloc_size,
577 			  tre_ring->pre_aligned, tre_ring->dma_handle);
578 	vfree(buf_ring->base);
579 
580 	buf_ring->base = tre_ring->base = NULL;
581 	tre_ring->ctxt_wp = NULL;
582 	chan_ctxt->rbase = 0;
583 	chan_ctxt->rlen = 0;
584 	chan_ctxt->rp = 0;
585 	chan_ctxt->wp = 0;
586 
587 	tmp = le32_to_cpu(chan_ctxt->chcfg);
588 	tmp &= ~CHAN_CTX_CHSTATE_MASK;
589 	tmp |= FIELD_PREP(CHAN_CTX_CHSTATE_MASK, MHI_CH_STATE_DISABLED);
590 	chan_ctxt->chcfg = cpu_to_le32(tmp);
591 
592 	/* Update to all cores */
593 	smp_wmb();
594 }
595 
596 int mhi_init_chan_ctxt(struct mhi_controller *mhi_cntrl,
597 		       struct mhi_chan *mhi_chan)
598 {
599 	struct mhi_ring *buf_ring;
600 	struct mhi_ring *tre_ring;
601 	struct mhi_chan_ctxt *chan_ctxt;
602 	u32 tmp;
603 	int ret;
604 
605 	buf_ring = &mhi_chan->buf_ring;
606 	tre_ring = &mhi_chan->tre_ring;
607 	tre_ring->el_size = sizeof(struct mhi_ring_element);
608 	tre_ring->len = tre_ring->el_size * tre_ring->elements;
609 	chan_ctxt = &mhi_cntrl->mhi_ctxt->chan_ctxt[mhi_chan->chan];
610 	ret = mhi_alloc_aligned_ring(mhi_cntrl, tre_ring, tre_ring->len);
611 	if (ret)
612 		return -ENOMEM;
613 
614 	buf_ring->el_size = sizeof(struct mhi_buf_info);
615 	buf_ring->len = buf_ring->el_size * buf_ring->elements;
616 	buf_ring->base = vzalloc(buf_ring->len);
617 
618 	if (!buf_ring->base) {
619 		dma_free_coherent(mhi_cntrl->cntrl_dev, tre_ring->alloc_size,
620 				  tre_ring->pre_aligned, tre_ring->dma_handle);
621 		return -ENOMEM;
622 	}
623 
624 	tmp = le32_to_cpu(chan_ctxt->chcfg);
625 	tmp &= ~CHAN_CTX_CHSTATE_MASK;
626 	tmp |= FIELD_PREP(CHAN_CTX_CHSTATE_MASK, MHI_CH_STATE_ENABLED);
627 	chan_ctxt->chcfg = cpu_to_le32(tmp);
628 
629 	chan_ctxt->rbase = cpu_to_le64(tre_ring->iommu_base);
630 	chan_ctxt->rp = chan_ctxt->wp = chan_ctxt->rbase;
631 	chan_ctxt->rlen = cpu_to_le64(tre_ring->len);
632 	tre_ring->ctxt_wp = &chan_ctxt->wp;
633 
634 	tre_ring->rp = tre_ring->wp = tre_ring->base;
635 	buf_ring->rp = buf_ring->wp = buf_ring->base;
636 	mhi_chan->db_cfg.db_mode = 1;
637 
638 	/* Update to all cores */
639 	smp_wmb();
640 
641 	return 0;
642 }
643 
644 static int parse_ev_cfg(struct mhi_controller *mhi_cntrl,
645 			const struct mhi_controller_config *config)
646 {
647 	struct mhi_event *mhi_event;
648 	const struct mhi_event_config *event_cfg;
649 	struct device *dev = mhi_cntrl->cntrl_dev;
650 	int i, num;
651 
652 	num = config->num_events;
653 	mhi_cntrl->total_ev_rings = num;
654 	mhi_cntrl->mhi_event = kcalloc(num, sizeof(*mhi_cntrl->mhi_event),
655 				       GFP_KERNEL);
656 	if (!mhi_cntrl->mhi_event)
657 		return -ENOMEM;
658 
659 	/* Populate event ring */
660 	mhi_event = mhi_cntrl->mhi_event;
661 	for (i = 0; i < num; i++) {
662 		event_cfg = &config->event_cfg[i];
663 
664 		mhi_event->er_index = i;
665 		mhi_event->ring.elements = event_cfg->num_elements;
666 		mhi_event->intmod = event_cfg->irq_moderation_ms;
667 		mhi_event->irq = event_cfg->irq;
668 
669 		if (event_cfg->channel != U32_MAX) {
670 			/* This event ring has a dedicated channel */
671 			mhi_event->chan = event_cfg->channel;
672 			if (mhi_event->chan >= mhi_cntrl->max_chan) {
673 				dev_err(dev,
674 					"Event Ring channel not available\n");
675 				goto error_ev_cfg;
676 			}
677 
678 			mhi_event->mhi_chan =
679 				&mhi_cntrl->mhi_chan[mhi_event->chan];
680 		}
681 
682 		/* Priority is fixed to 1 for now */
683 		mhi_event->priority = 1;
684 
685 		mhi_event->db_cfg.brstmode = event_cfg->mode;
686 		if (MHI_INVALID_BRSTMODE(mhi_event->db_cfg.brstmode))
687 			goto error_ev_cfg;
688 
689 		if (mhi_event->db_cfg.brstmode == MHI_DB_BRST_ENABLE)
690 			mhi_event->db_cfg.process_db = mhi_db_brstmode;
691 		else
692 			mhi_event->db_cfg.process_db = mhi_db_brstmode_disable;
693 
694 		mhi_event->data_type = event_cfg->data_type;
695 
696 		switch (mhi_event->data_type) {
697 		case MHI_ER_DATA:
698 			mhi_event->process_event = mhi_process_data_event_ring;
699 			break;
700 		case MHI_ER_CTRL:
701 			mhi_event->process_event = mhi_process_ctrl_ev_ring;
702 			break;
703 		default:
704 			dev_err(dev, "Event Ring type not supported\n");
705 			goto error_ev_cfg;
706 		}
707 
708 		mhi_event->hw_ring = event_cfg->hardware_event;
709 		if (mhi_event->hw_ring)
710 			mhi_cntrl->hw_ev_rings++;
711 		else
712 			mhi_cntrl->sw_ev_rings++;
713 
714 		mhi_event->cl_manage = event_cfg->client_managed;
715 		mhi_event->offload_ev = event_cfg->offload_channel;
716 		mhi_event++;
717 	}
718 
719 	return 0;
720 
721 error_ev_cfg:
722 
723 	kfree(mhi_cntrl->mhi_event);
724 	return -EINVAL;
725 }
726 
727 static int parse_ch_cfg(struct mhi_controller *mhi_cntrl,
728 			const struct mhi_controller_config *config)
729 {
730 	const struct mhi_channel_config *ch_cfg;
731 	struct device *dev = mhi_cntrl->cntrl_dev;
732 	int i;
733 	u32 chan;
734 
735 	mhi_cntrl->max_chan = config->max_channels;
736 
737 	/*
738 	 * The allocation of MHI channels can exceed 32KB in some scenarios,
739 	 * so to avoid any memory possible allocation failures, vzalloc is
740 	 * used here
741 	 */
742 	mhi_cntrl->mhi_chan = vzalloc(mhi_cntrl->max_chan *
743 				      sizeof(*mhi_cntrl->mhi_chan));
744 	if (!mhi_cntrl->mhi_chan)
745 		return -ENOMEM;
746 
747 	INIT_LIST_HEAD(&mhi_cntrl->lpm_chans);
748 
749 	/* Populate channel configurations */
750 	for (i = 0; i < config->num_channels; i++) {
751 		struct mhi_chan *mhi_chan;
752 
753 		ch_cfg = &config->ch_cfg[i];
754 
755 		chan = ch_cfg->num;
756 		if (chan >= mhi_cntrl->max_chan) {
757 			dev_err(dev, "Channel %d not available\n", chan);
758 			goto error_chan_cfg;
759 		}
760 
761 		mhi_chan = &mhi_cntrl->mhi_chan[chan];
762 		mhi_chan->name = ch_cfg->name;
763 		mhi_chan->chan = chan;
764 
765 		mhi_chan->tre_ring.elements = ch_cfg->num_elements;
766 		if (!mhi_chan->tre_ring.elements)
767 			goto error_chan_cfg;
768 
769 		/*
770 		 * For some channels, local ring length should be bigger than
771 		 * the transfer ring length due to internal logical channels
772 		 * in device. So host can queue much more buffers than transfer
773 		 * ring length. Example, RSC channels should have a larger local
774 		 * channel length than transfer ring length.
775 		 */
776 		mhi_chan->buf_ring.elements = ch_cfg->local_elements;
777 		if (!mhi_chan->buf_ring.elements)
778 			mhi_chan->buf_ring.elements = mhi_chan->tre_ring.elements;
779 		mhi_chan->er_index = ch_cfg->event_ring;
780 		mhi_chan->dir = ch_cfg->dir;
781 
782 		/*
783 		 * For most channels, chtype is identical to channel directions.
784 		 * So, if it is not defined then assign channel direction to
785 		 * chtype
786 		 */
787 		mhi_chan->type = ch_cfg->type;
788 		if (!mhi_chan->type)
789 			mhi_chan->type = (enum mhi_ch_type)mhi_chan->dir;
790 
791 		mhi_chan->ee_mask = ch_cfg->ee_mask;
792 		mhi_chan->db_cfg.pollcfg = ch_cfg->pollcfg;
793 		mhi_chan->lpm_notify = ch_cfg->lpm_notify;
794 		mhi_chan->offload_ch = ch_cfg->offload_channel;
795 		mhi_chan->db_cfg.reset_req = ch_cfg->doorbell_mode_switch;
796 		mhi_chan->pre_alloc = ch_cfg->auto_queue;
797 		mhi_chan->wake_capable = ch_cfg->wake_capable;
798 
799 		/*
800 		 * If MHI host allocates buffers, then the channel direction
801 		 * should be DMA_FROM_DEVICE
802 		 */
803 		if (mhi_chan->pre_alloc && mhi_chan->dir != DMA_FROM_DEVICE) {
804 			dev_err(dev, "Invalid channel configuration\n");
805 			goto error_chan_cfg;
806 		}
807 
808 		/*
809 		 * Bi-directional and direction less channel must be an
810 		 * offload channel
811 		 */
812 		if ((mhi_chan->dir == DMA_BIDIRECTIONAL ||
813 		     mhi_chan->dir == DMA_NONE) && !mhi_chan->offload_ch) {
814 			dev_err(dev, "Invalid channel configuration\n");
815 			goto error_chan_cfg;
816 		}
817 
818 		if (!mhi_chan->offload_ch) {
819 			mhi_chan->db_cfg.brstmode = ch_cfg->doorbell;
820 			if (MHI_INVALID_BRSTMODE(mhi_chan->db_cfg.brstmode)) {
821 				dev_err(dev, "Invalid Door bell mode\n");
822 				goto error_chan_cfg;
823 			}
824 		}
825 
826 		if (mhi_chan->db_cfg.brstmode == MHI_DB_BRST_ENABLE)
827 			mhi_chan->db_cfg.process_db = mhi_db_brstmode;
828 		else
829 			mhi_chan->db_cfg.process_db = mhi_db_brstmode_disable;
830 
831 		mhi_chan->configured = true;
832 
833 		if (mhi_chan->lpm_notify)
834 			list_add_tail(&mhi_chan->node, &mhi_cntrl->lpm_chans);
835 	}
836 
837 	return 0;
838 
839 error_chan_cfg:
840 	vfree(mhi_cntrl->mhi_chan);
841 
842 	return -EINVAL;
843 }
844 
845 static int parse_config(struct mhi_controller *mhi_cntrl,
846 			const struct mhi_controller_config *config)
847 {
848 	int ret;
849 
850 	/* Parse MHI channel configuration */
851 	ret = parse_ch_cfg(mhi_cntrl, config);
852 	if (ret)
853 		return ret;
854 
855 	/* Parse MHI event configuration */
856 	ret = parse_ev_cfg(mhi_cntrl, config);
857 	if (ret)
858 		goto error_ev_cfg;
859 
860 	mhi_cntrl->timeout_ms = config->timeout_ms;
861 	if (!mhi_cntrl->timeout_ms)
862 		mhi_cntrl->timeout_ms = MHI_TIMEOUT_MS;
863 
864 	mhi_cntrl->bounce_buf = config->use_bounce_buf;
865 	mhi_cntrl->buffer_len = config->buf_len;
866 	if (!mhi_cntrl->buffer_len)
867 		mhi_cntrl->buffer_len = MHI_MAX_MTU;
868 
869 	/* By default, host is allowed to ring DB in both M0 and M2 states */
870 	mhi_cntrl->db_access = MHI_PM_M0 | MHI_PM_M2;
871 	if (config->m2_no_db)
872 		mhi_cntrl->db_access &= ~MHI_PM_M2;
873 
874 	return 0;
875 
876 error_ev_cfg:
877 	vfree(mhi_cntrl->mhi_chan);
878 
879 	return ret;
880 }
881 
882 int mhi_register_controller(struct mhi_controller *mhi_cntrl,
883 			    const struct mhi_controller_config *config)
884 {
885 	struct mhi_event *mhi_event;
886 	struct mhi_chan *mhi_chan;
887 	struct mhi_cmd *mhi_cmd;
888 	struct mhi_device *mhi_dev;
889 	u32 soc_info;
890 	int ret, i;
891 
892 	if (!mhi_cntrl || !mhi_cntrl->cntrl_dev || !mhi_cntrl->regs ||
893 	    !mhi_cntrl->runtime_get || !mhi_cntrl->runtime_put ||
894 	    !mhi_cntrl->status_cb || !mhi_cntrl->read_reg ||
895 	    !mhi_cntrl->write_reg || !mhi_cntrl->nr_irqs ||
896 	    !mhi_cntrl->irq || !mhi_cntrl->reg_len)
897 		return -EINVAL;
898 
899 	ret = parse_config(mhi_cntrl, config);
900 	if (ret)
901 		return -EINVAL;
902 
903 	mhi_cntrl->mhi_cmd = kcalloc(NR_OF_CMD_RINGS,
904 				     sizeof(*mhi_cntrl->mhi_cmd), GFP_KERNEL);
905 	if (!mhi_cntrl->mhi_cmd) {
906 		ret = -ENOMEM;
907 		goto err_free_event;
908 	}
909 
910 	INIT_LIST_HEAD(&mhi_cntrl->transition_list);
911 	mutex_init(&mhi_cntrl->pm_mutex);
912 	rwlock_init(&mhi_cntrl->pm_lock);
913 	spin_lock_init(&mhi_cntrl->transition_lock);
914 	spin_lock_init(&mhi_cntrl->wlock);
915 	INIT_WORK(&mhi_cntrl->st_worker, mhi_pm_st_worker);
916 	init_waitqueue_head(&mhi_cntrl->state_event);
917 
918 	mhi_cntrl->hiprio_wq = alloc_ordered_workqueue("mhi_hiprio_wq", WQ_HIGHPRI);
919 	if (!mhi_cntrl->hiprio_wq) {
920 		dev_err(mhi_cntrl->cntrl_dev, "Failed to allocate workqueue\n");
921 		ret = -ENOMEM;
922 		goto err_free_cmd;
923 	}
924 
925 	mhi_cmd = mhi_cntrl->mhi_cmd;
926 	for (i = 0; i < NR_OF_CMD_RINGS; i++, mhi_cmd++)
927 		spin_lock_init(&mhi_cmd->lock);
928 
929 	mhi_event = mhi_cntrl->mhi_event;
930 	for (i = 0; i < mhi_cntrl->total_ev_rings; i++, mhi_event++) {
931 		/* Skip for offload events */
932 		if (mhi_event->offload_ev)
933 			continue;
934 
935 		mhi_event->mhi_cntrl = mhi_cntrl;
936 		spin_lock_init(&mhi_event->lock);
937 		if (mhi_event->data_type == MHI_ER_CTRL)
938 			tasklet_init(&mhi_event->task, mhi_ctrl_ev_task,
939 				     (ulong)mhi_event);
940 		else
941 			tasklet_init(&mhi_event->task, mhi_ev_task,
942 				     (ulong)mhi_event);
943 	}
944 
945 	mhi_chan = mhi_cntrl->mhi_chan;
946 	for (i = 0; i < mhi_cntrl->max_chan; i++, mhi_chan++) {
947 		mutex_init(&mhi_chan->mutex);
948 		init_completion(&mhi_chan->completion);
949 		rwlock_init(&mhi_chan->lock);
950 
951 		/* used in setting bei field of TRE */
952 		mhi_event = &mhi_cntrl->mhi_event[mhi_chan->er_index];
953 		mhi_chan->intmod = mhi_event->intmod;
954 	}
955 
956 	if (mhi_cntrl->bounce_buf) {
957 		mhi_cntrl->map_single = mhi_map_single_use_bb;
958 		mhi_cntrl->unmap_single = mhi_unmap_single_use_bb;
959 	} else {
960 		mhi_cntrl->map_single = mhi_map_single_no_bb;
961 		mhi_cntrl->unmap_single = mhi_unmap_single_no_bb;
962 	}
963 
964 	/* Read the MHI device info */
965 	ret = mhi_read_reg(mhi_cntrl, mhi_cntrl->regs,
966 			   SOC_HW_VERSION_OFFS, &soc_info);
967 	if (ret)
968 		goto err_destroy_wq;
969 
970 	mhi_cntrl->family_number = FIELD_GET(SOC_HW_VERSION_FAM_NUM_BMSK, soc_info);
971 	mhi_cntrl->device_number = FIELD_GET(SOC_HW_VERSION_DEV_NUM_BMSK, soc_info);
972 	mhi_cntrl->major_version = FIELD_GET(SOC_HW_VERSION_MAJOR_VER_BMSK, soc_info);
973 	mhi_cntrl->minor_version = FIELD_GET(SOC_HW_VERSION_MINOR_VER_BMSK, soc_info);
974 
975 	mhi_cntrl->index = ida_alloc(&mhi_controller_ida, GFP_KERNEL);
976 	if (mhi_cntrl->index < 0) {
977 		ret = mhi_cntrl->index;
978 		goto err_destroy_wq;
979 	}
980 
981 	/* Register controller with MHI bus */
982 	mhi_dev = mhi_alloc_device(mhi_cntrl);
983 	if (IS_ERR(mhi_dev)) {
984 		dev_err(mhi_cntrl->cntrl_dev, "Failed to allocate MHI device\n");
985 		ret = PTR_ERR(mhi_dev);
986 		goto err_ida_free;
987 	}
988 
989 	mhi_dev->dev_type = MHI_DEVICE_CONTROLLER;
990 	mhi_dev->mhi_cntrl = mhi_cntrl;
991 	dev_set_name(&mhi_dev->dev, "mhi%d", mhi_cntrl->index);
992 	mhi_dev->name = dev_name(&mhi_dev->dev);
993 
994 	/* Init wakeup source */
995 	device_init_wakeup(&mhi_dev->dev, true);
996 
997 	ret = device_add(&mhi_dev->dev);
998 	if (ret)
999 		goto err_release_dev;
1000 
1001 	mhi_cntrl->mhi_dev = mhi_dev;
1002 
1003 	mhi_create_debugfs(mhi_cntrl);
1004 
1005 	return 0;
1006 
1007 err_release_dev:
1008 	put_device(&mhi_dev->dev);
1009 err_ida_free:
1010 	ida_free(&mhi_controller_ida, mhi_cntrl->index);
1011 err_destroy_wq:
1012 	destroy_workqueue(mhi_cntrl->hiprio_wq);
1013 err_free_cmd:
1014 	kfree(mhi_cntrl->mhi_cmd);
1015 err_free_event:
1016 	kfree(mhi_cntrl->mhi_event);
1017 	vfree(mhi_cntrl->mhi_chan);
1018 
1019 	return ret;
1020 }
1021 EXPORT_SYMBOL_GPL(mhi_register_controller);
1022 
1023 void mhi_unregister_controller(struct mhi_controller *mhi_cntrl)
1024 {
1025 	struct mhi_device *mhi_dev = mhi_cntrl->mhi_dev;
1026 	struct mhi_chan *mhi_chan = mhi_cntrl->mhi_chan;
1027 	unsigned int i;
1028 
1029 	mhi_destroy_debugfs(mhi_cntrl);
1030 
1031 	destroy_workqueue(mhi_cntrl->hiprio_wq);
1032 	kfree(mhi_cntrl->mhi_cmd);
1033 	kfree(mhi_cntrl->mhi_event);
1034 
1035 	/* Drop the references to MHI devices created for channels */
1036 	for (i = 0; i < mhi_cntrl->max_chan; i++, mhi_chan++) {
1037 		if (!mhi_chan->mhi_dev)
1038 			continue;
1039 
1040 		put_device(&mhi_chan->mhi_dev->dev);
1041 	}
1042 	vfree(mhi_cntrl->mhi_chan);
1043 
1044 	device_del(&mhi_dev->dev);
1045 	put_device(&mhi_dev->dev);
1046 
1047 	ida_free(&mhi_controller_ida, mhi_cntrl->index);
1048 }
1049 EXPORT_SYMBOL_GPL(mhi_unregister_controller);
1050 
1051 struct mhi_controller *mhi_alloc_controller(void)
1052 {
1053 	struct mhi_controller *mhi_cntrl;
1054 
1055 	mhi_cntrl = kzalloc(sizeof(*mhi_cntrl), GFP_KERNEL);
1056 
1057 	return mhi_cntrl;
1058 }
1059 EXPORT_SYMBOL_GPL(mhi_alloc_controller);
1060 
1061 void mhi_free_controller(struct mhi_controller *mhi_cntrl)
1062 {
1063 	kfree(mhi_cntrl);
1064 }
1065 EXPORT_SYMBOL_GPL(mhi_free_controller);
1066 
1067 int mhi_prepare_for_power_up(struct mhi_controller *mhi_cntrl)
1068 {
1069 	struct device *dev = &mhi_cntrl->mhi_dev->dev;
1070 	u32 bhi_off, bhie_off;
1071 	int ret;
1072 
1073 	mutex_lock(&mhi_cntrl->pm_mutex);
1074 
1075 	ret = mhi_init_dev_ctxt(mhi_cntrl);
1076 	if (ret)
1077 		goto error_dev_ctxt;
1078 
1079 	ret = mhi_read_reg(mhi_cntrl, mhi_cntrl->regs, BHIOFF, &bhi_off);
1080 	if (ret) {
1081 		dev_err(dev, "Error getting BHI offset\n");
1082 		goto error_reg_offset;
1083 	}
1084 
1085 	if (bhi_off >= mhi_cntrl->reg_len) {
1086 		dev_err(dev, "BHI offset: 0x%x is out of range: 0x%zx\n",
1087 			bhi_off, mhi_cntrl->reg_len);
1088 		ret = -EINVAL;
1089 		goto error_reg_offset;
1090 	}
1091 	mhi_cntrl->bhi = mhi_cntrl->regs + bhi_off;
1092 
1093 	if (mhi_cntrl->fbc_download || mhi_cntrl->rddm_size) {
1094 		ret = mhi_read_reg(mhi_cntrl, mhi_cntrl->regs, BHIEOFF,
1095 				   &bhie_off);
1096 		if (ret) {
1097 			dev_err(dev, "Error getting BHIE offset\n");
1098 			goto error_reg_offset;
1099 		}
1100 
1101 		if (bhie_off >= mhi_cntrl->reg_len) {
1102 			dev_err(dev,
1103 				"BHIe offset: 0x%x is out of range: 0x%zx\n",
1104 				bhie_off, mhi_cntrl->reg_len);
1105 			ret = -EINVAL;
1106 			goto error_reg_offset;
1107 		}
1108 		mhi_cntrl->bhie = mhi_cntrl->regs + bhie_off;
1109 	}
1110 
1111 	if (mhi_cntrl->rddm_size) {
1112 		/*
1113 		 * This controller supports RDDM, so we need to manually clear
1114 		 * BHIE RX registers since POR values are undefined.
1115 		 */
1116 		memset_io(mhi_cntrl->bhie + BHIE_RXVECADDR_LOW_OFFS,
1117 			  0, BHIE_RXVECSTATUS_OFFS - BHIE_RXVECADDR_LOW_OFFS +
1118 			  4);
1119 		/*
1120 		 * Allocate RDDM table for debugging purpose if specified
1121 		 */
1122 		mhi_alloc_bhie_table(mhi_cntrl, &mhi_cntrl->rddm_image,
1123 				     mhi_cntrl->rddm_size);
1124 		if (mhi_cntrl->rddm_image) {
1125 			ret = mhi_rddm_prepare(mhi_cntrl,
1126 					       mhi_cntrl->rddm_image);
1127 			if (ret) {
1128 				mhi_free_bhie_table(mhi_cntrl,
1129 						    mhi_cntrl->rddm_image);
1130 				goto error_reg_offset;
1131 			}
1132 		}
1133 	}
1134 
1135 	mutex_unlock(&mhi_cntrl->pm_mutex);
1136 
1137 	return 0;
1138 
1139 error_reg_offset:
1140 	mhi_deinit_dev_ctxt(mhi_cntrl);
1141 
1142 error_dev_ctxt:
1143 	mutex_unlock(&mhi_cntrl->pm_mutex);
1144 
1145 	return ret;
1146 }
1147 EXPORT_SYMBOL_GPL(mhi_prepare_for_power_up);
1148 
1149 void mhi_unprepare_after_power_down(struct mhi_controller *mhi_cntrl)
1150 {
1151 	if (mhi_cntrl->fbc_image) {
1152 		mhi_free_bhie_table(mhi_cntrl, mhi_cntrl->fbc_image);
1153 		mhi_cntrl->fbc_image = NULL;
1154 	}
1155 
1156 	if (mhi_cntrl->rddm_image) {
1157 		mhi_free_bhie_table(mhi_cntrl, mhi_cntrl->rddm_image);
1158 		mhi_cntrl->rddm_image = NULL;
1159 	}
1160 
1161 	mhi_cntrl->bhi = NULL;
1162 	mhi_cntrl->bhie = NULL;
1163 
1164 	mhi_deinit_dev_ctxt(mhi_cntrl);
1165 }
1166 EXPORT_SYMBOL_GPL(mhi_unprepare_after_power_down);
1167 
1168 static void mhi_release_device(struct device *dev)
1169 {
1170 	struct mhi_device *mhi_dev = to_mhi_device(dev);
1171 
1172 	/*
1173 	 * We need to set the mhi_chan->mhi_dev to NULL here since the MHI
1174 	 * devices for the channels will only get created if the mhi_dev
1175 	 * associated with it is NULL. This scenario will happen during the
1176 	 * controller suspend and resume.
1177 	 */
1178 	if (mhi_dev->ul_chan)
1179 		mhi_dev->ul_chan->mhi_dev = NULL;
1180 
1181 	if (mhi_dev->dl_chan)
1182 		mhi_dev->dl_chan->mhi_dev = NULL;
1183 
1184 	kfree(mhi_dev);
1185 }
1186 
1187 struct mhi_device *mhi_alloc_device(struct mhi_controller *mhi_cntrl)
1188 {
1189 	struct mhi_device *mhi_dev;
1190 	struct device *dev;
1191 
1192 	mhi_dev = kzalloc(sizeof(*mhi_dev), GFP_KERNEL);
1193 	if (!mhi_dev)
1194 		return ERR_PTR(-ENOMEM);
1195 
1196 	dev = &mhi_dev->dev;
1197 	device_initialize(dev);
1198 	dev->bus = &mhi_bus_type;
1199 	dev->release = mhi_release_device;
1200 
1201 	if (mhi_cntrl->mhi_dev) {
1202 		/* for MHI client devices, parent is the MHI controller device */
1203 		dev->parent = &mhi_cntrl->mhi_dev->dev;
1204 	} else {
1205 		/* for MHI controller device, parent is the bus device (e.g. pci device) */
1206 		dev->parent = mhi_cntrl->cntrl_dev;
1207 	}
1208 
1209 	mhi_dev->mhi_cntrl = mhi_cntrl;
1210 	mhi_dev->dev_wake = 0;
1211 
1212 	return mhi_dev;
1213 }
1214 
1215 static int mhi_driver_probe(struct device *dev)
1216 {
1217 	struct mhi_device *mhi_dev = to_mhi_device(dev);
1218 	struct mhi_controller *mhi_cntrl = mhi_dev->mhi_cntrl;
1219 	struct device_driver *drv = dev->driver;
1220 	struct mhi_driver *mhi_drv = to_mhi_driver(drv);
1221 	struct mhi_event *mhi_event;
1222 	struct mhi_chan *ul_chan = mhi_dev->ul_chan;
1223 	struct mhi_chan *dl_chan = mhi_dev->dl_chan;
1224 	int ret;
1225 
1226 	/* Bring device out of LPM */
1227 	ret = mhi_device_get_sync(mhi_dev);
1228 	if (ret)
1229 		return ret;
1230 
1231 	ret = -EINVAL;
1232 
1233 	if (ul_chan) {
1234 		/*
1235 		 * If channel supports LPM notifications then status_cb should
1236 		 * be provided
1237 		 */
1238 		if (ul_chan->lpm_notify && !mhi_drv->status_cb)
1239 			goto exit_probe;
1240 
1241 		/* For non-offload channels then xfer_cb should be provided */
1242 		if (!ul_chan->offload_ch && !mhi_drv->ul_xfer_cb)
1243 			goto exit_probe;
1244 
1245 		ul_chan->xfer_cb = mhi_drv->ul_xfer_cb;
1246 	}
1247 
1248 	ret = -EINVAL;
1249 	if (dl_chan) {
1250 		/*
1251 		 * If channel supports LPM notifications then status_cb should
1252 		 * be provided
1253 		 */
1254 		if (dl_chan->lpm_notify && !mhi_drv->status_cb)
1255 			goto exit_probe;
1256 
1257 		/* For non-offload channels then xfer_cb should be provided */
1258 		if (!dl_chan->offload_ch && !mhi_drv->dl_xfer_cb)
1259 			goto exit_probe;
1260 
1261 		mhi_event = &mhi_cntrl->mhi_event[dl_chan->er_index];
1262 
1263 		/*
1264 		 * If the channel event ring is managed by client, then
1265 		 * status_cb must be provided so that the framework can
1266 		 * notify pending data
1267 		 */
1268 		if (mhi_event->cl_manage && !mhi_drv->status_cb)
1269 			goto exit_probe;
1270 
1271 		dl_chan->xfer_cb = mhi_drv->dl_xfer_cb;
1272 	}
1273 
1274 	/* Call the user provided probe function */
1275 	ret = mhi_drv->probe(mhi_dev, mhi_dev->id);
1276 	if (ret)
1277 		goto exit_probe;
1278 
1279 	mhi_device_put(mhi_dev);
1280 
1281 	return ret;
1282 
1283 exit_probe:
1284 	mhi_unprepare_from_transfer(mhi_dev);
1285 
1286 	mhi_device_put(mhi_dev);
1287 
1288 	return ret;
1289 }
1290 
1291 static int mhi_driver_remove(struct device *dev)
1292 {
1293 	struct mhi_device *mhi_dev = to_mhi_device(dev);
1294 	struct mhi_driver *mhi_drv = to_mhi_driver(dev->driver);
1295 	struct mhi_controller *mhi_cntrl = mhi_dev->mhi_cntrl;
1296 	struct mhi_chan *mhi_chan;
1297 	enum mhi_ch_state ch_state[] = {
1298 		MHI_CH_STATE_DISABLED,
1299 		MHI_CH_STATE_DISABLED
1300 	};
1301 	int dir;
1302 
1303 	/* Skip if it is a controller device */
1304 	if (mhi_dev->dev_type == MHI_DEVICE_CONTROLLER)
1305 		return 0;
1306 
1307 	/* Reset both channels */
1308 	for (dir = 0; dir < 2; dir++) {
1309 		mhi_chan = dir ? mhi_dev->ul_chan : mhi_dev->dl_chan;
1310 
1311 		if (!mhi_chan)
1312 			continue;
1313 
1314 		/* Wake all threads waiting for completion */
1315 		write_lock_irq(&mhi_chan->lock);
1316 		mhi_chan->ccs = MHI_EV_CC_INVALID;
1317 		complete_all(&mhi_chan->completion);
1318 		write_unlock_irq(&mhi_chan->lock);
1319 
1320 		/* Set the channel state to disabled */
1321 		mutex_lock(&mhi_chan->mutex);
1322 		write_lock_irq(&mhi_chan->lock);
1323 		ch_state[dir] = mhi_chan->ch_state;
1324 		mhi_chan->ch_state = MHI_CH_STATE_SUSPENDED;
1325 		write_unlock_irq(&mhi_chan->lock);
1326 
1327 		/* Reset the non-offload channel */
1328 		if (!mhi_chan->offload_ch)
1329 			mhi_reset_chan(mhi_cntrl, mhi_chan);
1330 
1331 		mutex_unlock(&mhi_chan->mutex);
1332 	}
1333 
1334 	mhi_drv->remove(mhi_dev);
1335 
1336 	/* De-init channel if it was enabled */
1337 	for (dir = 0; dir < 2; dir++) {
1338 		mhi_chan = dir ? mhi_dev->ul_chan : mhi_dev->dl_chan;
1339 
1340 		if (!mhi_chan)
1341 			continue;
1342 
1343 		mutex_lock(&mhi_chan->mutex);
1344 
1345 		if ((ch_state[dir] == MHI_CH_STATE_ENABLED ||
1346 		     ch_state[dir] == MHI_CH_STATE_STOP) &&
1347 		    !mhi_chan->offload_ch)
1348 			mhi_deinit_chan_ctxt(mhi_cntrl, mhi_chan);
1349 
1350 		mhi_chan->ch_state = MHI_CH_STATE_DISABLED;
1351 
1352 		mutex_unlock(&mhi_chan->mutex);
1353 	}
1354 
1355 	while (mhi_dev->dev_wake)
1356 		mhi_device_put(mhi_dev);
1357 
1358 	return 0;
1359 }
1360 
1361 int __mhi_driver_register(struct mhi_driver *mhi_drv, struct module *owner)
1362 {
1363 	struct device_driver *driver = &mhi_drv->driver;
1364 
1365 	if (!mhi_drv->probe || !mhi_drv->remove)
1366 		return -EINVAL;
1367 
1368 	driver->bus = &mhi_bus_type;
1369 	driver->owner = owner;
1370 	driver->probe = mhi_driver_probe;
1371 	driver->remove = mhi_driver_remove;
1372 
1373 	return driver_register(driver);
1374 }
1375 EXPORT_SYMBOL_GPL(__mhi_driver_register);
1376 
1377 void mhi_driver_unregister(struct mhi_driver *mhi_drv)
1378 {
1379 	driver_unregister(&mhi_drv->driver);
1380 }
1381 EXPORT_SYMBOL_GPL(mhi_driver_unregister);
1382 
1383 static int mhi_uevent(struct device *dev, struct kobj_uevent_env *env)
1384 {
1385 	struct mhi_device *mhi_dev = to_mhi_device(dev);
1386 
1387 	return add_uevent_var(env, "MODALIAS=" MHI_DEVICE_MODALIAS_FMT,
1388 					mhi_dev->name);
1389 }
1390 
1391 static int mhi_match(struct device *dev, struct device_driver *drv)
1392 {
1393 	struct mhi_device *mhi_dev = to_mhi_device(dev);
1394 	struct mhi_driver *mhi_drv = to_mhi_driver(drv);
1395 	const struct mhi_device_id *id;
1396 
1397 	/*
1398 	 * If the device is a controller type then there is no client driver
1399 	 * associated with it
1400 	 */
1401 	if (mhi_dev->dev_type == MHI_DEVICE_CONTROLLER)
1402 		return 0;
1403 
1404 	for (id = mhi_drv->id_table; id->chan[0]; id++)
1405 		if (!strcmp(mhi_dev->name, id->chan)) {
1406 			mhi_dev->id = id;
1407 			return 1;
1408 		}
1409 
1410 	return 0;
1411 };
1412 
1413 struct bus_type mhi_bus_type = {
1414 	.name = "mhi",
1415 	.dev_name = "mhi",
1416 	.match = mhi_match,
1417 	.uevent = mhi_uevent,
1418 	.dev_groups = mhi_dev_groups,
1419 };
1420 
1421 static int __init mhi_init(void)
1422 {
1423 	mhi_debugfs_init();
1424 	return bus_register(&mhi_bus_type);
1425 }
1426 
1427 static void __exit mhi_exit(void)
1428 {
1429 	mhi_debugfs_exit();
1430 	bus_unregister(&mhi_bus_type);
1431 }
1432 
1433 postcore_initcall(mhi_init);
1434 module_exit(mhi_exit);
1435 
1436 MODULE_LICENSE("GPL v2");
1437 MODULE_DESCRIPTION("MHI Host Interface");
1438