aacs.c 27.2 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];
55 56
    /* PMSN is cached for BD-J */
    uint8_t   pmsn[16];
npzacs's avatar
npzacs committed
57 58 59 60

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

npzacs's avatar
npzacs committed
62 63 64 65 66 67
    /* 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 ... */
68 69 70 71

    /* bus encryption */
    int       bee;        /* bus encryption enabled flag in content certificate */
    int       bec;        /* bus encryption capable flag in drive certificate */
npzacs's avatar
npzacs committed
72 73
};

74 75 76
static const uint8_t empty_key[] = "\x00\x00\x00\x00\x00\x00\x00\x00"
                                   "\x00\x00\x00\x00\x00\x00\x00\x00";

77 78
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
79
                        uint8_t *mk)
cRTrn13's avatar
cRTrn13 committed
80
{
81
    gcry_cipher_hd_t gcry_h;
npzacs's avatar
npzacs committed
82
    int a;
cRTrn13's avatar
cRTrn13 committed
83
    uint8_t dec_vd[16];
84
    char str[40];
cRTrn13's avatar
cRTrn13 committed
85

86
    DEBUG(DBG_AACS, "Validate processing key %s...\n", print_hex(str, pk, 16));
cRTrn13's avatar
cRTrn13 committed
87
    DEBUG(DBG_AACS, " Using:\n");
88 89 90
    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));
91

92 93 94
    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);
95

cRTrn13's avatar
cRTrn13 committed
96 97 98 99
    for (a = 0; a < 4; a++) {
        mk[a + 12] ^= uv[a];
    }

100 101 102
    gcry_cipher_setkey(gcry_h, mk, 16);
    gcry_cipher_decrypt (gcry_h, dec_vd, 16, vd, 16);
    gcry_cipher_close(gcry_h);
103

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

npzacs's avatar
npzacs committed
109
    return AACS_ERROR_NO_PK;
cRTrn13's avatar
cRTrn13 committed
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 146 147 148 149 150 151
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);
        }
    }
}

