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