#include <errno.h>
#include <string.h>
-#include "inputstream.h"
-#include "outputstream.h"
+#include "sinkinput.h"
+#include "sourceoutput.h"
#include "protocol-simple.h"
#include "client.h"
+#include "sample-util.h"
struct connection {
struct protocol_simple *protocol;
struct iochannel *io;
- struct input_stream *istream;
- struct output_stream *ostream;
+ struct sink_input *sink_input;
+ struct source_output *source_output;
struct client *client;
+ struct memblockq *input_memblockq, *output_memblockq;
};
struct protocol_simple {
struct connection *c = data;
assert(data);
- if (c->istream)
- input_stream_free(c->istream);
- if (c->ostream)
- output_stream_free(c->ostream);
-
- client_free(c->client);
-
- iochannel_free(c->io);
+ if (c->sink_input)
+ sink_input_free(c->sink_input);
+ if (c->source_output)
+ source_output_free(c->source_output);
+ if (c->client)
+ client_free(c->client);
+ if (c->io)
+ iochannel_free(c->io);
+ if (c->input_memblockq)
+ memblockq_free(c->input_memblockq);
+ if (c->output_memblockq)
+ memblockq_free(c->output_memblockq);
free(c);
}
free_connection(c, NULL);
}
-static void istream_kill_cb(struct input_stream *i, void *userdata) {
- struct connection *c = userdata;
- assert(i && c);
- destroy_connection(c);
-}
+static int do_read(struct connection *c) {
+ struct memchunk chunk;
+ ssize_t r;
-static void ostream_kill_cb(struct output_stream *o, void *userdata) {
- struct connection *c = userdata;
- assert(o && c);
- destroy_connection(c);
-}
+ if (!iochannel_is_readable(c->io))
+ return 0;
+
+ if (!c->sink_input || !memblockq_is_writable(c->input_memblockq, BUFSIZE))
+ return 0;
+
+ chunk.memblock = memblock_new(BUFSIZE);
+ assert(chunk.memblock);
-static void client_kill_cb(struct client *client, void*userdata) {
- struct connection *c= userdata;
- assert(client && c);
- destroy_connection(c);
-}
+ if ((r = iochannel_read(c->io, chunk.memblock->data, BUFSIZE)) <= 0) {
+ fprintf(stderr, "read(): %s\n", r == 0 ? "EOF" : strerror(errno));
+ memblock_unref(chunk.memblock);
+ return -1;
+ }
-static void io_callback(struct iochannel*io, void *userdata) {
- struct connection *c = userdata;
- assert(io && c);
+ chunk.memblock->length = chunk.length = r;
+ chunk.index = 0;
- if (c->istream && iochannel_is_readable(io)) {
- struct memchunk chunk;
- ssize_t r;
+ assert(c->input_memblockq);
+ memblockq_push(c->input_memblockq, &chunk, 0);
+ memblock_unref(chunk.memblock);
+ assert(c->sink_input);
+ sink_notify(c->sink_input->sink);
+
+ return 0;
+}
- chunk.memblock = memblock_new(BUFSIZE);
- assert(chunk.memblock);
+static int do_write(struct connection *c) {
+ struct memchunk chunk;
+ ssize_t r;
- if ((r = iochannel_read(io, chunk.memblock->data, BUFSIZE)) <= 0) {
- fprintf(stderr, "read(): %s\n", r == 0 ? "EOF" : strerror(errno));
- memblock_unref(chunk.memblock);
- goto fail;
- }
-
- chunk.memblock->length = r;
- chunk.length = r;
- chunk.index = 0;
+ if (!iochannel_is_writable(c->io))
+ return 0;
+
+ if (!c->source_output)
+ return 0;
+
+ assert(c->output_memblockq);
+ if (memblockq_peek(c->output_memblockq, &chunk) < 0)
+ return 0;
- memblockq_push(c->istream->memblockq, &chunk, 0);
- input_stream_notify_sink(c->istream);
+ assert(chunk.memblock && chunk.length);
+
+ if ((r = iochannel_write(c->io, chunk.memblock->data+chunk.index, chunk.length)) < 0) {
+ fprintf(stderr, "write(): %s\n", strerror(errno));
memblock_unref(chunk.memblock);
+ return -1;
}
+
+ memblockq_drop(c->output_memblockq, r);
+ memblock_unref(chunk.memblock);
+ return 0;
+}
- if (c->ostream && iochannel_is_writable(io)) {
- struct memchunk chunk;
- ssize_t r;
+/*** sink_input callbacks ***/
- memblockq_peek(c->ostream->memblockq, &chunk);
- assert(chunk.memblock && chunk.length);
+static int sink_input_peek_cb(struct sink_input *i, struct memchunk *chunk) {
+ struct connection*c;
+ assert(i && i->userdata && chunk);
+ c = i->userdata;
+
+ if (memblockq_peek(c->input_memblockq, chunk) < 0)
+ return -1;
- if ((r = iochannel_write(io, chunk.memblock->data+chunk.index, chunk.length)) < 0) {
- fprintf(stderr, "write(): %s\n", strerror(errno));
- memblock_unref(chunk.memblock);
- goto fail;
- }
-
- memblockq_drop(c->ostream->memblockq, r);
- memblock_unref(chunk.memblock);
- }
+ return 0;
+}
- return;
+static void sink_input_drop_cb(struct sink_input *i, size_t length) {
+ struct connection*c = i->userdata;
+ assert(i && c && length);
+
+ memblockq_drop(c->input_memblockq, length);
-fail:
- destroy_connection(c);
+ if (do_read(c) < 0)
+ destroy_connection(c);
+}
+
+static void sink_input_kill_cb(struct sink_input *i) {
+ assert(i && i->userdata);
+ destroy_connection((struct connection *) i->userdata);
+}
+
+
+static uint32_t sink_input_get_latency_cb(struct sink_input *i) {
+ struct connection*c = i->userdata;
+ assert(i && c);
+ return pa_samples_usec(memblockq_get_length(c->input_memblockq), &c->sink_input->sample_spec);
+}
+
+/*** source_output callbacks ***/
+
+static void source_output_push_cb(struct source_output *o, struct memchunk *chunk) {
+ struct connection *c = o->userdata;
+ assert(o && c && chunk);
+
+ memblockq_push(c->output_memblockq, chunk, 0);
+
+ if (do_write(c) < 0)
+ destroy_connection(c);
}
+static void source_output_kill_cb(struct source_output *o) {
+ assert(o && o->userdata);
+ destroy_connection((struct connection *) o->userdata);
+}
+
+/*** client callbacks ***/
+
+static void client_kill_cb(struct client *c) {
+ assert(c && c->userdata);
+ destroy_connection((struct connection *) c->userdata);
+}
+
+/*** iochannel callbacks ***/
+
+static void io_callback(struct iochannel*io, void *userdata) {
+ struct connection *c = userdata;
+ assert(io && c && c->io == io);
+
+ if (do_read(c) < 0 || do_write(c) < 0)
+ destroy_connection(c);
+}
+
+/*** socket_server callbacks */
+
static void on_connection(struct socket_server*s, struct iochannel *io, void *userdata) {
struct protocol_simple *p = userdata;
struct connection *c = NULL;
+ char cname[256];
assert(s && io && p);
c = malloc(sizeof(struct connection));
assert(c);
c->io = io;
- c->istream = NULL;
- c->ostream = NULL;
+ c->sink_input = NULL;
+ c->source_output = NULL;
+ c->input_memblockq = c->output_memblockq = NULL;
c->protocol = p;
- c->client = client_new(p->core, "SIMPLE", "Client");
+ iochannel_peer_to_string(io, cname, sizeof(cname));
+ c->client = client_new(p->core, "SIMPLE", cname);
assert(c->client);
- client_set_kill_callback(c->client, client_kill_cb, c);
+ c->client->kill = client_kill_cb;
+ c->client->userdata = c;
if (p->mode & PROTOCOL_SIMPLE_RECORD) {
struct source *source;
+ size_t l;
- if (!(source = core_get_default_source(p->core))) {
+ if (!(source = source_get_default(p->core))) {
fprintf(stderr, "Failed to get default source.\n");
goto fail;
}
- c->ostream = output_stream_new(source, &DEFAULT_SAMPLE_SPEC, c->client->name);
- assert(c->ostream);
- output_stream_set_kill_callback(c->ostream, ostream_kill_cb, c);
+ c->source_output = source_output_new(source, &DEFAULT_SAMPLE_SPEC, c->client->name);
+ assert(c->source_output);
+ c->source_output->push = source_output_push_cb;
+ c->source_output->kill = source_output_kill_cb;
+ c->source_output->userdata = c;
+
+ l = 5*pa_bytes_per_second(&DEFAULT_SAMPLE_SPEC); /* 5s */
+ c->output_memblockq = memblockq_new(l, pa_sample_size(&DEFAULT_SAMPLE_SPEC), l/2);
}
if (p->mode & PROTOCOL_SIMPLE_PLAYBACK) {
struct sink *sink;
+ size_t l;
- if (!(sink = core_get_default_sink(p->core))) {
+ if (!(sink = sink_get_default(p->core))) {
fprintf(stderr, "Failed to get default sink.\n");
goto fail;
}
- c->istream = input_stream_new(sink, &DEFAULT_SAMPLE_SPEC, c->client->name);
- assert(c->istream);
- input_stream_set_kill_callback(c->istream, istream_kill_cb, c);
+ c->sink_input = sink_input_new(sink, &DEFAULT_SAMPLE_SPEC, c->client->name);
+ assert(c->sink_input);
+ c->sink_input->peek = sink_input_peek_cb;
+ c->sink_input->drop = sink_input_drop_cb;
+ c->sink_input->kill = sink_input_kill_cb;
+ c->sink_input->get_latency = sink_input_get_latency_cb;
+ c->sink_input->userdata = c;
+
+ l = pa_bytes_per_second(&DEFAULT_SAMPLE_SPEC)/2; /* half a second */
+ c->input_memblockq = memblockq_new(l, pa_sample_size(&DEFAULT_SAMPLE_SPEC), l/2);
}
fail:
if (c) {
- if (c->client)
- client_free(c->client);
- if (c->istream)
- input_stream_free(c->istream);
- if (c->ostream)
- output_stream_free(c->ostream);
-
+ free_connection(c, NULL);
iochannel_free(c->io);
free(c);
}