aacs.c 21.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];
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
static int _read_vid(AACS *aacs, cert_list *hcl)
cRTrn13's avatar
openssl  
cRTrn13 committed
207
{
208 209
    /* Use VID given in config file if available */
    if (memcmp(aacs->vid, empty_key, 16)) {
npzacs's avatar
npzacs committed
210
        return AACS_SUCCESS;
211 212
    }

cRTrn13's avatar
cRTrn13 committed
213
    MMC* mmc = NULL;
npzacs's avatar
npzacs committed
214
    if (!(mmc = mmc_open(aacs->path))) {
npzacs's avatar
npzacs committed
215
        return AACS_ERROR_MMC_OPEN;
216 217
    }

npzacs's avatar
npzacs committed
218 219
    int error_code = AACS_ERROR_NO_CERT;

220
    for (;hcl && hcl->host_priv_key && hcl->host_cert; hcl = hcl->next) {
221 222 223

        char tmp_str[2*92+1];
        uint8_t priv_key[20], cert[92];
224 225
        hexstring_to_hex_array(priv_key, sizeof(priv_key), hcl->host_priv_key);
        hexstring_to_hex_array(cert,     sizeof(cert),     hcl->host_cert);
226 227

        if (!crypto_aacs_verify_host_cert(cert)) {
npzacs's avatar
npzacs committed
228 229 230 231
            DEBUG(DBG_AACS, "Not using invalid host certificate %s.\n",
                  print_hex(tmp_str, cert, 92));
            continue;
        }
232 233 234 235

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

npzacs's avatar
npzacs committed
236
        int mmc_result = mmc_read_vid(mmc, priv_key, cert, aacs->vid);
npzacs's avatar
npzacs committed
237 238 239 240 241 242 243 244 245 246 247
        switch (mmc_result) {
            case MMC_SUCCESS:
                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;
248 249 250 251 252 253
        }
    }

    mmc_close(mmc);

    DEBUG(DBG_AACS, "Error reading VID!\n");
npzacs's avatar
npzacs committed
254
    return error_code;
255
}
cRTrn13's avatar
openssl  
cRTrn13 committed
256

257 258
static int _calc_vuk(AACS *aacs, uint8_t *mk, uint8_t *vuk,
                     pk_list *pk_list, cert_list *host_cert_list)
259
{
npzacs's avatar
npzacs committed
260 261
    int error_code;

262
    /* Skip if retrieved from config file */
263
    if (memcmp(vuk, empty_key, 16)) {
npzacs's avatar
npzacs committed
264 265 266
        DEBUG(DBG_AACS, "Using VUK from config file\n");
        return AACS_SUCCESS;
    }
267

npzacs's avatar
npzacs committed
268
    /* get cached vuk */
269
    if (keycache_find("vuk", aacs->disc_id, vuk, 16)) {
npzacs's avatar
npzacs committed
270
        DEBUG(DBG_AACS, "Using cached VUK\n");
npzacs's avatar
npzacs committed
271
        return AACS_SUCCESS;
npzacs's avatar
npzacs committed
272 273
    }

274
    /* make sure we have media key */
275
    error_code = _calc_mk(aacs, mk, pk_list);
npzacs's avatar
npzacs committed
276 277
    if (error_code != AACS_SUCCESS) {
        return error_code;
278 279
    }

npzacs's avatar
npzacs committed
280
    /* acquire VID */
281
    error_code = _read_vid(aacs, host_cert_list);
npzacs's avatar
npzacs committed
282 283 284
    if (error_code != AACS_SUCCESS) {
        return error_code;
    }
285

npzacs's avatar
npzacs committed
286 287
    int a;
    gcry_cipher_hd_t gcry_h;
288

npzacs's avatar
npzacs committed
289
    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 0);
290 291
    gcry_cipher_setkey(gcry_h, mk, 16);
    gcry_cipher_decrypt(gcry_h, vuk, 16, aacs->vid, 16);
