tools.c

   1 // Copyright 2007,2008  Segher Boessenkool  <segher@kernel.crashing.org>
   2 // Licensed under the terms of the GNU GPL, version 2
   3 // http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt
   4
   5 #include "tools.h"
   6
   7 #include <openssl/aes.h>
   8 #include <openssl/sha.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include <stdio.h>
  12
  13 //
  14 // basic data types
  15 //
  16
  17 u32 be32(u8 *p)
  18 {
  19         return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
  20 }
  21
  22 u64 be64(u8 *p)
  23 {
  24         return ((u64)be32(p) << 32) | be32(p + 4);
  25 }
  26
  27 u64 be34(u8 *p)
  28 {
  29         return 4 * (u64)be32(p);
  30 }
  31
  32 //
  33 // crypto
  34 //
  35
  36 void sha(u8 *data, u32 len, u8 *hash)
  37 {
  38         SHA1(data, len, hash);
  39 }
  40
  41 void get_key(const char *name, u8 *key, u32 len)
  42 {
  43         char path[256];
  44         char *home;
  45         FILE *fp;
  46
  47         home = getenv("HOME");
  48         if (home == 0)
  49                 fatal("cannot find HOME");
  50         snprintf(path, sizeof path, "%s/.wii/%s", home, name);
  51
  52         fp = fopen(path, "rb");
  53         if (fp == 0)
  54                 fatal("cannot open common-key");
  55         if (fread(key, len, 1, fp) != 1)
  56                 fatal("error reading common-key");
  57         fclose(fp);
  58 }
  59
  60 void aes_cbc_dec(u8 *key, u8 *iv, u8 *in, u32 len, u8 *out)
  61 {
  62         AES_KEY aes_key;
  63
  64         AES_set_decrypt_key(key, 128, &aes_key);
  65         AES_cbc_encrypt(in, out, len, &aes_key, iv, AES_DECRYPT);
  66 }
  67
  68 void decrypt_title_key(u8 *title_key_crypted, u8 *title_id, u8 *title_key)
  69 {
  70         u8 common_key[16];
  71         u8 iv[16];
  72
  73         get_key("common-key", common_key, 16);
  74
  75         memset(iv, 0, sizeof iv);
  76         memcpy(iv, title_id, 8);
  77         aes_cbc_dec(common_key, iv, title_key_crypted, 16, title_key);
  78 }
  79
  80 int check_cert(u8 *data, u32 data_len, u8 *cert, u32 cert_len)
  81 {
  82         return -1;
  83 }
  84
  85 //
  86 // compression
  87 //
  88
  89 void do_yaz0(u8 *in, u32 in_size, u8 *out, u32 out_size)
  90 {
  91         u32 nout;
  92         u8 bits;
  93         u32 nbits;
  94         u32 n, d, i;
  95
  96         nbits = 0;
  97         in += 0x10;
  98         for (nout = 0; nout < out_size; ) {
  99                 if (nbits == 0) {
 100                         bits = *in++;
 101                         nbits = 8;
 102                 }
 103
 104                 if ((bits & 0x80) != 0) {
 105                         *out++ = *in++;
 106                         nout++;
 107                 } else {
 108                         n = *in++;
 109                         d = *in++;
 110                         d |= (n << 8) & 0xf00;
 111                         n >>= 4;
 112                         if (n == 0)
 113                                 n = 0x10 + *in++;
 114                         n += 2;
 115                         d++;
 116
 117                         for (i = 0; i < n; i++) {
 118                                 *out = *(out - d);
 119                                 out++;
 120                         }
 121                         nout += n;
 122                 }
 123
 124                 nbits--;
 125                 bits <<= 1;
 126         };
 127 }
 128
 129 //
 130 // error handling
 131 //
 132
 133 void fatal(const char *s)
 134 {
 135         perror(s);
 136
 137         exit(1);
 138 }
 139
 140 //
 141 // output formatting
 142 //
 143
 144 void print_bytes(u8 *x, u32 n)
 145 {
 146         u32 i;
 147
 148         for (i = 0; i < n; i++)
 149                 fprintf(stderr, "%02x", x[i]);
 150 }
 151
 152 void hexdump(u8 *x, u32 n)
 153 {
 154         u32 i, j;
 155
 156         for (i = 0; i < n; i += 16) {
 157                 fprintf(stderr, "%04x:", i);
 158                 for (j = 0; j < 16 && i + j < n; j++) {
 159                         if ((j & 3) == 0)
 160                                 fprintf(stderr, " ");
 161                         fprintf(stderr, "%02x", *x++);
 162                 }
 163                 fprintf(stderr, "\n");
 164         }
 165         if (n & 15)
 166                 fprintf(stderr, "\n");
 167 }