152
static int _calc_mk(AACS *aacs, uint8_t *mk, pk_list *pkl)
cRTrn13's avatar
keyfile  
cRTrn13 committed
153 154 155
{
    int a, num_uvs = 0;
    size_t len;
156
    uint8_t *buf = NULL;
cRTrn13's avatar
keyfile  
cRTrn13 committed
157
    MKB *mkb = NULL;
158
    const uint8_t *rec, *uvs;
cRTrn13's avatar
keyfile  
cRTrn13 committed
159

160
    /* Skip if retrieved from config file */
161 162 163
    if (memcmp(mk, empty_key, 16)) {
        return AACS_SUCCESS;
    }
164

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

npzacs's avatar
npzacs committed
167
    if ((mkb = mkb_open(aacs->path))) {
168
        DEBUG(DBG_AACS, "Get UVS...\n");
169
        _update_drl(mkb);
npzacs's avatar
npzacs committed
170
        aacs->mkb_version = mkb_version(mkb);
171 172 173 174 175 176 177 178
        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
179

180 181
        DEBUG(DBG_AACS, "Get cvalues...\n");
        rec = mkb_cvalues(mkb, &len);
182 183

        for (; pkl && pkl->key; pkl = pkl->next) {
184
                uint8_t pk[16];
185
                hexstring_to_hex_array(pk, sizeof(pk), pkl->key);
186 187 188
                DEBUG(DBG_AACS, "Trying processing key...\n");

                for (a = 0; a < num_uvs; a++) {
189
                    if (AACS_SUCCESS == _validate_pk(pk, rec + a * 16, uvs + 1 + a * 5,
190
                      mkb_mk_dv(mkb), mk)) {
191 192
                        mkb_close(mkb);
                        X_FREE(buf);
193

194
                        char str[40];
195
                        DEBUG(DBG_AACS, "Media key: %s\n", print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
196
                        return AACS_SUCCESS;
197 198
                    }
                }
cRTrn13's avatar
keyfile  
cRTrn13 committed
199 200
            }

201 202
        mkb_close(mkb);
        X_FREE(buf);
cRTrn13's avatar
keyfile  
cRTrn13 committed
203

npzacs's avatar
npzacs committed
204
        DEBUG(DBG_AACS | DBG_CRIT, "Error calculating media key. Missing right processing key ?\n");
npzacs's avatar
npzacs committed
205
        return AACS_ERROR_NO_PK;
npzacs's avatar
npzacs committed
206
    }
npzacs's avatar
npzacs committed
207

npzacs's avatar
npzacs committed
208
    DEBUG(DBG_AACS | DBG_CRIT, "Error opening %s/AACS/MKB_RO.inf\n", aacs->path);
npzacs's avatar
npzacs committed
209
    return AACS_ERROR_CORRUPTED_DISC;
cRTrn13's avatar
keyfile  
cRTrn13 committed
210
}
211

212 213 214 215
static MKB *_get_hrl_mkb(MMC *mmc)
{
    MKB     *mkb = NULL;
    uint8_t *data;
npzacs's avatar
npzacs committed
216
    int      mkb_size;
217

npzacs's avatar
npzacs committed
218
    data = mmc_read_mkb(mmc, 0, &mkb_size);
219 220

    /* check acquired hrl signature */
npzacs's avatar
npzacs committed
221 222 223
    if (data && mkb_size > 0) {
        if (_rl_verify_signature(data, mkb_size)) {
            mkb = mkb_init(data, mkb_size);
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 263 264 265 266 267 268
            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;
}

269
static int _read_vid(AACS *aacs, cert_list *hcl, int force_mmc)
cRTrn13's avatar
openssl  
cRTrn13 committed
270
{
271
    /* Use VID given in config file if available */
272
    if (!force_mmc && memcmp(aacs->vid, empty_key, 16)) {
npzacs's avatar
npzacs committed
273
        return AACS_SUCCESS;
274 275
    }

cRTrn13's avatar
cRTrn13 committed
276
    MMC* mmc = NULL;
npzacs's avatar
npzacs committed
277
    if (!(mmc = mmc_open(aacs->path))) {
npzacs's avatar
npzacs committed
278
        return AACS_ERROR_MMC_OPEN;
279 280
    }

npzacs's avatar
npzacs committed
281 282
    int error_code = AACS_ERROR_NO_CERT;

283 284
    MKB *hrl_mkb = _get_hrl_mkb(mmc);

285
    for (;hcl && hcl->host_priv_key && hcl->host_cert; hcl = hcl->next) {
286 287 288

        char tmp_str[2*92+1];
        uint8_t priv_key[20], cert[92];
289 290
        hexstring_to_hex_array(priv_key, sizeof(priv_key), hcl->host_priv_key);
        hexstring_to_hex_array(cert,     sizeof(cert),     hcl->host_cert);
291 292

        if (!crypto_aacs_verify_host_cert(cert)) {
npzacs's avatar
npzacs committed
293 294 295 296
            DEBUG(DBG_AACS, "Not using invalid host certificate %s.\n",
                  print_hex(tmp_str, cert, 92));
            continue;
        }
297

298 299 300 301 302 303 304
        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;
        }

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

308
        int mmc_result = mmc_read_vid(mmc, priv_key, cert, aacs->vid, aacs->pmsn);
npzacs's avatar
npzacs committed
309 310
        switch (mmc_result) {
            case MMC_SUCCESS:
311
                mkb_close(hrl_mkb);
npzacs's avatar
npzacs committed
312
                mmc_close(mmc);
npzacs's avatar
npzacs committed
313 314
                /* cache vid */
                keycache_save("vid", aacs->disc_id, aacs->vid, 16);
npzacs's avatar
npzacs committed
315 316 317 318 319 320 321 322
                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;
323 324 325
        }
    }

326
    mkb_close(hrl_mkb);
327 328 329
    mmc_close(mmc);

    DEBUG(DBG_AACS, "Error reading VID!\n");
npzacs's avatar
npzacs committed
330
    return error_code;
331
}
cRTrn13's avatar
openssl  
cRTrn13 committed
332

333
static int _calc_vuk(AACS *aacs, uint8_t *mk, uint8_t *vuk,
npzacs's avatar
npzacs committed
334
                     pk_list *pkl, cert_list *host_cert_list)
335
{
npzacs's avatar
npzacs committed
336 337
    int error_code;

338
    /* Skip if retrieved from config file */
339
    if (memcmp(vuk, empty_key, 16)) {
npzacs's avatar
npzacs committed
340 341 342
        DEBUG(DBG_AACS, "Using VUK from config file\n");
        return AACS_SUCCESS;
    }
343

npzacs's avatar
npzacs committed
344
    /* get cached vuk */
345
    if (keycache_find("vuk", aacs->disc_id, vuk, 16)) {
npzacs's avatar
npzacs committed
346
        DEBUG(DBG_AACS, "Using cached VUK\n");
npzacs's avatar
npzacs committed
347
        return AACS_SUCCESS;
npzacs's avatar
npzacs committed
348 349
    }

350
    /* make sure we have media key */
npzacs's avatar
npzacs committed
351
    error_code = _calc_mk(aacs, mk, pkl);
npzacs's avatar
npzacs committed
352 353
    if (error_code != AACS_SUCCESS) {
        return error_code;
354 355
    }

npzacs's avatar
npzacs committed
356
    /* acquire VID */
357
    error_code = _read_vid(aacs, host_cert_list, 0);
npzacs's avatar
npzacs committed
358 359 360
    if (error_code != AACS_SUCCESS) {
        return error_code;
    }
361

npzacs's avatar
npzacs committed
362 363
    int a;
    gcry_cipher_hd_t gcry_h;
364

npzacs's avatar
npzacs committed
365
    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 0);
366 367
    gcry_cipher_setkey(gcry_h, mk, 16);
    gcry_cipher_decrypt(gcry_h, vuk, 16, aacs->vid, 16);
npzacs's avatar
npzacs committed
368
    gcry_cipher_close(gcry_h);
369

npzacs's avatar
npzacs committed
370
    for (a = 0; a < 16; a++) {
371
        vuk[a] ^= aacs->vid[a];
npzacs's avatar
npzacs committed
372
    }
373

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

npzacs's avatar
npzacs committed
377
    /* cache vuk */
378
    keycache_save("vuk", aacs->disc_id, vuk, 16);
npzacs's avatar
npzacs committed
379

npzacs's avatar
npzacs committed
380
    return AACS_SUCCESS;
cRTrn13's avatar
openssl  
cRTrn13 committed
381 382
}

383 384
static uint16_t _read_u16(AACS_FILE_H *fp)
{
npzacs's avatar
npzacs committed
385
  uint8_t data[2] = {0, 0};
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 424 425 426 427 428 429 430 431

  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
432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
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;
}

448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463
static AACS_FILE_H *_open_content_certificate_file(const char *path)
{
    AACS_FILE_H *fp;
    char        *f_name;

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

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

    X_FREE(f_name);
    return fp;
}

npzacs's avatar
npzacs committed
464
/* Function that collects keys from keydb config entry */
465 466
static void _find_config_entry(AACS *aacs, title_entry_list *ce,
                               uint8_t *mk, uint8_t *vuk)
npzacs's avatar
npzacs committed
467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490
{
    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) {
491
            hexstring_to_hex_array(mk, 16, ce->entry.mek);
npzacs's avatar
npzacs committed
492 493

            DEBUG(DBG_AACS, "Found media key for %s: %s\n",
494
                  ce->entry.discid, print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
495 496 497 498 499 500 501 502 503 504 505
        }

        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) {
506
            hexstring_to_hex_array(vuk, 16, ce->entry.vuk);
npzacs's avatar
npzacs committed
507 508

            DEBUG(DBG_AACS, "Found volume unique key for %s: %s\n",
509
                  ce->entry.discid, print_hex(str, vuk, 16));
npzacs's avatar
npzacs committed
510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531
        }

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

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

532
static int _calc_uks(AACS *aacs, const char *configfile_path)
cRTrn13's avatar
openssl  
cRTrn13 committed
533
{
534
    AACS_FILE_H *fp = NULL;
535
    uint8_t  buf[16];
536
    uint64_t f_pos;
537
    unsigned int i;
npzacs's avatar
npzacs committed
538
    int error_code;
cRTrn13's avatar
openssl  
cRTrn13 committed
539

540 541 542 543 544 545 546 547 548 549 550
    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);

551
    /* Skip if retrieved from config file */
552 553
    if (aacs->uks) {
        keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
554
        return AACS_SUCCESS;
555
    }
556

557
    /* Make sure we have VUK */
558 559
    error_code = _calc_vuk(aacs, mk, vuk, cf->pkl, cf->host_cert_list);
    keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
560 561
    if (error_code != AACS_SUCCESS) {
        return error_code;
562
    }
563

564 565
    DEBUG(DBG_AACS, "Calculate CPS unit keys...\n");

npzacs's avatar
npzacs committed
566
    fp = _open_unit_key_file(aacs->path);
npzacs's avatar
npzacs committed
567 568 569
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
    }
