4 This file is part of polypaudio.
6 polypaudio 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 polypaudio 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 polypaudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
30 #include <sys/types.h>
37 #ifdef HAVE_SYS_WAIT_H
41 #ifdef HAVE_SYS_SOCKET_H
42 #include <sys/socket.h>
48 #include "../polypcore/winsock.h"
50 #include <polyp/version.h>
52 #include <polypcore/native-common.h>
53 #include <polypcore/pdispatch.h>
54 #include <polypcore/pstream.h>
55 #include <polypcore/dynarray.h>
56 #include <polypcore/socket-client.h>
57 #include <polypcore/pstream-util.h>
58 #include <polypcore/util.h>
59 #include <polypcore/xmalloc.h>
60 #include <polypcore/log.h>
61 #include <polypcore/socket-util.h>
65 #include "client-conf.h"
68 #include "client-conf-x11.h"
73 #define AUTOSPAWN_LOCK "autospawn.lock"
75 static const pa_pdispatch_cb_t command_table[PA_COMMAND_MAX] = {
76 [PA_COMMAND_REQUEST] = pa_command_request,
77 [PA_COMMAND_OVERFLOW] = pa_command_overflow_or_underflow,
78 [PA_COMMAND_UNDERFLOW] = pa_command_overflow_or_underflow,
79 [PA_COMMAND_PLAYBACK_STREAM_KILLED] = pa_command_stream_killed,
80 [PA_COMMAND_RECORD_STREAM_KILLED] = pa_command_stream_killed,
81 [PA_COMMAND_SUBSCRIBE_EVENT] = pa_command_subscribe_event
84 static void unlock_autospawn_lock_file(pa_context *c) {
87 if (c->autospawn_lock_fd >= 0) {
89 pa_runtime_path(AUTOSPAWN_LOCK, lf, sizeof(lf));
91 pa_unlock_lockfile(lf, c->autospawn_lock_fd);
92 c->autospawn_lock_fd = -1;
96 pa_context *pa_context_new(pa_mainloop_api *mainloop, const char *name) {
102 c = pa_xnew(pa_context, 1);
104 c->name = pa_xstrdup(name);
105 c->mainloop = mainloop;
109 c->playback_streams = pa_dynarray_new();
110 c->record_streams = pa_dynarray_new();
112 PA_LLIST_HEAD_INIT(pa_stream, c->streams);
113 PA_LLIST_HEAD_INIT(pa_operation, c->operations);
116 c->state = PA_CONTEXT_UNCONNECTED;
120 c->state_callback = NULL;
121 c->state_userdata = NULL;
123 c->subscribe_callback = NULL;
124 c->subscribe_userdata = NULL;
126 c->memblock_stat = pa_memblock_stat_new();
128 c->server_list = NULL;
130 c->autospawn_lock_fd = -1;
131 memset(&c->spawn_api, 0, sizeof(c->spawn_api));
135 pa_check_signal_is_blocked(SIGPIPE);
138 c->conf = pa_client_conf_new();
139 pa_client_conf_load(c->conf, NULL);
141 pa_client_conf_from_x11(c->conf, NULL);
143 pa_client_conf_env(c->conf);
148 static void context_free(pa_context *c) {
151 unlock_autospawn_lock_file(c);
153 while (c->operations)
154 pa_operation_cancel(c->operations);
157 pa_stream_set_state(c->streams, PA_STREAM_TERMINATED);
160 pa_socket_client_unref(c->client);
162 pa_pdispatch_unref(c->pdispatch);
164 pa_pstream_close(c->pstream);
165 pa_pstream_unref(c->pstream);
168 if (c->record_streams)
169 pa_dynarray_free(c->record_streams, NULL, NULL);
170 if (c->playback_streams)
171 pa_dynarray_free(c->playback_streams, NULL, NULL);
173 pa_memblock_stat_unref(c->memblock_stat);
176 pa_client_conf_free(c->conf);
178 pa_strlist_free(c->server_list);
185 pa_context* pa_context_ref(pa_context *c) {
193 void pa_context_unref(pa_context *c) {
201 void pa_context_set_state(pa_context *c, pa_context_state_t st) {
210 if (st == PA_CONTEXT_FAILED || st == PA_CONTEXT_TERMINATED) {
213 s = c->streams ? pa_stream_ref(c->streams) : NULL;
215 pa_stream *n = s->next ? pa_stream_ref(s->next) : NULL;
216 pa_stream_set_state(s, st == PA_CONTEXT_FAILED ? PA_STREAM_FAILED : PA_STREAM_TERMINATED);
222 pa_pdispatch_unref(c->pdispatch);
226 pa_pstream_close(c->pstream);
227 pa_pstream_unref(c->pstream);
232 pa_socket_client_unref(c->client);
237 if (c->state_callback)
238 c->state_callback(c, c->state_userdata);
243 void pa_context_fail(pa_context *c, int error) {
247 pa_context_set_error(c, error);
248 pa_context_set_state(c, PA_CONTEXT_FAILED);
251 int pa_context_set_error(pa_context *c, int error) {
253 assert(error < PA_ERR_MAX);
261 static void pstream_die_callback(pa_pstream *p, void *userdata) {
262 pa_context *c = userdata;
267 pa_context_fail(c, PA_ERR_CONNECTIONTERMINATED);
270 static void pstream_packet_callback(pa_pstream *p, pa_packet *packet, const void *creds, void *userdata) {
271 pa_context *c = userdata;
279 if (pa_pdispatch_run(c->pdispatch, packet, creds, c) < 0)
280 pa_context_fail(c, PA_ERR_PROTOCOL);
285 static void pstream_memblock_callback(pa_pstream *p, uint32_t channel, int64_t offset, pa_seek_mode_t seek, const pa_memchunk *chunk, void *userdata) {
286 pa_context *c = userdata;
291 assert(chunk->memblock);
292 assert(chunk->length);
298 if ((s = pa_dynarray_get(c->record_streams, channel))) {
300 pa_memblockq_seek(s->record_memblockq, offset, seek);
301 pa_memblockq_push_align(s->record_memblockq, chunk);
303 if (s->read_callback) {
306 if ((l = pa_memblockq_get_length(s->record_memblockq)) > 0)
307 s->read_callback(s, l, s->read_userdata);
314 int pa_context_handle_error(pa_context *c, uint32_t command, pa_tagstruct *t) {
318 if (command == PA_COMMAND_ERROR) {
321 if (pa_tagstruct_getu32(t, &c->error) < 0) {
322 pa_context_fail(c, PA_ERR_PROTOCOL);
326 } else if (command == PA_COMMAND_TIMEOUT)
327 c->error = PA_ERR_TIMEOUT;
329 pa_context_fail(c, PA_ERR_PROTOCOL);
336 static void setup_complete_callback(pa_pdispatch *pd, uint32_t command, uint32_t tag, pa_tagstruct *t, void *userdata) {
337 pa_context *c = userdata;
341 assert(c->state == PA_CONTEXT_AUTHORIZING || c->state == PA_CONTEXT_SETTING_NAME);
345 if (command != PA_COMMAND_REPLY) {
347 if (pa_context_handle_error(c, command, t) < 0)
348 pa_context_fail(c, PA_ERR_PROTOCOL);
350 pa_context_fail(c, c->error);
355 case PA_CONTEXT_AUTHORIZING: {
358 if (pa_tagstruct_getu32(t, &c->version) < 0 ||
359 !pa_tagstruct_eof(t)) {
360 pa_context_fail(c, PA_ERR_PROTOCOL);
364 /* Minimum supported version */
365 if (c->version < 8) {
366 pa_context_fail(c, PA_ERR_VERSION);
370 reply = pa_tagstruct_command(c, PA_COMMAND_SET_CLIENT_NAME, &tag);
371 pa_tagstruct_puts(reply, c->name);
372 pa_pstream_send_tagstruct(c->pstream, reply);
373 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, setup_complete_callback, c);
375 pa_context_set_state(c, PA_CONTEXT_SETTING_NAME);
379 case PA_CONTEXT_SETTING_NAME :
380 pa_context_set_state(c, PA_CONTEXT_READY);
391 static void setup_context(pa_context *c, pa_iochannel *io) {
401 c->pstream = pa_pstream_new(c->mainloop, io, c->memblock_stat);
403 pa_pstream_set_die_callback(c->pstream, pstream_die_callback, c);
404 pa_pstream_set_recieve_packet_callback(c->pstream, pstream_packet_callback, c);
405 pa_pstream_set_recieve_memblock_callback(c->pstream, pstream_memblock_callback, c);
407 assert(!c->pdispatch);
408 c->pdispatch = pa_pdispatch_new(c->mainloop, command_table, PA_COMMAND_MAX);
410 if (!c->conf->cookie_valid) {
411 pa_context_fail(c, PA_ERR_AUTHKEY);
415 t = pa_tagstruct_command(c, PA_COMMAND_AUTH, &tag);
416 pa_tagstruct_putu32(t, PA_PROTOCOL_VERSION);
417 pa_tagstruct_put_arbitrary(t, c->conf->cookie, sizeof(c->conf->cookie));
418 pa_pstream_send_tagstruct_with_creds(c->pstream, t, 1);
419 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, setup_complete_callback, c);
421 pa_context_set_state(c, PA_CONTEXT_AUTHORIZING);
428 static void on_connection(pa_socket_client *client, pa_iochannel*io, void *userdata);
432 static int context_connect_spawn(pa_context *c) {
435 int fds[2] = { -1, -1} ;
440 if (socketpair(AF_UNIX, SOCK_STREAM, 0, fds) < 0) {
441 pa_log(__FILE__": socketpair() failed: %s", strerror(errno));
442 pa_context_fail(c, PA_ERR_INTERNAL);
446 pa_fd_set_cloexec(fds[0], 1);
448 pa_socket_low_delay(fds[0]);
449 pa_socket_low_delay(fds[1]);
451 if (c->spawn_api.prefork)
452 c->spawn_api.prefork();
454 if ((pid = fork()) < 0) {
455 pa_log(__FILE__": fork() failed: %s", strerror(errno));
456 pa_context_fail(c, PA_ERR_INTERNAL);
458 if (c->spawn_api.postfork)
459 c->spawn_api.postfork();
466 const char *state = NULL;
468 const char * argv[MAX_ARGS+1];
471 /* Not required, since fds[0] has CLOEXEC enabled anyway */
474 if (c->spawn_api.atfork)
475 c->spawn_api.atfork();
481 argv[n++] = c->conf->daemon_binary;
482 argv[n++] = "--daemonize=yes";
484 snprintf(t, sizeof(t), "-Lmodule-native-protocol-fd fd=%i", fds[1]);
485 argv[n++] = strdup(t);
487 while (n < MAX_ARGS) {
490 if (!(a = pa_split_spaces(c->conf->extra_arguments, &state)))
498 execv(argv[0], (char * const *) argv);
505 r = waitpid(pid, &status, 0);
507 if (c->spawn_api.postfork)
508 c->spawn_api.postfork();
511 pa_log(__FILE__": waitpid() failed: %s", strerror(errno));
512 pa_context_fail(c, PA_ERR_INTERNAL);
514 } else if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
515 pa_context_fail(c, PA_ERR_CONNECTIONREFUSED);
523 io = pa_iochannel_new(c->mainloop, fds[0], fds[0]);
525 setup_context(c, io);
526 unlock_autospawn_lock_file(c);
538 unlock_autospawn_lock_file(c);
545 #endif /* OS_IS_WIN32 */
547 static int try_next_connection(pa_context *c) {
558 c->server_list = pa_strlist_pop(c->server_list, &u);
563 if (c->do_autospawn) {
564 r = context_connect_spawn(c);
569 pa_context_fail(c, PA_ERR_CONNECTIONREFUSED);
573 pa_log_debug(__FILE__": Trying to connect to %s...", u);
576 c->server = pa_xstrdup(u);
578 if (!(c->client = pa_socket_client_new_string(c->mainloop, u, PA_NATIVE_DEFAULT_PORT)))
581 c->local = pa_socket_client_is_local(c->client);
582 pa_socket_client_set_callback(c->client, on_connection, c);
594 static void on_connection(pa_socket_client *client, pa_iochannel*io, void *userdata) {
595 pa_context *c = userdata;
599 assert(c->state == PA_CONTEXT_CONNECTING);
603 pa_socket_client_unref(client);
607 /* Try the item in the list */
608 if (errno == ECONNREFUSED || errno == ETIMEDOUT || errno == EHOSTUNREACH) {
609 try_next_connection(c);
613 pa_context_fail(c, PA_ERR_CONNECTIONREFUSED);
617 unlock_autospawn_lock_file(c);
618 setup_context(c, io);
624 int pa_context_connect(
627 pa_context_flags_t flags,
628 const pa_spawn_api *api) {
635 PA_CHECK_VALIDITY(c, c->state == PA_CONTEXT_UNCONNECTED, PA_ERR_BADSTATE);
636 PA_CHECK_VALIDITY(c, !(flags & ~PA_CONTEXT_NOAUTOSPAWN), PA_ERR_INVALID);
637 PA_CHECK_VALIDITY(c, !server || *server, PA_ERR_INVALID);
640 server = c->conf->default_server;
644 assert(!c->server_list);
647 if (!(c->server_list = pa_strlist_parse(server))) {
648 pa_context_fail(c, PA_ERR_INVALIDSERVER);
655 /* Prepend in reverse order */
657 if ((d = getenv("DISPLAY"))) {
660 if ((e = strchr(d, ':')))
664 c->server_list = pa_strlist_prepend(c->server_list, d);
669 c->server_list = pa_strlist_prepend(c->server_list, "tcp6:localhost");
670 c->server_list = pa_strlist_prepend(c->server_list, "localhost");
671 c->server_list = pa_strlist_prepend(c->server_list, pa_runtime_path(PA_NATIVE_DEFAULT_UNIX_SOCKET, ufn, sizeof(ufn)));
673 /* Wrap the connection attempts in a single transaction for sane autospawn locking */
674 if (!(flags & PA_CONTEXT_NOAUTOSPAWN) && c->conf->autospawn) {
677 pa_runtime_path(AUTOSPAWN_LOCK, lf, sizeof(lf));
678 pa_make_secure_parent_dir(lf);
679 assert(c->autospawn_lock_fd <= 0);
680 c->autospawn_lock_fd = pa_lock_lockfile(lf);
689 pa_context_set_state(c, PA_CONTEXT_CONNECTING);
690 r = try_next_connection(c);
698 void pa_context_disconnect(pa_context *c) {
702 pa_context_set_state(c, PA_CONTEXT_TERMINATED);
705 pa_context_state_t pa_context_get_state(pa_context *c) {
712 int pa_context_errno(pa_context *c) {
719 void pa_context_set_state_callback(pa_context *c, pa_context_notify_cb_t cb, void *userdata) {
723 c->state_callback = cb;
724 c->state_userdata = userdata;
727 int pa_context_is_pending(pa_context *c) {
732 c->state == PA_CONTEXT_CONNECTING ||
733 c->state == PA_CONTEXT_AUTHORIZING ||
734 c->state == PA_CONTEXT_SETTING_NAME ||
735 c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
737 return (c->pstream && pa_pstream_is_pending(c->pstream)) ||
738 (c->pdispatch && pa_pdispatch_is_pending(c->pdispatch)) ||
742 static void set_dispatch_callbacks(pa_operation *o);
744 static void pdispatch_drain_callback(PA_GCC_UNUSED pa_pdispatch*pd, void *userdata) {
745 set_dispatch_callbacks(userdata);
748 static void pstream_drain_callback(PA_GCC_UNUSED pa_pstream *s, void *userdata) {
749 set_dispatch_callbacks(userdata);
752 static void set_dispatch_callbacks(pa_operation *o) {
758 assert(o->context->ref >= 1);
759 assert(o->context->state == PA_CONTEXT_READY);
761 pa_pstream_set_drain_callback(o->context->pstream, NULL, NULL);
762 pa_pdispatch_set_drain_callback(o->context->pdispatch, NULL, NULL);
764 if (pa_pdispatch_is_pending(o->context->pdispatch)) {
765 pa_pdispatch_set_drain_callback(o->context->pdispatch, pdispatch_drain_callback, o);
769 if (pa_pstream_is_pending(o->context->pstream)) {
770 pa_pstream_set_drain_callback(o->context->pstream, pstream_drain_callback, o);
778 pa_context_notify_cb_t cb = (pa_context_notify_cb_t) o->callback;
779 cb(o->context, o->userdata);
782 pa_operation_done(o);
785 pa_operation_unref(o);
788 pa_operation* pa_context_drain(pa_context *c, pa_context_notify_cb_t cb, void *userdata) {
794 PA_CHECK_VALIDITY_RETURN_NULL(c, c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
795 PA_CHECK_VALIDITY_RETURN_NULL(c, pa_context_is_pending(c), PA_ERR_BADSTATE);
797 o = pa_operation_new(c, NULL, (pa_operation_cb_t) cb, userdata);
798 set_dispatch_callbacks(pa_operation_ref(o));
803 void pa_context_simple_ack_callback(pa_pdispatch *pd, uint32_t command, PA_GCC_UNUSED uint32_t tag, pa_tagstruct *t, void *userdata) {
804 pa_operation *o = userdata;
812 if (command != PA_COMMAND_REPLY) {
813 if (pa_context_handle_error(o->context, command, t) < 0)
817 } else if (!pa_tagstruct_eof(t)) {
818 pa_context_fail(o->context, PA_ERR_PROTOCOL);
823 pa_context_success_cb_t cb = (pa_context_success_cb_t) o->callback;
824 cb(o->context, success, o->userdata);
828 pa_operation_done(o);
829 pa_operation_unref(o);
832 pa_operation* pa_context_exit_daemon(pa_context *c, pa_context_success_cb_t cb, void *userdata) {
840 PA_CHECK_VALIDITY_RETURN_NULL(c, c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
842 o = pa_operation_new(c, NULL, (pa_operation_cb_t) cb, userdata);
844 t = pa_tagstruct_command(c, PA_COMMAND_EXIT, &tag);
845 pa_pstream_send_tagstruct(c->pstream, t);
846 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, pa_operation_ref(o));
851 pa_operation* pa_context_send_simple_command(pa_context *c, uint32_t command, pa_pdispatch_cb_t internal_cb, pa_operation_cb_t cb, void *userdata) {
859 PA_CHECK_VALIDITY_RETURN_NULL(c, c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
861 o = pa_operation_new(c, NULL, cb, userdata);
863 t = pa_tagstruct_command(c, command, &tag);
864 pa_pstream_send_tagstruct(c->pstream, t);
865 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, internal_cb, pa_operation_ref(o));
870 pa_operation* pa_context_set_default_sink(pa_context *c, const char *name, pa_context_success_cb_t cb, void *userdata) {
878 PA_CHECK_VALIDITY_RETURN_NULL(c, c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
880 o = pa_operation_new(c, NULL, (pa_operation_cb_t) cb, userdata);
882 t = pa_tagstruct_command(c, PA_COMMAND_SET_DEFAULT_SINK, &tag);
883 pa_tagstruct_puts(t, name);
884 pa_pstream_send_tagstruct(c->pstream, t);
885 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, pa_operation_ref(o));
890 pa_operation* pa_context_set_default_source(pa_context *c, const char *name, pa_context_success_cb_t cb, void *userdata) {
898 PA_CHECK_VALIDITY_RETURN_NULL(c, c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
900 o = pa_operation_new(c, NULL, (pa_operation_cb_t) cb, userdata);
902 t = pa_tagstruct_command(c, PA_COMMAND_SET_DEFAULT_SOURCE, &tag);
903 pa_tagstruct_puts(t, name);
904 pa_pstream_send_tagstruct(c->pstream, t);
905 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, pa_operation_ref(o));
910 int pa_context_is_local(pa_context *c) {
916 pa_operation* pa_context_set_name(pa_context *c, const char *name, pa_context_success_cb_t cb, void *userdata) {
925 PA_CHECK_VALIDITY_RETURN_NULL(c, c->state == PA_CONTEXT_READY, PA_ERR_BADSTATE);
927 o = pa_operation_new(c, NULL, (pa_operation_cb_t) cb, userdata);
929 t = pa_tagstruct_command(c, PA_COMMAND_SET_CLIENT_NAME, &tag);
930 pa_tagstruct_puts(t, name);
931 pa_pstream_send_tagstruct(c->pstream, t);
932 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, pa_operation_ref(o));
937 const char* pa_get_library_version(void) {
938 return PACKAGE_VERSION;
941 const char* pa_context_get_server(pa_context *c) {
948 if (*c->server == '{') {
949 char *e = strchr(c->server+1, '}');
950 return e ? e+1 : c->server;
956 uint32_t pa_context_get_protocol_version(pa_context *c) {
957 return PA_PROTOCOL_VERSION;
960 uint32_t pa_context_get_server_protocol_version(pa_context *c) {
967 pa_tagstruct *pa_tagstruct_command(pa_context *c, uint32_t command, uint32_t *tag) {
973 t = pa_tagstruct_new(NULL, 0);
974 pa_tagstruct_putu32(t, command);
975 pa_tagstruct_putu32(t, *tag = c->ctag++);