1 /* 2 * fat_write.c 3 * 4 * R/W (V)FAT 12/16/32 filesystem implementation by Donggeun Kim 5 * 6 * See file CREDITS for list of people who contributed to this 7 * project. 8 * 9 * This program is free software; you can redistribute it and/or 10 * modify it under the terms of the GNU General Public License as 11 * published by the Free Software Foundation; either version 2 of 12 * the License, or (at your option) any later version. 13 * 14 * This program is distributed in the hope that it will be useful, 15 * but WITHOUT ANY WARRANTY; without even the implied warranty of 16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 17 * GNU General Public License for more details. 18 * 19 * You should have received a copy of the GNU General Public License 20 * along with this program; if not, write to the Free Software 21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, 22 * MA 02111-1307 USA 23 */ 24 25 #include <common.h> 26 #include <command.h> 27 #include <config.h> 28 #include <fat.h> 29 #include <asm/byteorder.h> 30 #include <part.h> 31 #include "fat.c" 32 33 static void uppercase(char *str, int len) 34 { 35 int i; 36 37 for (i = 0; i < len; i++) { 38 TOUPPER(*str); 39 str++; 40 } 41 } 42 43 static int total_sector; 44 static int disk_write(__u32 startblock, __u32 getsize, __u8 *bufptr) 45 { 46 if (cur_dev == NULL) 47 return -1; 48 49 if (startblock + getsize > total_sector) { 50 printf("error: overflow occurs\n"); 51 return -1; 52 } 53 54 startblock += part_offset; 55 56 if (cur_dev->block_read) { 57 return cur_dev->block_write(cur_dev->dev, startblock, getsize, 58 (unsigned long *) bufptr); 59 } 60 return -1; 61 } 62 63 /* 64 * Set short name in directory entry 65 */ 66 static void set_name(dir_entry *dirent, const char *filename) 67 { 68 char s_name[VFAT_MAXLEN_BYTES]; 69 char *period; 70 int period_location, len, i, ext_num; 71 72 if (filename == NULL) 73 return; 74 75 len = strlen(filename); 76 if (len == 0) 77 return; 78 79 memcpy(s_name, filename, len); 80 uppercase(s_name, len); 81 82 period = strchr(s_name, '.'); 83 if (period == NULL) { 84 period_location = len; 85 ext_num = 0; 86 } else { 87 period_location = period - s_name; 88 ext_num = len - period_location - 1; 89 } 90 91 /* Pad spaces when the length of file name is shorter than eight */ 92 if (period_location < 8) { 93 memcpy(dirent->name, s_name, period_location); 94 for (i = period_location; i < 8; i++) 95 dirent->name[i] = ' '; 96 } else if (period_location == 8) { 97 memcpy(dirent->name, s_name, period_location); 98 } else { 99 memcpy(dirent->name, s_name, 6); 100 dirent->name[6] = '~'; 101 dirent->name[7] = '1'; 102 } 103 104 if (ext_num < 3) { 105 memcpy(dirent->ext, s_name + period_location + 1, ext_num); 106 for (i = ext_num; i < 3; i++) 107 dirent->ext[i] = ' '; 108 } else 109 memcpy(dirent->ext, s_name + period_location + 1, 3); 110 111 debug("name : %s\n", dirent->name); 112 debug("ext : %s\n", dirent->ext); 113 } 114 115 /* 116 * Write fat buffer into block device 117 */ 118 static int flush_fat_buffer(fsdata *mydata) 119 { 120 int getsize = FATBUFBLOCKS; 121 __u32 fatlength = mydata->fatlength; 122 __u8 *bufptr = mydata->fatbuf; 123 __u32 startblock = mydata->fatbufnum * FATBUFBLOCKS; 124 125 fatlength *= mydata->sect_size; 126 startblock += mydata->fat_sect; 127 128 if (getsize > fatlength) 129 getsize = fatlength; 130 131 /* Write FAT buf */ 132 if (disk_write(startblock, getsize, bufptr) < 0) { 133 debug("error: writing FAT blocks\n"); 134 return -1; 135 } 136 137 return 0; 138 } 139 140 /* 141 * Get the entry at index 'entry' in a FAT (12/16/32) table. 142 * On failure 0x00 is returned. 143 * When bufnum is changed, write back the previous fatbuf to the disk. 144 */ 145 static __u32 get_fatent_value(fsdata *mydata, __u32 entry) 146 { 147 __u32 bufnum; 148 __u32 off16, offset; 149 __u32 ret = 0x00; 150 __u16 val1, val2; 151 152 switch (mydata->fatsize) { 153 case 32: 154 bufnum = entry / FAT32BUFSIZE; 155 offset = entry - bufnum * FAT32BUFSIZE; 156 break; 157 case 16: 158 bufnum = entry / FAT16BUFSIZE; 159 offset = entry - bufnum * FAT16BUFSIZE; 160 break; 161 case 12: 162 bufnum = entry / FAT12BUFSIZE; 163 offset = entry - bufnum * FAT12BUFSIZE; 164 break; 165 166 default: 167 /* Unsupported FAT size */ 168 return ret; 169 } 170 171 debug("FAT%d: entry: 0x%04x = %d, offset: 0x%04x = %d\n", 172 mydata->fatsize, entry, entry, offset, offset); 173 174 /* Read a new block of FAT entries into the cache. */ 175 if (bufnum != mydata->fatbufnum) { 176 int getsize = FATBUFBLOCKS; 177 __u8 *bufptr = mydata->fatbuf; 178 __u32 fatlength = mydata->fatlength; 179 __u32 startblock = bufnum * FATBUFBLOCKS; 180 181 if (getsize > fatlength) 182 getsize = fatlength; 183 184 fatlength *= mydata->sect_size; /* We want it in bytes now */ 185 startblock += mydata->fat_sect; /* Offset from start of disk */ 186 187 /* Write back the fatbuf to the disk */ 188 if (mydata->fatbufnum != -1) { 189 if (flush_fat_buffer(mydata) < 0) 190 return -1; 191 } 192 193 if (disk_read(startblock, getsize, bufptr) < 0) { 194 debug("Error reading FAT blocks\n"); 195 return ret; 196 } 197 mydata->fatbufnum = bufnum; 198 } 199 200 /* Get the actual entry from the table */ 201 switch (mydata->fatsize) { 202 case 32: 203 ret = FAT2CPU32(((__u32 *) mydata->fatbuf)[offset]); 204 break; 205 case 16: 206 ret = FAT2CPU16(((__u16 *) mydata->fatbuf)[offset]); 207 break; 208 case 12: 209 off16 = (offset * 3) / 4; 210 211 switch (offset & 0x3) { 212 case 0: 213 ret = FAT2CPU16(((__u16 *) mydata->fatbuf)[off16]); 214 ret &= 0xfff; 215 break; 216 case 1: 217 val1 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16]); 218 val1 &= 0xf000; 219 val2 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16 + 1]); 220 val2 &= 0x00ff; 221 ret = (val2 << 4) | (val1 >> 12); 222 break; 223 case 2: 224 val1 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16]); 225 val1 &= 0xff00; 226 val2 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16 + 1]); 227 val2 &= 0x000f; 228 ret = (val2 << 8) | (val1 >> 8); 229 break; 230 case 3: 231 ret = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16]); 232 ret = (ret & 0xfff0) >> 4; 233 break; 234 default: 235 break; 236 } 237 break; 238 } 239 debug("FAT%d: ret: %08x, entry: %08x, offset: %04x\n", 240 mydata->fatsize, ret, entry, offset); 241 242 return ret; 243 } 244 245 #ifdef CONFIG_SUPPORT_VFAT 246 /* 247 * Set the file name information from 'name' into 'slotptr', 248 */ 249 static int str2slot(dir_slot *slotptr, const char *name, int *idx) 250 { 251 int j, end_idx = 0; 252 253 for (j = 0; j <= 8; j += 2) { 254 if (name[*idx] == 0x00) { 255 slotptr->name0_4[j] = 0; 256 slotptr->name0_4[j + 1] = 0; 257 end_idx++; 258 goto name0_4; 259 } 260 slotptr->name0_4[j] = name[*idx]; 261 (*idx)++; 262 end_idx++; 263 } 264 for (j = 0; j <= 10; j += 2) { 265 if (name[*idx] == 0x00) { 266 slotptr->name5_10[j] = 0; 267 slotptr->name5_10[j + 1] = 0; 268 end_idx++; 269 goto name5_10; 270 } 271 slotptr->name5_10[j] = name[*idx]; 272 (*idx)++; 273 end_idx++; 274 } 275 for (j = 0; j <= 2; j += 2) { 276 if (name[*idx] == 0x00) { 277 slotptr->name11_12[j] = 0; 278 slotptr->name11_12[j + 1] = 0; 279 end_idx++; 280 goto name11_12; 281 } 282 slotptr->name11_12[j] = name[*idx]; 283 (*idx)++; 284 end_idx++; 285 } 286 287 if (name[*idx] == 0x00) 288 return 1; 289 290 return 0; 291 /* Not used characters are filled with 0xff 0xff */ 292 name0_4: 293 for (; end_idx < 5; end_idx++) { 294 slotptr->name0_4[end_idx * 2] = 0xff; 295 slotptr->name0_4[end_idx * 2 + 1] = 0xff; 296 } 297 end_idx = 5; 298 name5_10: 299 end_idx -= 5; 300 for (; end_idx < 6; end_idx++) { 301 slotptr->name5_10[end_idx * 2] = 0xff; 302 slotptr->name5_10[end_idx * 2 + 1] = 0xff; 303 } 304 end_idx = 11; 305 name11_12: 306 end_idx -= 11; 307 for (; end_idx < 2; end_idx++) { 308 slotptr->name11_12[end_idx * 2] = 0xff; 309 slotptr->name11_12[end_idx * 2 + 1] = 0xff; 310 } 311 312 return 1; 313 } 314 315 static int is_next_clust(fsdata *mydata, dir_entry *dentptr); 316 static void flush_dir_table(fsdata *mydata, dir_entry **dentptr); 317 318 /* 319 * Fill dir_slot entries with appropriate name, id, and attr 320 * The real directory entry is returned by 'dentptr' 321 */ 322 static void 323 fill_dir_slot(fsdata *mydata, dir_entry **dentptr, const char *l_name) 324 { 325 dir_slot *slotptr = (dir_slot *)get_vfatname_block; 326 __u8 counter, checksum; 327 int idx = 0, ret; 328 char s_name[16]; 329 330 /* Get short file name and checksum value */ 331 strncpy(s_name, (*dentptr)->name, 16); 332 checksum = mkcksum(s_name); 333 334 do { 335 memset(slotptr, 0x00, sizeof(dir_slot)); 336 ret = str2slot(slotptr, l_name, &idx); 337 slotptr->id = ++counter; 338 slotptr->attr = ATTR_VFAT; 339 slotptr->alias_checksum = checksum; 340 slotptr++; 341 } while (ret == 0); 342 343 slotptr--; 344 slotptr->id |= LAST_LONG_ENTRY_MASK; 345 346 while (counter >= 1) { 347 if (is_next_clust(mydata, *dentptr)) { 348 /* A new cluster is allocated for directory table */ 349 flush_dir_table(mydata, dentptr); 350 } 351 memcpy(*dentptr, slotptr, sizeof(dir_slot)); 352 (*dentptr)++; 353 slotptr--; 354 counter--; 355 } 356 357 if (is_next_clust(mydata, *dentptr)) { 358 /* A new cluster is allocated for directory table */ 359 flush_dir_table(mydata, dentptr); 360 } 361 } 362 363 static __u32 dir_curclust; 364 365 /* 366 * Extract the full long filename starting at 'retdent' (which is really 367 * a slot) into 'l_name'. If successful also copy the real directory entry 368 * into 'retdent' 369 * If additional adjacent cluster for directory entries is read into memory, 370 * then 'get_vfatname_block' is copied into 'get_dentfromdir_block' and 371 * the location of the real directory entry is returned by 'retdent' 372 * Return 0 on success, -1 otherwise. 373 */ 374 static int 375 get_long_file_name(fsdata *mydata, int curclust, __u8 *cluster, 376 dir_entry **retdent, char *l_name) 377 { 378 dir_entry *realdent; 379 dir_slot *slotptr = (dir_slot *)(*retdent); 380 dir_slot *slotptr2 = NULL; 381 __u8 *buflimit = cluster + mydata->sect_size * ((curclust == 0) ? 382 PREFETCH_BLOCKS : 383 mydata->clust_size); 384 __u8 counter = (slotptr->id & ~LAST_LONG_ENTRY_MASK) & 0xff; 385 int idx = 0, cur_position = 0; 386 387 if (counter > VFAT_MAXSEQ) { 388 debug("Error: VFAT name is too long\n"); 389 return -1; 390 } 391 392 while ((__u8 *)slotptr < buflimit) { 393 if (counter == 0) 394 break; 395 if (((slotptr->id & ~LAST_LONG_ENTRY_MASK) & 0xff) != counter) 396 return -1; 397 slotptr++; 398 counter--; 399 } 400 401 if ((__u8 *)slotptr >= buflimit) { 402 if (curclust == 0) 403 return -1; 404 curclust = get_fatent_value(mydata, dir_curclust); 405 if (CHECK_CLUST(curclust, mydata->fatsize)) { 406 debug("curclust: 0x%x\n", curclust); 407 printf("Invalid FAT entry\n"); 408 return -1; 409 } 410 411 dir_curclust = curclust; 412 413 if (get_cluster(mydata, curclust, get_vfatname_block, 414 mydata->clust_size * mydata->sect_size) != 0) { 415 debug("Error: reading directory block\n"); 416 return -1; 417 } 418 419 slotptr2 = (dir_slot *)get_vfatname_block; 420 while (counter > 0) { 421 if (((slotptr2->id & ~LAST_LONG_ENTRY_MASK) 422 & 0xff) != counter) 423 return -1; 424 slotptr2++; 425 counter--; 426 } 427 428 /* Save the real directory entry */ 429 realdent = (dir_entry *)slotptr2; 430 while ((__u8 *)slotptr2 > get_vfatname_block) { 431 slotptr2--; 432 slot2str(slotptr2, l_name, &idx); 433 } 434 } else { 435 /* Save the real directory entry */ 436 realdent = (dir_entry *)slotptr; 437 } 438 439 do { 440 slotptr--; 441 if (slot2str(slotptr, l_name, &idx)) 442 break; 443 } while (!(slotptr->id & LAST_LONG_ENTRY_MASK)); 444 445 l_name[idx] = '\0'; 446 if (*l_name == DELETED_FLAG) 447 *l_name = '\0'; 448 else if (*l_name == aRING) 449 *l_name = DELETED_FLAG; 450 downcase(l_name); 451 452 /* Return the real directory entry */ 453 *retdent = realdent; 454 455 if (slotptr2) { 456 memcpy(get_dentfromdir_block, get_vfatname_block, 457 mydata->clust_size * mydata->sect_size); 458 cur_position = (__u8 *)realdent - get_vfatname_block; 459 *retdent = (dir_entry *) &get_dentfromdir_block[cur_position]; 460 } 461 462 return 0; 463 } 464 465 #endif 466 467 /* 468 * Set the entry at index 'entry' in a FAT (16/32) table. 469 */ 470 static int set_fatent_value(fsdata *mydata, __u32 entry, __u32 entry_value) 471 { 472 __u32 bufnum, offset; 473 474 switch (mydata->fatsize) { 475 case 32: 476 bufnum = entry / FAT32BUFSIZE; 477 offset = entry - bufnum * FAT32BUFSIZE; 478 break; 479 case 16: 480 bufnum = entry / FAT16BUFSIZE; 481 offset = entry - bufnum * FAT16BUFSIZE; 482 break; 483 default: 484 /* Unsupported FAT size */ 485 return -1; 486 } 487 488 /* Read a new block of FAT entries into the cache. */ 489 if (bufnum != mydata->fatbufnum) { 490 int getsize = FATBUFBLOCKS; 491 __u8 *bufptr = mydata->fatbuf; 492 __u32 fatlength = mydata->fatlength; 493 __u32 startblock = bufnum * FATBUFBLOCKS; 494 495 fatlength *= mydata->sect_size; 496 startblock += mydata->fat_sect; 497 498 if (getsize > fatlength) 499 getsize = fatlength; 500 501 if (mydata->fatbufnum != -1) { 502 if (flush_fat_buffer(mydata) < 0) 503 return -1; 504 } 505 506 if (disk_read(startblock, getsize, bufptr) < 0) { 507 debug("Error reading FAT blocks\n"); 508 return -1; 509 } 510 mydata->fatbufnum = bufnum; 511 } 512 513 /* Set the actual entry */ 514 switch (mydata->fatsize) { 515 case 32: 516 ((__u32 *) mydata->fatbuf)[offset] = cpu_to_le32(entry_value); 517 break; 518 case 16: 519 ((__u16 *) mydata->fatbuf)[offset] = cpu_to_le16(entry_value); 520 break; 521 default: 522 return -1; 523 } 524 525 return 0; 526 } 527 528 /* 529 * Determine the entry value at index 'entry' in a FAT (16/32) table 530 */ 531 static __u32 determine_fatent(fsdata *mydata, __u32 entry) 532 { 533 __u32 next_fat, next_entry = entry + 1; 534 535 while (1) { 536 next_fat = get_fatent_value(mydata, next_entry); 537 if (next_fat == 0) { 538 set_fatent_value(mydata, entry, next_entry); 539 break; 540 } 541 next_entry++; 542 } 543 debug("FAT%d: entry: %08x, entry_value: %04x\n", 544 mydata->fatsize, entry, next_entry); 545 546 return next_entry; 547 } 548 549 /* 550 * Write at most 'size' bytes from 'buffer' into the specified cluster. 551 * Return 0 on success, -1 otherwise. 552 */ 553 static int 554 set_cluster(fsdata *mydata, __u32 clustnum, __u8 *buffer, 555 unsigned long size) 556 { 557 int idx = 0; 558 __u32 startsect; 559 560 if (clustnum > 0) 561 startsect = mydata->data_begin + 562 clustnum * mydata->clust_size; 563 else 564 startsect = mydata->rootdir_sect; 565 566 debug("clustnum: %d, startsect: %d\n", clustnum, startsect); 567 568 if (disk_write(startsect, size / mydata->sect_size, buffer) < 0) { 569 debug("Error writing data\n"); 570 return -1; 571 } 572 573 if (size % mydata->sect_size) { 574 __u8 tmpbuf[mydata->sect_size]; 575 576 idx = size / mydata->sect_size; 577 buffer += idx * mydata->sect_size; 578 memcpy(tmpbuf, buffer, size % mydata->sect_size); 579 580 if (disk_write(startsect + idx, 1, tmpbuf) < 0) { 581 debug("Error writing data\n"); 582 return -1; 583 } 584 585 return 0; 586 } 587 588 return 0; 589 } 590 591 /* 592 * Find the first empty cluster 593 */ 594 static int find_empty_cluster(fsdata *mydata) 595 { 596 __u32 fat_val, entry = 3; 597 598 while (1) { 599 fat_val = get_fatent_value(mydata, entry); 600 if (fat_val == 0) 601 break; 602 entry++; 603 } 604 605 return entry; 606 } 607 608 /* 609 * Write directory entries in 'get_dentfromdir_block' to block device 610 */ 611 static void flush_dir_table(fsdata *mydata, dir_entry **dentptr) 612 { 613 int dir_newclust = 0; 614 615 if (set_cluster(mydata, dir_curclust, 616 get_dentfromdir_block, 617 mydata->clust_size * mydata->sect_size) != 0) { 618 printf("error: wrinting directory entry\n"); 619 return; 620 } 621 dir_newclust = find_empty_cluster(mydata); 622 set_fatent_value(mydata, dir_curclust, dir_newclust); 623 if (mydata->fatsize == 32) 624 set_fatent_value(mydata, dir_newclust, 0xffffff8); 625 else if (mydata->fatsize == 16) 626 set_fatent_value(mydata, dir_newclust, 0xfff8); 627 628 dir_curclust = dir_newclust; 629 630 if (flush_fat_buffer(mydata) < 0) 631 return; 632 633 memset(get_dentfromdir_block, 0x00, 634 mydata->clust_size * mydata->sect_size); 635 636 *dentptr = (dir_entry *) get_dentfromdir_block; 637 } 638 639 /* 640 * Set empty cluster from 'entry' to the end of a file 641 */ 642 static int clear_fatent(fsdata *mydata, __u32 entry) 643 { 644 __u32 fat_val; 645 646 while (1) { 647 fat_val = get_fatent_value(mydata, entry); 648 if (fat_val != 0) 649 set_fatent_value(mydata, entry, 0); 650 else 651 break; 652 653 if (fat_val == 0xfffffff || fat_val == 0xffff) 654 break; 655 656 entry = fat_val; 657 } 658 659 /* Flush fat buffer */ 660 if (flush_fat_buffer(mydata) < 0) 661 return -1; 662 663 return 0; 664 } 665 666 /* 667 * Write at most 'maxsize' bytes from 'buffer' into 668 * the file associated with 'dentptr' 669 * Return the number of bytes read or -1 on fatal errors. 670 */ 671 static int 672 set_contents(fsdata *mydata, dir_entry *dentptr, __u8 *buffer, 673 unsigned long maxsize) 674 { 675 unsigned long filesize = FAT2CPU32(dentptr->size), gotsize = 0; 676 unsigned int bytesperclust = mydata->clust_size * mydata->sect_size; 677 __u32 curclust = START(dentptr); 678 __u32 endclust = 0, newclust = 0; 679 unsigned long actsize; 680 681 debug("Filesize: %ld bytes\n", filesize); 682 683 if (maxsize > 0 && filesize > maxsize) 684 filesize = maxsize; 685 686 debug("%ld bytes\n", filesize); 687 688 actsize = bytesperclust; 689 endclust = curclust; 690 do { 691 /* search for consecutive clusters */ 692 while (actsize < filesize) { 693 newclust = determine_fatent(mydata, endclust); 694 695 if ((newclust - 1) != endclust) 696 goto getit; 697 698 if (CHECK_CLUST(newclust, mydata->fatsize)) { 699 debug("curclust: 0x%x\n", newclust); 700 debug("Invalid FAT entry\n"); 701 return gotsize; 702 } 703 endclust = newclust; 704 actsize += bytesperclust; 705 } 706 /* actsize >= file size */ 707 actsize -= bytesperclust; 708 /* set remaining clusters */ 709 if (set_cluster(mydata, curclust, buffer, (int)actsize) != 0) { 710 debug("error: writing cluster\n"); 711 return -1; 712 } 713 714 /* set remaining bytes */ 715 gotsize += (int)actsize; 716 filesize -= actsize; 717 buffer += actsize; 718 actsize = filesize; 719 720 if (set_cluster(mydata, endclust, buffer, (int)actsize) != 0) { 721 debug("error: writing cluster\n"); 722 return -1; 723 } 724 gotsize += actsize; 725 726 /* Mark end of file in FAT */ 727 if (mydata->fatsize == 16) 728 newclust = 0xffff; 729 else if (mydata->fatsize == 32) 730 newclust = 0xfffffff; 731 set_fatent_value(mydata, endclust, newclust); 732 733 return gotsize; 734 getit: 735 if (set_cluster(mydata, curclust, buffer, (int)actsize) != 0) { 736 debug("error: writing cluster\n"); 737 return -1; 738 } 739 gotsize += (int)actsize; 740 filesize -= actsize; 741 buffer += actsize; 742 743 if (CHECK_CLUST(curclust, mydata->fatsize)) { 744 debug("curclust: 0x%x\n", curclust); 745 debug("Invalid FAT entry\n"); 746 return gotsize; 747 } 748 actsize = bytesperclust; 749 curclust = endclust = newclust; 750 } while (1); 751 } 752 753 /* 754 * Fill dir_entry 755 */ 756 static void fill_dentry(fsdata *mydata, dir_entry *dentptr, 757 const char *filename, __u32 start_cluster, __u32 size, __u8 attr) 758 { 759 if (mydata->fatsize == 32) 760 dentptr->starthi = 761 cpu_to_le16((start_cluster & 0xffff0000) >> 16); 762 dentptr->start = cpu_to_le16(start_cluster & 0xffff); 763 dentptr->size = cpu_to_le32(size); 764 765 dentptr->attr = attr; 766 767 set_name(dentptr, filename); 768 } 769 770 /* 771 * Check whether adding a file makes the file system to 772 * exceed the size of the block device 773 * Return -1 when overflow occurs, otherwise return 0 774 */ 775 static int check_overflow(fsdata *mydata, __u32 clustnum, unsigned long size) 776 { 777 __u32 startsect, sect_num; 778 779 if (clustnum > 0) { 780 startsect = mydata->data_begin + 781 clustnum * mydata->clust_size; 782 } else { 783 startsect = mydata->rootdir_sect; 784 } 785 786 sect_num = size / mydata->sect_size; 787 if (size % mydata->sect_size) 788 sect_num++; 789 790 if (startsect + sect_num > total_sector) 791 return -1; 792 793 return 0; 794 } 795 796 /* 797 * Check if adding several entries exceed one cluster boundary 798 */ 799 static int is_next_clust(fsdata *mydata, dir_entry *dentptr) 800 { 801 int cur_position; 802 803 cur_position = (__u8 *)dentptr - get_dentfromdir_block; 804 805 if (cur_position >= mydata->clust_size * mydata->sect_size) 806 return 1; 807 else 808 return 0; 809 } 810 811 static dir_entry *empty_dentptr; 812 /* 813 * Find a directory entry based on filename or start cluster number 814 * If the directory entry is not found, 815 * the new position for writing a directory entry will be returned 816 */ 817 static dir_entry *find_directory_entry(fsdata *mydata, int startsect, 818 char *filename, dir_entry *retdent, __u32 start) 819 { 820 __u16 prevcksum = 0xffff; 821 __u32 curclust = (startsect - mydata->data_begin) / mydata->clust_size; 822 823 debug("get_dentfromdir: %s\n", filename); 824 825 while (1) { 826 dir_entry *dentptr; 827 828 int i; 829 830 if (get_cluster(mydata, curclust, get_dentfromdir_block, 831 mydata->clust_size * mydata->sect_size) != 0) { 832 printf("Error: reading directory block\n"); 833 return NULL; 834 } 835 836 dentptr = (dir_entry *)get_dentfromdir_block; 837 838 dir_curclust = curclust; 839 840 for (i = 0; i < DIRENTSPERCLUST; i++) { 841 char s_name[14], l_name[VFAT_MAXLEN_BYTES]; 842 843 l_name[0] = '\0'; 844 if (dentptr->name[0] == DELETED_FLAG) { 845 dentptr++; 846 if (is_next_clust(mydata, dentptr)) 847 break; 848 continue; 849 } 850 if ((dentptr->attr & ATTR_VOLUME)) { 851 #ifdef CONFIG_SUPPORT_VFAT 852 if ((dentptr->attr & ATTR_VFAT) && 853 (dentptr->name[0] & LAST_LONG_ENTRY_MASK)) { 854 prevcksum = 855 ((dir_slot *)dentptr)->alias_checksum; 856 get_long_file_name(mydata, curclust, 857 get_dentfromdir_block, 858 &dentptr, l_name); 859 debug("vfatname: |%s|\n", l_name); 860 } else 861 #endif 862 { 863 /* Volume label or VFAT entry */ 864 dentptr++; 865 if (is_next_clust(mydata, dentptr)) 866 break; 867 continue; 868 } 869 } 870 if (dentptr->name[0] == 0) { 871 debug("Dentname == NULL - %d\n", i); 872 empty_dentptr = dentptr; 873 return NULL; 874 } 875 876 get_name(dentptr, s_name); 877 878 if (strcmp(filename, s_name) 879 && strcmp(filename, l_name)) { 880 debug("Mismatch: |%s|%s|\n", 881 s_name, l_name); 882 dentptr++; 883 if (is_next_clust(mydata, dentptr)) 884 break; 885 continue; 886 } 887 888 memcpy(retdent, dentptr, sizeof(dir_entry)); 889 890 debug("DentName: %s", s_name); 891 debug(", start: 0x%x", START(dentptr)); 892 debug(", size: 0x%x %s\n", 893 FAT2CPU32(dentptr->size), 894 (dentptr->attr & ATTR_DIR) ? 895 "(DIR)" : ""); 896 897 return dentptr; 898 } 899 900 curclust = get_fatent_value(mydata, dir_curclust); 901 if ((curclust >= 0xffffff8) || (curclust >= 0xfff8)) { 902 empty_dentptr = dentptr; 903 return NULL; 904 } 905 if (CHECK_CLUST(curclust, mydata->fatsize)) { 906 debug("curclust: 0x%x\n", curclust); 907 debug("Invalid FAT entry\n"); 908 return NULL; 909 } 910 } 911 912 return NULL; 913 } 914 915 static int do_fat_write(const char *filename, void *buffer, 916 unsigned long size) 917 { 918 dir_entry *dentptr, *retdent; 919 dir_slot *slotptr; 920 __u32 startsect; 921 __u32 start_cluster; 922 boot_sector bs; 923 volume_info volinfo; 924 fsdata datablock; 925 fsdata *mydata = &datablock; 926 int cursect; 927 int root_cluster, ret = -1, name_len; 928 char l_filename[VFAT_MAXLEN_BYTES]; 929 930 dir_curclust = 0; 931 932 if (read_bootsectandvi(&bs, &volinfo, &mydata->fatsize)) { 933 debug("error: reading boot sector\n"); 934 return -1; 935 } 936 937 total_sector = bs.total_sect; 938 if (total_sector == 0) 939 total_sector = part_size; 940 941 root_cluster = bs.root_cluster; 942 943 if (mydata->fatsize == 32) 944 mydata->fatlength = bs.fat32_length; 945 else 946 mydata->fatlength = bs.fat_length; 947 948 mydata->fat_sect = bs.reserved; 949 950 cursect = mydata->rootdir_sect 951 = mydata->fat_sect + mydata->fatlength * bs.fats; 952 953 mydata->sect_size = (bs.sector_size[1] << 8) + bs.sector_size[0]; 954 mydata->clust_size = bs.cluster_size; 955 956 if (mydata->fatsize == 32) { 957 mydata->data_begin = mydata->rootdir_sect - 958 (mydata->clust_size * 2); 959 } else { 960 int rootdir_size; 961 962 rootdir_size = ((bs.dir_entries[1] * (int)256 + 963 bs.dir_entries[0]) * 964 sizeof(dir_entry)) / 965 mydata->sect_size; 966 mydata->data_begin = mydata->rootdir_sect + 967 rootdir_size - 968 (mydata->clust_size * 2); 969 } 970 971 mydata->fatbufnum = -1; 972 mydata->fatbuf = malloc(FATBUFSIZE); 973 if (mydata->fatbuf == NULL) { 974 debug("Error: allocating memory\n"); 975 return -1; 976 } 977 978 if (disk_read(cursect, 979 (mydata->fatsize == 32) ? 980 (mydata->clust_size) : 981 PREFETCH_BLOCKS, do_fat_read_block) < 0) { 982 debug("Error: reading rootdir block\n"); 983 goto exit; 984 } 985 dentptr = (dir_entry *) do_fat_read_block; 986 987 name_len = strlen(filename); 988 memcpy(l_filename, filename, name_len); 989 downcase(l_filename); 990 991 startsect = mydata->rootdir_sect; 992 retdent = find_directory_entry(mydata, startsect, 993 l_filename, dentptr, 0); 994 if (retdent) { 995 /* Update file size and start_cluster in a directory entry */ 996 retdent->size = cpu_to_le32(size); 997 start_cluster = FAT2CPU16(retdent->start); 998 if (mydata->fatsize == 32) 999 start_cluster |= 1000 (FAT2CPU16(retdent->starthi) << 16); 1001 1002 ret = check_overflow(mydata, start_cluster, size); 1003 if (ret) { 1004 printf("Error: %ld overflow\n", size); 1005 goto exit; 1006 } 1007 1008 ret = clear_fatent(mydata, start_cluster); 1009 if (ret) { 1010 printf("Error: clearing FAT entries\n"); 1011 goto exit; 1012 } 1013 1014 ret = set_contents(mydata, retdent, buffer, size); 1015 if (ret) { 1016 printf("Error: writing contents\n"); 1017 goto exit; 1018 } 1019 1020 /* Flush fat buffer */ 1021 ret = flush_fat_buffer(mydata); 1022 if (ret) { 1023 printf("Error: flush fat buffer\n"); 1024 goto exit; 1025 } 1026 1027 /* Write directory table to device */ 1028 ret = set_cluster(mydata, dir_curclust, 1029 get_dentfromdir_block, 1030 mydata->clust_size * mydata->sect_size); 1031 if (ret) { 1032 printf("Error: wrinting directory entry\n"); 1033 goto exit; 1034 } 1035 } else { 1036 slotptr = (dir_slot *)empty_dentptr; 1037 1038 /* Set short name to set alias checksum field in dir_slot */ 1039 set_name(empty_dentptr, filename); 1040 fill_dir_slot(mydata, &empty_dentptr, filename); 1041 1042 ret = start_cluster = find_empty_cluster(mydata); 1043 if (ret < 0) { 1044 printf("Error: finding empty cluster\n"); 1045 goto exit; 1046 } 1047 1048 ret = check_overflow(mydata, start_cluster, size); 1049 if (ret) { 1050 printf("Error: %ld overflow\n", size); 1051 goto exit; 1052 } 1053 1054 /* Set attribute as archieve for regular file */ 1055 fill_dentry(mydata, empty_dentptr, filename, 1056 start_cluster, size, 0x20); 1057 1058 ret = set_contents(mydata, empty_dentptr, buffer, size); 1059 if (ret) { 1060 printf("Error: writing contents\n"); 1061 goto exit; 1062 } 1063 1064 /* Flush fat buffer */ 1065 ret = flush_fat_buffer(mydata); 1066 if (ret) { 1067 printf("Error: flush fat buffer\n"); 1068 goto exit; 1069 } 1070 1071 /* Write directory table to device */ 1072 ret = set_cluster(mydata, dir_curclust, 1073 get_dentfromdir_block, 1074 mydata->clust_size * mydata->sect_size); 1075 if (ret) { 1076 printf("Error: writing directory entry\n"); 1077 goto exit; 1078 } 1079 } 1080 1081 exit: 1082 free(mydata->fatbuf); 1083 return ret; 1084 } 1085 1086 int file_fat_write(const char *filename, void *buffer, unsigned long maxsize) 1087 { 1088 printf("writing %s\n", filename); 1089 return do_fat_write(filename, buffer, maxsize); 1090 } 1091