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 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 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>
39 #include "subscribe.h"
41 #include "sink-input.h"
42 #include "pdispatch.h"
44 #include "pstream-util.h"
46 #include "socket-client.h"
47 #include "socket-util.h"
48 #include "authkey-prop.h"
51 #include "module-tunnel-sink-symdef.h"
52 PA_MODULE_DESCRIPTION("Tunnel module for sinks")
53 PA_MODULE_USAGE("server=<address> sink=<remote sink name> cookie=<filename> format=<sample format> channels=<number of channels> rate=<sample rate> sink_name=<name for the local sink>")
55 #include "module-tunnel-source-symdef.h"
56 PA_MODULE_DESCRIPTION("Tunnel module for sources")
57 PA_MODULE_USAGE("server=<address> source=<remote source name> cookie=<filename> format=<sample format> channels=<number of channels> rate=<sample rate> source_name=<name for the local source>")
60 PA_MODULE_AUTHOR("Lennart Poettering")
61 PA_MODULE_VERSION(PACKAGE_VERSION
)
63 #define DEFAULT_SINK_NAME "tunnel"
64 #define DEFAULT_SOURCE_NAME "tunnel"
66 #define DEFAULT_TLENGTH (44100*2*2/10) //(10240*8)
67 #define DEFAULT_MAXLENGTH ((DEFAULT_TLENGTH*3)/2)
68 #define DEFAULT_MINREQ 512
69 #define DEFAULT_PREBUF (DEFAULT_TLENGTH-DEFAULT_MINREQ)
70 #define DEFAULT_FRAGSIZE 1024
72 #define DEFAULT_TIMEOUT 5
74 #define LATENCY_INTERVAL 10
76 static const char* const valid_modargs
[] = {
92 static void command_stream_killed(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
);
95 static void command_request(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
);
98 static const struct pa_pdispatch_command command_table
[PA_COMMAND_MAX
] = {
100 [PA_COMMAND_REQUEST
] = { command_request
},
102 [PA_COMMAND_PLAYBACK_STREAM_KILLED
] = { command_stream_killed
},
103 [PA_COMMAND_RECORD_STREAM_KILLED
] = { command_stream_killed
},
107 struct pa_socket_client
*client
;
108 struct pa_pstream
*pstream
;
109 struct pa_pdispatch
*pdispatch
;
114 struct pa_sink
*sink
;
115 uint32_t requested_bytes
;
118 struct pa_source
*source
;
121 struct pa_module
*module
;
122 struct pa_core
*core
;
124 uint8_t auth_cookie
[PA_NATIVE_COOKIE_LENGTH
];
127 uint32_t device_index
;
130 pa_usec_t host_latency
;
132 struct pa_time_event
*time_event
;
134 int auth_cookie_in_property
;
137 static void close_stuff(struct userdata
*u
) {
141 pa_pstream_close(u
->pstream
);
142 pa_pstream_unref(u
->pstream
);
147 pa_pdispatch_unref(u
->pdispatch
);
152 pa_socket_client_unref(u
->client
);
158 pa_sink_disconnect(u
->sink
);
159 pa_sink_unref(u
->sink
);
164 pa_source_disconnect(u
->source
);
165 pa_source_unref(u
->source
);
171 u
->core
->mainloop
->time_free(u
->time_event
);
172 u
->time_event
= NULL
;
176 static void die(struct userdata
*u
) {
179 pa_module_unload_request(u
->module
);
182 static void command_stream_killed(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
) {
183 struct userdata
*u
= userdata
;
184 assert(pd
&& t
&& u
&& u
->pdispatch
== pd
);
186 pa_log(__FILE__
": stream killed\n");
191 static void send_prebuf_request(struct userdata
*u
) {
192 struct pa_tagstruct
*t
;
194 t
= pa_tagstruct_new(NULL
, 0);
195 pa_tagstruct_putu32(t
, PA_COMMAND_PREBUF_PLAYBACK_STREAM
);
196 pa_tagstruct_putu32(t
, u
->ctag
++);
197 pa_tagstruct_putu32(t
, u
->channel
);
198 pa_pstream_send_tagstruct(u
->pstream
, t
);
201 static void send_bytes(struct userdata
*u
) {
207 while (u
->requested_bytes
> 0) {
208 struct pa_memchunk chunk
;
209 if (pa_sink_render(u
->sink
, u
->requested_bytes
, &chunk
) < 0) {
212 if (u
->requested_bytes
>= DEFAULT_TLENGTH
-DEFAULT_PREBUF
)
213 send_prebuf_request(u
);
218 pa_pstream_send_memblock(u
->pstream
, u
->channel
, 0, &chunk
);
219 pa_memblock_unref(chunk
.memblock
);
221 if (chunk
.length
> u
->requested_bytes
)
222 u
->requested_bytes
= 0;
224 u
->requested_bytes
-= chunk
.length
;
228 static void command_request(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
) {
229 struct userdata
*u
= userdata
;
230 uint32_t bytes
, channel
;
231 assert(pd
&& command
== PA_COMMAND_REQUEST
&& t
&& u
&& u
->pdispatch
== pd
);
233 if (pa_tagstruct_getu32(t
, &channel
) < 0 ||
234 pa_tagstruct_getu32(t
, &bytes
) < 0 ||
235 !pa_tagstruct_eof(t
)) {
236 pa_log(__FILE__
": invalid protocol reply\n");
241 if (channel
!= u
->channel
) {
242 pa_log(__FILE__
": recieved data for invalid channel\n");
247 u
->requested_bytes
+= bytes
;
253 static void stream_get_latency_callback(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
) {
254 struct userdata
*u
= userdata
;
255 pa_usec_t buffer_usec
, sink_usec
, source_usec
, transport_usec
;
257 uint32_t queue_length
;
258 struct timeval local
, remote
, now
;
259 assert(pd
&& u
&& t
);
261 if (command
!= PA_COMMAND_REPLY
) {
262 if (command
== PA_COMMAND_ERROR
)
263 pa_log(__FILE__
": failed to get latency.\n");
265 pa_log(__FILE__
": protocol error.\n");
270 if (pa_tagstruct_get_usec(t
, &buffer_usec
) < 0 ||
271 pa_tagstruct_get_usec(t
, &sink_usec
) < 0 ||
272 pa_tagstruct_get_usec(t
, &source_usec
) < 0 ||
273 pa_tagstruct_get_boolean(t
, &playing
) < 0 ||
274 pa_tagstruct_getu32(t
, &queue_length
) < 0 ||
275 pa_tagstruct_get_timeval(t
, &local
) < 0 ||
276 pa_tagstruct_get_timeval(t
, &remote
) < 0 ||
277 !pa_tagstruct_eof(t
)) {
278 pa_log(__FILE__
": invalid reply.\n");
283 gettimeofday(&now
, NULL
);
285 if (pa_timeval_cmp(&local
, &remote
) < 0 && pa_timeval_cmp(&remote
, &now
)) {
286 /* local and remote seem to have synchronized clocks */
288 transport_usec
= pa_timeval_diff(&remote
, &local
);
290 transport_usec
= pa_timeval_diff(&now
, &remote
);
293 transport_usec
= pa_timeval_diff(&now
, &local
)/2;
296 u
->host_latency
= sink_usec
+ transport_usec
;
298 u
->host_latency
= source_usec
+ transport_usec
;
299 if (u
->host_latency
> sink_usec
)
300 u
->host_latency
-= sink_usec
;
305 /* pa_log(__FILE__": estimated host latency: %0.0f usec\n", (double) u->host_latency); */
308 static void request_latency(struct userdata
*u
) {
309 struct pa_tagstruct
*t
;
314 t
= pa_tagstruct_new(NULL
, 0);
316 pa_tagstruct_putu32(t
, PA_COMMAND_GET_PLAYBACK_LATENCY
);
318 pa_tagstruct_putu32(t
, PA_COMMAND_GET_RECORD_LATENCY
);
320 pa_tagstruct_putu32(t
, tag
= u
->ctag
++);
321 pa_tagstruct_putu32(t
, u
->channel
);
323 gettimeofday(&now
, NULL
);
324 pa_tagstruct_put_timeval(t
, &now
);
326 pa_pstream_send_tagstruct(u
->pstream
, t
);
327 pa_pdispatch_register_reply(u
->pdispatch
, tag
, DEFAULT_TIMEOUT
, stream_get_latency_callback
, u
);
330 static void create_stream_callback(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
) {
331 struct userdata
*u
= userdata
;
332 assert(pd
&& u
&& u
->pdispatch
== pd
);
334 if (command
!= PA_COMMAND_REPLY
) {
335 if (command
== PA_COMMAND_ERROR
)
336 pa_log(__FILE__
": failed to create stream.\n");
338 pa_log(__FILE__
": protocol error.\n");
343 if (pa_tagstruct_getu32(t
, &u
->channel
) < 0 ||
344 pa_tagstruct_getu32(t
, &u
->device_index
) < 0 ||
346 pa_tagstruct_getu32(t
, &u
->requested_bytes
) < 0 ||
348 !pa_tagstruct_eof(t
)) {
349 pa_log(__FILE__
": invalid reply.\n");
360 static void setup_complete_callback(struct pa_pdispatch
*pd
, uint32_t command
, uint32_t tag
, struct pa_tagstruct
*t
, void *userdata
) {
361 struct userdata
*u
= userdata
;
362 struct pa_tagstruct
*reply
;
363 char name
[256], un
[128], hn
[128];
364 assert(pd
&& u
&& u
->pdispatch
== pd
);
366 if (command
!= PA_COMMAND_REPLY
|| !pa_tagstruct_eof(t
)) {
367 if (command
== PA_COMMAND_ERROR
)
368 pa_log(__FILE__
": failed to authenticate\n");
370 pa_log(__FILE__
": protocol error.\n");
375 snprintf(name
, sizeof(name
), "Tunnel from host '%s', user '%s', sink '%s'",
376 pa_get_host_name(hn
, sizeof(hn
)),
377 pa_get_user_name(un
, sizeof(un
)),
380 snprintf(name
, sizeof(name
), "Tunnel from host '%s', user '%s', source '%s'",
381 pa_get_host_name(hn
, sizeof(hn
)),
382 pa_get_user_name(un
, sizeof(un
)),
386 reply
= pa_tagstruct_new(NULL
, 0);
387 pa_tagstruct_putu32(reply
, PA_COMMAND_SET_CLIENT_NAME
);
388 pa_tagstruct_putu32(reply
, tag
= u
->ctag
++);
389 pa_tagstruct_puts(reply
, name
);
390 pa_pstream_send_tagstruct(u
->pstream
, reply
);
391 /* We ignore the server's reply here */
393 reply
= pa_tagstruct_new(NULL
, 0);
395 pa_tagstruct_putu32(reply
, PA_COMMAND_CREATE_PLAYBACK_STREAM
);
396 pa_tagstruct_putu32(reply
, tag
= u
->ctag
++);
397 pa_tagstruct_puts(reply
, name
);
398 pa_tagstruct_put_sample_spec(reply
, &u
->sink
->sample_spec
);
399 pa_tagstruct_putu32(reply
, PA_INVALID_INDEX
);
400 pa_tagstruct_puts(reply
, u
->sink_name
);
401 pa_tagstruct_putu32(reply
, DEFAULT_MAXLENGTH
);
402 pa_tagstruct_put_boolean(reply
, 0);
403 pa_tagstruct_putu32(reply
, DEFAULT_TLENGTH
);
404 pa_tagstruct_putu32(reply
, DEFAULT_PREBUF
);
405 pa_tagstruct_putu32(reply
, DEFAULT_MINREQ
);
406 pa_tagstruct_putu32(reply
, PA_VOLUME_NORM
);
408 pa_tagstruct_putu32(reply
, PA_COMMAND_CREATE_RECORD_STREAM
);
409 pa_tagstruct_putu32(reply
, tag
= u
->ctag
++);
410 pa_tagstruct_puts(reply
, name
);
411 pa_tagstruct_put_sample_spec(reply
, &u
->source
->sample_spec
);
412 pa_tagstruct_putu32(reply
, PA_INVALID_INDEX
);
413 pa_tagstruct_puts(reply
, u
->source_name
);
414 pa_tagstruct_putu32(reply
, DEFAULT_MAXLENGTH
);
415 pa_tagstruct_put_boolean(reply
, 0);
416 pa_tagstruct_putu32(reply
, DEFAULT_FRAGSIZE
);
419 pa_pstream_send_tagstruct(u
->pstream
, reply
);
420 pa_pdispatch_register_reply(u
->pdispatch
, tag
, DEFAULT_TIMEOUT
, create_stream_callback
, u
);
423 static void pstream_die_callback(struct pa_pstream
*p
, void *userdata
) {
424 struct userdata
*u
= userdata
;
427 pa_log(__FILE__
": stream died.\n");
432 static void pstream_packet_callback(struct pa_pstream
*p
, struct pa_packet
*packet
, void *userdata
) {
433 struct userdata
*u
= userdata
;
434 assert(p
&& packet
&& u
);
436 if (pa_pdispatch_run(u
->pdispatch
, packet
, u
) < 0) {
437 pa_log(__FILE__
": invalid packet\n");
443 static void pstream_memblock_callback(struct pa_pstream
*p
, uint32_t channel
, uint32_t delta
, const struct pa_memchunk
*chunk
, void *userdata
) {
444 struct userdata
*u
= userdata
;
445 assert(p
&& chunk
&& u
);
447 if (channel
!= u
->channel
) {
448 pa_log(__FILE__
": recieved memory block on bad channel.\n");
453 pa_source_post(u
->source
, chunk
);
457 static void on_connection(struct pa_socket_client
*sc
, struct pa_iochannel
*io
, void *userdata
) {
458 struct userdata
*u
= userdata
;
459 struct pa_tagstruct
*t
;
461 assert(sc
&& u
&& u
->client
== sc
);
463 pa_socket_client_unref(u
->client
);
467 pa_log(__FILE__
": connection failed.\n");
468 pa_module_unload_request(u
->module
);
472 u
->pstream
= pa_pstream_new(u
->core
->mainloop
, io
, u
->core
->memblock_stat
);
473 u
->pdispatch
= pa_pdispatch_new(u
->core
->mainloop
, command_table
, PA_COMMAND_MAX
);
475 pa_pstream_set_die_callback(u
->pstream
, pstream_die_callback
, u
);
476 pa_pstream_set_recieve_packet_callback(u
->pstream
, pstream_packet_callback
, u
);
478 pa_pstream_set_recieve_memblock_callback(u
->pstream
, pstream_memblock_callback
, u
);
481 t
= pa_tagstruct_new(NULL
, 0);
482 pa_tagstruct_putu32(t
, PA_COMMAND_AUTH
);
483 pa_tagstruct_putu32(t
, tag
= u
->ctag
++);
484 pa_tagstruct_put_arbitrary(t
, u
->auth_cookie
, sizeof(u
->auth_cookie
));
485 pa_pstream_send_tagstruct(u
->pstream
, t
);
486 pa_pdispatch_register_reply(u
->pdispatch
, tag
, DEFAULT_TIMEOUT
, setup_complete_callback
, u
);
491 static void sink_notify(struct pa_sink
*sink
) {
493 assert(sink
&& sink
->userdata
);
499 static pa_usec_t
sink_get_latency(struct pa_sink
*sink
) {
503 assert(sink
&& sink
->userdata
);
508 if (l
> u
->requested_bytes
) {
509 l
-= u
->requested_bytes
;
510 usec
+= pa_bytes_to_usec(l
, &u
->sink
->sample_spec
);
513 usec
+= u
->host_latency
;
518 static pa_usec_t
source_get_latency(struct pa_source
*source
) {
520 assert(source
&& source
->userdata
);
521 u
= source
->userdata
;
523 return u
->host_latency
;
527 static void timeout_callback(struct pa_mainloop_api
*m
, struct pa_time_event
*e
, const struct timeval
*tv
, void *userdata
) {
528 struct userdata
*u
= userdata
;
534 gettimeofday(&ntv
, NULL
);
535 ntv
.tv_sec
+= LATENCY_INTERVAL
;
536 m
->time_restart(e
, &ntv
);
539 static int load_key(struct userdata
*u
, const char*fn
) {
542 u
->auth_cookie_in_property
= 0;
544 if (!fn
&& pa_authkey_prop_get(u
->core
, PA_NATIVE_COOKIE_PROPERTY_NAME
, u
->auth_cookie
, sizeof(u
->auth_cookie
)) >= 0) {
545 pa_log(__FILE__
": using already loaded auth cookie.\n");
546 pa_authkey_prop_ref(u
->core
, PA_NATIVE_COOKIE_PROPERTY_NAME
);
547 u
->auth_cookie_in_property
= 1;
552 fn
= PA_NATIVE_COOKIE_FILE
;
554 if (pa_authkey_load_from_home(fn
, u
->auth_cookie
, sizeof(u
->auth_cookie
)) < 0)
557 pa_log(__FILE__
": loading cookie from disk.\n");
559 if (pa_authkey_prop_put(u
->core
, PA_NATIVE_COOKIE_PROPERTY_NAME
, u
->auth_cookie
, sizeof(u
->auth_cookie
)) >= 0)
560 u
->auth_cookie_in_property
= 1;
565 int pa__init(struct pa_core
*c
, struct pa_module
*m
) {
566 struct pa_modargs
*ma
= NULL
;
567 struct userdata
*u
= NULL
;
568 struct pa_sample_spec ss
;
572 if (!(ma
= pa_modargs_new(m
->argument
, valid_modargs
))) {
573 pa_log(__FILE__
": failed to parse module arguments\n");
577 u
= pa_xmalloc(sizeof(struct userdata
));
584 u
->server_name
= NULL
;
586 u
->sink_name
= pa_xstrdup(pa_modargs_get_value(ma
, "sink", NULL
));;
588 u
->requested_bytes
= 0;
590 u
->source_name
= pa_xstrdup(pa_modargs_get_value(ma
, "source", NULL
));;
594 u
->device_index
= u
->channel
= PA_INVALID_INDEX
;
596 u
->auth_cookie_in_property
= 0;
597 u
->time_event
= NULL
;
599 if (load_key(u
, pa_modargs_get_value(ma
, "cookie", NULL
)) < 0)
602 if (!(u
->server_name
= pa_xstrdup(pa_modargs_get_value(ma
, "server", NULL
)))) {
603 pa_log(__FILE__
": no server specified.\n");
607 ss
= c
->default_sample_spec
;
608 if (pa_modargs_get_sample_spec(ma
, &ss
) < 0) {
609 pa_log(__FILE__
": invalid sample format specification\n");
613 if (u
->server_name
[0] == '/')
614 u
->client
= pa_socket_client_new_unix(c
->mainloop
, u
->server_name
);
619 if (!(sa
= pa_resolve_server(u
->server_name
, &len
, PA_NATIVE_DEFAULT_PORT
))) {
620 pa_log(__FILE__
": failed to resolve server '%s'\n", u
->server_name
);
624 u
->client
= pa_socket_client_new_sockaddr(c
->mainloop
, sa
, len
);
631 pa_socket_client_set_callback(u
->client
, on_connection
, u
);
634 if (!(u
->sink
= pa_sink_new(c
, pa_modargs_get_value(ma
, "sink_name", DEFAULT_SINK_NAME
), 0, &ss
))) {
635 pa_log(__FILE__
": failed to create sink.\n");
639 u
->sink
->notify
= sink_notify
;
640 u
->sink
->get_latency
= sink_get_latency
;
641 u
->sink
->userdata
= u
;
642 u
->sink
->description
= pa_sprintf_malloc("Tunnel to '%s%s%s'", u
->sink_name
? u
->sink_name
: "", u
->sink_name
? "@" : "", u
->server_name
);
644 pa_sink_set_owner(u
->sink
, m
);
646 if (!(u
->source
= pa_source_new(c
, pa_modargs_get_value(ma
, "source_name", DEFAULT_SOURCE_NAME
), 0, &ss
))) {
647 pa_log(__FILE__
": failed to create source.\n");
651 u
->source
->get_latency
= source_get_latency
;
652 u
->source
->userdata
= u
;
653 u
->source
->description
= pa_sprintf_malloc("Tunnel to '%s%s%s'", u
->source_name
? u
->source_name
: "", u
->source_name
? "@" : "", u
->server_name
);
655 pa_source_set_owner(u
->source
, m
);
658 gettimeofday(&ntv
, NULL
);
659 ntv
.tv_sec
+= LATENCY_INTERVAL
;
660 u
->time_event
= c
->mainloop
->time_new(c
->mainloop
, &ntv
, timeout_callback
, u
);
674 void pa__done(struct pa_core
*c
, struct pa_module
*m
) {
678 if (!(u
= m
->userdata
))
683 if (u
->auth_cookie_in_property
)
684 pa_authkey_prop_unref(c
, PA_NATIVE_COOKIE_PROPERTY_NAME
);
687 pa_xfree(u
->sink_name
);
689 pa_xfree(u
->source_name
);
691 pa_xfree(u
->server_name
);