lib/configuration.c, lib/home.c: Introduce environment variables.

[disorder] / lib / uaudio-rtp.c

diff --git a/lib/uaudio-rtp.c b/lib/uaudio-rtp.c
index 3ce1cbe6b54cb3c25f6cb4c017d35c0a10c7e822..9e7abde229dbb22a411ea4c81830c2aa84720daf 100644 (file)
--- a/lib/uaudio-rtp.c
+++ b/lib/uaudio-rtp.c
@@ -1,6 +1,6 @@
 /*
  * This file is part of DisOrder.
 /*
  * This file is part of DisOrder.

- * Copyright (C) 2009 Richard Kettlewell
+ * Copyright (C) 2009, 2013 Richard Kettlewell

  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by


@@ -29,6 +29,7 @@
 #include <unistd.h>
 #include <time.h>
 #include <sys/uio.h>
 #include <unistd.h>
 #include <time.h>
 #include <sys/uio.h>

+#include <pthread.h>

 
 #include "uaudio.h"
 #include "mem.h"
 
 #include "uaudio.h"
 #include "mem.h"


@@ -40,21 +41,20 @@
 #include "timeval.h"
 #include "configuration.h"
 
 #include "timeval.h"
 #include "configuration.h"
 

-/** @brief Bytes to send per network packet
- *
- * This is the maximum number of bytes we pass to write(2); to determine actual
- * packet sizes, add a UDP header and an IP header (and a link layer header if
- * it's the link layer size you care about).
- *
- * Don't make this too big or arithmetic will start to overflow.
- */
-#define NETWORK_BYTES (1500-8/*UDP*/-40/*IP*/-8/*conservatism*/)
+/** @brief Bytes to send per network packet */
+static int rtp_max_payload;

 
 /** @brief RTP payload type */
 static int rtp_payload;
 
 
 /** @brief RTP payload type */
 static int rtp_payload;
 

-/** @brief RTP output socket */
-static int rtp_fd;
+/** @brief RTP broadcast/multicast output socket */
+static int rtp_fd = -1;
+
+/** @brief RTP unicast output socket (IPv4) */
+static int rtp_fd4 = -1;
+
+/** @brief RTP unicast output socket (IPv6) */
+static int rtp_fd6 = -1;

 
 /** @brief RTP SSRC */
 static uint32_t rtp_id;
 
 /** @brief RTP SSRC */
 static uint32_t rtp_id;


@@ -71,8 +71,26 @@ static uint16_t rtp_sequence;
  */
 static int rtp_errors;
 
  */
 static int rtp_errors;
 

-/** @brief Set while paused */
-static volatile int rtp_paused;
+/** @brief RTP mode */
+static int rtp_mode;
+
+#define RTP_BROADCAST 1
+#define RTP_MULTICAST 2
+#define RTP_UNICAST 3
+#define RTP_REQUEST 4
+#define RTP_AUTO 5
+
+/** @brief A unicast client */
+struct rtp_recipient {
+  struct rtp_recipient *next;
+  struct sockaddr_storage sa;
+};
+
+/** @brief List of unicast clients */
+static struct rtp_recipient *rtp_recipient_list;
+
+/** @brief Mutex protecting data structures */
+static pthread_mutex_t rtp_lock = PTHREAD_MUTEX_INITIALIZER;

 
 static const char *const rtp_options[] = {
   "rtp-destination",
 
 static const char *const rtp_options[] = {
   "rtp-destination",


@@ -81,6 +99,9 @@ static const char *const rtp_options[] = {
   "rtp-source-port",
   "multicast-ttl",
   "multicast-loop",
   "rtp-source-port",
   "multicast-ttl",
   "multicast-loop",

+  "rtp-mode",
+  "rtp-max-payload",
+  "rtp-mtu-discovery",

   NULL
 };
 
   NULL
 };
 


@@ -183,32 +204,80 @@ static size_t rtp_play(void *buffer, size_t nsamples, unsigned flags) {
     uaudio_schedule_sent(nsamples);
     return nsamples;
   }
     uaudio_schedule_sent(nsamples);
     return nsamples;
   }

