xref: /linux/sound/soc/codecs/wm_adsp.c (revision 67f8bc848ee31831336bd478e57d2f993551902e)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * wm_adsp.c  --  Wolfson ADSP support
4  *
5  * Copyright 2012 Wolfson Microelectronics plc
6  *
7  * Author: Mark Brown <broonie@opensource.wolfsonmicro.com>
8  */
9 
10 #include <kunit/static_stub.h>
11 #include <kunit/visibility.h>
12 #include <linux/array_size.h>
13 #include <linux/cleanup.h>
14 #include <linux/ctype.h>
15 #include <linux/module.h>
16 #include <linux/moduleparam.h>
17 #include <linux/init.h>
18 #include <linux/delay.h>
19 #include <linux/firmware.h>
20 #include <linux/list.h>
21 #include <linux/pm.h>
22 #include <linux/regmap.h>
23 #include <linux/regulator/consumer.h>
24 #include <linux/slab.h>
25 #include <linux/string.h>
26 #include <linux/workqueue.h>
27 #include <linux/debugfs.h>
28 #include <sound/core.h>
29 #include <sound/pcm.h>
30 #include <sound/pcm_params.h>
31 #include <sound/soc.h>
32 #include <sound/jack.h>
33 #include <sound/initval.h>
34 #include <sound/tlv.h>
35 
36 #include "wm_adsp.h"
37 
38 #define adsp_crit(_dsp, fmt, ...) \
39 	dev_crit(_dsp->cs_dsp.dev, "%s: " fmt, _dsp->cs_dsp.name, ##__VA_ARGS__)
40 #define adsp_err(_dsp, fmt, ...) \
41 	dev_err(_dsp->cs_dsp.dev, "%s: " fmt, _dsp->cs_dsp.name, ##__VA_ARGS__)
42 #define adsp_warn(_dsp, fmt, ...) \
43 	dev_warn(_dsp->cs_dsp.dev, "%s: " fmt, _dsp->cs_dsp.name, ##__VA_ARGS__)
44 #define adsp_info(_dsp, fmt, ...) \
45 	dev_info(_dsp->cs_dsp.dev, "%s: " fmt, _dsp->cs_dsp.name, ##__VA_ARGS__)
46 #define adsp_dbg(_dsp, fmt, ...) \
47 	dev_dbg(_dsp->cs_dsp.dev, "%s: " fmt, _dsp->cs_dsp.name, ##__VA_ARGS__)
48 
49 #define compr_err(_obj, fmt, ...) \
50 	adsp_err(_obj->dsp, "%s: " fmt, _obj->name ? _obj->name : "legacy", \
51 		 ##__VA_ARGS__)
52 #define compr_dbg(_obj, fmt, ...) \
53 	adsp_dbg(_obj->dsp, "%s: " fmt, _obj->name ? _obj->name : "legacy", \
54 		 ##__VA_ARGS__)
55 
56 #define ADSP_MAX_STD_CTRL_SIZE               512
57 
58 static const struct cs_dsp_client_ops wm_adsp1_client_ops;
59 static const struct cs_dsp_client_ops wm_adsp2_client_ops;
60 
61 #define WM_ADSP_FW_MBC_VSS  0
62 #define WM_ADSP_FW_HIFI     1
63 #define WM_ADSP_FW_TX       2
64 #define WM_ADSP_FW_TX_SPK   3
65 #define WM_ADSP_FW_RX       4
66 #define WM_ADSP_FW_RX_ANC   5
67 #define WM_ADSP_FW_CTRL     6
68 #define WM_ADSP_FW_ASR      7
69 #define WM_ADSP_FW_TRACE    8
70 #define WM_ADSP_FW_SPK_PROT 9
71 #define WM_ADSP_FW_SPK_CALI 10
72 #define WM_ADSP_FW_SPK_DIAG 11
73 #define WM_ADSP_FW_MISC     12
74 
75 #define WM_ADSP_NUM_FW      13
76 
77 static const char * const wm_adsp_fw_text[WM_ADSP_NUM_FW] = {
78 	[WM_ADSP_FW_MBC_VSS] =  "MBC/VSS",
79 	[WM_ADSP_FW_HIFI] =     "MasterHiFi",
80 	[WM_ADSP_FW_TX] =       "Tx",
81 	[WM_ADSP_FW_TX_SPK] =   "Tx Speaker",
82 	[WM_ADSP_FW_RX] =       "Rx",
83 	[WM_ADSP_FW_RX_ANC] =   "Rx ANC",
84 	[WM_ADSP_FW_CTRL] =     "Voice Ctrl",
85 	[WM_ADSP_FW_ASR] =      "ASR Assist",
86 	[WM_ADSP_FW_TRACE] =    "Dbg Trace",
87 	[WM_ADSP_FW_SPK_PROT] = "Protection",
88 	[WM_ADSP_FW_SPK_CALI] = "Calibration",
89 	[WM_ADSP_FW_SPK_DIAG] = "Diagnostic",
90 	[WM_ADSP_FW_MISC] =     "Misc",
91 };
92 
93 struct wm_adsp_system_config_xm_hdr {
94 	__be32 sys_enable;
95 	__be32 fw_id;
96 	__be32 fw_rev;
97 	__be32 boot_status;
98 	__be32 watchdog;
99 	__be32 dma_buffer_size;
100 	__be32 rdma[6];
101 	__be32 wdma[8];
102 	__be32 build_job_name[3];
103 	__be32 build_job_number;
104 } __packed;
105 
106 struct wm_halo_system_config_xm_hdr {
107 	__be32 halo_heartbeat;
108 	__be32 build_job_name[3];
109 	__be32 build_job_number;
110 } __packed;
111 
112 struct wm_adsp_alg_xm_struct {
113 	__be32 magic;
114 	__be32 smoothing;
115 	__be32 threshold;
116 	__be32 host_buf_ptr;
117 	__be32 start_seq;
118 	__be32 high_water_mark;
119 	__be32 low_water_mark;
120 	__be64 smoothed_power;
121 } __packed;
122 
123 struct wm_adsp_host_buf_coeff_v1 {
124 	__be32 host_buf_ptr;		/* Host buffer pointer */
125 	__be32 versions;		/* Version numbers */
126 	__be32 name[4];			/* The buffer name */
127 } __packed;
128 
129 struct wm_adsp_buffer {
130 	__be32 buf1_base;		/* Base addr of first buffer area */
131 	__be32 buf1_size;		/* Size of buf1 area in DSP words */
132 	__be32 buf2_base;		/* Base addr of 2nd buffer area */
133 	__be32 buf1_buf2_size;		/* Size of buf1+buf2 in DSP words */
134 	__be32 buf3_base;		/* Base addr of buf3 area */
135 	__be32 buf_total_size;		/* Size of buf1+buf2+buf3 in DSP words */
136 	__be32 high_water_mark;		/* Point at which IRQ is asserted */
137 	__be32 irq_count;		/* bits 1-31 count IRQ assertions */
138 	__be32 irq_ack;			/* acked IRQ count, bit 0 enables IRQ */
139 	__be32 next_write_index;	/* word index of next write */
140 	__be32 next_read_index;		/* word index of next read */
141 	__be32 error;			/* error if any */
142 	__be32 oldest_block_index;	/* word index of oldest surviving */
143 	__be32 requested_rewind;	/* how many blocks rewind was done */
144 	__be32 reserved_space;		/* internal */
145 	__be32 min_free;		/* min free space since stream start */
146 	__be32 blocks_written[2];	/* total blocks written (64 bit) */
147 	__be32 words_written[2];	/* total words written (64 bit) */
148 } __packed;
149 
150 struct wm_adsp_compr;
151 
152 struct wm_adsp_compr_buf {
153 	struct list_head list;
154 	struct wm_adsp *dsp;
155 	struct wm_adsp_compr *compr;
156 
157 	struct wm_adsp_buffer_region *regions;
158 	u32 host_buf_ptr;
159 
160 	u32 error;
161 	u32 irq_count;
162 	int read_index;
163 	int avail;
164 	int host_buf_mem_type;
165 
166 	char *name;
167 };
168 
169 struct wm_adsp_compr {
170 	struct list_head list;
171 	struct wm_adsp *dsp;
172 	struct wm_adsp_compr_buf *buf;
173 
174 	struct snd_compr_stream *stream;
175 	struct snd_compressed_buffer size;
176 
177 	u32 *raw_buf;
178 	u64 copied_total;
179 
180 	unsigned int sample_rate;
181 
182 	const char *name;
183 };
184 
185 #define WM_ADSP_MIN_FRAGMENTS          1
186 #define WM_ADSP_MAX_FRAGMENTS          256
187 #define WM_ADSP_MIN_FRAGMENT_SIZE      (16 * CS_DSP_DATA_WORD_SIZE)
188 #define WM_ADSP_MAX_FRAGMENT_SIZE      (4096 * CS_DSP_DATA_WORD_SIZE)
189 
190 #define WM_ADSP_ALG_XM_STRUCT_MAGIC    0x49aec7
191 
192 #define HOST_BUFFER_FIELD(field) \
193 	(offsetof(struct wm_adsp_buffer, field) / sizeof(__be32))
194 
195 #define ALG_XM_FIELD(field) \
196 	(offsetof(struct wm_adsp_alg_xm_struct, field) / sizeof(__be32))
197 
198 #define HOST_BUF_COEFF_SUPPORTED_COMPAT_VER	1
199 
200 #define HOST_BUF_COEFF_COMPAT_VER_MASK		0xFF00
201 #define HOST_BUF_COEFF_COMPAT_VER_SHIFT		8
202 
203 static int wm_adsp_buffer_init(struct wm_adsp *dsp);
204 static int wm_adsp_buffer_free(struct wm_adsp *dsp);
205 
206 struct wm_adsp_buffer_region {
207 	unsigned int offset;
208 	unsigned int cumulative_size;
209 	unsigned int mem_type;
210 	unsigned int base_addr;
211 };
212 
213 struct wm_adsp_buffer_region_def {
214 	unsigned int mem_type;
215 	unsigned int base_offset;
216 	unsigned int size_offset;
217 };
218 
219 static const struct wm_adsp_buffer_region_def default_regions[] = {
220 	{
221 		.mem_type = WMFW_ADSP2_XM,
222 		.base_offset = HOST_BUFFER_FIELD(buf1_base),
223 		.size_offset = HOST_BUFFER_FIELD(buf1_size),
224 	},
225 	{
226 		.mem_type = WMFW_ADSP2_XM,
227 		.base_offset = HOST_BUFFER_FIELD(buf2_base),
228 		.size_offset = HOST_BUFFER_FIELD(buf1_buf2_size),
229 	},
230 	{
231 		.mem_type = WMFW_ADSP2_YM,
232 		.base_offset = HOST_BUFFER_FIELD(buf3_base),
233 		.size_offset = HOST_BUFFER_FIELD(buf_total_size),
234 	},
235 };
236 
237 struct wm_adsp_fw_caps {
238 	u32 id;
239 	struct snd_codec_desc desc;
240 	int num_regions;
241 	const struct wm_adsp_buffer_region_def *region_defs;
242 };
243 
244 static const struct wm_adsp_fw_caps ctrl_caps[] = {
245 	{
246 		.id = SND_AUDIOCODEC_BESPOKE,
247 		.desc = {
248 			.max_ch = 8,
249 			.sample_rates = { 16000 },
250 			.num_sample_rates = 1,
251 			.formats = SNDRV_PCM_FMTBIT_S16_LE,
252 		},
253 		.num_regions = ARRAY_SIZE(default_regions),
254 		.region_defs = default_regions,
255 	},
256 };
257 
258 static const struct wm_adsp_fw_caps trace_caps[] = {
259 	{
260 		.id = SND_AUDIOCODEC_BESPOKE,
261 		.desc = {
262 			.max_ch = 8,
263 			.sample_rates = {
264 				4000, 8000, 11025, 12000, 16000, 22050,
265 				24000, 32000, 44100, 48000, 64000, 88200,
266 				96000, 176400, 192000
267 			},
268 			.num_sample_rates = 15,
269 			.formats = SNDRV_PCM_FMTBIT_S16_LE,
270 		},
271 		.num_regions = ARRAY_SIZE(default_regions),
272 		.region_defs = default_regions,
273 	},
274 };
275 
276 static const struct {
277 	const char *file;
278 	int compr_direction;
279 	int num_caps;
280 	const struct wm_adsp_fw_caps *caps;
281 	bool voice_trigger;
282 } wm_adsp_fw[WM_ADSP_NUM_FW] = {
283 	[WM_ADSP_FW_MBC_VSS] =  { .file = "mbc-vss" },
284 	[WM_ADSP_FW_HIFI] =     { .file = "hifi" },
285 	[WM_ADSP_FW_TX] =       { .file = "tx" },
286 	[WM_ADSP_FW_TX_SPK] =   { .file = "tx-spk" },
287 	[WM_ADSP_FW_RX] =       { .file = "rx" },
288 	[WM_ADSP_FW_RX_ANC] =   { .file = "rx-anc" },
289 	[WM_ADSP_FW_CTRL] =     {
290 		.file = "ctrl",
291 		.compr_direction = SND_COMPRESS_CAPTURE,
292 		.num_caps = ARRAY_SIZE(ctrl_caps),
293 		.caps = ctrl_caps,
294 		.voice_trigger = true,
295 	},
296 	[WM_ADSP_FW_ASR] =      { .file = "asr" },
297 	[WM_ADSP_FW_TRACE] =    {
298 		.file = "trace",
299 		.compr_direction = SND_COMPRESS_CAPTURE,
300 		.num_caps = ARRAY_SIZE(trace_caps),
301 		.caps = trace_caps,
302 	},
303 	[WM_ADSP_FW_SPK_PROT] = {
304 		.file = "spk-prot",
305 		.compr_direction = SND_COMPRESS_CAPTURE,
306 		.num_caps = ARRAY_SIZE(trace_caps),
307 		.caps = trace_caps,
308 	},
309 	[WM_ADSP_FW_SPK_CALI] = { .file = "spk-cali" },
310 	[WM_ADSP_FW_SPK_DIAG] = { .file = "spk-diag" },
311 	[WM_ADSP_FW_MISC] =     { .file = "misc" },
312 };
313 
314 struct wm_coeff_ctl {
315 	const char *name;
316 	struct cs_dsp_coeff_ctl *cs_ctl;
317 	struct soc_bytes_ext bytes_ext;
318 	struct work_struct work;
319 };
320 
321 #if IS_ENABLED(CONFIG_KUNIT)
322 const char *wm_adsp_get_fwf_name_by_index(int index)
323 {
324 	if (index < ARRAY_SIZE(wm_adsp_fw))
325 		return wm_adsp_fw[index].file;
326 
327 	return NULL;
328 }
329 EXPORT_SYMBOL_IF_KUNIT(wm_adsp_get_fwf_name_by_index);
330 #endif
331 
332 int wm_adsp_fw_get(struct snd_kcontrol *kcontrol,
333 		   struct snd_ctl_elem_value *ucontrol)
334 {
335 	struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
336 	struct soc_enum *e = (struct soc_enum *)kcontrol->private_value;
337 	struct wm_adsp *dsp = snd_soc_component_get_drvdata(component);
338 
339 	ucontrol->value.enumerated.item[0] = dsp[e->shift_l].fw;
340 
341 	return 0;
342 }
343 EXPORT_SYMBOL_GPL(wm_adsp_fw_get);
344 
345 int wm_adsp_fw_put(struct snd_kcontrol *kcontrol,
346 		   struct snd_ctl_elem_value *ucontrol)
347 {
348 	struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
349 	struct soc_enum *e = (struct soc_enum *)kcontrol->private_value;
350 	struct wm_adsp *dsp = snd_soc_component_get_drvdata(component);
351 
352 	if (ucontrol->value.enumerated.item[0] == dsp[e->shift_l].fw)
353 		return 0;
354 
355 	if (ucontrol->value.enumerated.item[0] >= WM_ADSP_NUM_FW)
356 		return -EINVAL;
357 
358 	guard(mutex)(&dsp[e->shift_l].cs_dsp.pwr_lock);
359 
360 	if (dsp[e->shift_l].cs_dsp.booted || !list_empty(&dsp[e->shift_l].compr_list))
361 		return -EBUSY;
362 	else
363 		dsp[e->shift_l].fw = ucontrol->value.enumerated.item[0];
364 
365 	return 1;
366 }
367 EXPORT_SYMBOL_GPL(wm_adsp_fw_put);
368 
369 const struct soc_enum wm_adsp_fw_enum[] = {
370 	SOC_ENUM_SINGLE(0, 0, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
371 	SOC_ENUM_SINGLE(0, 1, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
372 	SOC_ENUM_SINGLE(0, 2, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
373 	SOC_ENUM_SINGLE(0, 3, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
374 	SOC_ENUM_SINGLE(0, 4, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
375 	SOC_ENUM_SINGLE(0, 5, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
376 	SOC_ENUM_SINGLE(0, 6, ARRAY_SIZE(wm_adsp_fw_text), wm_adsp_fw_text),
377 };
378 EXPORT_SYMBOL_GPL(wm_adsp_fw_enum);
379 
380 static inline struct wm_coeff_ctl *bytes_ext_to_ctl(struct soc_bytes_ext *ext)
381 {
382 	return container_of(ext, struct wm_coeff_ctl, bytes_ext);
383 }
384 
385 static int wm_coeff_info(struct snd_kcontrol *kctl,
386 			 struct snd_ctl_elem_info *uinfo)
387 {
388 	struct soc_bytes_ext *bytes_ext =
389 		(struct soc_bytes_ext *)kctl->private_value;
390 	struct wm_coeff_ctl *ctl = bytes_ext_to_ctl(bytes_ext);
391 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
392 
393 	switch (cs_ctl->type) {
394 	case WMFW_CTL_TYPE_ACKED:
395 		uinfo->type = SNDRV_CTL_ELEM_TYPE_INTEGER;
396 		uinfo->value.integer.min = CS_DSP_ACKED_CTL_MIN_VALUE;
397 		uinfo->value.integer.max = CS_DSP_ACKED_CTL_MAX_VALUE;
398 		uinfo->value.integer.step = 1;
399 		uinfo->count = 1;
400 		break;
401 	default:
402 		uinfo->type = SNDRV_CTL_ELEM_TYPE_BYTES;
403 		uinfo->count = cs_ctl->len;
404 		break;
405 	}
406 
407 	return 0;
408 }
409 
410 static int wm_coeff_put(struct snd_kcontrol *kctl,
411 			struct snd_ctl_elem_value *ucontrol)
412 {
413 	struct soc_bytes_ext *bytes_ext =
414 		(struct soc_bytes_ext *)kctl->private_value;
415 	struct wm_coeff_ctl *ctl = bytes_ext_to_ctl(bytes_ext);
416 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
417 	char *p = ucontrol->value.bytes.data;
418 
419 	return cs_dsp_coeff_lock_and_write_ctrl(cs_ctl, 0, p, cs_ctl->len);
420 }
421 
422 static int wm_coeff_tlv_put(struct snd_kcontrol *kctl,
423 			    const unsigned int __user *bytes, unsigned int size)
424 {
425 	struct soc_bytes_ext *bytes_ext =
426 		(struct soc_bytes_ext *)kctl->private_value;
427 	struct wm_coeff_ctl *ctl = bytes_ext_to_ctl(bytes_ext);
428 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
429 	void *scratch __free(kvfree) = vmemdup_user(bytes, size);
430 
431 	if (IS_ERR(scratch))
432 		return PTR_ERR(scratch);
433 
434 	return cs_dsp_coeff_lock_and_write_ctrl(cs_ctl, 0, scratch, size);
435 }
436 
437 static int wm_coeff_put_acked(struct snd_kcontrol *kctl,
438 			      struct snd_ctl_elem_value *ucontrol)
439 {
440 	struct soc_bytes_ext *bytes_ext =
441 		(struct soc_bytes_ext *)kctl->private_value;
442 	struct wm_coeff_ctl *ctl = bytes_ext_to_ctl(bytes_ext);
443 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
444 	unsigned int val = ucontrol->value.integer.value[0];
445 	int ret;
446 
447 	if (val == 0)
448 		return 0;	/* 0 means no event */
449 
450 	guard(mutex)(&cs_ctl->dsp->pwr_lock);
451 
452 	if (cs_ctl->enabled)
453 		ret = cs_dsp_coeff_write_acked_control(cs_ctl, val);
454 	else
455 		ret = -EPERM;
456 
457 	if (ret < 0)
458 		return ret;
459 
460 	return 1;
461 }
462 
463 static int wm_coeff_get(struct snd_kcontrol *kctl,
464 			struct snd_ctl_elem_value *ucontrol)
465 {
466 	struct soc_bytes_ext *bytes_ext =
467 		(struct soc_bytes_ext *)kctl->private_value;
468 	struct wm_coeff_ctl *ctl = bytes_ext_to_ctl(bytes_ext);
469 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
470 	char *p = ucontrol->value.bytes.data;
471 
472 	return cs_dsp_coeff_lock_and_read_ctrl(cs_ctl, 0, p, cs_ctl->len);
473 }
474 
475 static int wm_coeff_tlv_get(struct snd_kcontrol *kctl,
476 			    unsigned int __user *bytes, unsigned int size)
477 {
478 	struct soc_bytes_ext *bytes_ext =
479 		(struct soc_bytes_ext *)kctl->private_value;
480 	struct wm_coeff_ctl *ctl = bytes_ext_to_ctl(bytes_ext);
481 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
482 	int ret = 0;
483 
484 	guard(mutex)(&cs_ctl->dsp->pwr_lock);
485 
486 	ret = cs_dsp_coeff_read_ctrl(cs_ctl, 0, cs_ctl->cache, size);
487 
488 	if (!ret && copy_to_user(bytes, cs_ctl->cache, size))
489 		ret = -EFAULT;
490 
491 	return ret;
492 }
493 
494 static int wm_coeff_get_acked(struct snd_kcontrol *kcontrol,
495 			      struct snd_ctl_elem_value *ucontrol)
496 {
497 	/*
498 	 * Although it's not useful to read an acked control, we must satisfy
499 	 * user-side assumptions that all controls are readable and that a
500 	 * write of the same value should be filtered out (it's valid to send
501 	 * the same event number again to the firmware). We therefore return 0,
502 	 * meaning "no event" so valid event numbers will always be a change
503 	 */
504 	ucontrol->value.integer.value[0] = 0;
505 
506 	return 0;
507 }
508 
509 static unsigned int wmfw_convert_flags(unsigned int in, unsigned int len)
510 {
511 	unsigned int out, rd, wr, vol;
512 
513 	if (len > ADSP_MAX_STD_CTRL_SIZE) {
514 		rd = SNDRV_CTL_ELEM_ACCESS_TLV_READ;
515 		wr = SNDRV_CTL_ELEM_ACCESS_TLV_WRITE;
516 		vol = SNDRV_CTL_ELEM_ACCESS_VOLATILE;
517 
518 		out = SNDRV_CTL_ELEM_ACCESS_TLV_CALLBACK;
519 	} else {
520 		rd = SNDRV_CTL_ELEM_ACCESS_READ;
521 		wr = SNDRV_CTL_ELEM_ACCESS_WRITE;
522 		vol = SNDRV_CTL_ELEM_ACCESS_VOLATILE;
523 
524 		out = 0;
525 	}
526 
527 	if (in) {
528 		out |= rd;
529 		if (in & WMFW_CTL_FLAG_WRITEABLE)
530 			out |= wr;
531 		if (in & WMFW_CTL_FLAG_VOLATILE)
532 			out |= vol;
533 	} else {
534 		out |= rd | wr | vol;
535 	}
536 
537 	return out;
538 }
539 
540 static void wm_adsp_ctl_work(struct work_struct *work)
541 {
542 	struct wm_coeff_ctl *ctl = container_of(work,
543 						struct wm_coeff_ctl,
544 						work);
545 	struct cs_dsp_coeff_ctl *cs_ctl = ctl->cs_ctl;
546 	struct wm_adsp *dsp = container_of(cs_ctl->dsp,
547 					   struct wm_adsp,
548 					   cs_dsp);
549 	struct snd_kcontrol_new *kcontrol;
550 
551 	kcontrol = kzalloc_obj(*kcontrol);
552 	if (!kcontrol)
553 		return;
554 
555 	kcontrol->name = ctl->name;
556 	kcontrol->info = wm_coeff_info;
557 	kcontrol->iface = SNDRV_CTL_ELEM_IFACE_MIXER;
558 	kcontrol->tlv.c = snd_soc_bytes_tlv_callback;
559 	kcontrol->private_value = (unsigned long)&ctl->bytes_ext;
560 	kcontrol->access = wmfw_convert_flags(cs_ctl->flags, cs_ctl->len);
561 
562 	switch (cs_ctl->type) {
563 	case WMFW_CTL_TYPE_ACKED:
564 		kcontrol->get = wm_coeff_get_acked;
565 		kcontrol->put = wm_coeff_put_acked;
566 		break;
567 	default:
568 		if (kcontrol->access & SNDRV_CTL_ELEM_ACCESS_TLV_CALLBACK) {
569 			ctl->bytes_ext.max = cs_ctl->len;
570 			ctl->bytes_ext.get = wm_coeff_tlv_get;
571 			ctl->bytes_ext.put = wm_coeff_tlv_put;
572 		} else {
573 			kcontrol->get = wm_coeff_get;
574 			kcontrol->put = wm_coeff_put;
575 		}
576 		break;
577 	}
578 
579 	snd_soc_add_component_controls(dsp->component, kcontrol, 1);
580 
581 	kfree(kcontrol);
582 }
583 
584 int wm_adsp_control_add(struct cs_dsp_coeff_ctl *cs_ctl)
585 {
586 	struct wm_adsp *dsp = container_of(cs_ctl->dsp, struct wm_adsp, cs_dsp);
587 	struct cs_dsp *cs_dsp = &dsp->cs_dsp;
588 	struct wm_coeff_ctl *ctl;
589 	char name[SNDRV_CTL_ELEM_ID_NAME_MAXLEN];
590 	const char *region_name;
591 	int ret;
592 
593 	if (cs_ctl->flags & WMFW_CTL_FLAG_SYS)
594 		return 0;
595 
596 	region_name = cs_dsp_mem_region_name(cs_ctl->alg_region.type);
597 	if (!region_name) {
598 		adsp_err(dsp, "Unknown region type: %d\n", cs_ctl->alg_region.type);
599 		return -EINVAL;
600 	}
601 
602 	switch (cs_dsp->wmfw_ver) {
603 	case 0:
604 	case 1:
605 		ret = scnprintf(name, SNDRV_CTL_ELEM_ID_NAME_MAXLEN,
606 				"%s %s %x", cs_dsp->name, region_name,
607 				cs_ctl->alg_region.alg);
608 		break;
609 	case 2:
610 		ret = scnprintf(name, SNDRV_CTL_ELEM_ID_NAME_MAXLEN,
611 				"%s%c %.12s %x", cs_dsp->name, *region_name,
612 				wm_adsp_fw_text[dsp->fw], cs_ctl->alg_region.alg);
613 		break;
614 	default:
615 		ret = scnprintf(name, SNDRV_CTL_ELEM_ID_NAME_MAXLEN,
616 				"%s %.12s %x", cs_dsp->name,
617 				wm_adsp_fw_text[dsp->fw], cs_ctl->alg_region.alg);
618 		break;
619 	}
620 
621 	if (cs_ctl->subname) {
622 		int avail = SNDRV_CTL_ELEM_ID_NAME_MAXLEN - ret - 2;
623 		int skip = 0;
624 
625 		if (dsp->component->name_prefix)
626 			avail -= strlen(dsp->component->name_prefix) + 1;
627 
628 		/* Truncate the subname from the start if it is too long */
629 		if (cs_ctl->subname_len > avail)
630 			skip = cs_ctl->subname_len - avail;
631 
632 		snprintf(name + ret, SNDRV_CTL_ELEM_ID_NAME_MAXLEN - ret,
633 			 " %.*s", cs_ctl->subname_len - skip, cs_ctl->subname + skip);
634 	}
635 
636 	ctl = kzalloc_obj(*ctl);
637 	if (!ctl)
638 		return -ENOMEM;
639 	ctl->cs_ctl = cs_ctl;
640 
641 	ctl->name = kmemdup(name, strlen(name) + 1, GFP_KERNEL);
642 	if (!ctl->name) {
643 		ret = -ENOMEM;
644 		goto err_ctl;
645 	}
646 
647 	cs_ctl->priv = ctl;
648 
649 	INIT_WORK(&ctl->work, wm_adsp_ctl_work);
650 	schedule_work(&ctl->work);
651 
652 	return 0;
653 
654 err_ctl:
655 	kfree(ctl);
656 
657 	return ret;
658 }
659 EXPORT_SYMBOL_GPL(wm_adsp_control_add);
660 
661 static int wm_adsp_control_add_cb(struct cs_dsp_coeff_ctl *cs_ctl)
662 {
663 	struct wm_adsp *dsp = container_of(cs_ctl->dsp, struct wm_adsp, cs_dsp);
664 
665 	if (dsp->control_add)
666 		return (dsp->control_add)(dsp, cs_ctl);
667 	else
668 		return wm_adsp_control_add(cs_ctl);
669 }
670 
671 static void wm_adsp_control_remove(struct cs_dsp_coeff_ctl *cs_ctl)
672 {
673 	struct wm_coeff_ctl *ctl = cs_ctl->priv;
674 
675 	if (!ctl)
676 		return;
677 
678 	cancel_work_sync(&ctl->work);
679 
680 	kfree(ctl->name);
681 	kfree(ctl);
682 }
683 
684 int wm_adsp_write_ctl(struct wm_adsp *dsp, const char *name, int type,
685 		      unsigned int alg, void *buf, size_t len)
686 {
687 	struct cs_dsp_coeff_ctl *cs_ctl;
688 	int ret;
689 
690 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
691 	cs_ctl = cs_dsp_get_ctl(&dsp->cs_dsp, name, type, alg);
692 	ret = cs_dsp_coeff_write_ctrl(cs_ctl, 0, buf, len);
693 
694 	if (ret < 0)
695 		return ret;
696 
697 	return 0;
698 }
699 EXPORT_SYMBOL_GPL(wm_adsp_write_ctl);
700 
701 int wm_adsp_read_ctl(struct wm_adsp *dsp, const char *name, int type,
702 		     unsigned int alg, void *buf, size_t len)
703 {
704 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
705 
706 	return cs_dsp_coeff_read_ctrl(cs_dsp_get_ctl(&dsp->cs_dsp, name, type, alg),
707 				     0, buf, len);
708 }
709 EXPORT_SYMBOL_GPL(wm_adsp_read_ctl);
710 
711 VISIBLE_IF_KUNIT void wm_adsp_release_firmware_files(struct wm_adsp_fw_files *fw)
712 {
713 	KUNIT_STATIC_STUB_REDIRECT(wm_adsp_release_firmware_files, fw);
714 
715 	release_firmware(fw->wmfw.firmware);
716 	kfree(fw->wmfw.filename);
717 
718 	release_firmware(fw->coeff.firmware);
719 	kfree(fw->coeff.filename);
720 }
721 EXPORT_SYMBOL_IF_KUNIT(wm_adsp_release_firmware_files);
722 
723 VISIBLE_IF_KUNIT int wm_adsp_firmware_request(const struct firmware **firmware,
724 					      const char *filename,
725 					      struct device *dev)
726 {
727 	KUNIT_STATIC_STUB_REDIRECT(wm_adsp_firmware_request, firmware, filename, dev);
728 
729 	return firmware_request_nowarn(firmware, filename, dev);
730 }
731 EXPORT_SYMBOL_IF_KUNIT(wm_adsp_firmware_request);
732 
733 static int wm_adsp_request_firmware_file(struct wm_adsp *dsp,
734 					 struct wm_adsp_fw_file *fw,
735 					 const char *dir, const char *system_name,
736 					 const char *asoc_component_prefix,
737 					 const char *filetype)
738 {
739 	struct cs_dsp *cs_dsp = &dsp->cs_dsp;
740 	const char *fwf;
741 	char *s, c;
742 	int ret;
743 
744 	if (dsp->fwf_name)
745 		fwf = dsp->fwf_name;
746 	else
747 		fwf = dsp->cs_dsp.name;
748 
749 	if (system_name && asoc_component_prefix)
750 		fw->filename = kasprintf(GFP_KERNEL, "%s%s-%s-%s-%s-%s.%s", dir, dsp->part,
751 					 fwf, wm_adsp_fw[dsp->fw].file, system_name,
752 					 asoc_component_prefix, filetype);
753 	else if (system_name)
754 		fw->filename = kasprintf(GFP_KERNEL, "%s%s-%s-%s-%s.%s", dir, dsp->part,
755 					 fwf, wm_adsp_fw[dsp->fw].file, system_name,
756 					 filetype);
757 	else
758 		fw->filename = kasprintf(GFP_KERNEL, "%s%s-%s-%s.%s", dir, dsp->part, fwf,
759 					 wm_adsp_fw[dsp->fw].file, filetype);
760 
761 	if (!fw->filename)
762 		return -ENOMEM;
763 
764 	/*
765 	 * Make sure that filename after dir is lower-case and any non-alpha-numeric
766 	 * characters except full-stop are replaced with hyphens.
767 	 */
768 	s = fw->filename + strlen(dir);
769 	while (*s) {
770 		c = *s;
771 		if (isalnum(c))
772 			*s = tolower(c);
773 		else if (c != '.')
774 			*s = '-';
775 		s++;
776 	}
777 
778 	ret = wm_adsp_firmware_request(&fw->firmware, fw->filename, cs_dsp->dev);
779 	if (ret < 0) {
780 		adsp_dbg(dsp, "Failed to request '%s': %d\n", fw->filename, ret);
781 		kfree(fw->filename);
782 		fw->filename = NULL;
783 		if (ret != -ENOENT)
784 			return ret;
785 	} else {
786 		adsp_dbg(dsp, "Found '%s'\n", fw->filename);
787 	}
788 
789 	return 0;
790 }
791 
792 static const char * const cirrus_dir = "cirrus/";
793 VISIBLE_IF_KUNIT int wm_adsp_request_firmware_files(struct wm_adsp *dsp,
794 						    struct wm_adsp_fw_files *fw)
795 {
796 	const char *system_name = dsp->system_name;
797 	const char *suffix = dsp->component->name_prefix;
798 	bool require_bin_suffix = false;
799 	int ret = 0;
800 
801 	if (dsp->fwf_suffix)
802 		suffix = dsp->fwf_suffix;
803 
804 	if (system_name) {
805 		ret = wm_adsp_request_firmware_file(dsp, &fw->wmfw,
806 						    cirrus_dir, system_name,
807 						    suffix, "wmfw");
808 		if (ret < 0)
809 			goto err;
810 
811 		if (suffix) {
812 			if (fw->wmfw.firmware) {
813 				require_bin_suffix = true;
814 			} else {
815 				/* Fallback to name without suffix */
816 				ret = wm_adsp_request_firmware_file(dsp, &fw->wmfw,
817 								    cirrus_dir, system_name,
818 								    NULL, "wmfw");
819 				if (ret < 0)
820 					goto err;
821 			}
822 		}
823 
824 		/* Look for matching .bin file */
825 		if (fw->wmfw.firmware || dsp->wmfw_optional) {
826 			ret = wm_adsp_request_firmware_file(dsp, &fw->coeff,
827 							    cirrus_dir, system_name,
828 							    suffix, "bin");
829 			if (ret < 0)
830 				goto err;
831 
832 			if (suffix && !fw->coeff.firmware && !require_bin_suffix) {
833 				/* Fallback to name without suffix */
834 				ret = wm_adsp_request_firmware_file(dsp,
835 								    &fw->coeff,
836 								    cirrus_dir, system_name,
837 								    NULL, "bin");
838 				if (ret < 0)
839 					goto err;
840 			}
841 		}
842 
843 		if (fw->wmfw.firmware || (dsp->wmfw_optional && fw->coeff.firmware))
844 			return 0;
845 	}
846 
847 	/* Check legacy location */
848 	ret = wm_adsp_request_firmware_file(dsp, &fw->wmfw, "", NULL, NULL, "wmfw");
849 	if (ret < 0)
850 		goto err;
851 
852 	if (fw->wmfw.firmware) {
853 		ret = wm_adsp_request_firmware_file(dsp, &fw->coeff, "", NULL, NULL, "bin");
854 		if (ret < 0)
855 			goto err;
856 
857 		return 0;
858 	}
859 
860 	/* Fall back to generic wmfw and optional matching bin */
861 	ret = wm_adsp_request_firmware_file(dsp, &fw->wmfw,
862 					    cirrus_dir, NULL, NULL, "wmfw");
863 	if (ret < 0)
864 		goto err;
865 
866 	if (fw->wmfw.firmware || dsp->wmfw_optional) {
867 		ret = wm_adsp_request_firmware_file(dsp, &fw->coeff,
868 						    cirrus_dir, NULL, NULL, "bin");
869 		if (ret < 0)
870 			goto err;
871 
872 		return 0;
873 	}
874 
875 	adsp_err(dsp, "Failed to request firmware <%s>%s-%s-%s<-%s<%s>>.wmfw\n",
876 		 cirrus_dir, dsp->part,
877 		 dsp->fwf_name ? dsp->fwf_name : dsp->cs_dsp.name,
878 		 wm_adsp_fw[dsp->fw].file, system_name, suffix);
879 
880 	ret = -ENOENT;
881 err:
882 	wm_adsp_release_firmware_files(fw);
883 
884 	return ret;
885 }
886 EXPORT_SYMBOL_IF_KUNIT(wm_adsp_request_firmware_files);
887 
888 static int wm_adsp_common_init(struct wm_adsp *dsp)
889 {
890 	INIT_LIST_HEAD(&dsp->compr_list);
891 	INIT_LIST_HEAD(&dsp->buffer_list);
892 
893 	return 0;
894 }
895 
896 int wm_adsp1_init(struct wm_adsp *dsp)
897 {
898 	int ret;
899 
900 	dsp->cs_dsp.client_ops = &wm_adsp1_client_ops;
901 
902 	ret = cs_dsp_adsp1_init(&dsp->cs_dsp);
903 	if (ret)
904 		return ret;
905 
906 	return wm_adsp_common_init(dsp);
907 }
908 EXPORT_SYMBOL_GPL(wm_adsp1_init);
909 
910 int wm_adsp1_event(struct snd_soc_dapm_widget *w,
911 		   struct snd_kcontrol *kcontrol,
912 		   int event)
913 {
914 	struct snd_soc_component *component = snd_soc_dapm_to_component(w->dapm);
915 	struct wm_adsp *dsps = snd_soc_component_get_drvdata(component);
916 	struct wm_adsp *dsp = &dsps[w->shift];
917 	struct wm_adsp_fw_files fw = { 0 };
918 	int ret = 0;
919 
920 	dsp->component = component;
921 
922 	switch (event) {
923 	case SND_SOC_DAPM_POST_PMU:
924 		ret = wm_adsp_request_firmware_files(dsp, &fw);
925 		if (ret)
926 			break;
927 
928 		ret = cs_dsp_adsp1_power_up(&dsp->cs_dsp,
929 					    fw.wmfw.firmware, fw.wmfw.filename,
930 					    fw.coeff.firmware, fw.coeff.filename,
931 					    wm_adsp_fw_text[dsp->fw]);
932 
933 		wm_adsp_release_firmware_files(&fw);
934 		break;
935 	case SND_SOC_DAPM_PRE_PMD:
936 		cs_dsp_adsp1_power_down(&dsp->cs_dsp);
937 		break;
938 	default:
939 		break;
940 	}
941 
942 	return ret;
943 }
944 EXPORT_SYMBOL_GPL(wm_adsp1_event);
945 
946 int wm_adsp2_set_dspclk(struct snd_soc_dapm_widget *w, unsigned int freq)
947 {
948 	struct snd_soc_component *component = snd_soc_dapm_to_component(w->dapm);
949 	struct wm_adsp *dsps = snd_soc_component_get_drvdata(component);
950 	struct wm_adsp *dsp = &dsps[w->shift];
951 
952 	return cs_dsp_set_dspclk(&dsp->cs_dsp, freq);
953 }
954 EXPORT_SYMBOL_GPL(wm_adsp2_set_dspclk);
955 
956 int wm_adsp2_preloader_get(struct snd_kcontrol *kcontrol,
957 			   struct snd_ctl_elem_value *ucontrol)
958 {
959 	struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
960 	struct wm_adsp *dsps = snd_soc_component_get_drvdata(component);
961 	struct soc_mixer_control *mc =
962 		(struct soc_mixer_control *)kcontrol->private_value;
963 	struct wm_adsp *dsp = &dsps[mc->shift - 1];
964 
965 	ucontrol->value.integer.value[0] = dsp->preloaded;
966 
967 	return 0;
968 }
969 EXPORT_SYMBOL_GPL(wm_adsp2_preloader_get);
970 
971 int wm_adsp2_preloader_put(struct snd_kcontrol *kcontrol,
972 			   struct snd_ctl_elem_value *ucontrol)
973 {
974 	struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
975 	struct wm_adsp *dsps = snd_soc_component_get_drvdata(component);
976 	struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);
977 	struct soc_mixer_control *mc =
978 		(struct soc_mixer_control *)kcontrol->private_value;
979 	struct wm_adsp *dsp = &dsps[mc->shift - 1];
980 	char preload[32];
981 
982 	if (dsp->preloaded == ucontrol->value.integer.value[0])
983 		return 0;
984 
985 	snprintf(preload, ARRAY_SIZE(preload), "%s Preload", dsp->cs_dsp.name);
986 
987 	if (ucontrol->value.integer.value[0] || dsp->toggle_preload)
988 		snd_soc_dapm_force_enable_pin(dapm, preload);
989 	else
990 		snd_soc_dapm_disable_pin(dapm, preload);
991 
992 	snd_soc_dapm_sync(dapm);
993 
994 	flush_work(&dsp->boot_work);
995 
996 	dsp->preloaded = ucontrol->value.integer.value[0];
997 
998 	if (dsp->toggle_preload) {
999 		snd_soc_dapm_disable_pin(dapm, preload);
1000 		snd_soc_dapm_sync(dapm);
1001 	}
1002 
1003 	return 1;
1004 }
1005 EXPORT_SYMBOL_GPL(wm_adsp2_preloader_put);
1006 
1007 int wm_adsp_power_up(struct wm_adsp *dsp, bool load_firmware)
1008 {
1009 	struct wm_adsp_fw_files fw = { 0 };
1010 	int ret = 0;
1011 
1012 	if (load_firmware) {
1013 		ret = wm_adsp_request_firmware_files(dsp, &fw);
1014 		if (ret)
1015 			return ret;
1016 	}
1017 
1018 	if (dsp->bin_mandatory && !fw.coeff.firmware) {
1019 		ret = -ENOENT;
1020 		goto err;
1021 	}
1022 
1023 	ret = cs_dsp_power_up(&dsp->cs_dsp,
1024 			      fw.wmfw.firmware, fw.wmfw.filename,
1025 			      fw.coeff.firmware, fw.coeff.filename,
1026 			      wm_adsp_fw_text[dsp->fw]);
1027 
1028 err:
1029 	wm_adsp_release_firmware_files(&fw);
1030 
1031 	return ret;
1032 }
1033 EXPORT_SYMBOL_GPL(wm_adsp_power_up);
1034 
1035 void wm_adsp_power_down(struct wm_adsp *dsp)
1036 {
1037 	cs_dsp_power_down(&dsp->cs_dsp);
1038 }
1039 EXPORT_SYMBOL_GPL(wm_adsp_power_down);
1040 
1041 static void wm_adsp_boot_work(struct work_struct *work)
1042 {
1043 	struct wm_adsp *dsp = container_of(work,
1044 					   struct wm_adsp,
1045 					   boot_work);
1046 
1047 	wm_adsp_power_up(dsp, true);
1048 }
1049 
1050 int wm_adsp_early_event(struct snd_soc_dapm_widget *w,
1051 			struct snd_kcontrol *kcontrol, int event)
1052 {
1053 	struct snd_soc_component *component = snd_soc_dapm_to_component(w->dapm);
1054 	struct wm_adsp *dsps = snd_soc_component_get_drvdata(component);
1055 	struct wm_adsp *dsp = &dsps[w->shift];
1056 
1057 	switch (event) {
1058 	case SND_SOC_DAPM_PRE_PMU:
1059 		queue_work(system_dfl_wq, &dsp->boot_work);
1060 		break;
1061 	case SND_SOC_DAPM_PRE_PMD:
1062 		wm_adsp_power_down(dsp);
1063 		break;
1064 	default:
1065 		break;
1066 	}
1067 
1068 	return 0;
1069 }
1070 EXPORT_SYMBOL_GPL(wm_adsp_early_event);
1071 
1072 static int wm_adsp_pre_run(struct cs_dsp *cs_dsp)
1073 {
1074 	struct wm_adsp *dsp = container_of(cs_dsp, struct wm_adsp, cs_dsp);
1075 
1076 	if (!dsp->pre_run)
1077 		return 0;
1078 
1079 	return (*dsp->pre_run)(dsp);
1080 }
1081 
1082 static int wm_adsp_event_post_run(struct cs_dsp *cs_dsp)
1083 {
1084 	struct wm_adsp *dsp = container_of(cs_dsp, struct wm_adsp, cs_dsp);
1085 
1086 	if (wm_adsp_fw[dsp->fw].num_caps != 0)
1087 		return wm_adsp_buffer_init(dsp);
1088 
1089 	return 0;
1090 }
1091 
1092 static void wm_adsp_event_post_stop(struct cs_dsp *cs_dsp)
1093 {
1094 	struct wm_adsp *dsp = container_of(cs_dsp, struct wm_adsp, cs_dsp);
1095 
1096 	if (wm_adsp_fw[dsp->fw].num_caps != 0)
1097 		wm_adsp_buffer_free(dsp);
1098 
1099 	dsp->fatal_error = false;
1100 }
1101 
1102 int wm_adsp_run(struct wm_adsp *dsp)
1103 {
1104 	flush_work(&dsp->boot_work);
1105 
1106 	return cs_dsp_run(&dsp->cs_dsp);
1107 }
1108 EXPORT_SYMBOL_GPL(wm_adsp_run);
1109 
1110 void wm_adsp_stop(struct wm_adsp *dsp)
1111 {
1112 	cs_dsp_stop(&dsp->cs_dsp);
1113 }
1114 EXPORT_SYMBOL_GPL(wm_adsp_stop);
1115 
1116 void wm_adsp_hibernate(struct wm_adsp *dsp, bool hibernate)
1117 {
1118 	cs_dsp_hibernate(&dsp->cs_dsp, hibernate);
1119 }
1120 EXPORT_SYMBOL_GPL(wm_adsp_hibernate);
1121 
1122 int wm_adsp_event(struct snd_soc_dapm_widget *w,
1123 		  struct snd_kcontrol *kcontrol, int event)
1124 {
1125 	struct snd_soc_component *component = snd_soc_dapm_to_component(w->dapm);
1126 	struct wm_adsp *dsps = snd_soc_component_get_drvdata(component);
1127 	struct wm_adsp *dsp = &dsps[w->shift];
1128 
1129 	switch (event) {
1130 	case SND_SOC_DAPM_POST_PMU:
1131 		return wm_adsp_run(dsp);
1132 	case SND_SOC_DAPM_PRE_PMD:
1133 		wm_adsp_stop(dsp);
1134 		return 0;
1135 	default:
1136 		return 0;
1137 	}
1138 }
1139 EXPORT_SYMBOL_GPL(wm_adsp_event);
1140 
1141 int wm_adsp2_component_probe(struct wm_adsp *dsp, struct snd_soc_component *component)
1142 {
1143 	struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);
1144 	char preload[32];
1145 
1146 	if (!dsp->cs_dsp.no_core_startstop) {
1147 		snprintf(preload, ARRAY_SIZE(preload), "%s Preload", dsp->cs_dsp.name);
1148 		snd_soc_dapm_disable_pin(dapm, preload);
1149 	}
1150 
1151 	cs_dsp_init_debugfs(&dsp->cs_dsp, component->debugfs_root);
1152 
1153 	dsp->component = component;
1154 
1155 	return 0;
1156 }
1157 EXPORT_SYMBOL_GPL(wm_adsp2_component_probe);
1158 
1159 int wm_adsp2_component_remove(struct wm_adsp *dsp, struct snd_soc_component *component)
1160 {
1161 	if (!dsp)
1162 		return 0;
1163 
1164 	if (!dsp->component)
1165 		return 0;
1166 
1167 	cs_dsp_cleanup_debugfs(&dsp->cs_dsp);
1168 	dsp->component = NULL;
1169 
1170 	return 0;
1171 }
1172 EXPORT_SYMBOL_GPL(wm_adsp2_component_remove);
1173 
1174 int wm_adsp2_init(struct wm_adsp *dsp)
1175 {
1176 	int ret;
1177 
1178 	INIT_WORK(&dsp->boot_work, wm_adsp_boot_work);
1179 
1180 	dsp->sys_config_size = sizeof(struct wm_adsp_system_config_xm_hdr);
1181 	dsp->cs_dsp.client_ops = &wm_adsp2_client_ops;
1182 
1183 	ret = cs_dsp_adsp2_init(&dsp->cs_dsp);
1184 	if (ret)
1185 		return ret;
1186 
1187 	return wm_adsp_common_init(dsp);
1188 }
1189 EXPORT_SYMBOL_GPL(wm_adsp2_init);
1190 
1191 int wm_halo_init(struct wm_adsp *dsp)
1192 {
1193 	int ret;
1194 
1195 	INIT_WORK(&dsp->boot_work, wm_adsp_boot_work);
1196 
1197 	dsp->sys_config_size = sizeof(struct wm_halo_system_config_xm_hdr);
1198 	dsp->cs_dsp.client_ops = &wm_adsp2_client_ops;
1199 
1200 	ret = cs_dsp_halo_init(&dsp->cs_dsp);
1201 	if (ret)
1202 		return ret;
1203 
1204 	return wm_adsp_common_init(dsp);
1205 }
1206 EXPORT_SYMBOL_GPL(wm_halo_init);
1207 
1208 void wm_adsp2_remove(struct wm_adsp *dsp)
1209 {
1210 	cs_dsp_remove(&dsp->cs_dsp);
1211 }
1212 EXPORT_SYMBOL_GPL(wm_adsp2_remove);
1213 
1214 static inline int wm_adsp_compr_attached(struct wm_adsp_compr *compr)
1215 {
1216 	return compr->buf != NULL;
1217 }
1218 
1219 static int wm_adsp_compr_attach(struct wm_adsp_compr *compr)
1220 {
1221 	struct wm_adsp_compr_buf *buf = NULL, *tmp;
1222 
1223 	if (compr->dsp->fatal_error)
1224 		return -EINVAL;
1225 
1226 	list_for_each_entry(tmp, &compr->dsp->buffer_list, list) {
1227 		if (!tmp->name || !strcmp(compr->name, tmp->name)) {
1228 			buf = tmp;
1229 			break;
1230 		}
1231 	}
1232 
1233 	if (!buf)
1234 		return -EINVAL;
1235 
1236 	compr->buf = buf;
1237 	buf->compr = compr;
1238 
1239 	return 0;
1240 }
1241 
1242 static void wm_adsp_compr_detach(struct wm_adsp_compr *compr)
1243 {
1244 	if (!compr)
1245 		return;
1246 
1247 	/* Wake the poll so it can see buffer is no longer attached */
1248 	if (compr->stream)
1249 		snd_compr_fragment_elapsed(compr->stream);
1250 
1251 	if (wm_adsp_compr_attached(compr)) {
1252 		compr->buf->compr = NULL;
1253 		compr->buf = NULL;
1254 	}
1255 }
1256 
1257 int wm_adsp_compr_open(struct wm_adsp *dsp, struct snd_compr_stream *stream)
1258 {
1259 	struct wm_adsp_compr *compr, *tmp;
1260 	struct snd_soc_pcm_runtime *rtd = stream->private_data;
1261 
1262 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
1263 
1264 	if (wm_adsp_fw[dsp->fw].num_caps == 0) {
1265 		adsp_err(dsp, "%s: Firmware does not support compressed API\n",
1266 			 snd_soc_rtd_to_codec(rtd, 0)->name);
1267 		return -ENXIO;
1268 	}
1269 
1270 	if (wm_adsp_fw[dsp->fw].compr_direction != stream->direction) {
1271 		adsp_err(dsp, "%s: Firmware does not support stream direction\n",
1272 			 snd_soc_rtd_to_codec(rtd, 0)->name);
1273 		return -EINVAL;
1274 	}
1275 
1276 	list_for_each_entry(tmp, &dsp->compr_list, list) {
1277 		if (!strcmp(tmp->name, snd_soc_rtd_to_codec(rtd, 0)->name)) {
1278 			adsp_err(dsp, "%s: Only a single stream supported per dai\n",
1279 				 snd_soc_rtd_to_codec(rtd, 0)->name);
1280 			return -EBUSY;
1281 		}
1282 	}
1283 
1284 	compr = kzalloc_obj(*compr);
1285 	if (!compr)
1286 		return -ENOMEM;
1287 
1288 	compr->dsp = dsp;
1289 	compr->stream = stream;
1290 	compr->name = snd_soc_rtd_to_codec(rtd, 0)->name;
1291 
1292 	list_add_tail(&compr->list, &dsp->compr_list);
1293 
1294 	stream->runtime->private_data = compr;
1295 
1296 	return 0;
1297 }
1298 EXPORT_SYMBOL_GPL(wm_adsp_compr_open);
1299 
1300 int wm_adsp_compr_free(struct snd_soc_component *component,
1301 		       struct snd_compr_stream *stream)
1302 {
1303 	struct wm_adsp_compr *compr = stream->runtime->private_data;
1304 	struct wm_adsp *dsp = compr->dsp;
1305 
1306 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
1307 
1308 	wm_adsp_compr_detach(compr);
1309 	list_del(&compr->list);
1310 
1311 	kfree(compr->raw_buf);
1312 	kfree(compr);
1313 
1314 	return 0;
1315 }
1316 EXPORT_SYMBOL_GPL(wm_adsp_compr_free);
1317 
1318 static int wm_adsp_compr_check_params(struct snd_compr_stream *stream,
1319 				      struct snd_compr_params *params)
1320 {
1321 	struct wm_adsp_compr *compr = stream->runtime->private_data;
1322 	struct wm_adsp *dsp = compr->dsp;
1323 	const struct wm_adsp_fw_caps *caps;
1324 	const struct snd_codec_desc *desc;
1325 	int i, j;
1326 
1327 	if (params->buffer.fragment_size < WM_ADSP_MIN_FRAGMENT_SIZE ||
1328 	    params->buffer.fragment_size > WM_ADSP_MAX_FRAGMENT_SIZE ||
1329 	    params->buffer.fragments < WM_ADSP_MIN_FRAGMENTS ||
1330 	    params->buffer.fragments > WM_ADSP_MAX_FRAGMENTS ||
1331 	    params->buffer.fragment_size % CS_DSP_DATA_WORD_SIZE) {
1332 		compr_err(compr, "Invalid buffer fragsize=%d fragments=%d\n",
1333 			  params->buffer.fragment_size,
1334 			  params->buffer.fragments);
1335 
1336 		return -EINVAL;
1337 	}
1338 
1339 	for (i = 0; i < wm_adsp_fw[dsp->fw].num_caps; i++) {
1340 		caps = &wm_adsp_fw[dsp->fw].caps[i];
1341 		desc = &caps->desc;
1342 
1343 		if (caps->id != params->codec.id)
1344 			continue;
1345 
1346 		if (stream->direction == SND_COMPRESS_PLAYBACK) {
1347 			if (desc->max_ch < params->codec.ch_out)
1348 				continue;
1349 		} else {
1350 			if (desc->max_ch < params->codec.ch_in)
1351 				continue;
1352 		}
1353 
1354 		if (!(desc->formats & (1 << params->codec.format)))
1355 			continue;
1356 
1357 		for (j = 0; j < desc->num_sample_rates; ++j)
1358 			if (desc->sample_rates[j] == params->codec.sample_rate)
1359 				return 0;
1360 	}
1361 
1362 	compr_err(compr, "Invalid params id=%u ch=%u,%u rate=%u fmt=%u\n",
1363 		  params->codec.id, params->codec.ch_in, params->codec.ch_out,
1364 		  params->codec.sample_rate, params->codec.format);
1365 	return -EINVAL;
1366 }
1367 
1368 static inline unsigned int wm_adsp_compr_frag_words(struct wm_adsp_compr *compr)
1369 {
1370 	return compr->size.fragment_size / CS_DSP_DATA_WORD_SIZE;
1371 }
1372 
1373 int wm_adsp_compr_set_params(struct snd_soc_component *component,
1374 			     struct snd_compr_stream *stream,
1375 			     struct snd_compr_params *params)
1376 {
1377 	struct wm_adsp_compr *compr = stream->runtime->private_data;
1378 	unsigned int size;
1379 	int ret;
1380 
1381 	ret = wm_adsp_compr_check_params(stream, params);
1382 	if (ret)
1383 		return ret;
1384 
1385 	compr->size = params->buffer;
1386 
1387 	compr_dbg(compr, "fragment_size=%d fragments=%d\n",
1388 		  compr->size.fragment_size, compr->size.fragments);
1389 
1390 	size = wm_adsp_compr_frag_words(compr) * sizeof(*compr->raw_buf);
1391 	compr->raw_buf = kmalloc(size, GFP_DMA | GFP_KERNEL);
1392 	if (!compr->raw_buf)
1393 		return -ENOMEM;
1394 
1395 	compr->sample_rate = params->codec.sample_rate;
1396 
1397 	return 0;
1398 }
1399 EXPORT_SYMBOL_GPL(wm_adsp_compr_set_params);
1400 
1401 int wm_adsp_compr_get_caps(struct snd_soc_component *component,
1402 			   struct snd_compr_stream *stream,
1403 			   struct snd_compr_caps *caps)
1404 {
1405 	struct wm_adsp_compr *compr = stream->runtime->private_data;
1406 	int fw = compr->dsp->fw;
1407 	int i;
1408 
1409 	if (wm_adsp_fw[fw].caps) {
1410 		for (i = 0; i < wm_adsp_fw[fw].num_caps; i++)
1411 			caps->codecs[i] = wm_adsp_fw[fw].caps[i].id;
1412 
1413 		caps->num_codecs = i;
1414 		caps->direction = wm_adsp_fw[fw].compr_direction;
1415 
1416 		caps->min_fragment_size = WM_ADSP_MIN_FRAGMENT_SIZE;
1417 		caps->max_fragment_size = WM_ADSP_MAX_FRAGMENT_SIZE;
1418 		caps->min_fragments = WM_ADSP_MIN_FRAGMENTS;
1419 		caps->max_fragments = WM_ADSP_MAX_FRAGMENTS;
1420 	}
1421 
1422 	return 0;
1423 }
1424 EXPORT_SYMBOL_GPL(wm_adsp_compr_get_caps);
1425 
1426 static inline int wm_adsp_buffer_read(struct wm_adsp_compr_buf *buf,
1427 				      unsigned int field_offset, u32 *data)
1428 {
1429 	return cs_dsp_read_data_word(&buf->dsp->cs_dsp, buf->host_buf_mem_type,
1430 				     buf->host_buf_ptr + field_offset, data);
1431 }
1432 
1433 static inline int wm_adsp_buffer_write(struct wm_adsp_compr_buf *buf,
1434 				       unsigned int field_offset, u32 data)
1435 {
1436 	return cs_dsp_write_data_word(&buf->dsp->cs_dsp, buf->host_buf_mem_type,
1437 				      buf->host_buf_ptr + field_offset,
1438 				      data);
1439 }
1440 
1441 static int wm_adsp_buffer_populate(struct wm_adsp_compr_buf *buf)
1442 {
1443 	const struct wm_adsp_fw_caps *caps = wm_adsp_fw[buf->dsp->fw].caps;
1444 	struct wm_adsp_buffer_region *region;
1445 	u32 offset = 0;
1446 	int i, ret;
1447 
1448 	buf->regions = kzalloc_objs(*buf->regions, caps->num_regions);
1449 	if (!buf->regions)
1450 		return -ENOMEM;
1451 
1452 	for (i = 0; i < caps->num_regions; ++i) {
1453 		region = &buf->regions[i];
1454 
1455 		region->offset = offset;
1456 		region->mem_type = caps->region_defs[i].mem_type;
1457 
1458 		ret = wm_adsp_buffer_read(buf, caps->region_defs[i].base_offset,
1459 					  &region->base_addr);
1460 		if (ret < 0)
1461 			goto err;
1462 
1463 		ret = wm_adsp_buffer_read(buf, caps->region_defs[i].size_offset,
1464 					  &offset);
1465 		if (ret < 0)
1466 			goto err;
1467 
1468 		region->cumulative_size = offset;
1469 
1470 		compr_dbg(buf,
1471 			  "region=%d type=%d base=%08x off=%08x size=%08x\n",
1472 			  i, region->mem_type, region->base_addr,
1473 			  region->offset, region->cumulative_size);
1474 	}
1475 
1476 	return 0;
1477 
1478 err:
1479 	kfree(buf->regions);
1480 	return ret;
1481 }
1482 
1483 static void wm_adsp_buffer_clear(struct wm_adsp_compr_buf *buf)
1484 {
1485 	buf->irq_count = 0xFFFFFFFF;
1486 	buf->read_index = -1;
1487 	buf->avail = 0;
1488 }
1489 
1490 static struct wm_adsp_compr_buf *wm_adsp_buffer_alloc(struct wm_adsp *dsp)
1491 {
1492 	struct wm_adsp_compr_buf *buf;
1493 
1494 	buf = kzalloc_obj(*buf);
1495 	if (!buf)
1496 		return NULL;
1497 
1498 	buf->dsp = dsp;
1499 
1500 	wm_adsp_buffer_clear(buf);
1501 
1502 	return buf;
1503 }
1504 
1505 static int wm_adsp_buffer_parse_legacy(struct wm_adsp *dsp)
1506 {
1507 	struct cs_dsp_alg_region *alg_region;
1508 	struct wm_adsp_compr_buf *buf;
1509 	u32 xmalg, addr, magic;
1510 	int i, ret;
1511 
1512 	alg_region = cs_dsp_find_alg_region(&dsp->cs_dsp, WMFW_ADSP2_XM, dsp->cs_dsp.fw_id);
1513 	if (!alg_region) {
1514 		adsp_err(dsp, "No algorithm region found\n");
1515 		return -EINVAL;
1516 	}
1517 
1518 	xmalg = dsp->sys_config_size / sizeof(__be32);
1519 
1520 	addr = alg_region->base + xmalg + ALG_XM_FIELD(magic);
1521 	ret = cs_dsp_read_data_word(&dsp->cs_dsp, WMFW_ADSP2_XM, addr, &magic);
1522 	if (ret < 0)
1523 		return ret;
1524 
1525 	if (magic != WM_ADSP_ALG_XM_STRUCT_MAGIC)
1526 		return -ENODEV;
1527 
1528 	buf = wm_adsp_buffer_alloc(dsp);
1529 	if (!buf)
1530 		return -ENOMEM;
1531 
1532 	addr = alg_region->base + xmalg + ALG_XM_FIELD(host_buf_ptr);
1533 	for (i = 0; i < 5; ++i) {
1534 		ret = cs_dsp_read_data_word(&dsp->cs_dsp, WMFW_ADSP2_XM, addr,
1535 					    &buf->host_buf_ptr);
1536 		if (ret < 0)
1537 			goto err;
1538 
1539 		if (buf->host_buf_ptr)
1540 			break;
1541 
1542 		usleep_range(1000, 2000);
1543 	}
1544 
1545 	if (!buf->host_buf_ptr) {
1546 		ret = -EIO;
1547 		goto err;
1548 	}
1549 
1550 	buf->host_buf_mem_type = WMFW_ADSP2_XM;
1551 
1552 	ret = wm_adsp_buffer_populate(buf);
1553 	if (ret < 0)
1554 		goto err;
1555 
1556 	list_add_tail(&buf->list, &dsp->buffer_list);
1557 
1558 	compr_dbg(buf, "legacy host_buf_ptr=%x\n", buf->host_buf_ptr);
1559 
1560 	return 0;
1561 
1562 err:
1563 	kfree(buf);
1564 
1565 	return ret;
1566 }
1567 
1568 static int wm_adsp_buffer_parse_coeff(struct cs_dsp_coeff_ctl *cs_ctl)
1569 {
1570 	struct wm_adsp_host_buf_coeff_v1 coeff_v1;
1571 	struct wm_adsp_compr_buf *buf;
1572 	struct wm_adsp *dsp = container_of(cs_ctl->dsp, struct wm_adsp, cs_dsp);
1573 	unsigned int version = 0;
1574 	int ret, i;
1575 
1576 	for (i = 0; i < 5; ++i) {
1577 		ret = cs_dsp_coeff_read_ctrl(cs_ctl, 0, &coeff_v1,
1578 					     min((size_t)cs_ctl->len, sizeof(coeff_v1)));
1579 		if (ret < 0)
1580 			return ret;
1581 
1582 		if (coeff_v1.host_buf_ptr)
1583 			break;
1584 
1585 		usleep_range(1000, 2000);
1586 	}
1587 
1588 	if (!coeff_v1.host_buf_ptr) {
1589 		adsp_err(dsp, "Failed to acquire host buffer\n");
1590 		return -EIO;
1591 	}
1592 
1593 	buf = wm_adsp_buffer_alloc(dsp);
1594 	if (!buf)
1595 		return -ENOMEM;
1596 
1597 	buf->host_buf_mem_type = cs_ctl->alg_region.type;
1598 	buf->host_buf_ptr = be32_to_cpu(coeff_v1.host_buf_ptr);
1599 
1600 	ret = wm_adsp_buffer_populate(buf);
1601 	if (ret < 0)
1602 		goto err;
1603 
1604 	/*
1605 	 * v0 host_buffer coefficients didn't have versioning, so if the
1606 	 * control is one word, assume version 0.
1607 	 */
1608 	if (cs_ctl->len == 4)
1609 		goto done;
1610 
1611 	version = be32_to_cpu(coeff_v1.versions) & HOST_BUF_COEFF_COMPAT_VER_MASK;
1612 	version >>= HOST_BUF_COEFF_COMPAT_VER_SHIFT;
1613 
1614 	if (version > HOST_BUF_COEFF_SUPPORTED_COMPAT_VER) {
1615 		adsp_err(dsp,
1616 			 "Host buffer coeff ver %u > supported version %u\n",
1617 			 version, HOST_BUF_COEFF_SUPPORTED_COMPAT_VER);
1618 		ret = -EINVAL;
1619 		goto err;
1620 	}
1621 
1622 	cs_dsp_remove_padding((u32 *)&coeff_v1.name, ARRAY_SIZE(coeff_v1.name));
1623 
1624 	buf->name = kasprintf(GFP_KERNEL, "%s-dsp-%s", dsp->part,
1625 			      (char *)&coeff_v1.name);
1626 
1627 done:
1628 	list_add_tail(&buf->list, &dsp->buffer_list);
1629 
1630 	compr_dbg(buf, "host_buf_ptr=%x coeff version %u\n",
1631 		  buf->host_buf_ptr, version);
1632 
1633 	return version;
1634 
1635 err:
1636 	kfree(buf);
1637 
1638 	return ret;
1639 }
1640 
1641 static int wm_adsp_buffer_init(struct wm_adsp *dsp)
1642 {
1643 	struct cs_dsp_coeff_ctl *cs_ctl;
1644 	int ret;
1645 
1646 	list_for_each_entry(cs_ctl, &dsp->cs_dsp.ctl_list, list) {
1647 		if (cs_ctl->type != WMFW_CTL_TYPE_HOST_BUFFER)
1648 			continue;
1649 
1650 		if (!cs_ctl->enabled)
1651 			continue;
1652 
1653 		ret = wm_adsp_buffer_parse_coeff(cs_ctl);
1654 		if (ret < 0) {
1655 			adsp_err(dsp, "Failed to parse coeff: %d\n", ret);
1656 			goto error;
1657 		} else if (ret == 0) {
1658 			/* Only one buffer supported for version 0 */
1659 			return 0;
1660 		}
1661 	}
1662 
1663 	if (list_empty(&dsp->buffer_list)) {
1664 		/* Fall back to legacy support */
1665 		ret = wm_adsp_buffer_parse_legacy(dsp);
1666 		if (ret == -ENODEV)
1667 			adsp_info(dsp, "Legacy support not available\n");
1668 		else if (ret)
1669 			adsp_warn(dsp, "Failed to parse legacy: %d\n", ret);
1670 	}
1671 
1672 	return 0;
1673 
1674 error:
1675 	wm_adsp_buffer_free(dsp);
1676 	return ret;
1677 }
1678 
1679 static int wm_adsp_buffer_free(struct wm_adsp *dsp)
1680 {
1681 	struct wm_adsp_compr_buf *buf, *tmp;
1682 
1683 	list_for_each_entry_safe(buf, tmp, &dsp->buffer_list, list) {
1684 		wm_adsp_compr_detach(buf->compr);
1685 
1686 		kfree(buf->name);
1687 		kfree(buf->regions);
1688 		list_del(&buf->list);
1689 		kfree(buf);
1690 	}
1691 
1692 	return 0;
1693 }
1694 
1695 static int wm_adsp_buffer_get_error(struct wm_adsp_compr_buf *buf)
1696 {
1697 	int ret;
1698 
1699 	ret = wm_adsp_buffer_read(buf, HOST_BUFFER_FIELD(error), &buf->error);
1700 	if (ret < 0) {
1701 		compr_err(buf, "Failed to check buffer error: %d\n", ret);
1702 		return ret;
1703 	}
1704 	if (buf->error != 0) {
1705 		compr_err(buf, "Buffer error occurred: %d\n", buf->error);
1706 		return -EIO;
1707 	}
1708 
1709 	return 0;
1710 }
1711 
1712 int wm_adsp_compr_trigger(struct snd_soc_component *component,
1713 			  struct snd_compr_stream *stream, int cmd)
1714 {
1715 	struct wm_adsp_compr *compr = stream->runtime->private_data;
1716 	struct wm_adsp *dsp = compr->dsp;
1717 	int ret = 0;
1718 
1719 	compr_dbg(compr, "Trigger: %d\n", cmd);
1720 
1721 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
1722 
1723 	switch (cmd) {
1724 	case SNDRV_PCM_TRIGGER_START:
1725 		if (!wm_adsp_compr_attached(compr)) {
1726 			ret = wm_adsp_compr_attach(compr);
1727 			if (ret < 0) {
1728 				compr_err(compr, "Failed to link buffer and stream: %d\n",
1729 					  ret);
1730 				break;
1731 			}
1732 		}
1733 
1734 		ret = wm_adsp_buffer_get_error(compr->buf);
1735 		if (ret < 0)
1736 			break;
1737 
1738 		/* Trigger the IRQ at one fragment of data */
1739 		ret = wm_adsp_buffer_write(compr->buf,
1740 					   HOST_BUFFER_FIELD(high_water_mark),
1741 					   wm_adsp_compr_frag_words(compr));
1742 		if (ret < 0) {
1743 			compr_err(compr, "Failed to set high water mark: %d\n",
1744 				  ret);
1745 			break;
1746 		}
1747 		break;
1748 	case SNDRV_PCM_TRIGGER_STOP:
1749 		if (wm_adsp_compr_attached(compr))
1750 			wm_adsp_buffer_clear(compr->buf);
1751 		break;
1752 	default:
1753 		ret = -EINVAL;
1754 		break;
1755 	}
1756 
1757 	return ret;
1758 }
1759 EXPORT_SYMBOL_GPL(wm_adsp_compr_trigger);
1760 
1761 static inline int wm_adsp_buffer_size(struct wm_adsp_compr_buf *buf)
1762 {
1763 	int last_region = wm_adsp_fw[buf->dsp->fw].caps->num_regions - 1;
1764 
1765 	return buf->regions[last_region].cumulative_size;
1766 }
1767 
1768 static int wm_adsp_buffer_update_avail(struct wm_adsp_compr_buf *buf)
1769 {
1770 	u32 next_read_index, next_write_index;
1771 	int write_index, read_index, avail;
1772 	int ret;
1773 
1774 	/* Only sync read index if we haven't already read a valid index */
1775 	if (buf->read_index < 0) {
1776 		ret = wm_adsp_buffer_read(buf,
1777 				HOST_BUFFER_FIELD(next_read_index),
1778 				&next_read_index);
1779 		if (ret < 0)
1780 			return ret;
1781 
1782 		read_index = sign_extend32(next_read_index, 23);
1783 
1784 		if (read_index < 0) {
1785 			compr_dbg(buf, "Avail check on unstarted stream\n");
1786 			return 0;
1787 		}
1788 
1789 		buf->read_index = read_index;
1790 	}
1791 
1792 	ret = wm_adsp_buffer_read(buf, HOST_BUFFER_FIELD(next_write_index),
1793 			&next_write_index);
1794 	if (ret < 0)
1795 		return ret;
1796 
1797 	write_index = sign_extend32(next_write_index, 23);
1798 
1799 	avail = write_index - buf->read_index;
1800 	if (avail < 0)
1801 		avail += wm_adsp_buffer_size(buf);
1802 
1803 	compr_dbg(buf, "readindex=0x%x, writeindex=0x%x, avail=%d\n",
1804 		  buf->read_index, write_index, avail * CS_DSP_DATA_WORD_SIZE);
1805 
1806 	buf->avail = avail;
1807 
1808 	return 0;
1809 }
1810 
1811 int wm_adsp_compr_handle_irq(struct wm_adsp *dsp)
1812 {
1813 	struct wm_adsp_compr_buf *buf;
1814 	struct wm_adsp_compr *compr;
1815 	int ret = 0;
1816 
1817 	mutex_lock(&dsp->cs_dsp.pwr_lock);
1818 
1819 	if (list_empty(&dsp->buffer_list)) {
1820 		ret = -ENODEV;
1821 		goto out;
1822 	}
1823 
1824 	adsp_dbg(dsp, "Handling buffer IRQ\n");
1825 
1826 	list_for_each_entry(buf, &dsp->buffer_list, list) {
1827 		compr = buf->compr;
1828 
1829 		ret = wm_adsp_buffer_get_error(buf);
1830 		if (ret < 0)
1831 			goto out_notify; /* Wake poll to report error */
1832 
1833 		ret = wm_adsp_buffer_read(buf, HOST_BUFFER_FIELD(irq_count),
1834 					  &buf->irq_count);
1835 		if (ret < 0) {
1836 			compr_err(buf, "Failed to get irq_count: %d\n", ret);
1837 			goto out;
1838 		}
1839 
1840 		ret = wm_adsp_buffer_update_avail(buf);
1841 		if (ret < 0) {
1842 			compr_err(buf, "Error reading avail: %d\n", ret);
1843 			goto out;
1844 		}
1845 
1846 		if (wm_adsp_fw[dsp->fw].voice_trigger && buf->irq_count == 2)
1847 			ret = WM_ADSP_COMPR_VOICE_TRIGGER;
1848 
1849 out_notify:
1850 		if (compr && compr->stream)
1851 			snd_compr_fragment_elapsed(compr->stream);
1852 	}
1853 
1854 out:
1855 	mutex_unlock(&dsp->cs_dsp.pwr_lock);
1856 
1857 	return ret;
1858 }
1859 EXPORT_SYMBOL_GPL(wm_adsp_compr_handle_irq);
1860 
1861 static int wm_adsp_buffer_reenable_irq(struct wm_adsp_compr_buf *buf)
1862 {
1863 	if (buf->irq_count & 0x01)
1864 		return 0;
1865 
1866 	compr_dbg(buf, "Enable IRQ(0x%x) for next fragment\n", buf->irq_count);
1867 
1868 	buf->irq_count |= 0x01;
1869 
1870 	return wm_adsp_buffer_write(buf, HOST_BUFFER_FIELD(irq_ack),
1871 				    buf->irq_count);
1872 }
1873 
1874 int wm_adsp_compr_pointer(struct snd_soc_component *component,
1875 			  struct snd_compr_stream *stream,
1876 			  struct snd_compr_tstamp64 *tstamp)
1877 {
1878 	struct wm_adsp_compr *compr = stream->runtime->private_data;
1879 	struct wm_adsp *dsp = compr->dsp;
1880 	struct wm_adsp_compr_buf *buf;
1881 	int ret = 0;
1882 
1883 	compr_dbg(compr, "Pointer request\n");
1884 
1885 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
1886 
1887 	buf = compr->buf;
1888 
1889 	if (dsp->fatal_error || !buf || buf->error) {
1890 		snd_compr_stop_error(stream, SNDRV_PCM_STATE_XRUN);
1891 		return -EIO;
1892 	}
1893 
1894 	if (buf->avail < wm_adsp_compr_frag_words(compr)) {
1895 		ret = wm_adsp_buffer_update_avail(buf);
1896 		if (ret < 0) {
1897 			compr_err(compr, "Error reading avail: %d\n", ret);
1898 			return ret;
1899 		}
1900 
1901 		/*
1902 		 * If we really have less than 1 fragment available tell the
1903 		 * DSP to inform us once a whole fragment is available.
1904 		 */
1905 		if (buf->avail < wm_adsp_compr_frag_words(compr)) {
1906 			ret = wm_adsp_buffer_get_error(buf);
1907 			if (ret < 0) {
1908 				if (buf->error)
1909 					snd_compr_stop_error(stream,
1910 							SNDRV_PCM_STATE_XRUN);
1911 				return ret;
1912 			}
1913 
1914 			ret = wm_adsp_buffer_reenable_irq(buf);
1915 			if (ret < 0) {
1916 				compr_err(compr, "Failed to re-enable buffer IRQ: %d\n",
1917 					  ret);
1918 				return ret;
1919 			}
1920 		}
1921 	}
1922 
1923 	tstamp->copied_total = compr->copied_total;
1924 	tstamp->copied_total += buf->avail * CS_DSP_DATA_WORD_SIZE;
1925 	tstamp->sampling_rate = compr->sample_rate;
1926 
1927 	return ret;
1928 }
1929 EXPORT_SYMBOL_GPL(wm_adsp_compr_pointer);
1930 
1931 static int wm_adsp_buffer_capture_block(struct wm_adsp_compr *compr, int target)
1932 {
1933 	struct wm_adsp_compr_buf *buf = compr->buf;
1934 	unsigned int adsp_addr;
1935 	int mem_type, nwords, max_read;
1936 	int i, ret;
1937 
1938 	/* Calculate read parameters */
1939 	for (i = 0; i < wm_adsp_fw[buf->dsp->fw].caps->num_regions; ++i)
1940 		if (buf->read_index < buf->regions[i].cumulative_size)
1941 			break;
1942 
1943 	if (i == wm_adsp_fw[buf->dsp->fw].caps->num_regions)
1944 		return -EINVAL;
1945 
1946 	mem_type = buf->regions[i].mem_type;
1947 	adsp_addr = buf->regions[i].base_addr +
1948 		    (buf->read_index - buf->regions[i].offset);
1949 
1950 	max_read = wm_adsp_compr_frag_words(compr);
1951 	nwords = buf->regions[i].cumulative_size - buf->read_index;
1952 
1953 	if (nwords > target)
1954 		nwords = target;
1955 	if (nwords > buf->avail)
1956 		nwords = buf->avail;
1957 	if (nwords > max_read)
1958 		nwords = max_read;
1959 	if (!nwords)
1960 		return 0;
1961 
1962 	/* Read data from DSP */
1963 	ret = cs_dsp_read_raw_data_block(&buf->dsp->cs_dsp, mem_type, adsp_addr,
1964 					 nwords, (__be32 *)compr->raw_buf);
1965 	if (ret < 0)
1966 		return ret;
1967 
1968 	cs_dsp_remove_padding(compr->raw_buf, nwords);
1969 
1970 	/* update read index to account for words read */
1971 	buf->read_index += nwords;
1972 	if (buf->read_index == wm_adsp_buffer_size(buf))
1973 		buf->read_index = 0;
1974 
1975 	ret = wm_adsp_buffer_write(buf, HOST_BUFFER_FIELD(next_read_index),
1976 				   buf->read_index);
1977 	if (ret < 0)
1978 		return ret;
1979 
1980 	/* update avail to account for words read */
1981 	buf->avail -= nwords;
1982 
1983 	return nwords;
1984 }
1985 
1986 static int wm_adsp_compr_read(struct wm_adsp_compr *compr,
1987 			      char __user *buf, size_t count)
1988 {
1989 	struct wm_adsp *dsp = compr->dsp;
1990 	int ntotal = 0;
1991 	int nwords, nbytes;
1992 
1993 	compr_dbg(compr, "Requested read of %zu bytes\n", count);
1994 
1995 	if (dsp->fatal_error || !compr->buf || compr->buf->error) {
1996 		snd_compr_stop_error(compr->stream, SNDRV_PCM_STATE_XRUN);
1997 		return -EIO;
1998 	}
1999 
2000 	count /= CS_DSP_DATA_WORD_SIZE;
2001 
2002 	do {
2003 		nwords = wm_adsp_buffer_capture_block(compr, count);
2004 		if (nwords < 0) {
2005 			compr_err(compr, "Failed to capture block: %d\n",
2006 				  nwords);
2007 			return nwords;
2008 		}
2009 
2010 		nbytes = nwords * CS_DSP_DATA_WORD_SIZE;
2011 
2012 		compr_dbg(compr, "Read %d bytes\n", nbytes);
2013 
2014 		if (copy_to_user(buf + ntotal, compr->raw_buf, nbytes)) {
2015 			compr_err(compr, "Failed to copy data to user: %d, %d\n",
2016 				  ntotal, nbytes);
2017 			return -EFAULT;
2018 		}
2019 
2020 		count -= nwords;
2021 		ntotal += nbytes;
2022 	} while (nwords > 0 && count > 0);
2023 
2024 	compr->copied_total += ntotal;
2025 
2026 	return ntotal;
2027 }
2028 
2029 int wm_adsp_compr_copy(struct snd_soc_component *component,
2030 		       struct snd_compr_stream *stream, char __user *buf,
2031 		       size_t count)
2032 {
2033 	struct wm_adsp_compr *compr = stream->runtime->private_data;
2034 	struct wm_adsp *dsp = compr->dsp;
2035 	int ret;
2036 
2037 	guard(mutex)(&dsp->cs_dsp.pwr_lock);
2038 
2039 	if (stream->direction == SND_COMPRESS_CAPTURE)
2040 		ret = wm_adsp_compr_read(compr, buf, count);
2041 	else
2042 		ret = -ENOTSUPP;
2043 
2044 	return ret;
2045 }
2046 EXPORT_SYMBOL_GPL(wm_adsp_compr_copy);
2047 
2048 static void wm_adsp_fatal_error(struct cs_dsp *cs_dsp)
2049 {
2050 	struct wm_adsp *dsp = container_of(cs_dsp, struct wm_adsp, cs_dsp);
2051 	struct wm_adsp_compr *compr;
2052 
2053 	dsp->fatal_error = true;
2054 
2055 	list_for_each_entry(compr, &dsp->compr_list, list) {
2056 		if (compr->stream)
2057 			snd_compr_fragment_elapsed(compr->stream);
2058 	}
2059 }
2060 
2061 irqreturn_t wm_adsp2_bus_error(int irq, void *data)
2062 {
2063 	struct wm_adsp *dsp = (struct wm_adsp *)data;
2064 
2065 	cs_dsp_adsp2_bus_error(&dsp->cs_dsp);
2066 
2067 	return IRQ_HANDLED;
2068 }
2069 EXPORT_SYMBOL_GPL(wm_adsp2_bus_error);
2070 
2071 irqreturn_t wm_halo_bus_error(int irq, void *data)
2072 {
2073 	struct wm_adsp *dsp = (struct wm_adsp *)data;
2074 
2075 	cs_dsp_halo_bus_error(&dsp->cs_dsp);
2076 
2077 	return IRQ_HANDLED;
2078 }
2079 EXPORT_SYMBOL_GPL(wm_halo_bus_error);
2080 
2081 irqreturn_t wm_halo_wdt_expire(int irq, void *data)
2082 {
2083 	struct wm_adsp *dsp = data;
2084 
2085 	cs_dsp_halo_wdt_expire(&dsp->cs_dsp);
2086 
2087 	return IRQ_HANDLED;
2088 }
2089 EXPORT_SYMBOL_GPL(wm_halo_wdt_expire);
2090 
2091 static const struct cs_dsp_client_ops wm_adsp1_client_ops = {
2092 	.control_add = wm_adsp_control_add_cb,
2093 	.control_remove = wm_adsp_control_remove,
2094 };
2095 
2096 static const struct cs_dsp_client_ops wm_adsp2_client_ops = {
2097 	.control_add = wm_adsp_control_add_cb,
2098 	.control_remove = wm_adsp_control_remove,
2099 	.pre_run = wm_adsp_pre_run,
2100 	.post_run = wm_adsp_event_post_run,
2101 	.post_stop = wm_adsp_event_post_stop,
2102 	.watchdog_expired = wm_adsp_fatal_error,
2103 };
2104 
2105 MODULE_DESCRIPTION("Cirrus Logic ASoC DSP Support");
2106 MODULE_LICENSE("GPL v2");
2107 MODULE_IMPORT_NS("FW_CS_DSP");
2108