xref: /openbmc/linux/sound/usb/caiaq/audio.c (revision 3ce311af)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *   Copyright (c) 2006-2008 Daniel Mack, Karsten Wiese
4 */
5 
6 #include <linux/device.h>
7 #include <linux/spinlock.h>
8 #include <linux/slab.h>
9 #include <linux/init.h>
10 #include <linux/usb.h>
11 #include <sound/core.h>
12 #include <sound/pcm.h>
13 
14 #include "device.h"
15 #include "audio.h"
16 
17 #define N_URBS			32
18 #define CLOCK_DRIFT_TOLERANCE	5
19 #define FRAMES_PER_URB		8
20 #define BYTES_PER_FRAME		512
21 #define CHANNELS_PER_STREAM	2
22 #define BYTES_PER_SAMPLE	3
23 #define BYTES_PER_SAMPLE_USB	4
24 #define MAX_BUFFER_SIZE		(128*1024)
25 #define MAX_ENDPOINT_SIZE	512
26 
27 #define ENDPOINT_CAPTURE	2
28 #define ENDPOINT_PLAYBACK	6
29 
30 #define MAKE_CHECKBYTE(cdev,stream,i) \
31 	(stream << 1) | (~(i / (cdev->n_streams * BYTES_PER_SAMPLE_USB)) & 1)
32 
33 static struct snd_pcm_hardware snd_usb_caiaq_pcm_hardware = {
34 	.info 		= (SNDRV_PCM_INFO_MMAP | SNDRV_PCM_INFO_INTERLEAVED |
35 			   SNDRV_PCM_INFO_BLOCK_TRANSFER),
36 	.formats 	= SNDRV_PCM_FMTBIT_S24_3BE,
37 	.rates 		= (SNDRV_PCM_RATE_44100 | SNDRV_PCM_RATE_48000 |
38 			   SNDRV_PCM_RATE_96000),
39 	.rate_min	= 44100,
40 	.rate_max	= 0, /* will overwrite later */
41 	.channels_min	= CHANNELS_PER_STREAM,
42 	.channels_max	= CHANNELS_PER_STREAM,
43 	.buffer_bytes_max = MAX_BUFFER_SIZE,
44 	.period_bytes_min = 128,
45 	.period_bytes_max = MAX_BUFFER_SIZE,
46 	.periods_min	= 1,
47 	.periods_max	= 1024,
48 };
49 
50 static void
51 activate_substream(struct snd_usb_caiaqdev *cdev,
52 	           struct snd_pcm_substream *sub)
53 {
54 	spin_lock(&cdev->spinlock);
55 
56 	if (sub->stream == SNDRV_PCM_STREAM_PLAYBACK)
57 		cdev->sub_playback[sub->number] = sub;
58 	else
59 		cdev->sub_capture[sub->number] = sub;
60 
61 	spin_unlock(&cdev->spinlock);
62 }
63 
64 static void
65 deactivate_substream(struct snd_usb_caiaqdev *cdev,
66 		     struct snd_pcm_substream *sub)
67 {
68 	unsigned long flags;
69 	spin_lock_irqsave(&cdev->spinlock, flags);
70 
71 	if (sub->stream == SNDRV_PCM_STREAM_PLAYBACK)
72 		cdev->sub_playback[sub->number] = NULL;
73 	else
74 		cdev->sub_capture[sub->number] = NULL;
75 
76 	spin_unlock_irqrestore(&cdev->spinlock, flags);
77 }
78 
79 static int
80 all_substreams_zero(struct snd_pcm_substream **subs)
81 {
82 	int i;
83 	for (i = 0; i < MAX_STREAMS; i++)
84 		if (subs[i] != NULL)
85 			return 0;
86 	return 1;
87 }
88 
89 static int stream_start(struct snd_usb_caiaqdev *cdev)
90 {
91 	int i, ret;
92 	struct device *dev = caiaqdev_to_dev(cdev);
93 
94 	dev_dbg(dev, "%s(%p)\n", __func__, cdev);
95 
96 	if (cdev->streaming)
97 		return -EINVAL;
98 
99 	memset(cdev->sub_playback, 0, sizeof(cdev->sub_playback));
100 	memset(cdev->sub_capture, 0, sizeof(cdev->sub_capture));
101 	cdev->input_panic = 0;
102 	cdev->output_panic = 0;
103 	cdev->first_packet = 4;
104 	cdev->streaming = 1;
105 	cdev->warned = 0;
106 
107 	for (i = 0; i < N_URBS; i++) {
108 		ret = usb_submit_urb(cdev->data_urbs_in[i], GFP_ATOMIC);
109 		if (ret) {
110 			dev_err(dev, "unable to trigger read #%d! (ret %d)\n",
111 				i, ret);
112 			cdev->streaming = 0;
113 			return -EPIPE;
114 		}
115 	}
116 
117 	return 0;
118 }
119 
120 static void stream_stop(struct snd_usb_caiaqdev *cdev)
121 {
122 	int i;
123 	struct device *dev = caiaqdev_to_dev(cdev);
124 
125 	dev_dbg(dev, "%s(%p)\n", __func__, cdev);
126 	if (!cdev->streaming)
127 		return;
128 
129 	cdev->streaming = 0;
130 
131 	for (i = 0; i < N_URBS; i++) {
132 		usb_kill_urb(cdev->data_urbs_in[i]);
133 
134 		if (test_bit(i, &cdev->outurb_active_mask))
135 			usb_kill_urb(cdev->data_urbs_out[i]);
136 	}
137 
138 	cdev->outurb_active_mask = 0;
139 }
140 
141 static int snd_usb_caiaq_substream_open(struct snd_pcm_substream *substream)
142 {
143 	struct snd_usb_caiaqdev *cdev = snd_pcm_substream_chip(substream);
144 	struct device *dev = caiaqdev_to_dev(cdev);
145 
146 	dev_dbg(dev, "%s(%p)\n", __func__, substream);
147 	substream->runtime->hw = cdev->pcm_info;
148 	snd_pcm_limit_hw_rates(substream->runtime);
149 
150 	return 0;
151 }
152 
153 static int snd_usb_caiaq_substream_close(struct snd_pcm_substream *substream)
154 {
155 	struct snd_usb_caiaqdev *cdev = snd_pcm_substream_chip(substream);
156 	struct device *dev = caiaqdev_to_dev(cdev);
157 
158 	dev_dbg(dev, "%s(%p)\n", __func__, substream);
159 	if (all_substreams_zero(cdev->sub_playback) &&
160 	    all_substreams_zero(cdev->sub_capture)) {
161 		/* when the last client has stopped streaming,
162 		 * all sample rates are allowed again */
163 		stream_stop(cdev);
164 		cdev->pcm_info.rates = cdev->samplerates;
165 	}
166 
167 	return 0;
168 }
169 
170 static int snd_usb_caiaq_pcm_hw_params(struct snd_pcm_substream *sub,
171 				       struct snd_pcm_hw_params *hw_params)
172 {
173 	return snd_pcm_lib_malloc_pages(sub, params_buffer_bytes(hw_params));
174 }
175 
176 static int snd_usb_caiaq_pcm_hw_free(struct snd_pcm_substream *sub)
177 {
178 	struct snd_usb_caiaqdev *cdev = snd_pcm_substream_chip(sub);
179 	deactivate_substream(cdev, sub);
180 	return snd_pcm_lib_free_pages(sub);
181 }
182 
183 /* this should probably go upstream */
184 #if SNDRV_PCM_RATE_5512 != 1 << 0 || SNDRV_PCM_RATE_192000 != 1 << 12
185 #error "Change this table"
186 #endif
187 
188 static unsigned int rates[] = { 5512, 8000, 11025, 16000, 22050, 32000, 44100,
189 				48000, 64000, 88200, 96000, 176400, 192000 };
190 
191 static int snd_usb_caiaq_pcm_prepare(struct snd_pcm_substream *substream)
192 {
193 	int bytes_per_sample, bpp, ret, i;
194 	int index = substream->number;
195 	struct snd_usb_caiaqdev *cdev = snd_pcm_substream_chip(substream);
196 	struct snd_pcm_runtime *runtime = substream->runtime;
197 	struct device *dev = caiaqdev_to_dev(cdev);
198 
199 	dev_dbg(dev, "%s(%p)\n", __func__, substream);
200 
201 	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK) {
202 		int out_pos;
203 
204 		switch (cdev->spec.data_alignment) {
205 		case 0:
206 		case 2:
207 			out_pos = BYTES_PER_SAMPLE + 1;
208 			break;
209 		case 3:
210 		default:
211 			out_pos = 0;
212 			break;
213 		}
214 
215 		cdev->period_out_count[index] = out_pos;
216 		cdev->audio_out_buf_pos[index] = out_pos;
217 	} else {
218 		int in_pos;
219 
220 		switch (cdev->spec.data_alignment) {
221 		case 0:
222 			in_pos = BYTES_PER_SAMPLE + 2;
223 			break;
224 		case 2:
225 			in_pos = BYTES_PER_SAMPLE;
226 			break;
227 		case 3:
228 		default:
229 			in_pos = 0;
230 			break;
231 		}
232 
233 		cdev->period_in_count[index] = in_pos;
234 		cdev->audio_in_buf_pos[index] = in_pos;
235 	}
236 
237 	if (cdev->streaming)
238 		return 0;
239 
240 	/* the first client that opens a stream defines the sample rate
241 	 * setting for all subsequent calls, until the last client closed. */
242 	for (i=0; i < ARRAY_SIZE(rates); i++)
243 		if (runtime->rate == rates[i])
244 			cdev->pcm_info.rates = 1 << i;
245 
246 	snd_pcm_limit_hw_rates(runtime);
247 
248 	bytes_per_sample = BYTES_PER_SAMPLE;
249 	if (cdev->spec.data_alignment >= 2)
250 		bytes_per_sample++;
251 
252 	bpp = ((runtime->rate / 8000) + CLOCK_DRIFT_TOLERANCE)
253 		* bytes_per_sample * CHANNELS_PER_STREAM * cdev->n_streams;
254 
255 	if (bpp > MAX_ENDPOINT_SIZE)
256 		bpp = MAX_ENDPOINT_SIZE;
257 
258 	ret = snd_usb_caiaq_set_audio_params(cdev, runtime->rate,
259 					     runtime->sample_bits, bpp);
260 	if (ret)
261 		return ret;
262 
263 	ret = stream_start(cdev);
264 	if (ret)
265 		return ret;
266 
267 	cdev->output_running = 0;
268 	wait_event_timeout(cdev->prepare_wait_queue, cdev->output_running, HZ);
269 	if (!cdev->output_running) {
270 		stream_stop(cdev);
271 		return -EPIPE;
272 	}
273 
274 	return 0;
275 }
276 
277 static int snd_usb_caiaq_pcm_trigger(struct snd_pcm_substream *sub, int cmd)
278 {
279 	struct snd_usb_caiaqdev *cdev = snd_pcm_substream_chip(sub);
280 	struct device *dev = caiaqdev_to_dev(cdev);
281 
282 	dev_dbg(dev, "%s(%p) cmd %d\n", __func__, sub, cmd);
283 
284 	switch (cmd) {
285 	case SNDRV_PCM_TRIGGER_START:
286 	case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
287 		activate_substream(cdev, sub);
288 		break;
289 	case SNDRV_PCM_TRIGGER_STOP:
290 	case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
291 		deactivate_substream(cdev, sub);
292 		break;
293 	default:
294 		return -EINVAL;
295 	}
296 
297 	return 0;
298 }
299 
300 static snd_pcm_uframes_t
301 snd_usb_caiaq_pcm_pointer(struct snd_pcm_substream *sub)
302 {
303 	int index = sub->number;
304 	struct snd_usb_caiaqdev *cdev = snd_pcm_substream_chip(sub);
305 	snd_pcm_uframes_t ptr;
306 
307 	spin_lock(&cdev->spinlock);
308 
309 	if (cdev->input_panic || cdev->output_panic) {
310 		ptr = SNDRV_PCM_POS_XRUN;
311 		goto unlock;
312 	}
313 
314 	if (sub->stream == SNDRV_PCM_STREAM_PLAYBACK)
315 		ptr = bytes_to_frames(sub->runtime,
316 					cdev->audio_out_buf_pos[index]);
317 	else
318 		ptr = bytes_to_frames(sub->runtime,
319 					cdev->audio_in_buf_pos[index]);
320 
321 unlock:
322 	spin_unlock(&cdev->spinlock);
323 	return ptr;
324 }
325 
326 /* operators for both playback and capture */
327 static const struct snd_pcm_ops snd_usb_caiaq_ops = {
328 	.open =		snd_usb_caiaq_substream_open,
329 	.close =	snd_usb_caiaq_substream_close,
330 	.ioctl =	snd_pcm_lib_ioctl,
331 	.hw_params =	snd_usb_caiaq_pcm_hw_params,
332 	.hw_free =	snd_usb_caiaq_pcm_hw_free,
333 	.prepare =	snd_usb_caiaq_pcm_prepare,
334 	.trigger =	snd_usb_caiaq_pcm_trigger,
335 	.pointer =	snd_usb_caiaq_pcm_pointer,
336 };
337 
338 static void check_for_elapsed_periods(struct snd_usb_caiaqdev *cdev,
339 				      struct snd_pcm_substream **subs)
340 {
341 	int stream, pb, *cnt;
342 	struct snd_pcm_substream *sub;
343 
344 	for (stream = 0; stream < cdev->n_streams; stream++) {
345 		sub = subs[stream];
346 		if (!sub)
347 			continue;
348 
349 		pb = snd_pcm_lib_period_bytes(sub);
350 		cnt = (sub->stream == SNDRV_PCM_STREAM_PLAYBACK) ?
351 					&cdev->period_out_count[stream] :
352 					&cdev->period_in_count[stream];
353 
354 		if (*cnt >= pb) {
355 			snd_pcm_period_elapsed(sub);
356 			*cnt %= pb;
357 		}
358 	}
359 }
360 
361 static void read_in_urb_mode0(struct snd_usb_caiaqdev *cdev,
362 			      const struct urb *urb,
363 			      const struct usb_iso_packet_descriptor *iso)
364 {
365 	unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
366 	struct snd_pcm_substream *sub;
367 	int stream, i;
368 
369 	if (all_substreams_zero(cdev->sub_capture))
370 		return;
371 
372 	for (i = 0; i < iso->actual_length;) {
373 		for (stream = 0; stream < cdev->n_streams; stream++, i++) {
374 			sub = cdev->sub_capture[stream];
375 			if (sub) {
376 				struct snd_pcm_runtime *rt = sub->runtime;
377 				char *audio_buf = rt->dma_area;
378 				int sz = frames_to_bytes(rt, rt->buffer_size);
379 				audio_buf[cdev->audio_in_buf_pos[stream]++]
380 					= usb_buf[i];
381 				cdev->period_in_count[stream]++;
382 				if (cdev->audio_in_buf_pos[stream] == sz)
383 					cdev->audio_in_buf_pos[stream] = 0;
384 			}
385 		}
386 	}
387 }
388 
389 static void read_in_urb_mode2(struct snd_usb_caiaqdev *cdev,
390 			      const struct urb *urb,
391 			      const struct usb_iso_packet_descriptor *iso)
392 {
393 	unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
394 	unsigned char check_byte;
395 	struct snd_pcm_substream *sub;
396 	int stream, i;
397 
398 	for (i = 0; i < iso->actual_length;) {
399 		if (i % (cdev->n_streams * BYTES_PER_SAMPLE_USB) == 0) {
400 			for (stream = 0;
401 			     stream < cdev->n_streams;
402 			     stream++, i++) {
403 				if (cdev->first_packet)
404 					continue;
405 
406 				check_byte = MAKE_CHECKBYTE(cdev, stream, i);
407 
408 				if ((usb_buf[i] & 0x3f) != check_byte)
409 					cdev->input_panic = 1;
410 
411 				if (usb_buf[i] & 0x80)
412 					cdev->output_panic = 1;
413 			}
414 		}
415 		cdev->first_packet = 0;
416 
417 		for (stream = 0; stream < cdev->n_streams; stream++, i++) {
418 			sub = cdev->sub_capture[stream];
419 			if (cdev->input_panic)
420 				usb_buf[i] = 0;
421 
422 			if (sub) {
423 				struct snd_pcm_runtime *rt = sub->runtime;
424 				char *audio_buf = rt->dma_area;
425 				int sz = frames_to_bytes(rt, rt->buffer_size);
426 				audio_buf[cdev->audio_in_buf_pos[stream]++] =
427 					usb_buf[i];
428 				cdev->period_in_count[stream]++;
429 				if (cdev->audio_in_buf_pos[stream] == sz)
430 					cdev->audio_in_buf_pos[stream] = 0;
431 			}
432 		}
433 	}
434 }
435 
436 static void read_in_urb_mode3(struct snd_usb_caiaqdev *cdev,
437 			      const struct urb *urb,
438 			      const struct usb_iso_packet_descriptor *iso)
439 {
440 	unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
441 	struct device *dev = caiaqdev_to_dev(cdev);
442 	int stream, i;
443 
444 	/* paranoia check */
445 	if (iso->actual_length % (BYTES_PER_SAMPLE_USB * CHANNELS_PER_STREAM))
446 		return;
447 
448 	for (i = 0; i < iso->actual_length;) {
449 		for (stream = 0; stream < cdev->n_streams; stream++) {
450 			struct snd_pcm_substream *sub = cdev->sub_capture[stream];
451 			char *audio_buf = NULL;
452 			int c, n, sz = 0;
453 
454 			if (sub && !cdev->input_panic) {
455 				struct snd_pcm_runtime *rt = sub->runtime;
456 				audio_buf = rt->dma_area;
457 				sz = frames_to_bytes(rt, rt->buffer_size);
458 			}
459 
460 			for (c = 0; c < CHANNELS_PER_STREAM; c++) {
461 				/* 3 audio data bytes, followed by 1 check byte */
462 				if (audio_buf) {
463 					for (n = 0; n < BYTES_PER_SAMPLE; n++) {
464 						audio_buf[cdev->audio_in_buf_pos[stream]++] = usb_buf[i+n];
465 
466 						if (cdev->audio_in_buf_pos[stream] == sz)
467 							cdev->audio_in_buf_pos[stream] = 0;
468 					}
469 
470 					cdev->period_in_count[stream] += BYTES_PER_SAMPLE;
471 				}
472 
473 				i += BYTES_PER_SAMPLE;
474 
475 				if (usb_buf[i] != ((stream << 1) | c) &&
476 				    !cdev->first_packet) {
477 					if (!cdev->input_panic)
478 						dev_warn(dev, " EXPECTED: %02x got %02x, c %d, stream %d, i %d\n",
479 							 ((stream << 1) | c), usb_buf[i], c, stream, i);
480 					cdev->input_panic = 1;
481 				}
482 
483 				i++;
484 			}
485 		}
486 	}
487 
488 	if (cdev->first_packet > 0)
489 		cdev->first_packet--;
490 }
491 
492 static void read_in_urb(struct snd_usb_caiaqdev *cdev,
493 			const struct urb *urb,
494 			const struct usb_iso_packet_descriptor *iso)
495 {
496 	struct device *dev = caiaqdev_to_dev(cdev);
497 
498 	if (!cdev->streaming)
499 		return;
500 
501 	if (iso->actual_length < cdev->bpp)
502 		return;
503 
504 	switch (cdev->spec.data_alignment) {
505 	case 0:
506 		read_in_urb_mode0(cdev, urb, iso);
507 		break;
508 	case 2:
509 		read_in_urb_mode2(cdev, urb, iso);
510 		break;
511 	case 3:
512 		read_in_urb_mode3(cdev, urb, iso);
513 		break;
514 	}
515 
516 	if ((cdev->input_panic || cdev->output_panic) && !cdev->warned) {
517 		dev_warn(dev, "streaming error detected %s %s\n",
518 				cdev->input_panic ? "(input)" : "",
519 				cdev->output_panic ? "(output)" : "");
520 		cdev->warned = 1;
521 	}
522 }
523 
524 static void fill_out_urb_mode_0(struct snd_usb_caiaqdev *cdev,
525 				struct urb *urb,
526 				const struct usb_iso_packet_descriptor *iso)
527 {
528 	unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
529 	struct snd_pcm_substream *sub;
530 	int stream, i;
531 
532 	for (i = 0; i < iso->length;) {
533 		for (stream = 0; stream < cdev->n_streams; stream++, i++) {
534 			sub = cdev->sub_playback[stream];
535 			if (sub) {
536 				struct snd_pcm_runtime *rt = sub->runtime;
537 				char *audio_buf = rt->dma_area;
538 				int sz = frames_to_bytes(rt, rt->buffer_size);
539 				usb_buf[i] =
540 					audio_buf[cdev->audio_out_buf_pos[stream]];
541 				cdev->period_out_count[stream]++;
542 				cdev->audio_out_buf_pos[stream]++;
543 				if (cdev->audio_out_buf_pos[stream] == sz)
544 					cdev->audio_out_buf_pos[stream] = 0;
545 			} else
546 				usb_buf[i] = 0;
547 		}
548 
549 		/* fill in the check bytes */
550 		if (cdev->spec.data_alignment == 2 &&
551 		    i % (cdev->n_streams * BYTES_PER_SAMPLE_USB) ==
552 		        (cdev->n_streams * CHANNELS_PER_STREAM))
553 			for (stream = 0; stream < cdev->n_streams; stream++, i++)
554 				usb_buf[i] = MAKE_CHECKBYTE(cdev, stream, i);
555 	}
556 }
557 
558 static void fill_out_urb_mode_3(struct snd_usb_caiaqdev *cdev,
559 				struct urb *urb,
560 				const struct usb_iso_packet_descriptor *iso)
561 {
562 	unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
563 	int stream, i;
564 
565 	for (i = 0; i < iso->length;) {
566 		for (stream = 0; stream < cdev->n_streams; stream++) {
567 			struct snd_pcm_substream *sub = cdev->sub_playback[stream];
568 			char *audio_buf = NULL;
569 			int c, n, sz = 0;
570 
571 			if (sub) {
572 				struct snd_pcm_runtime *rt = sub->runtime;
573 				audio_buf = rt->dma_area;
574 				sz = frames_to_bytes(rt, rt->buffer_size);
575 			}
576 
577 			for (c = 0; c < CHANNELS_PER_STREAM; c++) {
578 				for (n = 0; n < BYTES_PER_SAMPLE; n++) {
579 					if (audio_buf) {
580 						usb_buf[i+n] = audio_buf[cdev->audio_out_buf_pos[stream]++];
581 
582 						if (cdev->audio_out_buf_pos[stream] == sz)
583 							cdev->audio_out_buf_pos[stream] = 0;
584 					} else {
585 						usb_buf[i+n] = 0;
586 					}
587 				}
588 
589 				if (audio_buf)
590 					cdev->period_out_count[stream] += BYTES_PER_SAMPLE;
591 
592 				i += BYTES_PER_SAMPLE;
593 
594 				/* fill in the check byte pattern */
595 				usb_buf[i++] = (stream << 1) | c;
596 			}
597 		}
598 	}
599 }
600 
601 static inline void fill_out_urb(struct snd_usb_caiaqdev *cdev,
602 				struct urb *urb,
603 				const struct usb_iso_packet_descriptor *iso)
604 {
605 	switch (cdev->spec.data_alignment) {
606 	case 0:
607 	case 2:
608 		fill_out_urb_mode_0(cdev, urb, iso);
609 		break;
610 	case 3:
611 		fill_out_urb_mode_3(cdev, urb, iso);
612 		break;
613 	}
614 }
615 
616 static void read_completed(struct urb *urb)
617 {
618 	struct snd_usb_caiaq_cb_info *info = urb->context;
619 	struct snd_usb_caiaqdev *cdev;
620 	struct device *dev;
621 	struct urb *out = NULL;
622 	int i, frame, len, send_it = 0, outframe = 0;
623 	unsigned long flags;
624 	size_t offset = 0;
625 
626 	if (urb->status || !info)
627 		return;
628 
629 	cdev = info->cdev;
630 	dev = caiaqdev_to_dev(cdev);
631 
632 	if (!cdev->streaming)
633 		return;
634 
635 	/* find an unused output urb that is unused */
636 	for (i = 0; i < N_URBS; i++)
637 		if (test_and_set_bit(i, &cdev->outurb_active_mask) == 0) {
638 			out = cdev->data_urbs_out[i];
639 			break;
640 		}
641 
642 	if (!out) {
643 		dev_err(dev, "Unable to find an output urb to use\n");
644 		goto requeue;
645 	}
646 
647 	/* read the recently received packet and send back one which has
648 	 * the same layout */
649 	for (frame = 0; frame < FRAMES_PER_URB; frame++) {
650 		if (urb->iso_frame_desc[frame].status)
651 			continue;
652 
653 		len = urb->iso_frame_desc[outframe].actual_length;
654 		out->iso_frame_desc[outframe].length = len;
655 		out->iso_frame_desc[outframe].actual_length = 0;
656 		out->iso_frame_desc[outframe].offset = offset;
657 		offset += len;
658 
659 		if (len > 0) {
660 			spin_lock_irqsave(&cdev->spinlock, flags);
661 			fill_out_urb(cdev, out, &out->iso_frame_desc[outframe]);
662 			read_in_urb(cdev, urb, &urb->iso_frame_desc[frame]);
663 			spin_unlock_irqrestore(&cdev->spinlock, flags);
664 			check_for_elapsed_periods(cdev, cdev->sub_playback);
665 			check_for_elapsed_periods(cdev, cdev->sub_capture);
666 			send_it = 1;
667 		}
668 
669 		outframe++;
670 	}
671 
672 	if (send_it) {
673 		out->number_of_packets = outframe;
674 		usb_submit_urb(out, GFP_ATOMIC);
675 	} else {
676 		struct snd_usb_caiaq_cb_info *oinfo = out->context;
677 		clear_bit(oinfo->index, &cdev->outurb_active_mask);
678 	}
679 
680 requeue:
681 	/* re-submit inbound urb */
682 	for (frame = 0; frame < FRAMES_PER_URB; frame++) {
683 		urb->iso_frame_desc[frame].offset = BYTES_PER_FRAME * frame;
684 		urb->iso_frame_desc[frame].length = BYTES_PER_FRAME;
685 		urb->iso_frame_desc[frame].actual_length = 0;
686 	}
687 
688 	urb->number_of_packets = FRAMES_PER_URB;
689 	usb_submit_urb(urb, GFP_ATOMIC);
690 }
691 
692 static void write_completed(struct urb *urb)
693 {
694 	struct snd_usb_caiaq_cb_info *info = urb->context;
695 	struct snd_usb_caiaqdev *cdev = info->cdev;
696 
697 	if (!cdev->output_running) {
698 		cdev->output_running = 1;
699 		wake_up(&cdev->prepare_wait_queue);
700 	}
701 
702 	clear_bit(info->index, &cdev->outurb_active_mask);
703 }
704 
705 static struct urb **alloc_urbs(struct snd_usb_caiaqdev *cdev, int dir, int *ret)
706 {
707 	int i, frame;
708 	struct urb **urbs;
709 	struct usb_device *usb_dev = cdev->chip.dev;
710 	unsigned int pipe;
711 
712 	pipe = (dir == SNDRV_PCM_STREAM_PLAYBACK) ?
713 		usb_sndisocpipe(usb_dev, ENDPOINT_PLAYBACK) :
714 		usb_rcvisocpipe(usb_dev, ENDPOINT_CAPTURE);
715 
716 	urbs = kmalloc_array(N_URBS, sizeof(*urbs), GFP_KERNEL);
717 	if (!urbs) {
718 		*ret = -ENOMEM;
719 		return NULL;
720 	}
721 
722 	for (i = 0; i < N_URBS; i++) {
723 		urbs[i] = usb_alloc_urb(FRAMES_PER_URB, GFP_KERNEL);
724 		if (!urbs[i]) {
725 			*ret = -ENOMEM;
726 			return urbs;
727 		}
728 
729 		urbs[i]->transfer_buffer =
730 			kmalloc_array(BYTES_PER_FRAME, FRAMES_PER_URB,
731 				      GFP_KERNEL);
732 		if (!urbs[i]->transfer_buffer) {
733 			*ret = -ENOMEM;
734 			return urbs;
735 		}
736 
737 		for (frame = 0; frame < FRAMES_PER_URB; frame++) {
738 			struct usb_iso_packet_descriptor *iso =
739 				&urbs[i]->iso_frame_desc[frame];
740 
741 			iso->offset = BYTES_PER_FRAME * frame;
742 			iso->length = BYTES_PER_FRAME;
743 		}
744 
745 		urbs[i]->dev = usb_dev;
746 		urbs[i]->pipe = pipe;
747 		urbs[i]->transfer_buffer_length = FRAMES_PER_URB
748 						* BYTES_PER_FRAME;
749 		urbs[i]->context = &cdev->data_cb_info[i];
750 		urbs[i]->interval = 1;
751 		urbs[i]->number_of_packets = FRAMES_PER_URB;
752 		urbs[i]->complete = (dir == SNDRV_PCM_STREAM_CAPTURE) ?
753 					read_completed : write_completed;
754 	}
755 
756 	*ret = 0;
757 	return urbs;
758 }
759 
760 static void free_urbs(struct urb **urbs)
761 {
762 	int i;
763 
764 	if (!urbs)
765 		return;
766 
767 	for (i = 0; i < N_URBS; i++) {
768 		if (!urbs[i])
769 			continue;
770 
771 		usb_kill_urb(urbs[i]);
772 		kfree(urbs[i]->transfer_buffer);
773 		usb_free_urb(urbs[i]);
774 	}
775 
776 	kfree(urbs);
777 }
778 
779 int snd_usb_caiaq_audio_init(struct snd_usb_caiaqdev *cdev)
780 {
781 	int i, ret;
782 	struct device *dev = caiaqdev_to_dev(cdev);
783 
784 	cdev->n_audio_in  = max(cdev->spec.num_analog_audio_in,
785 			       cdev->spec.num_digital_audio_in) /
786 				CHANNELS_PER_STREAM;
787 	cdev->n_audio_out = max(cdev->spec.num_analog_audio_out,
788 			       cdev->spec.num_digital_audio_out) /
789 				CHANNELS_PER_STREAM;
790 	cdev->n_streams = max(cdev->n_audio_in, cdev->n_audio_out);
791 
792 	dev_dbg(dev, "cdev->n_audio_in = %d\n", cdev->n_audio_in);
793 	dev_dbg(dev, "cdev->n_audio_out = %d\n", cdev->n_audio_out);
794 	dev_dbg(dev, "cdev->n_streams = %d\n", cdev->n_streams);
795 
796 	if (cdev->n_streams > MAX_STREAMS) {
797 		dev_err(dev, "unable to initialize device, too many streams.\n");
798 		return -EINVAL;
799 	}
800 
801 	if (cdev->n_streams < 1) {
802 		dev_err(dev, "bogus number of streams: %d\n", cdev->n_streams);
803 		return -EINVAL;
804 	}
805 
806 	ret = snd_pcm_new(cdev->chip.card, cdev->product_name, 0,
807 			cdev->n_audio_out, cdev->n_audio_in, &cdev->pcm);
808 
809 	if (ret < 0) {
810 		dev_err(dev, "snd_pcm_new() returned %d\n", ret);
811 		return ret;
812 	}
813 
814 	cdev->pcm->private_data = cdev;
815 	strlcpy(cdev->pcm->name, cdev->product_name, sizeof(cdev->pcm->name));
816 
817 	memset(cdev->sub_playback, 0, sizeof(cdev->sub_playback));
818 	memset(cdev->sub_capture, 0, sizeof(cdev->sub_capture));
819 
820 	memcpy(&cdev->pcm_info, &snd_usb_caiaq_pcm_hardware,
821 			sizeof(snd_usb_caiaq_pcm_hardware));
822 
823 	/* setup samplerates */
824 	cdev->samplerates = cdev->pcm_info.rates;
825 	switch (cdev->chip.usb_id) {
826 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AK1):
827 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_RIGKONTROL3):
828 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_SESSIONIO):
829 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_GUITARRIGMOBILE):
830 		cdev->samplerates |= SNDRV_PCM_RATE_192000;
831 		/* fall thru */
832 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AUDIO2DJ):
833 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AUDIO4DJ):
834 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AUDIO8DJ):
835 	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORAUDIO2):
836 		cdev->samplerates |= SNDRV_PCM_RATE_88200;
837 		break;
838 	}
839 
840 	snd_pcm_set_ops(cdev->pcm, SNDRV_PCM_STREAM_PLAYBACK,
841 				&snd_usb_caiaq_ops);
842 	snd_pcm_set_ops(cdev->pcm, SNDRV_PCM_STREAM_CAPTURE,
843 				&snd_usb_caiaq_ops);
844 	snd_pcm_lib_preallocate_pages_for_all(cdev->pcm, SNDRV_DMA_TYPE_VMALLOC,
845 					      NULL, 0, 0);
846 
847 	cdev->data_cb_info =
848 		kmalloc_array(N_URBS, sizeof(struct snd_usb_caiaq_cb_info),
849 					GFP_KERNEL);
850 
851 	if (!cdev->data_cb_info)
852 		return -ENOMEM;
853 
854 	cdev->outurb_active_mask = 0;
855 	BUILD_BUG_ON(N_URBS > (sizeof(cdev->outurb_active_mask) * 8));
856 
857 	for (i = 0; i < N_URBS; i++) {
858 		cdev->data_cb_info[i].cdev = cdev;
859 		cdev->data_cb_info[i].index = i;
860 	}
861 
862 	cdev->data_urbs_in = alloc_urbs(cdev, SNDRV_PCM_STREAM_CAPTURE, &ret);
863 	if (ret < 0) {
864 		kfree(cdev->data_cb_info);
865 		free_urbs(cdev->data_urbs_in);
866 		return ret;
867 	}
868 
869 	cdev->data_urbs_out = alloc_urbs(cdev, SNDRV_PCM_STREAM_PLAYBACK, &ret);
870 	if (ret < 0) {
871 		kfree(cdev->data_cb_info);
872 		free_urbs(cdev->data_urbs_in);
873 		free_urbs(cdev->data_urbs_out);
874 		return ret;
875 	}
876 
877 	return 0;
878 }
879 
880 void snd_usb_caiaq_audio_free(struct snd_usb_caiaqdev *cdev)
881 {
882 	struct device *dev = caiaqdev_to_dev(cdev);
883 
884 	dev_dbg(dev, "%s(%p)\n", __func__, cdev);
885 	stream_stop(cdev);
886 	free_urbs(cdev->data_urbs_in);
887 	free_urbs(cdev->data_urbs_out);
888 	kfree(cdev->data_cb_info);
889 }
890 
891