1 files changed, 492 insertions, 0 deletions

diff --git a/src/polypcore/pstream.c b/src/polypcore/pstream.c
new file mode 100644
index 00000000..bd00ba4f
--- /dev/null
+++ b/src/polypcore/pstream.c
@@ -0,0 +1,492 @@
+/* $Id$ */
+
+/***
+  This file is part of polypaudio.
+ 
+  polypaudio is free software; you can redistribute it and/or modify
+  it under the terms of the GNU Lesser General Public License as
+  published by the Free Software Foundation; either version 2.1 of the
+  License, or (at your option) any later version.
+ 
+  polypaudio is distributed in the hope that it will be useful, but
+  WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+  Lesser General Public License for more details.
+ 
+  You should have received a copy of the GNU Lesser General Public
+  License along with polypaudio; if not, write to the Free Software
+  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+  USA.
+***/
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+#include <unistd.h>
+
+#ifdef HAVE_NETINET_IN_H
+#include <netinet/in.h>
+#endif
+
+#include "winsock.h"
+
+#include "pstream.h"
+#include "queue.h"
+#include "xmalloc.h"
+#include "log.h"
+
+typedef enum pa_pstream_descriptor_index {
+    PA_PSTREAM_DESCRIPTOR_LENGTH,
+    PA_PSTREAM_DESCRIPTOR_CHANNEL,
+    PA_PSTREAM_DESCRIPTOR_DELTA,
+    PA_PSTREAM_DESCRIPTOR_MAX
+} pa_pstream_descriptor_index;
+
+typedef uint32_t pa_pstream_descriptor[PA_PSTREAM_DESCRIPTOR_MAX];
+
+#define PA_PSTREAM_DESCRIPTOR_SIZE (PA_PSTREAM_DESCRIPTOR_MAX*sizeof(uint32_t))
+#define FRAME_SIZE_MAX (1024*500) /* half a megabyte */
+
+struct item_info {
+    enum { PA_PSTREAM_ITEM_PACKET, PA_PSTREAM_ITEM_MEMBLOCK } type;
+
+    /* memblock info */
+    pa_memchunk chunk;
+    uint32_t channel;
+    uint32_t delta;
+
+    /* packet info */
+    pa_packet *packet;
+};
+
+struct pa_pstream {
+    int ref;
+    
+    pa_mainloop_api *mainloop;
+    pa_defer_event *defer_event;
+    pa_iochannel *io;
+    pa_queue *send_queue;
+
+    int dead;
+    void (*die_callback) (pa_pstream *p, void *userdata);
+    void *die_callback_userdata;
+
+    struct {
+        struct item_info* current;
+        pa_pstream_descriptor descriptor;
+        void *data;
+        size_t index;
+    } write;
+
+    struct {
+        pa_memblock *memblock;
+        pa_packet *packet;
+        pa_pstream_descriptor descriptor;
+        void *data;
+        size_t index;
+    } read;
+
+    void (*recieve_packet_callback) (pa_pstream *p, pa_packet *packet, void *userdata);
+    void *recieve_packet_callback_userdata;
+
+    void (*recieve_memblock_callback) (pa_pstream *p, uint32_t channel, uint32_t delta, const pa_memchunk *chunk, void *userdata);
+    void *recieve_memblock_callback_userdata;
+
+    void (*drain_callback)(pa_pstream *p, void *userdata);
+    void *drain_userdata;
+
+    pa_memblock_stat *memblock_stat;
+};
+
+static void do_write(pa_pstream *p);
+static void do_read(pa_pstream *p);
+
+static void do_something(pa_pstream *p) {
+    assert(p);
+
+    p->mainloop->defer_enable(p->defer_event, 0);
+
+    pa_pstream_ref(p);
+    
+    if (!p->dead && pa_iochannel_is_readable(p->io))
+        do_read(p);
+
+    if (!p->dead && pa_iochannel_is_writable(p->io))
+        do_write(p);
+
+    /* In case the line was hungup, make sure to rerun this function
+       as soon as possible, until all data has been read. */
+    
+    if (!p->dead && pa_iochannel_is_hungup(p->io))
+        p->mainloop->defer_enable(p->defer_event, 1);
+    
+    pa_pstream_unref(p);
+}
+
+static void io_callback(pa_iochannel*io, void *userdata) {
+    pa_pstream *p = userdata;
+    assert(p && p->io == io);
+    do_something(p);
+}
+
+static void defer_callback(pa_mainloop_api *m, pa_defer_event *e, void*userdata) {
+    pa_pstream *p = userdata;
+    assert(p && p->defer_event == e && p->mainloop == m);
+    do_something(p);
+}
+
+pa_pstream *pa_pstream_new(pa_mainloop_api *m, pa_iochannel *io, pa_memblock_stat *s) {
+    pa_pstream *p;
+    assert(io);
+
+    p = pa_xmalloc(sizeof(pa_pstream));
+    p->ref = 1;
+    p->io = io;
+    pa_iochannel_set_callback(io, io_callback, p);
+
+    p->dead = 0;
+    p->die_callback = NULL;
+    p->die_callback_userdata = NULL;
+
+    p->mainloop = m;
+    p->defer_event = m->defer_new(m, defer_callback, p);
+    m->defer_enable(p->defer_event, 0);
+    
+    p->send_queue = pa_queue_new();
+    assert(p->send_queue);
+
+    p->write.current = NULL;
+    p->write.index = 0;
+
+    p->read.memblock = NULL;
+    p->read.packet = NULL;
+    p->read.index = 0;
+
+    p->recieve_packet_callback = NULL;
+    p->recieve_packet_callback_userdata = NULL;
+    
+    p->recieve_memblock_callback = NULL;
+    p->recieve_memblock_callback_userdata = NULL;
+
+    p->drain_callback = NULL;
+    p->drain_userdata = NULL;
+
+    p->memblock_stat = s;
+
+    pa_iochannel_socket_set_rcvbuf(io, 1024*8); 
+    pa_iochannel_socket_set_sndbuf(io, 1024*8); 
+
+    return p;
+}
+
+static void item_free(void *item, PA_GCC_UNUSED void *p) {
+    struct item_info *i = item;
+    assert(i);
+
+    if (i->type == PA_PSTREAM_ITEM_MEMBLOCK) {
+        assert(i->chunk.memblock);
+        pa_memblock_unref(i->chunk.memblock);
+    } else {
+        assert(i->type == PA_PSTREAM_ITEM_PACKET);
+        assert(i->packet);
+        pa_packet_unref(i->packet);
+    }
+
+    pa_xfree(i);
+}
+
+static void pstream_free(pa_pstream *p) {
+    assert(p);
+
+    pa_pstream_close(p);
+    
+    pa_queue_free(p->send_queue, item_free, NULL);
+
+    if (p->write.current)
+        item_free(p->write.current, NULL);
+
+    if (p->read.memblock)
+        pa_memblock_unref(p->read.memblock);
+    
+    if (p->read.packet)
+        pa_packet_unref(p->read.packet);
+
+    pa_xfree(p);
+}
+
+void pa_pstream_send_packet(pa_pstream*p, pa_packet *packet) {
+    struct item_info *i;
+    assert(p && packet && p->ref >= 1);
+
+    if (p->dead)
+        return;
+    
+/*     pa_log(__FILE__": push-packet %p\n", packet); */
+    
+    i = pa_xmalloc(sizeof(struct item_info));
+    i->type = PA_PSTREAM_ITEM_PACKET;
+    i->packet = pa_packet_ref(packet);
+
+    pa_queue_push(p->send_queue, i);
+    p->mainloop->defer_enable(p->defer_event, 1);
+}
+
+void pa_pstream_send_memblock(pa_pstream*p, uint32_t channel, uint32_t delta, const pa_memchunk *chunk) {
+    struct item_info *i;
+    assert(p && channel != (uint32_t) -1 && chunk && p->ref >= 1);
+
+    if (p->dead)
+        return;
+    
+/*     pa_log(__FILE__": push-memblock %p\n", chunk); */
+    
+    i = pa_xmalloc(sizeof(struct item_info));
+    i->type = PA_PSTREAM_ITEM_MEMBLOCK;
+    i->chunk = *chunk;
+    i->channel = channel;
+    i->delta = delta;
+
+    pa_memblock_ref(i->chunk.memblock);
+
+    pa_queue_push(p->send_queue, i);
+    p->mainloop->defer_enable(p->defer_event, 1);
+}
+
+void pa_pstream_set_recieve_packet_callback(pa_pstream *p, void (*callback) (pa_pstream *p, pa_packet *packet, void *userdata), void *userdata) {
+    assert(p && callback);
+
+    p->recieve_packet_callback = callback;
+    p->recieve_packet_callback_userdata = userdata;
+}
+
+void pa_pstream_set_recieve_memblock_callback(pa_pstream *p, void (*callback) (pa_pstream *p, uint32_t channel, uint32_t delta, const pa_memchunk *chunk, void *userdata), void *userdata) {
+    assert(p && callback);
+
+    p->recieve_memblock_callback = callback;
+    p->recieve_memblock_callback_userdata = userdata;
+}
+
+static void prepare_next_write_item(pa_pstream *p) {
+    assert(p);
+
+    if (!(p->write.current = pa_queue_pop(p->send_queue)))
+        return;
+    
+    p->write.index = 0;
+    
+    if (p->write.current->type == PA_PSTREAM_ITEM_PACKET) {
+        /*pa_log(__FILE__": pop-packet %p\n", p->write.current->packet);*/
+        
+        assert(p->write.current->packet);
+        p->write.data = p->write.current->packet->data;
+        p->write.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH] = htonl(p->write.current->packet->length);
+        p->write.descriptor[PA_PSTREAM_DESCRIPTOR_CHANNEL] = htonl((uint32_t) -1);
+        p->write.descriptor[PA_PSTREAM_DESCRIPTOR_DELTA] = 0;
+    } else {
+        assert(p->write.current->type == PA_PSTREAM_ITEM_MEMBLOCK && p->write.current->chunk.memblock);
+        p->write.data = (uint8_t*) p->write.current->chunk.memblock->data + p->write.current->chunk.index;
+        p->write.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH] = htonl(p->write.current->chunk.length);
+        p->write.descriptor[PA_PSTREAM_DESCRIPTOR_CHANNEL] = htonl(p->write.current->channel);
+        p->write.descriptor[PA_PSTREAM_DESCRIPTOR_DELTA] = htonl(p->write.current->delta);
+    }
+}
+
+static void do_write(pa_pstream *p) {
+    void *d;
+    size_t l;
+    ssize_t r;
+    assert(p);
+
+    if (!p->write.current)
+        prepare_next_write_item(p);
+
+    if (!p->write.current)
+        return;
+
+    assert(p->write.data);
+
+    if (p->write.index < PA_PSTREAM_DESCRIPTOR_SIZE) {
+        d = (uint8_t*) p->write.descriptor + p->write.index;
+        l = PA_PSTREAM_DESCRIPTOR_SIZE - p->write.index;
+    } else {
+        d = (uint8_t*) p->write.data + p->write.index - PA_PSTREAM_DESCRIPTOR_SIZE;
+        l = ntohl(p->write.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH]) - (p->write.index - PA_PSTREAM_DESCRIPTOR_SIZE);
+    }
+
+    if ((r = pa_iochannel_write(p->io, d, l)) < 0)
+        goto die;
+
+    p->write.index += r;
+
+    if (p->write.index >= PA_PSTREAM_DESCRIPTOR_SIZE+ntohl(p->write.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH])) {
+        assert(p->write.current);
+        item_free(p->write.current, (void *) 1);
+        p->write.current = NULL;
+
+        if (p->drain_callback && !pa_pstream_is_pending(p))
+            p->drain_callback(p, p->drain_userdata);
+    }
+
+    return;
+    
+die:
+    p->dead = 1;
+    if (p->die_callback)
+        p->die_callback(p, p->die_callback_userdata);
+}
+
+static void do_read(pa_pstream *p) {
+    void *d;
+    size_t l; 
+    ssize_t r;
+    assert(p);
+
+    if (p->read.index < PA_PSTREAM_DESCRIPTOR_SIZE) {
+        d = (uint8_t*) p->read.descriptor + p->read.index;
+        l = PA_PSTREAM_DESCRIPTOR_SIZE - p->read.index;
+    } else {
+        assert(p->read.data);
+        d = (uint8_t*) p->read.data + p->read.index - PA_PSTREAM_DESCRIPTOR_SIZE;
+        l = ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH]) - (p->read.index - PA_PSTREAM_DESCRIPTOR_SIZE);
+    }
+
+    if ((r = pa_iochannel_read(p->io, d, l)) <= 0)
+        goto die;
+    
+    p->read.index += r;
+
+    if (p->read.index == PA_PSTREAM_DESCRIPTOR_SIZE) {
+        /* Reading of frame descriptor complete */
+
+        /* Frame size too large */
+        if (ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH]) > FRAME_SIZE_MAX) {
+            pa_log(__FILE__": Frame size too large\n");
+            goto die;
+        }
+        
+        assert(!p->read.packet && !p->read.memblock);
+
+        if (ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_CHANNEL]) == (uint32_t) -1) {
+            /* Frame is a packet frame */
+            p->read.packet = pa_packet_new(ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH]));
+            assert(p->read.packet);
+            p->read.data = p->read.packet->data;
+        } else {
+            /* Frame is a memblock frame */
+            p->read.memblock = pa_memblock_new(ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH]), p->memblock_stat);
+            assert(p->read.memblock);
+            p->read.data = p->read.memblock->data;
+        }
+            
+    } else if (p->read.index > PA_PSTREAM_DESCRIPTOR_SIZE) {
+        /* Frame payload available */
+        
+        if (p->read.memblock && p->recieve_memblock_callback) { /* Is this memblock data? Than pass it to the user */
+            l = (p->read.index - r) < PA_PSTREAM_DESCRIPTOR_SIZE ? p->read.index - PA_PSTREAM_DESCRIPTOR_SIZE : (size_t) r;
+                
+            if (l > 0) {
+                pa_memchunk chunk;
+                
+                chunk.memblock = p->read.memblock;
+                chunk.index = p->read.index - PA_PSTREAM_DESCRIPTOR_SIZE - l;
+                chunk.length = l;
+
+                if (p->recieve_memblock_callback)
+                    p->recieve_memblock_callback(
+                        p,
+                        ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_CHANNEL]),
+                        ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_DELTA]),
+                        &chunk,
+                        p->recieve_memblock_callback_userdata);
+            }
+        }
+
+        /* Frame complete */
+        if (p->read.index >= ntohl(p->read.descriptor[PA_PSTREAM_DESCRIPTOR_LENGTH]) + PA_PSTREAM_DESCRIPTOR_SIZE) {
+            if (p->read.memblock) {
+                assert(!p->read.packet);
+                
+                pa_memblock_unref(p->read.memblock);
+                p->read.memblock = NULL;
+            } else {
+                assert(p->read.packet);
+                
+                if (p->recieve_packet_callback)
+                    p->recieve_packet_callback(p, p->read.packet, p->recieve_packet_callback_userdata);
+
+                pa_packet_unref(p->read.packet);
+                p->read.packet = NULL;
+            }
+
+            p->read.index = 0;
+        }
+    }
+
+    return;
+
+die:
+    p->dead = 1;
+    if (p->die_callback)
+        p->die_callback(p, p->die_callback_userdata);
+   
+}
+
+void pa_pstream_set_die_callback(pa_pstream *p, void (*callback)(pa_pstream *p, void *userdata), void *userdata) {
+    assert(p && callback);
+    p->die_callback = callback;
+    p->die_callback_userdata = userdata;
+}
+
+int pa_pstream_is_pending(pa_pstream *p) {
+    assert(p);
+
+    if (p->dead)
+        return 0;
+
+    return p->write.current || !pa_queue_is_empty(p->send_queue);
+}
+
+void pa_pstream_set_drain_callback(pa_pstream *p, void (*cb)(pa_pstream *p, void *userdata), void *userdata) {
+    assert(p);
+
+    p->drain_callback = cb;
+    p->drain_userdata = userdata;
+}
+
+void pa_pstream_unref(pa_pstream*p) {
+    assert(p && p->ref >= 1);
+
+    if (!(--(p->ref)))
+        pstream_free(p);
+}
+
+pa_pstream* pa_pstream_ref(pa_pstream*p) {
+    assert(p && p->ref >= 1);
+    p->ref++;
+    return p;
+}
+
+void pa_pstream_close(pa_pstream *p) {
+    assert(p);
+
+    p->dead = 1;
+
+    if (p->io) {
+        pa_iochannel_free(p->io);
+        p->io = NULL;
+    }
+
+    if (p->defer_event) {
+        p->mainloop->defer_free(p->defer_event);
+        p->defer_event = NULL;
+    }
+
+    p->die_callback = NULL;
+    p->drain_callback = NULL;
+    p->recieve_packet_callback = NULL;
+    p->recieve_memblock_callback = NULL;
+}