bluray.c 13 KB
Newer Older
Accident's avatar
 
Accident committed
1 2 3 4 5 6 7 8 9 10 11
#if HAVE_CONFIG_H
#include "config.h"
#endif

#if HAVE_MALLOC_H
#include <malloc.h>
#endif

#if HAVE_STDLIB_H
#include <stdlib.h>
#endif
cRTrn13's avatar
cRTrn13 committed
12

Accident's avatar
 
Accident committed
13 14 15 16
#if HAVE_INTTYPES_H
#include <inttypes.h>
#endif

cRTrn13's avatar
cRTrn13 committed
17
#include <stdio.h>
cRTrn13's avatar
cRTrn13 committed
18

cRTrn13's avatar
cRTrn13 committed
19
#include "bluray.h"
cRTrn13's avatar
cRTrn13 committed
20
#include "util/macro.h"
cRTrn13's avatar
cRTrn13 committed
21
#include "util/logging.h"
22 23
#include "util/strutl.h"
#include "file/dl.h"
j45's avatar
j45 committed
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
#include "libbdnav/navigation.h"

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

    f_name = str_printf("%s" DIR_SEP "BDMV" DIR_SEP "STREAM" DIR_SEP "%s",
                        bd->device_path, bd->clip->name);

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

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

            if (bd->h_libbdplus && bd->bdplus) {
                fptr_p_void bdplus_set_title;
                bdplus_set_title = dl_dlsym(bd->h_libbdplus, "bdplus_set_title");
                if (bdplus_set_title)
                    bdplus_set_title(bd->bdplus, bd->clip->clip_id);
            }
            return 1;
        }

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

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

    X_FREE(f_name);
    return 0;
}
cRTrn13's avatar
cRTrn13 committed
64

cRTrn13's avatar
cRTrn13 committed
65 66
BLURAY *bd_open(const char* device_path, const char* keyfile_path)
{
67
    BLURAY *bd = calloc(1, sizeof(BLURAY));
cRTrn13's avatar
cRTrn13 committed
68

cRTrn13's avatar
cRTrn13 committed
69
    if (device_path) {
Accident's avatar
 
Accident committed
70 71

        bd->device_path = strdup(device_path);
cRTrn13's avatar
cRTrn13 committed
72

cRTrn13's avatar
cRTrn13 committed
73 74 75 76
        bd->aacs = NULL;
        bd->h_libaacs = NULL;
        bd->fp = NULL;

cRTrn13's avatar
Fix  
cRTrn13 committed
77
        if (keyfile_path) {
78 79
            //if ((bd->h_libaacs = dlopen("libaacs.so", RTLD_LAZY))) {
            if ((bd->h_libaacs = dl_dlopen("aacs"))) {
cRTrn13's avatar
cRTrn13 committed
80 81
                fptr_p_void fptr;

cRTrn13's avatar
cRTrn13 committed
82
                DEBUG(DBG_BLURAY, "Downloaded libaacs (0x%08x)\n", bd->h_libaacs);
cRTrn13's avatar
cRTrn13 committed
83

Accident's avatar
 
Accident committed
84
                fptr = dl_dlsym(bd->h_libaacs, "aacs_open");
85
                bd->aacs = fptr(device_path, keyfile_path);
cRTrn13's avatar
cRTrn13 committed
86
            } else {
87
                DEBUG(DBG_BLURAY, "libaacs not found!\n");
cRTrn13's avatar
cRTrn13 committed
88 89
            }
        } else {
cRTrn13's avatar
cRTrn13 committed
90
            DEBUG(DBG_BLURAY | DBG_CRIT, "No keyfile provided. You will not be able to make use of crypto functionality (0x%08x)\n", bd);
cRTrn13's avatar
cRTrn13 committed
91 92
        }

93 94 95 96

        // Take a quick stab to see if we want/need bdplus
        // we should fix this, and add various string functions.
        {
Accident's avatar
 
Accident committed
97 98 99
            uint8_t vid[16] = {
                0xC5,0x43,0xEF,0x2A,0x15,0x0E,0x50,0xC4,0xE2,0xCA,
                0x71,0x65,0xB1,0x7C,0xA7,0xCB}; // FIXME
100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
            FILE_H *fd;
            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");
                if ((bd->h_libbdplus = dl_dlopen("bdplus"))) {
                    DEBUG(DBG_BLURAY, "Downloaded libbdplus (0x%08x)\n",
                          bd->h_libbdplus);

                    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(device_path, keyfile_path, vid);

                    // 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");

                } // dlopen
            } // file_open
            X_FREE(tmp);
        }


127

cRTrn13's avatar
Fix  
cRTrn13 committed
128
        DEBUG(DBG_BLURAY, "BLURAY initialized! (0x%08x)\n", bd);
cRTrn13's avatar
cRTrn13 committed
129
    } else {
cRTrn13's avatar
cRTrn13 committed
130 131
        X_FREE(bd);

cRTrn13's avatar
cRTrn13 committed
132
        DEBUG(DBG_BLURAY | DBG_CRIT, "No device path provided!\n");
cRTrn13's avatar
cRTrn13 committed
133 134 135 136 137 138 139
    }

    return bd;
}

