bluray.c 46.3 KB
Newer Older
1 2
/*
 * This file is part of libbluray
3
 * Copyright (C) 2009-2010  Obliter0n
4
 * Copyright (C) 2009-2010  John Stebbins
hpi1's avatar
hpi1 committed
5
 * Copyright (C) 2010       hpi1
6
 *
gates's avatar
gates committed
7 8 9 10
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
11
 *
gates's avatar
gates committed
12
 * This library is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
gates's avatar
gates committed
14 15
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
16
 *
gates's avatar
gates committed
17 18 19
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library. If not, see
 * <http://www.gnu.org/licenses/>.
20 21
 */

Accident's avatar
 
Accident committed
22 23 24 25
#if HAVE_CONFIG_H
#include "config.h"
#endif

26
#include "bluray.h"
hpi1's avatar
hpi1 committed
27
#include "register.h"
28 29 30 31 32
#include "util/macro.h"
#include "util/logging.h"
#include "util/strutl.h"
#include "bdnav/navigation.h"
#include "bdnav/index_parse.h"
hpi1's avatar
hpi1 committed
33
#include "hdmv/hdmv_vm.h"
34
#include "decoders/graphics_controller.h"
35
#include "file/file.h"
36
#ifdef DLOPEN_CRYPTO_LIBS
37 38
#include "file/dl.h"
#endif
39 40 41
#ifdef USING_BDJAVA
#include "bdj/bdj.h"
#endif
42

43
#ifndef DLOPEN_CRYPTO_LIBS
44
#include <libaacs/aacs.h>
45
#include <libbdplus/bdplus.h>
46
#endif
47
#include <stdlib.h>
48 49
#include <inttypes.h>
#include <string.h>
50

hpi1's avatar
hpi1 committed
51 52 53
typedef int     (*fptr_int)();
typedef int32_t (*fptr_int32)();
typedef void*   (*fptr_p_void)();
hpi1's avatar
hpi1 committed
54 55

#define MAX_EVENTS 31  /* 2^n - 1 */
hpi1's avatar
hpi1 committed
56
typedef struct bd_event_queue_s {
hpi1's avatar
hpi1 committed
57 58
    unsigned in;  /* next free slot */
    unsigned out; /* next event */
john's avatar
john committed
59
    BD_EVENT ev[MAX_EVENTS+1];
hpi1's avatar
hpi1 committed
60 61 62 63 64 65 66 67
} BD_EVENT_QUEUE;

typedef enum {
    title_undef = 0,
    title_hdmv,
    title_bdj,
} BD_TITLE_TYPE;

68 69 70 71 72 73 74 75 76 77 78 79 80
typedef struct {
    /* current clip */
    NAV_CLIP       *clip;
    BD_FILE_H      *fp;
    uint64_t       clip_size;
    uint64_t       clip_block_pos;
    uint64_t       clip_pos;

    /* current aligned unit */
    uint16_t       int_buf_off;

} BD_STREAM;

hpi1's avatar
hpi1 committed
81 82 83 84 85 86
typedef struct {
    NAV_CLIP *clip;
    uint64_t  clip_size;
    uint8_t  *buf;
} BD_PRELOAD;

hpi1's avatar
hpi1 committed
87 88 89 90 91 92 93 94 95
struct bluray {

    /* current disc */
    char           *device_path;
    INDX_ROOT      *index;
    NAV_TITLE_LIST *title_list;

    /* current playlist */
    NAV_TITLE      *title;
96
    uint32_t       title_idx;
hpi1's avatar
hpi1 committed
97 98
    uint64_t       s_pos;

99 100
    /* streams */
    BD_STREAM      st0; /* main path */
hpi1's avatar
hpi1 committed
101
    BD_PRELOAD     st_ig; /* preloaded IG stream sub path */
hpi1's avatar
hpi1 committed
102

103
    /* buffer for bd_read(): current aligned unit of main stream (st0) */
hpi1's avatar
hpi1 committed
104 105 106 107 108 109 110 111
    uint8_t        int_buf[6144];

    /* seamless angle change request */
    int            seamless_angle_change;
    uint32_t       angle_change_pkt;
    uint32_t       angle_change_time;
    unsigned       request_angle;

112
    /* chapter tracking */
john's avatar
john committed
113
    uint64_t       next_chapter_start;
114

hpi1's avatar
hpi1 committed
115
    /* aacs */
116
#ifdef DLOPEN_CRYPTO_LIBS
hpi1's avatar
hpi1 committed
117
    void           *h_libaacs;   // library handle
118
#endif
hpi1's avatar
hpi1 committed
119
    void           *aacs;
120
    fptr_p_void    libaacs_open;
hpi1's avatar
hpi1 committed
121 122 123
    fptr_int       libaacs_decrypt_unit;

    /* BD+ */
124
#ifdef DLOPEN_CRYPTO_LIBS
hpi1's avatar
hpi1 committed
125
    void           *h_libbdplus; // library handle
126
#endif
hpi1's avatar
hpi1 committed
127
    void           *bdplus;
128
    fptr_p_void    bdplus_init;
hpi1's avatar
hpi1 committed
129 130 131 132 133 134 135 136
    fptr_int32     bdplus_seek;
    fptr_int32     bdplus_fixup;