npzacs's avatar
npzacs committed
292
    gcry_cipher_close(gcry_h);
293

npzacs's avatar
npzacs committed
294
    for (a = 0; a < 16; a++) {
295
        vuk[a] ^= aacs->vid[a];
npzacs's avatar
npzacs committed
296
    }
297

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

npzacs's avatar
npzacs committed
301
    /* cache vuk */
302
    keycache_save("vuk", aacs->disc_id, vuk, 16);
npzacs's avatar
npzacs committed
303

npzacs's avatar
npzacs committed
304
    return AACS_SUCCESS;
cRTrn13's avatar
openssl  
cRTrn13 committed
305 306
}

307 308
static uint16_t _read_u16(AACS_FILE_H *fp)
{
npzacs's avatar
npzacs committed
309
  uint8_t data[2] = {0, 0};
310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355

  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
356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371
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
372
/* Function that collects keys from keydb config entry */
373 374
static void _find_config_entry(AACS *aacs, title_entry_list *ce,
                               uint8_t *mk, uint8_t *vuk)
npzacs's avatar
npzacs committed
375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398
{
    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) {
399
            hexstring_to_hex_array(mk, 16, ce->entry.mek);
npzacs's avatar
npzacs committed
400 401

            DEBUG(DBG_AACS, "Found media key for %s: %s\n",
402
                  ce->entry.discid, print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
403 404 405 406 407 408 409 410 411 412 413
        }

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

            DEBUG(DBG_AACS, "Found volume unique key for %s: %s\n",
417
                  ce->entry.discid, print_hex(str, vuk, 16));
npzacs's avatar
npzacs committed
418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440
        }

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

441
static int _calc_uks(AACS *aacs, const char *configfile_path)
cRTrn13's avatar
openssl  
cRTrn13 committed
442
{
443
    AACS_FILE_H *fp = NULL;
444
    uint8_t  buf[16];
445
    uint64_t f_pos;
446
    unsigned int i;
npzacs's avatar
npzacs committed
447
    int error_code;
cRTrn13's avatar
openssl  
cRTrn13 committed
448

449 450 451 452 453 454 455 456 457 458 459
    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);

460
    /* Skip if retrieved from config file */
461 462
    if (aacs->uks) {
        keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
463
        return AACS_SUCCESS;
464
    }
465

466
    /* Make sure we have VUK */
467 468
    error_code = _calc_vuk(aacs, mk, vuk, cf->pkl, cf->host_cert_list);
    keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
469 470
    if (error_code != AACS_SUCCESS) {
        return error_code;
471
    }
472

473 474
    DEBUG(DBG_AACS, "Calculate CPS unit keys...\n");

npzacs's avatar
npzacs committed
475
    fp = _open_unit_key_file(aacs->path);
npzacs's avatar
npzacs committed
476 477 478
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
    }
cRTrn13's avatar
openssl  
cRTrn13 committed
479

npzacs's avatar
npzacs committed
480 481
    if ((file_read(fp, buf, 4)) == 4) {
        f_pos = MKINT_BE32(buf);
npzacs's avatar
npzacs committed
482

npzacs's avatar
npzacs committed
483 484 485 486
        // 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
487

npzacs's avatar
npzacs committed
488 489
            X_FREE(aacs->uks);
            aacs->uks = calloc(aacs->num_uks, 16);
npzacs's avatar
npzacs committed
490

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

npzacs's avatar
npzacs committed
493 494 495 496 497
        } 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
498

npzacs's avatar
npzacs committed
499 500 501
        // Read keys
        for (i = 0; i < aacs->num_uks; i++) {
            f_pos += 48;
cRTrn13's avatar
openssl  
cRTrn13 committed
502

npzacs's avatar
npzacs committed
503 504 505 506 507 508
            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
509
            }
cRTrn13's avatar
openssl  
cRTrn13 committed
510

