aacs.c 24.5 KB
Newer Older
1 2
/*
 * This file is part of libaacs
3
 * Copyright (C) 2009-2010  Obliter0n
npzacs's avatar
npzacs committed
4
 * Copyright (C) 2009-2010  npzacs
5
 *
gates's avatar
gates committed
6 7 8 9
 * 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.
10
 *
gates's avatar
gates committed
11
 * This library is distributed in the hope that it will be useful,
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
gates's avatar
gates committed
13 14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
15
 *
gates's avatar
gates committed
16 17 18
 * 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/>.
19 20
 */

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

npzacs's avatar
npzacs committed
25 26
#include <util/attributes.h>

npzacs's avatar
npzacs committed
27
#include "aacs-version.h"
cRTrn13's avatar
cRTrn13 committed
28
#include "aacs.h"
29
#include "crypto.h"
cRTrn13's avatar
cRTrn13 committed
30
#include "mmc.h"
31
#include "mkb.h"
gates's avatar
gates committed
32
#include "file/file.h"
gates's avatar
gates committed
33
#include "file/keydbcfg.h"
34 35
#include "util/macro.h"
#include "util/logging.h"
36
#include "util/strutl.h"
cRTrn13's avatar
cRTrn13 committed
37

38
#include <inttypes.h>
39
#include <string.h>
40
#include <stdio.h>
npzacs's avatar
npzacs committed
41 42 43
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
44
#include <gcrypt.h>
45

npzacs's avatar
npzacs committed
46

npzacs's avatar
npzacs committed
47
struct aacs {
npzacs's avatar
npzacs committed
48 49 50 51 52
    /* current disc */
    char     *path;
    int       mkb_version;
    uint8_t   disc_id[20];

53 54
    /* VID is cached for BD-J */
    uint8_t   vid[16];
npzacs's avatar
npzacs committed
55 56 57 58

    /* unit key for each CPS unit */
    uint32_t  num_uks;
    uint8_t  *uks;
npzacs's avatar
npzacs committed
59

npzacs's avatar
npzacs committed
60 61 62 63 64 65
    /* CPS unit of currently selected title */
    uint16_t  current_cps_unit;

    /* title -> CPS unit mappings */
    uint32_t  num_titles;
    uint16_t *cps_units;  /* [0] = first play ; [1] = top menu ; [2] = title 1 ... */
npzacs's avatar
npzacs committed
66 67
};

68 69 70
static const uint8_t empty_key[] = "\x00\x00\x00\x00\x00\x00\x00\x00"
                                   "\x00\x00\x00\x00\x00\x00\x00\x00";

71 72
static int _validate_pk(const uint8_t *pk,
                        const uint8_t *cvalue, const uint8_t *uv, const uint8_t *vd,
npzacs's avatar
npzacs committed
73
                        uint8_t *mk)
cRTrn13's avatar
cRTrn13 committed
74
{
75
    gcry_cipher_hd_t gcry_h;
npzacs's avatar
npzacs committed
76
    int a;
cRTrn13's avatar
cRTrn13 committed
77
    uint8_t dec_vd[16];
78
    char str[40];
cRTrn13's avatar
cRTrn13 committed
79

80
    DEBUG(DBG_AACS, "Validate processing key %s...\n", print_hex(str, pk, 16));
cRTrn13's avatar
cRTrn13 committed
81
    DEBUG(DBG_AACS, " Using:\n");
82 83 84
    DEBUG(DBG_AACS, "   UV: %s\n", print_hex(str, uv, 4));
    DEBUG(DBG_AACS, "   cvalue: %s\n", print_hex(str, cvalue, 16));
    DEBUG(DBG_AACS, "   Verification data: %s\n", print_hex(str, vd, 16));
85

86 87 88
    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 0);
    gcry_cipher_setkey(gcry_h, pk, 16);
    gcry_cipher_decrypt(gcry_h, mk, 16, cvalue, 16);
89

cRTrn13's avatar
cRTrn13 committed
90 91 92 93
    for (a = 0; a < 4; a++) {
        mk[a + 12] ^= uv[a];
    }

94 95 96
    gcry_cipher_setkey(gcry_h, mk, 16);
    gcry_cipher_decrypt (gcry_h, dec_vd, 16, vd, 16);
    gcry_cipher_close(gcry_h);
97

cRTrn13's avatar
cRTrn13 committed
98
    if (!memcmp(dec_vd, "\x01\x23\x45\x67\x89\xAB\xCD\xEF", 8)) {
npzacs's avatar
npzacs committed
99
        DEBUG(DBG_AACS, "Processing key %s is valid!\n", print_hex(str, pk, 16));
npzacs's avatar
npzacs committed
100
        return AACS_SUCCESS;
cRTrn13's avatar
cRTrn13 committed
101 102
    }

npzacs's avatar
npzacs committed
103
    return AACS_ERROR_NO_PK;
cRTrn13's avatar
cRTrn13 committed
104
}
105

