Set udp socket address

Author: viktorxda

include/smolrtsp/transport.h

@@ -69,6 +69,19 @@ SmolRTSP_Transport smolrtsp_transport_tcp(
  */
 SmolRTSP_Transport smolrtsp_transport_udp(int fd) SMOLRTSP_PRIV_MUST_USE;
 
+/**
+ * Creates a new UDP transport with address.
+ *
+ * @param[in] fd The socket file descriptor to be provided with data.
+ * @param[in] addr Pointer to address data (e.g., sockaddr_un or sockaddr_in).
+ * @param[in] len Length of the address in bytes.
+ *
+ * @pre `fd >= 0`
+ * @pre `addr && len > 0 && len < sizeof(self->addr)`
+ */
+SmolRTSP_Transport smolrtsp_transport_udp_address(
+    int fd, void *addr, size_t len) SMOLRTSP_PRIV_MUST_USE;
+
 /**
  * Creates a new datagram socket suitable for #smolrtsp_transport_udp.
  *

src/transport/udp.c

@@ -14,6 +14,8 @@
 
 typedef struct {
     int fd;
+    char addr[128];
+    size_t len;
 } SmolRTSP_UdpTransport;
 
 declImpl(SmolRTSP_Transport, SmolRTSP_UdpTransport);
@@ -28,6 +30,21 @@ SmolRTSP_Transport smolrtsp_transport_udp(int fd) {
     SmolRTSP_UdpTransport *self = malloc(sizeof *self);
     assert(self);
     self->fd = fd;
+    self->len = 0;
+
+    return DYN(SmolRTSP_UdpTransport, SmolRTSP_Transport, self);
+}
+
+SmolRTSP_Transport
+smolrtsp_transport_udp_address(int fd, void *addr, size_t len) {
+    assert(fd >= 0);
+    assert(addr && len > 0 && len < sizeof(self->addr));
+
+    SmolRTSP_UdpTransport *self = malloc(sizeof *self);
+    assert(self);
+    self->fd = fd;
+    self->len = len;
+    memcpy(&self->addr, addr, len);
 
     return DYN(SmolRTSP_UdpTransport, SmolRTSP_Transport, self);
 }
@@ -46,8 +63,8 @@ static int SmolRTSP_UdpTransport_transmit(VSelf, SmolRTSP_IoVecSlice bufs) {
     assert(self);
 
     const struct msghdr msg = {
-        .msg_name = NULL,
-        .msg_namelen = 0,
+        .msg_name = self->addr,
+        .msg_namelen = self->len,
         .msg_iov = bufs.ptr,
         .msg_iovlen = bufs.len,
         .msg_control = NULL,