cRTrn13's avatar
openssl  
cRTrn13 committed
570

npzacs's avatar
npzacs committed
571 572
    if ((file_read(fp, buf, 4)) == 4) {
        f_pos = MKINT_BE32(buf);
npzacs's avatar
npzacs committed
573

npzacs's avatar
npzacs committed
574 575 576 577
        // 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
578

npzacs's avatar
npzacs committed
579 580
            X_FREE(aacs->uks);
            aacs->uks = calloc(aacs->num_uks, 16);
npzacs's avatar
npzacs committed
581

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

npzacs's avatar
npzacs committed
584 585 586 587 588
        } 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
589

npzacs's avatar
npzacs committed
590 591 592
        // Read keys
        for (i = 0; i < aacs->num_uks; i++) {
            f_pos += 48;
cRTrn13's avatar
openssl  
cRTrn13 committed
593

npzacs's avatar
npzacs committed
594 595 596 597 598 599
            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
600
            }
cRTrn13's avatar
openssl  
cRTrn13 committed
601

npzacs's avatar
npzacs committed
602 603 604
            gcry_cipher_hd_t gcry_h;
            gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES,
                             GCRY_CIPHER_MODE_ECB, 0);
605
            gcry_cipher_setkey(gcry_h, vuk, 16);
npzacs's avatar
npzacs committed
606 607
            gcry_cipher_decrypt(gcry_h, aacs->uks + 16*i, 16, buf, 16);
            gcry_cipher_close(gcry_h);