106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
static int _rl_verify_signature(const uint8_t *rl, size_t size)
{
    int    entries = MKINT_BE32(rl + 12 + 8);
    size_t len     = 12 + 12 + 8 * entries; /* type_and_version_rec=12, rl_header=12, rl=entries*8 */

    if (len + 40 > size) {
        DEBUG(DBG_AACS, "revocation list size mismatch\n");
        return 0;
    }

    return crypto_aacs_verify_aacsla(rl + len, rl, len);
}

static void _update_drl(MKB *mkb)
{
    uint32_t version = mkb_version(mkb);
    uint32_t cache_version;

    if (!cache_get("drl", &cache_version, NULL, NULL) || cache_version < version) {
        size_t drl_len;
        const uint8_t *drl_rec = mkb_drive_revokation_entries(mkb, &drl_len);
        const uint8_t *v_rec   = mkb_type_and_version_record(mkb);

        if (drl_rec && v_rec) {
            drl_rec -= 4;
            drl_len += 4;

            uint8_t *data = malloc(12 + drl_len);
            memcpy(data,      v_rec,   12);
            memcpy(data + 12, drl_rec, drl_len);
            if (!_rl_verify_signature(data, drl_len + 12)) {
                DEBUG(DBG_AACS | DBG_CRIT, "invalid drive revocation list signature, not using it\n");
            } else {
                cache_save("drl", version, data, drl_len + 12);
            }
            X_FREE(data);
        }
    }
}

