bluray.c 39.6 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 "file/file.h"
35
#ifdef DLOPEN_CRYPTO_LIBS
36 37
#include "file/dl.h"
#endif
38 39 40
#ifdef USING_BDJAVA
#include "bdj/bdj.h"
#endif
41

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

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

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

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

67 68 69 70 71 72 73 74 75 76 77 78 79
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
80 81 82 83 84 85 86 87 88
struct bluray {

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

    /* current playlist */
    NAV_TITLE      *title;
89
    uint32_t       title_idx;
hpi1's avatar
hpi1 committed
90 91 92
    uint64_t       s_size;
    uint64_t       s_pos;

93 94
    /* streams */
    BD_STREAM      st0; /* main path */
hpi1's avatar
hpi1 committed
95

96
    /* buffer for bd_read(): current aligned unit of main stream (st0) */
hpi1's avatar
hpi1 committed
97 98 99 100 101 102 103 104
    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;

105 106 107
    /* chapter tracking */
    uint32_t       next_chapter_start;

hpi1's avatar
hpi1 committed
108
    /* aacs */
109
#ifdef DLOPEN_CRYPTO_LIBS
hpi1's avatar
hpi1 committed
110
    void           *h_libaacs;   // library handle
111
#endif
hpi1's avatar
hpi1 committed
112 113 114 115
    void           *aacs;
    fptr_int       libaacs_decrypt_unit;

    /* BD+ */
116
#ifdef DLOPEN_CRYPTO_LIBS
hpi1's avatar
hpi1 committed
117
    void           *h_libbdplus; // library handle
118
#endif
hpi1's avatar
hpi1 committed
119 120 121 122 123 124 125 126 127
    void           *bdplus;
    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
128 129 130
    HDMV_VM        *hdmv_vm;
    uint8_t        hdmv_suspended;

hpi1's avatar
hpi1 committed
131
    void           *bdjava;
hpi1's avatar
hpi1 committed
132 133
};

134 135 136 137 138 139 140 141 142 143 144 145 146
#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
147 148 149 150
/*
 * Navigation mode event queue
 */

hpi1's avatar
hpi1 committed
151 152 153 154 155 156 157 158 159 160 161 162 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
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;
            return 0;
        }

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

    return -1;
}

/*
hpi1's avatar
hpi1 committed
197
 * open / close
hpi1's avatar
hpi1 committed
198 199
 */

hpi1's avatar
hpi1 committed
200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216
static void _libaacs_close(BLURAY *bd)
{
    if (bd->aacs) {
        DL_CALL(bd->h_libaacs, aacs_close, bd->aacs);
        bd->aacs = NULL;
    }

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

    bd->libaacs_decrypt_unit = NULL;
}

217 218
static int _libaacs_open(BLURAY *bd, const char *keyfile_path)
{
hpi1's avatar
hpi1 committed
219 220
    _libaacs_close(bd);

221
#ifdef DLOPEN_CRYPTO_LIBS
222
    if ((bd->h_libaacs = dl_dlopen("libaacs", "0"))) {
223 224
        DEBUG(DBG_BLURAY, "Downloaded libaacs (%p)\n", bd->h_libaacs);

225
        fptr_p_void fptr = dl_dlsym(bd->h_libaacs, "aacs_open");
226 227 228 229 230 231 232 233 234 235 236 237
        bd->libaacs_decrypt_unit = dl_dlsym(bd->h_libaacs, "aacs_decrypt_unit");

        if (fptr && bd->libaacs_decrypt_unit) {
            if ((bd->aacs = fptr(bd->device_path, keyfile_path))) {
                DEBUG(DBG_BLURAY, "Opened libaacs (%p)\n", bd->aacs);
                return 1;
            }
            DEBUG(DBG_BLURAY, "aacs_open() failed!\n");
        } else {
            DEBUG(DBG_BLURAY, "libaacs dlsym failed!\n");
        }
        dl_dlclose(bd->h_libaacs);
238 239
        bd->h_libaacs = NULL;

240 241 242
    } else {
        DEBUG(DBG_BLURAY, "libaacs not found!\n");
    }
243
#else
244
    DEBUG(DBG_BLURAY, "Using libaacs via normal linking\n");
245

246
    bd->libaacs_decrypt_unit = &aacs_decrypt_unit;
247

248 249 250 251
    if ((bd->aacs = aacs_open(bd->device_path, keyfile_path))) {

        DEBUG(DBG_BLURAY, "Opened libaacs (%p)\n", bd->aacs);
        return 1;
252
    }
253
    DEBUG(DBG_BLURAY, "aacs_open() failed!\n");
254
#endif
255 256 257 258 259 260

    bd->libaacs_decrypt_unit = NULL;

    return 0;
}

