aacs.c 20 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
static int _calc_mk(AACS *aacs, uint8_t *mk, pk_list *pkl)
cRTrn13's avatar
keyfile  
cRTrn13 committed
107 108 109
{
    int a, num_uvs = 0;
    size_t len;
110
    uint8_t *buf = NULL;
cRTrn13's avatar
keyfile  
cRTrn13 committed
111
    MKB *mkb = NULL;
112
    const uint8_t *rec, *uvs;
cRTrn13's avatar
keyfile  
cRTrn13 committed
113

114
    /* Skip if retrieved from config file */
115 116 117
    if (memcmp(mk, empty_key, 16)) {
        return AACS_SUCCESS;
    }
118

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

npzacs's avatar
npzacs committed
121
    if ((mkb = mkb_open(aacs->path))) {
122
        DEBUG(DBG_AACS, "Get UVS...\n");
npzacs's avatar
npzacs committed
123
        aacs->mkb_version = mkb_version(mkb);
124 125 126 127 128 129 130 131
        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
132

133 134
        DEBUG(DBG_AACS, "Get cvalues...\n");
        rec = mkb_cvalues(mkb, &len);
135 136

        for (; pkl && pkl->key; pkl = pkl->next) {
137
                uint8_t pk[16];
138
                hexstring_to_hex_array(pk, sizeof(pk), pkl->key);
139 140 141
                DEBUG(DBG_AACS, "Trying processing key...\n");

                for (a = 0; a < num_uvs; a++) {
142
                    if (AACS_SUCCESS == _validate_pk(pk, rec + a * 16, uvs + 1 + a * 5,
143
                      mkb_mk_dv(mkb), mk)) {
144 145
                        mkb_close(mkb);
                        X_FREE(buf);
146

147
                        char str[40];
148
                        DEBUG(DBG_AACS, "Media key: %s\n", print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
149
                        return AACS_SUCCESS;
150 151
                    }
                }
cRTrn13's avatar
keyfile  
cRTrn13 committed
152 153
            }

154 155
        mkb_close(mkb);
        X_FREE(buf);
cRTrn13's avatar
keyfile  
cRTrn13 committed
156

npzacs's avatar
npzacs committed
157
        DEBUG(DBG_AACS | DBG_CRIT, "Error calculating media key. Missing right processing key ?\n");
npzacs's avatar
npzacs committed
158
        return AACS_ERROR_NO_PK;
npzacs's avatar
npzacs committed
159
    }
npzacs's avatar
npzacs committed
160

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

165
static int _read_vid(AACS *aacs, cert_list *hcl)
cRTrn13's avatar
openssl  
cRTrn13 committed
166
{
167 168
    /* Use VID given in config file if available */
    if (memcmp(aacs->vid, empty_key, 16)) {
npzacs's avatar
npzacs committed
169
        return AACS_SUCCESS;
170 171
    }

cRTrn13's avatar
cRTrn13 committed
172
    MMC* mmc = NULL;
npzacs's avatar
npzacs committed
173
    if (!(mmc = mmc_open(aacs->path))) {
npzacs's avatar
npzacs committed
174
        return AACS_ERROR_MMC_OPEN;
175 176
    }

npzacs's avatar
npzacs committed
177 178
    int error_code = AACS_ERROR_NO_CERT;

179
    for (;hcl && hcl->host_priv_key && hcl->host_cert; hcl = hcl->next) {
180 181 182

        char tmp_str[2*92+1];
        uint8_t priv_key[20], cert[92];
183 184
        hexstring_to_hex_array(priv_key, sizeof(priv_key), hcl->host_priv_key);
        hexstring_to_hex_array(cert,     sizeof(cert),     hcl->host_cert);
185 186

        if (!crypto_aacs_verify_host_cert(cert)) {
npzacs's avatar
npzacs committed
187 188 189 190
            DEBUG(DBG_AACS, "Not using invalid host certificate %s.\n",
                  print_hex(tmp_str, cert, 92));
            continue;
        }
191 192 193 194

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

npzacs's avatar
npzacs committed
195
        int mmc_result = mmc_read_vid(mmc, priv_key, cert, aacs->vid);
npzacs's avatar
npzacs committed
196 197 198 199 200 201 202 203 204 205 206
        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;
207 208 209 210 211 212
        }
    }

    mmc_close(mmc);

    DEBUG(DBG_AACS, "Error reading VID!\n");
npzacs's avatar
npzacs committed
213
    return error_code;
214
}
cRTrn13's avatar
openssl  
cRTrn13 committed
215

216 217
static int _calc_vuk(AACS *aacs, uint8_t *mk, uint8_t *vuk,
                     pk_list *pk_list, cert_list *host_cert_list)
218
{
npzacs's avatar
npzacs committed
219 220
    int error_code;

221
    /* Skip if retrieved from config file */
222
    if (memcmp(vuk, empty_key, 16)) {
npzacs's avatar
npzacs committed
223 224 225
        DEBUG(DBG_AACS, "Using VUK from config file\n");
        return AACS_SUCCESS;
    }
226

npzacs's avatar
npzacs committed
227
    /* get cached vuk */
228
    if (keycache_find("vuk", aacs->disc_id, vuk, 16)) {
npzacs's avatar
npzacs committed
229
        DEBUG(DBG_AACS, "Using cached VUK\n");
npzacs's avatar
npzacs committed
230
        return AACS_SUCCESS;
npzacs's avatar
npzacs committed
231 232
    }

233
    /* make sure we have media key */
234
    error_code = _calc_mk(aacs, mk, pk_list);
npzacs's avatar
npzacs committed
235 236
    if (error_code != AACS_SUCCESS) {
        return error_code;
237 238
    }

npzacs's avatar
npzacs committed
239
    /* acquire VID */
240
    error_code = _read_vid(aacs, host_cert_list);
npzacs's avatar
npzacs committed
241 242 243
    if (error_code != AACS_SUCCESS) {
        return error_code;
    }
244

npzacs's avatar
npzacs committed
245 246
    int a;
    gcry_cipher_hd_t gcry_h;
247

npzacs's avatar
npzacs committed
248
    gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 0);
249 250
    gcry_cipher_setkey(gcry_h, mk, 16);
    gcry_cipher_decrypt(gcry_h, vuk, 16, aacs->vid, 16);
npzacs's avatar
npzacs committed
251
    gcry_cipher_close(gcry_h);
252

npzacs's avatar
npzacs committed
253
    for (a = 0; a < 16; a++) {
254
        vuk[a] ^= aacs->vid[a];
npzacs's avatar
npzacs committed
255
    }
256

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

npzacs's avatar
npzacs committed
260
    /* cache vuk */
261
    keycache_save("vuk", aacs->disc_id, vuk, 16);
npzacs's avatar
npzacs committed
262

npzacs's avatar
npzacs committed
263
    return AACS_SUCCESS;
cRTrn13's avatar
openssl  
cRTrn13 committed
264 265
}