146
static int _calc_mk(AACS *aacs, uint8_t *mk, pk_list *pkl)
cRTrn13's avatar
keyfile  
cRTrn13 committed
147 148 149
{
    int a, num_uvs = 0;
    size_t len;
150
    uint8_t *buf = NULL;
cRTrn13's avatar
keyfile  
cRTrn13 committed
151
    MKB *mkb = NULL;
152
    const uint8_t *rec, *uvs;
cRTrn13's avatar
keyfile  
cRTrn13 committed
153

154
    /* Skip if retrieved from config file */
155 156 157
    if (memcmp(mk, empty_key, 16)) {
        return AACS_SUCCESS;
    }
158

cRTrn13's avatar
cRTrn13 committed
159 160
    DEBUG(DBG_AACS, "Calculate media key...\n");

npzacs's avatar
npzacs committed
161
    if ((mkb = mkb_open(aacs->path))) {
162
        DEBUG(DBG_AACS, "Get UVS...\n");
163
        _update_drl(mkb);
npzacs's avatar
npzacs committed
164
        aacs->mkb_version = mkb_version(mkb);
165 166 167 168 169 170 171 172
        uvs = mkb_subdiff_records(mkb, &len);
        rec = uvs;
        while (rec < uvs + len) {
            if (rec[0] & 0xc0)
                break;
            rec += 5;
            num_uvs++;
        }
cRTrn13's avatar
keyfile  
cRTrn13 committed
173

174 175
        DEBUG(DBG_AACS, "Get cvalues...\n");
        rec = mkb_cvalues(mkb, &len);
176 177

        for (; pkl && pkl->key; pkl = pkl->next) {
178
                uint8_t pk[16];
179
                hexstring_to_hex_array(pk, sizeof(pk), pkl->key);
180 181 182
                DEBUG(DBG_AACS, "Trying processing key...\n");

                for (a = 0; a < num_uvs; a++) {
183
                    if (AACS_SUCCESS == _validate_pk(pk, rec + a * 16, uvs + 1 + a * 5,
184
                      mkb_mk_dv(mkb), mk)) {
185 186
                        mkb_close(mkb);
                        X_FREE(buf);
187

188
                        char str[40];
189
                        DEBUG(DBG_AACS, "Media key: %s\n", print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
190
                        return AACS_SUCCESS;
191 192
                    }
                }
cRTrn13's avatar
keyfile  
cRTrn13 committed
193 194
            }

195 196
        mkb_close(mkb);
        X_FREE(buf);
cRTrn13's avatar
keyfile  
cRTrn13 committed
197

npzacs's avatar
npzacs committed
198
        DEBUG(DBG_AACS | DBG_CRIT, "Error calculating media key. Missing right processing key ?\n");
npzacs's avatar
npzacs committed
199
        return AACS_ERROR_NO_PK;
npzacs's avatar
npzacs committed
200
    }
npzacs's avatar
npzacs committed
201

npzacs's avatar
npzacs committed
202
    DEBUG(DBG_AACS | DBG_CRIT, "Error opening %s/AACS/MKB_RO.inf\n", aacs->path);
npzacs's avatar
npzacs committed
203
    return AACS_ERROR_CORRUPTED_DISC;
cRTrn13's avatar
keyfile  
cRTrn13 committed
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 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262
static MKB *_get_hrl_mkb(MMC *mmc)
{
    MKB     *mkb = NULL;
    uint8_t *data;
    int      size;

    data = mmc_read_mkb(mmc, 0, &size);

    /* check acquired hrl signature */
    if (data && size > 0) {
        if (_rl_verify_signature(data, size)) {
            mkb = mkb_init(data, size);
            DEBUG(DBG_AACS, "Partial hrl mkb read. Version: %d\n", mkb_version(mkb));
        } else {
            DEBUG(DBG_AACS | DBG_CRIT, "invalid host revocation list signature, not using it\n");
            X_FREE(data);
        }
    }

    if (mkb) {
        /* use latest version, keep cache up-to-date */
        uint32_t size;
        size = mkb_data_size(mkb);
        data = cache_get_or_update("hrl", mkb_data(mkb), &size, mkb_version(mkb));

        if (!_rl_verify_signature(data, size)) {
            DEBUG(DBG_AACS | DBG_CRIT, "invalid cached revocation list signature, replacing it\n");
            cache_save("hrl", mkb_version(mkb), mkb_data(mkb), mkb_data_size(mkb));
            X_FREE(data);
        } else {
            /* use cached version */
            mkb_close(mkb);
            mkb = mkb_init(data, size);
        }

    } else {
        /* use cached version */
        uint32_t size;
        data = cache_get_or_update("hrl", NULL, &size, 0);
        if (data && size > 0) {
            if (!_rl_verify_signature(data, size)) {
                mkb = mkb_init(data, size);
            } else {
                DEBUG(DBG_AACS | DBG_CRIT, "invalid cached revocation list signature, deleting cache\n");
                cache_remove("hrl");
            }
        }
    }


    if (mkb) {
        DEBUG(DBG_AACS, "Using hrl version %d\n", mkb_version(mkb));
    }

    return mkb;
}

263
static int _read_vid(AACS *aacs, cert_list *hcl)
cRTrn13's avatar
openssl  
cRTrn13 committed
264
{
265 266
    /* Use VID given in config file if available */
    if (memcmp(aacs->vid, empty_key, 16)) {
npzacs's avatar
npzacs committed
267
        return AACS_SUCCESS;
268 269
    }

cRTrn13's avatar
cRTrn13 committed
270
    MMC* mmc = NULL;
npzacs's avatar
npzacs committed
271
    if (!(mmc = mmc_open(aacs->path))) {
npzacs's avatar
npzacs committed
272
        return AACS_ERROR_MMC_OPEN;
273 274
    }

npzacs's avatar
npzacs committed
275 276
    int error_code = AACS_ERROR_NO_CERT;

277 278
    MKB *hrl_mkb = _get_hrl_mkb(mmc);

279
    for (;hcl && hcl->host_priv_key && hcl->host_cert; hcl = hcl->next) {
280 281 282

        char tmp_str[2*92+1];
        uint8_t priv_key[20], cert[92];
283 284
        hexstring_to_hex_array(priv_key, sizeof(priv_key), hcl->host_priv_key);
        hexstring_to_hex_array(cert,     sizeof(cert),     hcl->host_cert);
285 286

        if (!crypto_aacs_verify_host_cert(cert)) {
npzacs's avatar
npzacs committed
287 288 289 290
            DEBUG(DBG_AACS, "Not using invalid host certificate %s.\n",
                  print_hex(tmp_str, cert, 92));
            continue;
        }
291

292 293 294 295 296 297 298
        if (mkb_host_cert_is_revoked(hrl_mkb, cert + 4)) {
            DEBUG(DBG_AACS | DBG_CRIT, "Host certificate %s has been revoked.\n",
                  print_hex(tmp_str, cert + 4, 6));
            error_code = AACS_ERROR_CERT_REVOKED;
            //continue;
        }

299 300 301
        DEBUG(DBG_AACS, "Trying host certificate (id 0x%s)...\n",
              print_hex(tmp_str, cert + 4, 6));

npzacs's avatar
npzacs committed
302
        int mmc_result = mmc_read_vid(mmc, priv_key, cert, aacs->vid);
npzacs's avatar
npzacs committed
303 304
        switch (mmc_result) {
            case MMC_SUCCESS:
305
                mkb_close(hrl_mkb);
npzacs's avatar
npzacs committed
306 307 308 309 310 311 312 313 314
                mmc_close(mmc);
                return AACS_SUCCESS;
            case MMC_ERROR_CERT_REVOKED:
                error_code = AACS_ERROR_CERT_REVOKED;
                break;
            case MMC_ERROR:
            default:
                error_code = AACS_ERROR_MMC_FAILURE;
                break;
315 316 317
        }
    }

318
    mkb_close(hrl_mkb);
319 320 321
    mmc_close(mmc);

    DEBUG(DBG_AACS, "Error reading VID!\n");
npzacs's avatar
npzacs committed
322
    return error_code;
323
}
cRTrn13's avatar
openssl  
cRTrn13 committed
324

325 326
static int _calc_vuk(AACS *aacs, uint8_t *mk, uint8_t *vuk,
                     pk_list *pk_list, cert_list *host_cert_list)
327
{
npzacs's avatar
npzacs committed
328 329
    int error_code;

330
    /* Skip if retrieved from config file */
331
    if (memcmp(vuk, empty_key, 16)) {
npzacs's avatar
npzacs committed
332 333 334
        DEBUG(DBG_AACS, "Using VUK from config file\n");
        return AACS_SUCCESS;
    }
335

npzacs's avatar
npzacs committed
336
    /* get cached vuk */
337
    if (keycache_find("vuk", aacs->disc_id, vuk, 16)) {
npzacs's avatar
npzacs committed
338
        DEBUG(DBG_AACS, "Using cached VUK\n");
npzacs's avatar
npzacs committed
339
        return AACS_SUCCESS;
npzacs's avatar
npzacs committed
340 341
    }

342
    /* make sure we have media key */
343
    error_code = _calc_mk(aacs, mk, pk_list);
npzacs's avatar
npzacs committed
344 345
    if (error_code != AACS_SUCCESS) {
        return error_code;
346 347
    }

npzacs's avatar
npzacs committed
348
    /* acquire VID */
349
    error_code = _read_vid(aacs, host_cert_list);
npzacs's avatar
npzacs committed
350 351 352
    if (error_code != AACS_SUCCESS) {
        return error_code;
    }
353

npzacs's avatar
npzacs committed
354 355
    int a;
    gcry_cipher_hd_t gcry_h;
356

npzacs's avatar
npzacs committed
357
    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 0);
358 359
    gcry_cipher_setkey(gcry_h, mk, 16);
    gcry_cipher_decrypt(gcry_h, vuk, 16, aacs->vid, 16);
npzacs's avatar
npzacs committed
360
    gcry_cipher_close(gcry_h);
361

npzacs's avatar
npzacs committed
362
    for (a = 0; a < 16; a++) {
363
        vuk[a] ^= aacs->vid[a];
npzacs's avatar
npzacs committed
364
    }
365

npzacs's avatar
npzacs committed
366
    char str[40];
367
    DEBUG(DBG_AACS, "Volume unique key: %s\n", print_hex(str, vuk, 16));
368

npzacs's avatar
npzacs committed
369
    /* cache vuk */
370
    keycache_save("vuk", aacs->disc_id, vuk, 16);
npzacs's avatar
npzacs committed
371

npzacs's avatar
npzacs committed
372
    return AACS_SUCCESS;
cRTrn13's avatar
openssl  
cRTrn13 committed
373 374
}

375 376
static uint16_t _read_u16(AACS_FILE_H *fp)
{
npzacs's avatar
npzacs committed
377
  uint8_t data[2] = {0, 0};
378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 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

  file_read(fp, data, sizeof(uint16_t));

  return MKINT_BE16(data);
}

static void _read_uks_map(AACS *aacs, AACS_FILE_H *fp)
{
    uint16_t first_play, top_menu;
    unsigned i;

    DEBUG(DBG_AACS, "Assigning CPS units to titles ...\n");

    X_FREE(aacs->cps_units);
    aacs->current_cps_unit = 0;

    file_seek(fp, 16 + 4, SEEK_SET);

    first_play = _read_u16(fp);
    top_menu   = _read_u16(fp);

    DEBUG(DBG_AACS, "Title FP : CPS unit %d\n", first_play);
    DEBUG(DBG_AACS, "Title TM : CPS unit %d\n", top_menu);

    aacs->num_titles   = _read_u16(fp);
    aacs->cps_units    = calloc(sizeof(uint16_t), aacs->num_titles + 2);
    aacs->cps_units[0] = first_play;
    aacs->cps_units[1] = top_menu;

    for (i = 2; i < aacs->num_titles + 2; i++) {
        _read_u16(fp); /* reserved */
        aacs->cps_units[i] = _read_u16(fp);
        DEBUG(DBG_AACS, "Title %02d : CPS unit %d\n", i - 1, aacs->cps_units[i]);
    }

    /* validate */
    for (i = 0; i < aacs->num_titles + 2; i++) {
        if (aacs->cps_units[i])
            aacs->cps_units[i]--; /* number [1...N] --> index [0...N-1] */
        if (aacs->cps_units[i] >= aacs->num_uks) {
            DEBUG(DBG_AACS, " *** Invalid CPS unit for title %d: %d !\n", (int) i - 1, aacs->cps_units[i]);
            aacs->cps_units[i] = 0;
        }
    }
}

npzacs's avatar
npzacs committed
424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439
static AACS_FILE_H *_open_unit_key_file(const char *path)
{
    AACS_FILE_H *fp;
    char        *f_name;

    f_name = str_printf("%s/AACS/Unit_Key_RO.inf", path);
    fp = file_open(f_name, "rb");

    if (!fp) {
        DEBUG(DBG_AACS | DBG_CRIT, "Error opening unit key file %s\n", f_name);
    }

    X_FREE(f_name);
    return fp;
}

npzacs's avatar
npzacs committed
440
/* Function that collects keys from keydb config entry */
441 442
static void _find_config_entry(AACS *aacs, title_entry_list *ce,
                               uint8_t *mk, uint8_t *vuk)
npzacs's avatar
npzacs committed
443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466
{
    uint8_t discid[20];
    char str[48];

    aacs->uks = NULL;
    aacs->num_uks = 0;

        while (ce && ce->entry.discid) {
            memset(discid, 0, sizeof(discid));
            hexstring_to_hex_array(discid, sizeof(discid),
                                   ce->entry.discid);
            if (!memcmp(aacs->disc_id, discid, 20)) {
                DEBUG(DBG_AACS, "Found config entry for discid %s\n",
                      ce->entry.discid);
                break;
            }

            ce = ce->next;
        }
        if (!ce) {
            return;
        }

        if (ce->entry.mek) {
467
            hexstring_to_hex_array(mk, 16, ce->entry.mek);
npzacs's avatar
npzacs committed
468 469

            DEBUG(DBG_AACS, "Found media key for %s: %s\n",
470
                  ce->entry.discid, print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
471 472 473 474 475 476 477 478 479 480 481
        }

        if (ce->entry.vid) {
            hexstring_to_hex_array(aacs->vid, sizeof(aacs->vid),
                                    ce->entry.vid);

            DEBUG(DBG_AACS, "Found volume id for %s: %s\n",
                  ce->entry.discid, print_hex(str, aacs->vid, 16));
        }

        if (ce->entry.vuk) {
482
            hexstring_to_hex_array(vuk, 16, ce->entry.vuk);
npzacs's avatar
npzacs committed
483 484

            DEBUG(DBG_AACS, "Found volume unique key for %s: %s\n",
485
                  ce->entry.discid, print_hex(str, vuk, 16));
npzacs's avatar
npzacs committed
486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508
        }

        if (ce->entry.uk) {
            DEBUG(DBG_AACS, "Acquire CPS unit keys from keydb config file...\n");

            digit_key_pair_list *ukcursor = ce->entry.uk;
            while (ukcursor && ukcursor->key_pair.key) {
                aacs->num_uks++;

                aacs->uks = (uint8_t*)realloc(aacs->uks, 16 * aacs->num_uks);
                hexstring_to_hex_array(aacs->uks + (16 * (aacs->num_uks - 1)), 16,
                                      ukcursor->key_pair.key);

                char str[40];
                DEBUG(DBG_AACS, "Unit key %d from keydb entry: %s\n",
                      aacs->num_uks,
                      print_hex(str, aacs->uks + (16 * (aacs->num_uks - 1)), 16));

                ukcursor = ukcursor->next;
            }
        }
}

509
static int _calc_uks(AACS *aacs, const char *configfile_path)
cRTrn13's avatar
openssl  
cRTrn13 committed
510
{
511
    AACS_FILE_H *fp = NULL;
512
    uint8_t  buf[16];
513
    uint64_t f_pos;
514
    unsigned int i;
npzacs's avatar
npzacs committed
515
    int error_code;
cRTrn13's avatar
openssl  
cRTrn13 committed
516

517 518 519 520 521 522 523 524 525 526 527
    config_file *cf = NULL;
    uint8_t mk[16] = {0}, vuk[16] = {0};

    cf = keydbcfg_config_load(configfile_path);
    if (!cf) {
        return AACS_ERROR_NO_CONFIG;
    }

    DEBUG(DBG_AACS, "Searching for keydb config entry...\n");
    _find_config_entry(aacs, cf->list, mk, vuk);

528
    /* Skip if retrieved from config file */
529 530
    if (aacs->uks) {
        keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
531
        return AACS_SUCCESS;
532
    }
533

534
    /* Make sure we have VUK */
535 536
    error_code = _calc_vuk(aacs, mk, vuk, cf->pkl, cf->host_cert_list);
    keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
537 538
    if (error_code != AACS_SUCCESS) {
        return error_code;
539
    }
540

541 542
    DEBUG(DBG_AACS, "Calculate CPS unit keys...\n");

npzacs's avatar
npzacs committed
543
    fp = _open_unit_key_file(aacs->path);
npzacs's avatar
npzacs committed
544 545 546
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
    }
cRTrn13's avatar
openssl  
cRTrn13 committed
547

npzacs's avatar
npzacs committed
548 549
    if ((file_read(fp, buf, 4)) == 4) {
        f_pos = MKINT_BE32(buf);
npzacs's avatar
npzacs committed
550

npzacs's avatar
npzacs committed
551 552 553 554
        // Read number of keys
        file_seek(fp, f_pos, SEEK_SET);
        if ((file_read(fp, buf, 2)) == 2) {
            aacs->num_uks = MKINT_BE16(buf);
npzacs's avatar
npzacs committed
555

npzacs's avatar
npzacs committed
556 557
            X_FREE(aacs->uks);
            aacs->uks = calloc(aacs->num_uks, 16);
npzacs's avatar
npzacs committed
558

npzacs's avatar
npzacs committed
559
            DEBUG(DBG_AACS, "%d CPS unit keys\n", aacs->num_uks);
npzacs's avatar
npzacs committed
560

npzacs's avatar
npzacs committed
561 562 563 564 565
        } else {
            aacs->num_uks = 0;
            DEBUG(DBG_AACS | DBG_CRIT, "Error reading number of unit keys\n");
            error_code = AACS_ERROR_CORRUPTED_DISC;
        }
cRTrn13's avatar
openssl  
cRTrn13 committed
566

npzacs's avatar
npzacs committed
567 568 569
        // Read keys
        for (i = 0; i < aacs->num_uks; i++) {
            f_pos += 48;
cRTrn13's avatar
openssl  
cRTrn13 committed
570

npzacs's avatar
npzacs committed
571 572 573 574 575 576
            file_seek(fp, f_pos, SEEK_SET);
            if ((file_read(fp, buf, 16)) != 16) {
                DEBUG(DBG_AACS, "Unit key %d: read error\n", i);
                aacs->num_uks = i;
                error_code = AACS_ERROR_CORRUPTED_DISC;
                break;
npzacs's avatar
npzacs committed
577
            }
cRTrn13's avatar
openssl  
cRTrn13 committed
578

npzacs's avatar
npzacs committed
579 580 581
            gcry_cipher_hd_t gcry_h;
            gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES,
                             GCRY_CIPHER_MODE_ECB, 0);
582
            gcry_cipher_setkey(gcry_h, vuk, 16);
npzacs's avatar
npzacs committed
583 584
            gcry_cipher_decrypt(gcry_h, aacs->uks + 16*i, 16, buf, 16);
            gcry_cipher_close(gcry_h);
cRTrn13's avatar
openssl  
cRTrn13 committed
585

npzacs's avatar
npzacs committed
586 587 588
            char str[40];
            DEBUG(DBG_AACS, "Unit key %d: %s\n", i,
                  print_hex(str, aacs->uks + 16*i, 16));
589 590
        }

npzacs's avatar
npzacs committed
591 592 593
        /* failing next is not fatal, it just slows down things */
        _read_uks_map(aacs, fp);

594
        file_close(fp);
cRTrn13's avatar
openssl  
cRTrn13 committed
595

npzacs's avatar
npzacs committed
596 597 598 599 600 601 602
        return error_code;
    }

    file_close(fp);

    DEBUG(DBG_AACS | DBG_CRIT, "Error reading unit keys\n");
    return AACS_ERROR_CORRUPTED_DISC;
cRTrn13's avatar
openssl  
cRTrn13 committed
603 604
}

