]> git.sesse.net Git - vlc/blobdiff - src/stream_output/sap.c
linsys_hdsdi: do not fail on EINTR
[vlc] / src / stream_output / sap.c
index 4f59b1e5c6c75cc9795edaa2802141b8f9de95f3..0a7874835026eb3e76da156eb14fe71260fea686 100644 (file)
 /* SAP is always on that port */
 #define IPPORT_SAP 9875
 
+/* A SAP session descriptor, enqueued in the SAP handler queue */
 struct session_descriptor_t
 {
-    struct sockaddr_storage addr;
-    socklen_t addrlen;
-
-    bool b_ssm;
+    struct session_descriptor_t *next;
+    size_t  length;
+    uint8_t data[];
 };
 
-/* A SAP session descriptor, enqueued in the SAP handler queue */
-typedef struct sap_session_t
-{
-    struct sap_session_t *next;
-    const session_descriptor_t *p_sd;
-    size_t                length;
-    uint8_t               data[];
-} sap_session_t;
-
 /* A SAP announce address. For each of these, we run the
  * control flow algorithm */
 typedef struct sap_address_t
@@ -76,17 +67,11 @@ typedef struct sap_address_t
     unsigned                interval;
 
     unsigned                session_count;
-    sap_session_t          *first;
+    session_descriptor_t   *first;
 } sap_address_t;
 
-/* The SAP handler, running in a separate thread */
-struct sap_handler_t
-{
-    VLC_COMMON_MEMBERS
-
-    vlc_mutex_t    lock;
-    sap_address_t *first;
-};
+static sap_address_t *sap_addrs = NULL;
+static vlc_mutex_t sap_mutex = VLC_STATIC_MUTEX;
 
 #define SAP_MAX_BUFFER 65534
 #define MIN_INTERVAL 2
@@ -94,32 +79,6 @@ struct sap_handler_t
 
 static void *RunThread (void *);
 