266 267
static uint16_t _read_u16(AACS_FILE_H *fp)
{
npzacs's avatar
npzacs committed
268
  uint8_t data[2] = {0, 0};
269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314

  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
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330
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
331
/* Function that collects keys from keydb config entry */
332 333
static void _find_config_entry(AACS *aacs, title_entry_list *ce,
                               uint8_t *mk, uint8_t *vuk)
npzacs's avatar
npzacs committed
334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357
{
    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) {
358
            hexstring_to_hex_array(mk, 16, ce->entry.mek);
npzacs's avatar
npzacs committed
359 360

            DEBUG(DBG_AACS, "Found media key for %s: %s\n",
361
                  ce->entry.discid, print_hex(str, mk, 16));
npzacs's avatar
npzacs committed
362 363 364 365 366 367 368 369 370 371 372
        }

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

            DEBUG(DBG_AACS, "Found volume unique key for %s: %s\n",
376
                  ce->entry.discid, print_hex(str, vuk, 16));
npzacs's avatar
npzacs committed
377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399
        }

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

400
static int _calc_uks(AACS *aacs, const char *configfile_path)
cRTrn13's avatar
openssl  
cRTrn13 committed
401
{
402
    AACS_FILE_H *fp = NULL;
403
    uint8_t  buf[16];
404
    uint64_t f_pos;
405
    unsigned int i;
npzacs's avatar
npzacs committed
406
    int error_code;
cRTrn13's avatar
openssl  
cRTrn13 committed
407

408 409 410 411 412 413 414 415 416 417 418
    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);

419
    /* Skip if retrieved from config file */
420 421
    if (aacs->uks) {
        keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
422
        return AACS_SUCCESS;
423
    }
424

425
    /* Make sure we have VUK */
426 427
    error_code = _calc_vuk(aacs, mk, vuk, cf->pkl, cf->host_cert_list);
    keydbcfg_config_file_close(cf);
npzacs's avatar
npzacs committed
428 429
    if (error_code != AACS_SUCCESS) {
        return error_code;
430
    }
431

432 433
    DEBUG(DBG_AACS, "Calculate CPS unit keys...\n");

npzacs's avatar
npzacs committed
434
    fp = _open_unit_key_file(aacs->path);