605 606
static int _calc_title_hash(const char *path, uint8_t *title_hash)
{
607
    AACS_FILE_H *fp = NULL;
608 609 610
    uint8_t *ukf_buf;
    char     str[48];
    int64_t  f_size;
npzacs's avatar
npzacs committed
611
    int      result = AACS_SUCCESS;
612

npzacs's avatar
npzacs committed
613 614 615
    fp = _open_unit_key_file(path);
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
616 617 618 619 620 621 622 623
    }

    file_seek(fp, 0, SEEK_END);
    f_size = file_tell(fp);
    file_seek(fp, 0, SEEK_SET);

    ukf_buf = malloc(f_size);

npzacs's avatar
npzacs committed
624 625 626
    if ((file_read(fp, ukf_buf, f_size)) == f_size) {
        crypto_aacs_title_hash(ukf_buf, f_size, title_hash);
        DEBUG(DBG_AACS, "Disc ID: %s\n", print_hex(str, title_hash, 20));
cRTrn13's avatar
cRTrn13 committed
627

npzacs's avatar
npzacs committed
628 629 630
    } else {
        result = AACS_ERROR_CORRUPTED_DISC;
        DEBUG(DBG_AACS | DBG_CRIT, "Failed to read %"PRIu64" bytes from unit key file %s/AACS/Unit_Key_RO.inf", f_size, path);
631 632 633 634 635
    }

    file_close(fp);
    X_FREE(ukf_buf);

