bluray.c 13.2 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
#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;
        }

j45's avatar
j45 committed
55
        DEBUG(DBG_BLURAY, "Clip %s empty! (%p)\n", f_name, bd);
j45's avatar
j45 committed
56 57
    }

j45's avatar
j45 committed
58
    DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open clip %s! (%p)\n", 
j45's avatar
j45 committed
59 60 61 62 63
          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;

j45's avatar
j45 committed
82
                DEBUG(DBG_BLURAY, "Downloaded libaacs (%p)\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 {
j45's avatar
j45 committed
90
            DEBUG(DBG_BLURAY | DBG_CRIT, "No keyfile provided. You will not be able to make use of crypto functionality (%p)\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
            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"))) {
j45's avatar
j45 committed
108
                    DEBUG(DBG_BLURAY, "Downloaded libbdplus (%p)\n",
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
                          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

j45's avatar
j45 committed
128
        DEBUG(DBG_BLURAY, "BLURAY initialized! (%p)\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);

j45's avatar
j45 committed
172
    DEBUG(DBG_BLURAY, "BLURAY destroyed! (%p)\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
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)
j45's avatar
j45 committed
190
                    DEBUG(DBG_BLURAY | DBG_CRIT, "Read %d bytes at %"PRIu64" ; requested %d ! (%p)\n", read_len, bd->clip_block_pos, len, bd);
j45's avatar
j45 committed
191 192 193 194 195

                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)) {
j45's avatar
j45 committed
196
                            DEBUG(DBG_BLURAY, "Unable decrypt unit! (%p)\n", bd);
j45's avatar
j45 committed
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

                            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;

j45's avatar
j45 committed
263
        DEBUG(DBG_BLURAY, "Seek to %"PRIu64" (%p)\n",
Accident's avatar
 
Accident committed
264
              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
    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) {
j45's avatar
j45 committed
294
                        DEBUG(DBG_BLURAY, "End of title (%p)\n", bd);
j45's avatar
j45 committed
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
                        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
    // Open the playlist
    if (bd->title_list == NULL) {
j45's avatar
j45 committed
342
        DEBUG(DBG_BLURAY, "Title list not yet read! (%p)\n", bd);
j45's avatar
j45 committed
343 344 345
        return 0;
    }
    if (bd->title_list->count <= title_idx) {
j45's avatar
j45 committed
346
        DEBUG(DBG_BLURAY, "Invalid title index %d! (%p)\n", title_idx, bd);
j45's avatar
j45 committed
347 348 349 350 351
        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) {
j45's avatar
j45 committed
352
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (%p)\n", 
j45's avatar
j45 committed
353 354 355
              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
    // Get the initial clip of the playlist
    bd->clip = nav_next_clip(bd->title, NULL);
    if (_open_m2ts(bd)) {
j45's avatar
j45 committed
363
        DEBUG(DBG_BLURAY, "Title %s selected! (%p)\n", f_name, bd);
j45's avatar
j45 committed
364 365 366 367
        return 1;
    }
    return 0;
}
368

369 370 371 372 373 374 375 376 377 378
int bd_select_angle(BLURAY *bd, int angle)
{
    if (bd->title == NULL) {
        DEBUG(DBG_BLURAY, "Title not yet selected! (%p)\n", bd);
        return 0;
    }
    bd->clip = nav_set_angle(bd->title, bd->clip, angle);
    return 1;
}

j45's avatar
j45 committed
379 380 381 382
uint64_t bd_get_title_size(BLURAY *bd)
{
    return bd ? bd->s_size : UINT64_C(0);
}
cRTrn13's avatar
cRTrn13 committed
383

j45's avatar
j45 committed
384 385 386 387
uint64_t bd_tell(BLURAY *bd)
{
    return bd ? bd->s_pos : INT64_C(0);
}
388

j45's avatar
j45 committed
389 390 391 392 393 394 395 396 397 398 399 400 401 402 403
// 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;
    }
404

j45's avatar
j45 committed
405 406
    if (bd->title_list != NULL) {
        nav_free_title_list(bd->title_list);
cRTrn13's avatar
cRTrn13 committed
407
    }
j45's avatar
j45 committed
408
    bd->title_list = nav_get_title_list(bd->device_path, flags);
cRTrn13's avatar
cRTrn13 committed
409

j45's avatar
j45 committed
410 411 412 413
    if (!bd->title_list) {
        DEBUG(DBG_BLURAY | DBG_CRIT, "nav_get_title_list(%s) failed (%p)\n", bd->device_path, bd);
        return 0;
    }
414

j45's avatar
j45 committed
415
    return bd->title_list->count;
cRTrn13's avatar
cRTrn13 committed
416
}
417

j45's avatar
j45 committed
418
BD_TITLE_INFO* bd_get_title_info(BLURAY *bd, uint32_t title_idx)
419
{
j45's avatar
j45 committed
420 421 422 423 424
    NAV_TITLE *title;
    BD_TITLE_INFO *title_info;
    int ii;

    if (bd->title_list == NULL) {
j45's avatar
j45 committed
425
        DEBUG(DBG_BLURAY, "Title list not yet read! (%p)\n", bd);
j45's avatar
j45 committed
426 427 428
        return NULL;
    }
    if (bd->title_list->count <= title_idx) {
j45's avatar
j45 committed
429
        DEBUG(DBG_BLURAY, "Invalid title index %d! (%p)\n", title_idx, bd);
j45's avatar
j45 committed
430 431 432 433
        return NULL;
    }
    title = nav_title_open(bd->device_path, bd->title_list->title_info[title_idx].name);
    if (title == NULL) {
j45's avatar
j45 committed
434
        DEBUG(DBG_BLURAY | DBG_CRIT, "Unable to open title %s! (%p)\n", 
j45's avatar
j45 committed
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451
              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;
452
}
cRTrn13's avatar
cRTrn13 committed
453

j45's avatar
j45 committed
454
void bd_free_title_info(BD_TITLE_INFO *title_info)
cRTrn13's avatar
cRTrn13 committed
455
{
j45's avatar
j45 committed
456 457
    X_FREE(title_info->chapters);
    X_FREE(title_info);
cRTrn13's avatar
cRTrn13 committed
458
}
j45's avatar
j45 committed
459