hpi1's avatar
hpi1 committed
261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
static void _libbdplus_close(BLURAY *bd)
{
    if (bd->bdplus) {
        DL_CALL(bd->h_libbdplus, bdplus_free, bd->bdplus);
        bd->bdplus = NULL;
    }

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

    bd->bdplus_seek  = NULL;
    bd->bdplus_fixup = NULL;
}

279 280
static void _libbdplus_open(BLURAY *bd, const char *keyfile_path)
{
hpi1's avatar
hpi1 committed
281 282
    _libbdplus_close(bd);

283 284 285 286 287
    // 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
288
    BD_FILE_H *fd;
289 290 291 292 293 294
    char *tmp = NULL;
    tmp = str_printf("%s/BDSVM/00000.svm", bd->device_path);
    if ((fd = file_open(tmp, "rb"))) {
        file_close(fd);

        DEBUG(DBG_BDPLUS, "attempting to load libbdplus\n");
295
#ifdef DLOPEN_CRYPTO_LIBS
296
        if ((bd->h_libbdplus = dl_dlopen("libbdplus", "0"))) {
297
            DEBUG(DBG_BLURAY, "Downloaded libbdplus (%p)\n", bd->h_libbdplus);
298 299 300 301 302 303

            fptr_p_void bdplus_init = dl_dlsym(bd->h_libbdplus, "bdplus_init");
            //bdplus_t *bdplus_init(path,configfile_path,*vid );
            if (bdplus_init)
                bd->bdplus = bdplus_init(bd->device_path, keyfile_path, vid);

304 305 306 307 308 309 310 311 312
            if (bd->bdplus) {
                // Since we will call these functions a lot, we assign them
                // now.
                bd->bdplus_seek  = dl_dlsym(bd->h_libbdplus, "bdplus_seek");
                bd->bdplus_fixup = dl_dlsym(bd->h_libbdplus, "bdplus_fixup");
            } else {
                dl_dlclose(bd->h_libbdplus);
                bd->h_libbdplus = NULL;
            }
313 314 315 316 317 318 319 320 321 322 323 324 325 326 327
        }
#else
        DEBUG(DBG_BLURAY,"Using libbdplus via normal linking\n");

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

        // Since we will call these functions a lot, we assign them
        // now.
        bd->bdplus_seek  = &bdplus_seek;
        bd->bdplus_fixup = &bdplus_fixup;
#endif
    } // file_open
    X_FREE(tmp);
}

hpi1's avatar
hpi1 committed
328
static int _index_open(BLURAY *bd)
329 330 331 332 333 334 335
{
    char *file;

    file = str_printf("%s/BDMV/index.bdmv", bd->device_path);
    bd->index = indx_parse(file);
    X_FREE(file);

hpi1's avatar
hpi1 committed
336
    return !!bd->index;
337 338
}

cRTrn13's avatar
cRTrn13 committed
339 340
BLURAY *bd_open(const char* device_path, const char* keyfile_path)
{
341
    BLURAY *bd = calloc(1, sizeof(BLURAY));
cRTrn13's avatar
cRTrn13 committed
342

cRTrn13's avatar
cRTrn13 committed
343
    if (device_path) {
Accident's avatar
 
Accident committed
344

345 346
        bd->device_path = (char*)malloc(strlen(device_path) + 1);
        strcpy(bd->device_path, device_path);
cRTrn13's avatar
cRTrn13 committed
347

348
        _libaacs_open(bd, keyfile_path);
349

350
        _libbdplus_open(bd, keyfile_path);
351

352
        _index_open(bd);
353

hpi1's avatar
hpi1 committed
354 355
        bd->regs = bd_registers_init();

j45's avatar
j45 committed
356
        DEBUG(DBG_BLURAY, "BLURAY initialized! (%p)\n", bd);
cRTrn13's avatar
cRTrn13 committed
357
    } else {
cRTrn13's avatar
cRTrn13 committed
358 359
        X_FREE(bd);

cRTrn13's avatar
cRTrn13 committed
360
        DEBUG(DBG_BLURAY | DBG_CRIT, "No device path provided!\n");
cRTrn13's avatar
cRTrn13 committed
361 362 363 364 365 366 367
    }

    return bd;
}

