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