diff options
Diffstat (limited to 'rbutil/jztool/src/identify_file.c')
-rw-r--r-- | rbutil/jztool/src/identify_file.c | 179 |
1 files changed, 179 insertions, 0 deletions
diff --git a/rbutil/jztool/src/identify_file.c b/rbutil/jztool/src/identify_file.c new file mode 100644 index 0000000000..3bf4a9ced7 --- /dev/null +++ b/rbutil/jztool/src/identify_file.c | |||
@@ -0,0 +1,179 @@ | |||
1 | /*************************************************************************** | ||
2 | * __________ __ ___. | ||
3 | * Open \______ \ ____ ____ | | _\_ |__ _______ ___ | ||
4 | * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / | ||
5 | * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < | ||
6 | * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ | ||
7 | * \/ \/ \/ \/ \/ | ||
8 | * $Id$ | ||
9 | * | ||
10 | * Copyright (C) 2021 Aidan MacDonald | ||
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 | |||
22 | #include "jztool.h" | ||
23 | #include <string.h> | ||
24 | |||
25 | /* Following is copied from mkspl-x1000, basically */ | ||
26 | struct x1000_spl_header { | ||
27 | uint8_t magic[8]; | ||
28 | uint8_t type; | ||
29 | uint8_t crc7; | ||
30 | uint8_t ppb; | ||
31 | uint8_t bpp; | ||
32 | uint32_t length; | ||
33 | }; | ||
34 | |||
35 | static const uint8_t x1000_spl_header_magic[8] = | ||
36 | {0x06, 0x05, 0x04, 0x03, 0x02, 0x55, 0xaa, 0x55}; | ||
37 | |||
38 | static const size_t X1000_SPL_HEADER_SIZE = 2 * 1024; | ||
39 | |||
40 | static uint8_t crc7(const uint8_t* buf, size_t len) | ||
41 | { | ||
42 | /* table-based computation of CRC7 */ | ||
43 | static const uint8_t t[256] = { | ||
44 | 0x00, 0x09, 0x12, 0x1b, 0x24, 0x2d, 0x36, 0x3f, | ||
45 | 0x48, 0x41, 0x5a, 0x53, 0x6c, 0x65, 0x7e, 0x77, | ||
46 | 0x19, 0x10, 0x0b, 0x02, 0x3d, 0x34, 0x2f, 0x26, | ||
47 | 0x51, 0x58, 0x43, 0x4a, 0x75, 0x7c, 0x67, 0x6e, | ||
48 | 0x32, 0x3b, 0x20, 0x29, 0x16, 0x1f, 0x04, 0x0d, | ||
49 | 0x7a, 0x73, 0x68, 0x61, 0x5e, 0x57, 0x4c, 0x45, | ||
50 | 0x2b, 0x22, 0x39, 0x30, 0x0f, 0x06, 0x1d, 0x14, | ||
51 | 0x63, 0x6a, 0x71, 0x78, 0x47, 0x4e, 0x55, 0x5c, | ||
52 | 0x64, 0x6d, 0x76, 0x7f, 0x40, 0x49, 0x52, 0x5b, | ||
53 | 0x2c, 0x25, 0x3e, 0x37, 0x08, 0x01, 0x1a, 0x13, | ||
54 | 0x7d, 0x74, 0x6f, 0x66, 0x59, 0x50, 0x4b, 0x42, | ||
55 | 0x35, 0x3c, 0x27, 0x2e, 0x11, 0x18, 0x03, 0x0a, | ||
56 | 0x56, 0x5f, 0x44, 0x4d, 0x72, 0x7b, 0x60, 0x69, | ||
57 | 0x1e, 0x17, 0x0c, 0x05, 0x3a, 0x33, 0x28, 0x21, | ||
58 | 0x4f, 0x46, 0x5d, 0x54, 0x6b, 0x62, 0x79, 0x70, | ||
59 | 0x07, 0x0e, 0x15, 0x1c, 0x23, 0x2a, 0x31, 0x38, | ||
60 | 0x41, 0x48, 0x53, 0x5a, 0x65, 0x6c, 0x77, 0x7e, | ||
61 | 0x09, 0x00, 0x1b, 0x12, 0x2d, 0x24, 0x3f, 0x36, | ||
62 | 0x58, 0x51, 0x4a, 0x43, 0x7c, 0x75, 0x6e, 0x67, | ||
63 | 0x10, 0x19, 0x02, 0x0b, 0x34, 0x3d, 0x26, 0x2f, | ||
64 | 0x73, 0x7a, 0x61, 0x68, 0x57, 0x5e, 0x45, 0x4c, | ||
65 | 0x3b, 0x32, 0x29, 0x20, 0x1f, 0x16, 0x0d, 0x04, | ||
66 | 0x6a, 0x63, 0x78, 0x71, 0x4e, 0x47, 0x5c, 0x55, | ||
67 | 0x22, 0x2b, 0x30, 0x39, 0x06, 0x0f, 0x14, 0x1d, | ||
68 | 0x25, 0x2c, 0x37, 0x3e, 0x01, 0x08, 0x13, 0x1a, | ||
69 | 0x6d, 0x64, 0x7f, 0x76, 0x49, 0x40, 0x5b, 0x52, | ||
70 | 0x3c, 0x35, 0x2e, 0x27, 0x18, 0x11, 0x0a, 0x03, | ||
71 | 0x74, 0x7d, 0x66, 0x6f, 0x50, 0x59, 0x42, 0x4b, | ||
72 | 0x17, 0x1e, 0x05, 0x0c, 0x33, 0x3a, 0x21, 0x28, | ||
73 | 0x5f, 0x56, 0x4d, 0x44, 0x7b, 0x72, 0x69, 0x60, | ||
74 | 0x0e, 0x07, 0x1c, 0x15, 0x2a, 0x23, 0x38, 0x31, | ||
75 | 0x46, 0x4f, 0x54, 0x5d, 0x62, 0x6b, 0x70, 0x79 | ||
76 | }; | ||
77 | |||
78 | uint8_t crc = 0; | ||
79 | while(len--) | ||
80 | crc = t[(crc << 1) ^ *buf++]; | ||
81 | return crc; | ||
82 | } | ||
83 | |||
84 | int jz_identify_x1000_spl(const void* data, size_t len) | ||
85 | { | ||
86 | /* Use <= check because a header-only file is not really valid, | ||
87 | * it should have at least one byte in it... */ | ||
88 | if(len <= X1000_SPL_HEADER_SIZE) | ||
89 | return JZ_IDERR_WRONG_SIZE; | ||
90 | |||
91 | /* Look for header magic bytes */ | ||
92 | const struct x1000_spl_header* header = (const struct x1000_spl_header*)data; | ||
93 | if(memcmp(header->magic, x1000_spl_header_magic, 8)) | ||
94 | return JZ_IDERR_BAD_HEADER; | ||
95 | |||
96 | /* Length stored in the header should equal the length of the file */ | ||
97 | if(header->length != len) | ||
98 | return JZ_IDERR_WRONG_SIZE; | ||
99 | |||
100 | /* Compute the CRC7 checksum; it only covers the SPL code */ | ||
101 | const uint8_t* dat = (const uint8_t*)data; | ||
102 | uint8_t sum = crc7(&dat[X1000_SPL_HEADER_SIZE], len - X1000_SPL_HEADER_SIZE); | ||
103 | if(header->crc7 != sum) | ||
104 | return JZ_IDERR_BAD_CHECKSUM; | ||
105 | |||
106 | return JZ_SUCCESS; | ||
107 | |||
108 | } | ||
109 | |||
110 | static const struct scramble_model_info { | ||
111 | const char* name; | ||
112 | int model_num; | ||
113 | } scramble_models[] = { | ||
114 | {"fiio", 114}, | ||
115 | {NULL, 0}, | ||
116 | }; | ||
117 | |||
118 | int jz_identify_scramble_image(const void* data, size_t len) | ||
119 | { | ||
120 | /* 4 bytes checksum + 4 bytes player model */ | ||
121 | if(len < 8) | ||
122 | return JZ_IDERR_WRONG_SIZE; | ||
123 | |||
124 | /* Look up the model number */ | ||
125 | const uint8_t* dat = (const uint8_t*)data; | ||
126 | const struct scramble_model_info* model_info = &scramble_models[0]; | ||
127 | for(; model_info->name != NULL; ++model_info) | ||
128 | if(!memcmp(&dat[4], model_info->name, 4)) | ||
129 | break; | ||
130 | |||
131 | if(model_info->name == NULL) | ||
132 | return JZ_IDERR_UNRECOGNIZED_MODEL; | ||
133 | |||
134 | /* Compute the checksum */ | ||
135 | uint32_t sum = model_info->model_num; | ||
136 | for(size_t i = 8; i < len; ++i) | ||
137 | sum += dat[i]; | ||
138 | |||
139 | /* Compare with file's checksum, it's stored in big-endian form */ | ||
140 | uint32_t fsum = (dat[0] << 24) | (dat[1] << 16) | (dat[2] << 8) | dat[3]; | ||
141 | if(sum != fsum) | ||
142 | return JZ_IDERR_BAD_CHECKSUM; | ||
143 | |||
144 | return JZ_SUCCESS; | ||
145 | } | ||
146 | |||
147 | int jz_identify_fiiom3k_bootimage(const void* data, size_t len) | ||
148 | { | ||
149 | /* The bootloader image is simply a dump of the first NAND eraseblock, | ||
150 | * so it has a fixed 128 KiB size */ | ||
151 | if(len != 128*1024) | ||
152 | return JZ_IDERR_WRONG_SIZE; | ||
153 | |||
154 | /* We'll verify the embedded SPL, but we have to drag out the correct | ||
155 | * length from the header. Length should be more than 12 KiB, due to | ||
156 | * limitations of the hardware */ | ||
157 | const struct x1000_spl_header* spl_header; | ||
158 | spl_header = (const struct x1000_spl_header*)data; | ||
159 | if(spl_header->length > 12 * 1024) | ||
160 | return JZ_IDERR_BAD_HEADER; | ||
161 | |||
162 | int rc = jz_identify_x1000_spl(data, spl_header->length); | ||
163 | if(rc < 0) | ||
164 | return rc; | ||
165 | |||
166 | const uint8_t* dat = (const uint8_t*)data; | ||
167 | |||
168 | /* Check the partition table is present */ | ||
169 | if(memcmp(&dat[0x3c00], "nand", 4)) | ||
170 | return JZ_IDERR_OTHER; | ||
171 | |||
172 | /* Check first bytes of PDMA firmware. It doesn't change | ||
173 | * between OF versions, and Rockbox doesn't modify it. */ | ||
174 | static const uint8_t pdma_fw[] = {0x54, 0x25, 0x42, 0xb3, 0x70, 0x25, 0x42, 0xb3}; | ||
175 | if(memcmp(&dat[0x4000], pdma_fw, sizeof(pdma_fw))) | ||
176 | return JZ_IDERR_OTHER; | ||
177 | |||
178 | return JZ_SUCCESS; | ||
179 | } | ||