npzacs's avatar
npzacs committed
636
    return result;
637
}
cRTrn13's avatar
cRTrn13 committed
638

npzacs's avatar
npzacs committed
639
static int _verify_ts(uint8_t *buf, size_t size)
640
{
641 642
    uint8_t *ptr;

cRTrn13's avatar
cRTrn13 committed
643 644 645 646 647 648 649
    if (size < 192) {
        return 1;
    }

    for (ptr=buf; ptr < buf+192; ptr++) {
        int failed = 0;
        if (*ptr == 0x47) {
650 651 652 653
            uint8_t *ptr2;

            for (ptr2=ptr; ptr2 < buf + size; ptr2 += 192) {
                if (*ptr2 != 0x47) {
cRTrn13's avatar
cRTrn13 committed
654 655
                    failed = 1;
                    break;
656 657
                }
            }
cRTrn13's avatar
cRTrn13 committed
658 659 660
            if (!failed) {
                return 1;
            }
661 662 663 664
        }
        ptr++;
    }

cRTrn13's avatar
cRTrn13 committed
665
    DEBUG(DBG_AACS, "Failed to verify TS!\n");
666

cRTrn13's avatar
cRTrn13 committed
667
    return 0;
668 669
}

670
#define ALIGNED_UNIT_LEN 6144
671
static int _decrypt_unit(AACS *aacs, uint8_t *out_buf, const uint8_t *in_buf, uint32_t curr_uk)
cRTrn13's avatar
cRTrn13 committed
672
{
673 674 675 676 677 678 679
    gcry_cipher_hd_t gcry_h;
    int a;
    uint8_t key[16], iv[] = "\x0b\xa0\xf8\xdd\xfe\xa6\x1f\xb3"
                            "\xd8\xdf\x9f\x56\x6a\x05\x0f\x78";

    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 0);
    gcry_cipher_setkey(gcry_h, aacs->uks + curr_uk * 16, 16);
