1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21#include "volume_id_internal.h"
22
23
24#define FAT12_MAX 0xff4
25#define FAT16_MAX 0xfff4
26#define FAT32_MAX 0x0ffffff6
27
28#define FAT_ATTR_VOLUME_ID 0x08
29#define FAT_ATTR_DIR 0x10
30#define FAT_ATTR_LONG_NAME 0x0f
31#define FAT_ATTR_MASK 0x3f
32#define FAT_ENTRY_FREE 0xe5
33
34struct vfat_super_block {
35 uint8_t boot_jump[3];
36 uint8_t sysid[8];
37 uint16_t sector_size_bytes;
38 uint8_t sectors_per_cluster;
39 uint16_t reserved_sct;
40 uint8_t fats;
41 uint16_t dir_entries;
42 uint16_t sectors;
43 uint8_t media;
44 uint16_t fat_length;
45 uint16_t secs_track;
46 uint16_t heads;
47 uint32_t hidden;
48 uint32_t total_sect;
49 union {
50 struct fat_super_block {
51 uint8_t unknown[3];
52 uint8_t serno[4];
53 uint8_t label[11];
54 uint8_t magic[8];
55 uint8_t dummy2[192];
56 uint8_t pmagic[2];
57 } __attribute__((__packed__)) fat;
58 struct fat32_super_block {
59 uint32_t fat32_length;
60 uint16_t flags;
61 uint8_t version[2];
62 uint32_t root_cluster;
63 uint16_t insfo_sector;
64 uint16_t backup_boot;
65 uint16_t reserved2[6];
66 uint8_t unknown[3];
67 uint8_t serno[4];
68 uint8_t label[11];
69 uint8_t magic[8];
70 uint8_t dummy2[164];
71 uint8_t pmagic[2];
72 } __attribute__((__packed__)) fat32;
73 } __attribute__((__packed__)) type;
74} __attribute__((__packed__));
75
76struct vfat_dir_entry {
77 uint8_t name[11];
78 uint8_t attr;
79 uint16_t time_creat;
80 uint16_t date_creat;
81 uint16_t time_acc;
82 uint16_t date_acc;
83 uint16_t cluster_high;
84 uint16_t time_write;
85 uint16_t date_write;
86 uint16_t cluster_low;
87 uint32_t size;
88} __attribute__((__packed__));
89
90static uint8_t *get_attr_volume_id(struct vfat_dir_entry *dir, int count)
91{
92 for (;--count >= 0; dir++) {
93
94 if (dir->name[0] == 0x00) {
95 dbg("end of dir");
96 break;
97 }
98
99
100 if (dir->name[0] == FAT_ENTRY_FREE)
101 continue;
102
103
104 if ((dir->attr & FAT_ATTR_MASK) == FAT_ATTR_LONG_NAME)
105 continue;
106
107 if ((dir->attr & (FAT_ATTR_VOLUME_ID | FAT_ATTR_DIR)) == FAT_ATTR_VOLUME_ID) {
108
109 if (dir->cluster_high != 0 || dir->cluster_low != 0)
110 continue;
111
112 dbg("found ATTR_VOLUME_ID id in root dir");
113 return dir->name;
114 }
115
116 dbg("skip dir entry");
117 }
118
119 return NULL;
120}
121
122int volume_id_probe_vfat(struct volume_id *id )
123{
124#define fat_partition_off ((uint64_t)0)
125 struct vfat_super_block *vs;
126 struct vfat_dir_entry *dir;
127 uint16_t sector_size_bytes;
128 uint16_t dir_entries;
129 uint32_t sect_count;
130 uint16_t reserved_sct;
131 uint32_t fat_size_sct;
132 uint32_t root_cluster;
133 uint32_t dir_size_sct;
134 uint32_t cluster_count;
135 uint64_t root_start_off;
136 uint32_t start_data_sct;
137 uint8_t *buf;
138 uint32_t buf_size;
139 uint8_t *label = NULL;
140 uint32_t next_cluster;
141 int maxloop;
142
143 dbg("probing at offset 0x%llx", (unsigned long long) fat_partition_off);
144
145 vs = volume_id_get_buffer(id, fat_partition_off, 0x200);
146 if (vs == NULL)
147 return -1;
148
149
150
151
152 if (memcmp(vs->sysid, "NTFS", 4) == 0)
153 return -1;
154
155 if (memcmp(vs->type.fat32.magic, "MSWIN", 5) == 0)
156 goto valid;
157
158 if (memcmp(vs->type.fat32.magic, "FAT32 ", 8) == 0)
159 goto valid;
160
161 if (memcmp(vs->type.fat.magic, "FAT16 ", 8) == 0)
162 goto valid;
163
164 if (memcmp(vs->type.fat.magic, "MSDOS", 5) == 0)
165 goto valid;
166
167 if (memcmp(vs->type.fat.magic, "FAT12 ", 8) == 0)
168 goto valid;
169
170
171
172
173
174
175
176 if ((vs->boot_jump[0] != 0xeb || vs->boot_jump[2] != 0x90) &&
177 vs->boot_jump[0] != 0xe9)
178 return -1;
179
180
181 if (vs->heads == 0)
182 return -1;
183
184
185 if (vs->sectors_per_cluster == 0 ||
186 (vs->sectors_per_cluster & (vs->sectors_per_cluster-1)))
187 return -1;
188
189
190 if (vs->media < 0xf8 && vs->media != 0xf0)
191 return -1;
192
193
194 if (vs->fats != 2)
195 return -1;
196
197 valid:
198
199 sector_size_bytes = le16_to_cpu(vs->sector_size_bytes);
200 if (sector_size_bytes != 0x200 && sector_size_bytes != 0x400 &&
201 sector_size_bytes != 0x800 && sector_size_bytes != 0x1000)
202 return -1;
203
204 dbg("sector_size_bytes 0x%x", sector_size_bytes);
205 dbg("sectors_per_cluster 0x%x", vs->sectors_per_cluster);
206
207 reserved_sct = le16_to_cpu(vs->reserved_sct);
208 dbg("reserved_sct 0x%x", reserved_sct);
209
210 sect_count = le16_to_cpu(vs->sectors);
211 if (sect_count == 0)
212 sect_count = le32_to_cpu(vs->total_sect);
213 dbg("sect_count 0x%x", sect_count);
214
215 fat_size_sct = le16_to_cpu(vs->fat_length);
216 if (fat_size_sct == 0)
217 fat_size_sct = le32_to_cpu(vs->type.fat32.fat32_length);
218 fat_size_sct *= vs->fats;
219 dbg("fat_size_sct 0x%x", fat_size_sct);
220
221 dir_entries = le16_to_cpu(vs->dir_entries);
222 dir_size_sct = ((dir_entries * sizeof(struct vfat_dir_entry)) +
223 (sector_size_bytes-1)) / sector_size_bytes;
224 dbg("dir_size_sct 0x%x", dir_size_sct);
225
226 cluster_count = sect_count - (reserved_sct + fat_size_sct + dir_size_sct);
227 cluster_count /= vs->sectors_per_cluster;
228 dbg("cluster_count 0x%x", cluster_count);
229
230
231
232
233
234
235
236
237
238 if (cluster_count >= FAT16_MAX)
239 goto fat32;
240
241
242 root_start_off = (reserved_sct + fat_size_sct) * sector_size_bytes;
243 dbg("root dir start 0x%llx", (unsigned long long) root_start_off);
244 dbg("expected entries 0x%x", dir_entries);
245
246 buf_size = dir_entries * sizeof(struct vfat_dir_entry);
247 buf = volume_id_get_buffer(id, fat_partition_off + root_start_off, buf_size);
248 if (buf == NULL)
249 goto ret;
250
251 label = get_attr_volume_id((struct vfat_dir_entry*) buf, dir_entries);
252
253 vs = volume_id_get_buffer(id, fat_partition_off, 0x200);
254 if (vs == NULL)
255 return -1;
256
257 if (label != NULL && memcmp(label, "NO NAME ", 11) != 0) {
258
259 volume_id_set_label_string(id, label, 11);
260 } else if (memcmp(vs->type.fat.label, "NO NAME ", 11) != 0) {
261
262 volume_id_set_label_string(id, vs->type.fat.label, 11);
263 }
264 volume_id_set_uuid(id, vs->type.fat.serno, UUID_DOS);
265 goto ret;
266
267 fat32:
268
269 buf_size = vs->sectors_per_cluster * sector_size_bytes;
270 root_cluster = le32_to_cpu(vs->type.fat32.root_cluster);
271 start_data_sct = reserved_sct + fat_size_sct;
272
273 next_cluster = root_cluster;
274 maxloop = 100;
275 while (--maxloop) {
276 uint64_t next_off_sct;
277 uint64_t next_off;
278 uint64_t fat_entry_off;
279 int count;
280
281 dbg("next_cluster 0x%x", (unsigned)next_cluster);
282 next_off_sct = (uint64_t)(next_cluster - 2) * vs->sectors_per_cluster;
283 next_off = (start_data_sct + next_off_sct) * sector_size_bytes;
284 dbg("cluster offset 0x%llx", (unsigned long long) next_off);
285
286
287 buf = volume_id_get_buffer(id, fat_partition_off + next_off, buf_size);
288 if (buf == NULL)
289 goto ret;
290
291 dir = (struct vfat_dir_entry*) buf;
292 count = buf_size / sizeof(struct vfat_dir_entry);
293 dbg("expected entries 0x%x", count);
294
295 label = get_attr_volume_id(dir, count);
296 if (label)
297 break;
298
299
300 fat_entry_off = (reserved_sct * sector_size_bytes) + (next_cluster * sizeof(uint32_t));
301 dbg("fat_entry_off 0x%llx", (unsigned long long)fat_entry_off);
302 buf = volume_id_get_buffer(id, fat_partition_off + fat_entry_off, buf_size);
303 if (buf == NULL)
304 goto ret;
305
306
307 next_cluster = le32_to_cpu(*(uint32_t*)buf) & 0x0fffffff;
308 if (next_cluster < 2 || next_cluster > FAT32_MAX)
309 break;
310 }
311 if (maxloop == 0)
312 dbg("reached maximum follow count of root cluster chain, give up");
313
314 vs = volume_id_get_buffer(id, fat_partition_off, 0x200);
315 if (vs == NULL)
316 return -1;
317
318 if (label != NULL && memcmp(label, "NO NAME ", 11) != 0) {
319
320 volume_id_set_label_string(id, label, 11);
321 } else if (memcmp(vs->type.fat32.label, "NO NAME ", 11) != 0) {
322
323 volume_id_set_label_string(id, vs->type.fat32.label, 11);
324 }
325 volume_id_set_uuid(id, vs->type.fat32.serno, UUID_DOS);
326
327 ret:
328
329
330
331 return 0;
332}
333