1 | #include "sysincludes.h"
|
---|
2 | #include "msdos.h"
|
---|
3 | #include "stream.h"
|
---|
4 | #include "mtools.h"
|
---|
5 | #include "fsP.h"
|
---|
6 |
|
---|
7 | extern Stream_t *default_drive;
|
---|
8 |
|
---|
9 | #ifdef HAVE_LONG_LONG
|
---|
10 | typedef long long fatBitMask;
|
---|
11 | #else
|
---|
12 | typedef long fatBitMask;
|
---|
13 | #endif
|
---|
14 |
|
---|
15 | typedef struct FatMap_t {
|
---|
16 | unsigned char *data;
|
---|
17 | fatBitMask dirty;
|
---|
18 | fatBitMask valid;
|
---|
19 | } FatMap_t;
|
---|
20 |
|
---|
21 | #define SECT_PER_ENTRY (sizeof(fatBitMask)*8)
|
---|
22 | #define ONE ((fatBitMask) 1)
|
---|
23 |
|
---|
24 | static inline int readSector(Fs_t *This, char *buf, unsigned int off,
|
---|
25 | size_t size)
|
---|
26 | {
|
---|
27 | return READS(This->Next, buf, sectorsToBytes((Stream_t *)This, off),
|
---|
28 | size << This->sectorShift);
|
---|
29 | }
|
---|
30 |
|
---|
31 |
|
---|
32 | static inline int forceReadSector(Fs_t *This, char *buf, unsigned int off,
|
---|
33 | size_t size)
|
---|
34 | {
|
---|
35 | return force_read(This->Next, buf, sectorsToBytes((Stream_t *)This, off),
|
---|
36 | size << This->sectorShift);
|
---|
37 | }
|
---|
38 |
|
---|
39 |
|
---|
40 | static inline int writeSector(Fs_t *This, char *buf, unsigned int off,
|
---|
41 | size_t size)
|
---|
42 | {
|
---|
43 | return WRITES(This->Next, buf, sectorsToBytes((Stream_t*)This, off),
|
---|
44 | size << This->sectorShift);
|
---|
45 | }
|
---|
46 |
|
---|
47 | static inline int forceWriteSector(Fs_t *This, char *buf, unsigned int off,
|
---|
48 | size_t size)
|
---|
49 | {
|
---|
50 | return force_write(This->Next, buf, sectorsToBytes((Stream_t*)This, off),
|
---|
51 | size << This->sectorShift);
|
---|
52 | }
|
---|
53 |
|
---|
54 |
|
---|
55 | static FatMap_t *GetFatMap(Fs_t *Stream)
|
---|
56 | {
|
---|
57 | int nr_entries,i;
|
---|
58 | FatMap_t *map;
|
---|
59 |
|
---|
60 | Stream->fat_error = 0;
|
---|
61 | nr_entries = (Stream->fat_len + SECT_PER_ENTRY - 1) / SECT_PER_ENTRY;
|
---|
62 | map = NewArray(nr_entries, FatMap_t);
|
---|
63 | if(!map)
|
---|
64 | return 0;
|
---|
65 |
|
---|
66 | for(i=0; i< nr_entries; i++) {
|
---|
67 | map[i].data = 0;
|
---|
68 | map[i].valid = 0;
|
---|
69 | map[i].dirty = 0;
|
---|
70 | }
|
---|
71 |
|
---|
72 | return map;
|
---|
73 | }
|
---|
74 |
|
---|
75 | static inline int locate(Fs_t *Stream, int offset, int *slot, int *bit)
|
---|
76 | {
|
---|
77 | if(offset >= Stream->fat_len)
|
---|
78 | return -1;
|
---|
79 | *slot = offset / SECT_PER_ENTRY;
|
---|
80 | *bit = offset % SECT_PER_ENTRY;
|
---|
81 | return 0;
|
---|
82 | }
|
---|
83 |
|
---|
84 | static inline int fatReadSector(Fs_t *This, int sector, int slot,
|
---|
85 | int bit, int dupe)
|
---|
86 | {
|
---|
87 | int fat_start, ret;
|
---|
88 |
|
---|
89 | dupe = (dupe + This->primaryFat) % This->num_fat;
|
---|
90 | fat_start = This->fat_start + This->fat_len * dupe;
|
---|
91 |
|
---|
92 | /* first, read as much as the buffer can give us */
|
---|
93 | ret = readSector(This,
|
---|
94 | (char *)(This->FatMap[slot].data+(bit<<This->sectorShift)),
|
---|
95 | fat_start+sector,
|
---|
96 | (SECT_PER_ENTRY - bit%SECT_PER_ENTRY));
|
---|
97 | if(ret < 0)
|
---|
98 | return 0;
|
---|
99 |
|
---|
100 | if(ret < This->sector_size) {
|
---|
101 | /* if we got less than one sector's worth, insist to get at
|
---|
102 | * least one sector */
|
---|
103 | ret = forceReadSector(This,
|
---|
104 | (char *) (This->FatMap[slot].data +
|
---|
105 | (bit << This->sectorShift)),
|
---|
106 | fat_start+sector, 1);
|
---|
107 | if(ret < This->sector_size)
|
---|
108 | return 0;
|
---|
109 | return 1;
|
---|
110 | }
|
---|
111 |
|
---|
112 | return ret >> This->sectorShift;
|
---|
113 | }
|
---|
114 |
|
---|
115 |
|
---|
116 | static int fatWriteSector(Fs_t *This, int sector, int slot, int bit, int dupe)
|
---|
117 | {
|
---|
118 | int fat_start;
|
---|
119 |
|
---|
120 | dupe = (dupe + This->primaryFat) % This->num_fat;
|
---|
121 | if(dupe && !This->writeAllFats)
|
---|
122 | return This->sector_size;
|
---|
123 |
|
---|
124 | fat_start = This->fat_start + This->fat_len * dupe;
|
---|
125 |
|
---|
126 | return forceWriteSector(This,
|
---|
127 | (char *)
|
---|
128 | (This->FatMap[slot].data + bit * This->sector_size),
|
---|
129 | fat_start+sector, 1);
|
---|
130 | }
|
---|
131 |
|
---|
132 | static unsigned char *loadSector(Fs_t *This,
|
---|
133 | unsigned int sector, fatAccessMode_t mode,
|
---|
134 | int recurs)
|
---|
135 | {
|
---|
136 | int slot, bit, i, ret;
|
---|
137 |
|
---|
138 | if(locate(This,sector, &slot, &bit) < 0)
|
---|
139 | return 0;
|
---|
140 | #if 0
|
---|
141 | if (((This->fat_len + SECT_PER_ENTRY - 1) / SECT_PER_ENTRY) <= slot) {
|
---|
142 | fprintf(stderr,"This should not happen\n");
|
---|
143 | fprintf(stderr, "fat_len = %d\n", This->fat_len);
|
---|
144 | fprintf(stderr, "SECT_PER_ENTRY=%d\n", (int)SECT_PER_ENTRY);
|
---|
145 | fprintf(stderr, "sector = %d slot = %d bit=%d\n",
|
---|
146 | sector, slot, bit);
|
---|
147 | fprintf(stderr, "left = %d",(int)
|
---|
148 | ((This->fat_len+SECT_PER_ENTRY-1) / SECT_PER_ENTRY));
|
---|
149 | return 0;
|
---|
150 | }
|
---|
151 | #endif
|
---|
152 | if(!This->FatMap[slot].data) {
|
---|
153 | /* allocate the storage space */
|
---|
154 | This->FatMap[slot].data =
|
---|
155 | malloc(This->sector_size * SECT_PER_ENTRY);
|
---|
156 | if(!This->FatMap[slot].data)
|
---|
157 | return 0;
|
---|
158 | memset(This->FatMap[slot].data, 0xee,
|
---|
159 | This->sector_size * SECT_PER_ENTRY);
|
---|
160 | }
|
---|
161 |
|
---|
162 | if(! (This->FatMap[slot].valid & (ONE << bit))) {
|
---|
163 | ret = -1;
|
---|
164 | for(i=0; i< This->num_fat; i++) {
|
---|
165 | /* read the sector */
|
---|
166 | ret = fatReadSector(This, sector, slot, bit, i);
|
---|
167 |
|
---|
168 | if(ret == 0) {
|
---|
169 | fprintf(stderr,
|
---|
170 | "Error reading fat number %d\n", i);
|
---|
171 | continue;
|
---|
172 | }
|
---|
173 | break;
|
---|
174 | }
|
---|
175 |
|
---|
176 | /* all copies bad. Return error */
|
---|
177 | if(ret == 0)
|
---|
178 | return 0;
|
---|
179 |
|
---|
180 | for(i=0; i < ret; i++)
|
---|
181 | This->FatMap[slot].valid |= ONE << (bit + i);
|
---|
182 |
|
---|
183 | if(!recurs && ret == 1)
|
---|
184 | /* do some prefetching, if we happened to only
|
---|
185 | * get one sector */
|
---|
186 | loadSector(This, sector+1, mode, 1);
|
---|
187 | if(!recurs && batchmode)
|
---|
188 | for(i=0; i < 1024; i++)
|
---|
189 | loadSector(This, sector+i, mode, 1);
|
---|
190 | }
|
---|
191 |
|
---|
192 | if(mode == FAT_ACCESS_WRITE) {
|
---|
193 | This->FatMap[slot].dirty |= ONE << bit;
|
---|
194 | This->fat_dirty = 1;
|
---|
195 | }
|
---|
196 | return This->FatMap[slot].data + (bit << This->sectorShift);
|
---|
197 | }
|
---|
198 |
|
---|
199 |
|
---|
200 | static unsigned char *getAddress(Fs_t *Stream,
|
---|
201 | unsigned int num, fatAccessMode_t mode)
|
---|
202 | {
|
---|
203 | unsigned char *ret;
|
---|
204 | int sector;
|
---|
205 | int offset;
|
---|
206 |
|
---|
207 | sector = num >> Stream->sectorShift;
|
---|
208 | ret = 0;
|
---|
209 | if(sector == Stream->lastFatSectorNr &&
|
---|
210 | Stream->lastFatAccessMode >= mode)
|
---|
211 | ret = Stream->lastFatSectorData;
|
---|
212 | if(!ret) {
|
---|
213 | ret = loadSector(Stream, sector, mode, 0);
|
---|
214 | if(!ret)
|
---|
215 | return 0;
|
---|
216 | Stream->lastFatSectorNr = sector;
|
---|
217 | Stream->lastFatSectorData = ret;
|
---|
218 | Stream->lastFatAccessMode = mode;
|
---|
219 | }
|
---|
220 | offset = num & Stream->sectorMask;
|
---|
221 | return ret+offset;
|
---|
222 | }
|
---|
223 |
|
---|
224 |
|
---|
225 | static int readByte(Fs_t *Stream, int start)
|
---|
226 | {
|
---|
227 | unsigned char *address;
|
---|
228 |
|
---|
229 | address = getAddress(Stream, start, FAT_ACCESS_READ);
|
---|
230 | if(!address)
|
---|
231 | return -1;
|
---|
232 | return *address;
|
---|
233 | }
|
---|
234 |
|
---|
235 |
|
---|
236 | /*
|
---|
237 | * Fat 12 encoding:
|
---|
238 | * | byte n | byte n+1 | byte n+2 |
|
---|
239 | * |7|6|5|4|3|2|1|0|7|6|5|4|3|2|1|0|7|6|5|4|3|2|1|0|
|
---|
240 | * | | | | | | | | | | | | | | | | | | | | | | | | |
|
---|
241 | * | n+0.0 | n+0.5 | n+1.0 | n+1.5 | n+2.0 | n+2.5 |
|
---|
242 | * \_____ \____ \______/________/_____ /
|
---|
243 | * ____\______\________/ _____/ ____\_/
|
---|
244 | * / \ \ / / \
|
---|
245 | * | n+1.5 | n+0.0 | n+0.5 | n+2.0 | n+2.5 | n+1.0 |
|
---|
246 | * | FAT entry k | FAT entry k+1 |
|
---|
247 | */
|
---|
248 |
|
---|
249 | /*
|
---|
250 | * Get and decode a FAT (file allocation table) entry. Returns the cluster
|
---|
251 | * number on success or 1 on failure.
|
---|
252 | */
|
---|
253 |
|
---|
254 | static unsigned int fat12_decode(Fs_t *Stream, unsigned int num)
|
---|
255 | {
|
---|
256 | unsigned int start = num * 3 / 2;
|
---|
257 | int byte0 = readByte(Stream, start);
|
---|
258 | int byte1 = readByte(Stream, start+1);
|
---|
259 |
|
---|
260 | if (num < 2 || byte0 < 0 || byte1 < 0 || num > Stream->num_clus+1) {
|
---|
261 | fprintf(stderr,"[1] Bad address %d\n", num);
|
---|
262 | return 1;
|
---|
263 | }
|
---|
264 |
|
---|
265 | if (num & 1)
|
---|
266 | return (byte1 << 4) | ((byte0 & 0xf0)>>4);
|
---|
267 | else
|
---|
268 | return ((byte1 & 0xf) << 8) | byte0;
|
---|
269 | }
|
---|
270 |
|
---|
271 |
|
---|
272 | /*
|
---|
273 | * Puts a code into the FAT table. Is the opposite of fat_decode(). No
|
---|
274 | * sanity checking is done on the code. Returns a 1 on error.
|
---|
275 | */
|
---|
276 | static void fat12_encode(Fs_t *Stream, unsigned int num, unsigned int code)
|
---|
277 | {
|
---|
278 | int start = num * 3 / 2;
|
---|
279 | unsigned char *address0 = getAddress(Stream, start, FAT_ACCESS_WRITE);
|
---|
280 | unsigned char *address1 = getAddress(Stream, start+1, FAT_ACCESS_WRITE);
|
---|
281 |
|
---|
282 | if (num & 1) {
|
---|
283 | /* (odd) not on byte boundary */
|
---|
284 | *address0 = (*address0 & 0x0f) | ((code << 4) & 0xf0);
|
---|
285 | *address1 = (code >> 4) & 0xff;
|
---|
286 | } else {
|
---|
287 | /* (even) on byte boundary */
|
---|
288 | *address0 = code & 0xff;
|
---|
289 | *address1 = (*address1 & 0xf0) | ((code >> 8) & 0x0f);
|
---|
290 | }
|
---|
291 | }
|
---|
292 |
|
---|
293 |
|
---|
294 | /*
|
---|
295 | * Fat 16 encoding:
|
---|
296 | * | byte n | byte n+1 |
|
---|
297 | * |7|6|5|4|3|2|1|0|7|6|5|4|3|2|1|0|
|
---|
298 | * | | | | | | | | | | | | | | | | |
|
---|
299 | * | FAT entry k |
|
---|
300 | */
|
---|
301 |
|
---|
302 | static unsigned int fat16_decode(Fs_t *Stream, unsigned int num)
|
---|
303 | {
|
---|
304 | unsigned char *address = getAddress(Stream, num << 1, FAT_ACCESS_READ);
|
---|
305 | return _WORD(address);
|
---|
306 | }
|
---|
307 |
|
---|
308 | static void fat16_encode(Fs_t *Stream, unsigned int num, unsigned int code)
|
---|
309 | {
|
---|
310 | unsigned char *address = getAddress(Stream, num << 1, FAT_ACCESS_WRITE);
|
---|
311 | set_word(address, code);
|
---|
312 | }
|
---|
313 |
|
---|
314 |
|
---|
315 | static unsigned int fast_fat16_decode(Fs_t *Stream, unsigned int num)
|
---|
316 | {
|
---|
317 | unsigned short *address =
|
---|
318 | (unsigned short *) getAddress(Stream, num << 1,
|
---|
319 | FAT_ACCESS_READ);
|
---|
320 | return *address;
|
---|
321 | }
|
---|
322 |
|
---|
323 | static void fast_fat16_encode(Fs_t *Stream, unsigned int num, unsigned int code)
|
---|
324 | {
|
---|
325 | unsigned short *address =
|
---|
326 | (unsigned short *) getAddress(Stream, num << 1,
|
---|
327 | FAT_ACCESS_WRITE);
|
---|
328 | *address = code;
|
---|
329 | }
|
---|
330 |
|
---|
331 |
|
---|
332 |
|
---|
333 |
|
---|
334 | /*
|
---|
335 | * Fat 32 encoding
|
---|
336 | */
|
---|
337 | static unsigned int fat32_decode(Fs_t *Stream, unsigned int num)
|
---|
338 | {
|
---|
339 | unsigned char *address = getAddress(Stream, num << 2, FAT_ACCESS_READ);
|
---|
340 | return _DWORD(address);
|
---|
341 | }
|
---|
342 |
|
---|
343 | static void fat32_encode(Fs_t *Stream, unsigned int num, unsigned int code)
|
---|
344 | {
|
---|
345 | unsigned char *address = getAddress(Stream, num << 2, FAT_ACCESS_WRITE);
|
---|
346 | set_dword(address, code);
|
---|
347 | }
|
---|
348 |
|
---|
349 |
|
---|
350 | static unsigned int fast_fat32_decode(Fs_t *Stream, unsigned int num)
|
---|
351 | {
|
---|
352 | unsigned int *address =
|
---|
353 | (unsigned int *) getAddress(Stream, num << 2,
|
---|
354 | FAT_ACCESS_READ);
|
---|
355 | return *address;
|
---|
356 | }
|
---|
357 |
|
---|
358 | static void fast_fat32_encode(Fs_t *Stream, unsigned int num, unsigned int code)
|
---|
359 | {
|
---|
360 | unsigned int *address =
|
---|
361 | (unsigned int *) getAddress(Stream, num << 2,
|
---|
362 | FAT_ACCESS_WRITE);
|
---|
363 | *address = code;
|
---|
364 | }
|
---|
365 |
|
---|
366 |
|
---|
367 | /*
|
---|
368 | * Write the FAT table to the disk. Up to now the FAT manipulation has
|
---|
369 | * been done in memory. All errors are fatal. (Might not be too smart
|
---|
370 | * to wait till the end of the program to write the table. Oh well...)
|
---|
371 | */
|
---|
372 |
|
---|
373 | void fat_write(Fs_t *This)
|
---|
374 | {
|
---|
375 | int i, j, dups, ret, bit, slot;
|
---|
376 | int fat_start;
|
---|
377 |
|
---|
378 | /*fprintf(stderr, "Fat write\n");*/
|
---|
379 |
|
---|
380 | if (!This->fat_dirty)
|
---|
381 | return;
|
---|
382 |
|
---|
383 | dups = This->num_fat;
|
---|
384 | if (This->fat_error)
|
---|
385 | dups = 1;
|
---|
386 |
|
---|
387 |
|
---|
388 | for(i=0; i<dups; i++){
|
---|
389 | j = 0;
|
---|
390 | fat_start = This->fat_start + i*This->fat_len;
|
---|
391 | for(slot=0;j<This->fat_len;slot++) {
|
---|
392 | if(!This->FatMap[slot].dirty) {
|
---|
393 | j += SECT_PER_ENTRY;
|
---|
394 | continue;
|
---|
395 | }
|
---|
396 | for(bit=0;
|
---|
397 | bit < SECT_PER_ENTRY && j<This->fat_len;
|
---|
398 | bit++,j++) {
|
---|
399 | if(!(This->FatMap[slot].dirty & (ONE << bit)))
|
---|
400 | continue;
|
---|
401 | ret = fatWriteSector(This,j,slot, bit, i);
|
---|
402 | if (ret < This->sector_size){
|
---|
403 | if (ret < 0 ){
|
---|
404 | perror("error in fat_write");
|
---|
405 | exit(1);
|
---|
406 | } else {
|
---|
407 | fprintf(stderr,
|
---|
408 | "end of file in fat_write\n");
|
---|
409 | exit(1);
|
---|
410 | }
|
---|
411 | }
|
---|
412 | /* if last dupe, zero it out */
|
---|
413 | if(i==dups-1)
|
---|
414 | This->FatMap[slot].dirty &= ~(1<<bit);
|
---|
415 | }
|
---|
416 | }
|
---|
417 | }
|
---|
418 | /* write the info sector, if any */
|
---|
419 | if(This->infoSectorLoc && This->infoSectorLoc != MAX32) {
|
---|
420 | /* initialize info sector */
|
---|
421 | InfoSector_t *infoSector;
|
---|
422 | infoSector = (InfoSector_t *) safe_malloc(This->sector_size);
|
---|
423 | set_dword(infoSector->signature1, INFOSECT_SIGNATURE1);
|
---|
424 | memset(infoSector->filler1, sizeof(infoSector->filler1),0);
|
---|
425 | memset(infoSector->filler2, sizeof(infoSector->filler2),0);
|
---|
426 | set_dword(infoSector->signature2, INFOSECT_SIGNATURE2);
|
---|
427 | set_dword(infoSector->pos, This->last);
|
---|
428 | set_dword(infoSector->count, This->freeSpace);
|
---|
429 | set_dword(infoSector->signature3, 0xaa55);
|
---|
430 | if(forceWriteSector(This, (char *)infoSector, This->infoSectorLoc, 1) !=
|
---|
431 | This->sector_size)
|
---|
432 | fprintf(stderr,"Trouble writing the info sector\n");
|
---|
433 | free(infoSector);
|
---|
434 | }
|
---|
435 | This->fat_dirty = 0;
|
---|
436 | This->lastFatAccessMode = FAT_ACCESS_READ;
|
---|
437 | }
|
---|
438 |
|
---|
439 |
|
---|
440 |
|
---|
441 | /*
|
---|
442 | * Zero-Fat
|
---|
443 | * Used by mformat.
|
---|
444 | */
|
---|
445 | int zero_fat(Fs_t *Stream, int media_descriptor)
|
---|
446 | {
|
---|
447 | int i, j;
|
---|
448 | int fat_start;
|
---|
449 | unsigned char *buf;
|
---|
450 |
|
---|
451 | buf = malloc(Stream->sector_size);
|
---|
452 | if(!buf) {
|
---|
453 | perror("alloc fat sector buffer");
|
---|
454 | return -1;
|
---|
455 | }
|
---|
456 | for(i=0; i< Stream->num_fat; i++) {
|
---|
457 | fat_start = Stream->fat_start + i*Stream->fat_len;
|
---|
458 | for(j = 0; j < Stream->fat_len; j++) {
|
---|
459 | if(j <= 1)
|
---|
460 | memset(buf, 0, Stream->sector_size);
|
---|
461 | if(!j) {
|
---|
462 | buf[0] = media_descriptor;
|
---|
463 | buf[2] = buf[1] = 0xff;
|
---|
464 | if(Stream->fat_bits > 12)
|
---|
465 | buf[3] = 0xff;
|
---|
466 | if(Stream->fat_bits > 16) {
|
---|
467 | buf[4] = 0xff;
|
---|
468 | buf[5] = 0xff;
|
---|
469 | buf[6] = 0xff;
|
---|
470 | buf[7] = 0x0f;
|
---|
471 | }
|
---|
472 | }
|
---|
473 |
|
---|
474 | if(forceWriteSector(Stream, (char *)buf,
|
---|
475 | fat_start + j, 1) !=
|
---|
476 | Stream->sector_size) {
|
---|
477 | fprintf(stderr,
|
---|
478 | "Trouble initializing a FAT sector\n");
|
---|
479 | free(buf);
|
---|
480 | return -1;
|
---|
481 | }
|
---|
482 | }
|
---|
483 | }
|
---|
484 |
|
---|
485 | free(buf);
|
---|
486 | Stream->FatMap = GetFatMap(Stream);
|
---|
487 | if (Stream->FatMap == NULL) {
|
---|
488 | perror("alloc fat map");
|
---|
489 | return -1;
|
---|
490 | }
|
---|
491 | return 0;
|
---|
492 | }
|
---|
493 |
|
---|
494 |
|
---|
495 | void set_fat12(Fs_t *This)
|
---|
496 | {
|
---|
497 | This->fat_bits = 12;
|
---|
498 | This->end_fat = 0xfff;
|
---|
499 | This->last_fat = 0xff6;
|
---|
500 | This->fat_decode = fat12_decode;
|
---|
501 | This->fat_encode = fat12_encode;
|
---|
502 | }
|
---|
503 |
|
---|
504 | static char word_endian_test[] = { 0x34, 0x12 };
|
---|
505 |
|
---|
506 | void set_fat16(Fs_t *This)
|
---|
507 | {
|
---|
508 | This->fat_bits = 16;
|
---|
509 | This->end_fat = 0xffff;
|
---|
510 | This->last_fat = 0xfff6;
|
---|
511 |
|
---|
512 | if(sizeof(unsigned short) == 2 &&
|
---|
513 | * (unsigned short *) word_endian_test == 0x1234) {
|
---|
514 | This->fat_decode = fast_fat16_decode;
|
---|
515 | This->fat_encode = fast_fat16_encode;
|
---|
516 | } else {
|
---|
517 | This->fat_decode = fat16_decode;
|
---|
518 | This->fat_encode = fat16_encode;
|
---|
519 | }
|
---|
520 | }
|
---|
521 |
|
---|
522 | static char dword_endian_test[] = { 0x78, 0x56, 0x34, 0x12 };
|
---|
523 |
|
---|
524 | void set_fat32(Fs_t *This)
|
---|
525 | {
|
---|
526 | This->fat_bits = 32;
|
---|
527 | This->end_fat = 0xfffffff;
|
---|
528 | This->last_fat = 0xffffff6;
|
---|
529 |
|
---|
530 | if(sizeof(unsigned int) == 4 &&
|
---|
531 | * (unsigned int *) dword_endian_test == 0x12345678) {
|
---|
532 | This->fat_decode = fast_fat32_decode;
|
---|
533 | This->fat_encode = fast_fat32_encode;
|
---|
534 | } else {
|
---|
535 | This->fat_decode = fat32_decode;
|
---|
536 | This->fat_encode = fat32_encode;
|
---|
537 | }
|
---|
538 | }
|
---|
539 |
|
---|
540 |
|
---|
541 | static int check_fat(Fs_t *This)
|
---|
542 | {
|
---|
543 | /*
|
---|
544 | * This is only a sanity check. For disks with really big FATs,
|
---|
545 | * there is no point in checking the whole FAT.
|
---|
546 | */
|
---|
547 |
|
---|
548 | int i, f, tocheck;
|
---|
549 | if(mtools_skip_check)
|
---|
550 | return 0;
|
---|
551 |
|
---|
552 | /* too few sectors in the FAT */
|
---|
553 | if(This->fat_len < NEEDED_FAT_SIZE(This))
|
---|
554 | return -1;
|
---|
555 | /* we do not warn about too much sectors in FAT, which may
|
---|
556 | * happen when a partition has been shrunk using FIPS, or on
|
---|
557 | * other occurrences */
|
---|
558 |
|
---|
559 | tocheck = This->num_clus;
|
---|
560 | if (tocheck < 0 || tocheck + 1 >= This->last_fat) {
|
---|
561 | fprintf(stderr, "Too many clusters in FAT\n");
|
---|
562 | return -1;
|
---|
563 | }
|
---|
564 |
|
---|
565 | if(tocheck > 4096)
|
---|
566 | tocheck = 4096;
|
---|
567 |
|
---|
568 | for ( i= 3 ; i < tocheck; i++){
|
---|
569 | f = This->fat_decode(This,i);
|
---|
570 | if (f == 1 || (f < This->last_fat && f > This->num_clus)){
|
---|
571 | fprintf(stderr,
|
---|
572 | "Cluster # at %d too big(%#x)\n", i,f);
|
---|
573 | fprintf(stderr,"Probably non MS-DOS disk\n");
|
---|
574 | return -1;
|
---|
575 | }
|
---|
576 | }
|
---|
577 | return 0;
|
---|
578 | }
|
---|
579 |
|
---|
580 |
|
---|
581 | /*
|
---|
582 | * Read the first sector of FAT table into memory. Crude error detection on
|
---|
583 | * wrong FAT encoding scheme.
|
---|
584 | */
|
---|
585 | static int check_media_type(Fs_t *This, struct bootsector *boot,
|
---|
586 | unsigned int tot_sectors)
|
---|
587 | {
|
---|
588 | unsigned char *address;
|
---|
589 |
|
---|
590 | This->num_clus = (tot_sectors - This->clus_start) / This->cluster_size;
|
---|
591 |
|
---|
592 | This->FatMap = GetFatMap(This);
|
---|
593 | if (This->FatMap == NULL) {
|
---|
594 | perror("alloc fat map");
|
---|
595 | return -1;
|
---|
596 | }
|
---|
597 |
|
---|
598 | address = getAddress(This, 0, FAT_ACCESS_READ);
|
---|
599 | if(!address) {
|
---|
600 | fprintf(stderr,
|
---|
601 | "Could not read first FAT sector\n");
|
---|
602 | return -1;
|
---|
603 | }
|
---|
604 |
|
---|
605 | if(mtools_skip_check)
|
---|
606 | return 0;
|
---|
607 |
|
---|
608 | if(!address[0] && !address[1] && !address[2])
|
---|
609 | /* Some Atari disks have zeroes where Dos has media descriptor
|
---|
610 | * and 0xff. Do not consider this as an error */
|
---|
611 | return 0;
|
---|
612 |
|
---|
613 | if((address[0] != boot->descr && boot->descr >= 0xf0 &&
|
---|
614 | ((address[0] != 0xf9 && address[0] != 0xf7)
|
---|
615 | || boot->descr != 0xf0)) || address[0] < 0xf0) {
|
---|
616 | fprintf(stderr,
|
---|
617 | "Bad media types %02x/%02x, probably non-MSDOS disk\n",
|
---|
618 | address[0],
|
---|
619 | boot->descr);
|
---|
620 | return -1;
|
---|
621 | }
|
---|
622 |
|
---|
623 | if(address[1] != 0xff || address[2] != 0xff){
|
---|
624 | fprintf(stderr,"Initial byte of fat is not 0xff\n");
|
---|
625 | return -1;
|
---|
626 | }
|
---|
627 |
|
---|
628 | return 0;
|
---|
629 | }
|
---|
630 |
|
---|
631 | static int fat_32_read(Fs_t *This, struct bootsector *boot,
|
---|
632 | unsigned int tot_sectors)
|
---|
633 | {
|
---|
634 | int size;
|
---|
635 |
|
---|
636 | This->fat_len = DWORD(ext.fat32.bigFat);
|
---|
637 | This->writeAllFats = !(boot->ext.fat32.extFlags[0] & 0x80);
|
---|
638 | This->primaryFat = boot->ext.fat32.extFlags[0] & 0xf;
|
---|
639 | This->rootCluster = DWORD(ext.fat32.rootCluster);
|
---|
640 | This->clus_start = This->fat_start + This->num_fat * This->fat_len;
|
---|
641 |
|
---|
642 | /* read the info sector */
|
---|
643 | size = This->sector_size;
|
---|
644 | This->infoSectorLoc = WORD(ext.fat32.infoSector);
|
---|
645 | if(This->sector_size >= 512 &&
|
---|
646 | This->infoSectorLoc && This->infoSectorLoc != MAX32) {
|
---|
647 | InfoSector_t *infoSector;
|
---|
648 | infoSector = (InfoSector_t *) safe_malloc(size);
|
---|
649 | if(forceReadSector(This, (char *)infoSector,
|
---|
650 | This->infoSectorLoc, 1) == This->sector_size &&
|
---|
651 | _DWORD(infoSector->signature1) == INFOSECT_SIGNATURE1 &&
|
---|
652 | _DWORD(infoSector->signature2) == INFOSECT_SIGNATURE2) {
|
---|
653 | This->freeSpace = _DWORD(infoSector->count);
|
---|
654 | This->last = _DWORD(infoSector->pos);
|
---|
655 | }
|
---|
656 | free(infoSector);
|
---|
657 | }
|
---|
658 |
|
---|
659 | set_fat32(This);
|
---|
660 | return(check_media_type(This,boot, tot_sectors) ||
|
---|
661 | check_fat(This));
|
---|
662 | }
|
---|
663 |
|
---|
664 |
|
---|
665 | static int old_fat_read(Fs_t *This, struct bootsector *boot,
|
---|
666 | int config_fat_bits,
|
---|
667 | size_t tot_sectors, int nodups)
|
---|
668 | {
|
---|
669 | This->writeAllFats = 1;
|
---|
670 | This->primaryFat = 0;
|
---|
671 | This->dir_start = This->fat_start + This->num_fat * This->fat_len;
|
---|
672 | This->clus_start = This->dir_start + This->dir_len;
|
---|
673 | This->infoSectorLoc = MAX32;
|
---|
674 |
|
---|
675 | if(nodups)
|
---|
676 | This->num_fat = 1;
|
---|
677 |
|
---|
678 | if(check_media_type(This,boot, tot_sectors))
|
---|
679 | return -1;
|
---|
680 |
|
---|
681 | if(This->num_clus > FAT12) {
|
---|
682 | set_fat16(This);
|
---|
683 | /* third FAT byte must be 0xff */
|
---|
684 | if(!mtools_skip_check && readByte(This, 3) != 0xff)
|
---|
685 | return -1;
|
---|
686 | } else
|
---|
687 | set_fat12(This);
|
---|
688 |
|
---|
689 | return check_fat(This);
|
---|
690 | }
|
---|
691 |
|
---|
692 | /*
|
---|
693 | * Read the first sector of the FAT table into memory and initialize
|
---|
694 | * structures.
|
---|
695 | */
|
---|
696 | int fat_read(Fs_t *This, struct bootsector *boot, int fat_bits,
|
---|
697 | size_t tot_sectors, int nodups)
|
---|
698 | {
|
---|
699 | This->fat_error = 0;
|
---|
700 | This->fat_dirty = 0;
|
---|
701 | This->last = MAX32;
|
---|
702 | This->freeSpace = MAX32;
|
---|
703 | This->lastFatSectorNr = 0;
|
---|
704 | This->lastFatSectorData = 0;
|
---|
705 |
|
---|
706 | if(This->fat_len)
|
---|
707 | return old_fat_read(This, boot, fat_bits, tot_sectors, nodups);
|
---|
708 | else
|
---|
709 | return fat_32_read(This, boot, tot_sectors);
|
---|
710 | }
|
---|
711 |
|
---|
712 |
|
---|
713 | unsigned int fatDecode(Fs_t *This, unsigned int pos)
|
---|
714 | {
|
---|
715 | int ret;
|
---|
716 |
|
---|
717 | ret = This->fat_decode(This, pos);
|
---|
718 | if(ret && (ret < 2 || ret > This->num_clus+1) && ret < This->last_fat) {
|
---|
719 | fprintf(stderr, "Bad FAT entry %d at %d\n", ret, pos);
|
---|
720 | This->fat_error++;
|
---|
721 | }
|
---|
722 | return ret;
|
---|
723 | }
|
---|
724 |
|
---|
725 | /* append a new cluster */
|
---|
726 | void fatAppend(Fs_t *This, unsigned int pos, unsigned int newpos)
|
---|
727 | {
|
---|
728 | This->fat_encode(This, pos, newpos);
|
---|
729 | This->fat_encode(This, newpos, This->end_fat);
|
---|
730 | if(This->freeSpace != MAX32)
|
---|
731 | This->freeSpace--;
|
---|
732 | }
|
---|
733 |
|
---|
734 | /* de-allocates the given cluster */
|
---|
735 | void fatDeallocate(Fs_t *This, unsigned int pos)
|
---|
736 | {
|
---|
737 | This->fat_encode(This, pos, 0);
|
---|
738 | if(This->freeSpace != MAX32)
|
---|
739 | This->freeSpace++;
|
---|
740 | }
|
---|
741 |
|
---|
742 | /* allocate a new cluster */
|
---|
743 | void fatAllocate(Fs_t *This, unsigned int pos, unsigned int value)
|
---|
744 | {
|
---|
745 | This->fat_encode(This, pos, value);
|
---|
746 | if(This->freeSpace != MAX32)
|
---|
747 | This->freeSpace--;
|
---|
748 | }
|
---|
749 |
|
---|
750 | void fatEncode(Fs_t *This, unsigned int pos, unsigned int value)
|
---|
751 | {
|
---|
752 | unsigned int oldvalue = This->fat_decode(This, pos);
|
---|
753 | This->fat_encode(This, pos, value);
|
---|
754 | if(This->freeSpace != MAX32) {
|
---|
755 | if(oldvalue)
|
---|
756 | This->freeSpace++;
|
---|
757 | if(value)
|
---|
758 | This->freeSpace--;
|
---|
759 | }
|
---|
760 | }
|
---|
761 |
|
---|
762 | unsigned int get_next_free_cluster(Fs_t *This, unsigned int last)
|
---|
763 | {
|
---|
764 | int i;
|
---|
765 |
|
---|
766 | if(This->last != MAX32)
|
---|
767 | last = This->last;
|
---|
768 |
|
---|
769 | if (last < 2 ||
|
---|
770 | last >= This->num_clus+1)
|
---|
771 | last = 1;
|
---|
772 |
|
---|
773 | for (i=last+1; i< This->num_clus+2; i++) {
|
---|
774 | if (!fatDecode(This, i)) {
|
---|
775 | This->last = i;
|
---|
776 | return i;
|
---|
777 | }
|
---|
778 | }
|
---|
779 |
|
---|
780 | for(i=2; i < last+1; i++) {
|
---|
781 | if (!fatDecode(This, i)) {
|
---|
782 | This->last = i;
|
---|
783 | return i;
|
---|
784 | }
|
---|
785 | }
|
---|
786 |
|
---|
787 |
|
---|
788 | fprintf(stderr,"No free cluster %d %d\n", This->preallocatedClusters,
|
---|
789 | This->last);
|
---|
790 | return 1;
|
---|
791 | }
|
---|
792 |
|
---|
793 | int fat_error(Stream_t *Dir)
|
---|
794 | {
|
---|
795 | Stream_t *Stream = GetFs(Dir);
|
---|
796 | DeclareThis(Fs_t);
|
---|
797 |
|
---|
798 | if(This->fat_error)
|
---|
799 | fprintf(stderr,"Fat error detected\n");
|
---|
800 |
|
---|
801 | return This->fat_error;
|
---|
802 | }
|
---|
803 |
|
---|
804 | int fat32RootCluster(Stream_t *Dir)
|
---|
805 | {
|
---|
806 | Stream_t *Stream = GetFs(Dir);
|
---|
807 | DeclareThis(Fs_t);
|
---|
808 |
|
---|
809 | if(This->fat_bits == 32)
|
---|
810 | return This->rootCluster;
|
---|
811 | else
|
---|
812 | return 0;
|
---|
813 | }
|
---|
814 |
|
---|
815 |
|
---|
816 | /*
|
---|
817 | * Get the amount of free space on the diskette
|
---|
818 | */
|
---|
819 |
|
---|
820 | mt_size_t getfree(Stream_t *Dir)
|
---|
821 | {
|
---|
822 | Stream_t *Stream = GetFs(Dir);
|
---|
823 | DeclareThis(Fs_t);
|
---|
824 |
|
---|
825 | if(This->freeSpace == MAX32 || This->freeSpace == 0) {
|
---|
826 | register unsigned int i;
|
---|
827 | size_t total;
|
---|
828 |
|
---|
829 | total = 0L;
|
---|
830 | for (i = 2; i < This->num_clus + 2; i++)
|
---|
831 | if (!fatDecode(This,i))
|
---|
832 | total++;
|
---|
833 | This->freeSpace = total;
|
---|
834 | }
|
---|
835 | return sectorsToBytes((Stream_t*)This,
|
---|
836 | This->freeSpace * This->cluster_size);
|
---|
837 | }
|
---|
838 |
|
---|
839 |
|
---|
840 | /*
|
---|
841 | * Ensure that there is a minimum of total sectors free
|
---|
842 | */
|
---|
843 | int getfreeMinClusters(Stream_t *Dir, size_t size)
|
---|
844 | {
|
---|
845 | Stream_t *Stream = GetFs(Dir);
|
---|
846 | DeclareThis(Fs_t);
|
---|
847 | register unsigned int i, last;
|
---|
848 | size_t total;
|
---|
849 |
|
---|
850 | if(batchmode && This->freeSpace == MAX32)
|
---|
851 | getfree(Stream);
|
---|
852 |
|
---|
853 | if(This->freeSpace != MAX32) {
|
---|
854 | if(This->freeSpace >= size)
|
---|
855 | return 1;
|
---|
856 | else {
|
---|
857 | fprintf(stderr, "Disk full\n");
|
---|
858 | got_signal = 1;
|
---|
859 | return 0;
|
---|
860 | }
|
---|
861 | }
|
---|
862 |
|
---|
863 | total = 0L;
|
---|
864 |
|
---|
865 | /* we start at the same place where we'll start later to actually
|
---|
866 | * allocate the sectors. That way, the same sectors of the FAT, which
|
---|
867 | * are already loaded during getfreeMin will be able to be reused
|
---|
868 | * during get_next_free_cluster */
|
---|
869 | last = This->last;
|
---|
870 |
|
---|
871 | if ( last < 2 || last >= This->num_clus + 2)
|
---|
872 | last = 1;
|
---|
873 | for (i=last+1; i< This->num_clus+2; i++){
|
---|
874 | if (!fatDecode(This, i))
|
---|
875 | total++;
|
---|
876 | if(total >= size)
|
---|
877 | return 1;
|
---|
878 | }
|
---|
879 | for(i=2; i < last+1; i++){
|
---|
880 | if (!fatDecode(This, i))
|
---|
881 | total++;
|
---|
882 | if(total >= size)
|
---|
883 | return 1;
|
---|
884 | }
|
---|
885 | fprintf(stderr, "Disk full\n");
|
---|
886 | got_signal = 1;
|
---|
887 | return 0;
|
---|
888 | }
|
---|
889 |
|
---|
890 |
|
---|
891 | int getfreeMinBytes(Stream_t *Dir, mt_size_t size)
|
---|
892 | {
|
---|
893 | Stream_t *Stream = GetFs(Dir);
|
---|
894 | DeclareThis(Fs_t);
|
---|
895 | size_t size2;
|
---|
896 |
|
---|
897 | size2 = size / (This->sector_size * This->cluster_size);
|
---|
898 | if(size % (This->sector_size * This->cluster_size))
|
---|
899 | size2++;
|
---|
900 | return getfreeMinClusters(Dir, size2);
|
---|
901 | }
|
---|
902 |
|
---|
903 |
|
---|
904 | unsigned int getStart(Stream_t *Dir, struct directory *dir)
|
---|
905 | {
|
---|
906 | Stream_t *Stream = GetFs(Dir);
|
---|
907 | unsigned int first;
|
---|
908 |
|
---|
909 | first = START(dir);
|
---|
910 | if(fat32RootCluster(Stream))
|
---|
911 | first |= STARTHI(dir) << 16;
|
---|
912 | return first;
|
---|
913 | }
|
---|
914 |
|
---|
915 | int fs_free(Stream_t *Stream)
|
---|
916 | {
|
---|
917 | DeclareThis(Fs_t);
|
---|
918 |
|
---|
919 | if(This->FatMap) {
|
---|
920 | int i, nr_entries;
|
---|
921 | nr_entries = (This->fat_len + SECT_PER_ENTRY - 1) /
|
---|
922 | SECT_PER_ENTRY;
|
---|
923 | for(i=0; i< nr_entries; i++)
|
---|
924 | if(This->FatMap[i].data)
|
---|
925 | free(This->FatMap[i].data);
|
---|
926 | free(This->FatMap);
|
---|
927 | }
|
---|
928 | return 0;
|
---|
929 | }
|
---|