Change sbc_initialized to pa_bool_t
[pulseaudio-mirror.git] / src / modules / module-bt-device.c
blob053ef080c7511bdc7c34b1eb91b3f385731bd707
1 /***
2 This file is part of PulseAudio.
4 Copyright 2008 Joao Paulo Rechi Vita
6 PulseAudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU Lesser General Public License as published
8 by the Free Software Foundation; either version 2 of the License,
9 or (at your option) any later version.
11 PulseAudio is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
16 You should have received a copy of the GNU Lesser General Public License
17 along with PulseAudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19 USA.
20 ***/
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
26 #include <string.h>
27 #include <errno.h>
28 #include <poll.h>
30 #include <pulse/xmalloc.h>
31 #include <pulse/timeval.h>
32 #include <pulse/sample.h>
33 #include <pulsecore/module.h>
34 #include <pulsecore/modargs.h>
35 #include <pulsecore/core-util.h>
36 #include <pulsecore/core-error.h>
37 #include <pulsecore/socket-util.h>
38 #include <pulsecore/thread.h>
39 #include <pulsecore/thread-mq.h>
40 #include <pulsecore/rtpoll.h>
41 #include <pulsecore/time-smoother.h>
42 #include <pulsecore/rtclock.h>
44 #include "dbus-util.h"
45 #include "module-bt-device-symdef.h"
46 #include "bt-ipc.h"
47 #include "bt-sbc.h"
48 #include "bt-rtp.h"
50 #define DEFAULT_SINK_NAME "bluetooth_sink"
51 #define BUFFER_SIZE 2048
52 #define MAX_BITPOOL 64
53 #define MIN_BITPOOL 2
54 #define SOL_SCO 17
55 #define SCO_TXBUFS 0x03
56 #define SCO_RXBUFS 0x04
58 PA_MODULE_AUTHOR("Joao Paulo Rechi Vita");
59 PA_MODULE_DESCRIPTION("Bluetooth audio sink and source");
60 PA_MODULE_VERSION(PACKAGE_VERSION);
61 PA_MODULE_LOAD_ONCE(FALSE);
62 PA_MODULE_USAGE(
63 "name=<name of the device> "
64 "addr=<address of the device> "
65 "profile=<a2dp|hsp>");
67 struct bt_a2dp {
68 sbc_capabilities_t sbc_capabilities;
69 sbc_t sbc; /* Codec data */
70 pa_bool_t sbc_initialized; /* Keep track if the encoder is initialized */
71 int codesize; /* SBC codesize */
72 int samples; /* Number of encoded samples */
73 uint8_t buffer[BUFFER_SIZE]; /* Codec transfer buffer */
74 int count; /* Codec transfer buffer counter */
76 int nsamples; /* Cumulative number of codec samples */
77 uint16_t seq_num; /* Cumulative packet sequence */
78 int frame_count; /* Current frames in buffer*/
81 struct userdata {
82 pa_core *core;
83 pa_module *module;
84 pa_sink *sink;
86 pa_thread_mq thread_mq;
87 pa_rtpoll *rtpoll;
88 pa_rtpoll_item *rtpoll_item;
89 pa_thread *thread;
91 int64_t offset;
92 pa_smoother *smoother;
94 pa_memchunk memchunk;
95 pa_mempool *mempool;
97 const char *name;
98 const char *addr;
99 const char *profile;
100 int rate;
101 int channels;
102 pa_sample_spec ss;
104 int audioservice_fd;
105 int stream_fd;
107 int transport;
108 int link_mtu;
109 size_t block_size;
110 pa_usec_t latency;
112 struct bt_a2dp a2dp;
115 static const char* const valid_modargs[] = {
116 "name",
117 "addr",
118 "profile",
119 "rate",
120 "channels",
121 NULL
124 static int bt_audioservice_send(int sk, const bt_audio_msg_header_t *msg) {
125 int e;
126 pa_log_debug("sending %s", bt_audio_strmsg(msg->msg_type));
127 if (send(sk, msg, BT_AUDIO_IPC_PACKET_SIZE, 0) > 0)
128 e = 0;
129 else {
130 e = -errno;
131 pa_log_error("Error sending data to audio service: %s(%d)", pa_cstrerror(errno), errno);
133 return e;
136 static int bt_audioservice_recv(int sk, bt_audio_msg_header_t *inmsg) {
137 int e;
138 const char *type;
140 pa_log_debug("trying to receive msg from audio service...");
141 if (recv(sk, inmsg, BT_AUDIO_IPC_PACKET_SIZE, 0) > 0) {
142 type = bt_audio_strmsg(inmsg->msg_type);
143 if (type) {
144 pa_log_debug("Received %s", type);
145 e = 0;
147 else {
148 e = -EINVAL;
149 pa_log_error("Bogus message type %d received from audio service", inmsg->msg_type);
152 else {
153 e = -errno;
154 pa_log_error("Error receiving data from audio service: %s(%d)", pa_cstrerror(errno), errno);
157 return e;
160 static int bt_audioservice_expect(int sk, bt_audio_msg_header_t *rsp_hdr, int expected_type) {
161 int e = bt_audioservice_recv(sk, rsp_hdr);
162 if (e == 0) {
163 if (rsp_hdr->msg_type != expected_type) {
164 e = -EINVAL;
165 pa_log_error("Bogus message %s received while %s was expected", bt_audio_strmsg(rsp_hdr->msg_type),
166 bt_audio_strmsg(expected_type));
169 return e;
172 static int bt_getcaps(struct userdata *u) {
173 int e;
174 char buf[BT_AUDIO_IPC_PACKET_SIZE];
175 bt_audio_rsp_msg_header_t *rsp_hdr = (void*) buf;
176 struct bt_getcapabilities_req *getcaps_req = (void*) buf;
177 struct bt_getcapabilities_rsp *getcaps_rsp = (void*) buf;
179 memset(getcaps_req, 0, BT_AUDIO_IPC_PACKET_SIZE);
180 getcaps_req->h.msg_type = BT_GETCAPABILITIES_REQ;
181 strncpy(getcaps_req->device, u->addr, 18);
182 if (strcasecmp(u->profile, "a2dp") == 0)
183 getcaps_req->transport = BT_CAPABILITIES_TRANSPORT_A2DP;
184 else if (strcasecmp(u->profile, "hsp") == 0)
185 getcaps_req->transport = BT_CAPABILITIES_TRANSPORT_SCO;
186 else {
187 pa_log_error("invalid profile argument: %s", u->profile);
188 return -1;
190 getcaps_req->flags = 0;
191 getcaps_req->flags |= BT_FLAG_AUTOCONNECT;
193 e = bt_audioservice_send(u->audioservice_fd, &getcaps_req->h);
194 if (e < 0) {
195 pa_log_error("failed to send GETCAPABILITIES_REQ");
196 return e;
199 e = bt_audioservice_expect(u->audioservice_fd, &rsp_hdr->msg_h, BT_GETCAPABILITIES_RSP);
200 if (e < 0) {
201 pa_log_error("failed to expect for GETCAPABILITIES_RSP");
202 return e;
204 if (rsp_hdr->posix_errno != 0) {
205 pa_log_error("BT_GETCAPABILITIES failed : %s (%d)", pa_cstrerror(rsp_hdr->posix_errno), rsp_hdr->posix_errno);
206 return -rsp_hdr->posix_errno;
209 if ((u->transport = getcaps_rsp->transport) == BT_CAPABILITIES_TRANSPORT_A2DP)
210 u->a2dp.sbc_capabilities = getcaps_rsp->sbc_capabilities;
212 return 0;
215 static uint8_t default_bitpool(uint8_t freq, uint8_t mode) {
216 switch (freq) {
217 case BT_SBC_SAMPLING_FREQ_16000:
218 case BT_SBC_SAMPLING_FREQ_32000:
219 return 53;
220 case BT_SBC_SAMPLING_FREQ_44100:
221 switch (mode) {
222 case BT_A2DP_CHANNEL_MODE_MONO:
223 case BT_A2DP_CHANNEL_MODE_DUAL_CHANNEL:
224 return 31;
225 case BT_A2DP_CHANNEL_MODE_STEREO:
226 case BT_A2DP_CHANNEL_MODE_JOINT_STEREO:
227 return 53;
228 default:
229 pa_log_warn("Invalid channel mode %u", mode);
230 return 53;
232 case BT_SBC_SAMPLING_FREQ_48000:
233 switch (mode) {
234 case BT_A2DP_CHANNEL_MODE_MONO:
235 case BT_A2DP_CHANNEL_MODE_DUAL_CHANNEL:
236 return 29;
237 case BT_A2DP_CHANNEL_MODE_STEREO:
238 case BT_A2DP_CHANNEL_MODE_JOINT_STEREO:
239 return 51;
240 default:
241 pa_log_warn("Invalid channel mode %u", mode);
242 return 51;
244 default:
245 pa_log_warn("Invalid sampling freq %u", freq);
246 return 53;
250 static int bt_a2dp_init(struct userdata *u) {
251 sbc_capabilities_t *cap = &u->a2dp.sbc_capabilities;
252 unsigned int max_bitpool, min_bitpool;
254 switch (u->rate) {
255 case 48000:
256 cap->frequency = BT_SBC_SAMPLING_FREQ_48000;
257 break;
258 case 44100:
259 cap->frequency = BT_SBC_SAMPLING_FREQ_44100;
260 break;
261 case 32000:
262 cap->frequency = BT_SBC_SAMPLING_FREQ_32000;
263 break;
264 case 16000:
265 cap->frequency = BT_SBC_SAMPLING_FREQ_16000;
266 break;
267 default:
268 pa_log_error("Rate %d not supported", u->rate);
269 return -1;
272 // if (cfg->has_channel_mode)
273 // cap->channel_mode = cfg->channel_mode;
274 // else
275 if (u->channels == 2) {
276 if (cap->channel_mode & BT_A2DP_CHANNEL_MODE_JOINT_STEREO)
277 cap->channel_mode = BT_A2DP_CHANNEL_MODE_JOINT_STEREO;
278 else if (cap->channel_mode & BT_A2DP_CHANNEL_MODE_STEREO)
279 cap->channel_mode = BT_A2DP_CHANNEL_MODE_STEREO;
280 else if (cap->channel_mode & BT_A2DP_CHANNEL_MODE_DUAL_CHANNEL)
281 cap->channel_mode = BT_A2DP_CHANNEL_MODE_DUAL_CHANNEL;
282 } else {
283 if (cap->channel_mode & BT_A2DP_CHANNEL_MODE_MONO)
284 cap->channel_mode = BT_A2DP_CHANNEL_MODE_MONO;
287 if (!cap->channel_mode) {
288 pa_log_error("No supported channel modes");
289 return -1;
292 // if (cfg->has_block_length)
293 // cap->block_length = cfg->block_length;
294 // else
295 if (cap->block_length & BT_A2DP_BLOCK_LENGTH_16)
296 cap->block_length = BT_A2DP_BLOCK_LENGTH_16;
297 else if (cap->block_length & BT_A2DP_BLOCK_LENGTH_12)
298 cap->block_length = BT_A2DP_BLOCK_LENGTH_12;
299 else if (cap->block_length & BT_A2DP_BLOCK_LENGTH_8)
300 cap->block_length = BT_A2DP_BLOCK_LENGTH_8;
301 else if (cap->block_length & BT_A2DP_BLOCK_LENGTH_4)
302 cap->block_length = BT_A2DP_BLOCK_LENGTH_4;
303 else {
304 pa_log_error("No supported block lengths");
305 return -1;
308 // if (cfg->has_subbands)
309 // cap->subbands = cfg->subbands;
310 if (cap->subbands & BT_A2DP_SUBBANDS_8)
311 cap->subbands = BT_A2DP_SUBBANDS_8;
312 else if (cap->subbands & BT_A2DP_SUBBANDS_4)
313 cap->subbands = BT_A2DP_SUBBANDS_4;
314 else {
315 pa_log_error("No supported subbands");
316 return -1;
319 // if (cfg->has_allocation_method)
320 // cap->allocation_method = cfg->allocation_method;
321 if (cap->allocation_method & BT_A2DP_ALLOCATION_LOUDNESS)
322 cap->allocation_method = BT_A2DP_ALLOCATION_LOUDNESS;
323 else if (cap->allocation_method & BT_A2DP_ALLOCATION_SNR)
324 cap->allocation_method = BT_A2DP_ALLOCATION_SNR;
326 // if (cfg->has_bitpool)
327 // min_bitpool = max_bitpool = cfg->bitpool;
328 // else {
329 min_bitpool = MAX(MIN_BITPOOL, cap->min_bitpool);
330 max_bitpool = MIN(default_bitpool(cap->frequency, cap->channel_mode), cap->max_bitpool);
331 // }
333 cap->min_bitpool = min_bitpool;
334 cap->max_bitpool = max_bitpool;
336 return 0;
339 static void bt_a2dp_setup(struct bt_a2dp *a2dp) {
340 sbc_capabilities_t active_capabilities = a2dp->sbc_capabilities;
342 if (a2dp->sbc_initialized)
343 sbc_reinit(&a2dp->sbc, 0);
344 else
345 sbc_init(&a2dp->sbc, 0);
346 a2dp->sbc_initialized = TRUE;
348 if (active_capabilities.frequency & BT_SBC_SAMPLING_FREQ_16000)
349 a2dp->sbc.frequency = SBC_FREQ_16000;
351 if (active_capabilities.frequency & BT_SBC_SAMPLING_FREQ_32000)
352 a2dp->sbc.frequency = SBC_FREQ_32000;
354 if (active_capabilities.frequency & BT_SBC_SAMPLING_FREQ_44100)
355 a2dp->sbc.frequency = SBC_FREQ_44100;
357 if (active_capabilities.frequency & BT_SBC_SAMPLING_FREQ_48000)
358 a2dp->sbc.frequency = SBC_FREQ_48000;
360 if (active_capabilities.channel_mode & BT_A2DP_CHANNEL_MODE_MONO)
361 a2dp->sbc.mode = SBC_MODE_MONO;
363 if (active_capabilities.channel_mode & BT_A2DP_CHANNEL_MODE_DUAL_CHANNEL)
364 a2dp->sbc.mode = SBC_MODE_DUAL_CHANNEL;
366 if (active_capabilities.channel_mode & BT_A2DP_CHANNEL_MODE_STEREO)
367 a2dp->sbc.mode = SBC_MODE_STEREO;
369 if (active_capabilities.channel_mode & BT_A2DP_CHANNEL_MODE_JOINT_STEREO)
370 a2dp->sbc.mode = SBC_MODE_JOINT_STEREO;
372 a2dp->sbc.allocation = (active_capabilities.allocation_method == BT_A2DP_ALLOCATION_SNR ? SBC_AM_SNR : SBC_AM_LOUDNESS);
374 switch (active_capabilities.subbands) {
375 case BT_A2DP_SUBBANDS_4:
376 a2dp->sbc.subbands = SBC_SB_4;
377 break;
378 case BT_A2DP_SUBBANDS_8:
379 a2dp->sbc.subbands = SBC_SB_8;
380 break;
383 switch (active_capabilities.block_length) {
384 case BT_A2DP_BLOCK_LENGTH_4:
385 a2dp->sbc.blocks = SBC_BLK_4;
386 break;
387 case BT_A2DP_BLOCK_LENGTH_8:
388 a2dp->sbc.blocks = SBC_BLK_8;
389 break;
390 case BT_A2DP_BLOCK_LENGTH_12:
391 a2dp->sbc.blocks = SBC_BLK_12;
392 break;
393 case BT_A2DP_BLOCK_LENGTH_16:
394 a2dp->sbc.blocks = SBC_BLK_16;
395 break;
398 a2dp->sbc.bitpool = active_capabilities.max_bitpool;
399 a2dp->codesize = sbc_get_codesize(&a2dp->sbc);
400 a2dp->count = sizeof(struct rtp_header) + sizeof(struct rtp_payload);
403 static int bt_setconf(struct userdata *u) {
404 int e;
405 char buf[BT_AUDIO_IPC_PACKET_SIZE];
406 bt_audio_rsp_msg_header_t *rsp_hdr = (void*) buf;
407 struct bt_setconfiguration_req *setconf_req = (void*) buf;
408 struct bt_setconfiguration_rsp *setconf_rsp = (void*) buf;
410 if (u->transport == BT_CAPABILITIES_TRANSPORT_A2DP) {
411 e = bt_a2dp_init(u);
412 if (e < 0) {
413 pa_log_error("a2dp_init error");
414 return e;
416 u->ss.format = PA_SAMPLE_S16LE;
418 else
419 u->ss.format = PA_SAMPLE_U8;
421 u->ss.rate = u->rate;
422 u->ss.channels = u->channels;
424 memset(setconf_req, 0, BT_AUDIO_IPC_PACKET_SIZE);
425 setconf_req->h.msg_type = BT_SETCONFIGURATION_REQ;
426 strncpy(setconf_req->device, u->addr, 18);
427 setconf_req->transport = u->transport;
428 if (u->transport == BT_CAPABILITIES_TRANSPORT_A2DP)
429 setconf_req->sbc_capabilities = u->a2dp.sbc_capabilities;
430 setconf_req->access_mode = BT_CAPABILITIES_ACCESS_MODE_WRITE;
432 e = bt_audioservice_send(u->audioservice_fd, &setconf_req->h);
433 if (e < 0) {
434 pa_log_error("failed to send BT_SETCONFIGURATION_REQ");
435 return e;
438 e = bt_audioservice_expect(u->audioservice_fd, &rsp_hdr->msg_h, BT_SETCONFIGURATION_RSP);
439 if (e < 0) {
440 pa_log_error("failed to expect BT_SETCONFIGURATION_RSP");
441 return e;
444 if (rsp_hdr->posix_errno != 0) {
445 pa_log_error("BT_SETCONFIGURATION failed : %s(%d)", pa_cstrerror(rsp_hdr->posix_errno), rsp_hdr->posix_errno);
446 return -rsp_hdr->posix_errno;
449 u->transport = setconf_rsp->transport;
450 u->link_mtu = setconf_rsp->link_mtu;
452 /* setup SBC encoder now we agree on parameters */
453 if (u->transport == BT_CAPABILITIES_TRANSPORT_A2DP) {
454 bt_a2dp_setup(&u->a2dp);
455 u->block_size = u->a2dp.codesize;
456 pa_log_info("sbc parameters:\n\tallocation=%u\n\tsubbands=%u\n\tblocks=%u\n\tbitpool=%u\n",
457 u->a2dp.sbc.allocation, u->a2dp.sbc.subbands, u->a2dp.sbc.blocks, u->a2dp.sbc.bitpool);
459 else
460 u->block_size = u->link_mtu;
462 return 0;
465 static int bt_getstreamfd(struct userdata *u) {
466 int e;
467 // uint32_t period_count = io->buffer_size / io->period_size;
468 char buf[BT_AUDIO_IPC_PACKET_SIZE];
469 struct bt_streamstart_req *start_req = (void*) buf;
470 bt_audio_rsp_msg_header_t *rsp_hdr = (void*) buf;
471 struct bt_streamfd_ind *streamfd_ind = (void*) buf;
473 memset(start_req, 0, BT_AUDIO_IPC_PACKET_SIZE);
474 start_req->h.msg_type = BT_STREAMSTART_REQ;
476 e = bt_audioservice_send(u->audioservice_fd, &start_req->h);
477 if (e < 0) {
478 pa_log_error("failed to send BT_STREAMSTART_REQ");
479 return e;
482 e = bt_audioservice_expect(u->audioservice_fd, &rsp_hdr->msg_h, BT_STREAMSTART_RSP);
483 if (e < 0) {
484 pa_log_error("failed to expect BT_STREAMSTART_RSP");
485 return e;
488 if (rsp_hdr->posix_errno != 0) {
489 pa_log_error("BT_START failed : %s(%d)", pa_cstrerror(rsp_hdr->posix_errno), rsp_hdr->posix_errno);
490 return -rsp_hdr->posix_errno;
493 e = bt_audioservice_expect(u->audioservice_fd, &streamfd_ind->h, BT_STREAMFD_IND);
494 if (e < 0) {
495 pa_log_error("failed to expect BT_STREAMFD_IND");
496 return e;
499 if (u->stream_fd >= 0)
500 pa_close(u->stream_fd);
502 u->stream_fd = bt_audio_service_get_data_fd(u->audioservice_fd);
503 if (u->stream_fd < 0) {
504 pa_log_error("failed to get data fd: %s (%d)",pa_cstrerror(errno), errno);
505 return -errno;
508 if (u->transport == BT_CAPABILITIES_TRANSPORT_A2DP) {
509 if (pa_socket_set_sndbuf(u->stream_fd, 10*u->link_mtu) < 0) {
510 pa_log_error("failed to set socket options for A2DP: %s (%d)",pa_cstrerror(errno), errno);
511 return -errno;
515 // if (setsockopt(u->stream_fd, SOL_SCO, SCO_TXBUFS, &period_count, sizeof(period_count)) == 0)
516 // return 0;
517 // if (setsockopt(u->stream_fd, SOL_SCO, SO_SNDBUF, &period_count, sizeof(period_count)) == 0)
518 // return 0;
519 // /* FIXME : handle error codes */
520 pa_make_fd_nonblock(u->stream_fd);
521 // pa_make_socket_low_delay(u->stream_fd);
523 return 0;
526 static int sink_process_msg(pa_msgobject *o, int code, void *data, int64_t offset, pa_memchunk *chunk) {
527 struct userdata *u = PA_SINK(o)->userdata;
529 pa_log_debug("got message: %d", code);
530 switch (code) {
532 case PA_SINK_MESSAGE_SET_STATE:
533 switch ((pa_sink_state_t) PA_PTR_TO_UINT(data)) {
534 case PA_SINK_SUSPENDED:
535 pa_assert(PA_SINK_IS_OPENED(u->sink->thread_info.state));
536 pa_smoother_pause(u->smoother, pa_rtclock_usec());
537 break;
538 case PA_SINK_IDLE:
539 case PA_SINK_RUNNING:
540 if (u->sink->thread_info.state == PA_SINK_SUSPENDED)
541 pa_smoother_resume(u->smoother, pa_rtclock_usec());
542 break;
543 case PA_SINK_UNLINKED:
544 case PA_SINK_INIT:
547 break;
549 case PA_SINK_MESSAGE_GET_LATENCY: {
550 pa_usec_t w, r;
551 r = pa_smoother_get(u->smoother, pa_rtclock_usec());
552 w = pa_bytes_to_usec(u->offset + u->memchunk.length, &u->sink->sample_spec);
553 *((pa_usec_t*) data) = w > r ? w - r : 0;
554 return 0;
559 return pa_sink_process_msg(o, code, data, offset, chunk);
562 static void sco_thread_func(void *userdata) {
563 struct userdata *u = userdata;
564 int write_type = 0;
566 pa_assert(u);
568 pa_log_debug("SCO thread starting up");
570 pa_thread_mq_install(&u->thread_mq);
571 pa_rtpoll_install(u->rtpoll);
573 pa_smoother_set_time_offset(u->smoother, pa_rtclock_usec());
575 for (;;) {
576 int ret;
577 struct pollfd *pollfd;
579 if (PA_SINK_IS_OPENED(u->sink->thread_info.state)) {
580 if (u->sink->thread_info.rewind_requested) {
581 pa_sink_process_rewind(u->sink, 0);
585 pollfd = pa_rtpoll_item_get_pollfd(u->rtpoll_item, NULL);
587 if (PA_SINK_IS_OPENED(u->sink->thread_info.state) && pollfd->revents) {
588 pa_usec_t usec;
589 int64_t n;
591 for (;;) {
592 /* Render some data and write it to the fifo */
593 ssize_t l;
594 void *p;
596 u->memchunk.memblock = pa_memblock_new(u->mempool, u->block_size);
597 pa_log_debug("memblock asked size %d", u->block_size);
598 u->memchunk.length = pa_memblock_get_length(u->memchunk.memblock);
599 pa_log_debug("memchunk length %d", u->memchunk.length);
600 pa_sink_render_into_full(u->sink, &u->memchunk);
602 pa_assert(u->memchunk.length > 0);
604 p = pa_memblock_acquire(u->memchunk.memblock);
606 l = pa_write(u->stream_fd, (uint8_t*) p, u->memchunk.length, &write_type);
607 pa_log_debug("memblock written to socket: %d bytes", l);
608 pa_memblock_release(u->memchunk.memblock);
609 pa_memblock_unref(u->memchunk.memblock);
610 pa_memchunk_reset(&u->memchunk);
611 pa_log("memchunk reseted");
613 pa_assert(l != 0);
615 if (l < 0) {
616 if (errno == EINTR) {
617 pa_log_debug("EINTR");
618 continue;
620 else if (errno == EAGAIN) {
621 pa_log_debug("EAGAIN");
622 goto filled_up;
624 else {
625 pa_log("Failed to write data to FIFO: %s", pa_cstrerror(errno));
626 goto fail;
628 } else {
629 u->offset += l;
630 pollfd->revents = 0;
634 filled_up:
635 n = u->offset;
636 #ifdef SIOCOUTQ
638 int l;
639 if (ioctl(u->fd, SIOCOUTQ, &l) >= 0 && l > 0)
640 n -= l;
642 #endif
643 usec = pa_bytes_to_usec(n, &u->sink->sample_spec);
644 if (usec > u->latency)
645 usec -= u->latency;
646 else
647 usec = 0;
648 pa_smoother_put(u->smoother, pa_rtclock_usec(), usec);
651 /* Hmm, nothing to do. Let's sleep */
652 pa_log_debug("SCO thread going to sleep");
653 pollfd->events = PA_SINK_IS_OPENED(u->sink->thread_info.state) ? POLLOUT : 0;
654 if ((ret = pa_rtpoll_run(u->rtpoll, TRUE)) < 0) {
655 pa_log("rtpoll_run < 0");
656 goto fail;
658 pa_log_debug("SCO thread waking up");
660 if (ret == 0) {
661 pa_log_debug("rtpoll_run == 0");
662 goto finish;
665 pollfd = pa_rtpoll_item_get_pollfd(u->rtpoll_item, NULL);
666 if (pollfd->revents & ~POLLOUT) {
667 pa_log_error("FIFO shutdown.");
668 goto fail;
672 fail:
673 /* If this was no regular exit from the loop we have to continue processing messages until we receive PA_MESSAGE_SHUTDOWN */
674 pa_log_debug("SCO thread failed");
675 pa_asyncmsgq_post(u->thread_mq.outq, PA_MSGOBJECT(u->core), PA_CORE_MESSAGE_UNLOAD_MODULE, u->module, 0, NULL, NULL);
676 pa_asyncmsgq_wait_for(u->thread_mq.inq, PA_MESSAGE_SHUTDOWN);
678 finish:
679 pa_log_debug("SCO thread shutting down");
682 static void a2dp_thread_func(void *userdata) {
683 struct userdata *u = userdata;
685 pa_assert(u);
687 pa_log_debug("A2DP Thread starting up");
689 pa_thread_mq_install(&u->thread_mq);
690 pa_rtpoll_install(u->rtpoll);
692 pa_smoother_set_time_offset(u->smoother, pa_rtclock_usec());
694 for (;;) {
695 int ret;
696 struct pollfd *pollfd;
698 if (PA_SINK_IS_OPENED(u->sink->thread_info.state)) {
699 if (u->sink->thread_info.rewind_requested) {
700 pa_sink_process_rewind(u->sink, 0);
704 pollfd = pa_rtpoll_item_get_pollfd(u->rtpoll_item, NULL);
706 if (PA_SINK_IS_OPENED(u->sink->thread_info.state) && pollfd->revents) {
707 pa_usec_t usec;
708 int64_t n;
709 ssize_t l;
710 int write_type = 0, written;
711 struct bt_a2dp *a2dp = &u->a2dp;
712 struct rtp_header *header = (void *) a2dp->buffer;
713 struct rtp_payload *payload = (void *) (a2dp->buffer + sizeof(*header));
715 do {
716 /* Render some data */
717 int frame_size, encoded;
718 void *p;
720 u->memchunk.memblock = pa_memblock_new(u->mempool, u->block_size);
721 pa_log_debug("memblock asked size %d", u->block_size);
722 u->memchunk.length = pa_memblock_get_length(u->memchunk.memblock);
723 pa_log_debug("memchunk length %d", u->memchunk.length);
724 pa_sink_render_into_full(u->sink, &u->memchunk);
726 pa_assert(u->memchunk.length > 0);
728 p = pa_memblock_acquire(u->memchunk.memblock);
729 frame_size = sbc_get_frame_length(&a2dp->sbc);
730 pa_log_debug("SBC frame_size: %d", frame_size);
732 encoded = sbc_encode(&a2dp->sbc, (uint8_t*) p, a2dp->codesize, a2dp->buffer + a2dp->count,
733 sizeof(a2dp->buffer) - a2dp->count, &written);
734 pa_log_debug("SBC: encoded: %d; written: %d", encoded, written);
735 if (encoded <= 0) {
736 pa_log_error("SBC encoding error (%d)", encoded);
737 goto fail;
739 pa_memblock_release(u->memchunk.memblock);
740 pa_memblock_unref(u->memchunk.memblock);
741 pa_memchunk_reset(&u->memchunk);
742 pa_log_debug("memchunk reseted");
744 a2dp->count += written;
745 a2dp->frame_count++;
746 a2dp->samples += encoded / frame_size;
747 a2dp->nsamples += encoded / frame_size;
749 } while (a2dp->count + written <= u->link_mtu);
751 /* write it to the fifo */
752 memset(a2dp->buffer, 0, sizeof(*header) + sizeof(*payload));
753 payload->frame_count = a2dp->frame_count;
754 header->v = 2;
755 header->pt = 1;
756 header->sequence_number = htons(a2dp->seq_num);
757 header->timestamp = htonl(a2dp->nsamples);
758 header->ssrc = htonl(1);
760 avdtp_write:
761 l = pa_write(u->stream_fd, a2dp->buffer, a2dp->count, write_type);
762 pa_log_debug("avdtp_write: requested %d bytes; written %d bytes", a2dp->count, l);
764 pa_assert(l != 0);
766 if (l < 0) {
767 if (errno == EINTR) {
768 pa_log_debug("EINTR");
769 continue;
771 else if (errno == EAGAIN) {
772 pa_log_debug("EAGAIN");
773 goto avdtp_write;
775 else {
776 pa_log_error("Failed to write data to FIFO: %s", pa_cstrerror(errno));
777 goto fail;
781 u->offset += a2dp->codesize*a2dp->frame_count;
782 pollfd->revents = 0;
784 /* Reset buffer of data to send */
785 a2dp->count = sizeof(struct rtp_header) + sizeof(struct rtp_payload);
786 a2dp->frame_count = 0;
787 a2dp->samples = 0;
788 a2dp->seq_num++;
790 /* feed the time smoother */
791 n = u->offset;
792 #ifdef SIOCOUTQ
794 int ll;
795 if (ioctl(u->fd, SIOCOUTQ, &ll) >= 0 && ll > 0)
796 n -= ll;
798 #endif
799 usec = pa_bytes_to_usec(n, &u->sink->sample_spec);
800 if (usec > u->latency)
801 usec -= u->latency;
802 else
803 usec = 0;
804 pa_smoother_put(u->smoother, pa_rtclock_usec(), usec);
807 /* Hmm, nothing to do. Let's sleep */
808 pa_log_debug("A2DP thread going to sleep");
809 pollfd->events = PA_SINK_IS_OPENED(u->sink->thread_info.state) ? POLLOUT : 0;
810 if ((ret = pa_rtpoll_run(u->rtpoll, TRUE)) < 0) {
811 pa_log_error("rtpoll_run < 0");
812 goto fail;
814 pa_log_debug("A2DP thread waking up");
816 if (ret == 0) {
817 pa_log_debug("rtpoll_run == 0");
818 goto finish;
821 pollfd = pa_rtpoll_item_get_pollfd(u->rtpoll_item, NULL);
822 if (pollfd->revents & ~POLLOUT) {
823 pa_log_error("FIFO shutdown.");
824 goto fail;
828 fail:
829 /* If this was no regular exit from the loop we have to continue processing messages until we receive PA_MESSAGE_SHUTDOWN */
830 pa_log_debug("A2DP thread failed");
831 pa_asyncmsgq_post(u->thread_mq.outq, PA_MSGOBJECT(u->core), PA_CORE_MESSAGE_UNLOAD_MODULE, u->module, 0, NULL, NULL);
832 pa_asyncmsgq_wait_for(u->thread_mq.inq, PA_MESSAGE_SHUTDOWN);
834 finish:
835 pa_log_debug("A2DP thread shutting down");
838 int pa__init(pa_module* m) {
839 int e;
840 const char *rate, *channels;
841 pa_modargs *ma;
842 pa_sink_new_data data;
843 struct pollfd *pollfd;
844 struct userdata *u;
846 pa_assert(m);
847 m->userdata = u = pa_xnew0(struct userdata, 1);
848 u->module = m;
849 u->core = m->core;
850 u->audioservice_fd = -1;
851 u->stream_fd = -1;
852 u->transport = -1;
853 u->offset = 0;
854 u->latency = 0;
855 u->a2dp.sbc_initialized = FALSE;
856 u->smoother = pa_smoother_new(PA_USEC_PER_SEC, PA_USEC_PER_SEC*2, TRUE, 10);
857 u->mempool = pa_mempool_new(FALSE);
858 pa_memchunk_reset(&u->memchunk);
859 u->rtpoll = pa_rtpoll_new();
860 pa_thread_mq_init(&u->thread_mq, u->core->mainloop, u->rtpoll);
861 u->rtpoll_item = NULL;
863 if (!(ma = pa_modargs_new(m->argument, valid_modargs))) {
864 pa_log_error("failed to parse module arguments");
865 goto fail;
867 if (!(u->name = pa_xstrdup(pa_modargs_get_value(ma, "name", DEFAULT_SINK_NAME)))) {
868 pa_log_error("failed to get device name from module arguments");
869 goto fail;
871 if (!(u->addr = pa_xstrdup(pa_modargs_get_value(ma, "addr", NULL)))) {
872 pa_log_error("failed to get device address from module arguments");
873 goto fail;
875 if (!(u->profile = pa_xstrdup(pa_modargs_get_value(ma, "profile", NULL)))) {
876 pa_log_error("failed to get profile from module arguments");
877 goto fail;
879 if (pa_modargs_get_value_u32(ma, "rate", &u->rate) < 0) {
880 pa_log_error("failed to get rate from module arguments");
881 goto fail;
883 if (pa_modargs_get_value_u32(ma, "channels", &u->channels) < 0) {
884 pa_log_error("failed to get channels from module arguments");
885 goto fail;
888 /* connect to the bluez audio service */
889 u->audioservice_fd = bt_audio_service_open();
890 if (u->audioservice_fd <= 0) {
891 pa_log_error("couldn't connect to bluetooth audio service");
892 goto fail;
894 pa_log_debug("connected to the bluetooth audio service");
896 /* queries device capabilities */
897 e = bt_getcaps(u);
898 if (e < 0) {
899 pa_log_error("failed to get device capabilities");
900 goto fail;
902 pa_log_debug("got device capabilities");
904 /* configures the connection */
905 e = bt_setconf(u);
906 if (e < 0) {
907 pa_log_error("failed to set config");
908 goto fail;
910 pa_log_debug("connection to the device configured");
912 /* gets the device socket */
913 e = bt_getstreamfd(u);
914 if (e < 0) {
915 pa_log_error("failed to get stream fd (%d)", e);
916 goto fail;
918 pa_log_debug("got the device socket");
920 /* create sink */
921 pa_sink_new_data_init(&data);
922 data.driver = __FILE__;
923 data.module = m;
924 pa_sink_new_data_set_name(&data, u->name);
925 pa_sink_new_data_set_sample_spec(&data, &u->ss);
926 pa_proplist_sets(data.proplist, PA_PROP_DEVICE_STRING, u->name);
927 pa_proplist_setf(data.proplist, PA_PROP_DEVICE_DESCRIPTION, "Bluetooth sink '%s' (%s)", u->name, u->addr);
928 u->sink = pa_sink_new(m->core, &data, PA_SINK_HARDWARE|PA_SINK_LATENCY);
929 pa_sink_new_data_done(&data);
930 if (!u->sink) {
931 pa_log_error("failed to create sink");
932 goto fail;
934 u->sink->userdata = u;
935 u->sink->parent.process_msg = sink_process_msg;
936 pa_sink_set_asyncmsgq(u->sink, u->thread_mq.inq);
937 pa_sink_set_rtpoll(u->sink, u->rtpoll);
939 u->rtpoll_item = pa_rtpoll_item_new(u->rtpoll, PA_RTPOLL_NEVER, 1);
940 pollfd = pa_rtpoll_item_get_pollfd(u->rtpoll_item, NULL);
941 pollfd->fd = u->stream_fd;
942 pollfd->events = pollfd->revents = 0;
944 /* start rt thread */
945 if (u->transport == BT_CAPABILITIES_TRANSPORT_A2DP) {
946 if (!(u->thread = pa_thread_new(a2dp_thread_func, u))) {
947 pa_log_error("failed to create A2DP thread");
948 goto fail;
951 else {
952 if (!(u->thread = pa_thread_new(sco_thread_func, u))) {
953 pa_log_error("failed to create SCO thread");
954 goto fail;
957 pa_sink_put(u->sink);
959 pa_modargs_free(ma);
960 return 0;
962 fail:
963 if (ma)
964 pa_modargs_free(ma);
965 pa__done(m);
966 return -1;
969 void pa__done(pa_module *m) {
970 struct userdata *u;
971 pa_assert(m);
973 if (!(u = m->userdata))
974 return;
976 if (u->sink)
977 pa_sink_unlink(u->sink);
979 if (u->thread) {
980 pa_asyncmsgq_send(u->thread_mq.inq, NULL, PA_MESSAGE_SHUTDOWN, NULL, 0, NULL);
981 pa_thread_free(u->thread);
984 pa_thread_mq_done(&u->thread_mq);
986 if (u->sink)
987 pa_sink_unref(u->sink);
989 if (u->rtpoll_item)
990 pa_rtpoll_item_free(u->rtpoll_item);
992 if (u->rtpoll)
993 pa_rtpoll_free(u->rtpoll);
995 if (u->memchunk.memblock)
996 pa_memblock_unref(u->memchunk.memblock);
998 if (u->smoother)
999 pa_smoother_free(u->smoother);
1001 if (u->stream_fd >= 0)
1002 pa_close(u->stream_fd);
1004 if (u->audioservice_fd >= 0)
1005 pa_close(u->audioservice_fd);
1007 pa_xfree(u);