void bd_close(BLURAY *bd)
{
140 141
    if (bd->h_libaacs && bd->aacs) {
        fptr_p_void fptr = dl_dlsym(bd->h_libaacs, "aacs_close");
Accident's avatar
 
Accident committed
142
        fptr(bd->aacs);  // FIXME: NULL
cRTrn13's avatar
cRTrn13 committed
143

144
        dl_dlclose(bd->h_libaacs);
cRTrn13's avatar
cRTrn13 committed
145 146
    }

147 148 149 150 151 152 153
    if (bd->h_libbdplus && bd->bdplus) {
        fptr_p_void bdplus_free = dl_dlsym(bd->h_libbdplus, "bdplus_free");
        if (bdplus_free) bdplus_free(bd->bdplus);
        bd->bdplus = NULL;

        dl_dlclose(bd->h_libbdplus);
        bd->h_libbdplus = NULL;
cRTrn13's avatar
cRTrn13 committed
154

155 156
        bd->bdplus_seek  = NULL;
        bd->bdplus_fixup = NULL;
157
    }
cRTrn13's avatar
cRTrn13 committed
158

cRTrn13's avatar
cRTrn13 committed
159 160 161
    if (bd->fp) {
        file_close(bd->fp);
    }
cRTrn13's avatar
cRTrn13 committed
162

j45's avatar
j45 committed
163 164 165 166 167 168 169
    if (bd->title_list != NULL) {
        nav_free_title_list(bd->title_list);
    }
    if (bd->title != NULL) {
        nav_title_close(bd->title);
    }

170 171
    X_FREE(bd->device_path);

cRTrn13's avatar
cRTrn13 committed
172
    DEBUG(DBG_BLURAY, "BLURAY destroyed! (0x%08x)\n", bd);
cRTrn13's avatar
cRTrn13 committed
173 174

    X_FREE(bd);
cRTrn13's avatar
cRTrn13 committed
175
}
cRTrn13's avatar
cRTrn13 committed
176

j45's avatar
j45 committed
177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238
int _read_block(BLURAY *bd)
{
    const int len = 6144;

    if (bd->fp) {
        DEBUG(DBG_BLURAY, "Reading unit [%d bytes] at %"PRIu64"... (%p)\n",
              len, bd->clip_block_pos, bd);

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

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

                if (bd->h_libaacs && bd->aacs) {
                    // FIXME: calling dlsym for every read() call.
                    if ((bd->libaacs_decrypt_unit = dl_dlsym(bd->h_libaacs, "aacs_decrypt_unit"))) {
                        if (!bd->libaacs_decrypt_unit(bd->aacs, bd->int_buf, len, bd->clip_block_pos)) {
                            DEBUG(DBG_BLURAY, "Unable decrypt unit! (0x%08x)\n", bd);

                            return 0;
                        } // decrypt
                    } // dlsym
                } // aacs

                bd->clip_block_pos += len;

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

                }

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

                return 1;
            }

            DEBUG(DBG_BLURAY | DBG_CRIT, "Read %d bytes at %"PRIu64" failed ! (%p)\n", len, bd->clip_block_pos, bd);

            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;
}