-  int written_bytes;
-  do {
-    written_bytes = writev(rtp_fd, vec, 2);
-  } while(written_bytes < 0 && errno == EINTR);
-  if(written_bytes < 0) {
-    disorder_error(errno, "error transmitting audio data");
-    ++rtp_errors;
-    if(rtp_errors == 10)
-      disorder_fatal(0, "too many audio tranmission errors");
-    return 0;
-  } else
-    rtp_errors /= 2;                    /* gradual decay */
+  /* Send stuff to explicitly registerd unicast addresses unconditionally */
+  struct rtp_recipient *r;
+  struct msghdr m;
+  memset(&m, 0, sizeof m);
+  m.msg_iov = vec;
+  m.msg_iovlen = 2;
+  pthread_mutex_lock(&rtp_lock);
+  for(r = rtp_recipient_list; r; r = r->next) {
+    m.msg_name = &r->sa;
+    m.msg_namelen = r->sa.ss_family == AF_INET ? 
+      sizeof(struct sockaddr_in) : sizeof (struct sockaddr_in6);
+    sendmsg(r->sa.ss_family == AF_INET ? rtp_fd4 : rtp_fd6,
+            &m, MSG_DONTWAIT|MSG_NOSIGNAL);
+    // TODO similar error handling to other case?
+  }
+  pthread_mutex_unlock(&rtp_lock);
+  if(rtp_mode != RTP_REQUEST) {
+    int written_bytes;
+    do {
+      written_bytes = writev(rtp_fd, vec, 2);
+    } while(written_bytes < 0 && errno == EINTR);
+    if(written_bytes < 0) {
+      disorder_error(errno, "error transmitting audio data");
+      ++rtp_errors;
+      if(rtp_errors == 10)
+        disorder_fatal(0, "too many audio transmission errors");
+      return 0;
+    } else
+      rtp_errors /= 2;                    /* gradual decay */
+  }

   /* TODO what can we sensibly do about short writes here?  Really that's just
    * an error and we ought to be using smaller packets. */
   uaudio_schedule_sent(nsamples);
   return nsamples;
 }
 
   /* TODO what can we sensibly do about short writes here?  Really that's just
    * an error and we ought to be using smaller packets. */
   uaudio_schedule_sent(nsamples);
   return nsamples;
 }
 

