add new tool paplay
[profile/ivi/pulseaudio-panda.git] / polyp / protocol-native.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 <string.h>
27 #include <stdio.h>
28 #include <assert.h>
29 #include <stdlib.h>
30
31 #include "protocol-native.h"
32 #include "native-common.h"
33 #include "packet.h"
34 #include "client.h"
35 #include "source-output.h"
36 #include "sink-input.h"
37 #include "pstream.h"
38 #include "tagstruct.h"
39 #include "pdispatch.h"
40 #include "pstream-util.h"
41 #include "authkey.h"
42 #include "namereg.h"
43 #include "scache.h"
44 #include "xmalloc.h"
45 #include "util.h"
46 #include "subscribe.h"
47 #include "log.h"
48 #include "autoload.h"
49
50 struct connection;
51 struct pa_protocol_native;
52
53 struct record_stream {
54     struct connection *connection;
55     uint32_t index;
56     struct pa_source_output *source_output;
57     struct pa_memblockq *memblockq;
58     size_t fragment_size;
59 };
60
61 struct playback_stream {
62     int type;
63     struct connection *connection;
64     uint32_t index;
65     struct pa_sink_input *sink_input;
66     struct pa_memblockq *memblockq;
67     size_t requested_bytes;
68     int drain_request;
69     uint32_t drain_tag;
70 };
71
72 struct upload_stream {
73     int type;
74     struct connection *connection;
75     uint32_t index;
76     struct pa_memchunk memchunk;
77     size_t length;
78     char *name;
79     struct pa_sample_spec sample_spec;
80 };
81
82 struct output_stream {
83     int type;
84 };
85
86 enum {
87     UPLOAD_STREAM,
88     PLAYBACK_STREAM
89 };
90
91 struct connection {
92     int authorized;
93     struct pa_protocol_native *protocol;
94     struct pa_client *client;
95     struct pa_pstream *pstream;
96     struct pa_pdispatch *pdispatch;
97     struct pa_idxset *record_streams, *output_streams;
98     uint32_t rrobin_index;
99     struct pa_subscription *subscription;
100 };
101
102 struct pa_protocol_native {
103     struct pa_module *module;
104     int public;
105     struct pa_core *core;
106     struct pa_socket_server *server;
107     struct pa_idxset *connections;
108     uint8_t auth_cookie[PA_NATIVE_COOKIE_LENGTH];
109 };
110
111 static int sink_input_peek_cb(struct pa_sink_input *i, struct pa_memchunk *chunk);
112 static void sink_input_drop_cb(struct pa_sink_input *i, const struct pa_memchunk *chunk, size_t length);
113 static void sink_input_kill_cb(struct pa_sink_input *i);
114 static pa_usec_t sink_input_get_latency_cb(struct pa_sink_input *i);
115
116 static void request_bytes(struct playback_stream*s);
117
118 static void source_output_kill_cb(struct pa_source_output *o);
119 static void source_output_push_cb(struct pa_source_output *o, const struct pa_memchunk *chunk);
120 static pa_usec_t source_output_get_latency_cb(struct pa_source_output *o);
121
122 static void command_exit(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
123 static void command_create_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
124 static void command_drain_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
125 static void command_create_record_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
126 static void command_delete_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
127 static void command_auth(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
128 static void command_set_client_name(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
129 static void command_lookup(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
130 static void command_stat(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
131 static void command_get_playback_latency(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
132 static void command_get_record_latency(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
133 static void command_create_upload_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
134 static void command_finish_upload_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
135 static void command_play_sample(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
136 static void command_remove_sample(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
137 static void command_get_info(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
138 static void command_get_info_list(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
139 static void command_get_server_info(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
140 static void command_subscribe(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
141 static void command_set_volume(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
142 static void command_cork_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
143 static void command_flush_or_trigger_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
144 static void command_set_default_sink_or_source(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
145 static void command_set_stream_name(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
146 static void command_kill(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
147 static void command_load_module(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
148 static void command_unload_module(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
149 static void command_add_autoload(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
150 static void command_remove_autoload(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
151 static void command_get_autoload_info(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
152 static void command_get_autoload_info_list(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
153 static void command_cork_record_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
154 static void command_flush_record_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata);
155
156 static const struct pa_pdispatch_command command_table[PA_COMMAND_MAX] = {
157     [PA_COMMAND_ERROR] = { NULL },
158     [PA_COMMAND_TIMEOUT] = { NULL },
159     [PA_COMMAND_REPLY] = { NULL },
160     [PA_COMMAND_CREATE_PLAYBACK_STREAM] = { command_create_playback_stream },
161     [PA_COMMAND_DELETE_PLAYBACK_STREAM] = { command_delete_stream },
162     [PA_COMMAND_DRAIN_PLAYBACK_STREAM] = { command_drain_playback_stream },
163     [PA_COMMAND_CREATE_RECORD_STREAM] = { command_create_record_stream },
164     [PA_COMMAND_DELETE_RECORD_STREAM] = { command_delete_stream },
165     [PA_COMMAND_AUTH] = { command_auth },
166     [PA_COMMAND_REQUEST] = { NULL },
167     [PA_COMMAND_EXIT] = { command_exit },
168     [PA_COMMAND_SET_CLIENT_NAME] = { command_set_client_name },
169     [PA_COMMAND_LOOKUP_SINK] = { command_lookup },
170     [PA_COMMAND_LOOKUP_SOURCE] = { command_lookup },
171     [PA_COMMAND_STAT] = { command_stat },
172     [PA_COMMAND_GET_PLAYBACK_LATENCY] = { command_get_playback_latency },
173     [PA_COMMAND_GET_RECORD_LATENCY] = { command_get_record_latency },
174     [PA_COMMAND_CREATE_UPLOAD_STREAM] = { command_create_upload_stream },
175     [PA_COMMAND_DELETE_UPLOAD_STREAM] = { command_delete_stream },
176     [PA_COMMAND_FINISH_UPLOAD_STREAM] = { command_finish_upload_stream },
177     [PA_COMMAND_PLAY_SAMPLE] = { command_play_sample },
178     [PA_COMMAND_REMOVE_SAMPLE] = { command_remove_sample },
179     [PA_COMMAND_GET_SINK_INFO] = { command_get_info },
180     [PA_COMMAND_GET_SOURCE_INFO] = { command_get_info },
181     [PA_COMMAND_GET_CLIENT_INFO] = { command_get_info },
182     [PA_COMMAND_GET_MODULE_INFO] = { command_get_info },
183     [PA_COMMAND_GET_SINK_INPUT_INFO] = { command_get_info },
184     [PA_COMMAND_GET_SOURCE_OUTPUT_INFO] = { command_get_info },
185     [PA_COMMAND_GET_SAMPLE_INFO] = { command_get_info },
186     [PA_COMMAND_GET_SINK_INFO_LIST] = { command_get_info_list },
187     [PA_COMMAND_GET_SOURCE_INFO_LIST] = { command_get_info_list },
188     [PA_COMMAND_GET_MODULE_INFO_LIST] = { command_get_info_list },
189     [PA_COMMAND_GET_CLIENT_INFO_LIST] = { command_get_info_list },
190     [PA_COMMAND_GET_SINK_INPUT_INFO_LIST] = { command_get_info_list },
191     [PA_COMMAND_GET_SOURCE_OUTPUT_INFO_LIST] = { command_get_info_list },
192     [PA_COMMAND_GET_SAMPLE_INFO_LIST] = { command_get_info_list },
193     [PA_COMMAND_GET_SERVER_INFO] = { command_get_server_info },
194     [PA_COMMAND_SUBSCRIBE] = { command_subscribe },
195
196     [PA_COMMAND_SET_SINK_VOLUME] = { command_set_volume },
197     [PA_COMMAND_SET_SINK_INPUT_VOLUME] = { command_set_volume },
198     
199     [PA_COMMAND_CORK_PLAYBACK_STREAM] = { command_cork_playback_stream },
200     [PA_COMMAND_FLUSH_PLAYBACK_STREAM] = { command_flush_or_trigger_playback_stream },
201     [PA_COMMAND_TRIGGER_PLAYBACK_STREAM] = { command_flush_or_trigger_playback_stream },
202     [PA_COMMAND_PREBUF_PLAYBACK_STREAM] = { command_flush_or_trigger_playback_stream },
203     
204     [PA_COMMAND_CORK_RECORD_STREAM] = { command_cork_record_stream },
205     [PA_COMMAND_FLUSH_RECORD_STREAM] = { command_flush_record_stream },
206     
207     [PA_COMMAND_SET_DEFAULT_SINK] = { command_set_default_sink_or_source },
208     [PA_COMMAND_SET_DEFAULT_SOURCE] = { command_set_default_sink_or_source },
209     [PA_COMMAND_SET_PLAYBACK_STREAM_NAME] = { command_set_stream_name }, 
210     [PA_COMMAND_SET_RECORD_STREAM_NAME] = { command_set_stream_name },
211     [PA_COMMAND_KILL_CLIENT] = { command_kill },
212     [PA_COMMAND_KILL_SINK_INPUT] = { command_kill },
213     [PA_COMMAND_KILL_SOURCE_OUTPUT] = { command_kill },
214     [PA_COMMAND_LOAD_MODULE] = { command_load_module },
215     [PA_COMMAND_UNLOAD_MODULE] = { command_unload_module },
216     [PA_COMMAND_GET_AUTOLOAD_INFO] = { command_get_autoload_info },
217     [PA_COMMAND_GET_AUTOLOAD_INFO_LIST] = { command_get_autoload_info_list },
218     [PA_COMMAND_ADD_AUTOLOAD] = { command_add_autoload },
219     [PA_COMMAND_REMOVE_AUTOLOAD] = { command_remove_autoload },
220
221 };
222
223 /* structure management */
224
225 static struct upload_stream* upload_stream_new(struct connection *c, const struct pa_sample_spec *ss, const char *name, size_t length) {
226     struct upload_stream *s;
227     assert(c && ss && name && length);
228     
229     s = pa_xmalloc(sizeof(struct upload_stream));
230     s->type = UPLOAD_STREAM;
231     s->connection = c;
232     s->sample_spec = *ss;
233     s->name = pa_xstrdup(name);
234
235     s->memchunk.memblock = NULL;
236     s->memchunk.index = 0;
237     s->memchunk.length = 0;
238
239     s->length = length;
240     
241     pa_idxset_put(c->output_streams, s, &s->index);
242     return s;
243 }
244
245 static void upload_stream_free(struct upload_stream *o) {
246     assert(o && o->connection);
247
248     pa_idxset_remove_by_data(o->connection->output_streams, o, NULL);
249
250     pa_xfree(o->name);
251     
252     if (o->memchunk.memblock)
253         pa_memblock_unref(o->memchunk.memblock);
254     
255     pa_xfree(o);
256 }
257
258 static struct record_stream* record_stream_new(struct connection *c, struct pa_source *source, const struct pa_sample_spec *ss, const char *name, size_t maxlength, size_t fragment_size) {
259     struct record_stream *s;
260     struct pa_source_output *source_output;
261     size_t base;
262     assert(c && source && ss && name && maxlength);
263
264     if (!(source_output = pa_source_output_new(source, name, ss, -1)))
265         return NULL;
266
267     s = pa_xmalloc(sizeof(struct record_stream));
268     s->connection = c;
269     s->source_output = source_output;
270     s->source_output->push = source_output_push_cb;
271     s->source_output->kill = source_output_kill_cb;
272     s->source_output->get_latency = source_output_get_latency_cb;
273     s->source_output->userdata = s;
274     s->source_output->owner = c->protocol->module;
275     s->source_output->client = c->client;
276
277     s->memblockq = pa_memblockq_new(maxlength, 0, base = pa_frame_size(ss), 0, 0, c->protocol->core->memblock_stat);
278     assert(s->memblockq);
279
280     s->fragment_size = (fragment_size/base)*base;
281     if (!s->fragment_size)
282         s->fragment_size = base;
283
284     pa_idxset_put(c->record_streams, s, &s->index);
285     return s;
286 }
287
288 static void record_stream_free(struct record_stream* r) {
289     assert(r && r->connection);
290
291     pa_idxset_remove_by_data(r->connection->record_streams, r, NULL);
292     pa_source_output_disconnect(r->source_output);
293     pa_source_output_unref(r->source_output);
294     pa_memblockq_free(r->memblockq);
295     pa_xfree(r);
296 }
297
298 static struct playback_stream* playback_stream_new(struct connection *c, struct pa_sink *sink, const struct pa_sample_spec *ss, const char *name,
299                                                    size_t maxlength,
300                                                    size_t tlength,
301                                                    size_t prebuf,
302                                                    size_t minreq,
303                                                    pa_volume_t volume) {
304     struct playback_stream *s;
305     struct pa_sink_input *sink_input;
306     assert(c && sink && ss && name && maxlength);
307
308     if (!(sink_input = pa_sink_input_new(sink, name, ss, 0, -1)))
309         return NULL;
310     
311     s = pa_xmalloc(sizeof(struct playback_stream));
312     s->type = PLAYBACK_STREAM;
313     s->connection = c;
314     s->sink_input = sink_input;
315     
316     s->sink_input->peek = sink_input_peek_cb;
317     s->sink_input->drop = sink_input_drop_cb;
318     s->sink_input->kill = sink_input_kill_cb;
319     s->sink_input->get_latency = sink_input_get_latency_cb;
320     s->sink_input->userdata = s;
321     s->sink_input->owner = c->protocol->module;
322     s->sink_input->client = c->client;
323     
324     s->memblockq = pa_memblockq_new(maxlength, tlength, pa_frame_size(ss), prebuf, minreq, c->protocol->core->memblock_stat);
325     assert(s->memblockq);
326
327     s->requested_bytes = 0;
328     s->drain_request = 0;
329
330     s->sink_input->volume = volume;
331     
332     pa_idxset_put(c->output_streams, s, &s->index);
333     return s;
334 }
335
336 static void playback_stream_free(struct playback_stream* p) {
337     assert(p && p->connection);
338
339     if (p->drain_request)
340         pa_pstream_send_error(p->connection->pstream, p->drain_tag, PA_ERROR_NOENTITY);
341
342     pa_idxset_remove_by_data(p->connection->output_streams, p, NULL);
343     pa_sink_input_disconnect(p->sink_input);
344     pa_sink_input_unref(p->sink_input);
345     pa_memblockq_free(p->memblockq);
346     pa_xfree(p);
347 }
348
349 static void connection_free(struct connection *c) {
350     struct record_stream *r;
351     struct output_stream *o;
352     assert(c && c->protocol);
353
354     pa_idxset_remove_by_data(c->protocol->connections, c, NULL);
355     while ((r = pa_idxset_first(c->record_streams, NULL)))
356         record_stream_free(r);
357     pa_idxset_free(c->record_streams, NULL, NULL);
358
359     while ((o = pa_idxset_first(c->output_streams, NULL)))
360         if (o->type == PLAYBACK_STREAM)
361             playback_stream_free((struct playback_stream*) o);
362         else
363             upload_stream_free((struct upload_stream*) o);
364     pa_idxset_free(c->output_streams, NULL, NULL);
365
366     pa_pdispatch_unref(c->pdispatch);
367     pa_pstream_close(c->pstream);
368     pa_pstream_unref(c->pstream);
369     pa_client_free(c->client);
370
371     if (c->subscription)
372         pa_subscription_free(c->subscription);
373     
374     pa_xfree(c);
375 }
376
377 static void request_bytes(struct playback_stream *s) {
378     struct pa_tagstruct *t;
379     size_t l;
380     assert(s);
381
382     if (!(l = pa_memblockq_missing(s->memblockq)))
383         return;
384     
385     if (l <= s->requested_bytes)
386         return;
387
388     l -= s->requested_bytes;
389
390     if (l < pa_memblockq_get_minreq(s->memblockq))
391         return;
392     
393     s->requested_bytes += l;
394
395     t = pa_tagstruct_new(NULL, 0);
396     assert(t);
397     pa_tagstruct_putu32(t, PA_COMMAND_REQUEST);
398     pa_tagstruct_putu32(t, (uint32_t) -1); /* tag */
399     pa_tagstruct_putu32(t, s->index);
400     pa_tagstruct_putu32(t, l);
401     pa_pstream_send_tagstruct(s->connection->pstream, t);
402
403 /*     pa_log(__FILE__": Requesting %u bytes\n", l); */
404 }
405
406 static void send_memblock(struct connection *c) {
407     uint32_t start;
408     struct record_stream *r;
409
410     start = PA_IDXSET_INVALID;
411     for (;;) {
412         struct pa_memchunk chunk;
413         
414         if (!(r = pa_idxset_rrobin(c->record_streams, &c->rrobin_index)))
415             return;
416
417         if (start == PA_IDXSET_INVALID)
418             start = c->rrobin_index;
419         else if (start == c->rrobin_index)
420             return;
421
422         if (pa_memblockq_peek(r->memblockq,  &chunk) >= 0) {
423             struct pa_memchunk schunk = chunk;
424             
425             if (schunk.length > r->fragment_size)
426                 schunk.length = r->fragment_size;
427
428             pa_pstream_send_memblock(c->pstream, r->index, 0, &schunk);
429             pa_memblockq_drop(r->memblockq, &chunk, schunk.length);
430             pa_memblock_unref(schunk.memblock);
431             
432             return;
433         }
434     }
435 }
436
437 static void send_playback_stream_killed(struct playback_stream *p) {
438     struct pa_tagstruct *t;
439     assert(p);
440
441     t = pa_tagstruct_new(NULL, 0);
442     assert(t);
443     pa_tagstruct_putu32(t, PA_COMMAND_PLAYBACK_STREAM_KILLED);
444     pa_tagstruct_putu32(t, (uint32_t) -1); /* tag */
445     pa_tagstruct_putu32(t, p->index);
446     pa_pstream_send_tagstruct(p->connection->pstream, t);
447 }
448
449 static void send_record_stream_killed(struct record_stream *r) {
450     struct pa_tagstruct *t;
451     assert(r);
452
453     t = pa_tagstruct_new(NULL, 0);
454     assert(t);
455     pa_tagstruct_putu32(t, PA_COMMAND_RECORD_STREAM_KILLED);
456     pa_tagstruct_putu32(t, (uint32_t) -1); /* tag */
457     pa_tagstruct_putu32(t, r->index);
458     pa_pstream_send_tagstruct(r->connection->pstream, t);
459 }
460
461 /*** sinkinput callbacks ***/
462
463 static int sink_input_peek_cb(struct pa_sink_input *i, struct pa_memchunk *chunk) {
464     struct playback_stream *s;
465     assert(i && i->userdata && chunk);
466     s = i->userdata;
467
468     if (pa_memblockq_peek(s->memblockq, chunk) < 0)
469         return -1;
470
471     return 0;
472 }
473
474 static void sink_input_drop_cb(struct pa_sink_input *i, const struct pa_memchunk *chunk, size_t length) {
475     struct playback_stream *s;
476     assert(i && i->userdata && length);
477     s = i->userdata;
478
479     pa_memblockq_drop(s->memblockq, chunk, length);
480     request_bytes(s);
481
482     if (s->drain_request && !pa_memblockq_is_readable(s->memblockq)) {
483         pa_pstream_send_simple_ack(s->connection->pstream, s->drain_tag);
484         s->drain_request = 0;
485     }
486
487 /*     pa_log(__FILE__": after_drop: %u\n", pa_memblockq_get_length(s->memblockq)); */
488 }
489
490 static void sink_input_kill_cb(struct pa_sink_input *i) {
491     assert(i && i->userdata);
492     send_playback_stream_killed((struct playback_stream *) i->userdata);
493     playback_stream_free((struct playback_stream *) i->userdata);
494 }
495
496 static pa_usec_t sink_input_get_latency_cb(struct pa_sink_input *i) {
497     struct playback_stream *s;
498     assert(i && i->userdata);
499     s = i->userdata;
500
501     /*pa_log(__FILE__": get_latency: %u\n", pa_memblockq_get_length(s->memblockq));*/
502     
503     return pa_bytes_to_usec(pa_memblockq_get_length(s->memblockq), &s->sink_input->sample_spec);
504 }
505
506 /*** source_output callbacks ***/
507
508 static void source_output_push_cb(struct pa_source_output *o, const struct pa_memchunk *chunk) {
509     struct record_stream *s;
510     assert(o && o->userdata && chunk);
511     s = o->userdata;
512     
513     pa_memblockq_push_align(s->memblockq, chunk, 0);
514     if (!pa_pstream_is_pending(s->connection->pstream))
515         send_memblock(s->connection);
516 }
517
518 static void source_output_kill_cb(struct pa_source_output *o) {
519     assert(o && o->userdata);
520     send_record_stream_killed((struct record_stream *) o->userdata);
521     record_stream_free((struct record_stream *) o->userdata);
522 }
523
524 static pa_usec_t source_output_get_latency_cb(struct pa_source_output *o) {
525     struct record_stream *s;
526     assert(o && o->userdata);
527     s = o->userdata;
528
529     /*pa_log(__FILE__": get_latency: %u\n", pa_memblockq_get_length(s->memblockq));*/
530     
531     return pa_bytes_to_usec(pa_memblockq_get_length(s->memblockq), &o->sample_spec);
532 }
533
534 /*** pdispatch callbacks ***/
535
536 static void protocol_error(struct connection *c) {
537     pa_log(__FILE__": protocol error, kicking client\n");
538     connection_free(c);
539 }
540
541 static void command_create_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
542     struct connection *c = userdata;
543     struct playback_stream *s;
544     size_t maxlength, tlength, prebuf, minreq;
545     uint32_t sink_index;
546     const char *name, *sink_name;
547     struct pa_sample_spec ss;
548     struct pa_tagstruct *reply;
549     struct pa_sink *sink;
550     pa_volume_t volume;
551     int corked;
552     assert(c && t && c->protocol && c->protocol->core);
553     
554     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
555         pa_tagstruct_get_sample_spec(t, &ss) < 0 ||
556         pa_tagstruct_getu32(t, &sink_index) < 0 ||
557         pa_tagstruct_gets(t, &sink_name) < 0 ||
558         pa_tagstruct_getu32(t, &maxlength) < 0 ||
559         pa_tagstruct_get_boolean(t, &corked) < 0 ||
560         pa_tagstruct_getu32(t, &tlength) < 0 ||
561         pa_tagstruct_getu32(t, &prebuf) < 0 ||
562         pa_tagstruct_getu32(t, &minreq) < 0 ||
563         pa_tagstruct_getu32(t, &volume) < 0 ||
564         !pa_tagstruct_eof(t)) {
565         protocol_error(c);
566         return;
567     }
568
569
570     if (!c->authorized) {
571         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
572         return;
573     }
574
575     if (sink_index != (uint32_t) -1)
576         sink = pa_idxset_get_by_index(c->protocol->core->sinks, sink_index);
577     else
578         sink = pa_namereg_get(c->protocol->core, sink_name, PA_NAMEREG_SINK, 1);
579
580     if (!sink) {
581         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
582         return;
583     }
584     
585     if (!(s = playback_stream_new(c, sink, &ss, name, maxlength, tlength, prebuf, minreq, volume))) {
586         pa_pstream_send_error(c->pstream, tag, PA_ERROR_INVALID);
587         return;
588     }
589
590     pa_sink_input_cork(s->sink_input, corked);
591     
592     reply = pa_tagstruct_new(NULL, 0);
593     assert(reply);
594     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
595     pa_tagstruct_putu32(reply, tag);
596     pa_tagstruct_putu32(reply, s->index);
597     assert(s->sink_input);
598     pa_tagstruct_putu32(reply, s->sink_input->index);
599     pa_tagstruct_putu32(reply, s->requested_bytes = pa_memblockq_missing(s->memblockq));
600     pa_pstream_send_tagstruct(c->pstream, reply);
601     request_bytes(s);
602 }
603
604 static void command_delete_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
605     struct connection *c = userdata;
606     uint32_t channel;
607     assert(c && t);
608     
609     if (pa_tagstruct_getu32(t, &channel) < 0 ||
610         !pa_tagstruct_eof(t)) {
611         protocol_error(c);
612         return;
613     }
614
615     if (!c->authorized) {
616         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
617         return;
618     }
619
620     if (command == PA_COMMAND_DELETE_PLAYBACK_STREAM) {
621         struct playback_stream *s;
622         if (!(s = pa_idxset_get_by_index(c->output_streams, channel)) || (s->type != PLAYBACK_STREAM)) {
623             pa_pstream_send_error(c->pstream, tag, PA_ERROR_EXIST);
624             return;
625         }
626
627         playback_stream_free(s);
628     } else if (command == PA_COMMAND_DELETE_RECORD_STREAM) {
629         struct record_stream *s;
630         if (!(s = pa_idxset_get_by_index(c->record_streams, channel))) {
631             pa_pstream_send_error(c->pstream, tag, PA_ERROR_EXIST);
632             return;
633         }
634
635         record_stream_free(s);
636     } else {
637         struct upload_stream *s;
638         assert(command == PA_COMMAND_DELETE_UPLOAD_STREAM);
639         if (!(s = pa_idxset_get_by_index(c->output_streams, channel)) || (s->type != UPLOAD_STREAM)) {
640             pa_pstream_send_error(c->pstream, tag, PA_ERROR_EXIST);
641             return;
642         }
643
644         upload_stream_free(s);
645     }
646             
647     pa_pstream_send_simple_ack(c->pstream, tag);
648 }
649
650 static void command_create_record_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
651     struct connection *c = userdata;
652     struct record_stream *s;
653     size_t maxlength, fragment_size;
654     uint32_t source_index;
655     const char *name, *source_name;
656     struct pa_sample_spec ss;
657     struct pa_tagstruct *reply;
658     struct pa_source *source;
659     assert(c && t && c->protocol && c->protocol->core);
660     
661     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
662         pa_tagstruct_get_sample_spec(t, &ss) < 0 ||
663         pa_tagstruct_getu32(t, &source_index) < 0 ||
664         pa_tagstruct_gets(t, &source_name) < 0 ||
665         pa_tagstruct_getu32(t, &maxlength) < 0 ||
666         pa_tagstruct_getu32(t, &fragment_size) < 0 ||
667         !pa_tagstruct_eof(t)) {
668         protocol_error(c);
669         return;
670     }
671
672     if (!c->authorized) {
673         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
674         return;
675     }
676
677     if (source_index != (uint32_t) -1)
678         source = pa_idxset_get_by_index(c->protocol->core->sources, source_index);
679     else
680         source = pa_namereg_get(c->protocol->core, source_name, PA_NAMEREG_SOURCE, 1);
681
682     if (!source) {
683         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
684         return;
685     }
686     
687     if (!(s = record_stream_new(c, source, &ss, name, maxlength, fragment_size))) {
688         pa_pstream_send_error(c->pstream, tag, PA_ERROR_INVALID);
689         return;
690     }
691     
692     reply = pa_tagstruct_new(NULL, 0);
693     assert(reply);
694     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
695     pa_tagstruct_putu32(reply, tag);
696     pa_tagstruct_putu32(reply, s->index);
697     assert(s->source_output);
698     pa_tagstruct_putu32(reply, s->source_output->index);
699     pa_pstream_send_tagstruct(c->pstream, reply);
700 }
701
702 static void command_exit(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
703     struct connection *c = userdata;
704     assert(c && t);
705     
706     if (!pa_tagstruct_eof(t)) {
707         protocol_error(c);
708         return;
709     }
710
711     if (!c->authorized) {
712         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
713         return;
714     }
715     
716     assert(c->protocol && c->protocol->core && c->protocol->core->mainloop);
717     c->protocol->core->mainloop->quit(c->protocol->core->mainloop, 0);
718     pa_pstream_send_simple_ack(c->pstream, tag); /* nonsense */
719     return;
720 }
721
722 static void command_auth(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
723     struct connection *c = userdata;
724     const void*cookie;
725     assert(c && t);
726
727     if (pa_tagstruct_get_arbitrary(t, &cookie, PA_NATIVE_COOKIE_LENGTH) < 0 ||
728         !pa_tagstruct_eof(t)) {
729         protocol_error(c);
730         return;
731     }
732
733     if (!c->authorized) {
734         if (memcmp(c->protocol->auth_cookie, cookie, PA_NATIVE_COOKIE_LENGTH) != 0) {
735             pa_log(__FILE__": Denied access to client with invalid authorization key.\n");
736             pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
737             return;
738         }
739         
740         c->authorized = 1;
741     }
742     
743     pa_pstream_send_simple_ack(c->pstream, tag);
744     return;
745 }
746
747 static void command_set_client_name(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
748     struct connection *c = userdata;
749     const char *name;
750     assert(c && t);
751
752     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
753         !pa_tagstruct_eof(t)) {
754         protocol_error(c);
755         return;
756     }
757
758     pa_client_set_name(c->client, name);
759     pa_pstream_send_simple_ack(c->pstream, tag);
760     return;
761 }
762
763 static void command_lookup(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
764     struct connection *c = userdata;
765     const char *name;
766     uint32_t index = PA_IDXSET_INVALID;
767     assert(c && t);
768
769     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
770         !pa_tagstruct_eof(t)) {
771         protocol_error(c);
772         return;
773     }
774
775     if (!c->authorized) {
776         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
777         return;
778     }
779
780     if (command == PA_COMMAND_LOOKUP_SINK) {
781         struct pa_sink *sink;
782         if ((sink = pa_namereg_get(c->protocol->core, name, PA_NAMEREG_SINK, 1)))
783             index = sink->index;
784     } else {
785         struct pa_source *source;
786         assert(command == PA_COMMAND_LOOKUP_SOURCE);
787         if ((source = pa_namereg_get(c->protocol->core, name, PA_NAMEREG_SOURCE, 1)))
788             index = source->index;
789     }
790
791     if (index == PA_IDXSET_INVALID)
792         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
793     else {
794         struct pa_tagstruct *reply;
795         reply = pa_tagstruct_new(NULL, 0);
796         assert(reply);
797         pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
798         pa_tagstruct_putu32(reply, tag);
799         pa_tagstruct_putu32(reply, index);
800         pa_pstream_send_tagstruct(c->pstream, reply);
801     }
802 }
803
804 static void command_drain_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
805     struct connection *c = userdata;
806     uint32_t index;
807     struct playback_stream *s;
808     assert(c && t);
809
810     if (pa_tagstruct_getu32(t, &index) < 0 ||
811         !pa_tagstruct_eof(t)) {
812         protocol_error(c);
813         return;
814     }
815
816     if (!c->authorized) {
817         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
818         return;
819     }
820
821     if (!(s = pa_idxset_get_by_index(c->output_streams, index)) || s->type != PLAYBACK_STREAM) {
822         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
823         return;
824     }
825
826     s->drain_request = 0;
827
828     pa_memblockq_prebuf_disable(s->memblockq);
829     
830     if (!pa_memblockq_is_readable(s->memblockq)) {
831 /*         pa_log("immediate drain: %u\n", pa_memblockq_get_length(s->memblockq)); */
832         pa_pstream_send_simple_ack(c->pstream, tag);
833     } else {
834 /*         pa_log("slow drain triggered\n"); */
835         s->drain_request = 1;
836         s->drain_tag = tag;
837
838         pa_sink_notify(s->sink_input->sink);
839     }
840
841
842 static void command_stat(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
843     struct connection *c = userdata;
844     struct pa_tagstruct *reply;
845     assert(c && t);
846
847     if (!pa_tagstruct_eof(t)) {
848         protocol_error(c);
849         return;
850     }
851
852     if (!c->authorized) {
853         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
854         return;
855     }
856
857     reply = pa_tagstruct_new(NULL, 0);
858     assert(reply);
859     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
860     pa_tagstruct_putu32(reply, tag);
861     pa_tagstruct_putu32(reply, c->protocol->core->memblock_stat->total);
862     pa_tagstruct_putu32(reply, c->protocol->core->memblock_stat->total_size);
863     pa_tagstruct_putu32(reply, c->protocol->core->memblock_stat->allocated);
864     pa_tagstruct_putu32(reply, c->protocol->core->memblock_stat->allocated_size);
865     pa_tagstruct_putu32(reply, pa_scache_total_size(c->protocol->core));
866     pa_pstream_send_tagstruct(c->pstream, reply);
867 }
868
869 static void command_get_playback_latency(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
870     struct connection *c = userdata;
871     struct pa_tagstruct *reply;
872     struct playback_stream *s;
873     struct timeval tv, now;
874     uint32_t index;
875     assert(c && t);
876     
877     if (pa_tagstruct_getu32(t, &index) < 0 ||
878         pa_tagstruct_get_timeval(t, &tv) < 0 ||
879         !pa_tagstruct_eof(t)) {
880         protocol_error(c);
881         return;
882     }
883
884     if (!c->authorized) {
885         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
886         return;
887     }
888
889     if (!(s = pa_idxset_get_by_index(c->output_streams, index)) || s->type != PLAYBACK_STREAM) {
890         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
891         return;
892     }
893
894     reply = pa_tagstruct_new(NULL, 0);
895     assert(reply);
896     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
897     pa_tagstruct_putu32(reply, tag);
898     pa_tagstruct_put_usec(reply, pa_sink_input_get_latency(s->sink_input));
899     pa_tagstruct_put_usec(reply, pa_sink_get_latency(s->sink_input->sink));
900     pa_tagstruct_put_usec(reply, 0);
901     pa_tagstruct_put_boolean(reply, pa_memblockq_is_readable(s->memblockq));
902     pa_tagstruct_putu32(reply, pa_memblockq_get_length(s->memblockq));
903     pa_tagstruct_put_timeval(reply, &tv);
904     gettimeofday(&now, NULL);
905     pa_tagstruct_put_timeval(reply, &now);
906     pa_pstream_send_tagstruct(c->pstream, reply);
907 }
908
909 static void command_get_record_latency(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
910     struct connection *c = userdata;
911     struct pa_tagstruct *reply;
912     struct record_stream *s;
913     struct timeval tv, now;
914     uint32_t index;
915     assert(c && t);
916
917     if (pa_tagstruct_getu32(t, &index) < 0 ||
918         pa_tagstruct_get_timeval(t, &tv) < 0 ||
919         !pa_tagstruct_eof(t)) {
920         protocol_error(c);
921         return;
922     }
923
924     if (!c->authorized) {
925         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
926         return;
927     }
928
929     if (!(s = pa_idxset_get_by_index(c->record_streams, index))) {
930         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
931         return;
932     }
933
934     reply = pa_tagstruct_new(NULL, 0);
935     assert(reply);
936     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
937     pa_tagstruct_putu32(reply, tag);
938     pa_tagstruct_put_usec(reply, pa_source_output_get_latency(s->source_output));
939     pa_tagstruct_put_usec(reply, s->source_output->source->monitor_of ? pa_sink_get_latency(s->source_output->source->monitor_of) : 0);
940     pa_tagstruct_put_usec(reply, pa_source_get_latency(s->source_output->source));
941     pa_tagstruct_put_boolean(reply, 0);
942     pa_tagstruct_putu32(reply, pa_memblockq_get_length(s->memblockq));
943     pa_tagstruct_put_timeval(reply, &tv);
944     gettimeofday(&now, NULL);
945     pa_tagstruct_put_timeval(reply, &now);
946     pa_pstream_send_tagstruct(c->pstream, reply);
947 }
948
949
950 static void command_create_upload_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
951     struct connection *c = userdata;
952     struct upload_stream *s;
953     size_t length;
954     const char *name;
955     struct pa_sample_spec ss;
956     struct pa_tagstruct *reply;
957     assert(c && t && c->protocol && c->protocol->core);
958     
959     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
960         pa_tagstruct_get_sample_spec(t, &ss) < 0 ||
961         pa_tagstruct_getu32(t, &length) < 0 ||
962         !pa_tagstruct_eof(t)) {
963         protocol_error(c);
964         return;
965     }
966
967     if (!c->authorized) {
968         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
969         return;
970     }
971
972     if ((length % pa_frame_size(&ss)) != 0 || length <= 0 || !*name) {
973         pa_pstream_send_error(c->pstream, tag, PA_ERROR_INVALID);
974         return;
975     }
976     
977     if (!(s = upload_stream_new(c, &ss, name, length))) {
978         pa_pstream_send_error(c->pstream, tag, PA_ERROR_INVALID);
979         return;
980     }
981     
982     reply = pa_tagstruct_new(NULL, 0);
983     assert(reply);
984     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
985     pa_tagstruct_putu32(reply, tag);
986     pa_tagstruct_putu32(reply, s->index);
987     pa_tagstruct_putu32(reply, length);
988     pa_pstream_send_tagstruct(c->pstream, reply);
989 }
990
991 static void command_finish_upload_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
992     struct connection *c = userdata;
993     uint32_t channel;
994     struct upload_stream *s;
995     uint32_t index;
996     assert(c && t);
997     
998     if (pa_tagstruct_getu32(t, &channel) < 0 ||
999         !pa_tagstruct_eof(t)) {
1000         protocol_error(c);
1001         return;
1002     }
1003
1004     if (!c->authorized) {
1005         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1006         return;
1007     }
1008
1009     if (!(s = pa_idxset_get_by_index(c->output_streams, channel)) || (s->type != UPLOAD_STREAM)) {
1010         pa_pstream_send_error(c->pstream, tag, PA_ERROR_EXIST);
1011         return;
1012     }
1013
1014     pa_scache_add_item(c->protocol->core, s->name, &s->sample_spec, &s->memchunk, &index);
1015     pa_pstream_send_simple_ack(c->pstream, tag);
1016     upload_stream_free(s);
1017 }
1018
1019 static void command_play_sample(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1020     struct connection *c = userdata;
1021     uint32_t sink_index, volume;
1022     struct pa_sink *sink;
1023     const char *name, *sink_name;
1024     assert(c && t);
1025
1026     if (pa_tagstruct_getu32(t, &sink_index) < 0 ||
1027         pa_tagstruct_gets(t, &sink_name) < 0 ||
1028         pa_tagstruct_getu32(t, &volume) < 0 ||
1029         pa_tagstruct_gets(t, &name) < 0 || !name || 
1030         !pa_tagstruct_eof(t)) {
1031         protocol_error(c);
1032         return;
1033     }
1034     
1035     if (!c->authorized) {
1036         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1037         return;
1038     }
1039
1040     if (sink_index != (uint32_t) -1)
1041         sink = pa_idxset_get_by_index(c->protocol->core->sinks, sink_index);
1042     else
1043         sink = pa_namereg_get(c->protocol->core, sink_name, PA_NAMEREG_SINK, 1);
1044
1045     if (!sink) {
1046         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1047         return;
1048     }
1049
1050     if (pa_scache_play_item(c->protocol->core, name, sink, volume) < 0) {
1051         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1052         return;
1053     }
1054
1055     pa_pstream_send_simple_ack(c->pstream, tag);
1056 }
1057
1058 static void command_remove_sample(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1059     struct connection *c = userdata;
1060     const char *name;
1061     assert(c && t);
1062
1063     if (pa_tagstruct_gets(t, &name) < 0 || !name || 
1064         !pa_tagstruct_eof(t)) {
1065         protocol_error(c);
1066         return;
1067     }
1068
1069     if (!c->authorized) {
1070         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1071         return;
1072     }
1073
1074     if (pa_scache_remove_item(c->protocol->core, name) < 0) {
1075         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1076         return;
1077     }
1078
1079     pa_pstream_send_simple_ack(c->pstream, tag);
1080 }
1081
1082 static void sink_fill_tagstruct(struct pa_tagstruct *t, struct pa_sink *sink) {
1083     assert(t && sink);
1084     pa_tagstruct_putu32(t, sink->index);
1085     pa_tagstruct_puts(t, sink->name);
1086     pa_tagstruct_puts(t, sink->description);
1087     pa_tagstruct_put_sample_spec(t, &sink->sample_spec);
1088     pa_tagstruct_putu32(t, sink->owner ? sink->owner->index : (uint32_t) -1);
1089     pa_tagstruct_putu32(t, sink->volume);
1090     pa_tagstruct_putu32(t, sink->monitor_source->index);
1091     pa_tagstruct_puts(t, sink->monitor_source->name);
1092     pa_tagstruct_put_usec(t, pa_sink_get_latency(sink));
1093 }
1094
1095 static void source_fill_tagstruct(struct pa_tagstruct *t, struct pa_source *source) {
1096     assert(t && source);
1097     pa_tagstruct_putu32(t, source->index);
1098     pa_tagstruct_puts(t, source->name);
1099     pa_tagstruct_puts(t, source->description);
1100     pa_tagstruct_put_sample_spec(t, &source->sample_spec);
1101     pa_tagstruct_putu32(t, source->owner ? source->owner->index : (uint32_t) -1);
1102     pa_tagstruct_putu32(t, source->monitor_of ? source->monitor_of->index : (uint32_t) -1);
1103     pa_tagstruct_puts(t, source->monitor_of ? source->monitor_of->name : NULL);
1104     pa_tagstruct_put_usec(t, pa_source_get_latency(source));
1105 }
1106
1107 static void client_fill_tagstruct(struct pa_tagstruct *t, struct pa_client *client) {
1108     assert(t && client);
1109     pa_tagstruct_putu32(t, client->index);
1110     pa_tagstruct_puts(t, client->name);
1111     pa_tagstruct_puts(t, client->protocol_name);
1112     pa_tagstruct_putu32(t, client->owner ? client->owner->index : (uint32_t) -1);
1113 }
1114
1115 static void module_fill_tagstruct(struct pa_tagstruct *t, struct pa_module *module) {
1116     assert(t && module);
1117     pa_tagstruct_putu32(t, module->index);
1118     pa_tagstruct_puts(t, module->name);
1119     pa_tagstruct_puts(t, module->argument);
1120     pa_tagstruct_putu32(t, module->n_used);
1121     pa_tagstruct_put_boolean(t, module->auto_unload);
1122 }
1123
1124 static void sink_input_fill_tagstruct(struct pa_tagstruct *t, struct pa_sink_input *s) {
1125     assert(t && s);
1126     pa_tagstruct_putu32(t, s->index);
1127     pa_tagstruct_puts(t, s->name);
1128     pa_tagstruct_putu32(t, s->owner ? s->owner->index : (uint32_t) -1);
1129     pa_tagstruct_putu32(t, s->client ? s->client->index : (uint32_t) -1);
1130     pa_tagstruct_putu32(t, s->sink->index);
1131     pa_tagstruct_put_sample_spec(t, &s->sample_spec);
1132     pa_tagstruct_putu32(t, s->volume);
1133     pa_tagstruct_put_usec(t, pa_sink_input_get_latency(s));
1134     pa_tagstruct_put_usec(t, pa_sink_get_latency(s->sink));
1135 }
1136
1137 static void source_output_fill_tagstruct(struct pa_tagstruct *t, struct pa_source_output *s) {
1138     assert(t && s);
1139     pa_tagstruct_putu32(t, s->index);
1140     pa_tagstruct_puts(t, s->name);
1141     pa_tagstruct_putu32(t, s->owner ? s->owner->index : (uint32_t) -1);
1142     pa_tagstruct_putu32(t, s->client ? s->client->index : (uint32_t) -1);
1143     pa_tagstruct_putu32(t, s->source->index);
1144     pa_tagstruct_put_sample_spec(t, &s->sample_spec);
1145     pa_tagstruct_put_usec(t, pa_source_output_get_latency(s));
1146     pa_tagstruct_put_usec(t, pa_source_get_latency(s->source));
1147 }
1148
1149 static void scache_fill_tagstruct(struct pa_tagstruct *t, struct pa_scache_entry *e) {
1150     assert(t && e);
1151     pa_tagstruct_putu32(t, e->index);
1152     pa_tagstruct_puts(t, e->name);
1153     pa_tagstruct_putu32(t, e->volume);
1154     pa_tagstruct_put_usec(t, pa_bytes_to_usec(e->memchunk.length, &e->sample_spec));
1155     pa_tagstruct_put_sample_spec(t, &e->sample_spec);
1156     pa_tagstruct_putu32(t, e->memchunk.length);
1157     pa_tagstruct_put_boolean(t, e->lazy);
1158     pa_tagstruct_puts(t, e->filename);
1159 }
1160
1161 static void command_get_info(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1162     struct connection *c = userdata;
1163     uint32_t index;
1164     struct pa_sink *sink = NULL;
1165     struct pa_source *source = NULL;
1166     struct pa_client *client = NULL;
1167     struct pa_module *module = NULL;
1168     struct pa_sink_input *si = NULL;
1169     struct pa_source_output *so = NULL;
1170     struct pa_scache_entry *sce = NULL;
1171     const char *name;
1172     struct pa_tagstruct *reply;
1173     assert(c && t);
1174
1175     
1176     if (pa_tagstruct_getu32(t, &index) < 0 ||
1177         (command != PA_COMMAND_GET_CLIENT_INFO &&
1178          command != PA_COMMAND_GET_MODULE_INFO &&
1179          command != PA_COMMAND_GET_SINK_INPUT_INFO &&
1180          command != PA_COMMAND_GET_SOURCE_OUTPUT_INFO &&
1181          pa_tagstruct_gets(t, &name) < 0) ||
1182         !pa_tagstruct_eof(t)) {
1183         protocol_error(c);
1184         return;
1185     }
1186     
1187     if (!c->authorized) {
1188         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1189         return;
1190     }
1191
1192     if (command == PA_COMMAND_GET_SINK_INFO) {
1193         if (index != (uint32_t) -1)
1194             sink = pa_idxset_get_by_index(c->protocol->core->sinks, index);
1195         else
1196             sink = pa_namereg_get(c->protocol->core, name, PA_NAMEREG_SINK, 1);
1197     } else if (command == PA_COMMAND_GET_SOURCE_INFO) {
1198         if (index != (uint32_t) -1)
1199             source = pa_idxset_get_by_index(c->protocol->core->sources, index);
1200         else
1201             source = pa_namereg_get(c->protocol->core, name, PA_NAMEREG_SOURCE, 1);
1202     } else if (command == PA_COMMAND_GET_CLIENT_INFO)
1203         client = pa_idxset_get_by_index(c->protocol->core->clients, index);
1204     else if (command == PA_COMMAND_GET_MODULE_INFO) 
1205         module = pa_idxset_get_by_index(c->protocol->core->modules, index);
1206     else if (command == PA_COMMAND_GET_SINK_INPUT_INFO)
1207         si = pa_idxset_get_by_index(c->protocol->core->sink_inputs, index);
1208     else if (command == PA_COMMAND_GET_SOURCE_OUTPUT_INFO)
1209         so = pa_idxset_get_by_index(c->protocol->core->source_outputs, index);
1210     else {
1211         assert(command == PA_COMMAND_GET_SAMPLE_INFO);
1212         if (index != (uint32_t) -1)
1213             sce = pa_idxset_get_by_index(c->protocol->core->scache, index);
1214         else
1215             sce = pa_namereg_get(c->protocol->core, name, PA_NAMEREG_SAMPLE, 0);
1216     }
1217             
1218     if (!sink && !source && !client && !module && !si && !so && !sce) {
1219         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1220         return;
1221     }
1222
1223     reply = pa_tagstruct_new(NULL, 0);
1224     assert(reply);
1225     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
1226     pa_tagstruct_putu32(reply, tag); 
1227     if (sink)
1228         sink_fill_tagstruct(reply, sink);
1229     else if (source)
1230         source_fill_tagstruct(reply, source);
1231     else if (client)
1232         client_fill_tagstruct(reply, client);
1233     else if (module)
1234         module_fill_tagstruct(reply, module);
1235     else if (si)
1236         sink_input_fill_tagstruct(reply, si);
1237     else if (so)
1238         source_output_fill_tagstruct(reply, so);
1239     else
1240         scache_fill_tagstruct(reply, sce);
1241     pa_pstream_send_tagstruct(c->pstream, reply);
1242 }
1243
1244 static void command_get_info_list(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1245     struct connection *c = userdata;
1246     struct pa_idxset *i;
1247     uint32_t index;
1248     void *p;
1249     struct pa_tagstruct *reply;
1250     assert(c && t);
1251
1252     if (!pa_tagstruct_eof(t)) {
1253         protocol_error(c);
1254         return;
1255     }
1256     
1257     if (!c->authorized) {
1258         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1259         return;
1260     }
1261
1262     reply = pa_tagstruct_new(NULL, 0);
1263     assert(reply);
1264     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
1265     pa_tagstruct_putu32(reply, tag);
1266
1267     if (command == PA_COMMAND_GET_SINK_INFO_LIST)
1268         i = c->protocol->core->sinks;
1269     else if (command == PA_COMMAND_GET_SOURCE_INFO_LIST)
1270         i = c->protocol->core->sources;
1271     else if (command == PA_COMMAND_GET_CLIENT_INFO_LIST)
1272         i = c->protocol->core->clients;
1273     else if (command == PA_COMMAND_GET_MODULE_INFO_LIST)
1274         i = c->protocol->core->modules;
1275     else if (command == PA_COMMAND_GET_SINK_INPUT_INFO_LIST)
1276         i = c->protocol->core->sink_inputs;
1277     else if (command == PA_COMMAND_GET_SOURCE_OUTPUT_INFO_LIST)
1278         i = c->protocol->core->source_outputs;
1279     else {
1280         assert(command == PA_COMMAND_GET_SAMPLE_INFO_LIST);
1281         i = c->protocol->core->scache;
1282     }
1283
1284     if (i) {
1285         for (p = pa_idxset_first(i, &index); p; p = pa_idxset_next(i, &index)) {
1286             if (command == PA_COMMAND_GET_SINK_INFO_LIST)
1287                 sink_fill_tagstruct(reply, p);
1288             else if (command == PA_COMMAND_GET_SOURCE_INFO_LIST)
1289                 source_fill_tagstruct(reply, p);
1290             else if (command == PA_COMMAND_GET_CLIENT_INFO_LIST)
1291                 client_fill_tagstruct(reply, p);
1292             else if (command == PA_COMMAND_GET_MODULE_INFO_LIST)
1293                 module_fill_tagstruct(reply, p);
1294             else if (command == PA_COMMAND_GET_SINK_INPUT_INFO_LIST)
1295                 sink_input_fill_tagstruct(reply, p);
1296             else if (command == PA_COMMAND_GET_SOURCE_OUTPUT_INFO_LIST) 
1297                 source_output_fill_tagstruct(reply, p);
1298             else {
1299                 assert(command == PA_COMMAND_GET_SAMPLE_INFO_LIST);
1300                 scache_fill_tagstruct(reply, p);
1301             }
1302         }
1303     }
1304     
1305     pa_pstream_send_tagstruct(c->pstream, reply);
1306 }
1307
1308 static void command_get_server_info(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1309     struct connection *c = userdata;
1310     struct pa_tagstruct *reply;
1311     char txt[256];
1312     const char *n;
1313     assert(c && t);
1314
1315     if (!pa_tagstruct_eof(t)) {
1316         protocol_error(c);
1317         return;
1318     }
1319     
1320     if (!c->authorized) {
1321         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1322         return;
1323     }
1324
1325     reply = pa_tagstruct_new(NULL, 0);
1326     assert(reply);
1327     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
1328     pa_tagstruct_putu32(reply, tag);
1329     pa_tagstruct_puts(reply, PACKAGE_NAME);
1330     pa_tagstruct_puts(reply, PACKAGE_VERSION);
1331     pa_tagstruct_puts(reply, pa_get_user_name(txt, sizeof(txt)));
1332     pa_tagstruct_puts(reply, pa_get_host_name(txt, sizeof(txt)));
1333     pa_tagstruct_put_sample_spec(reply, &c->protocol->core->default_sample_spec);
1334
1335     n = pa_namereg_get_default_sink_name(c->protocol->core);
1336     pa_tagstruct_puts(reply, n);
1337     n = pa_namereg_get_default_source_name(c->protocol->core);
1338     pa_tagstruct_puts(reply, n);
1339     pa_pstream_send_tagstruct(c->pstream, reply);
1340 }
1341
1342 static void subscription_cb(struct pa_core *core, enum pa_subscription_event_type e, uint32_t index, void *userdata) {
1343     struct pa_tagstruct *t;
1344     struct connection *c = userdata;
1345     assert(c && core);
1346
1347     t = pa_tagstruct_new(NULL, 0);
1348     assert(t);
1349     pa_tagstruct_putu32(t, PA_COMMAND_SUBSCRIBE_EVENT);
1350     pa_tagstruct_putu32(t, (uint32_t) -1);
1351     pa_tagstruct_putu32(t, e);
1352     pa_tagstruct_putu32(t, index);
1353     pa_pstream_send_tagstruct(c->pstream, t);
1354 }
1355
1356 static void command_subscribe(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1357     struct connection *c = userdata;
1358     enum pa_subscription_mask m;
1359     assert(c && t);
1360
1361     if (pa_tagstruct_getu32(t, &m) < 0 ||
1362         !pa_tagstruct_eof(t)) {
1363         protocol_error(c);
1364         return;
1365     }
1366     
1367     if (!c->authorized) {
1368         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1369         return;
1370     }
1371
1372     if (c->subscription)
1373         pa_subscription_free(c->subscription);
1374
1375     if (m != 0) {
1376         c->subscription = pa_subscription_new(c->protocol->core, m, subscription_cb, c);
1377         assert(c->subscription);
1378     } else
1379         c->subscription = NULL;
1380
1381     pa_pstream_send_simple_ack(c->pstream, tag);
1382 }
1383
1384 static void command_set_volume(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1385     struct connection *c = userdata;
1386     uint32_t index, volume;
1387     struct pa_sink *sink = NULL;
1388     struct pa_sink_input *si = NULL;
1389     const char *name = NULL;
1390     assert(c && t);
1391
1392     if (pa_tagstruct_getu32(t, &index) < 0 ||
1393         (command == PA_COMMAND_SET_SINK_VOLUME && pa_tagstruct_gets(t, &name) < 0) ||
1394         pa_tagstruct_getu32(t, &volume) ||
1395         !pa_tagstruct_eof(t)) {
1396         protocol_error(c);
1397         return;
1398     }
1399     
1400     if (!c->authorized) {
1401         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1402         return;
1403     }
1404
1405     if (command == PA_COMMAND_SET_SINK_VOLUME) {
1406         if (index != (uint32_t) -1)
1407             sink = pa_idxset_get_by_index(c->protocol->core->sinks, index);
1408         else
1409             sink = pa_namereg_get(c->protocol->core, name, PA_NAMEREG_SINK, 1);
1410     }  else {
1411         assert(command == PA_COMMAND_SET_SINK_INPUT_VOLUME);
1412         si = pa_idxset_get_by_index(c->protocol->core->sink_inputs, index);
1413     }
1414
1415     if (!si && !sink) {
1416         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1417         return;
1418     }
1419
1420     if (sink)
1421         pa_sink_set_volume(sink, volume);
1422     else if (si)
1423         pa_sink_input_set_volume(si, volume);
1424
1425     pa_pstream_send_simple_ack(c->pstream, tag);
1426 }
1427
1428 static void command_cork_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1429     struct connection *c = userdata;
1430     uint32_t index;
1431     int b;
1432     struct playback_stream *s;
1433     assert(c && t);
1434
1435     if (pa_tagstruct_getu32(t, &index) < 0 ||
1436         pa_tagstruct_get_boolean(t, &b) < 0 ||
1437         !pa_tagstruct_eof(t)) {
1438         protocol_error(c);
1439         return;
1440     }
1441
1442     if (!c->authorized) {
1443         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1444         return;
1445     }
1446
1447     if (!(s = pa_idxset_get_by_index(c->output_streams, index)) || s->type != PLAYBACK_STREAM) {
1448         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1449         return;
1450     }
1451
1452     pa_sink_input_cork(s->sink_input, b);
1453     pa_pstream_send_simple_ack(c->pstream, tag);
1454 }
1455
1456 static void command_flush_or_trigger_playback_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1457     struct connection *c = userdata;
1458     uint32_t index;
1459     struct playback_stream *s;
1460     assert(c && t);
1461
1462     if (pa_tagstruct_getu32(t, &index) < 0 ||
1463         !pa_tagstruct_eof(t)) {
1464         protocol_error(c);
1465         return;
1466     }
1467
1468     if (!c->authorized) {
1469         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1470         return;
1471     }
1472
1473     if (!(s = pa_idxset_get_by_index(c->output_streams, index)) || s->type != PLAYBACK_STREAM) {
1474         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1475         return;
1476     }
1477
1478     if (command == PA_COMMAND_PREBUF_PLAYBACK_STREAM)
1479         pa_memblockq_prebuf_reenable(s->memblockq);
1480     else if (command == PA_COMMAND_TRIGGER_PLAYBACK_STREAM)
1481         pa_memblockq_prebuf_disable(s->memblockq);
1482     else {
1483         assert(command == PA_COMMAND_FLUSH_PLAYBACK_STREAM);
1484         pa_memblockq_flush(s->memblockq);
1485         /*pa_log(__FILE__": flush: %u\n", pa_memblockq_get_length(s->memblockq));*/
1486     }
1487
1488     pa_sink_notify(s->sink_input->sink);
1489     pa_pstream_send_simple_ack(c->pstream, tag);
1490     request_bytes(s);
1491 }
1492
1493 static void command_cork_record_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1494     struct connection *c = userdata;
1495     uint32_t index;
1496     struct record_stream *s;
1497     int b;
1498     assert(c && t);
1499
1500     if (pa_tagstruct_getu32(t, &index) < 0 ||
1501         pa_tagstruct_get_boolean(t, &b) < 0 ||
1502         !pa_tagstruct_eof(t)) {
1503         protocol_error(c);
1504         return;
1505     }
1506
1507     if (!c->authorized) {
1508         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1509         return;
1510     }
1511
1512     if (!(s = pa_idxset_get_by_index(c->record_streams, index))) {
1513         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1514         return;
1515     }
1516
1517     pa_source_output_cork(s->source_output, b);
1518     pa_pstream_send_simple_ack(c->pstream, tag);
1519 }
1520
1521 static void command_flush_record_stream(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1522     struct connection *c = userdata;
1523     uint32_t index;
1524     struct record_stream *s;
1525     assert(c && t);
1526
1527     if (pa_tagstruct_getu32(t, &index) < 0 ||
1528         !pa_tagstruct_eof(t)) {
1529         protocol_error(c);
1530         return;
1531     }
1532
1533     if (!c->authorized) {
1534         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1535         return;
1536     }
1537
1538     if (!(s = pa_idxset_get_by_index(c->record_streams, index))) {
1539         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1540         return;
1541     }
1542
1543     pa_memblockq_flush(s->memblockq);
1544     pa_pstream_send_simple_ack(c->pstream, tag);
1545 }
1546
1547 static void command_set_default_sink_or_source(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1548     struct connection *c = userdata;
1549     uint32_t index;
1550     const char *s;
1551     assert(c && t);
1552
1553     if (pa_tagstruct_getu32(t, &index) < 0 ||
1554         pa_tagstruct_gets(t, &s) < 0 || !s ||
1555         !pa_tagstruct_eof(t)) {
1556         protocol_error(c);
1557         return;
1558     }
1559
1560     if (!c->authorized) {
1561         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1562         return;
1563     }
1564
1565     pa_namereg_set_default(c->protocol->core, s, command == PA_COMMAND_SET_DEFAULT_SOURCE ? PA_NAMEREG_SOURCE : PA_NAMEREG_SINK);
1566     pa_pstream_send_simple_ack(c->pstream, tag);
1567 }
1568
1569 static void command_set_stream_name(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1570     struct connection *c = userdata;
1571     uint32_t index;
1572     const char *name;
1573     assert(c && t);
1574
1575     if (pa_tagstruct_getu32(t, &index) < 0 ||
1576         pa_tagstruct_gets(t, &name) < 0 || !name || 
1577         !pa_tagstruct_eof(t)) {
1578         protocol_error(c);
1579         return;
1580     }
1581     
1582     if (!c->authorized) {
1583         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1584         return;
1585     }
1586
1587     if (command == PA_COMMAND_SET_PLAYBACK_STREAM_NAME) {
1588         struct playback_stream *s;
1589         
1590         if (!(s = pa_idxset_get_by_index(c->output_streams, index)) || s->type != PLAYBACK_STREAM) {
1591             pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1592             return;
1593         }
1594
1595         pa_sink_input_set_name(s->sink_input, name);
1596         
1597     } else {
1598         struct record_stream *s;
1599         
1600         if (!(s = pa_idxset_get_by_index(c->record_streams, index))) {
1601             pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1602             return;
1603         }
1604
1605         pa_source_output_set_name(s->source_output, name);
1606     }
1607
1608     pa_pstream_send_simple_ack(c->pstream, tag);
1609 }
1610
1611 static void command_kill(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1612     struct connection *c = userdata;
1613     uint32_t index;
1614     assert(c && t);
1615
1616     if (pa_tagstruct_getu32(t, &index) < 0 ||
1617         !pa_tagstruct_eof(t)) {
1618         protocol_error(c);
1619         return;
1620     }
1621     
1622     if (!c->authorized) {
1623         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1624         return;
1625     }
1626
1627     if (command == PA_COMMAND_KILL_CLIENT) {
1628         struct pa_client *client;
1629         
1630         if (!(client = pa_idxset_get_by_index(c->protocol->core->clients, index))) {
1631             pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1632             return;
1633         }
1634
1635         pa_client_kill(client);
1636     } else if (command == PA_COMMAND_KILL_SINK_INPUT) {
1637         struct pa_sink_input *s;
1638         
1639         if (!(s = pa_idxset_get_by_index(c->protocol->core->sink_inputs, index))) {
1640             pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1641             return;
1642         }
1643
1644         pa_sink_input_kill(s);
1645     } else {
1646         struct pa_source_output *s;
1647
1648         assert(command == PA_COMMAND_KILL_SOURCE_OUTPUT);
1649         
1650         if (!(s = pa_idxset_get_by_index(c->protocol->core->source_outputs, index))) {
1651             pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1652             return;
1653         }
1654
1655         pa_source_output_kill(s);
1656     }
1657
1658     pa_pstream_send_simple_ack(c->pstream, tag);
1659 }
1660
1661 static void command_load_module(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1662     struct connection *c = userdata;
1663     struct pa_module *m;
1664     const char *name, *argument;
1665     struct pa_tagstruct *reply;
1666     assert(c && t);
1667
1668     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
1669         pa_tagstruct_gets(t, &argument) < 0 ||
1670         !pa_tagstruct_eof(t)) {
1671         protocol_error(c);
1672         return;
1673     }
1674     
1675     if (!c->authorized) {
1676         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1677         return;
1678     }
1679
1680     if (!(m = pa_module_load(c->protocol->core, name, argument))) {
1681         pa_pstream_send_error(c->pstream, tag, PA_ERROR_INITFAILED);
1682         return;
1683     }
1684
1685     reply = pa_tagstruct_new(NULL, 0);
1686     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
1687     pa_tagstruct_putu32(reply, tag);
1688     pa_tagstruct_putu32(reply, m->index);
1689     pa_pstream_send_tagstruct(c->pstream, reply);
1690 }
1691
1692 static void command_unload_module(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1693     struct connection *c = userdata;
1694     uint32_t index;
1695     struct pa_module *m;
1696     assert(c && t);
1697
1698     if (pa_tagstruct_getu32(t, &index) < 0 ||
1699         !pa_tagstruct_eof(t)) {
1700         protocol_error(c);
1701         return;
1702     }
1703     
1704     if (!c->authorized) {
1705         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1706         return;
1707     }
1708
1709     if (!(m = pa_idxset_get_by_index(c->protocol->core->modules, index))) {
1710         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1711         return;
1712     }
1713
1714     pa_module_unload_request(m);
1715     pa_pstream_send_simple_ack(c->pstream, tag);
1716 }
1717
1718 static void command_add_autoload(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1719     struct connection *c = userdata;
1720     const char *name, *module, *argument;
1721     uint32_t type;
1722     assert(c && t);
1723
1724     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
1725         pa_tagstruct_getu32(t, &type) < 0 || type > 1 ||
1726         pa_tagstruct_gets(t, &module) < 0 || !module ||
1727         pa_tagstruct_gets(t, &argument) < 0 ||
1728         !pa_tagstruct_eof(t)) {
1729         protocol_error(c);
1730         return;
1731     }
1732     
1733     if (!c->authorized) {
1734         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1735         return;
1736     }
1737
1738     if (pa_autoload_add(c->protocol->core, name, type == 0 ? PA_NAMEREG_SINK : PA_NAMEREG_SOURCE, module, argument) < 0) {
1739         pa_pstream_send_error(c->pstream, tag, PA_ERROR_EXIST);
1740         return;
1741     }
1742
1743     pa_pstream_send_simple_ack(c->pstream, tag);
1744 }
1745
1746 static void command_remove_autoload(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1747     struct connection *c = userdata;
1748     const char *name;
1749     uint32_t type;
1750     assert(c && t);
1751
1752     if (pa_tagstruct_gets(t, &name) < 0 || !name ||
1753         pa_tagstruct_getu32(t, &type) < 0 || type > 1 ||
1754         !pa_tagstruct_eof(t)) {
1755         protocol_error(c);
1756         return;
1757     }
1758     
1759     if (!c->authorized) {
1760         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1761         return;
1762     }
1763
1764     if (pa_autoload_remove(c->protocol->core, name, type == 0 ? PA_NAMEREG_SINK : PA_NAMEREG_SOURCE) < 0) {
1765         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1766         return;
1767     }
1768
1769     pa_pstream_send_simple_ack(c->pstream, tag);
1770 }
1771
1772 static void autoload_fill_tagstruct(struct pa_tagstruct *t, struct pa_autoload_entry *e) {
1773     assert(t && e);
1774     pa_tagstruct_puts(t, e->name);
1775     pa_tagstruct_putu32(t, e->type == PA_NAMEREG_SINK ? 0 : 1);
1776     pa_tagstruct_puts(t, e->module);
1777     pa_tagstruct_puts(t, e->argument);
1778 }
1779
1780 static void command_get_autoload_info(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1781     struct connection *c = userdata;
1782     struct pa_autoload_entry *a = NULL;
1783     uint32_t type;
1784     const char *name;
1785     struct pa_tagstruct *reply;
1786     assert(c && t);
1787     
1788     if (pa_tagstruct_gets(t, &name) < 0 || name ||
1789         pa_tagstruct_getu32(t, &type) < 0 || type > 1 ||
1790         !pa_tagstruct_eof(t)) {
1791         protocol_error(c);
1792         return;
1793     }
1794
1795     if (!c->authorized) {
1796         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1797         return;
1798     }
1799
1800     if (!c->protocol->core->autoload_hashmap || !(a = pa_hashmap_get(c->protocol->core->autoload_hashmap, name)) || (a->type == PA_NAMEREG_SINK && type != 0) || (a->type == PA_NAMEREG_SOURCE && type != 1)) {
1801         pa_pstream_send_error(c->pstream, tag, PA_ERROR_NOENTITY);
1802         return;
1803     }
1804
1805     reply = pa_tagstruct_new(NULL, 0);
1806     assert(reply);
1807     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
1808     pa_tagstruct_putu32(reply, tag);
1809     autoload_fill_tagstruct(reply, a);
1810     pa_pstream_send_tagstruct(c->pstream, reply);
1811 }
1812
1813 static void command_get_autoload_info_list(struct pa_pdispatch *pd, uint32_t command, uint32_t tag, struct pa_tagstruct *t, void *userdata) {
1814     struct connection *c = userdata;
1815     struct pa_tagstruct *reply;
1816     assert(c && t);
1817
1818     if (!pa_tagstruct_eof(t)) {
1819         protocol_error(c);
1820         return;
1821     }
1822     
1823     if (!c->authorized) {
1824         pa_pstream_send_error(c->pstream, tag, PA_ERROR_ACCESS);
1825         return;
1826     }
1827
1828     reply = pa_tagstruct_new(NULL, 0);
1829     pa_tagstruct_putu32(reply, PA_COMMAND_REPLY);
1830     pa_tagstruct_putu32(reply, tag);
1831
1832     if (c->protocol->core->autoload_hashmap) {
1833         struct pa_autoload_entry *a;
1834         void *state = NULL;
1835
1836         while ((a = pa_hashmap_iterate(c->protocol->core->autoload_hashmap, &state)))
1837             autoload_fill_tagstruct(reply, a);
1838     }
1839     
1840     pa_pstream_send_tagstruct(c->pstream, reply);
1841 }
1842
1843 /*** pstream callbacks ***/
1844
1845 static void pstream_packet_callback(struct pa_pstream *p, struct pa_packet *packet, void *userdata) {
1846     struct connection *c = userdata;
1847     assert(p && packet && packet->data && c);
1848
1849     if (pa_pdispatch_run(c->pdispatch, packet, c) < 0) {
1850         pa_log(__FILE__": invalid packet.\n");
1851         connection_free(c);
1852     }
1853 }
1854
1855 static void pstream_memblock_callback(struct pa_pstream *p, uint32_t channel, uint32_t delta, const struct pa_memchunk *chunk, void *userdata) {
1856     struct connection *c = userdata;
1857     struct output_stream *stream;
1858     assert(p && chunk && userdata);
1859     
1860     if (!(stream = pa_idxset_get_by_index(c->output_streams, channel))) {
1861         pa_log(__FILE__": client sent block for invalid stream.\n");
1862         connection_free(c);
1863         return;
1864     }
1865
1866     if (stream->type == PLAYBACK_STREAM) {
1867         struct playback_stream *p = (struct playback_stream*) stream;
1868         if (chunk->length >= p->requested_bytes)
1869             p->requested_bytes = 0;
1870         else
1871             p->requested_bytes -= chunk->length;
1872         
1873         pa_memblockq_push_align(p->memblockq, chunk, delta);
1874         assert(p->sink_input);
1875 /*         pa_log(__FILE__": after_recv: %u\n", pa_memblockq_get_length(p->memblockq)); */
1876
1877         pa_sink_notify(p->sink_input->sink);
1878 /*          pa_log(__FILE__": Recieved %u bytes.\n", chunk->length);  */
1879
1880     } else {
1881         struct upload_stream *u = (struct upload_stream*) stream;
1882         size_t l;
1883         assert(u->type == UPLOAD_STREAM);
1884
1885         if (!u->memchunk.memblock) {
1886             if (u->length == chunk->length) {
1887                 u->memchunk = *chunk;
1888                 pa_memblock_ref(u->memchunk.memblock);
1889                 u->length = 0;
1890             } else {
1891                 u->memchunk.memblock = pa_memblock_new(u->length, c->protocol->core->memblock_stat);
1892                 u->memchunk.index = u->memchunk.length = 0;
1893             }
1894         }
1895         
1896         assert(u->memchunk.memblock);
1897         
1898         l = u->length; 
1899         if (l > chunk->length)
1900             l = chunk->length;
1901
1902         if (l > 0) {
1903             memcpy((uint8_t*) u->memchunk.memblock->data + u->memchunk.index + u->memchunk.length,
1904                    (uint8_t*) chunk->memblock->data+chunk->index, l);
1905             u->memchunk.length += l;
1906             u->length -= l;
1907         }
1908     }
1909 }
1910
1911 static void pstream_die_callback(struct pa_pstream *p, void *userdata) {
1912     struct connection *c = userdata;
1913     assert(p && c);
1914     connection_free(c);
1915
1916 /*    pa_log(__FILE__": connection died.\n");*/
1917 }
1918
1919
1920 static void pstream_drain_callback(struct pa_pstream *p, void *userdata) {
1921     struct connection *c = userdata;
1922     assert(p && c);
1923
1924     send_memblock(c);
1925 }
1926
1927 /*** client callbacks ***/
1928
1929 static void client_kill_cb(struct pa_client *c) {
1930     assert(c && c->userdata);
1931     connection_free(c->userdata);
1932 }
1933
1934 /*** socket server callbacks ***/
1935
1936 static void on_connection(struct pa_socket_server*s, struct pa_iochannel *io, void *userdata) {
1937     struct pa_protocol_native *p = userdata;
1938     struct connection *c;
1939     assert(io && p);
1940
1941     c = pa_xmalloc(sizeof(struct connection));
1942
1943     c->authorized =!! p->public;
1944     c->protocol = p;
1945     assert(p->core);
1946     c->client = pa_client_new(p->core, "NATIVE", "Client");
1947     assert(c->client);
1948     c->client->kill = client_kill_cb;
1949     c->client->userdata = c;
1950     c->client->owner = p->module;
1951     
1952     c->pstream = pa_pstream_new(p->core->mainloop, io, p->core->memblock_stat);
1953     assert(c->pstream);
1954
1955     pa_pstream_set_recieve_packet_callback(c->pstream, pstream_packet_callback, c);
1956     pa_pstream_set_recieve_memblock_callback(c->pstream, pstream_memblock_callback, c);
1957     pa_pstream_set_die_callback(c->pstream, pstream_die_callback, c);
1958     pa_pstream_set_drain_callback(c->pstream, pstream_drain_callback, c);
1959
1960     c->pdispatch = pa_pdispatch_new(p->core->mainloop, command_table, PA_COMMAND_MAX);
1961     assert(c->pdispatch);
1962
1963     c->record_streams = pa_idxset_new(NULL, NULL);
1964     c->output_streams = pa_idxset_new(NULL, NULL);
1965     assert(c->record_streams && c->output_streams);
1966
1967     c->rrobin_index = PA_IDXSET_INVALID;
1968     c->subscription = NULL;
1969
1970     pa_idxset_put(p->connections, c, NULL);
1971 }
1972
1973 /*** module entry points ***/
1974
1975 static struct pa_protocol_native* protocol_new_internal(struct pa_core *c, struct pa_module *m, struct pa_modargs *ma) {
1976     struct pa_protocol_native *p;
1977     int public = 0;
1978     assert(c && ma);
1979
1980     if (pa_modargs_get_value_boolean(ma, "public", &public) < 0) {
1981         pa_log(__FILE__": public= expects a boolean argument.\n");
1982         return NULL;
1983     }
1984     
1985     p = pa_xmalloc(sizeof(struct pa_protocol_native));
1986
1987     if (pa_authkey_load_from_home(pa_modargs_get_value(ma, "cookie", PA_NATIVE_COOKIE_FILE), p->auth_cookie, sizeof(p->auth_cookie)) < 0) {
1988         pa_xfree(p);
1989         return NULL;
1990     }
1991
1992     p->module = m;
1993     p->public = public;
1994     p->server = NULL;
1995     p->core = c;
1996     p->connections = pa_idxset_new(NULL, NULL);
1997     assert(p->connections);
1998
1999     return p;
2000 }
2001
2002 struct pa_protocol_native* pa_protocol_native_new(struct pa_core *core, struct pa_socket_server *server, struct pa_module *m, struct pa_modargs *ma) {
2003     struct pa_protocol_native *p;
2004
2005     if (!(p = protocol_new_internal(core, m, ma)))
2006         return NULL;
2007     
2008     p->server = server;
2009     pa_socket_server_set_callback(p->server, on_connection, p);
2010     
2011     return p;
2012 }
2013
2014 void pa_protocol_native_free(struct pa_protocol_native *p) {
2015     struct connection *c;
2016     assert(p);
2017
2018     while ((c = pa_idxset_first(p->connections, NULL)))
2019         connection_free(c);
2020     pa_idxset_free(p->connections, NULL, NULL);
2021
2022     if (p->server)
2023         pa_socket_server_unref(p->server);
2024     
2025     pa_xfree(p);
2026 }
2027
2028 struct pa_protocol_native* pa_protocol_native_new_iochannel(struct pa_core*core, struct pa_iochannel *io, struct pa_module *m, struct pa_modargs *ma) {
2029     struct pa_protocol_native *p;
2030
2031     if (!(p = protocol_new_internal(core, m, ma)))
2032         return NULL;
2033
2034     on_connection(NULL, io, p);
2035     
2036     return p;
2037 }