void bd_close(BLURAY *bd)
{
hpi1's avatar
hpi1 committed
368 369
    bd_stop_bdj(bd);

hpi1's avatar
hpi1 committed
370
    _libaacs_close(bd);
371

hpi1's avatar
hpi1 committed
372
    _libbdplus_close(bd);
cRTrn13's avatar
cRTrn13 committed
373

374 375
    if (bd->st0.fp) {
        file_close(bd->st0.fp);
cRTrn13's avatar
cRTrn13 committed
376
    }
cRTrn13's avatar
cRTrn13 committed
377

j45's avatar
j45 committed
378 379 380 381 382 383 384
    if (bd->title_list != NULL) {
        nav_free_title_list(bd->title_list);
    }
    if (bd->title != NULL) {
        nav_title_close(bd->title);
    }

hpi1's avatar
hpi1 committed
385
    hdmv_vm_free(bd->hdmv_vm);
hpi1's avatar
hpi1 committed
386
    indx_free(bd->index);
hpi1's avatar
hpi1 committed
387 388
    bd_registers_free(bd->regs);

hpi1's avatar
hpi1 committed
389
    X_FREE(bd->event_queue);
390 391
    X_FREE(bd->device_path);

j45's avatar
j45 committed
392
    DEBUG(DBG_BLURAY, "BLURAY destroyed! (%p)\n", bd);
cRTrn13's avatar
cRTrn13 committed
393 394

    X_FREE(bd);
cRTrn13's avatar
cRTrn13 committed
395
}
cRTrn13's avatar
cRTrn13 committed
396

hpi1's avatar
hpi1 committed
397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
/*
 * clip access
 */

static int _open_m2ts(BLURAY *bd, BD_STREAM *st)
{
    char *f_name;

    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 != NULL) {
        file_close(st->fp);
    }
    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;
}

448
static int _read_block(BLURAY *bd, BD_STREAM *st, uint8_t *buf)
j45's avatar
j45 committed
449 450 451
{
    const int len = 6144;

452
    if (st->fp) {
j45's avatar
j45 committed
453
        DEBUG(DBG_BLURAY, "Reading unit [%d bytes] at %"PRIu64"... (%p)\n",
454
              len, st->clip_block_pos, bd);
j45's avatar
j45 committed
455

456
        if (len + st->clip_block_pos <= st->clip_size) {
j45's avatar
j45 committed
457 458
            int read_len;

459
            if ((read_len = file_read(st->fp, buf, len))) {
j45's avatar
j45 committed
460
                if (read_len != len)
461
                    DEBUG(DBG_BLURAY | DBG_CRIT, "Read %d bytes at %"PRIu64" ; requested %d ! (%p)\n", read_len, st->clip_block_pos, len, bd);
j45's avatar
j45 committed
462

463
                if (bd->libaacs_decrypt_unit) {
hpi1's avatar
hpi1 committed
464
                    if (!bd->libaacs_decrypt_unit(bd->aacs, buf)) {
465 466 467 468
                        DEBUG(DBG_BLURAY, "Unable decrypt unit! (%p)\n", bd);

                        return 0;
                    } // decrypt
j45's avatar
j45 committed
469 470
                } // aacs

471
                st->clip_block_pos += len;
j45's avatar
j45 committed
472 473 474 475

                // bdplus fixup, if required.
                if (bd->bdplus_fixup && bd->bdplus) {
                    int32_t numFixes;
hpi1's avatar
hpi1 committed
476
                    numFixes = bd->bdplus_fixup(bd->bdplus, len, buf);
Accident's avatar
Accident committed
477
#if 1
j45's avatar
j45 committed
478
                    if (numFixes) {
Accident's avatar
Accident committed
479
                        DEBUG(DBG_BDPLUS,
j45's avatar
j45 committed
480 481 482 483 484 485 486 487 488 489 490
                              "BDPLUS did %u fixups\n", numFixes);
                    }
#endif

                }

                DEBUG(DBG_BLURAY, "Read unit OK! (%p)\n", bd);

                return 1;
            }

491
            DEBUG(DBG_BLURAY | DBG_CRIT, "Read %d bytes at %"PRIu64" failed ! (%p)\n", len, st->clip_block_pos, bd);
j45's avatar
j45 committed
492 493 494 495 496 497 498 499 500 501 502 503 504 505

            return 0;
        }

        DEBUG(DBG_BLURAY | DBG_CRIT, "Read past EOF ! (%p)\n", bd);

        return 0;
    }

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

    return 0;
}

hpi1's avatar
hpi1 committed
506 507 508 509
/*
 * seeking and current position
 */

510 511
static int64_t _seek_internal(BLURAY *bd, BD_STREAM *st,
                              NAV_CLIP *clip, uint32_t title_pkt, uint32_t clip_pkt)
