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 f36fd90a 2022-07-09 op #include "config.h"
 18 3baa2617 2022-02-16 op 
 19 3baa2617 2022-02-16 op #include <inttypes.h>
 20 bb3f279f 2022-02-16 op #include <limits.h>
 21 3baa2617 2022-02-16 op #include <stdio.h>
 22 3baa2617 2022-02-16 op #include <stdlib.h>
 23 3baa2617 2022-02-16 op #include <stdint.h>
 24 791d3db3 2022-07-09 op #include <string.h>
 25 3baa2617 2022-02-16 op 
 26 3baa2617 2022-02-16 op #include <FLAC/stream_decoder.h>
 27 3baa2617 2022-02-16 op 
 28 3baa2617 2022-02-16 op #include "amused.h"
 29 bee9cc8d 2022-02-18 op #include "log.h"
 30 3baa2617 2022-02-16 op 
 31 791d3db3 2022-07-09 op struct write_args {
 32 791d3db3 2022-07-09 op 	FLAC__StreamDecoder *decoder;
 33 791d3db3 2022-07-09 op 	int seek_failed;
 34 791d3db3 2022-07-09 op };
 35 791d3db3 2022-07-09 op 
 36 791d3db3 2022-07-09 op static int
 37 791d3db3 2022-07-09 op sample_seek(struct write_args *wa, int64_t seek)
 38 791d3db3 2022-07-09 op {
 39 791d3db3 2022-07-09 op 	int ok;
 40 791d3db3 2022-07-09 op 
 41 791d3db3 2022-07-09 op 	ok = FLAC__stream_decoder_seek_absolute(wa->decoder, seek);
 42 791d3db3 2022-07-09 op 	if (ok)
 43 791d3db3 2022-07-09 op 		player_setpos(seek);
 44 791d3db3 2022-07-09 op 	else
 45 791d3db3 2022-07-09 op 		wa->seek_failed = 1;
 46 791d3db3 2022-07-09 op 	return ok;
 47 791d3db3 2022-07-09 op }
 48 791d3db3 2022-07-09 op 
 49 3baa2617 2022-02-16 op static FLAC__StreamDecoderWriteStatus
 50 3baa2617 2022-02-16 op writecb(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame,
 51 44cf9512 2022-03-25 op     const int32_t * const *src, void *data)
 52 3baa2617 2022-02-16 op {
 53 791d3db3 2022-07-09 op 	struct write_args *wa = data;
 54 3baa2617 2022-02-16 op 	static uint8_t buf[BUFSIZ];
 55 791d3db3 2022-07-09 op 	int64_t seek;
 56 44cf9512 2022-03-25 op 	int c, i, bps, chans;
 57 3baa2617 2022-02-16 op 	size_t len;
 58 3baa2617 2022-02-16 op 
 59 44cf9512 2022-03-25 op 	bps = frame->header.bits_per_sample;
 60 44cf9512 2022-03-25 op 	chans = frame->header.channels;
 61 44cf9512 2022-03-25 op 
 62 3baa2617 2022-02-16 op 	for (i = 0, len = 0; i < frame->header.blocksize; ++i) {
 63 d9cc6713 2022-03-25 op 		if (len + 4*chans >= sizeof(buf)) {
 64 791d3db3 2022-07-09 op 			if (!play(buf, len, &seek))
 65 2139c525 2022-03-09 op 				goto quit;
 66 791d3db3 2022-07-09 op 			if (seek != -1) {
 67 791d3db3 2022-07-09 op 				if (sample_seek(wa, seek))
 68 791d3db3 2022-07-09 op 					break;
 69 791d3db3 2022-07-09 op 				else
 70 791d3db3 2022-07-09 op 					goto quit;
 71 791d3db3 2022-07-09 op 			}
 72 ac02581f 2022-03-13 op 			len = 0;
 73 3baa2617 2022-02-16 op 		}
 74 3baa2617 2022-02-16 op 
 75 44cf9512 2022-03-25 op 		for (c = 0; c < chans; ++c) {
 76 44cf9512 2022-03-25 op 			switch (bps) {
 77 44cf9512 2022-03-25 op 			case 8:
 78 44cf9512 2022-03-25 op 				buf[len++] = src[c][i] & 0xff;
 79 44cf9512 2022-03-25 op 				break;
 80 44cf9512 2022-03-25 op 			case 16:
 81 44cf9512 2022-03-25 op 				buf[len++] = src[c][i] & 0xff;
 82 44cf9512 2022-03-25 op 				buf[len++] = (src[c][i] >> 8) & 0xff;
 83 44cf9512 2022-03-25 op 				break;
 84 44cf9512 2022-03-25 op 			case 24:
 85 44cf9512 2022-03-25 op 			case 32:
 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 				buf[len++] = (src[c][i] >> 16) & 0xff;
 89 44cf9512 2022-03-25 op 				buf[len++] = (src[c][i] >> 24) & 0xff;
 90 44cf9512 2022-03-25 op 				break;
 91 44cf9512 2022-03-25 op 			default:
 92 44cf9512 2022-03-25 op 				log_warnx("unsupported flac bps=%d", bps);
 93 44cf9512 2022-03-25 op 				goto quit;
 94 44cf9512 2022-03-25 op 			}
 95 44cf9512 2022-03-25 op 		}
 96 3baa2617 2022-02-16 op 	}
 97 3baa2617 2022-02-16 op 
 98 791d3db3 2022-07-09 op 	if (len != 0 && !play(buf, len, &seek))
 99 2139c525 2022-03-09 op 		goto quit;
100 3baa2617 2022-02-16 op 
101 791d3db3 2022-07-09 op 	if (seek != -1 && !sample_seek(wa, seek))
102 791d3db3 2022-07-09 op 		goto quit;
103 791d3db3 2022-07-09 op 
104 3baa2617 2022-02-16 op 	return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
105 2139c525 2022-03-09 op quit:
106 2139c525 2022-03-09 op 	return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
107 3baa2617 2022-02-16 op }
108 3baa2617 2022-02-16 op 
109 3baa2617 2022-02-16 op static void
110 3baa2617 2022-02-16 op metacb(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *meta,
111 3baa2617 2022-02-16 op     void *d)
112 3baa2617 2022-02-16 op {
113 3baa2617 2022-02-16 op 	uint32_t sample_rate;
114 44cf9512 2022-03-25 op 	int channels, bits;
115 3baa2617 2022-02-16 op 
116 3baa2617 2022-02-16 op 	if (meta->type == FLAC__METADATA_TYPE_STREAMINFO) {
117 44cf9512 2022-03-25 op 		bits = meta->data.stream_info.bits_per_sample;
118 3baa2617 2022-02-16 op 		sample_rate = meta->data.stream_info.sample_rate;
119 7fc831ea 2022-02-18 op 		channels = meta->data.stream_info.channels;
120 3baa2617 2022-02-16 op 
121 44cf9512 2022-03-25 op 		if (player_setup(bits, sample_rate, channels) == -1)
122 1fb06c31 2022-06-10 op 			err(1, "player_setup");
123 ff06024f 2022-07-08 op 
124 ff06024f 2022-07-08 op 		player_setduration(meta->data.stream_info.total_samples);
125 3baa2617 2022-02-16 op 	}
126 3baa2617 2022-02-16 op }
127 3baa2617 2022-02-16 op 
128 3baa2617 2022-02-16 op static void
129 1771f738 2022-06-09 op errcb(const FLAC__StreamDecoder *decoder,
130 1771f738 2022-06-09 op     FLAC__StreamDecoderErrorStatus status, void *data)
131 3baa2617 2022-02-16 op {
132 1771f738 2022-06-09 op 	log_warnx("flac error: %s",
133 1771f738 2022-06-09 op 	    FLAC__StreamDecoderErrorStatusString[status]);
134 3baa2617 2022-02-16 op }
135 3baa2617 2022-02-16 op 
136 0da0ad46 2022-03-09 op int
137 17ef54d6 2022-06-22 op play_flac(int fd, const char **errstr)
138 3baa2617 2022-02-16 op {
139 3baa2617 2022-02-16 op 	FILE *f;
140 791d3db3 2022-07-09 op 	struct write_args wa;
141 bee9cc8d 2022-02-18 op 	int s, ok = 1;
142 3baa2617 2022-02-16 op 	FLAC__StreamDecoder *decoder = NULL;
143 3baa2617 2022-02-16 op 	FLAC__StreamDecoderInitStatus init_status;
144 3baa2617 2022-02-16 op 
145 986b215c 2022-07-09 op 	if ((f = fdopen(fd, "r")) == NULL) {
146 986b215c 2022-07-09 op 		*errstr = "fdopen failed";
147 986b215c 2022-07-09 op 		close(fd);
148 986b215c 2022-07-09 op 		return -1;
149 986b215c 2022-07-09 op 	}
150 3baa2617 2022-02-16 op 
151 3baa2617 2022-02-16 op 	decoder = FLAC__stream_decoder_new();
152 986b215c 2022-07-09 op 	if (decoder == NULL) {
153 986b215c 2022-07-09 op 		*errstr = "FLAC__stream_decoder_new() failed";
154 986b215c 2022-07-09 op 		fclose(f);
155 986b215c 2022-07-09 op 		return -1;
156 986b215c 2022-07-09 op 	}
157 3baa2617 2022-02-16 op 
158 3baa2617 2022-02-16 op 	FLAC__stream_decoder_set_md5_checking(decoder, 1);
159 3baa2617 2022-02-16 op 
160 791d3db3 2022-07-09 op 	memset(&wa, 0, sizeof(wa));
161 791d3db3 2022-07-09 op 	wa.decoder = decoder;
162 791d3db3 2022-07-09 op 
163 3baa2617 2022-02-16 op 	init_status = FLAC__stream_decoder_init_FILE(decoder, f, writecb,
164 791d3db3 2022-07-09 op 	    metacb, errcb, &wa);
165 3baa2617 2022-02-16 op 	if (init_status != FLAC__STREAM_DECODER_INIT_STATUS_OK)
166 3baa2617 2022-02-16 op 		errx(1, "flac decoder: %s",
167 3baa2617 2022-02-16 op 		    FLAC__StreamDecoderInitStatusString[init_status]);
168 3baa2617 2022-02-16 op 
169 3baa2617 2022-02-16 op 	ok = FLAC__stream_decoder_process_until_end_of_stream(decoder);
170 0da0ad46 2022-03-09 op 
171 791d3db3 2022-07-09 op 	s = FLAC__stream_decoder_get_state(decoder);
172 0da0ad46 2022-03-09 op 	FLAC__stream_decoder_delete(decoder);
173 0da0ad46 2022-03-09 op 	fclose(f);
174 0da0ad46 2022-03-09 op 
175 791d3db3 2022-07-09 op 	if (s == FLAC__STREAM_DECODER_ABORTED && !wa.seek_failed)
176 0923bedc 2022-03-13 op 		return 1;
177 791d3db3 2022-07-09 op 	else if (!ok && !wa.seek_failed) {
178 17ef54d6 2022-06-22 op 		*errstr = "flac decoding error";
179 0da0ad46 2022-03-09 op 		return -1;
180 0923bedc 2022-03-13 op 	} else
181 0923bedc 2022-03-13 op 		return 0;
182 3baa2617 2022-02-16 op }