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>
31 #include <sys/socket.h>
39 #include "polyplib-internal.h"
40 #include "polyplib-context.h"
41 #include "native-common.h"
42 #include "pdispatch.h"
45 #include "socket-client.h"
46 #include "pstream-util.h"
50 #include "client-conf.h"
51 #include "socket-util.h"
54 #include "client-conf-x11.h"
57 #define AUTOSPAWN_LOCK "autospawn.lock"
59 static const struct pa_pdispatch_command command_table[PA_COMMAND_MAX] = {
60 [PA_COMMAND_REQUEST] = { pa_command_request },
61 [PA_COMMAND_PLAYBACK_STREAM_KILLED] = { pa_command_stream_killed },
62 [PA_COMMAND_RECORD_STREAM_KILLED] = { pa_command_stream_killed },
63 [PA_COMMAND_SUBSCRIBE_EVENT] = { pa_command_subscribe_event },
66 static void unlock_autospawn_lock_file(struct pa_context *c) {
69 if (c->autospawn_lock_fd >= 0) {
71 pa_runtime_path(AUTOSPAWN_LOCK, lf, sizeof(lf));
73 pa_unlock_lockfile(lf, c->autospawn_lock_fd);
74 c->autospawn_lock_fd = -1;
78 struct pa_context *pa_context_new(struct pa_mainloop_api *mainloop, const char *name) {
80 assert(mainloop && name);
82 c = pa_xmalloc(sizeof(struct pa_context));
84 c->name = pa_xstrdup(name);
85 c->mainloop = mainloop;
89 c->playback_streams = pa_dynarray_new();
90 c->record_streams = pa_dynarray_new();
91 assert(c->playback_streams && c->record_streams);
93 PA_LLIST_HEAD_INIT(struct pa_stream, c->streams);
94 PA_LLIST_HEAD_INIT(struct pa_operation, c->operations);
96 c->error = PA_ERROR_OK;
97 c->state = PA_CONTEXT_UNCONNECTED;
100 c->state_callback = NULL;
101 c->state_userdata = NULL;
103 c->subscribe_callback = NULL;
104 c->subscribe_userdata = NULL;
106 c->memblock_stat = pa_memblock_stat_new();
108 c->server_list = NULL;
110 c->autospawn_lock_fd = -1;
111 memset(&c->spawn_api, 0, sizeof(c->spawn_api));
114 pa_check_signal_is_blocked(SIGPIPE);
116 c->conf = pa_client_conf_new();
117 pa_client_conf_load(c->conf, NULL);
119 pa_client_conf_from_x11(c->conf, NULL);
121 pa_client_conf_env(c->conf);
126 static void context_free(struct pa_context *c) {
129 unlock_autospawn_lock_file(c);
131 while (c->operations)
132 pa_operation_cancel(c->operations);
135 pa_stream_set_state(c->streams, PA_STREAM_TERMINATED);
138 pa_socket_client_unref(c->client);
140 pa_pdispatch_unref(c->pdispatch);
142 pa_pstream_close(c->pstream);
143 pa_pstream_unref(c->pstream);
146 if (c->record_streams)
147 pa_dynarray_free(c->record_streams, NULL, NULL);
148 if (c->playback_streams)
149 pa_dynarray_free(c->playback_streams, NULL, NULL);
151 pa_memblock_stat_unref(c->memblock_stat);
154 pa_client_conf_free(c->conf);
156 pa_strlist_free(c->server_list);
163 struct pa_context* pa_context_ref(struct pa_context *c) {
164 assert(c && c->ref >= 1);
169 void pa_context_unref(struct pa_context *c) {
170 assert(c && c->ref >= 1);
172 if ((--(c->ref)) == 0)
176 void pa_context_set_state(struct pa_context *c, enum pa_context_state st) {
184 if (st == PA_CONTEXT_FAILED || st == PA_CONTEXT_TERMINATED) {
187 s = c->streams ? pa_stream_ref(c->streams) : NULL;
189 struct pa_stream *n = s->next ? pa_stream_ref(s->next) : NULL;
190 pa_stream_set_state(s, st == PA_CONTEXT_FAILED ? PA_STREAM_FAILED : PA_STREAM_TERMINATED);
196 pa_pdispatch_unref(c->pdispatch);
200 pa_pstream_close(c->pstream);
201 pa_pstream_unref(c->pstream);
206 pa_socket_client_unref(c->client);
211 if (c->state_callback)
212 c->state_callback(c, c->state_userdata);
217 void pa_context_fail(struct pa_context *c, int error) {
220 pa_context_set_state(c, PA_CONTEXT_FAILED);
223 static void pstream_die_callback(struct pa_pstream *p, void *userdata) {
224 struct pa_context *c = userdata;
226 pa_context_fail(c, PA_ERROR_CONNECTIONTERMINATED);
229 static void pstream_packet_callback(struct pa_pstream *p, struct pa_packet *packet, void *userdata) {
230 struct pa_context *c = userdata;
231 assert(p && packet && c);
235 if (pa_pdispatch_run(c->pdispatch, packet, c) < 0) {
236 pa_log(__FILE__": invalid packet.\n");
237 pa_context_fail(c, PA_ERROR_PROTOCOL);
243 static void pstream_memblock_callback(struct pa_pstream *p, uint32_t channel, uint32_t delta, const struct pa_memchunk *chunk, void *userdata) {
244 struct pa_context *c = userdata;
246 assert(p && chunk && c && chunk->memblock && chunk->memblock->data);
250 if ((s = pa_dynarray_get(c->record_streams, channel))) {
251 pa_mcalign_push(s->mcalign, chunk);
254 struct pa_memchunk t;
256 if (pa_mcalign_pop(s->mcalign, &t) < 0)
259 if (s->read_callback) {
260 s->read_callback(s, (uint8_t*) t.memblock->data + t.index, t.length, s->read_userdata);
261 s->counter += chunk->length;
264 pa_memblock_unref(t.memblock);
271 int pa_context_handle_error(struct pa_context *c, uint32_t command, struct pa_tagstruct *t) {
274 if (command == PA_COMMAND_ERROR) {
277 if (pa_tagstruct_getu32(t, &c->error) < 0) {
278 pa_context_fail(c, PA_ERROR_PROTOCOL);
282 } else if (command == PA_COMMAND_TIMEOUT)
283 c->error = PA_ERROR_TIMEOUT;
285 pa_context_fail(c, PA_ERROR_PROTOCOL);
292 static void setup_complete_callback(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
293 struct pa_context *c = userdata;
294 assert(pd && c && (c->state == PA_CONTEXT_AUTHORIZING || c->state == PA_CONTEXT_SETTING_NAME));
298 if (command != PA_COMMAND_REPLY) {
300 if (pa_context_handle_error(c, command, t) < 0)
301 pa_context_fail(c, PA_ERROR_PROTOCOL);
303 pa_context_fail(c, c->error);
308 case PA_CONTEXT_AUTHORIZING: {
309 struct pa_tagstruct *t;
310 t = pa_tagstruct_new(NULL, 0);
312 pa_tagstruct_putu32(t, PA_COMMAND_SET_CLIENT_NAME);
313 pa_tagstruct_putu32(t, tag = c->ctag++);
314 pa_tagstruct_puts(t, c->name);
315 pa_pstream_send_tagstruct(c->pstream, t);
316 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, setup_complete_callback, c);
318 pa_context_set_state(c, PA_CONTEXT_SETTING_NAME);
322 case PA_CONTEXT_SETTING_NAME :
323 pa_context_set_state(c, PA_CONTEXT_READY);
334 static void setup_context(struct pa_context *c, struct pa_iochannel *io) {
335 struct pa_tagstruct *t;
342 c->pstream = pa_pstream_new(c->mainloop, io, c->memblock_stat);
345 pa_pstream_set_die_callback(c->pstream, pstream_die_callback, c);
346 pa_pstream_set_recieve_packet_callback(c->pstream, pstream_packet_callback, c);
347 pa_pstream_set_recieve_memblock_callback(c->pstream, pstream_memblock_callback, c);
349 assert(!c->pdispatch);
350 c->pdispatch = pa_pdispatch_new(c->mainloop, command_table, PA_COMMAND_MAX);
351 assert(c->pdispatch);
353 if (!c->conf->cookie_valid) {
354 pa_context_fail(c, PA_ERROR_AUTHKEY);
358 t = pa_tagstruct_new(NULL, 0);
360 pa_tagstruct_putu32(t, PA_COMMAND_AUTH);
361 pa_tagstruct_putu32(t, tag = c->ctag++);
362 pa_tagstruct_put_arbitrary(t, c->conf->cookie, sizeof(c->conf->cookie));
363 pa_pstream_send_tagstruct(c->pstream, t);
364 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, setup_complete_callback, c);
366 pa_context_set_state(c, PA_CONTEXT_AUTHORIZING);
373 static void on_connection(struct pa_socket_client *client, struct pa_iochannel*io, void *userdata);
375 static int context_connect_spawn(struct pa_context *c) {
378 int fds[2] = { -1, -1} ;
379 struct pa_iochannel *io;
383 if (socketpair(AF_UNIX, SOCK_STREAM, 0, fds) < 0) {
384 pa_log(__FILE__": socketpair() failed: %s\n", strerror(errno));
385 pa_context_fail(c, PA_ERROR_INTERNAL);
389 pa_fd_set_cloexec(fds[0], 1);
391 pa_socket_low_delay(fds[0]);
392 pa_socket_low_delay(fds[1]);
394 if (c->spawn_api.prefork)
395 c->spawn_api.prefork();
397 if ((pid = fork()) < 0) {
398 pa_log(__FILE__": fork() failed: %s\n", strerror(errno));
399 pa_context_fail(c, PA_ERROR_INTERNAL);
401 if (c->spawn_api.postfork)
402 c->spawn_api.postfork();
409 const char *state = NULL;
411 char *argv[MAX_ARGS+1];
414 /* Not required, since fds[0] has CLOEXEC enabled anyway */
417 if (c->spawn_api.atfork)
418 c->spawn_api.atfork();
424 argv[n++] = c->conf->daemon_binary;
425 argv[n++] = "--daemonize=yes";
427 snprintf(t, sizeof(t), "-Lmodule-native-protocol-fd fd=%i", fds[1]);
428 argv[n++] = strdup(t);
430 while (n < MAX_ARGS) {
433 if (!(a = pa_split_spaces(c->conf->extra_arguments, &state)))
441 execv(argv[0], argv);
448 r = waitpid(pid, &status, 0);
450 if (c->spawn_api.postfork)
451 c->spawn_api.postfork();
454 pa_log(__FILE__": waitpid() failed: %s\n", strerror(errno));
455 pa_context_fail(c, PA_ERROR_INTERNAL);
457 } else if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
458 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
466 io = pa_iochannel_new(c->mainloop, fds[0], fds[0]);
468 setup_context(c, io);
469 unlock_autospawn_lock_file(c);
481 unlock_autospawn_lock_file(c);
488 static int try_next_connection(struct pa_context *c) {
491 assert(c && !c->client);
498 c->server_list = pa_strlist_pop(c->server_list, &u);
502 if (c->do_autospawn) {
503 r = context_connect_spawn(c);
507 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
511 /* pa_log(__FILE__": Trying to connect to %s...\n", u); */
514 c->server = pa_xstrdup(u);
516 if (!(c->client = pa_socket_client_new_string(c->mainloop, u, PA_NATIVE_DEFAULT_PORT)))
519 c->local = pa_socket_client_is_local(c->client);
520 pa_socket_client_set_callback(c->client, on_connection, c);
533 static void on_connection(struct pa_socket_client *client, struct pa_iochannel*io, void *userdata) {
534 struct pa_context *c = userdata;
535 assert(client && c && c->state == PA_CONTEXT_CONNECTING);
539 pa_socket_client_unref(client);
543 /* Try the item in the list */
544 if (errno == ECONNREFUSED || errno == ETIMEDOUT || errno == EHOSTUNREACH) {
545 try_next_connection(c);
549 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
553 unlock_autospawn_lock_file(c);
554 setup_context(c, io);
560 int pa_context_connect(struct pa_context *c, const char *server, int spawn, const struct pa_spawn_api *api) {
562 assert(c && c->ref >= 1 && c->state == PA_CONTEXT_UNCONNECTED);
565 server = c->conf->default_server;
569 assert(!c->server_list);
572 if (!(c->server_list = pa_strlist_parse(server))) {
573 pa_context_fail(c, PA_ERROR_INVALIDSERVER);
580 /* Prepend in reverse order */
582 if ((d = getenv("DISPLAY"))) {
585 if ((e = strchr(d, ':')))
589 c->server_list = pa_strlist_prepend(c->server_list, d);
594 c->server_list = pa_strlist_prepend(c->server_list, "tcp6:localhost");
595 c->server_list = pa_strlist_prepend(c->server_list, "localhost");
596 c->server_list = pa_strlist_prepend(c->server_list, pa_runtime_path(PA_NATIVE_DEFAULT_UNIX_SOCKET, ufn, sizeof(ufn)));
598 /* Wrap the connection attempts in a single transaction for sane autospawn locking */
599 if (spawn && c->conf->autospawn) {
602 pa_runtime_path(AUTOSPAWN_LOCK, lf, sizeof(lf));
603 pa_make_secure_parent_dir(lf);
604 assert(c->autospawn_lock_fd <= 0);
605 c->autospawn_lock_fd = pa_lock_lockfile(lf);
614 pa_context_set_state(c, PA_CONTEXT_CONNECTING);
615 r = try_next_connection(c);
623 void pa_context_disconnect(struct pa_context *c) {
625 pa_context_set_state(c, PA_CONTEXT_TERMINATED);
628 enum pa_context_state pa_context_get_state(struct pa_context *c) {
629 assert(c && c->ref >= 1);
633 int pa_context_errno(struct pa_context *c) {
634 assert(c && c->ref >= 1);
638 void pa_context_set_state_callback(struct pa_context *c, void (*cb)(struct pa_context *c, void *userdata), void *userdata) {
639 assert(c && c->ref >= 1);
640 c->state_callback = cb;
641 c->state_userdata = userdata;
644 int pa_context_is_pending(struct pa_context *c) {
645 assert(c && c->ref >= 1);
647 /* pa_log("pstream: %i\n", pa_pstream_is_pending(c->pstream)); */
648 /* pa_log("pdispatch: %i\n", pa_pdispatch_is_pending(c->pdispatch)); */
650 return (c->pstream && pa_pstream_is_pending(c->pstream)) ||
651 (c->pdispatch && pa_pdispatch_is_pending(c->pdispatch)) ||
655 static void set_dispatch_callbacks(struct pa_operation *o);
657 static void pdispatch_drain_callback(struct pa_pdispatch*pd, void *userdata) {
658 set_dispatch_callbacks(userdata);
661 static void pstream_drain_callback(struct pa_pstream *s, void *userdata) {
662 set_dispatch_callbacks(userdata);
665 static void set_dispatch_callbacks(struct pa_operation *o) {
667 assert(o && o->context && o->context->ref >= 1 && o->ref >= 1 && o->context->state == PA_CONTEXT_READY);
669 pa_pstream_set_drain_callback(o->context->pstream, NULL, NULL);
670 pa_pdispatch_set_drain_callback(o->context->pdispatch, NULL, NULL);
672 if (pa_pdispatch_is_pending(o->context->pdispatch)) {
673 pa_pdispatch_set_drain_callback(o->context->pdispatch, pdispatch_drain_callback, o);
677 if (pa_pstream_is_pending(o->context->pstream)) {
678 pa_pstream_set_drain_callback(o->context->pstream, pstream_drain_callback, o);
686 void (*cb)(struct pa_context *c, void *userdata);
687 cb = (void (*)(struct pa_context*, void*)) o->callback;
688 cb(o->context, o->userdata);
691 pa_operation_done(o);
694 pa_operation_unref(o);
697 struct pa_operation* pa_context_drain(struct pa_context *c, void (*cb) (struct pa_context*c, void *userdata), void *userdata) {
698 struct pa_operation *o;
699 assert(c && c->ref >= 1);
701 if (c->state != PA_CONTEXT_READY)
704 if (!pa_context_is_pending(c))
707 o = pa_operation_new(c, NULL);
710 o->userdata = userdata;
712 set_dispatch_callbacks(pa_operation_ref(o));
717 void pa_context_exit_daemon(struct pa_context *c) {
718 struct pa_tagstruct *t;
719 assert(c && c->ref >= 1);
721 t = pa_tagstruct_new(NULL, 0);
723 pa_tagstruct_putu32(t, PA_COMMAND_EXIT);
724 pa_tagstruct_putu32(t, c->ctag++);
725 pa_pstream_send_tagstruct(c->pstream, t);
728 void pa_context_simple_ack_callback(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
729 struct pa_operation *o = userdata;
731 assert(pd && o && o->context && o->ref >= 1);
733 if (command != PA_COMMAND_REPLY) {
734 if (pa_context_handle_error(o->context, command, t) < 0)
738 } else if (!pa_tagstruct_eof(t)) {
739 pa_context_fail(o->context, PA_ERROR_PROTOCOL);
744 void (*cb)(struct pa_context *c, int success, void *userdata) = o->callback;
745 cb(o->context, success, o->userdata);
749 pa_operation_done(o);
750 pa_operation_unref(o);
753 struct pa_operation* pa_context_send_simple_command(struct pa_context *c, uint32_t command, void (*internal_callback)(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata), void (*cb)(), void *userdata) {
754 struct pa_tagstruct *t;
755 struct pa_operation *o;
759 o = pa_operation_new(c, NULL);
761 o->userdata = userdata;
763 t = pa_tagstruct_new(NULL, 0);
764 pa_tagstruct_putu32(t, command);
765 pa_tagstruct_putu32(t, tag = c->ctag++);
766 pa_pstream_send_tagstruct(c->pstream, t);
767 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, internal_callback, o);
769 return pa_operation_ref(o);
772 struct pa_operation* pa_context_set_default_sink(struct pa_context *c, const char *name, void(*cb)(struct pa_context*c, int success, void *userdata), void *userdata) {
773 struct pa_tagstruct *t;
774 struct pa_operation *o;
778 o = pa_operation_new(c, NULL);
780 o->userdata = userdata;
782 t = pa_tagstruct_new(NULL, 0);
783 pa_tagstruct_putu32(t, PA_COMMAND_SET_DEFAULT_SINK);
784 pa_tagstruct_putu32(t, tag = c->ctag++);
785 pa_tagstruct_puts(t, name);
786 pa_pstream_send_tagstruct(c->pstream, t);
787 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, o);
789 return pa_operation_ref(o);
792 struct pa_operation* pa_context_set_default_source(struct pa_context *c, const char *name, void(*cb)(struct pa_context*c, int success, void *userdata), void *userdata) {
793 struct pa_tagstruct *t;
794 struct pa_operation *o;
798 o = pa_operation_new(c, NULL);
800 o->userdata = userdata;
802 t = pa_tagstruct_new(NULL, 0);
803 pa_tagstruct_putu32(t, PA_COMMAND_SET_DEFAULT_SOURCE);
804 pa_tagstruct_putu32(t, tag = c->ctag++);
805 pa_tagstruct_puts(t, name);
806 pa_pstream_send_tagstruct(c->pstream, t);
807 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, o);
809 return pa_operation_ref(o);
812 int pa_context_is_local(struct pa_context *c) {
817 struct pa_operation* pa_context_set_name(struct pa_context *c, const char *name, void(*cb)(struct pa_context*c, int success, void *userdata), void *userdata) {
818 struct pa_tagstruct *t;
819 struct pa_operation *o;
821 assert(c && name && cb);
823 o = pa_operation_new(c, NULL);
825 o->userdata = userdata;
827 t = pa_tagstruct_new(NULL, 0);
828 pa_tagstruct_putu32(t, PA_COMMAND_SET_CLIENT_NAME);
829 pa_tagstruct_putu32(t, tag = c->ctag++);
830 pa_tagstruct_puts(t, name);
831 pa_pstream_send_tagstruct(c->pstream, t);
832 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, o);
834 return pa_operation_ref(o);
837 const char* pa_get_library_version(void) {
838 return PACKAGE_VERSION;
841 const char* pa_context_get_server(struct pa_context *c) {
846 if (*c->server == '{') {
847 char *e = strchr(c->server+1, '}');
848 return e ? e+1 : c->server;