    /* player state */
    BD_REGISTERS   *regs;       // player registers
    BD_EVENT_QUEUE *event_queue; // navigation mode event queue
    BD_TITLE_TYPE  title_type;  // type of current title (in navigation mode)

hpi1's avatar
hpi1 committed
137 138 139
    HDMV_VM        *hdmv_vm;
    uint8_t        hdmv_suspended;

hpi1's avatar
hpi1 committed
140
    void           *bdjava;
141 142 143

    /* graphics */
    GRAPHICS_CONTROLLER *graphics_controller;
hpi1's avatar
hpi1 committed
144 145
};

146 147 148 149 150 151 152 153 154 155 156 157 158
#ifdef DLOPEN_CRYPTO_LIBS
#    define DL_CALL(lib,func,param,...)             \
     do {                                           \
          fptr_p_void fptr = dl_dlsym(lib, #func);  \
          if (fptr) {                               \
              fptr(param, ##__VA_ARGS__);           \
          }                                         \
      } while (0)
#else
#    define DL_CALL(lib,func,param,...)         \
     func (param, ##__VA_ARGS__)
#endif

hpi1's avatar
hpi1 committed
159 160 161 162
/*
 * Navigation mode event queue
 */

hpi1's avatar
hpi1 committed
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198
static void _init_event_queue(BLURAY *bd)
{
    if (!bd->event_queue) {
        bd->event_queue = calloc(1, sizeof(struct bd_event_queue_s));
    } else {
        memset(bd->event_queue, 0, sizeof(struct bd_event_queue_s));
    }
}

static int _get_event(BLURAY *bd, BD_EVENT *ev)
{
    struct bd_event_queue_s *eq = bd->event_queue;

    if (eq) {
        if (eq->in != eq->out) {
            *ev = eq->ev[eq->out];
            eq->out = (eq->out + 1) & MAX_EVENTS;
            return 1;
        }
    }

    ev->event = BD_EVENT_NONE;

    return 0;
}

static int _queue_event(BLURAY *bd, BD_EVENT ev)
{
    struct bd_event_queue_s *eq = bd->event_queue;

    if (eq) {
        unsigned new_in = (eq->in + 1) & MAX_EVENTS;

        if (new_in != eq->out) {
            eq->ev[eq->in] = ev;
            eq->in = new_in;
hpi1's avatar
hpi1 committed
199
            return 1;
hpi1's avatar
hpi1 committed
200 201 202 203 204
        }

        DEBUG(DBG_BLURAY|DBG_CRIT, "_queue_event(%d, %d): queue overflow !\n", ev.event, ev.param);
    }

hpi1's avatar
hpi1 committed
205
    return 0;
hpi1's avatar
hpi1 committed
206 207
}

208
/*
209
 * clip access (BD_STREAM)
210 211
 */

hpi1's avatar
hpi1 committed
212 213 214 215 216 217 218 219
static void _close_m2ts(BD_STREAM *st)
{
    if (st->fp != NULL) {
        file_close(st->fp);
        st->fp = NULL;
    }
}

220 221 222 223
static int _open_m2ts(BLURAY *bd, BD_STREAM *st)
{
    char *f_name;

hpi1's avatar
hpi1 committed
224 225
    _close_m2ts(st);

226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270
    f_name = str_printf("%s" DIR_SEP "BDMV" DIR_SEP "STREAM" DIR_SEP "%s",
                        bd->device_path, st->clip->name);

    st->clip_pos = (uint64_t)st->clip->start_pkt * 192;
    st->clip_block_pos = (st->clip_pos / 6144) * 6144;

    if ((st->fp = file_open(f_name, "rb"))) {
        file_seek(st->fp, 0, SEEK_END);
        if ((st->clip_size = file_tell(st->fp))) {
            file_seek(st->fp, st->clip_block_pos, SEEK_SET);
            st->int_buf_off = 6144;
            X_FREE(f_name);

            if (bd->bdplus) {
                DL_CALL(bd->h_libbdplus, bdplus_set_title,
                        bd->bdplus, st->clip->clip_id);
            }

            if (bd->aacs) {
                uint32_t title = bd_psr_read(bd->regs, PSR_TITLE_NUMBER);
                DL_CALL(bd->h_libaacs, aacs_select_title,
                        bd->aacs, title);
            }

            if (st == &bd->st0)
                bd_psr_write(bd->regs, PSR_PLAYITEM, st->clip->ref);

            return 1;
        }

        DEBUG(DBG_BLURAY, "Clip %s empty! (%p)\n", f_name, bd);
    }

    DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open clip %s! (%p)\n",
          f_name, bd);

    X_FREE(f_name);
    return 0;
}

static int _read_block(BLURAY *bd, BD_STREAM *st, uint8_t *buf)
{
    const int len = 6144;

    if (st->fp) {
271
        DEBUG(DBG_STREAM, "Reading unit [%d bytes] at %"PRIu64"... (%p)\n",
272 273 274 275 276 277 278
              len, st->clip_block_pos, bd);

        if (len + st->clip_block_pos <= st->clip_size) {
            int read_len;

            if ((read_len = file_read(st->fp, buf, len))) {
                if (read_len != len)
279
                    DEBUG(DBG_STREAM | DBG_CRIT, "Read %d bytes at %"PRIu64" ; requested %d ! (%p)\n", read_len, st->clip_block_pos, len, bd);
280

281
                if (bd->aacs && bd->libaacs_decrypt_unit) {
282
                    if (!bd->libaacs_decrypt_unit(bd->aacs, buf)) {
283
                        DEBUG(DBG_AACS | DBG_CRIT, "Unable decrypt unit (AACS)! (%p)\n", bd);
284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303

                        return 0;
                    } // decrypt
                } // aacs

                st->clip_block_pos += len;

                // bdplus fixup, if required.
                if (bd->bdplus_fixup && bd->bdplus) {
                    int32_t numFixes;
                    numFixes = bd->bdplus_fixup(bd->bdplus, len, buf);
#if 1
                    if (numFixes) {
                        DEBUG(DBG_BDPLUS,
                              "BDPLUS did %u fixups\n", numFixes);
                    }
#endif

                }

304
                DEBUG(DBG_STREAM, "Read unit OK! (%p)\n", bd);
305 306 307 308

                return 1;
            }

309
            DEBUG(DBG_STREAM | DBG_CRIT, "Read %d bytes at %"PRIu64" failed ! (%p)\n", len, st->clip_block_pos, bd);
310 311 312 313

            return 0;
        }

314
        DEBUG(DBG_STREAM | DBG_CRIT, "Read past EOF ! (%p)\n", bd);
315 316 317 318 319 320 321 322 323

        return 0;
    }

    DEBUG(DBG_BLURAY, "No valid title selected! (%p)\n", bd);

    return 0;
}

hpi1's avatar
hpi1 committed
324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375
/*
 * clip preload (BD_PRELOAD)
 */

static void _close_preload(BD_PRELOAD *p)
{
    X_FREE(p->buf);
    memset(p, 0, sizeof(*p));
}

static int _preload_m2ts(BLURAY *bd, BD_PRELOAD *p)
{
    /* setup and open BD_STREAM */

    BD_STREAM st;

    memset(&st, 0, sizeof(st));
    st.clip = p->clip;

    if (!_open_m2ts(bd, &st)) {
        return 0;
    }

    /* allocate buffer */
    p->clip_size = st.clip_size;
    p->buf       = realloc(p->buf, p->clip_size);

    /* read clip to buffer */

    uint8_t *buf = p->buf;
    uint8_t *end = p->buf + p->clip_size;

    for (; buf < end; buf += 6144) {
        if (!_read_block(bd, &st, buf)) {
            DEBUG(DBG_BLURAY|DBG_CRIT, "_preload_m2ts(): error loading %s at %"PRIu64"\n",
                  st.clip->name, (uint64_t)(buf - p->buf));
            _close_m2ts(&st);
            _close_preload(p);
            return 0;
        }
    }

    /* */

    DEBUG(DBG_BLURAY, "_preload_m2ts(): loaded %"PRIu64" bytes from %s\n",
          st.clip_size, st.clip->name);

    _close_m2ts(&st);

    return 1;
}

376 377 378 379 380 381
static int64_t _seek_stream(BLURAY *bd, BD_STREAM *st,
                            NAV_CLIP *clip, uint32_t clip_pkt)
{
    if (!clip)
        return -1;

382
    if (!st->fp || !st->clip || clip->ref != st->clip->ref) {
383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399
        // The position is in a new clip
        st->clip = clip;
        if (!_open_m2ts(bd, st)) {
            return -1;
        }
    }

    st->clip_pos = (uint64_t)clip_pkt * 192;
    st->clip_block_pos = (st->clip_pos / 6144) * 6144;

    file_seek(st->fp, st->clip_block_pos, SEEK_SET);

    st->int_buf_off = 6144;

    return st->clip_pos;
}

hpi1's avatar
hpi1 committed
400
/*
401
 * libaacs and libbdplus open / close
hpi1's avatar
hpi1 committed
402 403
 */

hpi1's avatar
hpi1 committed
404 405 406 407 408 409
static void _libaacs_close(BLURAY *bd)
{
    if (bd->aacs) {
        DL_CALL(bd->h_libaacs, aacs_close, bd->aacs);
        bd->aacs = NULL;
    }
410 411 412 413 414
}

static void _libaacs_unload(BLURAY *bd)
{
    _libaacs_close(bd);
hpi1's avatar
hpi1 committed
415 416 417 418 419 420 421 422

#ifdef DLOPEN_CRYPTO_LIBS
    if (bd->h_libaacs) {
        dl_dlclose(bd->h_libaacs);
        bd->h_libaacs = NULL;
    }
#endif

423
    bd->libaacs_open         = NULL;
hpi1's avatar
hpi1 committed
424 425 426
    bd->libaacs_decrypt_unit = NULL;
}

427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446
static int _libaacs_required(BLURAY *bd)
{
    BD_FILE_H *fd;
    char      *tmp;

    tmp = str_printf("%s/AACS/Unit_Key_RO.inf", bd->device_path);
    fd = file_open(tmp, "rb");
    X_FREE(tmp);

    if (fd) {
        file_close(fd);

        DEBUG(DBG_BLURAY, "AACS/Unit_Key_RO.inf found. Disc seems to be AACS protected (%p)\n", bd);
        return 1;
    }

    DEBUG(DBG_BLURAY, "AACS/Unit_Key_RO.inf not found. No AACS protection (%p)\n", bd);
    return 0;
}

447
static int _libaacs_load(BLURAY *bd)
448
{
449 450 451
#ifdef DLOPEN_CRYPTO_LIBS
    if (bd->h_libaacs) {
        return 1;
452 453
    }

454
    if ((bd->h_libaacs = dl_dlopen("libaacs", "0"))) {
455

456 457 458
        DEBUG(DBG_BLURAY, "Loading libaacs (%p)\n", bd->h_libaacs);

        bd->libaacs_open         = dl_dlsym(bd->h_libaacs, "aacs_open");
459 460
        bd->libaacs_decrypt_unit = dl_dlsym(bd->h_libaacs, "aacs_decrypt_unit");

461 462 463 464
        if (bd->libaacs_open && bd->libaacs_decrypt_unit) {
            DEBUG(DBG_BLURAY, "Loaded libaacs (%p)\n", bd->h_libaacs);
            return 1;

465
        } else {
466
            DEBUG(DBG_BLURAY, "libaacs dlsym failed! (%p)\n", bd->h_libaacs);
467
        }
468

469
    } else {
470
        DEBUG(DBG_BLURAY, "libaacs not found! (%p)\n", bd);
471
    }
472 473 474 475 476

    _libaacs_unload(bd);

    return 0;

477
#else
478
    DEBUG(DBG_BLURAY, "Using libaacs via normal linking\n");
479

480
    bd->libaacs_open         = &aacs_open;
481
    bd->libaacs_decrypt_unit = &aacs_decrypt_unit;
482

483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501
    return 1;
#endif
}

static int _libaacs_open(BLURAY *bd, const char *keyfile_path)
{
    _libaacs_close(bd);

    if (!_libaacs_required(bd)) {
        /* no AACS */
        return 1; /* no error if libaacs is not needed */
    }

    if (!_libaacs_load(bd)) {
        /* no libaacs */
        return 0;
    }

    bd->aacs = bd->libaacs_open(bd->device_path, keyfile_path);
502

503
    if (bd->aacs) {
504 505
        DEBUG(DBG_BLURAY, "Opened libaacs (%p)\n", bd->aacs);
        return 1;
506
    }
507

508
    DEBUG(DBG_BLURAY, "aacs_open() failed!\n");
509

510
    _libaacs_unload(bd);
511 512 513
    return 0;
}

hpi1's avatar
hpi1 committed
514 515 516 517 518 519
static void _libbdplus_close(BLURAY *bd)
{
    if (bd->bdplus) {
        DL_CALL(bd->h_libbdplus, bdplus_free, bd->bdplus);
        bd->bdplus = NULL;
    }
520 521 522 523 524
}

static void _libbdplus_unload(BLURAY *bd)
{
    _libbdplus_close(bd);
hpi1's avatar
hpi1 committed
525 526 527 528 529 530 531 532

#ifdef DLOPEN_CRYPTO_LIBS
    if (bd->h_libbdplus) {
        dl_dlclose(bd->h_libbdplus);
        bd->h_libbdplus = NULL;
    }
#endif

533
    bd->bdplus_init  = NULL;
hpi1's avatar
hpi1 committed
534 535 536 537
    bd->bdplus_seek  = NULL;
    bd->bdplus_fixup = NULL;
}

538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
static int _libbdplus_required(BLURAY *bd)
{
    BD_FILE_H *fd;
    char      *tmp;

    tmp = str_printf("%s/BDSVM/00000.svm", bd->device_path);
    fd = file_open(tmp, "rb");
    X_FREE(tmp);

    if (fd) {
        file_close(fd);

        DEBUG(DBG_BLURAY, "BDSVM/00000.svm found. Disc seems to be BD+ protected (%p)\n", bd);
        return 1;
    }

    DEBUG(DBG_BLURAY, "BDSVM/00000.svm not found. No BD+ protection (%p)\n", bd);
    return 0;
}

558
static int _libbdplus_load(BLURAY *bd)
559
{
560
    DEBUG(DBG_BDPLUS, "attempting to load libbdplus\n");
hpi1's avatar
hpi1 committed
561

562 563 564
#ifdef DLOPEN_CRYPTO_LIBS
    if (bd->h_libbdplus) {
        return 1;
565 566
    }

hpi1's avatar
hpi1 committed
567
    if ((bd->h_libbdplus = dl_dlopen("libbdplus", "0"))) {
568

569 570 571 572 573 574 575 576 577
        DEBUG(DBG_BLURAY, "Loading libbdplus (%p)\n", bd->h_libbdplus);

        bd->bdplus_init  = dl_dlsym(bd->h_libbdplus, "bdplus_init");
        bd->bdplus_seek  = dl_dlsym(bd->h_libbdplus, "bdplus_seek");
        bd->bdplus_fixup = dl_dlsym(bd->h_libbdplus, "bdplus_fixup");

        if (bd->bdplus_init && bd->bdplus_seek && bd->bdplus_fixup) {
            DEBUG(DBG_BLURAY, "Loaded libbdplus (%p)\n", bd->h_libbdplus);
            return 1;
hpi1's avatar
hpi1 committed
578
        }
579 580 581 582 583

        DEBUG(DBG_BLURAY, "libbdplus dlsym failed! (%p)\n", bd->h_libbdplus);

    } else {
        DEBUG(DBG_BLURAY, "libbdplus not found! (%p)\n", bd);
hpi1's avatar
hpi1 committed
584
    }
585 586 587 588 589

    _libbdplus_unload(bd);

    return 0;

590
#else
hpi1's avatar
hpi1 committed
591
    DEBUG(DBG_BLURAY,"Using libbdplus via normal linking\n");
592

593
    bd->bdplus_init  = &bdplus_init;
hpi1's avatar
hpi1 committed
594 595
    bd->bdplus_seek  = &bdplus_seek;
    bd->bdplus_fixup = &bdplus_fixup;
596 597

    return 1;
598
#endif
599
}
hpi1's avatar
hpi1 committed
600

601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631
static int _libbdplus_open(BLURAY *bd, const char *keyfile_path)
{
    // Take a quick stab to see if we want/need bdplus
    // we should fix this, and add various string functions.
    uint8_t vid[16] = {
        0xC5,0x43,0xEF,0x2A,0x15,0x0E,0x50,0xC4,0xE2,0xCA,
        0x71,0x65,0xB1,0x7C,0xA7,0xCB}; // FIXME

    _libbdplus_close(bd);

    if (!_libbdplus_required(bd)) {
        /* no BD+ */
        return 1; /* no error if libbdplus is not needed */
    }

    if (!_libbdplus_load(bd)) {
        /* no libbdplus */
        return 0;
    }

    bd->bdplus = bd->bdplus_init(bd->device_path, keyfile_path, vid);

    if (bd->bdplus) {
        DEBUG(DBG_BLURAY,"libbdplus initialized\n");
        return 1;
    }

    DEBUG(DBG_BLURAY,"bdplus_init() failed\n");

    _libbdplus_unload(bd);
    return 0;
632 633
}

634 635 636 637
/*
 * index open
 */

hpi1's avatar
hpi1 committed
638
static int _index_open(BLURAY *bd)
639
{
hpi1's avatar
hpi1 committed
640 641
    if (!bd->index) {
        char *file;
642

hpi1's avatar
hpi1 committed
643 644 645 646
        file = str_printf("%s/BDMV/index.bdmv", bd->device_path);
        bd->index = indx_parse(file);
        X_FREE(file);
    }
647

hpi1's avatar
hpi1 committed
648
    return !!bd->index;
649 650
}

cRTrn13's avatar
cRTrn13 committed
651 652
BLURAY *bd_open(const char* device_path, const char* keyfile_path)
{
653
    BLURAY *bd = calloc(1, sizeof(BLURAY));
cRTrn13's avatar
cRTrn13 committed
654

cRTrn13's avatar
cRTrn13 committed
655
    if (device_path) {
Accident's avatar
 
Accident committed
656

657 658
        bd->device_path = (char*)malloc(strlen(device_path) + 1);
        strcpy(bd->device_path, device_path);
cRTrn13's avatar
cRTrn13 committed
659

660
        _libaacs_open(bd, keyfile_path);
661

662
        _libbdplus_open(bd, keyfile_path);
663

664
        _index_open(bd);
665

hpi1's avatar
hpi1 committed
666 667
        bd->regs = bd_registers_init();

j45's avatar
j45 committed
668
        DEBUG(DBG_BLURAY, "BLURAY initialized! (%p)\n", bd);
cRTrn13's avatar
cRTrn13 committed
669
    } else {
cRTrn13's avatar
cRTrn13 committed
670 671
        X_FREE(bd);

cRTrn13's avatar
cRTrn13 committed
672
        DEBUG(DBG_BLURAY | DBG_CRIT, "No device path provided!\n");
cRTrn13's avatar
cRTrn13 committed
673 674 675 676 677 678 679
    }

    return bd;
}

void bd_close(BLURAY *bd)
{
hpi1's avatar
hpi1 committed
680 681
    bd_stop_bdj(bd);

682
    _libaacs_unload(bd);
683

684
    _libbdplus_unload(bd);
cRTrn13's avatar
cRTrn13 committed
685

hpi1's avatar
hpi1 committed
686
    _close_m2ts(&bd->st0);
hpi1's avatar
hpi1 committed
687
    _close_preload(&bd->st_ig);
cRTrn13's avatar
cRTrn13 committed
688

j45's avatar
j45 committed
689 690 691 692 693 694 695
    if (bd->title_list != NULL) {
        nav_free_title_list(bd->title_list);
    }
    if (bd->title != NULL) {
        nav_title_close(bd->title);
    }

696
    hdmv_vm_free(&bd->hdmv_vm);
697

698
    gc_free(&bd->graphics_controller);
699
    indx_free(&bd->index);
hpi1's avatar
hpi1 committed
700 701
    bd_registers_free(bd->regs);

hpi1's avatar
hpi1 committed
702
    X_FREE(bd->event_queue);
703 704
    X_FREE(bd->device_path);

j45's avatar
j45 committed
705
    DEBUG(DBG_BLURAY, "BLURAY destroyed! (%p)\n", bd);
cRTrn13's avatar
cRTrn13 committed
706 707

    X_FREE(bd);
cRTrn13's avatar
cRTrn13 committed
708
}
cRTrn13's avatar
cRTrn13 committed
709

hpi1's avatar
hpi1 committed
710 711 712 713
/*
 * seeking and current position
 */

714
static int64_t _seek_internal(BLURAY *bd,
715
                              NAV_CLIP *clip, uint32_t title_pkt, uint32_t clip_pkt)
716
{
717
    if (_seek_stream(bd, &bd->st0, clip, clip_pkt) >= 0) {
718

719 720
        /* update title position */
        bd->s_pos = (uint64_t)title_pkt * 192;
721

722 723 724 725
        /* chapter tracking */
        uint32_t current_chapter = bd_get_current_chapter(bd);
        bd->next_chapter_start = bd_chapter_pos(bd, current_chapter + 1);
        bd_psr_write(bd->regs, PSR_CHAPTER,  current_chapter + 1);
726

727 728
        DEBUG(DBG_BLURAY, "Seek to %"PRIu64" (%p)\n",
              bd->s_pos, bd);
729

730 731 732 733
        if (bd->bdplus_seek && bd->bdplus) {
            bd->bdplus_seek(bd->bdplus, bd->st0.clip_block_pos);
        }
    }
734 735 736 737

    return bd->s_pos;
}

738
/* _change_angle() should be used only before call to _seek_internal() ! */
739
static void _change_angle(BLURAY *bd)
740 741
{
    if (bd->seamless_angle_change) {
742
        bd->st0.clip = nav_set_angle(bd->title, bd->st0.clip, bd->request_angle);
743
        bd->seamless_angle_change = 0;
744
        bd_psr_write(bd->regs, PSR_ANGLE_NUMBER, bd->title->angle + 1);
745 746 747

        /* force re-opening .m2ts file in _seek_internal() */
        _close_m2ts(&bd->st0);
748
    }
749 750 751 752 753 754
}

int64_t bd_seek_time(BLURAY *bd, uint64_t tick)
{
    uint32_t clip_pkt, out_pkt;
    NAV_CLIP *clip;
hpi1's avatar
hpi1 committed
755 756 757 758

    tick /= 2;

    if (tick < bd->title->duration) {
759

760 761
        _change_angle(bd);

762 763 764
        // Find the closest access unit to the requested position
        clip = nav_time_search(bd->title, tick, &clip_pkt, &out_pkt);

765
        return _seek_internal(bd, clip, out_pkt, clip_pkt);
766 767 768 769 770
    }

    return bd->s_pos;
}

hpi1's avatar
hpi1 committed
771 772 773 774 775 776 777 778 779 780 781
uint64_t bd_tell_time(BLURAY *bd)
{
    uint32_t clip_pkt = 0, out_pkt = 0, out_time = 0;

    if (bd && bd->title) {
        nav_packet_search(bd->title, bd->s_pos / 192, &clip_pkt, &out_pkt, &out_time);
    }

    return ((uint64_t)out_time) * 2;
}

hpi1's avatar
hpi1 committed
782
int64_t bd_seek_chapter(BLURAY *bd, unsigned chapter)
j45's avatar
j45 committed
783 784 785 786
{
    uint32_t clip_pkt, out_pkt;
    NAV_CLIP *clip;

787 788
    if (chapter < bd->title->chap_list.count) {

789 790
        _change_angle(bd);

791 792
        // Find the closest access unit to the requested position
        clip = nav_chapter_search(bd->title, chapter, &clip_pkt, &out_pkt);
j45's avatar
j45 committed
793

794
        return _seek_internal(bd, clip, out_pkt, clip_pkt);
795 796 797
    }

    return bd->s_pos;
j45's avatar
j45 committed
798 799
}

hpi1's avatar
hpi1 committed
800
int64_t bd_chapter_pos(BLURAY *bd, unsigned chapter)
801 802 803
{
    uint32_t clip_pkt, out_pkt;

hpi1's avatar
hpi1 committed
804 805 806 807 808 809 810
    if (chapter < bd->title->chap_list.count) {
        // Find the closest access unit to the requested position
        nav_chapter_search(bd->title, chapter, &clip_pkt, &out_pkt);
        return (int64_t)out_pkt * 192;
    }

    return -1;
811 812
}

813 814
uint32_t bd_get_current_chapter(BLURAY *bd)
{
815
    return nav_chapter_get_current(bd->st0.clip, bd->st0.clip_pos / 192);
816 817
}

hpi1's avatar
hpi1 committed
818
int64_t bd_seek_mark(BLURAY *bd, unsigned mark)
j45's avatar
j45 committed
819 820 821 822
{
    uint32_t clip_pkt, out_pkt;
    NAV_CLIP *clip;

823 824
    if (mark < bd->title->mark_list.count) {

825 826
        _change_angle(bd);

827 828
        // Find the closest access unit to the requested position
        clip = nav_mark_search(bd->title, mark, &clip_pkt, &out_pkt);
j45's avatar
j45 committed
829

830
        return _seek_internal(bd, clip, out_pkt, clip_pkt);
831 832 833
    }

    return bd->s_pos;
j45's avatar
j45 committed
834 835
}

j45's avatar
j45 committed
836
int64_t bd_seek(BLURAY *bd, uint64_t pos)
cRTrn13's avatar
cRTrn13 committed
837
{
j45's avatar
j45 committed
838 839 840
    uint32_t pkt, clip_pkt, out_pkt, out_time;
    NAV_CLIP *clip;

hpi1's avatar
hpi1 committed
841
    if (pos < (uint64_t)bd->title->packets * 192) {
j45's avatar
j45 committed
842 843
        pkt = pos / 192;

844 845
        _change_angle(bd);

j45's avatar
j45 committed
846 847
        // Find the closest access unit to the requested position
        clip = nav_packet_search(bd->title, pkt, &clip_pkt, &out_pkt, &out_time);
848

849
        return _seek_internal(bd, clip, out_pkt, clip_pkt);
850 851 852 853 854
    }

    return bd->s_pos;
}

hpi1's avatar
hpi1 committed
855 856
uint64_t bd_get_title_size(BLURAY *bd)
{
hpi1's avatar
hpi1 committed
857 858 859 860
    if (bd && bd->title) {
        return (uint64_t)bd->title->packets * 192;
    }
    return UINT64_C(0);
hpi1's avatar
hpi1 committed
861 862 863 864 865 866 867 868 869 870 871
}

uint64_t bd_tell(BLURAY *bd)
{
    return bd ? bd->s_pos : INT64_C(0);
}

/*
 * read
 */

872
static int64_t _clip_seek_time(BLURAY *bd, uint64_t tick)
873 874 875
{
    uint32_t clip_pkt, out_pkt;

876
    if (tick < bd->st0.clip->out_time) {
877 878

        // Find the closest access unit to the requested position
879
        nav_clip_time_search(bd->st0.clip, tick, &clip_pkt, &out_pkt);
880

881
        return _seek_internal(bd, bd->st0.clip, out_pkt, clip_pkt);
cRTrn13's avatar
cRTrn13 committed
882 883 884 885 886 887 888
    }

    return bd->s_pos;
}

int bd_read(BLURAY *bd, unsigned char *buf, int len)
{
889
    BD_STREAM *st = &bd->st0;
j45's avatar
j45 committed
890
    int out_len;
cRTrn13's avatar
cRTrn13 committed
891

892
    if (st->fp) {
j45's avatar
j45 committed
893
        out_len = 0;
894
        DEBUG(DBG_STREAM, "Reading [%d bytes] at %"PRIu64"... (%p)\n", len, bd->s_pos, bd);
j45's avatar
j45 committed
895 896 897 898

        while (len > 0) {
            uint32_t clip_pkt;

899
            unsigned int size = len;
j45's avatar
j45 committed
900
            // Do we need to read more data?
901
            clip_pkt = st->clip_pos / 192;
902 903
            if (bd->seamless_angle_change) {
                if (clip_pkt >= bd->angle_change_pkt) {
904 905 906
                    if (clip_pkt >= st->clip->end_pkt) {
                        st->clip = nav_next_clip(bd->title, st->clip);
                        if (!_open_m2ts(bd, st)) {
907 908
                            return -1;
                        }
909
                        bd->s_pos = st->clip->pos;
910
                    } else {
911
                        _change_angle(bd);
912
                        _clip_seek_time(bd, bd->angle_change_time);
913 914 915
                    }
                    bd->seamless_angle_change = 0;
                } else {
916
                    uint64_t angle_pos;
917

918
                    angle_pos = bd->angle_change_pkt * 192;
919
                    if (angle_pos - st->clip_pos < size)
920
                    {
921
                        size = angle_pos - st->clip_pos;
922 923 924
                    }
                }
            }
925
            if (st->int_buf_off == 6144 || clip_pkt >= st->clip->end_pkt) {
j45's avatar
j45 committed
926 927

                // Do we need to get the next clip?
928
                if (st->clip == NULL) {
929 930
                    // We previously reached the last clip.  Nothing
                    // else to read.
hpi1's avatar
hpi1 committed
931
                    return 0;
932
                }
933 934 935
                if (clip_pkt >= st->clip->end_pkt) {
                    st->clip = nav_next_clip(bd->title, st->clip);
                    if (st->clip == NULL) {
936
                        DEBUG(DBG_BLURAY|DBG_STREAM, "End of title (%p)\n", bd);
j45's avatar
j45 committed
937 938
                        return out_len;
                    }
939
                    if (!_open_m2ts(bd, st)) {
j45's avatar
j45 committed
940 941 942
                        return -1;
                    }
                }
943
                if (_read_block(bd, st, bd->int_buf)) {
j45's avatar
j45 committed
944

945
                    st->int_buf_off = st->clip_pos % 6144;
j45's avatar
j45 committed
946 947 948 949 950

                } else {
                    return out_len;
                }
            }
951 952
            if (size > (unsigned int)6144 - st->int_buf_off) {
                size = 6144 - st->int_buf_off;
j45's avatar
j45 committed
953
            }
954
            memcpy(buf, bd->int_buf + st->int_buf_off, size);
j45's avatar
j45 committed
955 956 957
            buf += size;
            len -= size;
            out_len += size;
958 959
            st->clip_pos += size;
            st->int_buf_off += size;
j45's avatar
j45 committed
960 961
            bd->s_pos += size;
        }
962

963 964 965 966 967 968 969
        /* chapter tracking */
        if (bd->s_pos > bd->next_chapter_start) {
            uint32_t current_chapter = bd_get_current_chapter(bd);
            bd->next_chapter_start = bd_chapter_pos(bd, current_chapter + 1);
            bd_psr_write(bd->regs, PSR_CHAPTER, current_chapter + 1);
        }

970
        DEBUG(DBG_STREAM, "%d bytes read OK! (%p)\n", out_len, bd);
cRTrn13's avatar
cRTrn13 committed
971

j45's avatar
j45 committed
972 973
        return out_len;
    }
Accident's avatar
 
Accident committed
974

975
    DEBUG(DBG_STREAM | DBG_CRIT, "bd_read(): no valid title selected! (%p)\n", bd);
976

j45's avatar
j45 committed
977
    return -1;
cRTrn13's avatar
cRTrn13 committed
978
}
cRTrn13's avatar
cRTrn13 committed
979

hpi1's avatar
hpi1 committed
980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040
/*
 * preloader for asynchronous sub paths
 */

static int _find_ig_stream(BLURAY *bd, uint16_t *pid, int *sub_path_idx)
{
    MPLS_PI  *pi        = &bd->title->pl->play_item[0];
    unsigned  ig_stream = bd_psr_read(bd->regs, PSR_IG_STREAM_ID);
    unsigned  ii;

    for (ii = 0; ii < pi->stn.num_ig; ii++) {
        if (ii + 1 == ig_stream) {
            if (pi->stn.ig[ii].stream_type == 2) {
                *sub_path_idx = pi->stn.ig[ii].subpath_id;
            }
            *pid = pi->stn.ig[ii].pid;

            DEBUG(DBG_BLURAY, "_find_ig_stream(): current IG stream pid 0x%04x sub-path %d\n",
                  *pid, *sub_path_idx);
            return 1;
        }
    }

    return 0;
}

static int _preload_ig_subpath(BLURAY *bd)
{
    int      ig_subpath = -1;
    uint16_t ig_pid     = 0;

    _find_ig_stream(bd, &ig_pid, &ig_subpath);

    if (!bd->graphics_controller) {
        return 0;
    }

    if (ig_subpath < 0) {
        return 0;
    }

    bd->st_ig.clip = &bd->title->sub_path[ig_subpath].clip_list.clip[0];

    if (!_preload_m2ts(bd, &bd->st_ig)) {
        return 0;
    }

    gc_decode_ts(bd->graphics_controller, ig_pid, bd->st_ig.buf, bd->st_ig.clip_size / 6144, -1);

    return 1;
}

static int _preload_subpaths(BLURAY *bd)
{
    if (bd->title->pl->sub_count <= 0) {
        return 0;
    }

    return _preload_ig_subpath(bd);
}

hpi1's avatar
hpi1 committed
1041 1042 1043 1044
/*
 * select title / angle
 */

hpi1's avatar
hpi1 committed
1045
static void _close_playlist(BLURAY *bd)
cRTrn13's avatar
cRTrn13 committed
1046
{
1047 1048 1049 1050
    if (bd->graphics_controller) {
        gc_run(bd->graphics_controller, GC_CTRL_RESET, 0, NULL);
    }

hpi1's avatar
hpi1 committed
1051
    _close_m2ts(&bd->st0);
hpi1's avatar
hpi1 committed
1052
    _close_preload(&bd->st_ig);
hpi1's avatar
hpi1 committed
1053

1054 1055
    if (bd->title) {
        nav_title_close(bd->title);
hpi1's avatar
hpi1 committed
1056
        bd->title = NULL;
1057
    }
hpi1's avatar
hpi1 committed
1058 1059 1060 1061 1062
}

static int _open_playlist(BLURAY *bd, const char *f_name)
{
    _close_playlist(bd);
1063

j45's avatar
j45 committed
1064 1065
    bd->title = nav_title_open(bd->device_path, f_name);
    if (bd->title == NULL) {
1066
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (%p)\n",
j45's avatar
j45 committed
1067 1068 1069
              f_name, bd);
        return 0;
    }
cRTrn13's avatar
cRTrn13 committed
1070

1071
    bd->seamless_angle_change = 0;
cRTrn13's avatar
cRTrn13 committed
1072 1073
    bd->s_pos = 0;

1074 1075 1076 1077 1078 1079
    bd->next_chapter_start = bd_chapter_pos(bd, 1);

    bd_psr_write(bd->regs, PSR_PLAYLIST, atoi(bd->title->name));
    bd_psr_write(bd->regs, PSR_ANGLE_NUMBER, bd->title->angle + 1);
    bd_psr_write(bd->regs, PSR_CHAPTER, 1);

j45's avatar
j45 committed
1080
    // Get the initial clip of the playlist
1081 1082
    bd->st0.clip = nav_next_clip(bd->title, NULL);
    if (_open_m2ts(bd, &bd->st0)) {
j45's avatar
j45 committed
1083
        DEBUG(DBG_BLURAY, "Title %s selected! (%p)\n", f_name, bd);
hpi1's avatar
hpi1 committed
1084 1085 1086

        _preload_subpaths(bd);

j45's avatar
j45 committed
1087 1088 1089 1090
        return 1;
    }
    return 0;
}
1091

hpi1's avatar
hpi1 committed
1092 1093 1094 1095 1096
int bd_select_playlist(BLURAY *bd, uint32_t playlist)
{
    char *f_name = str_printf("%05d.mpls", playlist);
    int result;

1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
    if (bd->title_list) {
        /* update current title */
        unsigned i;
        for (i = 0; i < bd->title_list->count; i++) {
            if (playlist == bd->title_list->title_info[i].mpls_id) {
                bd->title_idx = i;
                break;
            }
        }
    }

hpi1's avatar
hpi1 committed
1108 1109 1110 1111 1112