configure: Don't say target_nptl="no" if there is no linux-user target
[qemu/ar7.git] / block / vvfat.c
blobcd3b8edd9f5938e8e89b2aadec471ec9bb9c5ca3
1 /* vim:set shiftwidth=4 ts=4: */
2 /*
3 * QEMU Block driver for virtual VFAT (shadows a local directory)
5 * Copyright (c) 2004,2005 Johannes E. Schindelin
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:
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
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.
25 #include <sys/stat.h>
26 #include <dirent.h>
27 #include "qemu-common.h"
28 #include "block/block_int.h"
29 #include "qemu/module.h"
30 #include "migration/migration.h"
31 #include "qapi/qmp/qint.h"
32 #include "qapi/qmp/qbool.h"
34 #ifndef S_IWGRP
35 #define S_IWGRP 0
36 #endif
37 #ifndef S_IWOTH
38 #define S_IWOTH 0
39 #endif
41 /* TODO: add ":bootsector=blabla.img:" */
42 /* LATER TODO: add automatic boot sector generation from
43 BOOTEASY.ASM and Ranish Partition Manager
44 Note that DOS assumes the system files to be the first files in the
45 file system (test if the boot sector still relies on that fact)! */
46 /* MAYBE TODO: write block-visofs.c */
47 /* TODO: call try_commit() only after a timeout */
49 /* #define DEBUG */
51 #ifdef DEBUG
53 #define DLOG(a) a
55 #undef stderr
56 #define stderr STDERR
57 FILE* stderr = NULL;
59 static void checkpoint(void);
61 #ifdef __MINGW32__
62 void nonono(const char* file, int line, const char* msg) {
63 fprintf(stderr, "Nonono! %s:%d %s\n", file, line, msg);
64 exit(-5);
66 #undef assert
67 #define assert(a) do {if (!(a)) nonono(__FILE__, __LINE__, #a);}while(0)
68 #endif
70 #else
72 #define DLOG(a)
74 #endif
76 /* dynamic array functions */
77 typedef struct array_t {
78 char* pointer;
79 unsigned int size,next,item_size;
80 } array_t;
82 static inline void array_init(array_t* array,unsigned int item_size)
84 array->pointer = NULL;
85 array->size=0;
86 array->next=0;
87 array->item_size=item_size;
90 static inline void array_free(array_t* array)
92 g_free(array->pointer);
93 array->size=array->next=0;
96 /* does not automatically grow */
97 static inline void* array_get(array_t* array,unsigned int index) {
98 assert(index < array->next);
99 return array->pointer + index * array->item_size;
102 static inline int array_ensure_allocated(array_t* array, int index)
104 if((index + 1) * array->item_size > array->size) {
105 int new_size = (index + 32) * array->item_size;
106 array->pointer = g_realloc(array->pointer, new_size);
107 if (!array->pointer)
108 return -1;
109 array->size = new_size;
110 array->next = index + 1;
113 return 0;
116 static inline void* array_get_next(array_t* array) {
117 unsigned int next = array->next;
118 void* result;
120 if (array_ensure_allocated(array, next) < 0)
121 return NULL;
123 array->next = next + 1;
124 result = array_get(array, next);
126 return result;
129 static inline void* array_insert(array_t* array,unsigned int index,unsigned int count) {
130 if((array->next+count)*array->item_size>array->size) {
131 int increment=count*array->item_size;
132 array->pointer=g_realloc(array->pointer,array->size+increment);
133 if(!array->pointer)
134 return NULL;
135 array->size+=increment;
137 memmove(array->pointer+(index+count)*array->item_size,
138 array->pointer+index*array->item_size,
139 (array->next-index)*array->item_size);
140 array->next+=count;
141 return array->pointer+index*array->item_size;
144 /* this performs a "roll", so that the element which was at index_from becomes
145 * index_to, but the order of all other elements is preserved. */
146 static inline int array_roll(array_t* array,int index_to,int index_from,int count)
148 char* buf;
149 char* from;
150 char* to;
151 int is;
153 if(!array ||
154 index_to<0 || index_to>=array->next ||
155 index_from<0 || index_from>=array->next)
156 return -1;
158 if(index_to==index_from)
159 return 0;
161 is=array->item_size;
162 from=array->pointer+index_from*is;
163 to=array->pointer+index_to*is;
164 buf=g_malloc(is*count);
165 memcpy(buf,from,is*count);
167 if(index_to<index_from)
168 memmove(to+is*count,to,from-to);
169 else
170 memmove(from,from+is*count,to-from);
172 memcpy(to,buf,is*count);
174 g_free(buf);
176 return 0;
179 static inline int array_remove_slice(array_t* array,int index, int count)
181 assert(index >=0);
182 assert(count > 0);
183 assert(index + count <= array->next);
184 if(array_roll(array,array->next-1,index,count))
185 return -1;
186 array->next -= count;
187 return 0;
190 static int array_remove(array_t* array,int index)
192 return array_remove_slice(array, index, 1);
195 /* return the index for a given member */
196 static int array_index(array_t* array, void* pointer)
198 size_t offset = (char*)pointer - array->pointer;
199 assert((offset % array->item_size) == 0);
200 assert(offset/array->item_size < array->next);
201 return offset/array->item_size;
204 /* These structures are used to fake a disk and the VFAT filesystem.
205 * For this reason we need to use QEMU_PACKED. */
207 typedef struct bootsector_t {
208 uint8_t jump[3];
209 uint8_t name[8];
210 uint16_t sector_size;
211 uint8_t sectors_per_cluster;
212 uint16_t reserved_sectors;
213 uint8_t number_of_fats;
214 uint16_t root_entries;
215 uint16_t total_sectors16;
216 uint8_t media_type;
217 uint16_t sectors_per_fat;
218 uint16_t sectors_per_track;
219 uint16_t number_of_heads;
220 uint32_t hidden_sectors;
221 uint32_t total_sectors;
222 union {
223 struct {
224 uint8_t drive_number;
225 uint8_t current_head;
226 uint8_t signature;
227 uint32_t id;
228 uint8_t volume_label[11];
229 } QEMU_PACKED fat16;
230 struct {
231 uint32_t sectors_per_fat;
232 uint16_t flags;
233 uint8_t major,minor;
234 uint32_t first_cluster_of_root_directory;
235 uint16_t info_sector;
236 uint16_t backup_boot_sector;
237 uint16_t ignored;
238 } QEMU_PACKED fat32;
239 } u;
240 uint8_t fat_type[8];
241 uint8_t ignored[0x1c0];
242 uint8_t magic[2];
243 } QEMU_PACKED bootsector_t;
245 typedef struct {
246 uint8_t head;
247 uint8_t sector;
248 uint8_t cylinder;
249 } mbr_chs_t;
251 typedef struct partition_t {
252 uint8_t attributes; /* 0x80 = bootable */
253 mbr_chs_t start_CHS;
254 uint8_t fs_type; /* 0x1 = FAT12, 0x6 = FAT16, 0xe = FAT16_LBA, 0xb = FAT32, 0xc = FAT32_LBA */
255 mbr_chs_t end_CHS;
256 uint32_t start_sector_long;
257 uint32_t length_sector_long;
258 } QEMU_PACKED partition_t;
260 typedef struct mbr_t {
261 uint8_t ignored[0x1b8];
262 uint32_t nt_id;
263 uint8_t ignored2[2];
264 partition_t partition[4];
265 uint8_t magic[2];
266 } QEMU_PACKED mbr_t;
268 typedef struct direntry_t {
269 uint8_t name[8];
270 uint8_t extension[3];
271 uint8_t attributes;
272 uint8_t reserved[2];
273 uint16_t ctime;
274 uint16_t cdate;
275 uint16_t adate;
276 uint16_t begin_hi;
277 uint16_t mtime;
278 uint16_t mdate;
279 uint16_t begin;
280 uint32_t size;
281 } QEMU_PACKED direntry_t;
283 /* this structure are used to transparently access the files */
285 typedef struct mapping_t {
286 /* begin is the first cluster, end is the last+1 */
287 uint32_t begin,end;
288 /* as s->directory is growable, no pointer may be used here */
289 unsigned int dir_index;
290 /* the clusters of a file may be in any order; this points to the first */
291 int first_mapping_index;
292 union {
293 /* offset is
294 * - the offset in the file (in clusters) for a file, or
295 * - the next cluster of the directory for a directory, and
296 * - the address of the buffer for a faked entry
298 struct {
299 uint32_t offset;
300 } file;
301 struct {
302 int parent_mapping_index;
303 int first_dir_index;
304 } dir;
305 } info;
306 /* path contains the full path, i.e. it always starts with s->path */
307 char* path;
309 enum { MODE_UNDEFINED = 0, MODE_NORMAL = 1, MODE_MODIFIED = 2,
310 MODE_DIRECTORY = 4, MODE_FAKED = 8,
311 MODE_DELETED = 16, MODE_RENAMED = 32 } mode;
312 int read_only;
313 } mapping_t;
315 #ifdef DEBUG
316 static void print_direntry(const struct direntry_t*);
317 static void print_mapping(const struct mapping_t* mapping);
318 #endif
320 /* here begins the real VVFAT driver */
322 typedef struct BDRVVVFATState {
323 CoMutex lock;
324 BlockDriverState* bs; /* pointer to parent */
325 unsigned int first_sectors_number; /* 1 for a single partition, 0x40 for a disk with partition table */
326 unsigned char first_sectors[0x40*0x200];
328 int fat_type; /* 16 or 32 */
329 array_t fat,directory,mapping;
331 unsigned int cluster_size;
332 unsigned int sectors_per_cluster;
333 unsigned int sectors_per_fat;
334 unsigned int sectors_of_root_directory;
335 uint32_t last_cluster_of_root_directory;
336 unsigned int faked_sectors; /* how many sectors are faked before file data */
337 uint32_t sector_count; /* total number of sectors of the partition */
338 uint32_t cluster_count; /* total number of clusters of this partition */
339 uint32_t max_fat_value;
341 int current_fd;
342 mapping_t* current_mapping;
343 unsigned char* cluster; /* points to current cluster */
344 unsigned char* cluster_buffer; /* points to a buffer to hold temp data */
345 unsigned int current_cluster;
347 /* write support */
348 BlockDriverState* write_target;
349 char* qcow_filename;
350 BlockDriverState* qcow;
351 void* fat2;
352 char* used_clusters;
353 array_t commits;
354 const char* path;
355 int downcase_short_names;
357 Error *migration_blocker;
358 } BDRVVVFATState;
360 /* take the sector position spos and convert it to Cylinder/Head/Sector position
361 * if the position is outside the specified geometry, fill maximum value for CHS
362 * and return 1 to signal overflow.
364 static int sector2CHS(mbr_chs_t *chs, int spos, int cyls, int heads, int secs)
366 int head,sector;
367 sector = spos % secs; spos /= secs;
368 head = spos % heads; spos /= heads;
369 if (spos >= cyls) {
370 /* Overflow,
371 it happens if 32bit sector positions are used, while CHS is only 24bit.
372 Windows/Dos is said to take 1023/255/63 as nonrepresentable CHS */
373 chs->head = 0xFF;
374 chs->sector = 0xFF;
375 chs->cylinder = 0xFF;
376 return 1;
378 chs->head = (uint8_t)head;
379 chs->sector = (uint8_t)( (sector+1) | ((spos>>8)<<6) );
380 chs->cylinder = (uint8_t)spos;
381 return 0;
384 static void init_mbr(BDRVVVFATState *s, int cyls, int heads, int secs)
386 /* TODO: if the files mbr.img and bootsect.img exist, use them */
387 mbr_t* real_mbr=(mbr_t*)s->first_sectors;
388 partition_t* partition = &(real_mbr->partition[0]);
389 int lba;
391 memset(s->first_sectors,0,512);
393 /* Win NT Disk Signature */
394 real_mbr->nt_id= cpu_to_le32(0xbe1afdfa);
396 partition->attributes=0x80; /* bootable */
398 /* LBA is used when partition is outside the CHS geometry */
399 lba = sector2CHS(&partition->start_CHS, s->first_sectors_number - 1,
400 cyls, heads, secs);
401 lba |= sector2CHS(&partition->end_CHS, s->bs->total_sectors - 1,
402 cyls, heads, secs);
404 /*LBA partitions are identified only by start/length_sector_long not by CHS*/
405 partition->start_sector_long = cpu_to_le32(s->first_sectors_number - 1);
406 partition->length_sector_long = cpu_to_le32(s->bs->total_sectors
407 - s->first_sectors_number + 1);
409 /* FAT12/FAT16/FAT32 */
410 /* DOS uses different types when partition is LBA,
411 probably to prevent older versions from using CHS on them */
412 partition->fs_type= s->fat_type==12 ? 0x1:
413 s->fat_type==16 ? (lba?0xe:0x06):
414 /*fat_tyoe==32*/ (lba?0xc:0x0b);
416 real_mbr->magic[0]=0x55; real_mbr->magic[1]=0xaa;
419 /* direntry functions */
421 /* dest is assumed to hold 258 bytes, and pads with 0xffff up to next multiple of 26 */
422 static inline int short2long_name(char* dest,const char* src)
424 int i;
425 int len;
426 for(i=0;i<129 && src[i];i++) {
427 dest[2*i]=src[i];
428 dest[2*i+1]=0;
430 len=2*i;
431 dest[2*i]=dest[2*i+1]=0;
432 for(i=2*i+2;(i%26);i++)
433 dest[i]=0xff;
434 return len;
437 static inline direntry_t* create_long_filename(BDRVVVFATState* s,const char* filename)
439 char buffer[258];
440 int length=short2long_name(buffer,filename),
441 number_of_entries=(length+25)/26,i;
442 direntry_t* entry;
444 for(i=0;i<number_of_entries;i++) {
445 entry=array_get_next(&(s->directory));
446 entry->attributes=0xf;
447 entry->reserved[0]=0;
448 entry->begin=0;
449 entry->name[0]=(number_of_entries-i)|(i==0?0x40:0);
451 for(i=0;i<26*number_of_entries;i++) {
452 int offset=(i%26);
453 if(offset<10) offset=1+offset;
454 else if(offset<22) offset=14+offset-10;
455 else offset=28+offset-22;
456 entry=array_get(&(s->directory),s->directory.next-1-(i/26));
457 entry->name[offset]=buffer[i];
459 return array_get(&(s->directory),s->directory.next-number_of_entries);
462 static char is_free(const direntry_t* direntry)
464 return direntry->name[0]==0xe5 || direntry->name[0]==0x00;
467 static char is_volume_label(const direntry_t* direntry)
469 return direntry->attributes == 0x28;
472 static char is_long_name(const direntry_t* direntry)
474 return direntry->attributes == 0xf;
477 static char is_short_name(const direntry_t* direntry)
479 return !is_volume_label(direntry) && !is_long_name(direntry)
480 && !is_free(direntry);
483 static char is_directory(const direntry_t* direntry)
485 return direntry->attributes & 0x10 && direntry->name[0] != 0xe5;
488 static inline char is_dot(const direntry_t* direntry)
490 return is_short_name(direntry) && direntry->name[0] == '.';
493 static char is_file(const direntry_t* direntry)
495 return is_short_name(direntry) && !is_directory(direntry);
498 static inline uint32_t begin_of_direntry(const direntry_t* direntry)
500 return le16_to_cpu(direntry->begin)|(le16_to_cpu(direntry->begin_hi)<<16);
503 static inline uint32_t filesize_of_direntry(const direntry_t* direntry)
505 return le32_to_cpu(direntry->size);
508 static void set_begin_of_direntry(direntry_t* direntry, uint32_t begin)
510 direntry->begin = cpu_to_le16(begin & 0xffff);
511 direntry->begin_hi = cpu_to_le16((begin >> 16) & 0xffff);
514 /* fat functions */
516 static inline uint8_t fat_chksum(const direntry_t* entry)
518 uint8_t chksum=0;
519 int i;
521 for(i=0;i<11;i++) {
522 unsigned char c;
524 c = (i < 8) ? entry->name[i] : entry->extension[i-8];
525 chksum=(((chksum&0xfe)>>1)|((chksum&0x01)?0x80:0)) + c;
528 return chksum;
531 /* if return_time==0, this returns the fat_date, else the fat_time */
532 static uint16_t fat_datetime(time_t time,int return_time) {
533 struct tm* t;
534 struct tm t1;
535 t = &t1;
536 localtime_r(&time,t);
537 if(return_time)
538 return cpu_to_le16((t->tm_sec/2)|(t->tm_min<<5)|(t->tm_hour<<11));
539 return cpu_to_le16((t->tm_mday)|((t->tm_mon+1)<<5)|((t->tm_year-80)<<9));
542 static inline void fat_set(BDRVVVFATState* s,unsigned int cluster,uint32_t value)
544 if(s->fat_type==32) {
545 uint32_t* entry=array_get(&(s->fat),cluster);
546 *entry=cpu_to_le32(value);
547 } else if(s->fat_type==16) {
548 uint16_t* entry=array_get(&(s->fat),cluster);
549 *entry=cpu_to_le16(value&0xffff);
550 } else {
551 int offset = (cluster*3/2);
552 unsigned char* p = array_get(&(s->fat), offset);
553 switch (cluster&1) {
554 case 0:
555 p[0] = value&0xff;
556 p[1] = (p[1]&0xf0) | ((value>>8)&0xf);
557 break;
558 case 1:
559 p[0] = (p[0]&0xf) | ((value&0xf)<<4);
560 p[1] = (value>>4);
561 break;
566 static inline uint32_t fat_get(BDRVVVFATState* s,unsigned int cluster)
568 if(s->fat_type==32) {
569 uint32_t* entry=array_get(&(s->fat),cluster);
570 return le32_to_cpu(*entry);
571 } else if(s->fat_type==16) {
572 uint16_t* entry=array_get(&(s->fat),cluster);
573 return le16_to_cpu(*entry);
574 } else {
575 const uint8_t* x=(uint8_t*)(s->fat.pointer)+cluster*3/2;
576 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
580 static inline int fat_eof(BDRVVVFATState* s,uint32_t fat_entry)
582 if(fat_entry>s->max_fat_value-8)
583 return -1;
584 return 0;
587 static inline void init_fat(BDRVVVFATState* s)
589 if (s->fat_type == 12) {
590 array_init(&(s->fat),1);
591 array_ensure_allocated(&(s->fat),
592 s->sectors_per_fat * 0x200 * 3 / 2 - 1);
593 } else {
594 array_init(&(s->fat),(s->fat_type==32?4:2));
595 array_ensure_allocated(&(s->fat),
596 s->sectors_per_fat * 0x200 / s->fat.item_size - 1);
598 memset(s->fat.pointer,0,s->fat.size);
600 switch(s->fat_type) {
601 case 12: s->max_fat_value=0xfff; break;
602 case 16: s->max_fat_value=0xffff; break;
603 case 32: s->max_fat_value=0x0fffffff; break;
604 default: s->max_fat_value=0; /* error... */
609 /* TODO: in create_short_filename, 0xe5->0x05 is not yet handled! */
610 /* TODO: in parse_short_filename, 0x05->0xe5 is not yet handled! */
611 static inline direntry_t* create_short_and_long_name(BDRVVVFATState* s,
612 unsigned int directory_start, const char* filename, int is_dot)
614 int i,j,long_index=s->directory.next;
615 direntry_t* entry = NULL;
616 direntry_t* entry_long = NULL;
618 if(is_dot) {
619 entry=array_get_next(&(s->directory));
620 memset(entry->name,0x20,11);
621 memcpy(entry->name,filename,strlen(filename));
622 return entry;
625 entry_long=create_long_filename(s,filename);
627 i = strlen(filename);
628 for(j = i - 1; j>0 && filename[j]!='.';j--);
629 if (j > 0)
630 i = (j > 8 ? 8 : j);
631 else if (i > 8)
632 i = 8;
634 entry=array_get_next(&(s->directory));
635 memset(entry->name,0x20,11);
636 memcpy(entry->name, filename, i);
638 if(j > 0)
639 for (i = 0; i < 3 && filename[j+1+i]; i++)
640 entry->extension[i] = filename[j+1+i];
642 /* upcase & remove unwanted characters */
643 for(i=10;i>=0;i--) {
644 if(i==10 || i==7) for(;i>0 && entry->name[i]==' ';i--);
645 if(entry->name[i]<=' ' || entry->name[i]>0x7f
646 || strchr(".*?<>|\":/\\[];,+='",entry->name[i]))
647 entry->name[i]='_';
648 else if(entry->name[i]>='a' && entry->name[i]<='z')
649 entry->name[i]+='A'-'a';
652 /* mangle duplicates */
653 while(1) {
654 direntry_t* entry1=array_get(&(s->directory),directory_start);
655 int j;
657 for(;entry1<entry;entry1++)
658 if(!is_long_name(entry1) && !memcmp(entry1->name,entry->name,11))
659 break; /* found dupe */
660 if(entry1==entry) /* no dupe found */
661 break;
663 /* use all 8 characters of name */
664 if(entry->name[7]==' ') {
665 int j;
666 for(j=6;j>0 && entry->name[j]==' ';j--)
667 entry->name[j]='~';
670 /* increment number */
671 for(j=7;j>0 && entry->name[j]=='9';j--)
672 entry->name[j]='0';
673 if(j>0) {
674 if(entry->name[j]<'0' || entry->name[j]>'9')
675 entry->name[j]='0';
676 else
677 entry->name[j]++;
681 /* calculate checksum; propagate to long name */
682 if(entry_long) {
683 uint8_t chksum=fat_chksum(entry);
685 /* calculate anew, because realloc could have taken place */
686 entry_long=array_get(&(s->directory),long_index);
687 while(entry_long<entry && is_long_name(entry_long)) {
688 entry_long->reserved[1]=chksum;
689 entry_long++;
693 return entry;
697 * Read a directory. (the index of the corresponding mapping must be passed).
699 static int read_directory(BDRVVVFATState* s, int mapping_index)
701 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
702 direntry_t* direntry;
703 const char* dirname = mapping->path;
704 int first_cluster = mapping->begin;
705 int parent_index = mapping->info.dir.parent_mapping_index;
706 mapping_t* parent_mapping = (mapping_t*)
707 (parent_index >= 0 ? array_get(&(s->mapping), parent_index) : NULL);
708 int first_cluster_of_parent = parent_mapping ? parent_mapping->begin : -1;
710 DIR* dir=opendir(dirname);
711 struct dirent* entry;
712 int i;
714 assert(mapping->mode & MODE_DIRECTORY);
716 if(!dir) {
717 mapping->end = mapping->begin;
718 return -1;
721 i = mapping->info.dir.first_dir_index =
722 first_cluster == 0 ? 0 : s->directory.next;
724 /* actually read the directory, and allocate the mappings */
725 while((entry=readdir(dir))) {
726 unsigned int length=strlen(dirname)+2+strlen(entry->d_name);
727 char* buffer;
728 direntry_t* direntry;
729 struct stat st;
730 int is_dot=!strcmp(entry->d_name,".");
731 int is_dotdot=!strcmp(entry->d_name,"..");
733 if(first_cluster == 0 && (is_dotdot || is_dot))
734 continue;
736 buffer=(char*)g_malloc(length);
737 snprintf(buffer,length,"%s/%s",dirname,entry->d_name);
739 if(stat(buffer,&st)<0) {
740 g_free(buffer);
741 continue;
744 /* create directory entry for this file */
745 direntry=create_short_and_long_name(s, i, entry->d_name,
746 is_dot || is_dotdot);
747 direntry->attributes=(S_ISDIR(st.st_mode)?0x10:0x20);
748 direntry->reserved[0]=direntry->reserved[1]=0;
749 direntry->ctime=fat_datetime(st.st_ctime,1);
750 direntry->cdate=fat_datetime(st.st_ctime,0);
751 direntry->adate=fat_datetime(st.st_atime,0);
752 direntry->begin_hi=0;
753 direntry->mtime=fat_datetime(st.st_mtime,1);
754 direntry->mdate=fat_datetime(st.st_mtime,0);
755 if(is_dotdot)
756 set_begin_of_direntry(direntry, first_cluster_of_parent);
757 else if(is_dot)
758 set_begin_of_direntry(direntry, first_cluster);
759 else
760 direntry->begin=0; /* do that later */
761 if (st.st_size > 0x7fffffff) {
762 fprintf(stderr, "File %s is larger than 2GB\n", buffer);
763 g_free(buffer);
764 closedir(dir);
765 return -2;
767 direntry->size=cpu_to_le32(S_ISDIR(st.st_mode)?0:st.st_size);
769 /* create mapping for this file */
770 if(!is_dot && !is_dotdot && (S_ISDIR(st.st_mode) || st.st_size)) {
771 s->current_mapping=(mapping_t*)array_get_next(&(s->mapping));
772 s->current_mapping->begin=0;
773 s->current_mapping->end=st.st_size;
775 * we get the direntry of the most recent direntry, which
776 * contains the short name and all the relevant information.
778 s->current_mapping->dir_index=s->directory.next-1;
779 s->current_mapping->first_mapping_index = -1;
780 if (S_ISDIR(st.st_mode)) {
781 s->current_mapping->mode = MODE_DIRECTORY;
782 s->current_mapping->info.dir.parent_mapping_index =
783 mapping_index;
784 } else {
785 s->current_mapping->mode = MODE_UNDEFINED;
786 s->current_mapping->info.file.offset = 0;
788 s->current_mapping->path=buffer;
789 s->current_mapping->read_only =
790 (st.st_mode & (S_IWUSR | S_IWGRP | S_IWOTH)) == 0;
793 closedir(dir);
795 /* fill with zeroes up to the end of the cluster */
796 while(s->directory.next%(0x10*s->sectors_per_cluster)) {
797 direntry_t* direntry=array_get_next(&(s->directory));
798 memset(direntry,0,sizeof(direntry_t));
801 /* TODO: if there are more entries, bootsector has to be adjusted! */
802 #define ROOT_ENTRIES (0x02 * 0x10 * s->sectors_per_cluster)
803 if (mapping_index == 0 && s->directory.next < ROOT_ENTRIES) {
804 /* root directory */
805 int cur = s->directory.next;
806 array_ensure_allocated(&(s->directory), ROOT_ENTRIES - 1);
807 s->directory.next = ROOT_ENTRIES;
808 memset(array_get(&(s->directory), cur), 0,
809 (ROOT_ENTRIES - cur) * sizeof(direntry_t));
812 /* reget the mapping, since s->mapping was possibly realloc()ed */
813 mapping = (mapping_t*)array_get(&(s->mapping), mapping_index);
814 first_cluster += (s->directory.next - mapping->info.dir.first_dir_index)
815 * 0x20 / s->cluster_size;
816 mapping->end = first_cluster;
818 direntry = (direntry_t*)array_get(&(s->directory), mapping->dir_index);
819 set_begin_of_direntry(direntry, mapping->begin);
821 return 0;
824 static inline uint32_t sector2cluster(BDRVVVFATState* s,off_t sector_num)
826 return (sector_num-s->faked_sectors)/s->sectors_per_cluster;
829 static inline off_t cluster2sector(BDRVVVFATState* s, uint32_t cluster_num)
831 return s->faked_sectors + s->sectors_per_cluster * cluster_num;
834 static int init_directories(BDRVVVFATState* s,
835 const char *dirname, int heads, int secs)
837 bootsector_t* bootsector;
838 mapping_t* mapping;
839 unsigned int i;
840 unsigned int cluster;
842 memset(&(s->first_sectors[0]),0,0x40*0x200);
844 s->cluster_size=s->sectors_per_cluster*0x200;
845 s->cluster_buffer=g_malloc(s->cluster_size);
848 * The formula: sc = spf+1+spf*spc*(512*8/fat_type),
849 * where sc is sector_count,
850 * spf is sectors_per_fat,
851 * spc is sectors_per_clusters, and
852 * fat_type = 12, 16 or 32.
854 i = 1+s->sectors_per_cluster*0x200*8/s->fat_type;
855 s->sectors_per_fat=(s->sector_count+i)/i; /* round up */
857 array_init(&(s->mapping),sizeof(mapping_t));
858 array_init(&(s->directory),sizeof(direntry_t));
860 /* add volume label */
862 direntry_t* entry=array_get_next(&(s->directory));
863 entry->attributes=0x28; /* archive | volume label */
864 memcpy(entry->name,"QEMU VVF",8);
865 memcpy(entry->extension,"AT ",3);
868 /* Now build FAT, and write back information into directory */
869 init_fat(s);
871 s->faked_sectors=s->first_sectors_number+s->sectors_per_fat*2;
872 s->cluster_count=sector2cluster(s, s->sector_count);
874 mapping = array_get_next(&(s->mapping));
875 mapping->begin = 0;
876 mapping->dir_index = 0;
877 mapping->info.dir.parent_mapping_index = -1;
878 mapping->first_mapping_index = -1;
879 mapping->path = g_strdup(dirname);
880 i = strlen(mapping->path);
881 if (i > 0 && mapping->path[i - 1] == '/')
882 mapping->path[i - 1] = '\0';
883 mapping->mode = MODE_DIRECTORY;
884 mapping->read_only = 0;
885 s->path = mapping->path;
887 for (i = 0, cluster = 0; i < s->mapping.next; i++) {
888 /* MS-DOS expects the FAT to be 0 for the root directory
889 * (except for the media byte). */
890 /* LATER TODO: still true for FAT32? */
891 int fix_fat = (i != 0);
892 mapping = array_get(&(s->mapping), i);
894 if (mapping->mode & MODE_DIRECTORY) {
895 mapping->begin = cluster;
896 if(read_directory(s, i)) {
897 fprintf(stderr, "Could not read directory %s\n",
898 mapping->path);
899 return -1;
901 mapping = array_get(&(s->mapping), i);
902 } else {
903 assert(mapping->mode == MODE_UNDEFINED);
904 mapping->mode=MODE_NORMAL;
905 mapping->begin = cluster;
906 if (mapping->end > 0) {
907 direntry_t* direntry = array_get(&(s->directory),
908 mapping->dir_index);
910 mapping->end = cluster + 1 + (mapping->end-1)/s->cluster_size;
911 set_begin_of_direntry(direntry, mapping->begin);
912 } else {
913 mapping->end = cluster + 1;
914 fix_fat = 0;
918 assert(mapping->begin < mapping->end);
920 /* next free cluster */
921 cluster = mapping->end;
923 if(cluster > s->cluster_count) {
924 fprintf(stderr,"Directory does not fit in FAT%d (capacity %.2f MB)\n",
925 s->fat_type, s->sector_count / 2000.0);
926 return -EINVAL;
929 /* fix fat for entry */
930 if (fix_fat) {
931 int j;
932 for(j = mapping->begin; j < mapping->end - 1; j++)
933 fat_set(s, j, j+1);
934 fat_set(s, mapping->end - 1, s->max_fat_value);
938 mapping = array_get(&(s->mapping), 0);
939 s->sectors_of_root_directory = mapping->end * s->sectors_per_cluster;
940 s->last_cluster_of_root_directory = mapping->end;
942 /* the FAT signature */
943 fat_set(s,0,s->max_fat_value);
944 fat_set(s,1,s->max_fat_value);
946 s->current_mapping = NULL;
948 bootsector=(bootsector_t*)(s->first_sectors+(s->first_sectors_number-1)*0x200);
949 bootsector->jump[0]=0xeb;
950 bootsector->jump[1]=0x3e;
951 bootsector->jump[2]=0x90;
952 memcpy(bootsector->name,"QEMU ",8);
953 bootsector->sector_size=cpu_to_le16(0x200);
954 bootsector->sectors_per_cluster=s->sectors_per_cluster;
955 bootsector->reserved_sectors=cpu_to_le16(1);
956 bootsector->number_of_fats=0x2; /* number of FATs */
957 bootsector->root_entries=cpu_to_le16(s->sectors_of_root_directory*0x10);
958 bootsector->total_sectors16=s->sector_count>0xffff?0:cpu_to_le16(s->sector_count);
959 bootsector->media_type=(s->first_sectors_number>1?0xf8:0xf0); /* media descriptor (f8=hd, f0=3.5 fd)*/
960 s->fat.pointer[0] = bootsector->media_type;
961 bootsector->sectors_per_fat=cpu_to_le16(s->sectors_per_fat);
962 bootsector->sectors_per_track = cpu_to_le16(secs);
963 bootsector->number_of_heads = cpu_to_le16(heads);
964 bootsector->hidden_sectors=cpu_to_le32(s->first_sectors_number==1?0:0x3f);
965 bootsector->total_sectors=cpu_to_le32(s->sector_count>0xffff?s->sector_count:0);
967 /* LATER TODO: if FAT32, this is wrong */
968 bootsector->u.fat16.drive_number=s->first_sectors_number==1?0:0x80; /* fda=0, hda=0x80 */
969 bootsector->u.fat16.current_head=0;
970 bootsector->u.fat16.signature=0x29;
971 bootsector->u.fat16.id=cpu_to_le32(0xfabe1afd);
973 memcpy(bootsector->u.fat16.volume_label,"QEMU VVFAT ",11);
974 memcpy(bootsector->fat_type,(s->fat_type==12?"FAT12 ":s->fat_type==16?"FAT16 ":"FAT32 "),8);
975 bootsector->magic[0]=0x55; bootsector->magic[1]=0xaa;
977 return 0;
980 #ifdef DEBUG
981 static BDRVVVFATState *vvv = NULL;
982 #endif
984 static int enable_write_target(BDRVVVFATState *s);
985 static int is_consistent(BDRVVVFATState *s);
987 static void vvfat_rebind(BlockDriverState *bs)
989 BDRVVVFATState *s = bs->opaque;
990 s->bs = bs;
993 static QemuOptsList runtime_opts = {
994 .name = "vvfat",
995 .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
996 .desc = {
998 .name = "dir",
999 .type = QEMU_OPT_STRING,
1000 .help = "Host directory to map to the vvfat device",
1003 .name = "fat-type",
1004 .type = QEMU_OPT_NUMBER,
1005 .help = "FAT type (12, 16 or 32)",
1008 .name = "floppy",
1009 .type = QEMU_OPT_BOOL,
1010 .help = "Create a floppy rather than a hard disk image",
1013 .name = "rw",
1014 .type = QEMU_OPT_BOOL,
1015 .help = "Make the image writable",
1017 { /* end of list */ }
1021 static void vvfat_parse_filename(const char *filename, QDict *options,
1022 Error **errp)
1024 int fat_type = 0;
1025 bool floppy = false;
1026 bool rw = false;
1027 int i;
1029 if (!strstart(filename, "fat:", NULL)) {
1030 error_setg(errp, "File name string must start with 'fat:'");
1031 return;
1034 /* Parse options */
1035 if (strstr(filename, ":32:")) {
1036 fat_type = 32;
1037 } else if (strstr(filename, ":16:")) {
1038 fat_type = 16;
1039 } else if (strstr(filename, ":12:")) {
1040 fat_type = 12;
1043 if (strstr(filename, ":floppy:")) {
1044 floppy = true;
1047 if (strstr(filename, ":rw:")) {
1048 rw = true;
1051 /* Get the directory name without options */
1052 i = strrchr(filename, ':') - filename;
1053 assert(i >= 3);
1054 if (filename[i - 2] == ':' && qemu_isalpha(filename[i - 1])) {
1055 /* workaround for DOS drive names */
1056 filename += i - 1;
1057 } else {
1058 filename += i + 1;
1061 /* Fill in the options QDict */
1062 qdict_put(options, "dir", qstring_from_str(filename));
1063 qdict_put(options, "fat-type", qint_from_int(fat_type));
1064 qdict_put(options, "floppy", qbool_from_int(floppy));
1065 qdict_put(options, "rw", qbool_from_int(rw));
1068 static int vvfat_open(BlockDriverState *bs, QDict *options, int flags)
1070 BDRVVVFATState *s = bs->opaque;
1071 int cyls, heads, secs;
1072 bool floppy;
1073 const char *dirname;
1074 QemuOpts *opts;
1075 Error *local_err = NULL;
1076 int ret;
1078 #ifdef DEBUG
1079 vvv = s;
1080 #endif
1082 DLOG(if (stderr == NULL) {
1083 stderr = fopen("vvfat.log", "a");
1084 setbuf(stderr, NULL);
1087 opts = qemu_opts_create_nofail(&runtime_opts);
1088 qemu_opts_absorb_qdict(opts, options, &local_err);
1089 if (error_is_set(&local_err)) {
1090 qerror_report_err(local_err);
1091 error_free(local_err);
1092 ret = -EINVAL;
1093 goto fail;
1096 dirname = qemu_opt_get(opts, "dir");
1097 if (!dirname) {
1098 qerror_report(ERROR_CLASS_GENERIC_ERROR, "vvfat block driver requires "
1099 "a 'dir' option");
1100 ret = -EINVAL;
1101 goto fail;
1104 s->fat_type = qemu_opt_get_number(opts, "fat-type", 0);
1105 floppy = qemu_opt_get_bool(opts, "floppy", false);
1107 if (floppy) {
1108 /* 1.44MB or 2.88MB floppy. 2.88MB can be FAT12 (default) or FAT16. */
1109 if (!s->fat_type) {
1110 s->fat_type = 12;
1111 secs = 36;
1112 s->sectors_per_cluster = 2;
1113 } else {
1114 secs = s->fat_type == 12 ? 18 : 36;
1115 s->sectors_per_cluster = 1;
1117 s->first_sectors_number = 1;
1118 cyls = 80;
1119 heads = 2;
1120 } else {
1121 /* 32MB or 504MB disk*/
1122 if (!s->fat_type) {
1123 s->fat_type = 16;
1125 cyls = s->fat_type == 12 ? 64 : 1024;
1126 heads = 16;
1127 secs = 63;
1130 switch (s->fat_type) {
1131 case 32:
1132 fprintf(stderr, "Big fat greek warning: FAT32 has not been tested. "
1133 "You are welcome to do so!\n");
1134 break;
1135 case 16:
1136 case 12:
1137 break;
1138 default:
1139 qerror_report(ERROR_CLASS_GENERIC_ERROR, "Valid FAT types are only "
1140 "12, 16 and 32");
1141 ret = -EINVAL;
1142 goto fail;
1146 s->bs = bs;
1148 /* LATER TODO: if FAT32, adjust */
1149 s->sectors_per_cluster=0x10;
1151 s->current_cluster=0xffffffff;
1153 s->first_sectors_number=0x40;
1154 /* read only is the default for safety */
1155 bs->read_only = 1;
1156 s->qcow = s->write_target = NULL;
1157 s->qcow_filename = NULL;
1158 s->fat2 = NULL;
1159 s->downcase_short_names = 1;
1161 fprintf(stderr, "vvfat %s chs %d,%d,%d\n",
1162 dirname, cyls, heads, secs);
1164 s->sector_count = cyls * heads * secs - (s->first_sectors_number - 1);
1166 if (qemu_opt_get_bool(opts, "rw", false)) {
1167 ret = enable_write_target(s);
1168 if (ret < 0) {
1169 goto fail;
1171 bs->read_only = 0;
1174 bs->total_sectors = cyls * heads * secs;
1176 if (init_directories(s, dirname, heads, secs)) {
1177 ret = -EIO;
1178 goto fail;
1181 s->sector_count = s->faked_sectors + s->sectors_per_cluster*s->cluster_count;
1183 if (s->first_sectors_number == 0x40) {
1184 init_mbr(s, cyls, heads, secs);
1187 // assert(is_consistent(s));
1188 qemu_co_mutex_init(&s->lock);
1190 /* Disable migration when vvfat is used rw */
1191 if (s->qcow) {
1192 error_set(&s->migration_blocker,
1193 QERR_BLOCK_FORMAT_FEATURE_NOT_SUPPORTED,
1194 "vvfat (rw)", bs->device_name, "live migration");
1195 migrate_add_blocker(s->migration_blocker);
1198 ret = 0;
1199 fail:
1200 qemu_opts_del(opts);
1201 return ret;
1204 static inline void vvfat_close_current_file(BDRVVVFATState *s)
1206 if(s->current_mapping) {
1207 s->current_mapping = NULL;
1208 if (s->current_fd) {
1209 qemu_close(s->current_fd);
1210 s->current_fd = 0;
1213 s->current_cluster = -1;
1216 /* mappings between index1 and index2-1 are supposed to be ordered
1217 * return value is the index of the last mapping for which end>cluster_num
1219 static inline int find_mapping_for_cluster_aux(BDRVVVFATState* s,int cluster_num,int index1,int index2)
1221 while(1) {
1222 int index3;
1223 mapping_t* mapping;
1224 index3=(index1+index2)/2;
1225 mapping=array_get(&(s->mapping),index3);
1226 assert(mapping->begin < mapping->end);
1227 if(mapping->begin>=cluster_num) {
1228 assert(index2!=index3 || index2==0);
1229 if(index2==index3)
1230 return index1;
1231 index2=index3;
1232 } else {
1233 if(index1==index3)
1234 return mapping->end<=cluster_num ? index2 : index1;
1235 index1=index3;
1237 assert(index1<=index2);
1238 DLOG(mapping=array_get(&(s->mapping),index1);
1239 assert(mapping->begin<=cluster_num);
1240 assert(index2 >= s->mapping.next ||
1241 ((mapping = array_get(&(s->mapping),index2)) &&
1242 mapping->end>cluster_num)));
1246 static inline mapping_t* find_mapping_for_cluster(BDRVVVFATState* s,int cluster_num)
1248 int index=find_mapping_for_cluster_aux(s,cluster_num,0,s->mapping.next);
1249 mapping_t* mapping;
1250 if(index>=s->mapping.next)
1251 return NULL;
1252 mapping=array_get(&(s->mapping),index);
1253 if(mapping->begin>cluster_num)
1254 return NULL;
1255 assert(mapping->begin<=cluster_num && mapping->end>cluster_num);
1256 return mapping;
1259 static int open_file(BDRVVVFATState* s,mapping_t* mapping)
1261 if(!mapping)
1262 return -1;
1263 if(!s->current_mapping ||
1264 strcmp(s->current_mapping->path,mapping->path)) {
1265 /* open file */
1266 int fd = qemu_open(mapping->path, O_RDONLY | O_BINARY | O_LARGEFILE);
1267 if(fd<0)
1268 return -1;
1269 vvfat_close_current_file(s);
1270 s->current_fd = fd;
1271 s->current_mapping = mapping;
1273 return 0;
1276 static inline int read_cluster(BDRVVVFATState *s,int cluster_num)
1278 if(s->current_cluster != cluster_num) {
1279 int result=0;
1280 off_t offset;
1281 assert(!s->current_mapping || s->current_fd || (s->current_mapping->mode & MODE_DIRECTORY));
1282 if(!s->current_mapping
1283 || s->current_mapping->begin>cluster_num
1284 || s->current_mapping->end<=cluster_num) {
1285 /* binary search of mappings for file */
1286 mapping_t* mapping=find_mapping_for_cluster(s,cluster_num);
1288 assert(!mapping || (cluster_num>=mapping->begin && cluster_num<mapping->end));
1290 if (mapping && mapping->mode & MODE_DIRECTORY) {
1291 vvfat_close_current_file(s);
1292 s->current_mapping = mapping;
1293 read_cluster_directory:
1294 offset = s->cluster_size*(cluster_num-s->current_mapping->begin);
1295 s->cluster = (unsigned char*)s->directory.pointer+offset
1296 + 0x20*s->current_mapping->info.dir.first_dir_index;
1297 assert(((s->cluster-(unsigned char*)s->directory.pointer)%s->cluster_size)==0);
1298 assert((char*)s->cluster+s->cluster_size <= s->directory.pointer+s->directory.next*s->directory.item_size);
1299 s->current_cluster = cluster_num;
1300 return 0;
1303 if(open_file(s,mapping))
1304 return -2;
1305 } else if (s->current_mapping->mode & MODE_DIRECTORY)
1306 goto read_cluster_directory;
1308 assert(s->current_fd);
1310 offset=s->cluster_size*(cluster_num-s->current_mapping->begin)+s->current_mapping->info.file.offset;
1311 if(lseek(s->current_fd, offset, SEEK_SET)!=offset)
1312 return -3;
1313 s->cluster=s->cluster_buffer;
1314 result=read(s->current_fd,s->cluster,s->cluster_size);
1315 if(result<0) {
1316 s->current_cluster = -1;
1317 return -1;
1319 s->current_cluster = cluster_num;
1321 return 0;
1324 #ifdef DEBUG
1325 static void print_direntry(const direntry_t* direntry)
1327 int j = 0;
1328 char buffer[1024];
1330 fprintf(stderr, "direntry %p: ", direntry);
1331 if(!direntry)
1332 return;
1333 if(is_long_name(direntry)) {
1334 unsigned char* c=(unsigned char*)direntry;
1335 int i;
1336 for(i=1;i<11 && c[i] && c[i]!=0xff;i+=2)
1337 #define ADD_CHAR(c) {buffer[j] = (c); if (buffer[j] < ' ') buffer[j] = 0xb0; j++;}
1338 ADD_CHAR(c[i]);
1339 for(i=14;i<26 && c[i] && c[i]!=0xff;i+=2)
1340 ADD_CHAR(c[i]);
1341 for(i=28;i<32 && c[i] && c[i]!=0xff;i+=2)
1342 ADD_CHAR(c[i]);
1343 buffer[j] = 0;
1344 fprintf(stderr, "%s\n", buffer);
1345 } else {
1346 int i;
1347 for(i=0;i<11;i++)
1348 ADD_CHAR(direntry->name[i]);
1349 buffer[j] = 0;
1350 fprintf(stderr,"%s attributes=0x%02x begin=%d size=%d\n",
1351 buffer,
1352 direntry->attributes,
1353 begin_of_direntry(direntry),le32_to_cpu(direntry->size));
1357 static void print_mapping(const mapping_t* mapping)
1359 fprintf(stderr, "mapping (%p): begin, end = %d, %d, dir_index = %d, "
1360 "first_mapping_index = %d, name = %s, mode = 0x%x, " ,
1361 mapping, mapping->begin, mapping->end, mapping->dir_index,
1362 mapping->first_mapping_index, mapping->path, mapping->mode);
1364 if (mapping->mode & MODE_DIRECTORY)
1365 fprintf(stderr, "parent_mapping_index = %d, first_dir_index = %d\n", mapping->info.dir.parent_mapping_index, mapping->info.dir.first_dir_index);
1366 else
1367 fprintf(stderr, "offset = %d\n", mapping->info.file.offset);
1369 #endif
1371 static int vvfat_read(BlockDriverState *bs, int64_t sector_num,
1372 uint8_t *buf, int nb_sectors)
1374 BDRVVVFATState *s = bs->opaque;
1375 int i;
1377 for(i=0;i<nb_sectors;i++,sector_num++) {
1378 if (sector_num >= bs->total_sectors)
1379 return -1;
1380 if (s->qcow) {
1381 int n;
1382 if (bdrv_is_allocated(s->qcow, sector_num, nb_sectors-i, &n)) {
1383 DLOG(fprintf(stderr, "sectors %d+%d allocated\n", (int)sector_num, n));
1384 if (bdrv_read(s->qcow, sector_num, buf + i*0x200, n)) {
1385 return -1;
1387 i += n - 1;
1388 sector_num += n - 1;
1389 continue;
1391 DLOG(fprintf(stderr, "sector %d not allocated\n", (int)sector_num));
1393 if(sector_num<s->faked_sectors) {
1394 if(sector_num<s->first_sectors_number)
1395 memcpy(buf+i*0x200,&(s->first_sectors[sector_num*0x200]),0x200);
1396 else if(sector_num-s->first_sectors_number<s->sectors_per_fat)
1397 memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number)*0x200]),0x200);
1398 else if(sector_num-s->first_sectors_number-s->sectors_per_fat<s->sectors_per_fat)
1399 memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number-s->sectors_per_fat)*0x200]),0x200);
1400 } else {
1401 uint32_t sector=sector_num-s->faked_sectors,
1402 sector_offset_in_cluster=(sector%s->sectors_per_cluster),
1403 cluster_num=sector/s->sectors_per_cluster;
1404 if(cluster_num > s->cluster_count || read_cluster(s, cluster_num) != 0) {
1405 /* LATER TODO: strict: return -1; */
1406 memset(buf+i*0x200,0,0x200);
1407 continue;
1409 memcpy(buf+i*0x200,s->cluster+sector_offset_in_cluster*0x200,0x200);
1412 return 0;
1415 static coroutine_fn int vvfat_co_read(BlockDriverState *bs, int64_t sector_num,
1416 uint8_t *buf, int nb_sectors)
1418 int ret;
1419 BDRVVVFATState *s = bs->opaque;
1420 qemu_co_mutex_lock(&s->lock);
1421 ret = vvfat_read(bs, sector_num, buf, nb_sectors);
1422 qemu_co_mutex_unlock(&s->lock);
1423 return ret;
1426 /* LATER TODO: statify all functions */
1429 * Idea of the write support (use snapshot):
1431 * 1. check if all data is consistent, recording renames, modifications,
1432 * new files and directories (in s->commits).
1434 * 2. if the data is not consistent, stop committing
1436 * 3. handle renames, and create new files and directories (do not yet
1437 * write their contents)
1439 * 4. walk the directories, fixing the mapping and direntries, and marking
1440 * the handled mappings as not deleted
1442 * 5. commit the contents of the files
1444 * 6. handle deleted files and directories
1448 typedef struct commit_t {
1449 char* path;
1450 union {
1451 struct { uint32_t cluster; } rename;
1452 struct { int dir_index; uint32_t modified_offset; } writeout;
1453 struct { uint32_t first_cluster; } new_file;
1454 struct { uint32_t cluster; } mkdir;
1455 } param;
1456 /* DELETEs and RMDIRs are handled differently: see handle_deletes() */
1457 enum {
1458 ACTION_RENAME, ACTION_WRITEOUT, ACTION_NEW_FILE, ACTION_MKDIR
1459 } action;
1460 } commit_t;
1462 static void clear_commits(BDRVVVFATState* s)
1464 int i;
1465 DLOG(fprintf(stderr, "clear_commits (%d commits)\n", s->commits.next));
1466 for (i = 0; i < s->commits.next; i++) {
1467 commit_t* commit = array_get(&(s->commits), i);
1468 assert(commit->path || commit->action == ACTION_WRITEOUT);
1469 if (commit->action != ACTION_WRITEOUT) {
1470 assert(commit->path);
1471 g_free(commit->path);
1472 } else
1473 assert(commit->path == NULL);
1475 s->commits.next = 0;
1478 static void schedule_rename(BDRVVVFATState* s,
1479 uint32_t cluster, char* new_path)
1481 commit_t* commit = array_get_next(&(s->commits));
1482 commit->path = new_path;
1483 commit->param.rename.cluster = cluster;
1484 commit->action = ACTION_RENAME;
1487 static void schedule_writeout(BDRVVVFATState* s,
1488 int dir_index, uint32_t modified_offset)
1490 commit_t* commit = array_get_next(&(s->commits));
1491 commit->path = NULL;
1492 commit->param.writeout.dir_index = dir_index;
1493 commit->param.writeout.modified_offset = modified_offset;
1494 commit->action = ACTION_WRITEOUT;
1497 static void schedule_new_file(BDRVVVFATState* s,
1498 char* path, uint32_t first_cluster)
1500 commit_t* commit = array_get_next(&(s->commits));
1501 commit->path = path;
1502 commit->param.new_file.first_cluster = first_cluster;
1503 commit->action = ACTION_NEW_FILE;
1506 static void schedule_mkdir(BDRVVVFATState* s, uint32_t cluster, char* path)
1508 commit_t* commit = array_get_next(&(s->commits));
1509 commit->path = path;
1510 commit->param.mkdir.cluster = cluster;
1511 commit->action = ACTION_MKDIR;
1514 typedef struct {
1516 * Since the sequence number is at most 0x3f, and the filename
1517 * length is at most 13 times the sequence number, the maximal
1518 * filename length is 0x3f * 13 bytes.
1520 unsigned char name[0x3f * 13 + 1];
1521 int checksum, len;
1522 int sequence_number;
1523 } long_file_name;
1525 static void lfn_init(long_file_name* lfn)
1527 lfn->sequence_number = lfn->len = 0;
1528 lfn->checksum = 0x100;
1531 /* return 0 if parsed successfully, > 0 if no long name, < 0 if error */
1532 static int parse_long_name(long_file_name* lfn,
1533 const direntry_t* direntry)
1535 int i, j, offset;
1536 const unsigned char* pointer = (const unsigned char*)direntry;
1538 if (!is_long_name(direntry))
1539 return 1;
1541 if (pointer[0] & 0x40) {
1542 lfn->sequence_number = pointer[0] & 0x3f;
1543 lfn->checksum = pointer[13];
1544 lfn->name[0] = 0;
1545 lfn->name[lfn->sequence_number * 13] = 0;
1546 } else if ((pointer[0] & 0x3f) != --lfn->sequence_number)
1547 return -1;
1548 else if (pointer[13] != lfn->checksum)
1549 return -2;
1550 else if (pointer[12] || pointer[26] || pointer[27])
1551 return -3;
1553 offset = 13 * (lfn->sequence_number - 1);
1554 for (i = 0, j = 1; i < 13; i++, j+=2) {
1555 if (j == 11)
1556 j = 14;
1557 else if (j == 26)
1558 j = 28;
1560 if (pointer[j+1] == 0)
1561 lfn->name[offset + i] = pointer[j];
1562 else if (pointer[j+1] != 0xff || (pointer[0] & 0x40) == 0)
1563 return -4;
1564 else
1565 lfn->name[offset + i] = 0;
1568 if (pointer[0] & 0x40)
1569 lfn->len = offset + strlen((char*)lfn->name + offset);
1571 return 0;
1574 /* returns 0 if successful, >0 if no short_name, and <0 on error */
1575 static int parse_short_name(BDRVVVFATState* s,
1576 long_file_name* lfn, direntry_t* direntry)
1578 int i, j;
1580 if (!is_short_name(direntry))
1581 return 1;
1583 for (j = 7; j >= 0 && direntry->name[j] == ' '; j--);
1584 for (i = 0; i <= j; i++) {
1585 if (direntry->name[i] <= ' ' || direntry->name[i] > 0x7f)
1586 return -1;
1587 else if (s->downcase_short_names)
1588 lfn->name[i] = qemu_tolower(direntry->name[i]);
1589 else
1590 lfn->name[i] = direntry->name[i];
1593 for (j = 2; j >= 0 && direntry->extension[j] == ' '; j--);
1594 if (j >= 0) {
1595 lfn->name[i++] = '.';
1596 lfn->name[i + j + 1] = '\0';
1597 for (;j >= 0; j--) {
1598 if (direntry->extension[j] <= ' ' || direntry->extension[j] > 0x7f)
1599 return -2;
1600 else if (s->downcase_short_names)
1601 lfn->name[i + j] = qemu_tolower(direntry->extension[j]);
1602 else
1603 lfn->name[i + j] = direntry->extension[j];
1605 } else
1606 lfn->name[i + j + 1] = '\0';
1608 lfn->len = strlen((char*)lfn->name);
1610 return 0;
1613 static inline uint32_t modified_fat_get(BDRVVVFATState* s,
1614 unsigned int cluster)
1616 if (cluster < s->last_cluster_of_root_directory) {
1617 if (cluster + 1 == s->last_cluster_of_root_directory)
1618 return s->max_fat_value;
1619 else
1620 return cluster + 1;
1623 if (s->fat_type==32) {
1624 uint32_t* entry=((uint32_t*)s->fat2)+cluster;
1625 return le32_to_cpu(*entry);
1626 } else if (s->fat_type==16) {
1627 uint16_t* entry=((uint16_t*)s->fat2)+cluster;
1628 return le16_to_cpu(*entry);
1629 } else {
1630 const uint8_t* x=s->fat2+cluster*3/2;
1631 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
1635 static inline int cluster_was_modified(BDRVVVFATState* s, uint32_t cluster_num)
1637 int was_modified = 0;
1638 int i, dummy;
1640 if (s->qcow == NULL)
1641 return 0;
1643 for (i = 0; !was_modified && i < s->sectors_per_cluster; i++)
1644 was_modified = bdrv_is_allocated(s->qcow,
1645 cluster2sector(s, cluster_num) + i, 1, &dummy);
1647 return was_modified;
1650 static const char* get_basename(const char* path)
1652 char* basename = strrchr(path, '/');
1653 if (basename == NULL)
1654 return path;
1655 else
1656 return basename + 1; /* strip '/' */
1660 * The array s->used_clusters holds the states of the clusters. If it is
1661 * part of a file, it has bit 2 set, in case of a directory, bit 1. If it
1662 * was modified, bit 3 is set.
1663 * If any cluster is allocated, but not part of a file or directory, this
1664 * driver refuses to commit.
1666 typedef enum {
1667 USED_DIRECTORY = 1, USED_FILE = 2, USED_ANY = 3, USED_ALLOCATED = 4
1668 } used_t;
1671 * get_cluster_count_for_direntry() not only determines how many clusters
1672 * are occupied by direntry, but also if it was renamed or modified.
1674 * A file is thought to be renamed *only* if there already was a file with
1675 * exactly the same first cluster, but a different name.
1677 * Further, the files/directories handled by this function are
1678 * assumed to be *not* deleted (and *only* those).
1680 static uint32_t get_cluster_count_for_direntry(BDRVVVFATState* s,
1681 direntry_t* direntry, const char* path)
1684 * This is a little bit tricky:
1685 * IF the guest OS just inserts a cluster into the file chain,
1686 * and leaves the rest alone, (i.e. the original file had clusters
1687 * 15 -> 16, but now has 15 -> 32 -> 16), then the following happens:
1689 * - do_commit will write the cluster into the file at the given
1690 * offset, but
1692 * - the cluster which is overwritten should be moved to a later
1693 * position in the file.
1695 * I am not aware that any OS does something as braindead, but this
1696 * situation could happen anyway when not committing for a long time.
1697 * Just to be sure that this does not bite us, detect it, and copy the
1698 * contents of the clusters to-be-overwritten into the qcow.
1700 int copy_it = 0;
1701 int was_modified = 0;
1702 int32_t ret = 0;
1704 uint32_t cluster_num = begin_of_direntry(direntry);
1705 uint32_t offset = 0;
1706 int first_mapping_index = -1;
1707 mapping_t* mapping = NULL;
1708 const char* basename2 = NULL;
1710 vvfat_close_current_file(s);
1712 /* the root directory */
1713 if (cluster_num == 0)
1714 return 0;
1716 /* write support */
1717 if (s->qcow) {
1718 basename2 = get_basename(path);
1720 mapping = find_mapping_for_cluster(s, cluster_num);
1722 if (mapping) {
1723 const char* basename;
1725 assert(mapping->mode & MODE_DELETED);
1726 mapping->mode &= ~MODE_DELETED;
1728 basename = get_basename(mapping->path);
1730 assert(mapping->mode & MODE_NORMAL);
1732 /* rename */
1733 if (strcmp(basename, basename2))
1734 schedule_rename(s, cluster_num, g_strdup(path));
1735 } else if (is_file(direntry))
1736 /* new file */
1737 schedule_new_file(s, g_strdup(path), cluster_num);
1738 else {
1739 abort();
1740 return 0;
1744 while(1) {
1745 if (s->qcow) {
1746 if (!copy_it && cluster_was_modified(s, cluster_num)) {
1747 if (mapping == NULL ||
1748 mapping->begin > cluster_num ||
1749 mapping->end <= cluster_num)
1750 mapping = find_mapping_for_cluster(s, cluster_num);
1753 if (mapping &&
1754 (mapping->mode & MODE_DIRECTORY) == 0) {
1756 /* was modified in qcow */
1757 if (offset != mapping->info.file.offset + s->cluster_size
1758 * (cluster_num - mapping->begin)) {
1759 /* offset of this cluster in file chain has changed */
1760 abort();
1761 copy_it = 1;
1762 } else if (offset == 0) {
1763 const char* basename = get_basename(mapping->path);
1765 if (strcmp(basename, basename2))
1766 copy_it = 1;
1767 first_mapping_index = array_index(&(s->mapping), mapping);
1770 if (mapping->first_mapping_index != first_mapping_index
1771 && mapping->info.file.offset > 0) {
1772 abort();
1773 copy_it = 1;
1776 /* need to write out? */
1777 if (!was_modified && is_file(direntry)) {
1778 was_modified = 1;
1779 schedule_writeout(s, mapping->dir_index, offset);
1784 if (copy_it) {
1785 int i, dummy;
1787 * This is horribly inefficient, but that is okay, since
1788 * it is rarely executed, if at all.
1790 int64_t offset = cluster2sector(s, cluster_num);
1792 vvfat_close_current_file(s);
1793 for (i = 0; i < s->sectors_per_cluster; i++) {
1794 if (!bdrv_is_allocated(s->qcow, offset + i, 1, &dummy)) {
1795 if (vvfat_read(s->bs, offset, s->cluster_buffer, 1)) {
1796 return -1;
1798 if (bdrv_write(s->qcow, offset, s->cluster_buffer, 1)) {
1799 return -2;
1806 ret++;
1807 if (s->used_clusters[cluster_num] & USED_ANY)
1808 return 0;
1809 s->used_clusters[cluster_num] = USED_FILE;
1811 cluster_num = modified_fat_get(s, cluster_num);
1813 if (fat_eof(s, cluster_num))
1814 return ret;
1815 else if (cluster_num < 2 || cluster_num > s->max_fat_value - 16)
1816 return -1;
1818 offset += s->cluster_size;
1823 * This function looks at the modified data (qcow).
1824 * It returns 0 upon inconsistency or error, and the number of clusters
1825 * used by the directory, its subdirectories and their files.
1827 static int check_directory_consistency(BDRVVVFATState *s,
1828 int cluster_num, const char* path)
1830 int ret = 0;
1831 unsigned char* cluster = g_malloc(s->cluster_size);
1832 direntry_t* direntries = (direntry_t*)cluster;
1833 mapping_t* mapping = find_mapping_for_cluster(s, cluster_num);
1835 long_file_name lfn;
1836 int path_len = strlen(path);
1837 char path2[PATH_MAX + 1];
1839 assert(path_len < PATH_MAX); /* len was tested before! */
1840 pstrcpy(path2, sizeof(path2), path);
1841 path2[path_len] = '/';
1842 path2[path_len + 1] = '\0';
1844 if (mapping) {
1845 const char* basename = get_basename(mapping->path);
1846 const char* basename2 = get_basename(path);
1848 assert(mapping->mode & MODE_DIRECTORY);
1850 assert(mapping->mode & MODE_DELETED);
1851 mapping->mode &= ~MODE_DELETED;
1853 if (strcmp(basename, basename2))
1854 schedule_rename(s, cluster_num, g_strdup(path));
1855 } else
1856 /* new directory */
1857 schedule_mkdir(s, cluster_num, g_strdup(path));
1859 lfn_init(&lfn);
1860 do {
1861 int i;
1862 int subret = 0;
1864 ret++;
1866 if (s->used_clusters[cluster_num] & USED_ANY) {
1867 fprintf(stderr, "cluster %d used more than once\n", (int)cluster_num);
1868 return 0;
1870 s->used_clusters[cluster_num] = USED_DIRECTORY;
1872 DLOG(fprintf(stderr, "read cluster %d (sector %d)\n", (int)cluster_num, (int)cluster2sector(s, cluster_num)));
1873 subret = vvfat_read(s->bs, cluster2sector(s, cluster_num), cluster,
1874 s->sectors_per_cluster);
1875 if (subret) {
1876 fprintf(stderr, "Error fetching direntries\n");
1877 fail:
1878 g_free(cluster);
1879 return 0;
1882 for (i = 0; i < 0x10 * s->sectors_per_cluster; i++) {
1883 int cluster_count = 0;
1885 DLOG(fprintf(stderr, "check direntry %d:\n", i); print_direntry(direntries + i));
1886 if (is_volume_label(direntries + i) || is_dot(direntries + i) ||
1887 is_free(direntries + i))
1888 continue;
1890 subret = parse_long_name(&lfn, direntries + i);
1891 if (subret < 0) {
1892 fprintf(stderr, "Error in long name\n");
1893 goto fail;
1895 if (subret == 0 || is_free(direntries + i))
1896 continue;
1898 if (fat_chksum(direntries+i) != lfn.checksum) {
1899 subret = parse_short_name(s, &lfn, direntries + i);
1900 if (subret < 0) {
1901 fprintf(stderr, "Error in short name (%d)\n", subret);
1902 goto fail;
1904 if (subret > 0 || !strcmp((char*)lfn.name, ".")
1905 || !strcmp((char*)lfn.name, ".."))
1906 continue;
1908 lfn.checksum = 0x100; /* cannot use long name twice */
1910 if (path_len + 1 + lfn.len >= PATH_MAX) {
1911 fprintf(stderr, "Name too long: %s/%s\n", path, lfn.name);
1912 goto fail;
1914 pstrcpy(path2 + path_len + 1, sizeof(path2) - path_len - 1,
1915 (char*)lfn.name);
1917 if (is_directory(direntries + i)) {
1918 if (begin_of_direntry(direntries + i) == 0) {
1919 DLOG(fprintf(stderr, "invalid begin for directory: %s\n", path2); print_direntry(direntries + i));
1920 goto fail;
1922 cluster_count = check_directory_consistency(s,
1923 begin_of_direntry(direntries + i), path2);
1924 if (cluster_count == 0) {
1925 DLOG(fprintf(stderr, "problem in directory %s:\n", path2); print_direntry(direntries + i));
1926 goto fail;
1928 } else if (is_file(direntries + i)) {
1929 /* check file size with FAT */
1930 cluster_count = get_cluster_count_for_direntry(s, direntries + i, path2);
1931 if (cluster_count !=
1932 (le32_to_cpu(direntries[i].size) + s->cluster_size
1933 - 1) / s->cluster_size) {
1934 DLOG(fprintf(stderr, "Cluster count mismatch\n"));
1935 goto fail;
1937 } else
1938 abort(); /* cluster_count = 0; */
1940 ret += cluster_count;
1943 cluster_num = modified_fat_get(s, cluster_num);
1944 } while(!fat_eof(s, cluster_num));
1946 g_free(cluster);
1947 return ret;
1950 /* returns 1 on success */
1951 static int is_consistent(BDRVVVFATState* s)
1953 int i, check;
1954 int used_clusters_count = 0;
1956 DLOG(checkpoint());
1958 * - get modified FAT
1959 * - compare the two FATs (TODO)
1960 * - get buffer for marking used clusters
1961 * - recurse direntries from root (using bs->bdrv_read to make
1962 * sure to get the new data)
1963 * - check that the FAT agrees with the size
1964 * - count the number of clusters occupied by this directory and
1965 * its files
1966 * - check that the cumulative used cluster count agrees with the
1967 * FAT
1968 * - if all is fine, return number of used clusters
1970 if (s->fat2 == NULL) {
1971 int size = 0x200 * s->sectors_per_fat;
1972 s->fat2 = g_malloc(size);
1973 memcpy(s->fat2, s->fat.pointer, size);
1975 check = vvfat_read(s->bs,
1976 s->first_sectors_number, s->fat2, s->sectors_per_fat);
1977 if (check) {
1978 fprintf(stderr, "Could not copy fat\n");
1979 return 0;
1981 assert (s->used_clusters);
1982 for (i = 0; i < sector2cluster(s, s->sector_count); i++)
1983 s->used_clusters[i] &= ~USED_ANY;
1985 clear_commits(s);
1987 /* mark every mapped file/directory as deleted.
1988 * (check_directory_consistency() will unmark those still present). */
1989 if (s->qcow)
1990 for (i = 0; i < s->mapping.next; i++) {
1991 mapping_t* mapping = array_get(&(s->mapping), i);
1992 if (mapping->first_mapping_index < 0)
1993 mapping->mode |= MODE_DELETED;
1996 used_clusters_count = check_directory_consistency(s, 0, s->path);
1997 if (used_clusters_count <= 0) {
1998 DLOG(fprintf(stderr, "problem in directory\n"));
1999 return 0;
2002 check = s->last_cluster_of_root_directory;
2003 for (i = check; i < sector2cluster(s, s->sector_count); i++) {
2004 if (modified_fat_get(s, i)) {
2005 if(!s->used_clusters[i]) {
2006 DLOG(fprintf(stderr, "FAT was modified (%d), but cluster is not used?\n", i));
2007 return 0;
2009 check++;
2012 if (s->used_clusters[i] == USED_ALLOCATED) {
2013 /* allocated, but not used... */
2014 DLOG(fprintf(stderr, "unused, modified cluster: %d\n", i));
2015 return 0;
2019 if (check != used_clusters_count)
2020 return 0;
2022 return used_clusters_count;
2025 static inline void adjust_mapping_indices(BDRVVVFATState* s,
2026 int offset, int adjust)
2028 int i;
2030 for (i = 0; i < s->mapping.next; i++) {
2031 mapping_t* mapping = array_get(&(s->mapping), i);
2033 #define ADJUST_MAPPING_INDEX(name) \
2034 if (mapping->name >= offset) \
2035 mapping->name += adjust
2037 ADJUST_MAPPING_INDEX(first_mapping_index);
2038 if (mapping->mode & MODE_DIRECTORY)
2039 ADJUST_MAPPING_INDEX(info.dir.parent_mapping_index);
2043 /* insert or update mapping */
2044 static mapping_t* insert_mapping(BDRVVVFATState* s,
2045 uint32_t begin, uint32_t end)
2048 * - find mapping where mapping->begin >= begin,
2049 * - if mapping->begin > begin: insert
2050 * - adjust all references to mappings!
2051 * - else: adjust
2052 * - replace name
2054 int index = find_mapping_for_cluster_aux(s, begin, 0, s->mapping.next);
2055 mapping_t* mapping = NULL;
2056 mapping_t* first_mapping = array_get(&(s->mapping), 0);
2058 if (index < s->mapping.next && (mapping = array_get(&(s->mapping), index))
2059 && mapping->begin < begin) {
2060 mapping->end = begin;
2061 index++;
2062 mapping = array_get(&(s->mapping), index);
2064 if (index >= s->mapping.next || mapping->begin > begin) {
2065 mapping = array_insert(&(s->mapping), index, 1);
2066 mapping->path = NULL;
2067 adjust_mapping_indices(s, index, +1);
2070 mapping->begin = begin;
2071 mapping->end = end;
2073 DLOG(mapping_t* next_mapping;
2074 assert(index + 1 >= s->mapping.next ||
2075 ((next_mapping = array_get(&(s->mapping), index + 1)) &&
2076 next_mapping->begin >= end)));
2078 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
2079 s->current_mapping = array_get(&(s->mapping),
2080 s->current_mapping - first_mapping);
2082 return mapping;
2085 static int remove_mapping(BDRVVVFATState* s, int mapping_index)
2087 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
2088 mapping_t* first_mapping = array_get(&(s->mapping), 0);
2090 /* free mapping */
2091 if (mapping->first_mapping_index < 0) {
2092 g_free(mapping->path);
2095 /* remove from s->mapping */
2096 array_remove(&(s->mapping), mapping_index);
2098 /* adjust all references to mappings */
2099 adjust_mapping_indices(s, mapping_index, -1);
2101 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
2102 s->current_mapping = array_get(&(s->mapping),
2103 s->current_mapping - first_mapping);
2105 return 0;
2108 static void adjust_dirindices(BDRVVVFATState* s, int offset, int adjust)
2110 int i;
2111 for (i = 0; i < s->mapping.next; i++) {
2112 mapping_t* mapping = array_get(&(s->mapping), i);
2113 if (mapping->dir_index >= offset)
2114 mapping->dir_index += adjust;
2115 if ((mapping->mode & MODE_DIRECTORY) &&
2116 mapping->info.dir.first_dir_index >= offset)
2117 mapping->info.dir.first_dir_index += adjust;
2121 static direntry_t* insert_direntries(BDRVVVFATState* s,
2122 int dir_index, int count)
2125 * make room in s->directory,
2126 * adjust_dirindices
2128 direntry_t* result = array_insert(&(s->directory), dir_index, count);
2129 if (result == NULL)
2130 return NULL;
2131 adjust_dirindices(s, dir_index, count);
2132 return result;
2135 static int remove_direntries(BDRVVVFATState* s, int dir_index, int count)
2137 int ret = array_remove_slice(&(s->directory), dir_index, count);
2138 if (ret)
2139 return ret;
2140 adjust_dirindices(s, dir_index, -count);
2141 return 0;
2145 * Adapt the mappings of the cluster chain starting at first cluster
2146 * (i.e. if a file starts at first_cluster, the chain is followed according
2147 * to the modified fat, and the corresponding entries in s->mapping are
2148 * adjusted)
2150 static int commit_mappings(BDRVVVFATState* s,
2151 uint32_t first_cluster, int dir_index)
2153 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2154 direntry_t* direntry = array_get(&(s->directory), dir_index);
2155 uint32_t cluster = first_cluster;
2157 vvfat_close_current_file(s);
2159 assert(mapping);
2160 assert(mapping->begin == first_cluster);
2161 mapping->first_mapping_index = -1;
2162 mapping->dir_index = dir_index;
2163 mapping->mode = (dir_index <= 0 || is_directory(direntry)) ?
2164 MODE_DIRECTORY : MODE_NORMAL;
2166 while (!fat_eof(s, cluster)) {
2167 uint32_t c, c1;
2169 for (c = cluster, c1 = modified_fat_get(s, c); c + 1 == c1;
2170 c = c1, c1 = modified_fat_get(s, c1));
2172 c++;
2173 if (c > mapping->end) {
2174 int index = array_index(&(s->mapping), mapping);
2175 int i, max_i = s->mapping.next - index;
2176 for (i = 1; i < max_i && mapping[i].begin < c; i++);
2177 while (--i > 0)
2178 remove_mapping(s, index + 1);
2180 assert(mapping == array_get(&(s->mapping), s->mapping.next - 1)
2181 || mapping[1].begin >= c);
2182 mapping->end = c;
2184 if (!fat_eof(s, c1)) {
2185 int i = find_mapping_for_cluster_aux(s, c1, 0, s->mapping.next);
2186 mapping_t* next_mapping = i >= s->mapping.next ? NULL :
2187 array_get(&(s->mapping), i);
2189 if (next_mapping == NULL || next_mapping->begin > c1) {
2190 int i1 = array_index(&(s->mapping), mapping);
2192 next_mapping = insert_mapping(s, c1, c1+1);
2194 if (c1 < c)
2195 i1++;
2196 mapping = array_get(&(s->mapping), i1);
2199 next_mapping->dir_index = mapping->dir_index;
2200 next_mapping->first_mapping_index =
2201 mapping->first_mapping_index < 0 ?
2202 array_index(&(s->mapping), mapping) :
2203 mapping->first_mapping_index;
2204 next_mapping->path = mapping->path;
2205 next_mapping->mode = mapping->mode;
2206 next_mapping->read_only = mapping->read_only;
2207 if (mapping->mode & MODE_DIRECTORY) {
2208 next_mapping->info.dir.parent_mapping_index =
2209 mapping->info.dir.parent_mapping_index;
2210 next_mapping->info.dir.first_dir_index =
2211 mapping->info.dir.first_dir_index +
2212 0x10 * s->sectors_per_cluster *
2213 (mapping->end - mapping->begin);
2214 } else
2215 next_mapping->info.file.offset = mapping->info.file.offset +
2216 mapping->end - mapping->begin;
2218 mapping = next_mapping;
2221 cluster = c1;
2224 return 0;
2227 static int commit_direntries(BDRVVVFATState* s,
2228 int dir_index, int parent_mapping_index)
2230 direntry_t* direntry = array_get(&(s->directory), dir_index);
2231 uint32_t first_cluster = dir_index == 0 ? 0 : begin_of_direntry(direntry);
2232 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2234 int factor = 0x10 * s->sectors_per_cluster;
2235 int old_cluster_count, new_cluster_count;
2236 int current_dir_index = mapping->info.dir.first_dir_index;
2237 int first_dir_index = current_dir_index;
2238 int ret, i;
2239 uint32_t c;
2241 DLOG(fprintf(stderr, "commit_direntries for %s, parent_mapping_index %d\n", mapping->path, parent_mapping_index));
2243 assert(direntry);
2244 assert(mapping);
2245 assert(mapping->begin == first_cluster);
2246 assert(mapping->info.dir.first_dir_index < s->directory.next);
2247 assert(mapping->mode & MODE_DIRECTORY);
2248 assert(dir_index == 0 || is_directory(direntry));
2250 mapping->info.dir.parent_mapping_index = parent_mapping_index;
2252 if (first_cluster == 0) {
2253 old_cluster_count = new_cluster_count =
2254 s->last_cluster_of_root_directory;
2255 } else {
2256 for (old_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2257 c = fat_get(s, c))
2258 old_cluster_count++;
2260 for (new_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2261 c = modified_fat_get(s, c))
2262 new_cluster_count++;
2265 if (new_cluster_count > old_cluster_count) {
2266 if (insert_direntries(s,
2267 current_dir_index + factor * old_cluster_count,
2268 factor * (new_cluster_count - old_cluster_count)) == NULL)
2269 return -1;
2270 } else if (new_cluster_count < old_cluster_count)
2271 remove_direntries(s,
2272 current_dir_index + factor * new_cluster_count,
2273 factor * (old_cluster_count - new_cluster_count));
2275 for (c = first_cluster; !fat_eof(s, c); c = modified_fat_get(s, c)) {
2276 void* direntry = array_get(&(s->directory), current_dir_index);
2277 int ret = vvfat_read(s->bs, cluster2sector(s, c), direntry,
2278 s->sectors_per_cluster);
2279 if (ret)
2280 return ret;
2281 assert(!strncmp(s->directory.pointer, "QEMU", 4));
2282 current_dir_index += factor;
2285 ret = commit_mappings(s, first_cluster, dir_index);
2286 if (ret)
2287 return ret;
2289 /* recurse */
2290 for (i = 0; i < factor * new_cluster_count; i++) {
2291 direntry = array_get(&(s->directory), first_dir_index + i);
2292 if (is_directory(direntry) && !is_dot(direntry)) {
2293 mapping = find_mapping_for_cluster(s, first_cluster);
2294 assert(mapping->mode & MODE_DIRECTORY);
2295 ret = commit_direntries(s, first_dir_index + i,
2296 array_index(&(s->mapping), mapping));
2297 if (ret)
2298 return ret;
2302 return 0;
2305 /* commit one file (adjust contents, adjust mapping),
2306 return first_mapping_index */
2307 static int commit_one_file(BDRVVVFATState* s,
2308 int dir_index, uint32_t offset)
2310 direntry_t* direntry = array_get(&(s->directory), dir_index);
2311 uint32_t c = begin_of_direntry(direntry);
2312 uint32_t first_cluster = c;
2313 mapping_t* mapping = find_mapping_for_cluster(s, c);
2314 uint32_t size = filesize_of_direntry(direntry);
2315 char* cluster = g_malloc(s->cluster_size);
2316 uint32_t i;
2317 int fd = 0;
2319 assert(offset < size);
2320 assert((offset % s->cluster_size) == 0);
2322 for (i = s->cluster_size; i < offset; i += s->cluster_size)
2323 c = modified_fat_get(s, c);
2325 fd = qemu_open(mapping->path, O_RDWR | O_CREAT | O_BINARY, 0666);
2326 if (fd < 0) {
2327 fprintf(stderr, "Could not open %s... (%s, %d)\n", mapping->path,
2328 strerror(errno), errno);
2329 g_free(cluster);
2330 return fd;
2332 if (offset > 0) {
2333 if (lseek(fd, offset, SEEK_SET) != offset) {
2334 qemu_close(fd);
2335 g_free(cluster);
2336 return -3;
2340 while (offset < size) {
2341 uint32_t c1;
2342 int rest_size = (size - offset > s->cluster_size ?
2343 s->cluster_size : size - offset);
2344 int ret;
2346 c1 = modified_fat_get(s, c);
2348 assert((size - offset == 0 && fat_eof(s, c)) ||
2349 (size > offset && c >=2 && !fat_eof(s, c)));
2351 ret = vvfat_read(s->bs, cluster2sector(s, c),
2352 (uint8_t*)cluster, (rest_size + 0x1ff) / 0x200);
2354 if (ret < 0) {
2355 qemu_close(fd);
2356 g_free(cluster);
2357 return ret;
2360 if (write(fd, cluster, rest_size) < 0) {
2361 qemu_close(fd);
2362 g_free(cluster);
2363 return -2;
2366 offset += rest_size;
2367 c = c1;
2370 if (ftruncate(fd, size)) {
2371 perror("ftruncate()");
2372 qemu_close(fd);
2373 g_free(cluster);
2374 return -4;
2376 qemu_close(fd);
2377 g_free(cluster);
2379 return commit_mappings(s, first_cluster, dir_index);
2382 #ifdef DEBUG
2383 /* test, if all mappings point to valid direntries */
2384 static void check1(BDRVVVFATState* s)
2386 int i;
2387 for (i = 0; i < s->mapping.next; i++) {
2388 mapping_t* mapping = array_get(&(s->mapping), i);
2389 if (mapping->mode & MODE_DELETED) {
2390 fprintf(stderr, "deleted\n");
2391 continue;
2393 assert(mapping->dir_index < s->directory.next);
2394 direntry_t* direntry = array_get(&(s->directory), mapping->dir_index);
2395 assert(mapping->begin == begin_of_direntry(direntry) || mapping->first_mapping_index >= 0);
2396 if (mapping->mode & MODE_DIRECTORY) {
2397 assert(mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster * (mapping->end - mapping->begin) <= s->directory.next);
2398 assert((mapping->info.dir.first_dir_index % (0x10 * s->sectors_per_cluster)) == 0);
2403 /* test, if all direntries have mappings */
2404 static void check2(BDRVVVFATState* s)
2406 int i;
2407 int first_mapping = -1;
2409 for (i = 0; i < s->directory.next; i++) {
2410 direntry_t* direntry = array_get(&(s->directory), i);
2412 if (is_short_name(direntry) && begin_of_direntry(direntry)) {
2413 mapping_t* mapping = find_mapping_for_cluster(s, begin_of_direntry(direntry));
2414 assert(mapping);
2415 assert(mapping->dir_index == i || is_dot(direntry));
2416 assert(mapping->begin == begin_of_direntry(direntry) || is_dot(direntry));
2419 if ((i % (0x10 * s->sectors_per_cluster)) == 0) {
2420 /* cluster start */
2421 int j, count = 0;
2423 for (j = 0; j < s->mapping.next; j++) {
2424 mapping_t* mapping = array_get(&(s->mapping), j);
2425 if (mapping->mode & MODE_DELETED)
2426 continue;
2427 if (mapping->mode & MODE_DIRECTORY) {
2428 if (mapping->info.dir.first_dir_index <= i && mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster > i) {
2429 assert(++count == 1);
2430 if (mapping->first_mapping_index == -1)
2431 first_mapping = array_index(&(s->mapping), mapping);
2432 else
2433 assert(first_mapping == mapping->first_mapping_index);
2434 if (mapping->info.dir.parent_mapping_index < 0)
2435 assert(j == 0);
2436 else {
2437 mapping_t* parent = array_get(&(s->mapping), mapping->info.dir.parent_mapping_index);
2438 assert(parent->mode & MODE_DIRECTORY);
2439 assert(parent->info.dir.first_dir_index < mapping->info.dir.first_dir_index);
2444 if (count == 0)
2445 first_mapping = -1;
2449 #endif
2451 static int handle_renames_and_mkdirs(BDRVVVFATState* s)
2453 int i;
2455 #ifdef DEBUG
2456 fprintf(stderr, "handle_renames\n");
2457 for (i = 0; i < s->commits.next; i++) {
2458 commit_t* commit = array_get(&(s->commits), i);
2459 fprintf(stderr, "%d, %s (%d, %d)\n", i, commit->path ? commit->path : "(null)", commit->param.rename.cluster, commit->action);
2461 #endif
2463 for (i = 0; i < s->commits.next;) {
2464 commit_t* commit = array_get(&(s->commits), i);
2465 if (commit->action == ACTION_RENAME) {
2466 mapping_t* mapping = find_mapping_for_cluster(s,
2467 commit->param.rename.cluster);
2468 char* old_path = mapping->path;
2470 assert(commit->path);
2471 mapping->path = commit->path;
2472 if (rename(old_path, mapping->path))
2473 return -2;
2475 if (mapping->mode & MODE_DIRECTORY) {
2476 int l1 = strlen(mapping->path);
2477 int l2 = strlen(old_path);
2478 int diff = l1 - l2;
2479 direntry_t* direntry = array_get(&(s->directory),
2480 mapping->info.dir.first_dir_index);
2481 uint32_t c = mapping->begin;
2482 int i = 0;
2484 /* recurse */
2485 while (!fat_eof(s, c)) {
2486 do {
2487 direntry_t* d = direntry + i;
2489 if (is_file(d) || (is_directory(d) && !is_dot(d))) {
2490 mapping_t* m = find_mapping_for_cluster(s,
2491 begin_of_direntry(d));
2492 int l = strlen(m->path);
2493 char* new_path = g_malloc(l + diff + 1);
2495 assert(!strncmp(m->path, mapping->path, l2));
2497 pstrcpy(new_path, l + diff + 1, mapping->path);
2498 pstrcpy(new_path + l1, l + diff + 1 - l1,
2499 m->path + l2);
2501 schedule_rename(s, m->begin, new_path);
2503 i++;
2504 } while((i % (0x10 * s->sectors_per_cluster)) != 0);
2505 c = fat_get(s, c);
2509 g_free(old_path);
2510 array_remove(&(s->commits), i);
2511 continue;
2512 } else if (commit->action == ACTION_MKDIR) {
2513 mapping_t* mapping;
2514 int j, parent_path_len;
2516 #ifdef __MINGW32__
2517 if (mkdir(commit->path))
2518 return -5;
2519 #else
2520 if (mkdir(commit->path, 0755))
2521 return -5;
2522 #endif
2524 mapping = insert_mapping(s, commit->param.mkdir.cluster,
2525 commit->param.mkdir.cluster + 1);
2526 if (mapping == NULL)
2527 return -6;
2529 mapping->mode = MODE_DIRECTORY;
2530 mapping->read_only = 0;
2531 mapping->path = commit->path;
2532 j = s->directory.next;
2533 assert(j);
2534 insert_direntries(s, s->directory.next,
2535 0x10 * s->sectors_per_cluster);
2536 mapping->info.dir.first_dir_index = j;
2538 parent_path_len = strlen(commit->path)
2539 - strlen(get_basename(commit->path)) - 1;
2540 for (j = 0; j < s->mapping.next; j++) {
2541 mapping_t* m = array_get(&(s->mapping), j);
2542 if (m->first_mapping_index < 0 && m != mapping &&
2543 !strncmp(m->path, mapping->path, parent_path_len) &&
2544 strlen(m->path) == parent_path_len)
2545 break;
2547 assert(j < s->mapping.next);
2548 mapping->info.dir.parent_mapping_index = j;
2550 array_remove(&(s->commits), i);
2551 continue;
2554 i++;
2556 return 0;
2560 * TODO: make sure that the short name is not matching *another* file
2562 static int handle_commits(BDRVVVFATState* s)
2564 int i, fail = 0;
2566 vvfat_close_current_file(s);
2568 for (i = 0; !fail && i < s->commits.next; i++) {
2569 commit_t* commit = array_get(&(s->commits), i);
2570 switch(commit->action) {
2571 case ACTION_RENAME: case ACTION_MKDIR:
2572 abort();
2573 fail = -2;
2574 break;
2575 case ACTION_WRITEOUT: {
2576 #ifndef NDEBUG
2577 /* these variables are only used by assert() below */
2578 direntry_t* entry = array_get(&(s->directory),
2579 commit->param.writeout.dir_index);
2580 uint32_t begin = begin_of_direntry(entry);
2581 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2582 #endif
2584 assert(mapping);
2585 assert(mapping->begin == begin);
2586 assert(commit->path == NULL);
2588 if (commit_one_file(s, commit->param.writeout.dir_index,
2589 commit->param.writeout.modified_offset))
2590 fail = -3;
2592 break;
2594 case ACTION_NEW_FILE: {
2595 int begin = commit->param.new_file.first_cluster;
2596 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2597 direntry_t* entry;
2598 int i;
2600 /* find direntry */
2601 for (i = 0; i < s->directory.next; i++) {
2602 entry = array_get(&(s->directory), i);
2603 if (is_file(entry) && begin_of_direntry(entry) == begin)
2604 break;
2607 if (i >= s->directory.next) {
2608 fail = -6;
2609 continue;
2612 /* make sure there exists an initial mapping */
2613 if (mapping && mapping->begin != begin) {
2614 mapping->end = begin;
2615 mapping = NULL;
2617 if (mapping == NULL) {
2618 mapping = insert_mapping(s, begin, begin+1);
2620 /* most members will be fixed in commit_mappings() */
2621 assert(commit->path);
2622 mapping->path = commit->path;
2623 mapping->read_only = 0;
2624 mapping->mode = MODE_NORMAL;
2625 mapping->info.file.offset = 0;
2627 if (commit_one_file(s, i, 0))
2628 fail = -7;
2630 break;
2632 default:
2633 abort();
2636 if (i > 0 && array_remove_slice(&(s->commits), 0, i))
2637 return -1;
2638 return fail;
2641 static int handle_deletes(BDRVVVFATState* s)
2643 int i, deferred = 1, deleted = 1;
2645 /* delete files corresponding to mappings marked as deleted */
2646 /* handle DELETEs and unused mappings (modified_fat_get(s, mapping->begin) == 0) */
2647 while (deferred && deleted) {
2648 deferred = 0;
2649 deleted = 0;
2651 for (i = 1; i < s->mapping.next; i++) {
2652 mapping_t* mapping = array_get(&(s->mapping), i);
2653 if (mapping->mode & MODE_DELETED) {
2654 direntry_t* entry = array_get(&(s->directory),
2655 mapping->dir_index);
2657 if (is_free(entry)) {
2658 /* remove file/directory */
2659 if (mapping->mode & MODE_DIRECTORY) {
2660 int j, next_dir_index = s->directory.next,
2661 first_dir_index = mapping->info.dir.first_dir_index;
2663 if (rmdir(mapping->path) < 0) {
2664 if (errno == ENOTEMPTY) {
2665 deferred++;
2666 continue;
2667 } else
2668 return -5;
2671 for (j = 1; j < s->mapping.next; j++) {
2672 mapping_t* m = array_get(&(s->mapping), j);
2673 if (m->mode & MODE_DIRECTORY &&
2674 m->info.dir.first_dir_index >
2675 first_dir_index &&
2676 m->info.dir.first_dir_index <
2677 next_dir_index)
2678 next_dir_index =
2679 m->info.dir.first_dir_index;
2681 remove_direntries(s, first_dir_index,
2682 next_dir_index - first_dir_index);
2684 deleted++;
2686 } else {
2687 if (unlink(mapping->path))
2688 return -4;
2689 deleted++;
2691 DLOG(fprintf(stderr, "DELETE (%d)\n", i); print_mapping(mapping); print_direntry(entry));
2692 remove_mapping(s, i);
2697 return 0;
2701 * synchronize mapping with new state:
2703 * - copy FAT (with bdrv_read)
2704 * - mark all filenames corresponding to mappings as deleted
2705 * - recurse direntries from root (using bs->bdrv_read)
2706 * - delete files corresponding to mappings marked as deleted
2708 static int do_commit(BDRVVVFATState* s)
2710 int ret = 0;
2712 /* the real meat are the commits. Nothing to do? Move along! */
2713 if (s->commits.next == 0)
2714 return 0;
2716 vvfat_close_current_file(s);
2718 ret = handle_renames_and_mkdirs(s);
2719 if (ret) {
2720 fprintf(stderr, "Error handling renames (%d)\n", ret);
2721 abort();
2722 return ret;
2725 /* copy FAT (with bdrv_read) */
2726 memcpy(s->fat.pointer, s->fat2, 0x200 * s->sectors_per_fat);
2728 /* recurse direntries from root (using bs->bdrv_read) */
2729 ret = commit_direntries(s, 0, -1);
2730 if (ret) {
2731 fprintf(stderr, "Fatal: error while committing (%d)\n", ret);
2732 abort();
2733 return ret;
2736 ret = handle_commits(s);
2737 if (ret) {
2738 fprintf(stderr, "Error handling commits (%d)\n", ret);
2739 abort();
2740 return ret;
2743 ret = handle_deletes(s);
2744 if (ret) {
2745 fprintf(stderr, "Error deleting\n");
2746 abort();
2747 return ret;
2750 if (s->qcow->drv->bdrv_make_empty) {
2751 s->qcow->drv->bdrv_make_empty(s->qcow);
2754 memset(s->used_clusters, 0, sector2cluster(s, s->sector_count));
2756 DLOG(checkpoint());
2757 return 0;
2760 static int try_commit(BDRVVVFATState* s)
2762 vvfat_close_current_file(s);
2763 DLOG(checkpoint());
2764 if(!is_consistent(s))
2765 return -1;
2766 return do_commit(s);
2769 static int vvfat_write(BlockDriverState *bs, int64_t sector_num,
2770 const uint8_t *buf, int nb_sectors)
2772 BDRVVVFATState *s = bs->opaque;
2773 int i, ret;
2775 DLOG(checkpoint());
2777 /* Check if we're operating in read-only mode */
2778 if (s->qcow == NULL) {
2779 return -EACCES;
2782 vvfat_close_current_file(s);
2785 * Some sanity checks:
2786 * - do not allow writing to the boot sector
2787 * - do not allow to write non-ASCII filenames
2790 if (sector_num < s->first_sectors_number)
2791 return -1;
2793 for (i = sector2cluster(s, sector_num);
2794 i <= sector2cluster(s, sector_num + nb_sectors - 1);) {
2795 mapping_t* mapping = find_mapping_for_cluster(s, i);
2796 if (mapping) {
2797 if (mapping->read_only) {
2798 fprintf(stderr, "Tried to write to write-protected file %s\n",
2799 mapping->path);
2800 return -1;
2803 if (mapping->mode & MODE_DIRECTORY) {
2804 int begin = cluster2sector(s, i);
2805 int end = begin + s->sectors_per_cluster, k;
2806 int dir_index;
2807 const direntry_t* direntries;
2808 long_file_name lfn;
2810 lfn_init(&lfn);
2812 if (begin < sector_num)
2813 begin = sector_num;
2814 if (end > sector_num + nb_sectors)
2815 end = sector_num + nb_sectors;
2816 dir_index = mapping->dir_index +
2817 0x10 * (begin - mapping->begin * s->sectors_per_cluster);
2818 direntries = (direntry_t*)(buf + 0x200 * (begin - sector_num));
2820 for (k = 0; k < (end - begin) * 0x10; k++) {
2821 /* do not allow non-ASCII filenames */
2822 if (parse_long_name(&lfn, direntries + k) < 0) {
2823 fprintf(stderr, "Warning: non-ASCII filename\n");
2824 return -1;
2826 /* no access to the direntry of a read-only file */
2827 else if (is_short_name(direntries+k) &&
2828 (direntries[k].attributes & 1)) {
2829 if (memcmp(direntries + k,
2830 array_get(&(s->directory), dir_index + k),
2831 sizeof(direntry_t))) {
2832 fprintf(stderr, "Warning: tried to write to write-protected file\n");
2833 return -1;
2838 i = mapping->end;
2839 } else
2840 i++;
2844 * Use qcow backend. Commit later.
2846 DLOG(fprintf(stderr, "Write to qcow backend: %d + %d\n", (int)sector_num, nb_sectors));
2847 ret = bdrv_write(s->qcow, sector_num, buf, nb_sectors);
2848 if (ret < 0) {
2849 fprintf(stderr, "Error writing to qcow backend\n");
2850 return ret;
2853 for (i = sector2cluster(s, sector_num);
2854 i <= sector2cluster(s, sector_num + nb_sectors - 1); i++)
2855 if (i >= 0)
2856 s->used_clusters[i] |= USED_ALLOCATED;
2858 DLOG(checkpoint());
2859 /* TODO: add timeout */
2860 try_commit(s);
2862 DLOG(checkpoint());
2863 return 0;
2866 static coroutine_fn int vvfat_co_write(BlockDriverState *bs, int64_t sector_num,
2867 const uint8_t *buf, int nb_sectors)
2869 int ret;
2870 BDRVVVFATState *s = bs->opaque;
2871 qemu_co_mutex_lock(&s->lock);
2872 ret = vvfat_write(bs, sector_num, buf, nb_sectors);
2873 qemu_co_mutex_unlock(&s->lock);
2874 return ret;
2877 static int coroutine_fn vvfat_co_is_allocated(BlockDriverState *bs,
2878 int64_t sector_num, int nb_sectors, int* n)
2880 BDRVVVFATState* s = bs->opaque;
2881 *n = s->sector_count - sector_num;
2882 if (*n > nb_sectors)
2883 *n = nb_sectors;
2884 else if (*n < 0)
2885 return 0;
2886 return 1;
2889 static int write_target_commit(BlockDriverState *bs, int64_t sector_num,
2890 const uint8_t* buffer, int nb_sectors) {
2891 BDRVVVFATState* s = *((BDRVVVFATState**) bs->opaque);
2892 return try_commit(s);
2895 static void write_target_close(BlockDriverState *bs) {
2896 BDRVVVFATState* s = *((BDRVVVFATState**) bs->opaque);
2897 bdrv_delete(s->qcow);
2898 g_free(s->qcow_filename);
2901 static BlockDriver vvfat_write_target = {
2902 .format_name = "vvfat_write_target",
2903 .bdrv_write = write_target_commit,
2904 .bdrv_close = write_target_close,
2907 static int enable_write_target(BDRVVVFATState *s)
2909 BlockDriver *bdrv_qcow;
2910 QEMUOptionParameter *options;
2911 int ret;
2912 int size = sector2cluster(s, s->sector_count);
2913 s->used_clusters = calloc(size, 1);
2915 array_init(&(s->commits), sizeof(commit_t));
2917 s->qcow_filename = g_malloc(1024);
2918 ret = get_tmp_filename(s->qcow_filename, 1024);
2919 if (ret < 0) {
2920 goto err;
2923 bdrv_qcow = bdrv_find_format("qcow");
2924 options = parse_option_parameters("", bdrv_qcow->create_options, NULL);
2925 set_option_parameter_int(options, BLOCK_OPT_SIZE, s->sector_count * 512);
2926 set_option_parameter(options, BLOCK_OPT_BACKING_FILE, "fat:");
2928 ret = bdrv_create(bdrv_qcow, s->qcow_filename, options);
2929 if (ret < 0) {
2930 goto err;
2933 s->qcow = bdrv_new("");
2935 ret = bdrv_open(s->qcow, s->qcow_filename, NULL,
2936 BDRV_O_RDWR | BDRV_O_CACHE_WB | BDRV_O_NO_FLUSH, bdrv_qcow);
2937 if (ret < 0) {
2938 bdrv_delete(s->qcow);
2939 goto err;
2942 #ifndef _WIN32
2943 unlink(s->qcow_filename);
2944 #endif
2946 s->bs->backing_hd = calloc(sizeof(BlockDriverState), 1);
2947 s->bs->backing_hd->drv = &vvfat_write_target;
2948 s->bs->backing_hd->opaque = g_malloc(sizeof(void*));
2949 *(void**)s->bs->backing_hd->opaque = s;
2951 return 0;
2953 err:
2954 g_free(s->qcow_filename);
2955 s->qcow_filename = NULL;
2956 return ret;
2959 static void vvfat_close(BlockDriverState *bs)
2961 BDRVVVFATState *s = bs->opaque;
2963 vvfat_close_current_file(s);
2964 array_free(&(s->fat));
2965 array_free(&(s->directory));
2966 array_free(&(s->mapping));
2967 g_free(s->cluster_buffer);
2969 if (s->qcow) {
2970 migrate_del_blocker(s->migration_blocker);
2971 error_free(s->migration_blocker);
2975 static BlockDriver bdrv_vvfat = {
2976 .format_name = "vvfat",
2977 .protocol_name = "fat",
2978 .instance_size = sizeof(BDRVVVFATState),
2980 .bdrv_parse_filename = vvfat_parse_filename,
2981 .bdrv_file_open = vvfat_open,
2982 .bdrv_close = vvfat_close,
2983 .bdrv_rebind = vvfat_rebind,
2985 .bdrv_read = vvfat_co_read,
2986 .bdrv_write = vvfat_co_write,
2987 .bdrv_co_is_allocated = vvfat_co_is_allocated,
2990 static void bdrv_vvfat_init(void)
2992 bdrv_register(&bdrv_vvfat);
2995 block_init(bdrv_vvfat_init);
2997 #ifdef DEBUG
2998 static void checkpoint(void) {
2999 assert(((mapping_t*)array_get(&(vvv->mapping), 0))->end == 2);
3000 check1(vvv);
3001 check2(vvv);
3002 assert(!vvv->current_mapping || vvv->current_fd || (vvv->current_mapping->mode & MODE_DIRECTORY));
3003 #if 0
3004 if (((direntry_t*)vvv->directory.pointer)[1].attributes != 0xf)
3005 fprintf(stderr, "Nonono!\n");
3006 mapping_t* mapping;
3007 direntry_t* direntry;
3008 assert(vvv->mapping.size >= vvv->mapping.item_size * vvv->mapping.next);
3009 assert(vvv->directory.size >= vvv->directory.item_size * vvv->directory.next);
3010 if (vvv->mapping.next<47)
3011 return;
3012 assert((mapping = array_get(&(vvv->mapping), 47)));
3013 assert(mapping->dir_index < vvv->directory.next);
3014 direntry = array_get(&(vvv->directory), mapping->dir_index);
3015 assert(!memcmp(direntry->name, "USB H ", 11) || direntry->name[0]==0);
3016 #endif
3018 #endif