udpsrc: store source address and destination IPV6 address

Author: funman

include/upipe/uref_block.h

@@ -47,6 +47,8 @@ UREF_ATTR_VOID_UREF(block, start, UREF_FLAG_BLOCK_START, start of logical block)
 UREF_ATTR_VOID_UREF(block, end, UREF_FLAG_BLOCK_END, end of logical block)
 UREF_ATTR_UNSIGNED(block, header_size, "b.header", global headers size)
 UREF_ATTR_UNSIGNED(block, net_ifindex, "b.ifindex", network interface index)
+UREF_ATTR_SOCKADDR(block, net_srcaddr, "b.srcaddr", source address)
+UREF_ATTR_SOCKADDR(block, net_ipi6_addr, "b.ipi6_addr", destination address when packet was received)
 
 /** @This returns a new uref pointing to a new ubuf pointing to a block.
  * This is equivalent to the two operations sequentially, and is a shortcut.

lib/upipe-modules/upipe_udp_source.c

@@ -239,6 +239,14 @@ static void upipe_udpsrc_worker(struct upump *upump)
         } else if (c->cmsg_level == IPPROTO_IPV6 && c->cmsg_type == IPV6_PKTINFO) {
             struct in6_pktinfo *info = (void*)CMSG_DATA(c);
             ifindex = info->ipi6_ifindex;
+            struct sockaddr_in6 in6 = {
+                .sin6_family = AF_INET6,
+                .sin6_port = 0,
+                .sin6_flowinfo = 0,
+                .sin6_scope_id = 0,
+            };
+            memcpy(&in6.sin6_addr, &info->ipi6_addr, sizeof(in6.sin6_addr));
+            uref_block_set_net_ipi6_addr(uref, (struct sockaddr*)&in6, sizeof(in6));
         }
     }
 
@@ -275,6 +283,8 @@ static void upipe_udpsrc_worker(struct upump *upump)
         memcpy(&upipe_udpsrc->addr, &addr, addrlen);
     }
 
+    uref_block_set_net_srcaddr(uref, (struct sockaddr*)&addr, addrlen);
+
     if (unlikely(ret == 0)) {
         uref_free(uref);
         if (likely(upipe_udpsrc->uclock == NULL)) {