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