1 3baa2617 2022-02-16 op /*
  2 3baa2617 2022-02-16 op  * Copyright (c) 2022 Omar Polo <op@openbsd.org>
  3 3baa2617 2022-02-16 op  *
  4 3baa2617 2022-02-16 op  * Permission to use, copy, modify, and distribute this software for any
  5 3baa2617 2022-02-16 op  * purpose with or without fee is hereby granted, provided that the above
  6 3baa2617 2022-02-16 op  * copyright notice and this permission notice appear in all copies.
  7 3baa2617 2022-02-16 op  *
  8 3baa2617 2022-02-16 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 3baa2617 2022-02-16 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 3baa2617 2022-02-16 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 3baa2617 2022-02-16 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 3baa2617 2022-02-16 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 3baa2617 2022-02-16 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 3baa2617 2022-02-16 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 3baa2617 2022-02-16 op  */
 16 3baa2617 2022-02-16 op 
 17 3baa2617 2022-02-16 op #include <sys/types.h>
 18 3baa2617 2022-02-16 op #include <sys/queue.h>
 19 3baa2617 2022-02-16 op #include <sys/uio.h>
 20 3baa2617 2022-02-16 op 
 21 3baa2617 2022-02-16 op #include <err.h>
 22 3baa2617 2022-02-16 op #include <event.h>
 23 3baa2617 2022-02-16 op #include <inttypes.h>
 24 bb3f279f 2022-02-16 op #include <limits.h>
 25 3baa2617 2022-02-16 op #include <stdio.h>
 26 3baa2617 2022-02-16 op #include <stdlib.h>
 27 3baa2617 2022-02-16 op #include <stdint.h>
 28 791d3db3 2022-07-09 op #include <string.h>
 29 3baa2617 2022-02-16 op #include <imsg.h>
 30 3baa2617 2022-02-16 op 
 31 3baa2617 2022-02-16 op #include <FLAC/stream_decoder.h>
 32 3baa2617 2022-02-16 op 
 33 3baa2617 2022-02-16 op #include "amused.h"
 34 bee9cc8d 2022-02-18 op #include "log.h"
 35 3baa2617 2022-02-16 op 
 36 791d3db3 2022-07-09 op struct write_args {
 37 791d3db3 2022-07-09 op 	FLAC__StreamDecoder *decoder;
 38 791d3db3 2022-07-09 op 	int seek_failed;
 39 791d3db3 2022-07-09 op };
 40 791d3db3 2022-07-09 op 
 41 791d3db3 2022-07-09 op static int
 42 791d3db3 2022-07-09 op sample_seek(struct write_args *wa, int64_t seek)
 43 791d3db3 2022-07-09 op {
 44 791d3db3 2022-07-09 op 	int ok;
 45 791d3db3 2022-07-09 op 
 46 791d3db3 2022-07-09 op 	ok = FLAC__stream_decoder_seek_absolute(wa->decoder, seek);
 47 791d3db3 2022-07-09 op 	if (ok)
 48 791d3db3 2022-07-09 op 		player_setpos(seek);
 49 791d3db3 2022-07-09 op 	else
 50 791d3db3 2022-07-09 op 		wa->seek_failed = 1;
 51 791d3db3 2022-07-09 op 	return ok;
 52 791d3db3 2022-07-09 op }
 53 791d3db3 2022-07-09 op 
 54 3baa2617 2022-02-16 op static FLAC__StreamDecoderWriteStatus
 55 3baa2617 2022-02-16 op writecb(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame,
 56 44cf9512 2022-03-25 op     const int32_t * const *src, void *data)
 57 3baa2617 2022-02-16 op {
 58 791d3db3 2022-07-09 op 	struct write_args *wa = data;
 59 3baa2617 2022-02-16 op 	static uint8_t buf[BUFSIZ];
 60 791d3db3 2022-07-09 op 	int64_t seek;
 61 44cf9512 2022-03-25 op 	int c, i, bps, chans;
 62 3baa2617 2022-02-16 op 	size_t len;
 63 3baa2617 2022-02-16 op 
 64 44cf9512 2022-03-25 op 	bps = frame->header.bits_per_sample;
 65 44cf9512 2022-03-25 op 	chans = frame->header.channels;
 66 44cf9512 2022-03-25 op 
 67 3baa2617 2022-02-16 op 	for (i = 0, len = 0; i < frame->header.blocksize; ++i) {
 68 d9cc6713 2022-03-25 op 		if (len + 4*chans >= sizeof(buf)) {
 69 791d3db3 2022-07-09 op 			if (!play(buf, len, &seek))
 70 2139c525 2022-03-09 op 				goto quit;
 71 791d3db3 2022-07-09 op 			if (seek != -1) {
 72 791d3db3 2022-07-09 op 				if (sample_seek(wa, seek))
 73 791d3db3 2022-07-09 op 					break;
 74 791d3db3 2022-07-09 op 				else
 75 791d3db3 2022-07-09 op 					goto quit;
 76 791d3db3 2022-07-09 op 			}
 77 ac02581f 2022-03-13 op 			len = 0;
 78 3baa2617 2022-02-16 op 		}
 79 3baa2617 2022-02-16 op 
 80 44cf9512 2022-03-25 op 		for (c = 0; c < chans; ++c) {
 81 44cf9512 2022-03-25 op 			switch (bps) {
 82 44cf9512 2022-03-25 op 			case 8:
 83 44cf9512 2022-03-25 op 				buf[len++] = src[c][i] & 0xff;
 84 44cf9512 2022-03-25 op 				break;
 85 44cf9512 2022-03-25 op 			case 16:
 86 44cf9512 2022-03-25 op 				buf[len++] = src[c][i] & 0xff;
 87 44cf9512 2022-03-25 op 				buf[len++] = (src[c][i] >> 8) & 0xff;
 88 44cf9512 2022-03-25 op 				break;
 89 44cf9512 2022-03-25 op 			case 24:
 90 44cf9512 2022-03-25 op 			case 32:
 91 44cf9512 2022-03-25 op 				buf[len++] = src[c][i] & 0xff;
 92 44cf9512 2022-03-25 op 				buf[len++] = (src[c][i] >> 8) & 0xff;
 93 44cf9512 2022-03-25 op 				buf[len++] = (src[c][i] >> 16) & 0xff;
 94 44cf9512 2022-03-25 op 				buf[len++] = (src[c][i] >> 24) & 0xff;
 95 44cf9512 2022-03-25 op 				break;
 96 44cf9512 2022-03-25 op 			default:
 97 44cf9512 2022-03-25 op 				log_warnx("unsupported flac bps=%d", bps);
 98 44cf9512 2022-03-25 op 				goto quit;
 99 44cf9512 2022-03-25 op 			}
100 44cf9512 2022-03-25 op 		}
101 3baa2617 2022-02-16 op 	}
102 3baa2617 2022-02-16 op 
103 791d3db3 2022-07-09 op 	if (len != 0 && !play(buf, len, &seek))
104 2139c525 2022-03-09 op 		goto quit;
105 3baa2617 2022-02-16 op 
106 791d3db3 2022-07-09 op 	if (seek != -1 && !sample_seek(wa, seek))
107 791d3db3 2022-07-09 op 		goto quit;
108 791d3db3 2022-07-09 op 
109 3baa2617 2022-02-16 op 	return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
110 2139c525 2022-03-09 op quit:
111 2139c525 2022-03-09 op 	return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
112 3baa2617 2022-02-16 op }
113 3baa2617 2022-02-16 op 
114 3baa2617 2022-02-16 op static void
115 3baa2617 2022-02-16 op metacb(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *meta,
116 3baa2617 2022-02-16 op     void *d)
117 3baa2617 2022-02-16 op {
118 3baa2617 2022-02-16 op 	uint32_t sample_rate;
119 44cf9512 2022-03-25 op 	int channels, bits;
120 3baa2617 2022-02-16 op 
121 3baa2617 2022-02-16 op 	if (meta->type == FLAC__METADATA_TYPE_STREAMINFO) {
122 44cf9512 2022-03-25 op 		bits = meta->data.stream_info.bits_per_sample;
123 3baa2617 2022-02-16 op 		sample_rate = meta->data.stream_info.sample_rate;
124 7fc831ea 2022-02-18 op 		channels = meta->data.stream_info.channels;
125 3baa2617 2022-02-16 op 
126 44cf9512 2022-03-25 op 		if (player_setup(bits, sample_rate, channels) == -1)
127 1fb06c31 2022-06-10 op 			err(1, "player_setup");
128 ff06024f 2022-07-08 op 
129 ff06024f 2022-07-08 op 		player_setduration(meta->data.stream_info.total_samples);
130 3baa2617 2022-02-16 op 	}
131 3baa2617 2022-02-16 op }
132 3baa2617 2022-02-16 op 
133 3baa2617 2022-02-16 op static void
134 1771f738 2022-06-09 op errcb(const FLAC__StreamDecoder *decoder,
135 1771f738 2022-06-09 op     FLAC__StreamDecoderErrorStatus status, void *data)
136 3baa2617 2022-02-16 op {
137 1771f738 2022-06-09 op 	log_warnx("flac error: %s",
138 1771f738 2022-06-09 op 	    FLAC__StreamDecoderErrorStatusString[status]);
139 3baa2617 2022-02-16 op }
140 3baa2617 2022-02-16 op 
141 0da0ad46 2022-03-09 op int
142 17ef54d6 2022-06-22 op play_flac(int fd, const char **errstr)
143 3baa2617 2022-02-16 op {
144 3baa2617 2022-02-16 op 	FILE *f;
145 791d3db3 2022-07-09 op 	struct write_args wa;
146 bee9cc8d 2022-02-18 op 	int s, ok = 1;
147 3baa2617 2022-02-16 op 	FLAC__StreamDecoder *decoder = NULL;
148 3baa2617 2022-02-16 op 	FLAC__StreamDecoderInitStatus init_status;
149 3baa2617 2022-02-16 op 
150 3baa2617 2022-02-16 op 	if ((f = fdopen(fd, "r")) == NULL)
151 3baa2617 2022-02-16 op 		err(1, "fdopen");
152 3baa2617 2022-02-16 op 
153 3baa2617 2022-02-16 op 	decoder = FLAC__stream_decoder_new();
154 3baa2617 2022-02-16 op 	if (decoder == NULL)
155 3baa2617 2022-02-16 op 		err(1, "flac stream decoder");
156 3baa2617 2022-02-16 op 
157 3baa2617 2022-02-16 op 	FLAC__stream_decoder_set_md5_checking(decoder, 1);
158 3baa2617 2022-02-16 op 
159 791d3db3 2022-07-09 op 	memset(&wa, 0, sizeof(wa));
160 791d3db3 2022-07-09 op 	wa.decoder = decoder;
161 791d3db3 2022-07-09 op 
162 3baa2617 2022-02-16 op 	init_status = FLAC__stream_decoder_init_FILE(decoder, f, writecb,
163 791d3db3 2022-07-09 op 	    metacb, errcb, &wa);
164 3baa2617 2022-02-16 op 	if (init_status != FLAC__STREAM_DECODER_INIT_STATUS_OK)
165 3baa2617 2022-02-16 op 		errx(1, "flac decoder: %s",
166 3baa2617 2022-02-16 op 		    FLAC__StreamDecoderInitStatusString[init_status]);
167 3baa2617 2022-02-16 op 
168 3baa2617 2022-02-16 op 	ok = FLAC__stream_decoder_process_until_end_of_stream(decoder);
169 0da0ad46 2022-03-09 op 
170 791d3db3 2022-07-09 op 	s = FLAC__stream_decoder_get_state(decoder);
171 0da0ad46 2022-03-09 op 	FLAC__stream_decoder_delete(decoder);
172 0da0ad46 2022-03-09 op 	fclose(f);
173 0da0ad46 2022-03-09 op 
174 791d3db3 2022-07-09 op 	if (s == FLAC__STREAM_DECODER_ABORTED && !wa.seek_failed)
175 0923bedc 2022-03-13 op 		return 1;
176 791d3db3 2022-07-09 op 	else if (!ok && !wa.seek_failed) {
177 17ef54d6 2022-06-22 op 		*errstr = "flac decoding error";
178 0da0ad46 2022-03-09 op 		return -1;
179 0923bedc 2022-03-13 op 	} else
180 0923bedc 2022-03-13 op 		return 0;
181 3baa2617 2022-02-16 op }