npzacs's avatar
npzacs committed
435 436 437
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
    }
cRTrn13's avatar
openssl  
cRTrn13 committed
438

npzacs's avatar
npzacs committed
439 440
    if ((file_read(fp, buf, 4)) == 4) {
        f_pos = MKINT_BE32(buf);
npzacs's avatar
npzacs committed
441

npzacs's avatar
npzacs committed
442 443 444 445
        // 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
446

npzacs's avatar
npzacs committed
447 448
            X_FREE(aacs->uks);
            aacs->uks = calloc(aacs->num_uks, 16);
npzacs's avatar
npzacs committed
449

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

npzacs's avatar
npzacs committed
452 453 454 455 456
        } 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
457

npzacs's avatar
npzacs committed
458 459 460
        // Read keys
        for (i = 0; i < aacs->num_uks; i++) {
            f_pos += 48;
cRTrn13's avatar
openssl  
cRTrn13 committed
461

npzacs's avatar
npzacs committed
462 463 464 465 466 467
            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
468
            }
cRTrn13's avatar
openssl  
cRTrn13 committed
469

npzacs's avatar
npzacs committed
470 471 472
            gcry_cipher_hd_t gcry_h;
            gcry_cipher_open(&gcry_h, GCRY_CIPHER_AES,
                             GCRY_CIPHER_MODE_ECB, 0);
473
            gcry_cipher_setkey(gcry_h, vuk, 16);
npzacs's avatar
npzacs committed
474 475
            gcry_cipher_decrypt(gcry_h, aacs->uks + 16*i, 16, buf, 16);
            gcry_cipher_close(gcry_h);
cRTrn13's avatar
openssl  
cRTrn13 committed
476

npzacs's avatar
npzacs committed
477 478 479
            char str[40];
            DEBUG(DBG_AACS, "Unit key %d: %s\n", i,
                  print_hex(str, aacs->uks + 16*i, 16));
480 481
        }

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

485
        file_close(fp);
cRTrn13's avatar
openssl  
cRTrn13 committed
486

npzacs's avatar
npzacs committed
487 488 489 490 491 492 493
        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
494 495
}

496 497
static int _calc_title_hash(const char *path, uint8_t *title_hash)
{
498
    AACS_FILE_H *fp = NULL;
499 500 501
    uint8_t *ukf_buf;
    char     str[48];
    int64_t  f_size;
npzacs's avatar
npzacs committed
502
    int      result = AACS_SUCCESS;
503

npzacs's avatar
npzacs committed
504 505 506
    fp = _open_unit_key_file(path);
    if (!fp) {
        return AACS_ERROR_CORRUPTED_DISC;
507 508 509 510 511 512 513 514
    }

    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
515 516 517
    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
518

npzacs's avatar
npzacs committed
519 520 521
    } 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);
522 523 524 525 526
    }

    file_close(fp);
    X_FREE(ukf_buf);

npzacs's avatar
npzacs committed
527
    return result;
528
}
cRTrn13's avatar
cRTrn13 committed
529

npzacs's avatar
npzacs committed
530
static int _verify_ts(uint8_t *buf, size_t size)
531
{
532 533
    uint8_t *ptr;

cRTrn13's avatar
cRTrn13 committed
534 535 536 537 538 539 540
    if (size < 192) {
        return 1;
    }

    for (ptr=buf; ptr < buf+192; ptr++) {
        int failed = 0;
        if (*ptr == 0x47) {
541 542 543 544
            uint8_t *ptr2;

            for (ptr2=ptr; ptr2 < buf + size; ptr2 += 192) {
                if (*ptr2 != 0x47) {
cRTrn13's avatar
cRTrn13 committed
545 546
                    failed = 1;
                    break;
547 548
                }
            }
cRTrn13's avatar
cRTrn13 committed
549 550 551
            if (!failed) {
                return 1;
            }
552 553 554 555
        }
        ptr++;
    }

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

cRTrn13's avatar
cRTrn13 committed
558
    return 0;
559 560
}

561
#define ALIGNED_UNIT_LEN 6144
562
static int _decrypt_unit(AACS *aacs, uint8_t *out_buf, const uint8_t *in_buf, uint32_t curr_uk)
cRTrn13's avatar
cRTrn13 committed
563
{
564 565 566 567 568 569 570
    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);
571
    gcry_cipher_encrypt(gcry_h, key, 16, in_buf, 16);
572
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
573

cRTrn13's avatar
cRTrn13 committed
574
    for (a = 0; a < 16; a++) {
575
        key[a] ^= in_buf[a];
cRTrn13's avatar
cRTrn13 committed
576 577
    }

