pipewire/src/modules/module-pipe-tunnel.c
Wim Taymans cd81b5f39a spa: add spa_json_begin_array/object and relaxed versions
Add spa_json_begin_array/object to replace
spa_json_init+spa_json_begin_array/object

This function is better because it does not waste a useless spa_json
structure as an iterator. The relaxed versions also error out when the
container is mismatched because parsing a mismatched container is not
going to give any results anyway.
2024-09-16 09:50:33 +02:00

1015 lines
27 KiB
C

/* PipeWire */
/* SPDX-FileCopyrightText: Copyright © 2021 Sanchayan Maity <sanchayan@asymptotic.io> */
/* SPDX-FileCopyrightText: Copyright © 2022 Wim Taymans */
/* SPDX-License-Identifier: MIT */
#include <string.h>
#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/uio.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <limits.h>
#include <math.h>
#include "config.h"
#include <spa/utils/result.h>
#include <spa/utils/string.h>
#include <spa/utils/json.h>
#include <spa/utils/ringbuffer.h>
#include <spa/utils/dll.h>
#include <spa/debug/types.h>
#include <spa/pod/builder.h>
#include <spa/param/audio/format-utils.h>
#include <spa/param/latency-utils.h>
#include <spa/param/audio/raw.h>
#include <pipewire/impl.h>
#include <pipewire/i18n.h>
/** \page page_module_pipe_tunnel Unix Pipe Tunnel
*
* The pipe-tunnel module provides a source or sink that tunnels all audio to
* or from a unix pipe respectively.
*
* ## Module Name
*
* `libpipewire-module-pipe-tunnel`
*
* ## Module Options
*
* - `tunnel.mode`: the desired tunnel to create. (Default `playback`)
* - `tunnel.may-pause`: if the tunnel stream is allowed to pause on xrun
* - `pipe.filename`: the filename of the pipe.
* - `stream.props`: Extra properties for the local stream.
*
* When `tunnel.mode` is `capture`, a capture stream on the default source is
* created. The samples captured from the source will be written to the pipe.
*
* When `tunnel.mode` is `sink`, a sink node is created. Samples played on the
* sink will be written to the pipe.
*
* When `tunnel.mode` is `playback`, a playback stream on the default sink is
* created. The samples read from the pipe will be played on the sink.
*
* When `tunnel.mode` is `source`, a source node is created. Samples read from
* the the pipe will be made available on the source.
*
* `tunnel.may-pause` allows the tunnel stream to become inactive (paused) when
* there is no data in the fifo or when the fifo is full. For `capture` and
* `playback` `tunnel.mode` this is by default true. For `source` and `sink`
* `tunnel.mode`, this is by default false. A paused stream will consume no
* CPU and will resume when the fifo becomes readable or writable again.
*
* When `pipe.filename` is not given, a default fifo in `/tmp/fifo_input` or
* `/tmp/fifo_output` will be created that can be written and read respectively,
* depending on the selected `tunnel.mode`.
*
* ## General options
*
* Options with well-known behavior.
*
* - \ref PW_KEY_REMOTE_NAME
* - \ref PW_KEY_AUDIO_FORMAT
* - \ref PW_KEY_AUDIO_RATE
* - \ref PW_KEY_AUDIO_CHANNELS
* - \ref SPA_KEY_AUDIO_POSITION
* - \ref PW_KEY_NODE_LATENCY
* - \ref PW_KEY_NODE_NAME
* - \ref PW_KEY_NODE_DESCRIPTION
* - \ref PW_KEY_NODE_GROUP
* - \ref PW_KEY_NODE_VIRTUAL
* - \ref PW_KEY_MEDIA_CLASS
* - \ref PW_KEY_TARGET_OBJECT to specify the remote name or serial id to link to
*
* When not otherwise specified, the pipe will accept or produce a
* 16 bits, stereo, 48KHz sample stream.
*
* ## Example configuration of a pipe playback stream
*
*\code{.unparsed}
* context.modules = [
* { name = libpipewire-module-pipe-tunnel
* args = {
* tunnel.mode = playback
* #tunnel.may-pause = true
* # Set the pipe name to tunnel to
* pipe.filename = "/tmp/fifo_output"
* #audio.format=<sample format>
* #audio.rate=<sample rate>
* #audio.channels=<number of channels>
* #audio.position=<channel map>
* #target.object=<remote target node>
* stream.props = {
* # extra sink properties
* }
* }
* }
* ]
*\endcode
*/
#define NAME "pipe-tunnel"
#define DEFAULT_CAPTURE_FILENAME "/tmp/fifo_input"
#define DEFAULT_PLAYBACK_FILENAME "/tmp/fifo_output"
#define DEFAULT_FORMAT "S16"
#define DEFAULT_RATE 48000
#define DEFAULT_CHANNELS 2
#define DEFAULT_POSITION "[ FL FR ]"
#define RINGBUFFER_SIZE (1u << 22)
#define RINGBUFFER_MASK (RINGBUFFER_SIZE-1)
PW_LOG_TOPIC_STATIC(mod_topic, "mod." NAME);
#define PW_LOG_TOPIC_DEFAULT mod_topic
#define MODULE_USAGE "( remote.name=<remote> ) " \
"( node.latency=<latency as fraction> ) " \
"( node.name=<name of the nodes> ) " \
"( node.description=<description of the nodes> ) " \
"( target.object=<remote node target name or serial> ) "\
"( audio.format=<sample format> ) " \
"( audio.rate=<sample rate> ) " \
"( audio.channels=<number of channels> ) " \
"( audio.position=<channel map> ) " \
"( tunnel.mode=capture|playback|sink|source )" \
"( tunnel.may-pause=<bool, if the stream can pause> )" \
"( pipe.filename=<filename> )" \
"( stream.props=<properties> ) "
static const struct spa_dict_item module_props[] = {
{ PW_KEY_MODULE_AUTHOR, "Wim Taymans <wim.taymans@gmail.com>" },
{ PW_KEY_MODULE_DESCRIPTION, "Create a UNIX pipe tunnel" },
{ PW_KEY_MODULE_USAGE, MODULE_USAGE },
{ PW_KEY_MODULE_VERSION, PACKAGE_VERSION },
};
struct impl {
struct pw_context *context;
struct pw_loop *main_loop;
struct pw_loop *data_loop;
#define MODE_PLAYBACK 0
#define MODE_CAPTURE 1
#define MODE_SINK 2
#define MODE_SOURCE 3
uint32_t mode;
struct pw_properties *props;
struct pw_impl_module *module;
struct spa_hook module_listener;
struct pw_core *core;
struct spa_hook core_proxy_listener;
struct spa_hook core_listener;
char *filename;
unsigned int unlink_fifo;
int fd;
struct spa_source *socket;
struct spa_source *timer;
struct pw_properties *stream_props;
enum pw_direction direction;
struct pw_stream *stream;
struct spa_hook stream_listener;
struct spa_audio_info_raw info;
uint32_t frame_size;
unsigned int do_disconnect:1;
unsigned int driving:1;
unsigned int may_pause:1;
unsigned int paused:1;
struct spa_ringbuffer ring;
void *buffer;
uint32_t target_buffer;
struct spa_io_rate_match *rate_match;
struct spa_io_position *position;
struct spa_dll dll;
float max_error;
double corr;
uint64_t next_time;
unsigned int have_sync:1;
unsigned int underrun:1;
};
static uint64_t get_time_ns(struct impl *impl)
{
struct timespec now;
if (spa_system_clock_gettime(impl->data_loop->system, CLOCK_MONOTONIC, &now) < 0)
return 0;
return SPA_TIMESPEC_TO_NSEC(&now);
}
static int set_timeout(struct impl *impl, uint64_t time)
{
struct timespec timeout, interval;
timeout.tv_sec = time / SPA_NSEC_PER_SEC;
timeout.tv_nsec = time % SPA_NSEC_PER_SEC;
interval.tv_sec = 0;
interval.tv_nsec = 0;
pw_loop_update_timer(impl->data_loop,
impl->timer, &timeout, &interval, true);
return 0;
}
static void on_timeout(void *d, uint64_t expirations)
{
struct impl *impl = d;
uint64_t duration, current_time;
uint32_t rate, index;
int32_t avail;
struct spa_io_position *pos = impl->position;
if (SPA_LIKELY(pos)) {
duration = pos->clock.target_duration;
rate = pos->clock.target_rate.denom;
} else {
duration = 1024;
rate = 48000;
}
pw_log_debug("timeout %"PRIu64, duration);
current_time = impl->next_time;
impl->next_time += (uint64_t)(duration / impl->corr * 1e9 / rate);
avail = spa_ringbuffer_get_read_index(&impl->ring, &index);
if (SPA_LIKELY(pos)) {
pos->clock.nsec = current_time;
pos->clock.rate = pos->clock.target_rate;
pos->clock.position += pos->clock.duration;
pos->clock.duration = pos->clock.target_duration;
pos->clock.delay = SPA_SCALE32_UP(avail, rate, impl->info.rate);
pos->clock.rate_diff = impl->corr;
pos->clock.next_nsec = impl->next_time;
}
set_timeout(impl, impl->next_time);
pw_stream_trigger_process(impl->stream);
}
static void stream_destroy(void *d)
{
struct impl *impl = d;
spa_hook_remove(&impl->stream_listener);
impl->stream = NULL;
}
static void stream_state_changed(void *d, enum pw_stream_state old,
enum pw_stream_state state, const char *error)
{
struct impl *impl = d;
switch (state) {
case PW_STREAM_STATE_ERROR:
case PW_STREAM_STATE_UNCONNECTED:
pw_impl_module_schedule_destroy(impl->module);
break;
case PW_STREAM_STATE_PAUSED:
if (impl->direction == PW_DIRECTION_OUTPUT) {
pw_loop_update_io(impl->data_loop, impl->socket, impl->paused ? SPA_IO_IN : 0);
set_timeout(impl, 0);
}
break;
case PW_STREAM_STATE_STREAMING:
if (impl->direction == PW_DIRECTION_OUTPUT) {
pw_loop_update_io(impl->data_loop, impl->socket, SPA_IO_IN);
impl->driving = pw_stream_is_driving(impl->stream);
if (impl->driving) {
impl->next_time = get_time_ns(impl);
set_timeout(impl, impl->next_time);
}
}
break;
default:
break;
}
}
static int do_pause(struct spa_loop *loop, bool async, uint32_t seq, const void *data,
size_t size, void *user_data)
{
struct impl *impl = user_data;
const bool *paused = data;
pw_log_info("set paused: %d", *paused);
impl->paused = *paused;
pw_stream_set_active(impl->stream, !*paused);
return 0;
}
static void pause_stream(struct impl *impl, bool paused)
{
if (!impl->may_pause)
return;
if (impl->direction == PW_DIRECTION_INPUT)
pw_loop_update_io(impl->data_loop, impl->socket, paused ? SPA_IO_OUT : 0);
pw_loop_invoke(impl->main_loop, do_pause, 1, &paused, sizeof(bool), false, impl);
}
static void playback_stream_process(void *data)
{
struct impl *impl = data;
struct pw_buffer *buf;
uint32_t i, size, offs;
ssize_t written;
if ((buf = pw_stream_dequeue_buffer(impl->stream)) == NULL) {
pw_log_debug("out of buffers: %m");
return;
}
for (i = 0; i < buf->buffer->n_datas; i++) {
struct spa_data *d;
d = &buf->buffer->datas[i];
offs = SPA_MIN(d->chunk->offset, d->maxsize);
size = SPA_MIN(d->chunk->size, d->maxsize - offs);
while (size > 0) {
written = write(impl->fd, SPA_MEMBER(d->data, offs, void), size);
if (written < 0) {
if (errno == EINTR) {
/* retry if interrupted */
continue;
} else if (errno == EAGAIN || errno == EWOULDBLOCK) {
/* Don't continue writing */
pw_log_debug("pipe (%s) overrun: %m", impl->filename);
pause_stream(impl, true);
break;
} else {
pw_log_warn("Failed to write to pipe (%s): %m",
impl->filename);
}
}
offs += written;
size -= written;
}
}
pw_stream_queue_buffer(impl->stream, buf);
}
static void update_rate(struct impl *impl, uint32_t filled)
{
float error;
if (impl->rate_match == NULL)
return;
error = (float)impl->target_buffer - (float)(filled);
error = SPA_CLAMP(error, -impl->max_error, impl->max_error);
impl->corr = spa_dll_update(&impl->dll, error);
pw_log_debug("error:%f corr:%f current:%u target:%u",
error, impl->corr, filled, impl->target_buffer);
if (!impl->driving) {
SPA_FLAG_SET(impl->rate_match->flags, SPA_IO_RATE_MATCH_FLAG_ACTIVE);
impl->rate_match->rate = 1.0 / impl->corr;
}
}
static void capture_stream_process(void *data)
{
struct impl *impl = data;
struct pw_buffer *buf;
struct spa_data *bd;
uint32_t req, index, size;
int32_t avail;
if ((buf = pw_stream_dequeue_buffer(impl->stream)) == NULL) {
pw_log_warn("out of buffers: %m");
return;
}
bd = &buf->buffer->datas[0];
if ((req = buf->requested * impl->frame_size) == 0)
req = 4096 * impl->frame_size;
size = SPA_MIN(req, bd->maxsize);
size = SPA_ROUND_DOWN(size, impl->frame_size);
avail = spa_ringbuffer_get_read_index(&impl->ring, &index);
pw_log_debug("avail %d %u %u", avail, index, size);
if (avail < (int32_t)size) {
memset(bd->data, 0, size);
if (avail >= 0) {
if (!impl->underrun) {
pw_log_warn("underrun %d < %u", avail, size);
impl->underrun = true;
}
pause_stream(impl, true);
}
impl->have_sync = false;
}
if (avail > (int32_t)(impl->target_buffer * 3)) {
pw_log_warn("resync %d > %u", avail, (int32_t)(impl->target_buffer * 3));
impl->have_sync = false;
}
if (avail > (int32_t)RINGBUFFER_SIZE) {
index += avail - impl->target_buffer;
avail = impl->target_buffer;
pw_log_warn("overrun %d > %u", avail, RINGBUFFER_SIZE);
}
if (avail > 0) {
avail = SPA_ROUND_DOWN(avail, impl->frame_size);
update_rate(impl, avail);
avail = SPA_MIN(size, (uint32_t)avail);
spa_ringbuffer_read_data(&impl->ring,
impl->buffer, RINGBUFFER_SIZE,
index & RINGBUFFER_MASK,
bd->data, avail);
index += avail;
spa_ringbuffer_read_update(&impl->ring, index);
impl->underrun = false;
}
bd->chunk->offset = 0;
bd->chunk->size = size;
bd->chunk->stride = impl->frame_size;
pw_stream_queue_buffer(impl->stream, buf);
}
static void stream_io_changed(void *data, uint32_t id, void *area, uint32_t size)
{
struct impl *impl = data;
switch (id) {
case SPA_IO_RateMatch:
impl->rate_match = area;
break;
case SPA_IO_Position:
impl->position = area;
break;
}
}
static const struct pw_stream_events playback_stream_events = {
PW_VERSION_STREAM_EVENTS,
.destroy = stream_destroy,
.state_changed = stream_state_changed,
.process = playback_stream_process
};
static const struct pw_stream_events capture_stream_events = {
PW_VERSION_STREAM_EVENTS,
.destroy = stream_destroy,
.io_changed = stream_io_changed,
.state_changed = stream_state_changed,
.process = capture_stream_process,
};
static int create_stream(struct impl *impl)
{
int res;
uint32_t n_params;
const struct spa_pod *params[1];
uint8_t buffer[1024];
struct spa_pod_builder b;
impl->stream = pw_stream_new(impl->core, "pipe", impl->stream_props);
impl->stream_props = NULL;
if (impl->stream == NULL)
return -errno;
if (impl->direction == PW_DIRECTION_OUTPUT) {
pw_stream_add_listener(impl->stream,
&impl->stream_listener,
&capture_stream_events, impl);
} else {
pw_stream_add_listener(impl->stream,
&impl->stream_listener,
&playback_stream_events, impl);
}
n_params = 0;
spa_pod_builder_init(&b, buffer, sizeof(buffer));
params[n_params++] = spa_format_audio_raw_build(&b,
SPA_PARAM_EnumFormat, &impl->info);
impl->paused = false;
if ((res = pw_stream_connect(impl->stream,
impl->direction,
PW_ID_ANY,
PW_STREAM_FLAG_AUTOCONNECT |
PW_STREAM_FLAG_MAP_BUFFERS |
PW_STREAM_FLAG_RT_PROCESS,
params, n_params)) < 0)
return res;
return 0;
}
static inline void
set_iovec(struct spa_ringbuffer *rbuf, void *buffer, uint32_t size,
uint32_t offset, struct iovec *iov, uint32_t len)
{
iov[0].iov_len = SPA_MIN(len, size - offset);
iov[0].iov_base = SPA_PTROFF(buffer, offset, void);
iov[1].iov_len = len - iov[0].iov_len;
iov[1].iov_base = buffer;
}
static int handle_pipe_read(struct impl *impl)
{
ssize_t nread;
int32_t filled;
uint32_t index;
struct iovec iov[2];
filled = spa_ringbuffer_get_write_index(&impl->ring, &index);
if (!impl->have_sync) {
memset(impl->buffer, 0, RINGBUFFER_SIZE);
}
if (filled < 0) {
pw_log_warn("%p: underrun write:%u filled:%d",
impl, index, filled);
}
set_iovec(&impl->ring,
impl->buffer, RINGBUFFER_SIZE,
index & RINGBUFFER_MASK,
iov, RINGBUFFER_SIZE);
nread = read(impl->fd, iov[0].iov_base, iov[0].iov_len);
if (nread > 0) {
index += nread;
filled += nread;
if (nread == (ssize_t)iov[0].iov_len) {
nread = read(impl->fd, iov[1].iov_base, iov[1].iov_len);
if (nread > 0) {
index += nread;
filled += nread;
}
}
}
if (!impl->have_sync) {
impl->ring.readindex = index - impl->target_buffer;
spa_dll_init(&impl->dll);
spa_dll_set_bw(&impl->dll, SPA_DLL_BW_MIN, 256.f, impl->info.rate);
impl->corr = 1.0f;
pw_log_info("resync");
impl->have_sync = true;
}
spa_ringbuffer_write_update(&impl->ring, index);
if (nread < 0) {
const bool important = !(errno == EINTR
|| errno == EAGAIN
|| errno == EWOULDBLOCK);
if (important)
pw_log_warn("failed to read from pipe (%s): %m",
impl->filename);
else
pw_log_debug("pipe (%s) underrun: %m", impl->filename);
}
pw_log_debug("filled %d %u %d", filled, index, impl->target_buffer);
return 0;
}
static void on_pipe_io(void *data, int fd, uint32_t mask)
{
struct impl *impl = data;
if (mask & (SPA_IO_ERR | SPA_IO_HUP)) {
pw_log_warn("error:%08x", mask);
pw_loop_update_io(impl->data_loop, impl->socket, 0);
return;
}
if (impl->paused)
pause_stream(impl, false);
if (mask & SPA_IO_IN)
handle_pipe_read(impl);
}
static int create_fifo(struct impl *impl)
{
struct stat st;
const char *filename;
bool do_unlink_fifo = false;
int fd = -1, res;
if ((filename = pw_properties_get(impl->props, "pipe.filename")) == NULL)
filename = impl->direction == PW_DIRECTION_INPUT ?
DEFAULT_CAPTURE_FILENAME :
DEFAULT_PLAYBACK_FILENAME;
if (mkfifo(filename, 0666) < 0) {
if (errno != EEXIST) {
res = -errno;
pw_log_error("mkfifo('%s'): %s", filename, spa_strerror(res));
goto error;
}
} else {
/*
* Our umask is 077, so the pipe won't be created with the
* requested permissions. Let's fix the permissions with chmod().
*/
if (chmod(filename, 0666) < 0)
pw_log_warn("chmod('%s'): %s", filename, spa_strerror(-errno));
do_unlink_fifo = true;
}
if ((fd = open(filename, O_RDWR | O_CLOEXEC | O_NONBLOCK, 0)) < 0) {
res = -errno;
pw_log_error("open('%s'): %s", filename, spa_strerror(res));
goto error;
}
if (fstat(fd, &st) < 0) {
res = -errno;
pw_log_error("fstat('%s'): %s", filename, spa_strerror(res));
goto error;
}
if (!S_ISFIFO(st.st_mode)) {
res = -EINVAL;
pw_log_error("'%s' is not a FIFO.", filename);
goto error;
}
impl->socket = pw_loop_add_io(impl->data_loop, fd,
0, false, on_pipe_io, impl);
if (impl->socket == NULL) {
res = -errno;
pw_log_error("can't create socket");
goto error;
}
impl->timer = pw_loop_add_timer(impl->data_loop, on_timeout, impl);
if (impl->timer == NULL) {
res = -errno;
pw_log_error("can't create timer");
goto error;
}
pw_log_info("%s fifo '%s' with format:%s channels:%d rate:%d",
impl->direction == PW_DIRECTION_OUTPUT ? "reading from" : "writing to",
filename,
spa_debug_type_find_name(spa_type_audio_format, impl->info.format),
impl->info.channels, impl->info.rate);
impl->filename = strdup(filename);
impl->unlink_fifo = do_unlink_fifo;
impl->fd = fd;
return 0;
error:
if (do_unlink_fifo)
unlink(filename);
if (fd >= 0)
close(fd);
return res;
}
static void core_error(void *data, uint32_t id, int seq, int res, const char *message)
{
struct impl *impl = data;
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 = core_error,
};
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)
{
if (impl->stream)
pw_stream_destroy(impl->stream);
if (impl->core && impl->do_disconnect)
pw_core_disconnect(impl->core);
if (impl->filename) {
if (impl->unlink_fifo)
unlink(impl->filename);
free(impl->filename);
}
if (impl->socket)
pw_loop_destroy_source(impl->data_loop, impl->socket);
if (impl->timer)
pw_loop_destroy_source(impl->data_loop, impl->timer);
if (impl->fd >= 0)
close(impl->fd);
pw_context_release_loop(impl->context, impl->data_loop);
pw_properties_free(impl->stream_props);
pw_properties_free(impl->props);
free(impl->buffer);
free(impl);
}
static void module_destroy(void *data)
{
struct impl *impl = data;
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 uint32_t channel_from_name(const char *name)
{
int i;
for (i = 0; spa_type_audio_channel[i].name; i++) {
if (spa_streq(name, spa_debug_type_short_name(spa_type_audio_channel[i].name)))
return spa_type_audio_channel[i].type;
}
return SPA_AUDIO_CHANNEL_UNKNOWN;
}
static void parse_position(struct spa_audio_info_raw *info, const char *val, size_t len)
{
struct spa_json it[1];
char v[256];
if (spa_json_begin_array_relax(&it[0], val, len) <= 0)
return;
info->channels = 0;
while (spa_json_get_string(&it[0], v, sizeof(v)) > 0 &&
info->channels < SPA_AUDIO_MAX_CHANNELS) {
info->position[info->channels++] = channel_from_name(v);
}
}
static inline uint32_t format_from_name(const char *name, size_t len)
{
int i;
for (i = 0; spa_type_audio_format[i].name; i++) {
if (strncmp(name, spa_debug_type_short_name(spa_type_audio_format[i].name), len) == 0)
return spa_type_audio_format[i].type;
}
return SPA_AUDIO_FORMAT_UNKNOWN;
}
static void parse_audio_info(const struct pw_properties *props, struct spa_audio_info_raw *info)
{
const char *str;
spa_zero(*info);
if ((str = pw_properties_get(props, PW_KEY_AUDIO_FORMAT)) == NULL)
str = DEFAULT_FORMAT;
info->format = format_from_name(str, strlen(str));
info->rate = pw_properties_get_uint32(props, PW_KEY_AUDIO_RATE, info->rate);
if (info->rate == 0)
info->rate = DEFAULT_RATE;
info->channels = pw_properties_get_uint32(props, PW_KEY_AUDIO_CHANNELS, info->channels);
info->channels = SPA_MIN(info->channels, SPA_AUDIO_MAX_CHANNELS);
if ((str = pw_properties_get(props, SPA_KEY_AUDIO_POSITION)) != NULL)
parse_position(info, str, strlen(str));
if (info->channels == 0)
parse_position(info, DEFAULT_POSITION, strlen(DEFAULT_POSITION));
}
static int calc_frame_size(const struct spa_audio_info_raw *info)
{
int res = info->channels;
switch (info->format) {
case SPA_AUDIO_FORMAT_U8:
case SPA_AUDIO_FORMAT_S8:
case SPA_AUDIO_FORMAT_ALAW:
case SPA_AUDIO_FORMAT_ULAW:
return res;
case SPA_AUDIO_FORMAT_S16:
case SPA_AUDIO_FORMAT_S16_OE:
case SPA_AUDIO_FORMAT_U16:
return res * 2;
case SPA_AUDIO_FORMAT_S24:
case SPA_AUDIO_FORMAT_S24_OE:
case SPA_AUDIO_FORMAT_U24:
return res * 3;
case SPA_AUDIO_FORMAT_S24_32:
case SPA_AUDIO_FORMAT_S24_32_OE:
case SPA_AUDIO_FORMAT_S32:
case SPA_AUDIO_FORMAT_S32_OE:
case SPA_AUDIO_FORMAT_U32:
case SPA_AUDIO_FORMAT_U32_OE:
case SPA_AUDIO_FORMAT_F32:
case SPA_AUDIO_FORMAT_F32_OE:
return res * 4;
case SPA_AUDIO_FORMAT_F64:
case SPA_AUDIO_FORMAT_F64_OE:
return res * 8;
default:
return 0;
}
}
static void copy_props(struct impl *impl, struct pw_properties *props, const char *key)
{
const char *str;
if ((str = pw_properties_get(props, key)) != NULL) {
if (pw_properties_get(impl->stream_props, key) == NULL)
pw_properties_set(impl->stream_props, key, str);
}
}
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 pw_properties *props = NULL;
struct impl *impl;
const char *str, *media_class = NULL;
int res;
PW_LOG_TOPIC_INIT(mod_topic);
impl = calloc(1, sizeof(struct impl));
if (impl == NULL)
return -errno;
impl->fd = -1;
pw_log_debug("module %p: new %s", impl, args);
if (args == NULL)
args = "";
props = pw_properties_new_string(args);
if (props == NULL) {
res = -errno;
pw_log_error( "can't create properties: %m");
goto error;
}
impl->props = props;
impl->stream_props = pw_properties_new(NULL, NULL);
if (impl->stream_props == NULL) {
res = -errno;
pw_log_error( "can't create properties: %m");
goto error;
}
impl->module = module;
impl->context = context;
impl->main_loop = pw_context_get_main_loop(context);
impl->data_loop = pw_context_acquire_loop(context, &props->dict);
if ((str = pw_properties_get(props, "tunnel.mode")) == NULL)
str = "playback";
if (spa_streq(str, "capture")) {
impl->mode = MODE_CAPTURE;
impl->direction = PW_DIRECTION_INPUT;
impl->may_pause = true;
} else if (spa_streq(str, "playback")) {
impl->mode = MODE_PLAYBACK;
impl->direction = PW_DIRECTION_OUTPUT;
impl->may_pause = true;
}else if (spa_streq(str, "sink")) {
impl->mode = MODE_SINK;
impl->direction = PW_DIRECTION_INPUT;
impl->may_pause = false;
media_class = "Audio/Sink";
} else if (spa_streq(str, "source")) {
impl->mode = MODE_SOURCE;
impl->direction = PW_DIRECTION_OUTPUT;
impl->may_pause = false;
media_class = "Audio/Source";
} else {
pw_log_error("invalid tunnel.mode '%s'", str);
res = -EINVAL;
goto error;
}
if ((str = pw_properties_get(props, "tunnel.may-pause")) != NULL)
impl->may_pause = spa_atob(str);
pw_properties_set(props, PW_KEY_NODE_LOOP_NAME, impl->data_loop->name);
if (pw_properties_get(props, PW_KEY_NODE_VIRTUAL) == NULL)
pw_properties_set(props, PW_KEY_NODE_VIRTUAL, "true");
if (pw_properties_get(props, PW_KEY_MEDIA_CLASS) == NULL)
pw_properties_set(props, PW_KEY_MEDIA_CLASS, media_class);
if ((str = pw_properties_get(props, "stream.props")) != NULL)
pw_properties_update_string(impl->stream_props, str, strlen(str));
copy_props(impl, props, PW_KEY_NODE_LOOP_NAME);
copy_props(impl, props, PW_KEY_AUDIO_FORMAT);
copy_props(impl, props, PW_KEY_AUDIO_RATE);
copy_props(impl, props, PW_KEY_AUDIO_CHANNELS);
copy_props(impl, props, SPA_KEY_AUDIO_POSITION);
copy_props(impl, props, PW_KEY_NODE_NAME);
copy_props(impl, props, PW_KEY_NODE_DESCRIPTION);
copy_props(impl, props, PW_KEY_NODE_GROUP);
copy_props(impl, props, PW_KEY_NODE_LATENCY);
copy_props(impl, props, PW_KEY_NODE_VIRTUAL);
copy_props(impl, props, PW_KEY_MEDIA_CLASS);
copy_props(impl, props, PW_KEY_TARGET_OBJECT);
copy_props(impl, props, "pipe.filename");
parse_audio_info(impl->stream_props, &impl->info);
impl->frame_size = calc_frame_size(&impl->info);
if (impl->frame_size == 0) {
pw_log_error("unsupported audio format:%d channels:%d",
impl->info.format, impl->info.channels);
res = -EINVAL;
goto error;
}
if (impl->info.rate != 0 &&
pw_properties_get(props, PW_KEY_NODE_RATE) == NULL)
pw_properties_setf(props, PW_KEY_NODE_RATE,
"1/%u", impl->info.rate);
copy_props(impl, props, PW_KEY_NODE_RATE);
impl->buffer = calloc(1, RINGBUFFER_SIZE);
if (impl->buffer == NULL) {
res = -errno;
pw_log_error("can't alloc ringbuffer: %m");
goto error;
}
spa_ringbuffer_init(&impl->ring);
impl->target_buffer = 8192 * impl->frame_size;
spa_dll_init(&impl->dll);
spa_dll_set_bw(&impl->dll, SPA_DLL_BW_MIN, 256.f, impl->info.rate);
impl->max_error = 256.0f * impl->frame_size;
impl->corr = 1.0f;
impl->core = pw_context_get_object(impl->context, PW_TYPE_INTERFACE_Core);
if (impl->core == NULL) {
str = pw_properties_get(props, PW_KEY_REMOTE_NAME);
impl->core = pw_context_connect(impl->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 error;
}
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);
if ((res = create_fifo(impl)) < 0)
goto error;
if ((res = create_stream(impl)) < 0)
goto error;
pw_impl_module_add_listener(module, &impl->module_listener, &module_events, impl);
pw_impl_module_update_properties(module, &SPA_DICT_INIT_ARRAY(module_props));
return 0;
error:
impl_destroy(impl);
return res;
}