pipewire/src/modules/module-rtp-source.c

1623 lines
41 KiB
C
Raw Normal View History

/* PipeWire */
/* SPDX-FileCopyrightText: Copyright © 2022 Wim Taymans <wim.taymans@gmail.com> */
/* SPDX-License-Identifier: MIT */
2022-10-04 09:16:17 +02:00
2022-10-04 13:19:14 +02:00
#include "config.h"
2022-10-04 09:16:17 +02:00
#include <limits.h>
#include <string.h>
2022-10-04 09:16:17 +02:00
#include <unistd.h>
2022-10-05 13:04:08 +02:00
#include <sys/stat.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
2022-10-04 09:16:17 +02:00
#include <arpa/inet.h>
#include <netinet/in.h>
2022-10-05 13:04:08 +02:00
#include <net/if.h>
2022-10-04 18:12:48 +02:00
#include <ctype.h>
2022-10-04 09:16:17 +02:00
2022-10-04 13:19:14 +02:00
#include <spa/utils/hook.h>
#include <spa/utils/result.h>
2022-10-04 20:44:07 +02:00
#include <spa/utils/ringbuffer.h>
2022-10-05 13:04:08 +02:00
#include <spa/utils/dll.h>
#include <spa/param/audio/format-utils.h>
#include <spa/control/control.h>
2022-10-04 09:16:17 +02:00
#include <pipewire/pipewire.h>
#include <pipewire/impl.h>
2022-10-04 13:19:14 +02:00
#include <module-rtp/sap.h>
#include <module-rtp/rtp.h>
2022-11-14 21:03:22 +03:00
#ifdef __FreeBSD__
#define ifr_ifindex ifr_index
#endif
2022-10-04 09:16:17 +02:00
/** \page page_module_rtp_source PipeWire Module: RTP source
*
* The `rtp-source` module creates a PipeWire source that receives audio
* RTP packets.
*
* ## Module Options
*
* Options specific to the behavior of this module
*
* - `sap.ip = <str>`: IP address of the SAP messages, default "224.0.0.56"
* - `sap.port = <str>`: port of the SAP messages, default 9875
2022-10-05 18:00:11 +02:00
* - `local.ifname = <str>`: interface name to use
* - `sess.latency.msec = <str>`: target network latency in milliseconds, default 100
* - `stream.props = {}`: properties to be passed to the stream
2022-10-04 09:16:17 +02:00
*
* ## General options
*
* Options with well-known behavior:
*
* - \ref PW_KEY_NODE_NAME
* - \ref PW_KEY_NODE_DESCRIPTION
* - \ref PW_KEY_MEDIA_NAME
* - \ref PW_KEY_MEDIA_CLASS
2022-10-04 09:16:17 +02:00
*
* ## Example configuration
*\code{.unparsed}
* context.modules = [
* { name = libpipewire-module-rtp-source
* args = {
* #sap.ip = 224.0.0.56
* #sap.port = 9875
* #local.ifname = eth0
* sess.latency.msec = 100
* #node.always-process = false # true to receive even when not running
* stream.props = {
* #media.class = "Audio/Source"
* #node.name = "rtp-source"
* }
* stream.rules = [
* { matches = [
* # any of the items in matches needs to match, if one does,
* # actions are emited.
* { # all keys must match the value. ~ in value starts regex.
* #rtp.origin = "wim 3883629975 0 IN IP4 0.0.0.0"
* #rtp.payload = "127"
* #rtp.fmt = "L16/48000/2"
* #rtp.session = "PipeWire RTP Stream on fedora"
* #rtp.ts-offset = 0
* #rtp.ts-refclk = "private"
* }
* ]
* actions = {
* create-stream = {
* #sess.latency.msec = 100
* #sess.ts-direct = false
* #target.object = ""
* }
* }
* }
* ]
* }
* }
* ]
2022-10-04 09:16:17 +02:00
*\endcode
*
2022-10-13 13:16:33 +02:00
* \since 0.3.60
2022-10-04 09:16:17 +02:00
*/
#define NAME "rtp-source"
PW_LOG_TOPIC_STATIC(mod_topic, "mod." NAME);
#define PW_LOG_TOPIC_DEFAULT mod_topic
#define SAP_MIME_TYPE "application/sdp"
#define ERROR_MSEC 2
#define MAX_SESSIONS 16
2022-10-05 13:04:08 +02:00
#define DEFAULT_CLEANUP_INTERVAL_SEC 90
#define DEFAULT_SAP_IP "224.0.0.56"
#define DEFAULT_SAP_PORT 9875
#define DEFAULT_SESS_LATENCY 100
#define BUFFER_SIZE (1u<<22)
#define BUFFER_MASK (BUFFER_SIZE-1)
#define BUFFER_SIZE2 (BUFFER_SIZE>>1)
#define BUFFER_MASK2 (BUFFER_SIZE2-1)
#define USAGE "sap.ip=<SAP IP address to listen on, default "DEFAULT_SAP_IP"> " \
"sap.port=<SAP port to listen on, default "SPA_STRINGIFY(DEFAULT_SAP_PORT)"> " \
"local.ifname=<local interface name to use> " \
"sess.latency.msec=<target network latency, default "SPA_STRINGIFY(DEFAULT_SESS_LATENCY)"> " \
"stream.props= { key=value ... } " \
"stream.rules=<rules> "
static const struct spa_dict_item module_info[] = {
{ PW_KEY_MODULE_AUTHOR, "Wim Taymans <wim.taymans@gmail.com>" },
{ PW_KEY_MODULE_DESCRIPTION, "RTP Source" },
{ PW_KEY_MODULE_USAGE, USAGE },
{ PW_KEY_MODULE_VERSION, PACKAGE_VERSION },
};
2022-10-05 18:00:11 +02:00
2022-10-04 09:16:17 +02:00
struct impl {
struct pw_impl_module *module;
struct spa_hook module_listener;
struct pw_properties *props;
struct pw_context *module_context;
struct pw_loop *loop;
struct pw_loop *data_loop;
2022-10-04 09:16:17 +02:00
struct pw_core *core;
struct spa_hook core_listener;
struct spa_hook core_proxy_listener;
struct spa_source *timer;
2022-10-04 09:16:17 +02:00
struct spa_source *sap_source;
2022-10-05 13:04:08 +02:00
struct pw_properties *stream_props;
2022-10-04 09:16:17 +02:00
unsigned int do_disconnect:1;
2022-10-05 13:04:08 +02:00
char *ifname;
2022-10-05 18:00:11 +02:00
char *sap_ip;
bool always_process;
2022-10-05 13:04:08 +02:00
int sap_port;
2022-10-04 09:16:17 +02:00
int sess_latency_msec;
uint32_t cleanup_interval;
2022-10-04 20:44:07 +02:00
struct spa_list sessions;
2022-10-05 13:04:08 +02:00
uint32_t n_sessions;
2022-10-04 09:16:17 +02:00
};
struct format_info {
uint32_t media_subtype;
2022-10-06 16:04:15 +02:00
uint32_t format;
uint32_t size;
const char *mime;
const char *media_type;
};
static const struct format_info audio_format_info[] = {
{ SPA_MEDIA_SUBTYPE_raw, SPA_AUDIO_FORMAT_U8, 1, "L8", "audio" },
{ SPA_MEDIA_SUBTYPE_raw, SPA_AUDIO_FORMAT_ALAW, 1, "PCMA", "audio" },
{ SPA_MEDIA_SUBTYPE_raw, SPA_AUDIO_FORMAT_ULAW, 1, "PCMU", "audio" },
{ SPA_MEDIA_SUBTYPE_raw, SPA_AUDIO_FORMAT_S16_BE, 2, "L16", "audio" },
{ SPA_MEDIA_SUBTYPE_raw, SPA_AUDIO_FORMAT_S24_BE, 3, "L24", "audio" },
{ SPA_MEDIA_SUBTYPE_control, 0, 1, "rtp-midi", "audio" },
2022-10-06 16:04:15 +02:00
};
static const struct format_info *find_format_info(const char *mime)
{
SPA_FOR_EACH_ELEMENT_VAR(audio_format_info, f)
2022-10-06 16:04:15 +02:00
if (spa_streq(f->mime, mime))
return f;
return NULL;
}
2022-10-04 20:44:07 +02:00
struct sdp_info {
2022-10-04 22:48:26 +02:00
uint16_t hash;
2022-10-04 20:44:07 +02:00
char origin[128];
char session[256];
char channelmap[512];
2022-10-04 09:16:17 +02:00
2022-10-04 20:44:07 +02:00
struct sockaddr_storage sa;
socklen_t salen;
2022-10-04 09:16:17 +02:00
2022-10-04 20:44:07 +02:00
uint16_t port;
uint8_t payload;
2022-10-04 09:16:17 +02:00
2022-10-06 16:04:15 +02:00
const struct format_info *format_info;
struct spa_audio_info info;
uint32_t rate;
2022-10-04 20:44:07 +02:00
uint32_t stride;
uint32_t ts_offset;
char refclk[64];
2022-10-04 20:44:07 +02:00
};
struct session {
struct impl *impl;
struct spa_list link;
uint64_t timestamp;
2022-10-04 20:44:07 +02:00
struct sdp_info info;
2022-10-04 09:16:17 +02:00
2022-10-04 20:44:07 +02:00
struct spa_source *source;
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
struct pw_stream *stream;
struct spa_hook stream_listener;
uint32_t expected_ssrc;
uint16_t expected_seq;
unsigned have_ssrc:1;
unsigned have_seq:1;
unsigned have_sync:1;
2022-10-04 09:16:17 +02:00
2022-10-04 20:44:07 +02:00
struct spa_ringbuffer ring;
uint8_t buffer[BUFFER_SIZE];
2022-10-05 13:04:08 +02:00
struct spa_io_rate_match *rate_match;
struct spa_io_position *position;
2022-10-05 13:04:08 +02:00
struct spa_dll dll;
double corr;
2022-10-05 13:04:08 +02:00
uint32_t target_buffer;
float max_error;
unsigned first:1;
unsigned receiving:1;
unsigned direct_timestamp:1;
float last_timestamp;
float last_time;
2022-10-04 20:44:07 +02:00
};
2022-10-04 09:16:17 +02:00
static void session_touch(struct session *sess)
2022-10-04 20:44:07 +02:00
{
struct timespec ts;
clock_gettime(CLOCK_MONOTONIC, &ts);
sess->timestamp = SPA_TIMESPEC_TO_NSEC(&ts);
2022-10-04 09:16:17 +02:00
}
static void process_audio(struct session *sess)
2022-10-04 09:16:17 +02:00
{
2022-10-04 20:44:07 +02:00
struct pw_buffer *buf;
struct spa_data *d;
uint32_t wanted, timestamp, target_buffer, stride, maxsize;
int32_t avail;
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
if ((buf = pw_stream_dequeue_buffer(sess->stream)) == NULL) {
pw_log_debug("Out of stream buffers: %m");
2022-10-04 20:44:07 +02:00
return;
}
d = buf->buffer->datas;
stride = sess->info.stride;
maxsize = d[0].maxsize / stride;
wanted = buf->requested ? SPA_MIN(buf->requested, maxsize) : maxsize;
2022-10-04 20:44:07 +02:00
if (sess->position && sess->direct_timestamp) {
/* in direct mode, read directly from the timestamp index,
* because sender and receiver are in sync, this would keep
* target_buffer of samples available. */
spa_ringbuffer_read_update(&sess->ring,
sess->position->clock.position);
}
avail = spa_ringbuffer_get_read_index(&sess->ring, &timestamp);
2022-10-04 20:44:07 +02:00
target_buffer = sess->target_buffer;
if (avail < (int32_t)wanted) {
enum spa_log_level level;
memset(d[0].data, 0, wanted * stride);
if (sess->have_sync) {
sess->have_sync = false;
level = SPA_LOG_LEVEL_WARN;
} else {
level = SPA_LOG_LEVEL_DEBUG;
2022-10-06 17:29:05 +02:00
}
pw_log(level, "underrun %d/%u < %u",
avail, target_buffer, wanted);
} else {
2022-10-05 13:04:08 +02:00
float error, corr;
if (sess->first) {
if ((uint32_t)avail > target_buffer) {
uint32_t skip = avail - target_buffer;
pw_log_debug("first: avail:%d skip:%u target:%u",
avail, skip, target_buffer);
timestamp += skip;
avail = target_buffer;
}
sess->first = false;
} else if (avail > (int32_t)SPA_MIN(target_buffer * 8, BUFFER_SIZE / stride)) {
pw_log_warn("overrun %u > %u", avail, target_buffer * 8);
timestamp += avail - target_buffer;
avail = target_buffer;
}
if (!sess->direct_timestamp) {
/* when not using direct timestamp and clocks are not
* in sync, try to adjust our playback rate to keep the
* requested target_buffer bytes in the ringbuffer */
error = (float)target_buffer - (float)avail;
2022-10-05 13:04:08 +02:00
error = SPA_CLAMP(error, -sess->max_error, sess->max_error);
corr = spa_dll_update(&sess->dll, error);
pw_log_debug("avail:%u target:%u error:%f corr:%f", avail,
target_buffer, error, corr);
2022-10-05 13:04:08 +02:00
if (sess->rate_match) {
SPA_FLAG_SET(sess->rate_match->flags,
SPA_IO_RATE_MATCH_FLAG_ACTIVE);
sess->rate_match->rate = 1.0f / corr;
}
2022-10-05 13:04:08 +02:00
}
spa_ringbuffer_read_data(&sess->ring,
sess->buffer,
2022-10-04 20:44:07 +02:00
BUFFER_SIZE,
(timestamp * stride) & BUFFER_MASK,
d[0].data, wanted * stride);
2022-10-05 13:04:08 +02:00
timestamp += wanted;
spa_ringbuffer_read_update(&sess->ring, timestamp);
}
d[0].chunk->size = wanted * stride;
d[0].chunk->stride = stride;
d[0].chunk->offset = 0;
buf->size = wanted;
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
pw_stream_queue_buffer(sess->stream, buf);
2022-10-04 09:16:17 +02:00
}
static void receive_audio(struct session *sess, uint8_t *packet,
uint32_t timestamp, uint32_t payload_offset, uint32_t len)
{
uint32_t plen = len - payload_offset;
uint8_t *payload = &packet[payload_offset];
uint32_t stride = sess->info.stride;
uint32_t samples = plen / stride;
uint32_t write, expected_write;
int32_t filled;
filled = spa_ringbuffer_get_write_index(&sess->ring, &expected_write);
/* we always write to timestamp + delay */
write = timestamp + sess->target_buffer;
if (!sess->have_sync) {
pw_log_info("sync to timestamp %u direct:%d", write, sess->direct_timestamp);
/* we read from timestamp, keeping target_buffer of data
* in the ringbuffer. */
sess->ring.readindex = timestamp;
sess->ring.writeindex = write;
filled = sess->target_buffer;
spa_dll_init(&sess->dll);
spa_dll_set_bw(&sess->dll, SPA_DLL_BW_MIN, 128, sess->info.rate);
memset(sess->buffer, 0, BUFFER_SIZE);
sess->have_sync = true;
} else if (expected_write != write) {
pw_log_debug("unexpected write (%u != %u)",
write, expected_write);
}
if (filled + samples > BUFFER_SIZE / stride) {
pw_log_debug("capture overrun %u + %u > %u", filled, samples,
BUFFER_SIZE / stride);
sess->have_sync = false;
} else {
pw_log_debug("got samples:%u", samples);
spa_ringbuffer_write_data(&sess->ring,
sess->buffer,
BUFFER_SIZE,
(write * stride) & BUFFER_MASK,
payload, (samples * stride));
write += samples;
spa_ringbuffer_write_update(&sess->ring, write);
}
}
static void process_midi(struct session *sess)
{
struct pw_buffer *buf;
struct spa_data *d;
uint32_t timestamp, duration, maxsize, read;
struct spa_pod_builder b;
struct spa_pod_frame f[1];
void *ptr;
struct spa_pod *pod;
struct spa_pod_control *c;
if ((buf = pw_stream_dequeue_buffer(sess->stream)) == NULL) {
pw_log_debug("Out of stream buffers: %m");
return;
}
d = buf->buffer->datas;
maxsize = d[0].maxsize;
/* we always use the graph position to select events, the receiver side is
* responsible for smoothing out the RTP timestamps to graph time */
duration = sess->position->clock.duration;
if (sess->position)
timestamp = sess->position->clock.position;
else
timestamp = 0;
/* we copy events into the buffer based on the rtp timestamp + delay. */
spa_pod_builder_init(&b, d[0].data, maxsize);
spa_pod_builder_push_sequence(&b, &f[0], 0);
while (true) {
int32_t avail = spa_ringbuffer_get_read_index(&sess->ring, &read);
if (avail <= 0)
break;
ptr = SPA_PTROFF(sess->buffer, read & BUFFER_MASK2, void);
if ((pod = spa_pod_from_data(ptr, avail, 0, avail)) == NULL)
goto done;
if (!spa_pod_is_sequence(pod))
goto done;
/* the ringbuffer contains series of sequences, one for each
* received packet */
SPA_POD_SEQUENCE_FOREACH((struct spa_pod_sequence*)pod, c) {
/* try to render with given delay */
uint32_t target = c->offset + sess->target_buffer;
if (timestamp != 0) {
/* skip old packets */
if (target < timestamp)
continue;
/* event for next cycle */
if (target >= timestamp + duration)
goto complete;
} else {
timestamp = target;
}
spa_pod_builder_control(&b, target - timestamp, SPA_CONTROL_Midi);
spa_pod_builder_bytes(&b,
SPA_POD_BODY(&c->value),
SPA_POD_BODY_SIZE(&c->value));
}
/* we completed a sequence (one RTP packet), advance ringbuffer
* and go to the next packet */
read += SPA_PTRDIFF(c, ptr);
spa_ringbuffer_read_update(&sess->ring, read);
}
complete:
spa_pod_builder_pop(&b, &f[0]);
if (b.state.offset > maxsize) {
pw_log_warn("overflow buffer %u %u", b.state.offset, maxsize);
b.state.offset = 0;
}
d[0].chunk->size = b.state.offset;
d[0].chunk->stride = 1;
d[0].chunk->offset = 0;
done:
pw_stream_queue_buffer(sess->stream, buf);
}
static int parse_varlen(uint8_t *p, uint32_t avail, uint32_t *result)
{
uint32_t value = 0, offs = 0;
while (offs < avail) {
uint8_t b = p[offs++];
value = (value << 7) | (b & 0x7f);
if ((b & 0x80) == 0)
break;
}
*result = value;
return offs;
}
static int get_midi_size(uint8_t *p, uint32_t avail)
{
int size;
uint32_t offs = 0, value;
switch (p[offs++]) {
case 0xc0 ... 0xdf:
size = 2;
break;
case 0x80 ... 0xbf:
case 0xe0 ... 0xef:
size = 3;
break;
case 0xff:
case 0xf0:
case 0xf7:
size = parse_varlen(&p[offs], avail - offs, &value);
size += value + 1;
break;
default:
return -EINVAL;
}
return size;
}
static double get_time(struct session *sess)
{
struct timespec ts;
double t;
clock_gettime(CLOCK_MONOTONIC, &ts);
t = sess->position->clock.position / (double) sess->position->clock.rate.denom;
t += (SPA_TIMESPEC_TO_NSEC(&ts) - sess->position->clock.nsec) / (double)SPA_NSEC_PER_SEC;
return t;
}
static void receive_midi(struct session *sess, uint8_t *packet,
uint32_t timestamp, uint32_t payload_offset, uint32_t plen)
{
uint32_t write;
struct rtp_midi_header *hdr;
int32_t filled;
struct spa_pod_builder b;
struct spa_pod_frame f[1];
void *ptr;
uint32_t offs = payload_offset, len, end;
bool first = true;
if (sess->direct_timestamp) {
/* in direct timestamp we attach the RTP timestamp directly on the
* midi events and render them in the corresponding cycle */
if (!sess->have_sync) {
pw_log_info("sync to timestamp %u/ direct:%d", timestamp,
sess->direct_timestamp);
sess->have_sync = true;
}
} else {
/* in non-direct timestamp mode, we relate the graph clock against
* the RTP timestamps */
double ts = timestamp / (float) sess->info.rate;
double t = get_time(sess);
double elapsed, estimated, diff;
/* the elapsed time between RTP timestamps */
elapsed = ts - sess->last_timestamp;
/* for that elapsed time, our clock should have advanced
* by this amount since the last estimation */
estimated = sess->last_time + elapsed * sess->corr;
/* calculate the diff between estimated and current clock time in
* samples */
diff = (estimated - t) * sess->info.rate;
/* no sync or we drifted too far, resync */
if (!sess->have_sync || fabs(diff) > sess->target_buffer) {
sess->corr = 1.0;
spa_dll_set_bw(&sess->dll, SPA_DLL_BW_MIN, 256, sess->info.rate);
pw_log_info("sync to timestamp %u/%f direct:%d", timestamp, t,
sess->direct_timestamp);
sess->have_sync = true;
2023-02-17 15:38:11 +01:00
sess->ring.readindex = sess->ring.writeindex;
} else {
/* update our new rate correction */
sess->corr = spa_dll_update(&sess->dll, diff);
/* our current time is now the estimated time */
t = estimated;
}
pw_log_debug("%f %f %f %f", t, estimated, diff, sess->corr);
timestamp = t * sess->info.rate;
sess->last_timestamp = ts;
sess->last_time = t;
}
filled = spa_ringbuffer_get_write_index(&sess->ring, &write);
if (filled > (int32_t)BUFFER_SIZE2)
return;
hdr = (struct rtp_midi_header *)&packet[offs++];
len = hdr->len;
if (hdr->b) {
len = (len << 8) | hdr->len_b;
offs++;
}
end = len + offs;
if (end > plen)
return;
ptr = SPA_PTROFF(sess->buffer, write & BUFFER_MASK2, void);
/* each packet is written as a sequence of events. The offset is
* the RTP timestamp */
spa_pod_builder_init(&b, ptr, BUFFER_SIZE2 - filled);
spa_pod_builder_push_sequence(&b, &f[0], 0);
while (offs < end) {
uint32_t delta;
int size;
if (first && !hdr->z)
delta = 0;
else
offs += parse_varlen(&packet[offs], end - offs, &delta);
timestamp += delta * sess->corr;
spa_pod_builder_control(&b, timestamp, SPA_CONTROL_Midi);
size = get_midi_size(&packet[offs], end - offs);
if (size <= 0 || offs + size > end) {
pw_log_warn("invalid size (%08x) %d (%u %u)",
packet[offs], size, offs, end);
break;
}
spa_pod_builder_bytes(&b, &packet[offs], size);
offs += size;
first = false;
}
spa_pod_builder_pop(&b, &f[0]);
write += b.state.offset;
spa_ringbuffer_write_update(&sess->ring, write);
}
static void stream_io_changed(void *data, uint32_t id, void *area, uint32_t size)
{
struct session *sess = data;
switch (id) {
case SPA_IO_RateMatch:
sess->rate_match = area;
break;
case SPA_IO_Position:
sess->position = area;
break;
}
}
static void stream_destroy(void *d)
{
struct session *sess = d;
spa_hook_remove(&sess->stream_listener);
sess->stream = NULL;
}
static void stream_process(void *data)
{
struct session *sess = data;
switch (sess->info.info.media_type) {
case SPA_MEDIA_TYPE_audio:
process_audio(sess);
break;
case SPA_MEDIA_TYPE_application:
process_midi(sess);
break;
}
}
2022-10-04 22:48:26 +02:00
static void
on_rtp_io(void *data, int fd, uint32_t mask)
{
struct session *sess = data;
2022-10-05 13:04:08 +02:00
struct rtp_header *hdr;
ssize_t len, hlen;
uint8_t buffer[2048];
2022-10-04 22:48:26 +02:00
if (mask & SPA_IO_IN) {
2022-10-05 13:04:08 +02:00
uint16_t seq;
uint32_t timestamp;
2022-10-04 22:48:26 +02:00
2022-10-05 13:04:08 +02:00
if ((len = recv(fd, buffer, sizeof(buffer), 0)) < 0)
goto receive_error;
2022-10-04 22:48:26 +02:00
if (len < 12)
2022-10-05 13:04:08 +02:00
goto short_packet;
2022-10-04 22:48:26 +02:00
hdr = (struct rtp_header*)buffer;
if (hdr->v != 2)
2022-10-05 13:04:08 +02:00
goto invalid_version;
2022-10-04 22:48:26 +02:00
hlen = 12 + hdr->cc * 4;
if (hlen > len)
2022-10-05 13:04:08 +02:00
goto invalid_len;
2022-10-04 22:48:26 +02:00
2022-10-05 13:04:08 +02:00
if (sess->have_ssrc && sess->expected_ssrc != hdr->ssrc)
goto unexpected_ssrc;
sess->expected_ssrc = hdr->ssrc;
sess->have_ssrc = true;
seq = ntohs(hdr->sequence_number);
if (sess->have_seq && sess->expected_seq != seq) {
pw_log_info("unexpected seq (%d != %d)", seq, sess->expected_seq);
sess->have_sync = false;
2022-10-05 13:04:08 +02:00
}
sess->expected_seq = seq + 1;
sess->have_seq = true;
timestamp = ntohl(hdr->timestamp) - sess->info.ts_offset;
2022-10-05 13:04:08 +02:00
switch (sess->info.info.media_type) {
case SPA_MEDIA_TYPE_audio:
receive_audio(sess, buffer, timestamp, hlen, len);
break;
case SPA_MEDIA_TYPE_application:
receive_midi(sess, buffer, timestamp, hlen, len);
2022-10-04 22:48:26 +02:00
}
sess->receiving = true;
2022-10-04 22:48:26 +02:00
}
2022-10-05 13:04:08 +02:00
return;
receive_error:
pw_log_warn("recv error: %m");
return;
short_packet:
pw_log_warn("short packet received");
return;
invalid_version:
pw_log_warn("invalid RTP version");
return;
invalid_len:
pw_log_warn("invalid RTP length");
return;
unexpected_ssrc:
pw_log_warn("unexpected SSRC (expected %u != %u)",
sess->expected_ssrc, hdr->ssrc);
return;
2022-10-04 22:48:26 +02:00
}
static int make_socket(const struct sockaddr* sa, socklen_t salen, char *ifname)
2022-10-04 22:48:26 +02:00
{
int af, fd, val, res;
2022-10-05 13:04:08 +02:00
struct ifreq req;
2022-10-04 22:48:26 +02:00
af = sa->sa_family;
if ((fd = socket(af, SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0)) < 0) {
pw_log_error("socket failed: %m");
return -errno;
}
#ifdef SO_TIMESTAMP
val = 1;
if (setsockopt(fd, SOL_SOCKET, SO_TIMESTAMP, &val, sizeof(val)) < 0) {
res = -errno;
pw_log_error("setsockopt failed: %m");
goto error;
}
#endif
val = 1;
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) < 0) {
res = -errno;
pw_log_error("setsockopt failed: %m");
goto error;
}
2022-10-05 13:04:08 +02:00
spa_zero(req);
if (ifname) {
snprintf(req.ifr_name, sizeof(req.ifr_name), "%s", ifname);
res = ioctl(fd, SIOCGIFINDEX, &req);
if (res < 0)
pw_log_warn("SIOCGIFINDEX %s failed: %m", ifname);
}
2022-10-04 22:48:26 +02:00
res = 0;
if (af == AF_INET) {
static const uint32_t ipv4_mcast_mask = 0xe0000000;
struct sockaddr_in *sa4 = (struct sockaddr_in*)sa;
2022-10-04 22:48:26 +02:00
if ((ntohl(sa4->sin_addr.s_addr) & ipv4_mcast_mask) == ipv4_mcast_mask) {
2022-10-05 13:04:08 +02:00
struct ip_mreqn mr4;
2022-10-04 22:48:26 +02:00
memset(&mr4, 0, sizeof(mr4));
mr4.imr_multiaddr = sa4->sin_addr;
2022-10-05 13:04:08 +02:00
mr4.imr_ifindex = req.ifr_ifindex;
2022-10-04 22:48:26 +02:00
res = setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mr4, sizeof(mr4));
} else {
sa4->sin_addr.s_addr = INADDR_ANY;
2022-10-04 22:48:26 +02:00
}
} else if (af == AF_INET6) {
struct sockaddr_in6 *sa6 = (struct sockaddr_in6*)sa;
2022-10-04 22:48:26 +02:00
if (sa6->sin6_addr.s6_addr[0] == 0xff) {
struct ipv6_mreq mr6;
memset(&mr6, 0, sizeof(mr6));
mr6.ipv6mr_multiaddr = sa6->sin6_addr;
2022-10-05 13:04:08 +02:00
mr6.ipv6mr_interface = req.ifr_ifindex;
2022-10-04 22:48:26 +02:00
res = setsockopt(fd, IPPROTO_IPV6, IPV6_JOIN_GROUP, &mr6, sizeof(mr6));
} else {
sa6->sin6_addr = in6addr_any;
2022-10-04 22:48:26 +02:00
}
} else {
res = -EINVAL;
goto error;
}
if (res < 0) {
res = -errno;
pw_log_error("join mcast failed: %m");
goto error;
}
if (bind(fd, sa, salen) < 0) {
res = -errno;
pw_log_error("bind() failed: %m");
2022-10-04 22:48:26 +02:00
goto error;
}
return fd;
error:
return res;
}
static uint32_t msec_to_samples(struct sdp_info *info, uint32_t msec)
2022-10-05 13:04:08 +02:00
{
return msec * info->rate / 1000;
2022-10-05 13:04:08 +02:00
}
static void session_free(struct session *sess)
{
if (sess->impl) {
pw_log_info("free session %s %s", sess->info.origin, sess->info.session);
sess->impl->n_sessions--;
spa_list_remove(&sess->link);
}
if (sess->stream)
pw_stream_destroy(sess->stream);
if (sess->source)
pw_loop_destroy_source(sess->impl->data_loop, sess->source);
free(sess);
}
struct session_info {
struct session *session;
struct pw_properties *props;
bool matched;
};
static int rule_matched(void *data, const char *location, const char *action,
const char *str, size_t len)
{
struct session_info *i = data;
int res = 0;
i->matched = true;
if (spa_streq(action, "create-stream")) {
pw_properties_update_string(i->props, str, len);
}
return res;
}
static int session_start(struct impl *impl, struct session *session) {
int fd;
if (session->source)
return 0;
pw_log_info("starting RTP listener");
if ((fd = make_socket((const struct sockaddr *)&session->info.sa,
session->info.salen, impl->ifname)) < 0) {
pw_log_error("failed to create socket: %m");
return fd;
}
session->source = pw_loop_add_io(impl->data_loop, fd,
SPA_IO_IN, true, on_rtp_io, session);
if (session->source == NULL) {
pw_log_error("can't create io source: %m");
close(fd);
return -errno;
}
return 0;
}
static void session_stop(struct impl *impl, struct session *session) {
if (!session->source)
return;
pw_log_info("stopping RTP listener");
pw_loop_destroy_source(
session->impl->data_loop,
session->source
);
session->source = NULL;
}
static void on_stream_state_changed(void *d, enum pw_stream_state old,
enum pw_stream_state state, const char *error)
{
struct session *sess = d;
struct impl *impl = sess->impl;
switch (state) {
case PW_STREAM_STATE_UNCONNECTED:
pw_log_info("stream disconnected, unloading");
pw_impl_module_schedule_destroy(impl->module);
break;
case PW_STREAM_STATE_ERROR:
pw_log_error("stream error: %s", error);
break;
case PW_STREAM_STATE_STREAMING:
if ((errno = -session_start(impl, sess)) < 0)
pw_log_error("failed to start RTP stream: %m");
break;
case PW_STREAM_STATE_PAUSED:
if (!impl->always_process)
session_stop(impl, sess);
break;
default:
break;
}
}
static const struct pw_stream_events out_stream_events = {
PW_VERSION_STREAM_EVENTS,
.destroy = stream_destroy,
.state_changed = on_stream_state_changed,
.io_changed = stream_io_changed,
.process = stream_process
};
2022-10-05 13:04:08 +02:00
static int session_new(struct impl *impl, struct sdp_info *info)
2022-10-04 09:16:17 +02:00
{
2022-10-04 20:44:07 +02:00
struct session *session;
2022-10-04 09:16:17 +02:00
const struct spa_pod *params[1];
struct spa_pod_builder b;
uint32_t n_params;
uint8_t buffer[1024];
2022-10-04 20:44:07 +02:00
struct pw_properties *props;
int res, sess_latency_msec;
const char *str;
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
if (impl->n_sessions >= MAX_SESSIONS) {
pw_log_warn("too many sessions (%u >= %u)", impl->n_sessions, MAX_SESSIONS);
return -EMFILE;
}
2022-10-04 20:44:07 +02:00
session = calloc(1, sizeof(struct session));
if (session == NULL)
return -errno;
2022-10-05 13:04:08 +02:00
session->info = *info;
session->first = true;
2022-10-04 20:44:07 +02:00
2022-10-05 13:04:08 +02:00
props = pw_properties_copy(impl->stream_props);
if (props == NULL) {
res = -errno;
goto error;
}
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
pw_properties_set(props, "rtp.origin", info->origin);
pw_properties_setf(props, "rtp.payload", "%u", info->payload);
pw_properties_setf(props, "rtp.fmt", "%s/%u/%u", info->format_info->mime,
info->rate, info->info.info.raw.channels);
2022-10-05 13:04:08 +02:00
if (info->session[0]) {
pw_properties_set(props, "rtp.session", info->session);
pw_properties_setf(props, PW_KEY_MEDIA_NAME, "RTP Stream (%s)",
info->session);
pw_properties_setf(props, PW_KEY_NODE_NAME, "%s",
info->session);
2022-10-05 13:04:08 +02:00
} else {
pw_properties_set(props, PW_KEY_MEDIA_NAME, "RTP Stream");
}
pw_properties_setf(props, "rtp.ts-offset", "%u", info->ts_offset);
pw_properties_set(props, "rtp.ts-refclk", info->refclk);
2022-10-04 23:27:44 +02:00
if ((str = pw_properties_get(impl->props, "stream.rules")) != NULL) {
struct session_info sinfo = {
.session = session,
.props = props,
};
pw_conf_match_rules(str, strlen(str), NAME, &props->dict,
rule_matched, &sinfo);
if (!sinfo.matched) {
res = 0;
pw_log_info("session '%s' was not matched", info->session);
goto error;
}
}
session->direct_timestamp = pw_properties_get_bool(props, "sess.ts-direct", false);
pw_log_info("new session %s %s direct:%d", info->origin, info->session,
session->direct_timestamp);
2022-10-04 09:16:17 +02:00
sess_latency_msec = pw_properties_get_uint32(props,
"sess.latency.msec", impl->sess_latency_msec);
session->target_buffer = msec_to_samples(info, sess_latency_msec);
session->max_error = msec_to_samples(info, ERROR_MSEC);
pw_properties_setf(props, PW_KEY_NODE_RATE, "1/%d", info->rate);
pw_properties_setf(props, PW_KEY_NODE_LATENCY, "%d/%d",
session->target_buffer / 2, info->rate);
spa_dll_init(&session->dll);
spa_dll_set_bw(&session->dll, SPA_DLL_BW_MIN, 128, info->rate);
session->corr = 1.0;
if (info->channelmap[0]) {
pw_properties_set(props, PW_KEY_NODE_CHANNELNAMES, info->channelmap);
pw_log_info("channelmap: %s", info->channelmap);
}
2022-10-05 13:04:08 +02:00
session->stream = pw_stream_new(impl->core,
2022-10-04 20:44:07 +02:00
"rtp-source playback", props);
if (session->stream == NULL) {
res = -errno;
pw_log_error("can't create stream: %m");
goto error;
}
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
pw_stream_add_listener(session->stream,
&session->stream_listener,
2022-10-04 20:44:07 +02:00
&out_stream_events, session);
2022-10-04 09:16:17 +02:00
n_params = 0;
spa_pod_builder_init(&b, buffer, sizeof(buffer));
switch (info->info.media_type) {
case SPA_MEDIA_TYPE_audio:
params[n_params++] = spa_format_audio_build(&b,
SPA_PARAM_EnumFormat, &info->info);
break;
case SPA_MEDIA_TYPE_application:
params[n_params++] = spa_pod_builder_add_object(&b,
SPA_TYPE_OBJECT_Format, SPA_PARAM_EnumFormat,
SPA_FORMAT_mediaType, SPA_POD_Id(SPA_MEDIA_TYPE_application),
SPA_FORMAT_mediaSubtype, SPA_POD_Id(SPA_MEDIA_SUBTYPE_control));
break;
default:
return -EINVAL;
}
2022-10-04 09:16:17 +02:00
2022-10-05 13:04:08 +02:00
if ((res = pw_stream_connect(session->stream,
2022-10-04 09:16:17 +02:00
PW_DIRECTION_OUTPUT,
PW_ID_ANY,
PW_STREAM_FLAG_MAP_BUFFERS |
PW_STREAM_FLAG_AUTOCONNECT |
PW_STREAM_FLAG_RT_PROCESS,
params, n_params)) < 0) {
pw_log_error("can't connect stream: %s", spa_strerror(res));
goto error;
}
2022-10-04 09:16:17 +02:00
if (impl->always_process &&
(res = session_start(impl, session)) < 0)
goto error;
2022-10-04 22:48:26 +02:00
session_touch(session);
session->impl = impl;
2022-10-04 22:48:26 +02:00
spa_list_append(&impl->sessions, &session->link);
2022-10-05 13:04:08 +02:00
impl->n_sessions++;
2022-10-04 22:48:26 +02:00
2022-10-04 09:16:17 +02:00
return 0;
error:
session_free(session);
return res;
2022-10-04 20:44:07 +02:00
}
2022-10-04 22:48:26 +02:00
static struct session *session_find(struct impl *impl, struct sdp_info *info)
{
struct session *sess;
spa_list_for_each(sess, &impl->sessions, link) {
if (info->hash == sess->info.hash &&
spa_streq(info->origin, sess->info.origin))
return sess;
}
return NULL;
}
2022-10-04 18:12:48 +02:00
static int parse_sdp_c(struct impl *impl, char *c, struct sdp_info *info)
{
int res;
c[strcspn(c, "/")] = 0;
if (spa_strstartswith(c, "c=IN IP4 ")) {
struct sockaddr_in *sa = (struct sockaddr_in*) &info->sa;
2022-10-04 22:48:26 +02:00
c += strlen("c=IN IP4 ");
2022-10-04 18:12:48 +02:00
if (inet_pton(AF_INET, c, &sa->sin_addr) <= 0) {
res = -errno;
pw_log_warn("inet_pton(%s) failed: %m", c);
goto error;
}
sa->sin_family = AF_INET;
info->salen = sizeof(struct sockaddr_in);
}
else if (spa_strstartswith(c, "c=IN IP6 ")) {
struct sockaddr_in6 *sa = (struct sockaddr_in6*) &info->sa;
2022-10-04 22:48:26 +02:00
c += strlen("c=IN IP6 ");
2022-10-04 18:12:48 +02:00
if (inet_pton(AF_INET6, c, &sa->sin6_addr) <= 0) {
res = -errno;
pw_log_warn("inet_pton(%s) failed: %m", c);
goto error;
}
sa->sin6_family = AF_INET6;
info->salen = sizeof(struct sockaddr_in6);
} else
return -EINVAL;
res= 0;
error:
return res;
}
static int parse_sdp_m(struct impl *impl, char *c, struct sdp_info *info)
2022-10-04 13:19:14 +02:00
{
2022-10-04 18:12:48 +02:00
int port, payload;
if (!spa_strstartswith(c, "m=audio "))
return -EINVAL;
2022-10-04 22:48:26 +02:00
c += strlen("m=audio ");
2022-10-04 18:12:48 +02:00
if (sscanf(c, "%i RTP/AVP %i", &port, &payload) != 2)
return -EINVAL;
if (port <= 0 || port > 0xFFFF)
return -EINVAL;
if (payload < 0 || payload > 127)
return -EINVAL;
info->port = (uint16_t) port;
info->payload = (uint8_t) payload;
2022-10-04 13:19:14 +02:00
return 0;
}
// some AES67 devices have channelmap encoded in i=*
// if `i` record is found, it matches the template
// and channel count matches, name the channels respectively
// `i=2 channels: 01, 08` is the format
static int parse_sdp_i(struct impl *impl, char *c, struct sdp_info *info)
{
if (!strstr(c, " channels: ")) {
return 0;
}
c += strlen("i=");
c[strcspn(c, " ")] = '\0';
uint32_t channels;
if (sscanf(c, "%u", &channels) != 1 || channels <= 0 || channels > SPA_AUDIO_MAX_CHANNELS)
return 0;
c += strcspn(c, "\0");
c += strlen(" channels: ");
strncpy(info->channelmap, c, sizeof(info->channelmap) - 1);
return 0;
}
static int parse_sdp_a_rtpmap(struct impl *impl, char *c, struct sdp_info *info)
2022-10-04 18:12:48 +02:00
{
int payload, len, rate, channels;
if (!spa_strstartswith(c, "a=rtpmap:"))
return 0;
2022-10-04 22:48:26 +02:00
c += strlen("a=rtpmap:");
2022-10-04 18:12:48 +02:00
if (sscanf(c, "%i %n", &payload, &len) != 1)
return -EINVAL;
if (payload < 0 || payload > 127)
return -EINVAL;
if (payload != info->payload)
return 0;
c += len;
2022-10-06 16:04:15 +02:00
c[strcspn(c, "/")] = 0;
info->format_info = find_format_info(c);
if (info->format_info == NULL)
2022-10-04 18:12:48 +02:00
return -EINVAL;
2022-10-06 16:04:15 +02:00
info->stride = info->format_info->size;
info->info.media_subtype = info->format_info->media_subtype;
2022-10-06 16:04:15 +02:00
c += strlen(c) + 1;
2022-10-04 18:12:48 +02:00
switch (info->info.media_subtype) {
case SPA_MEDIA_SUBTYPE_raw:
info->info.media_type = SPA_MEDIA_TYPE_audio;
info->info.info.raw.format = info->format_info->format;
if (sscanf(c, "%u/%u", &rate, &channels) == 2) {
info->info.info.raw.channels = channels;
} else if (sscanf(c, "%u", &rate) == 1) {
info->info.info.raw.channels = 1;
} else
return -EINVAL;
info->info.info.raw.rate = rate;
pw_log_debug("rate: %d, ch: %d", rate, channels);
2022-10-04 20:44:07 +02:00
if (channels == 1) {
info->info.info.raw.position[0] = SPA_AUDIO_CHANNEL_MONO;
} else if (channels == 2) {
info->info.info.raw.position[0] = SPA_AUDIO_CHANNEL_FL;
info->info.info.raw.position[1] = SPA_AUDIO_CHANNEL_FR;
}
info->stride *= channels;
info->rate = rate;
break;
case SPA_MEDIA_SUBTYPE_control:
info->info.media_type = SPA_MEDIA_TYPE_application;
if (sscanf(c, "%u", &rate) != 1)
return -EINVAL;
info->rate = rate;
break;
}
2022-10-04 18:12:48 +02:00
return 0;
}
static int parse_sdp_a_mediaclk(struct impl *impl, char *c, struct sdp_info *info)
{
if (!spa_strstartswith(c, "a=mediaclk:"))
return 0;
c += strlen("a=mediaclk:");
if (spa_strstartswith(c, "direct=")) {
int offset;
c += strlen("direct=");
if (sscanf(c, "%i", &offset) != 1)
return -EINVAL;
info->ts_offset = offset;
} else if (spa_strstartswith(c, "sender")) {
info->ts_offset = 0;
}
return 0;
}
static int parse_sdp_a_ts_refclk(struct impl *impl, char *c, struct sdp_info *info)
{
if (!spa_strstartswith(c, "a=ts-refclk:"))
return 0;
c += strlen("a=ts-refclk:");
snprintf(info->refclk, sizeof(info->refclk), "%s", c);
return 0;
}
2022-10-04 18:12:48 +02:00
static int parse_sdp(struct impl *impl, char *sdp, struct sdp_info *info)
{
char *s = sdp;
int count = 0, res = 0;
size_t l;
while (*s) {
if ((l = strcspn(s, "\r\n")) < 2)
goto too_short;
s[l] = 0;
2022-10-04 23:27:44 +02:00
pw_log_debug("%d: %s", count, s);
2022-10-04 18:12:48 +02:00
if (count++ == 0 && strcmp(s, "v=0") != 0)
goto invalid_version;
if (spa_strstartswith(s, "o="))
2022-10-04 20:44:07 +02:00
snprintf(info->origin, sizeof(info->origin), "%s", &s[2]);
2022-10-04 18:12:48 +02:00
else if (spa_strstartswith(s, "s="))
2022-10-04 20:44:07 +02:00
snprintf(info->session, sizeof(info->session), "%s", &s[2]);
2022-10-04 18:12:48 +02:00
else if (spa_strstartswith(s, "c="))
res = parse_sdp_c(impl, s, info);
else if (spa_strstartswith(s, "m="))
res = parse_sdp_m(impl, s, info);
else if (spa_strstartswith(s, "a=rtpmap:"))
res = parse_sdp_a_rtpmap(impl, s, info);
else if (spa_strstartswith(s, "a=mediaclk:"))
res = parse_sdp_a_mediaclk(impl, s, info);
else if (spa_strstartswith(s, "a=ts-refclk:"))
res = parse_sdp_a_ts_refclk(impl, s, info);
else if (spa_strstartswith(s, "i="))
res = parse_sdp_i(impl, s, info);
2022-10-04 18:12:48 +02:00
if (res < 0)
goto error;
s += l + 1;
while (isspace(*s))
s++;
}
2022-10-04 22:48:26 +02:00
if (((struct sockaddr*) &info->sa)->sa_family == AF_INET)
((struct sockaddr_in*) &info->sa)->sin_port = htons(info->port);
else
((struct sockaddr_in6*) &info->sa)->sin6_port = htons(info->port);
2022-10-04 18:12:48 +02:00
return 0;
too_short:
pw_log_warn("SDP: line starting with `%.6s...' too short", s);
return -EINVAL;
invalid_version:
pw_log_warn("SDP: invalid first version line `%*s'", (int)l, s);
return -EINVAL;
error:
2022-10-04 22:48:26 +02:00
pw_log_warn("SDP: error: %s", spa_strerror(res));
2022-10-04 18:12:48 +02:00
return res;
}
static int parse_sap(struct impl *impl, void *data, size_t len)
2022-10-04 13:19:14 +02:00
{
struct sap_header *header;
2022-10-04 18:12:48 +02:00
char *mime, *sdp;
struct sdp_info info;
2022-10-04 22:48:26 +02:00
struct session *sess;
2022-10-04 18:12:48 +02:00
int res;
size_t offs;
bool bye;
2022-10-04 13:19:14 +02:00
if (len < 8)
return -EINVAL;
header = (struct sap_header*) data;
if (header->v != 1)
return -EINVAL;
if (header->e)
return -ENOTSUP;
if (header->c)
return -ENOTSUP;
offs = header->a ? 12 : 8;
offs += header->auth_len * 4;
if (len <= offs)
return -EINVAL;
mime = SPA_PTROFF(data, offs, char);
2022-10-04 22:48:26 +02:00
if (spa_strstartswith(mime, "v=0")) {
2022-10-04 13:19:14 +02:00
sdp = mime;
mime = SAP_MIME_TYPE;
} else if (spa_streq(mime, SAP_MIME_TYPE))
2022-10-04 13:19:14 +02:00
sdp = SPA_PTROFF(mime, strlen(mime)+1, char);
else
return -EINVAL;
2022-10-04 23:27:44 +02:00
pw_log_debug("got sap: %s %s", mime, sdp);
2022-10-04 22:48:26 +02:00
2022-10-04 18:12:48 +02:00
spa_zero(info);
if ((res = parse_sdp(impl, sdp, &info)) < 0)
return res;
bye = header->t;
2022-10-04 22:48:26 +02:00
sess = session_find(impl, &info);
if (sess == NULL) {
if (!bye)
session_new(impl, &info);
2022-10-04 22:48:26 +02:00
} else {
if (bye)
session_free(sess);
else
session_touch(sess);
2022-10-04 22:48:26 +02:00
}
2022-10-04 18:12:48 +02:00
return res;
2022-10-04 13:19:14 +02:00
}
static void
on_sap_io(void *data, int fd, uint32_t mask)
2022-10-04 09:16:17 +02:00
{
2022-10-04 13:19:14 +02:00
struct impl *impl = data;
if (mask & SPA_IO_IN) {
uint8_t buffer[2048];
ssize_t len;
if ((len = recv(fd, buffer, sizeof(buffer), 0)) < 0) {
pw_log_warn("recv error: %m");
return;
}
if ((size_t)len >= sizeof(buffer))
return;
buffer[len] = 0;
parse_sap(impl, buffer, len);
}
}
static int start_sap_listener(struct impl *impl)
{
struct sockaddr_in sa4;
struct sockaddr_in6 sa6;
struct sockaddr *sa;
socklen_t salen;
2022-10-04 22:48:26 +02:00
int fd, res;
2022-10-04 09:16:17 +02:00
2022-10-05 18:00:11 +02:00
if (inet_pton(AF_INET, impl->sap_ip, &sa4.sin_addr) > 0) {
2022-10-04 22:48:26 +02:00
sa4.sin_family = AF_INET;
2022-10-05 13:04:08 +02:00
sa4.sin_port = htons(impl->sap_port);
2022-10-04 13:19:14 +02:00
sa = (struct sockaddr*) &sa4;
salen = sizeof(sa4);
2022-10-05 18:00:11 +02:00
} else if (inet_pton(AF_INET6, impl->sap_ip, &sa6.sin6_addr) > 0) {
2022-10-04 22:48:26 +02:00
sa6.sin6_family = AF_INET6;
2022-10-05 13:04:08 +02:00
sa6.sin6_port = htons(impl->sap_port);
2022-10-04 13:19:14 +02:00
sa = (struct sockaddr*) &sa6;
salen = sizeof(sa6);
} else
return -EINVAL;
2022-10-04 09:16:17 +02:00
if ((fd = make_socket(sa, salen, impl->ifname)) < 0)
2022-10-04 22:48:26 +02:00
return fd;
2022-10-04 13:19:14 +02:00
pw_log_info("starting SAP listener");
impl->sap_source = pw_loop_add_io(impl->loop, fd,
SPA_IO_IN, true, on_sap_io, impl);
if (impl->sap_source == NULL) {
res = -errno;
goto error;
}
return 0;
2022-10-04 09:16:17 +02:00
error:
close(fd);
return res;
}
static void on_timer_event(void *data, uint64_t expirations)
{
struct impl *impl = data;
struct timespec now;
struct session *sess, *tmp;
uint64_t timestamp, interval;
clock_gettime(CLOCK_MONOTONIC, &now);
timestamp = SPA_TIMESPEC_TO_NSEC(&now);
interval = impl->cleanup_interval * SPA_NSEC_PER_SEC;
spa_list_for_each_safe(sess, tmp, &impl->sessions, link) {
if (sess->timestamp + interval < timestamp) {
pw_log_debug("More than %lu elapsed from last advertisement at %lu",
interval, sess->timestamp);
if (!sess->receiving) {
pw_log_info("SAP timeout, closing inactive RTP source");
session_free(sess);
} else {
pw_log_info("SAP timeout, keeping active RTP source");
}
}
sess->receiving = false;
}
}
2022-10-04 20:44:07 +02:00
static void core_destroy(void *d)
{
struct impl *impl = d;
spa_hook_remove(&impl->core_listener);
impl->core = NULL;
pw_impl_module_schedule_destroy(impl->module);
}
static const struct pw_proxy_events core_proxy_events = {
.destroy = core_destroy,
};
static void impl_destroy(struct impl *impl)
{
struct session *sess;
spa_list_consume(sess, &impl->sessions, link)
session_free(sess);
if (impl->core && impl->do_disconnect)
pw_core_disconnect(impl->core);
if (impl->sap_source)
pw_loop_destroy_source(impl->loop, impl->sap_source);
if (impl->timer)
pw_loop_destroy_source(impl->loop, impl->timer);
2022-10-05 13:04:08 +02:00
pw_properties_free(impl->stream_props);
2022-10-04 20:44:07 +02:00
pw_properties_free(impl->props);
2022-10-05 13:04:08 +02:00
free(impl->ifname);
2022-10-05 18:00:11 +02:00
free(impl->sap_ip);
2022-10-04 20:44:07 +02:00
free(impl);
}
static void module_destroy(void *d)
{
struct impl *impl = d;
spa_hook_remove(&impl->module_listener);
impl_destroy(impl);
}
static const struct pw_impl_module_events module_events = {
PW_VERSION_IMPL_MODULE_EVENTS,
.destroy = module_destroy,
};
static void on_core_error(void *d, uint32_t id, int seq, int res, const char *message)
{
struct impl *impl = d;
pw_log_error("error id:%u seq:%d res:%d (%s): %s",
id, seq, res, spa_strerror(res), message);
if (id == PW_ID_CORE && res == -EPIPE)
pw_impl_module_schedule_destroy(impl->module);
}
static const struct pw_core_events core_events = {
PW_VERSION_CORE_EVENTS,
.error = on_core_error,
};
2022-10-04 09:16:17 +02:00
SPA_EXPORT
int pipewire__module_init(struct pw_impl_module *module, const char *args)
{
struct pw_context *context = pw_impl_module_get_context(module);
struct impl *impl;
const char *str;
struct timespec value, interval;
2022-10-04 09:16:17 +02:00
int res = 0;
PW_LOG_TOPIC_INIT(mod_topic);
impl = calloc(1, sizeof(struct impl));
if (impl == NULL)
return -errno;
spa_list_init(&impl->sessions);
2022-10-04 09:16:17 +02:00
if (args == NULL)
args = "";
impl->props = pw_properties_new_string(args);
impl->stream_props = pw_properties_new(NULL, NULL);
if (impl->props == NULL || impl->stream_props == NULL) {
2022-10-04 09:16:17 +02:00
res = -errno;
pw_log_error( "can't create properties: %m");
goto out;
}
impl->module = module;
impl->module_context = context;
impl->loop = pw_context_get_main_loop(context);
impl->data_loop = pw_data_loop_get_loop(pw_context_get_data_loop(context));
2022-10-04 09:16:17 +02:00
if (pw_properties_get(impl->stream_props, PW_KEY_NODE_VIRTUAL) == NULL)
pw_properties_set(impl->stream_props, PW_KEY_NODE_VIRTUAL, "true");
if (pw_properties_get(impl->stream_props, PW_KEY_NODE_NETWORK) == NULL)
pw_properties_set(impl->stream_props, PW_KEY_NODE_NETWORK, "true");
2022-10-05 13:04:08 +02:00
if ((str = pw_properties_get(impl->props, "stream.props")) != NULL)
pw_properties_update_string(impl->stream_props, str, strlen(str));
2022-10-07 16:12:22 +02:00
str = pw_properties_get(impl->props, "local.ifname");
2022-10-05 13:04:08 +02:00
impl->ifname = str ? strdup(str) : NULL;
impl->always_process = pw_properties_get_bool(impl->props, PW_KEY_NODE_ALWAYS_PROCESS, false);
str = pw_properties_get(impl->props, "sap.ip");
2022-10-05 18:00:11 +02:00
impl->sap_ip = strdup(str ? str : DEFAULT_SAP_IP);
impl->sap_port = pw_properties_get_uint32(impl->props,
2022-10-05 18:00:11 +02:00
"sap.port", DEFAULT_SAP_PORT);
impl->sess_latency_msec = pw_properties_get_uint32(impl->props,
2022-10-05 13:04:08 +02:00
"sess.latency.msec", DEFAULT_SESS_LATENCY);
impl->cleanup_interval = pw_properties_get_uint32(impl->props,
"sap.interval.sec", DEFAULT_CLEANUP_INTERVAL_SEC);
2022-10-04 09:16:17 +02:00
impl->core = pw_context_get_object(impl->module_context, PW_TYPE_INTERFACE_Core);
if (impl->core == NULL) {
str = pw_properties_get(impl->props, PW_KEY_REMOTE_NAME);
2022-10-04 09:16:17 +02:00
impl->core = pw_context_connect(impl->module_context,
pw_properties_new(
PW_KEY_REMOTE_NAME, str,
NULL),
0);
impl->do_disconnect = true;
}
if (impl->core == NULL) {
res = -errno;
pw_log_error("can't connect: %m");
goto out;
}
pw_proxy_add_listener((struct pw_proxy*)impl->core,
&impl->core_proxy_listener,
&core_proxy_events, impl);
pw_core_add_listener(impl->core,
&impl->core_listener,
&core_events, impl);
impl->timer = pw_loop_add_timer(impl->loop, on_timer_event, impl);
if (impl->timer == NULL) {
res = -errno;
pw_log_error("can't create timer source: %m");
goto out;
}
value.tv_sec = 0;
value.tv_nsec = 1;
interval.tv_sec = impl->cleanup_interval;
interval.tv_nsec = 0;
pw_loop_update_timer(impl->loop, impl->timer, &value, &interval, false);
2022-10-04 09:16:17 +02:00
if ((res = start_sap_listener(impl)) < 0)
goto out;
pw_impl_module_add_listener(module, &impl->module_listener, &module_events, impl);
pw_impl_module_update_properties(module, &SPA_DICT_INIT_ARRAY(module_info));
pw_log_info("Successfully loaded module-rtp-source");
return 0;
out:
impl_destroy(impl);
return res;
}