stream: Switch stream_populate() to byte-based
[qemu/ar7.git] / block / vvfat.c
blobf55104c338f159df234c16d91d1c113d7e98ea91
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 "qemu/osdep.h"
26 #include <dirent.h>
27 #include "qapi/error.h"
28 #include "block/block_int.h"
29 #include "qemu/module.h"
30 #include "qemu/bswap.h"
31 #include "migration/blocker.h"
32 #include "qapi/qmp/qbool.h"
33 #include "qapi/qmp/qstring.h"
34 #include "qemu/cutils.h"
36 #ifndef S_IWGRP
37 #define S_IWGRP 0
38 #endif
39 #ifndef S_IWOTH
40 #define S_IWOTH 0
41 #endif
43 /* TODO: add ":bootsector=blabla.img:" */
44 /* LATER TODO: add automatic boot sector generation from
45 BOOTEASY.ASM and Ranish Partition Manager
46 Note that DOS assumes the system files to be the first files in the
47 file system (test if the boot sector still relies on that fact)! */
48 /* MAYBE TODO: write block-visofs.c */
49 /* TODO: call try_commit() only after a timeout */
51 /* #define DEBUG */
53 #ifdef DEBUG
55 #define DLOG(a) a
57 static void checkpoint(void);
59 #ifdef __MINGW32__
60 void nonono(const char* file, int line, const char* msg) {
61 fprintf(stderr, "Nonono! %s:%d %s\n", file, line, msg);
62 exit(-5);
64 #undef assert
65 #define assert(a) do {if (!(a)) nonono(__FILE__, __LINE__, #a);}while(0)
66 #endif
68 #else
70 #define DLOG(a)
72 #endif
74 /* dynamic array functions */
75 typedef struct array_t {
76 char* pointer;
77 unsigned int size,next,item_size;
78 } array_t;
80 static inline void array_init(array_t* array,unsigned int item_size)
82 array->pointer = NULL;
83 array->size=0;
84 array->next=0;
85 array->item_size=item_size;
88 static inline void array_free(array_t* array)
90 g_free(array->pointer);
91 array->size=array->next=0;
94 /* does not automatically grow */
95 static inline void* array_get(array_t* array,unsigned int index) {
96 assert(index < array->next);
97 return array->pointer + index * array->item_size;
100 static inline int array_ensure_allocated(array_t* array, int index)
102 if((index + 1) * array->item_size > array->size) {
103 int new_size = (index + 32) * array->item_size;
104 array->pointer = g_realloc(array->pointer, new_size);
105 if (!array->pointer)
106 return -1;
107 array->size = new_size;
108 array->next = index + 1;
111 return 0;
114 static inline void* array_get_next(array_t* array) {
115 unsigned int next = array->next;
117 if (array_ensure_allocated(array, next) < 0)
118 return NULL;
120 array->next = next + 1;
121 return array_get(array, next);
124 static inline void* array_insert(array_t* array,unsigned int index,unsigned int count) {
125 if((array->next+count)*array->item_size>array->size) {
126 int increment=count*array->item_size;
127 array->pointer=g_realloc(array->pointer,array->size+increment);
128 if(!array->pointer)
129 return NULL;
130 array->size+=increment;
132 memmove(array->pointer+(index+count)*array->item_size,
133 array->pointer+index*array->item_size,
134 (array->next-index)*array->item_size);
135 array->next+=count;
136 return array->pointer+index*array->item_size;
139 /* this performs a "roll", so that the element which was at index_from becomes
140 * index_to, but the order of all other elements is preserved. */
141 static inline int array_roll(array_t* array,int index_to,int index_from,int count)
143 char* buf;
144 char* from;
145 char* to;
146 int is;
148 if(!array ||
149 index_to<0 || index_to>=array->next ||
150 index_from<0 || index_from>=array->next)
151 return -1;
153 if(index_to==index_from)
154 return 0;
156 is=array->item_size;
157 from=array->pointer+index_from*is;
158 to=array->pointer+index_to*is;
159 buf=g_malloc(is*count);
160 memcpy(buf,from,is*count);
162 if(index_to<index_from)
163 memmove(to+is*count,to,from-to);
164 else
165 memmove(from,from+is*count,to-from);
167 memcpy(to,buf,is*count);
169 g_free(buf);
171 return 0;
174 static inline int array_remove_slice(array_t* array,int index, int count)
176 assert(index >=0);
177 assert(count > 0);
178 assert(index + count <= array->next);
179 if(array_roll(array,array->next-1,index,count))
180 return -1;
181 array->next -= count;
182 return 0;
185 static int array_remove(array_t* array,int index)
187 return array_remove_slice(array, index, 1);
190 /* return the index for a given member */
191 static int array_index(array_t* array, void* pointer)
193 size_t offset = (char*)pointer - array->pointer;
194 assert((offset % array->item_size) == 0);
195 assert(offset/array->item_size < array->next);
196 return offset/array->item_size;
199 /* These structures are used to fake a disk and the VFAT filesystem.
200 * For this reason we need to use QEMU_PACKED. */
202 typedef struct bootsector_t {
203 uint8_t jump[3];
204 uint8_t name[8];
205 uint16_t sector_size;
206 uint8_t sectors_per_cluster;
207 uint16_t reserved_sectors;
208 uint8_t number_of_fats;
209 uint16_t root_entries;
210 uint16_t total_sectors16;
211 uint8_t media_type;
212 uint16_t sectors_per_fat;
213 uint16_t sectors_per_track;
214 uint16_t number_of_heads;
215 uint32_t hidden_sectors;
216 uint32_t total_sectors;
217 union {
218 struct {
219 uint8_t drive_number;
220 uint8_t reserved1;
221 uint8_t signature;
222 uint32_t id;
223 uint8_t volume_label[11];
224 uint8_t fat_type[8];
225 uint8_t ignored[0x1c0];
226 } QEMU_PACKED fat16;
227 struct {
228 uint32_t sectors_per_fat;
229 uint16_t flags;
230 uint8_t major,minor;
231 uint32_t first_cluster_of_root_dir;
232 uint16_t info_sector;
233 uint16_t backup_boot_sector;
234 uint8_t reserved[12];
235 uint8_t drive_number;
236 uint8_t reserved1;
237 uint8_t signature;
238 uint32_t id;
239 uint8_t volume_label[11];
240 uint8_t fat_type[8];
241 uint8_t ignored[0x1a4];
242 } QEMU_PACKED fat32;
243 } u;
244 uint8_t magic[2];
245 } QEMU_PACKED bootsector_t;
247 typedef struct {
248 uint8_t head;
249 uint8_t sector;
250 uint8_t cylinder;
251 } mbr_chs_t;
253 typedef struct partition_t {
254 uint8_t attributes; /* 0x80 = bootable */
255 mbr_chs_t start_CHS;
256 uint8_t fs_type; /* 0x1 = FAT12, 0x6 = FAT16, 0xe = FAT16_LBA, 0xb = FAT32, 0xc = FAT32_LBA */
257 mbr_chs_t end_CHS;
258 uint32_t start_sector_long;
259 uint32_t length_sector_long;
260 } QEMU_PACKED partition_t;
262 typedef struct mbr_t {
263 uint8_t ignored[0x1b8];
264 uint32_t nt_id;
265 uint8_t ignored2[2];
266 partition_t partition[4];
267 uint8_t magic[2];
268 } QEMU_PACKED mbr_t;
270 typedef struct direntry_t {
271 uint8_t name[8 + 3];
272 uint8_t attributes;
273 uint8_t reserved[2];
274 uint16_t ctime;
275 uint16_t cdate;
276 uint16_t adate;
277 uint16_t begin_hi;
278 uint16_t mtime;
279 uint16_t mdate;
280 uint16_t begin;
281 uint32_t size;
282 } QEMU_PACKED direntry_t;
284 /* this structure are used to transparently access the files */
286 typedef struct mapping_t {
287 /* begin is the first cluster, end is the last+1 */
288 uint32_t begin,end;
289 /* as s->directory is growable, no pointer may be used here */
290 unsigned int dir_index;
291 /* the clusters of a file may be in any order; this points to the first */
292 int first_mapping_index;
293 union {
294 /* offset is
295 * - the offset in the file (in clusters) for a file, or
296 * - the next cluster of the directory for a directory
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 {
310 MODE_UNDEFINED = 0,
311 MODE_NORMAL = 1,
312 MODE_MODIFIED = 2,
313 MODE_DIRECTORY = 4,
314 MODE_DELETED = 8,
315 } mode;
316 int read_only;
317 } mapping_t;
319 #ifdef DEBUG
320 static void print_direntry(const struct direntry_t*);
321 static void print_mapping(const struct mapping_t* mapping);
322 #endif
324 /* here begins the real VVFAT driver */
326 typedef struct BDRVVVFATState {
327 CoMutex lock;
328 BlockDriverState* bs; /* pointer to parent */
329 unsigned char first_sectors[0x40*0x200];
331 int fat_type; /* 16 or 32 */
332 array_t fat,directory,mapping;
333 char volume_label[11];
335 uint32_t offset_to_bootsector; /* 0 for floppy, 0x3f for disk */
337 unsigned int cluster_size;
338 unsigned int sectors_per_cluster;
339 unsigned int sectors_per_fat;
340 uint32_t last_cluster_of_root_directory;
341 /* how many entries are available in root directory (0 for FAT32) */
342 uint16_t root_entries;
343 uint32_t sector_count; /* total number of sectors of the partition */
344 uint32_t cluster_count; /* total number of clusters of this partition */
345 uint32_t max_fat_value;
346 uint32_t offset_to_fat;
347 uint32_t offset_to_root_dir;
349 int current_fd;
350 mapping_t* current_mapping;
351 unsigned char* cluster; /* points to current cluster */
352 unsigned char* cluster_buffer; /* points to a buffer to hold temp data */
353 unsigned int current_cluster;
355 /* write support */
356 char* qcow_filename;
357 BdrvChild* qcow;
358 void* fat2;
359 char* used_clusters;
360 array_t commits;
361 const char* path;
362 int downcase_short_names;
364 Error *migration_blocker;
365 } BDRVVVFATState;
367 /* take the sector position spos and convert it to Cylinder/Head/Sector position
368 * if the position is outside the specified geometry, fill maximum value for CHS
369 * and return 1 to signal overflow.
371 static int sector2CHS(mbr_chs_t *chs, int spos, int cyls, int heads, int secs)
373 int head,sector;
374 sector = spos % secs; spos /= secs;
375 head = spos % heads; spos /= heads;
376 if (spos >= cyls) {
377 /* Overflow,
378 it happens if 32bit sector positions are used, while CHS is only 24bit.
379 Windows/Dos is said to take 1023/255/63 as nonrepresentable CHS */
380 chs->head = 0xFF;
381 chs->sector = 0xFF;
382 chs->cylinder = 0xFF;
383 return 1;
385 chs->head = (uint8_t)head;
386 chs->sector = (uint8_t)( (sector+1) | ((spos>>8)<<6) );
387 chs->cylinder = (uint8_t)spos;
388 return 0;
391 static void init_mbr(BDRVVVFATState *s, int cyls, int heads, int secs)
393 /* TODO: if the files mbr.img and bootsect.img exist, use them */
394 mbr_t* real_mbr=(mbr_t*)s->first_sectors;
395 partition_t* partition = &(real_mbr->partition[0]);
396 int lba;
398 memset(s->first_sectors,0,512);
400 /* Win NT Disk Signature */
401 real_mbr->nt_id= cpu_to_le32(0xbe1afdfa);
403 partition->attributes=0x80; /* bootable */
405 /* LBA is used when partition is outside the CHS geometry */
406 lba = sector2CHS(&partition->start_CHS, s->offset_to_bootsector,
407 cyls, heads, secs);
408 lba |= sector2CHS(&partition->end_CHS, s->bs->total_sectors - 1,
409 cyls, heads, secs);
411 /*LBA partitions are identified only by start/length_sector_long not by CHS*/
412 partition->start_sector_long = cpu_to_le32(s->offset_to_bootsector);
413 partition->length_sector_long = cpu_to_le32(s->bs->total_sectors
414 - s->offset_to_bootsector);
416 /* FAT12/FAT16/FAT32 */
417 /* DOS uses different types when partition is LBA,
418 probably to prevent older versions from using CHS on them */
419 partition->fs_type = s->fat_type == 12 ? 0x1 :
420 s->fat_type == 16 ? (lba ? 0xe : 0x06) :
421 /*s->fat_type == 32*/ (lba ? 0xc : 0x0b);
423 real_mbr->magic[0]=0x55; real_mbr->magic[1]=0xaa;
426 /* direntry functions */
428 static direntry_t *create_long_filename(BDRVVVFATState *s, const char *filename)
430 int number_of_entries, i;
431 glong length;
432 direntry_t *entry;
434 gunichar2 *longname = g_utf8_to_utf16(filename, -1, NULL, &length, NULL);
435 if (!longname) {
436 fprintf(stderr, "vvfat: invalid UTF-8 name: %s\n", filename);
437 return NULL;
440 number_of_entries = (length * 2 + 25) / 26;
442 for(i=0;i<number_of_entries;i++) {
443 entry=array_get_next(&(s->directory));
444 entry->attributes=0xf;
445 entry->reserved[0]=0;
446 entry->begin=0;
447 entry->name[0]=(number_of_entries-i)|(i==0?0x40:0);
449 for(i=0;i<26*number_of_entries;i++) {
450 int offset=(i%26);
451 if(offset<10) offset=1+offset;
452 else if(offset<22) offset=14+offset-10;
453 else offset=28+offset-22;
454 entry=array_get(&(s->directory),s->directory.next-1-(i/26));
455 if (i >= 2 * length + 2) {
456 entry->name[offset] = 0xff;
457 } else if (i % 2 == 0) {
458 entry->name[offset] = longname[i / 2] & 0xff;
459 } else {
460 entry->name[offset] = longname[i / 2] >> 8;
463 g_free(longname);
464 return array_get(&(s->directory),s->directory.next-number_of_entries);
467 static char is_free(const direntry_t* direntry)
469 return direntry->name[0]==0xe5 || direntry->name[0]==0x00;
472 static char is_volume_label(const direntry_t* direntry)
474 return direntry->attributes == 0x28;
477 static char is_long_name(const direntry_t* direntry)
479 return direntry->attributes == 0xf;
482 static char is_short_name(const direntry_t* direntry)
484 return !is_volume_label(direntry) && !is_long_name(direntry)
485 && !is_free(direntry);
488 static char is_directory(const direntry_t* direntry)
490 return direntry->attributes & 0x10 && direntry->name[0] != 0xe5;
493 static inline char is_dot(const direntry_t* direntry)
495 return is_short_name(direntry) && direntry->name[0] == '.';
498 static char is_file(const direntry_t* direntry)
500 return is_short_name(direntry) && !is_directory(direntry);
503 static inline uint32_t begin_of_direntry(const direntry_t* direntry)
505 return le16_to_cpu(direntry->begin)|(le16_to_cpu(direntry->begin_hi)<<16);
508 static inline uint32_t filesize_of_direntry(const direntry_t* direntry)
510 return le32_to_cpu(direntry->size);
513 static void set_begin_of_direntry(direntry_t* direntry, uint32_t begin)
515 direntry->begin = cpu_to_le16(begin & 0xffff);
516 direntry->begin_hi = cpu_to_le16((begin >> 16) & 0xffff);
519 static uint8_t to_valid_short_char(gunichar c)
521 c = g_unichar_toupper(c);
522 if ((c >= '0' && c <= '9') ||
523 (c >= 'A' && c <= 'Z') ||
524 strchr("$%'-_@~`!(){}^#&", c) != 0) {
525 return c;
526 } else {
527 return 0;
531 static direntry_t *create_short_filename(BDRVVVFATState *s,
532 const char *filename,
533 unsigned int directory_start)
535 int i, j = 0;
536 direntry_t *entry = array_get_next(&(s->directory));
537 const gchar *p, *last_dot = NULL;
538 gunichar c;
539 bool lossy_conversion = false;
540 char tail[11];
542 if (!entry) {
543 return NULL;
545 memset(entry->name, 0x20, sizeof(entry->name));
547 /* copy filename and search last dot */
548 for (p = filename; ; p = g_utf8_next_char(p)) {
549 c = g_utf8_get_char(p);
550 if (c == '\0') {
551 break;
552 } else if (c == '.') {
553 if (j == 0) {
554 /* '.' at start of filename */
555 lossy_conversion = true;
556 } else {
557 if (last_dot) {
558 lossy_conversion = true;
560 last_dot = p;
562 } else if (!last_dot) {
563 /* first part of the name; copy it */
564 uint8_t v = to_valid_short_char(c);
565 if (j < 8 && v) {
566 entry->name[j++] = v;
567 } else {
568 lossy_conversion = true;
573 /* copy extension (if any) */
574 if (last_dot) {
575 j = 0;
576 for (p = g_utf8_next_char(last_dot); ; p = g_utf8_next_char(p)) {
577 c = g_utf8_get_char(p);
578 if (c == '\0') {
579 break;
580 } else {
581 /* extension; copy it */
582 uint8_t v = to_valid_short_char(c);
583 if (j < 3 && v) {
584 entry->name[8 + (j++)] = v;
585 } else {
586 lossy_conversion = true;
592 if (entry->name[0] == 0xe5) {
593 entry->name[0] = 0x05;
596 /* numeric-tail generation */
597 for (j = 0; j < 8; j++) {
598 if (entry->name[j] == ' ') {
599 break;
602 for (i = lossy_conversion ? 1 : 0; i < 999999; i++) {
603 direntry_t *entry1;
604 if (i > 0) {
605 int len = sprintf(tail, "~%d", i);
606 memcpy(entry->name + MIN(j, 8 - len), tail, len);
608 for (entry1 = array_get(&(s->directory), directory_start);
609 entry1 < entry; entry1++) {
610 if (!is_long_name(entry1) &&
611 !memcmp(entry1->name, entry->name, 11)) {
612 break; /* found dupe */
615 if (entry1 == entry) {
616 /* no dupe found */
617 return entry;
620 return NULL;
623 /* fat functions */
625 static inline uint8_t fat_chksum(const direntry_t* entry)
627 uint8_t chksum=0;
628 int i;
630 for (i = 0; i < ARRAY_SIZE(entry->name); i++) {
631 chksum = (((chksum & 0xfe) >> 1) |
632 ((chksum & 0x01) ? 0x80 : 0)) + entry->name[i];
635 return chksum;
638 /* if return_time==0, this returns the fat_date, else the fat_time */
639 static uint16_t fat_datetime(time_t time,int return_time) {
640 struct tm* t;
641 struct tm t1;
642 t = &t1;
643 localtime_r(&time,t);
644 if(return_time)
645 return cpu_to_le16((t->tm_sec/2)|(t->tm_min<<5)|(t->tm_hour<<11));
646 return cpu_to_le16((t->tm_mday)|((t->tm_mon+1)<<5)|((t->tm_year-80)<<9));
649 static inline void fat_set(BDRVVVFATState* s,unsigned int cluster,uint32_t value)
651 if(s->fat_type==32) {
652 uint32_t* entry=array_get(&(s->fat),cluster);
653 *entry=cpu_to_le32(value);
654 } else if(s->fat_type==16) {
655 uint16_t* entry=array_get(&(s->fat),cluster);
656 *entry=cpu_to_le16(value&0xffff);
657 } else {
658 int offset = (cluster*3/2);
659 unsigned char* p = array_get(&(s->fat), offset);
660 switch (cluster&1) {
661 case 0:
662 p[0] = value&0xff;
663 p[1] = (p[1]&0xf0) | ((value>>8)&0xf);
664 break;
665 case 1:
666 p[0] = (p[0]&0xf) | ((value&0xf)<<4);
667 p[1] = (value>>4);
668 break;
673 static inline uint32_t fat_get(BDRVVVFATState* s,unsigned int cluster)
675 if(s->fat_type==32) {
676 uint32_t* entry=array_get(&(s->fat),cluster);
677 return le32_to_cpu(*entry);
678 } else if(s->fat_type==16) {
679 uint16_t* entry=array_get(&(s->fat),cluster);
680 return le16_to_cpu(*entry);
681 } else {
682 const uint8_t* x=(uint8_t*)(s->fat.pointer)+cluster*3/2;
683 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
687 static inline int fat_eof(BDRVVVFATState* s,uint32_t fat_entry)
689 if(fat_entry>s->max_fat_value-8)
690 return -1;
691 return 0;
694 static inline void init_fat(BDRVVVFATState* s)
696 if (s->fat_type == 12) {
697 array_init(&(s->fat),1);
698 array_ensure_allocated(&(s->fat),
699 s->sectors_per_fat * 0x200 * 3 / 2 - 1);
700 } else {
701 array_init(&(s->fat),(s->fat_type==32?4:2));
702 array_ensure_allocated(&(s->fat),
703 s->sectors_per_fat * 0x200 / s->fat.item_size - 1);
705 memset(s->fat.pointer,0,s->fat.size);
707 switch(s->fat_type) {
708 case 12: s->max_fat_value=0xfff; break;
709 case 16: s->max_fat_value=0xffff; break;
710 case 32: s->max_fat_value=0x0fffffff; break;
711 default: s->max_fat_value=0; /* error... */
716 static inline direntry_t* create_short_and_long_name(BDRVVVFATState* s,
717 unsigned int directory_start, const char* filename, int is_dot)
719 int long_index = s->directory.next;
720 direntry_t* entry = NULL;
721 direntry_t* entry_long = NULL;
723 if(is_dot) {
724 entry=array_get_next(&(s->directory));
725 memset(entry->name, 0x20, sizeof(entry->name));
726 memcpy(entry->name,filename,strlen(filename));
727 return entry;
730 entry_long=create_long_filename(s,filename);
731 entry = create_short_filename(s, filename, directory_start);
733 /* calculate checksum; propagate to long name */
734 if(entry_long) {
735 uint8_t chksum=fat_chksum(entry);
737 /* calculate anew, because realloc could have taken place */
738 entry_long=array_get(&(s->directory),long_index);
739 while(entry_long<entry && is_long_name(entry_long)) {
740 entry_long->reserved[1]=chksum;
741 entry_long++;
745 return entry;
749 * Read a directory. (the index of the corresponding mapping must be passed).
751 static int read_directory(BDRVVVFATState* s, int mapping_index)
753 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
754 direntry_t* direntry;
755 const char* dirname = mapping->path;
756 int first_cluster = mapping->begin;
757 int parent_index = mapping->info.dir.parent_mapping_index;
758 mapping_t* parent_mapping = (mapping_t*)
759 (parent_index >= 0 ? array_get(&(s->mapping), parent_index) : NULL);
760 int first_cluster_of_parent = parent_mapping ? parent_mapping->begin : -1;
762 DIR* dir=opendir(dirname);
763 struct dirent* entry;
764 int i;
766 assert(mapping->mode & MODE_DIRECTORY);
768 if(!dir) {
769 mapping->end = mapping->begin;
770 return -1;
773 i = mapping->info.dir.first_dir_index =
774 first_cluster == 0 ? 0 : s->directory.next;
776 if (first_cluster != 0) {
777 /* create the top entries of a subdirectory */
778 (void)create_short_and_long_name(s, i, ".", 1);
779 (void)create_short_and_long_name(s, i, "..", 1);
782 /* actually read the directory, and allocate the mappings */
783 while((entry=readdir(dir))) {
784 unsigned int length=strlen(dirname)+2+strlen(entry->d_name);
785 char* buffer;
786 direntry_t* direntry;
787 struct stat st;
788 int is_dot=!strcmp(entry->d_name,".");
789 int is_dotdot=!strcmp(entry->d_name,"..");
791 if (first_cluster == 0 && s->directory.next >= s->root_entries - 1) {
792 fprintf(stderr, "Too many entries in root directory\n");
793 closedir(dir);
794 return -2;
797 if(first_cluster == 0 && (is_dotdot || is_dot))
798 continue;
800 buffer = g_malloc(length);
801 snprintf(buffer,length,"%s/%s",dirname,entry->d_name);
803 if(stat(buffer,&st)<0) {
804 g_free(buffer);
805 continue;
808 /* create directory entry for this file */
809 if (!is_dot && !is_dotdot) {
810 direntry = create_short_and_long_name(s, i, entry->d_name, 0);
811 } else {
812 direntry = array_get(&(s->directory), is_dot ? i : i + 1);
814 direntry->attributes=(S_ISDIR(st.st_mode)?0x10:0x20);
815 direntry->reserved[0]=direntry->reserved[1]=0;
816 direntry->ctime=fat_datetime(st.st_ctime,1);
817 direntry->cdate=fat_datetime(st.st_ctime,0);
818 direntry->adate=fat_datetime(st.st_atime,0);
819 direntry->begin_hi=0;
820 direntry->mtime=fat_datetime(st.st_mtime,1);
821 direntry->mdate=fat_datetime(st.st_mtime,0);
822 if(is_dotdot)
823 set_begin_of_direntry(direntry, first_cluster_of_parent);
824 else if(is_dot)
825 set_begin_of_direntry(direntry, first_cluster);
826 else
827 direntry->begin=0; /* do that later */
828 if (st.st_size > 0x7fffffff) {
829 fprintf(stderr, "File %s is larger than 2GB\n", buffer);
830 g_free(buffer);
831 closedir(dir);
832 return -2;
834 direntry->size=cpu_to_le32(S_ISDIR(st.st_mode)?0:st.st_size);
836 /* create mapping for this file */
837 if(!is_dot && !is_dotdot && (S_ISDIR(st.st_mode) || st.st_size)) {
838 s->current_mapping = array_get_next(&(s->mapping));
839 s->current_mapping->begin=0;
840 s->current_mapping->end=st.st_size;
842 * we get the direntry of the most recent direntry, which
843 * contains the short name and all the relevant information.
845 s->current_mapping->dir_index=s->directory.next-1;
846 s->current_mapping->first_mapping_index = -1;
847 if (S_ISDIR(st.st_mode)) {
848 s->current_mapping->mode = MODE_DIRECTORY;
849 s->current_mapping->info.dir.parent_mapping_index =
850 mapping_index;
851 } else {
852 s->current_mapping->mode = MODE_UNDEFINED;
853 s->current_mapping->info.file.offset = 0;
855 s->current_mapping->path=buffer;
856 s->current_mapping->read_only =
857 (st.st_mode & (S_IWUSR | S_IWGRP | S_IWOTH)) == 0;
858 } else {
859 g_free(buffer);
862 closedir(dir);
864 /* fill with zeroes up to the end of the cluster */
865 while(s->directory.next%(0x10*s->sectors_per_cluster)) {
866 direntry_t* direntry=array_get_next(&(s->directory));
867 memset(direntry,0,sizeof(direntry_t));
870 if (s->fat_type != 32 &&
871 mapping_index == 0 &&
872 s->directory.next < s->root_entries) {
873 /* root directory */
874 int cur = s->directory.next;
875 array_ensure_allocated(&(s->directory), s->root_entries - 1);
876 s->directory.next = s->root_entries;
877 memset(array_get(&(s->directory), cur), 0,
878 (s->root_entries - cur) * sizeof(direntry_t));
881 /* re-get the mapping, since s->mapping was possibly realloc()ed */
882 mapping = array_get(&(s->mapping), mapping_index);
883 first_cluster += (s->directory.next - mapping->info.dir.first_dir_index)
884 * 0x20 / s->cluster_size;
885 mapping->end = first_cluster;
887 direntry = array_get(&(s->directory), mapping->dir_index);
888 set_begin_of_direntry(direntry, mapping->begin);
890 return 0;
893 static inline uint32_t sector2cluster(BDRVVVFATState* s,off_t sector_num)
895 return (sector_num - s->offset_to_root_dir) / s->sectors_per_cluster;
898 static inline off_t cluster2sector(BDRVVVFATState* s, uint32_t cluster_num)
900 return s->offset_to_root_dir + s->sectors_per_cluster * cluster_num;
903 static int init_directories(BDRVVVFATState* s,
904 const char *dirname, int heads, int secs,
905 Error **errp)
907 bootsector_t* bootsector;
908 mapping_t* mapping;
909 unsigned int i;
910 unsigned int cluster;
912 memset(&(s->first_sectors[0]),0,0x40*0x200);
914 s->cluster_size=s->sectors_per_cluster*0x200;
915 s->cluster_buffer=g_malloc(s->cluster_size);
918 * The formula: sc = spf+1+spf*spc*(512*8/fat_type),
919 * where sc is sector_count,
920 * spf is sectors_per_fat,
921 * spc is sectors_per_clusters, and
922 * fat_type = 12, 16 or 32.
924 i = 1+s->sectors_per_cluster*0x200*8/s->fat_type;
925 s->sectors_per_fat=(s->sector_count+i)/i; /* round up */
927 s->offset_to_fat = s->offset_to_bootsector + 1;
928 s->offset_to_root_dir = s->offset_to_fat + s->sectors_per_fat * 2;
930 array_init(&(s->mapping),sizeof(mapping_t));
931 array_init(&(s->directory),sizeof(direntry_t));
933 /* add volume label */
935 direntry_t* entry=array_get_next(&(s->directory));
936 entry->attributes=0x28; /* archive | volume label */
937 memcpy(entry->name, s->volume_label, sizeof(entry->name));
940 /* Now build FAT, and write back information into directory */
941 init_fat(s);
943 /* TODO: if there are more entries, bootsector has to be adjusted! */
944 s->root_entries = 0x02 * 0x10 * s->sectors_per_cluster;
945 s->cluster_count=sector2cluster(s, s->sector_count);
947 mapping = array_get_next(&(s->mapping));
948 mapping->begin = 0;
949 mapping->dir_index = 0;
950 mapping->info.dir.parent_mapping_index = -1;
951 mapping->first_mapping_index = -1;
952 mapping->path = g_strdup(dirname);
953 i = strlen(mapping->path);
954 if (i > 0 && mapping->path[i - 1] == '/')
955 mapping->path[i - 1] = '\0';
956 mapping->mode = MODE_DIRECTORY;
957 mapping->read_only = 0;
958 s->path = mapping->path;
960 for (i = 0, cluster = 0; i < s->mapping.next; i++) {
961 /* MS-DOS expects the FAT to be 0 for the root directory
962 * (except for the media byte). */
963 /* LATER TODO: still true for FAT32? */
964 int fix_fat = (i != 0);
965 mapping = array_get(&(s->mapping), i);
967 if (mapping->mode & MODE_DIRECTORY) {
968 mapping->begin = cluster;
969 if(read_directory(s, i)) {
970 error_setg(errp, "Could not read directory %s",
971 mapping->path);
972 return -1;
974 mapping = array_get(&(s->mapping), i);
975 } else {
976 assert(mapping->mode == MODE_UNDEFINED);
977 mapping->mode=MODE_NORMAL;
978 mapping->begin = cluster;
979 if (mapping->end > 0) {
980 direntry_t* direntry = array_get(&(s->directory),
981 mapping->dir_index);
983 mapping->end = cluster + 1 + (mapping->end-1)/s->cluster_size;
984 set_begin_of_direntry(direntry, mapping->begin);
985 } else {
986 mapping->end = cluster + 1;
987 fix_fat = 0;
991 assert(mapping->begin < mapping->end);
993 /* next free cluster */
994 cluster = mapping->end;
996 if(cluster > s->cluster_count) {
997 error_setg(errp,
998 "Directory does not fit in FAT%d (capacity %.2f MB)",
999 s->fat_type, s->sector_count / 2000.0);
1000 return -1;
1003 /* fix fat for entry */
1004 if (fix_fat) {
1005 int j;
1006 for(j = mapping->begin; j < mapping->end - 1; j++)
1007 fat_set(s, j, j+1);
1008 fat_set(s, mapping->end - 1, s->max_fat_value);
1012 mapping = array_get(&(s->mapping), 0);
1013 s->last_cluster_of_root_directory = mapping->end;
1015 /* the FAT signature */
1016 fat_set(s,0,s->max_fat_value);
1017 fat_set(s,1,s->max_fat_value);
1019 s->current_mapping = NULL;
1021 bootsector = (bootsector_t *)(s->first_sectors
1022 + s->offset_to_bootsector * 0x200);
1023 bootsector->jump[0]=0xeb;
1024 bootsector->jump[1]=0x3e;
1025 bootsector->jump[2]=0x90;
1026 memcpy(bootsector->name, "MSWIN4.1", 8);
1027 bootsector->sector_size=cpu_to_le16(0x200);
1028 bootsector->sectors_per_cluster=s->sectors_per_cluster;
1029 bootsector->reserved_sectors=cpu_to_le16(1);
1030 bootsector->number_of_fats=0x2; /* number of FATs */
1031 bootsector->root_entries = cpu_to_le16(s->root_entries);
1032 bootsector->total_sectors16=s->sector_count>0xffff?0:cpu_to_le16(s->sector_count);
1033 /* media descriptor: hard disk=0xf8, floppy=0xf0 */
1034 bootsector->media_type = (s->offset_to_bootsector > 0 ? 0xf8 : 0xf0);
1035 s->fat.pointer[0] = bootsector->media_type;
1036 bootsector->sectors_per_fat=cpu_to_le16(s->sectors_per_fat);
1037 bootsector->sectors_per_track = cpu_to_le16(secs);
1038 bootsector->number_of_heads = cpu_to_le16(heads);
1039 bootsector->hidden_sectors = cpu_to_le32(s->offset_to_bootsector);
1040 bootsector->total_sectors=cpu_to_le32(s->sector_count>0xffff?s->sector_count:0);
1042 /* LATER TODO: if FAT32, this is wrong */
1043 /* drive_number: fda=0, hda=0x80 */
1044 bootsector->u.fat16.drive_number = s->offset_to_bootsector == 0 ? 0 : 0x80;
1045 bootsector->u.fat16.signature=0x29;
1046 bootsector->u.fat16.id=cpu_to_le32(0xfabe1afd);
1048 memcpy(bootsector->u.fat16.volume_label, s->volume_label,
1049 sizeof(bootsector->u.fat16.volume_label));
1050 memcpy(bootsector->u.fat16.fat_type,
1051 s->fat_type == 12 ? "FAT12 " : "FAT16 ", 8);
1052 bootsector->magic[0]=0x55; bootsector->magic[1]=0xaa;
1054 return 0;
1057 #ifdef DEBUG
1058 static BDRVVVFATState *vvv = NULL;
1059 #endif
1061 static int enable_write_target(BlockDriverState *bs, Error **errp);
1062 static int is_consistent(BDRVVVFATState *s);
1064 static QemuOptsList runtime_opts = {
1065 .name = "vvfat",
1066 .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
1067 .desc = {
1069 .name = "dir",
1070 .type = QEMU_OPT_STRING,
1071 .help = "Host directory to map to the vvfat device",
1074 .name = "fat-type",
1075 .type = QEMU_OPT_NUMBER,
1076 .help = "FAT type (12, 16 or 32)",
1079 .name = "floppy",
1080 .type = QEMU_OPT_BOOL,
1081 .help = "Create a floppy rather than a hard disk image",
1084 .name = "label",
1085 .type = QEMU_OPT_STRING,
1086 .help = "Use a volume label other than QEMU VVFAT",
1089 .name = "rw",
1090 .type = QEMU_OPT_BOOL,
1091 .help = "Make the image writable",
1093 { /* end of list */ }
1097 static void vvfat_parse_filename(const char *filename, QDict *options,
1098 Error **errp)
1100 int fat_type = 0;
1101 bool floppy = false;
1102 bool rw = false;
1103 int i;
1105 if (!strstart(filename, "fat:", NULL)) {
1106 error_setg(errp, "File name string must start with 'fat:'");
1107 return;
1110 /* Parse options */
1111 if (strstr(filename, ":32:")) {
1112 fat_type = 32;
1113 } else if (strstr(filename, ":16:")) {
1114 fat_type = 16;
1115 } else if (strstr(filename, ":12:")) {
1116 fat_type = 12;
1119 if (strstr(filename, ":floppy:")) {
1120 floppy = true;
1123 if (strstr(filename, ":rw:")) {
1124 rw = true;
1127 /* Get the directory name without options */
1128 i = strrchr(filename, ':') - filename;
1129 assert(i >= 3);
1130 if (filename[i - 2] == ':' && qemu_isalpha(filename[i - 1])) {
1131 /* workaround for DOS drive names */
1132 filename += i - 1;
1133 } else {
1134 filename += i + 1;
1137 /* Fill in the options QDict */
1138 qdict_put_str(options, "dir", filename);
1139 qdict_put_int(options, "fat-type", fat_type);
1140 qdict_put_bool(options, "floppy", floppy);
1141 qdict_put_bool(options, "rw", rw);
1144 static int vvfat_open(BlockDriverState *bs, QDict *options, int flags,
1145 Error **errp)
1147 BDRVVVFATState *s = bs->opaque;
1148 int cyls, heads, secs;
1149 bool floppy;
1150 const char *dirname, *label;
1151 QemuOpts *opts;
1152 Error *local_err = NULL;
1153 int ret;
1155 #ifdef DEBUG
1156 vvv = s;
1157 #endif
1159 opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
1160 qemu_opts_absorb_qdict(opts, options, &local_err);
1161 if (local_err) {
1162 error_propagate(errp, local_err);
1163 ret = -EINVAL;
1164 goto fail;
1167 dirname = qemu_opt_get(opts, "dir");
1168 if (!dirname) {
1169 error_setg(errp, "vvfat block driver requires a 'dir' option");
1170 ret = -EINVAL;
1171 goto fail;
1174 s->fat_type = qemu_opt_get_number(opts, "fat-type", 0);
1175 floppy = qemu_opt_get_bool(opts, "floppy", false);
1177 memset(s->volume_label, ' ', sizeof(s->volume_label));
1178 label = qemu_opt_get(opts, "label");
1179 if (label) {
1180 size_t label_length = strlen(label);
1181 if (label_length > 11) {
1182 error_setg(errp, "vvfat label cannot be longer than 11 bytes");
1183 ret = -EINVAL;
1184 goto fail;
1186 memcpy(s->volume_label, label, label_length);
1187 } else {
1188 memcpy(s->volume_label, "QEMU VVFAT", 10);
1191 if (floppy) {
1192 /* 1.44MB or 2.88MB floppy. 2.88MB can be FAT12 (default) or FAT16. */
1193 if (!s->fat_type) {
1194 s->fat_type = 12;
1195 secs = 36;
1196 s->sectors_per_cluster = 2;
1197 } else {
1198 secs = s->fat_type == 12 ? 18 : 36;
1199 s->sectors_per_cluster = 1;
1201 cyls = 80;
1202 heads = 2;
1203 } else {
1204 /* 32MB or 504MB disk*/
1205 if (!s->fat_type) {
1206 s->fat_type = 16;
1208 s->offset_to_bootsector = 0x3f;
1209 cyls = s->fat_type == 12 ? 64 : 1024;
1210 heads = 16;
1211 secs = 63;
1214 switch (s->fat_type) {
1215 case 32:
1216 fprintf(stderr, "Big fat greek warning: FAT32 has not been tested. "
1217 "You are welcome to do so!\n");
1218 break;
1219 case 16:
1220 case 12:
1221 break;
1222 default:
1223 error_setg(errp, "Valid FAT types are only 12, 16 and 32");
1224 ret = -EINVAL;
1225 goto fail;
1229 s->bs = bs;
1231 /* LATER TODO: if FAT32, adjust */
1232 s->sectors_per_cluster=0x10;
1234 s->current_cluster=0xffffffff;
1236 s->qcow = NULL;
1237 s->qcow_filename = NULL;
1238 s->fat2 = NULL;
1239 s->downcase_short_names = 1;
1241 fprintf(stderr, "vvfat %s chs %d,%d,%d\n",
1242 dirname, cyls, heads, secs);
1244 s->sector_count = cyls * heads * secs - s->offset_to_bootsector;
1246 if (qemu_opt_get_bool(opts, "rw", false)) {
1247 if (!bdrv_is_read_only(bs)) {
1248 ret = enable_write_target(bs, errp);
1249 if (ret < 0) {
1250 goto fail;
1252 } else {
1253 ret = -EPERM;
1254 error_setg(errp,
1255 "Unable to set VVFAT to 'rw' when drive is read-only");
1256 goto fail;
1258 } else {
1259 /* read only is the default for safety */
1260 ret = bdrv_set_read_only(bs, true, &local_err);
1261 if (ret < 0) {
1262 error_propagate(errp, local_err);
1263 goto fail;
1267 bs->total_sectors = cyls * heads * secs;
1269 if (init_directories(s, dirname, heads, secs, errp)) {
1270 ret = -EIO;
1271 goto fail;
1274 s->sector_count = s->offset_to_root_dir
1275 + s->sectors_per_cluster * s->cluster_count;
1277 /* Disable migration when vvfat is used rw */
1278 if (s->qcow) {
1279 error_setg(&s->migration_blocker,
1280 "The vvfat (rw) format used by node '%s' "
1281 "does not support live migration",
1282 bdrv_get_device_or_node_name(bs));
1283 ret = migrate_add_blocker(s->migration_blocker, &local_err);
1284 if (local_err) {
1285 error_propagate(errp, local_err);
1286 error_free(s->migration_blocker);
1287 goto fail;
1291 if (s->offset_to_bootsector > 0) {
1292 init_mbr(s, cyls, heads, secs);
1295 qemu_co_mutex_init(&s->lock);
1297 ret = 0;
1298 fail:
1299 qemu_opts_del(opts);
1300 return ret;
1303 static void vvfat_refresh_limits(BlockDriverState *bs, Error **errp)
1305 bs->bl.request_alignment = BDRV_SECTOR_SIZE; /* No sub-sector I/O */
1308 static inline void vvfat_close_current_file(BDRVVVFATState *s)
1310 if(s->current_mapping) {
1311 s->current_mapping = NULL;
1312 if (s->current_fd) {
1313 qemu_close(s->current_fd);
1314 s->current_fd = 0;
1317 s->current_cluster = -1;
1320 /* mappings between index1 and index2-1 are supposed to be ordered
1321 * return value is the index of the last mapping for which end>cluster_num
1323 static inline int find_mapping_for_cluster_aux(BDRVVVFATState* s,int cluster_num,int index1,int index2)
1325 while(1) {
1326 int index3;
1327 mapping_t* mapping;
1328 index3=(index1+index2)/2;
1329 mapping=array_get(&(s->mapping),index3);
1330 assert(mapping->begin < mapping->end);
1331 if(mapping->begin>=cluster_num) {
1332 assert(index2!=index3 || index2==0);
1333 if(index2==index3)
1334 return index1;
1335 index2=index3;
1336 } else {
1337 if(index1==index3)
1338 return mapping->end<=cluster_num ? index2 : index1;
1339 index1=index3;
1341 assert(index1<=index2);
1342 DLOG(mapping=array_get(&(s->mapping),index1);
1343 assert(mapping->begin<=cluster_num);
1344 assert(index2 >= s->mapping.next ||
1345 ((mapping = array_get(&(s->mapping),index2)) &&
1346 mapping->end>cluster_num)));
1350 static inline mapping_t* find_mapping_for_cluster(BDRVVVFATState* s,int cluster_num)
1352 int index=find_mapping_for_cluster_aux(s,cluster_num,0,s->mapping.next);
1353 mapping_t* mapping;
1354 if(index>=s->mapping.next)
1355 return NULL;
1356 mapping=array_get(&(s->mapping),index);
1357 if(mapping->begin>cluster_num)
1358 return NULL;
1359 assert(mapping->begin<=cluster_num && mapping->end>cluster_num);
1360 return mapping;
1363 static int open_file(BDRVVVFATState* s,mapping_t* mapping)
1365 if(!mapping)
1366 return -1;
1367 if(!s->current_mapping ||
1368 strcmp(s->current_mapping->path,mapping->path)) {
1369 /* open file */
1370 int fd = qemu_open(mapping->path, O_RDONLY | O_BINARY | O_LARGEFILE);
1371 if(fd<0)
1372 return -1;
1373 vvfat_close_current_file(s);
1374 s->current_fd = fd;
1375 s->current_mapping = mapping;
1377 return 0;
1380 static inline int read_cluster(BDRVVVFATState *s,int cluster_num)
1382 if(s->current_cluster != cluster_num) {
1383 int result=0;
1384 off_t offset;
1385 assert(!s->current_mapping || s->current_fd || (s->current_mapping->mode & MODE_DIRECTORY));
1386 if(!s->current_mapping
1387 || s->current_mapping->begin>cluster_num
1388 || s->current_mapping->end<=cluster_num) {
1389 /* binary search of mappings for file */
1390 mapping_t* mapping=find_mapping_for_cluster(s,cluster_num);
1392 assert(!mapping || (cluster_num>=mapping->begin && cluster_num<mapping->end));
1394 if (mapping && mapping->mode & MODE_DIRECTORY) {
1395 vvfat_close_current_file(s);
1396 s->current_mapping = mapping;
1397 read_cluster_directory:
1398 offset = s->cluster_size*(cluster_num-s->current_mapping->begin);
1399 s->cluster = (unsigned char*)s->directory.pointer+offset
1400 + 0x20*s->current_mapping->info.dir.first_dir_index;
1401 assert(((s->cluster-(unsigned char*)s->directory.pointer)%s->cluster_size)==0);
1402 assert((char*)s->cluster+s->cluster_size <= s->directory.pointer+s->directory.next*s->directory.item_size);
1403 s->current_cluster = cluster_num;
1404 return 0;
1407 if(open_file(s,mapping))
1408 return -2;
1409 } else if (s->current_mapping->mode & MODE_DIRECTORY)
1410 goto read_cluster_directory;
1412 assert(s->current_fd);
1414 offset=s->cluster_size*(cluster_num-s->current_mapping->begin)+s->current_mapping->info.file.offset;
1415 if(lseek(s->current_fd, offset, SEEK_SET)!=offset)
1416 return -3;
1417 s->cluster=s->cluster_buffer;
1418 result=read(s->current_fd,s->cluster,s->cluster_size);
1419 if(result<0) {
1420 s->current_cluster = -1;
1421 return -1;
1423 s->current_cluster = cluster_num;
1425 return 0;
1428 #ifdef DEBUG
1429 static void print_direntry(const direntry_t* direntry)
1431 int j = 0;
1432 char buffer[1024];
1434 fprintf(stderr, "direntry %p: ", direntry);
1435 if(!direntry)
1436 return;
1437 if(is_long_name(direntry)) {
1438 unsigned char* c=(unsigned char*)direntry;
1439 int i;
1440 for(i=1;i<11 && c[i] && c[i]!=0xff;i+=2)
1441 #define ADD_CHAR(c) {buffer[j] = (c); if (buffer[j] < ' ') buffer[j] = 0xb0; j++;}
1442 ADD_CHAR(c[i]);
1443 for(i=14;i<26 && c[i] && c[i]!=0xff;i+=2)
1444 ADD_CHAR(c[i]);
1445 for(i=28;i<32 && c[i] && c[i]!=0xff;i+=2)
1446 ADD_CHAR(c[i]);
1447 buffer[j] = 0;
1448 fprintf(stderr, "%s\n", buffer);
1449 } else {
1450 int i;
1451 for(i=0;i<11;i++)
1452 ADD_CHAR(direntry->name[i]);
1453 buffer[j] = 0;
1454 fprintf(stderr,"%s attributes=0x%02x begin=%d size=%d\n",
1455 buffer,
1456 direntry->attributes,
1457 begin_of_direntry(direntry),le32_to_cpu(direntry->size));
1461 static void print_mapping(const mapping_t* mapping)
1463 fprintf(stderr, "mapping (%p): begin, end = %d, %d, dir_index = %d, "
1464 "first_mapping_index = %d, name = %s, mode = 0x%x, " ,
1465 mapping, mapping->begin, mapping->end, mapping->dir_index,
1466 mapping->first_mapping_index, mapping->path, mapping->mode);
1468 if (mapping->mode & MODE_DIRECTORY)
1469 fprintf(stderr, "parent_mapping_index = %d, first_dir_index = %d\n", mapping->info.dir.parent_mapping_index, mapping->info.dir.first_dir_index);
1470 else
1471 fprintf(stderr, "offset = %d\n", mapping->info.file.offset);
1473 #endif
1475 static int vvfat_read(BlockDriverState *bs, int64_t sector_num,
1476 uint8_t *buf, int nb_sectors)
1478 BDRVVVFATState *s = bs->opaque;
1479 int i;
1481 for(i=0;i<nb_sectors;i++,sector_num++) {
1482 if (sector_num >= bs->total_sectors)
1483 return -1;
1484 if (s->qcow) {
1485 int n;
1486 int ret;
1487 ret = bdrv_is_allocated(s->qcow->bs, sector_num,
1488 nb_sectors - i, &n);
1489 if (ret < 0) {
1490 return ret;
1492 if (ret) {
1493 DLOG(fprintf(stderr, "sectors %d+%d allocated\n",
1494 (int)sector_num, n));
1495 if (bdrv_read(s->qcow, sector_num, buf + i * 0x200, n)) {
1496 return -1;
1498 i += n - 1;
1499 sector_num += n - 1;
1500 continue;
1502 DLOG(fprintf(stderr, "sector %d not allocated\n", (int)sector_num));
1504 if (sector_num < s->offset_to_root_dir) {
1505 if (sector_num < s->offset_to_fat) {
1506 memcpy(buf + i * 0x200,
1507 &(s->first_sectors[sector_num * 0x200]),
1508 0x200);
1509 } else if (sector_num < s->offset_to_fat + s->sectors_per_fat) {
1510 memcpy(buf + i * 0x200,
1511 &(s->fat.pointer[(sector_num
1512 - s->offset_to_fat) * 0x200]),
1513 0x200);
1514 } else if (sector_num < s->offset_to_root_dir) {
1515 memcpy(buf + i * 0x200,
1516 &(s->fat.pointer[(sector_num - s->offset_to_fat
1517 - s->sectors_per_fat) * 0x200]),
1518 0x200);
1520 } else {
1521 uint32_t sector = sector_num - s->offset_to_root_dir,
1522 sector_offset_in_cluster=(sector%s->sectors_per_cluster),
1523 cluster_num=sector/s->sectors_per_cluster;
1524 if(cluster_num > s->cluster_count || read_cluster(s, cluster_num) != 0) {
1525 /* LATER TODO: strict: return -1; */
1526 memset(buf+i*0x200,0,0x200);
1527 continue;
1529 memcpy(buf+i*0x200,s->cluster+sector_offset_in_cluster*0x200,0x200);
1532 return 0;
1535 static int coroutine_fn
1536 vvfat_co_preadv(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
1537 QEMUIOVector *qiov, int flags)
1539 int ret;
1540 BDRVVVFATState *s = bs->opaque;
1541 uint64_t sector_num = offset >> BDRV_SECTOR_BITS;
1542 int nb_sectors = bytes >> BDRV_SECTOR_BITS;
1543 void *buf;
1545 assert((offset & (BDRV_SECTOR_SIZE - 1)) == 0);
1546 assert((bytes & (BDRV_SECTOR_SIZE - 1)) == 0);
1548 buf = g_try_malloc(bytes);
1549 if (bytes && buf == NULL) {
1550 return -ENOMEM;
1553 qemu_co_mutex_lock(&s->lock);
1554 ret = vvfat_read(bs, sector_num, buf, nb_sectors);
1555 qemu_co_mutex_unlock(&s->lock);
1557 qemu_iovec_from_buf(qiov, 0, buf, bytes);
1558 g_free(buf);
1560 return ret;
1563 /* LATER TODO: statify all functions */
1566 * Idea of the write support (use snapshot):
1568 * 1. check if all data is consistent, recording renames, modifications,
1569 * new files and directories (in s->commits).
1571 * 2. if the data is not consistent, stop committing
1573 * 3. handle renames, and create new files and directories (do not yet
1574 * write their contents)
1576 * 4. walk the directories, fixing the mapping and direntries, and marking
1577 * the handled mappings as not deleted
1579 * 5. commit the contents of the files
1581 * 6. handle deleted files and directories
1585 typedef struct commit_t {
1586 char* path;
1587 union {
1588 struct { uint32_t cluster; } rename;
1589 struct { int dir_index; uint32_t modified_offset; } writeout;
1590 struct { uint32_t first_cluster; } new_file;
1591 struct { uint32_t cluster; } mkdir;
1592 } param;
1593 /* DELETEs and RMDIRs are handled differently: see handle_deletes() */
1594 enum {
1595 ACTION_RENAME, ACTION_WRITEOUT, ACTION_NEW_FILE, ACTION_MKDIR
1596 } action;
1597 } commit_t;
1599 static void clear_commits(BDRVVVFATState* s)
1601 int i;
1602 DLOG(fprintf(stderr, "clear_commits (%d commits)\n", s->commits.next));
1603 for (i = 0; i < s->commits.next; i++) {
1604 commit_t* commit = array_get(&(s->commits), i);
1605 assert(commit->path || commit->action == ACTION_WRITEOUT);
1606 if (commit->action != ACTION_WRITEOUT) {
1607 assert(commit->path);
1608 g_free(commit->path);
1609 } else
1610 assert(commit->path == NULL);
1612 s->commits.next = 0;
1615 static void schedule_rename(BDRVVVFATState* s,
1616 uint32_t cluster, char* new_path)
1618 commit_t* commit = array_get_next(&(s->commits));
1619 commit->path = new_path;
1620 commit->param.rename.cluster = cluster;
1621 commit->action = ACTION_RENAME;
1624 static void schedule_writeout(BDRVVVFATState* s,
1625 int dir_index, uint32_t modified_offset)
1627 commit_t* commit = array_get_next(&(s->commits));
1628 commit->path = NULL;
1629 commit->param.writeout.dir_index = dir_index;
1630 commit->param.writeout.modified_offset = modified_offset;
1631 commit->action = ACTION_WRITEOUT;
1634 static void schedule_new_file(BDRVVVFATState* s,
1635 char* path, uint32_t first_cluster)
1637 commit_t* commit = array_get_next(&(s->commits));
1638 commit->path = path;
1639 commit->param.new_file.first_cluster = first_cluster;
1640 commit->action = ACTION_NEW_FILE;
1643 static void schedule_mkdir(BDRVVVFATState* s, uint32_t cluster, char* path)
1645 commit_t* commit = array_get_next(&(s->commits));
1646 commit->path = path;
1647 commit->param.mkdir.cluster = cluster;
1648 commit->action = ACTION_MKDIR;
1651 typedef struct {
1653 * Since the sequence number is at most 0x3f, and the filename
1654 * length is at most 13 times the sequence number, the maximal
1655 * filename length is 0x3f * 13 bytes.
1657 unsigned char name[0x3f * 13 + 1];
1658 int checksum, len;
1659 int sequence_number;
1660 } long_file_name;
1662 static void lfn_init(long_file_name* lfn)
1664 lfn->sequence_number = lfn->len = 0;
1665 lfn->checksum = 0x100;
1668 /* return 0 if parsed successfully, > 0 if no long name, < 0 if error */
1669 static int parse_long_name(long_file_name* lfn,
1670 const direntry_t* direntry)
1672 int i, j, offset;
1673 const unsigned char* pointer = (const unsigned char*)direntry;
1675 if (!is_long_name(direntry))
1676 return 1;
1678 if (pointer[0] & 0x40) {
1679 lfn->sequence_number = pointer[0] & 0x3f;
1680 lfn->checksum = pointer[13];
1681 lfn->name[0] = 0;
1682 lfn->name[lfn->sequence_number * 13] = 0;
1683 } else if ((pointer[0] & 0x3f) != --lfn->sequence_number)
1684 return -1;
1685 else if (pointer[13] != lfn->checksum)
1686 return -2;
1687 else if (pointer[12] || pointer[26] || pointer[27])
1688 return -3;
1690 offset = 13 * (lfn->sequence_number - 1);
1691 for (i = 0, j = 1; i < 13; i++, j+=2) {
1692 if (j == 11)
1693 j = 14;
1694 else if (j == 26)
1695 j = 28;
1697 if (pointer[j+1] == 0)
1698 lfn->name[offset + i] = pointer[j];
1699 else if (pointer[j+1] != 0xff || (pointer[0] & 0x40) == 0)
1700 return -4;
1701 else
1702 lfn->name[offset + i] = 0;
1705 if (pointer[0] & 0x40)
1706 lfn->len = offset + strlen((char*)lfn->name + offset);
1708 return 0;
1711 /* returns 0 if successful, >0 if no short_name, and <0 on error */
1712 static int parse_short_name(BDRVVVFATState* s,
1713 long_file_name* lfn, direntry_t* direntry)
1715 int i, j;
1717 if (!is_short_name(direntry))
1718 return 1;
1720 for (j = 7; j >= 0 && direntry->name[j] == ' '; j--);
1721 for (i = 0; i <= j; i++) {
1722 if (direntry->name[i] <= ' ' || direntry->name[i] > 0x7f)
1723 return -1;
1724 else if (s->downcase_short_names)
1725 lfn->name[i] = qemu_tolower(direntry->name[i]);
1726 else
1727 lfn->name[i] = direntry->name[i];
1730 for (j = 2; j >= 0 && direntry->name[8 + j] == ' '; j--) {
1732 if (j >= 0) {
1733 lfn->name[i++] = '.';
1734 lfn->name[i + j + 1] = '\0';
1735 for (;j >= 0; j--) {
1736 uint8_t c = direntry->name[8 + j];
1737 if (c <= ' ' || c > 0x7f) {
1738 return -2;
1739 } else if (s->downcase_short_names) {
1740 lfn->name[i + j] = qemu_tolower(c);
1741 } else {
1742 lfn->name[i + j] = c;
1745 } else
1746 lfn->name[i + j + 1] = '\0';
1748 if (lfn->name[0] == 0x05) {
1749 lfn->name[0] = 0xe5;
1751 lfn->len = strlen((char*)lfn->name);
1753 return 0;
1756 static inline uint32_t modified_fat_get(BDRVVVFATState* s,
1757 unsigned int cluster)
1759 if (cluster < s->last_cluster_of_root_directory) {
1760 if (cluster + 1 == s->last_cluster_of_root_directory)
1761 return s->max_fat_value;
1762 else
1763 return cluster + 1;
1766 if (s->fat_type==32) {
1767 uint32_t* entry=((uint32_t*)s->fat2)+cluster;
1768 return le32_to_cpu(*entry);
1769 } else if (s->fat_type==16) {
1770 uint16_t* entry=((uint16_t*)s->fat2)+cluster;
1771 return le16_to_cpu(*entry);
1772 } else {
1773 const uint8_t* x=s->fat2+cluster*3/2;
1774 return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
1778 static inline bool cluster_was_modified(BDRVVVFATState *s,
1779 uint32_t cluster_num)
1781 int was_modified = 0;
1782 int i, dummy;
1784 if (s->qcow == NULL) {
1785 return 0;
1788 for (i = 0; !was_modified && i < s->sectors_per_cluster; i++) {
1789 was_modified = bdrv_is_allocated(s->qcow->bs,
1790 cluster2sector(s, cluster_num) + i,
1791 1, &dummy);
1795 * Note that this treats failures to learn allocation status the
1796 * same as if an allocation has occurred. It's as safe as
1797 * anything else, given that a failure to learn allocation status
1798 * will probably result in more failures.
1800 return !!was_modified;
1803 static const char* get_basename(const char* path)
1805 char* basename = strrchr(path, '/');
1806 if (basename == NULL)
1807 return path;
1808 else
1809 return basename + 1; /* strip '/' */
1813 * The array s->used_clusters holds the states of the clusters. If it is
1814 * part of a file, it has bit 2 set, in case of a directory, bit 1. If it
1815 * was modified, bit 3 is set.
1816 * If any cluster is allocated, but not part of a file or directory, this
1817 * driver refuses to commit.
1819 typedef enum {
1820 USED_DIRECTORY = 1, USED_FILE = 2, USED_ANY = 3, USED_ALLOCATED = 4
1821 } used_t;
1824 * get_cluster_count_for_direntry() not only determines how many clusters
1825 * are occupied by direntry, but also if it was renamed or modified.
1827 * A file is thought to be renamed *only* if there already was a file with
1828 * exactly the same first cluster, but a different name.
1830 * Further, the files/directories handled by this function are
1831 * assumed to be *not* deleted (and *only* those).
1833 static uint32_t get_cluster_count_for_direntry(BDRVVVFATState* s,
1834 direntry_t* direntry, const char* path)
1837 * This is a little bit tricky:
1838 * IF the guest OS just inserts a cluster into the file chain,
1839 * and leaves the rest alone, (i.e. the original file had clusters
1840 * 15 -> 16, but now has 15 -> 32 -> 16), then the following happens:
1842 * - do_commit will write the cluster into the file at the given
1843 * offset, but
1845 * - the cluster which is overwritten should be moved to a later
1846 * position in the file.
1848 * I am not aware that any OS does something as braindead, but this
1849 * situation could happen anyway when not committing for a long time.
1850 * Just to be sure that this does not bite us, detect it, and copy the
1851 * contents of the clusters to-be-overwritten into the qcow.
1853 int copy_it = 0;
1854 int was_modified = 0;
1855 int32_t ret = 0;
1857 uint32_t cluster_num = begin_of_direntry(direntry);
1858 uint32_t offset = 0;
1859 int first_mapping_index = -1;
1860 mapping_t* mapping = NULL;
1861 const char* basename2 = NULL;
1863 vvfat_close_current_file(s);
1865 /* the root directory */
1866 if (cluster_num == 0)
1867 return 0;
1869 /* write support */
1870 if (s->qcow) {
1871 basename2 = get_basename(path);
1873 mapping = find_mapping_for_cluster(s, cluster_num);
1875 if (mapping) {
1876 const char* basename;
1878 assert(mapping->mode & MODE_DELETED);
1879 mapping->mode &= ~MODE_DELETED;
1881 basename = get_basename(mapping->path);
1883 assert(mapping->mode & MODE_NORMAL);
1885 /* rename */
1886 if (strcmp(basename, basename2))
1887 schedule_rename(s, cluster_num, g_strdup(path));
1888 } else if (is_file(direntry))
1889 /* new file */
1890 schedule_new_file(s, g_strdup(path), cluster_num);
1891 else {
1892 abort();
1893 return 0;
1897 while(1) {
1898 if (s->qcow) {
1899 if (!copy_it && cluster_was_modified(s, cluster_num)) {
1900 if (mapping == NULL ||
1901 mapping->begin > cluster_num ||
1902 mapping->end <= cluster_num)
1903 mapping = find_mapping_for_cluster(s, cluster_num);
1906 if (mapping &&
1907 (mapping->mode & MODE_DIRECTORY) == 0) {
1909 /* was modified in qcow */
1910 if (offset != mapping->info.file.offset + s->cluster_size
1911 * (cluster_num - mapping->begin)) {
1912 /* offset of this cluster in file chain has changed */
1913 abort();
1914 copy_it = 1;
1915 } else if (offset == 0) {
1916 const char* basename = get_basename(mapping->path);
1918 if (strcmp(basename, basename2))
1919 copy_it = 1;
1920 first_mapping_index = array_index(&(s->mapping), mapping);
1923 if (mapping->first_mapping_index != first_mapping_index
1924 && mapping->info.file.offset > 0) {
1925 abort();
1926 copy_it = 1;
1929 /* need to write out? */
1930 if (!was_modified && is_file(direntry)) {
1931 was_modified = 1;
1932 schedule_writeout(s, mapping->dir_index, offset);
1937 if (copy_it) {
1938 int i, dummy;
1940 * This is horribly inefficient, but that is okay, since
1941 * it is rarely executed, if at all.
1943 int64_t offset = cluster2sector(s, cluster_num);
1945 vvfat_close_current_file(s);
1946 for (i = 0; i < s->sectors_per_cluster; i++) {
1947 int res;
1949 res = bdrv_is_allocated(s->qcow->bs, offset + i, 1, &dummy);
1950 if (res < 0) {
1951 return -1;
1953 if (!res) {
1954 res = vvfat_read(s->bs, offset, s->cluster_buffer, 1);
1955 if (res) {
1956 return -1;
1958 res = bdrv_write(s->qcow, offset, s->cluster_buffer, 1);
1959 if (res) {
1960 return -2;
1967 ret++;
1968 if (s->used_clusters[cluster_num] & USED_ANY)
1969 return 0;
1970 s->used_clusters[cluster_num] = USED_FILE;
1972 cluster_num = modified_fat_get(s, cluster_num);
1974 if (fat_eof(s, cluster_num))
1975 return ret;
1976 else if (cluster_num < 2 || cluster_num > s->max_fat_value - 16)
1977 return -1;
1979 offset += s->cluster_size;
1984 * This function looks at the modified data (qcow).
1985 * It returns 0 upon inconsistency or error, and the number of clusters
1986 * used by the directory, its subdirectories and their files.
1988 static int check_directory_consistency(BDRVVVFATState *s,
1989 int cluster_num, const char* path)
1991 int ret = 0;
1992 unsigned char* cluster = g_malloc(s->cluster_size);
1993 direntry_t* direntries = (direntry_t*)cluster;
1994 mapping_t* mapping = find_mapping_for_cluster(s, cluster_num);
1996 long_file_name lfn;
1997 int path_len = strlen(path);
1998 char path2[PATH_MAX + 1];
2000 assert(path_len < PATH_MAX); /* len was tested before! */
2001 pstrcpy(path2, sizeof(path2), path);
2002 path2[path_len] = '/';
2003 path2[path_len + 1] = '\0';
2005 if (mapping) {
2006 const char* basename = get_basename(mapping->path);
2007 const char* basename2 = get_basename(path);
2009 assert(mapping->mode & MODE_DIRECTORY);
2011 assert(mapping->mode & MODE_DELETED);
2012 mapping->mode &= ~MODE_DELETED;
2014 if (strcmp(basename, basename2))
2015 schedule_rename(s, cluster_num, g_strdup(path));
2016 } else
2017 /* new directory */
2018 schedule_mkdir(s, cluster_num, g_strdup(path));
2020 lfn_init(&lfn);
2021 do {
2022 int i;
2023 int subret = 0;
2025 ret++;
2027 if (s->used_clusters[cluster_num] & USED_ANY) {
2028 fprintf(stderr, "cluster %d used more than once\n", (int)cluster_num);
2029 goto fail;
2031 s->used_clusters[cluster_num] = USED_DIRECTORY;
2033 DLOG(fprintf(stderr, "read cluster %d (sector %d)\n", (int)cluster_num, (int)cluster2sector(s, cluster_num)));
2034 subret = vvfat_read(s->bs, cluster2sector(s, cluster_num), cluster,
2035 s->sectors_per_cluster);
2036 if (subret) {
2037 fprintf(stderr, "Error fetching direntries\n");
2038 fail:
2039 g_free(cluster);
2040 return 0;
2043 for (i = 0; i < 0x10 * s->sectors_per_cluster; i++) {
2044 int cluster_count = 0;
2046 DLOG(fprintf(stderr, "check direntry %d:\n", i); print_direntry(direntries + i));
2047 if (is_volume_label(direntries + i) || is_dot(direntries + i) ||
2048 is_free(direntries + i))
2049 continue;
2051 subret = parse_long_name(&lfn, direntries + i);
2052 if (subret < 0) {
2053 fprintf(stderr, "Error in long name\n");
2054 goto fail;
2056 if (subret == 0 || is_free(direntries + i))
2057 continue;
2059 if (fat_chksum(direntries+i) != lfn.checksum) {
2060 subret = parse_short_name(s, &lfn, direntries + i);
2061 if (subret < 0) {
2062 fprintf(stderr, "Error in short name (%d)\n", subret);
2063 goto fail;
2065 if (subret > 0 || !strcmp((char*)lfn.name, ".")
2066 || !strcmp((char*)lfn.name, ".."))
2067 continue;
2069 lfn.checksum = 0x100; /* cannot use long name twice */
2071 if (path_len + 1 + lfn.len >= PATH_MAX) {
2072 fprintf(stderr, "Name too long: %s/%s\n", path, lfn.name);
2073 goto fail;
2075 pstrcpy(path2 + path_len + 1, sizeof(path2) - path_len - 1,
2076 (char*)lfn.name);
2078 if (is_directory(direntries + i)) {
2079 if (begin_of_direntry(direntries + i) == 0) {
2080 DLOG(fprintf(stderr, "invalid begin for directory: %s\n", path2); print_direntry(direntries + i));
2081 goto fail;
2083 cluster_count = check_directory_consistency(s,
2084 begin_of_direntry(direntries + i), path2);
2085 if (cluster_count == 0) {
2086 DLOG(fprintf(stderr, "problem in directory %s:\n", path2); print_direntry(direntries + i));
2087 goto fail;
2089 } else if (is_file(direntries + i)) {
2090 /* check file size with FAT */
2091 cluster_count = get_cluster_count_for_direntry(s, direntries + i, path2);
2092 if (cluster_count !=
2093 DIV_ROUND_UP(le32_to_cpu(direntries[i].size), s->cluster_size)) {
2094 DLOG(fprintf(stderr, "Cluster count mismatch\n"));
2095 goto fail;
2097 } else
2098 abort(); /* cluster_count = 0; */
2100 ret += cluster_count;
2103 cluster_num = modified_fat_get(s, cluster_num);
2104 } while(!fat_eof(s, cluster_num));
2106 g_free(cluster);
2107 return ret;
2110 /* returns 1 on success */
2111 static int is_consistent(BDRVVVFATState* s)
2113 int i, check;
2114 int used_clusters_count = 0;
2116 DLOG(checkpoint());
2118 * - get modified FAT
2119 * - compare the two FATs (TODO)
2120 * - get buffer for marking used clusters
2121 * - recurse direntries from root (using bs->bdrv_read to make
2122 * sure to get the new data)
2123 * - check that the FAT agrees with the size
2124 * - count the number of clusters occupied by this directory and
2125 * its files
2126 * - check that the cumulative used cluster count agrees with the
2127 * FAT
2128 * - if all is fine, return number of used clusters
2130 if (s->fat2 == NULL) {
2131 int size = 0x200 * s->sectors_per_fat;
2132 s->fat2 = g_malloc(size);
2133 memcpy(s->fat2, s->fat.pointer, size);
2135 check = vvfat_read(s->bs,
2136 s->offset_to_fat, s->fat2, s->sectors_per_fat);
2137 if (check) {
2138 fprintf(stderr, "Could not copy fat\n");
2139 return 0;
2141 assert (s->used_clusters);
2142 for (i = 0; i < sector2cluster(s, s->sector_count); i++)
2143 s->used_clusters[i] &= ~USED_ANY;
2145 clear_commits(s);
2147 /* mark every mapped file/directory as deleted.
2148 * (check_directory_consistency() will unmark those still present). */
2149 if (s->qcow)
2150 for (i = 0; i < s->mapping.next; i++) {
2151 mapping_t* mapping = array_get(&(s->mapping), i);
2152 if (mapping->first_mapping_index < 0)
2153 mapping->mode |= MODE_DELETED;
2156 used_clusters_count = check_directory_consistency(s, 0, s->path);
2157 if (used_clusters_count <= 0) {
2158 DLOG(fprintf(stderr, "problem in directory\n"));
2159 return 0;
2162 check = s->last_cluster_of_root_directory;
2163 for (i = check; i < sector2cluster(s, s->sector_count); i++) {
2164 if (modified_fat_get(s, i)) {
2165 if(!s->used_clusters[i]) {
2166 DLOG(fprintf(stderr, "FAT was modified (%d), but cluster is not used?\n", i));
2167 return 0;
2169 check++;
2172 if (s->used_clusters[i] == USED_ALLOCATED) {
2173 /* allocated, but not used... */
2174 DLOG(fprintf(stderr, "unused, modified cluster: %d\n", i));
2175 return 0;
2179 if (check != used_clusters_count)
2180 return 0;
2182 return used_clusters_count;
2185 static inline void adjust_mapping_indices(BDRVVVFATState* s,
2186 int offset, int adjust)
2188 int i;
2190 for (i = 0; i < s->mapping.next; i++) {
2191 mapping_t* mapping = array_get(&(s->mapping), i);
2193 #define ADJUST_MAPPING_INDEX(name) \
2194 if (mapping->name >= offset) \
2195 mapping->name += adjust
2197 ADJUST_MAPPING_INDEX(first_mapping_index);
2198 if (mapping->mode & MODE_DIRECTORY)
2199 ADJUST_MAPPING_INDEX(info.dir.parent_mapping_index);
2203 /* insert or update mapping */
2204 static mapping_t* insert_mapping(BDRVVVFATState* s,
2205 uint32_t begin, uint32_t end)
2208 * - find mapping where mapping->begin >= begin,
2209 * - if mapping->begin > begin: insert
2210 * - adjust all references to mappings!
2211 * - else: adjust
2212 * - replace name
2214 int index = find_mapping_for_cluster_aux(s, begin, 0, s->mapping.next);
2215 mapping_t* mapping = NULL;
2216 mapping_t* first_mapping = array_get(&(s->mapping), 0);
2218 if (index < s->mapping.next && (mapping = array_get(&(s->mapping), index))
2219 && mapping->begin < begin) {
2220 mapping->end = begin;
2221 index++;
2222 mapping = array_get(&(s->mapping), index);
2224 if (index >= s->mapping.next || mapping->begin > begin) {
2225 mapping = array_insert(&(s->mapping), index, 1);
2226 mapping->path = NULL;
2227 adjust_mapping_indices(s, index, +1);
2230 mapping->begin = begin;
2231 mapping->end = end;
2233 DLOG(mapping_t* next_mapping;
2234 assert(index + 1 >= s->mapping.next ||
2235 ((next_mapping = array_get(&(s->mapping), index + 1)) &&
2236 next_mapping->begin >= end)));
2238 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
2239 s->current_mapping = array_get(&(s->mapping),
2240 s->current_mapping - first_mapping);
2242 return mapping;
2245 static int remove_mapping(BDRVVVFATState* s, int mapping_index)
2247 mapping_t* mapping = array_get(&(s->mapping), mapping_index);
2248 mapping_t* first_mapping = array_get(&(s->mapping), 0);
2250 /* free mapping */
2251 if (mapping->first_mapping_index < 0) {
2252 g_free(mapping->path);
2255 /* remove from s->mapping */
2256 array_remove(&(s->mapping), mapping_index);
2258 /* adjust all references to mappings */
2259 adjust_mapping_indices(s, mapping_index, -1);
2261 if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
2262 s->current_mapping = array_get(&(s->mapping),
2263 s->current_mapping - first_mapping);
2265 return 0;
2268 static void adjust_dirindices(BDRVVVFATState* s, int offset, int adjust)
2270 int i;
2271 for (i = 0; i < s->mapping.next; i++) {
2272 mapping_t* mapping = array_get(&(s->mapping), i);
2273 if (mapping->dir_index >= offset)
2274 mapping->dir_index += adjust;
2275 if ((mapping->mode & MODE_DIRECTORY) &&
2276 mapping->info.dir.first_dir_index >= offset)
2277 mapping->info.dir.first_dir_index += adjust;
2281 static direntry_t* insert_direntries(BDRVVVFATState* s,
2282 int dir_index, int count)
2285 * make room in s->directory,
2286 * adjust_dirindices
2288 direntry_t* result = array_insert(&(s->directory), dir_index, count);
2289 if (result == NULL)
2290 return NULL;
2291 adjust_dirindices(s, dir_index, count);
2292 return result;
2295 static int remove_direntries(BDRVVVFATState* s, int dir_index, int count)
2297 int ret = array_remove_slice(&(s->directory), dir_index, count);
2298 if (ret)
2299 return ret;
2300 adjust_dirindices(s, dir_index, -count);
2301 return 0;
2305 * Adapt the mappings of the cluster chain starting at first cluster
2306 * (i.e. if a file starts at first_cluster, the chain is followed according
2307 * to the modified fat, and the corresponding entries in s->mapping are
2308 * adjusted)
2310 static int commit_mappings(BDRVVVFATState* s,
2311 uint32_t first_cluster, int dir_index)
2313 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2314 direntry_t* direntry = array_get(&(s->directory), dir_index);
2315 uint32_t cluster = first_cluster;
2317 vvfat_close_current_file(s);
2319 assert(mapping);
2320 assert(mapping->begin == first_cluster);
2321 mapping->first_mapping_index = -1;
2322 mapping->dir_index = dir_index;
2323 mapping->mode = (dir_index <= 0 || is_directory(direntry)) ?
2324 MODE_DIRECTORY : MODE_NORMAL;
2326 while (!fat_eof(s, cluster)) {
2327 uint32_t c, c1;
2329 for (c = cluster, c1 = modified_fat_get(s, c); c + 1 == c1;
2330 c = c1, c1 = modified_fat_get(s, c1));
2332 c++;
2333 if (c > mapping->end) {
2334 int index = array_index(&(s->mapping), mapping);
2335 int i, max_i = s->mapping.next - index;
2336 for (i = 1; i < max_i && mapping[i].begin < c; i++);
2337 while (--i > 0)
2338 remove_mapping(s, index + 1);
2340 assert(mapping == array_get(&(s->mapping), s->mapping.next - 1)
2341 || mapping[1].begin >= c);
2342 mapping->end = c;
2344 if (!fat_eof(s, c1)) {
2345 int i = find_mapping_for_cluster_aux(s, c1, 0, s->mapping.next);
2346 mapping_t* next_mapping = i >= s->mapping.next ? NULL :
2347 array_get(&(s->mapping), i);
2349 if (next_mapping == NULL || next_mapping->begin > c1) {
2350 int i1 = array_index(&(s->mapping), mapping);
2352 next_mapping = insert_mapping(s, c1, c1+1);
2354 if (c1 < c)
2355 i1++;
2356 mapping = array_get(&(s->mapping), i1);
2359 next_mapping->dir_index = mapping->dir_index;
2360 next_mapping->first_mapping_index =
2361 mapping->first_mapping_index < 0 ?
2362 array_index(&(s->mapping), mapping) :
2363 mapping->first_mapping_index;
2364 next_mapping->path = mapping->path;
2365 next_mapping->mode = mapping->mode;
2366 next_mapping->read_only = mapping->read_only;
2367 if (mapping->mode & MODE_DIRECTORY) {
2368 next_mapping->info.dir.parent_mapping_index =
2369 mapping->info.dir.parent_mapping_index;
2370 next_mapping->info.dir.first_dir_index =
2371 mapping->info.dir.first_dir_index +
2372 0x10 * s->sectors_per_cluster *
2373 (mapping->end - mapping->begin);
2374 } else
2375 next_mapping->info.file.offset = mapping->info.file.offset +
2376 mapping->end - mapping->begin;
2378 mapping = next_mapping;
2381 cluster = c1;
2384 return 0;
2387 static int commit_direntries(BDRVVVFATState* s,
2388 int dir_index, int parent_mapping_index)
2390 direntry_t* direntry = array_get(&(s->directory), dir_index);
2391 uint32_t first_cluster = dir_index == 0 ? 0 : begin_of_direntry(direntry);
2392 mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2394 int factor = 0x10 * s->sectors_per_cluster;
2395 int old_cluster_count, new_cluster_count;
2396 int current_dir_index = mapping->info.dir.first_dir_index;
2397 int first_dir_index = current_dir_index;
2398 int ret, i;
2399 uint32_t c;
2401 DLOG(fprintf(stderr, "commit_direntries for %s, parent_mapping_index %d\n", mapping->path, parent_mapping_index));
2403 assert(direntry);
2404 assert(mapping);
2405 assert(mapping->begin == first_cluster);
2406 assert(mapping->info.dir.first_dir_index < s->directory.next);
2407 assert(mapping->mode & MODE_DIRECTORY);
2408 assert(dir_index == 0 || is_directory(direntry));
2410 mapping->info.dir.parent_mapping_index = parent_mapping_index;
2412 if (first_cluster == 0) {
2413 old_cluster_count = new_cluster_count =
2414 s->last_cluster_of_root_directory;
2415 } else {
2416 for (old_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2417 c = fat_get(s, c))
2418 old_cluster_count++;
2420 for (new_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2421 c = modified_fat_get(s, c))
2422 new_cluster_count++;
2425 if (new_cluster_count > old_cluster_count) {
2426 if (insert_direntries(s,
2427 current_dir_index + factor * old_cluster_count,
2428 factor * (new_cluster_count - old_cluster_count)) == NULL)
2429 return -1;
2430 } else if (new_cluster_count < old_cluster_count)
2431 remove_direntries(s,
2432 current_dir_index + factor * new_cluster_count,
2433 factor * (old_cluster_count - new_cluster_count));
2435 for (c = first_cluster; !fat_eof(s, c); c = modified_fat_get(s, c)) {
2436 direntry_t *first_direntry;
2437 void* direntry = array_get(&(s->directory), current_dir_index);
2438 int ret = vvfat_read(s->bs, cluster2sector(s, c), direntry,
2439 s->sectors_per_cluster);
2440 if (ret)
2441 return ret;
2443 /* The first directory entry on the filesystem is the volume name */
2444 first_direntry = (direntry_t*) s->directory.pointer;
2445 assert(!memcmp(first_direntry->name, s->volume_label, 11));
2447 current_dir_index += factor;
2450 ret = commit_mappings(s, first_cluster, dir_index);
2451 if (ret)
2452 return ret;
2454 /* recurse */
2455 for (i = 0; i < factor * new_cluster_count; i++) {
2456 direntry = array_get(&(s->directory), first_dir_index + i);
2457 if (is_directory(direntry) && !is_dot(direntry)) {
2458 mapping = find_mapping_for_cluster(s, first_cluster);
2459 assert(mapping->mode & MODE_DIRECTORY);
2460 ret = commit_direntries(s, first_dir_index + i,
2461 array_index(&(s->mapping), mapping));
2462 if (ret)
2463 return ret;
2467 return 0;
2470 /* commit one file (adjust contents, adjust mapping),
2471 return first_mapping_index */
2472 static int commit_one_file(BDRVVVFATState* s,
2473 int dir_index, uint32_t offset)
2475 direntry_t* direntry = array_get(&(s->directory), dir_index);
2476 uint32_t c = begin_of_direntry(direntry);
2477 uint32_t first_cluster = c;
2478 mapping_t* mapping = find_mapping_for_cluster(s, c);
2479 uint32_t size = filesize_of_direntry(direntry);
2480 char* cluster = g_malloc(s->cluster_size);
2481 uint32_t i;
2482 int fd = 0;
2484 assert(offset < size);
2485 assert((offset % s->cluster_size) == 0);
2487 for (i = s->cluster_size; i < offset; i += s->cluster_size)
2488 c = modified_fat_get(s, c);
2490 fd = qemu_open(mapping->path, O_RDWR | O_CREAT | O_BINARY, 0666);
2491 if (fd < 0) {
2492 fprintf(stderr, "Could not open %s... (%s, %d)\n", mapping->path,
2493 strerror(errno), errno);
2494 g_free(cluster);
2495 return fd;
2497 if (offset > 0) {
2498 if (lseek(fd, offset, SEEK_SET) != offset) {
2499 qemu_close(fd);
2500 g_free(cluster);
2501 return -3;
2505 while (offset < size) {
2506 uint32_t c1;
2507 int rest_size = (size - offset > s->cluster_size ?
2508 s->cluster_size : size - offset);
2509 int ret;
2511 c1 = modified_fat_get(s, c);
2513 assert((size - offset == 0 && fat_eof(s, c)) ||
2514 (size > offset && c >=2 && !fat_eof(s, c)));
2516 ret = vvfat_read(s->bs, cluster2sector(s, c),
2517 (uint8_t*)cluster, (rest_size + 0x1ff) / 0x200);
2519 if (ret < 0) {
2520 qemu_close(fd);
2521 g_free(cluster);
2522 return ret;
2525 if (write(fd, cluster, rest_size) < 0) {
2526 qemu_close(fd);
2527 g_free(cluster);
2528 return -2;
2531 offset += rest_size;
2532 c = c1;
2535 if (ftruncate(fd, size)) {
2536 perror("ftruncate()");
2537 qemu_close(fd);
2538 g_free(cluster);
2539 return -4;
2541 qemu_close(fd);
2542 g_free(cluster);
2544 return commit_mappings(s, first_cluster, dir_index);
2547 #ifdef DEBUG
2548 /* test, if all mappings point to valid direntries */
2549 static void check1(BDRVVVFATState* s)
2551 int i;
2552 for (i = 0; i < s->mapping.next; i++) {
2553 mapping_t* mapping = array_get(&(s->mapping), i);
2554 if (mapping->mode & MODE_DELETED) {
2555 fprintf(stderr, "deleted\n");
2556 continue;
2558 assert(mapping->dir_index < s->directory.next);
2559 direntry_t* direntry = array_get(&(s->directory), mapping->dir_index);
2560 assert(mapping->begin == begin_of_direntry(direntry) || mapping->first_mapping_index >= 0);
2561 if (mapping->mode & MODE_DIRECTORY) {
2562 assert(mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster * (mapping->end - mapping->begin) <= s->directory.next);
2563 assert((mapping->info.dir.first_dir_index % (0x10 * s->sectors_per_cluster)) == 0);
2568 /* test, if all direntries have mappings */
2569 static void check2(BDRVVVFATState* s)
2571 int i;
2572 int first_mapping = -1;
2574 for (i = 0; i < s->directory.next; i++) {
2575 direntry_t* direntry = array_get(&(s->directory), i);
2577 if (is_short_name(direntry) && begin_of_direntry(direntry)) {
2578 mapping_t* mapping = find_mapping_for_cluster(s, begin_of_direntry(direntry));
2579 assert(mapping);
2580 assert(mapping->dir_index == i || is_dot(direntry));
2581 assert(mapping->begin == begin_of_direntry(direntry) || is_dot(direntry));
2584 if ((i % (0x10 * s->sectors_per_cluster)) == 0) {
2585 /* cluster start */
2586 int j, count = 0;
2588 for (j = 0; j < s->mapping.next; j++) {
2589 mapping_t* mapping = array_get(&(s->mapping), j);
2590 if (mapping->mode & MODE_DELETED)
2591 continue;
2592 if (mapping->mode & MODE_DIRECTORY) {
2593 if (mapping->info.dir.first_dir_index <= i && mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster > i) {
2594 assert(++count == 1);
2595 if (mapping->first_mapping_index == -1)
2596 first_mapping = array_index(&(s->mapping), mapping);
2597 else
2598 assert(first_mapping == mapping->first_mapping_index);
2599 if (mapping->info.dir.parent_mapping_index < 0)
2600 assert(j == 0);
2601 else {
2602 mapping_t* parent = array_get(&(s->mapping), mapping->info.dir.parent_mapping_index);
2603 assert(parent->mode & MODE_DIRECTORY);
2604 assert(parent->info.dir.first_dir_index < mapping->info.dir.first_dir_index);
2609 if (count == 0)
2610 first_mapping = -1;
2614 #endif
2616 static int handle_renames_and_mkdirs(BDRVVVFATState* s)
2618 int i;
2620 #ifdef DEBUG
2621 fprintf(stderr, "handle_renames\n");
2622 for (i = 0; i < s->commits.next; i++) {
2623 commit_t* commit = array_get(&(s->commits), i);
2624 fprintf(stderr, "%d, %s (%d, %d)\n", i, commit->path ? commit->path : "(null)", commit->param.rename.cluster, commit->action);
2626 #endif
2628 for (i = 0; i < s->commits.next;) {
2629 commit_t* commit = array_get(&(s->commits), i);
2630 if (commit->action == ACTION_RENAME) {
2631 mapping_t* mapping = find_mapping_for_cluster(s,
2632 commit->param.rename.cluster);
2633 char* old_path = mapping->path;
2635 assert(commit->path);
2636 mapping->path = commit->path;
2637 if (rename(old_path, mapping->path))
2638 return -2;
2640 if (mapping->mode & MODE_DIRECTORY) {
2641 int l1 = strlen(mapping->path);
2642 int l2 = strlen(old_path);
2643 int diff = l1 - l2;
2644 direntry_t* direntry = array_get(&(s->directory),
2645 mapping->info.dir.first_dir_index);
2646 uint32_t c = mapping->begin;
2647 int i = 0;
2649 /* recurse */
2650 while (!fat_eof(s, c)) {
2651 do {
2652 direntry_t* d = direntry + i;
2654 if (is_file(d) || (is_directory(d) && !is_dot(d))) {
2655 mapping_t* m = find_mapping_for_cluster(s,
2656 begin_of_direntry(d));
2657 int l = strlen(m->path);
2658 char* new_path = g_malloc(l + diff + 1);
2660 assert(!strncmp(m->path, mapping->path, l2));
2662 pstrcpy(new_path, l + diff + 1, mapping->path);
2663 pstrcpy(new_path + l1, l + diff + 1 - l1,
2664 m->path + l2);
2666 schedule_rename(s, m->begin, new_path);
2668 i++;
2669 } while((i % (0x10 * s->sectors_per_cluster)) != 0);
2670 c = fat_get(s, c);
2674 g_free(old_path);
2675 array_remove(&(s->commits), i);
2676 continue;
2677 } else if (commit->action == ACTION_MKDIR) {
2678 mapping_t* mapping;
2679 int j, parent_path_len;
2681 #ifdef __MINGW32__
2682 if (mkdir(commit->path))
2683 return -5;
2684 #else
2685 if (mkdir(commit->path, 0755))
2686 return -5;
2687 #endif
2689 mapping = insert_mapping(s, commit->param.mkdir.cluster,
2690 commit->param.mkdir.cluster + 1);
2691 if (mapping == NULL)
2692 return -6;
2694 mapping->mode = MODE_DIRECTORY;
2695 mapping->read_only = 0;
2696 mapping->path = commit->path;
2697 j = s->directory.next;
2698 assert(j);
2699 insert_direntries(s, s->directory.next,
2700 0x10 * s->sectors_per_cluster);
2701 mapping->info.dir.first_dir_index = j;
2703 parent_path_len = strlen(commit->path)
2704 - strlen(get_basename(commit->path)) - 1;
2705 for (j = 0; j < s->mapping.next; j++) {
2706 mapping_t* m = array_get(&(s->mapping), j);
2707 if (m->first_mapping_index < 0 && m != mapping &&
2708 !strncmp(m->path, mapping->path, parent_path_len) &&
2709 strlen(m->path) == parent_path_len)
2710 break;
2712 assert(j < s->mapping.next);
2713 mapping->info.dir.parent_mapping_index = j;
2715 array_remove(&(s->commits), i);
2716 continue;
2719 i++;
2721 return 0;
2725 * TODO: make sure that the short name is not matching *another* file
2727 static int handle_commits(BDRVVVFATState* s)
2729 int i, fail = 0;
2731 vvfat_close_current_file(s);
2733 for (i = 0; !fail && i < s->commits.next; i++) {
2734 commit_t* commit = array_get(&(s->commits), i);
2735 switch(commit->action) {
2736 case ACTION_RENAME: case ACTION_MKDIR:
2737 abort();
2738 fail = -2;
2739 break;
2740 case ACTION_WRITEOUT: {
2741 #ifndef NDEBUG
2742 /* these variables are only used by assert() below */
2743 direntry_t* entry = array_get(&(s->directory),
2744 commit->param.writeout.dir_index);
2745 uint32_t begin = begin_of_direntry(entry);
2746 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2747 #endif
2749 assert(mapping);
2750 assert(mapping->begin == begin);
2751 assert(commit->path == NULL);
2753 if (commit_one_file(s, commit->param.writeout.dir_index,
2754 commit->param.writeout.modified_offset))
2755 fail = -3;
2757 break;
2759 case ACTION_NEW_FILE: {
2760 int begin = commit->param.new_file.first_cluster;
2761 mapping_t* mapping = find_mapping_for_cluster(s, begin);
2762 direntry_t* entry;
2763 int i;
2765 /* find direntry */
2766 for (i = 0; i < s->directory.next; i++) {
2767 entry = array_get(&(s->directory), i);
2768 if (is_file(entry) && begin_of_direntry(entry) == begin)
2769 break;
2772 if (i >= s->directory.next) {
2773 fail = -6;
2774 continue;
2777 /* make sure there exists an initial mapping */
2778 if (mapping && mapping->begin != begin) {
2779 mapping->end = begin;
2780 mapping = NULL;
2782 if (mapping == NULL) {
2783 mapping = insert_mapping(s, begin, begin+1);
2785 /* most members will be fixed in commit_mappings() */
2786 assert(commit->path);
2787 mapping->path = commit->path;
2788 mapping->read_only = 0;
2789 mapping->mode = MODE_NORMAL;
2790 mapping->info.file.offset = 0;
2792 if (commit_one_file(s, i, 0))
2793 fail = -7;
2795 break;
2797 default:
2798 abort();
2801 if (i > 0 && array_remove_slice(&(s->commits), 0, i))
2802 return -1;
2803 return fail;
2806 static int handle_deletes(BDRVVVFATState* s)
2808 int i, deferred = 1, deleted = 1;
2810 /* delete files corresponding to mappings marked as deleted */
2811 /* handle DELETEs and unused mappings (modified_fat_get(s, mapping->begin) == 0) */
2812 while (deferred && deleted) {
2813 deferred = 0;
2814 deleted = 0;
2816 for (i = 1; i < s->mapping.next; i++) {
2817 mapping_t* mapping = array_get(&(s->mapping), i);
2818 if (mapping->mode & MODE_DELETED) {
2819 direntry_t* entry = array_get(&(s->directory),
2820 mapping->dir_index);
2822 if (is_free(entry)) {
2823 /* remove file/directory */
2824 if (mapping->mode & MODE_DIRECTORY) {
2825 int j, next_dir_index = s->directory.next,
2826 first_dir_index = mapping->info.dir.first_dir_index;
2828 if (rmdir(mapping->path) < 0) {
2829 if (errno == ENOTEMPTY) {
2830 deferred++;
2831 continue;
2832 } else
2833 return -5;
2836 for (j = 1; j < s->mapping.next; j++) {
2837 mapping_t* m = array_get(&(s->mapping), j);
2838 if (m->mode & MODE_DIRECTORY &&
2839 m->info.dir.first_dir_index >
2840 first_dir_index &&
2841 m->info.dir.first_dir_index <
2842 next_dir_index)
2843 next_dir_index =
2844 m->info.dir.first_dir_index;
2846 remove_direntries(s, first_dir_index,
2847 next_dir_index - first_dir_index);
2849 deleted++;
2851 } else {
2852 if (unlink(mapping->path))
2853 return -4;
2854 deleted++;
2856 DLOG(fprintf(stderr, "DELETE (%d)\n", i); print_mapping(mapping); print_direntry(entry));
2857 remove_mapping(s, i);
2862 return 0;
2866 * synchronize mapping with new state:
2868 * - copy FAT (with bdrv_read)
2869 * - mark all filenames corresponding to mappings as deleted
2870 * - recurse direntries from root (using bs->bdrv_read)
2871 * - delete files corresponding to mappings marked as deleted
2873 static int do_commit(BDRVVVFATState* s)
2875 int ret = 0;
2877 /* the real meat are the commits. Nothing to do? Move along! */
2878 if (s->commits.next == 0)
2879 return 0;
2881 vvfat_close_current_file(s);
2883 ret = handle_renames_and_mkdirs(s);
2884 if (ret) {
2885 fprintf(stderr, "Error handling renames (%d)\n", ret);
2886 abort();
2887 return ret;
2890 /* copy FAT (with bdrv_read) */
2891 memcpy(s->fat.pointer, s->fat2, 0x200 * s->sectors_per_fat);
2893 /* recurse direntries from root (using bs->bdrv_read) */
2894 ret = commit_direntries(s, 0, -1);
2895 if (ret) {
2896 fprintf(stderr, "Fatal: error while committing (%d)\n", ret);
2897 abort();
2898 return ret;
2901 ret = handle_commits(s);
2902 if (ret) {
2903 fprintf(stderr, "Error handling commits (%d)\n", ret);
2904 abort();
2905 return ret;
2908 ret = handle_deletes(s);
2909 if (ret) {
2910 fprintf(stderr, "Error deleting\n");
2911 abort();
2912 return ret;
2915 if (s->qcow->bs->drv->bdrv_make_empty) {
2916 s->qcow->bs->drv->bdrv_make_empty(s->qcow->bs);
2919 memset(s->used_clusters, 0, sector2cluster(s, s->sector_count));
2921 DLOG(checkpoint());
2922 return 0;
2925 static int try_commit(BDRVVVFATState* s)
2927 vvfat_close_current_file(s);
2928 DLOG(checkpoint());
2929 if(!is_consistent(s))
2930 return -1;
2931 return do_commit(s);
2934 static int vvfat_write(BlockDriverState *bs, int64_t sector_num,
2935 const uint8_t *buf, int nb_sectors)
2937 BDRVVVFATState *s = bs->opaque;
2938 int i, ret;
2940 DLOG(checkpoint());
2942 /* Check if we're operating in read-only mode */
2943 if (s->qcow == NULL) {
2944 return -EACCES;
2947 vvfat_close_current_file(s);
2950 * Some sanity checks:
2951 * - do not allow writing to the boot sector
2952 * - do not allow to write non-ASCII filenames
2955 if (sector_num < s->offset_to_fat)
2956 return -1;
2958 for (i = sector2cluster(s, sector_num);
2959 i <= sector2cluster(s, sector_num + nb_sectors - 1);) {
2960 mapping_t* mapping = find_mapping_for_cluster(s, i);
2961 if (mapping) {
2962 if (mapping->read_only) {
2963 fprintf(stderr, "Tried to write to write-protected file %s\n",
2964 mapping->path);
2965 return -1;
2968 if (mapping->mode & MODE_DIRECTORY) {
2969 int begin = cluster2sector(s, i);
2970 int end = begin + s->sectors_per_cluster, k;
2971 int dir_index;
2972 const direntry_t* direntries;
2973 long_file_name lfn;
2975 lfn_init(&lfn);
2977 if (begin < sector_num)
2978 begin = sector_num;
2979 if (end > sector_num + nb_sectors)
2980 end = sector_num + nb_sectors;
2981 dir_index = mapping->dir_index +
2982 0x10 * (begin - mapping->begin * s->sectors_per_cluster);
2983 direntries = (direntry_t*)(buf + 0x200 * (begin - sector_num));
2985 for (k = 0; k < (end - begin) * 0x10; k++) {
2986 /* do not allow non-ASCII filenames */
2987 if (parse_long_name(&lfn, direntries + k) < 0) {
2988 fprintf(stderr, "Warning: non-ASCII filename\n");
2989 return -1;
2991 /* no access to the direntry of a read-only file */
2992 else if (is_short_name(direntries+k) &&
2993 (direntries[k].attributes & 1)) {
2994 if (memcmp(direntries + k,
2995 array_get(&(s->directory), dir_index + k),
2996 sizeof(direntry_t))) {
2997 fprintf(stderr, "Warning: tried to write to write-protected file\n");
2998 return -1;
3003 i = mapping->end;
3004 } else
3005 i++;
3009 * Use qcow backend. Commit later.
3011 DLOG(fprintf(stderr, "Write to qcow backend: %d + %d\n", (int)sector_num, nb_sectors));
3012 ret = bdrv_write(s->qcow, sector_num, buf, nb_sectors);
3013 if (ret < 0) {
3014 fprintf(stderr, "Error writing to qcow backend\n");
3015 return ret;
3018 for (i = sector2cluster(s, sector_num);
3019 i <= sector2cluster(s, sector_num + nb_sectors - 1); i++)
3020 if (i >= 0)
3021 s->used_clusters[i] |= USED_ALLOCATED;
3023 DLOG(checkpoint());
3024 /* TODO: add timeout */
3025 try_commit(s);
3027 DLOG(checkpoint());
3028 return 0;
3031 static int coroutine_fn
3032 vvfat_co_pwritev(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
3033 QEMUIOVector *qiov, int flags)
3035 int ret;
3036 BDRVVVFATState *s = bs->opaque;
3037 uint64_t sector_num = offset >> BDRV_SECTOR_BITS;
3038 int nb_sectors = bytes >> BDRV_SECTOR_BITS;
3039 void *buf;
3041 assert((offset & (BDRV_SECTOR_SIZE - 1)) == 0);
3042 assert((bytes & (BDRV_SECTOR_SIZE - 1)) == 0);
3044 buf = g_try_malloc(bytes);
3045 if (bytes && buf == NULL) {
3046 return -ENOMEM;
3048 qemu_iovec_to_buf(qiov, 0, buf, bytes);
3050 qemu_co_mutex_lock(&s->lock);
3051 ret = vvfat_write(bs, sector_num, buf, nb_sectors);
3052 qemu_co_mutex_unlock(&s->lock);
3054 g_free(buf);
3056 return ret;
3059 static int64_t coroutine_fn vvfat_co_get_block_status(BlockDriverState *bs,
3060 int64_t sector_num, int nb_sectors, int *n, BlockDriverState **file)
3062 *n = bs->total_sectors - sector_num;
3063 if (*n > nb_sectors) {
3064 *n = nb_sectors;
3065 } else if (*n < 0) {
3066 return 0;
3068 return BDRV_BLOCK_DATA;
3071 static int coroutine_fn
3072 write_target_commit(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
3073 QEMUIOVector *qiov, int flags)
3075 BDRVVVFATState* s = *((BDRVVVFATState**) bs->opaque);
3076 return try_commit(s);
3079 static void write_target_close(BlockDriverState *bs) {
3080 BDRVVVFATState* s = *((BDRVVVFATState**) bs->opaque);
3081 bdrv_unref_child(s->bs, s->qcow);
3082 g_free(s->qcow_filename);
3085 static BlockDriver vvfat_write_target = {
3086 .format_name = "vvfat_write_target",
3087 .instance_size = sizeof(void*),
3088 .bdrv_co_pwritev = write_target_commit,
3089 .bdrv_close = write_target_close,
3092 static void vvfat_qcow_options(int *child_flags, QDict *child_options,
3093 int parent_flags, QDict *parent_options)
3095 qdict_set_default_str(child_options, BDRV_OPT_READ_ONLY, "off");
3096 *child_flags = BDRV_O_NO_FLUSH;
3099 static const BdrvChildRole child_vvfat_qcow = {
3100 .inherit_options = vvfat_qcow_options,
3103 static int enable_write_target(BlockDriverState *bs, Error **errp)
3105 BDRVVVFATState *s = bs->opaque;
3106 BlockDriver *bdrv_qcow = NULL;
3107 BlockDriverState *backing;
3108 QemuOpts *opts = NULL;
3109 int ret;
3110 int size = sector2cluster(s, s->sector_count);
3111 QDict *options;
3113 s->used_clusters = calloc(size, 1);
3115 array_init(&(s->commits), sizeof(commit_t));
3117 s->qcow_filename = g_malloc(PATH_MAX);
3118 ret = get_tmp_filename(s->qcow_filename, PATH_MAX);
3119 if (ret < 0) {
3120 error_setg_errno(errp, -ret, "can't create temporary file");
3121 goto err;
3124 bdrv_qcow = bdrv_find_format("qcow");
3125 if (!bdrv_qcow) {
3126 error_setg(errp, "Failed to locate qcow driver");
3127 ret = -ENOENT;
3128 goto err;
3131 opts = qemu_opts_create(bdrv_qcow->create_opts, NULL, 0, &error_abort);
3132 qemu_opt_set_number(opts, BLOCK_OPT_SIZE, s->sector_count * 512,
3133 &error_abort);
3134 qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, "fat:", &error_abort);
3136 ret = bdrv_create(bdrv_qcow, s->qcow_filename, opts, errp);
3137 qemu_opts_del(opts);
3138 if (ret < 0) {
3139 goto err;
3142 options = qdict_new();
3143 qdict_put_str(options, "write-target.driver", "qcow");
3144 s->qcow = bdrv_open_child(s->qcow_filename, options, "write-target", bs,
3145 &child_vvfat_qcow, false, errp);
3146 QDECREF(options);
3147 if (!s->qcow) {
3148 ret = -EINVAL;
3149 goto err;
3152 #ifndef _WIN32
3153 unlink(s->qcow_filename);
3154 #endif
3156 backing = bdrv_new_open_driver(&vvfat_write_target, NULL, BDRV_O_ALLOW_RDWR,
3157 &error_abort);
3158 *(void**) backing->opaque = s;
3160 bdrv_set_backing_hd(s->bs, backing, &error_abort);
3161 bdrv_unref(backing);
3163 return 0;
3165 err:
3166 g_free(s->qcow_filename);
3167 s->qcow_filename = NULL;
3168 return ret;
3171 static void vvfat_child_perm(BlockDriverState *bs, BdrvChild *c,
3172 const BdrvChildRole *role,
3173 uint64_t perm, uint64_t shared,
3174 uint64_t *nperm, uint64_t *nshared)
3176 BDRVVVFATState *s = bs->opaque;
3178 assert(c == s->qcow || role == &child_backing);
3180 if (c == s->qcow) {
3181 /* This is a private node, nobody should try to attach to it */
3182 *nperm = BLK_PERM_CONSISTENT_READ | BLK_PERM_WRITE;
3183 *nshared = BLK_PERM_WRITE_UNCHANGED;
3184 } else {
3185 /* The backing file is there so 'commit' can use it. vvfat doesn't
3186 * access it in any way. */
3187 *nperm = 0;
3188 *nshared = BLK_PERM_ALL;
3192 static void vvfat_close(BlockDriverState *bs)
3194 BDRVVVFATState *s = bs->opaque;
3196 vvfat_close_current_file(s);
3197 array_free(&(s->fat));
3198 array_free(&(s->directory));
3199 array_free(&(s->mapping));
3200 g_free(s->cluster_buffer);
3202 if (s->qcow) {
3203 migrate_del_blocker(s->migration_blocker);
3204 error_free(s->migration_blocker);
3208 static BlockDriver bdrv_vvfat = {
3209 .format_name = "vvfat",
3210 .protocol_name = "fat",
3211 .instance_size = sizeof(BDRVVVFATState),
3213 .bdrv_parse_filename = vvfat_parse_filename,
3214 .bdrv_file_open = vvfat_open,
3215 .bdrv_refresh_limits = vvfat_refresh_limits,
3216 .bdrv_close = vvfat_close,
3217 .bdrv_child_perm = vvfat_child_perm,
3219 .bdrv_co_preadv = vvfat_co_preadv,
3220 .bdrv_co_pwritev = vvfat_co_pwritev,
3221 .bdrv_co_get_block_status = vvfat_co_get_block_status,
3224 static void bdrv_vvfat_init(void)
3226 bdrv_register(&bdrv_vvfat);
3229 block_init(bdrv_vvfat_init);
3231 #ifdef DEBUG
3232 static void checkpoint(void) {
3233 assert(((mapping_t*)array_get(&(vvv->mapping), 0))->end == 2);
3234 check1(vvv);
3235 check2(vvv);
3236 assert(!vvv->current_mapping || vvv->current_fd || (vvv->current_mapping->mode & MODE_DIRECTORY));
3237 #if 0
3238 if (((direntry_t*)vvv->directory.pointer)[1].attributes != 0xf)
3239 fprintf(stderr, "Nonono!\n");
3240 mapping_t* mapping;
3241 direntry_t* direntry;
3242 assert(vvv->mapping.size >= vvv->mapping.item_size * vvv->mapping.next);
3243 assert(vvv->directory.size >= vvv->directory.item_size * vvv->directory.next);
3244 if (vvv->mapping.next<47)
3245 return;
3246 assert((mapping = array_get(&(vvv->mapping), 47)));
3247 assert(mapping->dir_index < vvv->directory.next);
3248 direntry = array_get(&(vvv->directory), mapping->dir_index);
3249 assert(!memcmp(direntry->name, "USB H ", 11) || direntry->name[0]==0);
3250 #endif
3252 #endif