512 513 514 515
{
    if (!clip)
        return -1;

516
    if (clip->ref != st->clip->ref) {
517
        // The position is in a new clip
518 519
        st->clip = clip;
        if (!_open_m2ts(bd, st)) {
520 521 522 523
            return -1;
        }
    }
    bd->s_pos = (uint64_t)title_pkt * 192;
524 525
    st->clip_pos = (uint64_t)clip_pkt * 192;
    st->clip_block_pos = (st->clip_pos / 6144) * 6144;
526

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

529
    st->int_buf_off = 6144;
530

531 532 533 534 535
    /* 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);

536 537 538
    DEBUG(DBG_BLURAY, "Seek to %"PRIu64" (%p)\n",
          bd->s_pos, bd);
    if (bd->bdplus_seek && bd->bdplus)
539
        bd->bdplus_seek(bd->bdplus, st->clip_block_pos);
540 541 542 543

    return bd->s_pos;
}

544 545 546 547 548 549
int64_t bd_seek_time(BLURAY *bd, uint64_t tick)
{
    uint32_t clip_pkt, out_pkt;
    NAV_CLIP *clip;

    if (bd->seamless_angle_change) {
550
        bd->st0.clip = nav_set_angle(bd->title, bd->st0.clip, bd->request_angle);
551
        bd->seamless_angle_change = 0;
552
        bd_psr_write(bd->regs, PSR_ANGLE_NUMBER, bd->title->angle + 1);
553
    }
hpi1's avatar
hpi1 committed
554 555 556 557

    tick /= 2;

    if (tick < bd->title->duration) {
558 559 560 561

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

562
        return _seek_internal(bd, &bd->st0, clip, out_pkt, clip_pkt);
563 564 565 566 567
    }

    return bd->s_pos;
}

hpi1's avatar
hpi1 committed
568 569 570 571 572 573 574 575 576 577 578
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
579
int64_t bd_seek_chapter(BLURAY *bd, unsigned chapter)
j45's avatar
j45 committed
580 581 582 583
{
    uint32_t clip_pkt, out_pkt;
    NAV_CLIP *clip;

584 585 586 587
    if (chapter < bd->title->chap_list.count) {

        // 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
588

589
        return _seek_internal(bd, &bd->st0, clip, out_pkt, clip_pkt);
590 591 592
    }

    return bd->s_pos;
j45's avatar
j45 committed
593 594
}

hpi1's avatar
hpi1 committed
595
int64_t bd_chapter_pos(BLURAY *bd, unsigned chapter)
596 597 598
{
    uint32_t clip_pkt, out_pkt;

hpi1's avatar
hpi1 committed
599 600 601 602 603 604 605
    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;
606 607
}

608 609
uint32_t bd_get_current_chapter(BLURAY *bd)
{
610
    return nav_chapter_get_current(bd->st0.clip, bd->st0.clip_pos / 192);
611 612
}

hpi1's avatar
hpi1 committed
613
int64_t bd_seek_mark(BLURAY *bd, unsigned mark)
j45's avatar
j45 committed
614 615 616 617
{
    uint32_t clip_pkt, out_pkt;
    NAV_CLIP *clip;

618 619 620 621
    if (mark < bd->title->mark_list.count) {

        // 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
622

623
        return _seek_internal(bd, &bd->st0, clip, out_pkt, clip_pkt);
624 625 626
    }

    return bd->s_pos;
j45's avatar
j45 committed
627 628
}

j45's avatar
j45 committed
629
int64_t bd_seek(BLURAY *bd, uint64_t pos)
cRTrn13's avatar
cRTrn13 committed
630
{
j45's avatar
j45 committed
631 632 633
    uint32_t pkt, clip_pkt, out_pkt, out_time;
    NAV_CLIP *clip;

634
    if (bd->seamless_angle_change) {
635
        bd->st0.clip = nav_set_angle(bd->title, bd->st0.clip, bd->request_angle);
636
        bd->seamless_angle_change = 0;
637
        bd_psr_write(bd->regs, PSR_ANGLE_NUMBER, bd->title->angle + 1);
638
    }
cRTrn13's avatar
cRTrn13 committed
639
    if (pos < bd->s_size) {
j45's avatar
j45 committed
640 641 642 643
        pkt = pos / 192;

        // Find the closest access unit to the requested position
        clip = nav_packet_search(bd->title, pkt, &clip_pkt, &out_pkt, &out_time);
644

645
        return _seek_internal(bd, &bd->st0, clip, out_pkt, clip_pkt);
646 647 648 649 650
    }

    return bd->s_pos;
}

hpi1's avatar
hpi1 committed
651 652 653 654 655 656 657 658 659 660 661 662 663 664
uint64_t bd_get_title_size(BLURAY *bd)
{
    return bd ? bd->s_size : UINT64_C(0);
}

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

/*
 * read
 */

665
static int64_t _clip_seek_time(BLURAY *bd, BD_STREAM *st, uint64_t tick)
666 667 668
{
    uint32_t clip_pkt, out_pkt;

669
    if (tick < st->clip->out_time) {
670 671

        // Find the closest access unit to the requested position
672
        nav_clip_time_search(st->clip, tick, &clip_pkt, &out_pkt);
673

674
        return _seek_internal(bd, st, st->clip, out_pkt, clip_pkt);
cRTrn13's avatar
cRTrn13 committed
675 676 677 678 679 680 681
    }

    return bd->s_pos;
}

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

685
    if (st->fp) {
j45's avatar
j45 committed
686 687 688 689 690 691
        out_len = 0;
        DEBUG(DBG_BLURAY, "Reading [%d bytes] at %"PRIu64"... (%p)\n", len, bd->s_pos, bd);

        while (len > 0) {
            uint32_t clip_pkt;

692
            unsigned int size = len;
j45's avatar
j45 committed
693
            // Do we need to read more data?
694
            clip_pkt = st->clip_pos / 192;
695 696
            if (bd->seamless_angle_change) {
                if (clip_pkt >= bd->angle_change_pkt) {
697 698 699
                    if (clip_pkt >= st->clip->end_pkt) {
                        st->clip = nav_next_clip(bd->title, st->clip);
                        if (!_open_m2ts(bd, st)) {
700 701
                            return -1;
                        }
702
                        bd->s_pos = st->clip->pos;
703
                    } else {
704
                        st->clip = nav_set_angle(bd->title, st->clip, bd->request_angle);
705
                        bd_psr_write(bd->regs, PSR_ANGLE_NUMBER, bd->title->angle + 1);
706
                        _clip_seek_time(bd, st, bd->angle_change_time);
707 708 709
                    }
                    bd->seamless_angle_change = 0;
                } else {
710
                    uint64_t angle_pos;
711

712
                    angle_pos = bd->angle_change_pkt * 192;
713
                    if (angle_pos - st->clip_pos < size)
714
                    {
715
                        size = angle_pos - st->clip_pos;
716 717 718
                    }
                }
            }
719
            if (st->int_buf_off == 6144 || clip_pkt >= st->clip->end_pkt) {
j45's avatar
j45 committed
720 721

                // Do we need to get the next clip?
722
                if (st->clip == NULL) {
723 724
                    // We previously reached the last clip.  Nothing
                    // else to read.
hpi1's avatar
hpi1 committed
725
                    return 0;
726
                }
727 728 729
                if (clip_pkt >= st->clip->end_pkt) {
                    st->clip = nav_next_clip(bd->title, st->clip);
                    if (st->clip == NULL) {
j45's avatar
j45 committed
730
                        DEBUG(DBG_BLURAY, "End of title (%p)\n", bd);
j45's avatar
j45 committed
731 732
                        return out_len;
                    }
733
                    if (!_open_m2ts(bd, st)) {
j45's avatar
j45 committed
734 735 736
                        return -1;
                    }
                }
737
                if (_read_block(bd, st, bd->int_buf)) {
j45's avatar
j45 committed
738

739
                    st->int_buf_off = st->clip_pos % 6144;
j45's avatar
j45 committed
740 741 742 743 744

                } else {
                    return out_len;
                }
            }
745 746
            if (size > (unsigned int)6144 - st->int_buf_off) {
                size = 6144 - st->int_buf_off;
j45's avatar
j45 committed
747
            }
748
            memcpy(buf, bd->int_buf + st->int_buf_off, size);
j45's avatar
j45 committed
749 750 751
            buf += size;
            len -= size;
            out_len += size;
752 753
            st->clip_pos += size;
            st->int_buf_off += size;
j45's avatar
j45 committed
754 755
            bd->s_pos += size;
        }
756

757 758 759 760 761 762 763
        /* 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);
        }

j45's avatar
j45 committed
764
        DEBUG(DBG_BLURAY, "%d bytes read OK! (%p)\n", out_len, bd);
cRTrn13's avatar
cRTrn13 committed
765

j45's avatar
j45 committed
766 767
        return out_len;
    }
Accident's avatar
 
Accident committed
768

j45's avatar
j45 committed
769
    DEBUG(DBG_BLURAY, "No valid title selected! (%p)\n", bd);
770

j45's avatar
j45 committed
771
    return -1;
cRTrn13's avatar
cRTrn13 committed
772
}
cRTrn13's avatar
cRTrn13 committed
773

hpi1's avatar
hpi1 committed
774 775 776 777
/*
 * select title / angle
 */

hpi1's avatar
hpi1 committed
778
static int _open_playlist(BLURAY *bd, const char *f_name)
cRTrn13's avatar
cRTrn13 committed
779
{
780 781 782 783
    if (bd->title) {
        nav_title_close(bd->title);
    }

j45's avatar
j45 committed
784 785
    bd->title = nav_title_open(bd->device_path, f_name);
    if (bd->title == NULL) {
786
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (%p)\n",
j45's avatar
j45 committed
787 788 789
              f_name, bd);
        return 0;
    }
cRTrn13's avatar
cRTrn13 committed
790

791
    bd->seamless_angle_change = 0;
cRTrn13's avatar
cRTrn13 committed
792
    bd->s_pos = 0;
j45's avatar
j45 committed
793
    bd->s_size = (uint64_t)bd->title->packets * 192;
cRTrn13's avatar
cRTrn13 committed
794

795 796 797 798 799 800
    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
801
    // Get the initial clip of the playlist
802 803
    bd->st0.clip = nav_next_clip(bd->title, NULL);
    if (_open_m2ts(bd, &bd->st0)) {
j45's avatar
j45 committed
804
        DEBUG(DBG_BLURAY, "Title %s selected! (%p)\n", f_name, bd);
j45's avatar
j45 committed
805 806 807 808
        return 1;
    }
    return 0;
}
809

hpi1's avatar
hpi1 committed
810 811 812 813 814
int bd_select_playlist(BLURAY *bd, uint32_t playlist)
{
    char *f_name = str_printf("%05d.mpls", playlist);
    int result;

815 816 817 818 819 820 821 822 823 824 825
    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
826 827 828 829 830 831
    result = _open_playlist(bd, f_name);

    X_FREE(f_name);
    return result;
}

hpi1's avatar
hpi1 committed
832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848
// Select a title for playback
// The title index is an index into the list
// established by bd_get_titles()
int bd_select_title(BLURAY *bd, uint32_t title_idx)
{
    const char *f_name;

    // Open the playlist
    if (bd->title_list == NULL) {
        DEBUG(DBG_BLURAY, "Title list not yet read! (%p)\n", bd);
        return 0;
    }
    if (bd->title_list->count <= title_idx) {
        DEBUG(DBG_BLURAY, "Invalid title index %d! (%p)\n", title_idx, bd);
        return 0;
    }

849
    bd->title_idx = title_idx;
hpi1's avatar
hpi1 committed
850 851 852 853 854
    f_name = bd->title_list->title_info[title_idx].name;

    return _open_playlist(bd, f_name);
}

855 856 857 858 859
uint32_t bd_get_current_title(BLURAY *bd)
{
    return bd->title_idx;
}

hpi1's avatar
hpi1 committed
860
int bd_select_angle(BLURAY *bd, unsigned angle)
861 862 863 864 865
{
    if (bd->title == NULL) {
        DEBUG(DBG_BLURAY, "Title not yet selected! (%p)\n", bd);
        return 0;
    }
866
    bd->st0.clip = nav_set_angle(bd->title, bd->st0.clip, angle);
867 868
    bd_psr_write(bd->regs, PSR_ANGLE_NUMBER, bd->title->angle + 1);

869 870 871
    return 1;
}

872 873 874 875 876 877
unsigned bd_get_current_angle(BLURAY *bd)
{
    return bd->title->angle;
}


hpi1's avatar
hpi1 committed
878
void bd_seamless_angle_change(BLURAY *bd, unsigned angle)
879 880 881
{
    uint32_t clip_pkt;

882 883
    clip_pkt = (bd->st0.clip_pos + 191) / 192;
    bd->angle_change_pkt = nav_angle_change_search(bd->st0.clip, clip_pkt,
884 885 886 887 888
                                                   &bd->angle_change_time);
    bd->request_angle = angle;
    bd->seamless_angle_change = 1;
}

hpi1's avatar
hpi1 committed
889 890 891
/*
 * title lists
 */
hpi1's avatar
hpi1 committed
892

j45's avatar
j45 committed
893 894 895 896 897 898
uint32_t bd_get_titles(BLURAY *bd, uint8_t flags)
{
    if (!bd) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "bd_get_titles(NULL) failed (%p)\n", bd);
        return 0;
    }
