]> git.sesse.net Git - vlc/blob - modules/access/bluray.c
bluray: protection against incorrect paths
[vlc] / modules / access / bluray.c
1 /*****************************************************************************
2  * bluray.c: Blu-ray disc support plugin
3  *****************************************************************************
4  * Copyright © 2010-2011 VideoLAN, VLC authors and libbluray AUTHORS
5  *
6  * Authors: Jean-Baptiste Kempf <jb@videolan.org>
7  *
8  * This program is free software; you can redistribute it and/or modify it
9  * under the terms of the GNU Lesser General Public License as published by
10  * the Free Software Foundation; either version 2.1 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public License
19  * along with this program; if not, write to the Free Software Foundation,
20  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
21  *****************************************************************************/
22
23 #ifdef HAVE_CONFIG_H
24 # include "config.h"
25 #endif
26
27 #include <assert.h>
28 #include <limits.h>                         /* PATH_MAX */
29 #if defined (HAVE_MNTENT_H) && defined(HAVE_SYS_STAT_H)
30 #include <mntent.h>
31 #include <sys/stat.h>
32 #endif
33
34 #include <vlc_common.h>
35 #include <vlc_plugin.h>
36 #include <vlc_demux.h>                      /* demux_t */
37 #include <vlc_input.h>                      /* Seekpoints, chapters */
38 #include <vlc_dialog.h>                     /* BD+/AACS warnings */
39 #include <vlc_vout.h>                       /* vout_PutSubpicture / subpicture_t */
40
41 #include <libbluray/bluray.h>
42 #include <libbluray/keys.h>
43 #include <libbluray/meta_data.h>
44 #include <libbluray/overlay.h>
45
46 /*****************************************************************************
47  * Module descriptor
48  *****************************************************************************/
49
50 #define BD_MENU_TEXT        N_( "Bluray menus" )
51 #define BD_MENU_LONGTEXT    N_( "Use bluray menus. If disabled, "\
52                                 "the movie will start directly" )
53
54 /* Callbacks */
55 static int  blurayOpen ( vlc_object_t * );
56 static void blurayClose( vlc_object_t * );
57
58 vlc_module_begin ()
59     set_shortname( N_("BluRay") )
60     set_description( N_("Blu-Ray Disc support (libbluray)") )
61
62     set_category( CAT_INPUT )
63     set_subcategory( SUBCAT_INPUT_ACCESS )
64     set_capability( "access_demux", 200)
65     add_bool( "bluray-menu", false, BD_MENU_TEXT, BD_MENU_LONGTEXT, false )
66
67     add_shortcut( "bluray", "file" )
68
69     set_callbacks( blurayOpen, blurayClose )
70 vlc_module_end ()
71
72 /* libbluray's overlay.h defines 2 types of overlay (bd_overlay_plane_e). */
73 #define MAX_OVERLAY 2
74
75 typedef enum OverlayStatus {
76     Closed = 0,
77     ToDisplay,  //Used to mark the overlay to be displayed the first time.
78     Displayed,
79     Outdated    //used to update the overlay after it has been sent to the vout
80 } OverlayStatus;
81
82 typedef struct bluray_overlay_t
83 {
84     VLC_GC_MEMBERS
85
86     vlc_mutex_t         lock;
87     subpicture_t        *p_pic;
88     OverlayStatus       status;
89     subpicture_region_t *p_regions;
90 } bluray_overlay_t;
91
92 struct  demux_sys_t
93 {
94     BLURAY              *bluray;
95
96     /* Titles */
97     unsigned int        i_title;
98     unsigned int        i_longest_title;
99     input_title_t       **pp_title;
100
101     /* Menus */
102     bluray_overlay_t    *p_overlays[MAX_OVERLAY];
103     int                 current_overlay; // -1 if no current overlay;
104     bool                b_menu;
105
106     /* */
107     input_thread_t      *p_input;
108     vout_thread_t       *p_vout;
109
110     /* TS stream */
111     stream_t            *p_parser;
112 };
113
114 struct subpicture_updater_sys_t
115 {
116     bluray_overlay_t    *p_overlay;
117 };
118
119 /*****************************************************************************
120  * Local prototypes
121  *****************************************************************************/
122 static int     blurayControl(demux_t *, int, va_list);
123 static int     blurayDemux(demux_t *);
124
125 static int     blurayInitTitles(demux_t *p_demux );
126 static int     bluraySetTitle(demux_t *p_demux, int i_title);
127
128 static void    blurayOverlayProc(void *ptr, const BD_OVERLAY * const overlay);
129
130 static int     onMouseEvent( vlc_object_t *p_vout, const char *psz_var,
131                              vlc_value_t old, vlc_value_t val, void *p_data );
132
133 #define FROM_TICKS(a) (a*CLOCK_FREQ / INT64_C(90000))
134 #define TO_TICKS(a)   (a*INT64_C(90000)/CLOCK_FREQ)
135 #define CUR_LENGTH    p_sys->pp_title[p_demux->info.i_title]->i_length
136
137 /*****************************************************************************
138  * blurayOpen: module init function
139  *****************************************************************************/
140 static int blurayOpen( vlc_object_t *object )
141 {
142     demux_t *p_demux = (demux_t*)object;
143     demux_sys_t *p_sys;
144
145     char *pos_title;
146     int i_title = -1;
147     char bd_path[PATH_MAX] = { '\0' };
148     const char *error_msg = NULL;
149
150     if (strcmp(p_demux->psz_access, "bluray")) {
151         // TODO BDMV support, once we figure out what to do in libbluray
152         return VLC_EGENERIC;
153     }
154
155     /* */
156     p_demux->p_sys = p_sys = calloc(1, sizeof(*p_sys));
157     if (unlikely(!p_sys)) {
158         return VLC_ENOMEM;
159     }
160     p_sys->current_overlay = -1;
161
162     /* init demux info fields */
163     p_demux->info.i_update    = 0;
164     p_demux->info.i_title     = 0;
165     p_demux->info.i_seekpoint = 0;
166
167     TAB_INIT( p_sys->i_title, p_sys->pp_title );
168
169     /* store current bd_path */
170     if (p_demux->psz_file) {
171         strncpy(bd_path, p_demux->psz_file, sizeof(bd_path));
172         bd_path[PATH_MAX - 1] = '\0';
173     }
174
175 #if defined (HAVE_MNTENT_H) && defined (HAVE_SYS_STAT_H)
176     /* If we're passed a block device, try to convert it to the mount point. */
177     struct stat st;
178     if ( !stat (bd_path, &st)) {
179         if (S_ISBLK (st.st_mode)) {
180             FILE* mtab = setmntent ("/proc/self/mounts", "r");
181             struct mntent* m;
182             struct mntent mbuf;
183             char buf [8192];
184             while ((m = getmntent_r (mtab, &mbuf, buf, sizeof(buf))) != NULL) {
185                 if (!strcmp (m->mnt_fsname, bd_path)) {
186                     strncpy (bd_path, m->mnt_dir, sizeof(bd_path));
187                     bd_path[sizeof(bd_path) - 1] = '\0';
188                     break;
189                 }
190             }
191             endmntent (mtab);
192         }
193     }
194 #endif /* HAVE_MNTENT_H && HAVE_SYS_STAT_H */
195     p_sys->bluray = bd_open(bd_path, NULL);
196     if (!p_sys->bluray) {
197         free(p_sys);
198         return VLC_EGENERIC;
199     }
200
201     /* Warning the user about AACS/BD+ */
202     const BLURAY_DISC_INFO *disc_info = bd_get_disc_info(p_sys->bluray);
203
204     /* Is it a bluray? */
205     if (!disc_info->bluray_detected) {
206         error_msg = "Path doesn't appear to be a bluray";
207         goto error;
208     }
209
210     msg_Info(p_demux, "First play: %i, Top menu: %i\n"
211                       "HDMV Titles: %i, BD-J Titles: %i, Other: %i",
212              disc_info->first_play_supported, disc_info->top_menu_supported,
213              disc_info->num_hdmv_titles, disc_info->num_bdj_titles,
214              disc_info->num_unsupported_titles);
215
216     /* AACS */
217     if (disc_info->aacs_detected) {
218         if (!disc_info->libaacs_detected) {
219             error_msg = _("This Blu-Ray Disc needs a library for AACS decoding, "
220                       "and your system does not have it.");
221             goto error;
222         }
223         if (!disc_info->aacs_handled) {
224             error_msg = _("Your system AACS decoding library does not work. "
225                       "Missing keys?");
226             goto error;
227         }
228     }
229
230     /* BD+ */
231     if (disc_info->bdplus_detected) {
232         if (!disc_info->libbdplus_detected) {
233             error_msg = _("This Blu-Ray Disc needs a library for BD+ decoding, "
234                       "and your system does not have it.");
235             goto error;
236         }
237         if (!disc_info->bdplus_handled) {
238             error_msg = _("Your system BD+ decoding library does not work. "
239                       "Missing configuration?");
240             goto error;
241         }
242     }
243
244     /* Get titles and chapters */
245     if (blurayInitTitles(p_demux) != VLC_SUCCESS) {
246         goto error;
247     }
248
249     /*
250      * Initialize the event queue, so we can receive events in blurayDemux(Menu).
251      */
252     bd_get_event(p_sys->bluray, NULL);
253
254     p_sys->b_menu = var_InheritBool( p_demux, "bluray-menu" );
255     if ( p_sys->b_menu )
256     {
257         p_sys->p_input = demux_GetParentInput(p_demux);
258         if (unlikely(!p_sys->p_input))
259             goto error;
260
261         /* libbluray will start playback from "First-Title" title */
262         bd_play(p_sys->bluray);
263
264         /* Registering overlay event handler */
265         bd_register_overlay_proc(p_sys->bluray, p_demux, blurayOverlayProc);
266     }
267     else
268     {
269         /* get title request */
270         if ((pos_title = strrchr(bd_path, ':'))) {
271             /* found character ':' for title information */
272             *(pos_title++) = '\0';
273             i_title = atoi(pos_title);
274         }
275
276         /* set start title number */
277         if (bluraySetTitle(p_demux, i_title) != VLC_SUCCESS) {
278             msg_Err( p_demux, "Could not set the title %d", i_title );
279             goto error;
280         }
281     }
282
283     p_sys->p_parser   = stream_DemuxNew(p_demux, "ts", p_demux->out);
284     if (!p_sys->p_parser) {
285         msg_Err(p_demux, "Failed to create TS demuxer");
286         goto error;
287     }
288
289     p_demux->pf_control = blurayControl;
290     p_demux->pf_demux   = blurayDemux;
291
292     return VLC_SUCCESS;
293
294 error:
295     if (error_msg)
296         dialog_Fatal(p_demux, _("Blu-Ray error"), "%s", error_msg);
297     blurayClose(object);
298     return VLC_EGENERIC;
299 }
300
301
302 /*****************************************************************************
303  * blurayClose: module destroy function
304  *****************************************************************************/
305 static void blurayClose( vlc_object_t *object )
306 {
307     demux_t *p_demux = (demux_t*)object;
308     demux_sys_t *p_sys = p_demux->p_sys;
309
310     /*
311      * Close libbluray first.
312      * This will close all the overlays before we release p_vout
313      * bd_close( NULL ) can crash
314      */
315     assert(p_sys->bluray);
316     bd_close(p_sys->bluray);
317
318     if (p_sys->p_vout != NULL) {
319         var_DelCallback(p_sys->p_vout, "mouse-moved", &onMouseEvent, p_demux);
320         var_DelCallback(p_sys->p_vout, "mouse-clicked", &onMouseEvent, p_demux);
321         vlc_object_release(p_sys->p_vout);
322     }
323     if (p_sys->p_input != NULL)
324         vlc_object_release(p_sys->p_input);
325     if (p_sys->p_parser)
326         stream_Delete(p_sys->p_parser);
327
328     /* Titles */
329     for (unsigned int i = 0; i < p_sys->i_title; i++)
330         vlc_input_title_Delete(p_sys->pp_title[i]);
331     TAB_CLEAN( p_sys->i_title, p_sys->pp_title );
332
333     free(p_sys);
334 }
335
336 /*****************************************************************************
337  * subpicture_updater_t functions:
338  *****************************************************************************/
339 static int subpictureUpdaterValidate( subpicture_t *p_subpic,
340                                       bool b_fmt_src, const video_format_t *p_fmt_src,
341                                       bool b_fmt_dst, const video_format_t *p_fmt_dst,
342                                       mtime_t i_ts )
343 {
344     VLC_UNUSED( b_fmt_src );
345     VLC_UNUSED( b_fmt_dst );
346     VLC_UNUSED( p_fmt_src );
347     VLC_UNUSED( p_fmt_dst );
348     VLC_UNUSED( i_ts );
349
350     subpicture_updater_sys_t *p_upd_sys = p_subpic->updater.p_sys;
351     bluray_overlay_t         *p_overlay = p_upd_sys->p_overlay;
352
353     vlc_mutex_lock(&p_overlay->lock);
354     int res = p_overlay->status == Outdated;
355     vlc_mutex_unlock(&p_overlay->lock);
356     return res;
357 }
358
359 /* This should probably be moved to subpictures.c afterward */
360 static subpicture_region_t* subpicture_region_Clone(subpicture_region_t *p_region_src)
361 {
362     if (!p_region_src)
363         return NULL;
364     subpicture_region_t *p_region_dst = subpicture_region_New(&p_region_src->fmt);
365     if (unlikely(!p_region_dst))
366         return NULL;
367
368     p_region_dst->i_x      = p_region_src->i_x;
369     p_region_dst->i_y      = p_region_src->i_y;
370     p_region_dst->i_align  = p_region_src->i_align;
371     p_region_dst->i_alpha  = p_region_src->i_alpha;
372
373     p_region_dst->psz_text = p_region_src->psz_text ? strdup(p_region_src->psz_text) : NULL;
374     p_region_dst->psz_html = p_region_src->psz_html ? strdup(p_region_src->psz_html) : NULL;
375     if (p_region_src->p_style != NULL) {
376         p_region_dst->p_style = malloc(sizeof(*p_region_dst->p_style));
377         p_region_dst->p_style = text_style_Copy(p_region_dst->p_style,
378                                                 p_region_src->p_style);
379     }
380
381     //Palette is already copied by subpicture_region_New, we just have to duplicate p_pixels
382     for (int i = 0; i < p_region_src->p_picture->i_planes; i++)
383         memcpy(p_region_dst->p_picture->p[i].p_pixels,
384                p_region_src->p_picture->p[i].p_pixels,
385                p_region_src->p_picture->p[i].i_lines * p_region_src->p_picture->p[i].i_pitch);
386     return p_region_dst;
387 }
388
389 static void subpictureUpdaterUpdate(subpicture_t *p_subpic,
390                                     const video_format_t *p_fmt_src,
391                                     const video_format_t *p_fmt_dst,
392                                     mtime_t i_ts)
393 {
394     VLC_UNUSED(p_fmt_src);
395     VLC_UNUSED(p_fmt_dst);
396     VLC_UNUSED(i_ts);
397     subpicture_updater_sys_t *p_upd_sys = p_subpic->updater.p_sys;
398     bluray_overlay_t         *p_overlay = p_upd_sys->p_overlay;
399
400     /*
401      * When this function is called, all p_subpic regions are gone.
402      * We need to duplicate our regions (stored internaly) to this subpic.
403      */
404     vlc_mutex_lock(&p_overlay->lock);
405
406     subpicture_region_t *p_src = p_overlay->p_regions;
407     if (!p_src)
408         return;
409
410     subpicture_region_t **p_dst = &(p_subpic->p_region);
411     while (p_src != NULL) {
412         *p_dst = subpicture_region_Clone(p_src);
413         if (*p_dst == NULL)
414             break ;
415         p_dst = &((*p_dst)->p_next);
416         p_src = p_src->p_next;
417     }
418     if (*p_dst != NULL)
419         (*p_dst)->p_next = NULL;
420     p_overlay->status = Displayed;
421     vlc_mutex_unlock(&p_overlay->lock);
422 }
423
424 static void subpictureUpdaterDestroy(subpicture_t *p_subpic)
425 {
426     vlc_gc_decref(p_subpic->updater.p_sys->p_overlay);
427 }
428
429 /*****************************************************************************
430  * User input events:
431  *****************************************************************************/
432 static int onMouseEvent(vlc_object_t *p_vout, const char *psz_var, vlc_value_t old,
433                         vlc_value_t val, void *p_data)
434 {
435     demux_t     *p_demux = (demux_t*)p_data;
436     demux_sys_t *p_sys   = p_demux->p_sys;
437     mtime_t     now      = mdate();
438     VLC_UNUSED(old);
439     VLC_UNUSED(p_vout);
440
441     if (psz_var[6] == 'm')   //Mouse moved
442         bd_mouse_select(p_sys->bluray, now, val.coords.x, val.coords.y);
443     else if (psz_var[6] == 'c') {
444         bd_mouse_select(p_sys->bluray, now, val.coords.x, val.coords.y);
445         bd_user_input(p_sys->bluray, now, BD_VK_MOUSE_ACTIVATE);
446     } else {
447         assert(0);
448     }
449     return VLC_SUCCESS;
450 }
451
452 /*****************************************************************************
453  * libbluray overlay handling:
454  *****************************************************************************/
455 static void blurayCleanOverayStruct(gc_object_t *p_gc)
456 {
457     bluray_overlay_t *p_overlay = vlc_priv(p_gc, bluray_overlay_t);
458
459     /*
460      * This will be called when destroying the picture.
461      * Don't delete it again from here!
462      */
463     vlc_mutex_destroy(&p_overlay->lock);
464     subpicture_region_Delete(p_overlay->p_regions);
465     free(p_overlay);
466 }
467
468 static void blurayCloseAllOverlays(demux_t *p_demux)
469 {
470     demux_sys_t *p_sys = p_demux->p_sys;
471
472     p_demux->p_sys->current_overlay = -1;
473     if (p_sys->p_vout != NULL) {
474         for (int i = 0; i < 0; i++) {
475             if (p_sys->p_overlays[i] != NULL) {
476                 vout_FlushSubpictureChannel(p_sys->p_vout,
477                                             p_sys->p_overlays[i]->p_pic->i_channel);
478                 vlc_gc_decref(p_sys->p_overlays[i]);
479                 p_sys->p_overlays[i] = NULL;
480             }
481         }
482         var_DelCallback(p_sys->p_vout, "mouse-moved", &onMouseEvent, p_demux);
483         var_DelCallback(p_sys->p_vout, "mouse-clicked", &onMouseEvent, p_demux);
484         vlc_object_release(p_sys->p_vout);
485         p_sys->p_vout = NULL;
486     }
487 }
488
489 /*
490  * Mark the overlay as "ToDisplay" status.
491  * This will not send the overlay to the vout instantly, as the vout
492  * may not be acquired (not acquirable) yet.
493  * If is has already been acquired, the overlay has already been sent to it,
494  * therefore, we only flag the overlay as "Outdated"
495  */
496 static void blurayActivateOverlay(demux_t *p_demux, const BD_OVERLAY* const ov)
497 {
498     demux_sys_t *p_sys = p_demux->p_sys;
499
500     /*
501      * If the overlay is already displayed, mark the picture as outdated.
502      * We must NOT use vout_PutSubpicture if a picture is already displayed.
503      */
504     vlc_mutex_lock(&p_sys->p_overlays[ov->plane]->lock);
505     if ((p_sys->p_overlays[ov->plane]->status == Displayed ||
506             p_sys->p_overlays[ov->plane]->status == Outdated)
507             && p_sys->p_vout) {
508         p_sys->p_overlays[ov->plane]->status = Outdated;
509         vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
510         return ;
511     }
512     /*
513      * Mark the overlay as available, but don't display it right now.
514      * the blurayDemuxMenu will send it to vout, as it may be unavailable when
515      * the overlay is computed
516      */
517     p_sys->current_overlay = ov->plane;
518     p_sys->p_overlays[ov->plane]->status = ToDisplay;
519     vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
520 }
521
522 static void blurayInitOverlay(demux_t *p_demux, const BD_OVERLAY* const ov)
523 {
524     demux_sys_t *p_sys = p_demux->p_sys;
525
526     assert(p_sys->p_overlays[ov->plane] == NULL);
527
528     p_sys->p_overlays[ov->plane] = calloc(1, sizeof(**p_sys->p_overlays));
529     if (unlikely(!p_sys->p_overlays[ov->plane]))
530         return;
531
532     subpicture_updater_sys_t *p_upd_sys = malloc(sizeof(*p_upd_sys));
533     if (unlikely(!p_upd_sys)) {
534         free(p_sys->p_overlays[ov->plane]);
535         p_sys->p_overlays[ov->plane] = NULL;
536         return;
537     }
538     vlc_gc_init(p_sys->p_overlays[ov->plane], blurayCleanOverayStruct);
539     /* Incrementing refcounter: vout + demux */
540     vlc_gc_incref(p_sys->p_overlays[ov->plane]);
541
542     p_upd_sys->p_overlay = p_sys->p_overlays[ov->plane];
543     subpicture_updater_t updater = {
544         .pf_validate = subpictureUpdaterValidate,
545         .pf_update   = subpictureUpdaterUpdate,
546         .pf_destroy  = subpictureUpdaterDestroy,
547         .p_sys       = p_upd_sys,
548     };
549     p_sys->p_overlays[ov->plane]->p_pic = subpicture_New(&updater);
550     p_sys->p_overlays[ov->plane]->p_pic->i_original_picture_width = ov->w;
551     p_sys->p_overlays[ov->plane]->p_pic->i_original_picture_height = ov->h;
552     p_sys->p_overlays[ov->plane]->p_pic->b_ephemer = true;
553     p_sys->p_overlays[ov->plane]->p_pic->b_absolute = true;
554 }
555
556 /**
557  * Destroy every regions in the subpicture.
558  * This is done in two steps:
559  * - Wiping our private regions list
560  * - Flagging the overlay as outdated, so the changes are replicated from
561  *   the subpicture_updater_t::pf_update
562  * This doesn't destroy the subpicture, as the overlay may be used again by libbluray.
563  */
564 static void blurayClearOverlay(demux_t *p_demux, const BD_OVERLAY* const ov)
565 {
566     demux_sys_t *p_sys = p_demux->p_sys;
567
568     vlc_mutex_lock(&p_sys->p_overlays[ov->plane]->lock);
569
570     subpicture_region_ChainDelete(p_sys->p_overlays[ov->plane]->p_regions);
571     p_sys->p_overlays[ov->plane]->p_regions = NULL;
572     p_sys->p_overlays[ov->plane]->status = Outdated;
573     vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
574 }
575
576 /*
577  * This will draw to the overlay by adding a region to our region list
578  * This will have to be copied to the subpicture used to render the overlay.
579  */
580 static void blurayDrawOverlay(demux_t *p_demux, const BD_OVERLAY* const ov)
581 {
582     demux_sys_t *p_sys = p_demux->p_sys;
583
584     /*
585      * Compute a subpicture_region_t.
586      * It will be copied and sent to the vout later.
587      */
588     if (!ov->img)
589         return;
590
591     vlc_mutex_lock(&p_sys->p_overlays[ov->plane]->lock);
592
593     /* Find a region to update */
594     subpicture_region_t *p_reg = p_sys->p_overlays[ov->plane]->p_regions;
595     subpicture_region_t *p_last = NULL;
596     while (p_reg != NULL) {
597         p_last = p_reg;
598         if (p_reg->i_x == ov->x && p_reg->i_y == ov->y &&
599                 p_reg->fmt.i_width == ov->w && p_reg->fmt.i_height == ov->h)
600             break;
601         p_reg = p_reg->p_next;
602     }
603
604     /* If there is no region to update, create a new one. */
605     if (!p_reg) {
606         video_format_t fmt;
607         video_format_Init(&fmt, 0);
608         video_format_Setup(&fmt, VLC_CODEC_YUVP, ov->w, ov->h, 1, 1);
609
610         p_reg = subpicture_region_New(&fmt);
611         p_reg->i_x = ov->x;
612         p_reg->i_y = ov->y;
613         /* Append it to our list. */
614         if (p_last != NULL)
615             p_last->p_next = p_reg;
616         else /* If we don't have a last region, then our list empty */
617             p_sys->p_overlays[ov->plane]->p_regions = p_reg;
618     }
619
620     /* Now we can update the region, regardless it's an update or an insert */
621     const BD_PG_RLE_ELEM *img = ov->img;
622     for (int y = 0; y < ov->h; y++) {
623         for (int x = 0; x < ov->w;) {
624             memset(p_reg->p_picture->p[0].p_pixels +
625                    y * p_reg->p_picture->p[0].i_pitch + x,
626                    img->color, img->len);
627             x += img->len;
628             img++;
629         }
630     }
631     if (ov->palette) {
632         p_reg->fmt.p_palette->i_entries = 256;
633         for (int i = 0; i < 256; ++i) {
634             p_reg->fmt.p_palette->palette[i][0] = ov->palette[i].Y;
635             p_reg->fmt.p_palette->palette[i][1] = ov->palette[i].Cb;
636             p_reg->fmt.p_palette->palette[i][2] = ov->palette[i].Cr;
637             p_reg->fmt.p_palette->palette[i][3] = ov->palette[i].T;
638         }
639     }
640     vlc_mutex_unlock(&p_sys->p_overlays[ov->plane]->lock);
641     /*
642      * /!\ The region is now stored in our internal list, but not in the subpicture /!\
643      */
644 }
645
646 static void blurayOverlayProc(void *ptr, const BD_OVERLAY *const overlay)
647 {
648     demux_t *p_demux = (demux_t*)ptr;
649
650     if (!overlay) {
651         msg_Info(p_demux, "Closing overlay.");
652         blurayCloseAllOverlays(p_demux);
653         return;
654     }
655     switch (overlay->cmd) {
656         case BD_OVERLAY_INIT:
657             msg_Info(p_demux, "Initializing overlay");
658             blurayInitOverlay(p_demux, overlay);
659             break;
660         case BD_OVERLAY_CLEAR:
661             blurayClearOverlay(p_demux, overlay);
662             break;
663         case BD_OVERLAY_FLUSH:
664             blurayActivateOverlay(p_demux, overlay);
665             break;
666         case BD_OVERLAY_DRAW:
667             blurayDrawOverlay(p_demux, overlay);
668             break;
669         default:
670             msg_Warn(p_demux, "Unknown BD overlay command: %u", overlay->cmd);
671             break;
672     }
673 }
674
675 static void bluraySendOverlayToVout(demux_t *p_demux)
676 {
677     demux_sys_t *p_sys = p_demux->p_sys;
678
679     assert(p_sys->current_overlay >= 0 &&
680            p_sys->p_overlays[p_sys->current_overlay] != NULL &&
681            p_sys->p_overlays[p_sys->current_overlay]->p_pic != NULL);
682
683     p_sys->p_overlays[p_sys->current_overlay]->p_pic->i_start =
684         p_sys->p_overlays[p_sys->current_overlay]->p_pic->i_stop = mdate();
685     p_sys->p_overlays[p_sys->current_overlay]->p_pic->i_channel =
686         vout_RegisterSubpictureChannel(p_sys->p_vout);
687     /*
688      * After this point, the picture should not be accessed from the demux thread,
689      * as it's hold by the vout thread.
690      * This must be done only once per subpicture, ie. only once between each
691      * blurayInitOverlay & blurayCloseOverlay call.
692      */
693     vout_PutSubpicture(p_sys->p_vout, p_sys->p_overlays[p_sys->current_overlay]->p_pic);
694     /*
695      * Mark the picture as Outdated, as it contains no region for now.
696      * This will make the subpicture_updater_t call pf_update
697      */
698     p_sys->p_overlays[p_sys->current_overlay]->status = Outdated;
699 }
700
701 static int blurayInitTitles(demux_t *p_demux )
702 {
703     demux_sys_t *p_sys = p_demux->p_sys;
704
705     /* get and set the titles */
706     unsigned i_title = bd_get_titles(p_sys->bluray, TITLES_RELEVANT, 60);
707     int64_t duration = 0;
708
709     for (unsigned int i = 0; i < i_title; i++) {
710         input_title_t *t = vlc_input_title_New();
711         if (!t)
712             break;
713
714         BLURAY_TITLE_INFO *title_info = bd_get_title_info(p_sys->bluray, i, 0);
715         if (!title_info)
716             break;
717         t->i_length = FROM_TICKS(title_info->duration);
718
719         if (t->i_length > duration) {
720             duration = t->i_length;
721             p_sys->i_longest_title = i;
722         }
723
724         for ( unsigned int j = 0; j < title_info->chapter_count; j++) {
725             seekpoint_t *s = vlc_seekpoint_New();
726             if (!s)
727                 break;
728             s->i_time_offset = title_info->chapters[j].offset;
729
730             TAB_APPEND( t->i_seekpoint, t->seekpoint, s );
731         }
732         TAB_APPEND( p_sys->i_title, p_sys->pp_title, t );
733         bd_free_title_info(title_info);
734     }
735     return VLC_SUCCESS;
736 }
737
738 static void blurayResetParser( demux_t *p_demux )
739 {
740     /*
741      * This is a hack and will have to be removed.
742      * The parser should be flushed, and not destroy/created each time
743      * we are changing title.
744      */
745     demux_sys_t *p_sys = p_demux->p_sys;
746     if (!p_sys->p_parser)
747         return;
748     stream_Delete(p_sys->p_parser);
749     p_sys->p_parser = stream_DemuxNew(p_demux, "ts", p_demux->out);
750     if (!p_sys->p_parser) {
751         msg_Err(p_demux, "Failed to create TS demuxer");
752     }
753 }
754
755 static void blurayUpdateTitle( demux_t *p_demux, int i_title )
756 {
757     blurayResetParser(p_demux);
758     if (i_title >= p_demux->p_sys->i_title)
759         return;
760
761     /* read title info and init some values */
762     p_demux->info.i_title = i_title;
763     p_demux->info.i_seekpoint = 0;
764     p_demux->info.i_update |= INPUT_UPDATE_TITLE | INPUT_UPDATE_SEEKPOINT;
765 }
766
767 /*****************************************************************************
768  * bluraySetTitle: select new BD title
769  *****************************************************************************/
770 static int bluraySetTitle(demux_t *p_demux, int i_title)
771 {
772     demux_sys_t *p_sys = p_demux->p_sys;
773
774     /* Looking for the main title, ie the longest duration */
775     if (i_title < 0)
776         i_title = p_sys->i_longest_title;
777     else if ((unsigned)i_title > p_sys->i_title)
778         return VLC_EGENERIC;
779
780     msg_Dbg( p_demux, "Selecting Title %i", i_title);
781
782     /* Select Blu-Ray title */
783     if (bd_select_title(p_demux->p_sys->bluray, i_title) == 0 ) {
784         msg_Err(p_demux, "cannot select bd title '%d'", p_demux->info.i_title);
785         return VLC_EGENERIC;
786     }
787     blurayUpdateTitle( p_demux, i_title );
788
789     return VLC_SUCCESS;
790 }
791
792
793 /*****************************************************************************
794  * blurayControl: handle the controls
795  *****************************************************************************/
796 static int blurayControl(demux_t *p_demux, int query, va_list args)
797 {
798     demux_sys_t *p_sys = p_demux->p_sys;
799     bool     *pb_bool;
800     int64_t  *pi_64;
801
802     switch (query) {
803         case DEMUX_CAN_SEEK:
804         case DEMUX_CAN_PAUSE:
805         case DEMUX_CAN_CONTROL_PACE:
806              pb_bool = (bool*)va_arg( args, bool * );
807              *pb_bool = true;
808              break;
809
810         case DEMUX_GET_PTS_DELAY:
811             pi_64 = (int64_t*)va_arg( args, int64_t * );
812             *pi_64 =
813                 INT64_C(1000) * var_InheritInteger( p_demux, "disc-caching" );
814             break;
815
816         case DEMUX_SET_PAUSE_STATE:
817             /* Nothing to do */
818             break;
819
820         case DEMUX_SET_TITLE:
821         {
822             int i_title = (int)va_arg( args, int );
823             if (bluraySetTitle(p_demux, i_title) != VLC_SUCCESS)
824                 return VLC_EGENERIC;
825             break;
826         }
827         case DEMUX_SET_SEEKPOINT:
828         {
829             int i_chapter = (int)va_arg( args, int );
830             bd_seek_chapter( p_sys->bluray, i_chapter );
831             p_demux->info.i_update = INPUT_UPDATE_SEEKPOINT;
832             break;
833         }
834
835         case DEMUX_GET_TITLE_INFO:
836         {
837             input_title_t ***ppp_title = (input_title_t***)va_arg( args, input_title_t*** );
838             int *pi_int             = (int*)va_arg( args, int* );
839             int *pi_title_offset    = (int*)va_arg( args, int* );
840             int *pi_chapter_offset  = (int*)va_arg( args, int* );
841
842             /* */
843             *pi_title_offset   = 0;
844             *pi_chapter_offset = 0;
845
846             /* Duplicate local title infos */
847             *pi_int = p_sys->i_title;
848             *ppp_title = calloc( p_sys->i_title, sizeof(input_title_t **) );
849             for( unsigned int i = 0; i < p_sys->i_title; i++ )
850                 (*ppp_title)[i] = vlc_input_title_Duplicate( p_sys->pp_title[i]);
851
852             return VLC_SUCCESS;
853         }
854
855         case DEMUX_GET_LENGTH:
856         {
857             int64_t *pi_length = (int64_t*)va_arg(args, int64_t *);
858             *pi_length = p_demux->info.i_title < p_sys->i_title ? CUR_LENGTH : 0;
859             return VLC_SUCCESS;
860         }
861         case DEMUX_SET_TIME:
862         {
863             int64_t i_time = (int64_t)va_arg(args, int64_t);
864             bd_seek_time(p_sys->bluray, TO_TICKS(i_time));
865             return VLC_SUCCESS;
866         }
867         case DEMUX_GET_TIME:
868         {
869             int64_t *pi_time = (int64_t*)va_arg(args, int64_t *);
870             *pi_time = (int64_t)FROM_TICKS(bd_tell_time(p_sys->bluray));
871             return VLC_SUCCESS;
872         }
873
874         case DEMUX_GET_POSITION:
875         {
876             double *pf_position = (double*)va_arg( args, double * );
877             *pf_position = p_demux->info.i_title < p_sys->i_title ?
878                         (double)FROM_TICKS(bd_tell_time(p_sys->bluray))/CUR_LENGTH : 0.0;
879             return VLC_SUCCESS;
880         }
881         case DEMUX_SET_POSITION:
882         {
883             double f_position = (double)va_arg(args, double);
884             bd_seek_time(p_sys->bluray, TO_TICKS(f_position*CUR_LENGTH));
885             return VLC_SUCCESS;
886         }
887
888         case DEMUX_GET_META:
889         {
890             const struct meta_dl *meta = bd_get_meta(p_sys->bluray);
891             if(!meta)
892                 return VLC_EGENERIC;
893
894             vlc_meta_t *p_meta = (vlc_meta_t *) va_arg (args, vlc_meta_t*);
895
896             if (!EMPTY_STR(meta->di_name)) vlc_meta_SetTitle(p_meta, meta->di_name);
897
898             if (!EMPTY_STR(meta->language_code)) vlc_meta_AddExtra(p_meta, "Language", meta->language_code);
899             if (!EMPTY_STR(meta->filename)) vlc_meta_AddExtra(p_meta, "Filename", meta->filename);
900             if (!EMPTY_STR(meta->di_alternative)) vlc_meta_AddExtra(p_meta, "Alternative", meta->di_alternative);
901
902             // if (meta->di_set_number > 0) vlc_meta_SetTrackNum(p_meta, meta->di_set_number);
903             // if (meta->di_num_sets > 0) vlc_meta_AddExtra(p_meta, "Discs numbers in Set", meta->di_num_sets);
904
905             if (meta->thumb_count > 0 && meta->thumbnails) {
906                 vlc_meta_SetArtURL(p_meta, meta->thumbnails[0].path);
907             }
908
909             return VLC_SUCCESS;
910         }
911
912         case DEMUX_CAN_RECORD:
913         case DEMUX_GET_FPS:
914         case DEMUX_SET_GROUP:
915         case DEMUX_HAS_UNSUPPORTED_META:
916         case DEMUX_GET_ATTACHMENTS:
917             return VLC_EGENERIC;
918         default:
919             msg_Warn( p_demux, "unimplemented query (%d) in control", query );
920             return VLC_EGENERIC;
921     }
922     return VLC_SUCCESS;
923 }
924
925 static void blurayHandleEvent( demux_t *p_demux, const BD_EVENT *e )
926 {
927     demux_sys_t *p_sys = p_demux->p_sys;
928
929     switch (e->event)
930     {
931         case BD_EVENT_TITLE:
932             blurayUpdateTitle( p_demux, e->param );
933             break;
934         case BD_EVENT_PLAYITEM:
935             break;
936         case BD_EVENT_AUDIO_STREAM:
937             break;
938         case BD_EVENT_CHAPTER:
939             p_demux->info.i_update |= INPUT_UPDATE_SEEKPOINT;
940             p_demux->info.i_seekpoint = 0;
941             break;
942         case BD_EVENT_ANGLE:
943         case BD_EVENT_IG_STREAM:
944         default:
945             msg_Warn( p_demux, "event: %d param: %d", e->event, e->param );
946             break;
947     }
948 }
949
950 static void blurayHandleEvents( demux_t *p_demux )
951 {
952     BD_EVENT e;
953
954     while (bd_get_event(p_demux->p_sys->bluray, &e))
955     {
956         blurayHandleEvent(p_demux, &e);
957     }
958 }
959
960 #define BD_TS_PACKET_SIZE (192)
961 #define NB_TS_PACKETS (200)
962
963 static int blurayDemux(demux_t *p_demux)
964 {
965     demux_sys_t *p_sys = p_demux->p_sys;
966
967     block_t *p_block = block_New(p_demux, NB_TS_PACKETS * (int64_t)BD_TS_PACKET_SIZE);
968     if (!p_block) {
969         return -1;
970     }
971
972     int nread = -1;
973     if (p_sys->b_menu == false) {
974         blurayHandleEvents(p_demux);
975         nread = bd_read(p_sys->bluray, p_block->p_buffer,
976                 NB_TS_PACKETS * BD_TS_PACKET_SIZE);
977         if (nread < 0) {
978             block_Release(p_block);
979             return nread;
980         }
981     }
982     else {
983         BD_EVENT e;
984         nread = bd_read_ext( p_sys->bluray, p_block->p_buffer,
985                 NB_TS_PACKETS * BD_TS_PACKET_SIZE, &e );
986         if ( nread == 0 ) {
987             if ( e.event == BD_EVENT_NONE )
988                 msg_Info( p_demux, "We reached the end of a title" );
989             else
990                 blurayHandleEvent( p_demux, &e );
991             block_Release(p_block);
992             return 1;
993         }
994         if (p_sys->current_overlay != -1)
995         {
996             vlc_mutex_lock(&p_sys->p_overlays[p_sys->current_overlay]->lock);
997             if (p_sys->p_overlays[p_sys->current_overlay]->status == ToDisplay) {
998                 vlc_mutex_unlock(&p_sys->p_overlays[p_sys->current_overlay]->lock);
999                 if (p_sys->p_vout == NULL)
1000                     p_sys->p_vout = input_GetVout(p_sys->p_input);
1001                 if (p_sys->p_vout != NULL) {
1002                     var_AddCallback(p_sys->p_vout, "mouse-moved", &onMouseEvent, p_demux);
1003                     var_AddCallback(p_sys->p_vout, "mouse-clicked", &onMouseEvent, p_demux);
1004                     bluraySendOverlayToVout(p_demux);
1005                 }
1006             } else
1007                 vlc_mutex_unlock(&p_sys->p_overlays[p_sys->current_overlay]->lock);
1008         }
1009     }
1010
1011     p_block->i_buffer = nread;
1012
1013     stream_DemuxSend( p_sys->p_parser, p_block );
1014
1015     return 1;
1016 }