Avoid division by zero in setup_next_fec_group().
[paraslash.git] / vss.c
1 /*
2 * Copyright (C) 1997-2009 Andre Noll <maan@systemlinux.org>
3 *
4 * Licensed under the GPL v2. For licencing details see COPYING.
5 */
6
7 /** \file vss.c The virtual streaming system.
8 *
9 * This contains the audio streaming code of para_server which is independent
10 * of the current audio format, audio file selector and of the activated
11 * senders.
12 */
13
14 #include <dirent.h>
15
16 #include "para.h"
17 #include "error.h"
18 #include "portable_io.h"
19 #include "fec.h"
20 #include "string.h"
21 #include "afh.h"
22 #include "afs.h"
23 #include "server.h"
24 #include "net.h"
25 #include "server.cmdline.h"
26 #include "list.h"
27 #include "vss.h"
28 #include "send.h"
29 #include "ipc.h"
30 #include "fd.h"
31 #include "sched.h"
32
33 extern struct misc_meta_data *mmd;
34
35 extern void dccp_send_init(struct sender *);
36 extern void http_send_init(struct sender *);
37 extern void udp_send_init(struct sender *);
38
39 /** The list of supported senders. */
40 struct sender senders[] = {
41 {
42 .name = "http",
43 .init = http_send_init,
44 },
45 {
46 .name = "dccp",
47 .init = dccp_send_init,
48 },
49 {
50 .name = "udp",
51 .init = udp_send_init,
52 },
53 {
54 .name = NULL,
55 }
56 };
57
58 /** The possible states of the afs socket. */
59 enum afs_socket_status {
60 /** Socket is inactive. */
61 AFS_SOCKET_READY,
62 /** Socket fd was included in the write fd set for select(). */
63 AFS_SOCKET_CHECK_FOR_WRITE,
64 /** vss wrote a request to the socket and waits for reply from afs. */
65 AFS_SOCKET_AFD_PENDING
66 };
67
68 /** The task structure for the virtual streaming system. */
69 struct vss_task {
70 /** Copied from the -announce_time command line option. */
71 struct timeval announce_tv;
72 /** End of the announcing interval. */
73 struct timeval data_send_barrier;
74 /** End of the EOF interval. */
75 struct timeval eof_barrier;
76 /** Only used if --autoplay_delay was given. */
77 struct timeval autoplay_barrier;
78 /** Used for afs-server communication. */
79 int afs_socket;
80 /** The current state of \a afs_socket. */
81 enum afs_socket_status afsss;
82 /** The memory mapped audio file. */
83 char *map;
84 /** Used by the scheduler. */
85 struct task task;
86 /** Pointer to the header of the mapped audio file. */
87 const char *header_buf;
88 /** Length of the audio file header. */
89 size_t header_len;
90 };
91
92 /**
93 * The list of currently connected fec clients.
94 *
95 * Senders may use \ref vss_add_fec_client() to add entries to the list.
96 */
97 static struct list_head fec_client_list;
98
99 /**
100 * Describes one slice of a FEC group.
101 *
102 * FEC slices directly correspond to the data packages sent by the paraslash
103 * senders that use FEC. Each slice is identified by its group number and its
104 * number within the group. All slices have the same size, but the last slice
105 * of the group may not be filled entirely.
106 */
107 struct fec_slice {
108 /** The slice number within the FEC group. */
109 uint8_t num;
110 /** The number of used bytes in this slice. */
111 uint16_t bytes;
112 };
113
114 /**
115 * Data associated with one FEC group.
116 *
117 * A FEC group consists of a fixed number of slices and this number is given by
118 * the \a slices_per_group parameter of struct \ref fec_client_parms. Each FEC
119 * group contains a number of chunks of the current audio file.
120 */
121 struct fec_group {
122 /** The number of the FEC group. */
123 uint32_t num;
124 /** Number of bytes in this group. */
125 uint32_t bytes;
126 /** The first chunk of the current audio file belonging to the group. */
127 uint32_t first_chunk;
128 /** The number of chunks contained in this group. */
129 uint32_t num_chunks;
130 /** The time needed to play all chunks of the group. */
131 struct timeval duration;
132 /** When the first chunk was sent. */
133 struct timeval start;
134 /** \a The group duration divided by \a slices_per_group. */
135 struct timeval slice_duration;
136 };
137
138 /**
139 * Describes one connected FEC client.
140 */
141 struct fec_client {
142 /** Parameters requested by the client. */
143 struct fec_client_parms *fcp;
144 /** Used by the core FEC code. */
145 struct fec_parms *parms;
146 /** The position of this client in the fec client list. */
147 struct list_head node;
148 /** When the first slice for this client was sent. */
149 struct timeval stream_start;
150 /** The first chunk sent to this FEC client. */
151 int first_stream_chunk;
152 /** Describes the current group. */
153 struct fec_group group;
154 /** Describes the current slice. */
155 struct fec_slice slice;
156 /** The data to be FEC-encoded (point to a region within the mapped audio file). */
157 const unsigned char **src_data;
158 /** Used for the last source pointer of the last group. */
159 unsigned char *extra_src_buf;
160 /** The size of the buffer for the extra source pointer. */
161 size_t extra_src_buf_size;
162 /** Contains FEC-encoded data. */
163 unsigned char *enc_buf;
164 /** Size of \a enc_buf. */
165 size_t enc_buf_size;
166 };
167
168 /**
169 * Get the chunk time of the current audio file.
170 *
171 * \return A pointer to a struct containing the chunk time, or NULL,
172 * if currently no audio file is selected.
173 */
174 struct timeval *vss_chunk_time(void)
175 {
176 if (mmd->afd.afhi.chunk_tv.tv_sec == 0 &&
177 mmd->afd.afhi.chunk_tv.tv_usec == 0)
178 return NULL;
179 return &mmd->afd.afhi.chunk_tv;
180 }
181
182 /**
183 * Write a fec header to a buffer.
184 *
185 * \param buf The buffer to write to.
186 * \param h The fec header to write.
187 */
188 static void write_fec_header(struct fec_client *fc)
189 {
190 char *buf = (char *)fc->enc_buf;
191
192 write_u32(buf, FEC_MAGIC);
193
194 write_u8(buf + 4, fc->fcp->slices_per_group);
195 write_u8(buf + 5, fc->fcp->data_slices_per_group);
196 write_u32(buf + 6, (uint32_t)0); /* audio header len */
197
198 write_u32(buf + 10, fc->group.num);
199 write_u32(buf + 14, fc->group.bytes);
200
201 write_u8(buf + 18, fc->slice.num);
202 write_u16(buf + 20, fc->slice.bytes);
203 memset(buf + 22, 0, 11);
204 }
205
206 static int setup_next_fec_group(struct fec_client *fc, struct vss_task *vsst)
207 {
208 uint32_t max_group_size, last_payload_size;
209 int i, k = fc->fcp->data_slices_per_group;
210 size_t len;
211 const char *start_buf;
212 struct timeval tmp, *chunk_tv = vss_chunk_time();
213
214 assert(chunk_tv);
215 if (fc->first_stream_chunk < 0) {
216 fc->stream_start = *now;
217 fc->first_stream_chunk = mmd->current_chunk;
218 fc->group.first_chunk = mmd->current_chunk;
219 fc->group.num = 0;
220 } else {
221 fc->group.first_chunk += fc->group.num_chunks;
222 fc->group.num++;
223 }
224 if (fc->group.first_chunk >= mmd->afd.afhi.chunks_total)
225 return 0;
226 max_group_size = (fc->fcp->max_slice_bytes - FEC_HEADER_SIZE) * k;
227 afh_get_chunk(fc->group.first_chunk, &mmd->afd.afhi, vsst->map,
228 &start_buf, &len);
229 for (i = fc->group.first_chunk, fc->group.bytes = 0;
230 i < mmd->afd.afhi.chunks_total; i++) {
231 const char *buf;
232
233 afh_get_chunk(i, &mmd->afd.afhi, vsst->map, &buf, &len);
234 if (fc->group.bytes + len > max_group_size)
235 break;
236 fc->group.bytes += len;
237 }
238 fc->group.num_chunks = i - fc->group.first_chunk;
239 fc->slice.num = 0;
240 fc->slice.bytes = ROUND_UP(fc->group.bytes, k) / k;
241 /* FIXME: Avoid division by zero in a better way */
242 if (!fc->slice.bytes)
243 return 0;
244
245 /* The last slice will not be fully used */
246 last_payload_size = fc->group.bytes % fc->slice.bytes;
247 if (!last_payload_size)
248 last_payload_size = fc->slice.bytes;
249
250 tv_scale(fc->group.first_chunk - fc->first_stream_chunk, chunk_tv,
251 &tmp);
252 tv_add(&fc->stream_start, &tmp, &fc->group.start);
253 if (fc->group.num) /* quick hack to avoid buffer underruns */
254 fc->group.start.tv_sec--;
255 tv_scale(fc->group.num_chunks, chunk_tv, &fc->group.duration);
256 tv_divide(fc->fcp->slices_per_group, &fc->group.duration,
257 &fc->group.slice_duration);
258
259 for (i = 0; i < k; i++)
260 fc->src_data[i] = (const unsigned char *)start_buf
261 + i * fc->slice.bytes;
262
263 if (start_buf + k * fc->slice.bytes > vsst->map + mmd->size) {
264 /* can not use last slice as it goes beyond the map */
265 if (fc->extra_src_buf_size < fc->slice.bytes) {
266 fc->extra_src_buf = para_realloc(fc->extra_src_buf, fc->slice.bytes);
267 fc->extra_src_buf_size = fc->slice.bytes;
268 }
269 memcpy(fc->extra_src_buf, start_buf + (k - 1) * fc->slice.bytes,
270 last_payload_size);
271 memset(fc->extra_src_buf + last_payload_size, 0,
272 fc->slice.bytes - last_payload_size);
273 fc->src_data[k - 1] = fc->extra_src_buf;
274 }
275 if (fc->enc_buf_size < fc->slice.bytes + FEC_HEADER_SIZE) {
276 fc->enc_buf_size = fc->slice.bytes + FEC_HEADER_SIZE;
277 fc->enc_buf = para_realloc(fc->enc_buf, fc->enc_buf_size);
278 }
279 PARA_INFO_LOG("FEC group %d: %d chunks (%d - %d), duration: %lums\n",
280 fc->group.num, fc->group.num_chunks, fc->group.first_chunk,
281 fc->group.first_chunk + fc->group.num_chunks - 1,
282 tv2ms(&fc->group.duration));
283 return 1;
284 }
285
286 static int compute_next_fec_slice(struct fec_client *fc, struct vss_task *vsst)
287 {
288 if (fc->first_stream_chunk < 0 || fc->slice.num
289 == fc->fcp->slices_per_group) {
290 if (!setup_next_fec_group(fc, vsst))
291 return 0;
292 }
293 write_fec_header(fc);
294 fec_encode(fc->parms, fc->src_data, fc->enc_buf + FEC_HEADER_SIZE,
295 fc->slice.num, fc->slice.bytes);
296 return 1;
297 }
298
299 /**
300 * Return a buffer that marks the end of the stream.
301 *
302 * \param buf Result pointer.
303 * \return The length of the eof buffer.
304 *
305 * This is used for (multicast) udp streaming where closing the socket on the
306 * sender might not give rise to an eof condition at the peer.
307 */
308 size_t vss_get_fec_eof_packet(const char **buf)
309 {
310 static const char fec_eof_packet[FEC_HEADER_SIZE] = FEC_EOF_PACKET;
311 *buf = fec_eof_packet;
312 return FEC_HEADER_SIZE;
313 }
314
315 /**
316 * Add one entry to the list of active fec clients.
317 *
318 * \param fcp Describes the fec parameters to be used for this client.
319 * \param result An opaque pointer that must be used by remove the client later.
320 *
321 * \return Standard.
322 */
323 int vss_add_fec_client(struct fec_client_parms *fcp, struct fec_client **result)
324 {
325 int ret;
326 struct fec_client *fc;
327
328 if (fcp->max_slice_bytes < FEC_HEADER_SIZE + fcp->data_slices_per_group)
329 return -ERRNO_TO_PARA_ERROR(EINVAL);
330 fc = para_calloc(sizeof(*fc));
331 fc->fcp = fcp;
332 ret = fec_new(fcp->data_slices_per_group, fcp->slices_per_group,
333 &fc->parms);
334 if (ret < 0)
335 goto err;
336 fc->first_stream_chunk = -1; /* stream not yet started */
337 fc->src_data = para_malloc(fc->fcp->slices_per_group * sizeof(char *));
338 para_list_add(&fc->node, &fec_client_list);
339 *result = fc;
340 return 1;
341 err:
342 fec_free(fc->parms);
343 free(fc);
344 *result = NULL;
345 return ret;
346 }
347
348 /**
349 * Remove one entry from the list of active fec clients.
350 *
351 * \param fc The client to be removed.
352 */
353 void vss_del_fec_client(struct fec_client *fc)
354 {
355 list_del(&fc->node);
356 free(fc->src_data);
357 free(fc->enc_buf);
358 free(fc->extra_src_buf);
359 fec_free(fc->parms);
360 free(fc);
361 }
362
363 /*
364 * Compute if/when next slice is due. If it isn't due yet and \a diff is
365 * not \p Null, compute the time difference next - now, where
366 *
367 * next = stream_start + (first_group_chunk - first_stream_chunk)
368 * * chunk_time + slice_num * slice_time
369 */
370 static int next_slice_is_due(struct fec_client *fc, struct timeval *diff)
371 {
372 struct timeval tmp, next;
373 int ret;
374
375 if (fc->first_stream_chunk < 0)
376 return 1;
377 tv_scale(fc->slice.num, &fc->group.slice_duration, &tmp);
378 tv_add(&tmp, &fc->group.start, &next);
379 ret = tv_diff(&next, now, diff);
380 return ret < 0? 1 : 0;
381 }
382
383 static void compute_slice_timeout(struct timeval *timeout)
384 {
385 struct fec_client *fc;
386
387 assert(vss_playing());
388 list_for_each_entry(fc, &fec_client_list, node) {
389 struct timeval diff;
390
391 if (next_slice_is_due(fc, &diff)) {
392 timeout->tv_sec = 0;
393 timeout->tv_usec = 0;
394 return;
395 }
396 /* timeout = min(timeout, diff) */
397 if (tv_diff(&diff, timeout, NULL) < 0)
398 *timeout = diff;
399 }
400 }
401
402 /**
403 * Check if vss status flag \a P (playing) is set.
404 *
405 * \return Greater than zero if playing, zero otherwise.
406 *
407 */
408 unsigned int vss_playing(void)
409 {
410 return mmd->new_vss_status_flags & VSS_PLAYING;
411 }
412
413 /**
414 * Check if the \a N (next) status flag is set.
415 *
416 * \return Greater than zero if set, zero if not.
417 *
418 */
419 unsigned int vss_next(void)
420 {
421 return mmd->new_vss_status_flags & VSS_NEXT;
422 }
423
424 /**
425 * Check if a reposition request is pending.
426 *
427 * \return Greater than zero if true, zero otherwise.
428 *
429 */
430 unsigned int vss_repos(void)
431 {
432 return mmd->new_vss_status_flags & VSS_REPOS;
433 }
434
435 /**
436 * Check if the vss is currently paused.
437 *
438 * \return Greater than zero if paused, zero otherwise.
439 *
440 */
441 unsigned int vss_paused(void)
442 {
443 return !(mmd->new_vss_status_flags & VSS_NEXT)
444 && !(mmd->new_vss_status_flags & VSS_PLAYING);
445 }
446
447 /**
448 * Check if the vss is currently stopped.
449 *
450 * \return Greater than zero if paused, zero otherwise.
451 *
452 */
453 unsigned int vss_stopped(void)
454 {
455 return (mmd->new_vss_status_flags & VSS_NEXT)
456 && !(mmd->new_vss_status_flags & VSS_PLAYING);
457 }
458
459 static int chk_barrier(const char *bname, const struct timeval *barrier,
460 struct timeval *diff, int print_log)
461 {
462 long ms;
463
464 if (tv_diff(now, barrier, diff) > 0)
465 return 1;
466 ms = tv2ms(diff);
467 if (print_log && ms)
468 PARA_DEBUG_LOG("%s barrier: %lims left\n", bname, ms);
469 return -1;
470 }
471
472 /*
473 * != NULL: timeout for next chunk
474 * NULL: nothing to do
475 */
476 static struct timeval *vss_compute_timeout(struct vss_task *vsst)
477 {
478 static struct timeval the_timeout;
479 struct timeval next_chunk;
480
481 if (vss_next() && vsst->map) {
482 /* only sleep a bit, nec*/
483 the_timeout.tv_sec = 0;
484 the_timeout.tv_usec = 100;
485 return &the_timeout;
486 }
487 if (chk_barrier("autoplay_delay", &vsst->autoplay_barrier,
488 &the_timeout, 1) < 0)
489 return &the_timeout;
490 if (chk_barrier("eof", &vsst->eof_barrier, &the_timeout, 1) < 0)
491 return &the_timeout;
492 if (chk_barrier("data send", &vsst->data_send_barrier,
493 &the_timeout, 1) < 0)
494 return &the_timeout;
495 if (!vss_playing() || !vsst->map)
496 return NULL;
497 compute_chunk_time(mmd->chunks_sent, &mmd->afd.afhi.chunk_tv,
498 &mmd->stream_start, &next_chunk);
499 if (chk_barrier("chunk", &next_chunk, &the_timeout, 0) >= 0) {
500 /* chunk is due or bof */
501 the_timeout.tv_sec = 0;
502 the_timeout.tv_usec = 0;
503 return &the_timeout;
504 }
505 /* compute min of current timeout and next slice time */
506 compute_slice_timeout(&the_timeout);
507 return &the_timeout;
508 }
509
510 static void vss_eof(struct vss_task *vsst)
511 {
512
513 mmd->stream_start = *now;
514 if (!vsst->map)
515 return;
516 if (mmd->new_vss_status_flags & VSS_NOMORE)
517 mmd->new_vss_status_flags = VSS_NEXT;
518 tv_add(&mmd->afd.afhi.eof_tv, now, &vsst->eof_barrier);
519 para_munmap(vsst->map, mmd->size);
520 vsst->map = NULL;
521 mmd->chunks_sent = 0;
522 mmd->offset = 0;
523 mmd->afd.afhi.seconds_total = 0;
524 mmd->afd.afhi.chunk_tv.tv_sec = 0;
525 mmd->afd.afhi.chunk_tv.tv_usec = 0;
526 free(mmd->afd.afhi.chunk_table);
527 mmd->afd.afhi.chunk_table = NULL;
528 free(mmd->afd.afhi.info_string);
529 mmd->afd.afhi.info_string = make_message("%s:\n%s:\n%s:\n", status_item_list[SI_AUDIO_FILE_INFO],
530 status_item_list[SI_TAGINFO1], status_item_list[SI_TAGINFO2]);
531 make_empty_status_items(mmd->afd.verbose_ls_output);
532 mmd->mtime = 0;
533 mmd->size = 0;
534 mmd->events++;
535 }
536
537 /**
538 * Get the list of all supported audio formats.
539 *
540 * \return Aa space separated list of all supported audio formats
541 * It is not allocated at runtime, i.e. there is no need to free
542 * the returned string in the caller.
543 */
544 const char *supported_audio_formats(void)
545 {
546 return SUPPORTED_AUDIO_FORMATS;
547 }
548
549 static int need_to_request_new_audio_file(struct vss_task *vsst)
550 {
551 struct timeval diff;
552
553 if (vsst->map) /* have audio file */
554 return 0;
555 if (!vss_playing()) /* don't need one */
556 return 0;
557 if (mmd->new_vss_status_flags & VSS_NOMORE)
558 return 0;
559 if (vsst->afsss == AFS_SOCKET_AFD_PENDING) /* already requested one */
560 return 0;
561 if (chk_barrier("autoplay_delay", &vsst->autoplay_barrier,
562 &diff, 1) < 0)
563 return 0;
564 return 1;
565 }
566
567
568
569 /**
570 * Compute the timeout for para_server's main select-loop.
571 *
572 * This function gets called from para_server to determine the timeout value
573 * for its main select loop.
574 *
575 * \param s Pointer to the server scheduler.
576 * \param t Pointer to the vss task structure.
577 *
578 * Before the timeout is computed, the current vss status flags are evaluated
579 * and acted upon by calling appropriate functions from the lower layers.
580 * Possible actions include
581 *
582 * - request a new audio file from afs,
583 * - shutdown of all senders (stop/pause command),
584 * - reposition the stream (ff/jmp command).
585 */
586 static void vss_pre_select(struct sched *s, struct task *t)
587 {
588 int i;
589 struct timeval *tv, diff;
590 struct vss_task *vsst = container_of(t, struct vss_task, task);
591
592 if (!vsst->map || vss_next() || vss_paused() || vss_repos()) {
593 struct fec_client *fc, *tmp;
594 for (i = 0; senders[i].name; i++)
595 if (senders[i].shutdown_clients)
596 senders[i].shutdown_clients();
597 list_for_each_entry_safe(fc, tmp, &fec_client_list, node)
598 fc->first_stream_chunk = -1;
599 }
600 if (vss_next())
601 vss_eof(vsst);
602 else if (vss_paused()) {
603 if (mmd->chunks_sent)
604 tv_add(&mmd->afd.afhi.eof_tv, now, &vsst->eof_barrier);
605 mmd->chunks_sent = 0;
606 } else if (vss_repos()) {
607 tv_add(now, &vsst->announce_tv, &vsst->data_send_barrier);
608 tv_add(&mmd->afd.afhi.eof_tv, now, &vsst->eof_barrier);
609 mmd->chunks_sent = 0;
610 mmd->current_chunk = mmd->repos_request;
611 mmd->new_vss_status_flags &= ~VSS_REPOS;
612 }
613 if (need_to_request_new_audio_file(vsst)) {
614 PARA_DEBUG_LOG("ready and playing, but no audio file\n");
615 para_fd_set(vsst->afs_socket, &s->wfds, &s->max_fileno);
616 vsst->afsss = AFS_SOCKET_CHECK_FOR_WRITE;
617 } else
618 para_fd_set(vsst->afs_socket, &s->rfds, &s->max_fileno);
619 for (i = 0; senders[i].name; i++) {
620 if (!senders[i].pre_select)
621 continue;
622 senders[i].pre_select(&s->max_fileno, &s->rfds, &s->wfds);
623 }
624 tv = vss_compute_timeout(vsst);
625 if (tv && tv_diff(tv, &s->timeout, &diff) < 0)
626 s->timeout = *tv;
627 }
628
629 static int recv_afs_msg(int afs_socket, int *fd, uint32_t *code, uint32_t *data)
630 {
631 char control[255], buf[8];
632 struct msghdr msg = {.msg_iov = NULL};
633 struct cmsghdr *cmsg;
634 struct iovec iov;
635 int ret = 0;
636
637 *fd = -1;
638 iov.iov_base = buf;
639 iov.iov_len = sizeof(buf);
640 msg.msg_iov = &iov;
641 msg.msg_iovlen = 1;
642 msg.msg_control = control;
643 msg.msg_controllen = sizeof(control);
644 memset(buf, 0, sizeof(buf));
645 ret = recvmsg(afs_socket, &msg, 0);
646 if (ret < 0)
647 return -ERRNO_TO_PARA_ERROR(errno);
648 if (iov.iov_len != sizeof(buf))
649 return -E_AFS_SHORT_READ;
650 *code = *(uint32_t*)buf;
651 *data = *(uint32_t*)(buf + 4);
652 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
653 if (cmsg->cmsg_level != SOL_SOCKET
654 || cmsg->cmsg_type != SCM_RIGHTS)
655 continue;
656 if ((cmsg->cmsg_len - CMSG_LEN(0)) / sizeof(int) != 1)
657 continue;
658 *fd = *(int *)CMSG_DATA(cmsg);
659 }
660 return 1;
661 }
662
663 static void recv_afs_result(struct vss_task *vsst)
664 {
665 int ret, passed_fd, shmid;
666 uint32_t afs_code = 0, afs_data = 0;
667 struct stat statbuf;
668
669 vsst->afsss = AFS_SOCKET_READY;
670 mmd->afd.afhi.chunk_table = NULL;
671 ret = recv_afs_msg(vsst->afs_socket, &passed_fd, &afs_code, &afs_data);
672 if (ret < 0)
673 goto err;
674 PARA_DEBUG_LOG("fd: %d, code: %u, shmid: %u\n", passed_fd, afs_code,
675 afs_data);
676 ret = -E_NOFD;
677 if (afs_code != NEXT_AUDIO_FILE)
678 goto err;
679 if (passed_fd < 0)
680 goto err;
681 shmid = afs_data;
682 ret = load_afd(shmid, &mmd->afd);
683 if (ret < 0)
684 goto err;
685 shm_destroy(shmid);
686 ret = fstat(passed_fd, &statbuf);
687 if (ret < 0) {
688 PARA_ERROR_LOG("fstat error:\n");
689 ret = -ERRNO_TO_PARA_ERROR(errno);
690 goto err;
691 }
692 mmd->size = statbuf.st_size;
693 mmd->mtime = statbuf.st_mtime;
694 ret = para_mmap(mmd->size, PROT_READ, MAP_PRIVATE, passed_fd,
695 0, &vsst->map);
696 if (ret < 0)
697 goto err;
698 close(passed_fd);
699 mmd->chunks_sent = 0;
700 mmd->current_chunk = 0;
701 mmd->offset = 0;
702 mmd->events++;
703 mmd->num_played++;
704 mmd->new_vss_status_flags &= (~VSS_NEXT);
705 afh_get_header(&mmd->afd.afhi, vsst->map, &vsst->header_buf,
706 &vsst->header_len);
707 return;
708 err:
709 free(mmd->afd.afhi.chunk_table);
710 if (passed_fd >= 0)
711 close(passed_fd);
712 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
713 mmd->new_vss_status_flags = VSS_NEXT;
714 }
715
716 /**
717 * Main sending function.
718 *
719 * This function gets called from vss_post_select(). It checks whether the next
720 * chunk of data should be pushed out. It obtains a pointer to the data to be
721 * sent out as well as its length from mmd->afd.afhi. This information is then
722 * passed to each supported sender's send() function as well as to the send()
723 * functions of each registered fec client.
724 */
725 static void vss_send(struct vss_task *vsst)
726 {
727 int i;
728 struct timeval due;
729 struct fec_client *fc, *tmp_fc;
730
731 if (!vsst->map || !vss_playing())
732 return;
733 if (chk_barrier("eof", &vsst->eof_barrier, &due, 1) < 0)
734 return;
735 if (chk_barrier("data send", &vsst->data_send_barrier,
736 &due, 1) < 0)
737 return;
738 if (mmd->current_chunk >= mmd->afd.afhi.chunks_total) { /* eof */
739 mmd->new_vss_status_flags |= VSS_NEXT;
740 return;
741 }
742 if (!mmd->chunks_sent) {
743 struct timeval tmp;
744 mmd->stream_start = *now;
745 tv_scale(mmd->current_chunk, &mmd->afd.afhi.chunk_tv, &tmp);
746 mmd->offset = tv2ms(&tmp);
747 mmd->events++;
748 }
749 compute_chunk_time(mmd->chunks_sent, &mmd->afd.afhi.chunk_tv,
750 &mmd->stream_start, &due);
751 if (tv_diff(&due, now, NULL) <= 0) {
752 const char *buf;
753 size_t len;
754 /*
755 * We call the send function also in case of empty chunks as
756 * they might have still some data queued which can be sent in
757 * this case.
758 */
759 afh_get_chunk(mmd->current_chunk, &mmd->afd.afhi, vsst->map,
760 &buf, &len);
761 for (i = 0; senders[i].name; i++) {
762 if (!senders[i].send)
763 continue;
764 senders[i].send(mmd->current_chunk, mmd->chunks_sent,
765 buf, len, vsst->header_buf, vsst->header_len);
766 }
767 }
768 list_for_each_entry_safe(fc, tmp_fc, &fec_client_list, node) {
769 if (!next_slice_is_due(fc, NULL))
770 continue;
771 if (!compute_next_fec_slice(fc, vsst))
772 continue;
773 PARA_DEBUG_LOG("sending %d:%d (%u bytes)\n", fc->group.num,
774 fc->slice.num, fc->slice.bytes);
775 fc->fcp->send((char *)fc->enc_buf,
776 fc->slice.bytes + FEC_HEADER_SIZE,
777 fc->fcp->private_data);
778 fc->slice.num++;
779 }
780 mmd->new_vss_status_flags |= VSS_PLAYING;
781 mmd->chunks_sent++;
782 mmd->current_chunk++;
783 }
784
785 static void vss_post_select(struct sched *s, struct task *t)
786 {
787 int ret, i;
788 struct vss_task *vsst = container_of(t, struct vss_task, task);
789
790 if (mmd->sender_cmd_data.cmd_num >= 0) {
791 int num = mmd->sender_cmd_data.cmd_num,
792 sender_num = mmd->sender_cmd_data.sender_num;
793
794 if (senders[sender_num].client_cmds[num])
795 senders[sender_num].client_cmds[num](&mmd->sender_cmd_data);
796 mmd->sender_cmd_data.cmd_num = -1;
797 }
798 if (vsst->afsss != AFS_SOCKET_CHECK_FOR_WRITE) {
799 if (FD_ISSET(vsst->afs_socket, &s->rfds))
800 recv_afs_result(vsst);
801 } else if (FD_ISSET(vsst->afs_socket, &s->wfds)) {
802 PARA_NOTICE_LOG("requesting new fd from afs\n");
803 ret = send_buffer(vsst->afs_socket, "new");
804 vsst->afsss = AFS_SOCKET_AFD_PENDING;
805 }
806 for (i = 0; senders[i].name; i++) {
807 if (!senders[i].post_select)
808 continue;
809 senders[i].post_select(&s->rfds, &s->wfds);
810 }
811 if ((vss_playing() && !(mmd->vss_status_flags & VSS_PLAYING)) ||
812 (vss_next() && vss_playing()))
813 tv_add(now, &vsst->announce_tv, &vsst->data_send_barrier);
814 vss_send(vsst);
815 }
816
817 /**
818 * Initialize the virtual streaming system task.
819 *
820 * \param afs_socket The fd for communication with afs.
821 *
822 * This also initializes all supported senders and starts streaming
823 * if the --autoplay command line flag was given.
824 */
825 void init_vss_task(int afs_socket)
826 {
827 static struct vss_task vss_task_struct, *vsst = &vss_task_struct;
828 int i;
829 char *hn = para_hostname(), *home = para_homedir();
830 long unsigned announce_time = conf.announce_time_arg > 0?
831 conf.announce_time_arg : 300,
832 autoplay_delay = conf.autoplay_delay_arg > 0?
833 conf.autoplay_delay_arg : 0;
834
835 vsst->afs_socket = afs_socket;
836 vsst->task.pre_select = vss_pre_select;
837 vsst->task.post_select = vss_post_select;
838 ms2tv(announce_time, &vsst->announce_tv);
839 PARA_INFO_LOG("announce timeval: %lums\n", tv2ms(&vsst->announce_tv));
840 INIT_LIST_HEAD(&fec_client_list);
841 for (i = 0; senders[i].name; i++) {
842 PARA_NOTICE_LOG("initializing %s sender\n", senders[i].name);
843 senders[i].init(&senders[i]);
844 }
845 free(hn);
846 free(home);
847 mmd->sender_cmd_data.cmd_num = -1;
848 make_empty_status_items(mmd->afd.verbose_ls_output);
849 if (conf.autoplay_given) {
850 struct timeval tmp;
851 mmd->vss_status_flags |= VSS_PLAYING;
852 mmd->new_vss_status_flags |= VSS_PLAYING;
853 ms2tv(autoplay_delay, &tmp);
854 tv_add(now, &tmp, &vsst->autoplay_barrier);
855 tv_add(&vsst->autoplay_barrier, &vsst->announce_tv,
856 &vsst->data_send_barrier);
857 }
858 register_task(&vsst->task);
859 }