+static void hack_send_buffer_size(int fd, const char *what) {
+  int sndbuf, target_sndbuf = 131072;
+  socklen_t len = sizeof sndbuf;
+
+  if(getsockopt(fd, SOL_SOCKET, SO_SNDBUF,
+                &sndbuf, &len) < 0)
+    disorder_fatal(errno, "error getting SO_SNDBUF on %s socket", what);
+  if(target_sndbuf > sndbuf) {
+    if(setsockopt(fd, SOL_SOCKET, SO_SNDBUF,
+                  &target_sndbuf, sizeof target_sndbuf) < 0)
+      disorder_error(errno, "error setting SO_SNDBUF on %s socket to %d",
+                     what, target_sndbuf);
+    else
+      disorder_info("changed socket send buffer size on %socket from %d to %d",
+                    what, sndbuf, target_sndbuf);
+  } else
+    disorder_info("default socket send buffer on %s socket is %d",
+                  what, sndbuf);
+}
+

 static void rtp_open(void) {
 static void rtp_open(void) {

-  struct addrinfo *res, *sres;
+  struct addrinfo *dres, *sres;

   static const int one = 1;
   static const int one = 1;

-  int sndbuf, target_sndbuf = 131072;
-  socklen_t len;

   struct netaddress dst[1], src[1];
   struct netaddress dst[1], src[1];

+  const char *mode;
+#ifdef IP_MTU_DISCOVER
+  const char *mtu_disc;
+  int opt;
+#endif

   
   

-  /* Get configuration */
+  /* Get the mode */
+  mode = uaudio_get("rtp-mode", "auto");
+  if(!strcmp(mode, "broadcast")) rtp_mode = RTP_BROADCAST;
+  else if(!strcmp(mode, "multicast")) rtp_mode = RTP_MULTICAST;
+  else if(!strcmp(mode, "unicast")) rtp_mode = RTP_UNICAST;
+  else if(!strcmp(mode, "request")) rtp_mode = RTP_REQUEST;
+  else rtp_mode = RTP_AUTO;
+  /* Get the source and destination addresses (which might be missing) */

   rtp_get_netconfig("rtp-destination-af",
                     "rtp-destination",
                     "rtp-destination-port",
   rtp_get_netconfig("rtp-destination-af",
                     "rtp-destination",
                     "rtp-destination-port",


@@ -217,28 +286,66 @@ static void rtp_open(void) {
                     "rtp-source",
                     "rtp-source-port",
                     src);
                     "rtp-source",
                     "rtp-source-port",
                     src);

-  /* ...microseconds */
-
-  /* Resolve addresses */
-  res = netaddress_resolve(dst, 0, IPPROTO_UDP);
-  if(!res)
-    exit(-1);
+  if(dst->af != -1) {
+    dres = netaddress_resolve(dst, 0, IPPROTO_UDP);
+    if(!dres)
+      exit(-1);
+  } else
+    dres = 0;

   if(src->af != -1) {
     sres = netaddress_resolve(src, 1, IPPROTO_UDP);
     if(!sres)
       exit(-1);
   } else
     sres = 0;
   if(src->af != -1) {
     sres = netaddress_resolve(src, 1, IPPROTO_UDP);
     if(!sres)
       exit(-1);
   } else
     sres = 0;

-  /* Create the socket */
-  if((rtp_fd = socket(res->ai_family,
-                      res->ai_socktype,
-                      res->ai_protocol)) < 0)
-    disorder_fatal(errno, "error creating broadcast socket");
-  if(multicast(res->ai_addr)) {
+  /* _AUTO inspects the destination address and acts accordingly */
+  if(rtp_mode == RTP_AUTO) {
+    if(!dres)
+      rtp_mode = RTP_REQUEST;
+    else if(multicast(dres->ai_addr))
+      rtp_mode = RTP_MULTICAST;
+    else {
+      struct ifaddrs *ifs;
+
+      if(getifaddrs(&ifs) < 0)
+        disorder_fatal(errno, "error calling getifaddrs");
+      while(ifs) {
+        /* (At least on Darwin) IFF_BROADCAST might be set but ifa_broadaddr
+         * still a null pointer.  It turns out that there's a subsequent entry
+         * for he same interface which _does_ have ifa_broadaddr though... */
+        if((ifs->ifa_flags & IFF_BROADCAST)
+           && ifs->ifa_broadaddr
+           && sockaddr_equal(ifs->ifa_broadaddr, dres->ai_addr))
+          break;
+        ifs = ifs->ifa_next;
+      }
+      if(ifs) 
+        rtp_mode = RTP_BROADCAST;
+      else
+        rtp_mode = RTP_UNICAST;
+    }
+  }
+  rtp_max_payload = atoi(uaudio_get("rtp-max-payload", "-1"));
+  if(rtp_max_payload < 0)
+    rtp_max_payload = 1500 - 8/*UDP*/ - 40/*IP*/ - 8/*conservatism*/;
+  /* Create the sockets */
+  if(rtp_mode != RTP_REQUEST) {
+    if((rtp_fd = socket(dres->ai_family,
+                        dres->ai_socktype,
+                        dres->ai_protocol)) < 0)
+      disorder_fatal(errno, "error creating RTP transmission socket");
+  }
+  if((rtp_fd4 = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
+    disorder_fatal(errno, "error creating v4 RTP transmission socket");
+  if((rtp_fd6 = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP)) < 0)
+    disorder_fatal(errno, "error creating v6 RTP transmission socket");
+  /* Configure the socket according to the desired mode */
+  switch(rtp_mode) {
+  case RTP_MULTICAST: {

     /* Enable multicast options */
     const int ttl = atoi(uaudio_get("multicast-ttl", "1"));
     const int loop = !strcmp(uaudio_get("multicast-loop", "yes"), "yes");
     /* Enable multicast options */
     const int ttl = atoi(uaudio_get("multicast-ttl", "1"));
     const int loop = !strcmp(uaudio_get("multicast-loop", "yes"), "yes");

-    switch(res->ai_family) {
+    switch(dres->ai_family) {

     case PF_INET: {
       if(setsockopt(rtp_fd, IPPROTO_IP, IP_MULTICAST_TTL,
                     &ttl, sizeof ttl) < 0)
     case PF_INET: {
       if(setsockopt(rtp_fd, IPPROTO_IP, IP_MULTICAST_TTL,
                     &ttl, sizeof ttl) < 0)


@@ -258,55 +365,55 @@ static void rtp_open(void) {
       break;
     }
     default:
       break;
     }
     default:

-      disorder_fatal(0, "unsupported address family %d", res->ai_family);
+      disorder_fatal(0, "unsupported address family %d", dres->ai_family);

     }
     disorder_info("multicasting on %s TTL=%d loop=%s", 
     }
     disorder_info("multicasting on %s TTL=%d loop=%s", 

-                  format_sockaddr(res->ai_addr), ttl, loop ? "yes" : "no");
-  } else {
-    struct ifaddrs *ifs;
-
-    if(getifaddrs(&ifs) < 0)
-      disorder_fatal(errno, "error calling getifaddrs");
-    while(ifs) {
-      /* (At least on Darwin) IFF_BROADCAST might be set but ifa_broadaddr
-       * still a null pointer.  It turns out that there's a subsequent entry
-       * for he same interface which _does_ have ifa_broadaddr though... */
-      if((ifs->ifa_flags & IFF_BROADCAST)
-         && ifs->ifa_broadaddr
-         && sockaddr_equal(ifs->ifa_broadaddr, res->ai_addr))
-        break;
-      ifs = ifs->ifa_next;
-    }
-    if(ifs) {
-      if(setsockopt(rtp_fd, SOL_SOCKET, SO_BROADCAST, &one, sizeof one) < 0)
-        disorder_fatal(errno, "error setting SO_BROADCAST on broadcast socket");
-      disorder_info("broadcasting on %s (%s)", 
-           format_sockaddr(res->ai_addr), ifs->ifa_name);
-    } else
-      disorder_info("unicasting on %s", format_sockaddr(res->ai_addr));
+                  format_sockaddr(dres->ai_addr), ttl, loop ? "yes" : "no");
+    break;

   }
   }

-  /* Enlarge the socket buffer */
-  len = sizeof sndbuf;
-  if(getsockopt(rtp_fd, SOL_SOCKET, SO_SNDBUF,
-                &sndbuf, &len) < 0)
-    disorder_fatal(errno, "error getting SO_SNDBUF");
-  if(target_sndbuf > sndbuf) {
-    if(setsockopt(rtp_fd, SOL_SOCKET, SO_SNDBUF,
-                  &target_sndbuf, sizeof target_sndbuf) < 0)
-      disorder_error(errno, "error setting SO_SNDBUF to %d", target_sndbuf);
-    else
-      disorder_info("changed socket send buffer size from %d to %d",
-           sndbuf, target_sndbuf);
-  } else
-    disorder_info("default socket send buffer is %d", sndbuf);
+  case RTP_UNICAST: {
+    disorder_info("unicasting on %s", format_sockaddr(dres->ai_addr));
+    break;
+  }
+  case RTP_BROADCAST: {
+    if(setsockopt(rtp_fd, SOL_SOCKET, SO_BROADCAST, &one, sizeof one) < 0)
+      disorder_fatal(errno, "error setting SO_BROADCAST on broadcast socket");
+    disorder_info("broadcasting on %s", 
+                  format_sockaddr(dres->ai_addr));
+    break;
+  }
+  case RTP_REQUEST: {
+    disorder_info("will transmit on request");
+    break;
+  }
+  }
+  /* Enlarge the socket buffers */
+  if (rtp_fd != -1) hack_send_buffer_size(rtp_fd, "master socket");
+  hack_send_buffer_size(rtp_fd4, "IPv4 on-demand socket");
+  hack_send_buffer_size(rtp_fd6, "IPv6 on-demand socket");

   /* We might well want to set additional broadcast- or multicast-related
    * options here */
   /* We might well want to set additional broadcast- or multicast-related
    * options here */

-  if(sres && bind(rtp_fd, sres->ai_addr, sres->ai_addrlen) < 0)
-    disorder_fatal(errno, "error binding broadcast socket to %s", 
-                   format_sockaddr(sres->ai_addr));
-  if(connect(rtp_fd, res->ai_addr, res->ai_addrlen) < 0)
-    disorder_fatal(errno, "error connecting broadcast socket to %s", 
-                   format_sockaddr(res->ai_addr));
+  if(rtp_mode != RTP_REQUEST) {
+    if(sres && bind(rtp_fd, sres->ai_addr, sres->ai_addrlen) < 0)
+      disorder_fatal(errno, "error binding broadcast socket to %s", 
+                     format_sockaddr(sres->ai_addr));
+    if(connect(rtp_fd, dres->ai_addr, dres->ai_addrlen) < 0)
+      disorder_fatal(errno, "error connecting broadcast socket to %s", 
+                     format_sockaddr(dres->ai_addr));
+  }
+#ifdef IP_MTU_DISCOVER
+  mtu_disc = uaudio_get("rtp-mtu-discovery", "default");
+  do {
+    if(!strcmp(mtu_disc, "yes")) opt = IP_PMTUDISC_DO;
+    else if(!strcmp(mtu_disc, "no")) opt = IP_PMTUDISC_DONT;
+    else break;
+    if(setsockopt(rtp_fd4, IPPROTO_IP, IP_MTU_DISCOVER, &opt, sizeof opt))
+      disorder_fatal(errno, "error setting MTU discovery");
+    if(sres->ai_family == AF_INET &&
+        setsockopt(rtp_fd, IPPROTO_IP, IP_MTU_DISCOVER, &opt, sizeof opt))
+      disorder_fatal(errno, "error setting MTU discovery");
+  } while (0);
+#endif

   if(config->rtp_verbose)
     disorder_info("RTP: prepared socket");
 }
   if(config->rtp_verbose)
     disorder_info("RTP: prepared socket");
 }


@@ -345,7 +452,7 @@ static void rtp_start(uaudio_callback *callback,
                       userdata,
                       rtp_play,
                       256 / uaudio_sample_size,
                       userdata,
                       rtp_play,
                       256 / uaudio_sample_size,

-                      (NETWORK_BYTES - sizeof(struct rtp_header))
+                      (rtp_max_payload - sizeof(struct rtp_header))

                       / uaudio_sample_size,
                       0);
   if(config->rtp_verbose)
                       / uaudio_sample_size,
                       0);
   if(config->rtp_verbose)


@@ -354,13 +461,15 @@ static void rtp_start(uaudio_callback *callback,
 
 static void rtp_stop(void) {
   uaudio_thread_stop();
 
 static void rtp_stop(void) {
   uaudio_thread_stop();

-  close(rtp_fd);
-  rtp_fd = -1;
+  if(rtp_fd >= 0) { close(rtp_fd); rtp_fd = -1; }
+  if(rtp_fd4 >= 0) { close(rtp_fd4); rtp_fd4 = -1; }
+  if(rtp_fd6 >= 0) { close(rtp_fd6); rtp_fd6 = -1; }

 }
 
 static void rtp_configure(void) {
   char buffer[64];
 
 }
 
 static void rtp_configure(void) {
   char buffer[64];
 

+  uaudio_set("rtp-mode", config->rtp_mode);

   rtp_set_netconfig("rtp-destination-af",
                     "rtp-destination",
                     "rtp-destination-port", &config->broadcast);
   rtp_set_netconfig("rtp-destination-af",
                     "rtp-destination",
                     "rtp-destination-port", &config->broadcast);


@@ -370,10 +479,64 @@ static void rtp_configure(void) {
   snprintf(buffer, sizeof buffer, "%ld", config->multicast_ttl);
   uaudio_set("multicast-ttl", buffer);
   uaudio_set("multicast-loop", config->multicast_loop ? "yes" : "no");
   snprintf(buffer, sizeof buffer, "%ld", config->multicast_ttl);
   uaudio_set("multicast-ttl", buffer);
   uaudio_set("multicast-loop", config->multicast_loop ? "yes" : "no");

+  snprintf(buffer, sizeof buffer, "%ld", config->rtp_max_payload);
+  uaudio_set("rtp-max-payload", buffer);
+  uaudio_set("rtp-mtu-discovery", config->rtp_mtu_discovery);

   if(config->rtp_verbose)
     disorder_info("RTP: configured");
 }
 
   if(config->rtp_verbose)
     disorder_info("RTP: configured");
 }
 

+/** @brief Add an RTP recipient address
+ * @param sa Pointer to recipient address
+ * @return 0 on success, -1 on error
+ */
+int rtp_add_recipient(const struct sockaddr_storage *sa) {
+  struct rtp_recipient *r;
+  int rc;
+  pthread_mutex_lock(&rtp_lock);
+  for(r = rtp_recipient_list;
+      r && sockaddrcmp((struct sockaddr *)sa,
+                       (struct sockaddr *)&r->sa);
+      r = r->next)
+    ;
+  if(r)
+    rc = -1;
+  else {
+    r = xmalloc(sizeof *r);
+    memcpy(&r->sa, sa, sizeof *sa);
+    r->next = rtp_recipient_list;
+    rtp_recipient_list = r;
+    rc = 0;
+  }
+  pthread_mutex_unlock(&rtp_lock);
+  return rc;
+}
+
+/** @brief Remove an RTP recipient address
+ * @param sa Pointer to recipient address
+ * @return 0 on success, -1 on error
+ */
+int rtp_remove_recipient(const struct sockaddr_storage *sa) {
+  struct rtp_recipient *r, **rr;
+  int rc;
+  pthread_mutex_lock(&rtp_lock);
+  for(rr = &rtp_recipient_list;
+      (r = *rr) && sockaddrcmp((struct sockaddr *)sa,
+                               (struct sockaddr *)&r->sa);
+      rr = &r->next)
+    ;
+  if(r) {
+    *rr = r->next;
+    xfree(r);
+    rc = 0;
+  } else {
+    disorder_error(0, "bogus rtp_remove_recipient");
+    rc = -1;
+  }
+  pthread_mutex_unlock(&rtp_lock);
+  return rc;
+}
+

 const struct uaudio uaudio_rtp = {
   .name = "rtp",
   .options = rtp_options,
 const struct uaudio uaudio_rtp = {
   .name = "rtp",
   .options = rtp_options,


@@ -382,6 +545,7 @@ const struct uaudio uaudio_rtp = {
   .activate = uaudio_thread_activate,
   .deactivate = uaudio_thread_deactivate,
   .configure = rtp_configure,
   .activate = uaudio_thread_activate,
   .deactivate = uaudio_thread_deactivate,
   .configure = rtp_configure,

+  .flags = UAUDIO_API_SERVER,

 };
 
 /*
 };
 
 /*