1 /* vim:set shiftwidth=4 ts=8: */
3 * QEMU Block driver for virtual VFAT (shadows a local directory)
5 * Copyright (c) 2004,2005 Johannes E. Schindelin
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
28 #include "qemu-common.h"
29 #include "block_int.h"
38 /* TODO: add ":bootsector=blabla.img:" */
39 /* LATER TODO: add automatic boot sector generation from
40 BOOTEASY.ASM and Ranish Partition Manager
41 Note that DOS assumes the system files to be the first files in the
42 file system (test if the boot sector still relies on that fact)! */
43 /* MAYBE TODO: write block-visofs.c */
44 /* TODO: call try_commit() only after a timeout */
56 static void checkpoint(void);
59 void nonono(const char* file
, int line
, const char* msg
) {
60 fprintf(stderr
, "Nonono! %s:%d %s\n", file
, line
, msg
);
64 #define assert(a) do {if (!(a)) nonono(__FILE__, __LINE__, #a);}while(0)
73 /* dynamic array functions */
74 typedef struct array_t
{
76 unsigned int size
,next
,item_size
;
79 static inline void array_init(array_t
* array
,unsigned int item_size
)
84 array
->item_size
=item_size
;
87 static inline void array_free(array_t
* array
)
91 array
->size
=array
->next
=0;
94 /* does not automatically grow */
95 static inline void* array_get(array_t
* array
,unsigned int index
) {
97 assert(index
< array
->next
);
98 return array
->pointer
+ index
* array
->item_size
;
101 static inline int array_ensure_allocated(array_t
* array
, int index
)
103 if((index
+ 1) * array
->item_size
> array
->size
) {
104 int new_size
= (index
+ 32) * array
->item_size
;
105 array
->pointer
= realloc(array
->pointer
, new_size
);
108 array
->size
= new_size
;
109 array
->next
= index
+ 1;
115 static inline void* array_get_next(array_t
* array
) {
116 unsigned int next
= array
->next
;
119 if (array_ensure_allocated(array
, next
) < 0)
122 array
->next
= next
+ 1;
123 result
= array_get(array
, next
);
128 static inline void* array_insert(array_t
* array
,unsigned int index
,unsigned int count
) {
129 if((array
->next
+count
)*array
->item_size
>array
->size
) {
130 int increment
=count
*array
->item_size
;
131 array
->pointer
=realloc(array
->pointer
,array
->size
+increment
);
134 array
->size
+=increment
;
136 memmove(array
->pointer
+(index
+count
)*array
->item_size
,
137 array
->pointer
+index
*array
->item_size
,
138 (array
->next
-index
)*array
->item_size
);
140 return array
->pointer
+index
*array
->item_size
;
143 /* this performs a "roll", so that the element which was at index_from becomes
144 * index_to, but the order of all other elements is preserved. */
145 static inline int array_roll(array_t
* array
,int index_to
,int index_from
,int count
)
153 index_to
<0 || index_to
>=array
->next
||
154 index_from
<0 || index_from
>=array
->next
)
157 if(index_to
==index_from
)
161 from
=array
->pointer
+index_from
*is
;
162 to
=array
->pointer
+index_to
*is
;
163 buf
=malloc(is
*count
);
164 memcpy(buf
,from
,is
*count
);
166 if(index_to
<index_from
)
167 memmove(to
+is
*count
,to
,from
-to
);
169 memmove(from
,from
+is
*count
,to
-from
);
171 memcpy(to
,buf
,is
*count
);
178 static inline int array_remove_slice(array_t
* array
,int index
, int count
)
182 assert(index
+ count
<= array
->next
);
183 if(array_roll(array
,array
->next
-1,index
,count
))
185 array
->next
-= count
;
189 static int array_remove(array_t
* array
,int index
)
191 return array_remove_slice(array
, index
, 1);
194 /* return the index for a given member */
195 static int array_index(array_t
* array
, void* pointer
)
197 size_t offset
= (char*)pointer
- array
->pointer
;
199 assert((offset
% array
->item_size
) == 0);
200 assert(offset
/array
->item_size
< array
->next
);
201 return offset
/array
->item_size
;
204 /* These structures are used to fake a disk and the VFAT filesystem.
205 * For this reason we need to use __attribute__((packed)). */
207 typedef struct bootsector_t
{
210 uint16_t sector_size
;
211 uint8_t sectors_per_cluster
;
212 uint16_t reserved_sectors
;
213 uint8_t number_of_fats
;
214 uint16_t root_entries
;
215 uint16_t total_sectors16
;
217 uint16_t sectors_per_fat
;
218 uint16_t sectors_per_track
;
219 uint16_t number_of_heads
;
220 uint32_t hidden_sectors
;
221 uint32_t total_sectors
;
224 uint8_t drive_number
;
225 uint8_t current_head
;
228 uint8_t volume_label
[11];
229 } __attribute__((packed
)) fat16
;
231 uint32_t sectors_per_fat
;
234 uint32_t first_cluster_of_root_directory
;
235 uint16_t info_sector
;
236 uint16_t backup_boot_sector
;
238 } __attribute__((packed
)) fat32
;
241 uint8_t ignored
[0x1c0];
243 } __attribute__((packed
)) bootsector_t
;
251 typedef struct partition_t
{
252 uint8_t attributes
; /* 0x80 = bootable */
254 uint8_t fs_type
; /* 0x1 = FAT12, 0x6 = FAT16, 0xe = FAT16_LBA, 0xb = FAT32, 0xc = FAT32_LBA */
256 uint32_t start_sector_long
;
257 uint32_t length_sector_long
;
258 } __attribute__((packed
)) partition_t
;
260 typedef struct mbr_t
{
261 uint8_t ignored
[0x1b8];
264 partition_t partition
[4];
266 } __attribute__((packed
)) mbr_t
;
268 typedef struct direntry_t
{
270 uint8_t extension
[3];
281 } __attribute__((packed
)) direntry_t
;
283 /* this structure are used to transparently access the files */
285 typedef struct mapping_t
{
286 /* begin is the first cluster, end is the last+1 */
288 /* as s->directory is growable, no pointer may be used here */
289 unsigned int dir_index
;
290 /* the clusters of a file may be in any order; this points to the first */
291 int first_mapping_index
;
294 * - the offset in the file (in clusters) for a file, or
295 * - the next cluster of the directory for a directory, and
296 * - the address of the buffer for a faked entry
302 int parent_mapping_index
;
306 /* path contains the full path, i.e. it always starts with s->path */
309 enum { MODE_UNDEFINED
= 0, MODE_NORMAL
= 1, MODE_MODIFIED
= 2,
310 MODE_DIRECTORY
= 4, MODE_FAKED
= 8,
311 MODE_DELETED
= 16, MODE_RENAMED
= 32 } mode
;
316 static void print_direntry(const struct direntry_t
*);
317 static void print_mapping(const struct mapping_t
* mapping
);
320 /* here begins the real VVFAT driver */
322 typedef struct BDRVVVFATState
{
323 BlockDriverState
* bs
; /* pointer to parent */
324 unsigned int first_sectors_number
; /* 1 for a single partition, 0x40 for a disk with partition table */
325 unsigned char first_sectors
[0x40*0x200];
327 int fat_type
; /* 16 or 32 */
328 array_t fat
,directory
,mapping
;
330 unsigned int cluster_size
;
331 unsigned int sectors_per_cluster
;
332 unsigned int sectors_per_fat
;
333 unsigned int sectors_of_root_directory
;
334 uint32_t last_cluster_of_root_directory
;
335 unsigned int faked_sectors
; /* how many sectors are faked before file data */
336 uint32_t sector_count
; /* total number of sectors of the partition */
337 uint32_t cluster_count
; /* total number of clusters of this partition */
338 uint32_t max_fat_value
;
341 mapping_t
* current_mapping
;
342 unsigned char* cluster
; /* points to current cluster */
343 unsigned char* cluster_buffer
; /* points to a buffer to hold temp data */
344 unsigned int current_cluster
;
347 BlockDriverState
* write_target
;
349 BlockDriverState
* qcow
;
354 int downcase_short_names
;
357 /* take the sector position spos and convert it to Cylinder/Head/Sector position
358 * if the position is outside the specified geometry, fill maximum value for CHS
359 * and return 1 to signal overflow.
361 static int sector2CHS(BlockDriverState
* bs
, mbr_chs_t
* chs
, int spos
){
363 sector
= spos
% (bs
->secs
); spos
/= bs
->secs
;
364 head
= spos
% (bs
->heads
); spos
/= bs
->heads
;
365 if(spos
>= bs
->cyls
){
367 it happens if 32bit sector positions are used, while CHS is only 24bit.
368 Windows/Dos is said to take 1023/255/63 as nonrepresentable CHS */
371 chs
->cylinder
= 0xFF;
374 chs
->head
= (uint8_t)head
;
375 chs
->sector
= (uint8_t)( (sector
+1) | ((spos
>>8)<<6) );
376 chs
->cylinder
= (uint8_t)spos
;
380 static void init_mbr(BDRVVVFATState
* s
)
382 /* TODO: if the files mbr.img and bootsect.img exist, use them */
383 mbr_t
* real_mbr
=(mbr_t
*)s
->first_sectors
;
384 partition_t
* partition
=&(real_mbr
->partition
[0]);
387 memset(s
->first_sectors
,0,512);
389 /* Win NT Disk Signature */
390 real_mbr
->nt_id
= cpu_to_le32(0xbe1afdfa);
392 partition
->attributes
=0x80; /* bootable */
394 /* LBA is used when partition is outside the CHS geometry */
395 lba
= sector2CHS(s
->bs
, &partition
->start_CHS
, s
->first_sectors_number
-1);
396 lba
|= sector2CHS(s
->bs
, &partition
->end_CHS
, s
->sector_count
);
398 /*LBA partitions are identified only by start/length_sector_long not by CHS*/
399 partition
->start_sector_long
=cpu_to_le32(s
->first_sectors_number
-1);
400 partition
->length_sector_long
=cpu_to_le32(s
->sector_count
- s
->first_sectors_number
+1);
402 /* FAT12/FAT16/FAT32 */
403 /* DOS uses different types when partition is LBA,
404 probably to prevent older versions from using CHS on them */
405 partition
->fs_type
= s
->fat_type
==12 ? 0x1:
406 s
->fat_type
==16 ? (lba
?0xe:0x06):
407 /*fat_tyoe==32*/ (lba
?0xc:0x0b);
409 real_mbr
->magic
[0]=0x55; real_mbr
->magic
[1]=0xaa;
412 /* direntry functions */
414 /* dest is assumed to hold 258 bytes, and pads with 0xffff up to next multiple of 26 */
415 static inline int short2long_name(char* dest
,const char* src
)
419 for(i
=0;i
<129 && src
[i
];i
++) {
424 dest
[2*i
]=dest
[2*i
+1]=0;
425 for(i
=2*i
+2;(i
%26);i
++)
430 static inline direntry_t
* create_long_filename(BDRVVVFATState
* s
,const char* filename
)
433 int length
=short2long_name(buffer
,filename
),
434 number_of_entries
=(length
+25)/26,i
;
437 for(i
=0;i
<number_of_entries
;i
++) {
438 entry
=array_get_next(&(s
->directory
));
439 entry
->attributes
=0xf;
440 entry
->reserved
[0]=0;
442 entry
->name
[0]=(number_of_entries
-i
)|(i
==0?0x40:0);
444 for(i
=0;i
<26*number_of_entries
;i
++) {
446 if(offset
<10) offset
=1+offset
;
447 else if(offset
<22) offset
=14+offset
-10;
448 else offset
=28+offset
-22;
449 entry
=array_get(&(s
->directory
),s
->directory
.next
-1-(i
/26));
450 entry
->name
[offset
]=buffer
[i
];
452 return array_get(&(s
->directory
),s
->directory
.next
-number_of_entries
);
455 static char is_free(const direntry_t
* direntry
)
457 /* return direntry->name[0]==0 ; */
458 return direntry
->attributes
== 0 || direntry
->name
[0]==0xe5;
461 static char is_volume_label(const direntry_t
* direntry
)
463 return direntry
->attributes
== 0x28;
466 static char is_long_name(const direntry_t
* direntry
)
468 return direntry
->attributes
== 0xf;
471 static char is_short_name(const direntry_t
* direntry
)
473 return !is_volume_label(direntry
) && !is_long_name(direntry
)
474 && !is_free(direntry
);
477 static char is_directory(const direntry_t
* direntry
)
479 return direntry
->attributes
& 0x10 && direntry
->name
[0] != 0xe5;
482 static inline char is_dot(const direntry_t
* direntry
)
484 return is_short_name(direntry
) && direntry
->name
[0] == '.';
487 static char is_file(const direntry_t
* direntry
)
489 return is_short_name(direntry
) && !is_directory(direntry
);
492 static inline uint32_t begin_of_direntry(const direntry_t
* direntry
)
494 return le16_to_cpu(direntry
->begin
)|(le16_to_cpu(direntry
->begin_hi
)<<16);
497 static inline uint32_t filesize_of_direntry(const direntry_t
* direntry
)
499 return le32_to_cpu(direntry
->size
);
502 static void set_begin_of_direntry(direntry_t
* direntry
, uint32_t begin
)
504 direntry
->begin
= cpu_to_le16(begin
& 0xffff);
505 direntry
->begin_hi
= cpu_to_le16((begin
>> 16) & 0xffff);
510 static inline uint8_t fat_chksum(const direntry_t
* entry
)
516 chksum
=(((chksum
&0xfe)>>1)|((chksum
&0x01)?0x80:0))
517 +(unsigned char)entry
->name
[i
];
522 /* if return_time==0, this returns the fat_date, else the fat_time */
523 static uint16_t fat_datetime(time_t time
,int return_time
) {
526 t
=localtime(&time
); /* this is not thread safe */
530 localtime_r(&time
,t
);
533 return cpu_to_le16((t
->tm_sec
/2)|(t
->tm_min
<<5)|(t
->tm_hour
<<11));
534 return cpu_to_le16((t
->tm_mday
)|((t
->tm_mon
+1)<<5)|((t
->tm_year
-80)<<9));
537 static inline void fat_set(BDRVVVFATState
* s
,unsigned int cluster
,uint32_t value
)
539 if(s
->fat_type
==32) {
540 uint32_t* entry
=array_get(&(s
->fat
),cluster
);
541 *entry
=cpu_to_le32(value
);
542 } else if(s
->fat_type
==16) {
543 uint16_t* entry
=array_get(&(s
->fat
),cluster
);
544 *entry
=cpu_to_le16(value
&0xffff);
546 int offset
= (cluster
*3/2);
547 unsigned char* p
= array_get(&(s
->fat
), offset
);
551 p
[1] = (p
[1]&0xf0) | ((value
>>8)&0xf);
554 p
[0] = (p
[0]&0xf) | ((value
&0xf)<<4);
561 static inline uint32_t fat_get(BDRVVVFATState
* s
,unsigned int cluster
)
563 if(s
->fat_type
==32) {
564 uint32_t* entry
=array_get(&(s
->fat
),cluster
);
565 return le32_to_cpu(*entry
);
566 } else if(s
->fat_type
==16) {
567 uint16_t* entry
=array_get(&(s
->fat
),cluster
);
568 return le16_to_cpu(*entry
);
570 const uint8_t* x
=(uint8_t*)(s
->fat
.pointer
)+cluster
*3/2;
571 return ((x
[0]|(x
[1]<<8))>>(cluster
&1?4:0))&0x0fff;
575 static inline int fat_eof(BDRVVVFATState
* s
,uint32_t fat_entry
)
577 if(fat_entry
>s
->max_fat_value
-8)
582 static inline void init_fat(BDRVVVFATState
* s
)
584 if (s
->fat_type
== 12) {
585 array_init(&(s
->fat
),1);
586 array_ensure_allocated(&(s
->fat
),
587 s
->sectors_per_fat
* 0x200 * 3 / 2 - 1);
589 array_init(&(s
->fat
),(s
->fat_type
==32?4:2));
590 array_ensure_allocated(&(s
->fat
),
591 s
->sectors_per_fat
* 0x200 / s
->fat
.item_size
- 1);
593 memset(s
->fat
.pointer
,0,s
->fat
.size
);
595 switch(s
->fat_type
) {
596 case 12: s
->max_fat_value
=0xfff; break;
597 case 16: s
->max_fat_value
=0xffff; break;
598 case 32: s
->max_fat_value
=0x0fffffff; break;
599 default: s
->max_fat_value
=0; /* error... */
604 /* TODO: in create_short_filename, 0xe5->0x05 is not yet handled! */
605 /* TODO: in parse_short_filename, 0x05->0xe5 is not yet handled! */
606 static inline direntry_t
* create_short_and_long_name(BDRVVVFATState
* s
,
607 unsigned int directory_start
, const char* filename
, int is_dot
)
609 int i
,j
,long_index
=s
->directory
.next
;
611 direntry_t
* entry_long
=0;
614 entry
=array_get_next(&(s
->directory
));
615 memset(entry
->name
,0x20,11);
616 memcpy(entry
->name
,filename
,strlen(filename
));
620 entry_long
=create_long_filename(s
,filename
);
622 i
= strlen(filename
);
623 for(j
= i
- 1; j
>0 && filename
[j
]!='.';j
--);
629 entry
=array_get_next(&(s
->directory
));
630 memset(entry
->name
,0x20,11);
631 strncpy((char*)entry
->name
,filename
,i
);
634 for (i
= 0; i
< 3 && filename
[j
+1+i
]; i
++)
635 entry
->extension
[i
] = filename
[j
+1+i
];
637 /* upcase & remove unwanted characters */
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
]))
643 else if(entry
->name
[i
]>='a' && entry
->name
[i
]<='z')
644 entry
->name
[i
]+='A'-'a';
647 /* mangle duplicates */
649 direntry_t
* entry1
=array_get(&(s
->directory
),directory_start
);
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 */
658 /* use all 8 characters of name */
659 if(entry
->name
[7]==' ') {
661 for(j
=6;j
>0 && entry
->name
[j
]==' ';j
--)
665 /* increment number */
666 for(j
=7;j
>0 && entry
->name
[j
]=='9';j
--)
669 if(entry
->name
[j
]<'0' || entry
->name
[j
]>'9')
676 /* calculate checksum; propagate to long name */
678 uint8_t chksum
=fat_chksum(entry
);
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
;
692 * Read a directory. (the index of the corresponding mapping must be passed).
694 static int read_directory(BDRVVVFATState
* s
, int mapping_index
)
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
) : 0);
703 int first_cluster_of_parent
= parent_mapping
? parent_mapping
->begin
: -1;
705 DIR* dir
=opendir(dirname
);
706 struct dirent
* entry
;
709 assert(mapping
->mode
& MODE_DIRECTORY
);
712 mapping
->end
= mapping
->begin
;
716 i
= mapping
->info
.dir
.first_dir_index
=
717 first_cluster
== 0 ? 0 : s
->directory
.next
;
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
);
723 direntry_t
* direntry
;
725 int is_dot
=!strcmp(entry
->d_name
,".");
726 int is_dotdot
=!strcmp(entry
->d_name
,"..");
728 if(first_cluster
== 0 && (is_dotdot
|| is_dot
))
731 buffer
=(char*)malloc(length
);
733 snprintf(buffer
,length
,"%s/%s",dirname
,entry
->d_name
);
735 if(stat(buffer
,&st
)<0) {
740 /* create directory entry for this file */
741 direntry
=create_short_and_long_name(s
, i
, entry
->d_name
,
742 is_dot
|| is_dotdot
);
743 direntry
->attributes
=(S_ISDIR(st
.st_mode
)?0x10:0x20);
744 direntry
->reserved
[0]=direntry
->reserved
[1]=0;
745 direntry
->ctime
=fat_datetime(st
.st_ctime
,1);
746 direntry
->cdate
=fat_datetime(st
.st_ctime
,0);
747 direntry
->adate
=fat_datetime(st
.st_atime
,0);
748 direntry
->begin_hi
=0;
749 direntry
->mtime
=fat_datetime(st
.st_mtime
,1);
750 direntry
->mdate
=fat_datetime(st
.st_mtime
,0);
752 set_begin_of_direntry(direntry
, first_cluster_of_parent
);
754 set_begin_of_direntry(direntry
, first_cluster
);
756 direntry
->begin
=0; /* do that later */
757 if (st
.st_size
> 0x7fffffff) {
758 fprintf(stderr
, "File %s is larger than 2GB\n", buffer
);
762 direntry
->size
=cpu_to_le32(S_ISDIR(st
.st_mode
)?0:st
.st_size
);
764 /* create mapping for this file */
765 if(!is_dot
&& !is_dotdot
&& (S_ISDIR(st
.st_mode
) || st
.st_size
)) {
766 s
->current_mapping
=(mapping_t
*)array_get_next(&(s
->mapping
));
767 s
->current_mapping
->begin
=0;
768 s
->current_mapping
->end
=st
.st_size
;
770 * we get the direntry of the most recent direntry, which
771 * contains the short name and all the relevant information.
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
=
780 s
->current_mapping
->mode
= MODE_UNDEFINED
;
781 s
->current_mapping
->info
.file
.offset
= 0;
783 s
->current_mapping
->path
=buffer
;
784 s
->current_mapping
->read_only
=
785 (st
.st_mode
& (S_IWUSR
| S_IWGRP
| S_IWOTH
)) == 0;
790 /* fill with zeroes up to the end of the cluster */
791 while(s
->directory
.next
%(0x10*s
->sectors_per_cluster
)) {
792 direntry_t
* direntry
=array_get_next(&(s
->directory
));
793 memset(direntry
,0,sizeof(direntry_t
));
796 /* TODO: if there are more entries, bootsector has to be adjusted! */
797 #define ROOT_ENTRIES (0x02 * 0x10 * s->sectors_per_cluster)
798 if (mapping_index
== 0 && s
->directory
.next
< ROOT_ENTRIES
) {
800 int cur
= s
->directory
.next
;
801 array_ensure_allocated(&(s
->directory
), ROOT_ENTRIES
- 1);
802 memset(array_get(&(s
->directory
), cur
), 0,
803 (ROOT_ENTRIES
- cur
) * sizeof(direntry_t
));
806 /* reget the mapping, since s->mapping was possibly realloc()ed */
807 mapping
= (mapping_t
*)array_get(&(s
->mapping
), mapping_index
);
808 first_cluster
+= (s
->directory
.next
- mapping
->info
.dir
.first_dir_index
)
809 * 0x20 / s
->cluster_size
;
810 mapping
->end
= first_cluster
;
812 direntry
= (direntry_t
*)array_get(&(s
->directory
), mapping
->dir_index
);
813 set_begin_of_direntry(direntry
, mapping
->begin
);
818 static inline uint32_t sector2cluster(BDRVVVFATState
* s
,off_t sector_num
)
820 return (sector_num
-s
->faked_sectors
)/s
->sectors_per_cluster
;
823 static inline off_t
cluster2sector(BDRVVVFATState
* s
, uint32_t cluster_num
)
825 return s
->faked_sectors
+ s
->sectors_per_cluster
* cluster_num
;
828 static inline uint32_t sector_offset_in_cluster(BDRVVVFATState
* s
,off_t sector_num
)
830 return (sector_num
-s
->first_sectors_number
-2*s
->sectors_per_fat
)%s
->sectors_per_cluster
;
834 static direntry_t
* get_direntry_for_mapping(BDRVVVFATState
* s
,mapping_t
* mapping
)
836 if(mapping
->mode
==MODE_UNDEFINED
)
838 return (direntry_t
*)(s
->directory
.pointer
+sizeof(direntry_t
)*mapping
->dir_index
);
842 static int init_directories(BDRVVVFATState
* s
,
845 bootsector_t
* bootsector
;
848 unsigned int cluster
;
850 memset(&(s
->first_sectors
[0]),0,0x40*0x200);
852 s
->cluster_size
=s
->sectors_per_cluster
*0x200;
853 s
->cluster_buffer
=malloc(s
->cluster_size
);
854 assert(s
->cluster_buffer
);
857 * The formula: sc = spf+1+spf*spc*(512*8/fat_type),
858 * where sc is sector_count,
859 * spf is sectors_per_fat,
860 * spc is sectors_per_clusters, and
861 * fat_type = 12, 16 or 32.
863 i
= 1+s
->sectors_per_cluster
*0x200*8/s
->fat_type
;
864 s
->sectors_per_fat
=(s
->sector_count
+i
)/i
; /* round up */
866 array_init(&(s
->mapping
),sizeof(mapping_t
));
867 array_init(&(s
->directory
),sizeof(direntry_t
));
869 /* add volume label */
871 direntry_t
* entry
=array_get_next(&(s
->directory
));
872 entry
->attributes
=0x28; /* archive | volume label */
873 snprintf((char*)entry
->name
,11,"QEMU VVFAT");
876 /* Now build FAT, and write back information into directory */
879 s
->faked_sectors
=s
->first_sectors_number
+s
->sectors_per_fat
*2;
880 s
->cluster_count
=sector2cluster(s
, s
->sector_count
);
882 mapping
= array_get_next(&(s
->mapping
));
884 mapping
->dir_index
= 0;
885 mapping
->info
.dir
.parent_mapping_index
= -1;
886 mapping
->first_mapping_index
= -1;
887 mapping
->path
= strdup(dirname
);
888 i
= strlen(mapping
->path
);
889 if (i
> 0 && mapping
->path
[i
- 1] == '/')
890 mapping
->path
[i
- 1] = '\0';
891 mapping
->mode
= MODE_DIRECTORY
;
892 mapping
->read_only
= 0;
893 s
->path
= mapping
->path
;
895 for (i
= 0, cluster
= 0; i
< s
->mapping
.next
; i
++) {
897 /* MS-DOS expects the FAT to be 0 for the root directory
898 * (except for the media byte). */
899 /* LATER TODO: still true for FAT32? */
900 int fix_fat
= (i
!= 0);
901 mapping
= array_get(&(s
->mapping
), i
);
903 if (mapping
->mode
& MODE_DIRECTORY
) {
904 mapping
->begin
= cluster
;
905 if(read_directory(s
, i
)) {
906 fprintf(stderr
, "Could not read directory %s\n",
910 mapping
= array_get(&(s
->mapping
), i
);
912 assert(mapping
->mode
== MODE_UNDEFINED
);
913 mapping
->mode
=MODE_NORMAL
;
914 mapping
->begin
= cluster
;
915 if (mapping
->end
> 0) {
916 direntry_t
* direntry
= array_get(&(s
->directory
),
919 mapping
->end
= cluster
+ 1 + (mapping
->end
-1)/s
->cluster_size
;
920 set_begin_of_direntry(direntry
, mapping
->begin
);
922 mapping
->end
= cluster
+ 1;
927 assert(mapping
->begin
< mapping
->end
);
929 /* fix fat for entry */
931 for(j
= mapping
->begin
; j
< mapping
->end
- 1; j
++)
933 fat_set(s
, mapping
->end
- 1, s
->max_fat_value
);
936 /* next free cluster */
937 cluster
= mapping
->end
;
939 if(cluster
> s
->cluster_count
) {
940 fprintf(stderr
,"Directory does not fit in FAT%d\n",s
->fat_type
);
945 mapping
= array_get(&(s
->mapping
), 0);
946 s
->sectors_of_root_directory
= mapping
->end
* s
->sectors_per_cluster
;
947 s
->last_cluster_of_root_directory
= mapping
->end
;
949 /* the FAT signature */
950 fat_set(s
,0,s
->max_fat_value
);
951 fat_set(s
,1,s
->max_fat_value
);
953 s
->current_mapping
= NULL
;
955 bootsector
=(bootsector_t
*)(s
->first_sectors
+(s
->first_sectors_number
-1)*0x200);
956 bootsector
->jump
[0]=0xeb;
957 bootsector
->jump
[1]=0x3e;
958 bootsector
->jump
[2]=0x90;
959 memcpy(bootsector
->name
,"QEMU ",8);
960 bootsector
->sector_size
=cpu_to_le16(0x200);
961 bootsector
->sectors_per_cluster
=s
->sectors_per_cluster
;
962 bootsector
->reserved_sectors
=cpu_to_le16(1);
963 bootsector
->number_of_fats
=0x2; /* number of FATs */
964 bootsector
->root_entries
=cpu_to_le16(s
->sectors_of_root_directory
*0x10);
965 bootsector
->total_sectors16
=s
->sector_count
>0xffff?0:cpu_to_le16(s
->sector_count
);
966 bootsector
->media_type
=(s
->fat_type
!=12?0xf8:s
->sector_count
==5760?0xf9:0xf8); /* media descriptor */
967 s
->fat
.pointer
[0] = bootsector
->media_type
;
968 bootsector
->sectors_per_fat
=cpu_to_le16(s
->sectors_per_fat
);
969 bootsector
->sectors_per_track
=cpu_to_le16(s
->bs
->secs
);
970 bootsector
->number_of_heads
=cpu_to_le16(s
->bs
->heads
);
971 bootsector
->hidden_sectors
=cpu_to_le32(s
->first_sectors_number
==1?0:0x3f);
972 bootsector
->total_sectors
=cpu_to_le32(s
->sector_count
>0xffff?s
->sector_count
:0);
974 /* LATER TODO: if FAT32, this is wrong */
975 bootsector
->u
.fat16
.drive_number
=s
->fat_type
==12?0:0x80; /* assume this is hda (TODO) */
976 bootsector
->u
.fat16
.current_head
=0;
977 bootsector
->u
.fat16
.signature
=0x29;
978 bootsector
->u
.fat16
.id
=cpu_to_le32(0xfabe1afd);
980 memcpy(bootsector
->u
.fat16
.volume_label
,"QEMU VVFAT ",11);
981 memcpy(bootsector
->fat_type
,(s
->fat_type
==12?"FAT12 ":s
->fat_type
==16?"FAT16 ":"FAT32 "),8);
982 bootsector
->magic
[0]=0x55; bootsector
->magic
[1]=0xaa;
988 static BDRVVVFATState
*vvv
= NULL
;
991 static int enable_write_target(BDRVVVFATState
*s
);
992 static int is_consistent(BDRVVVFATState
*s
);
994 static int vvfat_open(BlockDriverState
*bs
, const char* dirname
, int flags
)
996 BDRVVVFATState
*s
= bs
->opaque
;
1004 DLOG(if (stderr
== NULL
) {
1005 stderr
= fopen("vvfat.log", "a");
1006 setbuf(stderr
, NULL
);
1012 /* LATER TODO: if FAT32, adjust */
1013 s
->sectors_per_cluster
=0x10;
1015 bs
->cyls
=1024; bs
->heads
=16; bs
->secs
=63;
1017 s
->current_cluster
=0xffffffff;
1019 s
->first_sectors_number
=0x40;
1020 /* read only is the default for safety */
1022 s
->qcow
= s
->write_target
= NULL
;
1023 s
->qcow_filename
= NULL
;
1025 s
->downcase_short_names
= 1;
1027 if (!strstart(dirname
, "fat:", NULL
))
1030 if (strstr(dirname
, ":floppy:")) {
1033 s
->first_sectors_number
= 1;
1034 s
->sectors_per_cluster
=2;
1035 bs
->cyls
= 80; bs
->heads
= 2; bs
->secs
= 36;
1038 s
->sector_count
=bs
->cyls
*bs
->heads
*bs
->secs
;
1040 if (strstr(dirname
, ":32:")) {
1041 fprintf(stderr
, "Big fat greek warning: FAT32 has not been tested. You are welcome to do so!\n");
1043 } else if (strstr(dirname
, ":16:")) {
1045 } else if (strstr(dirname
, ":12:")) {
1047 s
->sector_count
=2880;
1050 if (strstr(dirname
, ":rw:")) {
1051 if (enable_write_target(s
))
1056 i
= strrchr(dirname
, ':') - dirname
;
1058 if (dirname
[i
-2] == ':' && isalpha(dirname
[i
-1]))
1059 /* workaround for DOS drive names */
1064 bs
->total_sectors
=bs
->cyls
*bs
->heads
*bs
->secs
;
1066 if(init_directories(s
, dirname
))
1069 s
->sector_count
= s
->faked_sectors
+ s
->sectors_per_cluster
*s
->cluster_count
;
1071 if(s
->first_sectors_number
==0x40)
1074 /* for some reason or other, MS-DOS does not like to know about CHS... */
1076 bs
->heads
= bs
->cyls
= bs
->secs
= 0;
1078 // assert(is_consistent(s));
1082 static inline void vvfat_close_current_file(BDRVVVFATState
*s
)
1084 if(s
->current_mapping
) {
1085 s
->current_mapping
= NULL
;
1086 if (s
->current_fd
) {
1087 close(s
->current_fd
);
1091 s
->current_cluster
= -1;
1094 /* mappings between index1 and index2-1 are supposed to be ordered
1095 * return value is the index of the last mapping for which end>cluster_num
1097 static inline int find_mapping_for_cluster_aux(BDRVVVFATState
* s
,int cluster_num
,int index1
,int index2
)
1099 int index3
=index1
+1;
1102 index3
=(index1
+index2
)/2;
1103 mapping
=array_get(&(s
->mapping
),index3
);
1104 assert(mapping
->begin
< mapping
->end
);
1105 if(mapping
->begin
>=cluster_num
) {
1106 assert(index2
!=index3
|| index2
==0);
1112 return mapping
->end
<=cluster_num
? index2
: index1
;
1115 assert(index1
<=index2
);
1116 DLOG(mapping
=array_get(&(s
->mapping
),index1
);
1117 assert(mapping
->begin
<=cluster_num
);
1118 assert(index2
>= s
->mapping
.next
||
1119 ((mapping
= array_get(&(s
->mapping
),index2
)) &&
1120 mapping
->end
>cluster_num
)));
1124 static inline mapping_t
* find_mapping_for_cluster(BDRVVVFATState
* s
,int cluster_num
)
1126 int index
=find_mapping_for_cluster_aux(s
,cluster_num
,0,s
->mapping
.next
);
1128 if(index
>=s
->mapping
.next
)
1130 mapping
=array_get(&(s
->mapping
),index
);
1131 if(mapping
->begin
>cluster_num
)
1133 assert(mapping
->begin
<=cluster_num
&& mapping
->end
>cluster_num
);
1138 * This function simply compares path == mapping->path. Since the mappings
1139 * are sorted by cluster, this is expensive: O(n).
1141 static inline mapping_t
* find_mapping_for_path(BDRVVVFATState
* s
,
1146 for (i
= 0; i
< s
->mapping
.next
; i
++) {
1147 mapping_t
* mapping
= array_get(&(s
->mapping
), i
);
1148 if (mapping
->first_mapping_index
< 0 &&
1149 !strcmp(path
, mapping
->path
))
1156 static int open_file(BDRVVVFATState
* s
,mapping_t
* mapping
)
1160 if(!s
->current_mapping
||
1161 strcmp(s
->current_mapping
->path
,mapping
->path
)) {
1163 int fd
= open(mapping
->path
, O_RDONLY
| O_BINARY
| O_LARGEFILE
);
1166 vvfat_close_current_file(s
);
1168 s
->current_mapping
= mapping
;
1173 static inline int read_cluster(BDRVVVFATState
*s
,int cluster_num
)
1175 if(s
->current_cluster
!= cluster_num
) {
1178 assert(!s
->current_mapping
|| s
->current_fd
|| (s
->current_mapping
->mode
& MODE_DIRECTORY
));
1179 if(!s
->current_mapping
1180 || s
->current_mapping
->begin
>cluster_num
1181 || s
->current_mapping
->end
<=cluster_num
) {
1182 /* binary search of mappings for file */
1183 mapping_t
* mapping
=find_mapping_for_cluster(s
,cluster_num
);
1185 assert(!mapping
|| (cluster_num
>=mapping
->begin
&& cluster_num
<mapping
->end
));
1187 if (mapping
&& mapping
->mode
& MODE_DIRECTORY
) {
1188 vvfat_close_current_file(s
);
1189 s
->current_mapping
= mapping
;
1190 read_cluster_directory
:
1191 offset
= s
->cluster_size
*(cluster_num
-s
->current_mapping
->begin
);
1192 s
->cluster
= (unsigned char*)s
->directory
.pointer
+offset
1193 + 0x20*s
->current_mapping
->info
.dir
.first_dir_index
;
1194 assert(((s
->cluster
-(unsigned char*)s
->directory
.pointer
)%s
->cluster_size
)==0);
1195 assert((char*)s
->cluster
+s
->cluster_size
<= s
->directory
.pointer
+s
->directory
.next
*s
->directory
.item_size
);
1196 s
->current_cluster
= cluster_num
;
1200 if(open_file(s
,mapping
))
1202 } else if (s
->current_mapping
->mode
& MODE_DIRECTORY
)
1203 goto read_cluster_directory
;
1205 assert(s
->current_fd
);
1207 offset
=s
->cluster_size
*(cluster_num
-s
->current_mapping
->begin
)+s
->current_mapping
->info
.file
.offset
;
1208 if(lseek(s
->current_fd
, offset
, SEEK_SET
)!=offset
)
1210 s
->cluster
=s
->cluster_buffer
;
1211 result
=read(s
->current_fd
,s
->cluster
,s
->cluster_size
);
1213 s
->current_cluster
= -1;
1216 s
->current_cluster
= cluster_num
;
1222 static void hexdump(const void* address
, uint32_t len
)
1224 const unsigned char* p
= address
;
1227 for (i
= 0; i
< len
; i
+= 16) {
1228 for (j
= 0; j
< 16 && i
+ j
< len
; j
++)
1229 fprintf(stderr
, "%02x ", p
[i
+ j
]);
1231 fprintf(stderr
, " ");
1232 fprintf(stderr
, " ");
1233 for (j
= 0; j
< 16 && i
+ j
< len
; j
++)
1234 fprintf(stderr
, "%c", (p
[i
+ j
] < ' ' || p
[i
+ j
] > 0x7f) ? '.' : p
[i
+ j
]);
1235 fprintf(stderr
, "\n");
1239 static void print_direntry(const direntry_t
* direntry
)
1244 fprintf(stderr
, "direntry 0x%x: ", (int)direntry
);
1247 if(is_long_name(direntry
)) {
1248 unsigned char* c
=(unsigned char*)direntry
;
1250 for(i
=1;i
<11 && c
[i
] && c
[i
]!=0xff;i
+=2)
1251 #define ADD_CHAR(c) {buffer[j] = (c); if (buffer[j] < ' ') buffer[j] = '°'; j++;}
1253 for(i
=14;i
<26 && c
[i
] && c
[i
]!=0xff;i
+=2)
1255 for(i
=28;i
<32 && c
[i
] && c
[i
]!=0xff;i
+=2)
1258 fprintf(stderr
, "%s\n", buffer
);
1262 ADD_CHAR(direntry
->name
[i
]);
1264 fprintf(stderr
,"%s attributes=0x%02x begin=%d size=%d\n",
1266 direntry
->attributes
,
1267 begin_of_direntry(direntry
),le32_to_cpu(direntry
->size
));
1271 static void print_mapping(const mapping_t
* mapping
)
1273 fprintf(stderr
, "mapping (0x%x): begin, end = %d, %d, dir_index = %d, first_mapping_index = %d, name = %s, mode = 0x%x, " , (int)mapping
, mapping
->begin
, mapping
->end
, mapping
->dir_index
, mapping
->first_mapping_index
, mapping
->path
, mapping
->mode
);
1274 if (mapping
->mode
& MODE_DIRECTORY
)
1275 fprintf(stderr
, "parent_mapping_index = %d, first_dir_index = %d\n", mapping
->info
.dir
.parent_mapping_index
, mapping
->info
.dir
.first_dir_index
);
1277 fprintf(stderr
, "offset = %d\n", mapping
->info
.file
.offset
);
1281 static int vvfat_read(BlockDriverState
*bs
, int64_t sector_num
,
1282 uint8_t *buf
, int nb_sectors
)
1284 BDRVVVFATState
*s
= bs
->opaque
;
1287 for(i
=0;i
<nb_sectors
;i
++,sector_num
++) {
1288 if (sector_num
>= s
->sector_count
)
1292 if (s
->qcow
->drv
->bdrv_is_allocated(s
->qcow
,
1293 sector_num
, nb_sectors
-i
, &n
)) {
1294 DLOG(fprintf(stderr
, "sectors %d+%d allocated\n", (int)sector_num
, n
));
1295 if (s
->qcow
->drv
->bdrv_read(s
->qcow
, sector_num
, buf
+i
*0x200, n
))
1298 sector_num
+= n
- 1;
1301 DLOG(fprintf(stderr
, "sector %d not allocated\n", (int)sector_num
));
1303 if(sector_num
<s
->faked_sectors
) {
1304 if(sector_num
<s
->first_sectors_number
)
1305 memcpy(buf
+i
*0x200,&(s
->first_sectors
[sector_num
*0x200]),0x200);
1306 else if(sector_num
-s
->first_sectors_number
<s
->sectors_per_fat
)
1307 memcpy(buf
+i
*0x200,&(s
->fat
.pointer
[(sector_num
-s
->first_sectors_number
)*0x200]),0x200);
1308 else if(sector_num
-s
->first_sectors_number
-s
->sectors_per_fat
<s
->sectors_per_fat
)
1309 memcpy(buf
+i
*0x200,&(s
->fat
.pointer
[(sector_num
-s
->first_sectors_number
-s
->sectors_per_fat
)*0x200]),0x200);
1311 uint32_t sector
=sector_num
-s
->faked_sectors
,
1312 sector_offset_in_cluster
=(sector
%s
->sectors_per_cluster
),
1313 cluster_num
=sector
/s
->sectors_per_cluster
;
1314 if(read_cluster(s
, cluster_num
) != 0) {
1315 /* LATER TODO: strict: return -1; */
1316 memset(buf
+i
*0x200,0,0x200);
1319 memcpy(buf
+i
*0x200,s
->cluster
+sector_offset_in_cluster
*0x200,0x200);
1325 /* LATER TODO: statify all functions */
1328 * Idea of the write support (use snapshot):
1330 * 1. check if all data is consistent, recording renames, modifications,
1331 * new files and directories (in s->commits).
1333 * 2. if the data is not consistent, stop committing
1335 * 3. handle renames, and create new files and directories (do not yet
1336 * write their contents)
1338 * 4. walk the directories, fixing the mapping and direntries, and marking
1339 * the handled mappings as not deleted
1341 * 5. commit the contents of the files
1343 * 6. handle deleted files and directories
1347 typedef struct commit_t
{
1350 struct { uint32_t cluster
; } rename
;
1351 struct { int dir_index
; uint32_t modified_offset
; } writeout
;
1352 struct { uint32_t first_cluster
; } new_file
;
1353 struct { uint32_t cluster
; } mkdir
;
1355 /* DELETEs and RMDIRs are handled differently: see handle_deletes() */
1357 ACTION_RENAME
, ACTION_WRITEOUT
, ACTION_NEW_FILE
, ACTION_MKDIR
1361 static void clear_commits(BDRVVVFATState
* s
)
1364 DLOG(fprintf(stderr
, "clear_commits (%d commits)\n", s
->commits
.next
));
1365 for (i
= 0; i
< s
->commits
.next
; i
++) {
1366 commit_t
* commit
= array_get(&(s
->commits
), i
);
1367 assert(commit
->path
|| commit
->action
== ACTION_WRITEOUT
);
1368 if (commit
->action
!= ACTION_WRITEOUT
) {
1369 assert(commit
->path
);
1372 assert(commit
->path
== NULL
);
1374 s
->commits
.next
= 0;
1377 static void schedule_rename(BDRVVVFATState
* s
,
1378 uint32_t cluster
, char* new_path
)
1380 commit_t
* commit
= array_get_next(&(s
->commits
));
1381 commit
->path
= new_path
;
1382 commit
->param
.rename
.cluster
= cluster
;
1383 commit
->action
= ACTION_RENAME
;
1386 static void schedule_writeout(BDRVVVFATState
* s
,
1387 int dir_index
, uint32_t modified_offset
)
1389 commit_t
* commit
= array_get_next(&(s
->commits
));
1390 commit
->path
= NULL
;
1391 commit
->param
.writeout
.dir_index
= dir_index
;
1392 commit
->param
.writeout
.modified_offset
= modified_offset
;
1393 commit
->action
= ACTION_WRITEOUT
;
1396 static void schedule_new_file(BDRVVVFATState
* s
,
1397 char* path
, uint32_t first_cluster
)
1399 commit_t
* commit
= array_get_next(&(s
->commits
));
1400 commit
->path
= path
;
1401 commit
->param
.new_file
.first_cluster
= first_cluster
;
1402 commit
->action
= ACTION_NEW_FILE
;
1405 static void schedule_mkdir(BDRVVVFATState
* s
, uint32_t cluster
, char* path
)
1407 commit_t
* commit
= array_get_next(&(s
->commits
));
1408 commit
->path
= path
;
1409 commit
->param
.mkdir
.cluster
= cluster
;
1410 commit
->action
= ACTION_MKDIR
;
1414 unsigned char name
[1024];
1416 int sequence_number
;
1419 static void lfn_init(long_file_name
* lfn
)
1421 lfn
->sequence_number
= lfn
->len
= 0;
1422 lfn
->checksum
= 0x100;
1425 /* return 0 if parsed successfully, > 0 if no long name, < 0 if error */
1426 static int parse_long_name(long_file_name
* lfn
,
1427 const direntry_t
* direntry
)
1430 const unsigned char* pointer
= (const unsigned char*)direntry
;
1432 if (!is_long_name(direntry
))
1435 if (pointer
[0] & 0x40) {
1436 lfn
->sequence_number
= pointer
[0] & 0x3f;
1437 lfn
->checksum
= pointer
[13];
1439 } else if ((pointer
[0] & 0x3f) != --lfn
->sequence_number
)
1441 else if (pointer
[13] != lfn
->checksum
)
1443 else if (pointer
[12] || pointer
[26] || pointer
[27])
1446 offset
= 13 * (lfn
->sequence_number
- 1);
1447 for (i
= 0, j
= 1; i
< 13; i
++, j
+=2) {
1453 if (pointer
[j
+1] == 0)
1454 lfn
->name
[offset
+ i
] = pointer
[j
];
1455 else if (pointer
[j
+1] != 0xff || (pointer
[0] & 0x40) == 0)
1458 lfn
->name
[offset
+ i
] = 0;
1461 if (pointer
[0] & 0x40)
1462 lfn
->len
= offset
+ strlen((char*)lfn
->name
+ offset
);
1467 /* returns 0 if successful, >0 if no short_name, and <0 on error */
1468 static int parse_short_name(BDRVVVFATState
* s
,
1469 long_file_name
* lfn
, direntry_t
* direntry
)
1473 if (!is_short_name(direntry
))
1476 for (j
= 7; j
>= 0 && direntry
->name
[j
] == ' '; j
--);
1477 for (i
= 0; i
<= j
; i
++) {
1478 if (direntry
->name
[i
] <= ' ' || direntry
->name
[i
] > 0x7f)
1480 else if (s
->downcase_short_names
)
1481 lfn
->name
[i
] = tolower(direntry
->name
[i
]);
1483 lfn
->name
[i
] = direntry
->name
[i
];
1486 for (j
= 2; j
>= 0 && direntry
->extension
[j
] == ' '; j
--);
1488 lfn
->name
[i
++] = '.';
1489 lfn
->name
[i
+ j
+ 1] = '\0';
1490 for (;j
>= 0; j
--) {
1491 if (direntry
->extension
[j
] <= ' ' || direntry
->extension
[j
] > 0x7f)
1493 else if (s
->downcase_short_names
)
1494 lfn
->name
[i
+ j
] = tolower(direntry
->extension
[j
]);
1496 lfn
->name
[i
+ j
] = direntry
->extension
[j
];
1499 lfn
->name
[i
+ j
+ 1] = '\0';
1501 lfn
->len
= strlen((char*)lfn
->name
);
1506 static inline uint32_t modified_fat_get(BDRVVVFATState
* s
,
1507 unsigned int cluster
)
1509 if (cluster
< s
->last_cluster_of_root_directory
) {
1510 if (cluster
+ 1 == s
->last_cluster_of_root_directory
)
1511 return s
->max_fat_value
;
1516 if (s
->fat_type
==32) {
1517 uint32_t* entry
=((uint32_t*)s
->fat2
)+cluster
;
1518 return le32_to_cpu(*entry
);
1519 } else if (s
->fat_type
==16) {
1520 uint16_t* entry
=((uint16_t*)s
->fat2
)+cluster
;
1521 return le16_to_cpu(*entry
);
1523 const uint8_t* x
=s
->fat2
+cluster
*3/2;
1524 return ((x
[0]|(x
[1]<<8))>>(cluster
&1?4:0))&0x0fff;
1528 static inline int cluster_was_modified(BDRVVVFATState
* s
, uint32_t cluster_num
)
1530 int was_modified
= 0;
1533 if (s
->qcow
== NULL
)
1536 for (i
= 0; !was_modified
&& i
< s
->sectors_per_cluster
; i
++)
1537 was_modified
= s
->qcow
->drv
->bdrv_is_allocated(s
->qcow
,
1538 cluster2sector(s
, cluster_num
) + i
, 1, &dummy
);
1540 return was_modified
;
1543 static const char* get_basename(const char* path
)
1545 char* basename
= strrchr(path
, '/');
1546 if (basename
== NULL
)
1549 return basename
+ 1; /* strip '/' */
1553 * The array s->used_clusters holds the states of the clusters. If it is
1554 * part of a file, it has bit 2 set, in case of a directory, bit 1. If it
1555 * was modified, bit 3 is set.
1556 * If any cluster is allocated, but not part of a file or directory, this
1557 * driver refuses to commit.
1560 USED_DIRECTORY
= 1, USED_FILE
= 2, USED_ANY
= 3, USED_ALLOCATED
= 4
1564 * get_cluster_count_for_direntry() not only determines how many clusters
1565 * are occupied by direntry, but also if it was renamed or modified.
1567 * A file is thought to be renamed *only* if there already was a file with
1568 * exactly the same first cluster, but a different name.
1570 * Further, the files/directories handled by this function are
1571 * assumed to be *not* deleted (and *only* those).
1573 static uint32_t get_cluster_count_for_direntry(BDRVVVFATState
* s
,
1574 direntry_t
* direntry
, const char* path
)
1577 * This is a little bit tricky:
1578 * IF the guest OS just inserts a cluster into the file chain,
1579 * and leaves the rest alone, (i.e. the original file had clusters
1580 * 15 -> 16, but now has 15 -> 32 -> 16), then the following happens:
1582 * - do_commit will write the cluster into the file at the given
1585 * - the cluster which is overwritten should be moved to a later
1586 * position in the file.
1588 * I am not aware that any OS does something as braindead, but this
1589 * situation could happen anyway when not committing for a long time.
1590 * Just to be sure that this does not bite us, detect it, and copy the
1591 * contents of the clusters to-be-overwritten into the qcow.
1594 int was_modified
= 0;
1597 uint32_t cluster_num
= begin_of_direntry(direntry
);
1598 uint32_t offset
= 0;
1599 int first_mapping_index
= -1;
1600 mapping_t
* mapping
= NULL
;
1601 const char* basename2
= NULL
;
1603 vvfat_close_current_file(s
);
1605 /* the root directory */
1606 if (cluster_num
== 0)
1611 basename2
= get_basename(path
);
1613 mapping
= find_mapping_for_cluster(s
, cluster_num
);
1616 const char* basename
;
1618 assert(mapping
->mode
& MODE_DELETED
);
1619 mapping
->mode
&= ~MODE_DELETED
;
1621 basename
= get_basename(mapping
->path
);
1623 assert(mapping
->mode
& MODE_NORMAL
);
1626 if (strcmp(basename
, basename2
))
1627 schedule_rename(s
, cluster_num
, strdup(path
));
1628 } else if (is_file(direntry
))
1630 schedule_new_file(s
, strdup(path
), cluster_num
);
1639 if (!copy_it
&& cluster_was_modified(s
, cluster_num
)) {
1640 if (mapping
== NULL
||
1641 mapping
->begin
> cluster_num
||
1642 mapping
->end
<= cluster_num
)
1643 mapping
= find_mapping_for_cluster(s
, cluster_num
);
1647 (mapping
->mode
& MODE_DIRECTORY
) == 0) {
1649 /* was modified in qcow */
1650 if (offset
!= mapping
->info
.file
.offset
+ s
->cluster_size
1651 * (cluster_num
- mapping
->begin
)) {
1652 /* offset of this cluster in file chain has changed */
1655 } else if (offset
== 0) {
1656 const char* basename
= get_basename(mapping
->path
);
1658 if (strcmp(basename
, basename2
))
1660 first_mapping_index
= array_index(&(s
->mapping
), mapping
);
1663 if (mapping
->first_mapping_index
!= first_mapping_index
1664 && mapping
->info
.file
.offset
> 0) {
1669 /* need to write out? */
1670 if (!was_modified
&& is_file(direntry
)) {
1672 schedule_writeout(s
, mapping
->dir_index
, offset
);
1680 * This is horribly inefficient, but that is okay, since
1681 * it is rarely executed, if at all.
1683 int64_t offset
= cluster2sector(s
, cluster_num
);
1685 vvfat_close_current_file(s
);
1686 for (i
= 0; i
< s
->sectors_per_cluster
; i
++)
1687 if (!s
->qcow
->drv
->bdrv_is_allocated(s
->qcow
,
1688 offset
+ i
, 1, &dummy
)) {
1689 if (vvfat_read(s
->bs
,
1690 offset
, s
->cluster_buffer
, 1))
1692 if (s
->qcow
->drv
->bdrv_write(s
->qcow
,
1693 offset
, s
->cluster_buffer
, 1))
1700 if (s
->used_clusters
[cluster_num
] & USED_ANY
)
1702 s
->used_clusters
[cluster_num
] = USED_FILE
;
1704 cluster_num
= modified_fat_get(s
, cluster_num
);
1706 if (fat_eof(s
, cluster_num
))
1708 else if (cluster_num
< 2 || cluster_num
> s
->max_fat_value
- 16)
1711 offset
+= s
->cluster_size
;
1716 * This function looks at the modified data (qcow).
1717 * It returns 0 upon inconsistency or error, and the number of clusters
1718 * used by the directory, its subdirectories and their files.
1720 static int check_directory_consistency(BDRVVVFATState
*s
,
1721 int cluster_num
, const char* path
)
1724 unsigned char* cluster
= malloc(s
->cluster_size
);
1725 direntry_t
* direntries
= (direntry_t
*)cluster
;
1726 mapping_t
* mapping
= find_mapping_for_cluster(s
, cluster_num
);
1729 int path_len
= strlen(path
);
1730 char path2
[PATH_MAX
];
1732 assert(path_len
< PATH_MAX
); /* len was tested before! */
1733 strcpy(path2
, path
);
1734 path2
[path_len
] = '/';
1735 path2
[path_len
+ 1] = '\0';
1738 const char* basename
= get_basename(mapping
->path
);
1739 const char* basename2
= get_basename(path
);
1741 assert(mapping
->mode
& MODE_DIRECTORY
);
1743 assert(mapping
->mode
& MODE_DELETED
);
1744 mapping
->mode
&= ~MODE_DELETED
;
1746 if (strcmp(basename
, basename2
))
1747 schedule_rename(s
, cluster_num
, strdup(path
));
1750 schedule_mkdir(s
, cluster_num
, strdup(path
));
1759 if (s
->used_clusters
[cluster_num
] & USED_ANY
) {
1760 fprintf(stderr
, "cluster %d used more than once\n", (int)cluster_num
);
1763 s
->used_clusters
[cluster_num
] = USED_DIRECTORY
;
1765 DLOG(fprintf(stderr
, "read cluster %d (sector %d)\n", (int)cluster_num
, (int)cluster2sector(s
, cluster_num
)));
1766 subret
= vvfat_read(s
->bs
, cluster2sector(s
, cluster_num
), cluster
,
1767 s
->sectors_per_cluster
);
1769 fprintf(stderr
, "Error fetching direntries\n");
1775 for (i
= 0; i
< 0x10 * s
->sectors_per_cluster
; i
++) {
1778 DLOG(fprintf(stderr
, "check direntry %d: \n", i
); print_direntry(direntries
+ i
));
1779 if (is_volume_label(direntries
+ i
) || is_dot(direntries
+ i
) ||
1780 is_free(direntries
+ i
))
1783 subret
= parse_long_name(&lfn
, direntries
+ i
);
1785 fprintf(stderr
, "Error in long name\n");
1788 if (subret
== 0 || is_free(direntries
+ i
))
1791 if (fat_chksum(direntries
+i
) != lfn
.checksum
) {
1792 subret
= parse_short_name(s
, &lfn
, direntries
+ i
);
1794 fprintf(stderr
, "Error in short name (%d)\n", subret
);
1797 if (subret
> 0 || !strcmp((char*)lfn
.name
, ".")
1798 || !strcmp((char*)lfn
.name
, ".."))
1801 lfn
.checksum
= 0x100; /* cannot use long name twice */
1803 if (path_len
+ 1 + lfn
.len
>= PATH_MAX
) {
1804 fprintf(stderr
, "Name too long: %s/%s\n", path
, lfn
.name
);
1807 strcpy(path2
+ path_len
+ 1, (char*)lfn
.name
);
1809 if (is_directory(direntries
+ i
)) {
1810 if (begin_of_direntry(direntries
+ i
) == 0) {
1811 DLOG(fprintf(stderr
, "invalid begin for directory: %s\n", path2
); print_direntry(direntries
+ i
));
1814 cluster_count
= check_directory_consistency(s
,
1815 begin_of_direntry(direntries
+ i
), path2
);
1816 if (cluster_count
== 0) {
1817 DLOG(fprintf(stderr
, "problem in directory %s:\n", path2
); print_direntry(direntries
+ i
));
1820 } else if (is_file(direntries
+ i
)) {
1821 /* check file size with FAT */
1822 cluster_count
= get_cluster_count_for_direntry(s
, direntries
+ i
, path2
);
1823 if (cluster_count
!=
1824 (le32_to_cpu(direntries
[i
].size
) + s
->cluster_size
1825 - 1) / s
->cluster_size
) {
1826 DLOG(fprintf(stderr
, "Cluster count mismatch\n"));
1830 assert(0); /* cluster_count = 0; */
1832 ret
+= cluster_count
;
1835 cluster_num
= modified_fat_get(s
, cluster_num
);
1836 } while(!fat_eof(s
, cluster_num
));
1842 /* returns 1 on success */
1843 static int is_consistent(BDRVVVFATState
* s
)
1846 int used_clusters_count
= 0;
1850 * - get modified FAT
1851 * - compare the two FATs (TODO)
1852 * - get buffer for marking used clusters
1853 * - recurse direntries from root (using bs->bdrv_read to make
1854 * sure to get the new data)
1855 * - check that the FAT agrees with the size
1856 * - count the number of clusters occupied by this directory and
1858 * - check that the cumulative used cluster count agrees with the
1860 * - if all is fine, return number of used clusters
1862 if (s
->fat2
== NULL
) {
1863 int size
= 0x200 * s
->sectors_per_fat
;
1864 s
->fat2
= malloc(size
);
1865 memcpy(s
->fat2
, s
->fat
.pointer
, size
);
1867 check
= vvfat_read(s
->bs
,
1868 s
->first_sectors_number
, s
->fat2
, s
->sectors_per_fat
);
1870 fprintf(stderr
, "Could not copy fat\n");
1873 assert (s
->used_clusters
);
1874 for (i
= 0; i
< sector2cluster(s
, s
->sector_count
); i
++)
1875 s
->used_clusters
[i
] &= ~USED_ANY
;
1879 /* mark every mapped file/directory as deleted.
1880 * (check_directory_consistency() will unmark those still present). */
1882 for (i
= 0; i
< s
->mapping
.next
; i
++) {
1883 mapping_t
* mapping
= array_get(&(s
->mapping
), i
);
1884 if (mapping
->first_mapping_index
< 0)
1885 mapping
->mode
|= MODE_DELETED
;
1888 used_clusters_count
= check_directory_consistency(s
, 0, s
->path
);
1889 if (used_clusters_count
<= 0) {
1890 DLOG(fprintf(stderr
, "problem in directory\n"));
1894 check
= s
->last_cluster_of_root_directory
;
1895 for (i
= check
; i
< sector2cluster(s
, s
->sector_count
); i
++) {
1896 if (modified_fat_get(s
, i
)) {
1897 if(!s
->used_clusters
[i
]) {
1898 DLOG(fprintf(stderr
, "FAT was modified (%d), but cluster is not used?\n", i
));
1904 if (s
->used_clusters
[i
] == USED_ALLOCATED
) {
1905 /* allocated, but not used... */
1906 DLOG(fprintf(stderr
, "unused, modified cluster: %d\n", i
));
1911 if (check
!= used_clusters_count
)
1914 return used_clusters_count
;
1917 static inline void adjust_mapping_indices(BDRVVVFATState
* s
,
1918 int offset
, int adjust
)
1922 for (i
= 0; i
< s
->mapping
.next
; i
++) {
1923 mapping_t
* mapping
= array_get(&(s
->mapping
), i
);
1925 #define ADJUST_MAPPING_INDEX(name) \
1926 if (mapping->name >= offset) \
1927 mapping->name += adjust
1929 ADJUST_MAPPING_INDEX(first_mapping_index
);
1930 if (mapping
->mode
& MODE_DIRECTORY
)
1931 ADJUST_MAPPING_INDEX(info
.dir
.parent_mapping_index
);
1935 /* insert or update mapping */
1936 static mapping_t
* insert_mapping(BDRVVVFATState
* s
,
1937 uint32_t begin
, uint32_t end
)
1940 * - find mapping where mapping->begin >= begin,
1941 * - if mapping->begin > begin: insert
1942 * - adjust all references to mappings!
1946 int index
= find_mapping_for_cluster_aux(s
, begin
, 0, s
->mapping
.next
);
1947 mapping_t
* mapping
= NULL
;
1948 mapping_t
* first_mapping
= array_get(&(s
->mapping
), 0);
1950 if (index
< s
->mapping
.next
&& (mapping
= array_get(&(s
->mapping
), index
))
1951 && mapping
->begin
< begin
) {
1952 mapping
->end
= begin
;
1954 mapping
= array_get(&(s
->mapping
), index
);
1956 if (index
>= s
->mapping
.next
|| mapping
->begin
> begin
) {
1957 mapping
= array_insert(&(s
->mapping
), index
, 1);
1958 mapping
->path
= NULL
;
1959 adjust_mapping_indices(s
, index
, +1);
1962 mapping
->begin
= begin
;
1965 DLOG(mapping_t
* next_mapping
;
1966 assert(index
+ 1 >= s
->mapping
.next
||
1967 ((next_mapping
= array_get(&(s
->mapping
), index
+ 1)) &&
1968 next_mapping
->begin
>= end
)));
1970 if (s
->current_mapping
&& first_mapping
!= (mapping_t
*)s
->mapping
.pointer
)
1971 s
->current_mapping
= array_get(&(s
->mapping
),
1972 s
->current_mapping
- first_mapping
);
1977 static int remove_mapping(BDRVVVFATState
* s
, int mapping_index
)
1979 mapping_t
* mapping
= array_get(&(s
->mapping
), mapping_index
);
1980 mapping_t
* first_mapping
= array_get(&(s
->mapping
), 0);
1983 if (mapping
->first_mapping_index
< 0)
1984 free(mapping
->path
);
1986 /* remove from s->mapping */
1987 array_remove(&(s
->mapping
), mapping_index
);
1989 /* adjust all references to mappings */
1990 adjust_mapping_indices(s
, mapping_index
, -1);
1992 if (s
->current_mapping
&& first_mapping
!= (mapping_t
*)s
->mapping
.pointer
)
1993 s
->current_mapping
= array_get(&(s
->mapping
),
1994 s
->current_mapping
- first_mapping
);
1999 static void adjust_dirindices(BDRVVVFATState
* s
, int offset
, int adjust
)
2002 for (i
= 0; i
< s
->mapping
.next
; i
++) {
2003 mapping_t
* mapping
= array_get(&(s
->mapping
), i
);
2004 if (mapping
->dir_index
>= offset
)
2005 mapping
->dir_index
+= adjust
;
2006 if ((mapping
->mode
& MODE_DIRECTORY
) &&
2007 mapping
->info
.dir
.first_dir_index
>= offset
)
2008 mapping
->info
.dir
.first_dir_index
+= adjust
;
2012 static direntry_t
* insert_direntries(BDRVVVFATState
* s
,
2013 int dir_index
, int count
)
2016 * make room in s->directory,
2019 direntry_t
* result
= array_insert(&(s
->directory
), dir_index
, count
);
2022 adjust_dirindices(s
, dir_index
, count
);
2026 static int remove_direntries(BDRVVVFATState
* s
, int dir_index
, int count
)
2028 int ret
= array_remove_slice(&(s
->directory
), dir_index
, count
);
2031 adjust_dirindices(s
, dir_index
, -count
);
2036 * Adapt the mappings of the cluster chain starting at first cluster
2037 * (i.e. if a file starts at first_cluster, the chain is followed according
2038 * to the modified fat, and the corresponding entries in s->mapping are
2041 static int commit_mappings(BDRVVVFATState
* s
,
2042 uint32_t first_cluster
, int dir_index
)
2044 mapping_t
* mapping
= find_mapping_for_cluster(s
, first_cluster
);
2045 direntry_t
* direntry
= array_get(&(s
->directory
), dir_index
);
2046 uint32_t cluster
= first_cluster
;
2048 vvfat_close_current_file(s
);
2051 assert(mapping
->begin
== first_cluster
);
2052 mapping
->first_mapping_index
= -1;
2053 mapping
->dir_index
= dir_index
;
2054 mapping
->mode
= (dir_index
<= 0 || is_directory(direntry
)) ?
2055 MODE_DIRECTORY
: MODE_NORMAL
;
2057 while (!fat_eof(s
, cluster
)) {
2060 for (c
= cluster
, c1
= modified_fat_get(s
, c
); c
+ 1 == c1
;
2061 c
= c1
, c1
= modified_fat_get(s
, c1
));
2064 if (c
> mapping
->end
) {
2065 int index
= array_index(&(s
->mapping
), mapping
);
2066 int i
, max_i
= s
->mapping
.next
- index
;
2067 for (i
= 1; i
< max_i
&& mapping
[i
].begin
< c
; i
++);
2069 remove_mapping(s
, index
+ 1);
2071 assert(mapping
== array_get(&(s
->mapping
), s
->mapping
.next
- 1)
2072 || mapping
[1].begin
>= c
);
2075 if (!fat_eof(s
, c1
)) {
2076 int i
= find_mapping_for_cluster_aux(s
, c1
, 0, s
->mapping
.next
);
2077 mapping_t
* next_mapping
= i
>= s
->mapping
.next
? NULL
:
2078 array_get(&(s
->mapping
), i
);
2080 if (next_mapping
== NULL
|| next_mapping
->begin
> c1
) {
2081 int i1
= array_index(&(s
->mapping
), mapping
);
2083 next_mapping
= insert_mapping(s
, c1
, c1
+1);
2087 mapping
= array_get(&(s
->mapping
), i1
);
2090 next_mapping
->dir_index
= mapping
->dir_index
;
2091 next_mapping
->first_mapping_index
=
2092 mapping
->first_mapping_index
< 0 ?
2093 array_index(&(s
->mapping
), mapping
) :
2094 mapping
->first_mapping_index
;
2095 next_mapping
->path
= mapping
->path
;
2096 next_mapping
->mode
= mapping
->mode
;
2097 next_mapping
->read_only
= mapping
->read_only
;
2098 if (mapping
->mode
& MODE_DIRECTORY
) {
2099 next_mapping
->info
.dir
.parent_mapping_index
=
2100 mapping
->info
.dir
.parent_mapping_index
;
2101 next_mapping
->info
.dir
.first_dir_index
=
2102 mapping
->info
.dir
.first_dir_index
+
2103 0x10 * s
->sectors_per_cluster
*
2104 (mapping
->end
- mapping
->begin
);
2106 next_mapping
->info
.file
.offset
= mapping
->info
.file
.offset
+
2107 mapping
->end
- mapping
->begin
;
2109 mapping
= next_mapping
;
2118 static int commit_direntries(BDRVVVFATState
* s
,
2119 int dir_index
, int parent_mapping_index
)
2121 direntry_t
* direntry
= array_get(&(s
->directory
), dir_index
);
2122 uint32_t first_cluster
= dir_index
== 0 ? 0 : begin_of_direntry(direntry
);
2123 mapping_t
* mapping
= find_mapping_for_cluster(s
, first_cluster
);
2125 int factor
= 0x10 * s
->sectors_per_cluster
;
2126 int old_cluster_count
, new_cluster_count
;
2127 int current_dir_index
= mapping
->info
.dir
.first_dir_index
;
2128 int first_dir_index
= current_dir_index
;
2132 DLOG(fprintf(stderr
, "commit_direntries for %s, parent_mapping_index %d\n", mapping
->path
, parent_mapping_index
));
2136 assert(mapping
->begin
== first_cluster
);
2137 assert(mapping
->info
.dir
.first_dir_index
< s
->directory
.next
);
2138 assert(mapping
->mode
& MODE_DIRECTORY
);
2139 assert(dir_index
== 0 || is_directory(direntry
));
2141 mapping
->info
.dir
.parent_mapping_index
= parent_mapping_index
;
2143 if (first_cluster
== 0) {
2144 old_cluster_count
= new_cluster_count
=
2145 s
->last_cluster_of_root_directory
;
2147 for (old_cluster_count
= 0, c
= first_cluster
; !fat_eof(s
, c
);
2149 old_cluster_count
++;
2151 for (new_cluster_count
= 0, c
= first_cluster
; !fat_eof(s
, c
);
2152 c
= modified_fat_get(s
, c
))
2153 new_cluster_count
++;
2156 if (new_cluster_count
> old_cluster_count
) {
2157 if (insert_direntries(s
,
2158 current_dir_index
+ factor
* old_cluster_count
,
2159 factor
* (new_cluster_count
- old_cluster_count
)) == NULL
)
2161 } else if (new_cluster_count
< old_cluster_count
)
2162 remove_direntries(s
,
2163 current_dir_index
+ factor
* new_cluster_count
,
2164 factor
* (old_cluster_count
- new_cluster_count
));
2166 for (c
= first_cluster
; !fat_eof(s
, c
); c
= modified_fat_get(s
, c
)) {
2167 void* direntry
= array_get(&(s
->directory
), current_dir_index
);
2168 int ret
= vvfat_read(s
->bs
, cluster2sector(s
, c
), direntry
,
2169 s
->sectors_per_cluster
);
2172 assert(!strncmp(s
->directory
.pointer
, "QEMU", 4));
2173 current_dir_index
+= factor
;
2176 ret
= commit_mappings(s
, first_cluster
, dir_index
);
2181 for (i
= 0; i
< factor
* new_cluster_count
; i
++) {
2182 direntry
= array_get(&(s
->directory
), first_dir_index
+ i
);
2183 if (is_directory(direntry
) && !is_dot(direntry
)) {
2184 mapping
= find_mapping_for_cluster(s
, first_cluster
);
2185 assert(mapping
->mode
& MODE_DIRECTORY
);
2186 ret
= commit_direntries(s
, first_dir_index
+ i
,
2187 array_index(&(s
->mapping
), mapping
));
2196 /* commit one file (adjust contents, adjust mapping),
2197 return first_mapping_index */
2198 static int commit_one_file(BDRVVVFATState
* s
,
2199 int dir_index
, uint32_t offset
)
2201 direntry_t
* direntry
= array_get(&(s
->directory
), dir_index
);
2202 uint32_t c
= begin_of_direntry(direntry
);
2203 uint32_t first_cluster
= c
;
2204 mapping_t
* mapping
= find_mapping_for_cluster(s
, c
);
2205 uint32_t size
= filesize_of_direntry(direntry
);
2206 char* cluster
= malloc(s
->cluster_size
);
2210 assert(offset
< size
);
2211 assert((offset
% s
->cluster_size
) == 0);
2213 for (i
= s
->cluster_size
; i
< offset
; i
+= s
->cluster_size
)
2214 c
= modified_fat_get(s
, c
);
2216 fd
= open(mapping
->path
, O_RDWR
| O_CREAT
| O_BINARY
, 0666);
2218 fprintf(stderr
, "Could not open %s... (%s, %d)\n", mapping
->path
,
2219 strerror(errno
), errno
);
2223 if (lseek(fd
, offset
, SEEK_SET
) != offset
)
2226 while (offset
< size
) {
2228 int rest_size
= (size
- offset
> s
->cluster_size
?
2229 s
->cluster_size
: size
- offset
);
2232 c1
= modified_fat_get(s
, c
);
2234 assert((size
- offset
== 0 && fat_eof(s
, c
)) ||
2235 (size
> offset
&& c
>=2 && !fat_eof(s
, c
)));
2238 ret
= vvfat_read(s
->bs
, cluster2sector(s
, c
),
2239 (uint8_t*)cluster
, (rest_size
+ 0x1ff) / 0x200);
2244 if (write(fd
, cluster
, rest_size
) < 0)
2247 offset
+= rest_size
;
2251 ftruncate(fd
, size
);
2254 return commit_mappings(s
, first_cluster
, dir_index
);
2258 /* test, if all mappings point to valid direntries */
2259 static void check1(BDRVVVFATState
* s
)
2262 for (i
= 0; i
< s
->mapping
.next
; i
++) {
2263 mapping_t
* mapping
= array_get(&(s
->mapping
), i
);
2264 if (mapping
->mode
& MODE_DELETED
) {
2265 fprintf(stderr
, "deleted\n");
2268 assert(mapping
->dir_index
>= 0);
2269 assert(mapping
->dir_index
< s
->directory
.next
);
2270 direntry_t
* direntry
= array_get(&(s
->directory
), mapping
->dir_index
);
2271 assert(mapping
->begin
== begin_of_direntry(direntry
) || mapping
->first_mapping_index
>= 0);
2272 if (mapping
->mode
& MODE_DIRECTORY
) {
2273 assert(mapping
->info
.dir
.first_dir_index
+ 0x10 * s
->sectors_per_cluster
* (mapping
->end
- mapping
->begin
) <= s
->directory
.next
);
2274 assert((mapping
->info
.dir
.first_dir_index
% (0x10 * s
->sectors_per_cluster
)) == 0);
2279 /* test, if all direntries have mappings */
2280 static void check2(BDRVVVFATState
* s
)
2283 int first_mapping
= -1;
2285 for (i
= 0; i
< s
->directory
.next
; i
++) {
2286 direntry_t
* direntry
= array_get(&(s
->directory
), i
);
2288 if (is_short_name(direntry
) && begin_of_direntry(direntry
)) {
2289 mapping_t
* mapping
= find_mapping_for_cluster(s
, begin_of_direntry(direntry
));
2291 assert(mapping
->dir_index
== i
|| is_dot(direntry
));
2292 assert(mapping
->begin
== begin_of_direntry(direntry
) || is_dot(direntry
));
2295 if ((i
% (0x10 * s
->sectors_per_cluster
)) == 0) {
2299 for (j
= 0; j
< s
->mapping
.next
; j
++) {
2300 mapping_t
* mapping
= array_get(&(s
->mapping
), j
);
2301 if (mapping
->mode
& MODE_DELETED
)
2303 if (mapping
->mode
& MODE_DIRECTORY
) {
2304 if (mapping
->info
.dir
.first_dir_index
<= i
&& mapping
->info
.dir
.first_dir_index
+ 0x10 * s
->sectors_per_cluster
> i
) {
2305 assert(++count
== 1);
2306 if (mapping
->first_mapping_index
== -1)
2307 first_mapping
= array_index(&(s
->mapping
), mapping
);
2309 assert(first_mapping
== mapping
->first_mapping_index
);
2310 if (mapping
->info
.dir
.parent_mapping_index
< 0)
2313 mapping_t
* parent
= array_get(&(s
->mapping
), mapping
->info
.dir
.parent_mapping_index
);
2314 assert(parent
->mode
& MODE_DIRECTORY
);
2315 assert(parent
->info
.dir
.first_dir_index
< mapping
->info
.dir
.first_dir_index
);
2327 static int handle_renames_and_mkdirs(BDRVVVFATState
* s
)
2332 fprintf(stderr
, "handle_renames\n");
2333 for (i
= 0; i
< s
->commits
.next
; i
++) {
2334 commit_t
* commit
= array_get(&(s
->commits
), i
);
2335 fprintf(stderr
, "%d, %s (%d, %d)\n", i
, commit
->path
? commit
->path
: "(null)", commit
->param
.rename
.cluster
, commit
->action
);
2339 for (i
= 0; i
< s
->commits
.next
;) {
2340 commit_t
* commit
= array_get(&(s
->commits
), i
);
2341 if (commit
->action
== ACTION_RENAME
) {
2342 mapping_t
* mapping
= find_mapping_for_cluster(s
,
2343 commit
->param
.rename
.cluster
);
2344 char* old_path
= mapping
->path
;
2346 assert(commit
->path
);
2347 mapping
->path
= commit
->path
;
2348 if (rename(old_path
, mapping
->path
))
2351 if (mapping
->mode
& MODE_DIRECTORY
) {
2352 int l1
= strlen(mapping
->path
);
2353 int l2
= strlen(old_path
);
2355 direntry_t
* direntry
= array_get(&(s
->directory
),
2356 mapping
->info
.dir
.first_dir_index
);
2357 uint32_t c
= mapping
->begin
;
2361 while (!fat_eof(s
, c
)) {
2363 direntry_t
* d
= direntry
+ i
;
2365 if (is_file(d
) || (is_directory(d
) && !is_dot(d
))) {
2366 mapping_t
* m
= find_mapping_for_cluster(s
,
2367 begin_of_direntry(d
));
2368 int l
= strlen(m
->path
);
2369 char* new_path
= malloc(l
+ diff
+ 1);
2371 assert(!strncmp(m
->path
, mapping
->path
, l2
));
2373 strcpy(new_path
, mapping
->path
);
2374 strcpy(new_path
+ l1
, m
->path
+ l2
);
2376 schedule_rename(s
, m
->begin
, new_path
);
2379 } while((i
% (0x10 * s
->sectors_per_cluster
)) != 0);
2385 array_remove(&(s
->commits
), i
);
2387 } else if (commit
->action
== ACTION_MKDIR
) {
2389 int j
, parent_path_len
;
2392 if (mkdir(commit
->path
))
2395 if (mkdir(commit
->path
, 0755))
2399 mapping
= insert_mapping(s
, commit
->param
.mkdir
.cluster
,
2400 commit
->param
.mkdir
.cluster
+ 1);
2401 if (mapping
== NULL
)
2404 mapping
->mode
= MODE_DIRECTORY
;
2405 mapping
->read_only
= 0;
2406 mapping
->path
= commit
->path
;
2407 j
= s
->directory
.next
;
2409 insert_direntries(s
, s
->directory
.next
,
2410 0x10 * s
->sectors_per_cluster
);
2411 mapping
->info
.dir
.first_dir_index
= j
;
2413 parent_path_len
= strlen(commit
->path
)
2414 - strlen(get_basename(commit
->path
)) - 1;
2415 for (j
= 0; j
< s
->mapping
.next
; j
++) {
2416 mapping_t
* m
= array_get(&(s
->mapping
), j
);
2417 if (m
->first_mapping_index
< 0 && m
!= mapping
&&
2418 !strncmp(m
->path
, mapping
->path
, parent_path_len
) &&
2419 strlen(m
->path
) == parent_path_len
)
2422 assert(j
< s
->mapping
.next
);
2423 mapping
->info
.dir
.parent_mapping_index
= j
;
2425 array_remove(&(s
->commits
), i
);
2435 * TODO: make sure that the short name is not matching *another* file
2437 static int handle_commits(BDRVVVFATState
* s
)
2441 vvfat_close_current_file(s
);
2443 for (i
= 0; !fail
&& i
< s
->commits
.next
; i
++) {
2444 commit_t
* commit
= array_get(&(s
->commits
), i
);
2445 switch(commit
->action
) {
2446 case ACTION_RENAME
: case ACTION_MKDIR
:
2450 case ACTION_WRITEOUT
: {
2451 direntry_t
* entry
= array_get(&(s
->directory
),
2452 commit
->param
.writeout
.dir_index
);
2453 uint32_t begin
= begin_of_direntry(entry
);
2454 mapping_t
* mapping
= find_mapping_for_cluster(s
, begin
);
2457 assert(mapping
->begin
== begin
);
2458 assert(commit
->path
== NULL
);
2460 if (commit_one_file(s
, commit
->param
.writeout
.dir_index
,
2461 commit
->param
.writeout
.modified_offset
))
2466 case ACTION_NEW_FILE
: {
2467 int begin
= commit
->param
.new_file
.first_cluster
;
2468 mapping_t
* mapping
= find_mapping_for_cluster(s
, begin
);
2473 for (i
= 0; i
< s
->directory
.next
; i
++) {
2474 entry
= array_get(&(s
->directory
), i
);
2475 if (is_file(entry
) && begin_of_direntry(entry
) == begin
)
2479 if (i
>= s
->directory
.next
) {
2484 /* make sure there exists an initial mapping */
2485 if (mapping
&& mapping
->begin
!= begin
) {
2486 mapping
->end
= begin
;
2489 if (mapping
== NULL
) {
2490 mapping
= insert_mapping(s
, begin
, begin
+1);
2492 /* most members will be fixed in commit_mappings() */
2493 assert(commit
->path
);
2494 mapping
->path
= commit
->path
;
2495 mapping
->read_only
= 0;
2496 mapping
->mode
= MODE_NORMAL
;
2497 mapping
->info
.file
.offset
= 0;
2499 if (commit_one_file(s
, i
, 0))
2508 if (i
> 0 && array_remove_slice(&(s
->commits
), 0, i
))
2513 static int handle_deletes(BDRVVVFATState
* s
)
2515 int i
, deferred
= 1, deleted
= 1;
2517 /* delete files corresponding to mappings marked as deleted */
2518 /* handle DELETEs and unused mappings (modified_fat_get(s, mapping->begin) == 0) */
2519 while (deferred
&& deleted
) {
2523 for (i
= 1; i
< s
->mapping
.next
; i
++) {
2524 mapping_t
* mapping
= array_get(&(s
->mapping
), i
);
2525 if (mapping
->mode
& MODE_DELETED
) {
2526 direntry_t
* entry
= array_get(&(s
->directory
),
2527 mapping
->dir_index
);
2529 if (is_free(entry
)) {
2530 /* remove file/directory */
2531 if (mapping
->mode
& MODE_DIRECTORY
) {
2532 int j
, next_dir_index
= s
->directory
.next
,
2533 first_dir_index
= mapping
->info
.dir
.first_dir_index
;
2535 if (rmdir(mapping
->path
) < 0) {
2536 if (errno
== ENOTEMPTY
) {
2543 for (j
= 1; j
< s
->mapping
.next
; j
++) {
2544 mapping_t
* m
= array_get(&(s
->mapping
), j
);
2545 if (m
->mode
& MODE_DIRECTORY
&&
2546 m
->info
.dir
.first_dir_index
>
2548 m
->info
.dir
.first_dir_index
<
2551 m
->info
.dir
.first_dir_index
;
2553 remove_direntries(s
, first_dir_index
,
2554 next_dir_index
- first_dir_index
);
2559 if (unlink(mapping
->path
))
2563 DLOG(fprintf(stderr
, "DELETE (%d)\n", i
); print_mapping(mapping
); print_direntry(entry
));
2564 remove_mapping(s
, i
);
2573 * synchronize mapping with new state:
2575 * - copy FAT (with bdrv_read)
2576 * - mark all filenames corresponding to mappings as deleted
2577 * - recurse direntries from root (using bs->bdrv_read)
2578 * - delete files corresponding to mappings marked as deleted
2580 static int do_commit(BDRVVVFATState
* s
)
2584 /* the real meat are the commits. Nothing to do? Move along! */
2585 if (s
->commits
.next
== 0)
2588 vvfat_close_current_file(s
);
2590 ret
= handle_renames_and_mkdirs(s
);
2592 fprintf(stderr
, "Error handling renames (%d)\n", ret
);
2597 /* copy FAT (with bdrv_read) */
2598 memcpy(s
->fat
.pointer
, s
->fat2
, 0x200 * s
->sectors_per_fat
);
2600 /* recurse direntries from root (using bs->bdrv_read) */
2601 ret
= commit_direntries(s
, 0, -1);
2603 fprintf(stderr
, "Fatal: error while committing (%d)\n", ret
);
2608 ret
= handle_commits(s
);
2610 fprintf(stderr
, "Error handling commits (%d)\n", ret
);
2615 ret
= handle_deletes(s
);
2617 fprintf(stderr
, "Error deleting\n");
2622 s
->qcow
->drv
->bdrv_make_empty(s
->qcow
);
2624 memset(s
->used_clusters
, 0, sector2cluster(s
, s
->sector_count
));
2630 static int try_commit(BDRVVVFATState
* s
)
2632 vvfat_close_current_file(s
);
2634 if(!is_consistent(s
))
2636 return do_commit(s
);
2639 static int vvfat_write(BlockDriverState
*bs
, int64_t sector_num
,
2640 const uint8_t *buf
, int nb_sectors
)
2642 BDRVVVFATState
*s
= bs
->opaque
;
2647 vvfat_close_current_file(s
);
2650 * Some sanity checks:
2651 * - do not allow writing to the boot sector
2652 * - do not allow to write non-ASCII filenames
2655 if (sector_num
< s
->first_sectors_number
)
2658 for (i
= sector2cluster(s
, sector_num
);
2659 i
<= sector2cluster(s
, sector_num
+ nb_sectors
- 1);) {
2660 mapping_t
* mapping
= find_mapping_for_cluster(s
, i
);
2662 if (mapping
->read_only
) {
2663 fprintf(stderr
, "Tried to write to write-protected file %s\n",
2668 if (mapping
->mode
& MODE_DIRECTORY
) {
2669 int begin
= cluster2sector(s
, i
);
2670 int end
= begin
+ s
->sectors_per_cluster
, k
;
2672 const direntry_t
* direntries
;
2677 if (begin
< sector_num
)
2679 if (end
> sector_num
+ nb_sectors
)
2680 end
= sector_num
+ nb_sectors
;
2681 dir_index
= mapping
->dir_index
+
2682 0x10 * (begin
- mapping
->begin
* s
->sectors_per_cluster
);
2683 direntries
= (direntry_t
*)(buf
+ 0x200 * (begin
- sector_num
));
2685 for (k
= 0; k
< (end
- begin
) * 0x10; k
++) {
2686 /* do not allow non-ASCII filenames */
2687 if (parse_long_name(&lfn
, direntries
+ k
) < 0) {
2688 fprintf(stderr
, "Warning: non-ASCII filename\n");
2691 /* no access to the direntry of a read-only file */
2692 else if (is_short_name(direntries
+k
) &&
2693 (direntries
[k
].attributes
& 1)) {
2694 if (memcmp(direntries
+ k
,
2695 array_get(&(s
->directory
), dir_index
+ k
),
2696 sizeof(direntry_t
))) {
2697 fprintf(stderr
, "Warning: tried to write to write-protected file\n");
2709 * Use qcow backend. Commit later.
2711 DLOG(fprintf(stderr
, "Write to qcow backend: %d + %d\n", (int)sector_num
, nb_sectors
));
2712 ret
= s
->qcow
->drv
->bdrv_write(s
->qcow
, sector_num
, buf
, nb_sectors
);
2714 fprintf(stderr
, "Error writing to qcow backend\n");
2718 for (i
= sector2cluster(s
, sector_num
);
2719 i
<= sector2cluster(s
, sector_num
+ nb_sectors
- 1); i
++)
2721 s
->used_clusters
[i
] |= USED_ALLOCATED
;
2724 /* TODO: add timeout */
2731 static int vvfat_is_allocated(BlockDriverState
*bs
,
2732 int64_t sector_num
, int nb_sectors
, int* n
)
2734 BDRVVVFATState
* s
= bs
->opaque
;
2735 *n
= s
->sector_count
- sector_num
;
2736 if (*n
> nb_sectors
)
2743 static int write_target_commit(BlockDriverState
*bs
, int64_t sector_num
,
2744 const uint8_t* buffer
, int nb_sectors
) {
2745 BDRVVVFATState
* s
= bs
->opaque
;
2746 return try_commit(s
);
2749 static void write_target_close(BlockDriverState
*bs
) {
2750 BDRVVVFATState
* s
= bs
->opaque
;
2751 bdrv_delete(s
->qcow
);
2752 free(s
->qcow_filename
);
2755 static BlockDriver vvfat_write_target
= {
2756 "vvfat_write_target", 0, NULL
, NULL
, NULL
,
2757 write_target_commit
,
2762 static int enable_write_target(BDRVVVFATState
*s
)
2764 int size
= sector2cluster(s
, s
->sector_count
);
2765 s
->used_clusters
= calloc(size
, 1);
2767 array_init(&(s
->commits
), sizeof(commit_t
));
2769 s
->qcow_filename
= malloc(1024);
2770 get_tmp_filename(s
->qcow_filename
, 1024);
2771 if (bdrv_create(&bdrv_qcow
,
2772 s
->qcow_filename
, s
->sector_count
, "fat:", 0) < 0)
2774 s
->qcow
= bdrv_new("");
2775 if (s
->qcow
== NULL
|| bdrv_open(s
->qcow
, s
->qcow_filename
, 0) < 0)
2779 unlink(s
->qcow_filename
);
2782 s
->bs
->backing_hd
= calloc(sizeof(BlockDriverState
), 1);
2783 s
->bs
->backing_hd
->drv
= &vvfat_write_target
;
2784 s
->bs
->backing_hd
->opaque
= s
;
2789 static void vvfat_close(BlockDriverState
*bs
)
2791 BDRVVVFATState
*s
= bs
->opaque
;
2793 vvfat_close_current_file(s
);
2794 array_free(&(s
->fat
));
2795 array_free(&(s
->directory
));
2796 array_free(&(s
->mapping
));
2797 if(s
->cluster_buffer
)
2798 free(s
->cluster_buffer
);
2801 BlockDriver bdrv_vvfat
= {
2803 sizeof(BDRVVVFATState
),
2804 NULL
, /* no probe for protocols */
2809 NULL
, /* ??? Not sure if we can do any meaningful flushing. */
2812 .protocol_name
= "fat",
2816 static void checkpoint(void) {
2817 assert(((mapping_t
*)array_get(&(vvv
->mapping
), 0))->end
== 2);
2820 assert(!vvv
->current_mapping
|| vvv
->current_fd
|| (vvv
->current_mapping
->mode
& MODE_DIRECTORY
));
2822 if (((direntry_t
*)vvv
->directory
.pointer
)[1].attributes
!= 0xf)
2823 fprintf(stderr
, "Nonono!\n");
2825 direntry_t
* direntry
;
2826 assert(vvv
->mapping
.size
>= vvv
->mapping
.item_size
* vvv
->mapping
.next
);
2827 assert(vvv
->directory
.size
>= vvv
->directory
.item_size
* vvv
->directory
.next
);
2828 if (vvv
->mapping
.next
<47)
2830 assert((mapping
= array_get(&(vvv
->mapping
), 47)));
2831 assert(mapping
->dir_index
< vvv
->directory
.next
);
2832 direntry
= array_get(&(vvv
->directory
), mapping
->dir_index
);
2833 assert(!memcmp(direntry
->name
, "USB H ", 11) || direntry
->name
[0]==0);
2836 /* avoid compiler warnings: */
2838 remove_mapping(vvv
, NULL
);
2839 print_mapping(NULL
);
2840 print_direntry(NULL
);