Merge tag 'nfsd-6.0-1' of git://git.kernel.org/pub/scm/linux/kernel/git/cel/linux
[platform/kernel/linux-starfive.git] / drivers / soundwire / stream.c
1 // SPDX-License-Identifier: (GPL-2.0 OR BSD-3-Clause)
2 // Copyright(c) 2015-18 Intel Corporation.
3
4 /*
5  *  stream.c - SoundWire Bus stream operations.
6  */
7
8 #include <linux/delay.h>
9 #include <linux/device.h>
10 #include <linux/init.h>
11 #include <linux/module.h>
12 #include <linux/mod_devicetable.h>
13 #include <linux/slab.h>
14 #include <linux/soundwire/sdw_registers.h>
15 #include <linux/soundwire/sdw.h>
16 #include <linux/soundwire/sdw_type.h>
17 #include <sound/soc.h>
18 #include "bus.h"
19
20 /*
21  * Array of supported rows and columns as per MIPI SoundWire Specification 1.1
22  *
23  * The rows are arranged as per the array index value programmed
24  * in register. The index 15 has dummy value 0 in order to fill hole.
25  */
26 int sdw_rows[SDW_FRAME_ROWS] = {48, 50, 60, 64, 75, 80, 125, 147,
27                         96, 100, 120, 128, 150, 160, 250, 0,
28                         192, 200, 240, 256, 72, 144, 90, 180};
29 EXPORT_SYMBOL(sdw_rows);
30
31 int sdw_cols[SDW_FRAME_COLS] = {2, 4, 6, 8, 10, 12, 14, 16};
32 EXPORT_SYMBOL(sdw_cols);
33
34 int sdw_find_col_index(int col)
35 {
36         int i;
37
38         for (i = 0; i < SDW_FRAME_COLS; i++) {
39                 if (sdw_cols[i] == col)
40                         return i;
41         }
42
43         pr_warn("Requested column not found, selecting lowest column no: 2\n");
44         return 0;
45 }
46 EXPORT_SYMBOL(sdw_find_col_index);
47
48 int sdw_find_row_index(int row)
49 {
50         int i;
51
52         for (i = 0; i < SDW_FRAME_ROWS; i++) {
53                 if (sdw_rows[i] == row)
54                         return i;
55         }
56
57         pr_warn("Requested row not found, selecting lowest row no: 48\n");
58         return 0;
59 }
60 EXPORT_SYMBOL(sdw_find_row_index);
61
62 static int _sdw_program_slave_port_params(struct sdw_bus *bus,
63                                           struct sdw_slave *slave,
64                                           struct sdw_transport_params *t_params,
65                                           enum sdw_dpn_type type)
66 {
67         u32 addr1, addr2, addr3, addr4;
68         int ret;
69         u16 wbuf;
70
71         if (bus->params.next_bank) {
72                 addr1 = SDW_DPN_OFFSETCTRL2_B1(t_params->port_num);
73                 addr2 = SDW_DPN_BLOCKCTRL3_B1(t_params->port_num);
74                 addr3 = SDW_DPN_SAMPLECTRL2_B1(t_params->port_num);
75                 addr4 = SDW_DPN_HCTRL_B1(t_params->port_num);
76         } else {
77                 addr1 = SDW_DPN_OFFSETCTRL2_B0(t_params->port_num);
78                 addr2 = SDW_DPN_BLOCKCTRL3_B0(t_params->port_num);
79                 addr3 = SDW_DPN_SAMPLECTRL2_B0(t_params->port_num);
80                 addr4 = SDW_DPN_HCTRL_B0(t_params->port_num);
81         }
82
83         /* Program DPN_OffsetCtrl2 registers */
84         ret = sdw_write(slave, addr1, t_params->offset2);
85         if (ret < 0) {
86                 dev_err(bus->dev, "DPN_OffsetCtrl2 register write failed\n");
87                 return ret;
88         }
89
90         /* Program DPN_BlockCtrl3 register */
91         ret = sdw_write(slave, addr2, t_params->blk_pkg_mode);
92         if (ret < 0) {
93                 dev_err(bus->dev, "DPN_BlockCtrl3 register write failed\n");
94                 return ret;
95         }
96
97         /*
98          * Data ports are FULL, SIMPLE and REDUCED. This function handles
99          * FULL and REDUCED only and beyond this point only FULL is
100          * handled, so bail out if we are not FULL data port type
101          */
102         if (type != SDW_DPN_FULL)
103                 return ret;
104
105         /* Program DPN_SampleCtrl2 register */
106         wbuf = FIELD_GET(SDW_DPN_SAMPLECTRL_HIGH, t_params->sample_interval - 1);
107
108         ret = sdw_write(slave, addr3, wbuf);
109         if (ret < 0) {
110                 dev_err(bus->dev, "DPN_SampleCtrl2 register write failed\n");
111                 return ret;
112         }
113
114         /* Program DPN_HCtrl register */
115         wbuf = FIELD_PREP(SDW_DPN_HCTRL_HSTART, t_params->hstart);
116         wbuf |= FIELD_PREP(SDW_DPN_HCTRL_HSTOP, t_params->hstop);
117
118         ret = sdw_write(slave, addr4, wbuf);
119         if (ret < 0)
120                 dev_err(bus->dev, "DPN_HCtrl register write failed\n");
121
122         return ret;
123 }
124
125 static int sdw_program_slave_port_params(struct sdw_bus *bus,
126                                          struct sdw_slave_runtime *s_rt,
127                                          struct sdw_port_runtime *p_rt)
128 {
129         struct sdw_transport_params *t_params = &p_rt->transport_params;
130         struct sdw_port_params *p_params = &p_rt->port_params;
131         struct sdw_slave_prop *slave_prop = &s_rt->slave->prop;
132         u32 addr1, addr2, addr3, addr4, addr5, addr6;
133         struct sdw_dpn_prop *dpn_prop;
134         int ret;
135         u8 wbuf;
136
137         if (s_rt->slave->is_mockup_device)
138                 return 0;
139
140         dpn_prop = sdw_get_slave_dpn_prop(s_rt->slave,
141                                           s_rt->direction,
142                                           t_params->port_num);
143         if (!dpn_prop)
144                 return -EINVAL;
145
146         addr1 = SDW_DPN_PORTCTRL(t_params->port_num);
147         addr2 = SDW_DPN_BLOCKCTRL1(t_params->port_num);
148
149         if (bus->params.next_bank) {
150                 addr3 = SDW_DPN_SAMPLECTRL1_B1(t_params->port_num);
151                 addr4 = SDW_DPN_OFFSETCTRL1_B1(t_params->port_num);
152                 addr5 = SDW_DPN_BLOCKCTRL2_B1(t_params->port_num);
153                 addr6 = SDW_DPN_LANECTRL_B1(t_params->port_num);
154
155         } else {
156                 addr3 = SDW_DPN_SAMPLECTRL1_B0(t_params->port_num);
157                 addr4 = SDW_DPN_OFFSETCTRL1_B0(t_params->port_num);
158                 addr5 = SDW_DPN_BLOCKCTRL2_B0(t_params->port_num);
159                 addr6 = SDW_DPN_LANECTRL_B0(t_params->port_num);
160         }
161
162         /* Program DPN_PortCtrl register */
163         wbuf = FIELD_PREP(SDW_DPN_PORTCTRL_DATAMODE, p_params->data_mode);
164         wbuf |= FIELD_PREP(SDW_DPN_PORTCTRL_FLOWMODE, p_params->flow_mode);
165
166         ret = sdw_update(s_rt->slave, addr1, 0xF, wbuf);
167         if (ret < 0) {
168                 dev_err(&s_rt->slave->dev,
169                         "DPN_PortCtrl register write failed for port %d\n",
170                         t_params->port_num);
171                 return ret;
172         }
173
174         if (!dpn_prop->read_only_wordlength) {
175                 /* Program DPN_BlockCtrl1 register */
176                 ret = sdw_write(s_rt->slave, addr2, (p_params->bps - 1));
177                 if (ret < 0) {
178                         dev_err(&s_rt->slave->dev,
179                                 "DPN_BlockCtrl1 register write failed for port %d\n",
180                                 t_params->port_num);
181                         return ret;
182                 }
183         }
184
185         /* Program DPN_SampleCtrl1 register */
186         wbuf = (t_params->sample_interval - 1) & SDW_DPN_SAMPLECTRL_LOW;
187         ret = sdw_write(s_rt->slave, addr3, wbuf);
188         if (ret < 0) {
189                 dev_err(&s_rt->slave->dev,
190                         "DPN_SampleCtrl1 register write failed for port %d\n",
191                         t_params->port_num);
192                 return ret;
193         }
194
195         /* Program DPN_OffsetCtrl1 registers */
196         ret = sdw_write(s_rt->slave, addr4, t_params->offset1);
197         if (ret < 0) {
198                 dev_err(&s_rt->slave->dev,
199                         "DPN_OffsetCtrl1 register write failed for port %d\n",
200                         t_params->port_num);
201                 return ret;
202         }
203
204         /* Program DPN_BlockCtrl2 register*/
205         if (t_params->blk_grp_ctrl_valid) {
206                 ret = sdw_write(s_rt->slave, addr5, t_params->blk_grp_ctrl);
207                 if (ret < 0) {
208                         dev_err(&s_rt->slave->dev,
209                                 "DPN_BlockCtrl2 reg write failed for port %d\n",
210                                 t_params->port_num);
211                         return ret;
212                 }
213         }
214
215         /* program DPN_LaneCtrl register */
216         if (slave_prop->lane_control_support) {
217                 ret = sdw_write(s_rt->slave, addr6, t_params->lane_ctrl);
218                 if (ret < 0) {
219                         dev_err(&s_rt->slave->dev,
220                                 "DPN_LaneCtrl register write failed for port %d\n",
221                                 t_params->port_num);
222                         return ret;
223                 }
224         }
225
226         if (dpn_prop->type != SDW_DPN_SIMPLE) {
227                 ret = _sdw_program_slave_port_params(bus, s_rt->slave,
228                                                      t_params, dpn_prop->type);
229                 if (ret < 0)
230                         dev_err(&s_rt->slave->dev,
231                                 "Transport reg write failed for port: %d\n",
232                                 t_params->port_num);
233         }
234
235         return ret;
236 }
237
238 static int sdw_program_master_port_params(struct sdw_bus *bus,
239                                           struct sdw_port_runtime *p_rt)
240 {
241         int ret;
242
243         /*
244          * we need to set transport and port parameters for the port.
245          * Transport parameters refers to the sample interval, offsets and
246          * hstart/stop etc of the data. Port parameters refers to word
247          * length, flow mode etc of the port
248          */
249         ret = bus->port_ops->dpn_set_port_transport_params(bus,
250                                         &p_rt->transport_params,
251                                         bus->params.next_bank);
252         if (ret < 0)
253                 return ret;
254
255         return bus->port_ops->dpn_set_port_params(bus,
256                                                   &p_rt->port_params,
257                                                   bus->params.next_bank);
258 }
259
260 /**
261  * sdw_program_port_params() - Programs transport parameters of Master(s)
262  * and Slave(s)
263  *
264  * @m_rt: Master stream runtime
265  */
266 static int sdw_program_port_params(struct sdw_master_runtime *m_rt)
267 {
268         struct sdw_slave_runtime *s_rt;
269         struct sdw_bus *bus = m_rt->bus;
270         struct sdw_port_runtime *p_rt;
271         int ret = 0;
272
273         /* Program transport & port parameters for Slave(s) */
274         list_for_each_entry(s_rt, &m_rt->slave_rt_list, m_rt_node) {
275                 list_for_each_entry(p_rt, &s_rt->port_list, port_node) {
276                         ret = sdw_program_slave_port_params(bus, s_rt, p_rt);
277                         if (ret < 0)
278                                 return ret;
279                 }
280         }
281
282         /* Program transport & port parameters for Master(s) */
283         list_for_each_entry(p_rt, &m_rt->port_list, port_node) {
284                 ret = sdw_program_master_port_params(bus, p_rt);
285                 if (ret < 0)
286                         return ret;
287         }
288
289         return 0;
290 }
291
292 /**
293  * sdw_enable_disable_slave_ports: Enable/disable slave data port
294  *
295  * @bus: bus instance
296  * @s_rt: slave runtime
297  * @p_rt: port runtime
298  * @en: enable or disable operation
299  *
300  * This function only sets the enable/disable bits in the relevant bank, the
301  * actual enable/disable is done with a bank switch
302  */
303 static int sdw_enable_disable_slave_ports(struct sdw_bus *bus,
304                                           struct sdw_slave_runtime *s_rt,
305                                           struct sdw_port_runtime *p_rt,
306                                           bool en)
307 {
308         struct sdw_transport_params *t_params = &p_rt->transport_params;
309         u32 addr;
310         int ret;
311
312         if (bus->params.next_bank)
313                 addr = SDW_DPN_CHANNELEN_B1(p_rt->num);
314         else
315                 addr = SDW_DPN_CHANNELEN_B0(p_rt->num);
316
317         /*
318          * Since bus doesn't support sharing a port across two streams,
319          * it is safe to reset this register
320          */
321         if (en)
322                 ret = sdw_write(s_rt->slave, addr, p_rt->ch_mask);
323         else
324                 ret = sdw_write(s_rt->slave, addr, 0x0);
325
326         if (ret < 0)
327                 dev_err(&s_rt->slave->dev,
328                         "Slave chn_en reg write failed:%d port:%d\n",
329                         ret, t_params->port_num);
330
331         return ret;
332 }
333
334 static int sdw_enable_disable_master_ports(struct sdw_master_runtime *m_rt,
335                                            struct sdw_port_runtime *p_rt,
336                                            bool en)
337 {
338         struct sdw_transport_params *t_params = &p_rt->transport_params;
339         struct sdw_bus *bus = m_rt->bus;
340         struct sdw_enable_ch enable_ch;
341         int ret;
342
343         enable_ch.port_num = p_rt->num;
344         enable_ch.ch_mask = p_rt->ch_mask;
345         enable_ch.enable = en;
346
347         /* Perform Master port channel(s) enable/disable */
348         if (bus->port_ops->dpn_port_enable_ch) {
349                 ret = bus->port_ops->dpn_port_enable_ch(bus,
350                                                         &enable_ch,
351                                                         bus->params.next_bank);
352                 if (ret < 0) {
353                         dev_err(bus->dev,
354                                 "Master chn_en write failed:%d port:%d\n",
355                                 ret, t_params->port_num);
356                         return ret;
357                 }
358         } else {
359                 dev_err(bus->dev,
360                         "dpn_port_enable_ch not supported, %s failed\n",
361                         en ? "enable" : "disable");
362                 return -EINVAL;
363         }
364
365         return 0;
366 }
367
368 /**
369  * sdw_enable_disable_ports() - Enable/disable port(s) for Master and
370  * Slave(s)
371  *
372  * @m_rt: Master stream runtime
373  * @en: mode (enable/disable)
374  */
375 static int sdw_enable_disable_ports(struct sdw_master_runtime *m_rt, bool en)
376 {
377         struct sdw_port_runtime *s_port, *m_port;
378         struct sdw_slave_runtime *s_rt;
379         int ret = 0;
380
381         /* Enable/Disable Slave port(s) */
382         list_for_each_entry(s_rt, &m_rt->slave_rt_list, m_rt_node) {
383                 list_for_each_entry(s_port, &s_rt->port_list, port_node) {
384                         ret = sdw_enable_disable_slave_ports(m_rt->bus, s_rt,
385                                                              s_port, en);
386                         if (ret < 0)
387                                 return ret;
388                 }
389         }
390
391         /* Enable/Disable Master port(s) */
392         list_for_each_entry(m_port, &m_rt->port_list, port_node) {
393                 ret = sdw_enable_disable_master_ports(m_rt, m_port, en);
394                 if (ret < 0)
395                         return ret;
396         }
397
398         return 0;
399 }
400
401 static int sdw_do_port_prep(struct sdw_slave_runtime *s_rt,
402                             struct sdw_prepare_ch prep_ch,
403                             enum sdw_port_prep_ops cmd)
404 {
405         int ret = 0;
406         struct sdw_slave *slave = s_rt->slave;
407
408         mutex_lock(&slave->sdw_dev_lock);
409
410         if (slave->probed) {
411                 struct device *dev = &slave->dev;
412                 struct sdw_driver *drv = drv_to_sdw_driver(dev->driver);
413
414                 if (drv->ops && drv->ops->port_prep) {
415                         ret = drv->ops->port_prep(slave, &prep_ch, cmd);
416                         if (ret < 0)
417                                 dev_err(dev, "Slave Port Prep cmd %d failed: %d\n",
418                                         cmd, ret);
419                 }
420         }
421
422         mutex_unlock(&slave->sdw_dev_lock);
423
424         return ret;
425 }
426
427 static int sdw_prep_deprep_slave_ports(struct sdw_bus *bus,
428                                        struct sdw_slave_runtime *s_rt,
429                                        struct sdw_port_runtime *p_rt,
430                                        bool prep)
431 {
432         struct completion *port_ready;
433         struct sdw_dpn_prop *dpn_prop;
434         struct sdw_prepare_ch prep_ch;
435         bool intr = false;
436         int ret = 0, val;
437         u32 addr;
438
439         prep_ch.num = p_rt->num;
440         prep_ch.ch_mask = p_rt->ch_mask;
441
442         dpn_prop = sdw_get_slave_dpn_prop(s_rt->slave,
443                                           s_rt->direction,
444                                           prep_ch.num);
445         if (!dpn_prop) {
446                 dev_err(bus->dev,
447                         "Slave Port:%d properties not found\n", prep_ch.num);
448                 return -EINVAL;
449         }
450
451         prep_ch.prepare = prep;
452
453         prep_ch.bank = bus->params.next_bank;
454
455         if (dpn_prop->imp_def_interrupts || !dpn_prop->simple_ch_prep_sm ||
456             bus->params.s_data_mode != SDW_PORT_DATA_MODE_NORMAL)
457                 intr = true;
458
459         /*
460          * Enable interrupt before Port prepare.
461          * For Port de-prepare, it is assumed that port
462          * was prepared earlier
463          */
464         if (prep && intr) {
465                 ret = sdw_configure_dpn_intr(s_rt->slave, p_rt->num, prep,
466                                              dpn_prop->imp_def_interrupts);
467                 if (ret < 0)
468                         return ret;
469         }
470
471         /* Inform slave about the impending port prepare */
472         sdw_do_port_prep(s_rt, prep_ch, SDW_OPS_PORT_PRE_PREP);
473
474         /* Prepare Slave port implementing CP_SM */
475         if (!dpn_prop->simple_ch_prep_sm) {
476                 addr = SDW_DPN_PREPARECTRL(p_rt->num);
477
478                 if (prep)
479                         ret = sdw_write(s_rt->slave, addr, p_rt->ch_mask);
480                 else
481                         ret = sdw_write(s_rt->slave, addr, 0x0);
482
483                 if (ret < 0) {
484                         dev_err(&s_rt->slave->dev,
485                                 "Slave prep_ctrl reg write failed\n");
486                         return ret;
487                 }
488
489                 /* Wait for completion on port ready */
490                 port_ready = &s_rt->slave->port_ready[prep_ch.num];
491                 wait_for_completion_timeout(port_ready,
492                         msecs_to_jiffies(dpn_prop->ch_prep_timeout));
493
494                 val = sdw_read(s_rt->slave, SDW_DPN_PREPARESTATUS(p_rt->num));
495                 if ((val < 0) || (val & p_rt->ch_mask)) {
496                         ret = (val < 0) ? val : -ETIMEDOUT;
497                         dev_err(&s_rt->slave->dev,
498                                 "Chn prep failed for port %d: %d\n", prep_ch.num, ret);
499                         return ret;
500                 }
501         }
502
503         /* Inform slaves about ports prepared */
504         sdw_do_port_prep(s_rt, prep_ch, SDW_OPS_PORT_POST_PREP);
505
506         /* Disable interrupt after Port de-prepare */
507         if (!prep && intr)
508                 ret = sdw_configure_dpn_intr(s_rt->slave, p_rt->num, prep,
509                                              dpn_prop->imp_def_interrupts);
510
511         return ret;
512 }
513
514 static int sdw_prep_deprep_master_ports(struct sdw_master_runtime *m_rt,
515                                         struct sdw_port_runtime *p_rt,
516                                         bool prep)
517 {
518         struct sdw_transport_params *t_params = &p_rt->transport_params;
519         struct sdw_bus *bus = m_rt->bus;
520         const struct sdw_master_port_ops *ops = bus->port_ops;
521         struct sdw_prepare_ch prep_ch;
522         int ret = 0;
523
524         prep_ch.num = p_rt->num;
525         prep_ch.ch_mask = p_rt->ch_mask;
526         prep_ch.prepare = prep; /* Prepare/De-prepare */
527         prep_ch.bank = bus->params.next_bank;
528
529         /* Pre-prepare/Pre-deprepare port(s) */
530         if (ops->dpn_port_prep) {
531                 ret = ops->dpn_port_prep(bus, &prep_ch);
532                 if (ret < 0) {
533                         dev_err(bus->dev, "Port prepare failed for port:%d\n",
534                                 t_params->port_num);
535                         return ret;
536                 }
537         }
538
539         return ret;
540 }
541
542 /**
543  * sdw_prep_deprep_ports() - Prepare/De-prepare port(s) for Master(s) and
544  * Slave(s)
545  *
546  * @m_rt: Master runtime handle
547  * @prep: Prepare or De-prepare
548  */
549 static int sdw_prep_deprep_ports(struct sdw_master_runtime *m_rt, bool prep)
550 {
551         struct sdw_slave_runtime *s_rt;
552         struct sdw_port_runtime *p_rt;
553         int ret = 0;
554
555         /* Prepare/De-prepare Slave port(s) */
556         list_for_each_entry(s_rt, &m_rt->slave_rt_list, m_rt_node) {
557                 list_for_each_entry(p_rt, &s_rt->port_list, port_node) {
558                         ret = sdw_prep_deprep_slave_ports(m_rt->bus, s_rt,
559                                                           p_rt, prep);
560                         if (ret < 0)
561                                 return ret;
562                 }
563         }
564
565         /* Prepare/De-prepare Master port(s) */
566         list_for_each_entry(p_rt, &m_rt->port_list, port_node) {
567                 ret = sdw_prep_deprep_master_ports(m_rt, p_rt, prep);
568                 if (ret < 0)
569                         return ret;
570         }
571
572         return ret;
573 }
574
575 /**
576  * sdw_notify_config() - Notify bus configuration
577  *
578  * @m_rt: Master runtime handle
579  *
580  * This function notifies the Master(s) and Slave(s) of the
581  * new bus configuration.
582  */
583 static int sdw_notify_config(struct sdw_master_runtime *m_rt)
584 {
585         struct sdw_slave_runtime *s_rt;
586         struct sdw_bus *bus = m_rt->bus;
587         struct sdw_slave *slave;
588         int ret;
589
590         if (bus->ops->set_bus_conf) {
591                 ret = bus->ops->set_bus_conf(bus, &bus->params);
592                 if (ret < 0)
593                         return ret;
594         }
595
596         list_for_each_entry(s_rt, &m_rt->slave_rt_list, m_rt_node) {
597                 slave = s_rt->slave;
598
599                 mutex_lock(&slave->sdw_dev_lock);
600
601                 if (slave->probed) {
602                         struct device *dev = &slave->dev;
603                         struct sdw_driver *drv = drv_to_sdw_driver(dev->driver);
604
605                         if (drv->ops && drv->ops->bus_config) {
606                                 ret = drv->ops->bus_config(slave, &bus->params);
607                                 if (ret < 0) {
608                                         dev_err(dev, "Notify Slave: %d failed\n",
609                                                 slave->dev_num);
610                                         mutex_unlock(&slave->sdw_dev_lock);
611                                         return ret;
612                                 }
613                         }
614                 }
615
616                 mutex_unlock(&slave->sdw_dev_lock);
617         }
618
619         return 0;
620 }
621
622 /**
623  * sdw_program_params() - Program transport and port parameters for Master(s)
624  * and Slave(s)
625  *
626  * @bus: SDW bus instance
627  * @prepare: true if sdw_program_params() is called by _prepare.
628  */
629 static int sdw_program_params(struct sdw_bus *bus, bool prepare)
630 {
631         struct sdw_master_runtime *m_rt;
632         int ret = 0;
633
634         list_for_each_entry(m_rt, &bus->m_rt_list, bus_node) {
635
636                 /*
637                  * this loop walks through all master runtimes for a
638                  * bus, but the ports can only be configured while
639                  * explicitly preparing a stream or handling an
640                  * already-prepared stream otherwise.
641                  */
642                 if (!prepare &&
643                     m_rt->stream->state == SDW_STREAM_CONFIGURED)
644                         continue;
645
646                 ret = sdw_program_port_params(m_rt);
647                 if (ret < 0) {
648                         dev_err(bus->dev,
649                                 "Program transport params failed: %d\n", ret);
650                         return ret;
651                 }
652
653                 ret = sdw_notify_config(m_rt);
654                 if (ret < 0) {
655                         dev_err(bus->dev,
656                                 "Notify bus config failed: %d\n", ret);
657                         return ret;
658                 }
659
660                 /* Enable port(s) on alternate bank for all active streams */
661                 if (m_rt->stream->state != SDW_STREAM_ENABLED)
662                         continue;
663
664                 ret = sdw_enable_disable_ports(m_rt, true);
665                 if (ret < 0) {
666                         dev_err(bus->dev, "Enable channel failed: %d\n", ret);
667                         return ret;
668                 }
669         }
670
671         return ret;
672 }
673
674 static int sdw_bank_switch(struct sdw_bus *bus, int m_rt_count)
675 {
676         int col_index, row_index;
677         bool multi_link;
678         struct sdw_msg *wr_msg;
679         u8 *wbuf;
680         int ret;
681         u16 addr;
682
683         wr_msg = kzalloc(sizeof(*wr_msg), GFP_KERNEL);
684         if (!wr_msg)
685                 return -ENOMEM;
686
687         bus->defer_msg.msg = wr_msg;
688
689         wbuf = kzalloc(sizeof(*wbuf), GFP_KERNEL);
690         if (!wbuf) {
691                 ret = -ENOMEM;
692                 goto error_1;
693         }
694
695         /* Get row and column index to program register */
696         col_index = sdw_find_col_index(bus->params.col);
697         row_index = sdw_find_row_index(bus->params.row);
698         wbuf[0] = col_index | (row_index << 3);
699
700         if (bus->params.next_bank)
701                 addr = SDW_SCP_FRAMECTRL_B1;
702         else
703                 addr = SDW_SCP_FRAMECTRL_B0;
704
705         sdw_fill_msg(wr_msg, NULL, addr, 1, SDW_BROADCAST_DEV_NUM,
706                      SDW_MSG_FLAG_WRITE, wbuf);
707         wr_msg->ssp_sync = true;
708
709         /*
710          * Set the multi_link flag only when both the hardware supports
711          * and hardware-based sync is required
712          */
713         multi_link = bus->multi_link && (m_rt_count >= bus->hw_sync_min_links);
714
715         if (multi_link)
716                 ret = sdw_transfer_defer(bus, wr_msg, &bus->defer_msg);
717         else
718                 ret = sdw_transfer(bus, wr_msg);
719
720         if (ret < 0 && ret != -ENODATA) {
721                 dev_err(bus->dev, "Slave frame_ctrl reg write failed\n");
722                 goto error;
723         }
724
725         if (!multi_link) {
726                 kfree(wr_msg);
727                 kfree(wbuf);
728                 bus->defer_msg.msg = NULL;
729                 bus->params.curr_bank = !bus->params.curr_bank;
730                 bus->params.next_bank = !bus->params.next_bank;
731         }
732
733         return 0;
734
735 error:
736         kfree(wbuf);
737 error_1:
738         kfree(wr_msg);
739         bus->defer_msg.msg = NULL;
740         return ret;
741 }
742
743 /**
744  * sdw_ml_sync_bank_switch: Multilink register bank switch
745  *
746  * @bus: SDW bus instance
747  *
748  * Caller function should free the buffers on error
749  */
750 static int sdw_ml_sync_bank_switch(struct sdw_bus *bus)
751 {
752         unsigned long time_left;
753
754         if (!bus->multi_link)
755                 return 0;
756
757         /* Wait for completion of transfer */
758         time_left = wait_for_completion_timeout(&bus->defer_msg.complete,
759                                                 bus->bank_switch_timeout);
760
761         if (!time_left) {
762                 dev_err(bus->dev, "Controller Timed out on bank switch\n");
763                 return -ETIMEDOUT;
764         }
765
766         bus->params.curr_bank = !bus->params.curr_bank;
767         bus->params.next_bank = !bus->params.next_bank;
768
769         if (bus->defer_msg.msg) {
770                 kfree(bus->defer_msg.msg->buf);
771                 kfree(bus->defer_msg.msg);
772         }
773
774         return 0;
775 }
776
777 static int do_bank_switch(struct sdw_stream_runtime *stream)
778 {
779         struct sdw_master_runtime *m_rt;
780         const struct sdw_master_ops *ops;
781         struct sdw_bus *bus;
782         bool multi_link = false;
783         int m_rt_count;
784         int ret = 0;
785
786         m_rt_count = stream->m_rt_count;
787
788         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
789                 bus = m_rt->bus;
790                 ops = bus->ops;
791
792                 if (bus->multi_link && m_rt_count >= bus->hw_sync_min_links) {
793                         multi_link = true;
794                         mutex_lock(&bus->msg_lock);
795                 }
796
797                 /* Pre-bank switch */
798                 if (ops->pre_bank_switch) {
799                         ret = ops->pre_bank_switch(bus);
800                         if (ret < 0) {
801                                 dev_err(bus->dev,
802                                         "Pre bank switch op failed: %d\n", ret);
803                                 goto msg_unlock;
804                         }
805                 }
806
807                 /*
808                  * Perform Bank switch operation.
809                  * For multi link cases, the actual bank switch is
810                  * synchronized across all Masters and happens later as a
811                  * part of post_bank_switch ops.
812                  */
813                 ret = sdw_bank_switch(bus, m_rt_count);
814                 if (ret < 0) {
815                         dev_err(bus->dev, "Bank switch failed: %d\n", ret);
816                         goto error;
817                 }
818         }
819
820         /*
821          * For multi link cases, it is expected that the bank switch is
822          * triggered by the post_bank_switch for the first Master in the list
823          * and for the other Masters the post_bank_switch() should return doing
824          * nothing.
825          */
826         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
827                 bus = m_rt->bus;
828                 ops = bus->ops;
829
830                 /* Post-bank switch */
831                 if (ops->post_bank_switch) {
832                         ret = ops->post_bank_switch(bus);
833                         if (ret < 0) {
834                                 dev_err(bus->dev,
835                                         "Post bank switch op failed: %d\n",
836                                         ret);
837                                 goto error;
838                         }
839                 } else if (multi_link) {
840                         dev_err(bus->dev,
841                                 "Post bank switch ops not implemented\n");
842                         ret = -EINVAL;
843                         goto error;
844                 }
845
846                 /* Set the bank switch timeout to default, if not set */
847                 if (!bus->bank_switch_timeout)
848                         bus->bank_switch_timeout = DEFAULT_BANK_SWITCH_TIMEOUT;
849
850                 /* Check if bank switch was successful */
851                 ret = sdw_ml_sync_bank_switch(bus);
852                 if (ret < 0) {
853                         dev_err(bus->dev,
854                                 "multi link bank switch failed: %d\n", ret);
855                         goto error;
856                 }
857
858                 if (multi_link)
859                         mutex_unlock(&bus->msg_lock);
860         }
861
862         return ret;
863
864 error:
865         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
866                 bus = m_rt->bus;
867                 if (bus->defer_msg.msg) {
868                         kfree(bus->defer_msg.msg->buf);
869                         kfree(bus->defer_msg.msg);
870                 }
871         }
872
873 msg_unlock:
874
875         if (multi_link) {
876                 list_for_each_entry(m_rt, &stream->master_list, stream_node) {
877                         bus = m_rt->bus;
878                         if (mutex_is_locked(&bus->msg_lock))
879                                 mutex_unlock(&bus->msg_lock);
880                 }
881         }
882
883         return ret;
884 }
885
886 static struct sdw_port_runtime *sdw_port_alloc(struct list_head *port_list)
887 {
888         struct sdw_port_runtime *p_rt;
889
890         p_rt = kzalloc(sizeof(*p_rt), GFP_KERNEL);
891         if (!p_rt)
892                 return NULL;
893
894         list_add_tail(&p_rt->port_node, port_list);
895
896         return p_rt;
897 }
898
899 static int sdw_port_config(struct sdw_port_runtime *p_rt,
900                            struct sdw_port_config *port_config,
901                            int port_index)
902 {
903         p_rt->ch_mask = port_config[port_index].ch_mask;
904         p_rt->num = port_config[port_index].num;
905
906         /*
907          * TODO: Check port capabilities for requested configuration
908          */
909
910         return 0;
911 }
912
913 static void sdw_port_free(struct sdw_port_runtime *p_rt)
914 {
915         list_del(&p_rt->port_node);
916         kfree(p_rt);
917 }
918
919 static bool sdw_slave_port_allocated(struct sdw_slave_runtime *s_rt)
920 {
921         return !list_empty(&s_rt->port_list);
922 }
923
924 static void sdw_slave_port_free(struct sdw_slave *slave,
925                                 struct sdw_stream_runtime *stream)
926 {
927         struct sdw_port_runtime *p_rt, *_p_rt;
928         struct sdw_master_runtime *m_rt;
929         struct sdw_slave_runtime *s_rt;
930
931         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
932                 list_for_each_entry(s_rt, &m_rt->slave_rt_list, m_rt_node) {
933                         if (s_rt->slave != slave)
934                                 continue;
935
936                         list_for_each_entry_safe(p_rt, _p_rt,
937                                                  &s_rt->port_list, port_node) {
938                                 sdw_port_free(p_rt);
939                         }
940                 }
941         }
942 }
943
944 static int sdw_slave_port_alloc(struct sdw_slave *slave,
945                                 struct sdw_slave_runtime *s_rt,
946                                 unsigned int num_config)
947 {
948         struct sdw_port_runtime *p_rt;
949         int i;
950
951         /* Iterate for number of ports to perform initialization */
952         for (i = 0; i < num_config; i++) {
953                 p_rt = sdw_port_alloc(&s_rt->port_list);
954                 if (!p_rt)
955                         return -ENOMEM;
956         }
957
958         return 0;
959 }
960
961 static int sdw_slave_port_is_valid_range(struct device *dev, int num)
962 {
963         if (!SDW_VALID_PORT_RANGE(num)) {
964                 dev_err(dev, "SoundWire: Invalid port number :%d\n", num);
965                 return -EINVAL;
966         }
967
968         return 0;
969 }
970
971 static int sdw_slave_port_config(struct sdw_slave *slave,
972                                  struct sdw_slave_runtime *s_rt,
973                                  struct sdw_port_config *port_config)
974 {
975         struct sdw_port_runtime *p_rt;
976         int ret;
977         int i;
978
979         i = 0;
980         list_for_each_entry(p_rt, &s_rt->port_list, port_node) {
981                 /*
982                  * TODO: Check valid port range as defined by DisCo/
983                  * slave
984                  */
985                 ret = sdw_slave_port_is_valid_range(&slave->dev, port_config[i].num);
986                 if (ret < 0)
987                         return ret;
988
989                 ret = sdw_port_config(p_rt, port_config, i);
990                 if (ret < 0)
991                         return ret;
992                 i++;
993         }
994
995         return 0;
996 }
997
998 static bool sdw_master_port_allocated(struct sdw_master_runtime *m_rt)
999 {
1000         return !list_empty(&m_rt->port_list);
1001 }
1002
1003 static void sdw_master_port_free(struct sdw_master_runtime *m_rt)
1004 {
1005         struct sdw_port_runtime *p_rt, *_p_rt;
1006
1007         list_for_each_entry_safe(p_rt, _p_rt, &m_rt->port_list, port_node) {
1008                 sdw_port_free(p_rt);
1009         }
1010 }
1011
1012 static int sdw_master_port_alloc(struct sdw_master_runtime *m_rt,
1013                                  unsigned int num_ports)
1014 {
1015         struct sdw_port_runtime *p_rt;
1016         int i;
1017
1018         /* Iterate for number of ports to perform initialization */
1019         for (i = 0; i < num_ports; i++) {
1020                 p_rt = sdw_port_alloc(&m_rt->port_list);
1021                 if (!p_rt)
1022                         return -ENOMEM;
1023         }
1024
1025         return 0;
1026 }
1027
1028 static int sdw_master_port_config(struct sdw_master_runtime *m_rt,
1029                                   struct sdw_port_config *port_config)
1030 {
1031         struct sdw_port_runtime *p_rt;
1032         int ret;
1033         int i;
1034
1035         i = 0;
1036         list_for_each_entry(p_rt, &m_rt->port_list, port_node) {
1037                 ret = sdw_port_config(p_rt, port_config, i);
1038                 if (ret < 0)
1039                         return ret;
1040                 i++;
1041         }
1042
1043         return 0;
1044 }
1045
1046 /**
1047  * sdw_slave_rt_alloc() - Allocate a Slave runtime handle.
1048  *
1049  * @slave: Slave handle
1050  * @m_rt: Master runtime handle
1051  *
1052  * This function is to be called with bus_lock held.
1053  */
1054 static struct sdw_slave_runtime
1055 *sdw_slave_rt_alloc(struct sdw_slave *slave,
1056                     struct sdw_master_runtime *m_rt)
1057 {
1058         struct sdw_slave_runtime *s_rt;
1059
1060         s_rt = kzalloc(sizeof(*s_rt), GFP_KERNEL);
1061         if (!s_rt)
1062                 return NULL;
1063
1064         INIT_LIST_HEAD(&s_rt->port_list);
1065         s_rt->slave = slave;
1066
1067         list_add_tail(&s_rt->m_rt_node, &m_rt->slave_rt_list);
1068
1069         return s_rt;
1070 }
1071
1072 /**
1073  * sdw_slave_rt_config() - Configure a Slave runtime handle.
1074  *
1075  * @s_rt: Slave runtime handle
1076  * @stream_config: Stream configuration
1077  *
1078  * This function is to be called with bus_lock held.
1079  */
1080 static int sdw_slave_rt_config(struct sdw_slave_runtime *s_rt,
1081                                struct sdw_stream_config *stream_config)
1082 {
1083         s_rt->ch_count = stream_config->ch_count;
1084         s_rt->direction = stream_config->direction;
1085
1086         return 0;
1087 }
1088
1089 static struct sdw_slave_runtime *sdw_slave_rt_find(struct sdw_slave *slave,
1090                                                    struct sdw_stream_runtime *stream)
1091 {
1092         struct sdw_slave_runtime *s_rt, *_s_rt;
1093         struct sdw_master_runtime *m_rt;
1094
1095         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1096                 /* Retrieve Slave runtime handle */
1097                 list_for_each_entry_safe(s_rt, _s_rt,
1098                                          &m_rt->slave_rt_list, m_rt_node) {
1099                         if (s_rt->slave == slave)
1100                                 return s_rt;
1101                 }
1102         }
1103         return NULL;
1104 }
1105
1106 /**
1107  * sdw_slave_rt_free() - Free Slave(s) runtime handle
1108  *
1109  * @slave: Slave handle.
1110  * @stream: Stream runtime handle.
1111  *
1112  * This function is to be called with bus_lock held.
1113  */
1114 static void sdw_slave_rt_free(struct sdw_slave *slave,
1115                               struct sdw_stream_runtime *stream)
1116 {
1117         struct sdw_slave_runtime *s_rt;
1118
1119         s_rt = sdw_slave_rt_find(slave, stream);
1120         if (s_rt) {
1121                 list_del(&s_rt->m_rt_node);
1122                 kfree(s_rt);
1123         }
1124 }
1125
1126 static struct sdw_master_runtime
1127 *sdw_master_rt_find(struct sdw_bus *bus,
1128                     struct sdw_stream_runtime *stream)
1129 {
1130         struct sdw_master_runtime *m_rt;
1131
1132         /* Retrieve Bus handle if already available */
1133         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1134                 if (m_rt->bus == bus)
1135                         return m_rt;
1136         }
1137
1138         return NULL;
1139 }
1140
1141 /**
1142  * sdw_master_rt_alloc() - Allocates a Master runtime handle
1143  *
1144  * @bus: SDW bus instance
1145  * @stream: Stream runtime handle.
1146  *
1147  * This function is to be called with bus_lock held.
1148  */
1149 static struct sdw_master_runtime
1150 *sdw_master_rt_alloc(struct sdw_bus *bus,
1151                      struct sdw_stream_runtime *stream)
1152 {
1153         struct sdw_master_runtime *m_rt;
1154
1155         m_rt = kzalloc(sizeof(*m_rt), GFP_KERNEL);
1156         if (!m_rt)
1157                 return NULL;
1158
1159         /* Initialization of Master runtime handle */
1160         INIT_LIST_HEAD(&m_rt->port_list);
1161         INIT_LIST_HEAD(&m_rt->slave_rt_list);
1162         list_add_tail(&m_rt->stream_node, &stream->master_list);
1163
1164         list_add_tail(&m_rt->bus_node, &bus->m_rt_list);
1165
1166         m_rt->bus = bus;
1167         m_rt->stream = stream;
1168
1169         return m_rt;
1170 }
1171
1172 /**
1173  * sdw_master_rt_config() - Configure Master runtime handle
1174  *
1175  * @m_rt: Master runtime handle
1176  * @stream_config: Stream configuration
1177  *
1178  * This function is to be called with bus_lock held.
1179  */
1180
1181 static int sdw_master_rt_config(struct sdw_master_runtime *m_rt,
1182                                 struct sdw_stream_config *stream_config)
1183 {
1184         m_rt->ch_count = stream_config->ch_count;
1185         m_rt->direction = stream_config->direction;
1186
1187         return 0;
1188 }
1189
1190 /**
1191  * sdw_master_rt_free() - Free Master runtime handle
1192  *
1193  * @m_rt: Master runtime node
1194  * @stream: Stream runtime handle.
1195  *
1196  * This function is to be called with bus_lock held
1197  * It frees the Master runtime handle and associated Slave(s) runtime
1198  * handle. If this is called first then sdw_slave_rt_free() will have
1199  * no effect as Slave(s) runtime handle would already be freed up.
1200  */
1201 static void sdw_master_rt_free(struct sdw_master_runtime *m_rt,
1202                                struct sdw_stream_runtime *stream)
1203 {
1204         struct sdw_slave_runtime *s_rt, *_s_rt;
1205
1206         list_for_each_entry_safe(s_rt, _s_rt, &m_rt->slave_rt_list, m_rt_node) {
1207                 sdw_slave_port_free(s_rt->slave, stream);
1208                 sdw_slave_rt_free(s_rt->slave, stream);
1209         }
1210
1211         list_del(&m_rt->stream_node);
1212         list_del(&m_rt->bus_node);
1213         kfree(m_rt);
1214 }
1215
1216 /**
1217  * sdw_config_stream() - Configure the allocated stream
1218  *
1219  * @dev: SDW device
1220  * @stream: SoundWire stream
1221  * @stream_config: Stream configuration for audio stream
1222  * @is_slave: is API called from Slave or Master
1223  *
1224  * This function is to be called with bus_lock held.
1225  */
1226 static int sdw_config_stream(struct device *dev,
1227                              struct sdw_stream_runtime *stream,
1228                              struct sdw_stream_config *stream_config,
1229                              bool is_slave)
1230 {
1231         /*
1232          * Update the stream rate, channel and bps based on data
1233          * source. For more than one data source (multilink),
1234          * match the rate, bps, stream type and increment number of channels.
1235          *
1236          * If rate/bps is zero, it means the values are not set, so skip
1237          * comparison and allow the value to be set and stored in stream
1238          */
1239         if (stream->params.rate &&
1240             stream->params.rate != stream_config->frame_rate) {
1241                 dev_err(dev, "rate not matching, stream:%s\n", stream->name);
1242                 return -EINVAL;
1243         }
1244
1245         if (stream->params.bps &&
1246             stream->params.bps != stream_config->bps) {
1247                 dev_err(dev, "bps not matching, stream:%s\n", stream->name);
1248                 return -EINVAL;
1249         }
1250
1251         stream->type = stream_config->type;
1252         stream->params.rate = stream_config->frame_rate;
1253         stream->params.bps = stream_config->bps;
1254
1255         /* TODO: Update this check during Device-device support */
1256         if (is_slave)
1257                 stream->params.ch_count += stream_config->ch_count;
1258
1259         return 0;
1260 }
1261
1262 /**
1263  * sdw_get_slave_dpn_prop() - Get Slave port capabilities
1264  *
1265  * @slave: Slave handle
1266  * @direction: Data direction.
1267  * @port_num: Port number
1268  */
1269 struct sdw_dpn_prop *sdw_get_slave_dpn_prop(struct sdw_slave *slave,
1270                                             enum sdw_data_direction direction,
1271                                             unsigned int port_num)
1272 {
1273         struct sdw_dpn_prop *dpn_prop;
1274         u8 num_ports;
1275         int i;
1276
1277         if (direction == SDW_DATA_DIR_TX) {
1278                 num_ports = hweight32(slave->prop.source_ports);
1279                 dpn_prop = slave->prop.src_dpn_prop;
1280         } else {
1281                 num_ports = hweight32(slave->prop.sink_ports);
1282                 dpn_prop = slave->prop.sink_dpn_prop;
1283         }
1284
1285         for (i = 0; i < num_ports; i++) {
1286                 if (dpn_prop[i].num == port_num)
1287                         return &dpn_prop[i];
1288         }
1289
1290         return NULL;
1291 }
1292
1293 /**
1294  * sdw_acquire_bus_lock: Acquire bus lock for all Master runtime(s)
1295  *
1296  * @stream: SoundWire stream
1297  *
1298  * Acquire bus_lock for each of the master runtime(m_rt) part of this
1299  * stream to reconfigure the bus.
1300  * NOTE: This function is called from SoundWire stream ops and is
1301  * expected that a global lock is held before acquiring bus_lock.
1302  */
1303 static void sdw_acquire_bus_lock(struct sdw_stream_runtime *stream)
1304 {
1305         struct sdw_master_runtime *m_rt;
1306         struct sdw_bus *bus;
1307
1308         /* Iterate for all Master(s) in Master list */
1309         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1310                 bus = m_rt->bus;
1311
1312                 mutex_lock(&bus->bus_lock);
1313         }
1314 }
1315
1316 /**
1317  * sdw_release_bus_lock: Release bus lock for all Master runtime(s)
1318  *
1319  * @stream: SoundWire stream
1320  *
1321  * Release the previously held bus_lock after reconfiguring the bus.
1322  * NOTE: This function is called from SoundWire stream ops and is
1323  * expected that a global lock is held before releasing bus_lock.
1324  */
1325 static void sdw_release_bus_lock(struct sdw_stream_runtime *stream)
1326 {
1327         struct sdw_master_runtime *m_rt;
1328         struct sdw_bus *bus;
1329
1330         /* Iterate for all Master(s) in Master list */
1331         list_for_each_entry_reverse(m_rt, &stream->master_list, stream_node) {
1332                 bus = m_rt->bus;
1333                 mutex_unlock(&bus->bus_lock);
1334         }
1335 }
1336
1337 static int _sdw_prepare_stream(struct sdw_stream_runtime *stream,
1338                                bool update_params)
1339 {
1340         struct sdw_master_runtime *m_rt;
1341         struct sdw_bus *bus = NULL;
1342         struct sdw_master_prop *prop;
1343         struct sdw_bus_params params;
1344         int ret;
1345
1346         /* Prepare  Master(s) and Slave(s) port(s) associated with stream */
1347         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1348                 bus = m_rt->bus;
1349                 prop = &bus->prop;
1350                 memcpy(&params, &bus->params, sizeof(params));
1351
1352                 /* TODO: Support Asynchronous mode */
1353                 if ((prop->max_clk_freq % stream->params.rate) != 0) {
1354                         dev_err(bus->dev, "Async mode not supported\n");
1355                         return -EINVAL;
1356                 }
1357
1358                 if (!update_params)
1359                         goto program_params;
1360
1361                 /* Increment cumulative bus bandwidth */
1362                 /* TODO: Update this during Device-Device support */
1363                 bus->params.bandwidth += m_rt->stream->params.rate *
1364                         m_rt->ch_count * m_rt->stream->params.bps;
1365
1366                 /* Compute params */
1367                 if (bus->compute_params) {
1368                         ret = bus->compute_params(bus);
1369                         if (ret < 0) {
1370                                 dev_err(bus->dev, "Compute params failed: %d\n",
1371                                         ret);
1372                                 return ret;
1373                         }
1374                 }
1375
1376 program_params:
1377                 /* Program params */
1378                 ret = sdw_program_params(bus, true);
1379                 if (ret < 0) {
1380                         dev_err(bus->dev, "Program params failed: %d\n", ret);
1381                         goto restore_params;
1382                 }
1383         }
1384
1385         if (!bus) {
1386                 pr_err("Configuration error in %s\n", __func__);
1387                 return -EINVAL;
1388         }
1389
1390         ret = do_bank_switch(stream);
1391         if (ret < 0) {
1392                 dev_err(bus->dev, "Bank switch failed: %d\n", ret);
1393                 goto restore_params;
1394         }
1395
1396         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1397                 bus = m_rt->bus;
1398
1399                 /* Prepare port(s) on the new clock configuration */
1400                 ret = sdw_prep_deprep_ports(m_rt, true);
1401                 if (ret < 0) {
1402                         dev_err(bus->dev, "Prepare port(s) failed ret = %d\n",
1403                                 ret);
1404                         return ret;
1405                 }
1406         }
1407
1408         stream->state = SDW_STREAM_PREPARED;
1409
1410         return ret;
1411
1412 restore_params:
1413         memcpy(&bus->params, &params, sizeof(params));
1414         return ret;
1415 }
1416
1417 /**
1418  * sdw_prepare_stream() - Prepare SoundWire stream
1419  *
1420  * @stream: Soundwire stream
1421  *
1422  * Documentation/driver-api/soundwire/stream.rst explains this API in detail
1423  */
1424 int sdw_prepare_stream(struct sdw_stream_runtime *stream)
1425 {
1426         bool update_params = true;
1427         int ret;
1428
1429         if (!stream) {
1430                 pr_err("SoundWire: Handle not found for stream\n");
1431                 return -EINVAL;
1432         }
1433
1434         sdw_acquire_bus_lock(stream);
1435
1436         if (stream->state == SDW_STREAM_PREPARED) {
1437                 ret = 0;
1438                 goto state_err;
1439         }
1440
1441         if (stream->state != SDW_STREAM_CONFIGURED &&
1442             stream->state != SDW_STREAM_DEPREPARED &&
1443             stream->state != SDW_STREAM_DISABLED) {
1444                 pr_err("%s: %s: inconsistent state state %d\n",
1445                        __func__, stream->name, stream->state);
1446                 ret = -EINVAL;
1447                 goto state_err;
1448         }
1449
1450         /*
1451          * when the stream is DISABLED, this means sdw_prepare_stream()
1452          * is called as a result of an underflow or a resume operation.
1453          * In this case, the bus parameters shall not be recomputed, but
1454          * still need to be re-applied
1455          */
1456         if (stream->state == SDW_STREAM_DISABLED)
1457                 update_params = false;
1458
1459         ret = _sdw_prepare_stream(stream, update_params);
1460
1461 state_err:
1462         sdw_release_bus_lock(stream);
1463         return ret;
1464 }
1465 EXPORT_SYMBOL(sdw_prepare_stream);
1466
1467 static int _sdw_enable_stream(struct sdw_stream_runtime *stream)
1468 {
1469         struct sdw_master_runtime *m_rt;
1470         struct sdw_bus *bus = NULL;
1471         int ret;
1472
1473         /* Enable Master(s) and Slave(s) port(s) associated with stream */
1474         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1475                 bus = m_rt->bus;
1476
1477                 /* Program params */
1478                 ret = sdw_program_params(bus, false);
1479                 if (ret < 0) {
1480                         dev_err(bus->dev, "Program params failed: %d\n", ret);
1481                         return ret;
1482                 }
1483
1484                 /* Enable port(s) */
1485                 ret = sdw_enable_disable_ports(m_rt, true);
1486                 if (ret < 0) {
1487                         dev_err(bus->dev,
1488                                 "Enable port(s) failed ret: %d\n", ret);
1489                         return ret;
1490                 }
1491         }
1492
1493         if (!bus) {
1494                 pr_err("Configuration error in %s\n", __func__);
1495                 return -EINVAL;
1496         }
1497
1498         ret = do_bank_switch(stream);
1499         if (ret < 0) {
1500                 dev_err(bus->dev, "Bank switch failed: %d\n", ret);
1501                 return ret;
1502         }
1503
1504         stream->state = SDW_STREAM_ENABLED;
1505         return 0;
1506 }
1507
1508 /**
1509  * sdw_enable_stream() - Enable SoundWire stream
1510  *
1511  * @stream: Soundwire stream
1512  *
1513  * Documentation/driver-api/soundwire/stream.rst explains this API in detail
1514  */
1515 int sdw_enable_stream(struct sdw_stream_runtime *stream)
1516 {
1517         int ret;
1518
1519         if (!stream) {
1520                 pr_err("SoundWire: Handle not found for stream\n");
1521                 return -EINVAL;
1522         }
1523
1524         sdw_acquire_bus_lock(stream);
1525
1526         if (stream->state == SDW_STREAM_ENABLED) {
1527                 ret = 0;
1528                 goto state_err;
1529         }
1530
1531         if (stream->state != SDW_STREAM_PREPARED &&
1532             stream->state != SDW_STREAM_DISABLED) {
1533                 pr_err("%s: %s: inconsistent state state %d\n",
1534                        __func__, stream->name, stream->state);
1535                 ret = -EINVAL;
1536                 goto state_err;
1537         }
1538
1539         ret = _sdw_enable_stream(stream);
1540
1541 state_err:
1542         sdw_release_bus_lock(stream);
1543         return ret;
1544 }
1545 EXPORT_SYMBOL(sdw_enable_stream);
1546
1547 static int _sdw_disable_stream(struct sdw_stream_runtime *stream)
1548 {
1549         struct sdw_master_runtime *m_rt;
1550         int ret;
1551
1552         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1553                 struct sdw_bus *bus = m_rt->bus;
1554
1555                 /* Disable port(s) */
1556                 ret = sdw_enable_disable_ports(m_rt, false);
1557                 if (ret < 0) {
1558                         dev_err(bus->dev, "Disable port(s) failed: %d\n", ret);
1559                         return ret;
1560                 }
1561         }
1562         stream->state = SDW_STREAM_DISABLED;
1563
1564         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1565                 struct sdw_bus *bus = m_rt->bus;
1566
1567                 /* Program params */
1568                 ret = sdw_program_params(bus, false);
1569                 if (ret < 0) {
1570                         dev_err(bus->dev, "Program params failed: %d\n", ret);
1571                         return ret;
1572                 }
1573         }
1574
1575         ret = do_bank_switch(stream);
1576         if (ret < 0) {
1577                 pr_err("Bank switch failed: %d\n", ret);
1578                 return ret;
1579         }
1580
1581         /* make sure alternate bank (previous current) is also disabled */
1582         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1583                 struct sdw_bus *bus = m_rt->bus;
1584
1585                 /* Disable port(s) */
1586                 ret = sdw_enable_disable_ports(m_rt, false);
1587                 if (ret < 0) {
1588                         dev_err(bus->dev, "Disable port(s) failed: %d\n", ret);
1589                         return ret;
1590                 }
1591         }
1592
1593         return 0;
1594 }
1595
1596 /**
1597  * sdw_disable_stream() - Disable SoundWire stream
1598  *
1599  * @stream: Soundwire stream
1600  *
1601  * Documentation/driver-api/soundwire/stream.rst explains this API in detail
1602  */
1603 int sdw_disable_stream(struct sdw_stream_runtime *stream)
1604 {
1605         int ret;
1606
1607         if (!stream) {
1608                 pr_err("SoundWire: Handle not found for stream\n");
1609                 return -EINVAL;
1610         }
1611
1612         sdw_acquire_bus_lock(stream);
1613
1614         if (stream->state == SDW_STREAM_DISABLED) {
1615                 ret = 0;
1616                 goto state_err;
1617         }
1618
1619         if (stream->state != SDW_STREAM_ENABLED) {
1620                 pr_err("%s: %s: inconsistent state state %d\n",
1621                        __func__, stream->name, stream->state);
1622                 ret = -EINVAL;
1623                 goto state_err;
1624         }
1625
1626         ret = _sdw_disable_stream(stream);
1627
1628 state_err:
1629         sdw_release_bus_lock(stream);
1630         return ret;
1631 }
1632 EXPORT_SYMBOL(sdw_disable_stream);
1633
1634 static int _sdw_deprepare_stream(struct sdw_stream_runtime *stream)
1635 {
1636         struct sdw_master_runtime *m_rt;
1637         struct sdw_bus *bus;
1638         int ret = 0;
1639
1640         list_for_each_entry(m_rt, &stream->master_list, stream_node) {
1641                 bus = m_rt->bus;
1642                 /* De-prepare port(s) */
1643                 ret = sdw_prep_deprep_ports(m_rt, false);
1644                 if (ret < 0) {
1645                         dev_err(bus->dev,
1646                                 "De-prepare port(s) failed: %d\n", ret);
1647                         return ret;
1648                 }
1649
1650                 /* TODO: Update this during Device-Device support */
1651                 bus->params.bandwidth -= m_rt->stream->params.rate *
1652                         m_rt->ch_count * m_rt->stream->params.bps;
1653
1654                 /* Compute params */
1655                 if (bus->compute_params) {
1656                         ret = bus->compute_params(bus);
1657                         if (ret < 0) {
1658                                 dev_err(bus->dev, "Compute params failed: %d\n",
1659                                         ret);
1660                                 return ret;
1661                         }
1662                 }
1663
1664                 /* Program params */
1665                 ret = sdw_program_params(bus, false);
1666                 if (ret < 0) {
1667                         dev_err(bus->dev, "Program params failed: %d\n", ret);
1668                         return ret;
1669                 }
1670         }
1671
1672         stream->state = SDW_STREAM_DEPREPARED;
1673         return do_bank_switch(stream);
1674 }
1675
1676 /**
1677  * sdw_deprepare_stream() - Deprepare SoundWire stream
1678  *
1679  * @stream: Soundwire stream
1680  *
1681  * Documentation/driver-api/soundwire/stream.rst explains this API in detail
1682  */
1683 int sdw_deprepare_stream(struct sdw_stream_runtime *stream)
1684 {
1685         int ret;
1686
1687         if (!stream) {
1688                 pr_err("SoundWire: Handle not found for stream\n");
1689                 return -EINVAL;
1690         }
1691
1692         sdw_acquire_bus_lock(stream);
1693
1694         if (stream->state == SDW_STREAM_DEPREPARED) {
1695                 ret = 0;
1696                 goto state_err;
1697         }
1698
1699         if (stream->state != SDW_STREAM_PREPARED &&
1700             stream->state != SDW_STREAM_DISABLED) {
1701                 pr_err("%s: %s: inconsistent state state %d\n",
1702                        __func__, stream->name, stream->state);
1703                 ret = -EINVAL;
1704                 goto state_err;
1705         }
1706
1707         ret = _sdw_deprepare_stream(stream);
1708
1709 state_err:
1710         sdw_release_bus_lock(stream);
1711         return ret;
1712 }
1713 EXPORT_SYMBOL(sdw_deprepare_stream);
1714
1715 static int set_stream(struct snd_pcm_substream *substream,
1716                       struct sdw_stream_runtime *sdw_stream)
1717 {
1718         struct snd_soc_pcm_runtime *rtd = substream->private_data;
1719         struct snd_soc_dai *dai;
1720         int ret = 0;
1721         int i;
1722
1723         /* Set stream pointer on all DAIs */
1724         for_each_rtd_dais(rtd, i, dai) {
1725                 ret = snd_soc_dai_set_stream(dai, sdw_stream, substream->stream);
1726                 if (ret < 0) {
1727                         dev_err(rtd->dev, "failed to set stream pointer on dai %s\n", dai->name);
1728                         break;
1729                 }
1730         }
1731
1732         return ret;
1733 }
1734
1735 /**
1736  * sdw_alloc_stream() - Allocate and return stream runtime
1737  *
1738  * @stream_name: SoundWire stream name
1739  *
1740  * Allocates a SoundWire stream runtime instance.
1741  * sdw_alloc_stream should be called only once per stream. Typically
1742  * invoked from ALSA/ASoC machine/platform driver.
1743  */
1744 struct sdw_stream_runtime *sdw_alloc_stream(const char *stream_name)
1745 {
1746         struct sdw_stream_runtime *stream;
1747
1748         stream = kzalloc(sizeof(*stream), GFP_KERNEL);
1749         if (!stream)
1750                 return NULL;
1751
1752         stream->name = stream_name;
1753         INIT_LIST_HEAD(&stream->master_list);
1754         stream->state = SDW_STREAM_ALLOCATED;
1755         stream->m_rt_count = 0;
1756
1757         return stream;
1758 }
1759 EXPORT_SYMBOL(sdw_alloc_stream);
1760
1761 /**
1762  * sdw_startup_stream() - Startup SoundWire stream
1763  *
1764  * @sdw_substream: Soundwire stream
1765  *
1766  * Documentation/driver-api/soundwire/stream.rst explains this API in detail
1767  */
1768 int sdw_startup_stream(void *sdw_substream)
1769 {
1770         struct snd_pcm_substream *substream = sdw_substream;
1771         struct snd_soc_pcm_runtime *rtd = substream->private_data;
1772         struct sdw_stream_runtime *sdw_stream;
1773         char *name;
1774         int ret;
1775
1776         if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
1777                 name = kasprintf(GFP_KERNEL, "%s-Playback", substream->name);
1778         else
1779                 name = kasprintf(GFP_KERNEL, "%s-Capture", substream->name);
1780
1781         if (!name)
1782                 return -ENOMEM;
1783
1784         sdw_stream = sdw_alloc_stream(name);
1785         if (!sdw_stream) {
1786                 dev_err(rtd->dev, "alloc stream failed for substream DAI %s\n", substream->name);
1787                 ret = -ENOMEM;
1788                 goto error;
1789         }
1790
1791         ret = set_stream(substream, sdw_stream);
1792         if (ret < 0)
1793                 goto release_stream;
1794         return 0;
1795
1796 release_stream:
1797         sdw_release_stream(sdw_stream);
1798         set_stream(substream, NULL);
1799 error:
1800         kfree(name);
1801         return ret;
1802 }
1803 EXPORT_SYMBOL(sdw_startup_stream);
1804
1805 /**
1806  * sdw_shutdown_stream() - Shutdown SoundWire stream
1807  *
1808  * @sdw_substream: Soundwire stream
1809  *
1810  * Documentation/driver-api/soundwire/stream.rst explains this API in detail
1811  */
1812 void sdw_shutdown_stream(void *sdw_substream)
1813 {
1814         struct snd_pcm_substream *substream = sdw_substream;
1815         struct snd_soc_pcm_runtime *rtd = substream->private_data;
1816         struct sdw_stream_runtime *sdw_stream;
1817         struct snd_soc_dai *dai;
1818
1819         /* Find stream from first CPU DAI */
1820         dai = asoc_rtd_to_cpu(rtd, 0);
1821
1822         sdw_stream = snd_soc_dai_get_stream(dai, substream->stream);
1823
1824         if (IS_ERR(sdw_stream)) {
1825                 dev_err(rtd->dev, "no stream found for DAI %s\n", dai->name);
1826                 return;
1827         }
1828
1829         /* release memory */
1830         kfree(sdw_stream->name);
1831         sdw_release_stream(sdw_stream);
1832
1833         /* clear DAI data */
1834         set_stream(substream, NULL);
1835 }
1836 EXPORT_SYMBOL(sdw_shutdown_stream);
1837
1838 /**
1839  * sdw_release_stream() - Free the assigned stream runtime
1840  *
1841  * @stream: SoundWire stream runtime
1842  *
1843  * sdw_release_stream should be called only once per stream
1844  */
1845 void sdw_release_stream(struct sdw_stream_runtime *stream)
1846 {
1847         kfree(stream);
1848 }
1849 EXPORT_SYMBOL(sdw_release_stream);
1850
1851 /**
1852  * sdw_stream_add_master() - Allocate and add master runtime to a stream
1853  *
1854  * @bus: SDW Bus instance
1855  * @stream_config: Stream configuration for audio stream
1856  * @port_config: Port configuration for audio stream
1857  * @num_ports: Number of ports
1858  * @stream: SoundWire stream
1859  */
1860 int sdw_stream_add_master(struct sdw_bus *bus,
1861                           struct sdw_stream_config *stream_config,
1862                           struct sdw_port_config *port_config,
1863                           unsigned int num_ports,
1864                           struct sdw_stream_runtime *stream)
1865 {
1866         struct sdw_master_runtime *m_rt;
1867         bool alloc_master_rt = true;
1868         int ret;
1869
1870         mutex_lock(&bus->bus_lock);
1871
1872         /*
1873          * For multi link streams, add the second master only if
1874          * the bus supports it.
1875          * Check if bus->multi_link is set
1876          */
1877         if (!bus->multi_link && stream->m_rt_count > 0) {
1878                 dev_err(bus->dev,
1879                         "Multilink not supported, link %d\n", bus->link_id);
1880                 ret = -EINVAL;
1881                 goto unlock;
1882         }
1883
1884         /*
1885          * check if Master is already allocated (e.g. as a result of Slave adding
1886          * it first), if so skip allocation and go to configuration
1887          */
1888         m_rt = sdw_master_rt_find(bus, stream);
1889         if (m_rt) {
1890                 alloc_master_rt = false;
1891                 goto skip_alloc_master_rt;
1892         }
1893
1894         m_rt = sdw_master_rt_alloc(bus, stream);
1895         if (!m_rt) {
1896                 dev_err(bus->dev, "Master runtime alloc failed for stream:%s\n", stream->name);
1897                 ret = -ENOMEM;
1898                 goto unlock;
1899         }
1900 skip_alloc_master_rt:
1901
1902         if (sdw_master_port_allocated(m_rt))
1903                 goto skip_alloc_master_port;
1904
1905         ret = sdw_master_port_alloc(m_rt, num_ports);
1906         if (ret)
1907                 goto alloc_error;
1908
1909         stream->m_rt_count++;
1910
1911 skip_alloc_master_port:
1912
1913         ret = sdw_master_rt_config(m_rt, stream_config);
1914         if (ret < 0)
1915                 goto unlock;
1916
1917         ret = sdw_config_stream(bus->dev, stream, stream_config, false);
1918         if (ret)
1919                 goto unlock;
1920
1921         ret = sdw_master_port_config(m_rt, port_config);
1922
1923         goto unlock;
1924
1925 alloc_error:
1926         /*
1927          * we only cleanup what was allocated in this routine
1928          */
1929         if (alloc_master_rt)
1930                 sdw_master_rt_free(m_rt, stream);
1931 unlock:
1932         mutex_unlock(&bus->bus_lock);
1933         return ret;
1934 }
1935 EXPORT_SYMBOL(sdw_stream_add_master);
1936
1937 /**
1938  * sdw_stream_remove_master() - Remove master from sdw_stream
1939  *
1940  * @bus: SDW Bus instance
1941  * @stream: SoundWire stream
1942  *
1943  * This removes and frees port_rt and master_rt from a stream
1944  */
1945 int sdw_stream_remove_master(struct sdw_bus *bus,
1946                              struct sdw_stream_runtime *stream)
1947 {
1948         struct sdw_master_runtime *m_rt, *_m_rt;
1949
1950         mutex_lock(&bus->bus_lock);
1951
1952         list_for_each_entry_safe(m_rt, _m_rt,
1953                                  &stream->master_list, stream_node) {
1954                 if (m_rt->bus != bus)
1955                         continue;
1956
1957                 sdw_master_port_free(m_rt);
1958                 sdw_master_rt_free(m_rt, stream);
1959                 stream->m_rt_count--;
1960         }
1961
1962         if (list_empty(&stream->master_list))
1963                 stream->state = SDW_STREAM_RELEASED;
1964
1965         mutex_unlock(&bus->bus_lock);
1966
1967         return 0;
1968 }
1969 EXPORT_SYMBOL(sdw_stream_remove_master);
1970
1971 /**
1972  * sdw_stream_add_slave() - Allocate and add master/slave runtime to a stream
1973  *
1974  * @slave: SDW Slave instance
1975  * @stream_config: Stream configuration for audio stream
1976  * @stream: SoundWire stream
1977  * @port_config: Port configuration for audio stream
1978  * @num_ports: Number of ports
1979  *
1980  * It is expected that Slave is added before adding Master
1981  * to the Stream.
1982  *
1983  */
1984 int sdw_stream_add_slave(struct sdw_slave *slave,
1985                          struct sdw_stream_config *stream_config,
1986                          struct sdw_port_config *port_config,
1987                          unsigned int num_ports,
1988                          struct sdw_stream_runtime *stream)
1989 {
1990         struct sdw_slave_runtime *s_rt;
1991         struct sdw_master_runtime *m_rt;
1992         bool alloc_master_rt = true;
1993         bool alloc_slave_rt = true;
1994
1995         int ret;
1996
1997         mutex_lock(&slave->bus->bus_lock);
1998
1999         /*
2000          * check if Master is already allocated, if so skip allocation
2001          * and go to configuration
2002          */
2003         m_rt = sdw_master_rt_find(slave->bus, stream);
2004         if (m_rt) {
2005                 alloc_master_rt = false;
2006                 goto skip_alloc_master_rt;
2007         }
2008
2009         /*
2010          * If this API is invoked by Slave first then m_rt is not valid.
2011          * So, allocate m_rt and add Slave to it.
2012          */
2013         m_rt = sdw_master_rt_alloc(slave->bus, stream);
2014         if (!m_rt) {
2015                 dev_err(&slave->dev, "Master runtime alloc failed for stream:%s\n", stream->name);
2016                 ret = -ENOMEM;
2017                 goto unlock;
2018         }
2019
2020 skip_alloc_master_rt:
2021         s_rt = sdw_slave_rt_find(slave, stream);
2022         if (s_rt)
2023                 goto skip_alloc_slave_rt;
2024
2025         s_rt = sdw_slave_rt_alloc(slave, m_rt);
2026         if (!s_rt) {
2027                 dev_err(&slave->dev, "Slave runtime alloc failed for stream:%s\n", stream->name);
2028                 alloc_slave_rt = false;
2029                 ret = -ENOMEM;
2030                 goto alloc_error;
2031         }
2032
2033 skip_alloc_slave_rt:
2034         if (sdw_slave_port_allocated(s_rt))
2035                 goto skip_port_alloc;
2036
2037         ret = sdw_slave_port_alloc(slave, s_rt, num_ports);
2038         if (ret)
2039                 goto alloc_error;
2040
2041 skip_port_alloc:
2042         ret =  sdw_master_rt_config(m_rt, stream_config);
2043         if (ret)
2044                 goto unlock;
2045
2046         ret = sdw_slave_rt_config(s_rt, stream_config);
2047         if (ret)
2048                 goto unlock;
2049
2050         ret = sdw_config_stream(&slave->dev, stream, stream_config, true);
2051         if (ret)
2052                 goto unlock;
2053
2054         ret = sdw_slave_port_config(slave, s_rt, port_config);
2055         if (ret)
2056                 goto unlock;
2057
2058         /*
2059          * Change stream state to CONFIGURED on first Slave add.
2060          * Bus is not aware of number of Slave(s) in a stream at this
2061          * point so cannot depend on all Slave(s) to be added in order to
2062          * change stream state to CONFIGURED.
2063          */
2064         stream->state = SDW_STREAM_CONFIGURED;
2065         goto unlock;
2066
2067 alloc_error:
2068         /*
2069          * we only cleanup what was allocated in this routine. The 'else if'
2070          * is intentional, the 'master_rt_free' will call sdw_slave_rt_free()
2071          * internally.
2072          */
2073         if (alloc_master_rt)
2074                 sdw_master_rt_free(m_rt, stream);
2075         else if (alloc_slave_rt)
2076                 sdw_slave_rt_free(slave, stream);
2077 unlock:
2078         mutex_unlock(&slave->bus->bus_lock);
2079         return ret;
2080 }
2081 EXPORT_SYMBOL(sdw_stream_add_slave);
2082
2083 /**
2084  * sdw_stream_remove_slave() - Remove slave from sdw_stream
2085  *
2086  * @slave: SDW Slave instance
2087  * @stream: SoundWire stream
2088  *
2089  * This removes and frees port_rt and slave_rt from a stream
2090  */
2091 int sdw_stream_remove_slave(struct sdw_slave *slave,
2092                             struct sdw_stream_runtime *stream)
2093 {
2094         mutex_lock(&slave->bus->bus_lock);
2095
2096         sdw_slave_port_free(slave, stream);
2097         sdw_slave_rt_free(slave, stream);
2098
2099         mutex_unlock(&slave->bus->bus_lock);
2100
2101         return 0;
2102 }
2103 EXPORT_SYMBOL(sdw_stream_remove_slave);