| 1 | /* |
| 2 | * Copyright (C) 2009-2011 Gabor Juhos <juhosg@openwrt.org> |
| 3 | * |
| 4 | * This program is free software; you can redistribute it and/or modify it |
| 5 | * under the terms of the GNU General Public License version 2 as published |
| 6 | * by the Free Software Foundation. |
| 7 | * |
| 8 | */ |
| 9 | |
| 10 | #ifndef _BUFFALO_LIB_H |
| 11 | #define _BUFFALO_LIB_H |
| 12 | |
| 13 | #include <stdint.h> |
| 14 | |
| 15 | #define ARRAY_SIZE(_a) (sizeof((_a)) / sizeof((_a)[0])) |
| 16 | #define BIT(_x) (1UL << (_x)) |
| 17 | |
| 18 | #define TAG_BRAND_LEN 32 |
| 19 | #define TAG_PRODUCT_LEN 32 |
| 20 | #define TAG_VERSION_LEN 8 |
| 21 | #define TAG_REGION_LEN 2 |
| 22 | #define TAG_LANGUAGE_LEN 8 |
| 23 | #define TAG_PLATFORM_LEN 8 |
| 24 | #define TAG_HWVER_LEN 4 |
| 25 | #define TAG_HWVER_VAL_LEN 4 |
| 26 | |
| 27 | struct buffalo_tag { |
| 28 | unsigned char product[TAG_PRODUCT_LEN]; |
| 29 | unsigned char brand[TAG_BRAND_LEN]; |
| 30 | unsigned char ver_major[TAG_VERSION_LEN]; |
| 31 | unsigned char ver_minor[TAG_VERSION_LEN]; |
| 32 | unsigned char region_code[2]; |
| 33 | uint32_t region_mask; |
| 34 | unsigned char unknown0[2]; |
| 35 | unsigned char language[TAG_LANGUAGE_LEN]; |
| 36 | unsigned char platform[TAG_PLATFORM_LEN]; |
| 37 | unsigned char hwv[TAG_HWVER_LEN]; |
| 38 | unsigned char hwv_val[TAG_HWVER_VAL_LEN]; |
| 39 | uint8_t unknown1[24]; |
| 40 | |
| 41 | uint32_t len; |
| 42 | uint32_t crc; |
| 43 | uint32_t base1; |
| 44 | uint32_t base2; |
| 45 | uint32_t data_len; |
| 46 | uint8_t flag; |
| 47 | uint8_t unknown2[3]; |
| 48 | } __attribute ((packed)); |
| 49 | |
| 50 | struct buffalo_tag2 { |
| 51 | unsigned char product[TAG_PRODUCT_LEN]; |
| 52 | unsigned char brand[TAG_BRAND_LEN]; |
| 53 | unsigned char ver_major[TAG_VERSION_LEN]; |
| 54 | unsigned char ver_minor[TAG_VERSION_LEN]; |
| 55 | unsigned char region_code[2]; |
| 56 | uint32_t region_mask; |
| 57 | unsigned char unknown0[2]; |
| 58 | unsigned char language[TAG_LANGUAGE_LEN]; |
| 59 | unsigned char platform[TAG_PLATFORM_LEN]; |
| 60 | unsigned char hwv[TAG_HWVER_LEN]; |
| 61 | unsigned char hwv_val[TAG_HWVER_VAL_LEN]; |
| 62 | uint8_t unknown1[24]; |
| 63 | |
| 64 | uint32_t total_len; |
| 65 | uint32_t crc; |
| 66 | uint32_t len1; |
| 67 | uint32_t len2; |
| 68 | uint8_t flag; |
| 69 | uint8_t unknown2[3]; |
| 70 | } __attribute ((packed)); |
| 71 | |
| 72 | #define ENC_PRODUCT_LEN 32 |
| 73 | #define ENC_VERSION_LEN 8 |
| 74 | #define ENC_MAGIC_LEN 6 |
| 75 | |
| 76 | unsigned long enc_compute_header_len(char *product, char *version); |
| 77 | unsigned long enc_compute_buf_len(char *product, char *version, |
| 78 | unsigned long datalen); |
| 79 | |
| 80 | struct enc_param { |
| 81 | unsigned char *key; |
| 82 | unsigned char magic[ENC_MAGIC_LEN]; |
| 83 | unsigned char product[ENC_PRODUCT_LEN]; |
| 84 | unsigned char version[ENC_VERSION_LEN]; |
| 85 | unsigned char seed; |
| 86 | int longstate; |
| 87 | unsigned datalen; |
| 88 | uint32_t csum; |
| 89 | }; |
| 90 | |
| 91 | int encrypt_buf(struct enc_param *ep, unsigned char *hdr, |
| 92 | unsigned char *data); |
| 93 | int decrypt_buf(struct enc_param *ep, unsigned char *data, |
| 94 | unsigned long datalen); |
| 95 | |
| 96 | #define BCRYPT_DEFAULT_STATE_LEN 256 |
| 97 | #define BCRYPT_MAX_KEYLEN 254 |
| 98 | |
| 99 | struct bcrypt_ctx { |
| 100 | unsigned long i; |
| 101 | unsigned long j; |
| 102 | unsigned char *state; |
| 103 | unsigned long state_len; |
| 104 | }; |
| 105 | |
| 106 | int bcrypt_init(struct bcrypt_ctx *ctx, void *key, int keylen, |
| 107 | unsigned long state_len); |
| 108 | int bcrypt_process(struct bcrypt_ctx *ctx, unsigned char *src, |
| 109 | unsigned char *dst, unsigned long len); |
| 110 | void bcrypt_finish(struct bcrypt_ctx *ctx); |
| 111 | int bcrypt_buf(unsigned char seed, unsigned char *key, unsigned char *src, |
| 112 | unsigned char *dst, unsigned long len, int longstate); |
| 113 | |
| 114 | uint32_t buffalo_csum(uint32_t csum, void *buf, unsigned long len); |
| 115 | uint32_t buffalo_crc(void *buf, unsigned long len); |
| 116 | |
| 117 | ssize_t get_file_size(char *name); |
| 118 | int read_file_to_buf(char *name, void *buf, ssize_t buflen); |
| 119 | int write_buf_to_file(char *name, void *buf, ssize_t buflen); |
| 120 | |
| 121 | #endif /* _BUFFALO_LIB_H */ |
| 122 | |
