filter_chain.c: Cosmetics.
[paraslash.git] / vss.c
diff --git a/vss.c b/vss.c
index 8a9d5f613f308db356756b090017d0ef88162080..3e19b134c2670f6d729609799d4a6a445df989c2 100644 (file)
--- a/vss.c
+++ b/vss.c
@@ -11,9 +11,6 @@
  * senders.
  */
 
-#include <sys/mman.h> /* mmap */
-#include <sys/time.h> /* gettimeofday */
-#include <sys/types.h>
 #include <dirent.h>
 
 #include "para.h"
 #include "send.h"
 #include "ipc.h"
 #include "fd.h"
-
-static struct timeval announce_tv;
-static struct timeval data_send_barrier;
-static struct timeval eof_barrier;
-static struct timeval autoplay_barrier;
+#include "sched.h"
 
 extern struct misc_meta_data *mmd;
-extern struct sender senders[];
 
-static char *map;
+extern void dccp_send_init(struct sender *);
+extern void http_send_init(struct sender *);
+extern void ortp_send_init(struct sender *);
+
+/** The list of supported senders. */
+struct sender senders[] = {
+       {
+               .name = "http",
+               .init = http_send_init,
+       },
+       {
+               .name = "dccp",
+               .init = dccp_send_init,
+       },
+#ifdef HAVE_ORTP
+       {
+               .name = "ortp",
+               .init = ortp_send_init,
+       },
+#endif
+       {
+               .name = NULL,
+       }
+};
+
+/** 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 {
+       struct timeval announce_tv;
+       struct timeval data_send_barrier;
+       struct timeval eof_barrier;
+       struct timeval autoplay_barrier;
+       int afs_socket;
+       enum afs_socket_status afsss;
+       char *map;
+       struct task task;
+       const char *header_buf;
+       size_t header_len;
+};
 
 /**
  * Check if vss status flag \a P (playing) is set.
@@ -97,61 +136,8 @@ unsigned int vss_stopped(void)
                && !(mmd->new_vss_status_flags & VSS_PLAYING);
 }
 
-/**
- * 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 Standard.
- */
-int vss_get_chunk(long unsigned chunk_num, char **buf, size_t *len)
-{
-       if (!map || !vss_playing())
-               return -E_CHUNK;
-       if (chunk_num >= mmd->afd.afhi.chunks_total)
-               return -E_CHUNK;
-       afh_get_chunk(chunk_num, &mmd->afd.afhi, map, buf, len);
-       return 1;
-}
-
-/**
- * Initialize the virtual streaming system.
- *
- * This also initializes all supported senders and starts streaming
- * if the --autoplay command line flag was given.
- */
-void vss_init(void)
-{
-       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;
-       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]);
-       }
-       free(hn);
-       free(home);
-       mmd->sender_cmd_data.cmd_num = -1;
-       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);
-       }
-}
-
-static int chk_barrier(const char *bname, const struct timeval *now,
-               const struct timeval *barrier, struct timeval *diff,
-               int print_log)
+static int chk_barrier(const char *bname, const struct timeval *barrier,
+               struct timeval *diff, int print_log)
 {
        long ms;
 
@@ -167,31 +153,30 @@ static int chk_barrier(const char *bname, const struct timeval *now,
  * != NULL: timeout for next chunk
  * NULL: nothing to do
  */
-static struct timeval *vss_compute_timeout(void)
+static struct timeval *vss_compute_timeout(struct vss_task *vsst)
 {
        static struct timeval the_timeout;
-       struct timeval now, next_chunk;
+       struct timeval next_chunk;
 
-       if (vss_next() && map) {
+       if (vss_next() && vsst->map) {
                /* 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,
+       if (chk_barrier("autoplay_delay", &vsst->autoplay_barrier,
                        &the_timeout, 1) < 0)
                return &the_timeout;
-       if (chk_barrier("eof", &now, &eof_barrier, &the_timeout, 1) < 0)
+       if (chk_barrier("eof", &vsst->eof_barrier, &the_timeout, 1) < 0)
                return &the_timeout;
-       if (chk_barrier("data send", &now, &data_send_barrier,
+       if (chk_barrier("data send", &vsst->data_send_barrier,
                        &the_timeout, 1) < 0)
                return &the_timeout;
-       if (!vss_playing() || !map)
+       if (!vss_playing() || !vsst->map)
                return NULL;
        compute_chunk_time(mmd->chunks_sent, &mmd->afd.afhi.chunk_tv,
                &mmd->stream_start, &next_chunk);
-       if (chk_barrier("chunk", &now, &next_chunk, &the_timeout, 0) < 0)
+       if (chk_barrier("chunk", &next_chunk, &the_timeout, 0) < 0)
                return &the_timeout;
        /* chunk is due or bof */
        the_timeout.tv_sec = 0;
@@ -199,20 +184,23 @@ static struct timeval *vss_compute_timeout(void)
        return &the_timeout;
 }
 
-static void vss_eof(void)
+static void vss_eof(struct vss_task *vsst)
 {
-       struct timeval now;
        char *tmp;
 
+       mmd->stream_start = *now;
+       if (!vsst->map)
+               return;
        if (mmd->new_vss_status_flags & VSS_NOMORE)
                mmd->new_vss_status_flags = VSS_NEXT;
-       gettimeofday(&now, NULL);
-       tv_add(&mmd->afd.afhi.eof_tv, &now, &eof_barrier);
-       munmap(map, mmd->size);
-       map = NULL;
+       tv_add(&mmd->afd.afhi.eof_tv, now, &vsst->eof_barrier);
+       para_munmap(vsst->map, mmd->size);
+       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;
        tmp  = make_message("%s:\n%s:\n%s:\n", status_item_list[SI_AUDIO_FILE_INFO],
@@ -226,19 +214,6 @@ static void vss_eof(void)
        mmd->events++;
 }
 
-/**
- * Get the header of the current audio file.
- *
- * \param buf The length of the header is stored here.
- * \param len Points to a buffer containing the header on return.
- *
- * \sa afh_get_header.
- */
-void vss_get_header(char **buf, size_t *len)
-{
-       afh_get_header(&mmd->afd.afhi, map, buf, len);
-}
-
 /**
  * Get the list of all supported audio formats.
  *
@@ -259,32 +234,21 @@ const char *supported_audio_formats(void)
  */
 struct timeval *vss_chunk_time(void)
 {
-       if (!map)
+       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;
 }
 
-/** The possible states of the afs socket. See \ref 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 afs to reply. */
-       AFS_SOCKET_AFD_PENDING
-};
-
-static enum afs_socket_status afsss;
-
-static int need_to_request_new_audio_file(void)
+static int need_to_request_new_audio_file(struct vss_task *vsst)
 {
-       if (map) /* have audio file */
+       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 (afsss == AFS_SOCKET_AFD_PENDING) /* already requested one */
+       if (vsst->afsss == AFS_SOCKET_AFD_PENDING) /* already requested one */
                return 0;
        return 1;
 }
@@ -295,9 +259,8 @@ static int need_to_request_new_audio_file(void)
  * This function gets called from para_server to determine the timeout value
  * for its main select loop.
  *
- * \param rfds The set of file descriptors to be checked for reading.
- * \param wfds The set of file descriptors to be checked for writing.
- * \param max_fileno The highest-numbered file descriptor.
+ * \param s Pointer to the server scheduler.
+ * \param t Pointer to the vss task structure.
  *
  * Before the timeout is computed, the current vss status flags are evaluated
  * and acted upon by calling appropriate functions from the lower layers.
@@ -306,50 +269,47 @@ static int need_to_request_new_audio_file(void)
  *     - request a new audio file from afs,
  *     - shutdown of all senders (stop/pause command),
  *     - reposition the stream (ff/jmp command).
- *
- * \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(fd_set *rfds, fd_set *wfds, int *max_fileno)
+static void vss_pre_select(struct sched *s, struct task *t)
 {
        int i;
-       struct timeval now;
+       struct timeval *tv, diff;
+       struct vss_task *vsst = container_of(t, struct vss_task, task);
 
-       if (!map || vss_next() || vss_paused() || vss_repos())
+       if (!vsst->map || vss_next() || vss_paused() || vss_repos())
                for (i = 0; senders[i].name; i++)
                        senders[i].shutdown_clients();
        if (vss_next())
-               vss_eof();
+               vss_eof(vsst);
        else if (vss_paused()) {
-               if (mmd->chunks_sent) {
-                       gettimeofday(&now, NULL);
-                       tv_add(&mmd->afd.afhi.eof_tv, &now, &eof_barrier);
-               }
+               if (mmd->chunks_sent)
+                       tv_add(&mmd->afd.afhi.eof_tv, now, &vsst->eof_barrier);
                mmd->chunks_sent = 0;
        } else if (vss_repos()) {
-               gettimeofday(&now, NULL);
-               tv_add(&now, &announce_tv, &data_send_barrier);
-               tv_add(&mmd->afd.afhi.eof_tv, &now, &eof_barrier);
+               tv_add(now, &vsst->announce_tv, &vsst->data_send_barrier);
+               tv_add(&mmd->afd.afhi.eof_tv, now, &vsst->eof_barrier);
                mmd->chunks_sent = 0;
                mmd->current_chunk = mmd->repos_request;
                mmd->new_vss_status_flags &= ~VSS_REPOS;
        }
 
-       if (need_to_request_new_audio_file()) {
+       if (need_to_request_new_audio_file(vsst)) {
                PARA_DEBUG_LOG("ready and playing, but no audio file\n");
-               para_fd_set(afs_socket, wfds, max_fileno);
-               afsss = AFS_SOCKET_CHECK_FOR_WRITE;
+               para_fd_set(vsst->afs_socket, &s->wfds, &s->max_fileno);
+               vsst->afsss = AFS_SOCKET_CHECK_FOR_WRITE;
        } else
-               para_fd_set(afs_socket, rfds, max_fileno);
+               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(max_fileno, rfds, wfds);
+               senders[i].pre_select(&s->max_fileno, &s->rfds, &s->wfds);
        }
-       return vss_compute_timeout();
+       tv = vss_compute_timeout(vsst);
+       if (tv && tv_diff(tv, &s->timeout, &diff) < 0)
+               s->timeout = *tv;
 }
 
-static int recv_afs_msg(int *fd, uint32_t *code, uint32_t *data)
+static int recv_afs_msg(int afs_socket, int *fd, uint32_t *code, uint32_t *data)
 {
        char control[255], buf[8];
        struct msghdr msg = {.msg_iov = NULL};
@@ -368,7 +328,6 @@ static int recv_afs_msg(int *fd, uint32_t *code, uint32_t *data)
        ret = recvmsg(afs_socket, &msg, 0);
        if (ret < 0)
                return -ERRNO_TO_PARA_ERROR(errno);
-       afsss = AFS_SOCKET_READY;
        if (iov.iov_len != sizeof(buf))
                return -E_AFS_SHORT_READ;
        *code = *(uint32_t*)buf;
@@ -384,15 +343,15 @@ static int recv_afs_msg(int *fd, uint32_t *code, uint32_t *data)
        return 1;
 }
 
-static void recv_afs_result(void)
+static void recv_afs_result(struct vss_task *vsst)
 {
        int ret, passed_fd, shmid;
        uint32_t afs_code = 0, afs_data = 0;
        struct stat statbuf;
-       struct timeval now;
 
+       vsst->afsss = AFS_SOCKET_READY;
        mmd->afd.afhi.chunk_table = NULL;
-       ret = recv_afs_msg(&passed_fd, &afs_code, &afs_data);
+       ret = recv_afs_msg(vsst->afs_socket, &passed_fd, &afs_code, &afs_data);
        if (ret < 0)
                goto err;
        PARA_DEBUG_LOG("fd: %d, code: %u, shmid: %u\n", passed_fd, afs_code,
@@ -415,7 +374,7 @@ static void recv_afs_result(void)
        }
        mmd->size = statbuf.st_size;
        mmd->mtime = statbuf.st_mtime;
-       map = para_mmap(mmd->size, PROT_READ, MAP_PRIVATE,
+       vsst->map = para_mmap(mmd->size, PROT_READ, MAP_PRIVATE,
                passed_fd, 0);
        close(passed_fd);
        mmd->chunks_sent = 0;
@@ -424,8 +383,9 @@ static void recv_afs_result(void)
        mmd->events++;
        mmd->num_played++;
        mmd->new_vss_status_flags &= (~VSS_NEXT);
-       gettimeofday(&now, NULL);
-       tv_add(&now, &announce_tv, &data_send_barrier);
+       tv_add(now, &vsst->announce_tv, &vsst->data_send_barrier);
+       afh_get_header(&mmd->afd.afhi, vsst->map, &vsst->header_buf,
+               &vsst->header_len);
        return;
 err:
        free(mmd->afd.afhi.chunk_table);
@@ -444,23 +404,22 @@ err:
  * each supported sender's send() function which is supposed to send out the data
  * to all connected clients.
  */
-static void vss_send_chunk(void)
+static void vss_send_chunk(struct vss_task *vsst)
 {
        int i;
-       struct timeval now, due;
-       char *buf;
+       struct timeval due;
+       const char *buf;
        size_t len;
 
-       if (!map || !vss_playing())
+       if (!vsst->map || !vss_playing())
                return;
-       gettimeofday(&now, NULL);
        compute_chunk_time(mmd->chunks_sent, &mmd->afd.afhi.chunk_tv,
                &mmd->stream_start, &due);
-       if (tv_diff(&due, &now, NULL) > 0)
+       if (tv_diff(&due, now, NULL) > 0)
                return;
-       if (chk_barrier("eof", &now, &eof_barrier, &due, 1) < 0)
+       if (chk_barrier("eof", &vsst->eof_barrier, &due, 1) < 0)
                return;
-       if (chk_barrier("data send", &now, &data_send_barrier,
+       if (chk_barrier("data send", &vsst->data_send_barrier,
                        &due, 1) < 0)
                return;
        mmd->new_vss_status_flags &= ~VSS_REPOS;
@@ -474,43 +433,85 @@ static void vss_send_chunk(void)
         */
        if (!mmd->chunks_sent) {
                struct timeval tmp;
-               gettimeofday(&mmd->stream_start, NULL);
+               mmd->stream_start = *now;
                tv_scale(mmd->current_chunk, &mmd->afd.afhi.chunk_tv, &tmp);
                mmd->offset = tv2ms(&tmp);
                mmd->events++;
        }
-       afh_get_chunk(mmd->current_chunk, &mmd->afd.afhi, map, &buf, &len);
+       afh_get_chunk(mmd->current_chunk, &mmd->afd.afhi, vsst->map, &buf, &len);
        for (i = 0; senders[i].name; i++)
-               senders[i].send(mmd->current_chunk, mmd->chunks_sent, buf, len);
+               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++;
 }
 
-void vss_post_select(fd_set *rfds, fd_set *wfds)
+static void vss_post_select(struct sched *s, struct task *t)
 {
        int ret, i;
+       struct vss_task *vsst = container_of(t, struct vss_task, task);
 
        if (mmd->sender_cmd_data.cmd_num >= 0) {
                int num = mmd->sender_cmd_data.cmd_num,
-                       s = mmd->sender_cmd_data.sender_num;
+                       sender_num = mmd->sender_cmd_data.sender_num;
 
-               if (senders[s].client_cmds[num])
-                       senders[s].client_cmds[num](&mmd->sender_cmd_data);
+               if (senders[sender_num].client_cmds[num])
+                       senders[sender_num].client_cmds[num](&mmd->sender_cmd_data);
                mmd->sender_cmd_data.cmd_num = -1;
        }
-       if (afsss != AFS_SOCKET_CHECK_FOR_WRITE) {
-               if (FD_ISSET(afs_socket, rfds))
-                       recv_afs_result();
-       } else if (FD_ISSET(afs_socket, wfds)) {
+       if (vsst->afsss != AFS_SOCKET_CHECK_FOR_WRITE) {
+               if (FD_ISSET(vsst->afs_socket, &s->rfds))
+                       recv_afs_result(vsst);
+       } else if (FD_ISSET(vsst->afs_socket, &s->wfds)) {
                PARA_NOTICE_LOG("requesting new fd from afs\n");
-               ret = send_buffer(afs_socket, "new");
-               afsss = AFS_SOCKET_AFD_PENDING;
+               ret = send_buffer(vsst->afs_socket, "new");
+               vsst->afsss = AFS_SOCKET_AFD_PENDING;
        }
        for (i = 0; senders[i].name; i++) {
                if (!senders[i].post_select)
                        continue;
-               senders[i].post_select(rfds, wfds);
+               senders[i].post_select(&s->rfds, &s->wfds);
+       }
+       vss_send_chunk(vsst);
+}
+
+/**
+ * Initialize the virtual streaming system task.
+ *
+ * \param afs_socket The fd for communication with afs.
+ *
+ * This also initializes all supported senders and starts streaming
+ * if the --autoplay command line flag was given.
+ */
+void init_vss_task(int afs_socket)
+{
+       static struct vss_task vss_task_struct, *vsst = &vss_task_struct;
+       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;
+
+       vsst->afs_socket = afs_socket;
+       vsst->task.pre_select = vss_pre_select;
+       vsst->task.post_select = vss_post_select;
+       ms2tv(announce_time, &vsst->announce_tv);
+       PARA_INFO_LOG("announce timeval: %lums\n", tv2ms(&vsst->announce_tv));
+       for (i = 0; senders[i].name; i++) {
+               PARA_NOTICE_LOG("initializing %s sender\n", senders[i].name);
+               senders[i].init(&senders[i]);
+       }
+       free(hn);
+       free(home);
+       mmd->sender_cmd_data.cmd_num = -1;
+       if (conf.autoplay_given) {
+               struct timeval tmp;
+               mmd->vss_status_flags |= VSS_PLAYING;
+               mmd->new_vss_status_flags |= VSS_PLAYING;
+               ms2tv(autoplay_delay, &tmp);
+               tv_add(now, &tmp, &vsst->autoplay_barrier);
        }
-       vss_send_chunk();
+       register_task(&vsst->task);
 }