680
    gcry_cipher_encrypt(gcry_h, key, 16, in_buf, 16);
681
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
682

cRTrn13's avatar
cRTrn13 committed
683
    for (a = 0; a < 16; a++) {
684
        key[a] ^= in_buf[a];
cRTrn13's avatar
cRTrn13 committed
685 686
    }

687 688
    memcpy(out_buf, in_buf, 16); /* first 16 bytes are plain */

689 690 691
    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_CBC, 0);
    gcry_cipher_setkey(gcry_h, key, 16);
    gcry_cipher_setiv(gcry_h, iv, 16);
692
    gcry_cipher_decrypt(gcry_h, out_buf + 16, ALIGNED_UNIT_LEN - 16, in_buf + 16, ALIGNED_UNIT_LEN - 16);
693
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
694

695
    if (_verify_ts(out_buf, ALIGNED_UNIT_LEN)) {
cRTrn13's avatar
cRTrn13 committed
696 697 698
        return 1;
    }

699
    if (curr_uk < aacs->num_uks - 1) {
700
        return _decrypt_unit(aacs, out_buf, in_buf, curr_uk++);
npzacs's avatar
npzacs committed
701 702
    }

cRTrn13's avatar
cRTrn13 committed
703 704 705
    return 0;
}

npzacs's avatar
npzacs committed
706 707
void aacs_get_version(int *major, int *minor, int *micro)
{
npzacs's avatar
npzacs committed
708 709 710
    *major = AACS_VERSION_MAJOR;
    *minor = AACS_VERSION_MINOR;
    *micro = AACS_VERSION_MICRO;
npzacs's avatar
npzacs committed
711 712
}

