git.sesse.net Git - ffmpeg/blob - libavformat/bluray.c

   1 /*
   2  * BluRay (libbluray) protocol
   3  *
   4  * Copyright (c) 2012 Petri Hintukainen <phintuka <at> users.sourceforge.net>
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * FFmpeg 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 GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with FFmpeg; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include <libbluray/bluray.h>
  24
  25 #include "libavutil/avstring.h"
  26 #include "libavformat/avformat.h"
  27 #include "libavformat/url.h"
  28 #include "libavutil/opt.h"
  29
  30 #define BLURAY_PROTO_PREFIX     "bluray:"
  31 #define MIN_PLAYLIST_LENGTH     180     /* 3 min */
  32
  33 typedef struct {
  34     const AVClass *class;
  35
  36     BLURAY *bd;
  37
  38     int playlist;
  39     int angle;
  40     int chapter;
  41     /*int region;*/
  42 } BlurayContext;
  43
  44 #define OFFSET(x) offsetof(BlurayContext, x)
  45 static const AVOption options[] = {
  46 {"playlist", "", OFFSET(playlist), AV_OPT_TYPE_INT, { .i64=-1 }, -1,  99999, AV_OPT_FLAG_DECODING_PARAM },
  47 {"angle",    "", OFFSET(angle),    AV_OPT_TYPE_INT, { .i64=0 },   0,   0xfe, AV_OPT_FLAG_DECODING_PARAM },
  48 {"chapter",  "", OFFSET(chapter),  AV_OPT_TYPE_INT, { .i64=1 },   1, 0xfffe, AV_OPT_FLAG_DECODING_PARAM },
  49 /*{"region",   "bluray player region code (1 = region A, 2 = region B, 4 = region C)", OFFSET(region), AV_OPT_TYPE_INT, { .i64=0 }, 0, 3, AV_OPT_FLAG_DECODING_PARAM },*/
  50 {NULL}
  51 };
  52
  53 static const AVClass bluray_context_class = {
  54     .class_name     = "bluray",
  55     .item_name      = av_default_item_name,
  56     .option         = options,
  57     .version        = LIBAVUTIL_VERSION_INT,
  58 };
  59
  60
  61 static int check_disc_info(URLContext *h)
  62 {
  63     BlurayContext *bd = h->priv_data;
  64     const BLURAY_DISC_INFO *disc_info;
  65
  66     disc_info = bd_get_disc_info(bd->bd);
  67     if (!disc_info) {
  68         av_log(h, AV_LOG_ERROR, "bd_get_disc_info() failed\n");
  69         return -1;
  70     }
  71
  72     if (!disc_info->bluray_detected) {
  73         av_log(h, AV_LOG_ERROR, "BluRay disc not detected\n");
  74         return -1;
  75     }
  76
  77     /* AACS */
  78     if (disc_info->aacs_detected && !disc_info->aacs_handled) {
  79         if (!disc_info->libaacs_detected) {
  80             av_log(h, AV_LOG_ERROR,
  81                    "Media stream encrypted with AACS, install and configure libaacs\n");
  82         } else {
  83             av_log(h, AV_LOG_ERROR, "Your libaacs can't decrypt this media\n");
  84         }
  85         return -1;
  86     }
  87
  88     /* BD+ */
  89     if (disc_info->bdplus_detected && !disc_info->bdplus_handled) {
  90         /*
  91         if (!disc_info->libbdplus_detected) {
  92             av_log(h, AV_LOG_ERROR,
  93                    "Media stream encrypted with BD+, install and configure libbdplus");
  94         } else {
  95         */
  96             av_log(h, AV_LOG_ERROR, "Unable to decrypt BD+ encrypted media\n");
  97         /*}*/
  98         return -1;
  99     }
 100
 101     return 0;
 102 }
 103
 104 static int bluray_close(URLContext *h)
 105 {
 106     BlurayContext *bd = h->priv_data;
 107     if (bd->bd) {
 108         bd_close(bd->bd);
 109     }
 110
 111     return 0;
 112 }
 113
 114 static int bluray_open(URLContext *h, const char *path, int flags)
 115 {
 116     BlurayContext *bd = h->priv_data;
 117     int num_title_idx;
 118     const char *diskname = path;
 119
 120     av_strstart(path, BLURAY_PROTO_PREFIX, &diskname);
 121
 122     bd->bd = bd_open(diskname, NULL);
 123     if (!bd->bd) {
 124         av_log(h, AV_LOG_ERROR, "bd_open() failed\n");
 125         return AVERROR(EIO);
 126     }
 127
 128     /* check if disc can be played */
 129     if (check_disc_info(h) < 0) {
 130         return AVERROR(EIO);
 131     }
 132
 133     /* setup player registers */
 134     /* region code has no effect without menus
 135     if (bd->region > 0 && bd->region < 5) {
 136         av_log(h, AV_LOG_INFO, "setting region code to %d (%c)\n", bd->region, 'A' + (bd->region - 1));
 137         bd_set_player_setting(bd->bd, BLURAY_PLAYER_SETTING_REGION_CODE, bd->region);
 138     }
 139     */
 140
 141     /* load title list */
 142     num_title_idx = bd_get_titles(bd->bd, TITLES_RELEVANT, MIN_PLAYLIST_LENGTH);
 143     av_log(h, AV_LOG_INFO, "%d usable playlists:\n", num_title_idx);
 144     if (num_title_idx < 1) {
 145         return AVERROR(EIO);
 146     }
 147
 148     /* if playlist was not given, select longest playlist */
 149     if (bd->playlist < 0) {
 150         uint64_t duration = 0;
 151         int i;
 152         for (i = 0; i < num_title_idx; i++) {
 153             BLURAY_TITLE_INFO *info = bd_get_title_info(bd->bd, i, 0);
 154
 155             av_log(h, AV_LOG_INFO, "playlist %05d.mpls (%d:%02d:%02d)\n",
 156                    info->playlist,
 157                    ((int)(info->duration / 90000) / 3600),
 158                    ((int)(info->duration / 90000) % 3600) / 60,
 159                    ((int)(info->duration / 90000) % 60));
 160
 161             if (info->duration > duration) {
 162                 bd->playlist = info->playlist;
 163                 duration = info->duration;
 164             }
 165
 166             bd_free_title_info(info);
 167         }
 168         av_log(h, AV_LOG_INFO, "selected %05d.mpls\n", bd->playlist);
 169     }
 170
 171     /* select playlist */
 172     if (bd_select_playlist(bd->bd, bd->playlist) <= 0) {
 173         av_log(h, AV_LOG_ERROR, "bd_select_playlist(%05d.mpls) failed\n", bd->playlist);
 174         return AVERROR(EIO);
 175     }
 176
 177     /* select angle */
 178     if (bd->angle >= 0) {
 179         bd_select_angle(bd->bd, bd->angle);
 180     }
 181
 182     /* select chapter */
 183     if (bd->chapter > 1) {
 184         bd_seek_chapter(bd->bd, bd->chapter - 1);
 185     }
 186
 187     return 0;
 188 }
 189
 190 static int bluray_read(URLContext *h, unsigned char *buf, int size)
 191 {
 192     BlurayContext *bd = h->priv_data;
 193     int len;
 194
 195     if (!bd || !bd->bd) {
 196         return AVERROR(EFAULT);
 197     }
 198
 199     len = bd_read(bd->bd, buf, size);
 200
 201     return len == 0 ? AVERROR_EOF : len;
 202 }
 203
 204 static int64_t bluray_seek(URLContext *h, int64_t pos, int whence)
 205 {
 206     BlurayContext *bd = h->priv_data;
 207
 208     if (!bd || !bd->bd) {
 209         return AVERROR(EFAULT);
 210     }
 211
 212     switch (whence) {
 213     case SEEK_SET:
 214     case SEEK_CUR:
 215     case SEEK_END:
 216         return bd_seek(bd->bd, pos);
 217
 218     case AVSEEK_SIZE:
 219         return bd_get_title_size(bd->bd);
 220     }
 221
 222     av_log(h, AV_LOG_ERROR, "Unsupported whence operation %d\n", whence);
 223     return AVERROR(EINVAL);
 224 }
 225
 226
 227 const URLProtocol ff_bluray_protocol = {
 228     .name            = "bluray",
 229     .url_close       = bluray_close,
 230     .url_open        = bluray_open,
 231     .url_read        = bluray_read,
 232     .url_seek        = bluray_seek,
 233     .priv_data_size  = sizeof(BlurayContext),
 234     .priv_data_class = &bluray_context_class,
 235 };