You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
rtpengine/daemon/media_player.c

2855 lines
79 KiB

#include "media_player.h"
#include <glib.h>
#ifdef WITH_TRANSCODING
#include <mysql.h>
#include <mysql/errmsg.h>
#endif
#include "obj.h"
#include "log.h"
#include "timerthread.h"
#include "call.h"
#include "call_interfaces.h"
#include "str.h"
#include "rtplib.h"
#include "codec.h"
#include "media_socket.h"
#include "ssrc.h"
#include "log_funcs.h"
#include "main.h"
#include "rtcp.h"
#ifdef WITH_TRANSCODING
#include "fix_frame_channel_layout.h"
#endif
#include "kernel.h"
#include "bufferpool.h"
#include "uring.h"
#include "arena.h"
#define DEFAULT_AVIO_BUFSIZE 4096
typedef enum {
MPC_OK = 0,
MPC_ERR = -1,
MPC_CACHED = 1,
} mp_cached_code;
#ifdef WITH_TRANSCODING
static bool media_player_play_file(struct media_player *mp, media_player_opts_t opts);
static bool media_player_play_blob(struct media_player *mp, media_player_opts_t opts);
static bool media_player_play_db(struct media_player *mp, media_player_opts_t opts);
static bool media_player_add_file(struct media_player *mp, media_player_opts_t opts);
static bool media_player_add_blob(struct media_player *mp, media_player_opts_t opts);
static bool media_player_add_db(struct media_player *mp, media_player_opts_t opts);
#endif
#ifdef WITH_TRANSCODING
static struct timerthread media_player_thread;
static __thread MYSQL *mysql_conn;
struct media_player_cache_packet;
static void cache_packet_free(struct media_player_cache_packet *p);
TYPED_GPTRARRAY_FULL(cache_packet_arr, struct media_player_cache_packet, cache_packet_free)
static void __media_player_set_opts(struct media_player *mp, media_player_opts_t opts);
struct media_player_cache_index {
struct media_player_content_index index;
rtp_payload_type dst_pt;
};
TYPED_DIRECT_FUNCS(media_player_direct_hash, media_player_direct_eq, struct media_player)
TYPED_GHASHTABLE(media_player_ht, struct media_player, struct media_player, media_player_direct_hash,
media_player_direct_eq, NULL, NULL)
struct media_player_cache_entry {
struct obj obj;
struct media_player_cache_index index;
memory_arena_t arena;
volatile bool finished;
// "unfinished" elements, only used while decoding is active:
mutex_t lock;
cond_t cond; // to wait for more data to be decoded
cache_packet_arr *packets; // read-only except for decoder thread, which uses finished flags and locks
unsigned long duration; // cumulative in ms, summed up while decoding
unsigned int kernel_idx; // -1 if not in use
media_player_ht wait_queue; // players waiting on decoder to finish
struct codec_scheduler csch;
struct media_player_coder coder; // de/encoder data
char *info_str; // for logging
call_t *call_ref;
};
struct media_player_cache_packet {
char *buf;
str s;
long long pts;
long long duration; // us
long long duration_ts;
};
struct media_player_media_file {
struct obj obj; // must be first
str blob;
union {
str_list *str_link;
GList *gen_link;
};
int64_t mtime_us;
int64_t atime_us;
};
static mutex_t media_player_cache_lock = MUTEX_STATIC_INIT;
static unsigned int media_player_cache_entry_hash(const struct media_player_cache_index *p);
static gboolean media_player_cache_entry_eq(const struct media_player_cache_index *A,
const struct media_player_cache_index *B);
static void __media_player_cache_entry_free(struct media_player_cache_entry *p);
TYPED_GHASHTABLE(media_player_cache_ht, struct media_player_cache_index, struct media_player_cache_entry,
media_player_cache_entry_hash, media_player_cache_entry_eq,
NULL, __obj_put)
TYPED_GQUEUE(media_player_cache_entry, struct media_player_cache_entry)
static media_player_cache_ht media_player_cache; // keys and values only ever freed at shutdown
TYPED_GHASHTABLE(media_player_media_files_ht, str, struct media_player_media_file, str_hash, str_equal,
NULL, __obj_put);
static mutex_t media_player_media_files_lock = MUTEX_STATIC_INIT;
static media_player_media_files_ht media_player_media_files;
static rwlock_t media_player_media_files_names_lock = RWLOCK_STATIC_INIT;
static str_q media_player_media_files_names = TYPED_GQUEUE_INIT;
// lock order: media_player_media_files_names_lock first, media_player_media_files_lock second
TYPED_GHASHTABLE(media_player_db_media_ht, void, struct media_player_media_file, g_direct_hash, g_direct_equal,
NULL, __obj_put);
static mutex_t media_player_db_media_lock = MUTEX_STATIC_INIT;
static media_player_db_media_ht media_player_db_media;
static rwlock_t media_player_db_media_ids_lock = RWLOCK_STATIC_INIT;
static GQueue media_player_db_media_ids = G_QUEUE_INIT;
// lock order: media_player_db_media_ids_lock first, media_player_db_media_lock second
static bool media_player_read_packet(struct media_player *mp);
static mp_cached_code __media_player_add_blob_id(struct media_player *mp,
media_player_opts_t opts,
const rtp_payload_type *dst_pt);
#endif
static struct timerthread send_timer_thread;
static void send_timer_send_nolock(struct send_timer *st, struct codec_packet *cp);
static void send_timer_send_lock(struct send_timer *st, struct codec_packet *cp);
#ifdef WITH_TRANSCODING
// called with call->master lock in W
static unsigned int send_timer_flush(struct send_timer *st, void *ptr) {
if (!st)
return 0;
return timerthread_queue_flush(&st->ttq, ptr);
}
static void media_player_coder_shutdown(struct media_player_coder *c) {
avformat_close_input(&c->fmtctx);
codec_handler_free(&c->handler);
if (c->avioctx) {
if (c->avioctx->buffer)
av_freep(&c->avioctx->buffer);
av_freep(&c->avioctx);
}
c->avstream = NULL;
c->blob = STR_NULL;
c->read_pos = STR_NULL;
}
// called with call->master lock in W
static void media_player_shutdown(struct media_player *mp) {
if (!mp)
return;
//ilog(LOG_DEBUG, "shutting down media_player");
timerthread_obj_deschedule(&mp->tt_obj);
mp->next_run = 0;
if (mp->sink) {
unsigned int num = send_timer_flush(mp->sink->send_timer, mp->coder.handler);
ilog(LOG_DEBUG, "%u packets removed from send queue", num);
// roll back seq numbers already used
mp->ssrc_out->seq_diff -= num;
}
if (mp->opts.block_egress && mp->media)
MEDIA_CLEAR(mp->media, BLOCK_EGRESS);
mp->media = NULL;
media_player_coder_shutdown(&mp->coder);
if (mp->kernel_idx != -1)
kernel_stop_stream_player(mp->kernel_idx);
else if (mp->cache_entry) {
mutex_lock(&mp->cache_entry->lock);
if (t_hash_table_is_set(mp->cache_entry->wait_queue)) {
if (t_hash_table_remove(mp->cache_entry->wait_queue, mp))
obj_put(&mp->tt_obj);
}
mutex_unlock(&mp->cache_entry->lock);
}
mp->cache_index.type = MP_OTHER;
mp->cache_index.file = STR_NULL;// coverity[missing_lock : FALSE]
obj_release(mp->cache_entry); // coverity[missing_lock : FALSE]
mp->cache_read_idx = 0;
mp->kernel_idx = -1;
}
#endif
long long media_player_stop(struct media_player *mp) {
#ifdef WITH_TRANSCODING
media_player_shutdown(mp);
if (!mp)
return 0;
return mp->last_frame_ts;
#else
return 0;
#endif
}
#ifdef WITH_TRANSCODING
static void __media_player_free(struct media_player *mp) {
media_player_shutdown(mp);
ssrc_entry_release(mp->ssrc_out);
mutex_destroy(&mp->lock);
obj_put(mp->call);
av_packet_free(&mp->coder.pkt);
obj_release(mp->media_file);
obj_release(mp->cache_entry);
}
#endif
// call->master_lock held in W
void media_player_new(struct media_player **mpp, struct call_monologue *ml, struct ssrc_entry_call *prev_ssrc,
media_player_opts_t *opts)
{
#ifdef WITH_TRANSCODING
struct media_player *mp;
if (!(mp = *mpp)) {
//ilog(LOG_DEBUG, "creating media_player");
mp = *mpp = obj_alloc0(struct media_player, __media_player_free);
while (mp->ssrc == 0)
mp->ssrc = ssl_random();
mp->tt_obj.tt = &media_player_thread;
mutex_init(&mp->lock);
mp->kernel_idx = -1;
mp->run_func = media_player_read_packet; // default
mp->call = obj_get(ml->call);
mp->ml = ml;
if (prev_ssrc)
mp->seq = atomic_get_na(&prev_ssrc->stats->ext_seq) + 1;
else
mp->seq = ssl_random();
mp->buffer_ts = ssl_random();
}
/* add opts if given */
if (opts)
__media_player_set_opts(mp, *opts);
if (!mp->coder.pkt) {
mp->coder.pkt = av_packet_alloc();
mp->coder.pkt->data = NULL;
mp->coder.pkt->size = 0;
}
#else
return;
#endif
}
static void __send_timer_free(void *p) {
struct send_timer *st = p;
obj_put(st->call);
}
static void __send_timer_send_now(struct timerthread_queue *ttq, void *p) {
send_timer_send_nolock((void *) ttq, p);
};
static void __send_timer_send_later(struct timerthread_queue *ttq, void *p) {
send_timer_send_lock((void *) ttq, p);
};
// call->master_lock held in W
struct send_timer *send_timer_new(struct packet_stream *ps) {
struct send_timer *st = timerthread_queue_new("send_timer", sizeof(*st),
&send_timer_thread,
__send_timer_send_now,
__send_timer_send_later,
__send_timer_free, (GDestroyNotify) codec_packet_free);
st->call = obj_get(ps->call);
st->sink = ps;
return st;
}
// call is locked in R
static void send_timer_rtcp(struct send_timer *st, struct ssrc_entry_call *ssrc_out) {
struct call_media *media = st->sink ? st->sink->media : NULL;
if (!media)
return;
rtcp_send_report(media, ssrc_out);
ssrc_out->next_rtcp = rtpe_now;
ssrc_out->next_rtcp += 5000000 + (ssl_random() % 2000000);
}
struct async_send_req {
struct uring_req req; // must be first
struct iovec iov;
struct msghdr msg;
struct sockaddr_storage sin;
void *buf;
};
static void async_send_req_free(struct uring_req *p, int32_t res, uint32_t flags) {
struct async_send_req *req = (__typeof__(req)) p;
bufferpool_unref(req->buf);
uring_methods.free(p);
}
static bool __send_timer_send_1(struct rtp_header *rh, struct packet_stream *sink, struct codec_packet *cp) {
stream_fd *sink_fd = sink->selected_sfd;
if (!sink_fd || sink_fd->socket.fd == -1 || sink->endpoint.address.family == NULL)
return false;
log_info_stream_fd(sink->selected_sfd);
if (rh) {
ilog(LOG_DEBUG, "Forward to sink endpoint: local %s -> remote %s%s%s "
"(RTP seq %u TS %u SSRC %x)",
endpoint_print_buf(&sink_fd->socket.local),
FMT_M(endpoint_print_buf(&sink->endpoint)),
ntohs(rh->seq_num),
ntohl(rh->timestamp),
ntohl(rh->ssrc));
codec_calc_jitter(cp->ssrc_out, ntohl(rh->timestamp), cp->clockrate, rtpe_now);
}
else
ilog(LOG_DEBUG, "Forward to sink endpoint: local %s -> remote %s%s%s",
endpoint_print_buf(&sink_fd->socket.local),
FMT_M(endpoint_print_buf(&sink->endpoint)));
struct async_send_req req_s;
struct async_send_req *req = uring_alloc(&req_s, async_send_req_free);
req->iov = (__typeof(req->iov)) {
.iov_base = cp->s.s,
.iov_len = cp->s.len,
};
req->msg = (__typeof(req->msg)) {
.msg_iov = &req->iov,
.msg_iovlen = 1,
};
req->buf = bufferpool_ref(cp->s.s);
uring_methods.sendmsg(&sink_fd->socket, &req->msg, &sink->endpoint, &req->sin, &req->req);
if (sink->call->recording && rtpe_config.rec_egress) {
// fill in required members
struct media_packet mp = {
.call = sink->call,
.stream = sink,
.sfd = sink_fd,
.fsin = sink->endpoint,
};
dump_packet(&mp, cp->plain.s ? &cp->plain : &cp->s);
}
atomic64_inc_na(&sink->stats_out->packets);
atomic64_add_na(&sink->stats_out->bytes, cp->s.len);
atomic64_inc_na(&sink_fd->local_intf->stats->out.packets);
atomic64_add_na(&sink_fd->local_intf->stats->out.bytes, cp->s.len);
log_info_pop();
return true;
}
static void __send_timer_send_common(struct send_timer *st, struct codec_packet *cp) {
log_info_stream_fd(st->sink->selected_sfd);
if (!__send_timer_send_1(cp->rtp, st->sink, cp))
goto out;
if (cp->ssrc_out && cp->rtp) {
atomic64_inc_na(&cp->ssrc_out->stats->packets);
atomic64_add_na(&cp->ssrc_out->stats->bytes, cp->s.len);
if (cp->ts)
atomic_set_na(&cp->ssrc_out->stats->timestamp, cp->ts);
else
atomic_set_na(&cp->ssrc_out->stats->timestamp, ntohl(cp->rtp->timestamp));
payload_tracker_add(&cp->ssrc_out->tracker, cp->rtp->m_pt & 0x7f);
}
// do we send RTCP?
struct ssrc_entry_call *ssrc_out = cp->ssrc_out;
if (ssrc_out && ssrc_out->next_rtcp) {
mutex_lock(&ssrc_out->h.lock);
int64_t diff = ssrc_out->next_rtcp - rtpe_now;
mutex_unlock(&ssrc_out->h.lock);
if (diff < 0)
send_timer_rtcp(st, ssrc_out);
}
out:
codec_packet_free(cp);
log_info_pop();
}
static void send_timer_send_lock(struct send_timer *st, struct codec_packet *cp) {
call_t *call = st->call;
if (!call)
return;
log_info_call(call);
rwlock_lock_r(&call->master_lock);
mutex_lock(&st->sink->out_lock);
__send_timer_send_common(st, cp);
mutex_unlock(&st->sink->out_lock);
rwlock_unlock_r(&call->master_lock);
log_info_pop();
}
// st->stream->out_lock (or call->master_lock/W) must be held already
static void send_timer_send_nolock(struct send_timer *st, struct codec_packet *cp) {
call_t *call = st->call;
if (!call)
return;
log_info_call(call);
__send_timer_send_common(st, cp);
log_info_pop();
}
// st->stream->out_lock (or call->master_lock/W) must be held already
void send_timer_push(struct send_timer *st, struct codec_packet *cp) {
timerthread_queue_push(&st->ttq, &cp->ttq_entry);
}
#ifdef WITH_TRANSCODING
#if LIBAVFORMAT_VERSION_INT >= AV_VERSION_INT(57, 26, 0)
#define CODECPAR codecpar
#else
#define CODECPAR codec
#endif
typedef union {
struct media_player_cache_entry *entry;
struct media_player *mp;
} media_player_coder_add_packet_arg __attribute__((__transparent_union__));
static void media_player_coder_add_packet(struct media_player_coder *c,
void (*fn)(media_player_coder_add_packet_arg p, char *buf, size_t len,
int64_t us_dur, unsigned long long pts), media_player_coder_add_packet_arg p) {
// scale pts and duration according to sample rate
int64_t duration_scaled = c->pkt->duration * c->avstream->CODECPAR->sample_rate
* c->avstream->time_base.num / c->avstream->time_base.den;
unsigned long long pts_scaled = c->pkt->pts * c->avstream->CODECPAR->sample_rate
* c->avstream->time_base.num / c->avstream->time_base.den;
int64_t us_dur = c->pkt->duration * 1000000LL * c->avstream->time_base.num
/ c->avstream->time_base.den;
ilog(LOG_DEBUG, "read media packet: pts %llu duration %lli (scaled %llu/%" PRId64 ", %" PRId64 " us), "
"sample rate %i, time_base %i/%i",
(unsigned long long) c->pkt->pts,
(long long) c->pkt->duration,
pts_scaled,
duration_scaled,
us_dur,
c->avstream->CODECPAR->sample_rate,
c->avstream->time_base.num, c->avstream->time_base.den);
fn(p, (char *) c->pkt->data, c->pkt->size, us_dur, pts_scaled);
}
static bool media_player_read_decoded_packet(struct media_player *mp) {
struct media_player_cache_entry *entry = mp->cache_entry;
if (!entry)
return false;
unsigned int read_idx = mp->cache_read_idx;
ilog(LOG_DEBUG, "Buffered media player reading packet #%u", read_idx);
retry:;
bool finished = entry->finished; // hold lock or not
if (!finished) {
// slow track with locking
mutex_lock(&entry->lock);
// confirm that we are indeed not finished
if (entry->finished) {
// preempted, good to go
mutex_unlock(&entry->lock);
finished = true;
}
}
if (read_idx >= entry->packets->len) {
if (!finished) {
// wait for more
cond_wait(&entry->cond, &entry->lock);
mutex_unlock(&entry->lock);
goto retry;
}
// EOF
if (mp->opts.repeat <= 1) {
ilog(LOG_DEBUG, "EOF reading from media buffer (%s), stopping playback",
entry->info_str);
return true;
}
ilog(LOG_DEBUG, "EOF reading from media buffer (%s) but will repeat %i time",
entry->info_str, mp->opts.repeat);
mp->opts.repeat--;
read_idx = mp->cache_read_idx = 0;
goto retry;
}
// got a packet
struct media_player_cache_packet *pkt = entry->packets->pdata[read_idx];
int64_t us_dur = pkt->duration;
mp->cache_read_idx++;
if (!finished)
mutex_unlock(&entry->lock);
// make a copy to send out
size_t len = pkt->s.len + sizeof(struct rtp_header) + RTP_BUFFER_TAIL_ROOM;
char *buf = bufferpool_alloc(media_bufferpool, len);
memcpy(buf, pkt->buf, len);
struct media_packet packet = {
.tv = rtpe_now,
.call = mp->call,
.media = mp->media,
.media_out = mp->media,
.rtp = (void *) buf,
.ssrc_out = mp->ssrc_out,
};
mp->last_frame_ts = pkt->pts;
codec_output_rtp(&packet, &entry->csch, mp->coder.handler, buf, pkt->s.len, mp->buffer_ts,
read_idx == 0, mp->seq++, 0, -1, 0);
mp->buffer_ts += pkt->duration_ts;
mp->sync_ts_tv = rtpe_now;
media_packet_encrypt(mp->crypt_handler->out->rtp_crypt, mp->sink, &packet);
mutex_lock(&mp->sink->out_lock);
if (media_socket_dequeue(&packet, mp->sink))
ilog(LOG_ERR, "Error sending playback media to RTP sink");
mutex_unlock(&mp->sink->out_lock);
// schedule our next run
mp->next_run += us_dur;
timerthread_obj_schedule_abs(&mp->tt_obj, mp->next_run);
return false;
}
static void media_player_kernel_player_start_now(struct media_player *mp) {
struct media_player_cache_entry *entry = mp->cache_entry;
if (!entry)
return;
const rtp_payload_type *dst_pt = &entry->coder.handler->dest_pt;
ilog(LOG_DEBUG, "Starting kernel media player index %i (PT %i)", entry->kernel_idx, dst_pt->payload_type);
struct rtpengine_play_stream_info info = {
.packet_stream_idx = entry->kernel_idx,
.pt = dst_pt->payload_type,
.seq = mp->seq,
.ts = mp->buffer_ts,
.ssrc = mp->ssrc_out->h.ssrc,
.repeat = mp->opts.repeat,
.stats = mp->sink->stats_out,
.iface_stats = mp->sink->selected_sfd->local_intf->stats,
.ssrc_stats = mp->ssrc_out->stats,
};
mp->sink->endpoint.address.family->endpoint2kernel(&info.dst_addr, &mp->sink->endpoint); // XXX unify with __re_address_translate_ep
mp->sink->selected_sfd->socket.local.address.family->endpoint2kernel(&info.src_addr, &mp->sink->selected_sfd->socket.local); // XXX unify with __re_address_translate_ep
mp->crypt_handler->out->kernel(&info.encrypt, mp->sink);
unsigned int idx = kernel_start_stream_player(&info);
if (idx == -1)
ilog(LOG_ERR, "Failed to start kernel media player (index %i): %s", info.packet_stream_idx, strerror(errno));
else
mp->kernel_idx = idx;
}
static void media_player_kernel_player_start(struct media_player *mp) {
struct media_player_cache_entry *entry = mp->cache_entry;
if (!entry)
return;
// decoder finished yet? try unlocked read first
bool finished = entry->finished;
if (!finished) {
mutex_lock(&entry->lock);
// check flag again in case it happened just now
if (!entry->finished) {
// add us to wait list
ilog(LOG_DEBUG, "Decoder not finished yet, waiting to start kernel player index %i",
entry->kernel_idx);
if (t_hash_table_insert(entry->wait_queue, mp, mp))
obj_hold(&mp->tt_obj);
mutex_unlock(&entry->lock);
return;
}
// finished now, drop down below
mutex_unlock(&entry->lock);
}
media_player_kernel_player_start_now(mp);
}
static void media_player_cached_reader_start(struct media_player *mp, str_case_value_ht codec_set) {
struct media_player_cache_entry *entry = mp->cache_entry;
const rtp_payload_type *dst_pt = &entry->coder.handler->dest_pt;
if (entry->kernel_idx != -1) {
media_player_kernel_player_start(mp);
return;
}
// create dummy codec handler and start timer
mp->coder.handler = codec_handler_make_dummy(&entry->coder.handler->dest_pt, mp->media, codec_set);
mp->run_func = media_player_read_decoded_packet;
mp->next_run = rtpe_now;
mp->coder.duration = entry->coder.duration;
// if we played anything before, scale our sync TS according to the time
// that has passed
if (mp->sync_ts_tv) {
int64_t ts_diff_us = rtpe_now - mp->sync_ts_tv;
mp->buffer_ts += fraction_divl(ts_diff_us * dst_pt->clock_rate / 1000000, &dst_pt->codec_def->default_clockrate_fact);
}
mp->sync_ts_tv = rtpe_now;
media_player_read_decoded_packet(mp);
}
static void cache_packet_free(struct media_player_cache_packet *p) {
RTPE_GAUGE_ADD(player_cache, -1 * (ssize_t) p->s.len);
bufferpool_unref(p->buf);
g_free(p);
}
// returns: true = entry exists, decoding handled separately, use entry for playback
// false = no entry exists, OR entry is a new one, proceed to open decoder, then call _play_start
static bool media_player_cache_get_entry(struct media_player *mp,
const rtp_payload_type *dst_pt, str_case_value_ht codec_set)
{
if (!rtpe_config.player_cache)
return false;
if (mp->cache_index.type <= 0)
return false;
if (!dst_pt)
return false;
obj_release(mp->cache_entry);
struct media_player_cache_index lookup;
lookup.index = mp->cache_index;
lookup.dst_pt = *dst_pt;
LOCK(&media_player_cache_lock);
struct media_player_cache_entry *entry = mp->cache_entry
= t_hash_table_lookup(media_player_cache, &lookup);
if (entry) {
obj_hold(entry); // ref in mp->cache_entry
media_player_cached_reader_start(mp, codec_set);
return true;
}
// initialise object
call_memory_arena_release();
entry = mp->cache_entry = obj_alloc0(struct media_player_cache_entry, __media_player_cache_entry_free);
memory_arena_init(&entry->arena);
memory_arena = &entry->arena;
struct media_player_cache_index *ins_key = &entry->index;
*ins_key = lookup;
ins_key->index.file = str_dup_str(&lookup.index.file);
codec_init_payload_type(&ins_key->dst_pt, MT_UNKNOWN); // duplicate contents into memory arena
mutex_init(&entry->lock);
cond_init(&entry->cond);
entry->packets = cache_packet_arr_new_sized(64);
entry->wait_queue = media_player_ht_new();
switch (lookup.index.type) {
case MP_DB:
entry->info_str = g_strdup_printf("DB media file #%llu", lookup.index.db_id);
break;
case MP_FILE:
entry->info_str = g_strdup_printf("media file '" STR_FORMAT "'",
STR_FMT(&lookup.index.file));
break;
case MP_BLOB:
entry->info_str = g_strdup_printf("binary media blob");
break;
default:;
}
t_hash_table_insert(media_player_cache, ins_key, obj_get(entry));
entry->kernel_idx = -1;
if (kernel.use_player) {
entry->kernel_idx = kernel_get_packet_stream();
if (entry->kernel_idx == -1)
ilog(LOG_ERR, "Failed to get kernel packet stream entry (%s)", strerror(errno));
else
ilog(LOG_DEBUG, "Using kernel packet stream index %i", entry->kernel_idx);
}
call_memory_arena_set(mp->call);
return false;
}
static void media_player_cache_packet(struct media_player_cache_entry *entry, char *buf, size_t len,
int64_t us_dur, unsigned long long pts)
{
// synthesise fake RTP header and media_packet context
struct rtp_header rtp = {
.timestamp = pts, // taken verbatim by handler_func_playback w/o byte swap
};
struct media_packet packet = {
.rtp = &rtp,
.cache_entry = entry,
};
packet.raw = STR_LEN(buf, len);
packet.payload = packet.raw;
entry->coder.handler->handler_func(entry->coder.handler, &packet);
}
static void media_player_cache_entry_decoder_thread(void *p) {
struct media_player_cache_entry *entry = p;
ilog(LOG_DEBUG, "Launching media decoder thread for %s", entry->info_str);
while (true) {
// let us be cancelled
thread_cancel_enable();
pthread_testcancel();
thread_cancel_disable();
int ret = av_read_frame(entry->coder.fmtctx, entry->coder.pkt);
if (ret < 0) {
if (ret != AVERROR_EOF)
ilog(LOG_ERR, "Error while reading from media stream");
break;
}
media_player_coder_add_packet(&entry->coder, media_player_cache_packet, entry);
av_packet_unref(entry->coder.pkt);
}
ilog(LOG_DEBUG, "Decoder thread for %s finished", entry->info_str);
mutex_lock(&entry->lock);
entry->finished = true;
cond_broadcast(&entry->cond);
media_player_ht_iter iter;
t_hash_table_iter_init(&iter, entry->wait_queue);
struct media_player *mp;
while (t_hash_table_iter_next(&iter, &mp, NULL)) {
if (mp->media)
media_player_kernel_player_start_now(mp);
obj_put(&mp->tt_obj);
}
t_hash_table_destroy_ptr(&entry->wait_queue); // not needed any more
obj_release(entry->call_ref);
mutex_unlock(&entry->lock);
obj_release(entry);
}
static void packet_encoded_cache(AVPacket *pkt, struct codec_ssrc_handler *ch, struct media_packet *mp,
str *s, char *buf, unsigned int pkt_len, const struct fraction *cr_fact)
{
struct media_player_cache_entry *entry = mp->cache_entry;
struct media_player_cache_packet *ep = g_new0(__typeof(*ep), 1);
long duration = fraction_divl(pkt->duration, cr_fact);
*ep = (__typeof__(*ep)) {
.buf = buf,
.s = *s,
.pts = pkt->pts,
.duration_ts = duration,
.duration = (long long) duration * 1000000LL
/ entry->coder.handler->dest_pt.clock_rate,
};
mutex_lock(&entry->lock);
t_ptr_array_add(entry->packets, ep);
if (entry->kernel_idx != -1) {
ilog(LOG_DEBUG, "Adding media packet (length %zu, TS %" PRIu64 ", delay %lu ms) to kernel packet stream %i",
s->len, pkt->pts, entry->duration, entry->kernel_idx);
if (!kernel_add_stream_packet(entry->kernel_idx, s->s, s->len, entry->duration, pkt->pts,
pkt->duration))
ilog(LOG_ERR | LOG_FLAG_LIMIT, "Failed to add packet to kernel player (%s)", strerror(errno));
}
entry->duration += ep->duration / 1000;
cond_broadcast(&entry->cond);
mutex_unlock(&entry->lock);
RTPE_GAUGE_ADD(player_cache, s->len);
}
static int media_player_packet_cache(encoder_t *enc, void *u1, void *u2) {
struct codec_ssrc_handler *ch = u1;
struct media_packet *mp = u2;
packet_encoded_packetize(enc->avpkt, ch, mp, enc->def->packetizer, enc, &enc->clockrate_fact,
packet_encoded_cache);
return 0;
}
// called from media_player_play_start, which is called after media_player_cache_get_entry returned true.
// this can mean that either we don't have a cache entry and should continue normally, or if we
// do have a cache entry, initialise it, set up the thread, take over decoding, and then proceed as a
// media player consuming the data from the decoder thread.
// returns: false = continue normally decode in-thread, true = take data from other thread
static bool media_player_cache_entry_init(struct media_player *mp, const rtp_payload_type *dst_pt,
str_case_value_ht codec_set)
{
struct media_player_cache_entry *entry = mp->cache_entry;
if (!entry)
return false;
// steal coder data
entry->coder = mp->coder;
ZERO(mp->coder);
mp->coder.duration = entry->coder.duration; // retain this for reporting
entry->coder.avioctx->opaque = &entry->coder; // format context pointer must point to new coder
entry->call_ref = obj_get(mp->call); // hold reference until decoding is finished, as blob is owned by call
entry->coder.handler->packet_encoded = media_player_packet_cache;
// use low priority (10 nice)
thread_create_detach_prio(media_player_cache_entry_decoder_thread, obj_hold(entry), NULL, 10, "mp decoder");
media_player_cached_reader_start(mp, codec_set);
return true;
}
// find suitable output payload type
static rtp_payload_type *media_player_get_dst_pt(struct media_player *mp) {
rtp_payload_type *dst_pt = NULL;
for (__auto_type l = mp->media->codecs.codec_prefs.head; l; l = l->next) {
dst_pt = l->data;
ensure_codec_def(dst_pt, mp->media);
if (codec_def_supported(dst_pt->codec_def) && !dst_pt->codec_def->supplemental)
goto found;
dst_pt = NULL;
}
if (!dst_pt) {
ilog(LOG_ERR, "No supported output codec found in SDP");
return NULL;
}
found:
ilog(LOG_DEBUG, "Output codec for media playback is " STR_FORMAT,
STR_FMT(&dst_pt->encoding_with_params));
return dst_pt;
}
bool media_player_pt_match(const struct media_player *mp, const rtp_payload_type *src_pt,
const rtp_payload_type *dst_pt)
{
if (!mp->coder.handler)
return true; // not initialised yet -> doesn't need a reset
if (!rtp_payload_type_eq_exact(&mp->coder.handler->dest_pt, dst_pt))
return false;
if (!rtp_payload_type_eq_exact(&mp->coder.handler->source_pt, src_pt))
return false;
return true;
}
static int media_player_setup_common(struct media_player *mp, const rtp_payload_type *src_pt,
const rtp_payload_type **dst_pt)
{
if (!*dst_pt)
*dst_pt = media_player_get_dst_pt(mp);
if (!*dst_pt)
return -1;
// if we played anything before, scale our sync TS according to the time
// that has passed
if (mp->sync_ts_tv) {
int64_t ts_diff_us = rtpe_now - mp->sync_ts_tv;
mp->sync_ts += fraction_divl(ts_diff_us * (*dst_pt)->clock_rate / 1000000, &(*dst_pt)->codec_def->default_clockrate_fact);
}
// if we already have a handler, see if anything needs changing
if (!media_player_pt_match(mp, src_pt, *dst_pt)) {
ilog(LOG_DEBUG, "Resetting codec handler for media player");
codec_handler_free(&mp->coder.handler);
}
return 0;
}
// used for generic playback (audio_player, t38_gateway)
int media_player_setup(struct media_player *mp, const rtp_payload_type *src_pt,
const rtp_payload_type *dst_pt, str_case_value_ht codec_set)
{
int ret = media_player_setup_common(mp, src_pt, &dst_pt);
if (ret)
return ret;
if (!mp->coder.handler)
mp->coder.handler = codec_handler_make_playback(src_pt, dst_pt, mp->sync_ts, mp->media,
mp->ssrc, codec_set);
if (!mp->coder.handler)
return -1;
return 0;
}
// used for "play media" player
static int __media_player_setup_internal(struct media_player *mp, const rtp_payload_type *src_pt,
const rtp_payload_type *dst_pt, str_case_value_ht codec_set)
{
int ret = media_player_setup_common(mp, src_pt, &dst_pt);
if (ret)
return ret;
if (!mp->coder.handler)
mp->coder.handler = codec_handler_make_media_player(src_pt, dst_pt, mp->sync_ts, mp->media,
mp->ssrc, codec_set);
if (!mp->coder.handler)
return -1;
return 0;
}
static int __ensure_codec_handler(struct media_player *mp, const rtp_payload_type *dst_pt,
str_case_value_ht codec_set)
{
if (mp->coder.handler)
return 0;
// synthesise rtp payload type
rtp_payload_type src_pt = { .payload_type = -1 };
src_pt.codec_def = codec_find_by_av(mp->coder.avstream->CODECPAR->codec_id);
if (!src_pt.codec_def || !src_pt.codec_def->support_decoding) {
ilog(LOG_ERR, "Attempting to play media from an unsupported file format/codec");
return -1;
}
src_pt.encoding = src_pt.codec_def->rtpname_str;
src_pt.channels = GET_CHANNELS(mp->coder.avstream->CODECPAR);
src_pt.clock_rate = mp->coder.avstream->CODECPAR->sample_rate;
codec_init_payload_type(&src_pt, MT_AUDIO);
if (__media_player_setup_internal(mp, &src_pt, dst_pt, codec_set))
return -1;
mp->coder.duration = mp->coder.avstream->duration * 1000 * mp->coder.avstream->time_base.num
/ mp->coder.avstream->time_base.den;
payload_type_clear(&src_pt);
return 0;
}
// appropriate lock must be held
void media_player_add_packet(struct media_player *mp, char *buf, size_t len,
int64_t us_dur, unsigned long long pts)
{
// synthesise fake RTP header and media_packet context
struct rtp_header rtp = {
.timestamp = pts, // taken verbatim by handler_func_playback w/o byte swap
.seq_num = htons(mp->seq),
};
struct media_packet packet = {
.tv = rtpe_now,
.call = mp->call,
.media = mp->media,
.media_out = mp->media,
.rtp = &rtp,
.ssrc_out = mp->ssrc_out,
};
packet.raw = STR_LEN(buf, len);
packet.payload = packet.raw;
mp->coder.handler->handler_func(mp->coder.handler, &packet);
// as this is timing sensitive and we may have spent some time decoding,
// update our global "now" timestamp
rtpe_now = now_us();
// keep track of RTP timestamps and real clock. look at the last packet we received
// and update our sync TS.
if (packet.packets_out.head) {
struct codec_packet *p = packet.packets_out.head->data;
if (p->rtp) {
mp->sync_ts = ntohl(p->rtp->timestamp);
mp->sync_ts_tv = p->ttq_entry.when;
}
}
media_packet_encrypt(mp->crypt_handler->out->rtp_crypt, mp->sink, &packet);
mutex_lock(&mp->sink->out_lock);
if (media_socket_dequeue(&packet, mp->sink))
ilog(LOG_ERR, "Error sending playback media to RTP sink");
mutex_unlock(&mp->sink->out_lock);
mp->next_run += us_dur;
timerthread_obj_schedule_abs(&mp->tt_obj, mp->next_run);
}
static int media_player_find_file_begin(struct media_player *mp) {
int ret = 0;
int64_t ret64 = avio_seek(mp->coder.fmtctx->pb, 0, SEEK_SET);
if (ret64 != 0)
ilog(LOG_ERR, "Failed to seek to beginning of media file");
ret = av_seek_frame(mp->coder.fmtctx, -1, 0, 0);
if (ret < 0)
ilog(LOG_ERR, "Failed to seek to beginning of media file");
ret = av_read_frame(mp->coder.fmtctx, mp->coder.pkt);
return ret;
}
// appropriate lock must be held
static bool media_player_read_packet(struct media_player *mp) {
if (!mp->coder.fmtctx)
return true;
int ret = av_read_frame(mp->coder.fmtctx, mp->coder.pkt);
if (ret < 0) {
if (ret == AVERROR_EOF) {
/* Duration counter cannot underflow and is always aligned to 0 when used.
* By default is -1.
* If either a duration or repeats counter are done, then the reading process
* is considered EOF.
*/
if (mp->opts.duration_spent == 0 ||
mp->opts.repeat <= 1)
{
ilog(LOG_DEBUG, "EOF reading from media stream");
return true;
}
ret = media_player_find_file_begin(mp);
/* counter for the max spent duration (in milliseconds)
* duration takes precedence over repeats, if used together
*/
if (mp->opts.duration_spent > 0) {
ilog(LOG_DEBUG, "EOF reading from stream but will be played further due to available duration '%lld'",
mp->opts.duration_spent);
mp->opts.duration_spent = mp->opts.duration_spent - mp->coder.duration;
/* don't let the duration counter to underflow */
if (mp->opts.duration_spent < 0)
mp->opts.duration_spent = 0;
}
/* counter for the max repeats
* still count down each time, even if we are based on max duration in milliseconds */
if (mp->opts.repeat > 1) {
ilog(LOG_DEBUG, "EOF reading from stream but will be played further due to available repeats '%d'",
mp->opts.repeat);
mp->opts.repeat--;
}
}
if (ret < 0 && ret != AVERROR_EOF) {
ilog(LOG_ERR, "Error while reading from media stream");
return true;
}
}
mp->last_frame_ts = mp->coder.pkt->pts;
media_player_coder_add_packet(&mp->coder, media_player_add_packet, mp);
av_packet_unref(mp->coder.pkt);
return false;
}
// call->master_lock held in W
void media_player_set_media(struct media_player *mp, struct call_media *media) {
mp->media = media;
if (media->streams.head) {
mp->sink = media->streams.head->data;
mp->crypt_handler = determine_handler(&transport_protocols[PROTO_RTP_AVP], media, true);
}
if (!mp->ssrc_out || mp->ssrc_out->h.ssrc != mp->ssrc) {
struct ssrc_entry_call *ssrc_ctx = get_ssrc(mp->ssrc, &media->ssrc_hash_out);
ssrc_ctx->next_rtcp = rtpe_now;
mp->ssrc_out = ssrc_ctx;
}
}
// call->master_lock held in W
// returns destination payload type, or NULL on failure
static const rtp_payload_type *media_player_play_setup(struct media_player *mp) {
// find call media suitable for playback
struct call_media *media;
for (unsigned int i = 0; i < mp->ml->medias->len; i++) {
media = mp->ml->medias->pdata[i];
if (media->type_id != MT_AUDIO)
continue;
/* moh allows inactive */
if (!mp->opts.moh && !MEDIA_ISSET(media, SEND))
continue;
if (media->streams.length == 0)
continue;
goto found;
}
media = NULL;
found:
if (!media) {
ilog(LOG_ERR, "No suitable SDP section for media playback");
return NULL;
}
media_player_set_media(mp, media);
return media_player_get_dst_pt(mp);
}
// call->master_lock held in W
// returns destination payload type, or NULL on failure
static const rtp_payload_type *media_player_play_init(struct media_player *mp) {
media_player_shutdown(mp);
return media_player_play_setup(mp);
}
// call->master_lock held in W
static bool media_player_play_start(struct media_player *mp, const rtp_payload_type *dst_pt,
str_case_value_ht codec_set)
{
// needed to have usable duration for some formats. ignore errors.
if (!mp->coder.fmtctx->streams || !mp->coder.fmtctx->streams[0])
avformat_find_stream_info(mp->coder.fmtctx, NULL);
mp->coder.avstream = mp->coder.fmtctx->streams[0];
if (!mp->coder.avstream) {
ilog(LOG_ERR, "No AVStream present in format context");
return false;
}
if (__ensure_codec_handler(mp, dst_pt, codec_set))
return false;
if (mp->opts.block_egress)
MEDIA_SET(mp->media, BLOCK_EGRESS);
if (media_player_cache_entry_init(mp, dst_pt, codec_set))
return true;
mp->next_run = rtpe_now;
// give ourselves a bit of a head start with decoding
mp->next_run -= 50000;
// if start_pos is positive, try to seek to that position
if (mp->opts.start_pos > 0) {
ilog(LOG_DEBUG, "Seeking to position %lli", mp->opts.start_pos);
av_seek_frame(mp->coder.fmtctx, 0, mp->opts.start_pos, 0);
}
else // in case this is a repeated start
av_seek_frame(mp->coder.fmtctx, 0, 0, 0);
media_player_read_packet(mp);
return true;
}
static void media_player_media_file_free(struct media_player_media_file *fo) {
RTPE_GAUGE_ADD(media_cache, -1 * (ssize_t) fo->blob.len);
g_free(fo->blob.s);
}
static str media_player_read_file(const char *f) {
gchar *buf = NULL;
gsize len = -1;
GError *err = NULL;
gboolean ret = g_file_get_contents(f, &buf, &len, &err);
if (!ret) {
ilog(LOG_ERR, "Failed to read media file '%s' for caching: %s", f, err->message);
g_error_free(err);
return STR_NULL;
}
if (len <= 0) {
ilog(LOG_ERR, "Media file '%s' appears to be empty", f);
g_free(buf);
return STR_NULL;
}
return STR_LEN(buf, len);
}
static struct media_player_media_file *media_player_media_file_new(str blob) {
__auto_type fo = obj_alloc0(struct media_player_media_file,
media_player_media_file_free);
fo->blob = blob;
fo->blob.dup = call_ref; // string is allocated by reference on `fo`
RTPE_GAUGE_ADD(media_cache, blob.len);
fo->atime_us = fo->mtime_us = rtpe_now;
return fo;
}
static struct media_player_media_file *media_player_media_file_read_c(const char *fn) {
str blob = media_player_read_file(fn);
if (blob.len == 0)
return NULL;
return media_player_media_file_new(blob);
}
static struct media_player_media_file *media_player_media_file_read_str(const str *fn) {
char file_s[PATH_MAX];
snprintf(file_s, sizeof(file_s), STR_FORMAT, STR_FMT(fn));
return media_player_media_file_read_c(file_s);
}
static const char *media_player_get_db_id(str *out, unsigned long long id,
str (*dup_fn)(const char *, size_t),
bool (*cache_fn)(const char *s, size_t len, unsigned long long id));
static bool media_player_add_cache_file_dummy(const char *s, size_t len, unsigned long long id) {
return true;
}
static bool (*media_player_add_cache_file)(const char *s, size_t len, unsigned long long id) =
media_player_add_cache_file_dummy;
static struct media_player_media_file *media_player_db_id_read(unsigned long long id) {
str blob;
const char *err = media_player_get_db_id(&blob, id, str_dup_len, media_player_add_cache_file);
if (err || blob.len == 0)
return NULL;
return media_player_media_file_new(blob);
}
static struct media_player_media_file *media_player_media_files_get_only(const str *fn) {
struct media_player_media_file *fo;
{
LOCK(&media_player_media_files_lock);
if (!t_hash_table_is_set(media_player_media_files))
return NULL;
fo = t_hash_table_lookup(media_player_media_files, fn);
if (!fo)
return NULL;
obj_hold(fo);
fo->atime_us = rtpe_now;
}
return fo;
}
// lock must be held, reference will be taken over
static struct media_player_media_file *media_player_db_id_get_only(unsigned long long id) {
struct media_player_media_file *fo;
{
LOCK(&media_player_db_media_lock);
if (!t_hash_table_is_set(media_player_db_media))
return NULL;
fo = t_hash_table_lookup(media_player_db_media, GUINT_TO_POINTER(id));
if (!fo)
return NULL;
obj_hold(fo);
fo->atime_us = rtpe_now;
}
return fo;
}
// locks must be held, reference will be taken over
static void media_player_media_files_insert(const str *fn, struct media_player_media_file *fo) {
if (!t_hash_table_is_set(media_player_media_files))
media_player_media_files = media_player_media_files_ht_new();
str *dup = str_dup(fn);
t_hash_table_insert(media_player_media_files, dup, fo);
t_queue_push_tail(&media_player_media_files_names, dup);
fo->str_link = media_player_media_files_names.tail;
}
static mp_cached_code media_player_set_media_file(struct media_player *mp,
media_player_opts_t opts,
const rtp_payload_type *dst_pt,
struct media_player_media_file *fo)
{
// release old reference if any and take over this new one
if (mp->media_file)
obj_put(mp->media_file);
mp->media_file = fo;
// switch to blob playing
opts.file = STR_NULL;
opts.blob = fo->blob;
// db_id remains set if it was, so that the cache lookup can succeed
return __media_player_add_blob_id(mp, opts, dst_pt);
}
// locks must be held, reference will be taken over
static void media_player_db_id_insert(unsigned long long id, struct media_player_media_file *fo) {
if (!t_hash_table_is_set(media_player_db_media))
media_player_db_media = media_player_db_media_ht_new();
t_hash_table_insert(media_player_db_media, GUINT_TO_POINTER(id), fo);
g_queue_push_tail(&media_player_db_media_ids, GUINT_TO_POINTER(id));
fo->gen_link = media_player_db_media_ids.tail;
}
static struct media_player_media_file *media_player_media_files_get_create(const str *fn) {
__auto_type fo = media_player_media_files_get_only(fn);
if (fo)
return fo;
fo = media_player_media_file_read_str(fn);
if (!fo)
return NULL;
RWLOCK_W(&media_player_media_files_names_lock);
LOCK(&media_player_media_files_lock);
// someone else may have beaten us to it
if (t_hash_table_is_set(media_player_media_files) && t_hash_table_lookup(media_player_media_files, fn))
return fo; // return the only reference, will disappear once player finishes
// insert new reference
media_player_media_files_insert(fn, obj_get(fo));
return fo;
}
static struct media_player_media_file *media_player_db_id_get_create(unsigned long long id) {
__auto_type fo = media_player_db_id_get_only(id);
if (fo)
return fo;
fo = media_player_db_id_read(id);
if (!fo)
return NULL;
RWLOCK_W(&media_player_db_media_ids_lock);
LOCK(&media_player_db_media_lock);
// someone else may have beaten us to it
if (t_hash_table_is_set(media_player_db_media) && t_hash_table_lookup(media_player_db_media, GUINT_TO_POINTER(id)))
return fo; // return the only reference, will disappear once player finishes
// insert new reference
media_player_db_id_insert(id, obj_get(fo));
return fo;
}
static struct media_player_media_file *(*media_player_media_files_get)(const str *fn)
= media_player_media_files_get_only;
static struct media_player_media_file *(*media_player_db_id_get)(unsigned long long)
= media_player_db_id_get_only;
static void __media_player_set_opts(struct media_player *mp, media_player_opts_t opts) {
mp->opts = opts;
if (mp->media && mp->opts.block_egress)
MEDIA_SET(mp->media, BLOCK_EGRESS);
}
// call->master_lock held in W
static mp_cached_code __media_player_add_file(struct media_player *mp,
media_player_opts_t opts,
const rtp_payload_type *dst_pt)
{
mp->cache_index.type = MP_FILE;
mp->cache_index.file = call_str_cpy(&opts.file);
if (media_player_cache_get_entry(mp, dst_pt, opts.codec_set))
return MPC_CACHED;
// check if we have it in memory
struct media_player_media_file *fo = media_player_media_files_get(&opts.file);
if (fo) {
ilog(LOG_DEBUG, "Using cached media file for playback");
return media_player_set_media_file(mp, opts, dst_pt, fo);
}
char file_s[PATH_MAX];
snprintf(file_s, sizeof(file_s), STR_FORMAT, STR_FMT(&opts.file));
int ret = avformat_open_input(&mp->coder.fmtctx, file_s, NULL, NULL);
if (ret < 0) {
ilog(LOG_ERR, "Failed to open media file for playback: %s", av_error(ret));
return MPC_ERR;
}
return MPC_OK;
}
// call->master_lock held in W
static bool media_player_play_file(struct media_player *mp, media_player_opts_t opts) {
const rtp_payload_type *dst_pt = media_player_play_init(mp);
if (!dst_pt)
return false;
mp_cached_code ret = __media_player_add_file(mp, opts, dst_pt);
if (ret == MPC_CACHED)
return true;
if (ret == MPC_ERR)
return false;
return media_player_play_start(mp, dst_pt, opts.codec_set);
}
#endif
bool media_player_add(struct media_player *mp, media_player_opts_t opts) {
#ifdef WITH_TRANSCODING
if (opts.file.len)
return media_player_add_file(mp, opts);
else if (opts.blob.len)
return media_player_add_blob(mp, opts);
else if (opts.db_id > 0)
return media_player_add_db(mp, opts);
else
return false;
#else
return false;
#endif
}
#ifdef WITH_TRANSCODING
// call->master_lock held in W
static bool media_player_add_file(struct media_player *mp, media_player_opts_t opts) {
int ret = __media_player_add_file(mp, opts, NULL);
return ret == 0;
}
#endif
/**
* When to_ml is given, check MoH capabilities of it,
* because then an offerer wants to use them instead of its own (from_ml).
*/
bool call_ml_wants_moh(struct call_monologue *from_ml, struct call_monologue *to_ml,
enum ng_opmode opmode)
{
struct call_monologue *ml = to_ml ? : from_ml;
if (opmode == OP_OFFER && call_ml_sendonly_inactive(from_ml) &&
(ml->moh_db_id > 0 || ml->moh_file.len || ml->moh_blob.len))
{
return true;
}
return false;
}
bool call_ml_stops_moh(struct call_monologue *from_ml, struct call_monologue *to_ml,
enum ng_opmode opmode)
{
#ifdef WITH_TRANSCODING
if (opmode == OP_OFFER && !call_ml_sendonly_inactive(from_ml) && (to_ml->player && to_ml->player->opts.moh))
{
return true;
}
#endif
return false;
}
/**
* MOH_ZEROCONN: check if originator wants to advertise zero address during moh.
* Sets zero-connection for the first found subscription media with a sendonly state
* and audio type.
* MOH_SENDRECV: check if originator wants to use sendrecv state instead of the default
* one (sendonly) during moh.
* Sets the sendrecv state for the first found subscription media with a sendonly state
* and audio type.
*/
void call_ml_moh_handle_flags(struct call_monologue *from_ml, struct call_monologue *to_ml) {
#ifdef WITH_TRANSCODING
/* if from_ml not given, then it's a reflected MoH, use capabilities of to_ml */
struct call_monologue *moh_ml = from_ml ? : to_ml;
if (!to_ml->player ||
!ML_ISSET2(moh_ml, MOH_ZEROCONN, MOH_SENDRECV))
{
return;
}
struct call_media * media = to_ml->player->media;
if (media) {
/* check zero-connection */
if (ML_ISSET(moh_ml, MOH_ZEROCONN)) {
struct packet_stream *ps;
__auto_type msl = media->streams.head;
while (msl)
{
ps = msl->data;
if (PS_ISSET(ps, RTP)) { /* find RTP stream, and don't touch RTCP */
ilog(LOG_DEBUG, "Forced packet stream of '"STR_FORMAT"' (media index: '%d')"
"to zero_addr due to MoH zero-connection.",
STR_FMT(&media->monologue->tag), media->index);
PS_SET(ps, ZERO_ADDR);
goto check_next; /* stop */
}
msl = msl->next;
}
}
check_next:
/* check mode sendrecv */
if (ML_ISSET(moh_ml, MOH_SENDRECV)) {
bf_set(&media->media_flags, MEDIA_FLAG_SEND | MEDIA_FLAG_RECV);
bf_set(&media->media_flags, MEDIA_FLAG_FAKE_SENDRECV);
if (media->media_subscriptions.head) {
__auto_type sub = media->media_subscriptions.head->data;
__auto_type sub_m = sub->media;
/* mark real state of originators media (no flag set - inactive, real_sendonly - sendonly) */
if (MEDIA_ISSET(sub_m, RECV))
bf_set(&media->media_flags, MEDIA_FLAG_REAL_SENDONLY);
}
}
}
#endif
}
const char * call_check_moh(struct call_monologue *from_ml, struct call_monologue *to_ml,
sdp_ng_flags *flags)
{
#ifdef WITH_TRANSCODING
/* offerer might want to use MoH capabilities of the other monologue.
* e.g. when offerer lacks own MoH capabilities, but wants other side to hear MoH sound during the hold.
*/
bool reflected = flags->moh_reflect ? true : false;
if (!flags->moh_double_hold && call_ml_wants_moh(from_ml, (reflected ? to_ml : NULL), flags->opmode))
{
const char *errstr = NULL;
struct call_monologue *moh_ml = reflected ? to_ml : from_ml;
if (flags->repeat_duration != -1)
ilog(LOG_DEBUG, "Repeat-duration given via flags, but the configuration source will be used!");
media_player_opts_t opts = MPO(
.repeat = rtpe_config.moh_max_repeats,
/* MoH always has duration set (even if not defined) */
.duration_spent = rtpe_config.moh_max_duration,
.start_pos = 0,
.block_egress = 1,
.codec_set = flags->codec_set,
.file = moh_ml->moh_file,
.blob = moh_ml->moh_blob,
.db_id = moh_ml->moh_db_id,
.moh = 1, /* mark as moh enabled */
);
/* whom to play the moh audio */
errstr = call_play_media_for_ml(to_ml, opts, NULL);
if (errstr) {
to_ml->player->opts.moh = 0; /* initialization failed, mark accordingly */
return errstr;
}
/* handle MoH related flags */
call_ml_moh_handle_flags((reflected ? NULL : from_ml), to_ml);
ilog(LOG_DEBUG, "Music on hold triggered with coming SDP offer.");
} else if (call_ml_stops_moh(from_ml, to_ml, flags->opmode))
{
/* whom to stop the moh audio */
call_stop_media_for_ml(to_ml);
ilog(LOG_DEBUG, "Music on hold stopped with coming SDP offer.");
}
return NULL;
#else
return NULL;
#endif
}
const char * call_play_media_for_ml(struct call_monologue *ml,
media_player_opts_t opts, sdp_ng_flags *flags)
{
#ifdef WITH_TRANSCODING
/* if mixing is enabled, codec handlers of all sources must be updated */
codec_update_all_source_handlers(ml, flags);
/* this starts the audio player if needed */
update_init_monologue_subscribers(ml, OP_PLAY_MEDIA);
/* media_player_new() now knows that audio player is in use
* TODO: player options can have changed if already exists */
media_player_new(&ml->player, ml, NULL, &opts);
if (opts.file.len) {
if (!media_player_play_file(ml->player, opts))
return "Failed to start media playback from file";
}
else if (opts.blob.len) {
if (!media_player_play_blob(ml->player, opts))
return "Failed to start media playback from blob";
}
else if (opts.db_id > 0) {
if (!media_player_play_db(ml->player, opts))
return "Failed to start media playback from database";
}
else
return "No media file specified";
return NULL;
#else
return "Not implemented";
#endif
}
long long call_stop_media_for_ml(struct call_monologue *ml)
{
#ifdef WITH_TRANSCODING
long long ret = media_player_stop(ml->player);
/* restore to non-mixing if needed */
codec_update_all_source_handlers(ml, NULL);
update_init_monologue_subscribers(ml, OP_STOP_MEDIA);
/* mark MoH as already not used (it can be unset now) */
ml->player->opts.moh = 0;
return ret;
#else
return 0;
#endif
}
#ifdef WITH_TRANSCODING
static int __mp_avio_read_wrap(void *opaque, uint8_t *buf, int buf_size) {
struct media_player_coder *c = opaque;
if (buf_size < 0)
return AVERROR(EINVAL);
if (buf_size == 0)
return 0;
if (!c->read_pos.len)
return AVERROR_EOF;
int len = buf_size;
if (len > c->read_pos.len)
len = c->read_pos.len;
memcpy(buf, c->read_pos.s, len);
str_shift(&c->read_pos, len);
return len;
}
static int __mp_avio_read(void *opaque, uint8_t *buf, int buf_size) {
ilog(LOG_DEBUG, "__mp_avio_read(%i)", buf_size);
int ret = __mp_avio_read_wrap(opaque, buf, buf_size);
ilog(LOG_DEBUG, "__mp_avio_read(%i) = %i", buf_size, ret);
return ret;
}
static int64_t __mp_avio_seek_set(struct media_player_coder *c, int64_t offset) {
ilog(LOG_DEBUG, "__mp_avio_seek_set(%" PRIi64 ")", offset);
if (offset < 0)
return AVERROR(EINVAL);
c->read_pos = c->blob;
if (str_shift(&c->read_pos, offset))
return AVERROR_EOF;
return offset;
}
static int64_t __mp_avio_seek(void *opaque, int64_t offset, int whence) {
int64_t ret;
ilog(LOG_DEBUG, "__mp_avio_seek(%" PRIi64 ", %i)", offset, whence);
struct media_player_coder *c = opaque;
if (whence == SEEK_SET)
ret = __mp_avio_seek_set(c, offset);
else if (whence == SEEK_CUR)
ret = __mp_avio_seek_set(c, ((int64_t) (c->read_pos.s - c->blob.s)) + offset);
else if (whence == SEEK_END)
ret = __mp_avio_seek_set(c, ((int64_t) c->blob.len) + offset);
else
ret = AVERROR(EINVAL);
ilog(LOG_DEBUG, "__mp_avio_seek(%" PRIi64 ", %i) = %" PRIi64, offset, whence, ret);
return ret;
}
// call->master_lock held in W
static mp_cached_code __media_player_add_blob_id(struct media_player *mp,
media_player_opts_t opts,
const rtp_payload_type *dst_pt)
{
const char *err;
int av_ret = 0;
if (opts.db_id > 0) {
mp->cache_index.type = MP_DB;
mp->cache_index.db_id = opts.db_id;
if (media_player_cache_get_entry(mp, dst_pt, opts.codec_set))
return MPC_CACHED;
}
else {
mp->cache_index.type = MP_BLOB;
mp->cache_index.file = call_str_cpy(&opts.blob);
if (media_player_cache_get_entry(mp, dst_pt, opts.codec_set))
return MPC_CACHED;
}
mp->coder.blob = call_str_cpy(&opts.blob);
mp->coder.read_pos = mp->coder.blob;
err = "could not allocate AVFormatContext";
mp->coder.fmtctx = avformat_alloc_context();
if (!mp->coder.fmtctx)
goto err;
void *avio_buf = av_malloc(DEFAULT_AVIO_BUFSIZE);
err = "failed to allocate AVIO buffer";
if (!avio_buf)
goto err;
mp->coder.avioctx = avio_alloc_context(avio_buf, DEFAULT_AVIO_BUFSIZE, 0, &mp->coder, __mp_avio_read,
NULL, __mp_avio_seek);
err = "failed to allocate AVIOContext";
if (!mp->coder.avioctx)
goto err;
mp->coder.fmtctx->pb = mp->coder.avioctx;
// consumes allocated mp->coder.fmtctx
err = "failed to open AVFormatContext input";
av_ret = avformat_open_input(&mp->coder.fmtctx, "dummy", NULL, NULL);
if (av_ret < 0)
goto err;
return MPC_OK;
err:
ilog(LOG_ERR, "Failed to start media playback from memory: %s", err);
if (av_ret)
ilog(LOG_ERR, "Error returned from libav: %s", av_error(av_ret));
return MPC_ERR;
}
// call->master_lock held in W
static bool media_player_play_blob(struct media_player *mp, media_player_opts_t opts) {
const rtp_payload_type *dst_pt = media_player_play_init(mp);
if (!dst_pt)
return false;
/* make sure to reset db_id before using blob */
opts.db_id = 0;
mp_cached_code ret = __media_player_add_blob_id(mp, opts, dst_pt);
if (ret == MPC_CACHED)
return true;
if (ret == MPC_ERR)
return false;
return media_player_play_start(mp, dst_pt, opts.codec_set);
}
// call->master_lock held in W
static bool media_player_add_blob(struct media_player *mp, media_player_opts_t opts) {
/* make sure to reset db_id before using blob */
opts.db_id = 0;
int ret = __media_player_add_blob_id(mp, opts, NULL);
return ret != MPC_ERR;
}
static int __connect_db(void) {
if (mysql_conn) {
mysql_close(mysql_conn);
mysql_conn = NULL;
}
mysql_conn = mysql_init(NULL);
if (!mysql_conn)
return -1;
if (!mysql_real_connect(mysql_conn, rtpe_config.mysql_host, rtpe_config.mysql_user, rtpe_config.mysql_pass, NULL, rtpe_config.mysql_port,
NULL, CLIENT_IGNORE_SIGPIPE))
goto err;
return 0;
err:
ilog(LOG_ERR, "Couldn't connect to database: %s", mysql_error(mysql_conn));
mysql_close(mysql_conn);
mysql_conn = NULL;
return -1;
}
static char *media_player_make_cache_entry_name(unsigned long long id) {
return g_strdup_printf("%s/%llu.blob", rtpe_config.db_media_cache, id);
}
static bool media_player_add_cache_file_create(const char *s, size_t len, unsigned long long id) {
if (!rtpe_config.db_media_cache)
return false;
g_autoptr(char) fn = media_player_make_cache_entry_name(id);
GError *err = NULL;
gboolean ok = g_file_set_contents(fn, s, len, &err);
if (!ok) {
ilog(LOG_WARN, "Failed to write to cache file '%s': %s", fn, err->message);
g_error_free(err);
return false;
}
return true;
}
static str dummy_dup(const char *s, size_t l) {
return STR_NULL;
}
static const char *media_player_get_db_id(str *out, unsigned long long id,
str (*dup_fn)(const char *, size_t),
bool (*cache_fn)(const char *s, size_t len, unsigned long long id))
{
const char *err;
g_autoptr(char) query = NULL;
err = "missing configuration";
if (!rtpe_config.mysql_host || !rtpe_config.mysql_query)
goto err;
query = g_strdup_printf(rtpe_config.mysql_query, id);
size_t len = strlen(query);
for (int retries = 0; retries < 5; retries++) {
if (!mysql_conn || retries != 0) {
err = "failed to connect to database";
if (__connect_db())
goto err;
}
int ret = mysql_real_query(mysql_conn, query, len);
if (ret == 0)
goto success;
ret = mysql_errno(mysql_conn);
if (ret == CR_SERVER_GONE_ERROR || ret == CR_SERVER_LOST)
continue;
ilog(LOG_ERR, "Failed to query from database: %s", mysql_error(mysql_conn));
}
err = "exceeded max number of database retries";
goto err;
success:;
MYSQL_RES *res = mysql_store_result(mysql_conn);
err = "failed to get result from database";
if (!res)
goto err;
MYSQL_ROW row = mysql_fetch_row(res);
unsigned long *lengths = mysql_fetch_lengths(res);
err = "empty result from database";
if (!row || !lengths || !row[0] || !lengths[0]) {
mysql_free_result(res);
goto err;
}
err = "failed to insert data into cache";
if (!cache_fn(row[0], lengths[0], id))
goto err;
*out = dup_fn(row[0], lengths[0]);
return NULL;
err:
if (query)
ilog(LOG_ERR, "Failed to read media from database (used query '%s'): %s", query, err);
else
ilog(LOG_ERR, "Failed to read media from database: %s", err);
return err;
}
// call->master_lock held in W
static mp_cached_code __media_player_add_db(struct media_player *mp,
media_player_opts_t opts,
const rtp_payload_type *dst_pt)
{
const char *err;
// check if we have it in memory
__auto_type fo = media_player_db_id_get(opts.db_id);
if (fo) {
ilog(LOG_DEBUG, "Using cached DB media for playback");
return media_player_set_media_file(mp, opts, dst_pt, fo);
}
// or maybe we have a cache file for it
if (rtpe_config.db_media_cache) {
g_autoptr(char) fn = media_player_make_cache_entry_name(opts.db_id);
gchar *buf = NULL;
gsize len = -1;
gboolean ret = g_file_get_contents(fn, &buf, &len, NULL);
if (ret && len > 0) {
// use this as blob and play it
ilog(LOG_DEBUG, "Using cached file of DB media for playback");
// use a `media_player_media_file` object to hold a reference on the g_malloc'd
// data to avoid having to memcpy it
fo = media_player_media_file_new(STR_LEN(buf, len));
utimensat(AT_FDCWD, fn,
(struct timespec[2])
{ { .tv_nsec = UTIME_NOW }, { .tv_nsec = UTIME_OMIT } },
0);
return media_player_set_media_file(mp, opts, dst_pt, fo);
}
if (ret) // zero-length file
g_free(buf);
}
err = media_player_get_db_id(&opts.blob, opts.db_id, call_str_cpy_len, media_player_add_cache_file);
if (err)
return MPC_ERR;
return __media_player_add_blob_id(mp, opts, dst_pt);
}
// call->master_lock held in W
static bool media_player_play_db(struct media_player *mp, media_player_opts_t opts) {
const rtp_payload_type *dst_pt = media_player_play_init(mp);
if (!dst_pt)
return false;
mp_cached_code ret = __media_player_add_db(mp, opts, dst_pt);
if (ret == MPC_CACHED)
return true;
if (ret == MPC_ERR)
return false;
return media_player_play_start(mp, dst_pt, opts.codec_set);
}
// call->master_lock held in W
static bool media_player_add_db(struct media_player *mp, media_player_opts_t opts) {
int ret = __media_player_add_db(mp, opts, NULL);
return ret != MPC_ERR;
}
static void media_player_run(void *ptr) {
struct media_player *mp = ptr;
call_t *call = mp->call;
log_info_media(mp->media);
//ilog(LOG_DEBUG, "running scheduled media_player");
rwlock_lock_r(&call->master_lock);
mutex_lock(&mp->lock);
bool finished = false;
if (mp->next_run)
finished = mp->run_func(mp);
mutex_unlock(&mp->lock);
rwlock_unlock_r(&call->master_lock);
if (finished) {
rwlock_lock_w(&call->master_lock);
mp->next_run = 0;
if (mp->opts.block_egress)
MEDIA_CLEAR(mp->media, BLOCK_EGRESS);
codec_update_all_source_handlers(mp->media->monologue, NULL);
update_init_monologue_subscribers(mp->media->monologue, OP_PLAY_MEDIA);
rwlock_unlock_w(&call->master_lock);
}
log_info_pop();
}
bool media_player_is_active(struct call_monologue *ml) {
if (!ml)
return false;
if (!ml->player)
return false;
if (!ml->player->next_run)
return false;
return true;
}
static unsigned int media_player_cache_entry_hash(const struct media_player_cache_index *i) {
unsigned int ret;
switch (i->index.type) {
case MP_DB:
ret = i->index.db_id;
break;
case MP_FILE:
case MP_BLOB:
ret = str_hash(&i->index.file);
break;
default:
abort();
}
ret ^= str_hash(&i->dst_pt.encoding_with_full_params);
ret ^= i->index.type;
return ret;
}
static gboolean media_player_cache_entry_eq(const struct media_player_cache_index *a,
const struct media_player_cache_index *b)
{
if (a->index.type != b->index.type)
return FALSE;
switch (a->index.type) {
case MP_DB:
if (a->index.db_id != b->index.db_id)
return FALSE;
break;
case MP_FILE:
case MP_BLOB:
if (!str_equal(&a->index.file, &b->index.file))
return FALSE;
break;
default:
abort();
}
return str_equal(&a->dst_pt.encoding_with_full_params, &b->dst_pt.encoding_with_full_params);
}
static void __media_player_cache_entry_free(struct media_player_cache_entry *e) {
t_ptr_array_free(e->packets, true);
mutex_destroy(&e->lock);
g_free(e->info_str);
if (t_hash_table_is_set(e->wait_queue)) {
media_player_ht_iter iter;
t_hash_table_iter_init(&iter, e->wait_queue);
struct media_player *mp;
while (t_hash_table_iter_next(&iter, &mp, NULL))
obj_put(&mp->tt_obj);
t_hash_table_destroy(e->wait_queue);
}
media_player_coder_shutdown(&e->coder);
av_packet_free(&e->coder.pkt);
kernel_free_packet_stream(e->kernel_idx);
g_free(e->index.index.file.s);
payload_type_clear(&e->index.dst_pt);
memory_arena_free(&e->arena);
}
#endif
// call->master_lock held in W
bool media_player_start(struct media_player *mp) {
#ifdef WITH_TRANSCODING
if (!mp->coder.fmtctx) // initialised?
return false;
const rtp_payload_type *dst_pt = media_player_play_setup(mp);
if (!dst_pt)
return false;
return media_player_play_start(mp, dst_pt, str_case_value_ht_null());
#else
return false;
#endif
}
void media_player_init(void) {
#ifdef WITH_TRANSCODING
if (rtpe_config.player_cache)
media_player_cache = media_player_cache_ht_new();
timerthread_init(&media_player_thread, rtpe_config.media_num_threads, media_player_run);
#endif
timerthread_init(&send_timer_thread, rtpe_config.media_num_threads, timerthread_queue_run);
}
void media_player_free(void) {
#ifdef WITH_TRANSCODING
timerthread_free(&media_player_thread);
if (t_hash_table_is_set(media_player_cache))
t_hash_table_destroy(media_player_cache);
if (t_hash_table_is_set(media_player_media_files))
t_hash_table_destroy(media_player_media_files);
t_queue_clear_full(&media_player_media_files_names, str_free);
if (t_hash_table_is_set(media_player_db_media))
t_hash_table_destroy(media_player_db_media);
g_queue_clear(&media_player_db_media_ids);
#endif
timerthread_free(&send_timer_thread);
}
void media_player_launch(void) {
#ifdef WITH_TRANSCODING
timerthread_launch(&media_player_thread, rtpe_config.scheduling, rtpe_config.priority, "media player");
#endif
}
void send_timer_launch(void) {
//ilog(LOG_DEBUG, "send_timer_loop");
timerthread_launch(&send_timer_thread, rtpe_config.scheduling, rtpe_config.priority, "media player");
}
bool media_player_preload_files(char **files) {
#ifdef WITH_TRANSCODING
if (!files || !files[0])
return true;
for (char **filep = files; *filep; filep++) {
char *file = *filep;
while (*file == ' ')
file++;
if (!strcmp(file, "ondemand") || !strcmp(file, "on demand") || !strcmp(file, "on-demand")) {
media_player_media_files_get = media_player_media_files_get_create;
continue;
}
ilog(LOG_DEBUG, "Reading media file '%s' for caching", file);
str f = STR(file);
if (t_hash_table_is_set(media_player_media_files) && t_hash_table_lookup(media_player_media_files, &f)) {
ilog(LOG_CRIT, "Duplicate entry for caching media file '%s'", file);
return false;
}
__auto_type fo = media_player_media_file_read_c(file);
if (!fo)
return false;
media_player_media_files_insert(&f, fo);
}
#endif
return true;
}
bool media_player_preload_db(char **ids) {
#ifdef WITH_TRANSCODING
if (!ids || !ids[0])
return true;
for (char **idp = ids; *idp; idp++) {
char *id_s = *idp;
while (*id_s == ' ')
id_s++;
if (!strcmp(id_s, "ondemand") || !strcmp(id_s, "on demand") || !strcmp(id_s, "on-demand")) {
media_player_db_id_get = media_player_db_id_get_create;
continue;
}
char *endp = NULL;
unsigned long long id = strtoull(id_s, &endp, 0);
if (id == 0 || id == ULLONG_MAX || (endp && *endp != '\0')) {
ilog(LOG_CRIT, "Invalid DB ID string number: '%s'", id_s);
return false;
}
ilog(LOG_DEBUG, "Reading media ID %llu from DB for caching", id);
if (t_hash_table_is_set(media_player_db_media)
&& t_hash_table_lookup(media_player_db_media, GUINT_TO_POINTER(id)))
{
ilog(LOG_CRIT, "Duplicate entry for caching media ID %llu", id);
return false;
}
__auto_type fo = media_player_db_id_read(id);
if (!fo)
return false;
media_player_db_id_insert(id, fo);
}
#endif
return true;
}
bool media_player_add_cached_file(str *name) {
bool ret = false;
#ifdef WITH_TRANSCODING
__auto_type fonew = media_player_media_file_read_str(name);
if (!fonew)
return false;
RWLOCK_W(&media_player_media_files_names_lock);
LOCK(&media_player_media_files_lock);
__auto_type foold = t_hash_table_lookup(media_player_media_files, name);
if (foold) {
fonew->str_link = foold->str_link;
t_hash_table_replace(media_player_media_files, name, fonew);
obj_put(foold);
}
else
media_player_media_files_insert(name, fonew);
ilog(LOG_DEBUG, "Added cached media file '" STR_FORMAT "'",
STR_FMT(name));
ret = true;
#endif
return ret;
}
bool media_player_reload_file(str *name) {
bool ret = false;
#ifdef WITH_TRANSCODING
__auto_type fo = media_player_media_files_get_only(name);
assert(fo != NULL);
// get file mtime
char file_s[PATH_MAX];
snprintf(file_s, sizeof(file_s), STR_FORMAT, STR_FMT(name));
struct stat sb;
int fail = stat(file_s, &sb);
if (fail)
ilog(LOG_WARN, "Failed to stat() media file '" STR_FORMAT "': %s",
STR_FMT(name), strerror(errno));
else if (timespec_us(sb.st_mtim) > fo->mtime_us) {
__auto_type fonew = media_player_media_file_read_c(file_s);
if (fonew) {
// got a new entry. swap it out against the old one
LOCK(&media_player_media_files_lock);
if (t_hash_table_is_set(media_player_media_files)
&& t_hash_table_lookup(media_player_media_files, name) == fo)
{
t_hash_table_insert(media_player_media_files, name, fonew); // releases `fo` reference
ilog(LOG_DEBUG, "Reloaded cached media file '" STR_FORMAT "'",
STR_FMT(name));
ret = true;
}
else // somebody beat us to it
obj_put(fonew);
}
}
obj_put(fo);
#endif
return ret;
}
unsigned int media_player_reload_files(void) {
unsigned int ret = 0;
#ifdef WITH_TRANSCODING
RWLOCK_R(&media_player_media_files_names_lock);
for (__auto_type l = media_player_media_files_names.head; l; l = l->next) {
str *name = l->data;
if (media_player_reload_file(name))
ret++;
}
#endif
return ret;
}
bool media_player_add_db_media(unsigned long long id) {
bool ret = false;
#ifdef WITH_TRANSCODING
__auto_type fonew = media_player_db_id_read(id);
if (!fonew)
return false;
RWLOCK_W(&media_player_db_media_ids_lock);
LOCK(&media_player_db_media_lock);
__auto_type foold = t_hash_table_lookup(media_player_db_media, GUINT_TO_POINTER(id));
if (foold) {
fonew->gen_link = foold->gen_link;
t_hash_table_replace(media_player_db_media, GUINT_TO_POINTER(id), fonew);
obj_put(foold);
}
else
media_player_db_id_insert(id, fonew);
ilog(LOG_DEBUG, "Added cached media DB entry %llu", id);
ret = true;
#endif
return ret;
}
bool media_player_reload_db_media(unsigned long long id) {
bool ret = false;
#ifdef WITH_TRANSCODING
__auto_type fo = media_player_db_id_get(id);
if (!fo)
return false;
// read fresh copy
__auto_type fonew = media_player_db_id_read(id);
if (fonew) {
// got a new entry. swap it out against the old one
LOCK(&media_player_db_media_lock);
if (t_hash_table_is_set(media_player_db_media)
&& t_hash_table_lookup(media_player_db_media, GUINT_TO_POINTER(id)) == fo)
{
t_hash_table_insert(media_player_db_media, GUINT_TO_POINTER(id), fonew); // releases `fo` reference
ilog(LOG_DEBUG, "Reloaded cached media DB entry %llu", id);
ret = true;
}
else // somebody beat us to it
obj_put(fonew);
}
obj_put(fo);
#endif
return ret;
}
unsigned int media_player_reload_db_medias(void) {
unsigned int ret = 0;
#ifdef WITH_TRANSCODING
RWLOCK_R(&media_player_db_media_ids_lock);
for (__auto_type l = media_player_db_media_ids.head; l; l = l->next) {
unsigned long long id = GPOINTER_TO_UINT(l->data);
if (media_player_reload_db_media(id))
ret++;
}
#endif
return ret;
}
enum thread_looper_action media_player_refresh_timer(void) {
media_player_reload_files();
return TLA_CONTINUE;
}
enum thread_looper_action media_player_refresh_db(void) {
media_player_reload_db_medias();
return TLA_CONTINUE;
}
enum thread_looper_action media_player_refresh_cache(void) {
media_player_reload_caches();
return TLA_CONTINUE;
}
#ifdef WITH_TRANSCODING
// media_player_media_files_names_lock must be held
// media_player_media_files_lock must not be held
static bool __media_player_evict_file(str *name) {
str *key;
struct media_player_media_file *val;
{
// short lock: remove from containers, unlock, then free
LOCK(&media_player_media_files_lock);
if (!t_hash_table_is_set(media_player_media_files))
return false;
bool ret = t_hash_table_steal_extended(media_player_media_files, name, &key, &val);
if (!ret)
return false;
t_queue_delete_link(&media_player_media_files_names, val->str_link);
}
obj_put(val);
g_free(key);
return true;
}
#endif
bool media_player_evict_file(str *name) {
#ifdef WITH_TRANSCODING
RWLOCK_W(&media_player_media_files_names_lock);
return __media_player_evict_file(name);
#else
return false;
#endif
}
unsigned int media_player_evict_files(void) {
unsigned int ret = 0;
#ifdef WITH_TRANSCODING
while (true) {
// intermittently release lock as this is low priority
RWLOCK_W(&media_player_media_files_names_lock);
if (!media_player_media_files_names.head)
return ret;
str *name = media_player_media_files_names.head->data;
if (__media_player_evict_file(name))
ret++;
}
#endif
return ret;
}
#ifdef WITH_TRANSCODING
// media_player_db_media_ids_lock must be held
// media_player_db_media_lock must not be held
static bool __media_player_evict_db_media(unsigned long long id) {
struct media_player_media_file *val;
{
// short lock: remove from containers, unlock, then free
LOCK(&media_player_db_media_lock);
if (!t_hash_table_is_set(media_player_db_media))
return false;
bool ret = t_hash_table_steal_extended(media_player_db_media, GUINT_TO_POINTER(id), NULL, &val);
if (!ret)
return false;
g_queue_delete_link(&media_player_db_media_ids, val->gen_link);
}
obj_put(val);
return true;
}
#endif
bool media_player_evict_db_media(unsigned long long id) {
#ifdef WITH_TRANSCODING
RWLOCK_W(&media_player_db_media_ids_lock);
return __media_player_evict_db_media(id);
#else
return false;
#endif
}
unsigned int media_player_evict_db_medias(void) {
unsigned int ret = 0;
#ifdef WITH_TRANSCODING
while (true) {
// intermittently release lock as this is low priority
RWLOCK_W(&media_player_db_media_ids_lock);
if (!media_player_db_media_ids.head)
return ret;
unsigned long long id = GPOINTER_TO_UINT(media_player_db_media_ids.head->data);
if (__media_player_evict_db_media(id))
ret++;
}
#endif
return ret;
}
str_q media_player_list_files(void) {
str_q ret = TYPED_GQUEUE_INIT;
#ifdef WITH_TRANSCODING
RWLOCK_R(&media_player_media_files_names_lock);
for (__auto_type l = media_player_media_files_names.head; l; l = l->next)
t_queue_push_tail(&ret, str_dup(l->data));
#endif
return ret;
}
GQueue media_player_list_dbs(void) {
GQueue ret = G_QUEUE_INIT;
#ifdef WITH_TRANSCODING
RWLOCK_R(&media_player_db_media_ids_lock);
for (GList *l = media_player_db_media_ids.head; l; l = l->next)
g_queue_push_tail(&ret, l->data);
#endif
return ret;
}
bool media_player_get_file_times(const str *s, int64_t *mtime, int64_t *atime) {
#ifdef WITH_TRANSCODING
LOCK(&media_player_media_files_lock);
__auto_type fo = t_hash_table_lookup(media_player_media_files, s);
if (!fo)
return false;
*mtime = fo->mtime_us / 1000000L;
*atime = fo->atime_us / 1000000L;
return true;
#else
return false;
#endif
}
bool media_player_get_db_times(unsigned long long id, int64_t *mtime, int64_t *atime) {
#ifdef WITH_TRANSCODING
LOCK(&media_player_db_media_lock);
__auto_type fo = t_hash_table_lookup(media_player_db_media, GUINT_TO_POINTER(id));
if (!fo)
return false;
*mtime = fo->mtime_us / 1000000L;
*atime = fo->atime_us / 1000000L;
return true;
#else
return false;
#endif
}
G_DEFINE_AUTOPTR_CLEANUP_FUNC(DIR, closedir)
typedef union {
GQueue *q;
unsigned int *u;
} iterate_db_cache_arg __attribute__((__transparent_union__));
static void media_player_iterate_db_cache(void (*callback)(unsigned long long, iterate_db_cache_arg),
iterate_db_cache_arg arg)
{
#ifdef WITH_TRANSCODING
if (!rtpe_config.db_media_cache)
return;
g_autoptr(DIR) dir = opendir(rtpe_config.db_media_cache);
if (!dir) {
ilog(LOG_ERR, "Failed to open media cache directory: %s", strerror(errno));
return;
}
struct dirent *de;
while ((de = readdir(dir))) {
char *errp;
unsigned long long id = strtoull(de->d_name, &errp, 10);
if (!id || strcmp(errp, ".blob"))
continue;
callback(id, arg);
}
#endif
}
static void media_player_add_to_queue(unsigned long long id, GQueue *q) {
g_queue_push_tail(q, GUINT_TO_POINTER(id));
}
GQueue media_player_list_caches(void) {
GQueue ret = G_QUEUE_INIT;
media_player_iterate_db_cache(media_player_add_to_queue, &ret);
return ret;
}
bool media_player_get_cache_times(unsigned long long id, int64_t *mtime, int64_t *atime) {
#ifdef WITH_TRANSCODING
g_autoptr(char) fn = media_player_make_cache_entry_name(id);
struct stat sb;
int fail = stat(fn, &sb);
if (fail)
return false;
*mtime = sb.st_mtim.tv_sec;
*atime = sb.st_atim.tv_sec;
return true;
#else
return false;
#endif
}
bool media_player_evict_cache(unsigned long long id) {
#ifdef WITH_TRANSCODING
g_autoptr(char) fn = media_player_make_cache_entry_name(id);
int ret = unlink(fn);
if (ret == 0)
return true;
if (errno == ENOENT) // ignore these
return false;
ilog(LOG_WARN, "Failed to unlink media cache file '%s': %s", fn, strerror(errno));
#endif
return false;
}
static void media_player_evict_caches_all(unsigned long long id, unsigned int *u) {
if (media_player_evict_cache(id))
(*u)++;
}
unsigned int media_player_evict_caches(void) {
unsigned int ret = 0;
media_player_iterate_db_cache(media_player_evict_caches_all, &ret);
return ret;
}
bool media_player_preload_cache(char **ids) {
#ifdef WITH_TRANSCODING
if (!ids || !ids[0])
return true;
for (char **idp = ids; *idp; idp++) {
char *id_s = *idp;
while (*id_s == ' ')
id_s++;
if (!strcmp(id_s, "ondemand") || !strcmp(id_s, "on demand") || !strcmp(id_s, "on-demand")) {
media_player_add_cache_file = media_player_add_cache_file_create;
continue;
}
char *endp = NULL;
unsigned long long id = strtoull(id_s, &endp, 0);
if (id == 0 || id == ULLONG_MAX || (endp && *endp != '\0')) {
ilog(LOG_CRIT, "Invalid DB ID string number: '%s'", id_s);
return false;
}
ilog(LOG_DEBUG, "Reading media ID %llu from DB for caching", id);
str out;
const char *err = media_player_get_db_id(&out, id, dummy_dup,
media_player_add_cache_file_create);
if (err)
return false; // error has been logged already
}
#endif
return true;
}
bool media_player_add_cache(unsigned long long id) {
bool ret = false;
#ifdef WITH_TRANSCODING
str out;
const char *err = media_player_get_db_id(&out, id, dummy_dup,
media_player_add_cache_file_create);
if (!err)
ret = true;
#endif
return ret;
}
bool media_player_reload_cache(unsigned long long id) {
bool ret = false;
#ifdef WITH_TRANSCODING
// "reload" implies that the file has to exist already
g_autoptr(char) fn = media_player_make_cache_entry_name(id);
if (!g_file_test(fn, G_FILE_TEST_EXISTS))
return false;
str out;
const char *err = media_player_get_db_id(&out, id, dummy_dup,
media_player_add_cache_file_create);
if (!err)
ret = true;
#endif
return ret;
}
static void media_player_reload_caches_all(unsigned long long id, unsigned int *u) {
if (media_player_reload_cache(id))
(*u)++;
}
unsigned int media_player_reload_caches(void) {
unsigned int ret = 0;
media_player_iterate_db_cache(media_player_reload_caches_all, &ret);
return ret;
}
charp_q media_player_list_player_cache(void) {
charp_q ret = TYPED_GQUEUE_INIT;
#ifdef WITH_TRANSCODING
if (!t_hash_table_is_set(media_player_cache))
return ret;
media_player_cache_ht_iter iter;
LOCK(&media_player_cache_lock);
t_hash_table_iter_init(&iter, media_player_cache);
struct media_player_cache_entry *entry;
while (t_hash_table_iter_next(&iter, NULL, &entry))
t_queue_push_tail(&ret, g_strdup_printf("%s for PT " STR_FORMAT, entry->info_str,
STR_FMT(&entry->index.dst_pt.encoding_with_full_params)));
#endif
return ret;
}
#ifdef WITH_TRANSCODING
// lock must not be held
static bool media_player_evict_player_cache(struct media_player_cache_entry *entry) {
LOCK(&media_player_cache_lock);
if (t_hash_table_remove(media_player_cache, &entry->index))
return true;
return false;
}
#endif
unsigned int media_player_evict_player_caches(void) {
unsigned int ret = 0;
#ifdef WITH_TRANSCODING
if (!t_hash_table_is_set(media_player_cache))
return 0;
// grab references from hash table
media_player_cache_entry_q q = TYPED_GQUEUE_INIT;
media_player_cache_ht_iter iter;
{
LOCK(&media_player_cache_lock);
t_hash_table_iter_init(&iter, media_player_cache);
struct media_player_cache_entry *entry;
while (t_hash_table_iter_next(&iter, NULL, &entry))
t_queue_push_tail(&q, obj_get(entry));
}
// release references
while (q.head) {
__auto_type entry = t_queue_pop_head(&q);
if (media_player_evict_player_cache(entry))
ret++;
obj_put(entry);
}
#endif
return ret;
}
#ifdef WITH_TRANSCODING
static void media_player_expire_files(void) {
if (rtpe_config.media_expire_us <= 0)
return;
int64_t limit = rtpe_now - rtpe_config.media_expire_us;
unsigned int num = 0;
{
RWLOCK_R(&media_player_media_files_names_lock);
str_list *next;
for (__auto_type l = media_player_media_files_names.head; l; l = next) {
next = l->next;
g_auto(str) name = STR_NULL;
{
LOCK(&media_player_media_files_lock);
__auto_type fo = t_hash_table_lookup(media_player_media_files, l->data);
if (!fo)
continue;
if (fo->atime_us >= limit)
continue;
name = str_dup_str(l->data);
}
if (__media_player_evict_file(&name))
num++;
}
}
if (num)
ilog(LOG_DEBUG, "Removed %u old entries from media cache", num);
}
static void media_player_expire_dbs(void) {
if (rtpe_config.db_expire_us <= 0)
return;
int64_t limit = rtpe_now - rtpe_config.db_expire_us;
unsigned int num = 0;
{
RWLOCK_R(&media_player_db_media_ids_lock);
GList *next;
for (__auto_type l = media_player_db_media_ids.head; l; l = next) {
next = l->next;
unsigned long long id;
{
LOCK(&media_player_db_media_lock);
__auto_type fo = t_hash_table_lookup(media_player_db_media, l->data);
if (!fo)
continue;
if (fo->atime_us >= limit)
continue;
id = GPOINTER_TO_UINT(l->data);
}
if (__media_player_evict_db_media(GPOINTER_TO_UINT(id)))
num++;
}
}
if (num)
ilog(LOG_DEBUG, "Removed %u old entries from DB media cache", num);
}
static void media_player_expire_cache_entry(unsigned long long id, unsigned int *num) {
int64_t mtime, atime;
if (!media_player_get_cache_times(id, &mtime, &atime))
return;
int64_t limit = rtpe_now - rtpe_config.db_expire_us;
if (atime >= limit)
return;
if (media_player_evict_cache(id))
(*num)++;
}
static void media_player_expire_caches(void) {
if (rtpe_config.cache_expire <= 0)
return;
unsigned int ret = 0;
media_player_iterate_db_cache(media_player_expire_cache_entry, &ret);
}
#endif
enum thread_looper_action media_player_expire(void) {
#ifdef WITH_TRANSCODING
media_player_expire_files();
media_player_expire_dbs();
media_player_expire_caches();
return TLA_CONTINUE;
#else
return TLA_BREAK;
#endif
}