899

j45's avatar
j45 committed
900 901
    if (bd->title_list != NULL) {
        nav_free_title_list(bd->title_list);
cRTrn13's avatar
cRTrn13 committed
902
    }
j45's avatar
j45 committed
903
    bd->title_list = nav_get_title_list(bd->device_path, flags);
cRTrn13's avatar
cRTrn13 committed
904

j45's avatar
j45 committed
905 906 907 908
    if (!bd->title_list) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "nav_get_title_list(%s) failed (%p)\n", bd->device_path, bd);
        return 0;
    }
909

j45's avatar
j45 committed
910
    return bd->title_list->count;
cRTrn13's avatar
cRTrn13 committed
911
}
912

john's avatar
john committed
913
static void _copy_streams(BLURAY_STREAM_INFO *streams, MPLS_STREAM *si, int count)
914 915 916 917 918 919 920 921 922
{
    int ii;

    for (ii = 0; ii < count; ii++) {
        streams[ii].coding_type = si[ii].coding_type;
        streams[ii].format = si[ii].format;
        streams[ii].rate = si[ii].rate;
        streams[ii].char_code = si[ii].char_code;
        memcpy(streams[ii].lang, si[ii].lang, 4);
hpi1's avatar
hpi1 committed
923
        streams[ii].pid = si[ii].pid;
924 925 926
    }
}

