Add a RTSP client impelmentation.

I still need to adapt the header reading to move the concatenation code to the headerlist lib

git-svn-id: file:///home/lennart/svn/public/pulseaudio/branches/coling@2333 fefdeb5f-60dc-0310-8127-8f9354f1896f

2 files changed, 538 insertions, 0 deletions

diff --git a/src/modules/rtp/rtsp.c b/src/modules/rtp/rtsp.c
new file mode 100644
index 00000000..3fd1ba0f
--- /dev/null
+++ b/src/modules/rtp/rtsp.c
@@ -0,0 +1,472 @@
+/* $Id$ */
+
+/***
+  This file is part of PulseAudio.
+
+  Copyright 2008 Colin Guthrie
+
+  PulseAudio 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 of the License,
+  or (at your option) any later version.
+
+  PulseAudio 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
+  General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with PulseAudio; 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 <fcntl.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include <arpa/inet.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+
+#ifdef HAVE_SYS_FILIO_H
+#include <sys/filio.h>
+#endif
+
+#include <pulse/xmalloc.h>
+
+#include <pulsecore/core-error.h>
+#include <pulsecore/core-util.h>
+#include <pulsecore/socket-util.h>
+#include <pulsecore/log.h>
+#include <pulsecore/macro.h>
+#include <pulsecore/strbuf.h>
+#include <pulsecore/poll.h>
+
+#include "rtsp.h"
+
+/*
+ * read one line from the file descriptor
+ * timeout: msec unit, -1 for infinite
+ * if CR comes then following LF is expected
+ * returned string in line is always null terminated, maxlen-1 is maximum string length
+ */
+static int pa_read_line(int fd, char *line, int maxlen, int timeout)
+{
+    int i, rval;
+    int count;
+    char ch;
+    struct pollfd pfds;
+    count = 0;
+    *line = 0;
+    pfds.events = POLLIN;
+    pfds.fd = fd;
+
+    for (i=0; i<maxlen; ++i) {
+        if (!poll(&pfds, 1, timeout))
+            return 0;
+
+        rval = read(fd, &ch, 1);
+
+        if (-1 == rval) {
+            if (EAGAIN == errno)
+                return 0;
+            //ERRMSG("%s:read error: %s\n", __func__, strerror(errno));
+            return -1;
+        }
+
+        if (0 == rval) {
+            //INFMSG("%s:disconnected on the other end\n", __func__);
+            return -1;
+        }
+
+        if ('\n' == ch) {
+            *line = 0;
+            return count;
+        }
+
+        if ('\r' == ch)
+            continue;
+
+        *line++ = ch;
+        count++;
+
+        if (count >= maxlen-1)
+            break;
+    }
+
+    *line = 0;
+    return count;
+}
+
+
+static int pa_rtsp_exec(pa_rtsp_context* c, const char* cmd,
+                        const char* content_type, const char* content,
+                        int expect_response,
+                        pa_headerlist* headers, pa_headerlist** response_headers) {
+    pa_strbuf* buf;
+    char* hdrs;
+    ssize_t l;
+    char response[1024];
+    int timeout;
+    char* token;
+    const char* token_state;
+    char delimiters[2];
+    char* header;
+    char* delimpos;
+
+
+    pa_assert(c);
+    pa_assert(c->url);
+
+    if (!cmd)
+        return 0;
+
+    buf = pa_strbuf_new();
+    pa_strbuf_printf(buf, "%s %s RTSP/1.0\r\nCSeq: %d\r\n", cmd, c->url, ++c->cseq);
+    if (c->session)
+        pa_strbuf_printf(buf, "Session: %s\r\n", c->session);
+
+    // Add the headers
+    if (headers) {
+        hdrs = pa_headerlist_to_string(headers);
+        pa_strbuf_puts(buf, hdrs);
+        pa_xfree(hdrs);
+    }
+
+    if (content_type && content) {
+        pa_strbuf_printf(buf, "Content-Type: %s\r\nContent-Length: %d\r\n",
+          content_type, (int)strlen(content));
+    }
+
+    pa_strbuf_printf(buf, "User-Agent: %s\r\n", c->useragent);
+
+    if (c->headers) {
+        hdrs = pa_headerlist_to_string(c->headers);
+        pa_strbuf_puts(buf, hdrs);
+        pa_xfree(hdrs);
+    }
+
+    pa_strbuf_puts(buf, "\r\n");
+
+    if (content_type && content) {
+        pa_strbuf_puts(buf, content);
+    }
+
+    // Our packet is created... now we can send it :)
+    hdrs = pa_strbuf_tostring_free(buf);
+    l = pa_write(c->fd, hdrs, strlen(hdrs), NULL);
+    pa_xfree(hdrs);
+
+    // Do we expect a response?
+    if (!expect_response)
+        return 1;
+
+    timeout = 5000;
+    if (pa_read_line(c->fd, response, sizeof(response), timeout) <= 0) {
+        //ERRMSG("%s: request failed\n",__func__);
+        return 0;
+    }
+
+    delimiters[0] = ' ';
+    delimiters[1] = '\0';
+    token_state = NULL;
+    pa_xfree(pa_split(response, delimiters, &token_state));
+    token = pa_split(response, delimiters, &token_state);
+    if (!token || strcmp(token, "200")) {
+        pa_xfree(token);
+        //ERRMSG("%s: request failed, error %s\n",__func__,token);
+        return 0;
+    }
+    pa_xfree(token);
+
+    // We want to return the headers?
+    if (!response_headers)
+    {
+        // We have no storage, so just clear out the response.
+        while (pa_read_line(c->fd, response, sizeof(response), timeout) > 0) {
+            // Reduce timeout for future requests
+            timeout = 1000;
+        }
+        return 1;
+    }
+
+    header = NULL;
+    buf = pa_strbuf_new();
+    while (pa_read_line(c->fd, response, sizeof(response), timeout) > 0) {
+        // Reduce timeout for future requests
+        timeout = 1000;
+
+        // If the first character is a space, it's a continuation header
+        if (header && ' ' == response[0]) {
+            // Add this line to the buffer (sans the space.
+            pa_strbuf_puts(buf, &(response[1]));
+            continue;
+        }
+
+        if (header) {
+            // This is not a continuation header so let's dump the full header/value into our proplist
+            pa_headerlist_puts(*response_headers, header, pa_strbuf_tostring_free(buf));
+            pa_xfree(header);
+            //header = NULL;
+            buf = pa_strbuf_new();
+        }
+
+        delimpos = strstr(response, ":");
+        if (!delimpos) {
+            //ERRMSG("%s: Request failed, bad header\n",__func__);
+            return 0;
+        }
+
+        if (strlen(delimpos) > 1) {
+            // Cut our line off so we can copy the header name out
+            *delimpos++ = '\0';
+
+            // Trim the front of any spaces
+            while (' ' == *delimpos)
+                ++delimpos;
+
+            pa_strbuf_puts(buf, delimpos);
+        } else {
+            // Cut our line off so we can copy the header name out
+            *delimpos = '\0';
+        }
+
+        // Save the header name
+        header = pa_xstrdup(response);
+    }
+    // We will have a header left from our looping itteration, so add it in :)
+    if (header) {
+        // This is not a continuation header so let's dump it into our proplist
+        pa_headerlist_puts(*response_headers, header, pa_strbuf_tostring(buf));
+    }
+    pa_strbuf_free(buf);
+
+    return 1;
+}
+
+
+pa_rtsp_context* pa_rtsp_context_new(const char* useragent) {
+    pa_rtsp_context *c;
+
+    c = pa_xnew0(pa_rtsp_context, 1);
+    c->fd = -1;
+    c->headers = pa_headerlist_new();
+
+    if (useragent)
+        c->useragent = useragent;
+    else
+        c->useragent = "PulseAudio RTSP Client";
+
+    return c;
+}
+
+
+void pa_rtsp_context_destroy(pa_rtsp_context* c) {
+    if (c) {
+        pa_xfree(c->url);
+        pa_xfree(c->session);
+        pa_xfree(c->transport);
+        pa_headerlist_free(c->headers);
+    }
+    pa_xfree(c);
+}
+
+
+int pa_rtsp_connect(pa_rtsp_context *c, const char* hostname, uint16_t port, const char* sid) {
+    struct sockaddr_in sa;
+    struct sockaddr_in name;
+    socklen_t namelen = sizeof(name);
+    struct hostent *host = NULL;
+    int r;
+
+    pa_assert(c);
+    pa_assert(hostname);
+    pa_assert(port > 0);
+    pa_assert(sid);
+
+    memset(&sa, 0, sizeof(sa));
+    sa.sin_family = AF_INET;
+    sa.sin_port = htons(port);
+
+    host = gethostbyname(hostname);
+    if (!host) {
+        unsigned int addr = inet_addr(hostname);
+        if (addr != INADDR_NONE)
+            host = gethostbyaddr((char*)&addr, 4, AF_INET);
+        if (!host)
+            return 0;
+    }
+    memcpy(&sa.sin_addr, host->h_addr, sizeof(struct in_addr));
+
+    if ((c->fd = socket(sa.sin_family, SOCK_STREAM, 0)) < 0) {
+        pa_log("socket(): %s", pa_cstrerror(errno));
+        return 0;
+    }
+
+    // Q: is FD_CLOEXEC reqd?
+    pa_make_fd_cloexec(c->fd);
+    pa_make_tcp_socket_low_delay(c->fd);
+
+    if ((r = connect(c->fd, &sa, sizeof(struct sockaddr_in))) < 0) {
+#ifdef OS_IS_WIN32
+        if (WSAGetLastError() != EWOULDBLOCK) {
+            pa_log_debug("connect(): %d", WSAGetLastError());
+#else
+        if (errno != EINPROGRESS) {
+            pa_log_debug("connect(): %s (%d)", pa_cstrerror(errno), errno);
+#endif
+            pa_close(c->fd);
+            c->fd = -1;
+            return 0;
+        }
+    }
+
+    if (0 != getsockname(c->fd, (struct sockaddr*)&name, &namelen)) {
+        pa_close(c->fd);
+        c->fd = -1;
+        return 0;
+    }
+    memcpy(&c->local_addr, &name.sin_addr, sizeof(struct in_addr));
+    c->url = pa_sprintf_malloc("rtsp://%s/%s", inet_ntoa(name.sin_addr), sid);
+
+    return 1;
+}
+
+
+void pa_rtsp_disconnect(pa_rtsp_context *c) {
+    pa_assert(c);
+
+    if (c->fd < 0)
+      return;
+    pa_close(c->fd);
+    c->fd = -1;
+}
+
+
+const char* pa_rtsp_localip(pa_rtsp_context* c) {
+    pa_assert(c);
+
+    if (c->fd < 0)
+        return NULL;
+    return inet_ntoa(c->local_addr);
+}
+
+
+int pa_rtsp_announce(pa_rtsp_context *c, const char* sdp) {
+    pa_assert(c);
+    if (!sdp)
+        return 0;
+
+    return pa_rtsp_exec(c, "ANNOUNCE", "application/sdp", sdp, 1, NULL, NULL);
+}
+
+
+int pa_rtsp_setup(pa_rtsp_context* c, pa_headerlist** response_headers) {
+    pa_headerlist* headers;
+    pa_headerlist* rheaders;
+    char delimiters[2];
+    char* token;
+    const char* token_state;
+    const char* pc;
+
+    pa_assert(c);
+
+    headers = pa_headerlist_new();
+    rheaders = pa_headerlist_new();
+    pa_headerlist_puts(headers, "Transport", "RTP/AVP/TCP;unicast;interleaved=0-1;mode=record");
+
+    if (!pa_rtsp_exec(c, "SETUP", NULL, NULL, 1, headers, &rheaders)) {
+        pa_headerlist_free(headers);
+        pa_headerlist_free(rheaders);
+        return 0;
+    }
+    pa_headerlist_free(headers);
+
+    c->session = pa_xstrdup(pa_headerlist_gets(rheaders, "Session"));
+    c->transport = pa_xstrdup(pa_headerlist_gets(rheaders, "Transport"));
+
+    if (!c->session || !c->transport) {
+        pa_headerlist_free(rheaders);
+        return 0;
+    }
+
+    // Now parse out the server port component of the response.
+    c->port = 0;
+    delimiters[0] = ';';
+    delimiters[1] = '\0';
+    token_state = NULL;
+    while ((token = pa_split(c->transport, delimiters, &token_state))) {
+        if ((pc = strstr(token, "="))) {
+            if (0 == strncmp(token, "server_port", 11)) {
+                pa_atou(pc+1, &c->port);
+                pa_xfree(token);
+                break;
+            }
+        }
+        pa_xfree(token);
+    }
+    if (0 == c->port) {
+        // Error no server_port in response
+        pa_headerlist_free(rheaders);
+        return 0;
+    }
+
+    *response_headers = rheaders;
+    return 1;
+}
+
+
+int pa_rtsp_record(pa_rtsp_context* c) {
+    pa_headerlist* headers;
+    int rv;
+
+    pa_assert(c);
+    if (!c->session) {
+        // No seesion in progres
+        return 0;
+    }
+
+    headers = pa_headerlist_new();
+    pa_headerlist_puts(headers, "Range", "npt=0-");
+    pa_headerlist_puts(headers, "RTP-Info", "seq=0;rtptime=0");
+
+    rv = pa_rtsp_exec(c, "RECORD", NULL, NULL, 1, headers, NULL);
+    pa_headerlist_free(headers);
+    return rv;
+}
+
+
+int pa_rtsp_teardown(pa_rtsp_context *c) {
+    pa_assert(c);
+
+    return pa_rtsp_exec(c, "TEARDOWN", NULL, NULL, 0, NULL, NULL);
+}
+
+
+int pa_rtsp_setparameter(pa_rtsp_context *c, const char* param) {
+    pa_assert(c);
+    if (!param)
+        return 0;
+
+    return pa_rtsp_exec(c, "SET_PARAMETER", "text/parameters", param, 1, NULL, NULL);
+}
+
+
+int pa_rtsp_flush(pa_rtsp_context *c) {
+    pa_headerlist* headers;
+    int rv;
+
+    pa_assert(c);
+
+    headers = pa_headerlist_new();
+    pa_headerlist_puts(headers, "RTP-Info", "seq=0;rtptime=0");
+
+    rv = pa_rtsp_exec(c, "FLUSH", NULL, NULL, 1, headers, NULL);
+    pa_headerlist_free(headers);
+    return rv;
+}
diff --git a/src/modules/rtp/rtsp.h b/src/modules/rtp/rtsp.h
new file mode 100644
index 00000000..504f1445
--- /dev/null
+++ b/src/modules/rtp/rtsp.h
@@ -0,0 +1,66 @@
+#ifndef foortsphfoo
+#define foortsphfoo
+
+/* $Id: rtp.h 1465 2007-05-29 17:24:48Z lennart $ */
+
+/***
+  This file is part of PulseAudio.
+
+  Copyright 2008 Colin Guthrie
+
+  PulseAudio 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 of the License,
+  or (at your option) any later version.
+
+  PulseAudio 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
+  General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public License
+  along with PulseAudio; if not, write to the Free Software
+  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+  USA.
+***/
+
+#include <inttypes.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+#include <netdb.h>
+
+#include <pulsecore/memblockq.h>
+#include <pulsecore/memchunk.h>
+#include <pulsecore/socket-client.h>
+
+#include "headerlist.h"
+
+typedef struct pa_rtsp_context {
+    int fd;
+    const char* useragent;
+    pa_headerlist* headers;
+    char* url;
+    uint32_t port;
+    uint32_t cseq;
+    char* session;
+    char* transport;
+    struct in_addr local_addr;
+} pa_rtsp_context;
+
+pa_rtsp_context* pa_rtsp_context_new(const char* useragent);
+void pa_rtsp_context_destroy(pa_rtsp_context* c);
+
+int pa_rtsp_connect(pa_rtsp_context* c, const char* hostname, uint16_t port, const char* sid);
+void pa_rtsp_disconnect(pa_rtsp_context* c);
+
+const char* pa_rtsp_localip(pa_rtsp_context* c);
+int pa_rtsp_announce(pa_rtsp_context* c, const char* sdp);
+
+int pa_rtsp_setup(pa_rtsp_context* c, pa_headerlist** response_headers);
+int pa_rtsp_record(pa_rtsp_context* c);
+int pa_rtsp_teardown(pa_rtsp_context* c);
+
+int pa_rtsp_setparameter(pa_rtsp_context* c, const char* param);
+int pa_rtsp_flush(pa_rtsp_context* c);
+
+#endif