stream/stream_bluray.c

   1 /*
   2  * Copyright (C) 2010 Benjamin Zores <ben@geexbox.org>
   3  *
   4  * This file is part of MPlayer.
   5  *
   6  * MPlayer is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * MPlayer is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 /*
  22  * Blu-ray parser/reader using libbluray
  23  *  Use 'git clone git://git.videolan.org/libbluray' to get it.
  24  *
  25  * TODO:
  26  *  - Add libbdnav support for menus navigation
  27  *  - Add AACS/BD+ protection detection
  28  *  - Add descrambled keys database support (KEYDB.cfg)
  29  *
  30  */
  31
  32 #include <libbluray/bluray.h>
  33
  34 #include "config.h"
  35 #include "libavutil/common.h"
  36 #include "libmpdemux/demuxer.h"
  37 #include "mp_msg.h"
  38 #include "m_struct.h"
  39 #include "m_option.h"
  40 #include "stream.h"
  41
  42 #define BLURAY_SECTOR_SIZE     6144
  43
  44 #define BLURAY_DEFAULT_ANGLE      0
  45 #define BLURAY_DEFAULT_CHAPTER    0
  46 #define BLURAY_DEFAULT_TITLE      0
  47
  48 char *bluray_device  = NULL;
  49 int   bluray_angle   = 0;
  50 int   bluray_chapter = 0;
  51
  52 struct bluray_priv_s {
  53     BLURAY *bd;
  54     int current_angle;
  55     int current_chapter;
  56     int current_title;
  57 };
  58
  59 static struct stream_priv_s {
  60     int title;
  61     char *device;
  62 } bluray_stream_priv_dflts = {
  63     BLURAY_DEFAULT_TITLE,
  64     NULL
  65 };
  66
  67 #define ST_OFF(f) M_ST_OFF(struct stream_priv_s,f)
  68 static const m_option_t bluray_stream_opts_fields[] = {
  69     { "hostname", ST_OFF(title),  CONF_TYPE_INT, M_OPT_RANGE, 0, 99999, NULL},
  70     { "filename", ST_OFF(device), CONF_TYPE_STRING, 0, 0 ,0, NULL},
  71     { NULL, NULL, 0, 0, 0, 0,  NULL }
  72 };
  73
  74 static const struct m_struct_st bluray_stream_opts = {
  75     "bluray",
  76     sizeof(struct stream_priv_s),
  77     &bluray_stream_priv_dflts,
  78     bluray_stream_opts_fields
  79 };
  80
  81 static void bluray_stream_close(stream_t *s)
  82 {
  83     struct bluray_priv_s *b = s->priv;
  84
  85     bd_close(b->bd);
  86     b->bd = NULL;
  87     free(b);
  88 }
  89
  90 static int bluray_stream_seek(stream_t *s, off_t pos)
  91 {
  92     struct bluray_priv_s *b = s->priv;
  93     off_t p;
  94
  95     p = bd_seek(b->bd, pos);
  96     if (p == -1)
  97         return 0;
  98
  99     s->pos = p;
 100     return 1;
 101 }
 102
 103 static int bluray_stream_fill_buffer(stream_t *s, char *buf, int len)
 104 {
 105     struct bluray_priv_s *b = s->priv;
 106
 107     return bd_read(b->bd, buf, len);
 108 }
 109
 110 static int bluray_stream_control(stream_t *s, int cmd, void *arg)
 111 {
 112     struct bluray_priv_s *b = s->priv;
 113
 114     switch (cmd) {
 115
 116     case STREAM_CTRL_GET_NUM_CHAPTERS: {
 117         BLURAY_TITLE_INFO *ti;
 118
 119         ti = bd_get_title_info(b->bd, b->current_title, b->current_angle);
 120         if (!ti)
 121             return STREAM_UNSUPPORTED;
 122
 123         *((unsigned int *) arg) = ti->chapter_count;
 124         bd_free_title_info(ti);
 125
 126         return 1;
 127     }
 128
 129     case STREAM_CTRL_GET_CURRENT_CHAPTER: {
 130         *((unsigned int *) arg) = b->current_chapter;
 131         return 1;
 132     }
 133
 134     case STREAM_CTRL_SEEK_TO_CHAPTER: {
 135         BLURAY_TITLE_INFO *ti;
 136         int chapter = *((unsigned int *) arg);
 137         int64_t pos;
 138         int r;
 139
 140         ti = bd_get_title_info(b->bd, b->current_title, b->current_angle);
 141         if (!ti)
 142             return STREAM_UNSUPPORTED;
 143
 144         if (chapter < 0 || chapter > ti->chapter_count) {
 145             bd_free_title_info(ti);
 146             return STREAM_UNSUPPORTED;
 147         }
 148
 149         pos = bd_chapter_pos(b->bd, chapter);
 150         r = bluray_stream_seek(s, pos);
 151         bd_free_title_info(ti);
 152
 153         return r ? 1 : STREAM_UNSUPPORTED;
 154     }
 155
 156     case STREAM_CTRL_GET_NUM_ANGLES: {
 157         BLURAY_TITLE_INFO *ti;
 158
 159         ti = bd_get_title_info(b->bd, b->current_title, b->current_angle);
 160         if (!ti)
 161             return STREAM_UNSUPPORTED;
 162
 163         *((int *) arg) = ti->angle_count;
 164         bd_free_title_info(ti);
 165
 166         return 1;
 167     }
 168
 169     case STREAM_CTRL_GET_ANGLE: {
 170         *((int *) arg) = b->current_angle;
 171         return 1;
 172     }
 173
 174     case STREAM_CTRL_SET_ANGLE: {
 175         BLURAY_TITLE_INFO *ti;
 176         int angle = *((int *) arg);
 177
 178         ti = bd_get_title_info(b->bd, b->current_title, b->current_angle);
 179         if (!ti)
 180             return STREAM_UNSUPPORTED;
 181
 182         if (angle < 0 || angle > ti->angle_count) {
 183             bd_free_title_info(ti);
 184             return STREAM_UNSUPPORTED;
 185         }
 186
 187         b->current_angle = angle;
 188         bd_seamless_angle_change(b->bd, angle);
 189         bd_free_title_info(ti);
 190
 191         return 1;
 192     }
 193
 194     default:
 195         break;
 196     }
 197
 198     return STREAM_UNSUPPORTED;
 199 }
 200
 201 static int bluray_stream_open(stream_t *s, int mode,
 202                               void *opts, int *file_format)
 203 {
 204     struct stream_priv_s *p = opts;
 205     struct bluray_priv_s *b;
 206
 207     BLURAY_TITLE_INFO *info = NULL;
 208     BLURAY *bd;
 209
 210     int title, title_guess, title_count;
 211     uint64_t title_size;
 212
 213     unsigned int chapter = 0, angle = 0;
 214     uint64_t max_duration = 0;
 215     int64_t chapter_pos = 0;
 216
 217     char *device = NULL;
 218     int i;
 219
 220     /* find the requested device */
 221     if (p->device)
 222         device = p->device;
 223     else if (bluray_device)
 224         device = bluray_device;
 225
 226     if (!device) {
 227         mp_tmsg(MSGT_OPEN, MSGL_ERR,
 228                 "No Blu-ray device/location was specified ...\n");
 229         return STREAM_UNSUPPORTED;
 230     }
 231
 232     /* open device */
 233     bd = bd_open(device, NULL);
 234     if (!bd) {
 235         mp_tmsg(MSGT_OPEN, MSGL_ERR, "Couldn't open Blu-ray device: %s\n",
 236                device);
 237         return STREAM_UNSUPPORTED;
 238     }
 239
 240     /* check for available titles on disc */
 241     title_count = bd_get_titles(bd, TITLES_RELEVANT, angle);
 242     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_BLURAY_TITLES=%d\n", title_count);
 243     if (!title_count) {
 244         mp_msg(MSGT_OPEN, MSGL_ERR,
 245                "Can't find any Blu-ray-compatible title here.\n");
 246         bd_close(bd);
 247         return STREAM_UNSUPPORTED;
 248     }
 249
 250     /* parse titles information */
 251     title_guess = BLURAY_DEFAULT_TITLE;
 252     for (i = 0; i < title_count; i++) {
 253         BLURAY_TITLE_INFO *ti;
 254         int sec, msec;
 255
 256         ti = bd_get_title_info(bd, i, angle);
 257         if (!ti)
 258             continue;
 259
 260         sec  = ti->duration / 90000;
 261         msec = (ti->duration - sec) % 1000;
 262
 263         mp_msg(MSGT_IDENTIFY, MSGL_INFO,
 264                "ID_BLURAY_TITLE_%d_CHAPTERS=%d\n", i + 1, ti->chapter_count);
 265         mp_msg(MSGT_IDENTIFY, MSGL_INFO,
 266                "ID_BLURAY_TITLE_%d_ANGLE=%d\n", i + 1, ti->angle_count);
 267         mp_msg(MSGT_IDENTIFY, MSGL_V,
 268                "ID_BLURAY_TITLE_%d_LENGTH=%d.%03d\n", i + 1, sec, msec);
 269
 270         /* try to guess which title may contain the main movie */
 271         if (ti->duration > max_duration) {
 272             max_duration = ti->duration;
 273             title_guess = i;
 274         }
 275
 276         bd_free_title_info(ti);
 277     }
 278
 279     /* Select current title */
 280     title = p->title ? p->title - 1: title_guess;
 281     title = FFMIN(title, title_count - 1);
 282
 283     bd_select_title(bd, title);
 284
 285     title_size = bd_get_title_size(bd);
 286     mp_msg(MSGT_IDENTIFY, MSGL_INFO,
 287            "ID_BLURAY_CURRENT_TITLE=%d\n", title + 1);
 288
 289     /* Get current title information */
 290     info = bd_get_title_info(bd, title, angle);
 291     if (!info)
 292         goto err_no_info;
 293
 294     /* Select chapter */
 295     chapter = bluray_chapter ? bluray_chapter : BLURAY_DEFAULT_CHAPTER;
 296     chapter = FFMIN(chapter, info->chapter_count);
 297
 298     if (chapter)
 299         chapter_pos = bd_chapter_pos(bd, chapter);
 300
 301     mp_msg(MSGT_IDENTIFY, MSGL_INFO,
 302            "ID_BLURAY_CURRENT_CHAPTER=%d\n", chapter + 1);
 303
 304     /* Select angle */
 305     angle = bluray_angle ? bluray_angle : BLURAY_DEFAULT_ANGLE;
 306     angle = FFMIN(angle, info->angle_count);
 307
 308     if (angle)
 309         bd_select_angle(bd, angle);
 310
 311     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_BLURAY_CURRENT_ANGLE=%d\n", angle + 1);
 312
 313     bd_free_title_info(info);
 314
 315 err_no_info:
 316     s->fill_buffer = bluray_stream_fill_buffer;
 317     s->seek        = bluray_stream_seek;
 318     s->close       = bluray_stream_close;
 319     s->control     = bluray_stream_control;
 320
 321     b                  = calloc(1, sizeof(struct bluray_priv_s));
 322     b->bd              = bd;
 323     b->current_angle   = angle;
 324     b->current_chapter = chapter;
 325     b->current_title   = title;
 326
 327     s->start_pos   = chapter_pos;
 328     s->end_pos     = title_size;
 329     s->sector_size = BLURAY_SECTOR_SIZE;
 330     s->flags       = mode | MP_STREAM_SEEK;
 331     s->priv        = b;
 332     s->type        = STREAMTYPE_BLURAY;
 333     s->url         = strdup("br://");
 334
 335     mp_tmsg(MSGT_OPEN, MSGL_V, "Blu-ray successfully opened.\n");
 336
 337     return STREAM_OK;
 338 }
 339
 340 const stream_info_t stream_info_bluray = {
 341     "Blu-ray Disc",
 342     "bd",
 343     "Benjamin Zores",
 344     "Play Blu-ray discs through external libbluray",
 345     bluray_stream_open,
 346     { "br", "bluray", NULL },
 347     &bluray_stream_opts,
 348     1
 349 };