hpi1's avatar
hpi1 committed
927
static BLURAY_TITLE_INFO* _fill_title_info(NAV_TITLE* title, uint32_t title_idx, uint32_t playlist)
928
{
john's avatar
john committed
929
    BLURAY_TITLE_INFO *title_info;
930
    unsigned int ii;
j45's avatar
j45 committed
931

john's avatar
john committed
932
    title_info = calloc(1, sizeof(BLURAY_TITLE_INFO));
j45's avatar
j45 committed
933
    title_info->idx = title_idx;
934
    title_info->playlist = playlist;
j45's avatar
j45 committed
935 936 937
    title_info->duration = (uint64_t)title->duration * 2;
    title_info->angle_count = title->angle_count;
    title_info->chapter_count = title->chap_list.count;
john's avatar
john committed
938
    title_info->chapters = calloc(title_info->chapter_count, sizeof(BLURAY_TITLE_CHAPTER));
j45's avatar
j45 committed
939 940
    for (ii = 0; ii < title_info->chapter_count; ii++) {
        title_info->chapters[ii].idx = ii;
j45's avatar
j45 committed
941 942 943
        title_info->chapters[ii].start = (uint64_t)title->chap_list.mark[ii].title_time * 2;
        title_info->chapters[ii].duration = (uint64_t)title->chap_list.mark[ii].duration * 2;
        title_info->chapters[ii].offset = (uint64_t)title->chap_list.mark[ii].title_pkt * 192;
j45's avatar
j45 committed
944
    }
945
    title_info->clip_count = title->clip_list.count;
john's avatar
john committed
946
    title_info->clips = calloc(title_info->clip_count, sizeof(BLURAY_CLIP_INFO));
947 948
    for (ii = 0; ii < title_info->clip_count; ii++) {
        MPLS_PI *pi = &title->pl->play_item[ii];
john's avatar
john committed
949
        BLURAY_CLIP_INFO *ci = &title_info->clips[ii];
950 951 952 953 954 955
        ci->video_stream_count = pi->stn.num_video;
        ci->audio_stream_count = pi->stn.num_audio;
        ci->pg_stream_count = pi->stn.num_pg + pi->stn.num_pip_pg;
        ci->ig_stream_count = pi->stn.num_ig;
        ci->sec_video_stream_count = pi->stn.num_secondary_video;
        ci->sec_audio_stream_count = pi->stn.num_secondary_audio;
john's avatar
john committed
956 957 958 959 960 961
        ci->video_streams = calloc(ci->video_stream_count, sizeof(BLURAY_STREAM_INFO));
        ci->audio_streams = calloc(ci->audio_stream_count, sizeof(BLURAY_STREAM_INFO));
        ci->pg_streams = calloc(ci->pg_stream_count, sizeof(BLURAY_STREAM_INFO));
        ci->ig_streams = calloc(ci->ig_stream_count, sizeof(BLURAY_STREAM_INFO));
        ci->sec_video_streams = calloc(ci->sec_video_stream_count, sizeof(BLURAY_STREAM_INFO));
        ci->sec_audio_streams = calloc(ci->sec_audio_stream_count, sizeof(BLURAY_STREAM_INFO));
962 963 964 965 966 967 968
        _copy_streams(ci->video_streams, pi->stn.video, ci->video_stream_count);
        _copy_streams(ci->audio_streams, pi->stn.audio, ci->audio_stream_count);
        _copy_streams(ci->pg_streams, pi->stn.pg, ci->pg_stream_count);
        _copy_streams(ci->ig_streams, pi->stn.ig, ci->ig_stream_count);
        _copy_streams(ci->sec_video_streams, pi->stn.secondary_video, ci->sec_video_stream_count);
        _copy_streams(ci->sec_audio_streams, pi->stn.secondary_audio, ci->sec_audio_stream_count);
    }
969 970 971 972

    return title_info;
}