npzacs's avatar
npzacs committed
511 512 513
            gcry_cipher_hd_t gcry_h;
            gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES,
                             GCRY_CIPHER_MODE_ECB, 0);
514
            gcry_cipher_setkey(gcry_h, vuk, 16);
npzacs's avatar
npzacs committed
515 516
            gcry_cipher_decrypt(gcry_h, aacs->uks + 16*i, 16, buf, 16);
            gcry_cipher_close(gcry_h);
cRTrn13's avatar
openssl  
cRTrn13 committed
517

npzacs's avatar
npzacs committed
518 519 520
            char str[40];
            DEBUG(DBG_AACS, "Unit key %d: %s\n", i,
                  print_hex(str, aacs->uks + 16*i, 16));
521 522
        }

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

526
        file_close(fp);
cRTrn13's avatar
openssl  
cRTrn13 committed
527

npzacs's avatar
npzacs committed
528 529 530 531 532 533 534
        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
535 536
}

537 538
static int _calc_title_hash(const char *path, uint8_t *title_hash)
{
539
    AACS_FILE_H *fp = NULL;
540 541 542
    uint8_t *ukf_buf;
    char     str[48];
    int64_t  f_size;
npzacs's avatar
npzacs committed
543
    int      result = AACS_SUCCESS;
544

npzacs's avatar
npzacs committed
545 546 547
    fp = _open_unit_key_file(path);
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
548 549 550 551 552 553 554 555
    }

    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
556 557 558
    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
559

npzacs's avatar
npzacs committed
560 561 562
    } 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);
563 564 565 566 567
    }

    file_close(fp);
    X_FREE(ukf_buf);

npzacs's avatar
npzacs committed
568
    return result;
569
}
cRTrn13's avatar
cRTrn13 committed
570

npzacs's avatar
npzacs committed
571
static int _verify_ts(uint8_t *buf, size_t size)
572
{
573 574
    uint8_t *ptr;

cRTrn13's avatar
cRTrn13 committed
575 576 577 578 579 580 581
    if (size < 192) {
        return 1;
    }

    for (ptr=buf; ptr < buf+192; ptr++) {
        int failed = 0;
        if (*ptr == 0x47) {
582 583 584 585
            uint8_t *ptr2;

            for (ptr2=ptr; ptr2 < buf + size; ptr2 += 192) {
                if (*ptr2 != 0x47) {
cRTrn13's avatar
cRTrn13 committed
586 587
                    failed = 1;
                    break;
588 589
                }
            }
cRTrn13's avatar
cRTrn13 committed
590 591 592
            if (!failed) {
                return 1;
            }
593 594 595 596
        }
        ptr++;
    }

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

cRTrn13's avatar
cRTrn13 committed
599
    return 0;
600 601
}

602
#define ALIGNED_UNIT_LEN 6144
603
static int _decrypt_unit(AACS *aacs, uint8_t *out_buf, const uint8_t *in_buf, uint32_t curr_uk)
cRTrn13's avatar
cRTrn13 committed
604
{
605 606 607 608 609 610 611
    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);
612
    gcry_cipher_encrypt(gcry_h, key, 16, in_buf, 16);
613
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
614

cRTrn13's avatar
cRTrn13 committed
615
    for (a = 0; a < 16; a++) {
616
        key[a] ^= in_buf[a];
cRTrn13's avatar
cRTrn13 committed
617 618
    }

619 620
    memcpy(out_buf, in_buf, 16); /* first 16 bytes are plain */

621 622 623
    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);
624
    gcry_cipher_decrypt(gcry_h, out_buf + 16, ALIGNED_UNIT_LEN - 16, in_buf + 16, ALIGNED_UNIT_LEN - 16);
625
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
626

627
    if (_verify_ts(out_buf, ALIGNED_UNIT_LEN)) {
cRTrn13's avatar
cRTrn13 committed
628 629 630
        return 1;
    }