int64_t bd_seek(BLURAY *bd, uint64_t pos)
cRTrn13's avatar
cRTrn13 committed
239
{
j45's avatar
j45 committed
240 241 242
    uint32_t pkt, clip_pkt, out_pkt, out_time;
    NAV_CLIP *clip;

cRTrn13's avatar
cRTrn13 committed
243
    if (pos < bd->s_size) {
j45's avatar
j45 committed
244 245 246 247 248 249 250 251 252 253 254 255 256 257
        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);
        if (clip->ref != bd->clip->ref) {
            // The position is in a new clip
            bd->clip = clip;
            if (!_open_m2ts(bd)) {
                return -1;
            }
        }
        bd->s_pos = (uint64_t)out_pkt * 192;
        bd->clip_pos = (uint64_t)clip_pkt * 192;
        bd->clip_block_pos = (bd->clip_pos / 6144) * 6144;
cRTrn13's avatar
cRTrn13 committed
258

j45's avatar
j45 committed
259
        file_seek(bd->fp, bd->clip_block_pos, SEEK_SET);
260

261 262
        bd->int_buf_off = 6144;

Accident's avatar
 
Accident committed
263 264
        DEBUG(DBG_BLURAY, "Seek to %"PRIu64" (0x%08x)\n",
              bd->s_pos, bd);
265
        if (bd->bdplus_seek && bd->bdplus)
Accident's avatar
 
Accident committed
266
            bd->bdplus_seek(bd->bdplus, bd->s_pos);
267

j45's avatar
j45 committed
268
        return bd->s_pos;
cRTrn13's avatar
cRTrn13 committed
269 270 271 272 273 274 275
    }

    return bd->s_pos;
}

int bd_read(BLURAY *bd, unsigned char *buf, int len)
{
j45's avatar
j45 committed
276 277
    int size;
    int out_len;
cRTrn13's avatar
cRTrn13 committed
278

j45's avatar
j45 committed
279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321
    if (bd->fp) {
        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;

            // Do we need to read more data?
            clip_pkt = bd->clip_pos / 192;
            if (bd->int_buf_off == 6144 || clip_pkt >= bd->clip->end_pkt) {

                // Do we need to get the next clip?
                if (clip_pkt >= bd->clip->end_pkt) {
                    bd->clip = nav_next_clip(bd->title, bd->clip);
                    if (bd->clip == NULL) {
                        DEBUG(DBG_BLURAY, "End of title (0x%08x)\n", bd);
                        return out_len;
                    }
                    if (!_open_m2ts(bd)) {
                        return -1;
                    }
                }
                if (_read_block(bd)) {

                    bd->int_buf_off = bd->clip_pos % 6144;

                } else {
                    return out_len;
                }
            }
            if (len > 6144 - bd->int_buf_off) {
                size = 6144 - bd->int_buf_off;
            } else {
                size = len;
            }
            memcpy(buf, bd->int_buf + bd->int_buf_off, size);
            buf += size;
            len -= size;
            out_len += size;
            bd->clip_pos += size;
            bd->int_buf_off += size;
            bd->s_pos += size;
        }
322

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

j45's avatar
j45 committed
325 326
        return out_len;
    }
Accident's avatar
 
Accident committed
327

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

j45's avatar
j45 committed
330
    return -1;
cRTrn13's avatar
cRTrn13 committed
331
}
cRTrn13's avatar
cRTrn13 committed
332

j45's avatar
j45 committed
333 334 335 336
// 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)
cRTrn13's avatar
cRTrn13 committed
337
{
j45's avatar
j45 committed
338
    char *f_name;
cRTrn13's avatar
cRTrn13 committed
339

j45's avatar
j45 committed
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355
    // Open the playlist
    if (bd->title_list == NULL) {
        DEBUG(DBG_BLURAY, "Title list not yet read! (0x%08x)\n", bd);
        return 0;
    }
    if (bd->title_list->count <= title_idx) {
        DEBUG(DBG_BLURAY, "Invalid title index %d! (0x%08x)\n", title_idx, bd);
        return 0;
    }
    f_name = bd->title_list->title_info[title_idx].name;
    bd->title = nav_title_open(bd->device_path, f_name);
    if (bd->title == NULL) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (0x%08x)\n", 
              f_name, bd);
        return 0;
    }
cRTrn13's avatar
cRTrn13 committed
356 357

    bd->s_pos = 0;
j45's avatar
j45 committed
358
    bd->s_size = (uint64_t)bd->title->packets * 192;