cRTrn13's avatar
openssl  
cRTrn13 committed
608

npzacs's avatar
npzacs committed
609 610 611
            char str[40];
            DEBUG(DBG_AACS, "Unit key %d: %s\n", i,
                  print_hex(str, aacs->uks + 16*i, 16));
612 613
        }

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

617
        file_close(fp);
cRTrn13's avatar
openssl  
cRTrn13 committed
618

npzacs's avatar
npzacs committed
619 620 621 622 623 624 625
        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
626 627
}

628 629
static int _calc_title_hash(const char *path, uint8_t *title_hash)
{
630
    AACS_FILE_H *fp = NULL;
631 632 633
    uint8_t *ukf_buf;
    char     str[48];
    int64_t  f_size;
npzacs's avatar
npzacs committed
634
    int      result = AACS_SUCCESS;
635

npzacs's avatar
npzacs committed
636 637 638
    fp = _open_unit_key_file(path);
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
639 640 641 642 643 644 645 646
    }

    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
647 648 649
    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
650

npzacs's avatar
npzacs committed
651 652 653
    } 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);
654 655 656 657 658
    }

    file_close(fp);
    X_FREE(ukf_buf);

npzacs's avatar
npzacs committed
659
    return result;
660
}
cRTrn13's avatar
cRTrn13 committed
661

662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705
static int _get_bus_encryption_enabled(const char *path)
{
    AACS_FILE_H *fp = NULL;
    uint8_t buf[2];
    int bee = 0;

    fp = _open_content_certificate_file(path);
    if (!fp) {
        DEBUG(DBG_AACS | DBG_CRIT, "Unable to open content certificate\n");
        return 0;
    }

    if (file_read(fp, buf, 2) == 2) {
        bee = (buf[1] & 0x80) >> 7;
        DEBUG(DBG_AACS, "Bus Encryption Enabled flag in content certificate: %d\n", bee);
    } else {
        DEBUG(DBG_AACS | DBG_CRIT, "Failed to read Bus Encryption Enabled flag from content certificate file\n");
    }

    file_close(fp);
    return bee;
}