-/**
- * Create the SAP handler
- *
- * \param p_announce a VLC object
- * \return the newly created SAP handler or NULL on error
- */
-static sap_handler_t *SAP_Create (vlc_object_t *p_announce)
-{
-    sap_handler_t *p_sap;
-
-    p_sap = vlc_custom_create (p_announce, sizeof (*p_sap), "sap sender");
-    if (p_sap == NULL)
-        return NULL;
-
-    vlc_mutex_init (&p_sap->lock);
-    p_sap->first = NULL;
-    return p_sap;
-}
-
-static void SAP_Destroy (sap_handler_t *p_sap)
-{
-    assert (p_sap->first == NULL);
-    vlc_mutex_destroy (&p_sap->lock);
-    vlc_object_release (p_sap);
-}
-
 static sap_address_t *AddressCreate (vlc_object_t *obj, const char *group)
 {
     int fd = net_ConnectUDP (obj, group, IPPORT_SAP, 255);
@@ -181,7 +140,7 @@ static void *RunThread (void *self)
 
     for (;;)
     {
-        sap_session_t *p_session;
+        session_descriptor_t *p_session;
         mtime_t deadline;
 
         while (addr->first == NULL)
@@ -204,34 +163,47 @@ static void *RunThread (void *self)
     assert (0);
 }
 
+#undef sout_AnnounceRegisterSDP
 /**
- * Add a SAP announce
+ *  Registers a new session with the announce handler, using a pregenerated SDP
+ *
+ * \param obj a VLC object
+ * \param sdp the SDP to register
+ * \param dst session address (needed for SAP address auto detection)
+ * \return the new session descriptor structure
  */
-static int SAP_Add (sap_handler_t *p_sap, session_descriptor_t *p_session,
-                    const char *sdp)
+session_descriptor_t *
+sout_AnnounceRegisterSDP (vlc_object_t *obj, const char *sdp,
+                          const char *dst)
 {
     int i;
     char psz_addr[NI_MAXNUMERICHOST];
-    sap_session_t *p_sap_session;
-    mtime_t i_hash;
     union
     {
         struct sockaddr     a;
         struct sockaddr_in  in;
         struct sockaddr_in6 in6;
     } addr;
-    socklen_t addrlen;
+    socklen_t addrlen = 0;
+    struct addrinfo *res;
+
+    msg_Dbg (obj, "adding SAP session");
 
-    addrlen = p_session->addrlen;
-    if ((addrlen == 0) || (addrlen > sizeof (addr)))
+    if (vlc_getaddrinfo (dst, 0, NULL, &res) == 0)
     {
-        msg_Err( p_sap, "No/invalid address specified for SAP announce" );
-        return VLC_EGENERIC;
+        if (res->ai_addrlen <= sizeof (addr))
+            memcpy (&addr, res->ai_addr, res->ai_addrlen);
+        addrlen = res->ai_addrlen;
+        freeaddrinfo (res);
     }
 
-    /* Determine SAP multicast address automatically */
-    memcpy (&addr, &p_session->addr, addrlen);
+    if (addrlen == 0 || addrlen > sizeof (addr))
+    {
+        msg_Err (obj, "No/invalid address specified for SAP announce" );
+        return NULL;
+    }
 
+    /* Determine SAP multicast address automatically */
     switch (addr.a.sa_family)
     {
 #if defined (HAVE_INET_PTON) || defined (_WIN32)
@@ -277,9 +249,9 @@ static int SAP_Add (sap_handler_t *p_sap, session_descriptor_t *p_session,
 
             if( ipv4 == 0 )
             {
-                msg_Err( p_sap, "Out-of-scope multicast address "
-                         "not supported by SAP" );
-                return VLC_EGENERIC;
+                msg_Err (obj, "Out-of-scope multicast address "
+                         "not supported by SAP");
+                return NULL;
             }
 
             addr.in.sin_addr.s_addr = ipv4;
@@ -287,9 +259,9 @@ static int SAP_Add (sap_handler_t *p_sap, session_descriptor_t *p_session,
         }
 
         default:
-            msg_Err( p_sap, "Address family %d not supported by SAP",
-                     addr.a.sa_family );
-            return VLC_EGENERIC;
+            msg_Err (obj, "Address family %d not supported by SAP",
+                     addr.a.sa_family);
+            return NULL;
     }
 
     i = vlc_getnameinfo( &addr.a, addrlen,
@@ -297,75 +269,72 @@ static int SAP_Add (sap_handler_t *p_sap, session_descriptor_t *p_session,
 
     if( i )
     {
-        msg_Err( p_sap, "%s", gai_strerror( i ) );
-        return VLC_EGENERIC;
+        msg_Err (obj, "%s", gai_strerror (i));
+        return NULL;
     }
 
     /* Find/create SAP address thread */
-    msg_Dbg( p_sap, "using SAP address: %s", psz_addr);
-
-    vlc_mutex_lock (&p_sap->lock);
     sap_address_t *sap_addr;
-    for (sap_addr = p_sap->first; sap_addr; sap_addr = sap_addr->next)
+
+    msg_Dbg (obj, "using SAP address: %s", psz_addr);
+    vlc_mutex_lock (&sap_mutex);
+    for (sap_addr = sap_addrs; sap_addr; sap_addr = sap_addr->next)
         if (!strcmp (psz_addr, sap_addr->group))
             break;
 
     if (sap_addr == NULL)
     {
-        sap_addr = AddressCreate (VLC_OBJECT(p_sap), psz_addr);
+        sap_addr = AddressCreate (obj, psz_addr);
         if (sap_addr == NULL)
         {
-            vlc_mutex_unlock (&p_sap->lock);
-            return VLC_EGENERIC;
+            vlc_mutex_unlock (&sap_mutex);
+            return NULL;
         }
-        sap_addr->next = p_sap->first;
-        p_sap->first = sap_addr;
+        sap_addr->next = sap_addrs;
+        sap_addrs = sap_addr;
     }
     /* Switch locks.
      * NEVER take the global SAP lock when holding a SAP thread lock! */
     vlc_mutex_lock (&sap_addr->lock);
-    vlc_mutex_unlock (&p_sap->lock);
+    vlc_mutex_unlock (&sap_mutex);
 
-    size_t headsize = 20, length;
+    size_t length = 20;
     switch (sap_addr->orig.ss_family)
     {
 #ifdef AF_INET6
         case AF_INET6:
-            headsize += 16;
+            length += 16;
             break;
 #endif
         case AF_INET:
-            headsize += 4;
+            length += 4;
             break;
         default:
             assert (0);
     }
 
     /* XXX: Check for dupes */
-    length = headsize + strlen (sdp);
-    p_sap_session = malloc (sizeof (*p_sap_session) + length + 1);
-    if (p_sap_session == NULL)
+    length += strlen (sdp);
+
+    session_descriptor_t *session = malloc (sizeof (*session) + length);
+    if (unlikely(session == NULL))
     {
         vlc_mutex_unlock (&sap_addr->lock);
-        return VLC_EGENERIC; /* NOTE: we should destroy the thread if left unused */
+        return NULL; /* NOTE: we should destroy the thread if left unused */
     }
-    p_sap_session->next = sap_addr->first;
-    sap_addr->first = p_sap_session;
-    p_sap_session->p_sd = p_session;
-    p_sap_session->length = length;
+    session->next = sap_addr->first;
+    sap_addr->first = session;
+    session->length = length;
 
     /* Build the SAP Headers */
-    uint8_t *psz_head = p_sap_session->data;
+    uint8_t *buf = session->data;
 
     /* SAPv1, not encrypted, not compressed */
-    psz_head[0] = 0x20;
-    psz_head[1] = 0x00; /* No authentication length */
-
-    i_hash = mdate();
-    psz_head[2] = i_hash >> 8; /* Msg id hash */
-    psz_head[3] = i_hash;      /* Msg id hash 2 */
+    buf[0] = 0x20;
+    buf[1] = 0x00; /* No authentication length */
+    SetWBE(buf + 2, mdate()); /* ID hash */
 
-    headsize = 4;
+    size_t offset = 4;
     switch (sap_addr->orig.ss_family)
     {
 #ifdef AF_INET6
@@ -373,9 +342,9 @@ static int SAP_Add (sap_handler_t *p_sap, session_descriptor_t *p_session,
         {
             const struct in6_addr *a6 =
                 &((const struct sockaddr_in6 *)&sap_addr->orig)->sin6_addr;
-            memcpy (psz_head + headsize, a6, 16);
-            psz_head[0] |= 0x10; /* IPv6 flag */
-            headsize += 16;
+            memcpy (buf + offset, a6, 16);
+            buf[0] |= 0x10; /* IPv6 flag */
+            offset += 16;
             break;
         }
 #endif
@@ -383,52 +352,56 @@ static int SAP_Add (sap_handler_t *p_sap, session_descriptor_t *p_session,
         {
             const struct in_addr *a4 =
                 &((const struct sockaddr_in *)&sap_addr->orig)->sin_addr;
-            memcpy (psz_head + headsize, a4, 4);
-            headsize += 4;
+            memcpy (buf + offset, a4, 4);
+            offset += 4;
             break;
         }
 
     }
 
-    memcpy (psz_head + headsize, "application/sdp", 16);
-    headsize += 16;
+    memcpy (buf + offset, "application/sdp", 16);
+    offset += 16;
 
     /* Build the final message */
-    strcpy((char *)psz_head + headsize, sdp);
+    strcpy((char *)buf + offset, sdp);
 
     sap_addr->session_count++;
     vlc_cond_signal (&sap_addr->wait);
     vlc_mutex_unlock (&sap_addr->lock);
-    return VLC_SUCCESS;
+    return session;
 }
 
+#undef sout_AnnounceUnRegister
 /**
- * Remove a SAP Announce
+ *  Unregisters an existing session
+ *
+ * \param obj a VLC object
+ * \param session the session descriptor
  */
-static void SAP_Del (sap_handler_t *p_sap,
-                     const session_descriptor_t *p_session)
+void sout_AnnounceUnRegister (vlc_object_t *obj, session_descriptor_t *session)
 {
-    vlc_mutex_lock (&p_sap->lock);
-
-    /* TODO: give a handle back in SAP_Add, and use that... */
     sap_address_t *addr, **paddr;
-    sap_session_t *session, **psession;
+    session_descriptor_t **psession;
 
-    paddr = &p_sap->first;
-    for (addr = p_sap->first; addr; addr = addr->next)
+    msg_Dbg (obj, "removing SAP session");
+    vlc_mutex_lock (&sap_mutex);
+    paddr = &sap_addrs;
+    for (;;)
     {
+        addr = *paddr;
+        assert (addr != NULL);
+
         psession = &addr->first;
         vlc_mutex_lock (&addr->lock);
-        for (session = addr->first; session; session = session->next)
+        while (*psession != NULL)
         {
-            if (session->p_sd == p_session)
+            if (*psession == session)
                 goto found;
-            psession = &session->next;
+            psession = &(*psession)->next;
         }
         vlc_mutex_unlock (&addr->lock);
         paddr = &addr->next;
     }
-    assert (0);
 
 found:
     *psession = session->next;
@@ -436,7 +409,7 @@ found:
     if (addr->first == NULL)
         /* Last session for this address -> unlink the address */
         *paddr = addr->next;
-    vlc_mutex_unlock (&p_sap->lock);
+    vlc_mutex_unlock (&sap_mutex);
 
     if (addr->first == NULL)
     {
@@ -453,98 +426,3 @@ found:
 
     free (session);
 }
-
-/****************************************************************************
- * Sout-side functions
- ****************************************************************************/
-
-static void sap_destroy (vlc_object_t *p_this)
-{
-    libvlc_priv (p_this->p_libvlc)->p_sap = NULL;
-}
-
-#undef sout_AnnounceRegisterSDP
-
-static vlc_mutex_t sap_mutex = VLC_STATIC_MUTEX;
-
-/**
- *  Registers a new session with the announce handler, using a pregenerated SDP
- *
- * \param obj a VLC object
- * \param psz_sdp the SDP to register
- * \param psz_dst session address (needed for SAP address auto detection)
- * \return the new session descriptor structure
- */
-session_descriptor_t *
-sout_AnnounceRegisterSDP( vlc_object_t *obj, const char *psz_sdp,
-                          const char *psz_dst )
-{
-    session_descriptor_t *p_session = calloc( 1, sizeof (*p_session) );
-    if( !p_session )
-        return NULL;
-
-    /* GRUIK. We should not convert back-and-forth from string to numbers */
-    struct addrinfo *res;
-    if (vlc_getaddrinfo (psz_dst, 0, NULL, &res) == 0)
-    {
-        if (res->ai_addrlen <= sizeof (p_session->addr))
-            memcpy (&p_session->addr, res->ai_addr,
-                    p_session->addrlen = res->ai_addrlen);
-        freeaddrinfo (res);
-    }
-
-    vlc_mutex_lock (&sap_mutex);
-    sap_handler_t *p_sap = libvlc_priv (obj->p_libvlc)->p_sap;
-    if (p_sap == NULL)
-    {
-        p_sap = SAP_Create (VLC_OBJECT (obj->p_libvlc));
-        libvlc_priv (obj->p_libvlc)->p_sap = p_sap;
-        vlc_object_set_destructor ((vlc_object_t *)p_sap, sap_destroy);
-    }
-    else
-        vlc_object_hold ((vlc_object_t *)p_sap);
-    vlc_mutex_unlock (&sap_mutex);
-
-    if (p_sap == NULL)
-        goto error;
-
-    msg_Dbg (obj, "adding SAP session");
-    if (SAP_Add (p_sap, p_session, psz_sdp))
-    {
-        vlc_mutex_lock (&sap_mutex);
-        vlc_object_release ((vlc_object_t *)p_sap);
-        vlc_mutex_unlock (&sap_mutex);
-        goto error;
-    }
-
-    return p_session;
-
-error:
-    free (p_session);
-    return NULL;
-}
-
-#undef sout_AnnounceUnRegister
-/**
- *  Unregisters an existing session
- *
- * \param obj a VLC object
- * \param p_session the session descriptor
- * \return VLC_SUCCESS or an error
- */
-int sout_AnnounceUnRegister( vlc_object_t *obj,
-                             session_descriptor_t *p_session )
-{
-    sap_handler_t *p_sap = libvlc_priv (obj->p_libvlc)->p_sap;
-
-    msg_Dbg (obj, "removing SAP session");
-    SAP_Del (p_sap, p_session);
-
-    vlc_mutex_lock (&sap_mutex);
-    vlc_object_release ((vlc_object_t *)p_sap);
-    vlc_mutex_unlock (&sap_mutex);
-
-    free (p_session);
-
-    return 0;
-}