cRTrn13's avatar
cRTrn13 committed
359

j45's avatar
j45 committed
360 361 362 363 364 365 366 367
    // Get the initial clip of the playlist
    bd->clip = nav_next_clip(bd->title, NULL);
    if (_open_m2ts(bd)) {
        DEBUG(DBG_BLURAY, "Title %s selected! (0x%08x)\n", f_name, bd);
        return 1;
    }
    return 0;
}
368

j45's avatar
j45 committed
369 370 371 372
uint64_t bd_get_title_size(BLURAY *bd)
{
    return bd ? bd->s_size : UINT64_C(0);
}
cRTrn13's avatar
cRTrn13 committed
373

j45's avatar
j45 committed
374 375 376 377
uint64_t bd_tell(BLURAY *bd)
{
    return bd ? bd->s_pos : INT64_C(0);
}
378

j45's avatar
j45 committed
379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
// This must be called after bd_open() and before bd_select_title().
// Populates the title list in BLURAY.
// Filtering of the returned list is controled throught flags
//  TITLES_ALL              - all titles
//  TITLES_FILTER_DUP_TITLE - remove duplicate titles
//  TITLES_FILTER_DUP_CLIP  - remove titles that have duplicated clips
//  TITLES_RELEVANT         - remove dup titles and clips
//
//  Returns the number of titles found
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;
    }
394

j45's avatar
j45 committed
395 396
    if (bd->title_list != NULL) {
        nav_free_title_list(bd->title_list);
cRTrn13's avatar
cRTrn13 committed
397
    }
j45's avatar
j45 committed
398
    bd->title_list = nav_get_title_list(bd->device_path, flags);
cRTrn13's avatar
cRTrn13 committed
399

j45's avatar
j45 committed
400 401 402 403
    if (!bd->title_list) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "nav_get_title_list(%s) failed (%p)\n", bd->device_path, bd);
        return 0;
    }
404

j45's avatar
j45 committed
405
    return bd->title_list->count;
cRTrn13's avatar
cRTrn13 committed
406
}
407

j45's avatar
j45 committed
408
BD_TITLE_INFO* bd_get_title_info(BLURAY *bd, uint32_t title_idx)
409
{
j45's avatar
j45 committed
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
    NAV_TITLE *title;
    BD_TITLE_INFO *title_info;
    int ii;

    if (bd->title_list == NULL) {
        DEBUG(DBG_BLURAY, "Title list not yet read! (0x%08x)\n", bd);
        return NULL;
    }
    if (bd->title_list->count <= title_idx) {
        DEBUG(DBG_BLURAY, "Invalid title index %d! (0x%08x)\n", title_idx, bd);
        return NULL;
    }
    title = nav_title_open(bd->device_path, bd->title_list->title_info[title_idx].name);
    if (title == NULL) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (0x%08x)\n", 
              bd->title_list->title_info[title_idx].name, bd);
        return NULL;
    }
    title_info = calloc(1, sizeof(BD_TITLE_INFO));
    title_info->idx = title_idx;
    title_info->duration = (uint64_t)title->duration * 2;
    title_info->angle_count = title->angle_count;
    title_info->chapter_count = title->chap_list.count;
    title_info->chapters = calloc(title_info->chapter_count, sizeof(BD_TITLE_CHAPTER));
    for (ii = 0; ii < title_info->chapter_count; ii++) {
        title_info->chapters[ii].idx = ii;
        title_info->chapters[ii].start = (uint64_t)title->chap_list.chapter[ii].title_time * 2;
        title_info->chapters[ii].duration = (uint64_t)title->chap_list.chapter[ii].duration * 2;
        title_info->chapters[ii].offset = (uint64_t)title->chap_list.chapter[ii].title_pkt * 192;
    }
    nav_title_close(title);
    return title_info;
442
}
cRTrn13's avatar
cRTrn13 committed
443

j45's avatar
j45 committed
444
void bd_free_title_info(BD_TITLE_INFO *title_info)
cRTrn13's avatar
cRTrn13 committed
445
{
j45's avatar
j45 committed
446 447
    X_FREE(title_info->chapters);
    X_FREE(title_info);
cRTrn13's avatar
cRTrn13 committed
448
}
j45's avatar
j45 committed
449