tools.h 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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. #ifndef _TOOLS_H
  5. #define _TOOLS_H
  6. #ifdef WIN32
  7. // A bit untidy for now - g3power
  8. #define SPINNER_MOD
  9. #endif
  10. // basic data types
  11. typedef unsigned char u8;
  12. typedef unsigned short u16;
  13. typedef unsigned int u32;
  14. typedef unsigned long long u64;
  15. u16 be16(const u8 *p);
  16. u32 be32(const u8 *p);
  17. u64 be64(const u8 *p);
  18. u64 be34(const u8 *p);
  19. void wbe16(u8 *p, u16 x);
  20. void wbe32(u8 *p, u32 x);
  21. void wbe64(u8 *p, u64 x);
  22. //#define round_down(x,n) ((x) & -(n))
  23. #define round_up(x,n) (-(-(x) & -(n)))
  24. // bignum
  25. int bn_compare(u8 *a, u8 *b, u32 n);
  26. void bn_sub_modulus(u8 *a, u8 *N, u32 n);
  27. void bn_add(u8 *d, u8 *a, u8 *b, u8 *N, u32 n);
  28. void bn_mul(u8 *d, u8 *a, u8 *b, u8 *N, u32 n);
  29. void bn_inv(u8 *d, u8 *a, u8 *N, u32 n); // only for prime N
  30. void bn_exp(u8 *d, u8 *a, u8 *N, u32 n, u8 *e, u32 en);
  31. // crypto
  32. void md5(u8 *data, u32 len, u8 *hash);
  33. void sha(u8 *data, u32 len, u8 *hash);
  34. void get_key(const char *name, u8 *key, u32 len);
  35. void aes_cbc_dec(u8 *key, u8 *iv, u8 *in, u32 len, u8 *out);
  36. void aes_cbc_enc(u8 *key, u8 *iv, u8 *in, u32 len, u8 *out);
  37. void decrypt_title_key(u8 *tik, u8 *title_key);
  38. int check_cert_chain(u8 *data, u32 data_len, u8 *cert, u32 cert_len);
  39. int check_ec(u8 *ng, u8 *ap, u8 *sig, u8 *sig_hash);
  40. void generate_ecdsa(u8 *R, u8 *S, u8 *k, u8 *hash);
  41. int check_ecdsa(u8 *Q, u8 *R, u8 *S, u8 *hash);
  42. void ec_priv_to_pub(u8 *k, u8 *Q);
  43. // compression
  44. void do_yaz0(u8 *in, u32 in_size, u8 *out, u32 out_size);
  45. // error handling
  46. void fatal(const char *s, ...);
  47. void non_fatal(const char *s, ...);
  48. // output formatting
  49. void print_bytes(u8 *x, u32 n);
  50. void hexdump(u8 *x, u32 n);
  51. void dump_tmd(u8 *tmd);
  52. void spinner(u64 x, u64 max);
  53. void progress(u64 x, u64 max);
  54. #endif