npzacs's avatar
npzacs committed
713
/* aacs_open2() wrapper for backwards compability */
cRTrn13's avatar
cRTrn13 committed
714
AACS *aacs_open(const char *path, const char *configfile_path)
npzacs's avatar
npzacs committed
715 716 717 718 719 720 721 722 723 724 725 726 727 728
{
    int error_code;
    AACS *aacs;

    aacs = aacs_open2(path, configfile_path, &error_code);
    if (error_code == AACS_SUCCESS) {
        return aacs;
    }

    aacs_close(aacs);
    return NULL;
}

AACS *aacs_open2(const char *path, const char *configfile_path, int *error_code)
cRTrn13's avatar
cRTrn13 committed
729
{
npzacs's avatar
npzacs committed
730
    DEBUG(DBG_AACS, "libaacs "AACS_VERSION_STRING" [%zd]\n", sizeof(AACS));
cRTrn13's avatar
cRTrn13 committed
731

732
    DEBUG(DBG_AACS, "Initializing libgcrypt...\n");
npzacs's avatar
npzacs committed
733
    if (!crypto_init()) {
npzacs's avatar
npzacs committed
734
        DEBUG(DBG_AACS | DBG_CRIT, "Failed to initialize libgcrypt\n");
735 736
        return NULL;
    }
737

cRTrn13's avatar
cRTrn13 committed
738
    AACS *aacs = calloc(1, sizeof(AACS));
739

npzacs's avatar
npzacs committed
740 741
    aacs->path = str_printf("%s", path);

npzacs's avatar
npzacs committed
742 743 744 745 746
    *error_code = _calc_title_hash(path, aacs->disc_id);
    if (*error_code != AACS_SUCCESS) {
        aacs_close(aacs);
        return NULL;
    }
npzacs's avatar
npzacs committed
747

npzacs's avatar
npzacs committed
748
    DEBUG(DBG_AACS, "Starting AACS waterfall...\n");
749
    *error_code = _calc_uks(aacs, configfile_path);
cRTrn13's avatar
cRTrn13 committed
750

npzacs's avatar
npzacs committed
751 752 753 754 755
    if (*error_code == AACS_SUCCESS) {
        DEBUG(DBG_AACS, "AACS initialized! (%p)\n", aacs);
    } else {
        DEBUG(DBG_AACS, "Failed to initialize AACS! (%p)\n", aacs);
    }
756

npzacs's avatar
npzacs committed
757
    return aacs;
cRTrn13's avatar
cRTrn13 committed
758 759
}