973
static BLURAY_TITLE_INFO *_get_title_info(BLURAY *bd, uint32_t title_idx, uint32_t playlist, const char *mpls_name)
974 975 976 977
{
    NAV_TITLE *title;
    BLURAY_TITLE_INFO *title_info;

978 979 980 981 982 983 984 985 986 987 988 989 990 991 992
    title = nav_title_open(bd->device_path, mpls_name);
    if (title == NULL) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (%p)\n",
              mpls_name, bd);
        return NULL;
    }

    title_info = _fill_title_info(title, title_idx, playlist);

    nav_title_close(title);
    return title_info;
}

BLURAY_TITLE_INFO* bd_get_title_info(BLURAY *bd, uint32_t title_idx)
{
993 994 995 996 997 998 999 1000 1001
    if (bd->title_list == NULL) {
        DEBUG(DBG_BLURAY, "Title list not yet read! (%p)\n", bd);
        return NULL;
    }
    if (bd->title_list->count <= title_idx) {
        DEBUG(DBG_BLURAY, "Invalid title index %d! (%p)\n", title_idx, bd);
        return NULL;
    }

1002 1003 1004
    return _get_title_info(bd,
                           title_idx, bd->title_list->title_info[title_idx].mpls_id,
                           bd->title_list->title_info[title_idx].name);
1005 1006 1007 1008 1009 1010 1011
}