578 579
    memcpy(out_buf, in_buf, 16); /* first 16 bytes are plain */

580 581 582
    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);
583
    gcry_cipher_decrypt(gcry_h, out_buf + 16, ALIGNED_UNIT_LEN - 16, in_buf + 16, ALIGNED_UNIT_LEN - 16);
584
    gcry_cipher_close(gcry_h);
cRTrn13's avatar
cRTrn13 committed
585

586
    if (_verify_ts(out_buf, ALIGNED_UNIT_LEN)) {
cRTrn13's avatar
cRTrn13 committed
587 588 589
        return 1;
    }

590
    if (curr_uk < aacs->num_uks - 1) {
591
        return _decrypt_unit(aacs, out_buf, in_buf, curr_uk++);
npzacs's avatar
npzacs committed
592 593
    }

cRTrn13's avatar
cRTrn13 committed
594 595 596
    return 0;
}

npzacs's avatar
npzacs committed
597 598
void aacs_get_version(int *major, int *minor, int *micro)
{
npzacs's avatar
npzacs committed
599 600 601
    *major = AACS_VERSION_MAJOR;
    *minor = AACS_VERSION_MINOR;
    *micro = AACS_VERSION_MICRO;
npzacs's avatar
npzacs committed
602 603
}

npzacs's avatar
npzacs committed
604
/* aacs_open2() wrapper for backwards compability */
cRTrn13's avatar
cRTrn13 committed
605
AACS *aacs_open(const char *path, const char *configfile_path)
npzacs's avatar
npzacs committed
606 607 608 609 610 611 612 613 614 615 616 617 618 619
{
    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
620
{
npzacs's avatar
npzacs committed
621
    DEBUG(DBG_AACS, "libaacs "AACS_VERSION_STRING" [%zd]\n", sizeof(AACS));
cRTrn13's avatar
cRTrn13 committed
622

623
    DEBUG(DBG_AACS, "Initializing libgcrypt...\n");
npzacs's avatar
npzacs committed
624
    if (!crypto_init()) {
npzacs's avatar
npzacs committed
625
        DEBUG(DBG_AACS | DBG_CRIT, "Failed to initialize libgcrypt\n");
626 627
        return NULL;
    }
628

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

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

npzacs's avatar
npzacs committed
633 634 635 636 637
    *error_code = _calc_title_hash(path, aacs->disc_id);
    if (*error_code != AACS_SUCCESS) {
        aacs_close(aacs);
        return NULL;
    }
npzacs's avatar
npzacs committed
638

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

npzacs's avatar
npzacs committed
642 643 644 645 646
    if (*error_code == AACS_SUCCESS) {
        DEBUG(DBG_AACS, "AACS initialized! (%p)\n", aacs);
    } else {
        DEBUG(DBG_AACS, "Failed to initialize AACS! (%p)\n", aacs);
    }
647

npzacs's avatar
npzacs committed
648
    return aacs;
cRTrn13's avatar
cRTrn13 committed
649 650
}

cRTrn13's avatar
cRTrn13 committed
651
void aacs_close(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
652
{
npzacs's avatar
npzacs committed
653 654 655
    if (!aacs)
        return;

656
    X_FREE(aacs->uks);
657
    X_FREE(aacs->cps_units);
npzacs's avatar
npzacs committed
658
    X_FREE(aacs->path);
659

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

cRTrn13's avatar
cRTrn13 committed
662 663 664
    X_FREE(aacs);
}

665
int aacs_decrypt_unit(AACS *aacs, uint8_t *buf)
666
{
667 668
    uint8_t out_buf[ALIGNED_UNIT_LEN];

669 670 671 672 673
    if (!(buf[0] & 0xc0)) {
        // TP_extra_header Copy_permission_indicator == 0, unit is not encrypted
        return 1;
    }

674
    if (_decrypt_unit(aacs, out_buf, buf, aacs->current_cps_unit)) {
675
        memcpy(buf, out_buf, ALIGNED_UNIT_LEN);
676 677 678 679 680 681 682

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

683 684 685
        return 1;
    }

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

688
    return 0;
689
}
cRTrn13's avatar
cRTrn13 committed
690

npzacs's avatar
npzacs committed
691 692 693 694 695 696 697 698 699 700 701 702
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
703 704 705 706 707
const uint8_t *aacs_get_disc_id(AACS *aacs)
{
    return aacs->disc_id;
}

708
const uint8_t *aacs_get_vid(AACS *aacs)
cRTrn13's avatar
cRTrn13 committed
709
{
710 711 712 713 714 715 716 717
    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
718 719
    return aacs->vid;
}
720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746

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