static int _get_bus_encryption_capable(const char *path)
{
    MMC* mmc = NULL;
    uint8_t drive_cert[92];
    int bec = 0;

    if (!(mmc = mmc_open(path))) {
        return 0;
    }

    if (mmc_read_drive_cert(mmc, drive_cert) == MMC_SUCCESS) {
        bec = drive_cert[1] & 1;
        DEBUG(DBG_AACS, "Bus Encryption Capable flag in drive certificate: %d\n", bec);
    } else {
        DEBUG(DBG_AACS | DBG_CRIT, "Unable to read drive certificate\n");
    }

    mmc_close(mmc);
    return bec;
}

npzacs's avatar
npzacs committed
706
static int _verify_ts(uint8_t *buf, size_t size)
707
{
708 709
    uint8_t *ptr;

cRTrn13's avatar
cRTrn13 committed
710 711 712 713 714 715 716
    if (size < 192) {
        return 1;
    }

    for (ptr=buf; ptr < buf+192; ptr++) {
        int failed = 0;
        if (*ptr == 0x47) {
717 718 719 720
            uint8_t *ptr2;

            for (ptr2=ptr; ptr2 < buf + size; ptr2 += 192) {
                if (*ptr2 != 0x47) {
cRTrn13's avatar
cRTrn13 committed
721 722
                    failed = 1;
                    break;
723 724
                }
            }
cRTrn13's avatar
cRTrn13 committed
725 726 727
            if (!failed) {
                return 1;
            }
728 729 730 731
        }
        ptr++;
    }

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

cRTrn13's avatar
cRTrn13 committed
734
    return 0;
735 736
}

737
#define ALIGNED_UNIT_LEN 6144
738
static int _decrypt_unit(AACS *aacs, uint8_t *out_buf, const uint8_t *in_buf, uint32_t curr_uk)
cRTrn13's avatar
cRTrn13 committed
739
{
740 741 742 743 744 745 746
    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);
747
    gcry_cipher_encrypt(gcry_h, key, 16, in_buf, 16);
748
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
749

cRTrn13's avatar
cRTrn13 committed
750
    for (a = 0; a < 16; a++) {
751
        key[a] ^= in_buf[a];
cRTrn13's avatar
cRTrn13 committed
752 753
    }

754 755
    memcpy(out_buf, in_buf, 16); /* first 16 bytes are plain */

756 757 758
    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);
759
    gcry_cipher_decrypt(gcry_h, out_buf + 16, ALIGNED_UNIT_LEN - 16, in_buf + 16, ALIGNED_UNIT_LEN - 16);
760
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
761

762
    if (_verify_ts(out_buf, ALIGNED_UNIT_LEN)) {
cRTrn13's avatar
cRTrn13 committed
763 764 765
        return 1;
    }

766
    if (curr_uk < aacs->num_uks - 1) {
767
        return _decrypt_unit(aacs, out_buf, in_buf, curr_uk++);
npzacs's avatar
npzacs committed
768 769
    }

cRTrn13's avatar
cRTrn13 committed
770 771 772
    return 0;
}

npzacs's avatar
npzacs committed
773 774
void aacs_get_version(int *major, int *minor, int *micro)
{
npzacs's avatar
npzacs committed
775 776 777
    *major = AACS_VERSION_MAJOR;
    *minor = AACS_VERSION_MINOR;
    *micro = AACS_VERSION_MICRO;
npzacs's avatar
npzacs committed
778 779
}

