utils/imxtools/sbtools/crypto.h

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2010 Amaury Pouly
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  18  * KIND, either express or implied.
  19  *
  20  ****************************************************************************/
  21 #ifndef __CRYPTO_H__
  22 #define __CRYPTO_H__
  23
  24 #include <stdio.h>
  25 #include <stdint.h>
  26 #include <string.h>
  27
  28 typedef uint8_t byte;
  29
  30 /* aes128.c */
  31 void xor_(byte *a, byte *b, int n);
  32 void EncryptAES(byte *msg, byte *key, byte *c);
  33 void DecryptAES(byte *c, byte *key, byte *m);
  34 void Pretty(byte* b,int len,const char* label);
  35 void cbc_mac(
  36     byte *in_data, /* Input data */
  37     byte *out_data, /* Output data (or NULL) */
  38     int nr_blocks, /* Number of blocks to encrypt/decrypt (one block=16 bytes) */
  39     byte key[16], /* Key */
  40     byte iv[16], /* Initialisation Vector */
  41     byte (*out_cbc_mac)[16], /* CBC-MAC of the result (or NULL) */
  42     int encrypt /* 1 to encrypt, 0 to decrypt */
  43     );
  44
  45 /* crypto.c */
  46 enum crypto_method_t
  47 {
  48     CRYPTO_NONE, /* disable */
  49     CRYPTO_KEY, /* key */
  50     CRYPTO_XOR_KEY, /* XOR key */
  51     CRYPTO_USBOTP, /* use usbotp device */
  52 };
  53
  54 /* parameter can be:
  55  * - CRYPTO_KEY: array of 16-bytes (the key)
  56  * - CRYPTO_USBOTP: 32-bit integer: vid << 16 | pid */
  57 void crypto_setup(enum crypto_method_t method, void *param);
  58
  59 #define CRYPTO_ERROR_SUCCESS    0
  60 #define CRYPTO_ERROR_BADSETUP   -1  /* bad crypto setup */
  61 #define CRYPTO_ERROR_NODEVICE   -2 /* no device with vid:pid */
  62 #define CRYPTO_ERROR_BADENDP    -3 /* device doesn't have the required endpoints */
  63 #define CRYPTO_ERROR_CLAIMFAIL  -4 /* device interface claim error */
  64 #define CRYPTO_ERROR_DEVREJECT  -5 /* device rejected cypto operation */
  65 #define CRYPTO_ERROR_DEVSILENT  -6 /* device did not notify completion */
  66 #define CRYPTO_ERROR_DEVERR     -7 /* device did something wrong (like return too small buffer) */
  67 #define CRYPTO_NUM_ERRORS       8
  68 /* return 0 on success, <0 on error */
  69 int crypto_apply(
  70     byte *in_data, /* Input data */
  71     byte *out_data, /* Output data (or NULL) */
  72     int nr_blocks, /* Number of blocks (one block=16 bytes) */
  73     byte iv[16], /* IV */
  74     byte (*out_cbc_mac)[16], /* CBC-MAC of the result (or NULL) */
  75     int encrypt);
  76
  77 union xorcrypt_key_t
  78 {
  79     uint8_t key[64];
  80     uint32_t k[16];
  81 };
  82
  83 /* all-in-one function */
  84 struct crypto_key_t
  85 {
  86     enum crypto_method_t method;
  87     union
  88     {
  89         byte key[16];
  90         union xorcrypt_key_t xor_key[2];
  91         uint32_t vid_pid;
  92         byte param[0];
  93     }u;
  94 };
  95
  96 int crypto_cbc(
  97     byte *in_data, /* Input data */
  98     byte *out_data, /* Output data (or NULL) */
  99     int nr_blocks, /* Number of blocks (one block=16 bytes) */
 100     struct crypto_key_t *key, /* Key */
 101     byte iv[16], /* IV */
 102     byte (*out_cbc_mac)[16], /* CBC-MAC of the result (or NULL) */
 103     int encrypt);
 104
 105 /* crc.c */
 106 uint32_t crc(byte *data, int size);
 107 uint32_t crc_continue(uint32_t previous_crc, byte *data, int size);
 108
 109 /* sha1.c */
 110 struct sha_1_params_t
 111 {
 112     uint32_t hash[5];
 113     uint64_t buffer_nr_bits;
 114     uint32_t w[80];
 115 };
 116
 117 void sha_1_init(struct sha_1_params_t *params);
 118 void sha_1_block(struct sha_1_params_t *params, uint32_t cur_hash[5], byte *data);
 119 void sha_1_update(struct sha_1_params_t *params, byte *buffer, int size);
 120 void sha_1_finish(struct sha_1_params_t *params);
 121 void sha_1_output(struct sha_1_params_t *params, byte *out);
 122
 123 /* xorcrypt.c */
 124
 125 // WARNING those functions modifies the keys !!
 126 uint32_t xor_encrypt(union xorcrypt_key_t keys[2], void *data, int size);
 127 uint32_t xor_decrypt(union xorcrypt_key_t keys[2], void *data, int size);
 128 void xor_generate_key(uint32_t laserfuse[3], union xorcrypt_key_t key[2]);
 129
 130 #endif /* __CRYPTO_H__ */