]> git.proxmox.com Git - qemu.git/blame - block/vvfat.c
Merge remote branch 'mst/for_anthony' into staging
[qemu.git] / block / vvfat.c
CommitLineData
a046433a 1/* vim:set shiftwidth=4 ts=8: */
de167e41
FB
2/*
3 * QEMU Block driver for virtual VFAT (shadows a local directory)
5fafdf24 4 *
a046433a 5 * Copyright (c) 2004,2005 Johannes E. Schindelin
5fafdf24 6 *
de167e41
FB
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
13 *
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23 * THE SOFTWARE.
24 */
25#include <sys/stat.h>
26#include <dirent.h>
faf07963 27#include "qemu-common.h"
de167e41 28#include "block_int.h"
5efa9d5a 29#include "module.h"
de167e41 30
a046433a
FB
31#ifndef S_IWGRP
32#define S_IWGRP 0
33#endif
34#ifndef S_IWOTH
35#define S_IWOTH 0
36#endif
37
38/* TODO: add ":bootsector=blabla.img:" */
39/* LATER TODO: add automatic boot sector generation from
40 BOOTEASY.ASM and Ranish Partition Manager
5fafdf24 41 Note that DOS assumes the system files to be the first files in the
a046433a
FB
42 file system (test if the boot sector still relies on that fact)! */
43/* MAYBE TODO: write block-visofs.c */
44/* TODO: call try_commit() only after a timeout */
45
46/* #define DEBUG */
47
48#ifdef DEBUG
49
50#define DLOG(a) a
51
52#undef stderr
53#define stderr STDERR
54FILE* stderr = NULL;
de167e41 55
3f47aa8c 56static void checkpoint(void);
de167e41 57
a046433a
FB
58#ifdef __MINGW32__
59void nonono(const char* file, int line, const char* msg) {
60 fprintf(stderr, "Nonono! %s:%d %s\n", file, line, msg);
61 exit(-5);
62}
63#undef assert
6bcb76c3 64#define assert(a) do {if (!(a)) nonono(__FILE__, __LINE__, #a);}while(0)
a046433a
FB
65#endif
66
67#else
68
69#define DLOG(a)
70
71#endif
de167e41
FB
72
73/* dynamic array functions */
c227f099 74typedef struct array_t {
de167e41
FB
75 char* pointer;
76 unsigned int size,next,item_size;
c227f099 77} array_t;
de167e41 78
c227f099 79static inline void array_init(array_t* array,unsigned int item_size)
de167e41 80{
511d2b14 81 array->pointer = NULL;
de167e41
FB
82 array->size=0;
83 array->next=0;
84 array->item_size=item_size;
85}
86
c227f099 87static inline void array_free(array_t* array)
de167e41
FB
88{
89 if(array->pointer)
90 free(array->pointer);
91 array->size=array->next=0;
92}
93
a046433a 94/* does not automatically grow */
c227f099 95static inline void* array_get(array_t* array,unsigned int index) {
a046433a
FB
96 assert(index < array->next);
97 return array->pointer + index * array->item_size;
98}
99
c227f099 100static inline int array_ensure_allocated(array_t* array, int index)
a046433a
FB
101{
102 if((index + 1) * array->item_size > array->size) {
103 int new_size = (index + 32) * array->item_size;
2137b4cc 104 array->pointer = qemu_realloc(array->pointer, new_size);
a046433a
FB
105 if (!array->pointer)
106 return -1;
107 array->size = new_size;
108 array->next = index + 1;
de167e41 109 }
a046433a
FB
110
111 return 0;
de167e41
FB
112}
113
c227f099 114static inline void* array_get_next(array_t* array) {
a046433a
FB
115 unsigned int next = array->next;
116 void* result;
117
118 if (array_ensure_allocated(array, next) < 0)
119 return NULL;
120
121 array->next = next + 1;
122 result = array_get(array, next);
123
de167e41
FB
124 return result;
125}
126
c227f099 127static inline void* array_insert(array_t* array,unsigned int index,unsigned int count) {
de167e41
FB
128 if((array->next+count)*array->item_size>array->size) {
129 int increment=count*array->item_size;
2137b4cc 130 array->pointer=qemu_realloc(array->pointer,array->size+increment);
de167e41 131 if(!array->pointer)
511d2b14 132 return NULL;
de167e41
FB
133 array->size+=increment;
134 }
135 memmove(array->pointer+(index+count)*array->item_size,
136 array->pointer+index*array->item_size,
137 (array->next-index)*array->item_size);
138 array->next+=count;
139 return array->pointer+index*array->item_size;
140}
141
142/* this performs a "roll", so that the element which was at index_from becomes
143 * index_to, but the order of all other elements is preserved. */
c227f099 144static inline int array_roll(array_t* array,int index_to,int index_from,int count)
de167e41
FB
145{
146 char* buf;
147 char* from;
148 char* to;
149 int is;
150
151 if(!array ||
152 index_to<0 || index_to>=array->next ||
153 index_from<0 || index_from>=array->next)
154 return -1;
3b46e624 155
de167e41
FB
156 if(index_to==index_from)
157 return 0;
158
159 is=array->item_size;
160 from=array->pointer+index_from*is;
161 to=array->pointer+index_to*is;
3ec88e80 162 buf=qemu_malloc(is*count);
de167e41
FB
163 memcpy(buf,from,is*count);
164
165 if(index_to<index_from)
166 memmove(to+is*count,to,from-to);
167 else
168 memmove(from,from+is*count,to-from);
3b46e624 169
de167e41
FB
170 memcpy(to,buf,is*count);
171
172 free(buf);
173
174 return 0;
175}
176
c227f099 177static inline int array_remove_slice(array_t* array,int index, int count)
de167e41 178{
a046433a
FB
179 assert(index >=0);
180 assert(count > 0);
181 assert(index + count <= array->next);
182 if(array_roll(array,array->next-1,index,count))
de167e41 183 return -1;
a046433a 184 array->next -= count;
de167e41
FB
185 return 0;
186}
187
c227f099 188static int array_remove(array_t* array,int index)
a046433a
FB
189{
190 return array_remove_slice(array, index, 1);
191}
192
193/* return the index for a given member */
c227f099 194static int array_index(array_t* array, void* pointer)
a046433a
FB
195{
196 size_t offset = (char*)pointer - array->pointer;
a046433a
FB
197 assert((offset % array->item_size) == 0);
198 assert(offset/array->item_size < array->next);
199 return offset/array->item_size;
200}
201
de167e41
FB
202/* These structures are used to fake a disk and the VFAT filesystem.
203 * For this reason we need to use __attribute__((packed)). */
204
c227f099 205typedef struct bootsector_t {
de167e41
FB
206 uint8_t jump[3];
207 uint8_t name[8];
208 uint16_t sector_size;
209 uint8_t sectors_per_cluster;
210 uint16_t reserved_sectors;
211 uint8_t number_of_fats;
212 uint16_t root_entries;
a046433a 213 uint16_t total_sectors16;
de167e41
FB
214 uint8_t media_type;
215 uint16_t sectors_per_fat;
216 uint16_t sectors_per_track;
217 uint16_t number_of_heads;
218 uint32_t hidden_sectors;
219 uint32_t total_sectors;
220 union {
221 struct {
222 uint8_t drive_number;
223 uint8_t current_head;
224 uint8_t signature;
225 uint32_t id;
226 uint8_t volume_label[11];
227 } __attribute__((packed)) fat16;
228 struct {
229 uint32_t sectors_per_fat;
230 uint16_t flags;
231 uint8_t major,minor;
232 uint32_t first_cluster_of_root_directory;
233 uint16_t info_sector;
234 uint16_t backup_boot_sector;
235 uint16_t ignored;
236 } __attribute__((packed)) fat32;
237 } u;
238 uint8_t fat_type[8];
239 uint8_t ignored[0x1c0];
240 uint8_t magic[2];
c227f099 241} __attribute__((packed)) bootsector_t;
de167e41 242
b570094d
TS
243typedef struct {
244 uint8_t head;
245 uint8_t sector;
246 uint8_t cylinder;
c227f099 247} mbr_chs_t;
b570094d 248
c227f099 249typedef struct partition_t {
de167e41 250 uint8_t attributes; /* 0x80 = bootable */
c227f099 251 mbr_chs_t start_CHS;
b570094d 252 uint8_t fs_type; /* 0x1 = FAT12, 0x6 = FAT16, 0xe = FAT16_LBA, 0xb = FAT32, 0xc = FAT32_LBA */
c227f099 253 mbr_chs_t end_CHS;
de167e41 254 uint32_t start_sector_long;
b570094d 255 uint32_t length_sector_long;
c227f099 256} __attribute__((packed)) partition_t;
de167e41 257
c227f099 258typedef struct mbr_t {
b570094d
TS
259 uint8_t ignored[0x1b8];
260 uint32_t nt_id;
261 uint8_t ignored2[2];
c227f099 262 partition_t partition[4];
de167e41 263 uint8_t magic[2];
c227f099 264} __attribute__((packed)) mbr_t;
de167e41 265
c227f099 266typedef struct direntry_t {
de167e41
FB
267 uint8_t name[8];
268 uint8_t extension[3];
269 uint8_t attributes;
270 uint8_t reserved[2];
271 uint16_t ctime;
272 uint16_t cdate;
273 uint16_t adate;
274 uint16_t begin_hi;
275 uint16_t mtime;
276 uint16_t mdate;
277 uint16_t begin;
278 uint32_t size;
c227f099 279} __attribute__((packed)) direntry_t;
de167e41
FB
280
281/* this structure are used to transparently access the files */
282
c227f099 283typedef struct mapping_t {
a046433a
FB
284 /* begin is the first cluster, end is the last+1 */
285 uint32_t begin,end;
de167e41
FB
286 /* as s->directory is growable, no pointer may be used here */
287 unsigned int dir_index;
a046433a
FB
288 /* the clusters of a file may be in any order; this points to the first */
289 int first_mapping_index;
290 union {
291 /* offset is
292 * - the offset in the file (in clusters) for a file, or
293 * - the next cluster of the directory for a directory, and
294 * - the address of the buffer for a faked entry
295 */
296 struct {
297 uint32_t offset;
298 } file;
299 struct {
300 int parent_mapping_index;
301 int first_dir_index;
302 } dir;
303 } info;
304 /* path contains the full path, i.e. it always starts with s->path */
305 char* path;
306
307 enum { MODE_UNDEFINED = 0, MODE_NORMAL = 1, MODE_MODIFIED = 2,
308 MODE_DIRECTORY = 4, MODE_FAKED = 8,
309 MODE_DELETED = 16, MODE_RENAMED = 32 } mode;
310 int read_only;
c227f099 311} mapping_t;
de167e41 312
a046433a 313#ifdef DEBUG
c227f099
AL
314static void print_direntry(const struct direntry_t*);
315static void print_mapping(const struct mapping_t* mapping);
a046433a 316#endif
de167e41
FB
317
318/* here begins the real VVFAT driver */
319
320typedef struct BDRVVVFATState {
a046433a 321 BlockDriverState* bs; /* pointer to parent */
de167e41
FB
322 unsigned int first_sectors_number; /* 1 for a single partition, 0x40 for a disk with partition table */
323 unsigned char first_sectors[0x40*0x200];
3b46e624 324
de167e41 325 int fat_type; /* 16 or 32 */
c227f099 326 array_t fat,directory,mapping;
3b46e624 327
de167e41
FB
328 unsigned int cluster_size;
329 unsigned int sectors_per_cluster;
330 unsigned int sectors_per_fat;
331 unsigned int sectors_of_root_directory;
a046433a 332 uint32_t last_cluster_of_root_directory;
de167e41
FB
333 unsigned int faked_sectors; /* how many sectors are faked before file data */
334 uint32_t sector_count; /* total number of sectors of the partition */
335 uint32_t cluster_count; /* total number of clusters of this partition */
de167e41 336 uint32_t max_fat_value;
3b46e624 337
de167e41 338 int current_fd;
c227f099 339 mapping_t* current_mapping;
a046433a
FB
340 unsigned char* cluster; /* points to current cluster */
341 unsigned char* cluster_buffer; /* points to a buffer to hold temp data */
de167e41
FB
342 unsigned int current_cluster;
343
344 /* write support */
a046433a
FB
345 BlockDriverState* write_target;
346 char* qcow_filename;
347 BlockDriverState* qcow;
348 void* fat2;
349 char* used_clusters;
c227f099 350 array_t commits;
a046433a
FB
351 const char* path;
352 int downcase_short_names;
de167e41
FB
353} BDRVVVFATState;
354
b570094d
TS
355/* take the sector position spos and convert it to Cylinder/Head/Sector position
356 * if the position is outside the specified geometry, fill maximum value for CHS
357 * and return 1 to signal overflow.
358 */
c227f099 359static int sector2CHS(BlockDriverState* bs, mbr_chs_t * chs, int spos){
b570094d
TS
360 int head,sector;
361 sector = spos % (bs->secs); spos/= bs->secs;
362 head = spos % (bs->heads); spos/= bs->heads;
363 if(spos >= bs->cyls){
364 /* Overflow,
365 it happens if 32bit sector positions are used, while CHS is only 24bit.
366 Windows/Dos is said to take 1023/255/63 as nonrepresentable CHS */
367 chs->head = 0xFF;
368 chs->sector = 0xFF;
369 chs->cylinder = 0xFF;
370 return 1;
371 }
372 chs->head = (uint8_t)head;
373 chs->sector = (uint8_t)( (sector+1) | ((spos>>8)<<6) );
374 chs->cylinder = (uint8_t)spos;
375 return 0;
376}
de167e41 377
de167e41
FB
378static void init_mbr(BDRVVVFATState* s)
379{
380 /* TODO: if the files mbr.img and bootsect.img exist, use them */
c227f099
AL
381 mbr_t* real_mbr=(mbr_t*)s->first_sectors;
382 partition_t* partition = &(real_mbr->partition[0]);
b570094d 383 int lba;
de167e41
FB
384
385 memset(s->first_sectors,0,512);
3b46e624 386
b570094d
TS
387 /* Win NT Disk Signature */
388 real_mbr->nt_id= cpu_to_le32(0xbe1afdfa);
389
de167e41 390 partition->attributes=0x80; /* bootable */
b570094d
TS
391
392 /* LBA is used when partition is outside the CHS geometry */
393 lba = sector2CHS(s->bs, &partition->start_CHS, s->first_sectors_number-1);
394 lba|= sector2CHS(s->bs, &partition->end_CHS, s->sector_count);
395
396 /*LBA partitions are identified only by start/length_sector_long not by CHS*/
397 partition->start_sector_long =cpu_to_le32(s->first_sectors_number-1);
398 partition->length_sector_long=cpu_to_le32(s->sector_count - s->first_sectors_number+1);
399
a046433a 400 /* FAT12/FAT16/FAT32 */
b570094d
TS
401 /* DOS uses different types when partition is LBA,
402 probably to prevent older versions from using CHS on them */
403 partition->fs_type= s->fat_type==12 ? 0x1:
404 s->fat_type==16 ? (lba?0xe:0x06):
405 /*fat_tyoe==32*/ (lba?0xc:0x0b);
de167e41
FB
406
407 real_mbr->magic[0]=0x55; real_mbr->magic[1]=0xaa;
408}
409
a046433a
FB
410/* direntry functions */
411
de167e41 412/* dest is assumed to hold 258 bytes, and pads with 0xffff up to next multiple of 26 */
60fe76f3 413static inline int short2long_name(char* dest,const char* src)
de167e41
FB
414{
415 int i;
1e080d5d 416 int len;
de167e41
FB
417 for(i=0;i<129 && src[i];i++) {
418 dest[2*i]=src[i];
419 dest[2*i+1]=0;
420 }
1e080d5d 421 len=2*i;
de167e41
FB
422 dest[2*i]=dest[2*i+1]=0;
423 for(i=2*i+2;(i%26);i++)
424 dest[i]=0xff;
1e080d5d 425 return len;
de167e41
FB
426}
427
c227f099 428static inline direntry_t* create_long_filename(BDRVVVFATState* s,const char* filename)
de167e41
FB
429{
430 char buffer[258];
431 int length=short2long_name(buffer,filename),
432 number_of_entries=(length+25)/26,i;
c227f099 433 direntry_t* entry;
de167e41
FB
434
435 for(i=0;i<number_of_entries;i++) {
436 entry=array_get_next(&(s->directory));
437 entry->attributes=0xf;
438 entry->reserved[0]=0;
439 entry->begin=0;
440 entry->name[0]=(number_of_entries-i)|(i==0?0x40:0);
441 }
1e080d5d 442 for(i=0;i<26*number_of_entries;i++) {
de167e41
FB
443 int offset=(i%26);
444 if(offset<10) offset=1+offset;
445 else if(offset<22) offset=14+offset-10;
446 else offset=28+offset-22;
447 entry=array_get(&(s->directory),s->directory.next-1-(i/26));
448 entry->name[offset]=buffer[i];
449 }
450 return array_get(&(s->directory),s->directory.next-number_of_entries);
451}
452
c227f099 453static char is_free(const direntry_t* direntry)
a046433a 454{
ad1a897e 455 return direntry->name[0]==0xe5 || direntry->name[0]==0x00;
a046433a
FB
456}
457
c227f099 458static char is_volume_label(const direntry_t* direntry)
a046433a
FB
459{
460 return direntry->attributes == 0x28;
461}
462
c227f099 463static char is_long_name(const direntry_t* direntry)
a046433a
FB
464{
465 return direntry->attributes == 0xf;
466}
467
c227f099 468static char is_short_name(const direntry_t* direntry)
a046433a
FB
469{
470 return !is_volume_label(direntry) && !is_long_name(direntry)
471 && !is_free(direntry);
472}
473
c227f099 474static char is_directory(const direntry_t* direntry)
a046433a
FB
475{
476 return direntry->attributes & 0x10 && direntry->name[0] != 0xe5;
477}
478
c227f099 479static inline char is_dot(const direntry_t* direntry)
a046433a
FB
480{
481 return is_short_name(direntry) && direntry->name[0] == '.';
482}
483
c227f099 484static char is_file(const direntry_t* direntry)
a046433a
FB
485{
486 return is_short_name(direntry) && !is_directory(direntry);
487}
488
c227f099 489static inline uint32_t begin_of_direntry(const direntry_t* direntry)
a046433a
FB
490{
491 return le16_to_cpu(direntry->begin)|(le16_to_cpu(direntry->begin_hi)<<16);
492}
493
c227f099 494static inline uint32_t filesize_of_direntry(const direntry_t* direntry)
a046433a
FB
495{
496 return le32_to_cpu(direntry->size);
497}
498
c227f099 499static void set_begin_of_direntry(direntry_t* direntry, uint32_t begin)
a046433a
FB
500{
501 direntry->begin = cpu_to_le16(begin & 0xffff);
502 direntry->begin_hi = cpu_to_le16((begin >> 16) & 0xffff);
503}
504
de167e41
FB
505/* fat functions */
506
c227f099 507static inline uint8_t fat_chksum(const direntry_t* entry)
de167e41
FB
508{
509 uint8_t chksum=0;
510 int i;
511
5606c220
AJ
512 for(i=0;i<11;i++) {
513 unsigned char c;
514
2aa326be 515 c = (i < 8) ? entry->name[i] : entry->extension[i-8];
5606c220
AJ
516 chksum=(((chksum&0xfe)>>1)|((chksum&0x01)?0x80:0)) + c;
517 }
3b46e624 518
de167e41
FB
519 return chksum;
520}
521
522/* if return_time==0, this returns the fat_date, else the fat_time */
523static uint16_t fat_datetime(time_t time,int return_time) {
524 struct tm* t;
525#ifdef _WIN32
526 t=localtime(&time); /* this is not thread safe */
527#else
528 struct tm t1;
6ab00cee 529 t = &t1;
de167e41
FB
530 localtime_r(&time,t);
531#endif
532 if(return_time)
533 return cpu_to_le16((t->tm_sec/2)|(t->tm_min<<5)|(t->tm_hour<<11));
534 return cpu_to_le16((t->tm_mday)|((t->tm_mon+1)<<5)|((t->tm_year-80)<<9));
535}
536
537static inline void fat_set(BDRVVVFATState* s,unsigned int cluster,uint32_t value)
538{
a046433a
FB
539 if(s->fat_type==32) {
540 uint32_t* entry=array_get(&(s->fat),cluster);
541 *entry=cpu_to_le32(value);
de167e41
FB
542 } else if(s->fat_type==16) {
543 uint16_t* entry=array_get(&(s->fat),cluster);
544 *entry=cpu_to_le16(value&0xffff);
545 } else {
a046433a
FB
546 int offset = (cluster*3/2);
547 unsigned char* p = array_get(&(s->fat), offset);
548 switch (cluster&1) {
549 case 0:
550 p[0] = value&0xff;
551 p[1] = (p[1]&0xf0) | ((value>>8)&0xf);
552 break;
553 case 1:
554 p[0] = (p[0]&0xf) | ((value&0xf)<<4);
555 p[1] = (value>>4);
556 break;
557 }
de167e41
FB
558 }
559}
560
561static inline uint32_t fat_get(BDRVVVFATState* s,unsigned int cluster)
562{
a046433a
FB
563 if(s->fat_type==32) {
564 uint32_t* entry=array_get(&(s->fat),cluster);
565 return le32_to_cpu(*entry);
de167e41
FB
566 } else if(s->fat_type==16) {
567 uint16_t* entry=array_get(&(s->fat),cluster);
568 return le16_to_cpu(*entry);
569 } else {
ffe8ab83 570 const uint8_t* x=(uint8_t*)(s->fat.pointer)+cluster*3/2;
a046433a 571 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
de167e41
FB
572 }
573}
574
575static inline int fat_eof(BDRVVVFATState* s,uint32_t fat_entry)
576{
577 if(fat_entry>s->max_fat_value-8)
578 return -1;
579 return 0;
580}
581
582static inline void init_fat(BDRVVVFATState* s)
583{
a046433a
FB
584 if (s->fat_type == 12) {
585 array_init(&(s->fat),1);
586 array_ensure_allocated(&(s->fat),
587 s->sectors_per_fat * 0x200 * 3 / 2 - 1);
588 } else {
589 array_init(&(s->fat),(s->fat_type==32?4:2));
590 array_ensure_allocated(&(s->fat),
591 s->sectors_per_fat * 0x200 / s->fat.item_size - 1);
592 }
de167e41 593 memset(s->fat.pointer,0,s->fat.size);
3b46e624 594
de167e41
FB
595 switch(s->fat_type) {
596 case 12: s->max_fat_value=0xfff; break;
597 case 16: s->max_fat_value=0xffff; break;
a046433a 598 case 32: s->max_fat_value=0x0fffffff; break;
de167e41
FB
599 default: s->max_fat_value=0; /* error... */
600 }
601
602}
603
a046433a
FB
604/* TODO: in create_short_filename, 0xe5->0x05 is not yet handled! */
605/* TODO: in parse_short_filename, 0x05->0xe5 is not yet handled! */
c227f099 606static inline direntry_t* create_short_and_long_name(BDRVVVFATState* s,
a046433a 607 unsigned int directory_start, const char* filename, int is_dot)
de167e41 608{
a046433a 609 int i,j,long_index=s->directory.next;
c227f099
AL
610 direntry_t* entry = NULL;
611 direntry_t* entry_long = NULL;
de167e41
FB
612
613 if(is_dot) {
614 entry=array_get_next(&(s->directory));
615 memset(entry->name,0x20,11);
616 memcpy(entry->name,filename,strlen(filename));
617 return entry;
618 }
3b46e624 619
de167e41 620 entry_long=create_long_filename(s,filename);
3b46e624 621
5fafdf24 622 i = strlen(filename);
a046433a
FB
623 for(j = i - 1; j>0 && filename[j]!='.';j--);
624 if (j > 0)
625 i = (j > 8 ? 8 : j);
626 else if (i > 8)
627 i = 8;
628
de167e41
FB
629 entry=array_get_next(&(s->directory));
630 memset(entry->name,0x20,11);
51a0f568 631 memcpy(entry->name, filename, i);
3b46e624 632
a046433a
FB
633 if(j > 0)
634 for (i = 0; i < 3 && filename[j+1+i]; i++)
635 entry->extension[i] = filename[j+1+i];
de167e41
FB
636
637 /* upcase & remove unwanted characters */
638 for(i=10;i>=0;i--) {
a046433a 639 if(i==10 || i==7) for(;i>0 && entry->name[i]==' ';i--);
de167e41 640 if(entry->name[i]<=' ' || entry->name[i]>0x7f
a046433a 641 || strchr(".*?<>|\":/\\[];,+='",entry->name[i]))
de167e41
FB
642 entry->name[i]='_';
643 else if(entry->name[i]>='a' && entry->name[i]<='z')
644 entry->name[i]+='A'-'a';
645 }
646
647 /* mangle duplicates */
648 while(1) {
c227f099 649 direntry_t* entry1=array_get(&(s->directory),directory_start);
de167e41
FB
650 int j;
651
652 for(;entry1<entry;entry1++)
a046433a 653 if(!is_long_name(entry1) && !memcmp(entry1->name,entry->name,11))
de167e41
FB
654 break; /* found dupe */
655 if(entry1==entry) /* no dupe found */
656 break;
657
5fafdf24 658 /* use all 8 characters of name */
de167e41
FB
659 if(entry->name[7]==' ') {
660 int j;
661 for(j=6;j>0 && entry->name[j]==' ';j--)
662 entry->name[j]='~';
663 }
664
665 /* increment number */
666 for(j=7;j>0 && entry->name[j]=='9';j--)
667 entry->name[j]='0';
668 if(j>0) {
669 if(entry->name[j]<'0' || entry->name[j]>'9')
670 entry->name[j]='0';
671 else
672 entry->name[j]++;
673 }
674 }
675
676 /* calculate checksum; propagate to long name */
677 if(entry_long) {
678 uint8_t chksum=fat_chksum(entry);
679
680 /* calculate anew, because realloc could have taken place */
681 entry_long=array_get(&(s->directory),long_index);
a046433a 682 while(entry_long<entry && is_long_name(entry_long)) {
de167e41
FB
683 entry_long->reserved[1]=chksum;
684 entry_long++;
685 }
686 }
687
688 return entry;
689}
690
a046433a
FB
691/*
692 * Read a directory. (the index of the corresponding mapping must be passed).
693 */
694static int read_directory(BDRVVVFATState* s, int mapping_index)
de167e41 695{
c227f099
AL
696 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
697 direntry_t* direntry;
a046433a
FB
698 const char* dirname = mapping->path;
699 int first_cluster = mapping->begin;
700 int parent_index = mapping->info.dir.parent_mapping_index;
c227f099 701 mapping_t* parent_mapping = (mapping_t*)
511d2b14 702 (parent_index >= 0 ? array_get(&(s->mapping), parent_index) : NULL);
a046433a 703 int first_cluster_of_parent = parent_mapping ? parent_mapping->begin : -1;
de167e41
FB
704
705 DIR* dir=opendir(dirname);
706 struct dirent* entry;
de167e41
FB
707 int i;
708
a046433a
FB
709 assert(mapping->mode & MODE_DIRECTORY);
710
711 if(!dir) {
712 mapping->end = mapping->begin;
de167e41 713 return -1;
a046433a 714 }
3b46e624 715
a046433a
FB
716 i = mapping->info.dir.first_dir_index =
717 first_cluster == 0 ? 0 : s->directory.next;
718
5fafdf24 719 /* actually read the directory, and allocate the mappings */
de167e41
FB
720 while((entry=readdir(dir))) {
721 unsigned int length=strlen(dirname)+2+strlen(entry->d_name);
722 char* buffer;
c227f099 723 direntry_t* direntry;
a046433a 724 struct stat st;
de167e41
FB
725 int is_dot=!strcmp(entry->d_name,".");
726 int is_dotdot=!strcmp(entry->d_name,"..");
727
a046433a 728 if(first_cluster == 0 && (is_dotdot || is_dot))
de167e41 729 continue;
5fafdf24 730
3ec88e80 731 buffer=(char*)qemu_malloc(length);
de167e41
FB
732 snprintf(buffer,length,"%s/%s",dirname,entry->d_name);
733
734 if(stat(buffer,&st)<0) {
735 free(buffer);
736 continue;
737 }
738
739 /* create directory entry for this file */
a046433a
FB
740 direntry=create_short_and_long_name(s, i, entry->d_name,
741 is_dot || is_dotdot);
de167e41
FB
742 direntry->attributes=(S_ISDIR(st.st_mode)?0x10:0x20);
743 direntry->reserved[0]=direntry->reserved[1]=0;
744 direntry->ctime=fat_datetime(st.st_ctime,1);
745 direntry->cdate=fat_datetime(st.st_ctime,0);
746 direntry->adate=fat_datetime(st.st_atime,0);
747 direntry->begin_hi=0;
748 direntry->mtime=fat_datetime(st.st_mtime,1);
749 direntry->mdate=fat_datetime(st.st_mtime,0);
750 if(is_dotdot)
a046433a 751 set_begin_of_direntry(direntry, first_cluster_of_parent);
de167e41 752 else if(is_dot)
a046433a 753 set_begin_of_direntry(direntry, first_cluster);
de167e41 754 else
a046433a
FB
755 direntry->begin=0; /* do that later */
756 if (st.st_size > 0x7fffffff) {
757 fprintf(stderr, "File %s is larger than 2GB\n", buffer);
758 free(buffer);
759 return -2;
760 }
761 direntry->size=cpu_to_le32(S_ISDIR(st.st_mode)?0:st.st_size);
de167e41
FB
762
763 /* create mapping for this file */
a046433a 764 if(!is_dot && !is_dotdot && (S_ISDIR(st.st_mode) || st.st_size)) {
c227f099 765 s->current_mapping=(mapping_t*)array_get_next(&(s->mapping));
de167e41
FB
766 s->current_mapping->begin=0;
767 s->current_mapping->end=st.st_size;
a046433a
FB
768 /*
769 * we get the direntry of the most recent direntry, which
770 * contains the short name and all the relevant information.
771 */
de167e41 772 s->current_mapping->dir_index=s->directory.next-1;
a046433a
FB
773 s->current_mapping->first_mapping_index = -1;
774 if (S_ISDIR(st.st_mode)) {
775 s->current_mapping->mode = MODE_DIRECTORY;
776 s->current_mapping->info.dir.parent_mapping_index =
777 mapping_index;
778 } else {
779 s->current_mapping->mode = MODE_UNDEFINED;
780 s->current_mapping->info.file.offset = 0;
781 }
782 s->current_mapping->path=buffer;
783 s->current_mapping->read_only =
784 (st.st_mode & (S_IWUSR | S_IWGRP | S_IWOTH)) == 0;
de167e41
FB
785 }
786 }
787 closedir(dir);
788
789 /* fill with zeroes up to the end of the cluster */
790 while(s->directory.next%(0x10*s->sectors_per_cluster)) {
c227f099
AL
791 direntry_t* direntry=array_get_next(&(s->directory));
792 memset(direntry,0,sizeof(direntry_t));
de167e41
FB
793 }
794
a046433a
FB
795/* TODO: if there are more entries, bootsector has to be adjusted! */
796#define ROOT_ENTRIES (0x02 * 0x10 * s->sectors_per_cluster)
797 if (mapping_index == 0 && s->directory.next < ROOT_ENTRIES) {
798 /* root directory */
799 int cur = s->directory.next;
800 array_ensure_allocated(&(s->directory), ROOT_ENTRIES - 1);
801 memset(array_get(&(s->directory), cur), 0,
c227f099 802 (ROOT_ENTRIES - cur) * sizeof(direntry_t));
de167e41 803 }
5fafdf24 804
a046433a 805 /* reget the mapping, since s->mapping was possibly realloc()ed */
c227f099 806 mapping = (mapping_t*)array_get(&(s->mapping), mapping_index);
a046433a
FB
807 first_cluster += (s->directory.next - mapping->info.dir.first_dir_index)
808 * 0x20 / s->cluster_size;
809 mapping->end = first_cluster;
810
c227f099 811 direntry = (direntry_t*)array_get(&(s->directory), mapping->dir_index);
a046433a 812 set_begin_of_direntry(direntry, mapping->begin);
3b46e624 813
a046433a
FB
814 return 0;
815}
de167e41 816
a046433a
FB
817static inline uint32_t sector2cluster(BDRVVVFATState* s,off_t sector_num)
818{
819 return (sector_num-s->faked_sectors)/s->sectors_per_cluster;
820}
de167e41 821
a046433a
FB
822static inline off_t cluster2sector(BDRVVVFATState* s, uint32_t cluster_num)
823{
824 return s->faked_sectors + s->sectors_per_cluster * cluster_num;
825}
de167e41 826
a046433a
FB
827static inline uint32_t sector_offset_in_cluster(BDRVVVFATState* s,off_t sector_num)
828{
829 return (sector_num-s->first_sectors_number-2*s->sectors_per_fat)%s->sectors_per_cluster;
830}
de167e41 831
a046433a 832#ifdef DBG
c227f099 833static direntry_t* get_direntry_for_mapping(BDRVVVFATState* s,mapping_t* mapping)
a046433a
FB
834{
835 if(mapping->mode==MODE_UNDEFINED)
836 return 0;
c227f099 837 return (direntry_t*)(s->directory.pointer+sizeof(direntry_t)*mapping->dir_index);
de167e41 838}
a046433a 839#endif
de167e41 840
a046433a
FB
841static int init_directories(BDRVVVFATState* s,
842 const char* dirname)
de167e41 843{
c227f099
AL
844 bootsector_t* bootsector;
845 mapping_t* mapping;
de167e41
FB
846 unsigned int i;
847 unsigned int cluster;
848
849 memset(&(s->first_sectors[0]),0,0x40*0x200);
850
de167e41 851 s->cluster_size=s->sectors_per_cluster*0x200;
3ec88e80 852 s->cluster_buffer=qemu_malloc(s->cluster_size);
a046433a
FB
853
854 /*
855 * The formula: sc = spf+1+spf*spc*(512*8/fat_type),
856 * where sc is sector_count,
857 * spf is sectors_per_fat,
858 * spc is sectors_per_clusters, and
859 * fat_type = 12, 16 or 32.
860 */
861 i = 1+s->sectors_per_cluster*0x200*8/s->fat_type;
862 s->sectors_per_fat=(s->sector_count+i)/i; /* round up */
3b46e624 863
c227f099
AL
864 array_init(&(s->mapping),sizeof(mapping_t));
865 array_init(&(s->directory),sizeof(direntry_t));
de167e41
FB
866
867 /* add volume label */
868 {
c227f099 869 direntry_t* entry=array_get_next(&(s->directory));
de167e41 870 entry->attributes=0x28; /* archive | volume label */
2dedf83e
KS
871 memcpy(entry->name,"QEMU VVF",8);
872 memcpy(entry->extension,"AT ",3);
de167e41
FB
873 }
874
de167e41
FB
875 /* Now build FAT, and write back information into directory */
876 init_fat(s);
877
a046433a
FB
878 s->faked_sectors=s->first_sectors_number+s->sectors_per_fat*2;
879 s->cluster_count=sector2cluster(s, s->sector_count);
880
881 mapping = array_get_next(&(s->mapping));
882 mapping->begin = 0;
883 mapping->dir_index = 0;
884 mapping->info.dir.parent_mapping_index = -1;
885 mapping->first_mapping_index = -1;
6265eb26 886 mapping->path = qemu_strdup(dirname);
a046433a
FB
887 i = strlen(mapping->path);
888 if (i > 0 && mapping->path[i - 1] == '/')
889 mapping->path[i - 1] = '\0';
890 mapping->mode = MODE_DIRECTORY;
891 mapping->read_only = 0;
892 s->path = mapping->path;
893
894 for (i = 0, cluster = 0; i < s->mapping.next; i++) {
5fafdf24 895 /* MS-DOS expects the FAT to be 0 for the root directory
a046433a
FB
896 * (except for the media byte). */
897 /* LATER TODO: still true for FAT32? */
898 int fix_fat = (i != 0);
899 mapping = array_get(&(s->mapping), i);
900
901 if (mapping->mode & MODE_DIRECTORY) {
902 mapping->begin = cluster;
903 if(read_directory(s, i)) {
904 fprintf(stderr, "Could not read directory %s\n",
905 mapping->path);
de167e41
FB
906 return -1;
907 }
a046433a
FB
908 mapping = array_get(&(s->mapping), i);
909 } else {
910 assert(mapping->mode == MODE_UNDEFINED);
de167e41 911 mapping->mode=MODE_NORMAL;
a046433a
FB
912 mapping->begin = cluster;
913 if (mapping->end > 0) {
c227f099 914 direntry_t* direntry = array_get(&(s->directory),
a046433a
FB
915 mapping->dir_index);
916
917 mapping->end = cluster + 1 + (mapping->end-1)/s->cluster_size;
918 set_begin_of_direntry(direntry, mapping->begin);
919 } else {
920 mapping->end = cluster + 1;
921 fix_fat = 0;
de167e41 922 }
a046433a
FB
923 }
924
925 assert(mapping->begin < mapping->end);
926
8ce0f869
AZ
927 /* next free cluster */
928 cluster = mapping->end;
929
930 if(cluster > s->cluster_count) {
931 fprintf(stderr,"Directory does not fit in FAT%d (capacity %s)\n",
932 s->fat_type,
933 s->fat_type == 12 ? s->sector_count == 2880 ? "1.44 MB"
934 : "2.88 MB"
935 : "504MB");
936 return -EINVAL;
937 }
938
a046433a
FB
939 /* fix fat for entry */
940 if (fix_fat) {
8ce0f869 941 int j;
a046433a
FB
942 for(j = mapping->begin; j < mapping->end - 1; j++)
943 fat_set(s, j, j+1);
944 fat_set(s, mapping->end - 1, s->max_fat_value);
945 }
de167e41
FB
946 }
947
a046433a
FB
948 mapping = array_get(&(s->mapping), 0);
949 s->sectors_of_root_directory = mapping->end * s->sectors_per_cluster;
950 s->last_cluster_of_root_directory = mapping->end;
951
952 /* the FAT signature */
953 fat_set(s,0,s->max_fat_value);
954 fat_set(s,1,s->max_fat_value);
de167e41 955
a046433a
FB
956 s->current_mapping = NULL;
957
c227f099 958 bootsector=(bootsector_t*)(s->first_sectors+(s->first_sectors_number-1)*0x200);
de167e41
FB
959 bootsector->jump[0]=0xeb;
960 bootsector->jump[1]=0x3e;
961 bootsector->jump[2]=0x90;
962 memcpy(bootsector->name,"QEMU ",8);
963 bootsector->sector_size=cpu_to_le16(0x200);
964 bootsector->sectors_per_cluster=s->sectors_per_cluster;
965 bootsector->reserved_sectors=cpu_to_le16(1);
966 bootsector->number_of_fats=0x2; /* number of FATs */
967 bootsector->root_entries=cpu_to_le16(s->sectors_of_root_directory*0x10);
a046433a
FB
968 bootsector->total_sectors16=s->sector_count>0xffff?0:cpu_to_le16(s->sector_count);
969 bootsector->media_type=(s->fat_type!=12?0xf8:s->sector_count==5760?0xf9:0xf8); /* media descriptor */
970 s->fat.pointer[0] = bootsector->media_type;
de167e41 971 bootsector->sectors_per_fat=cpu_to_le16(s->sectors_per_fat);
a046433a
FB
972 bootsector->sectors_per_track=cpu_to_le16(s->bs->secs);
973 bootsector->number_of_heads=cpu_to_le16(s->bs->heads);
de167e41 974 bootsector->hidden_sectors=cpu_to_le32(s->first_sectors_number==1?0:0x3f);
a046433a 975 bootsector->total_sectors=cpu_to_le32(s->sector_count>0xffff?s->sector_count:0);
de167e41 976
a046433a
FB
977 /* LATER TODO: if FAT32, this is wrong */
978 bootsector->u.fat16.drive_number=s->fat_type==12?0:0x80; /* assume this is hda (TODO) */
de167e41
FB
979 bootsector->u.fat16.current_head=0;
980 bootsector->u.fat16.signature=0x29;
981 bootsector->u.fat16.id=cpu_to_le32(0xfabe1afd);
982
983 memcpy(bootsector->u.fat16.volume_label,"QEMU VVFAT ",11);
984 memcpy(bootsector->fat_type,(s->fat_type==12?"FAT12 ":s->fat_type==16?"FAT16 ":"FAT32 "),8);
985 bootsector->magic[0]=0x55; bootsector->magic[1]=0xaa;
986
987 return 0;
988}
989
83f64091 990#ifdef DEBUG
a046433a 991static BDRVVVFATState *vvv = NULL;
83f64091 992#endif
a046433a
FB
993
994static int enable_write_target(BDRVVVFATState *s);
995static int is_consistent(BDRVVVFATState *s);
996
83f64091 997static int vvfat_open(BlockDriverState *bs, const char* dirname, int flags)
de167e41
FB
998{
999 BDRVVVFATState *s = bs->opaque;
a046433a 1000 int floppy = 0;
de167e41
FB
1001 int i;
1002
83f64091 1003#ifdef DEBUG
a046433a 1004 vvv = s;
83f64091 1005#endif
a046433a
FB
1006
1007DLOG(if (stderr == NULL) {
1008 stderr = fopen("vvfat.log", "a");
1009 setbuf(stderr, NULL);
1010})
1011
1012 s->bs = bs;
1013
de167e41 1014 s->fat_type=16;
a046433a 1015 /* LATER TODO: if FAT32, adjust */
a046433a 1016 s->sectors_per_cluster=0x10;
b570094d
TS
1017 /* 504MB disk*/
1018 bs->cyls=1024; bs->heads=16; bs->secs=63;
de167e41
FB
1019
1020 s->current_cluster=0xffffffff;
de167e41 1021
de167e41 1022 s->first_sectors_number=0x40;
a046433a
FB
1023 /* read only is the default for safety */
1024 bs->read_only = 1;
1025 s->qcow = s->write_target = NULL;
1026 s->qcow_filename = NULL;
1027 s->fat2 = NULL;
1028 s->downcase_short_names = 1;
3b46e624 1029
a046433a
FB
1030 if (!strstart(dirname, "fat:", NULL))
1031 return -1;
1032
a046433a
FB
1033 if (strstr(dirname, ":floppy:")) {
1034 floppy = 1;
1035 s->fat_type = 12;
1036 s->first_sectors_number = 1;
1037 s->sectors_per_cluster=2;
1038 bs->cyls = 80; bs->heads = 2; bs->secs = 36;
1039 }
1040
b570094d
TS
1041 s->sector_count=bs->cyls*bs->heads*bs->secs;
1042
a046433a
FB
1043 if (strstr(dirname, ":32:")) {
1044 fprintf(stderr, "Big fat greek warning: FAT32 has not been tested. You are welcome to do so!\n");
1045 s->fat_type = 32;
1046 } else if (strstr(dirname, ":16:")) {
1047 s->fat_type = 16;
1048 } else if (strstr(dirname, ":12:")) {
1049 s->fat_type = 12;
1050 s->sector_count=2880;
de167e41 1051 }
a046433a 1052
b570094d
TS
1053 if (strstr(dirname, ":rw:")) {
1054 if (enable_write_target(s))
1055 return -1;
1056 bs->read_only = 0;
1057 }
1058
a046433a
FB
1059 i = strrchr(dirname, ':') - dirname;
1060 assert(i >= 3);
cd390083 1061 if (dirname[i-2] == ':' && qemu_isalpha(dirname[i-1]))
a046433a
FB
1062 /* workaround for DOS drive names */
1063 dirname += i-1;
1064 else
1065 dirname += i+1;
1066
1067 bs->total_sectors=bs->cyls*bs->heads*bs->secs;
b570094d 1068
a046433a 1069 if(init_directories(s, dirname))
de167e41
FB
1070 return -1;
1071
b570094d
TS
1072 s->sector_count = s->faked_sectors + s->sectors_per_cluster*s->cluster_count;
1073
de167e41
FB
1074 if(s->first_sectors_number==0x40)
1075 init_mbr(s);
1076
a046433a
FB
1077 /* for some reason or other, MS-DOS does not like to know about CHS... */
1078 if (floppy)
1079 bs->heads = bs->cyls = bs->secs = 0;
1080
1081 // assert(is_consistent(s));
de167e41
FB
1082 return 0;
1083}
1084
1085static inline void vvfat_close_current_file(BDRVVVFATState *s)
1086{
1087 if(s->current_mapping) {
a046433a
FB
1088 s->current_mapping = NULL;
1089 if (s->current_fd) {
1090 close(s->current_fd);
1091 s->current_fd = 0;
1092 }
de167e41 1093 }
a046433a 1094 s->current_cluster = -1;
de167e41
FB
1095}
1096
1097/* mappings between index1 and index2-1 are supposed to be ordered
1098 * return value is the index of the last mapping for which end>cluster_num
1099 */
1100static inline int find_mapping_for_cluster_aux(BDRVVVFATState* s,int cluster_num,int index1,int index2)
1101{
de167e41 1102 while(1) {
88bf7950 1103 int index3;
c227f099 1104 mapping_t* mapping;
de167e41
FB
1105 index3=(index1+index2)/2;
1106 mapping=array_get(&(s->mapping),index3);
a046433a
FB
1107 assert(mapping->begin < mapping->end);
1108 if(mapping->begin>=cluster_num) {
de167e41
FB
1109 assert(index2!=index3 || index2==0);
1110 if(index2==index3)
a046433a 1111 return index1;
de167e41
FB
1112 index2=index3;
1113 } else {
1114 if(index1==index3)
a046433a 1115 return mapping->end<=cluster_num ? index2 : index1;
de167e41
FB
1116 index1=index3;
1117 }
1118 assert(index1<=index2);
a046433a
FB
1119 DLOG(mapping=array_get(&(s->mapping),index1);
1120 assert(mapping->begin<=cluster_num);
5fafdf24 1121 assert(index2 >= s->mapping.next ||
a046433a
FB
1122 ((mapping = array_get(&(s->mapping),index2)) &&
1123 mapping->end>cluster_num)));
de167e41
FB
1124 }
1125}
1126
c227f099 1127static inline mapping_t* find_mapping_for_cluster(BDRVVVFATState* s,int cluster_num)
de167e41
FB
1128{
1129 int index=find_mapping_for_cluster_aux(s,cluster_num,0,s->mapping.next);
c227f099 1130 mapping_t* mapping;
de167e41 1131 if(index>=s->mapping.next)
511d2b14 1132 return NULL;
de167e41
FB
1133 mapping=array_get(&(s->mapping),index);
1134 if(mapping->begin>cluster_num)
511d2b14 1135 return NULL;
a046433a 1136 assert(mapping->begin<=cluster_num && mapping->end>cluster_num);
de167e41
FB
1137 return mapping;
1138}
1139
a046433a
FB
1140/*
1141 * This function simply compares path == mapping->path. Since the mappings
1142 * are sorted by cluster, this is expensive: O(n).
1143 */
c227f099 1144static inline mapping_t* find_mapping_for_path(BDRVVVFATState* s,
a046433a
FB
1145 const char* path)
1146{
1147 int i;
1148
1149 for (i = 0; i < s->mapping.next; i++) {
c227f099 1150 mapping_t* mapping = array_get(&(s->mapping), i);
a046433a
FB
1151 if (mapping->first_mapping_index < 0 &&
1152 !strcmp(path, mapping->path))
1153 return mapping;
1154 }
1155
1156 return NULL;
1157}
1158
c227f099 1159static int open_file(BDRVVVFATState* s,mapping_t* mapping)
de167e41
FB
1160{
1161 if(!mapping)
1162 return -1;
de167e41 1163 if(!s->current_mapping ||
a046433a 1164 strcmp(s->current_mapping->path,mapping->path)) {
de167e41 1165 /* open file */
a046433a 1166 int fd = open(mapping->path, O_RDONLY | O_BINARY | O_LARGEFILE);
de167e41
FB
1167 if(fd<0)
1168 return -1;
1169 vvfat_close_current_file(s);
1170 s->current_fd = fd;
de167e41
FB
1171 s->current_mapping = mapping;
1172 }
1173 return 0;
1174}
1175
1176static inline int read_cluster(BDRVVVFATState *s,int cluster_num)
1177{
1178 if(s->current_cluster != cluster_num) {
1179 int result=0;
1180 off_t offset;
a046433a 1181 assert(!s->current_mapping || s->current_fd || (s->current_mapping->mode & MODE_DIRECTORY));
de167e41
FB
1182 if(!s->current_mapping
1183 || s->current_mapping->begin>cluster_num
1184 || s->current_mapping->end<=cluster_num) {
1185 /* binary search of mappings for file */
c227f099 1186 mapping_t* mapping=find_mapping_for_cluster(s,cluster_num);
a046433a
FB
1187
1188 assert(!mapping || (cluster_num>=mapping->begin && cluster_num<mapping->end));
1189
1190 if (mapping && mapping->mode & MODE_DIRECTORY) {
1191 vvfat_close_current_file(s);
1192 s->current_mapping = mapping;
1193read_cluster_directory:
1194 offset = s->cluster_size*(cluster_num-s->current_mapping->begin);
ffe8ab83 1195 s->cluster = (unsigned char*)s->directory.pointer+offset
a046433a
FB
1196 + 0x20*s->current_mapping->info.dir.first_dir_index;
1197 assert(((s->cluster-(unsigned char*)s->directory.pointer)%s->cluster_size)==0);
1198 assert((char*)s->cluster+s->cluster_size <= s->directory.pointer+s->directory.next*s->directory.item_size);
1199 s->current_cluster = cluster_num;
1200 return 0;
1201 }
1202
1203 if(open_file(s,mapping))
de167e41 1204 return -2;
a046433a
FB
1205 } else if (s->current_mapping->mode & MODE_DIRECTORY)
1206 goto read_cluster_directory;
de167e41 1207
a046433a
FB
1208 assert(s->current_fd);
1209
1210 offset=s->cluster_size*(cluster_num-s->current_mapping->begin)+s->current_mapping->info.file.offset;
de167e41
FB
1211 if(lseek(s->current_fd, offset, SEEK_SET)!=offset)
1212 return -3;
a046433a 1213 s->cluster=s->cluster_buffer;
de167e41
FB
1214 result=read(s->current_fd,s->cluster,s->cluster_size);
1215 if(result<0) {
1216 s->current_cluster = -1;
1217 return -1;
1218 }
1219 s->current_cluster = cluster_num;
1220 }
1221 return 0;
1222}
1223
a046433a
FB
1224#ifdef DEBUG
1225static void hexdump(const void* address, uint32_t len)
de167e41 1226{
a046433a
FB
1227 const unsigned char* p = address;
1228 int i, j;
1229
1230 for (i = 0; i < len; i += 16) {
1231 for (j = 0; j < 16 && i + j < len; j++)
1232 fprintf(stderr, "%02x ", p[i + j]);
1233 for (; j < 16; j++)
1234 fprintf(stderr, " ");
1235 fprintf(stderr, " ");
1236 for (j = 0; j < 16 && i + j < len; j++)
1237 fprintf(stderr, "%c", (p[i + j] < ' ' || p[i + j] > 0x7f) ? '.' : p[i + j]);
1238 fprintf(stderr, "\n");
de167e41 1239 }
de167e41
FB
1240}
1241
c227f099 1242static void print_direntry(const direntry_t* direntry)
de167e41 1243{
a046433a
FB
1244 int j = 0;
1245 char buffer[1024];
1246
3e89cb04 1247 fprintf(stderr, "direntry %p: ", direntry);
de167e41
FB
1248 if(!direntry)
1249 return;
a046433a 1250 if(is_long_name(direntry)) {
de167e41
FB
1251 unsigned char* c=(unsigned char*)direntry;
1252 int i;
1253 for(i=1;i<11 && c[i] && c[i]!=0xff;i+=2)
3891b370 1254#define ADD_CHAR(c) {buffer[j] = (c); if (buffer[j] < ' ') buffer[j] = 0xb0; j++;}
a046433a 1255 ADD_CHAR(c[i]);
de167e41 1256 for(i=14;i<26 && c[i] && c[i]!=0xff;i+=2)
a046433a 1257 ADD_CHAR(c[i]);
de167e41 1258 for(i=28;i<32 && c[i] && c[i]!=0xff;i+=2)
a046433a
FB
1259 ADD_CHAR(c[i]);
1260 buffer[j] = 0;
1261 fprintf(stderr, "%s\n", buffer);
de167e41
FB
1262 } else {
1263 int i;
1264 for(i=0;i<11;i++)
a046433a
FB
1265 ADD_CHAR(direntry->name[i]);
1266 buffer[j] = 0;
1267 fprintf(stderr,"%s attributes=0x%02x begin=%d size=%d\n",
1268 buffer,
de167e41 1269 direntry->attributes,
a046433a 1270 begin_of_direntry(direntry),le32_to_cpu(direntry->size));
de167e41
FB
1271 }
1272}
1273
c227f099 1274static void print_mapping(const mapping_t* mapping)
de167e41 1275{
3e89cb04
KW
1276 fprintf(stderr, "mapping (%p): begin, end = %d, %d, dir_index = %d, "
1277 "first_mapping_index = %d, name = %s, mode = 0x%x, " ,
1278 mapping, mapping->begin, mapping->end, mapping->dir_index,
1279 mapping->first_mapping_index, mapping->path, mapping->mode);
1280
a046433a
FB
1281 if (mapping->mode & MODE_DIRECTORY)
1282 fprintf(stderr, "parent_mapping_index = %d, first_dir_index = %d\n", mapping->info.dir.parent_mapping_index, mapping->info.dir.first_dir_index);
1283 else
1284 fprintf(stderr, "offset = %d\n", mapping->info.file.offset);
de167e41 1285}
a046433a 1286#endif
de167e41 1287
5fafdf24 1288static int vvfat_read(BlockDriverState *bs, int64_t sector_num,
a046433a 1289 uint8_t *buf, int nb_sectors)
de167e41 1290{
a046433a 1291 BDRVVVFATState *s = bs->opaque;
de167e41 1292 int i;
de167e41 1293
a046433a
FB
1294 for(i=0;i<nb_sectors;i++,sector_num++) {
1295 if (sector_num >= s->sector_count)
1296 return -1;
1297 if (s->qcow) {
1298 int n;
1299 if (s->qcow->drv->bdrv_is_allocated(s->qcow,
1300 sector_num, nb_sectors-i, &n)) {
1301DLOG(fprintf(stderr, "sectors %d+%d allocated\n", (int)sector_num, n));
1302 if (s->qcow->drv->bdrv_read(s->qcow, sector_num, buf+i*0x200, n))
1303 return -1;
1304 i += n - 1;
1305 sector_num += n - 1;
1306 continue;
1307 }
1308DLOG(fprintf(stderr, "sector %d not allocated\n", (int)sector_num));
de167e41 1309 }
a046433a
FB
1310 if(sector_num<s->faked_sectors) {
1311 if(sector_num<s->first_sectors_number)
1312 memcpy(buf+i*0x200,&(s->first_sectors[sector_num*0x200]),0x200);
1313 else if(sector_num-s->first_sectors_number<s->sectors_per_fat)
1314 memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number)*0x200]),0x200);
1315 else if(sector_num-s->first_sectors_number-s->sectors_per_fat<s->sectors_per_fat)
1316 memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number-s->sectors_per_fat)*0x200]),0x200);
1317 } else {
1318 uint32_t sector=sector_num-s->faked_sectors,
1319 sector_offset_in_cluster=(sector%s->sectors_per_cluster),
1320 cluster_num=sector/s->sectors_per_cluster;
1321 if(read_cluster(s, cluster_num) != 0) {
1322 /* LATER TODO: strict: return -1; */
1323 memset(buf+i*0x200,0,0x200);
1324 continue;
de167e41 1325 }
a046433a 1326 memcpy(buf+i*0x200,s->cluster+sector_offset_in_cluster*0x200,0x200);
de167e41
FB
1327 }
1328 }
de167e41
FB
1329 return 0;
1330}
1331
a046433a 1332/* LATER TODO: statify all functions */
de167e41 1333
a046433a
FB
1334/*
1335 * Idea of the write support (use snapshot):
de167e41 1336 *
a046433a
FB
1337 * 1. check if all data is consistent, recording renames, modifications,
1338 * new files and directories (in s->commits).
de167e41 1339 *
a046433a 1340 * 2. if the data is not consistent, stop committing
de167e41 1341 *
a046433a
FB
1342 * 3. handle renames, and create new files and directories (do not yet
1343 * write their contents)
de167e41 1344 *
a046433a
FB
1345 * 4. walk the directories, fixing the mapping and direntries, and marking
1346 * the handled mappings as not deleted
de167e41 1347 *
a046433a 1348 * 5. commit the contents of the files
de167e41 1349 *
a046433a 1350 * 6. handle deleted files and directories
de167e41
FB
1351 *
1352 */
1353
c227f099 1354typedef struct commit_t {
a046433a
FB
1355 char* path;
1356 union {
1357 struct { uint32_t cluster; } rename;
1358 struct { int dir_index; uint32_t modified_offset; } writeout;
1359 struct { uint32_t first_cluster; } new_file;
1360 struct { uint32_t cluster; } mkdir;
1361 } param;
1362 /* DELETEs and RMDIRs are handled differently: see handle_deletes() */
1363 enum {
1364 ACTION_RENAME, ACTION_WRITEOUT, ACTION_NEW_FILE, ACTION_MKDIR
1365 } action;
c227f099 1366} commit_t;
de167e41 1367
a046433a 1368static void clear_commits(BDRVVVFATState* s)
de167e41
FB
1369{
1370 int i;
a046433a
FB
1371DLOG(fprintf(stderr, "clear_commits (%d commits)\n", s->commits.next));
1372 for (i = 0; i < s->commits.next; i++) {
c227f099 1373 commit_t* commit = array_get(&(s->commits), i);
a046433a
FB
1374 assert(commit->path || commit->action == ACTION_WRITEOUT);
1375 if (commit->action != ACTION_WRITEOUT) {
1376 assert(commit->path);
1377 free(commit->path);
1378 } else
1379 assert(commit->path == NULL);
de167e41 1380 }
a046433a 1381 s->commits.next = 0;
de167e41
FB
1382}
1383
a046433a
FB
1384static void schedule_rename(BDRVVVFATState* s,
1385 uint32_t cluster, char* new_path)
de167e41 1386{
c227f099 1387 commit_t* commit = array_get_next(&(s->commits));
a046433a
FB
1388 commit->path = new_path;
1389 commit->param.rename.cluster = cluster;
1390 commit->action = ACTION_RENAME;
de167e41
FB
1391}
1392
a046433a
FB
1393static void schedule_writeout(BDRVVVFATState* s,
1394 int dir_index, uint32_t modified_offset)
de167e41 1395{
c227f099 1396 commit_t* commit = array_get_next(&(s->commits));
a046433a
FB
1397 commit->path = NULL;
1398 commit->param.writeout.dir_index = dir_index;
1399 commit->param.writeout.modified_offset = modified_offset;
1400 commit->action = ACTION_WRITEOUT;
de167e41
FB
1401}
1402
a046433a
FB
1403static void schedule_new_file(BDRVVVFATState* s,
1404 char* path, uint32_t first_cluster)
de167e41 1405{
c227f099 1406 commit_t* commit = array_get_next(&(s->commits));
a046433a
FB
1407 commit->path = path;
1408 commit->param.new_file.first_cluster = first_cluster;
1409 commit->action = ACTION_NEW_FILE;
1410}
1411
1412static void schedule_mkdir(BDRVVVFATState* s, uint32_t cluster, char* path)
1413{
c227f099 1414 commit_t* commit = array_get_next(&(s->commits));
a046433a
FB
1415 commit->path = path;
1416 commit->param.mkdir.cluster = cluster;
1417 commit->action = ACTION_MKDIR;
1418}
1419
1420typedef struct {
64eaabda
TS
1421 /*
1422 * Since the sequence number is at most 0x3f, and the filename
1423 * length is at most 13 times the sequence number, the maximal
1424 * filename length is 0x3f * 13 bytes.
1425 */
1426 unsigned char name[0x3f * 13 + 1];
a046433a
FB
1427 int checksum, len;
1428 int sequence_number;
1429} long_file_name;
1430
1431static void lfn_init(long_file_name* lfn)
1432{
1433 lfn->sequence_number = lfn->len = 0;
1434 lfn->checksum = 0x100;
1435}
1436
1437/* return 0 if parsed successfully, > 0 if no long name, < 0 if error */
1438static int parse_long_name(long_file_name* lfn,
c227f099 1439 const direntry_t* direntry)
a046433a
FB
1440{
1441 int i, j, offset;
1442 const unsigned char* pointer = (const unsigned char*)direntry;
1443
1444 if (!is_long_name(direntry))
1445 return 1;
1446
1447 if (pointer[0] & 0x40) {
1448 lfn->sequence_number = pointer[0] & 0x3f;
1449 lfn->checksum = pointer[13];
1450 lfn->name[0] = 0;
59fdb018 1451 lfn->name[lfn->sequence_number * 13] = 0;
a046433a
FB
1452 } else if ((pointer[0] & 0x3f) != --lfn->sequence_number)
1453 return -1;
1454 else if (pointer[13] != lfn->checksum)
1455 return -2;
1456 else if (pointer[12] || pointer[26] || pointer[27])
1457 return -3;
1458
1459 offset = 13 * (lfn->sequence_number - 1);
1460 for (i = 0, j = 1; i < 13; i++, j+=2) {
1461 if (j == 11)
1462 j = 14;
1463 else if (j == 26)
1464 j = 28;
1465
1466 if (pointer[j+1] == 0)
1467 lfn->name[offset + i] = pointer[j];
1468 else if (pointer[j+1] != 0xff || (pointer[0] & 0x40) == 0)
1469 return -4;
1470 else
1471 lfn->name[offset + i] = 0;
de167e41 1472 }
a046433a
FB
1473
1474 if (pointer[0] & 0x40)
ffe8ab83 1475 lfn->len = offset + strlen((char*)lfn->name + offset);
a046433a 1476
de167e41
FB
1477 return 0;
1478}
1479
a046433a
FB
1480/* returns 0 if successful, >0 if no short_name, and <0 on error */
1481static int parse_short_name(BDRVVVFATState* s,
c227f099 1482 long_file_name* lfn, direntry_t* direntry)
de167e41 1483{
a046433a 1484 int i, j;
de167e41 1485
a046433a
FB
1486 if (!is_short_name(direntry))
1487 return 1;
1488
1489 for (j = 7; j >= 0 && direntry->name[j] == ' '; j--);
1490 for (i = 0; i <= j; i++) {
1491 if (direntry->name[i] <= ' ' || direntry->name[i] > 0x7f)
1492 return -1;
1493 else if (s->downcase_short_names)
47398b9c 1494 lfn->name[i] = qemu_tolower(direntry->name[i]);
a046433a
FB
1495 else
1496 lfn->name[i] = direntry->name[i];
de167e41
FB
1497 }
1498
a046433a
FB
1499 for (j = 2; j >= 0 && direntry->extension[j] == ' '; j--);
1500 if (j >= 0) {
1501 lfn->name[i++] = '.';
1502 lfn->name[i + j + 1] = '\0';
1503 for (;j >= 0; j--) {
1504 if (direntry->extension[j] <= ' ' || direntry->extension[j] > 0x7f)
1505 return -2;
1506 else if (s->downcase_short_names)
47398b9c 1507 lfn->name[i + j] = qemu_tolower(direntry->extension[j]);
a046433a
FB
1508 else
1509 lfn->name[i + j] = direntry->extension[j];
1510 }
1511 } else
1512 lfn->name[i + j + 1] = '\0';
1513
ffe8ab83 1514 lfn->len = strlen((char*)lfn->name);
a046433a
FB
1515
1516 return 0;
de167e41
FB
1517}
1518
a046433a
FB
1519static inline uint32_t modified_fat_get(BDRVVVFATState* s,
1520 unsigned int cluster)
de167e41 1521{
a046433a
FB
1522 if (cluster < s->last_cluster_of_root_directory) {
1523 if (cluster + 1 == s->last_cluster_of_root_directory)
1524 return s->max_fat_value;
1525 else
1526 return cluster + 1;
1527 }
1528
1529 if (s->fat_type==32) {
1530 uint32_t* entry=((uint32_t*)s->fat2)+cluster;
1531 return le32_to_cpu(*entry);
1532 } else if (s->fat_type==16) {
1533 uint16_t* entry=((uint16_t*)s->fat2)+cluster;
1534 return le16_to_cpu(*entry);
1535 } else {
1536 const uint8_t* x=s->fat2+cluster*3/2;
1537 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
1538 }
1539}
1540
1541static inline int cluster_was_modified(BDRVVVFATState* s, uint32_t cluster_num)
1542{
1543 int was_modified = 0;
1544 int i, dummy;
1545
1546 if (s->qcow == NULL)
de167e41 1547 return 0;
a046433a
FB
1548
1549 for (i = 0; !was_modified && i < s->sectors_per_cluster; i++)
1550 was_modified = s->qcow->drv->bdrv_is_allocated(s->qcow,
1551 cluster2sector(s, cluster_num) + i, 1, &dummy);
1552
1553 return was_modified;
de167e41
FB
1554}
1555
a046433a 1556static const char* get_basename(const char* path)
de167e41 1557{
a046433a
FB
1558 char* basename = strrchr(path, '/');
1559 if (basename == NULL)
1560 return path;
1561 else
1562 return basename + 1; /* strip '/' */
de167e41
FB
1563}
1564
a046433a
FB
1565/*
1566 * The array s->used_clusters holds the states of the clusters. If it is
1567 * part of a file, it has bit 2 set, in case of a directory, bit 1. If it
1568 * was modified, bit 3 is set.
1569 * If any cluster is allocated, but not part of a file or directory, this
1570 * driver refuses to commit.
1571 */
1572typedef enum {
1573 USED_DIRECTORY = 1, USED_FILE = 2, USED_ANY = 3, USED_ALLOCATED = 4
c227f099 1574} used_t;
de167e41 1575
a046433a
FB
1576/*
1577 * get_cluster_count_for_direntry() not only determines how many clusters
1578 * are occupied by direntry, but also if it was renamed or modified.
1579 *
1580 * A file is thought to be renamed *only* if there already was a file with
1581 * exactly the same first cluster, but a different name.
1582 *
1583 * Further, the files/directories handled by this function are
1584 * assumed to be *not* deleted (and *only* those).
1585 */
1586static uint32_t get_cluster_count_for_direntry(BDRVVVFATState* s,
c227f099 1587 direntry_t* direntry, const char* path)
de167e41 1588{
a046433a
FB
1589 /*
1590 * This is a little bit tricky:
1591 * IF the guest OS just inserts a cluster into the file chain,
1592 * and leaves the rest alone, (i.e. the original file had clusters
1593 * 15 -> 16, but now has 15 -> 32 -> 16), then the following happens:
1594 *
1595 * - do_commit will write the cluster into the file at the given
1596 * offset, but
1597 *
1598 * - the cluster which is overwritten should be moved to a later
1599 * position in the file.
1600 *
1601 * I am not aware that any OS does something as braindead, but this
1602 * situation could happen anyway when not committing for a long time.
1603 * Just to be sure that this does not bite us, detect it, and copy the
1604 * contents of the clusters to-be-overwritten into the qcow.
1605 */
1606 int copy_it = 0;
1607 int was_modified = 0;
1608 int32_t ret = 0;
1609
1610 uint32_t cluster_num = begin_of_direntry(direntry);
1611 uint32_t offset = 0;
1612 int first_mapping_index = -1;
c227f099 1613 mapping_t* mapping = NULL;
a046433a 1614 const char* basename2 = NULL;
de167e41 1615
a046433a 1616 vvfat_close_current_file(s);
de167e41 1617
a046433a
FB
1618 /* the root directory */
1619 if (cluster_num == 0)
de167e41 1620 return 0;
de167e41 1621
a046433a
FB
1622 /* write support */
1623 if (s->qcow) {
1624 basename2 = get_basename(path);
de167e41 1625
a046433a
FB
1626 mapping = find_mapping_for_cluster(s, cluster_num);
1627
1628 if (mapping) {
da2414e9
FB
1629 const char* basename;
1630
a046433a
FB
1631 assert(mapping->mode & MODE_DELETED);
1632 mapping->mode &= ~MODE_DELETED;
1633
da2414e9 1634 basename = get_basename(mapping->path);
a046433a
FB
1635
1636 assert(mapping->mode & MODE_NORMAL);
1637
1638 /* rename */
1639 if (strcmp(basename, basename2))
6265eb26 1640 schedule_rename(s, cluster_num, qemu_strdup(path));
a046433a
FB
1641 } else if (is_file(direntry))
1642 /* new file */
6265eb26 1643 schedule_new_file(s, qemu_strdup(path), cluster_num);
a046433a 1644 else {
43dc2a64 1645 abort();
a046433a
FB
1646 return 0;
1647 }
de167e41
FB
1648 }
1649
a046433a
FB
1650 while(1) {
1651 if (s->qcow) {
1652 if (!copy_it && cluster_was_modified(s, cluster_num)) {
1653 if (mapping == NULL ||
1654 mapping->begin > cluster_num ||
1655 mapping->end <= cluster_num)
1656 mapping = find_mapping_for_cluster(s, cluster_num);
de167e41 1657
a046433a
FB
1658
1659 if (mapping &&
1660 (mapping->mode & MODE_DIRECTORY) == 0) {
1661
1662 /* was modified in qcow */
1663 if (offset != mapping->info.file.offset + s->cluster_size
1664 * (cluster_num - mapping->begin)) {
1665 /* offset of this cluster in file chain has changed */
43dc2a64 1666 abort();
a046433a
FB
1667 copy_it = 1;
1668 } else if (offset == 0) {
1669 const char* basename = get_basename(mapping->path);
1670
1671 if (strcmp(basename, basename2))
1672 copy_it = 1;
1673 first_mapping_index = array_index(&(s->mapping), mapping);
1674 }
1675
1676 if (mapping->first_mapping_index != first_mapping_index
1677 && mapping->info.file.offset > 0) {
43dc2a64 1678 abort();
a046433a
FB
1679 copy_it = 1;
1680 }
1681
1682 /* need to write out? */
1683 if (!was_modified && is_file(direntry)) {
1684 was_modified = 1;
1685 schedule_writeout(s, mapping->dir_index, offset);
1686 }
1687 }
1688 }
1689
1690 if (copy_it) {
1691 int i, dummy;
1692 /*
1693 * This is horribly inefficient, but that is okay, since
1694 * it is rarely executed, if at all.
1695 */
1696 int64_t offset = cluster2sector(s, cluster_num);
1697
1698 vvfat_close_current_file(s);
1699 for (i = 0; i < s->sectors_per_cluster; i++)
1700 if (!s->qcow->drv->bdrv_is_allocated(s->qcow,
1701 offset + i, 1, &dummy)) {
1702 if (vvfat_read(s->bs,
1703 offset, s->cluster_buffer, 1))
1704 return -1;
1705 if (s->qcow->drv->bdrv_write(s->qcow,
1706 offset, s->cluster_buffer, 1))
1707 return -2;
1708 }
1709 }
1710 }
1711
1712 ret++;
1713 if (s->used_clusters[cluster_num] & USED_ANY)
1714 return 0;
1715 s->used_clusters[cluster_num] = USED_FILE;
1716
1717 cluster_num = modified_fat_get(s, cluster_num);
1718
1719 if (fat_eof(s, cluster_num))
1720 return ret;
1721 else if (cluster_num < 2 || cluster_num > s->max_fat_value - 16)
1722 return -1;
1723
1724 offset += s->cluster_size;
1725 }
de167e41
FB
1726}
1727
a046433a 1728/*
5fafdf24 1729 * This function looks at the modified data (qcow).
a046433a
FB
1730 * It returns 0 upon inconsistency or error, and the number of clusters
1731 * used by the directory, its subdirectories and their files.
1732 */
1733static int check_directory_consistency(BDRVVVFATState *s,
1734 int cluster_num, const char* path)
de167e41 1735{
a046433a 1736 int ret = 0;
3ec88e80 1737 unsigned char* cluster = qemu_malloc(s->cluster_size);
c227f099
AL
1738 direntry_t* direntries = (direntry_t*)cluster;
1739 mapping_t* mapping = find_mapping_for_cluster(s, cluster_num);
a046433a
FB
1740
1741 long_file_name lfn;
1742 int path_len = strlen(path);
1743 char path2[PATH_MAX];
1744
1745 assert(path_len < PATH_MAX); /* len was tested before! */
363a37d5 1746 pstrcpy(path2, sizeof(path2), path);
a046433a
FB
1747 path2[path_len] = '/';
1748 path2[path_len + 1] = '\0';
1749
1750 if (mapping) {
1751 const char* basename = get_basename(mapping->path);
1752 const char* basename2 = get_basename(path);
1753
1754 assert(mapping->mode & MODE_DIRECTORY);
1755
1756 assert(mapping->mode & MODE_DELETED);
1757 mapping->mode &= ~MODE_DELETED;
1758
1759 if (strcmp(basename, basename2))
6265eb26 1760 schedule_rename(s, cluster_num, qemu_strdup(path));
a046433a
FB
1761 } else
1762 /* new directory */
6265eb26 1763 schedule_mkdir(s, cluster_num, qemu_strdup(path));
3b46e624 1764
a046433a
FB
1765 lfn_init(&lfn);
1766 do {
de167e41 1767 int i;
a046433a
FB
1768 int subret = 0;
1769
1770 ret++;
1771
1772 if (s->used_clusters[cluster_num] & USED_ANY) {
1773 fprintf(stderr, "cluster %d used more than once\n", (int)cluster_num);
1774 return 0;
1775 }
1776 s->used_clusters[cluster_num] = USED_DIRECTORY;
1777
1778DLOG(fprintf(stderr, "read cluster %d (sector %d)\n", (int)cluster_num, (int)cluster2sector(s, cluster_num)));
1779 subret = vvfat_read(s->bs, cluster2sector(s, cluster_num), cluster,
1780 s->sectors_per_cluster);
1781 if (subret) {
1782 fprintf(stderr, "Error fetching direntries\n");
1783 fail:
1784 free(cluster);
1785 return 0;
1786 }
1787
1788 for (i = 0; i < 0x10 * s->sectors_per_cluster; i++) {
3f4cb3d3 1789 int cluster_count = 0;
a046433a
FB
1790
1791DLOG(fprintf(stderr, "check direntry %d: \n", i); print_direntry(direntries + i));
1792 if (is_volume_label(direntries + i) || is_dot(direntries + i) ||
1793 is_free(direntries + i))
1794 continue;
1795
1796 subret = parse_long_name(&lfn, direntries + i);
1797 if (subret < 0) {
1798 fprintf(stderr, "Error in long name\n");
1799 goto fail;
de167e41 1800 }
a046433a
FB
1801 if (subret == 0 || is_free(direntries + i))
1802 continue;
1803
1804 if (fat_chksum(direntries+i) != lfn.checksum) {
1805 subret = parse_short_name(s, &lfn, direntries + i);
1806 if (subret < 0) {
1807 fprintf(stderr, "Error in short name (%d)\n", subret);
1808 goto fail;
1809 }
ffe8ab83
TS
1810 if (subret > 0 || !strcmp((char*)lfn.name, ".")
1811 || !strcmp((char*)lfn.name, ".."))
a046433a
FB
1812 continue;
1813 }
1814 lfn.checksum = 0x100; /* cannot use long name twice */
1815
1816 if (path_len + 1 + lfn.len >= PATH_MAX) {
1817 fprintf(stderr, "Name too long: %s/%s\n", path, lfn.name);
1818 goto fail;
1819 }
363a37d5
BS
1820 pstrcpy(path2 + path_len + 1, sizeof(path2) - path_len - 1,
1821 (char*)lfn.name);
a046433a
FB
1822
1823 if (is_directory(direntries + i)) {
1824 if (begin_of_direntry(direntries + i) == 0) {
1825 DLOG(fprintf(stderr, "invalid begin for directory: %s\n", path2); print_direntry(direntries + i));
1826 goto fail;
1827 }
1828 cluster_count = check_directory_consistency(s,
1829 begin_of_direntry(direntries + i), path2);
1830 if (cluster_count == 0) {
1831 DLOG(fprintf(stderr, "problem in directory %s:\n", path2); print_direntry(direntries + i));
1832 goto fail;
1833 }
1834 } else if (is_file(direntries + i)) {
1835 /* check file size with FAT */
1836 cluster_count = get_cluster_count_for_direntry(s, direntries + i, path2);
1837 if (cluster_count !=
1838 (le32_to_cpu(direntries[i].size) + s->cluster_size
1839 - 1) / s->cluster_size) {
1840 DLOG(fprintf(stderr, "Cluster count mismatch\n"));
1841 goto fail;
1842 }
1843 } else
43dc2a64 1844 abort(); /* cluster_count = 0; */
a046433a
FB
1845
1846 ret += cluster_count;
de167e41 1847 }
de167e41 1848
a046433a
FB
1849 cluster_num = modified_fat_get(s, cluster_num);
1850 } while(!fat_eof(s, cluster_num));
de167e41 1851
a046433a
FB
1852 free(cluster);
1853 return ret;
1854}
1855
1856/* returns 1 on success */
1857static int is_consistent(BDRVVVFATState* s)
1858{
1859 int i, check;
1860 int used_clusters_count = 0;
1861
1862DLOG(checkpoint());
1863 /*
1864 * - get modified FAT
1865 * - compare the two FATs (TODO)
1866 * - get buffer for marking used clusters
1867 * - recurse direntries from root (using bs->bdrv_read to make
1868 * sure to get the new data)
1869 * - check that the FAT agrees with the size
1870 * - count the number of clusters occupied by this directory and
1871 * its files
1872 * - check that the cumulative used cluster count agrees with the
1873 * FAT
1874 * - if all is fine, return number of used clusters
1875 */
1876 if (s->fat2 == NULL) {
1877 int size = 0x200 * s->sectors_per_fat;
3ec88e80 1878 s->fat2 = qemu_malloc(size);
a046433a
FB
1879 memcpy(s->fat2, s->fat.pointer, size);
1880 }
1881 check = vvfat_read(s->bs,
1882 s->first_sectors_number, s->fat2, s->sectors_per_fat);
1883 if (check) {
1884 fprintf(stderr, "Could not copy fat\n");
1885 return 0;
1886 }
1887 assert (s->used_clusters);
1888 for (i = 0; i < sector2cluster(s, s->sector_count); i++)
1889 s->used_clusters[i] &= ~USED_ANY;
1890
1891 clear_commits(s);
1892
1893 /* mark every mapped file/directory as deleted.
1894 * (check_directory_consistency() will unmark those still present). */
1895 if (s->qcow)
1896 for (i = 0; i < s->mapping.next; i++) {
c227f099 1897 mapping_t* mapping = array_get(&(s->mapping), i);
a046433a
FB
1898 if (mapping->first_mapping_index < 0)
1899 mapping->mode |= MODE_DELETED;
de167e41 1900 }
a046433a
FB
1901
1902 used_clusters_count = check_directory_consistency(s, 0, s->path);
1903 if (used_clusters_count <= 0) {
1904 DLOG(fprintf(stderr, "problem in directory\n"));
1905 return 0;
de167e41
FB
1906 }
1907
a046433a
FB
1908 check = s->last_cluster_of_root_directory;
1909 for (i = check; i < sector2cluster(s, s->sector_count); i++) {
1910 if (modified_fat_get(s, i)) {
1911 if(!s->used_clusters[i]) {
1912 DLOG(fprintf(stderr, "FAT was modified (%d), but cluster is not used?\n", i));
1913 return 0;
1914 }
1915 check++;
1916 }
1917
1918 if (s->used_clusters[i] == USED_ALLOCATED) {
1919 /* allocated, but not used... */
1920 DLOG(fprintf(stderr, "unused, modified cluster: %d\n", i));
1921 return 0;
1922 }
1923 }
1924
1925 if (check != used_clusters_count)
1926 return 0;
1927
1928 return used_clusters_count;
1929}
1930
1931static inline void adjust_mapping_indices(BDRVVVFATState* s,
1932 int offset, int adjust)
1933{
1934 int i;
1935
1936 for (i = 0; i < s->mapping.next; i++) {
c227f099 1937 mapping_t* mapping = array_get(&(s->mapping), i);
a046433a
FB
1938
1939#define ADJUST_MAPPING_INDEX(name) \
1940 if (mapping->name >= offset) \
1941 mapping->name += adjust
1942
1943 ADJUST_MAPPING_INDEX(first_mapping_index);
1944 if (mapping->mode & MODE_DIRECTORY)
1945 ADJUST_MAPPING_INDEX(info.dir.parent_mapping_index);
de167e41 1946 }
a046433a
FB
1947}
1948
1949/* insert or update mapping */
c227f099 1950static mapping_t* insert_mapping(BDRVVVFATState* s,
a046433a
FB
1951 uint32_t begin, uint32_t end)
1952{
1953 /*
1954 * - find mapping where mapping->begin >= begin,
1955 * - if mapping->begin > begin: insert
1956 * - adjust all references to mappings!
1957 * - else: adjust
1958 * - replace name
1959 */
1960 int index = find_mapping_for_cluster_aux(s, begin, 0, s->mapping.next);
c227f099
AL
1961 mapping_t* mapping = NULL;
1962 mapping_t* first_mapping = array_get(&(s->mapping), 0);
a046433a
FB
1963
1964 if (index < s->mapping.next && (mapping = array_get(&(s->mapping), index))
1965 && mapping->begin < begin) {
1966 mapping->end = begin;
1967 index++;
1968 mapping = array_get(&(s->mapping), index);
1969 }
1970 if (index >= s->mapping.next || mapping->begin > begin) {
1971 mapping = array_insert(&(s->mapping), index, 1);
1972 mapping->path = NULL;
1973 adjust_mapping_indices(s, index, +1);
1974 }
1975
1976 mapping->begin = begin;
1977 mapping->end = end;
de167e41 1978
c227f099 1979DLOG(mapping_t* next_mapping;
a046433a
FB
1980assert(index + 1 >= s->mapping.next ||
1981((next_mapping = array_get(&(s->mapping), index + 1)) &&
1982 next_mapping->begin >= end)));
1983
c227f099 1984 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
a046433a
FB
1985 s->current_mapping = array_get(&(s->mapping),
1986 s->current_mapping - first_mapping);
1987
1988 return mapping;
1989}
1990
1991static int remove_mapping(BDRVVVFATState* s, int mapping_index)
1992{
c227f099
AL
1993 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
1994 mapping_t* first_mapping = array_get(&(s->mapping), 0);
a046433a
FB
1995
1996 /* free mapping */
1997 if (mapping->first_mapping_index < 0)
1998 free(mapping->path);
1999
2000 /* remove from s->mapping */
2001 array_remove(&(s->mapping), mapping_index);
2002
2003 /* adjust all references to mappings */
2004 adjust_mapping_indices(s, mapping_index, -1);
2005
c227f099 2006 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
a046433a
FB
2007 s->current_mapping = array_get(&(s->mapping),
2008 s->current_mapping - first_mapping);
de167e41 2009
de167e41
FB
2010 return 0;
2011}
2012
a046433a
FB
2013static void adjust_dirindices(BDRVVVFATState* s, int offset, int adjust)
2014{
2015 int i;
2016 for (i = 0; i < s->mapping.next; i++) {
c227f099 2017 mapping_t* mapping = array_get(&(s->mapping), i);
a046433a
FB
2018 if (mapping->dir_index >= offset)
2019 mapping->dir_index += adjust;
2020 if ((mapping->mode & MODE_DIRECTORY) &&
2021 mapping->info.dir.first_dir_index >= offset)
2022 mapping->info.dir.first_dir_index += adjust;
2023 }
2024}
de167e41 2025
c227f099 2026static direntry_t* insert_direntries(BDRVVVFATState* s,
a046433a 2027 int dir_index, int count)
de167e41 2028{
a046433a
FB
2029 /*
2030 * make room in s->directory,
2031 * adjust_dirindices
2032 */
c227f099 2033 direntry_t* result = array_insert(&(s->directory), dir_index, count);
a046433a
FB
2034 if (result == NULL)
2035 return NULL;
2036 adjust_dirindices(s, dir_index, count);
de167e41
FB
2037 return result;
2038}
2039
a046433a
FB
2040static int remove_direntries(BDRVVVFATState* s, int dir_index, int count)
2041{
2042 int ret = array_remove_slice(&(s->directory), dir_index, count);
2043 if (ret)
2044 return ret;
2045 adjust_dirindices(s, dir_index, -count);
2046 return 0;
2047}
de167e41 2048
a046433a
FB
2049/*
2050 * Adapt the mappings of the cluster chain starting at first cluster
2051 * (i.e. if a file starts at first_cluster, the chain is followed according
2052 * to the modified fat, and the corresponding entries in s->mapping are
2053 * adjusted)
2054 */
2055static int commit_mappings(BDRVVVFATState* s,
2056 uint32_t first_cluster, int dir_index)
de167e41 2057{
c227f099
AL
2058 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2059 direntry_t* direntry = array_get(&(s->directory), dir_index);
a046433a
FB
2060 uint32_t cluster = first_cluster;
2061
2062 vvfat_close_current_file(s);
2063
2064 assert(mapping);
2065 assert(mapping->begin == first_cluster);
2066 mapping->first_mapping_index = -1;
2067 mapping->dir_index = dir_index;
2068 mapping->mode = (dir_index <= 0 || is_directory(direntry)) ?
2069 MODE_DIRECTORY : MODE_NORMAL;
2070
2071 while (!fat_eof(s, cluster)) {
2072 uint32_t c, c1;
2073
2074 for (c = cluster, c1 = modified_fat_get(s, c); c + 1 == c1;
2075 c = c1, c1 = modified_fat_get(s, c1));
2076
2077 c++;
2078 if (c > mapping->end) {
2079 int index = array_index(&(s->mapping), mapping);
2080 int i, max_i = s->mapping.next - index;
2081 for (i = 1; i < max_i && mapping[i].begin < c; i++);
2082 while (--i > 0)
2083 remove_mapping(s, index + 1);
2084 }
2085 assert(mapping == array_get(&(s->mapping), s->mapping.next - 1)
2086 || mapping[1].begin >= c);
2087 mapping->end = c;
2088
2089 if (!fat_eof(s, c1)) {
2090 int i = find_mapping_for_cluster_aux(s, c1, 0, s->mapping.next);
c227f099 2091 mapping_t* next_mapping = i >= s->mapping.next ? NULL :
a046433a
FB
2092 array_get(&(s->mapping), i);
2093
2094 if (next_mapping == NULL || next_mapping->begin > c1) {
2095 int i1 = array_index(&(s->mapping), mapping);
2096
2097 next_mapping = insert_mapping(s, c1, c1+1);
2098
2099 if (c1 < c)
2100 i1++;
2101 mapping = array_get(&(s->mapping), i1);
2102 }
2103
2104 next_mapping->dir_index = mapping->dir_index;
5fafdf24 2105 next_mapping->first_mapping_index =
a046433a
FB
2106 mapping->first_mapping_index < 0 ?
2107 array_index(&(s->mapping), mapping) :
2108 mapping->first_mapping_index;
2109 next_mapping->path = mapping->path;
2110 next_mapping->mode = mapping->mode;
2111 next_mapping->read_only = mapping->read_only;
2112 if (mapping->mode & MODE_DIRECTORY) {
2113 next_mapping->info.dir.parent_mapping_index =
2114 mapping->info.dir.parent_mapping_index;
2115 next_mapping->info.dir.first_dir_index =
2116 mapping->info.dir.first_dir_index +
2117 0x10 * s->sectors_per_cluster *
2118 (mapping->end - mapping->begin);
2119 } else
2120 next_mapping->info.file.offset = mapping->info.file.offset +
2121 mapping->end - mapping->begin;
2122
2123 mapping = next_mapping;
2124 }
3b46e624 2125
a046433a
FB
2126 cluster = c1;
2127 }
de167e41 2128
de167e41
FB
2129 return 0;
2130}
2131
a046433a
FB
2132static int commit_direntries(BDRVVVFATState* s,
2133 int dir_index, int parent_mapping_index)
de167e41 2134{
c227f099 2135 direntry_t* direntry = array_get(&(s->directory), dir_index);
a046433a 2136 uint32_t first_cluster = dir_index == 0 ? 0 : begin_of_direntry(direntry);
c227f099 2137 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
a046433a
FB
2138
2139 int factor = 0x10 * s->sectors_per_cluster;
2140 int old_cluster_count, new_cluster_count;
2141 int current_dir_index = mapping->info.dir.first_dir_index;
2142 int first_dir_index = current_dir_index;
2143 int ret, i;
2144 uint32_t c;
2145
2146DLOG(fprintf(stderr, "commit_direntries for %s, parent_mapping_index %d\n", mapping->path, parent_mapping_index));
2147
2148 assert(direntry);
2149 assert(mapping);
2150 assert(mapping->begin == first_cluster);
2151 assert(mapping->info.dir.first_dir_index < s->directory.next);
2152 assert(mapping->mode & MODE_DIRECTORY);
2153 assert(dir_index == 0 || is_directory(direntry));
2154
2155 mapping->info.dir.parent_mapping_index = parent_mapping_index;
2156
2157 if (first_cluster == 0) {
2158 old_cluster_count = new_cluster_count =
2159 s->last_cluster_of_root_directory;
2160 } else {
2161 for (old_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2162 c = fat_get(s, c))
2163 old_cluster_count++;
de167e41 2164
a046433a
FB
2165 for (new_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2166 c = modified_fat_get(s, c))
2167 new_cluster_count++;
2168 }
de167e41 2169
a046433a
FB
2170 if (new_cluster_count > old_cluster_count) {
2171 if (insert_direntries(s,
2172 current_dir_index + factor * old_cluster_count,
2173 factor * (new_cluster_count - old_cluster_count)) == NULL)
2174 return -1;
2175 } else if (new_cluster_count < old_cluster_count)
2176 remove_direntries(s,
2177 current_dir_index + factor * new_cluster_count,
2178 factor * (old_cluster_count - new_cluster_count));
2179
2180 for (c = first_cluster; !fat_eof(s, c); c = modified_fat_get(s, c)) {
2181 void* direntry = array_get(&(s->directory), current_dir_index);
2182 int ret = vvfat_read(s->bs, cluster2sector(s, c), direntry,
2183 s->sectors_per_cluster);
2184 if (ret)
2185 return ret;
2186 assert(!strncmp(s->directory.pointer, "QEMU", 4));
2187 current_dir_index += factor;
2188 }
de167e41 2189
a046433a
FB
2190 ret = commit_mappings(s, first_cluster, dir_index);
2191 if (ret)
2192 return ret;
2193
2194 /* recurse */
2195 for (i = 0; i < factor * new_cluster_count; i++) {
2196 direntry = array_get(&(s->directory), first_dir_index + i);
2197 if (is_directory(direntry) && !is_dot(direntry)) {
2198 mapping = find_mapping_for_cluster(s, first_cluster);
2199 assert(mapping->mode & MODE_DIRECTORY);
2200 ret = commit_direntries(s, first_dir_index + i,
2201 array_index(&(s->mapping), mapping));
2202 if (ret)
2203 return ret;
2204 }
2205 }
de167e41 2206
a046433a
FB
2207 return 0;
2208}
de167e41 2209
a046433a
FB
2210/* commit one file (adjust contents, adjust mapping),
2211 return first_mapping_index */
2212static int commit_one_file(BDRVVVFATState* s,
2213 int dir_index, uint32_t offset)
2214{
c227f099 2215 direntry_t* direntry = array_get(&(s->directory), dir_index);
a046433a
FB
2216 uint32_t c = begin_of_direntry(direntry);
2217 uint32_t first_cluster = c;
c227f099 2218 mapping_t* mapping = find_mapping_for_cluster(s, c);
a046433a 2219 uint32_t size = filesize_of_direntry(direntry);
3ec88e80 2220 char* cluster = qemu_malloc(s->cluster_size);
a046433a
FB
2221 uint32_t i;
2222 int fd = 0;
2223
2224 assert(offset < size);
2225 assert((offset % s->cluster_size) == 0);
2226
2227 for (i = s->cluster_size; i < offset; i += s->cluster_size)
2228 c = modified_fat_get(s, c);
2229
6bcb76c3 2230 fd = open(mapping->path, O_RDWR | O_CREAT | O_BINARY, 0666);
a046433a
FB
2231 if (fd < 0) {
2232 fprintf(stderr, "Could not open %s... (%s, %d)\n", mapping->path,
2233 strerror(errno), errno);
2234 return fd;
de167e41 2235 }
a046433a
FB
2236 if (offset > 0)
2237 if (lseek(fd, offset, SEEK_SET) != offset)
2238 return -3;
2239
2240 while (offset < size) {
2241 uint32_t c1;
2242 int rest_size = (size - offset > s->cluster_size ?
2243 s->cluster_size : size - offset);
2244 int ret;
2245
2246 c1 = modified_fat_get(s, c);
2247
2248 assert((size - offset == 0 && fat_eof(s, c)) ||
2249 (size > offset && c >=2 && !fat_eof(s, c)));
a046433a
FB
2250
2251 ret = vvfat_read(s->bs, cluster2sector(s, c),
ffe8ab83 2252 (uint8_t*)cluster, (rest_size + 0x1ff) / 0x200);
a046433a
FB
2253
2254 if (ret < 0)
2255 return ret;
2256
2257 if (write(fd, cluster, rest_size) < 0)
2258 return -2;
2259
2260 offset += rest_size;
2261 c = c1;
2262 }
2263
2dedf83e
KS
2264 if (ftruncate(fd, size)) {
2265 perror("ftruncate()");
2266 close(fd);
2267 return -4;
2268 }
a046433a
FB
2269 close(fd);
2270
2271 return commit_mappings(s, first_cluster, dir_index);
2272}
2273
2274#ifdef DEBUG
2275/* test, if all mappings point to valid direntries */
2276static void check1(BDRVVVFATState* s)
2277{
2278 int i;
2279 for (i = 0; i < s->mapping.next; i++) {
c227f099 2280 mapping_t* mapping = array_get(&(s->mapping), i);
a046433a
FB
2281 if (mapping->mode & MODE_DELETED) {
2282 fprintf(stderr, "deleted\n");
2283 continue;
2284 }
2285 assert(mapping->dir_index >= 0);
2286 assert(mapping->dir_index < s->directory.next);
c227f099 2287 direntry_t* direntry = array_get(&(s->directory), mapping->dir_index);
a046433a
FB
2288 assert(mapping->begin == begin_of_direntry(direntry) || mapping->first_mapping_index >= 0);
2289 if (mapping->mode & MODE_DIRECTORY) {
2290 assert(mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster * (mapping->end - mapping->begin) <= s->directory.next);
2291 assert((mapping->info.dir.first_dir_index % (0x10 * s->sectors_per_cluster)) == 0);
de167e41
FB
2292 }
2293 }
de167e41
FB
2294}
2295
a046433a
FB
2296/* test, if all direntries have mappings */
2297static void check2(BDRVVVFATState* s)
de167e41 2298{
de167e41 2299 int i;
a046433a 2300 int first_mapping = -1;
de167e41 2301
a046433a 2302 for (i = 0; i < s->directory.next; i++) {
c227f099 2303 direntry_t* direntry = array_get(&(s->directory), i);
de167e41 2304
a046433a 2305 if (is_short_name(direntry) && begin_of_direntry(direntry)) {
c227f099 2306 mapping_t* mapping = find_mapping_for_cluster(s, begin_of_direntry(direntry));
a046433a
FB
2307 assert(mapping);
2308 assert(mapping->dir_index == i || is_dot(direntry));
2309 assert(mapping->begin == begin_of_direntry(direntry) || is_dot(direntry));
2310 }
de167e41 2311
a046433a
FB
2312 if ((i % (0x10 * s->sectors_per_cluster)) == 0) {
2313 /* cluster start */
2314 int j, count = 0;
de167e41 2315
a046433a 2316 for (j = 0; j < s->mapping.next; j++) {
c227f099 2317 mapping_t* mapping = array_get(&(s->mapping), j);
a046433a 2318 if (mapping->mode & MODE_DELETED)
de167e41 2319 continue;
a046433a
FB
2320 if (mapping->mode & MODE_DIRECTORY) {
2321 if (mapping->info.dir.first_dir_index <= i && mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster > i) {
2322 assert(++count == 1);
2323 if (mapping->first_mapping_index == -1)
2324 first_mapping = array_index(&(s->mapping), mapping);
2325 else
2326 assert(first_mapping == mapping->first_mapping_index);
2327 if (mapping->info.dir.parent_mapping_index < 0)
2328 assert(j == 0);
2329 else {
c227f099 2330 mapping_t* parent = array_get(&(s->mapping), mapping->info.dir.parent_mapping_index);
a046433a
FB
2331 assert(parent->mode & MODE_DIRECTORY);
2332 assert(parent->info.dir.first_dir_index < mapping->info.dir.first_dir_index);
2333 }
2334 }
de167e41 2335 }
a046433a
FB
2336 }
2337 if (count == 0)
2338 first_mapping = -1;
2339 }
2340 }
2341}
2342#endif
de167e41 2343
a046433a
FB
2344static int handle_renames_and_mkdirs(BDRVVVFATState* s)
2345{
2346 int i;
de167e41 2347
a046433a
FB
2348#ifdef DEBUG
2349 fprintf(stderr, "handle_renames\n");
2350 for (i = 0; i < s->commits.next; i++) {
c227f099 2351 commit_t* commit = array_get(&(s->commits), i);
a046433a
FB
2352 fprintf(stderr, "%d, %s (%d, %d)\n", i, commit->path ? commit->path : "(null)", commit->param.rename.cluster, commit->action);
2353 }
2354#endif
2355
2356 for (i = 0; i < s->commits.next;) {
c227f099 2357 commit_t* commit = array_get(&(s->commits), i);
a046433a 2358 if (commit->action == ACTION_RENAME) {
c227f099 2359 mapping_t* mapping = find_mapping_for_cluster(s,
a046433a
FB
2360 commit->param.rename.cluster);
2361 char* old_path = mapping->path;
2362
2363 assert(commit->path);
2364 mapping->path = commit->path;
2365 if (rename(old_path, mapping->path))
2366 return -2;
2367
2368 if (mapping->mode & MODE_DIRECTORY) {
2369 int l1 = strlen(mapping->path);
2370 int l2 = strlen(old_path);
2371 int diff = l1 - l2;
c227f099 2372 direntry_t* direntry = array_get(&(s->directory),
a046433a
FB
2373 mapping->info.dir.first_dir_index);
2374 uint32_t c = mapping->begin;
2375 int i = 0;
2376
2377 /* recurse */
2378 while (!fat_eof(s, c)) {
2379 do {
c227f099 2380 direntry_t* d = direntry + i;
a046433a
FB
2381
2382 if (is_file(d) || (is_directory(d) && !is_dot(d))) {
c227f099 2383 mapping_t* m = find_mapping_for_cluster(s,
a046433a
FB
2384 begin_of_direntry(d));
2385 int l = strlen(m->path);
3ec88e80 2386 char* new_path = qemu_malloc(l + diff + 1);
a046433a
FB
2387
2388 assert(!strncmp(m->path, mapping->path, l2));
2389
363a37d5
BS
2390 pstrcpy(new_path, l + diff + 1, mapping->path);
2391 pstrcpy(new_path + l1, l + diff + 1 - l1,
2392 m->path + l2);
a046433a
FB
2393
2394 schedule_rename(s, m->begin, new_path);
de167e41 2395 }
a046433a
FB
2396 i++;
2397 } while((i % (0x10 * s->sectors_per_cluster)) != 0);
2398 c = fat_get(s, c);
de167e41
FB
2399 }
2400 }
de167e41 2401
a046433a
FB
2402 free(old_path);
2403 array_remove(&(s->commits), i);
2404 continue;
2405 } else if (commit->action == ACTION_MKDIR) {
c227f099 2406 mapping_t* mapping;
a046433a
FB
2407 int j, parent_path_len;
2408
48c2f068
FB
2409#ifdef __MINGW32__
2410 if (mkdir(commit->path))
2411 return -5;
2412#else
2413 if (mkdir(commit->path, 0755))
2414 return -5;
2415#endif
a046433a
FB
2416
2417 mapping = insert_mapping(s, commit->param.mkdir.cluster,
2418 commit->param.mkdir.cluster + 1);
2419 if (mapping == NULL)
2420 return -6;
2421
2422 mapping->mode = MODE_DIRECTORY;
2423 mapping->read_only = 0;
2424 mapping->path = commit->path;
2425 j = s->directory.next;
2426 assert(j);
2427 insert_direntries(s, s->directory.next,
2428 0x10 * s->sectors_per_cluster);
2429 mapping->info.dir.first_dir_index = j;
2430
2431 parent_path_len = strlen(commit->path)
2432 - strlen(get_basename(commit->path)) - 1;
2433 for (j = 0; j < s->mapping.next; j++) {
c227f099 2434 mapping_t* m = array_get(&(s->mapping), j);
a046433a
FB
2435 if (m->first_mapping_index < 0 && m != mapping &&
2436 !strncmp(m->path, mapping->path, parent_path_len) &&
2437 strlen(m->path) == parent_path_len)
2438 break;
2439 }
2440 assert(j < s->mapping.next);
2441 mapping->info.dir.parent_mapping_index = j;
2442
2443 array_remove(&(s->commits), i);
2444 continue;
2445 }
2446
2447 i++;
2448 }
2449 return 0;
2450}
2451
2452/*
2453 * TODO: make sure that the short name is not matching *another* file
2454 */
2455static int handle_commits(BDRVVVFATState* s)
2456{
2457 int i, fail = 0;
2458
2459 vvfat_close_current_file(s);
2460
2461 for (i = 0; !fail && i < s->commits.next; i++) {
c227f099 2462 commit_t* commit = array_get(&(s->commits), i);
a046433a
FB
2463 switch(commit->action) {
2464 case ACTION_RENAME: case ACTION_MKDIR:
43dc2a64 2465 abort();
a046433a
FB
2466 fail = -2;
2467 break;
2468 case ACTION_WRITEOUT: {
a6c6f76c
BS
2469#ifndef NDEBUG
2470 /* these variables are only used by assert() below */
c227f099 2471 direntry_t* entry = array_get(&(s->directory),
a046433a
FB
2472 commit->param.writeout.dir_index);
2473 uint32_t begin = begin_of_direntry(entry);
c227f099 2474 mapping_t* mapping = find_mapping_for_cluster(s, begin);
a6c6f76c 2475#endif
a046433a
FB
2476
2477 assert(mapping);
2478 assert(mapping->begin == begin);
2479 assert(commit->path == NULL);
2480
2481 if (commit_one_file(s, commit->param.writeout.dir_index,
2482 commit->param.writeout.modified_offset))
2483 fail = -3;
2484
2485 break;
2486 }
2487 case ACTION_NEW_FILE: {
2488 int begin = commit->param.new_file.first_cluster;
c227f099
AL
2489 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2490 direntry_t* entry;
a046433a 2491 int i;
de167e41 2492
a046433a
FB
2493 /* find direntry */
2494 for (i = 0; i < s->directory.next; i++) {
2495 entry = array_get(&(s->directory), i);
2496 if (is_file(entry) && begin_of_direntry(entry) == begin)
2497 break;
de167e41 2498 }
de167e41 2499
a046433a
FB
2500 if (i >= s->directory.next) {
2501 fail = -6;
2502 continue;
2503 }
de167e41 2504
a046433a
FB
2505 /* make sure there exists an initial mapping */
2506 if (mapping && mapping->begin != begin) {
2507 mapping->end = begin;
2508 mapping = NULL;
2509 }
2510 if (mapping == NULL) {
2511 mapping = insert_mapping(s, begin, begin+1);
2512 }
2513 /* most members will be fixed in commit_mappings() */
2514 assert(commit->path);
2515 mapping->path = commit->path;
2516 mapping->read_only = 0;
2517 mapping->mode = MODE_NORMAL;
2518 mapping->info.file.offset = 0;
2519
2520 if (commit_one_file(s, i, 0))
2521 fail = -7;
2522
2523 break;
2524 }
2525 default:
43dc2a64 2526 abort();
a046433a
FB
2527 }
2528 }
2529 if (i > 0 && array_remove_slice(&(s->commits), 0, i))
2530 return -1;
2531 return fail;
2532}
2533
2534static int handle_deletes(BDRVVVFATState* s)
2535{
2536 int i, deferred = 1, deleted = 1;
2537
2538 /* delete files corresponding to mappings marked as deleted */
2539 /* handle DELETEs and unused mappings (modified_fat_get(s, mapping->begin) == 0) */
2540 while (deferred && deleted) {
2541 deferred = 0;
2542 deleted = 0;
2543
2544 for (i = 1; i < s->mapping.next; i++) {
c227f099 2545 mapping_t* mapping = array_get(&(s->mapping), i);
a046433a 2546 if (mapping->mode & MODE_DELETED) {
c227f099 2547 direntry_t* entry = array_get(&(s->directory),
a046433a
FB
2548 mapping->dir_index);
2549
2550 if (is_free(entry)) {
2551 /* remove file/directory */
2552 if (mapping->mode & MODE_DIRECTORY) {
2553 int j, next_dir_index = s->directory.next,
2554 first_dir_index = mapping->info.dir.first_dir_index;
2555
2556 if (rmdir(mapping->path) < 0) {
2557 if (errno == ENOTEMPTY) {
2558 deferred++;
2559 continue;
2560 } else
2561 return -5;
de167e41 2562 }
a046433a
FB
2563
2564 for (j = 1; j < s->mapping.next; j++) {
c227f099 2565 mapping_t* m = array_get(&(s->mapping), j);
a046433a
FB
2566 if (m->mode & MODE_DIRECTORY &&
2567 m->info.dir.first_dir_index >
2568 first_dir_index &&
2569 m->info.dir.first_dir_index <
2570 next_dir_index)
2571 next_dir_index =
2572 m->info.dir.first_dir_index;
de167e41 2573 }
a046433a
FB
2574 remove_direntries(s, first_dir_index,
2575 next_dir_index - first_dir_index);
de167e41 2576
a046433a 2577 deleted++;
de167e41 2578 }
a046433a
FB
2579 } else {
2580 if (unlink(mapping->path))
2581 return -4;
2582 deleted++;
de167e41 2583 }
a046433a
FB
2584 DLOG(fprintf(stderr, "DELETE (%d)\n", i); print_mapping(mapping); print_direntry(entry));
2585 remove_mapping(s, i);
de167e41
FB
2586 }
2587 }
2588 }
a046433a
FB
2589
2590 return 0;
2591}
2592
2593/*
2594 * synchronize mapping with new state:
2595 *
2596 * - copy FAT (with bdrv_read)
2597 * - mark all filenames corresponding to mappings as deleted
2598 * - recurse direntries from root (using bs->bdrv_read)
2599 * - delete files corresponding to mappings marked as deleted
2600 */
2601static int do_commit(BDRVVVFATState* s)
2602{
2603 int ret = 0;
2604
2605 /* the real meat are the commits. Nothing to do? Move along! */
2606 if (s->commits.next == 0)
2607 return 0;
2608
2609 vvfat_close_current_file(s);
2610
2611 ret = handle_renames_and_mkdirs(s);
2612 if (ret) {
2613 fprintf(stderr, "Error handling renames (%d)\n", ret);
43dc2a64 2614 abort();
a046433a
FB
2615 return ret;
2616 }
2617
5fafdf24 2618 /* copy FAT (with bdrv_read) */
a046433a
FB
2619 memcpy(s->fat.pointer, s->fat2, 0x200 * s->sectors_per_fat);
2620
2621 /* recurse direntries from root (using bs->bdrv_read) */
2622 ret = commit_direntries(s, 0, -1);
2623 if (ret) {
2624 fprintf(stderr, "Fatal: error while committing (%d)\n", ret);
43dc2a64 2625 abort();
a046433a
FB
2626 return ret;
2627 }
2628
2629 ret = handle_commits(s);
2630 if (ret) {
2631 fprintf(stderr, "Error handling commits (%d)\n", ret);
43dc2a64 2632 abort();
a046433a
FB
2633 return ret;
2634 }
2635
2636 ret = handle_deletes(s);
2637 if (ret) {
2638 fprintf(stderr, "Error deleting\n");
43dc2a64 2639 abort();
a046433a
FB
2640 return ret;
2641 }
2642
2643 s->qcow->drv->bdrv_make_empty(s->qcow);
2644
2645 memset(s->used_clusters, 0, sector2cluster(s, s->sector_count));
2646
2647DLOG(checkpoint());
2648 return 0;
2649}
2650
2651static int try_commit(BDRVVVFATState* s)
2652{
2653 vvfat_close_current_file(s);
2654DLOG(checkpoint());
2655 if(!is_consistent(s))
2656 return -1;
2657 return do_commit(s);
2658}
2659
5fafdf24 2660static int vvfat_write(BlockDriverState *bs, int64_t sector_num,
a046433a
FB
2661 const uint8_t *buf, int nb_sectors)
2662{
5fafdf24 2663 BDRVVVFATState *s = bs->opaque;
a046433a
FB
2664 int i, ret;
2665
2666DLOG(checkpoint());
2667
2668 vvfat_close_current_file(s);
2669
2670 /*
2671 * Some sanity checks:
2672 * - do not allow writing to the boot sector
2673 * - do not allow to write non-ASCII filenames
2674 */
2675
2676 if (sector_num < s->first_sectors_number)
2677 return -1;
2678
2679 for (i = sector2cluster(s, sector_num);
2680 i <= sector2cluster(s, sector_num + nb_sectors - 1);) {
c227f099 2681 mapping_t* mapping = find_mapping_for_cluster(s, i);
a046433a
FB
2682 if (mapping) {
2683 if (mapping->read_only) {
2684 fprintf(stderr, "Tried to write to write-protected file %s\n",
2685 mapping->path);
2686 return -1;
2687 }
2688
2689 if (mapping->mode & MODE_DIRECTORY) {
2690 int begin = cluster2sector(s, i);
2691 int end = begin + s->sectors_per_cluster, k;
2692 int dir_index;
c227f099 2693 const direntry_t* direntries;
a046433a
FB
2694 long_file_name lfn;
2695
2696 lfn_init(&lfn);
2697
2698 if (begin < sector_num)
2699 begin = sector_num;
2700 if (end > sector_num + nb_sectors)
2701 end = sector_num + nb_sectors;
5fafdf24 2702 dir_index = mapping->dir_index +
a046433a 2703 0x10 * (begin - mapping->begin * s->sectors_per_cluster);
c227f099 2704 direntries = (direntry_t*)(buf + 0x200 * (begin - sector_num));
a046433a
FB
2705
2706 for (k = 0; k < (end - begin) * 0x10; k++) {
2707 /* do not allow non-ASCII filenames */
2708 if (parse_long_name(&lfn, direntries + k) < 0) {
2709 fprintf(stderr, "Warning: non-ASCII filename\n");
2710 return -1;
2711 }
2712 /* no access to the direntry of a read-only file */
2713 else if (is_short_name(direntries+k) &&
2714 (direntries[k].attributes & 1)) {
2715 if (memcmp(direntries + k,
2716 array_get(&(s->directory), dir_index + k),
c227f099 2717 sizeof(direntry_t))) {
a046433a
FB
2718 fprintf(stderr, "Warning: tried to write to write-protected file\n");
2719 return -1;
2720 }
2721 }
2722 }
2723 }
2724 i = mapping->end;
2725 } else
2726 i++;
2727 }
2728
2729 /*
2730 * Use qcow backend. Commit later.
2731 */
2732DLOG(fprintf(stderr, "Write to qcow backend: %d + %d\n", (int)sector_num, nb_sectors));
2733 ret = s->qcow->drv->bdrv_write(s->qcow, sector_num, buf, nb_sectors);
2734 if (ret < 0) {
2735 fprintf(stderr, "Error writing to qcow backend\n");
2736 return ret;
2737 }
2738
2739 for (i = sector2cluster(s, sector_num);
2740 i <= sector2cluster(s, sector_num + nb_sectors - 1); i++)
2741 if (i >= 0)
2742 s->used_clusters[i] |= USED_ALLOCATED;
2743
2744DLOG(checkpoint());
2745 /* TODO: add timeout */
2746 try_commit(s);
2747
2748DLOG(checkpoint());
2749 return 0;
2750}
2751
2752static int vvfat_is_allocated(BlockDriverState *bs,
2753 int64_t sector_num, int nb_sectors, int* n)
2754{
2755 BDRVVVFATState* s = bs->opaque;
2756 *n = s->sector_count - sector_num;
2757 if (*n > nb_sectors)
2758 *n = nb_sectors;
2759 else if (*n < 0)
2760 return 0;
5fafdf24 2761 return 1;
a046433a
FB
2762}
2763
2764static int write_target_commit(BlockDriverState *bs, int64_t sector_num,
2765 const uint8_t* buffer, int nb_sectors) {
2766 BDRVVVFATState* s = bs->opaque;
2767 return try_commit(s);
2768}
2769
2770static void write_target_close(BlockDriverState *bs) {
2771 BDRVVVFATState* s = bs->opaque;
2772 bdrv_delete(s->qcow);
2773 free(s->qcow_filename);
2774}
2775
2776static BlockDriver vvfat_write_target = {
f9e96436
CH
2777 .format_name = "vvfat_write_target",
2778 .bdrv_write = write_target_commit,
2779 .bdrv_close = write_target_close,
a046433a
FB
2780};
2781
2782static int enable_write_target(BDRVVVFATState *s)
2783{
91a073a9
KW
2784 BlockDriver *bdrv_qcow;
2785 QEMUOptionParameter *options;
a046433a
FB
2786 int size = sector2cluster(s, s->sector_count);
2787 s->used_clusters = calloc(size, 1);
2788
c227f099 2789 array_init(&(s->commits), sizeof(commit_t));
a046433a 2790
3ec88e80 2791 s->qcow_filename = qemu_malloc(1024);
83f64091 2792 get_tmp_filename(s->qcow_filename, 1024);
91a073a9
KW
2793
2794 bdrv_qcow = bdrv_find_format("qcow");
2795 options = parse_option_parameters("", bdrv_qcow->create_options, NULL);
2796 set_option_parameter_int(options, BLOCK_OPT_SIZE, s->sector_count * 512);
2797 set_option_parameter(options, BLOCK_OPT_BACKING_FILE, "fat:");
2798
2799 if (bdrv_create(bdrv_qcow, s->qcow_filename, options) < 0)
a046433a
FB
2800 return -1;
2801 s->qcow = bdrv_new("");
d6e9098e
KW
2802 if (s->qcow == NULL ||
2803 bdrv_open(s->qcow, s->qcow_filename, BDRV_O_RDWR, bdrv_qcow) < 0)
2804 {
a046433a 2805 return -1;
d6e9098e 2806 }
a046433a
FB
2807
2808#ifndef _WIN32
2809 unlink(s->qcow_filename);
2810#endif
2811
2812 s->bs->backing_hd = calloc(sizeof(BlockDriverState), 1);
2813 s->bs->backing_hd->drv = &vvfat_write_target;
2814 s->bs->backing_hd->opaque = s;
2815
de167e41
FB
2816 return 0;
2817}
2818
2819static void vvfat_close(BlockDriverState *bs)
2820{
2821 BDRVVVFATState *s = bs->opaque;
2822
2823 vvfat_close_current_file(s);
2824 array_free(&(s->fat));
2825 array_free(&(s->directory));
2826 array_free(&(s->mapping));
a046433a
FB
2827 if(s->cluster_buffer)
2828 free(s->cluster_buffer);
de167e41
FB
2829}
2830
5efa9d5a 2831static BlockDriver bdrv_vvfat = {
e60f469c
AJ
2832 .format_name = "vvfat",
2833 .instance_size = sizeof(BDRVVVFATState),
66f82cee 2834 .bdrv_file_open = vvfat_open,
e60f469c
AJ
2835 .bdrv_read = vvfat_read,
2836 .bdrv_write = vvfat_write,
2837 .bdrv_close = vvfat_close,
2838 .bdrv_is_allocated = vvfat_is_allocated,
2839 .protocol_name = "fat",
de167e41
FB
2840};
2841
5efa9d5a
AL
2842static void bdrv_vvfat_init(void)
2843{
2844 bdrv_register(&bdrv_vvfat);
2845}
2846
2847block_init(bdrv_vvfat_init);
2848
a046433a 2849#ifdef DEBUG
3f47aa8c 2850static void checkpoint(void) {
c227f099 2851 assert(((mapping_t*)array_get(&(vvv->mapping), 0))->end == 2);
a046433a
FB
2852 check1(vvv);
2853 check2(vvv);
2854 assert(!vvv->current_mapping || vvv->current_fd || (vvv->current_mapping->mode & MODE_DIRECTORY));
2855#if 0
c227f099 2856 if (((direntry_t*)vvv->directory.pointer)[1].attributes != 0xf)
a046433a 2857 fprintf(stderr, "Nonono!\n");
c227f099
AL
2858 mapping_t* mapping;
2859 direntry_t* direntry;
a046433a
FB
2860 assert(vvv->mapping.size >= vvv->mapping.item_size * vvv->mapping.next);
2861 assert(vvv->directory.size >= vvv->directory.item_size * vvv->directory.next);
2862 if (vvv->mapping.next<47)
2863 return;
2864 assert((mapping = array_get(&(vvv->mapping), 47)));
2865 assert(mapping->dir_index < vvv->directory.next);
2866 direntry = array_get(&(vvv->directory), mapping->dir_index);
2867 assert(!memcmp(direntry->name, "USB H ", 11) || direntry->name[0]==0);
2868#endif
2869 return;
2870 /* avoid compiler warnings: */
2871 hexdump(NULL, 100);
792b45b1 2872 remove_mapping(vvv, 0);
a046433a
FB
2873 print_mapping(NULL);
2874 print_direntry(NULL);
2875}
2876#endif