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
27struct 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
50struct 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
76unsigned long enc_compute_header_len(char *product, char *version);
77unsigned long enc_compute_buf_len(char *product, char *version,
78				  unsigned long datalen);
79
80struct 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
91int encrypt_buf(struct enc_param *ep, unsigned char *hdr,
92	        unsigned char *data);
93int 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
99struct bcrypt_ctx {
100	unsigned long i;
101	unsigned long j;
102	unsigned char *state;
103	unsigned long state_len;
104};
105
106int bcrypt_init(struct bcrypt_ctx *ctx, void *key, int keylen,
107		unsigned long state_len);
108int bcrypt_process(struct bcrypt_ctx *ctx, unsigned char *src,
109		   unsigned char *dst, unsigned long len);
110void bcrypt_finish(struct bcrypt_ctx *ctx);
111int bcrypt_buf(unsigned char seed, unsigned char *key, unsigned char *src,
112	       unsigned char *dst, unsigned long len, int longstate);
113
114uint32_t buffalo_csum(uint32_t csum, void *buf, unsigned long len);
115uint32_t buffalo_crc(void *buf, unsigned long len);
116
117ssize_t get_file_size(char *name);
118int read_file_to_buf(char *name, void *buf, ssize_t buflen);
119int write_buf_to_file(char *name, void *buf, ssize_t buflen);
120
121#endif /* _BUFFALO_LIB_H */
122