BLURAY_TITLE_INFO* bd_get_playlist_info(BLURAY *bd, uint32_t playlist)
{
    char *f_name = str_printf("%05d.mpls", playlist);
    BLURAY_TITLE_INFO *title_info;

1012
    title_info = _get_title_info(bd, 0, playlist, f_name);
1013

hpi1's avatar
hpi1 committed
1014 1015
    X_FREE(f_name);

j45's avatar
j45 committed
1016
    return title_info;
1017
}
cRTrn13's avatar
cRTrn13 committed
1018

john's avatar
john committed
1019
void bd_free_title_info(BLURAY_TITLE_INFO *title_info)
cRTrn13's avatar
cRTrn13 committed
1020
{
1021 1022
    unsigned int ii;

j45's avatar
j45 committed
1023
    X_FREE(title_info->chapters);
1024 1025 1026 1027 1028 1029 1030 1031 1032
    for (ii = 0; ii < title_info->clip_count; ii++) {
        X_FREE(title_info->clips[ii].video_streams);
        X_FREE(title_info->clips[ii].audio_streams);
        X_FREE(title_info->clips[ii].pg_streams);
        X_FREE(title_info->clips[ii].ig_streams);
        X_FREE(title_info->clips[ii].sec_video_streams);
        X_FREE(title_info->clips[ii].sec_audio_streams);
    }
    X_FREE(title_info->clips);
j45's avatar
j45 committed
1033
    X_FREE(title_info);
cRTrn13's avatar
cRTrn13 committed
1034
}
j45's avatar
j45 committed
1035

hpi1's avatar
hpi1 committed
1036 1037 1038 1039 1040 1041 1042
/*
 * player settings
 */

int bd_set_player_setting(BLURAY *bd, uint32_t idx, uint32_t value)
{
    static const struct { uint32_t idx; uint32_t  psr; } map[] = {
john's avatar
john committed
1043 1044 1045 1046 1047 1048 1049 1050 1051 1052
        { BLURAY_PLAYER_SETTING_PARENTAL,       PSR_PARENTAL },
        { BLURAY_PLAYER_SETTING_AUDIO_CAP,      PSR_AUDIO_CAP },
        { BLURAY_PLAYER_SETTING_AUDIO_LANG,     PSR_AUDIO_LANG },
        { BLURAY_PLAYER_SETTING_PG_LANG,        PSR_PG_AND_SUB_LANG },
        { BLURAY_PLAYER_SETTING_MENU_LANG,      PSR_MENU_LANG },
        { BLURAY_PLAYER_SETTING_COUNTRY_CODE,   PSR_COUNTRY },
        { BLURAY_PLAYER_SETTING_REGION_CODE,    PSR_REGION },
        { BLURAY_PLAYER_SETTING_VIDEO_CAP,      PSR_VIDEO_CAP },
        { BLURAY_PLAYER_SETTING_TEXT_CAP,       PSR_TEXT_CAP },
        { BLURAY_PLAYER_SETTING_PLAYER_PROFILE, PSR_PROFILE_VERSION },
hpi1's avatar
hpi1 committed
1053 1054 1055 1056
    };

    unsigned i;

john's avatar
john committed