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