npzacs's avatar
npzacs committed
780
/* aacs_open2() wrapper for backwards compability */
cRTrn13's avatar
cRTrn13 committed
781
AACS *aacs_open(const char *path, const char *configfile_path)
npzacs's avatar
npzacs committed
782 783 784 785 786 787 788 789 790 791 792 793 794 795
{
    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
796
{
npzacs's avatar
npzacs committed
797
    DEBUG(DBG_AACS, "libaacs "AACS_VERSION_STRING" [%zd]\n", sizeof(AACS));
cRTrn13's avatar
cRTrn13 committed
798

799
    DEBUG(DBG_AACS, "Initializing libgcrypt...\n");
npzacs's avatar
npzacs committed
800
    if (!crypto_init()) {
npzacs's avatar
npzacs committed
801
        DEBUG(DBG_AACS | DBG_CRIT, "Failed to initialize libgcrypt\n");
802 803
        return NULL;
    }
804

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

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

npzacs's avatar
npzacs committed
809 810 811 812 813
    *error_code = _calc_title_hash(path, aacs->disc_id);
    if (*error_code != AACS_SUCCESS) {
        aacs_close(aacs);
        return NULL;
    }
npzacs's avatar
npzacs committed
814

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

818 819 820
    aacs->bee = _get_bus_encryption_enabled(path);
    aacs->bec = _get_bus_encryption_capable(path);

npzacs's avatar
npzacs committed
821
    if (*error_code == AACS_SUCCESS) {
822
        DEBUG(DBG_AACS, "AACS initialized!\n");
npzacs's avatar
npzacs committed
823
    } else {
824
        DEBUG(DBG_AACS, "Failed to initialize AACS!\n");
npzacs's avatar
npzacs committed
825
    }
826

npzacs's avatar
npzacs committed
827
    return aacs;
cRTrn13's avatar
cRTrn13 committed
828 829
}

cRTrn13's avatar
cRTrn13 committed
830
void aacs_close(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
831
{
npzacs's avatar
npzacs committed
832 833 834
    if (!aacs)
        return;

835
    X_FREE(aacs->uks);
836
    X_FREE(aacs->cps_units);
npzacs's avatar
npzacs committed
837
    X_FREE(aacs->path);
838

839
    DEBUG(DBG_AACS, "AACS destroyed!\n");
cRTrn13's avatar
cRTrn13 committed
840

cRTrn13's avatar
cRTrn13 committed
841 842 843
    X_FREE(aacs);
}

844
int aacs_decrypt_unit(AACS *aacs, uint8_t *buf)
845
{
846 847
    uint8_t out_buf[ALIGNED_UNIT_LEN];

848 849 850 851 852
    if (!(buf[0] & 0xc0)) {
        // TP_extra_header Copy_permission_indicator == 0, unit is not encrypted
        return 1;
    }

853
    if (_decrypt_unit(aacs, out_buf, buf, aacs->current_cps_unit)) {
854
        memcpy(buf, out_buf, ALIGNED_UNIT_LEN);
855 856 857 858 859 860 861

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

862 863 864
        return 1;
    }

865
    DEBUG(DBG_AACS, "Failed decrypting unit [6144 bytes]\n");
866

867
    return 0;
868
}
cRTrn13's avatar
cRTrn13 committed
869

npzacs's avatar
npzacs committed
870 871 872 873 874 875 876 877 878 879 880 881
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
882 883 884 885 886
const uint8_t *aacs_get_disc_id(AACS *aacs)
{
    return aacs->disc_id;
}

887
const uint8_t *aacs_get_vid(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
888
{
889
    if (!memcmp(aacs->vid, empty_key, 16)) {
npzacs's avatar
npzacs committed
890 891 892 893 894 895
        /* get cached vid */
        if (keycache_find("vid", aacs->disc_id, aacs->vid, 16)) {
            DEBUG(DBG_AACS, "Using cached VID\n");
            return aacs->vid;
        }

896 897
        config_file *cf = keydbcfg_config_load(NULL);
        if (cf) {
898
            _read_vid(aacs, cf->host_cert_list, 0);
899 900 901 902

            keydbcfg_config_file_close(cf);
        }
    }
903 904 905 906 907 908

    if (!memcmp(aacs->vid, empty_key, 16)) {
        DEBUG(DBG_AACS | DBG_CRIT, "aacs_get_vid() failed\n");
        return NULL;
    }

cRTrn13's avatar
cRTrn13 committed
909 910
    return aacs->vid;
}
911

912 913
const uint8_t *aacs_get_pmsn(AACS *aacs)
{
npzacs's avatar
npzacs committed
914
    if (!memcmp(aacs->pmsn, empty_key, 16)) {
915 916
        config_file *cf = keydbcfg_config_load(NULL);
        if (cf) {
917
            _read_vid(aacs, cf->host_cert_list, 1);
918 919 920 921

            keydbcfg_config_file_close(cf);
        }
    }
922 923