]> code.delx.au - pulseaudio/blob - polyp/polyplib-context.c
correct autospawning
[pulseaudio] / polyp / polyplib-context.c
1 /* $Id$ */
2
3 /***
4 This file is part of polypaudio.
5
6 polypaudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU 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.
10
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.
15
16 You should have received a copy of the GNU 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
19 USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <stdio.h>
27 #include <assert.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <sys/types.h>
31 #include <sys/socket.h>
32 #include <netdb.h>
33 #include <unistd.h>
34 #include <sys/stat.h>
35 #include <errno.h>
36 #include <sys/wait.h>
37
38 #include "polyplib-internal.h"
39 #include "polyplib-context.h"
40 #include "native-common.h"
41 #include "pdispatch.h"
42 #include "pstream.h"
43 #include "dynarray.h"
44 #include "socket-client.h"
45 #include "pstream-util.h"
46 #include "authkey.h"
47 #include "util.h"
48 #include "xmalloc.h"
49 #include "log.h"
50
51 static const struct pa_pdispatch_command command_table[PA_COMMAND_MAX] = {
52 [PA_COMMAND_REQUEST] = { pa_command_request },
53 [PA_COMMAND_PLAYBACK_STREAM_KILLED] = { pa_command_stream_killed },
54 [PA_COMMAND_RECORD_STREAM_KILLED] = { pa_command_stream_killed },
55 [PA_COMMAND_SUBSCRIBE_EVENT] = { pa_command_subscribe_event },
56 };
57
58 struct pa_context *pa_context_new(struct pa_mainloop_api *mainloop, const char *name) {
59 struct pa_context *c;
60 assert(mainloop && name);
61
62 c = pa_xmalloc(sizeof(struct pa_context));
63 c->ref = 1;
64 c->name = pa_xstrdup(name);
65 c->mainloop = mainloop;
66 c->client = NULL;
67 c->pstream = NULL;
68 c->pdispatch = NULL;
69 c->playback_streams = pa_dynarray_new();
70 c->record_streams = pa_dynarray_new();
71 assert(c->playback_streams && c->record_streams);
72
73 PA_LLIST_HEAD_INIT(struct pa_stream, c->streams);
74 PA_LLIST_HEAD_INIT(struct pa_operation, c->operations);
75
76 c->error = PA_ERROR_OK;
77 c->state = PA_CONTEXT_UNCONNECTED;
78 c->ctag = 0;
79
80 c->state_callback = NULL;
81 c->state_userdata = NULL;
82
83 c->subscribe_callback = NULL;
84 c->subscribe_userdata = NULL;
85
86 c->memblock_stat = pa_memblock_stat_new();
87
88 pa_check_signal_is_blocked(SIGPIPE);
89 return c;
90 }
91
92 static void context_free(struct pa_context *c) {
93 assert(c);
94
95 while (c->operations)
96 pa_operation_cancel(c->operations);
97
98 while (c->streams)
99 pa_stream_set_state(c->streams, PA_STREAM_TERMINATED);
100
101 if (c->client)
102 pa_socket_client_unref(c->client);
103 if (c->pdispatch)
104 pa_pdispatch_unref(c->pdispatch);
105 if (c->pstream) {
106 pa_pstream_close(c->pstream);
107 pa_pstream_unref(c->pstream);
108 }
109
110 if (c->record_streams)
111 pa_dynarray_free(c->record_streams, NULL, NULL);
112 if (c->playback_streams)
113 pa_dynarray_free(c->playback_streams, NULL, NULL);
114
115 pa_memblock_stat_unref(c->memblock_stat);
116
117 pa_xfree(c->name);
118 pa_xfree(c);
119 }
120
121 struct pa_context* pa_context_ref(struct pa_context *c) {
122 assert(c && c->ref >= 1);
123 c->ref++;
124 return c;
125 }
126
127 void pa_context_unref(struct pa_context *c) {
128 assert(c && c->ref >= 1);
129
130 if ((--(c->ref)) == 0)
131 context_free(c);
132 }
133
134 void pa_context_set_state(struct pa_context *c, enum pa_context_state st) {
135 assert(c);
136
137 if (c->state == st)
138 return;
139
140 pa_context_ref(c);
141
142 if (st == PA_CONTEXT_FAILED || st == PA_CONTEXT_TERMINATED) {
143 struct pa_stream *s;
144
145 s = c->streams ? pa_stream_ref(c->streams) : NULL;
146 while (s) {
147 struct pa_stream *n = s->next ? pa_stream_ref(s->next) : NULL;
148 pa_stream_set_state(s, st == PA_CONTEXT_FAILED ? PA_STREAM_FAILED : PA_STREAM_TERMINATED);
149 pa_stream_unref(s);
150 s = n;
151 }
152
153 if (c->pdispatch)
154 pa_pdispatch_unref(c->pdispatch);
155 c->pdispatch = NULL;
156
157 if (c->pstream) {
158 pa_pstream_close(c->pstream);
159 pa_pstream_unref(c->pstream);
160 }
161 c->pstream = NULL;
162
163 if (c->client)
164 pa_socket_client_unref(c->client);
165 c->client = NULL;
166 }
167
168 c->state = st;
169 if (c->state_callback)
170 c->state_callback(c, c->state_userdata);
171
172 pa_context_unref(c);
173 }
174
175 void pa_context_fail(struct pa_context *c, int error) {
176 assert(c);
177 c->error = error;
178 pa_context_set_state(c, PA_CONTEXT_FAILED);
179 }
180
181 static void pstream_die_callback(struct pa_pstream *p, void *userdata) {
182 struct pa_context *c = userdata;
183 assert(p && c);
184 pa_context_fail(c, PA_ERROR_CONNECTIONTERMINATED);
185 }
186
187 static void pstream_packet_callback(struct pa_pstream *p, struct pa_packet *packet, void *userdata) {
188 struct pa_context *c = userdata;
189 assert(p && packet && c);
190
191 pa_context_ref(c);
192
193 if (pa_pdispatch_run(c->pdispatch, packet, c) < 0) {
194 pa_log(__FILE__": invalid packet.\n");
195 pa_context_fail(c, PA_ERROR_PROTOCOL);
196 }
197
198 pa_context_unref(c);
199 }
200
201 static void pstream_memblock_callback(struct pa_pstream *p, uint32_t channel, uint32_t delta, const struct pa_memchunk *chunk, void *userdata) {
202 struct pa_context *c = userdata;
203 struct pa_stream *s;
204 assert(p && chunk && c && chunk->memblock && chunk->memblock->data);
205
206 pa_context_ref(c);
207
208 if ((s = pa_dynarray_get(c->record_streams, channel))) {
209 if (s->read_callback)
210 s->read_callback(s, (uint8_t*) chunk->memblock->data + chunk->index, chunk->length, s->read_userdata);
211 }
212
213 pa_context_unref(c);
214 }
215
216 int pa_context_handle_error(struct pa_context *c, uint32_t command, struct pa_tagstruct *t) {
217 assert(c && t);
218
219 if (command == PA_COMMAND_ERROR) {
220 if (pa_tagstruct_getu32(t, &c->error) < 0) {
221 pa_context_fail(c, PA_ERROR_PROTOCOL);
222 return -1;
223
224 }
225 } else if (command == PA_COMMAND_TIMEOUT)
226 c->error = PA_ERROR_TIMEOUT;
227 else {
228 pa_context_fail(c, PA_ERROR_PROTOCOL);
229 return -1;
230 }
231
232 return 0;
233 }
234
235 static void setup_complete_callback(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
236 struct pa_context *c = userdata;
237 assert(pd && c && (c->state == PA_CONTEXT_AUTHORIZING || c->state == PA_CONTEXT_SETTING_NAME));
238
239 pa_context_ref(c);
240
241 if (command != PA_COMMAND_REPLY) {
242
243 if (pa_context_handle_error(c, command, t) < 0)
244 pa_context_fail(c, PA_ERROR_PROTOCOL);
245
246 pa_context_fail(c, c->error);
247 goto finish;
248 }
249
250 switch(c->state) {
251 case PA_CONTEXT_AUTHORIZING: {
252 struct pa_tagstruct *t;
253 t = pa_tagstruct_new(NULL, 0);
254 assert(t);
255 pa_tagstruct_putu32(t, PA_COMMAND_SET_NAME);
256 pa_tagstruct_putu32(t, tag = c->ctag++);
257 pa_tagstruct_puts(t, c->name);
258 pa_pstream_send_tagstruct(c->pstream, t);
259 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, setup_complete_callback, c);
260
261 pa_context_set_state(c, PA_CONTEXT_SETTING_NAME);
262 break;
263 }
264
265 case PA_CONTEXT_SETTING_NAME :
266 pa_context_set_state(c, PA_CONTEXT_READY);
267 break;
268
269 default:
270 assert(0);
271 }
272
273 finish:
274 pa_context_unref(c);
275 }
276
277 static void setup_context(struct pa_context *c, struct pa_iochannel *io) {
278 struct pa_tagstruct *t;
279 uint32_t tag;
280 assert(c && io);
281
282 pa_context_ref(c);
283
284 assert(!c->pstream);
285 c->pstream = pa_pstream_new(c->mainloop, io, c->memblock_stat);
286 assert(c->pstream);
287
288 pa_pstream_set_die_callback(c->pstream, pstream_die_callback, c);
289 pa_pstream_set_recieve_packet_callback(c->pstream, pstream_packet_callback, c);
290 pa_pstream_set_recieve_memblock_callback(c->pstream, pstream_memblock_callback, c);
291
292 assert(!c->pdispatch);
293 c->pdispatch = pa_pdispatch_new(c->mainloop, command_table, PA_COMMAND_MAX);
294 assert(c->pdispatch);
295
296 if (pa_authkey_load_from_home(PA_NATIVE_COOKIE_FILE, c->auth_cookie, sizeof(c->auth_cookie)) < 0) {
297 pa_context_fail(c, PA_ERROR_AUTHKEY);
298 goto finish;
299 }
300
301 t = pa_tagstruct_new(NULL, 0);
302 assert(t);
303 pa_tagstruct_putu32(t, PA_COMMAND_AUTH);
304 pa_tagstruct_putu32(t, tag = c->ctag++);
305 pa_tagstruct_put_arbitrary(t, c->auth_cookie, sizeof(c->auth_cookie));
306 pa_pstream_send_tagstruct(c->pstream, t);
307 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, setup_complete_callback, c);
308
309 pa_context_set_state(c, PA_CONTEXT_AUTHORIZING);
310
311 finish:
312
313 pa_context_unref(c);
314 }
315
316 static void on_connection(struct pa_socket_client *client, struct pa_iochannel*io, void *userdata) {
317 struct pa_context *c = userdata;
318 assert(client && c && c->state == PA_CONTEXT_CONNECTING);
319
320 pa_context_ref(c);
321
322 pa_socket_client_unref(client);
323 c->client = NULL;
324
325 if (!io) {
326 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
327 goto finish;
328 }
329
330 setup_context(c, io);
331
332 finish:
333 pa_context_unref(c);
334 }
335
336 static struct sockaddr *resolve_server(const char *server, size_t *len) {
337 struct sockaddr *sa;
338 struct addrinfo hints, *result = NULL;
339 char *port, host[256];
340 assert(server && len);
341
342 snprintf(host, sizeof(host), "%s", server);
343 host[strcspn(host, ":")] = 0;
344
345 if ((port = strrchr(server, ':')))
346 port++;
347
348 if (!port)
349 port = DEFAULT_PORT;
350
351 memset(&hints, 0, sizeof(hints));
352 hints.ai_family = PF_UNSPEC;
353 hints.ai_socktype = SOCK_STREAM;
354 hints.ai_protocol = 0;
355
356 if (getaddrinfo(host, port, &hints, &result) != 0)
357 return NULL;
358 assert(result);
359
360 sa = pa_xmalloc(*len = result->ai_addrlen);
361 memcpy(sa, result->ai_addr, *len);
362
363 freeaddrinfo(result);
364
365 return sa;
366 }
367
368 static int is_running(void) {
369 struct stat st;
370
371 if (DEFAULT_SERVER[0] != '/')
372 return 1;
373
374 if (stat(DEFAULT_SERVER, &st) < 0)
375 return 0;
376
377 return 1;
378 }
379
380 static int context_connect_spawn(struct pa_context *c, const struct pa_spawn_api *api) {
381 pid_t pid;
382 int status, r;
383 int fds[2] = { -1, -1} ;
384 struct pa_iochannel *io;
385
386 if (socketpair(AF_UNIX, SOCK_STREAM, 0, fds) < 0) {
387 pa_log(__FILE__": socketpair() failed: %s\n", strerror(errno));
388 pa_context_fail(c, PA_ERROR_INTERNAL);
389 goto fail;
390 }
391
392 if (api && api->prefork)
393 api->prefork();
394
395 if ((pid = fork()) < 0) {
396 pa_log(__FILE__": fork() failed: %s\n", strerror(errno));
397 pa_context_fail(c, PA_ERROR_INTERNAL);
398
399 if (api && api->postfork)
400 api->postfork();
401
402 goto fail;
403 } else if (!pid) {
404 char t[128];
405 char *p;
406 /* Child */
407
408 close(fds[0]);
409
410 if (api && api->atfork)
411 api->atfork();
412
413 if (!(p = getenv(ENV_DEFAULT_BINARY)))
414 p = POLYPAUDIO_BINARY;
415
416 snprintf(t, sizeof(t), "%s=1", ENV_AUTOSPAWNED);
417 putenv(t);
418
419 snprintf(t, sizeof(t), "-Lmodule-native-protocol-fd fd=%i", fds[1]);
420 execl(p, p, t, NULL);
421
422 exit(1);
423 }
424
425 /* Parent */
426
427 r = waitpid(pid, &status, 0);
428
429 if (api && api->postfork)
430 api->postfork();
431
432 if (r < 0) {
433 pa_log(__FILE__": waitpid() failed: %s\n", strerror(errno));
434 pa_context_fail(c, PA_ERROR_INTERNAL);
435 goto fail;
436 } else if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
437 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
438 goto fail;
439 }
440
441 close(fds[1]);
442
443 io = pa_iochannel_new(c->mainloop, fds[0], fds[0]);
444 setup_context(c, io);
445 return 0;
446
447 fail:
448 if (fds[0] != -1)
449 close(fds[0]);
450 if (fds[1] != -1)
451 close(fds[1]);
452
453 return -1;
454 }
455
456
457
458 int pa_context_connect(struct pa_context *c, const char *server, int spawn, const struct pa_spawn_api *api) {
459 int r = -1;
460 assert(c && c->ref >= 1 && c->state == PA_CONTEXT_UNCONNECTED);
461
462 if (!server)
463 if (!(server = getenv(ENV_DEFAULT_SERVER))) {
464 if (spawn && !is_running()) {
465 char *b;
466
467 if ((b = getenv(ENV_DISABLE_AUTOSPAWN)))
468 if (pa_parse_boolean(b) > 1)
469 return -1;
470
471 return context_connect_spawn(c, api);
472 }
473
474 server = DEFAULT_SERVER;
475 }
476
477 pa_context_ref(c);
478
479 assert(!c->client);
480
481 if (*server == '/') {
482 if (!(c->client = pa_socket_client_new_unix(c->mainloop, server))) {
483 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
484 goto finish;
485 }
486 } else {
487 struct sockaddr* sa;
488 size_t sa_len;
489
490 if (!(sa = resolve_server(server, &sa_len))) {
491 pa_context_fail(c, PA_ERROR_INVALIDSERVER);
492 goto finish;
493 }
494
495 c->client = pa_socket_client_new_sockaddr(c->mainloop, sa, sa_len);
496 pa_xfree(sa);
497
498 if (!c->client) {
499 pa_context_fail(c, PA_ERROR_CONNECTIONREFUSED);
500 goto finish;
501 }
502 }
503
504 pa_socket_client_set_callback(c->client, on_connection, c);
505 pa_context_set_state(c, PA_CONTEXT_CONNECTING);
506
507 r = 0;
508
509 finish:
510 pa_context_unref(c);
511
512 return r;
513 }
514
515 void pa_context_disconnect(struct pa_context *c) {
516 assert(c);
517 pa_context_set_state(c, PA_CONTEXT_TERMINATED);
518 }
519
520 enum pa_context_state pa_context_get_state(struct pa_context *c) {
521 assert(c && c->ref >= 1);
522 return c->state;
523 }
524
525 int pa_context_errno(struct pa_context *c) {
526 assert(c && c->ref >= 1);
527 return c->error;
528 }
529
530 void pa_context_set_state_callback(struct pa_context *c, void (*cb)(struct pa_context *c, void *userdata), void *userdata) {
531 assert(c && c->ref >= 1);
532 c->state_callback = cb;
533 c->state_userdata = userdata;
534 }
535
536 int pa_context_is_pending(struct pa_context *c) {
537 assert(c && c->ref >= 1);
538
539 /* pa_log("pstream: %i\n", pa_pstream_is_pending(c->pstream)); */
540 /* pa_log("pdispatch: %i\n", pa_pdispatch_is_pending(c->pdispatch)); */
541
542 return (c->pstream && pa_pstream_is_pending(c->pstream)) || (c->pdispatch && pa_pdispatch_is_pending(c->pdispatch)) || c->client;
543 }
544
545 static void set_dispatch_callbacks(struct pa_operation *o);
546
547 static void pdispatch_drain_callback(struct pa_pdispatch*pd, void *userdata) {
548 set_dispatch_callbacks(userdata);
549 }
550
551 static void pstream_drain_callback(struct pa_pstream *s, void *userdata) {
552 set_dispatch_callbacks(userdata);
553 }
554
555 static void set_dispatch_callbacks(struct pa_operation *o) {
556 int done = 1;
557 assert(o && o->context && o->context->ref >= 1 && o->ref >= 1 && o->context->state == PA_CONTEXT_READY);
558
559 pa_pstream_set_drain_callback(o->context->pstream, NULL, NULL);
560 pa_pdispatch_set_drain_callback(o->context->pdispatch, NULL, NULL);
561
562 if (pa_pdispatch_is_pending(o->context->pdispatch)) {
563 pa_pdispatch_set_drain_callback(o->context->pdispatch, pdispatch_drain_callback, o);
564 done = 0;
565 }
566
567 if (pa_pstream_is_pending(o->context->pstream)) {
568 pa_pstream_set_drain_callback(o->context->pstream, pstream_drain_callback, o);
569 done = 0;
570 }
571
572 if (!done)
573 pa_operation_ref(o);
574 else {
575 if (o->callback) {
576 void (*cb)(struct pa_context *c, void *userdata);
577 cb = (void (*)(struct pa_context*, void*)) o->callback;
578 cb(o->context, o->userdata);
579 }
580
581 pa_operation_done(o);
582 }
583
584 pa_operation_unref(o);
585 }
586
587 struct pa_operation* pa_context_drain(struct pa_context *c, void (*cb) (struct pa_context*c, void *userdata), void *userdata) {
588 struct pa_operation *o;
589 assert(c && c->ref >= 1);
590
591 if (c->state != PA_CONTEXT_READY)
592 return NULL;
593
594 if (!pa_context_is_pending(c))
595 return NULL;
596
597 o = pa_operation_new(c, NULL);
598 assert(o);
599 o->callback = cb;
600 o->userdata = userdata;
601
602 set_dispatch_callbacks(pa_operation_ref(o));
603
604 return o;
605 }
606
607 void pa_context_exit_daemon(struct pa_context *c) {
608 struct pa_tagstruct *t;
609 assert(c && c->ref >= 1);
610
611 t = pa_tagstruct_new(NULL, 0);
612 assert(t);
613 pa_tagstruct_putu32(t, PA_COMMAND_EXIT);
614 pa_tagstruct_putu32(t, c->ctag++);
615 pa_pstream_send_tagstruct(c->pstream, t);
616 }
617
618 void pa_context_simple_ack_callback(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
619 struct pa_operation *o = userdata;
620 int success = 1;
621 assert(pd && o && o->context && o->ref >= 1);
622
623 if (command != PA_COMMAND_REPLY) {
624 if (pa_context_handle_error(o->context, command, t) < 0)
625 goto finish;
626
627 success = 0;
628 } else if (!pa_tagstruct_eof(t)) {
629 pa_context_fail(o->context, PA_ERROR_PROTOCOL);
630 goto finish;
631 }
632
633 if (o->callback) {
634 void (*cb)(struct pa_context *c, int success, void *userdata) = o->callback;
635 cb(o->context, success, o->userdata);
636 }
637
638 finish:
639 pa_operation_done(o);
640 pa_operation_unref(o);
641 }
642
643 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) {
644 struct pa_tagstruct *t;
645 struct pa_operation *o;
646 uint32_t tag;
647 assert(c && cb);
648
649 o = pa_operation_new(c, NULL);
650 o->callback = cb;
651 o->userdata = userdata;
652
653 t = pa_tagstruct_new(NULL, 0);
654 pa_tagstruct_putu32(t, command);
655 pa_tagstruct_putu32(t, tag = c->ctag++);
656 pa_pstream_send_tagstruct(c->pstream, t);
657 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, internal_callback, o);
658
659 return pa_operation_ref(o);
660 }
661
662 const char* pa_get_library_version(void) {
663 return PACKAGE_VERSION;
664 }
665
666 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) {
667 struct pa_tagstruct *t;
668 struct pa_operation *o;
669 uint32_t tag;
670 assert(c && cb);
671
672 o = pa_operation_new(c, NULL);
673 o->callback = cb;
674 o->userdata = userdata;
675
676 t = pa_tagstruct_new(NULL, 0);
677 pa_tagstruct_putu32(t, PA_COMMAND_SET_DEFAULT_SINK);
678 pa_tagstruct_putu32(t, tag = c->ctag++);
679 pa_tagstruct_puts(t, name);
680 pa_pstream_send_tagstruct(c->pstream, t);
681 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, o);
682
683 return pa_operation_ref(o);
684 }
685
686 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) {
687 struct pa_tagstruct *t;
688 struct pa_operation *o;
689 uint32_t tag;
690 assert(c && cb);
691
692 o = pa_operation_new(c, NULL);
693 o->callback = cb;
694 o->userdata = userdata;
695
696 t = pa_tagstruct_new(NULL, 0);
697 pa_tagstruct_putu32(t, PA_COMMAND_SET_DEFAULT_SOURCE);
698 pa_tagstruct_putu32(t, tag = c->ctag++);
699 pa_tagstruct_puts(t, name);
700 pa_pstream_send_tagstruct(c->pstream, t);
701 pa_pdispatch_register_reply(c->pdispatch, tag, DEFAULT_TIMEOUT, pa_context_simple_ack_callback, o);
702
703 return pa_operation_ref(o);
704 }