-/*
- * Copyright (C) 1997-2007 Andre Noll <maan@systemlinux.org>
- *
- * Licensed under the GPL v2. For licencing details see COPYING.
- */
+/* Copyright (C) 1997 Andre Noll <maan@tuebingen.mpg.de>, see file COPYING. */
-/** \file vss.c the virtual streaming system
+/** \file vss.c The virtual streaming system.
*
* This contains the audio streaming code of para_server which is independent
* of the current audio format, audio file selector and of the activated
* senders.
*/
-#include <sys/mman.h> /* mmap */
-#include <sys/time.h> /* gettimeofday */
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <regex.h>
+#include <osl.h>
#include <sys/types.h>
-#include <dirent.h>
+#include <arpa/inet.h>
+#include <sys/un.h>
+#include <netdb.h>
+#include <lopsub.h>
+#include "server.lsg.h"
#include "para.h"
+#include "error.h"
+#include "portable_io.h"
+#include "fec.h"
+#include "string.h"
#include "afh.h"
+#include "afs.h"
#include "server.h"
-#include "server.cmdline.h"
-#include "afs_common.h"
-#include "vss.h"
+#include "net.h"
+#include "list.h"
#include "send.h"
-#include "error.h"
-#include "string.h"
+#include "sched.h"
+#include "vss.h"
+#include "ipc.h"
#include "fd.h"
-extern const char *status_item_list[];
-
-static struct timeval announce_tv;
-static struct timeval data_send_barrier;
-static struct timeval eof_barrier;
-static struct timeval autoplay_barrier;
-
extern struct misc_meta_data *mmd;
-extern struct audio_file_selector selectors[];
-extern struct sender senders[];
-static int audio_file;
-static char *map;
+extern void dccp_send_init(struct sender *);
+extern void http_send_init(struct sender *);
+extern void udp_send_init(struct sender *);
-/* The mp3 audio format handler does not need any libs. */
-void mp3_init(struct audio_format_handler *);
-
-#ifdef HAVE_OGGVORBIS
- void ogg_init(struct audio_format_handler *);
-#endif
-#ifdef HAVE_FAAD
- void aac_afh_init(struct audio_format_handler *);
-#endif
-
-/**
- * The list of supported audio formats.
- *
- * We always define the full array of audio formats even if some audio formats
- * were not compiled in. This is because for each audio file the number of its
- * audio format is stored in the databse. We don't want that numbers to become
- * stale just because the user installed a new version of paraslash that
- * supports a different set of audio formats.
- *
- * It can still be easily detected whether an audio format is compiled in by
- * checking if the init function pointer is not \p NULL.
- */
-static struct audio_format_handler afl[] = {
+/** The list of supported senders. */
+struct sender senders[] = {
{
- .name = "mp3",
- .init = mp3_init,
+ .name = "http",
+ .init = http_send_init,
},
{
- .name = "ogg",
-#ifdef HAVE_OGGVORBIS
- .init = ogg_init,
-#endif
+ .name = "dccp",
+ .init = dccp_send_init,
},
{
- .name = "aac",
-#ifdef HAVE_FAAD
- .init = aac_afh_init,
-#endif
+ .name = "udp",
+ .init = udp_send_init,
},
{
.name = NULL,
}
};
-static inline int next_audio_format(int format)
-{
- for (;;) {
- if (!afl[format].name)
- return format;
- format++;
- if (afl[format].init)
- return format;
- }
-
-}
-
-/** Iterate over each supported audio format. */
-#define FOR_EACH_AUDIO_FORMAT(i) for (i = 0; afl[i].name; i = next_audio_format(i))
+/** The possible states of the afs socket. */
+enum afs_socket_status {
+ /** Socket is inactive. */
+ AFS_SOCKET_READY,
+ /** Socket fd was included in the write fd set for select(). */
+ AFS_SOCKET_CHECK_FOR_WRITE,
+ /** vss wrote a request to the socket and waits for reply from afs. */
+ AFS_SOCKET_AFD_PENDING
+};
+/** The task structure for the virtual streaming system. */
+struct vss_task {
+ /** Copied from the -announce_time command line option. */
+ struct timeval announce_tv;
+ /** End of the announcing interval. */
+ struct timeval data_send_barrier;
+ /** End of the EOF interval. */
+ struct timeval eof_barrier;
+ /** Only used if --autoplay_delay was given. */
+ struct timeval autoplay_barrier;
+ /** Used for afs-server communication. */
+ int afs_socket;
+ /** The current state of \a afs_socket. */
+ enum afs_socket_status afsss;
+ /** The memory mapped audio file. */
+ char *map;
+ /** The size of the memory mapping. */
+ size_t mapsize;
+ /** Used by the scheduler. */
+ struct task *task;
+ /** Pointer to the header of the mapped audio file. */
+ char *header_buf;
+ /** Length of the audio file header. */
+ size_t header_len;
+ /** Time between audio file headers are sent. */
+ struct timeval header_interval;
+ /* Only used if afh supports dynamic chunks. */
+ void *afh_context;
+};
/**
- * check if vss status flag \a P (playing) is set
- *
- * \return greater than zero if playing, zero otherwise.
+ * The list of currently connected fec clients.
*
+ * Senders may use \ref vss_add_fec_client() to add entries to the list.
*/
-unsigned int vss_playing(void)
-{
- return mmd->new_vss_status_flags & VSS_PLAYING;
-}
+static struct list_head fec_client_list;
/**
- * check if \a N (next) status flag is set
+ * Data associated with one FEC group.
*
- * \return greater than zero if set, zero if not.
+ * A FEC group consists of a fixed number of slices and this number is given by
+ * the \a slices_per_group parameter of struct \ref fec_client_parms. Each FEC
+ * group contains a number of chunks of the current audio file.
*
+ * FEC slices directly correspond to the data packages sent by the paraslash
+ * senders that use FEC. Each slice is identified by its group number and its
+ * number within the group. All slices have the same size, but the last slice
+ * of the group may not be filled entirely.
*/
-unsigned int vss_next(void)
-{
- return mmd->new_vss_status_flags & VSS_NEXT;
-}
+struct fec_group {
+ /** The number of the FEC group. */
+ uint32_t num;
+ /** Number of bytes in this group. */
+ uint32_t bytes;
+ /** The first chunk of the current audio file belonging to the group. */
+ uint32_t first_chunk;
+ /** The number of chunks contained in this group. */
+ uint32_t num_chunks;
+ /** When the first chunk was sent. */
+ struct timeval start;
+ /** The duration of the full group. */
+ struct timeval duration;
+ /** The group duration divided by the number of slices. */
+ struct timeval slice_duration;
+ /** Group contains the audio file header that occupies that many slices. */
+ uint8_t num_header_slices;
+ /** Number of bytes per slice for this group. */
+ uint16_t slice_bytes;
+};
+
+/** A FEC client is always in one of these states. */
+enum fec_client_state {
+ FEC_STATE_NONE = 0, /**< not initialized and not enabled */
+ FEC_STATE_DISABLED, /**< temporarily disabled */
+ FEC_STATE_READY_TO_RUN /**< initialized and enabled */
+};
/**
- * check if a reposition request is pending
- *
- * \return greater than zero if true, zero otherwise.
- *
+ * Describes one connected FEC client.
*/
-unsigned int vss_repos(void)
-{
- return mmd->new_vss_status_flags & VSS_REPOS;
-}
+struct fec_client {
+ /** Current state of the client */
+ enum fec_client_state state;
+ /** The connected sender client (transport layer). */
+ struct sender_client *sc;
+ /** Parameters requested by the client. */
+ struct fec_client_parms *fcp;
+ /** Used by the core FEC code. */
+ struct fec_parms *parms;
+ /** The position of this client in the fec client list. */
+ struct list_head node;
+ /** When the first slice for this client was sent. */
+ struct timeval stream_start;
+ /** The first chunk sent to this FEC client. */
+ int first_stream_chunk;
+ /** Describes the current group. */
+ struct fec_group group;
+ /** The current slice. */
+ uint8_t current_slice_num;
+ /** The data to be FEC-encoded (point to a region within the mapped audio file). */
+ const unsigned char **src_data;
+ /** Last time an audio header was sent. */
+ struct timeval next_header_time;
+ /** Used for the last source pointer of an audio file. */
+ unsigned char *extra_src_buf;
+ /** Needed for the last slice of the audio file header. */
+ unsigned char *extra_header_buf;
+ /** Extra slices needed to store largest chunk + header. */
+ int num_extra_slices;
+ /** Contains the FEC-encoded data. */
+ unsigned char *enc_buf;
+ /** Maximal packet size. */
+ int mps;
+};
/**
- * check if the vss is currently paused
- *
- * \return greater than zero if paused, zero otherwise.
+ * Get the chunk time of the current audio file.
*
+ * \return A pointer to a struct containing the chunk time, or NULL,
+ * if currently no audio file is selected.
*/
-unsigned int vss_paused(void)
+struct timeval *vss_chunk_time(void)
{
- return !(mmd->new_vss_status_flags & VSS_NEXT)
- && !(mmd->new_vss_status_flags & VSS_PLAYING);
+ if (mmd->afd.afhi.chunk_tv.tv_sec == 0 &&
+ mmd->afd.afhi.chunk_tv.tv_usec == 0)
+ return NULL;
+ return &mmd->afd.afhi.chunk_tv;
}
/**
- * Get the name of the given audio format.
- *
- * \param i The audio format number.
+ * Write a fec header to a buffer.
*
- * This returns a pointer to statically allocated memory so it
- * must not be freed by the caller.
+ * \param buf The buffer to write to.
+ * \param h The fec header to write.
*/
-const char *audio_format_name(int i)
+static void write_fec_header(struct fec_client *fc, struct vss_task *vsst)
{
- //PARA_NOTICE_LOG("array size: %u¸ requested: %d\n", ARRAY_SIZE(afl), i);
- assert(i < 0 || i < ARRAY_SIZE(afl) - 1);
- return i >= 0? afl[i].name : "(none)";
+ char *buf = (char *)fc->enc_buf;
+ struct fec_group *g = &fc->group;
+ struct fec_client_parms *p = fc->fcp;
+
+ write_u32(buf, FEC_MAGIC);
+
+ write_u8(buf + 4, p->slices_per_group + fc->num_extra_slices);
+ write_u8(buf + 5, p->data_slices_per_group + fc->num_extra_slices);
+ write_u32(buf + 6, g->num_header_slices? vsst->header_len : 0);
+
+ write_u32(buf + 10, g->num);
+ write_u32(buf + 14, g->bytes);
+
+ write_u8(buf + 18, fc->current_slice_num);
+ write_u8(buf + 19, 0); /* unused */
+ write_u16(buf + 20, g->slice_bytes);
+ write_u8(buf + 22, g->first_chunk? 0 : 1);
+ write_u8(buf + 23, vsst->header_len? 1 : 0);
+ memset(buf + 24, 0, 8);
}
-/**
- * initialize the virtual streaming system
- *
- * Call the init functions of all supported audio format handlers and
- * initialize all supported senders.
- */
-void vss_init(void)
+static bool need_audio_header(struct fec_client *fc, struct vss_task *vsst)
{
- int i;
- char *hn = para_hostname(), *home = para_homedir();
- long unsigned announce_time = conf.announce_time_arg > 0?
- conf.announce_time_arg : 300,
- autoplay_delay = conf.autoplay_delay_arg > 0?
- conf.autoplay_delay_arg : 0;
-
- PARA_DEBUG_LOG("supported audio formats: %s\n",
- SUPPORTED_AUDIO_FORMATS);
- FOR_EACH_AUDIO_FORMAT(i) {
- PARA_NOTICE_LOG("initializing %s handler\n",
- afl[i].name);
- afl[i].init(&afl[i]);
- }
- ms2tv(announce_time, &announce_tv);
- PARA_INFO_LOG("announce timeval: %lums\n", tv2ms(&announce_tv));
- for (i = 0; senders[i].name; i++) {
- PARA_NOTICE_LOG("initializing %s sender\n", senders[i].name);
- senders[i].init(&senders[i]);
+ if (!mmd->current_chunk) {
+ tv_add(now, &vsst->header_interval, &fc->next_header_time);
+ return false;
}
- free(hn);
- free(home);
- if (conf.autoplay_given) {
- struct timeval now, tmp;
- mmd->vss_status_flags |= VSS_PLAYING;
- mmd->new_vss_status_flags |= VSS_PLAYING;
- gettimeofday(&now, NULL);
- ms2tv(autoplay_delay, &tmp);
- tv_add(&now, &tmp, &autoplay_barrier);
+ if (!vsst->header_buf)
+ return false;
+ if (vsst->header_len == 0)
+ return false;
+ if (fc->group.num > 0) {
+ if (!fc->fcp->need_periodic_header)
+ return false;
+ if (tv_diff(&fc->next_header_time, now, NULL) > 0)
+ return false;
}
+ tv_add(now, &vsst->header_interval, &fc->next_header_time);
+ return true;
}
-static int get_file_info(int i)
+static bool need_data_slices(struct fec_client *fc, struct vss_task *vsst)
{
- return afl[i].get_file_info(map, mmd->size, &mmd->afi);
+ if (fc->group.num > 0)
+ return true;
+ if (!vsst->header_buf)
+ return true;
+ if (vsst->header_len == 0)
+ return true;
+ if (fc->fcp->need_periodic_header)
+ return true;
+ return false;
}
-/**
- * guess the audio format judging from filename
- *
- * \param name the filename
- *
- * \return This function returns -1 if it has no idea what kind of audio
- * file this might be. Otherwise the (non-negative) number of the audio format
- * is returned.
- */
-int guess_audio_format(const char *name)
+static int num_slices(long unsigned bytes, int max_payload, int rs)
{
- int i,j, len = strlen(name);
+ int ret;
- FOR_EACH_AUDIO_FORMAT(i) {
- for (j = 0; afl[i].suffixes[j]; j++) {
- const char *p = afl[i].suffixes[j];
- int plen = strlen(p);
- if (len < plen + 1)
- continue;
- if (name[len - plen - 1] != '.')
- continue;
- if (strcasecmp(name + len - plen, p))
- continue;
-// PARA_DEBUG_LOG("might be %s\n", audio_format_name(i));
- return i;
- }
+ assert(max_payload > 0);
+ assert(rs > 0);
+ ret = DIV_ROUND_UP(bytes, max_payload);
+ if (ret + rs > 255)
+ return -E_BAD_CT;
+ return ret;
+}
+
+/* set group start and group duration */
+static void set_group_timing(struct fec_client *fc, struct vss_task *vsst)
+{
+ struct fec_group *g = &fc->group;
+ struct timeval *chunk_tv = vss_chunk_time();
+
+ if (!need_data_slices(fc, vsst))
+ ms2tv(200, &g->duration);
+ else
+ tv_scale(g->num_chunks, chunk_tv, &g->duration);
+ tv_divide(fc->fcp->slices_per_group + fc->num_extra_slices,
+ &g->duration, &g->slice_duration);
+ PARA_DEBUG_LOG("durations (group/chunk/slice): %lu/%lu/%lu\n",
+ tv2ms(&g->duration), tv2ms(chunk_tv), tv2ms(&g->slice_duration));
+}
+
+static int initialize_fec_client(struct fec_client *fc, struct vss_task *vsst)
+{
+ int k, n, ret;
+ int hs, ds, rs; /* header/data/redundant slices */
+ struct fec_client_parms *fcp = fc->fcp;
+
+ /* set mps */
+ if (fcp->init_fec) {
+ /*
+ * Set the maximum slice size to the Maximum Packet Size if the
+ * transport protocol allows to determine this value. The user
+ * can specify a slice size up to this value.
+ */
+ ret = fcp->init_fec(fc->sc);
+ if (ret < 0)
+ return ret;
+ fc->mps = ret;
+ } else
+ fc->mps = generic_max_transport_msg_size(fc->sc->fd);
+ if (fc->mps <= FEC_HEADER_SIZE)
+ return -ERRNO_TO_PARA_ERROR(EINVAL);
+
+ rs = fc->fcp->slices_per_group - fc->fcp->data_slices_per_group;
+ ret = num_slices(vsst->header_len, fc->mps - FEC_HEADER_SIZE, rs);
+ if (ret < 0)
+ return ret;
+ hs = ret;
+ ret = num_slices(mmd->afd.max_chunk_size, fc->mps - FEC_HEADER_SIZE, rs);
+ if (ret < 0)
+ return ret;
+ ds = ret;
+ if (fc->fcp->need_periodic_header)
+ k = hs + ds;
+ else
+ k = PARA_MAX(hs, ds);
+ if (k < fc->fcp->data_slices_per_group)
+ k = fc->fcp->data_slices_per_group;
+ fc->num_extra_slices = k - fc->fcp->data_slices_per_group;
+ n = k + rs;
+ fec_free(fc->parms);
+ ret = fec_new(k, n, &fc->parms);
+ if (ret < 0)
+ return ret;
+ PARA_INFO_LOG("mps: %d, k: %d, n: %d, extra slices: %d\n",
+ fc->mps, k, n, fc->num_extra_slices);
+ fc->src_data = para_realloc(fc->src_data, k * sizeof(char *));
+ fc->enc_buf = para_realloc(fc->enc_buf, fc->mps);
+ fc->extra_src_buf = para_realloc(fc->extra_src_buf, fc->mps);
+ fc->extra_header_buf = para_realloc(fc->extra_header_buf, fc->mps);
+
+ fc->state = FEC_STATE_READY_TO_RUN;
+ fc->next_header_time.tv_sec = 0;
+ fc->stream_start = *now;
+ fc->first_stream_chunk = mmd->current_chunk;
+ return 1;
+}
+
+static int vss_get_chunk(int chunk_num, struct vss_task *vsst,
+ char **buf, size_t *sz)
+{
+ int ret;
+
+ /*
+ * Chunk zero is special for header streams: It is the first portion of
+ * the audio file which consists of the audio file header. It may be
+ * arbitrary large due to embedded meta data. Audio format handlers may
+ * replace the header by a stripped one with meta data omitted which is
+ * of bounded size. We always use the stripped header for streaming
+ * rather than the unmodified header (chunk zero).
+ */
+ if (chunk_num == 0 && vsst->header_len > 0) {
+ assert(vsst->header_buf);
+ *buf = vsst->header_buf; /* stripped header */
+ *sz = vsst->header_len;
+ return 0;
+ }
+ ret = afh_get_chunk(chunk_num, &mmd->afd.afhi,
+ mmd->afd.audio_format_id, vsst->map, vsst->mapsize,
+ (const char **)buf, sz, &vsst->afh_context);
+ if (ret < 0) {
+ *buf = NULL;
+ *sz = 0;
}
- return -E_BAD_AUDIO_FILE_SUFFIX;
+ return ret;
}
-static int get_audio_format(int omit)
+static int compute_group_size(struct vss_task *vsst, struct fec_group *g,
+ int max_bytes)
{
- int i;
+ char *buf;
+ size_t len;
+ int ret, i, max_chunks = PARA_MAX(1LU, 150 / tv2ms(vss_chunk_time()));
- FOR_EACH_AUDIO_FORMAT(i) {
- if (i == omit)
- continue;
- if (get_file_info(i) > 0)
- return i;
+ if (g->first_chunk == 0) {
+ g->num_chunks = 1;
+ ret = vss_get_chunk(0, vsst, &buf, &len);
+ if (ret < 0)
+ return ret;
+ g->bytes = len;
+ return 0;
}
- return -E_AUDIO_FORMAT;
+
+ g->num_chunks = 0;
+ g->bytes = 0;
+ /*
+ * Include chunks into the group until the group duration is at least
+ * 150ms. For ogg and wma, a single chunk's duration (ogg page/wma
+ * super frame) is already larger than 150ms, so a FEC group consists
+ * of exactly one chunk for these audio formats.
+ */
+ for (i = 0;; i++) {
+ int chunk_num = g->first_chunk + i;
+
+ if (g->bytes > 0 && i >= max_chunks) /* duration limit */
+ break;
+ if (chunk_num >= mmd->afd.afhi.chunks_total) /* eof */
+ break;
+ ret = vss_get_chunk(chunk_num, vsst, &buf, &len);
+ if (ret < 0)
+ return ret;
+ if (g->bytes + len > max_bytes)
+ break;
+ /* Include this chunk */
+ g->bytes += len;
+ g->num_chunks++;
+ }
+ assert(g->num_chunks);
+ return 1;
}
-/**
- * Call get_file_info() to obtain an afhi structure.
+/*
+ * Compute the slice size of the next group.
+ *
+ * The FEC parameters n and k are fixed but the slice size varies per
+ * FEC group. We'd like to choose slices as small as possible to avoid
+ * unnecessary FEC calculations but large enough to guarantee that the
+ * k data slices suffice to encode the header (if needed) and the data
+ * chunk(s).
+ *
+ * Once we know the payload of the next group, we define the number s
+ * of bytes per slice for this group by
+ *
+ * s = ceil(payload / k)
+ *
+ * However, for header streams, computing s is more complicated since no
+ * overlapping of header and data slices is possible. Hence we have k >=
+ * 2 and s must satisfy
+ *
+ * (*) ceil(h / s) + ceil(d / s) <= k
*
- * \param path The full path of the audio file.
- * \param data Pointer to the contents of the (mapped) file.
- * \param size The file size in bytes.
- * \param afhi Result pointer.
+ * where h and d are payload of the header and the data chunk(s)
+ * respectively. In general there is no value for s such that (*)
+ * becomes an equality, for example if h = 4000, d = 5000 and k = 10.
*
- * \return The number of the audio format on success, \p -E_AUDIO_FORMAT if no
- * compiled in audio format handler is able to handler the file.
+ * We use the following approach for computing a suitable value for s:
*
- * This function tries to find an audio format handler that can interpret the
- * file given by \a data and \a size.
+ * Let
+ * k1 := ceil(k * min(h, d) / (h + d)),
+ * k2 := k - k1.
*
- * It first tries to determine the audio format from the filename given by \a
- * path. If this doesn't work, all other audio format handlers are tried until
- * one is found that can handle the file.
+ * Note that k >= 2 implies k1 > 0 and k2 > 0, so
+ *
+ * s := max(ceil(min(h, d) / k1), ceil(max(h, d) / k2))
+ *
+ * is well-defined. Inequality (*) holds for this value of s since k1
+ * slices suffice to store min(h, d) while k2 slices suffice to store
+ * max(h, d), i.e. the first addent of (*) is bounded by k1 and the
+ * second by k2.
+ *
+ * For the above example we obtain
+ *
+ * k1 = ceil(10 * 4000 / 9000) = 5, k2 = 5,
+ * s = max(4000 / 5, 5000 / 5) = 1000,
+ *
+ * which is optimal since a slice size of 999 bytes would already require
+ * 11 slices.
*/
-int compute_afhi(const char *path, char *data, size_t size,
- struct audio_format_info *afhi)
+static int compute_slice_size(struct fec_client *fc, struct vss_task *vsst)
{
- int ret, i, format = guess_audio_format(path);
+ struct fec_group *g = &fc->group;
+ int k = fc->fcp->data_slices_per_group + fc->num_extra_slices;
+ int n = fc->fcp->slices_per_group + fc->num_extra_slices;
+ int ret, k1, k2, h, d, min, max, sum;
+ int max_slice_bytes = fc->mps - FEC_HEADER_SIZE;
+ int max_group_bytes;
- if (format >= 0) {
- ret = afl[format].get_file_info(data, size, afhi);
- if (ret >= 0)
- return format;
+ if (!need_audio_header(fc, vsst)) {
+ max_group_bytes = k * max_slice_bytes;
+ g->num_header_slices = 0;
+ ret = compute_group_size(vsst, g, max_group_bytes);
+ if (ret < 0)
+ return ret;
+ g->slice_bytes = DIV_ROUND_UP(g->bytes, k);
+ if (g->slice_bytes == 0)
+ g->slice_bytes = 1;
+ return 1;
}
- FOR_EACH_AUDIO_FORMAT(i) {
- if (i == format) /* we already tried this one to no avail */
- continue;
- ret = afl[i].get_file_info(data, size, afhi);
- if (ret >= 0)
- return i;
- PARA_WARNING_LOG("%s\n", PARA_STRERROR(-ret));
+ if (!need_data_slices(fc, vsst)) {
+ g->bytes = 0;
+ g->num_chunks = 0;
+ g->slice_bytes = DIV_ROUND_UP(vsst->header_len, k);
+ g->num_header_slices = k;
+ return 1;
}
- return -E_AUDIO_FORMAT;
-}
+ h = vsst->header_len;
+ max_group_bytes = (k - num_slices(h, max_slice_bytes, n - k))
+ * max_slice_bytes;
+ ret = compute_group_size(vsst, g, max_group_bytes);
+ if (ret < 0)
+ return ret;
+ d = g->bytes;
+ if (d == 0) {
+ g->slice_bytes = DIV_ROUND_UP(h, k);
+ ret = num_slices(vsst->header_len, g->slice_bytes, n - k);
+ if (ret < 0)
+ return ret;
+ g->num_header_slices = ret;
+ return 1;
+ }
+ min = PARA_MIN(h, d);
+ max = PARA_MAX(h, d);
+ sum = h + d;
+ k1 = DIV_ROUND_UP(k * min, sum);
+ k2 = k - k1;
+ assert(k1 > 0);
+ assert(k2 > 0);
-/*
- * upddate shared mem
- */
-static int update_mmd(void)
-{
- int i;
+ g->slice_bytes = PARA_MAX(DIV_ROUND_UP(min, k1), DIV_ROUND_UP(max, k2));
+ /*
+ * This value of s := g->slice_bytes satisfies inequality (*) above,
+ * but it might be larger than max_slice_bytes. However, we know that
+ * max_slice_bytes are sufficient to store header and data, so:
+ */
+ g->slice_bytes = PARA_MIN((int)g->slice_bytes, max_slice_bytes);
- i = guess_audio_format(mmd->filename);
- if (i < 0 || get_file_info(i) < 0)
- i = get_audio_format(i);
- if (i < 0)
- return i;
- mmd->audio_format = i;
- mmd->chunks_sent = 0;
- mmd->current_chunk = 0;
- mmd->offset = 0;
- mmd->events++;
+ ret = num_slices(vsst->header_len, g->slice_bytes, n - k);
+ if (ret < 0)
+ return ret;
+ g->num_header_slices = ret;
return 1;
}
-static void vss_get_audio_file(void)
+static int setup_next_fec_group(struct fec_client *fc, struct vss_task *vsst)
{
- char **sl = selectors[mmd->selector_num].get_audio_file_list(10);
- int i;
- struct stat file_status;
-
- if (!sl)
- goto err_out;
- for (i = 0; sl[i]; i++) {
- struct timeval now;
- PARA_INFO_LOG("trying %s\n", sl[i]);
- if (strlen(sl[i]) >= _POSIX_PATH_MAX)
- continue;
- audio_file = open(sl[i], O_RDONLY);
- if (audio_file < 0)
- continue;
- if (fstat(audio_file, &file_status) == -1 ||
- !file_status.st_size) {
- close(audio_file);
+ int ret, i, k, n, data_slices;
+ size_t len;
+ char *buf, *p;
+ struct fec_group *g = &fc->group;
+
+ if (fc->state == FEC_STATE_NONE) {
+ ret = initialize_fec_client(fc, vsst);
+ if (ret < 0)
+ return ret;
+ g->first_chunk = mmd->current_chunk;
+ g->num = 0;
+ g->start = *now;
+ } else {
+ struct timeval tmp;
+ if (g->first_chunk + g->num_chunks >= mmd->afd.afhi.chunks_total)
+ return 0;
+ /*
+ * Start and duration of this group depend only on the previous
+ * group. Compute the new group start as g->start += g->duration.
+ */
+ tmp = g->start;
+ tv_add(&tmp, &g->duration, &g->start);
+ set_group_timing(fc, vsst);
+ g->first_chunk += g->num_chunks;
+ g->num++;
+ }
+ k = fc->fcp->data_slices_per_group + fc->num_extra_slices;
+ n = fc->fcp->slices_per_group + fc->num_extra_slices;
+
+ compute_slice_size(fc, vsst);
+ assert(g->slice_bytes > 0);
+ ret = num_slices(g->bytes, g->slice_bytes, n - k);
+ if (ret < 0)
+ return ret;
+ data_slices = ret;
+ assert(g->num_header_slices + data_slices <= k);
+ fc->current_slice_num = 0;
+ if (g->num == 0)
+ set_group_timing(fc, vsst);
+ /* setup header slices */
+ buf = vsst->header_buf;
+ for (i = 0; i < g->num_header_slices; i++) {
+ uint32_t payload_size;
+ if (buf + g->slice_bytes <= vsst->header_buf + vsst->header_len) {
+ fc->src_data[i] = (const unsigned char *)buf;
+ buf += g->slice_bytes;
continue;
}
- mmd->size = file_status.st_size;
- mmd->mtime = file_status.st_mtime;
- map = para_mmap(mmd->size, PROT_READ, MAP_PRIVATE,
- audio_file, 0);
- strcpy(mmd->filename, sl[i]);
- mmd->afi.header_len = 0; /* default: no header */
- if (update_mmd() < 0) { /* invalid file */
- close(audio_file);
- munmap(map, mmd->size);
- map = NULL;
- continue;
+ /*
+ * Can not use vss->header_buf for this slice as it
+ * goes beyond the buffer. This slice will not be fully
+ * used.
+ */
+ payload_size = vsst->header_buf + vsst->header_len - buf;
+ memcpy(fc->extra_header_buf, buf, payload_size);
+ if (payload_size < g->slice_bytes)
+ memset(fc->extra_header_buf + payload_size, 0,
+ g->slice_bytes - payload_size);
+ /*
+ * There might be more than one header slice to fill although
+ * only the first one will be used. Set all header slices to
+ * our extra buffer.
+ */
+ while (i < g->num_header_slices)
+ fc->src_data[i++] = fc->extra_header_buf;
+ break; /* we don't want i to be increased. */
+ }
+
+ /*
+ * Setup data slices. Note that for ogg streams chunk 0 points to a
+ * buffer on the heap rather than to the mapped audio file.
+ */
+ ret = vss_get_chunk(g->first_chunk, vsst, &buf, &len);
+ if (ret < 0)
+ return ret;
+ for (p = buf; i < g->num_header_slices + data_slices; i++) {
+ if (p + g->slice_bytes > buf + g->bytes) {
+ /*
+ * We must make a copy for this slice since using p
+ * directly would exceed the buffer.
+ */
+ uint32_t payload_size = buf + g->bytes - p;
+ assert(payload_size + FEC_HEADER_SIZE <= fc->mps);
+ memcpy(fc->extra_src_buf, p, payload_size);
+ if (payload_size < g->slice_bytes)
+ memset(fc->extra_src_buf + payload_size, 0,
+ g->slice_bytes - payload_size);
+ fc->src_data[i] = fc->extra_src_buf;
+ i++;
+ break;
}
- mmd->num_played++;
- if (selectors[mmd->selector_num].update_audio_file)
- selectors[mmd->selector_num].update_audio_file(sl[i]);
- PARA_NOTICE_LOG("next audio file: %s\n", mmd->filename);
- mmd->new_vss_status_flags &= (~VSS_NEXT);
- gettimeofday(&now, NULL);
- tv_add(&now, &announce_tv, &data_send_barrier);
- goto free;
+ fc->src_data[i] = (const unsigned char *)p;
+ p += g->slice_bytes;
}
-err_out:
- PARA_ERROR_LOG("%s", "no valid files found\n");
- mmd->new_vss_status_flags = VSS_NEXT;
-free:
- if (sl) {
- for (i = 0; sl[i]; i++)
- free(sl[i]);
- free(sl);
+ if (i < k) {
+ /* use arbitrary data for all remaining slices */
+ buf = vsst->map;
+ for (; i < k; i++)
+ fc->src_data[i] = (const unsigned char *)buf;
}
+ PARA_DEBUG_LOG("FEC group %u: %u chunks (%u - %u), %u bytes\n",
+ g->num, g->num_chunks, g->first_chunk,
+ g->first_chunk + g->num_chunks - 1, g->bytes
+ );
+ PARA_DEBUG_LOG("slice_bytes: %d, %d header slices, %d data slices\n",
+ g->slice_bytes, g->num_header_slices, data_slices
+ );
+ return 1;
}
-static int chk_barrier(const char *bname, const struct timeval *now,
- const struct timeval *barrier, struct timeval *diff,
- int print_log)
+static int compute_next_fec_slice(struct fec_client *fc, struct vss_task *vsst)
{
- long ms;
+ if (fc->state == FEC_STATE_NONE || fc->current_slice_num
+ == fc->fcp->slices_per_group + fc->num_extra_slices) {
+ int ret = setup_next_fec_group(fc, vsst);
+ if (ret == 0)
+ return 0;
+ if (ret < 0) {
+ PARA_ERROR_LOG("%s\n", para_strerror(-ret));
+ PARA_ERROR_LOG("FEC client temporarily disabled\n");
+ fc->state = FEC_STATE_DISABLED;
+ return ret;
+ }
+ }
+ write_fec_header(fc, vsst);
+ fec_encode(fc->parms, fc->src_data, fc->enc_buf + FEC_HEADER_SIZE,
+ fc->current_slice_num, fc->group.slice_bytes);
+ return 1;
+}
- if (tv_diff(now, barrier, diff) > 0)
- return 1;
- ms = tv2ms(diff);
- if (print_log && ms)
- PARA_DEBUG_LOG("%s barrier: %lims left\n", bname, ms);
- return -1;
+/**
+ * Return a buffer that marks the end of the stream.
+ *
+ * \param buf Result pointer.
+ * \return The length of the eof buffer.
+ *
+ * This is used for (multicast) udp streaming where closing the socket on the
+ * sender might not give rise to an eof condition at the peer.
+ */
+size_t vss_get_fec_eof_packet(const char **buf)
+{
+ static const char fec_eof_packet[FEC_HEADER_SIZE] = FEC_EOF_PACKET;
+ *buf = fec_eof_packet;
+ return FEC_HEADER_SIZE;
}
-static void vss_next_chunk_time(struct timeval *due)
+/**
+ * Add one entry to the list of active fec clients.
+ *
+ * \param sc Generic sender_client data of the transport layer.
+ * \param fcp FEC parameters as supplied by the transport layer.
+ *
+ * \return Newly allocated fec_client struct.
+ */
+struct fec_client *vss_add_fec_client(struct sender_client *sc,
+ struct fec_client_parms *fcp)
{
- struct timeval tmp;
+ struct fec_client *fc = para_calloc(sizeof(*fc));
- tv_scale(mmd->chunks_sent, &mmd->afi.chunk_tv, &tmp);
- tv_add(&tmp, &mmd->stream_start, due);
+ fc->sc = sc;
+ fc->fcp = fcp;
+ para_list_add(&fc->node, &fec_client_list);
+ return fc;
+}
+
+/**
+ * Remove one entry from the list of active fec clients.
+ *
+ * \param fc The client to be removed.
+ */
+void vss_del_fec_client(struct fec_client *fc)
+{
+ list_del(&fc->node);
+ free(fc->src_data);
+ free(fc->enc_buf);
+ free(fc->extra_src_buf);
+ free(fc->extra_header_buf);
+ fec_free(fc->parms);
+ free(fc);
}
/*
- * != NULL: timeout for next chunk
- * NULL: nothing to do
+ * Compute if/when next slice is due. If it isn't due yet and \a diff is
+ * not \p Null, compute the time difference next - now, where
+ *
+ * next = stream_start + (first_group_chunk - first_stream_chunk)
+ * * chunk_time + slice_num * slice_time
*/
-static struct timeval *vss_compute_timeout(void)
+static int next_slice_is_due(struct fec_client *fc, struct timeval *diff)
{
- static struct timeval the_timeout;
- struct timeval now, next_chunk;
+ struct timeval tmp, next;
+ int ret;
- if (vss_next() && mmd->audio_format >= 0) {
- /* only sleep a bit, nec*/
- the_timeout.tv_sec = 0;
- the_timeout.tv_usec = 100;
- return &the_timeout;
- }
- gettimeofday(&now, NULL);
- if (chk_barrier("autoplay_delay", &now, &autoplay_barrier,
- &the_timeout, 1) < 0)
- return &the_timeout;
- if (chk_barrier("eof", &now, &eof_barrier, &the_timeout, 1) < 0)
- return &the_timeout;
- if (chk_barrier("data send", &now, &data_send_barrier,
- &the_timeout, 1) < 0)
- return &the_timeout;
- if (mmd->audio_format < 0 || !vss_playing() || !map)
- return NULL;
- vss_next_chunk_time(&next_chunk);
- if (chk_barrier(afl[mmd->audio_format].name, &now, &next_chunk,
- &the_timeout, 0) < 0)
- return &the_timeout;
- /* chunk is due or bof */
- the_timeout.tv_sec = 0;
- the_timeout.tv_usec = 0;
- return &the_timeout;
+ if (fc->state == FEC_STATE_NONE)
+ return 1;
+ tv_scale(fc->current_slice_num, &fc->group.slice_duration, &tmp);
+ tv_add(&tmp, &fc->group.start, &next);
+ ret = tv_diff(&next, now, diff);
+ return ret < 0? 1 : 0;
}
-static void vss_eof(void)
+static void set_eof_barrier(struct vss_task *vsst)
{
- struct timeval now;
- int i;
- char *tmp;
+ struct fec_client *fc;
+ struct timeval timeout = {1, 0}, *chunk_tv = vss_chunk_time();
- if (!map) {
- for (i = 0; senders[i].name; i++)
- senders[i].shutdown_clients();
- return;
+ if (!chunk_tv)
+ goto out;
+ list_for_each_entry(fc, &fec_client_list, node) {
+ struct timeval group_duration;
+
+ if (fc->state != FEC_STATE_READY_TO_RUN)
+ continue;
+ tv_scale(fc->group.num_chunks, chunk_tv, &group_duration);
+ if (tv_diff(&timeout, &group_duration, NULL) < 0)
+ timeout = group_duration;
}
- gettimeofday(&now, NULL);
- tv_add(&mmd->afi.eof_tv, &now, &eof_barrier);
- munmap(map, mmd->size);
- map = NULL;
- close(audio_file);
- mmd->audio_format = -1;
- mmd->chunks_sent = 0;
- mmd->offset = 0;
- mmd->afi.seconds_total = 0;
- free(mmd->afi.chunk_table);
- mmd->afi.chunk_table = NULL;
- tmp = make_message("%s:\n%s:\n%s:\n", status_item_list[SI_AUDIO_INFO1],
- status_item_list[SI_AUDIO_INFO2], status_item_list[SI_AUDIO_INFO3]);
- strcpy(mmd->afi.info_string, tmp);
- free(tmp);
- tmp = make_message("%s:\n%s:\n%s:\n", status_item_list[SI_DBINFO1],
- status_item_list[SI_DBINFO2], status_item_list[SI_DBINFO3]);
- strcpy(mmd->selector_info, tmp);
- free(tmp);
- mmd->filename[0] = '\0';
- mmd->size = 0;
- mmd->events++;
+out:
+ tv_add(now, &timeout, &vsst->eof_barrier);
}
/**
- * Get the header of the current audio file.
+ * Check if vss status flag \a P (playing) is set.
*
- * \param header_len the length of the header is stored here
- *
- * \return a pointer to a buffer containing the header, or NULL, if no audio
- * file is selected or if the current audio format does not need special header
- * treamtment.
+ * \return Greater than zero if playing, zero otherwise.
*
*/
-char *vss_get_header(size_t *header_len)
+unsigned int vss_playing(void)
{
- if (mmd->audio_format < 0 || !map || !mmd->afi.header_len)
- return NULL;
- *header_len = mmd->afi.header_len;
- return map + mmd->afi.header_offset;
+ return mmd->new_vss_status_flags & VSS_PLAYING;
}
/**
- * get the list of all supported audio formats
+ * Check if the \a N (next) status flag is set.
+ *
+ * \return Greater than zero if set, zero if not.
*
- * \return a space separated list of all supported audio formats
- * It is not allocated at runtime, i.e. there is no need to free
- * the returned string in the caller.
*/
-const char *supported_audio_formats(void)
+unsigned int vss_next(void)
{
- return SUPPORTED_AUDIO_FORMATS;
+ return mmd->new_vss_status_flags & VSS_NEXT;
}
/**
- * get the chunk time of the current audio file
+ * Check if a reposition request is pending.
+ *
+ * \return Greater than zero if true, zero otherwise.
*
- * \return a pointer to a struct containing the chunk time, or NULL,
- * if currently no audio file is selected.
*/
-struct timeval *vss_chunk_time(void)
+unsigned int vss_repos(void)
{
- if (mmd->audio_format < 0)
- return NULL;
- return &mmd->afi.chunk_tv;
+ return mmd->new_vss_status_flags & VSS_REPOS;
}
/**
- * compute the timeout for para_server's main select-loop
+ * Check if the vss is currently paused.
*
- * This function gets called from para_server to determine the timeout value
- * for its main select loop.
+ * \return Greater than zero if paused, zero otherwise.
*
- * Before the timeout is computed, the current vss status flags are evaluated
- * and acted upon by calling appropriate functions from the lower layers.
- * Possible actions include
+ */
+unsigned int vss_paused(void)
+{
+ return !(mmd->new_vss_status_flags & VSS_NEXT)
+ && !(mmd->new_vss_status_flags & VSS_PLAYING);
+}
+
+/**
+ * Check if the vss is currently stopped.
*
- * - request a new file list from the current audio file selector
- * - shutdown of all senders (stop/pause command)
- * - reposition the stream (ff/jmp command)
+ * \return Greater than zero if paused, zero otherwise.
*
- * \return A pointer to a struct timeval containing the timeout for the next
- * chunk of data to be sent, or NULL if we're not sending right now.
*/
-struct timeval *vss_preselect(void)
-{
- struct audio_format_handler *af = NULL;
- int i, format;
- struct timeval *ret;
-again:
- format = mmd->audio_format;
- if (format >= 0)
- af = afl + format;
- else
- for (i = 0; senders[i].name; i++)
- senders[i].shutdown_clients();
- if (vss_next() && af) {
- vss_eof();
- return vss_compute_timeout();
- }
- if (vss_paused() || vss_repos()) {
- for (i = 0; senders[i].name; i++)
- senders[i].shutdown_clients();
- if (af) {
- struct timeval now;
- gettimeofday(&now, NULL);
- if (!vss_paused() || mmd->chunks_sent)
- tv_add(&mmd->afi.eof_tv, &now, &eof_barrier);
- if (vss_repos())
- tv_add(&now, &announce_tv, &data_send_barrier);
- if (mmd->new_vss_status_flags & VSS_NOMORE)
- mmd->new_vss_status_flags = VSS_NEXT;
- }
- mmd->chunks_sent = 0;
- }
- if (vss_repos()) {
- mmd->new_vss_status_flags &= ~(VSS_REPOS);
- mmd->current_chunk = mmd->repos_request;
- }
- ret = vss_compute_timeout();
- if (!ret && !map && vss_playing() &&
- !(mmd->new_vss_status_flags & VSS_NOMORE)) {
- PARA_DEBUG_LOG("%s", "ready and playing, but no audio file\n");
- vss_get_audio_file();
- goto again;
+unsigned int vss_stopped(void)
+{
+ return (mmd->new_vss_status_flags & VSS_NEXT)
+ && !(mmd->new_vss_status_flags & VSS_PLAYING);
+}
+
+static int chk_barrier(const char *bname, const struct timeval *barrier,
+ struct timeval *diff, int print_log)
+{
+ long ms;
+
+ if (tv_diff(now, barrier, diff) > 0)
+ return 1;
+ ms = tv2ms(diff);
+ if (print_log && ms)
+ PARA_DEBUG_LOG("%s barrier: %lims left\n", bname, ms);
+ return -1;
+}
+
+static void vss_compute_timeout(struct sched *s, struct vss_task *vsst)
+{
+ struct timeval tv;
+ struct fec_client *fc;
+
+ if (!vss_playing() || !vsst->map)
+ return;
+ if (vss_next() && vsst->map) /* only sleep a bit, nec*/
+ return sched_request_timeout_ms(100, s);
+
+ /* Each of these barriers must have passed until we may proceed */
+ if (sched_request_barrier(&vsst->autoplay_barrier, s) == 1)
+ return;
+ if (sched_request_barrier(&vsst->eof_barrier, s) == 1)
+ return;
+ if (sched_request_barrier(&vsst->data_send_barrier, s) == 1)
+ return;
+ /*
+ * Compute the select timeout as the minimal time until the next
+ * chunk/slice is due for any client.
+ */
+ compute_chunk_time(mmd->chunks_sent, &mmd->afd.afhi.chunk_tv,
+ &mmd->stream_start, &tv);
+ if (sched_request_barrier_or_min_delay(&tv, s) == 0)
+ return;
+ list_for_each_entry(fc, &fec_client_list, node) {
+ if (fc->state != FEC_STATE_READY_TO_RUN)
+ continue;
+ if (next_slice_is_due(fc, &tv))
+ return sched_min_delay(s);
+ sched_request_timeout(&tv, s);
}
- return ret;
}
-static void get_chunk(long unsigned chunk_num, char **buf, size_t *len)
+static void vss_eof(struct vss_task *vsst)
{
- size_t pos = mmd->afi.chunk_table[chunk_num];
- *buf = map + pos;
- *len = mmd->afi.chunk_table[chunk_num + 1] - pos;
+
+ if (!vsst->map)
+ return;
+ if (mmd->new_vss_status_flags & VSS_NOMORE)
+ mmd->new_vss_status_flags = VSS_NEXT;
+ set_eof_barrier(vsst);
+ afh_free_header(vsst->header_buf, mmd->afd.audio_format_id);
+ vsst->header_buf = NULL;
+ para_munmap(vsst->map, vsst->mapsize);
+ vsst->map = NULL;
+ mmd->chunks_sent = 0;
+ //mmd->offset = 0;
+ mmd->afd.afhi.seconds_total = 0;
+ mmd->afd.afhi.chunk_tv.tv_sec = 0;
+ mmd->afd.afhi.chunk_tv.tv_usec = 0;
+ free(mmd->afd.afhi.chunk_table);
+ mmd->afd.afhi.chunk_table = NULL;
+ vsst->mapsize = 0;
+ afh_close(vsst->afh_context, mmd->afd.audio_format_id);
+ vsst->afh_context = NULL;
+ mmd->events++;
}
-/**
- * Get the data of the given chunk.
- *
- * \param chunk_num The number of the desired chunk.
- * \param buf Chunk data.
- * \param len Chunk length in bytes.
- *
- * \return Positive on success, negative on errors.
- */
-int vss_get_chunk(long unsigned chunk_num, char **buf, size_t *len)
+static int need_to_request_new_audio_file(struct vss_task *vsst)
{
- if (mmd->audio_format < 0 || !map || !vss_playing())
- return -E_CHUNK;
- if (chunk_num >= mmd->afi.chunks_total)
- return -E_CHUNK;
- get_chunk(chunk_num, buf, len);
+ struct timeval diff;
+
+ if (vsst->map) /* have audio file */
+ return 0;
+ if (!vss_playing()) /* don't need one */
+ return 0;
+ if (mmd->new_vss_status_flags & VSS_NOMORE)
+ return 0;
+ if (vsst->afsss == AFS_SOCKET_AFD_PENDING) /* already requested one */
+ return 0;
+ if (chk_barrier("autoplay_delay", &vsst->autoplay_barrier,
+ &diff, 1) < 0)
+ return 0;
return 1;
}
+static void set_mmd_offset(void)
+{
+ struct timeval offset;
+ tv_scale(mmd->current_chunk, &mmd->afd.afhi.chunk_tv, &offset);
+ mmd->offset = tv2ms(&offset);
+}
+
+static void vss_pre_select(struct sched *s, void *context)
+{
+ int i;
+ struct vss_task *vsst = context;
+
+ if (need_to_request_new_audio_file(vsst)) {
+ PARA_DEBUG_LOG("ready and playing, but no audio file\n");
+ para_fd_set(vsst->afs_socket, &s->wfds, &s->max_fileno);
+ vsst->afsss = AFS_SOCKET_CHECK_FOR_WRITE;
+ } else
+ para_fd_set(vsst->afs_socket, &s->rfds, &s->max_fileno);
+ for (i = 0; senders[i].name; i++) {
+ if (!senders[i].pre_select)
+ continue;
+ senders[i].pre_select(&s->max_fileno, &s->rfds, &s->wfds);
+ }
+ vss_compute_timeout(s, vsst);
+}
+
+static int recv_afs_msg(int afs_socket, int *fd, uint32_t *code, uint32_t *data)
+{
+ char control[255] __a_aligned(8), buf[8];
+ struct msghdr msg = {.msg_iov = NULL};
+ struct cmsghdr *cmsg;
+ struct iovec iov;
+ int ret = 0;
+
+ *fd = -1;
+ iov.iov_base = buf;
+ iov.iov_len = sizeof(buf);
+ msg.msg_iov = &iov;
+ msg.msg_iovlen = 1;
+ msg.msg_control = control;
+ msg.msg_controllen = sizeof(control);
+ memset(buf, 0, sizeof(buf));
+ ret = recvmsg(afs_socket, &msg, 0);
+ if (ret < 0)
+ return -ERRNO_TO_PARA_ERROR(errno);
+ if (iov.iov_len != sizeof(buf))
+ return -E_AFS_SHORT_READ;
+ *code = *(uint32_t*)buf;
+ *data = *(uint32_t*)(buf + 4);
+ for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
+ if (cmsg->cmsg_level != SOL_SOCKET
+ || cmsg->cmsg_type != SCM_RIGHTS)
+ continue;
+ if ((cmsg->cmsg_len - CMSG_LEN(0)) / sizeof(int) != 1)
+ continue;
+ *fd = *(int *)CMSG_DATA(cmsg);
+ }
+ return 1;
+}
+
+#ifndef MAP_POPULATE
+#define MAP_POPULATE 0
+#endif
+
+static void recv_afs_result(struct vss_task *vsst, fd_set *rfds)
+{
+ int ret, passed_fd, shmid;
+ uint32_t afs_code = 0, afs_data = 0;
+ struct stat statbuf;
+
+ if (!FD_ISSET(vsst->afs_socket, rfds))
+ return;
+ ret = recv_afs_msg(vsst->afs_socket, &passed_fd, &afs_code, &afs_data);
+ if (ret == -ERRNO_TO_PARA_ERROR(EAGAIN))
+ return;
+ if (ret < 0)
+ goto err;
+ vsst->afsss = AFS_SOCKET_READY;
+ PARA_DEBUG_LOG("fd: %d, code: %u, shmid: %u\n", passed_fd, afs_code,
+ afs_data);
+ ret = -E_NOFD;
+ if (afs_code != NEXT_AUDIO_FILE)
+ goto err;
+ if (passed_fd < 0)
+ goto err;
+ shmid = afs_data;
+ ret = load_afd(shmid, &mmd->afd);
+ if (ret < 0)
+ goto err;
+ shm_destroy(shmid);
+ ret = fstat(passed_fd, &statbuf);
+ if (ret < 0) {
+ PARA_ERROR_LOG("fstat error:\n");
+ ret = -ERRNO_TO_PARA_ERROR(errno);
+ goto err;
+ }
+ ret = para_mmap(statbuf.st_size, PROT_READ, MAP_PRIVATE | MAP_POPULATE,
+ passed_fd, 0, &vsst->map);
+ if (ret < 0)
+ goto err;
+ vsst->mapsize = statbuf.st_size;
+ close(passed_fd);
+ mmd->chunks_sent = 0;
+ mmd->current_chunk = 0;
+ mmd->offset = 0;
+ mmd->events++;
+ mmd->num_played++;
+ mmd->new_vss_status_flags &= (~VSS_NEXT);
+ afh_get_header(&mmd->afd.afhi, mmd->afd.audio_format_id,
+ vsst->map, vsst->mapsize, &vsst->header_buf, &vsst->header_len);
+ return;
+err:
+ free(mmd->afd.afhi.chunk_table);
+ if (passed_fd >= 0)
+ close(passed_fd);
+ PARA_ERROR_LOG("%s\n", para_strerror(-ret));
+ mmd->new_vss_status_flags = VSS_NEXT;
+}
+
/**
- * main sending function
+ * Main sending function.
*
- * This function gets called from para_server as soon as the next chunk of
- * data should be pushed out. It first calls the read_chunk() function of
- * the current audio format handler to obtain a pointer to the data to be
- * sent out as well as its length. This information is then passed to each
- * supported sender's send() function which does the actual sending.
+ * This function gets called from vss_post_select(). It checks whether the next
+ * chunk of data should be pushed out. It obtains a pointer to the data to be
+ * sent out as well as its length from mmd->afd.afhi. This information is then
+ * passed to each supported sender's send() function as well as to the send()
+ * functions of each registered fec client.
*/
-void vss_send_chunk(void)
+static void vss_send(struct vss_task *vsst)
{
- int i;
- struct audio_format_handler *af;
- struct timeval now, due;
+ int i, ret;
+ bool fec_active = false;
+ struct timeval due;
+ struct fec_client *fc, *tmp_fc;
+ char *buf;
+ size_t len;
- if (mmd->audio_format < 0 || !map || !vss_playing())
+ if (!vsst->map || !vss_playing())
return;
- af = &afl[mmd->audio_format];
- gettimeofday(&now, NULL);
- vss_next_chunk_time(&due);
- if (tv_diff(&due, &now, NULL) > 0)
+ if (chk_barrier("eof", &vsst->eof_barrier, &due, 1) < 0)
return;
- if (chk_barrier("eof", &now, &eof_barrier, &due, 1) < 0)
+ if (chk_barrier("data send", &vsst->data_send_barrier, &due, 1) < 0)
return;
- if (chk_barrier("data send", &now, &data_send_barrier,
- &due, 1) < 0)
+ list_for_each_entry_safe(fc, tmp_fc, &fec_client_list, node) {
+ if (fc->state == FEC_STATE_DISABLED)
+ continue;
+ if (!next_slice_is_due(fc, NULL)) {
+ fec_active = true;
+ continue;
+ }
+ if (compute_next_fec_slice(fc, vsst) <= 0)
+ continue;
+ PARA_DEBUG_LOG("sending %u:%u (%u bytes)\n", fc->group.num,
+ fc->current_slice_num, fc->group.slice_bytes);
+ fc->current_slice_num++;
+ fc->fcp->send_fec(fc->sc, (char *)fc->enc_buf,
+ fc->group.slice_bytes + FEC_HEADER_SIZE);
+ fec_active = true;
+ }
+ if (mmd->current_chunk >= mmd->afd.afhi.chunks_total) { /* eof */
+ if (!fec_active)
+ mmd->new_vss_status_flags |= VSS_NEXT;
return;
- mmd->new_vss_status_flags &= ~VSS_REPOS;
- if (mmd->current_chunk >= mmd->afi.chunks_total) { /* eof */
- mmd->new_vss_status_flags |= VSS_NEXT;
- return vss_eof();
}
- /*
- * We call the send function also in case of empty chunks as they
- * might have still some data queued which can be sent in this case.
- */
+ compute_chunk_time(mmd->chunks_sent, &mmd->afd.afhi.chunk_tv,
+ &mmd->stream_start, &due);
+ if (tv_diff(&due, now, NULL) > 0)
+ return;
if (!mmd->chunks_sent) {
- struct timeval tmp;
- gettimeofday(&mmd->stream_start, NULL);
- tv_scale(mmd->current_chunk, &mmd->afi.chunk_tv, &tmp);
- mmd->offset = tv2ms(&tmp);
+ mmd->stream_start = *now;
mmd->events++;
+ set_mmd_offset();
}
- for (i = 0; senders[i].name; i++) {
- char *buf;
- size_t len;
- get_chunk(mmd->current_chunk, &buf, &len);
- senders[i].send(mmd->current_chunk, mmd->chunks_sent, buf, len);
+ ret = vss_get_chunk(mmd->current_chunk, vsst, &buf, &len);
+ if (ret < 0) {
+ PARA_ERROR_LOG("could not get chunk %lu: %s\n",
+ mmd->current_chunk, para_strerror(-ret));
+ } else {
+ /*
+ * We call ->send() even if len is zero because senders might
+ * have data queued which can be sent now.
+ */
+ for (i = 0; senders[i].name; i++) {
+ if (!senders[i].send)
+ continue;
+ senders[i].send(mmd->current_chunk, mmd->chunks_sent,
+ buf, len, vsst->header_buf, vsst->header_len);
+ }
}
- mmd->new_vss_status_flags |= VSS_PLAYING;
mmd->chunks_sent++;
mmd->current_chunk++;
}
+
+static int vss_post_select(struct sched *s, void *context)
+{
+ int ret, i;
+ struct vss_task *vsst = context;
+
+ if (!vsst->map || vss_next() || vss_paused() || vss_repos()) {
+ /* shut down senders and fec clients */
+ struct fec_client *fc, *tmp;
+ for (i = 0; senders[i].name; i++)
+ if (senders[i].shutdown_clients)
+ senders[i].shutdown_clients();
+ list_for_each_entry_safe(fc, tmp, &fec_client_list, node)
+ fc->state = FEC_STATE_NONE;
+ mmd->stream_start.tv_sec = 0;
+ mmd->stream_start.tv_usec = 0;
+ }
+ if (vss_next())
+ vss_eof(vsst);
+ else if (vss_paused()) {
+ if (mmd->chunks_sent)
+ set_eof_barrier(vsst);
+ mmd->chunks_sent = 0;
+ } else if (vss_repos()) { /* repositioning due to ff/jmp command */
+ tv_add(now, &vsst->announce_tv, &vsst->data_send_barrier);
+ set_eof_barrier(vsst);
+ mmd->chunks_sent = 0;
+ mmd->current_chunk = afh_get_start_chunk(mmd->repos_request,
+ &mmd->afd.afhi, mmd->afd.audio_format_id);
+ mmd->new_vss_status_flags &= ~VSS_REPOS;
+ set_mmd_offset();
+ }
+ /* If a sender command is pending, run it. */
+ if (mmd->sender_cmd_data.cmd_num >= 0) {
+ int num = mmd->sender_cmd_data.cmd_num,
+ sender_num = mmd->sender_cmd_data.sender_num;
+
+ if (senders[sender_num].client_cmds[num]) {
+ ret = senders[sender_num].client_cmds[num]
+ (&mmd->sender_cmd_data);
+ if (ret < 0)
+ PARA_ERROR_LOG("%s\n", para_strerror(-ret));
+ }
+ mmd->sender_cmd_data.cmd_num = -1;
+ }
+ if (vsst->afsss != AFS_SOCKET_CHECK_FOR_WRITE)
+ recv_afs_result(vsst, &s->rfds);
+ else if (FD_ISSET(vsst->afs_socket, &s->wfds)) {
+ PARA_NOTICE_LOG("requesting new fd from afs\n");
+ ret = write_buffer(vsst->afs_socket, "new");
+ if (ret < 0)
+ PARA_CRIT_LOG("%s\n", para_strerror(-ret));
+ else
+ vsst->afsss = AFS_SOCKET_AFD_PENDING;
+ }
+ for (i = 0; senders[i].name; i++) {
+ if (!senders[i].post_select)
+ continue;
+ senders[i].post_select(&s->rfds, &s->wfds);
+ }
+ if ((vss_playing() && !(mmd->vss_status_flags & VSS_PLAYING)) ||
+ (vss_next() && vss_playing()))
+ tv_add(now, &vsst->announce_tv, &vsst->data_send_barrier);
+ vss_send(vsst);
+ return 0;
+}
+
+/**
+ * Initialize the virtual streaming system task.
+ *
+ * \param afs_socket The fd for communication with afs.
+ * \param s The scheduler to register the vss task to.
+ *
+ * This also initializes all supported senders and starts streaming
+ * if the --autoplay command line flag was given.
+ */
+void vss_init(int afs_socket, struct sched *s)
+{
+ static struct vss_task vss_task_struct, *vsst = &vss_task_struct;
+ int i;
+ long unsigned announce_time = OPT_UINT32_VAL(ANNOUNCE_TIME),
+ autoplay_delay = OPT_UINT32_VAL(AUTOPLAY_DELAY);
+ vsst->header_interval.tv_sec = 5; /* should this be configurable? */
+ vsst->afs_socket = afs_socket;
+ ms2tv(announce_time, &vsst->announce_tv);
+ PARA_INFO_LOG("announce timeval: %lums\n", tv2ms(&vsst->announce_tv));
+ INIT_LIST_HEAD(&fec_client_list);
+ for (i = 0; senders[i].name; i++) {
+ PARA_NOTICE_LOG("initializing %s sender\n", senders[i].name);
+ senders[i].init(&senders[i]);
+ }
+ mmd->sender_cmd_data.cmd_num = -1;
+ if (OPT_GIVEN(AUTOPLAY)) {
+ struct timeval tmp;
+ mmd->vss_status_flags |= VSS_PLAYING;
+ mmd->new_vss_status_flags |= VSS_PLAYING;
+ ms2tv(autoplay_delay, &tmp);
+ tv_add(clock_get_realtime(NULL), &tmp, &vsst->autoplay_barrier);
+ tv_add(&vsst->autoplay_barrier, &vsst->announce_tv,
+ &vsst->data_send_barrier);
+ }
+ vsst->task = task_register(&(struct task_info) {
+ .name = "vss",
+ .pre_select = vss_pre_select,
+ .post_select = vss_post_select,
+ .context = vsst,
+ }, s);
+}