ASoC: wm0010: Delete an error message for a failed memory allocation in wm0010_boot()
[muen/linux.git] / sound / soc / codecs / wm0010.c
1 /*
2  * wm0010.c  --  WM0010 DSP Driver
3  *
4  * Copyright 2012 Wolfson Microelectronics PLC.
5  *
6  * Authors: Mark Brown <broonie@opensource.wolfsonmicro.com>
7  *          Dimitris Papastamos <dp@opensource.wolfsonmicro.com>
8  *          Scott Ling <sl@opensource.wolfsonmicro.com>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License version 2 as
12  * published by the Free Software Foundation.
13  */
14
15 #include <linux/module.h>
16 #include <linux/moduleparam.h>
17 #include <linux/interrupt.h>
18 #include <linux/irqreturn.h>
19 #include <linux/init.h>
20 #include <linux/spi/spi.h>
21 #include <linux/firmware.h>
22 #include <linux/delay.h>
23 #include <linux/fs.h>
24 #include <linux/gpio.h>
25 #include <linux/regulator/consumer.h>
26 #include <linux/mutex.h>
27 #include <linux/workqueue.h>
28
29 #include <sound/soc.h>
30 #include <sound/wm0010.h>
31
32 #define DEVICE_ID_WM0010        10
33
34 /* We only support v1 of the .dfw INFO record */
35 #define INFO_VERSION            1
36
37 enum dfw_cmd {
38         DFW_CMD_FUSE = 0x01,
39         DFW_CMD_CODE_HDR,
40         DFW_CMD_CODE_DATA,
41         DFW_CMD_PLL,
42         DFW_CMD_INFO = 0xff
43 };
44
45 struct dfw_binrec {
46         u8 command;
47         u32 length:24;
48         u32 address;
49         uint8_t data[0];
50 } __packed;
51
52 struct dfw_inforec {
53         u8 info_version;
54         u8 tool_major_version;
55         u8 tool_minor_version;
56         u8 dsp_target;
57 };
58
59 struct dfw_pllrec {
60         u8 command;
61         u32 length:24;
62         u32 address;
63         u32 clkctrl1;
64         u32 clkctrl2;
65         u32 clkctrl3;
66         u32 ldetctrl;
67         u32 uart_div;
68         u32 spi_div;
69 } __packed;
70
71 static struct pll_clock_map {
72         int max_sysclk;
73         int max_pll_spi_speed;
74         u32 pll_clkctrl1;
75 } pll_clock_map[] = {                      /* Dividers */
76         { 22000000, 26000000, 0x00201f11 }, /* 2,32,2  */
77         { 18000000, 26000000, 0x00203f21 }, /* 2,64,4  */
78         { 14000000, 26000000, 0x00202620 }, /* 1,39,4  */
79         { 10000000, 22000000, 0x00203120 }, /* 1,50,4  */
80         {  6500000, 22000000, 0x00204520 }, /* 1,70,4  */
81         {  5500000, 22000000, 0x00103f10 }, /* 1,64,2  */
82 };
83
84 enum wm0010_state {
85         WM0010_POWER_OFF,
86         WM0010_OUT_OF_RESET,
87         WM0010_BOOTROM,
88         WM0010_STAGE2,
89         WM0010_FIRMWARE,
90 };
91
92 struct wm0010_priv {
93         struct snd_soc_codec *codec;
94
95         struct mutex lock;
96         struct device *dev;
97
98         struct wm0010_pdata pdata;
99
100         int gpio_reset;
101         int gpio_reset_value;
102
103         struct regulator_bulk_data core_supplies[2];
104         struct regulator *dbvdd;
105
106         int sysclk;
107
108         enum wm0010_state state;
109         bool boot_failed;
110         bool ready;
111         bool pll_running;
112         int max_spi_freq;
113         int board_max_spi_speed;
114         u32 pll_clkctrl1;
115
116         spinlock_t irq_lock;
117         int irq;
118
119         struct completion boot_completion;
120 };
121
122 struct wm0010_spi_msg {
123         struct spi_message m;
124         struct spi_transfer t;
125         u8 *tx_buf;
126         u8 *rx_buf;
127         size_t len;
128 };
129
130 static const struct snd_soc_dapm_widget wm0010_dapm_widgets[] = {
131 SND_SOC_DAPM_SUPPLY("CLKIN",  SND_SOC_NOPM, 0, 0, NULL, 0),
132 };
133
134 static const struct snd_soc_dapm_route wm0010_dapm_routes[] = {
135         { "SDI2 Capture", NULL, "SDI1 Playback" },
136         { "SDI1 Capture", NULL, "SDI2 Playback" },
137
138         { "SDI1 Capture", NULL, "CLKIN" },
139         { "SDI2 Capture", NULL, "CLKIN" },
140         { "SDI1 Playback", NULL, "CLKIN" },
141         { "SDI2 Playback", NULL, "CLKIN" },
142 };
143
144 static const char *wm0010_state_to_str(enum wm0010_state state)
145 {
146         static const char * const state_to_str[] = {
147                 "Power off",
148                 "Out of reset",
149                 "Boot ROM",
150                 "Stage2",
151                 "Firmware"
152         };
153
154         if (state < 0 || state >= ARRAY_SIZE(state_to_str))
155                 return "null";
156         return state_to_str[state];
157 }
158
159 /* Called with wm0010->lock held */
160 static void wm0010_halt(struct snd_soc_codec *codec)
161 {
162         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
163         unsigned long flags;
164         enum wm0010_state state;
165
166         /* Fetch the wm0010 state */
167         spin_lock_irqsave(&wm0010->irq_lock, flags);
168         state = wm0010->state;
169         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
170
171         switch (state) {
172         case WM0010_POWER_OFF:
173                 /* If there's nothing to do, bail out */
174                 return;
175         case WM0010_OUT_OF_RESET:
176         case WM0010_BOOTROM:
177         case WM0010_STAGE2:
178         case WM0010_FIRMWARE:
179                 /* Remember to put chip back into reset */
180                 gpio_set_value_cansleep(wm0010->gpio_reset,
181                                         wm0010->gpio_reset_value);
182                 /* Disable the regulators */
183                 regulator_disable(wm0010->dbvdd);
184                 regulator_bulk_disable(ARRAY_SIZE(wm0010->core_supplies),
185                                        wm0010->core_supplies);
186                 break;
187         }
188
189         spin_lock_irqsave(&wm0010->irq_lock, flags);
190         wm0010->state = WM0010_POWER_OFF;
191         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
192 }
193
194 struct wm0010_boot_xfer {
195         struct list_head list;
196         struct snd_soc_codec *codec;
197         struct completion *done;
198         struct spi_message m;
199         struct spi_transfer t;
200 };
201
202 /* Called with wm0010->lock held */
203 static void wm0010_mark_boot_failure(struct wm0010_priv *wm0010)
204 {
205         enum wm0010_state state;
206         unsigned long flags;
207
208         spin_lock_irqsave(&wm0010->irq_lock, flags);
209         state = wm0010->state;
210         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
211
212         dev_err(wm0010->dev, "Failed to transition from `%s' state to `%s' state\n",
213                 wm0010_state_to_str(state), wm0010_state_to_str(state + 1));
214
215         wm0010->boot_failed = true;
216 }
217
218 static void wm0010_boot_xfer_complete(void *data)
219 {
220         struct wm0010_boot_xfer *xfer = data;
221         struct snd_soc_codec *codec = xfer->codec;
222         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
223         u32 *out32 = xfer->t.rx_buf;
224         int i;
225
226         if (xfer->m.status != 0) {
227                 dev_err(codec->dev, "SPI transfer failed: %d\n",
228                         xfer->m.status);
229                 wm0010_mark_boot_failure(wm0010);
230                 if (xfer->done)
231                         complete(xfer->done);
232                 return;
233         }
234
235         for (i = 0; i < xfer->t.len / 4; i++) {
236                 dev_dbg(codec->dev, "%d: %04x\n", i, out32[i]);
237
238                 switch (be32_to_cpu(out32[i])) {
239                 case 0xe0e0e0e0:
240                         dev_err(codec->dev,
241                                 "%d: ROM error reported in stage 2\n", i);
242                         wm0010_mark_boot_failure(wm0010);
243                         break;
244
245                 case 0x55555555:
246                         if (wm0010->state < WM0010_STAGE2)
247                                 break;
248                         dev_err(codec->dev,
249                                 "%d: ROM bootloader running in stage 2\n", i);
250                         wm0010_mark_boot_failure(wm0010);
251                         break;
252
253                 case 0x0fed0000:
254                         dev_dbg(codec->dev, "Stage2 loader running\n");
255                         break;
256
257                 case 0x0fed0007:
258                         dev_dbg(codec->dev, "CODE_HDR packet received\n");
259                         break;
260
261                 case 0x0fed0008:
262                         dev_dbg(codec->dev, "CODE_DATA packet received\n");
263                         break;
264
265                 case 0x0fed0009:
266                         dev_dbg(codec->dev, "Download complete\n");
267                         break;
268
269                 case 0x0fed000c:
270                         dev_dbg(codec->dev, "Application start\n");
271                         break;
272
273                 case 0x0fed000e:
274                         dev_dbg(codec->dev, "PLL packet received\n");
275                         wm0010->pll_running = true;
276                         break;
277
278                 case 0x0fed0025:
279                         dev_err(codec->dev, "Device reports image too long\n");
280                         wm0010_mark_boot_failure(wm0010);
281                         break;
282
283                 case 0x0fed002c:
284                         dev_err(codec->dev, "Device reports bad SPI packet\n");
285                         wm0010_mark_boot_failure(wm0010);
286                         break;
287
288                 case 0x0fed0031:
289                         dev_err(codec->dev, "Device reports SPI read overflow\n");
290                         wm0010_mark_boot_failure(wm0010);
291                         break;
292
293                 case 0x0fed0032:
294                         dev_err(codec->dev, "Device reports SPI underclock\n");
295                         wm0010_mark_boot_failure(wm0010);
296                         break;
297
298                 case 0x0fed0033:
299                         dev_err(codec->dev, "Device reports bad header packet\n");
300                         wm0010_mark_boot_failure(wm0010);
301                         break;
302
303                 case 0x0fed0034:
304                         dev_err(codec->dev, "Device reports invalid packet type\n");
305                         wm0010_mark_boot_failure(wm0010);
306                         break;
307
308                 case 0x0fed0035:
309                         dev_err(codec->dev, "Device reports data before header error\n");
310                         wm0010_mark_boot_failure(wm0010);
311                         break;
312
313                 case 0x0fed0038:
314                         dev_err(codec->dev, "Device reports invalid PLL packet\n");
315                         break;
316
317                 case 0x0fed003a:
318                         dev_err(codec->dev, "Device reports packet alignment error\n");
319                         wm0010_mark_boot_failure(wm0010);
320                         break;
321
322                 default:
323                         dev_err(codec->dev, "Unrecognised return 0x%x\n",
324                             be32_to_cpu(out32[i]));
325                         wm0010_mark_boot_failure(wm0010);
326                         break;
327                 }
328
329                 if (wm0010->boot_failed)
330                         break;
331         }
332
333         if (xfer->done)
334                 complete(xfer->done);
335 }
336
337 static void byte_swap_64(u64 *data_in, u64 *data_out, u32 len)
338 {
339         int i;
340
341         for (i = 0; i < len / 8; i++)
342                 data_out[i] = cpu_to_be64(le64_to_cpu(data_in[i]));
343 }
344
345 static int wm0010_firmware_load(const char *name, struct snd_soc_codec *codec)
346 {
347         struct spi_device *spi = to_spi_device(codec->dev);
348         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
349         struct list_head xfer_list;
350         struct wm0010_boot_xfer *xfer;
351         int ret;
352         struct completion done;
353         const struct firmware *fw;
354         const struct dfw_binrec *rec;
355         const struct dfw_inforec *inforec;
356         u64 *img;
357         u8 *out, dsp;
358         u32 len, offset;
359
360         INIT_LIST_HEAD(&xfer_list);
361
362         ret = request_firmware(&fw, name, codec->dev);
363         if (ret != 0) {
364                 dev_err(codec->dev, "Failed to request application(%s): %d\n",
365                         name, ret);
366                 return ret;
367         }
368
369         rec = (const struct dfw_binrec *)fw->data;
370         inforec = (const struct dfw_inforec *)rec->data;
371         offset = 0;
372         dsp = inforec->dsp_target;
373         wm0010->boot_failed = false;
374         if (WARN_ON(!list_empty(&xfer_list)))
375                 return -EINVAL;
376         init_completion(&done);
377
378         /* First record should be INFO */
379         if (rec->command != DFW_CMD_INFO) {
380                 dev_err(codec->dev, "First record not INFO\r\n");
381                 ret = -EINVAL;
382                 goto abort;
383         }
384
385         if (inforec->info_version != INFO_VERSION) {
386                 dev_err(codec->dev,
387                         "Unsupported version (%02d) of INFO record\r\n",
388                         inforec->info_version);
389                 ret = -EINVAL;
390                 goto abort;
391         }
392
393         dev_dbg(codec->dev, "Version v%02d INFO record found\r\n",
394                 inforec->info_version);
395
396         /* Check it's a DSP file */
397         if (dsp != DEVICE_ID_WM0010) {
398                 dev_err(codec->dev, "Not a WM0010 firmware file.\r\n");
399                 ret = -EINVAL;
400                 goto abort;
401         }
402
403         /* Skip the info record as we don't need to send it */
404         offset += ((rec->length) + 8);
405         rec = (void *)&rec->data[rec->length];
406
407         while (offset < fw->size) {
408                 dev_dbg(codec->dev,
409                         "Packet: command %d, data length = 0x%x\r\n",
410                         rec->command, rec->length);
411                 len = rec->length + 8;
412
413                 xfer = kzalloc(sizeof(*xfer), GFP_KERNEL);
414                 if (!xfer) {
415                         ret = -ENOMEM;
416                         goto abort;
417                 }
418
419                 xfer->codec = codec;
420                 list_add_tail(&xfer->list, &xfer_list);
421
422                 out = kzalloc(len, GFP_KERNEL | GFP_DMA);
423                 if (!out) {
424                         ret = -ENOMEM;
425                         goto abort1;
426                 }
427                 xfer->t.rx_buf = out;
428
429                 img = kzalloc(len, GFP_KERNEL | GFP_DMA);
430                 if (!img) {
431                         ret = -ENOMEM;
432                         goto abort1;
433                 }
434                 xfer->t.tx_buf = img;
435
436                 byte_swap_64((u64 *)&rec->command, img, len);
437
438                 spi_message_init(&xfer->m);
439                 xfer->m.complete = wm0010_boot_xfer_complete;
440                 xfer->m.context = xfer;
441                 xfer->t.len = len;
442                 xfer->t.bits_per_word = 8;
443
444                 if (!wm0010->pll_running) {
445                         xfer->t.speed_hz = wm0010->sysclk / 6;
446                 } else {
447                         xfer->t.speed_hz = wm0010->max_spi_freq;
448
449                         if (wm0010->board_max_spi_speed &&
450                            (wm0010->board_max_spi_speed < wm0010->max_spi_freq))
451                                         xfer->t.speed_hz = wm0010->board_max_spi_speed;
452                 }
453
454                 /* Store max usable spi frequency for later use */
455                 wm0010->max_spi_freq = xfer->t.speed_hz;
456
457                 spi_message_add_tail(&xfer->t, &xfer->m);
458
459                 offset += ((rec->length) + 8);
460                 rec = (void *)&rec->data[rec->length];
461
462                 if (offset >= fw->size) {
463                         dev_dbg(codec->dev, "All transfers scheduled\n");
464                         xfer->done = &done;
465                 }
466
467                 ret = spi_async(spi, &xfer->m);
468                 if (ret != 0) {
469                         dev_err(codec->dev, "Write failed: %d\n", ret);
470                         goto abort1;
471                 }
472
473                 if (wm0010->boot_failed) {
474                         dev_dbg(codec->dev, "Boot fail!\n");
475                         ret = -EINVAL;
476                         goto abort1;
477                 }
478         }
479
480         wait_for_completion(&done);
481
482         ret = 0;
483
484 abort1:
485         while (!list_empty(&xfer_list)) {
486                 xfer = list_first_entry(&xfer_list, struct wm0010_boot_xfer,
487                                         list);
488                 kfree(xfer->t.rx_buf);
489                 kfree(xfer->t.tx_buf);
490                 list_del(&xfer->list);
491                 kfree(xfer);
492         }
493
494 abort:
495         release_firmware(fw);
496         return ret;
497 }
498
499 static int wm0010_stage2_load(struct snd_soc_codec *codec)
500 {
501         struct spi_device *spi = to_spi_device(codec->dev);
502         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
503         const struct firmware *fw;
504         struct spi_message m;
505         struct spi_transfer t;
506         u32 *img;
507         u8 *out;
508         int i;
509         int ret = 0;
510
511         ret = request_firmware(&fw, "wm0010_stage2.bin", codec->dev);
512         if (ret != 0) {
513                 dev_err(codec->dev, "Failed to request stage2 loader: %d\n",
514                         ret);
515                 return ret;
516         }
517
518         dev_dbg(codec->dev, "Downloading %zu byte stage 2 loader\n", fw->size);
519
520         /* Copy to local buffer first as vmalloc causes problems for dma */
521         img = kzalloc(fw->size, GFP_KERNEL | GFP_DMA);
522         if (!img) {
523                 ret = -ENOMEM;
524                 goto abort2;
525         }
526
527         out = kzalloc(fw->size, GFP_KERNEL | GFP_DMA);
528         if (!out) {
529                 ret = -ENOMEM;
530                 goto abort1;
531         }
532
533         memcpy(img, &fw->data[0], fw->size);
534
535         spi_message_init(&m);
536         memset(&t, 0, sizeof(t));
537         t.rx_buf = out;
538         t.tx_buf = img;
539         t.len = fw->size;
540         t.bits_per_word = 8;
541         t.speed_hz = wm0010->sysclk / 10;
542         spi_message_add_tail(&t, &m);
543
544         dev_dbg(codec->dev, "Starting initial download at %dHz\n",
545                 t.speed_hz);
546
547         ret = spi_sync(spi, &m);
548         if (ret != 0) {
549                 dev_err(codec->dev, "Initial download failed: %d\n", ret);
550                 goto abort;
551         }
552
553         /* Look for errors from the boot ROM */
554         for (i = 0; i < fw->size; i++) {
555                 if (out[i] != 0x55) {
556                         dev_err(codec->dev, "Boot ROM error: %x in %d\n",
557                                 out[i], i);
558                         wm0010_mark_boot_failure(wm0010);
559                         ret = -EBUSY;
560                         goto abort;
561                 }
562         }
563 abort:
564         kfree(out);
565 abort1:
566         kfree(img);
567 abort2:
568         release_firmware(fw);
569
570         return ret;
571 }
572
573 static int wm0010_boot(struct snd_soc_codec *codec)
574 {
575         struct spi_device *spi = to_spi_device(codec->dev);
576         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
577         unsigned long flags;
578         int ret;
579         struct spi_message m;
580         struct spi_transfer t;
581         struct dfw_pllrec pll_rec;
582         u32 *p, len;
583         u64 *img_swap;
584         u8 *out;
585         int i;
586
587         spin_lock_irqsave(&wm0010->irq_lock, flags);
588         if (wm0010->state != WM0010_POWER_OFF)
589                 dev_warn(wm0010->dev, "DSP already powered up!\n");
590         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
591
592         if (wm0010->sysclk > 26000000) {
593                 dev_err(codec->dev, "Max DSP clock frequency is 26MHz\n");
594                 ret = -ECANCELED;
595                 goto err;
596         }
597
598         mutex_lock(&wm0010->lock);
599         wm0010->pll_running = false;
600
601         dev_dbg(codec->dev, "max_spi_freq: %d\n", wm0010->max_spi_freq);
602
603         ret = regulator_bulk_enable(ARRAY_SIZE(wm0010->core_supplies),
604                                     wm0010->core_supplies);
605         if (ret != 0) {
606                 dev_err(&spi->dev, "Failed to enable core supplies: %d\n",
607                         ret);
608                 mutex_unlock(&wm0010->lock);
609                 goto err;
610         }
611
612         ret = regulator_enable(wm0010->dbvdd);
613         if (ret != 0) {
614                 dev_err(&spi->dev, "Failed to enable DBVDD: %d\n", ret);
615                 goto err_core;
616         }
617
618         /* Release reset */
619         gpio_set_value_cansleep(wm0010->gpio_reset, !wm0010->gpio_reset_value);
620         spin_lock_irqsave(&wm0010->irq_lock, flags);
621         wm0010->state = WM0010_OUT_OF_RESET;
622         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
623
624         if (!wait_for_completion_timeout(&wm0010->boot_completion,
625                                          msecs_to_jiffies(20)))
626                 dev_err(codec->dev, "Failed to get interrupt from DSP\n");
627
628         spin_lock_irqsave(&wm0010->irq_lock, flags);
629         wm0010->state = WM0010_BOOTROM;
630         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
631
632         ret = wm0010_stage2_load(codec);
633         if (ret)
634                 goto abort;
635
636         if (!wait_for_completion_timeout(&wm0010->boot_completion,
637                                          msecs_to_jiffies(20)))
638                 dev_err(codec->dev, "Failed to get interrupt from DSP loader.\n");
639
640         spin_lock_irqsave(&wm0010->irq_lock, flags);
641         wm0010->state = WM0010_STAGE2;
642         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
643
644         /* Only initialise PLL if max_spi_freq initialised */
645         if (wm0010->max_spi_freq) {
646
647                 /* Initialise a PLL record */
648                 memset(&pll_rec, 0, sizeof(pll_rec));
649                 pll_rec.command = DFW_CMD_PLL;
650                 pll_rec.length = (sizeof(pll_rec) - 8);
651
652                 /* On wm0010 only the CLKCTRL1 value is used */
653                 pll_rec.clkctrl1 = wm0010->pll_clkctrl1;
654
655                 ret = -ENOMEM;
656                 len = pll_rec.length + 8;
657                 out = kzalloc(len, GFP_KERNEL | GFP_DMA);
658                 if (!out)
659                         goto abort;
660
661                 img_swap = kzalloc(len, GFP_KERNEL | GFP_DMA);
662                 if (!img_swap)
663                         goto abort_out;
664
665                 /* We need to re-order for 0010 */
666                 byte_swap_64((u64 *)&pll_rec, img_swap, len);
667
668                 spi_message_init(&m);
669                 memset(&t, 0, sizeof(t));
670                 t.rx_buf = out;
671                 t.tx_buf = img_swap;
672                 t.len = len;
673                 t.bits_per_word = 8;
674                 t.speed_hz = wm0010->sysclk / 6;
675                 spi_message_add_tail(&t, &m);
676
677                 ret = spi_sync(spi, &m);
678                 if (ret) {
679                         dev_err(codec->dev, "First PLL write failed: %d\n", ret);
680                         goto abort_swap;
681                 }
682
683                 /* Use a second send of the message to get the return status */
684                 ret = spi_sync(spi, &m);
685                 if (ret) {
686                         dev_err(codec->dev, "Second PLL write failed: %d\n", ret);
687                         goto abort_swap;
688                 }
689
690                 p = (u32 *)out;
691
692                 /* Look for PLL active code from the DSP */
693                 for (i = 0; i < len / 4; i++) {
694                         if (*p == 0x0e00ed0f) {
695                                 dev_dbg(codec->dev, "PLL packet received\n");
696                                 wm0010->pll_running = true;
697                                 break;
698                         }
699                         p++;
700                 }
701
702                 kfree(img_swap);
703                 kfree(out);
704         } else
705                 dev_dbg(codec->dev, "Not enabling DSP PLL.");
706
707         ret = wm0010_firmware_load("wm0010.dfw", codec);
708
709         if (ret != 0)
710                 goto abort;
711
712         spin_lock_irqsave(&wm0010->irq_lock, flags);
713         wm0010->state = WM0010_FIRMWARE;
714         spin_unlock_irqrestore(&wm0010->irq_lock, flags);
715
716         mutex_unlock(&wm0010->lock);
717
718         return 0;
719
720 abort_swap:
721         kfree(img_swap);
722 abort_out:
723         kfree(out);
724 abort:
725         /* Put the chip back into reset */
726         wm0010_halt(codec);
727         mutex_unlock(&wm0010->lock);
728         return ret;
729
730 err_core:
731         mutex_unlock(&wm0010->lock);
732         regulator_bulk_disable(ARRAY_SIZE(wm0010->core_supplies),
733                                wm0010->core_supplies);
734 err:
735         return ret;
736 }
737
738 static int wm0010_set_bias_level(struct snd_soc_codec *codec,
739                                  enum snd_soc_bias_level level)
740 {
741         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
742
743         switch (level) {
744         case SND_SOC_BIAS_ON:
745                 if (snd_soc_codec_get_bias_level(codec) == SND_SOC_BIAS_PREPARE)
746                         wm0010_boot(codec);
747                 break;
748         case SND_SOC_BIAS_PREPARE:
749                 break;
750         case SND_SOC_BIAS_STANDBY:
751                 if (snd_soc_codec_get_bias_level(codec) == SND_SOC_BIAS_PREPARE) {
752                         mutex_lock(&wm0010->lock);
753                         wm0010_halt(codec);
754                         mutex_unlock(&wm0010->lock);
755                 }
756                 break;
757         case SND_SOC_BIAS_OFF:
758                 break;
759         }
760
761         return 0;
762 }
763
764 static int wm0010_set_sysclk(struct snd_soc_codec *codec, int source,
765                              int clk_id, unsigned int freq, int dir)
766 {
767         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
768         unsigned int i;
769
770         wm0010->sysclk = freq;
771
772         if (freq < pll_clock_map[ARRAY_SIZE(pll_clock_map)-1].max_sysclk) {
773                 wm0010->max_spi_freq = 0;
774         } else {
775                 for (i = 0; i < ARRAY_SIZE(pll_clock_map); i++)
776                         if (freq >= pll_clock_map[i].max_sysclk) {
777                                 wm0010->max_spi_freq = pll_clock_map[i].max_pll_spi_speed;
778                                 wm0010->pll_clkctrl1 = pll_clock_map[i].pll_clkctrl1;
779                                 break;
780                         }
781         }
782
783         return 0;
784 }
785
786 static int wm0010_probe(struct snd_soc_codec *codec);
787
788 static const struct snd_soc_codec_driver soc_codec_dev_wm0010 = {
789         .probe = wm0010_probe,
790         .set_bias_level = wm0010_set_bias_level,
791         .set_sysclk = wm0010_set_sysclk,
792         .idle_bias_off = true,
793
794         .component_driver = {
795                 .dapm_widgets           = wm0010_dapm_widgets,
796                 .num_dapm_widgets       = ARRAY_SIZE(wm0010_dapm_widgets),
797                 .dapm_routes            = wm0010_dapm_routes,
798                 .num_dapm_routes        = ARRAY_SIZE(wm0010_dapm_routes),
799         },
800 };
801
802 #define WM0010_RATES (SNDRV_PCM_RATE_44100 | SNDRV_PCM_RATE_48000)
803 #define WM0010_FORMATS (SNDRV_PCM_FMTBIT_S8 | SNDRV_PCM_FMTBIT_S16_LE |\
804                         SNDRV_PCM_FMTBIT_S20_3LE | SNDRV_PCM_FMTBIT_S24_LE |\
805                         SNDRV_PCM_FMTBIT_S32_LE)
806
807 static struct snd_soc_dai_driver wm0010_dai[] = {
808         {
809                 .name = "wm0010-sdi1",
810                 .playback = {
811                         .stream_name = "SDI1 Playback",
812                         .channels_min = 1,
813                         .channels_max = 2,
814                         .rates = WM0010_RATES,
815                         .formats = WM0010_FORMATS,
816                 },
817                 .capture = {
818                          .stream_name = "SDI1 Capture",
819                          .channels_min = 1,
820                          .channels_max = 2,
821                          .rates = WM0010_RATES,
822                          .formats = WM0010_FORMATS,
823                  },
824         },
825         {
826                 .name = "wm0010-sdi2",
827                 .playback = {
828                         .stream_name = "SDI2 Playback",
829                         .channels_min = 1,
830                         .channels_max = 2,
831                         .rates = WM0010_RATES,
832                         .formats = WM0010_FORMATS,
833                 },
834                 .capture = {
835                          .stream_name = "SDI2 Capture",
836                          .channels_min = 1,
837                          .channels_max = 2,
838                          .rates = WM0010_RATES,
839                          .formats = WM0010_FORMATS,
840                  },
841         },
842 };
843
844 static irqreturn_t wm0010_irq(int irq, void *data)
845 {
846         struct wm0010_priv *wm0010 = data;
847
848         switch (wm0010->state) {
849         case WM0010_OUT_OF_RESET:
850         case WM0010_BOOTROM:
851         case WM0010_STAGE2:
852                 spin_lock(&wm0010->irq_lock);
853                 complete(&wm0010->boot_completion);
854                 spin_unlock(&wm0010->irq_lock);
855                 return IRQ_HANDLED;
856         default:
857                 return IRQ_NONE;
858         }
859
860         return IRQ_NONE;
861 }
862
863 static int wm0010_probe(struct snd_soc_codec *codec)
864 {
865         struct wm0010_priv *wm0010 = snd_soc_codec_get_drvdata(codec);
866
867         wm0010->codec = codec;
868
869         return 0;
870 }
871
872 static int wm0010_spi_probe(struct spi_device *spi)
873 {
874         unsigned long gpio_flags;
875         int ret;
876         int trigger;
877         int irq;
878         struct wm0010_priv *wm0010;
879
880         wm0010 = devm_kzalloc(&spi->dev, sizeof(*wm0010),
881                               GFP_KERNEL);
882         if (!wm0010)
883                 return -ENOMEM;
884
885         mutex_init(&wm0010->lock);
886         spin_lock_init(&wm0010->irq_lock);
887
888         spi_set_drvdata(spi, wm0010);
889         wm0010->dev = &spi->dev;
890
891         if (dev_get_platdata(&spi->dev))
892                 memcpy(&wm0010->pdata, dev_get_platdata(&spi->dev),
893                        sizeof(wm0010->pdata));
894
895         init_completion(&wm0010->boot_completion);
896
897         wm0010->core_supplies[0].supply = "AVDD";
898         wm0010->core_supplies[1].supply = "DCVDD";
899         ret = devm_regulator_bulk_get(wm0010->dev, ARRAY_SIZE(wm0010->core_supplies),
900                                       wm0010->core_supplies);
901         if (ret != 0) {
902                 dev_err(wm0010->dev, "Failed to obtain core supplies: %d\n",
903                         ret);
904                 return ret;
905         }
906
907         wm0010->dbvdd = devm_regulator_get(wm0010->dev, "DBVDD");
908         if (IS_ERR(wm0010->dbvdd)) {
909                 ret = PTR_ERR(wm0010->dbvdd);
910                 dev_err(wm0010->dev, "Failed to obtain DBVDD: %d\n", ret);
911                 return ret;
912         }
913
914         if (wm0010->pdata.gpio_reset) {
915                 wm0010->gpio_reset = wm0010->pdata.gpio_reset;
916
917                 if (wm0010->pdata.reset_active_high)
918                         wm0010->gpio_reset_value = 1;
919                 else
920                         wm0010->gpio_reset_value = 0;
921
922                 if (wm0010->gpio_reset_value)
923                         gpio_flags = GPIOF_OUT_INIT_HIGH;
924                 else
925                         gpio_flags = GPIOF_OUT_INIT_LOW;
926
927                 ret = devm_gpio_request_one(wm0010->dev, wm0010->gpio_reset,
928                                             gpio_flags, "wm0010 reset");
929                 if (ret < 0) {
930                         dev_err(wm0010->dev,
931                                 "Failed to request GPIO for DSP reset: %d\n",
932                                 ret);
933                         return ret;
934                 }
935         } else {
936                 dev_err(wm0010->dev, "No reset GPIO configured\n");
937                 return -EINVAL;
938         }
939
940         wm0010->state = WM0010_POWER_OFF;
941
942         irq = spi->irq;
943         if (wm0010->pdata.irq_flags)
944                 trigger = wm0010->pdata.irq_flags;
945         else
946                 trigger = IRQF_TRIGGER_FALLING;
947         trigger |= IRQF_ONESHOT;
948
949         ret = request_threaded_irq(irq, NULL, wm0010_irq, trigger,
950                                    "wm0010", wm0010);
951         if (ret) {
952                 dev_err(wm0010->dev, "Failed to request IRQ %d: %d\n",
953                         irq, ret);
954                 return ret;
955         }
956         wm0010->irq = irq;
957
958         ret = irq_set_irq_wake(irq, 1);
959         if (ret) {
960                 dev_err(wm0010->dev, "Failed to set IRQ %d as wake source: %d\n",
961                         irq, ret);
962                 return ret;
963         }
964
965         if (spi->max_speed_hz)
966                 wm0010->board_max_spi_speed = spi->max_speed_hz;
967         else
968                 wm0010->board_max_spi_speed = 0;
969
970         ret = snd_soc_register_codec(&spi->dev,
971                                      &soc_codec_dev_wm0010, wm0010_dai,
972                                      ARRAY_SIZE(wm0010_dai));
973         if (ret < 0)
974                 return ret;
975
976         return 0;
977 }
978
979 static int wm0010_spi_remove(struct spi_device *spi)
980 {
981         struct wm0010_priv *wm0010 = spi_get_drvdata(spi);
982
983         snd_soc_unregister_codec(&spi->dev);
984
985         gpio_set_value_cansleep(wm0010->gpio_reset,
986                                 wm0010->gpio_reset_value);
987
988         irq_set_irq_wake(wm0010->irq, 0);
989
990         if (wm0010->irq)
991                 free_irq(wm0010->irq, wm0010);
992
993         return 0;
994 }
995
996 static struct spi_driver wm0010_spi_driver = {
997         .driver = {
998                 .name   = "wm0010",
999         },
1000         .probe          = wm0010_spi_probe,
1001         .remove         = wm0010_spi_remove,
1002 };
1003
1004 module_spi_driver(wm0010_spi_driver);
1005
1006 MODULE_DESCRIPTION("ASoC WM0010 driver");
1007 MODULE_AUTHOR("Mark Brown <broonie@opensource.wolfsonmicro.com>");
1008 MODULE_LICENSE("GPL");