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