631
    if (curr_uk < aacs->num_uks - 1) {
632
        return _decrypt_unit(aacs, out_buf, in_buf, curr_uk++);
npzacs's avatar
npzacs committed
633 634
    }

cRTrn13's avatar
cRTrn13 committed
635 636 637
    return 0;
}

npzacs's avatar
npzacs committed
638 639
void aacs_get_version(int *major, int *minor, int *micro)
{
npzacs's avatar
npzacs committed
640 641 642
    *major = AACS_VERSION_MAJOR;
    *minor = AACS_VERSION_MINOR;
    *micro = AACS_VERSION_MICRO;
npzacs's avatar
npzacs committed
643 644
}

npzacs's avatar
npzacs committed
645
/* aacs_open2() wrapper for backwards compability */
cRTrn13's avatar
cRTrn13 committed
646
AACS *aacs_open(const char *path, const char *configfile_path)
npzacs's avatar
npzacs committed
647 648 649 650 651 652 653 654 655 656 657 658 659 660
{
    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
661
{
npzacs's avatar
npzacs committed
662
    DEBUG(DBG_AACS, "libaacs "AACS_VERSION_STRING" [%zd]\n", sizeof(AACS));
cRTrn13's avatar
cRTrn13 committed
663

664
    DEBUG(DBG_AACS, "Initializing libgcrypt...\n");
npzacs's avatar
npzacs committed
665
    if (!crypto_init()) {
npzacs's avatar
npzacs committed
666
        DEBUG(DBG_AACS | DBG_CRIT, "Failed to initialize libgcrypt\n");
667 668
        return NULL;
    }
669

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

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

npzacs's avatar
npzacs committed
674 675 676 677 678
    *error_code = _calc_title_hash(path, aacs->disc_id);
    if (*error_code != AACS_SUCCESS) {
        aacs_close(aacs);
        return NULL;
    }
npzacs's avatar
npzacs committed
679

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

npzacs's avatar
npzacs committed
683 684 685 686 687
    if (*error_code == AACS_SUCCESS) {
        DEBUG(DBG_AACS, "AACS initialized! (%p)\n", aacs);
    } else {
        DEBUG(DBG_AACS, "Failed to initialize AACS! (%p)\n", aacs);
    }
688

npzacs's avatar
npzacs committed
689
    return aacs;
cRTrn13's avatar
cRTrn13 committed
690 691
}

cRTrn13's avatar
cRTrn13 committed
692
void aacs_close(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
693
{
npzacs's avatar
npzacs committed
694 695 696
    if (!aacs)
        return;

697
    X_FREE(aacs->uks);
698
    X_FREE(aacs->cps_units);
npzacs's avatar
npzacs committed
699
    X_FREE(aacs->path);
700

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

cRTrn13's avatar
cRTrn13 committed
703 704 705
    X_FREE(aacs);
}

706
int aacs_decrypt_unit(AACS *aacs, uint8_t *buf)
707
{
708 709
    uint8_t out_buf[ALIGNED_UNIT_LEN];

710 711 712 713 714
    if (!(buf[0] & 0xc0)) {
        // TP_extra_header Copy_permission_indicator == 0, unit is not encrypted
        return 1;
    }

715
    if (_decrypt_unit(aacs, out_buf, buf, aacs->current_cps_unit)) {
716
        memcpy(buf, out_buf, ALIGNED_UNIT_LEN);
717 718 719 720 721 722 723

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

724 725 726
        return 1;
    }

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

729
    return 0;
730
}
cRTrn13's avatar
cRTrn13 committed
731

npzacs's avatar
npzacs committed
732 733 734 735 736 737 738 739 740 741 742 743
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
744 745 746 747 748
const uint8_t *aacs_get_disc_id(AACS *aacs)
{
    return aacs->disc_id;
}

749
const uint8_t *aacs_get_vid(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
750
{
751 752 753 754 755 756 757 758
    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
759 760
    return aacs->vid;
}
761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787

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