cRTrn13's avatar
cRTrn13 committed
760
void aacs_close(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
761
{
npzacs's avatar
npzacs committed
762 763 764
    if (!aacs)
        return;

765
    X_FREE(aacs->uks);
766
    X_FREE(aacs->cps_units);
npzacs's avatar
npzacs committed
767
    X_FREE(aacs->path);
768

npzacs's avatar
npzacs committed
769
    DEBUG(DBG_AACS, "AACS destroyed! (%p)\n", aacs);
cRTrn13's avatar
cRTrn13 committed
770

cRTrn13's avatar
cRTrn13 committed
771 772 773
    X_FREE(aacs);
}

774
int aacs_decrypt_unit(AACS *aacs, uint8_t *buf)
775
{
776 777
    uint8_t out_buf[ALIGNED_UNIT_LEN];

778 779 780 781 782
    if (!(buf[0] & 0xc0)) {
        // TP_extra_header Copy_permission_indicator == 0, unit is not encrypted
        return 1;
    }

783
    if (_decrypt_unit(aacs, out_buf, buf, aacs->current_cps_unit)) {
784
        memcpy(buf, out_buf, ALIGNED_UNIT_LEN);
785 786 787 788 789 790 791

        // Clear copy_permission_indicator bits
        int i;
        for (i = 0; i < 6144; i += 192) {
            buf[i] &= ~0xc0;
        }

792 793 794
        return 1;
    }

795
    DEBUG(DBG_AACS, "Failed decrypting unit [6144 bytes] (%p)\n", aacs);
796

797
    return 0;
798
}
cRTrn13's avatar
cRTrn13 committed
799

npzacs's avatar
npzacs committed
800 801 802 803 804 805 806 807 808 809 810 811
int aacs_get_mkb_version(AACS *aacs)
{
    if (!aacs->mkb_version) {
        MKB *mkb;
        if ((mkb = mkb_open(aacs->path))) {
            aacs->mkb_version = mkb_version(mkb);
            mkb_close(mkb);
        }
    }
    return aacs->mkb_version;
}

npzacs's avatar
npzacs committed
812 813 814 815 816
const uint8_t *aacs_get_disc_id(AACS *aacs)
{
    return aacs->disc_id;
}

817
const uint8_t *aacs_get_vid(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
818
{
819 820 821 822 823 824 825 826
    if (!memcmp(aacs->vid, empty_key, 16)) {
        config_file *cf = keydbcfg_config_load(NULL);
        if (cf) {
            _read_vid(aacs, cf->host_cert_list);

            keydbcfg_config_file_close(cf);
        }
    }
cRTrn13's avatar
cRTrn13 committed
827 828
    return aacs->vid;
}
829

npzacs's avatar
npzacs committed
830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875
static AACS_RL_ENTRY *_get_rl(const char *type, int *num_records, int *mkb_version)
{
    uint32_t len, version;
    void *data = NULL;

    *num_records = *mkb_version = 0;

    cache_get(type, &version, &len, NULL);

    if (version > 0 && len > 24) {
        data = malloc(len);
        if (cache_get(type, &version, &len, data) && len > 24) {

            if (_rl_verify_signature(data, len)) {
                *mkb_version = version;
                *num_records = MKINT_BE32((uint8_t*)data + 20);
                memmove(data, (uint8_t*)data + 24, len - 24);

                int ii;
                AACS_RL_ENTRY *rl = data;
                for (ii = 0; ii < *num_records; ii++) {
                    rl[ii].range = MKINT_BE16((uint8_t*)&rl[ii].range);
                }

                return rl;
            }

            DEBUG(DBG_AACS | DBG_CRIT, "invalid signature in cached %s\n", type);
        }

        X_FREE(data);
    }

    return data;
}

AACS_RL_ENTRY *aacs_get_hrl(int *num_records, int *mkb_version)
{
    return _get_rl("hrl", num_records, mkb_version);
}

AACS_RL_ENTRY *aacs_get_drl(int *num_records, int *mkb_version)
{
    return _get_rl("drl", num_records, mkb_version);
}

876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901
void aacs_select_title(AACS *aacs, uint32_t title)
{
    if (!aacs) {
        return;
    }

    if (!aacs->cps_units) {
        DEBUG(DBG_AACS|DBG_CRIT, "aacs_select_title(): CPS units not read ! (%p)\n", aacs);
        return;
    }

    if (title == 0xffff) {
        /* first play */
        aacs->current_cps_unit = aacs->cps_units[0];
        DEBUG(DBG_AACS, "aacs_set_title(first_play): CPS unit %d (%p)\n", aacs->current_cps_unit, aacs);
        return;
    }

    if (title <= aacs->num_titles) {
        aacs->current_cps_unit = aacs->cps_units[title + 1];
        DEBUG(DBG_AACS, "aacs_set_title(%d): CPS unit %d (%p)\n", title, aacs->current_cps_unit, aacs);
        return;
    }

    DEBUG(DBG_AACS|DBG_CRIT, "aacs_set_title(%d): invalid title ! (%p)\n", title, aacs);
}