hw/acpi/aml-build: Add ToUUID macro
[qemu/ar7.git] / block / vvfat.c
blobe803589675e6db57a0b648ce8da2e04c6d8b9a2c
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 static void checkpoint(void);
57 #ifdef __MINGW32__
58 void nonono(const char* file, int line, const char* msg) {
59 fprintf(stderr, "Nonono! %s:%d %s\n", file, line, msg);
60 exit(-5);
62 #undef assert
63 #define assert(a) do {if (!(a)) nonono(__FILE__, __LINE__, #a);}while(0)
64 #endif
66 #else
68 #define DLOG(a)
70 #endif
72 /* dynamic array functions */
73 typedef struct array_t {
74 char* pointer;
75 unsigned int size,next,item_size;
76 } array_t;
78 static inline void array_init(array_t* array,unsigned int item_size)
80 array->pointer = NULL;
81 array->size=0;
82 array->next=0;
83 array->item_size=item_size;
86 static inline void array_free(array_t* array)
88 g_free(array->pointer);
89 array->size=array->next=0;
92 /* does not automatically grow */
93 static inline void* array_get(array_t* array,unsigned int index) {
94 assert(index < array->next);
95 return array->pointer + index * array->item_size;
98 static inline int array_ensure_allocated(array_t* array, int index)
100 if((index + 1) * array->item_size > array->size) {
101 int new_size = (index + 32) * array->item_size;
102 array->pointer = g_realloc(array->pointer, new_size);
103 if (!array->pointer)
104 return -1;
105 array->size = new_size;
106 array->next = index + 1;
109 return 0;
112 static inline void* array_get_next(array_t* array) {
113 unsigned int next = array->next;
114 void* result;
116 if (array_ensure_allocated(array, next) < 0)
117 return NULL;
119 array->next = next + 1;
120 result = array_get(array, next);
122 return result;
125 static inline void* array_insert(array_t* array,unsigned int index,unsigned int count) {
126 if((array->next+count)*array->item_size>array->size) {
127 int increment=count*array->item_size;
128 array->pointer=g_realloc(array->pointer,array->size+increment);
129 if(!array->pointer)
130 return NULL;
131 array->size+=increment;
133 memmove(array->pointer+(index+count)*array->item_size,
134 array->pointer+index*array->item_size,
135 (array->next-index)*array->item_size);
136 array->next+=count;
137 return array->pointer+index*array->item_size;
140 /* this performs a "roll", so that the element which was at index_from becomes
141 * index_to, but the order of all other elements is preserved. */
142 static inline int array_roll(array_t* array,int index_to,int index_from,int count)
144 char* buf;
145 char* from;
146 char* to;
147 int is;
149 if(!array ||
150 index_to<0 || index_to>=array->next ||
151 index_from<0 || index_from>=array->next)
152 return -1;
154 if(index_to==index_from)
155 return 0;
157 is=array->item_size;
158 from=array->pointer+index_from*is;
159 to=array->pointer+index_to*is;
160 buf=g_malloc(is*count);
161 memcpy(buf,from,is*count);
163 if(index_to<index_from)
164 memmove(to+is*count,to,from-to);
165 else
166 memmove(from,from+is*count,to-from);
168 memcpy(to,buf,is*count);
170 g_free(buf);
172 return 0;
175 static inline int array_remove_slice(array_t* array,int index, int count)
177 assert(index >=0);
178 assert(count > 0);
179 assert(index + count <= array->next);
180 if(array_roll(array,array->next-1,index,count))
181 return -1;
182 array->next -= count;
183 return 0;
186 static int array_remove(array_t* array,int index)
188 return array_remove_slice(array, index, 1);
191 /* return the index for a given member */
192 static int array_index(array_t* array, void* pointer)
194 size_t offset = (char*)pointer - array->pointer;
195 assert((offset % array->item_size) == 0);
196 assert(offset/array->item_size < array->next);
197 return offset/array->item_size;
200 /* These structures are used to fake a disk and the VFAT filesystem.
201 * For this reason we need to use QEMU_PACKED. */
203 typedef struct bootsector_t {
204 uint8_t jump[3];
205 uint8_t name[8];
206 uint16_t sector_size;
207 uint8_t sectors_per_cluster;
208 uint16_t reserved_sectors;
209 uint8_t number_of_fats;
210 uint16_t root_entries;
211 uint16_t total_sectors16;
212 uint8_t media_type;
213 uint16_t sectors_per_fat;
214 uint16_t sectors_per_track;
215 uint16_t number_of_heads;
216 uint32_t hidden_sectors;
217 uint32_t total_sectors;
218 union {
219 struct {
220 uint8_t drive_number;
221 uint8_t current_head;
222 uint8_t signature;
223 uint32_t id;
224 uint8_t volume_label[11];
225 } QEMU_PACKED fat16;
226 struct {
227 uint32_t sectors_per_fat;
228 uint16_t flags;
229 uint8_t major,minor;
230 uint32_t first_cluster_of_root_directory;
231 uint16_t info_sector;
232 uint16_t backup_boot_sector;
233 uint16_t ignored;
234 } QEMU_PACKED fat32;
235 } u;
236 uint8_t fat_type[8];
237 uint8_t ignored[0x1c0];
238 uint8_t magic[2];
239 } QEMU_PACKED bootsector_t;
241 typedef struct {
242 uint8_t head;
243 uint8_t sector;
244 uint8_t cylinder;
245 } mbr_chs_t;
247 typedef struct partition_t {
248 uint8_t attributes; /* 0x80 = bootable */
249 mbr_chs_t start_CHS;
250 uint8_t fs_type; /* 0x1 = FAT12, 0x6 = FAT16, 0xe = FAT16_LBA, 0xb = FAT32, 0xc = FAT32_LBA */
251 mbr_chs_t end_CHS;
252 uint32_t start_sector_long;
253 uint32_t length_sector_long;
254 } QEMU_PACKED partition_t;
256 typedef struct mbr_t {
257 uint8_t ignored[0x1b8];
258 uint32_t nt_id;
259 uint8_t ignored2[2];
260 partition_t partition[4];
261 uint8_t magic[2];
262 } QEMU_PACKED mbr_t;
264 typedef struct direntry_t {
265 uint8_t name[8 + 3];
266 uint8_t attributes;
267 uint8_t reserved[2];
268 uint16_t ctime;
269 uint16_t cdate;
270 uint16_t adate;
271 uint16_t begin_hi;
272 uint16_t mtime;
273 uint16_t mdate;
274 uint16_t begin;
275 uint32_t size;
276 } QEMU_PACKED direntry_t;
278 /* this structure are used to transparently access the files */
280 typedef struct mapping_t {
281 /* begin is the first cluster, end is the last+1 */
282 uint32_t begin,end;
283 /* as s->directory is growable, no pointer may be used here */
284 unsigned int dir_index;
285 /* the clusters of a file may be in any order; this points to the first */
286 int first_mapping_index;
287 union {
288 /* offset is
289 * - the offset in the file (in clusters) for a file, or
290 * - the next cluster of the directory for a directory, and
291 * - the address of the buffer for a faked entry
293 struct {
294 uint32_t offset;
295 } file;
296 struct {
297 int parent_mapping_index;
298 int first_dir_index;
299 } dir;
300 } info;
301 /* path contains the full path, i.e. it always starts with s->path */
302 char* path;
304 enum { MODE_UNDEFINED = 0, MODE_NORMAL = 1, MODE_MODIFIED = 2,
305 MODE_DIRECTORY = 4, MODE_FAKED = 8,
306 MODE_DELETED = 16, MODE_RENAMED = 32 } mode;
307 int read_only;
308 } mapping_t;
310 #ifdef DEBUG
311 static void print_direntry(const struct direntry_t*);
312 static void print_mapping(const struct mapping_t* mapping);
313 #endif
315 /* here begins the real VVFAT driver */
317 typedef struct BDRVVVFATState {
318 CoMutex lock;
319 BlockDriverState* bs; /* pointer to parent */
320 unsigned int first_sectors_number; /* 1 for a single partition, 0x40 for a disk with partition table */
321 unsigned char first_sectors[0x40*0x200];
323 int fat_type; /* 16 or 32 */
324 array_t fat,directory,mapping;
326 unsigned int cluster_size;
327 unsigned int sectors_per_cluster;
328 unsigned int sectors_per_fat;
329 unsigned int sectors_of_root_directory;
330 uint32_t last_cluster_of_root_directory;
331 unsigned int faked_sectors; /* how many sectors are faked before file data */
332 uint32_t sector_count; /* total number of sectors of the partition */
333 uint32_t cluster_count; /* total number of clusters of this partition */
334 uint32_t max_fat_value;
336 int current_fd;
337 mapping_t* current_mapping;
338 unsigned char* cluster; /* points to current cluster */
339 unsigned char* cluster_buffer; /* points to a buffer to hold temp data */
340 unsigned int current_cluster;
342 /* write support */
343 BlockDriverState* write_target;
344 char* qcow_filename;
345 BlockDriverState* qcow;
346 void* fat2;
347 char* used_clusters;
348 array_t commits;
349 const char* path;
350 int downcase_short_names;
352 Error *migration_blocker;
353 } BDRVVVFATState;
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.
359 static int sector2CHS(mbr_chs_t *chs, int spos, int cyls, int heads, int secs)
361 int head,sector;
362 sector = spos % secs; spos /= secs;
363 head = spos % heads; spos /= heads;
364 if (spos >= cyls) {
365 /* Overflow,
366 it happens if 32bit sector positions are used, while CHS is only 24bit.
367 Windows/Dos is said to take 1023/255/63 as nonrepresentable CHS */
368 chs->head = 0xFF;
369 chs->sector = 0xFF;
370 chs->cylinder = 0xFF;
371 return 1;
373 chs->head = (uint8_t)head;
374 chs->sector = (uint8_t)( (sector+1) | ((spos>>8)<<6) );
375 chs->cylinder = (uint8_t)spos;
376 return 0;
379 static void init_mbr(BDRVVVFATState *s, int cyls, int heads, int secs)
381 /* TODO: if the files mbr.img and bootsect.img exist, use them */
382 mbr_t* real_mbr=(mbr_t*)s->first_sectors;
383 partition_t* partition = &(real_mbr->partition[0]);
384 int lba;
386 memset(s->first_sectors,0,512);
388 /* Win NT Disk Signature */
389 real_mbr->nt_id= cpu_to_le32(0xbe1afdfa);
391 partition->attributes=0x80; /* bootable */
393 /* LBA is used when partition is outside the CHS geometry */
394 lba = sector2CHS(&partition->start_CHS, s->first_sectors_number - 1,
395 cyls, heads, secs);
396 lba |= sector2CHS(&partition->end_CHS, s->bs->total_sectors - 1,
397 cyls, heads, secs);
399 /*LBA partitions are identified only by start/length_sector_long not by CHS*/
400 partition->start_sector_long = cpu_to_le32(s->first_sectors_number - 1);
401 partition->length_sector_long = cpu_to_le32(s->bs->total_sectors
402 - s->first_sectors_number + 1);
404 /* FAT12/FAT16/FAT32 */
405 /* DOS uses different types when partition is LBA,
406 probably to prevent older versions from using CHS on them */
407 partition->fs_type= s->fat_type==12 ? 0x1:
408 s->fat_type==16 ? (lba?0xe:0x06):
409 /*fat_tyoe==32*/ (lba?0xc:0x0b);
411 real_mbr->magic[0]=0x55; real_mbr->magic[1]=0xaa;
414 /* direntry functions */
416 /* dest is assumed to hold 258 bytes, and pads with 0xffff up to next multiple of 26 */
417 static inline int short2long_name(char* dest,const char* src)
419 int i;
420 int len;
421 for(i=0;i<129 && src[i];i++) {
422 dest[2*i]=src[i];
423 dest[2*i+1]=0;
425 len=2*i;
426 dest[2*i]=dest[2*i+1]=0;
427 for(i=2*i+2;(i%26);i++)
428 dest[i]=0xff;
429 return len;
432 static inline direntry_t* create_long_filename(BDRVVVFATState* s,const char* filename)
434 char buffer[258];
435 int length=short2long_name(buffer,filename),
436 number_of_entries=(length+25)/26,i;
437 direntry_t* entry;
439 for(i=0;i<number_of_entries;i++) {
440 entry=array_get_next(&(s->directory));
441 entry->attributes=0xf;
442 entry->reserved[0]=0;
443 entry->begin=0;
444 entry->name[0]=(number_of_entries-i)|(i==0?0x40:0);
446 for(i=0;i<26*number_of_entries;i++) {
447 int offset=(i%26);
448 if(offset<10) offset=1+offset;
449 else if(offset<22) offset=14+offset-10;
450 else offset=28+offset-22;
451 entry=array_get(&(s->directory),s->directory.next-1-(i/26));
452 entry->name[offset]=buffer[i];
454 return array_get(&(s->directory),s->directory.next-number_of_entries);
457 static char is_free(const direntry_t* direntry)
459 return direntry->name[0]==0xe5 || direntry->name[0]==0x00;
462 static char is_volume_label(const direntry_t* direntry)
464 return direntry->attributes == 0x28;
467 static char is_long_name(const direntry_t* direntry)
469 return direntry->attributes == 0xf;
472 static char is_short_name(const direntry_t* direntry)
474 return !is_volume_label(direntry) && !is_long_name(direntry)
475 && !is_free(direntry);
478 static char is_directory(const direntry_t* direntry)
480 return direntry->attributes & 0x10 && direntry->name[0] != 0xe5;
483 static inline char is_dot(const direntry_t* direntry)
485 return is_short_name(direntry) && direntry->name[0] == '.';
488 static char is_file(const direntry_t* direntry)
490 return is_short_name(direntry) && !is_directory(direntry);
493 static inline uint32_t begin_of_direntry(const direntry_t* direntry)
495 return le16_to_cpu(direntry->begin)|(le16_to_cpu(direntry->begin_hi)<<16);
498 static inline uint32_t filesize_of_direntry(const direntry_t* direntry)
500 return le32_to_cpu(direntry->size);
503 static void set_begin_of_direntry(direntry_t* direntry, uint32_t begin)
505 direntry->begin = cpu_to_le16(begin & 0xffff);
506 direntry->begin_hi = cpu_to_le16((begin >> 16) & 0xffff);
509 /* fat functions */
511 static inline uint8_t fat_chksum(const direntry_t* entry)
513 uint8_t chksum=0;
514 int i;
516 for (i = 0; i < ARRAY_SIZE(entry->name); i++) {
517 chksum = (((chksum & 0xfe) >> 1) |
518 ((chksum & 0x01) ? 0x80 : 0)) + entry->name[i];
521 return chksum;
524 /* if return_time==0, this returns the fat_date, else the fat_time */
525 static uint16_t fat_datetime(time_t time,int return_time) {
526 struct tm* t;
527 struct tm t1;
528 t = &t1;
529 localtime_r(&time,t);
530 if(return_time)
531 return cpu_to_le16((t->tm_sec/2)|(t->tm_min<<5)|(t->tm_hour<<11));
532 return cpu_to_le16((t->tm_mday)|((t->tm_mon+1)<<5)|((t->tm_year-80)<<9));
535 static inline void fat_set(BDRVVVFATState* s,unsigned int cluster,uint32_t value)
537 if(s->fat_type==32) {
538 uint32_t* entry=array_get(&(s->fat),cluster);
539 *entry=cpu_to_le32(value);
540 } else if(s->fat_type==16) {
541 uint16_t* entry=array_get(&(s->fat),cluster);
542 *entry=cpu_to_le16(value&0xffff);
543 } else {
544 int offset = (cluster*3/2);
545 unsigned char* p = array_get(&(s->fat), offset);
546 switch (cluster&1) {
547 case 0:
548 p[0] = value&0xff;
549 p[1] = (p[1]&0xf0) | ((value>>8)&0xf);
550 break;
551 case 1:
552 p[0] = (p[0]&0xf) | ((value&0xf)<<4);
553 p[1] = (value>>4);
554 break;
559 static inline uint32_t fat_get(BDRVVVFATState* s,unsigned int cluster)
561 if(s->fat_type==32) {
562 uint32_t* entry=array_get(&(s->fat),cluster);
563 return le32_to_cpu(*entry);
564 } else if(s->fat_type==16) {
565 uint16_t* entry=array_get(&(s->fat),cluster);
566 return le16_to_cpu(*entry);
567 } else {
568 const uint8_t* x=(uint8_t*)(s->fat.pointer)+cluster*3/2;
569 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
573 static inline int fat_eof(BDRVVVFATState* s,uint32_t fat_entry)
575 if(fat_entry>s->max_fat_value-8)
576 return -1;
577 return 0;
580 static inline void init_fat(BDRVVVFATState* s)
582 if (s->fat_type == 12) {
583 array_init(&(s->fat),1);
584 array_ensure_allocated(&(s->fat),
585 s->sectors_per_fat * 0x200 * 3 / 2 - 1);
586 } else {
587 array_init(&(s->fat),(s->fat_type==32?4:2));
588 array_ensure_allocated(&(s->fat),
589 s->sectors_per_fat * 0x200 / s->fat.item_size - 1);
591 memset(s->fat.pointer,0,s->fat.size);
593 switch(s->fat_type) {
594 case 12: s->max_fat_value=0xfff; break;
595 case 16: s->max_fat_value=0xffff; break;
596 case 32: s->max_fat_value=0x0fffffff; break;
597 default: s->max_fat_value=0; /* error... */
602 /* TODO: in create_short_filename, 0xe5->0x05 is not yet handled! */
603 /* TODO: in parse_short_filename, 0x05->0xe5 is not yet handled! */
604 static inline direntry_t* create_short_and_long_name(BDRVVVFATState* s,
605 unsigned int directory_start, const char* filename, int is_dot)
607 int i,j,long_index=s->directory.next;
608 direntry_t* entry = NULL;
609 direntry_t* entry_long = NULL;
611 if(is_dot) {
612 entry=array_get_next(&(s->directory));
613 memset(entry->name, 0x20, sizeof(entry->name));
614 memcpy(entry->name,filename,strlen(filename));
615 return entry;
618 entry_long=create_long_filename(s,filename);
620 i = strlen(filename);
621 for(j = i - 1; j>0 && filename[j]!='.';j--);
622 if (j > 0)
623 i = (j > 8 ? 8 : j);
624 else if (i > 8)
625 i = 8;
627 entry=array_get_next(&(s->directory));
628 memset(entry->name, 0x20, sizeof(entry->name));
629 memcpy(entry->name, filename, i);
631 if (j > 0) {
632 for (i = 0; i < 3 && filename[j + 1 + i]; i++) {
633 entry->name[8 + i] = filename[j + 1 + i];
637 /* upcase & remove unwanted characters */
638 for(i=10;i>=0;i--) {
639 if(i==10 || i==7) for(;i>0 && entry->name[i]==' ';i--);
640 if(entry->name[i]<=' ' || entry->name[i]>0x7f
641 || strchr(".*?<>|\":/\\[];,+='",entry->name[i]))
642 entry->name[i]='_';
643 else if(entry->name[i]>='a' && entry->name[i]<='z')
644 entry->name[i]+='A'-'a';
647 /* mangle duplicates */
648 while(1) {
649 direntry_t* entry1=array_get(&(s->directory),directory_start);
650 int j;
652 for(;entry1<entry;entry1++)
653 if(!is_long_name(entry1) && !memcmp(entry1->name,entry->name,11))
654 break; /* found dupe */
655 if(entry1==entry) /* no dupe found */
656 break;
658 /* use all 8 characters of name */
659 if(entry->name[7]==' ') {
660 int j;
661 for(j=6;j>0 && entry->name[j]==' ';j--)
662 entry->name[j]='~';
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]++;
676 /* calculate checksum; propagate to long name */
677 if(entry_long) {
678 uint8_t chksum=fat_chksum(entry);
680 /* calculate anew, because realloc could have taken place */
681 entry_long=array_get(&(s->directory),long_index);
682 while(entry_long<entry && is_long_name(entry_long)) {
683 entry_long->reserved[1]=chksum;
684 entry_long++;
688 return entry;
692 * Read a directory. (the index of the corresponding mapping must be passed).
694 static int read_directory(BDRVVVFATState* s, int mapping_index)
696 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
697 direntry_t* direntry;
698 const char* dirname = mapping->path;
699 int first_cluster = mapping->begin;
700 int parent_index = mapping->info.dir.parent_mapping_index;
701 mapping_t* parent_mapping = (mapping_t*)
702 (parent_index >= 0 ? array_get(&(s->mapping), parent_index) : NULL);
703 int first_cluster_of_parent = parent_mapping ? parent_mapping->begin : -1;
705 DIR* dir=opendir(dirname);
706 struct dirent* entry;
707 int i;
709 assert(mapping->mode & MODE_DIRECTORY);
711 if(!dir) {
712 mapping->end = mapping->begin;
713 return -1;
716 i = mapping->info.dir.first_dir_index =
717 first_cluster == 0 ? 0 : s->directory.next;
719 /* actually read the directory, and allocate the mappings */
720 while((entry=readdir(dir))) {
721 unsigned int length=strlen(dirname)+2+strlen(entry->d_name);
722 char* buffer;
723 direntry_t* direntry;
724 struct stat st;
725 int is_dot=!strcmp(entry->d_name,".");
726 int is_dotdot=!strcmp(entry->d_name,"..");
728 if(first_cluster == 0 && (is_dotdot || is_dot))
729 continue;
731 buffer = g_malloc(length);
732 snprintf(buffer,length,"%s/%s",dirname,entry->d_name);
734 if(stat(buffer,&st)<0) {
735 g_free(buffer);
736 continue;
739 /* create directory entry for this file */
740 direntry=create_short_and_long_name(s, i, entry->d_name,
741 is_dot || is_dotdot);
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)
751 set_begin_of_direntry(direntry, first_cluster_of_parent);
752 else if(is_dot)
753 set_begin_of_direntry(direntry, first_cluster);
754 else
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 g_free(buffer);
759 closedir(dir);
760 return -2;
762 direntry->size=cpu_to_le32(S_ISDIR(st.st_mode)?0:st.st_size);
764 /* create mapping for this file */
765 if(!is_dot && !is_dotdot && (S_ISDIR(st.st_mode) || st.st_size)) {
766 s->current_mapping = array_get_next(&(s->mapping));
767 s->current_mapping->begin=0;
768 s->current_mapping->end=st.st_size;
770 * we get the direntry of the most recent direntry, which
771 * contains the short name and all the relevant information.
773 s->current_mapping->dir_index=s->directory.next-1;
774 s->current_mapping->first_mapping_index = -1;
775 if (S_ISDIR(st.st_mode)) {
776 s->current_mapping->mode = MODE_DIRECTORY;
777 s->current_mapping->info.dir.parent_mapping_index =
778 mapping_index;
779 } else {
780 s->current_mapping->mode = MODE_UNDEFINED;
781 s->current_mapping->info.file.offset = 0;
783 s->current_mapping->path=buffer;
784 s->current_mapping->read_only =
785 (st.st_mode & (S_IWUSR | S_IWGRP | S_IWOTH)) == 0;
786 } else {
787 g_free(buffer);
790 closedir(dir);
792 /* fill with zeroes up to the end of the cluster */
793 while(s->directory.next%(0x10*s->sectors_per_cluster)) {
794 direntry_t* direntry=array_get_next(&(s->directory));
795 memset(direntry,0,sizeof(direntry_t));
798 /* TODO: if there are more entries, bootsector has to be adjusted! */
799 #define ROOT_ENTRIES (0x02 * 0x10 * s->sectors_per_cluster)
800 if (mapping_index == 0 && s->directory.next < ROOT_ENTRIES) {
801 /* root directory */
802 int cur = s->directory.next;
803 array_ensure_allocated(&(s->directory), ROOT_ENTRIES - 1);
804 s->directory.next = ROOT_ENTRIES;
805 memset(array_get(&(s->directory), cur), 0,
806 (ROOT_ENTRIES - cur) * sizeof(direntry_t));
809 /* reget the mapping, since s->mapping was possibly realloc()ed */
810 mapping = array_get(&(s->mapping), mapping_index);
811 first_cluster += (s->directory.next - mapping->info.dir.first_dir_index)
812 * 0x20 / s->cluster_size;
813 mapping->end = first_cluster;
815 direntry = array_get(&(s->directory), mapping->dir_index);
816 set_begin_of_direntry(direntry, mapping->begin);
818 return 0;
821 static inline uint32_t sector2cluster(BDRVVVFATState* s,off_t sector_num)
823 return (sector_num-s->faked_sectors)/s->sectors_per_cluster;
826 static inline off_t cluster2sector(BDRVVVFATState* s, uint32_t cluster_num)
828 return s->faked_sectors + s->sectors_per_cluster * cluster_num;
831 static int init_directories(BDRVVVFATState* s,
832 const char *dirname, int heads, int secs,
833 Error **errp)
835 bootsector_t* bootsector;
836 mapping_t* mapping;
837 unsigned int i;
838 unsigned int cluster;
840 memset(&(s->first_sectors[0]),0,0x40*0x200);
842 s->cluster_size=s->sectors_per_cluster*0x200;
843 s->cluster_buffer=g_malloc(s->cluster_size);
846 * The formula: sc = spf+1+spf*spc*(512*8/fat_type),
847 * where sc is sector_count,
848 * spf is sectors_per_fat,
849 * spc is sectors_per_clusters, and
850 * fat_type = 12, 16 or 32.
852 i = 1+s->sectors_per_cluster*0x200*8/s->fat_type;
853 s->sectors_per_fat=(s->sector_count+i)/i; /* round up */
855 array_init(&(s->mapping),sizeof(mapping_t));
856 array_init(&(s->directory),sizeof(direntry_t));
858 /* add volume label */
860 direntry_t* entry=array_get_next(&(s->directory));
861 entry->attributes=0x28; /* archive | volume label */
862 memcpy(entry->name, "QEMU VVFAT ", sizeof(entry->name));
865 /* Now build FAT, and write back information into directory */
866 init_fat(s);
868 s->faked_sectors=s->first_sectors_number+s->sectors_per_fat*2;
869 s->cluster_count=sector2cluster(s, s->sector_count);
871 mapping = array_get_next(&(s->mapping));
872 mapping->begin = 0;
873 mapping->dir_index = 0;
874 mapping->info.dir.parent_mapping_index = -1;
875 mapping->first_mapping_index = -1;
876 mapping->path = g_strdup(dirname);
877 i = strlen(mapping->path);
878 if (i > 0 && mapping->path[i - 1] == '/')
879 mapping->path[i - 1] = '\0';
880 mapping->mode = MODE_DIRECTORY;
881 mapping->read_only = 0;
882 s->path = mapping->path;
884 for (i = 0, cluster = 0; i < s->mapping.next; i++) {
885 /* MS-DOS expects the FAT to be 0 for the root directory
886 * (except for the media byte). */
887 /* LATER TODO: still true for FAT32? */
888 int fix_fat = (i != 0);
889 mapping = array_get(&(s->mapping), i);
891 if (mapping->mode & MODE_DIRECTORY) {
892 mapping->begin = cluster;
893 if(read_directory(s, i)) {
894 error_setg(errp, "Could not read directory %s",
895 mapping->path);
896 return -1;
898 mapping = array_get(&(s->mapping), i);
899 } else {
900 assert(mapping->mode == MODE_UNDEFINED);
901 mapping->mode=MODE_NORMAL;
902 mapping->begin = cluster;
903 if (mapping->end > 0) {
904 direntry_t* direntry = array_get(&(s->directory),
905 mapping->dir_index);
907 mapping->end = cluster + 1 + (mapping->end-1)/s->cluster_size;
908 set_begin_of_direntry(direntry, mapping->begin);
909 } else {
910 mapping->end = cluster + 1;
911 fix_fat = 0;
915 assert(mapping->begin < mapping->end);
917 /* next free cluster */
918 cluster = mapping->end;
920 if(cluster > s->cluster_count) {
921 error_setg(errp,
922 "Directory does not fit in FAT%d (capacity %.2f MB)",
923 s->fat_type, s->sector_count / 2000.0);
924 return -1;
927 /* fix fat for entry */
928 if (fix_fat) {
929 int j;
930 for(j = mapping->begin; j < mapping->end - 1; j++)
931 fat_set(s, j, j+1);
932 fat_set(s, mapping->end - 1, s->max_fat_value);
936 mapping = array_get(&(s->mapping), 0);
937 s->sectors_of_root_directory = mapping->end * s->sectors_per_cluster;
938 s->last_cluster_of_root_directory = mapping->end;
940 /* the FAT signature */
941 fat_set(s,0,s->max_fat_value);
942 fat_set(s,1,s->max_fat_value);
944 s->current_mapping = NULL;
946 bootsector=(bootsector_t*)(s->first_sectors+(s->first_sectors_number-1)*0x200);
947 bootsector->jump[0]=0xeb;
948 bootsector->jump[1]=0x3e;
949 bootsector->jump[2]=0x90;
950 memcpy(bootsector->name,"QEMU ",8);
951 bootsector->sector_size=cpu_to_le16(0x200);
952 bootsector->sectors_per_cluster=s->sectors_per_cluster;
953 bootsector->reserved_sectors=cpu_to_le16(1);
954 bootsector->number_of_fats=0x2; /* number of FATs */
955 bootsector->root_entries=cpu_to_le16(s->sectors_of_root_directory*0x10);
956 bootsector->total_sectors16=s->sector_count>0xffff?0:cpu_to_le16(s->sector_count);
957 bootsector->media_type=(s->first_sectors_number>1?0xf8:0xf0); /* media descriptor (f8=hd, f0=3.5 fd)*/
958 s->fat.pointer[0] = bootsector->media_type;
959 bootsector->sectors_per_fat=cpu_to_le16(s->sectors_per_fat);
960 bootsector->sectors_per_track = cpu_to_le16(secs);
961 bootsector->number_of_heads = cpu_to_le16(heads);
962 bootsector->hidden_sectors=cpu_to_le32(s->first_sectors_number==1?0:0x3f);
963 bootsector->total_sectors=cpu_to_le32(s->sector_count>0xffff?s->sector_count:0);
965 /* LATER TODO: if FAT32, this is wrong */
966 bootsector->u.fat16.drive_number=s->first_sectors_number==1?0:0x80; /* fda=0, hda=0x80 */
967 bootsector->u.fat16.current_head=0;
968 bootsector->u.fat16.signature=0x29;
969 bootsector->u.fat16.id=cpu_to_le32(0xfabe1afd);
971 memcpy(bootsector->u.fat16.volume_label,"QEMU VVFAT ",11);
972 memcpy(bootsector->fat_type,(s->fat_type==12?"FAT12 ":s->fat_type==16?"FAT16 ":"FAT32 "),8);
973 bootsector->magic[0]=0x55; bootsector->magic[1]=0xaa;
975 return 0;
978 #ifdef DEBUG
979 static BDRVVVFATState *vvv = NULL;
980 #endif
982 static int enable_write_target(BDRVVVFATState *s, Error **errp);
983 static int is_consistent(BDRVVVFATState *s);
985 static void vvfat_rebind(BlockDriverState *bs)
987 BDRVVVFATState *s = bs->opaque;
988 s->bs = bs;
991 static QemuOptsList runtime_opts = {
992 .name = "vvfat",
993 .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
994 .desc = {
996 .name = "dir",
997 .type = QEMU_OPT_STRING,
998 .help = "Host directory to map to the vvfat device",
1001 .name = "fat-type",
1002 .type = QEMU_OPT_NUMBER,
1003 .help = "FAT type (12, 16 or 32)",
1006 .name = "floppy",
1007 .type = QEMU_OPT_BOOL,
1008 .help = "Create a floppy rather than a hard disk image",
1011 .name = "rw",
1012 .type = QEMU_OPT_BOOL,
1013 .help = "Make the image writable",
1015 { /* end of list */ }
1019 static void vvfat_parse_filename(const char *filename, QDict *options,
1020 Error **errp)
1022 int fat_type = 0;
1023 bool floppy = false;
1024 bool rw = false;
1025 int i;
1027 if (!strstart(filename, "fat:", NULL)) {
1028 error_setg(errp, "File name string must start with 'fat:'");
1029 return;
1032 /* Parse options */
1033 if (strstr(filename, ":32:")) {
1034 fat_type = 32;
1035 } else if (strstr(filename, ":16:")) {
1036 fat_type = 16;
1037 } else if (strstr(filename, ":12:")) {
1038 fat_type = 12;
1041 if (strstr(filename, ":floppy:")) {
1042 floppy = true;
1045 if (strstr(filename, ":rw:")) {
1046 rw = true;
1049 /* Get the directory name without options */
1050 i = strrchr(filename, ':') - filename;
1051 assert(i >= 3);
1052 if (filename[i - 2] == ':' && qemu_isalpha(filename[i - 1])) {
1053 /* workaround for DOS drive names */
1054 filename += i - 1;
1055 } else {
1056 filename += i + 1;
1059 /* Fill in the options QDict */
1060 qdict_put(options, "dir", qstring_from_str(filename));
1061 qdict_put(options, "fat-type", qint_from_int(fat_type));
1062 qdict_put(options, "floppy", qbool_from_int(floppy));
1063 qdict_put(options, "rw", qbool_from_int(rw));
1066 static int vvfat_open(BlockDriverState *bs, QDict *options, int flags,
1067 Error **errp)
1069 BDRVVVFATState *s = bs->opaque;
1070 int cyls, heads, secs;
1071 bool floppy;
1072 const char *dirname;
1073 QemuOpts *opts;
1074 Error *local_err = NULL;
1075 int ret;
1077 #ifdef DEBUG
1078 vvv = s;
1079 #endif
1081 opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
1082 qemu_opts_absorb_qdict(opts, options, &local_err);
1083 if (local_err) {
1084 error_propagate(errp, local_err);
1085 ret = -EINVAL;
1086 goto fail;
1089 dirname = qemu_opt_get(opts, "dir");
1090 if (!dirname) {
1091 error_setg(errp, "vvfat block driver requires a 'dir' option");
1092 ret = -EINVAL;
1093 goto fail;
1096 s->fat_type = qemu_opt_get_number(opts, "fat-type", 0);
1097 floppy = qemu_opt_get_bool(opts, "floppy", false);
1099 if (floppy) {
1100 /* 1.44MB or 2.88MB floppy. 2.88MB can be FAT12 (default) or FAT16. */
1101 if (!s->fat_type) {
1102 s->fat_type = 12;
1103 secs = 36;
1104 s->sectors_per_cluster = 2;
1105 } else {
1106 secs = s->fat_type == 12 ? 18 : 36;
1107 s->sectors_per_cluster = 1;
1109 s->first_sectors_number = 1;
1110 cyls = 80;
1111 heads = 2;
1112 } else {
1113 /* 32MB or 504MB disk*/
1114 if (!s->fat_type) {
1115 s->fat_type = 16;
1117 s->first_sectors_number = 0x40;
1118 cyls = s->fat_type == 12 ? 64 : 1024;
1119 heads = 16;
1120 secs = 63;
1123 switch (s->fat_type) {
1124 case 32:
1125 fprintf(stderr, "Big fat greek warning: FAT32 has not been tested. "
1126 "You are welcome to do so!\n");
1127 break;
1128 case 16:
1129 case 12:
1130 break;
1131 default:
1132 error_setg(errp, "Valid FAT types are only 12, 16 and 32");
1133 ret = -EINVAL;
1134 goto fail;
1138 s->bs = bs;
1140 /* LATER TODO: if FAT32, adjust */
1141 s->sectors_per_cluster=0x10;
1143 s->current_cluster=0xffffffff;
1145 /* read only is the default for safety */
1146 bs->read_only = 1;
1147 s->qcow = s->write_target = NULL;
1148 s->qcow_filename = NULL;
1149 s->fat2 = NULL;
1150 s->downcase_short_names = 1;
1152 fprintf(stderr, "vvfat %s chs %d,%d,%d\n",
1153 dirname, cyls, heads, secs);
1155 s->sector_count = cyls * heads * secs - (s->first_sectors_number - 1);
1157 if (qemu_opt_get_bool(opts, "rw", false)) {
1158 ret = enable_write_target(s, errp);
1159 if (ret < 0) {
1160 goto fail;
1162 bs->read_only = 0;
1165 bs->total_sectors = cyls * heads * secs;
1167 if (init_directories(s, dirname, heads, secs, errp)) {
1168 ret = -EIO;
1169 goto fail;
1172 s->sector_count = s->faked_sectors + s->sectors_per_cluster*s->cluster_count;
1174 if (s->first_sectors_number == 0x40) {
1175 init_mbr(s, cyls, heads, secs);
1178 // assert(is_consistent(s));
1179 qemu_co_mutex_init(&s->lock);
1181 /* Disable migration when vvfat is used rw */
1182 if (s->qcow) {
1183 error_setg(&s->migration_blocker,
1184 "The vvfat (rw) format used by node '%s' "
1185 "does not support live migration",
1186 bdrv_get_device_or_node_name(bs));
1187 migrate_add_blocker(s->migration_blocker);
1190 ret = 0;
1191 fail:
1192 qemu_opts_del(opts);
1193 return ret;
1196 static inline void vvfat_close_current_file(BDRVVVFATState *s)
1198 if(s->current_mapping) {
1199 s->current_mapping = NULL;
1200 if (s->current_fd) {
1201 qemu_close(s->current_fd);
1202 s->current_fd = 0;
1205 s->current_cluster = -1;
1208 /* mappings between index1 and index2-1 are supposed to be ordered
1209 * return value is the index of the last mapping for which end>cluster_num
1211 static inline int find_mapping_for_cluster_aux(BDRVVVFATState* s,int cluster_num,int index1,int index2)
1213 while(1) {
1214 int index3;
1215 mapping_t* mapping;
1216 index3=(index1+index2)/2;
1217 mapping=array_get(&(s->mapping),index3);
1218 assert(mapping->begin < mapping->end);
1219 if(mapping->begin>=cluster_num) {
1220 assert(index2!=index3 || index2==0);
1221 if(index2==index3)
1222 return index1;
1223 index2=index3;
1224 } else {
1225 if(index1==index3)
1226 return mapping->end<=cluster_num ? index2 : index1;
1227 index1=index3;
1229 assert(index1<=index2);
1230 DLOG(mapping=array_get(&(s->mapping),index1);
1231 assert(mapping->begin<=cluster_num);
1232 assert(index2 >= s->mapping.next ||
1233 ((mapping = array_get(&(s->mapping),index2)) &&
1234 mapping->end>cluster_num)));
1238 static inline mapping_t* find_mapping_for_cluster(BDRVVVFATState* s,int cluster_num)
1240 int index=find_mapping_for_cluster_aux(s,cluster_num,0,s->mapping.next);
1241 mapping_t* mapping;
1242 if(index>=s->mapping.next)
1243 return NULL;
1244 mapping=array_get(&(s->mapping),index);
1245 if(mapping->begin>cluster_num)
1246 return NULL;
1247 assert(mapping->begin<=cluster_num && mapping->end>cluster_num);
1248 return mapping;
1251 static int open_file(BDRVVVFATState* s,mapping_t* mapping)
1253 if(!mapping)
1254 return -1;
1255 if(!s->current_mapping ||
1256 strcmp(s->current_mapping->path,mapping->path)) {
1257 /* open file */
1258 int fd = qemu_open(mapping->path, O_RDONLY | O_BINARY | O_LARGEFILE);
1259 if(fd<0)
1260 return -1;
1261 vvfat_close_current_file(s);
1262 s->current_fd = fd;
1263 s->current_mapping = mapping;
1265 return 0;
1268 static inline int read_cluster(BDRVVVFATState *s,int cluster_num)
1270 if(s->current_cluster != cluster_num) {
1271 int result=0;
1272 off_t offset;
1273 assert(!s->current_mapping || s->current_fd || (s->current_mapping->mode & MODE_DIRECTORY));
1274 if(!s->current_mapping
1275 || s->current_mapping->begin>cluster_num
1276 || s->current_mapping->end<=cluster_num) {
1277 /* binary search of mappings for file */
1278 mapping_t* mapping=find_mapping_for_cluster(s,cluster_num);
1280 assert(!mapping || (cluster_num>=mapping->begin && cluster_num<mapping->end));
1282 if (mapping && mapping->mode & MODE_DIRECTORY) {
1283 vvfat_close_current_file(s);
1284 s->current_mapping = mapping;
1285 read_cluster_directory:
1286 offset = s->cluster_size*(cluster_num-s->current_mapping->begin);
1287 s->cluster = (unsigned char*)s->directory.pointer+offset
1288 + 0x20*s->current_mapping->info.dir.first_dir_index;
1289 assert(((s->cluster-(unsigned char*)s->directory.pointer)%s->cluster_size)==0);
1290 assert((char*)s->cluster+s->cluster_size <= s->directory.pointer+s->directory.next*s->directory.item_size);
1291 s->current_cluster = cluster_num;
1292 return 0;
1295 if(open_file(s,mapping))
1296 return -2;
1297 } else if (s->current_mapping->mode & MODE_DIRECTORY)
1298 goto read_cluster_directory;
1300 assert(s->current_fd);
1302 offset=s->cluster_size*(cluster_num-s->current_mapping->begin)+s->current_mapping->info.file.offset;
1303 if(lseek(s->current_fd, offset, SEEK_SET)!=offset)
1304 return -3;
1305 s->cluster=s->cluster_buffer;
1306 result=read(s->current_fd,s->cluster,s->cluster_size);
1307 if(result<0) {
1308 s->current_cluster = -1;
1309 return -1;
1311 s->current_cluster = cluster_num;
1313 return 0;
1316 #ifdef DEBUG
1317 static void print_direntry(const direntry_t* direntry)
1319 int j = 0;
1320 char buffer[1024];
1322 fprintf(stderr, "direntry %p: ", direntry);
1323 if(!direntry)
1324 return;
1325 if(is_long_name(direntry)) {
1326 unsigned char* c=(unsigned char*)direntry;
1327 int i;
1328 for(i=1;i<11 && c[i] && c[i]!=0xff;i+=2)
1329 #define ADD_CHAR(c) {buffer[j] = (c); if (buffer[j] < ' ') buffer[j] = 0xb0; j++;}
1330 ADD_CHAR(c[i]);
1331 for(i=14;i<26 && c[i] && c[i]!=0xff;i+=2)
1332 ADD_CHAR(c[i]);
1333 for(i=28;i<32 && c[i] && c[i]!=0xff;i+=2)
1334 ADD_CHAR(c[i]);
1335 buffer[j] = 0;
1336 fprintf(stderr, "%s\n", buffer);
1337 } else {
1338 int i;
1339 for(i=0;i<11;i++)
1340 ADD_CHAR(direntry->name[i]);
1341 buffer[j] = 0;
1342 fprintf(stderr,"%s attributes=0x%02x begin=%d size=%d\n",
1343 buffer,
1344 direntry->attributes,
1345 begin_of_direntry(direntry),le32_to_cpu(direntry->size));
1349 static void print_mapping(const mapping_t* mapping)
1351 fprintf(stderr, "mapping (%p): begin, end = %d, %d, dir_index = %d, "
1352 "first_mapping_index = %d, name = %s, mode = 0x%x, " ,
1353 mapping, mapping->begin, mapping->end, mapping->dir_index,
1354 mapping->first_mapping_index, mapping->path, mapping->mode);
1356 if (mapping->mode & MODE_DIRECTORY)
1357 fprintf(stderr, "parent_mapping_index = %d, first_dir_index = %d\n", mapping->info.dir.parent_mapping_index, mapping->info.dir.first_dir_index);
1358 else
1359 fprintf(stderr, "offset = %d\n", mapping->info.file.offset);
1361 #endif
1363 static int vvfat_read(BlockDriverState *bs, int64_t sector_num,
1364 uint8_t *buf, int nb_sectors)
1366 BDRVVVFATState *s = bs->opaque;
1367 int i;
1369 for(i=0;i<nb_sectors;i++,sector_num++) {
1370 if (sector_num >= bs->total_sectors)
1371 return -1;
1372 if (s->qcow) {
1373 int n;
1374 if (bdrv_is_allocated(s->qcow, sector_num, nb_sectors-i, &n)) {
1375 DLOG(fprintf(stderr, "sectors %d+%d allocated\n", (int)sector_num, n));
1376 if (bdrv_read(s->qcow, sector_num, buf + i*0x200, n)) {
1377 return -1;
1379 i += n - 1;
1380 sector_num += n - 1;
1381 continue;
1383 DLOG(fprintf(stderr, "sector %d not allocated\n", (int)sector_num));
1385 if(sector_num<s->faked_sectors) {
1386 if(sector_num<s->first_sectors_number)
1387 memcpy(buf+i*0x200,&(s->first_sectors[sector_num*0x200]),0x200);
1388 else if(sector_num-s->first_sectors_number<s->sectors_per_fat)
1389 memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number)*0x200]),0x200);
1390 else if(sector_num-s->first_sectors_number-s->sectors_per_fat<s->sectors_per_fat)
1391 memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number-s->sectors_per_fat)*0x200]),0x200);
1392 } else {
1393 uint32_t sector=sector_num-s->faked_sectors,
1394 sector_offset_in_cluster=(sector%s->sectors_per_cluster),
1395 cluster_num=sector/s->sectors_per_cluster;
1396 if(cluster_num > s->cluster_count || read_cluster(s, cluster_num) != 0) {
1397 /* LATER TODO: strict: return -1; */
1398 memset(buf+i*0x200,0,0x200);
1399 continue;
1401 memcpy(buf+i*0x200,s->cluster+sector_offset_in_cluster*0x200,0x200);
1404 return 0;
1407 static coroutine_fn int vvfat_co_read(BlockDriverState *bs, int64_t sector_num,
1408 uint8_t *buf, int nb_sectors)
1410 int ret;
1411 BDRVVVFATState *s = bs->opaque;
1412 qemu_co_mutex_lock(&s->lock);
1413 ret = vvfat_read(bs, sector_num, buf, nb_sectors);
1414 qemu_co_mutex_unlock(&s->lock);
1415 return ret;
1418 /* LATER TODO: statify all functions */
1421 * Idea of the write support (use snapshot):
1423 * 1. check if all data is consistent, recording renames, modifications,
1424 * new files and directories (in s->commits).
1426 * 2. if the data is not consistent, stop committing
1428 * 3. handle renames, and create new files and directories (do not yet
1429 * write their contents)
1431 * 4. walk the directories, fixing the mapping and direntries, and marking
1432 * the handled mappings as not deleted
1434 * 5. commit the contents of the files
1436 * 6. handle deleted files and directories
1440 typedef struct commit_t {
1441 char* path;
1442 union {
1443 struct { uint32_t cluster; } rename;
1444 struct { int dir_index; uint32_t modified_offset; } writeout;
1445 struct { uint32_t first_cluster; } new_file;
1446 struct { uint32_t cluster; } mkdir;
1447 } param;
1448 /* DELETEs and RMDIRs are handled differently: see handle_deletes() */
1449 enum {
1450 ACTION_RENAME, ACTION_WRITEOUT, ACTION_NEW_FILE, ACTION_MKDIR
1451 } action;
1452 } commit_t;
1454 static void clear_commits(BDRVVVFATState* s)
1456 int i;
1457 DLOG(fprintf(stderr, "clear_commits (%d commits)\n", s->commits.next));
1458 for (i = 0; i < s->commits.next; i++) {
1459 commit_t* commit = array_get(&(s->commits), i);
1460 assert(commit->path || commit->action == ACTION_WRITEOUT);
1461 if (commit->action != ACTION_WRITEOUT) {
1462 assert(commit->path);
1463 g_free(commit->path);
1464 } else
1465 assert(commit->path == NULL);
1467 s->commits.next = 0;
1470 static void schedule_rename(BDRVVVFATState* s,
1471 uint32_t cluster, char* new_path)
1473 commit_t* commit = array_get_next(&(s->commits));
1474 commit->path = new_path;
1475 commit->param.rename.cluster = cluster;
1476 commit->action = ACTION_RENAME;
1479 static void schedule_writeout(BDRVVVFATState* s,
1480 int dir_index, uint32_t modified_offset)
1482 commit_t* commit = array_get_next(&(s->commits));
1483 commit->path = NULL;
1484 commit->param.writeout.dir_index = dir_index;
1485 commit->param.writeout.modified_offset = modified_offset;
1486 commit->action = ACTION_WRITEOUT;
1489 static void schedule_new_file(BDRVVVFATState* s,
1490 char* path, uint32_t first_cluster)
1492 commit_t* commit = array_get_next(&(s->commits));
1493 commit->path = path;
1494 commit->param.new_file.first_cluster = first_cluster;
1495 commit->action = ACTION_NEW_FILE;
1498 static void schedule_mkdir(BDRVVVFATState* s, uint32_t cluster, char* path)
1500 commit_t* commit = array_get_next(&(s->commits));
1501 commit->path = path;
1502 commit->param.mkdir.cluster = cluster;
1503 commit->action = ACTION_MKDIR;
1506 typedef struct {
1508 * Since the sequence number is at most 0x3f, and the filename
1509 * length is at most 13 times the sequence number, the maximal
1510 * filename length is 0x3f * 13 bytes.
1512 unsigned char name[0x3f * 13 + 1];
1513 int checksum, len;
1514 int sequence_number;
1515 } long_file_name;
1517 static void lfn_init(long_file_name* lfn)
1519 lfn->sequence_number = lfn->len = 0;
1520 lfn->checksum = 0x100;
1523 /* return 0 if parsed successfully, > 0 if no long name, < 0 if error */
1524 static int parse_long_name(long_file_name* lfn,
1525 const direntry_t* direntry)
1527 int i, j, offset;
1528 const unsigned char* pointer = (const unsigned char*)direntry;
1530 if (!is_long_name(direntry))
1531 return 1;
1533 if (pointer[0] & 0x40) {
1534 lfn->sequence_number = pointer[0] & 0x3f;
1535 lfn->checksum = pointer[13];
1536 lfn->name[0] = 0;
1537 lfn->name[lfn->sequence_number * 13] = 0;
1538 } else if ((pointer[0] & 0x3f) != --lfn->sequence_number)
1539 return -1;
1540 else if (pointer[13] != lfn->checksum)
1541 return -2;
1542 else if (pointer[12] || pointer[26] || pointer[27])
1543 return -3;
1545 offset = 13 * (lfn->sequence_number - 1);
1546 for (i = 0, j = 1; i < 13; i++, j+=2) {
1547 if (j == 11)
1548 j = 14;
1549 else if (j == 26)
1550 j = 28;
1552 if (pointer[j+1] == 0)
1553 lfn->name[offset + i] = pointer[j];
1554 else if (pointer[j+1] != 0xff || (pointer[0] & 0x40) == 0)
1555 return -4;
1556 else
1557 lfn->name[offset + i] = 0;
1560 if (pointer[0] & 0x40)
1561 lfn->len = offset + strlen((char*)lfn->name + offset);
1563 return 0;
1566 /* returns 0 if successful, >0 if no short_name, and <0 on error */
1567 static int parse_short_name(BDRVVVFATState* s,
1568 long_file_name* lfn, direntry_t* direntry)
1570 int i, j;
1572 if (!is_short_name(direntry))
1573 return 1;
1575 for (j = 7; j >= 0 && direntry->name[j] == ' '; j--);
1576 for (i = 0; i <= j; i++) {
1577 if (direntry->name[i] <= ' ' || direntry->name[i] > 0x7f)
1578 return -1;
1579 else if (s->downcase_short_names)
1580 lfn->name[i] = qemu_tolower(direntry->name[i]);
1581 else
1582 lfn->name[i] = direntry->name[i];
1585 for (j = 2; j >= 0 && direntry->name[8 + j] == ' '; j--) {
1587 if (j >= 0) {
1588 lfn->name[i++] = '.';
1589 lfn->name[i + j + 1] = '\0';
1590 for (;j >= 0; j--) {
1591 uint8_t c = direntry->name[8 + j];
1592 if (c <= ' ' || c > 0x7f) {
1593 return -2;
1594 } else if (s->downcase_short_names) {
1595 lfn->name[i + j] = qemu_tolower(c);
1596 } else {
1597 lfn->name[i + j] = c;
1600 } else
1601 lfn->name[i + j + 1] = '\0';
1603 lfn->len = strlen((char*)lfn->name);
1605 return 0;
1608 static inline uint32_t modified_fat_get(BDRVVVFATState* s,
1609 unsigned int cluster)
1611 if (cluster < s->last_cluster_of_root_directory) {
1612 if (cluster + 1 == s->last_cluster_of_root_directory)
1613 return s->max_fat_value;
1614 else
1615 return cluster + 1;
1618 if (s->fat_type==32) {
1619 uint32_t* entry=((uint32_t*)s->fat2)+cluster;
1620 return le32_to_cpu(*entry);
1621 } else if (s->fat_type==16) {
1622 uint16_t* entry=((uint16_t*)s->fat2)+cluster;
1623 return le16_to_cpu(*entry);
1624 } else {
1625 const uint8_t* x=s->fat2+cluster*3/2;
1626 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
1630 static inline int cluster_was_modified(BDRVVVFATState* s, uint32_t cluster_num)
1632 int was_modified = 0;
1633 int i, dummy;
1635 if (s->qcow == NULL)
1636 return 0;
1638 for (i = 0; !was_modified && i < s->sectors_per_cluster; i++)
1639 was_modified = bdrv_is_allocated(s->qcow,
1640 cluster2sector(s, cluster_num) + i, 1, &dummy);
1642 return was_modified;
1645 static const char* get_basename(const char* path)
1647 char* basename = strrchr(path, '/');
1648 if (basename == NULL)
1649 return path;
1650 else
1651 return basename + 1; /* strip '/' */
1655 * The array s->used_clusters holds the states of the clusters. If it is
1656 * part of a file, it has bit 2 set, in case of a directory, bit 1. If it
1657 * was modified, bit 3 is set.
1658 * If any cluster is allocated, but not part of a file or directory, this
1659 * driver refuses to commit.
1661 typedef enum {
1662 USED_DIRECTORY = 1, USED_FILE = 2, USED_ANY = 3, USED_ALLOCATED = 4
1663 } used_t;
1666 * get_cluster_count_for_direntry() not only determines how many clusters
1667 * are occupied by direntry, but also if it was renamed or modified.
1669 * A file is thought to be renamed *only* if there already was a file with
1670 * exactly the same first cluster, but a different name.
1672 * Further, the files/directories handled by this function are
1673 * assumed to be *not* deleted (and *only* those).
1675 static uint32_t get_cluster_count_for_direntry(BDRVVVFATState* s,
1676 direntry_t* direntry, const char* path)
1679 * This is a little bit tricky:
1680 * IF the guest OS just inserts a cluster into the file chain,
1681 * and leaves the rest alone, (i.e. the original file had clusters
1682 * 15 -> 16, but now has 15 -> 32 -> 16), then the following happens:
1684 * - do_commit will write the cluster into the file at the given
1685 * offset, but
1687 * - the cluster which is overwritten should be moved to a later
1688 * position in the file.
1690 * I am not aware that any OS does something as braindead, but this
1691 * situation could happen anyway when not committing for a long time.
1692 * Just to be sure that this does not bite us, detect it, and copy the
1693 * contents of the clusters to-be-overwritten into the qcow.
1695 int copy_it = 0;
1696 int was_modified = 0;
1697 int32_t ret = 0;
1699 uint32_t cluster_num = begin_of_direntry(direntry);
1700 uint32_t offset = 0;
1701 int first_mapping_index = -1;
1702 mapping_t* mapping = NULL;
1703 const char* basename2 = NULL;
1705 vvfat_close_current_file(s);
1707 /* the root directory */
1708 if (cluster_num == 0)
1709 return 0;
1711 /* write support */
1712 if (s->qcow) {
1713 basename2 = get_basename(path);
1715 mapping = find_mapping_for_cluster(s, cluster_num);
1717 if (mapping) {
1718 const char* basename;
1720 assert(mapping->mode & MODE_DELETED);
1721 mapping->mode &= ~MODE_DELETED;
1723 basename = get_basename(mapping->path);
1725 assert(mapping->mode & MODE_NORMAL);
1727 /* rename */
1728 if (strcmp(basename, basename2))
1729 schedule_rename(s, cluster_num, g_strdup(path));
1730 } else if (is_file(direntry))
1731 /* new file */
1732 schedule_new_file(s, g_strdup(path), cluster_num);
1733 else {
1734 abort();
1735 return 0;
1739 while(1) {
1740 if (s->qcow) {
1741 if (!copy_it && cluster_was_modified(s, cluster_num)) {
1742 if (mapping == NULL ||
1743 mapping->begin > cluster_num ||
1744 mapping->end <= cluster_num)
1745 mapping = find_mapping_for_cluster(s, cluster_num);
1748 if (mapping &&
1749 (mapping->mode & MODE_DIRECTORY) == 0) {
1751 /* was modified in qcow */
1752 if (offset != mapping->info.file.offset + s->cluster_size
1753 * (cluster_num - mapping->begin)) {
1754 /* offset of this cluster in file chain has changed */
1755 abort();
1756 copy_it = 1;
1757 } else if (offset == 0) {
1758 const char* basename = get_basename(mapping->path);
1760 if (strcmp(basename, basename2))
1761 copy_it = 1;
1762 first_mapping_index = array_index(&(s->mapping), mapping);
1765 if (mapping->first_mapping_index != first_mapping_index
1766 && mapping->info.file.offset > 0) {
1767 abort();
1768 copy_it = 1;
1771 /* need to write out? */
1772 if (!was_modified && is_file(direntry)) {
1773 was_modified = 1;
1774 schedule_writeout(s, mapping->dir_index, offset);
1779 if (copy_it) {
1780 int i, dummy;
1782 * This is horribly inefficient, but that is okay, since
1783 * it is rarely executed, if at all.
1785 int64_t offset = cluster2sector(s, cluster_num);
1787 vvfat_close_current_file(s);
1788 for (i = 0; i < s->sectors_per_cluster; i++) {
1789 if (!bdrv_is_allocated(s->qcow, offset + i, 1, &dummy)) {
1790 if (vvfat_read(s->bs, offset, s->cluster_buffer, 1)) {
1791 return -1;
1793 if (bdrv_write(s->qcow, offset, s->cluster_buffer, 1)) {
1794 return -2;
1801 ret++;
1802 if (s->used_clusters[cluster_num] & USED_ANY)
1803 return 0;
1804 s->used_clusters[cluster_num] = USED_FILE;
1806 cluster_num = modified_fat_get(s, cluster_num);
1808 if (fat_eof(s, cluster_num))
1809 return ret;
1810 else if (cluster_num < 2 || cluster_num > s->max_fat_value - 16)
1811 return -1;
1813 offset += s->cluster_size;
1818 * This function looks at the modified data (qcow).
1819 * It returns 0 upon inconsistency or error, and the number of clusters
1820 * used by the directory, its subdirectories and their files.
1822 static int check_directory_consistency(BDRVVVFATState *s,
1823 int cluster_num, const char* path)
1825 int ret = 0;
1826 unsigned char* cluster = g_malloc(s->cluster_size);
1827 direntry_t* direntries = (direntry_t*)cluster;
1828 mapping_t* mapping = find_mapping_for_cluster(s, cluster_num);
1830 long_file_name lfn;
1831 int path_len = strlen(path);
1832 char path2[PATH_MAX + 1];
1834 assert(path_len < PATH_MAX); /* len was tested before! */
1835 pstrcpy(path2, sizeof(path2), path);
1836 path2[path_len] = '/';
1837 path2[path_len + 1] = '\0';
1839 if (mapping) {
1840 const char* basename = get_basename(mapping->path);
1841 const char* basename2 = get_basename(path);
1843 assert(mapping->mode & MODE_DIRECTORY);
1845 assert(mapping->mode & MODE_DELETED);
1846 mapping->mode &= ~MODE_DELETED;
1848 if (strcmp(basename, basename2))
1849 schedule_rename(s, cluster_num, g_strdup(path));
1850 } else
1851 /* new directory */
1852 schedule_mkdir(s, cluster_num, g_strdup(path));
1854 lfn_init(&lfn);
1855 do {
1856 int i;
1857 int subret = 0;
1859 ret++;
1861 if (s->used_clusters[cluster_num] & USED_ANY) {
1862 fprintf(stderr, "cluster %d used more than once\n", (int)cluster_num);
1863 goto fail;
1865 s->used_clusters[cluster_num] = USED_DIRECTORY;
1867 DLOG(fprintf(stderr, "read cluster %d (sector %d)\n", (int)cluster_num, (int)cluster2sector(s, cluster_num)));
1868 subret = vvfat_read(s->bs, cluster2sector(s, cluster_num), cluster,
1869 s->sectors_per_cluster);
1870 if (subret) {
1871 fprintf(stderr, "Error fetching direntries\n");
1872 fail:
1873 g_free(cluster);
1874 return 0;
1877 for (i = 0; i < 0x10 * s->sectors_per_cluster; i++) {
1878 int cluster_count = 0;
1880 DLOG(fprintf(stderr, "check direntry %d:\n", i); print_direntry(direntries + i));
1881 if (is_volume_label(direntries + i) || is_dot(direntries + i) ||
1882 is_free(direntries + i))
1883 continue;
1885 subret = parse_long_name(&lfn, direntries + i);
1886 if (subret < 0) {
1887 fprintf(stderr, "Error in long name\n");
1888 goto fail;
1890 if (subret == 0 || is_free(direntries + i))
1891 continue;
1893 if (fat_chksum(direntries+i) != lfn.checksum) {
1894 subret = parse_short_name(s, &lfn, direntries + i);
1895 if (subret < 0) {
1896 fprintf(stderr, "Error in short name (%d)\n", subret);
1897 goto fail;
1899 if (subret > 0 || !strcmp((char*)lfn.name, ".")
1900 || !strcmp((char*)lfn.name, ".."))
1901 continue;
1903 lfn.checksum = 0x100; /* cannot use long name twice */
1905 if (path_len + 1 + lfn.len >= PATH_MAX) {
1906 fprintf(stderr, "Name too long: %s/%s\n", path, lfn.name);
1907 goto fail;
1909 pstrcpy(path2 + path_len + 1, sizeof(path2) - path_len - 1,
1910 (char*)lfn.name);
1912 if (is_directory(direntries + i)) {
1913 if (begin_of_direntry(direntries + i) == 0) {
1914 DLOG(fprintf(stderr, "invalid begin for directory: %s\n", path2); print_direntry(direntries + i));
1915 goto fail;
1917 cluster_count = check_directory_consistency(s,
1918 begin_of_direntry(direntries + i), path2);
1919 if (cluster_count == 0) {
1920 DLOG(fprintf(stderr, "problem in directory %s:\n", path2); print_direntry(direntries + i));
1921 goto fail;
1923 } else if (is_file(direntries + i)) {
1924 /* check file size with FAT */
1925 cluster_count = get_cluster_count_for_direntry(s, direntries + i, path2);
1926 if (cluster_count !=
1927 (le32_to_cpu(direntries[i].size) + s->cluster_size
1928 - 1) / s->cluster_size) {
1929 DLOG(fprintf(stderr, "Cluster count mismatch\n"));
1930 goto fail;
1932 } else
1933 abort(); /* cluster_count = 0; */
1935 ret += cluster_count;
1938 cluster_num = modified_fat_get(s, cluster_num);
1939 } while(!fat_eof(s, cluster_num));
1941 g_free(cluster);
1942 return ret;
1945 /* returns 1 on success */
1946 static int is_consistent(BDRVVVFATState* s)
1948 int i, check;
1949 int used_clusters_count = 0;
1951 DLOG(checkpoint());
1953 * - get modified FAT
1954 * - compare the two FATs (TODO)
1955 * - get buffer for marking used clusters
1956 * - recurse direntries from root (using bs->bdrv_read to make
1957 * sure to get the new data)
1958 * - check that the FAT agrees with the size
1959 * - count the number of clusters occupied by this directory and
1960 * its files
1961 * - check that the cumulative used cluster count agrees with the
1962 * FAT
1963 * - if all is fine, return number of used clusters
1965 if (s->fat2 == NULL) {
1966 int size = 0x200 * s->sectors_per_fat;
1967 s->fat2 = g_malloc(size);
1968 memcpy(s->fat2, s->fat.pointer, size);
1970 check = vvfat_read(s->bs,
1971 s->first_sectors_number, s->fat2, s->sectors_per_fat);
1972 if (check) {
1973 fprintf(stderr, "Could not copy fat\n");
1974 return 0;
1976 assert (s->used_clusters);
1977 for (i = 0; i < sector2cluster(s, s->sector_count); i++)
1978 s->used_clusters[i] &= ~USED_ANY;
1980 clear_commits(s);
1982 /* mark every mapped file/directory as deleted.
1983 * (check_directory_consistency() will unmark those still present). */
1984 if (s->qcow)
1985 for (i = 0; i < s->mapping.next; i++) {
1986 mapping_t* mapping = array_get(&(s->mapping), i);
1987 if (mapping->first_mapping_index < 0)
1988 mapping->mode |= MODE_DELETED;
1991 used_clusters_count = check_directory_consistency(s, 0, s->path);
1992 if (used_clusters_count <= 0) {
1993 DLOG(fprintf(stderr, "problem in directory\n"));
1994 return 0;
1997 check = s->last_cluster_of_root_directory;
1998 for (i = check; i < sector2cluster(s, s->sector_count); i++) {
1999 if (modified_fat_get(s, i)) {
2000 if(!s->used_clusters[i]) {
2001 DLOG(fprintf(stderr, "FAT was modified (%d), but cluster is not used?\n", i));
2002 return 0;
2004 check++;
2007 if (s->used_clusters[i] == USED_ALLOCATED) {
2008 /* allocated, but not used... */
2009 DLOG(fprintf(stderr, "unused, modified cluster: %d\n", i));
2010 return 0;
2014 if (check != used_clusters_count)
2015 return 0;
2017 return used_clusters_count;
2020 static inline void adjust_mapping_indices(BDRVVVFATState* s,
2021 int offset, int adjust)
2023 int i;
2025 for (i = 0; i < s->mapping.next; i++) {
2026 mapping_t* mapping = array_get(&(s->mapping), i);
2028 #define ADJUST_MAPPING_INDEX(name) \
2029 if (mapping->name >= offset) \
2030 mapping->name += adjust
2032 ADJUST_MAPPING_INDEX(first_mapping_index);
2033 if (mapping->mode & MODE_DIRECTORY)
2034 ADJUST_MAPPING_INDEX(info.dir.parent_mapping_index);
2038 /* insert or update mapping */
2039 static mapping_t* insert_mapping(BDRVVVFATState* s,
2040 uint32_t begin, uint32_t end)
2043 * - find mapping where mapping->begin >= begin,
2044 * - if mapping->begin > begin: insert
2045 * - adjust all references to mappings!
2046 * - else: adjust
2047 * - replace name
2049 int index = find_mapping_for_cluster_aux(s, begin, 0, s->mapping.next);
2050 mapping_t* mapping = NULL;
2051 mapping_t* first_mapping = array_get(&(s->mapping), 0);
2053 if (index < s->mapping.next && (mapping = array_get(&(s->mapping), index))
2054 && mapping->begin < begin) {
2055 mapping->end = begin;
2056 index++;
2057 mapping = array_get(&(s->mapping), index);
2059 if (index >= s->mapping.next || mapping->begin > begin) {
2060 mapping = array_insert(&(s->mapping), index, 1);
2061 mapping->path = NULL;
2062 adjust_mapping_indices(s, index, +1);
2065 mapping->begin = begin;
2066 mapping->end = end;
2068 DLOG(mapping_t* next_mapping;
2069 assert(index + 1 >= s->mapping.next ||
2070 ((next_mapping = array_get(&(s->mapping), index + 1)) &&
2071 next_mapping->begin >= end)));
2073 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
2074 s->current_mapping = array_get(&(s->mapping),
2075 s->current_mapping - first_mapping);
2077 return mapping;
2080 static int remove_mapping(BDRVVVFATState* s, int mapping_index)
2082 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
2083 mapping_t* first_mapping = array_get(&(s->mapping), 0);
2085 /* free mapping */
2086 if (mapping->first_mapping_index < 0) {
2087 g_free(mapping->path);
2090 /* remove from s->mapping */
2091 array_remove(&(s->mapping), mapping_index);
2093 /* adjust all references to mappings */
2094 adjust_mapping_indices(s, mapping_index, -1);
2096 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
2097 s->current_mapping = array_get(&(s->mapping),
2098 s->current_mapping - first_mapping);
2100 return 0;
2103 static void adjust_dirindices(BDRVVVFATState* s, int offset, int adjust)
2105 int i;
2106 for (i = 0; i < s->mapping.next; i++) {
2107 mapping_t* mapping = array_get(&(s->mapping), i);
2108 if (mapping->dir_index >= offset)
2109 mapping->dir_index += adjust;
2110 if ((mapping->mode & MODE_DIRECTORY) &&
2111 mapping->info.dir.first_dir_index >= offset)
2112 mapping->info.dir.first_dir_index += adjust;
2116 static direntry_t* insert_direntries(BDRVVVFATState* s,
2117 int dir_index, int count)
2120 * make room in s->directory,
2121 * adjust_dirindices
2123 direntry_t* result = array_insert(&(s->directory), dir_index, count);
2124 if (result == NULL)
2125 return NULL;
2126 adjust_dirindices(s, dir_index, count);
2127 return result;
2130 static int remove_direntries(BDRVVVFATState* s, int dir_index, int count)
2132 int ret = array_remove_slice(&(s->directory), dir_index, count);
2133 if (ret)
2134 return ret;
2135 adjust_dirindices(s, dir_index, -count);
2136 return 0;
2140 * Adapt the mappings of the cluster chain starting at first cluster
2141 * (i.e. if a file starts at first_cluster, the chain is followed according
2142 * to the modified fat, and the corresponding entries in s->mapping are
2143 * adjusted)
2145 static int commit_mappings(BDRVVVFATState* s,
2146 uint32_t first_cluster, int dir_index)
2148 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2149 direntry_t* direntry = array_get(&(s->directory), dir_index);
2150 uint32_t cluster = first_cluster;
2152 vvfat_close_current_file(s);
2154 assert(mapping);
2155 assert(mapping->begin == first_cluster);
2156 mapping->first_mapping_index = -1;
2157 mapping->dir_index = dir_index;
2158 mapping->mode = (dir_index <= 0 || is_directory(direntry)) ?
2159 MODE_DIRECTORY : MODE_NORMAL;
2161 while (!fat_eof(s, cluster)) {
2162 uint32_t c, c1;
2164 for (c = cluster, c1 = modified_fat_get(s, c); c + 1 == c1;
2165 c = c1, c1 = modified_fat_get(s, c1));
2167 c++;
2168 if (c > mapping->end) {
2169 int index = array_index(&(s->mapping), mapping);
2170 int i, max_i = s->mapping.next - index;
2171 for (i = 1; i < max_i && mapping[i].begin < c; i++);
2172 while (--i > 0)
2173 remove_mapping(s, index + 1);
2175 assert(mapping == array_get(&(s->mapping), s->mapping.next - 1)
2176 || mapping[1].begin >= c);
2177 mapping->end = c;
2179 if (!fat_eof(s, c1)) {
2180 int i = find_mapping_for_cluster_aux(s, c1, 0, s->mapping.next);
2181 mapping_t* next_mapping = i >= s->mapping.next ? NULL :
2182 array_get(&(s->mapping), i);
2184 if (next_mapping == NULL || next_mapping->begin > c1) {
2185 int i1 = array_index(&(s->mapping), mapping);
2187 next_mapping = insert_mapping(s, c1, c1+1);
2189 if (c1 < c)
2190 i1++;
2191 mapping = array_get(&(s->mapping), i1);
2194 next_mapping->dir_index = mapping->dir_index;
2195 next_mapping->first_mapping_index =
2196 mapping->first_mapping_index < 0 ?
2197 array_index(&(s->mapping), mapping) :
2198 mapping->first_mapping_index;
2199 next_mapping->path = mapping->path;
2200 next_mapping->mode = mapping->mode;
2201 next_mapping->read_only = mapping->read_only;
2202 if (mapping->mode & MODE_DIRECTORY) {
2203 next_mapping->info.dir.parent_mapping_index =
2204 mapping->info.dir.parent_mapping_index;
2205 next_mapping->info.dir.first_dir_index =
2206 mapping->info.dir.first_dir_index +
2207 0x10 * s->sectors_per_cluster *
2208 (mapping->end - mapping->begin);
2209 } else
2210 next_mapping->info.file.offset = mapping->info.file.offset +
2211 mapping->end - mapping->begin;
2213 mapping = next_mapping;
2216 cluster = c1;
2219 return 0;
2222 static int commit_direntries(BDRVVVFATState* s,
2223 int dir_index, int parent_mapping_index)
2225 direntry_t* direntry = array_get(&(s->directory), dir_index);
2226 uint32_t first_cluster = dir_index == 0 ? 0 : begin_of_direntry(direntry);
2227 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2229 int factor = 0x10 * s->sectors_per_cluster;
2230 int old_cluster_count, new_cluster_count;
2231 int current_dir_index = mapping->info.dir.first_dir_index;
2232 int first_dir_index = current_dir_index;
2233 int ret, i;
2234 uint32_t c;
2236 DLOG(fprintf(stderr, "commit_direntries for %s, parent_mapping_index %d\n", mapping->path, parent_mapping_index));
2238 assert(direntry);
2239 assert(mapping);
2240 assert(mapping->begin == first_cluster);
2241 assert(mapping->info.dir.first_dir_index < s->directory.next);
2242 assert(mapping->mode & MODE_DIRECTORY);
2243 assert(dir_index == 0 || is_directory(direntry));
2245 mapping->info.dir.parent_mapping_index = parent_mapping_index;
2247 if (first_cluster == 0) {
2248 old_cluster_count = new_cluster_count =
2249 s->last_cluster_of_root_directory;
2250 } else {
2251 for (old_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2252 c = fat_get(s, c))
2253 old_cluster_count++;
2255 for (new_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2256 c = modified_fat_get(s, c))
2257 new_cluster_count++;
2260 if (new_cluster_count > old_cluster_count) {
2261 if (insert_direntries(s,
2262 current_dir_index + factor * old_cluster_count,
2263 factor * (new_cluster_count - old_cluster_count)) == NULL)
2264 return -1;
2265 } else if (new_cluster_count < old_cluster_count)
2266 remove_direntries(s,
2267 current_dir_index + factor * new_cluster_count,
2268 factor * (old_cluster_count - new_cluster_count));
2270 for (c = first_cluster; !fat_eof(s, c); c = modified_fat_get(s, c)) {
2271 void* direntry = array_get(&(s->directory), current_dir_index);
2272 int ret = vvfat_read(s->bs, cluster2sector(s, c), direntry,
2273 s->sectors_per_cluster);
2274 if (ret)
2275 return ret;
2276 assert(!strncmp(s->directory.pointer, "QEMU", 4));
2277 current_dir_index += factor;
2280 ret = commit_mappings(s, first_cluster, dir_index);
2281 if (ret)
2282 return ret;
2284 /* recurse */
2285 for (i = 0; i < factor * new_cluster_count; i++) {
2286 direntry = array_get(&(s->directory), first_dir_index + i);
2287 if (is_directory(direntry) && !is_dot(direntry)) {
2288 mapping = find_mapping_for_cluster(s, first_cluster);
2289 assert(mapping->mode & MODE_DIRECTORY);
2290 ret = commit_direntries(s, first_dir_index + i,
2291 array_index(&(s->mapping), mapping));
2292 if (ret)
2293 return ret;
2297 return 0;
2300 /* commit one file (adjust contents, adjust mapping),
2301 return first_mapping_index */
2302 static int commit_one_file(BDRVVVFATState* s,
2303 int dir_index, uint32_t offset)
2305 direntry_t* direntry = array_get(&(s->directory), dir_index);
2306 uint32_t c = begin_of_direntry(direntry);
2307 uint32_t first_cluster = c;
2308 mapping_t* mapping = find_mapping_for_cluster(s, c);
2309 uint32_t size = filesize_of_direntry(direntry);
2310 char* cluster = g_malloc(s->cluster_size);
2311 uint32_t i;
2312 int fd = 0;
2314 assert(offset < size);
2315 assert((offset % s->cluster_size) == 0);
2317 for (i = s->cluster_size; i < offset; i += s->cluster_size)
2318 c = modified_fat_get(s, c);
2320 fd = qemu_open(mapping->path, O_RDWR | O_CREAT | O_BINARY, 0666);
2321 if (fd < 0) {
2322 fprintf(stderr, "Could not open %s... (%s, %d)\n", mapping->path,
2323 strerror(errno), errno);
2324 g_free(cluster);
2325 return fd;
2327 if (offset > 0) {
2328 if (lseek(fd, offset, SEEK_SET) != offset) {
2329 qemu_close(fd);
2330 g_free(cluster);
2331 return -3;
2335 while (offset < size) {
2336 uint32_t c1;
2337 int rest_size = (size - offset > s->cluster_size ?
2338 s->cluster_size : size - offset);
2339 int ret;
2341 c1 = modified_fat_get(s, c);
2343 assert((size - offset == 0 && fat_eof(s, c)) ||
2344 (size > offset && c >=2 && !fat_eof(s, c)));
2346 ret = vvfat_read(s->bs, cluster2sector(s, c),
2347 (uint8_t*)cluster, (rest_size + 0x1ff) / 0x200);
2349 if (ret < 0) {
2350 qemu_close(fd);
2351 g_free(cluster);
2352 return ret;
2355 if (write(fd, cluster, rest_size) < 0) {
2356 qemu_close(fd);
2357 g_free(cluster);
2358 return -2;
2361 offset += rest_size;
2362 c = c1;
2365 if (ftruncate(fd, size)) {
2366 perror("ftruncate()");
2367 qemu_close(fd);
2368 g_free(cluster);
2369 return -4;
2371 qemu_close(fd);
2372 g_free(cluster);
2374 return commit_mappings(s, first_cluster, dir_index);
2377 #ifdef DEBUG
2378 /* test, if all mappings point to valid direntries */
2379 static void check1(BDRVVVFATState* s)
2381 int i;
2382 for (i = 0; i < s->mapping.next; i++) {
2383 mapping_t* mapping = array_get(&(s->mapping), i);
2384 if (mapping->mode & MODE_DELETED) {
2385 fprintf(stderr, "deleted\n");
2386 continue;
2388 assert(mapping->dir_index < s->directory.next);
2389 direntry_t* direntry = array_get(&(s->directory), mapping->dir_index);
2390 assert(mapping->begin == begin_of_direntry(direntry) || mapping->first_mapping_index >= 0);
2391 if (mapping->mode & MODE_DIRECTORY) {
2392 assert(mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster * (mapping->end - mapping->begin) <= s->directory.next);
2393 assert((mapping->info.dir.first_dir_index % (0x10 * s->sectors_per_cluster)) == 0);
2398 /* test, if all direntries have mappings */
2399 static void check2(BDRVVVFATState* s)
2401 int i;
2402 int first_mapping = -1;
2404 for (i = 0; i < s->directory.next; i++) {
2405 direntry_t* direntry = array_get(&(s->directory), i);
2407 if (is_short_name(direntry) && begin_of_direntry(direntry)) {
2408 mapping_t* mapping = find_mapping_for_cluster(s, begin_of_direntry(direntry));
2409 assert(mapping);
2410 assert(mapping->dir_index == i || is_dot(direntry));
2411 assert(mapping->begin == begin_of_direntry(direntry) || is_dot(direntry));
2414 if ((i % (0x10 * s->sectors_per_cluster)) == 0) {
2415 /* cluster start */
2416 int j, count = 0;
2418 for (j = 0; j < s->mapping.next; j++) {
2419 mapping_t* mapping = array_get(&(s->mapping), j);
2420 if (mapping->mode & MODE_DELETED)
2421 continue;
2422 if (mapping->mode & MODE_DIRECTORY) {
2423 if (mapping->info.dir.first_dir_index <= i && mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster > i) {
2424 assert(++count == 1);
2425 if (mapping->first_mapping_index == -1)
2426 first_mapping = array_index(&(s->mapping), mapping);
2427 else
2428 assert(first_mapping == mapping->first_mapping_index);
2429 if (mapping->info.dir.parent_mapping_index < 0)
2430 assert(j == 0);
2431 else {
2432 mapping_t* parent = array_get(&(s->mapping), mapping->info.dir.parent_mapping_index);
2433 assert(parent->mode & MODE_DIRECTORY);
2434 assert(parent->info.dir.first_dir_index < mapping->info.dir.first_dir_index);
2439 if (count == 0)
2440 first_mapping = -1;
2444 #endif
2446 static int handle_renames_and_mkdirs(BDRVVVFATState* s)
2448 int i;
2450 #ifdef DEBUG
2451 fprintf(stderr, "handle_renames\n");
2452 for (i = 0; i < s->commits.next; i++) {
2453 commit_t* commit = array_get(&(s->commits), i);
2454 fprintf(stderr, "%d, %s (%d, %d)\n", i, commit->path ? commit->path : "(null)", commit->param.rename.cluster, commit->action);
2456 #endif
2458 for (i = 0; i < s->commits.next;) {
2459 commit_t* commit = array_get(&(s->commits), i);
2460 if (commit->action == ACTION_RENAME) {
2461 mapping_t* mapping = find_mapping_for_cluster(s,
2462 commit->param.rename.cluster);
2463 char* old_path = mapping->path;
2465 assert(commit->path);
2466 mapping->path = commit->path;
2467 if (rename(old_path, mapping->path))
2468 return -2;
2470 if (mapping->mode & MODE_DIRECTORY) {
2471 int l1 = strlen(mapping->path);
2472 int l2 = strlen(old_path);
2473 int diff = l1 - l2;
2474 direntry_t* direntry = array_get(&(s->directory),
2475 mapping->info.dir.first_dir_index);
2476 uint32_t c = mapping->begin;
2477 int i = 0;
2479 /* recurse */
2480 while (!fat_eof(s, c)) {
2481 do {
2482 direntry_t* d = direntry + i;
2484 if (is_file(d) || (is_directory(d) && !is_dot(d))) {
2485 mapping_t* m = find_mapping_for_cluster(s,
2486 begin_of_direntry(d));
2487 int l = strlen(m->path);
2488 char* new_path = g_malloc(l + diff + 1);
2490 assert(!strncmp(m->path, mapping->path, l2));
2492 pstrcpy(new_path, l + diff + 1, mapping->path);
2493 pstrcpy(new_path + l1, l + diff + 1 - l1,
2494 m->path + l2);
2496 schedule_rename(s, m->begin, new_path);
2498 i++;
2499 } while((i % (0x10 * s->sectors_per_cluster)) != 0);
2500 c = fat_get(s, c);
2504 g_free(old_path);
2505 array_remove(&(s->commits), i);
2506 continue;
2507 } else if (commit->action == ACTION_MKDIR) {
2508 mapping_t* mapping;
2509 int j, parent_path_len;
2511 #ifdef __MINGW32__
2512 if (mkdir(commit->path))
2513 return -5;
2514 #else
2515 if (mkdir(commit->path, 0755))
2516 return -5;
2517 #endif
2519 mapping = insert_mapping(s, commit->param.mkdir.cluster,
2520 commit->param.mkdir.cluster + 1);
2521 if (mapping == NULL)
2522 return -6;
2524 mapping->mode = MODE_DIRECTORY;
2525 mapping->read_only = 0;
2526 mapping->path = commit->path;
2527 j = s->directory.next;
2528 assert(j);
2529 insert_direntries(s, s->directory.next,
2530 0x10 * s->sectors_per_cluster);
2531 mapping->info.dir.first_dir_index = j;
2533 parent_path_len = strlen(commit->path)
2534 - strlen(get_basename(commit->path)) - 1;
2535 for (j = 0; j < s->mapping.next; j++) {
2536 mapping_t* m = array_get(&(s->mapping), j);
2537 if (m->first_mapping_index < 0 && m != mapping &&
2538 !strncmp(m->path, mapping->path, parent_path_len) &&
2539 strlen(m->path) == parent_path_len)
2540 break;
2542 assert(j < s->mapping.next);
2543 mapping->info.dir.parent_mapping_index = j;
2545 array_remove(&(s->commits), i);
2546 continue;
2549 i++;
2551 return 0;
2555 * TODO: make sure that the short name is not matching *another* file
2557 static int handle_commits(BDRVVVFATState* s)
2559 int i, fail = 0;
2561 vvfat_close_current_file(s);
2563 for (i = 0; !fail && i < s->commits.next; i++) {
2564 commit_t* commit = array_get(&(s->commits), i);
2565 switch(commit->action) {
2566 case ACTION_RENAME: case ACTION_MKDIR:
2567 abort();
2568 fail = -2;
2569 break;
2570 case ACTION_WRITEOUT: {
2571 #ifndef NDEBUG
2572 /* these variables are only used by assert() below */
2573 direntry_t* entry = array_get(&(s->directory),
2574 commit->param.writeout.dir_index);
2575 uint32_t begin = begin_of_direntry(entry);
2576 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2577 #endif
2579 assert(mapping);
2580 assert(mapping->begin == begin);
2581 assert(commit->path == NULL);
2583 if (commit_one_file(s, commit->param.writeout.dir_index,
2584 commit->param.writeout.modified_offset))
2585 fail = -3;
2587 break;
2589 case ACTION_NEW_FILE: {
2590 int begin = commit->param.new_file.first_cluster;
2591 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2592 direntry_t* entry;
2593 int i;
2595 /* find direntry */
2596 for (i = 0; i < s->directory.next; i++) {
2597 entry = array_get(&(s->directory), i);
2598 if (is_file(entry) && begin_of_direntry(entry) == begin)
2599 break;
2602 if (i >= s->directory.next) {
2603 fail = -6;
2604 continue;
2607 /* make sure there exists an initial mapping */
2608 if (mapping && mapping->begin != begin) {
2609 mapping->end = begin;
2610 mapping = NULL;
2612 if (mapping == NULL) {
2613 mapping = insert_mapping(s, begin, begin+1);
2615 /* most members will be fixed in commit_mappings() */
2616 assert(commit->path);
2617 mapping->path = commit->path;
2618 mapping->read_only = 0;
2619 mapping->mode = MODE_NORMAL;
2620 mapping->info.file.offset = 0;
2622 if (commit_one_file(s, i, 0))
2623 fail = -7;
2625 break;
2627 default:
2628 abort();
2631 if (i > 0 && array_remove_slice(&(s->commits), 0, i))
2632 return -1;
2633 return fail;
2636 static int handle_deletes(BDRVVVFATState* s)
2638 int i, deferred = 1, deleted = 1;
2640 /* delete files corresponding to mappings marked as deleted */
2641 /* handle DELETEs and unused mappings (modified_fat_get(s, mapping->begin) == 0) */
2642 while (deferred && deleted) {
2643 deferred = 0;
2644 deleted = 0;
2646 for (i = 1; i < s->mapping.next; i++) {
2647 mapping_t* mapping = array_get(&(s->mapping), i);
2648 if (mapping->mode & MODE_DELETED) {
2649 direntry_t* entry = array_get(&(s->directory),
2650 mapping->dir_index);
2652 if (is_free(entry)) {
2653 /* remove file/directory */
2654 if (mapping->mode & MODE_DIRECTORY) {
2655 int j, next_dir_index = s->directory.next,
2656 first_dir_index = mapping->info.dir.first_dir_index;
2658 if (rmdir(mapping->path) < 0) {
2659 if (errno == ENOTEMPTY) {
2660 deferred++;
2661 continue;
2662 } else
2663 return -5;
2666 for (j = 1; j < s->mapping.next; j++) {
2667 mapping_t* m = array_get(&(s->mapping), j);
2668 if (m->mode & MODE_DIRECTORY &&
2669 m->info.dir.first_dir_index >
2670 first_dir_index &&
2671 m->info.dir.first_dir_index <
2672 next_dir_index)
2673 next_dir_index =
2674 m->info.dir.first_dir_index;
2676 remove_direntries(s, first_dir_index,
2677 next_dir_index - first_dir_index);
2679 deleted++;
2681 } else {
2682 if (unlink(mapping->path))
2683 return -4;
2684 deleted++;
2686 DLOG(fprintf(stderr, "DELETE (%d)\n", i); print_mapping(mapping); print_direntry(entry));
2687 remove_mapping(s, i);
2692 return 0;
2696 * synchronize mapping with new state:
2698 * - copy FAT (with bdrv_read)
2699 * - mark all filenames corresponding to mappings as deleted
2700 * - recurse direntries from root (using bs->bdrv_read)
2701 * - delete files corresponding to mappings marked as deleted
2703 static int do_commit(BDRVVVFATState* s)
2705 int ret = 0;
2707 /* the real meat are the commits. Nothing to do? Move along! */
2708 if (s->commits.next == 0)
2709 return 0;
2711 vvfat_close_current_file(s);
2713 ret = handle_renames_and_mkdirs(s);
2714 if (ret) {
2715 fprintf(stderr, "Error handling renames (%d)\n", ret);
2716 abort();
2717 return ret;
2720 /* copy FAT (with bdrv_read) */
2721 memcpy(s->fat.pointer, s->fat2, 0x200 * s->sectors_per_fat);
2723 /* recurse direntries from root (using bs->bdrv_read) */
2724 ret = commit_direntries(s, 0, -1);
2725 if (ret) {
2726 fprintf(stderr, "Fatal: error while committing (%d)\n", ret);
2727 abort();
2728 return ret;
2731 ret = handle_commits(s);
2732 if (ret) {
2733 fprintf(stderr, "Error handling commits (%d)\n", ret);
2734 abort();
2735 return ret;
2738 ret = handle_deletes(s);
2739 if (ret) {
2740 fprintf(stderr, "Error deleting\n");
2741 abort();
2742 return ret;
2745 if (s->qcow->drv->bdrv_make_empty) {
2746 s->qcow->drv->bdrv_make_empty(s->qcow);
2749 memset(s->used_clusters, 0, sector2cluster(s, s->sector_count));
2751 DLOG(checkpoint());
2752 return 0;
2755 static int try_commit(BDRVVVFATState* s)
2757 vvfat_close_current_file(s);
2758 DLOG(checkpoint());
2759 if(!is_consistent(s))
2760 return -1;
2761 return do_commit(s);
2764 static int vvfat_write(BlockDriverState *bs, int64_t sector_num,
2765 const uint8_t *buf, int nb_sectors)
2767 BDRVVVFATState *s = bs->opaque;
2768 int i, ret;
2770 DLOG(checkpoint());
2772 /* Check if we're operating in read-only mode */
2773 if (s->qcow == NULL) {
2774 return -EACCES;
2777 vvfat_close_current_file(s);
2780 * Some sanity checks:
2781 * - do not allow writing to the boot sector
2782 * - do not allow to write non-ASCII filenames
2785 if (sector_num < s->first_sectors_number)
2786 return -1;
2788 for (i = sector2cluster(s, sector_num);
2789 i <= sector2cluster(s, sector_num + nb_sectors - 1);) {
2790 mapping_t* mapping = find_mapping_for_cluster(s, i);
2791 if (mapping) {
2792 if (mapping->read_only) {
2793 fprintf(stderr, "Tried to write to write-protected file %s\n",
2794 mapping->path);
2795 return -1;
2798 if (mapping->mode & MODE_DIRECTORY) {
2799 int begin = cluster2sector(s, i);
2800 int end = begin + s->sectors_per_cluster, k;
2801 int dir_index;
2802 const direntry_t* direntries;
2803 long_file_name lfn;
2805 lfn_init(&lfn);
2807 if (begin < sector_num)
2808 begin = sector_num;
2809 if (end > sector_num + nb_sectors)
2810 end = sector_num + nb_sectors;
2811 dir_index = mapping->dir_index +
2812 0x10 * (begin - mapping->begin * s->sectors_per_cluster);
2813 direntries = (direntry_t*)(buf + 0x200 * (begin - sector_num));
2815 for (k = 0; k < (end - begin) * 0x10; k++) {
2816 /* do not allow non-ASCII filenames */
2817 if (parse_long_name(&lfn, direntries + k) < 0) {
2818 fprintf(stderr, "Warning: non-ASCII filename\n");
2819 return -1;
2821 /* no access to the direntry of a read-only file */
2822 else if (is_short_name(direntries+k) &&
2823 (direntries[k].attributes & 1)) {
2824 if (memcmp(direntries + k,
2825 array_get(&(s->directory), dir_index + k),
2826 sizeof(direntry_t))) {
2827 fprintf(stderr, "Warning: tried to write to write-protected file\n");
2828 return -1;
2833 i = mapping->end;
2834 } else
2835 i++;
2839 * Use qcow backend. Commit later.
2841 DLOG(fprintf(stderr, "Write to qcow backend: %d + %d\n", (int)sector_num, nb_sectors));
2842 ret = bdrv_write(s->qcow, sector_num, buf, nb_sectors);
2843 if (ret < 0) {
2844 fprintf(stderr, "Error writing to qcow backend\n");
2845 return ret;
2848 for (i = sector2cluster(s, sector_num);
2849 i <= sector2cluster(s, sector_num + nb_sectors - 1); i++)
2850 if (i >= 0)
2851 s->used_clusters[i] |= USED_ALLOCATED;
2853 DLOG(checkpoint());
2854 /* TODO: add timeout */
2855 try_commit(s);
2857 DLOG(checkpoint());
2858 return 0;
2861 static coroutine_fn int vvfat_co_write(BlockDriverState *bs, int64_t sector_num,
2862 const uint8_t *buf, int nb_sectors)
2864 int ret;
2865 BDRVVVFATState *s = bs->opaque;
2866 qemu_co_mutex_lock(&s->lock);
2867 ret = vvfat_write(bs, sector_num, buf, nb_sectors);
2868 qemu_co_mutex_unlock(&s->lock);
2869 return ret;
2872 static int64_t coroutine_fn vvfat_co_get_block_status(BlockDriverState *bs,
2873 int64_t sector_num, int nb_sectors, int* n)
2875 BDRVVVFATState* s = bs->opaque;
2876 *n = s->sector_count - sector_num;
2877 if (*n > nb_sectors) {
2878 *n = nb_sectors;
2879 } else if (*n < 0) {
2880 return 0;
2882 return BDRV_BLOCK_DATA;
2885 static int write_target_commit(BlockDriverState *bs, int64_t sector_num,
2886 const uint8_t* buffer, int nb_sectors) {
2887 BDRVVVFATState* s = *((BDRVVVFATState**) bs->opaque);
2888 return try_commit(s);
2891 static void write_target_close(BlockDriverState *bs) {
2892 BDRVVVFATState* s = *((BDRVVVFATState**) bs->opaque);
2893 bdrv_unref(s->qcow);
2894 g_free(s->qcow_filename);
2897 static BlockDriver vvfat_write_target = {
2898 .format_name = "vvfat_write_target",
2899 .bdrv_write = write_target_commit,
2900 .bdrv_close = write_target_close,
2903 static int enable_write_target(BDRVVVFATState *s, Error **errp)
2905 BlockDriver *bdrv_qcow = NULL;
2906 QemuOpts *opts = NULL;
2907 int ret;
2908 int size = sector2cluster(s, s->sector_count);
2909 s->used_clusters = calloc(size, 1);
2911 array_init(&(s->commits), sizeof(commit_t));
2913 s->qcow_filename = g_malloc(PATH_MAX);
2914 ret = get_tmp_filename(s->qcow_filename, PATH_MAX);
2915 if (ret < 0) {
2916 error_setg_errno(errp, -ret, "can't create temporary file");
2917 goto err;
2920 bdrv_qcow = bdrv_find_format("qcow");
2921 if (!bdrv_qcow) {
2922 error_setg(errp, "Failed to locate qcow driver");
2923 ret = -ENOENT;
2924 goto err;
2927 opts = qemu_opts_create(bdrv_qcow->create_opts, NULL, 0, &error_abort);
2928 qemu_opt_set_number(opts, BLOCK_OPT_SIZE, s->sector_count * 512,
2929 &error_abort);
2930 qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, "fat:", &error_abort);
2932 ret = bdrv_create(bdrv_qcow, s->qcow_filename, opts, errp);
2933 qemu_opts_del(opts);
2934 if (ret < 0) {
2935 goto err;
2938 s->qcow = NULL;
2939 ret = bdrv_open(&s->qcow, s->qcow_filename, NULL, NULL,
2940 BDRV_O_RDWR | BDRV_O_CACHE_WB | BDRV_O_NO_FLUSH,
2941 bdrv_qcow, errp);
2942 if (ret < 0) {
2943 goto err;
2946 #ifndef _WIN32
2947 unlink(s->qcow_filename);
2948 #endif
2950 bdrv_set_backing_hd(s->bs, bdrv_new());
2951 s->bs->backing_hd->drv = &vvfat_write_target;
2952 s->bs->backing_hd->opaque = g_new(void *, 1);
2953 *(void**)s->bs->backing_hd->opaque = s;
2955 return 0;
2957 err:
2958 g_free(s->qcow_filename);
2959 s->qcow_filename = NULL;
2960 return ret;
2963 static void vvfat_close(BlockDriverState *bs)
2965 BDRVVVFATState *s = bs->opaque;
2967 vvfat_close_current_file(s);
2968 array_free(&(s->fat));
2969 array_free(&(s->directory));
2970 array_free(&(s->mapping));
2971 g_free(s->cluster_buffer);
2973 if (s->qcow) {
2974 migrate_del_blocker(s->migration_blocker);
2975 error_free(s->migration_blocker);
2979 static BlockDriver bdrv_vvfat = {
2980 .format_name = "vvfat",
2981 .protocol_name = "fat",
2982 .instance_size = sizeof(BDRVVVFATState),
2984 .bdrv_parse_filename = vvfat_parse_filename,
2985 .bdrv_file_open = vvfat_open,
2986 .bdrv_close = vvfat_close,
2987 .bdrv_rebind = vvfat_rebind,
2989 .bdrv_read = vvfat_co_read,
2990 .bdrv_write = vvfat_co_write,
2991 .bdrv_co_get_block_status = vvfat_co_get_block_status,
2994 static void bdrv_vvfat_init(void)
2996 bdrv_register(&bdrv_vvfat);
2999 block_init(bdrv_vvfat_init);
3001 #ifdef DEBUG
3002 static void checkpoint(void) {
3003 assert(((mapping_t*)array_get(&(vvv->mapping), 0))->end == 2);
3004 check1(vvv);
3005 check2(vvv);
3006 assert(!vvv->current_mapping || vvv->current_fd || (vvv->current_mapping->mode & MODE_DIRECTORY));
3007 #if 0
3008 if (((direntry_t*)vvv->directory.pointer)[1].attributes != 0xf)
3009 fprintf(stderr, "Nonono!\n");
3010 mapping_t* mapping;
3011 direntry_t* direntry;
3012 assert(vvv->mapping.size >= vvv->mapping.item_size * vvv->mapping.next);
3013 assert(vvv->directory.size >= vvv->directory.item_size * vvv->directory.next);
3014 if (vvv->mapping.next<47)
3015 return;
3016 assert((mapping = array_get(&(vvv->mapping), 47)));
3017 assert(mapping->dir_index < vvv->directory.next);
3018 direntry = array_get(&(vvv->directory), mapping->dir_index);
3019 assert(!memcmp(direntry->name, "USB H ", 11) || direntry->name[0]==0);
3020 #endif
3022 #endif