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