tizen 2.4 release
[kernel/u-boot-tm1.git] / fs / fat / fat_write.c
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 block, __u32 nr_blocks, void *buf)
45 {
46         if (!cur_dev || !cur_dev->block_write)
47                 return -1;
48
49         if (cur_part_info.start + block + nr_blocks >
50                 cur_part_info.start + total_sector) {
51                 printf("error: overflow occurs\n");
52                 return -1;
53         }
54
55         return cur_dev->block_write(cur_dev->dev,
56                         cur_part_info.start + block, nr_blocks, buf);
57 }
58
59 /*
60  * Set short name in directory entry
61  */
62 static void set_name(dir_entry *dirent, const char *filename)
63 {
64         char s_name[VFAT_MAXLEN_BYTES];
65         char *period;
66         int period_location, len, i, ext_num;
67
68         if (filename == NULL)
69                 return;
70
71         len = strlen(filename);
72         if (len == 0)
73                 return;
74
75         memcpy(s_name, filename, len);
76         uppercase(s_name, len);
77
78         period = strchr(s_name, '.');
79         if (period == NULL) {
80                 period_location = len;
81                 ext_num = 0;
82         } else {
83                 period_location = period - s_name;
84                 ext_num = len - period_location - 1;
85         }
86
87         /* Pad spaces when the length of file name is shorter than eight */
88         if (period_location < 8) {
89                 memcpy(dirent->name, s_name, period_location);
90                 for (i = period_location; i < 8; i++)
91                         dirent->name[i] = ' ';
92         } else if (period_location == 8) {
93                 memcpy(dirent->name, s_name, period_location);
94         } else {
95                 memcpy(dirent->name, s_name, 6);
96                 dirent->name[6] = '~';
97                 dirent->name[7] = '1';
98         }
99
100         if (ext_num < 3) {
101                 memcpy(dirent->ext, s_name + period_location + 1, ext_num);
102                 for (i = ext_num; i < 3; i++)
103                         dirent->ext[i] = ' ';
104         } else
105                 memcpy(dirent->ext, s_name + period_location + 1, 3);
106
107         debug("name : %s\n", dirent->name);
108         debug("ext : %s\n", dirent->ext);
109 }
110
111 static __u8 num_of_fats;
112 /*
113  * Write fat buffer into block device
114  */
115 static int flush_fat_buffer(fsdata *mydata)
116 {
117         int getsize = FATBUFBLOCKS;
118         __u32 fatlength = mydata->fatlength;
119         __u8 *bufptr = mydata->fatbuf;
120         __u32 startblock = mydata->fatbufnum * FATBUFBLOCKS;
121
122         fatlength *= mydata->sect_size;
123         startblock += mydata->fat_sect;
124
125         if (getsize > fatlength)
126                 getsize = fatlength;
127
128         /* Write FAT buf */
129         if (disk_write(startblock, getsize, bufptr) < 0) {
130                 debug("error: writing FAT blocks\n");
131                 return -1;
132         }
133
134         if (num_of_fats == 2) {
135                 /* Update corresponding second FAT blocks */
136                 startblock += mydata->fatlength;
137                 if (disk_write(startblock, getsize, bufptr) < 0) {
138                         debug("error: writing second FAT blocks\n");
139                         return -1;
140                 }
141         }
142
143         return 0;
144 }
145
146 /*
147  * Get the entry at index 'entry' in a FAT (12/16/32) table.
148  * On failure 0x00 is returned.
149  * When bufnum is changed, write back the previous fatbuf to the disk.
150  */
151 static __u32 get_fatent_value(fsdata *mydata, __u32 entry)
152 {
153         __u32 bufnum;
154         __u32 off16, offset;
155         __u32 ret = 0x00;
156         __u16 val1, val2;
157
158         switch (mydata->fatsize) {
159         case 32:
160                 bufnum = entry / FAT32BUFSIZE;
161                 offset = entry - bufnum * FAT32BUFSIZE;
162                 break;
163         case 16:
164                 bufnum = entry / FAT16BUFSIZE;
165                 offset = entry - bufnum * FAT16BUFSIZE;
166                 break;
167         case 12:
168                 bufnum = entry / FAT12BUFSIZE;
169                 offset = entry - bufnum * FAT12BUFSIZE;
170                 break;
171
172         default:
173                 /* Unsupported FAT size */
174                 return ret;
175         }
176
177         debug("FAT%d: entry: 0x%04x = %d, offset: 0x%04x = %d\n",
178                mydata->fatsize, entry, entry, offset, offset);
179
180         /* Read a new block of FAT entries into the cache. */
181         if (bufnum != mydata->fatbufnum) {
182                 int getsize = FATBUFBLOCKS;
183                 __u8 *bufptr = mydata->fatbuf;
184                 __u32 fatlength = mydata->fatlength;
185                 __u32 startblock = bufnum * FATBUFBLOCKS;
186
187                 if (getsize > fatlength)
188                         getsize = fatlength;
189
190                 fatlength *= mydata->sect_size; /* We want it in bytes now */
191                 startblock += mydata->fat_sect; /* Offset from start of disk */
192
193                 /* Write back the fatbuf to the disk */
194                 if (mydata->fatbufnum != -1) {
195                         if (flush_fat_buffer(mydata) < 0)
196                                 return -1;
197                 }
198
199                 if (disk_read(startblock, getsize, bufptr) < 0) {
200                         debug("Error reading FAT blocks\n");
201                         return ret;
202                 }
203                 mydata->fatbufnum = bufnum;
204         }
205
206         /* Get the actual entry from the table */
207         switch (mydata->fatsize) {
208         case 32:
209                 ret = FAT2CPU32(((__u32 *) mydata->fatbuf)[offset]);
210                 break;
211         case 16:
212                 ret = FAT2CPU16(((__u16 *) mydata->fatbuf)[offset]);
213                 break;
214         case 12:
215                 off16 = (offset * 3) / 4;
216
217                 switch (offset & 0x3) {
218                 case 0:
219                         ret = FAT2CPU16(((__u16 *) mydata->fatbuf)[off16]);
220                         ret &= 0xfff;
221                         break;
222                 case 1:
223                         val1 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16]);
224                         val1 &= 0xf000;
225                         val2 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16 + 1]);
226                         val2 &= 0x00ff;
227                         ret = (val2 << 4) | (val1 >> 12);
228                         break;
229                 case 2:
230                         val1 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16]);
231                         val1 &= 0xff00;
232                         val2 = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16 + 1]);
233                         val2 &= 0x000f;
234                         ret = (val2 << 8) | (val1 >> 8);
235                         break;
236                 case 3:
237                         ret = FAT2CPU16(((__u16 *)mydata->fatbuf)[off16]);
238                         ret = (ret & 0xfff0) >> 4;
239                         break;
240                 default:
241                         break;
242                 }
243                 break;
244         }
245         debug("FAT%d: ret: %08x, entry: %08x, offset: %04x\n",
246                mydata->fatsize, ret, entry, offset);
247
248         return ret;
249 }
250
251 #ifdef CONFIG_SUPPORT_VFAT
252 /*
253  * Set the file name information from 'name' into 'slotptr',
254  */
255 static int str2slot(dir_slot *slotptr, const char *name, int *idx)
256 {
257         int j, end_idx = 0;
258
259         for (j = 0; j <= 8; j += 2) {
260                 if (name[*idx] == 0x00) {
261                         slotptr->name0_4[j] = 0;
262                         slotptr->name0_4[j + 1] = 0;
263                         end_idx++;
264                         goto name0_4;
265                 }
266                 slotptr->name0_4[j] = name[*idx];
267                 (*idx)++;
268                 end_idx++;
269         }
270         for (j = 0; j <= 10; j += 2) {
271                 if (name[*idx] == 0x00) {
272                         slotptr->name5_10[j] = 0;
273                         slotptr->name5_10[j + 1] = 0;
274                         end_idx++;
275                         goto name5_10;
276                 }
277                 slotptr->name5_10[j] = name[*idx];
278                 (*idx)++;
279                 end_idx++;
280         }
281         for (j = 0; j <= 2; j += 2) {
282                 if (name[*idx] == 0x00) {
283                         slotptr->name11_12[j] = 0;
284                         slotptr->name11_12[j + 1] = 0;
285                         end_idx++;
286                         goto name11_12;
287                 }
288                 slotptr->name11_12[j] = name[*idx];
289                 (*idx)++;
290                 end_idx++;
291         }
292
293         if (name[*idx] == 0x00)
294                 return 1;
295
296         return 0;
297 /* Not used characters are filled with 0xff 0xff */
298 name0_4:
299         for (; end_idx < 5; end_idx++) {
300                 slotptr->name0_4[end_idx * 2] = 0xff;
301                 slotptr->name0_4[end_idx * 2 + 1] = 0xff;
302         }
303         end_idx = 5;
304 name5_10:
305         end_idx -= 5;
306         for (; end_idx < 6; end_idx++) {
307                 slotptr->name5_10[end_idx * 2] = 0xff;
308                 slotptr->name5_10[end_idx * 2 + 1] = 0xff;
309         }
310         end_idx = 11;
311 name11_12:
312         end_idx -= 11;
313         for (; end_idx < 2; end_idx++) {
314                 slotptr->name11_12[end_idx * 2] = 0xff;
315                 slotptr->name11_12[end_idx * 2 + 1] = 0xff;
316         }
317
318         return 1;
319 }
320
321 static int is_next_clust(fsdata *mydata, dir_entry *dentptr);
322 static void flush_dir_table(fsdata *mydata, dir_entry **dentptr);
323
324 /*
325  * Fill dir_slot entries with appropriate name, id, and attr
326  * The real directory entry is returned by 'dentptr'
327  */
328 static void
329 fill_dir_slot(fsdata *mydata, dir_entry **dentptr, const char *l_name)
330 {
331         dir_slot *slotptr = (dir_slot *)get_vfatname_block;
332         __u8 counter = 0, checksum;
333         int idx = 0, ret;
334         char s_name[16];
335
336         /* Get short file name and checksum value */
337         strncpy(s_name, (*dentptr)->name, 16);
338         checksum = mkcksum(s_name);
339
340         do {
341                 memset(slotptr, 0x00, sizeof(dir_slot));
342                 ret = str2slot(slotptr, l_name, &idx);
343                 slotptr->id = ++counter;
344                 slotptr->attr = ATTR_VFAT;
345                 slotptr->alias_checksum = checksum;
346                 slotptr++;
347         } while (ret == 0);
348
349         slotptr--;
350         slotptr->id |= LAST_LONG_ENTRY_MASK;
351
352         while (counter >= 1) {
353                 if (is_next_clust(mydata, *dentptr)) {
354                         /* A new cluster is allocated for directory table */
355                         flush_dir_table(mydata, dentptr);
356                 }
357                 memcpy(*dentptr, slotptr, sizeof(dir_slot));
358                 (*dentptr)++;
359                 slotptr--;
360                 counter--;
361         }
362
363         if (is_next_clust(mydata, *dentptr)) {
364                 /* A new cluster is allocated for directory table */
365                 flush_dir_table(mydata, dentptr);
366         }
367 }
368
369 static __u32 dir_curclust;
370
371 /*
372  * Extract the full long filename starting at 'retdent' (which is really
373  * a slot) into 'l_name'. If successful also copy the real directory entry
374  * into 'retdent'
375  * If additional adjacent cluster for directory entries is read into memory,
376  * then 'get_vfatname_block' is copied into 'get_dentfromdir_block' and
377  * the location of the real directory entry is returned by 'retdent'
378  * Return 0 on success, -1 otherwise.
379  */
380 static int
381 get_long_file_name(fsdata *mydata, int curclust, __u8 *cluster,
382               dir_entry **retdent, char *l_name)
383 {
384         dir_entry *realdent;
385         dir_slot *slotptr = (dir_slot *)(*retdent);
386         dir_slot *slotptr2 = NULL;
387         __u8 *buflimit = cluster + mydata->sect_size * ((curclust == 0) ?
388                                                         PREFETCH_BLOCKS :
389                                                         mydata->clust_size);
390         __u8 counter = (slotptr->id & ~LAST_LONG_ENTRY_MASK) & 0xff;
391         int idx = 0, cur_position = 0;
392
393         if (counter > VFAT_MAXSEQ) {
394                 debug("Error: VFAT name is too long\n");
395                 return -1;
396         }
397
398         while ((__u8 *)slotptr < buflimit) {
399                 if (counter == 0)
400                         break;
401                 if (((slotptr->id & ~LAST_LONG_ENTRY_MASK) & 0xff) != counter)
402                         return -1;
403                 slotptr++;
404                 counter--;
405         }
406
407         if ((__u8 *)slotptr >= buflimit) {
408                 if (curclust == 0)
409                         return -1;
410                 curclust = get_fatent_value(mydata, dir_curclust);
411                 if (CHECK_CLUST(curclust, mydata->fatsize)) {
412                         debug("curclust: 0x%x\n", curclust);
413                         printf("Invalid FAT entry\n");
414                         return -1;
415                 }
416
417                 dir_curclust = curclust;
418
419                 if (get_cluster(mydata, curclust, get_vfatname_block,
420                                 mydata->clust_size * mydata->sect_size) != 0) {
421                         debug("Error: reading directory block\n");
422                         return -1;
423                 }
424
425                 slotptr2 = (dir_slot *)get_vfatname_block;
426                 while (counter > 0) {
427                         if (((slotptr2->id & ~LAST_LONG_ENTRY_MASK)
428                             & 0xff) != counter)
429                                 return -1;
430                         slotptr2++;
431                         counter--;
432                 }
433
434                 /* Save the real directory entry */
435                 realdent = (dir_entry *)slotptr2;
436                 while ((__u8 *)slotptr2 > get_vfatname_block) {
437                         slotptr2--;
438                         slot2str(slotptr2, l_name, &idx);
439                 }
440         } else {
441                 /* Save the real directory entry */
442                 realdent = (dir_entry *)slotptr;
443         }
444
445         do {
446                 slotptr--;
447                 if (slot2str(slotptr, l_name, &idx))
448                         break;
449         } while (!(slotptr->id & LAST_LONG_ENTRY_MASK));
450
451         l_name[idx] = '\0';
452         if (*l_name == DELETED_FLAG)
453                 *l_name = '\0';
454         else if (*l_name == aRING)
455                 *l_name = DELETED_FLAG;
456         downcase(l_name);
457
458         /* Return the real directory entry */
459         *retdent = realdent;
460
461         if (slotptr2) {
462                 memcpy(get_dentfromdir_block, get_vfatname_block,
463                         mydata->clust_size * mydata->sect_size);
464                 cur_position = (__u8 *)realdent - get_vfatname_block;
465                 *retdent = (dir_entry *) &get_dentfromdir_block[cur_position];
466         }
467
468         return 0;
469 }
470
471 static int
472 delete_long_file_name(fsdata *mydata, int curclust, __u8 *cluster,
473               dir_entry **retdent)
474 {
475         dir_entry *realdent;
476         dir_slot *slotptr = (dir_slot *)(*retdent);
477         dir_slot *slotptr2 = NULL;
478         __u8 *buflimit = cluster + mydata->sect_size * ((curclust == 0) ?
479                                                         PREFETCH_BLOCKS :
480                                                         mydata->clust_size);
481         __u8 counter = (slotptr->id & ~LAST_LONG_ENTRY_MASK) & 0xff;
482         int idx = 0, cur_position = 0;
483
484         if (counter > VFAT_MAXSEQ) {
485                 debug("Error: VFAT name is too long\n");
486                 return -1;
487         }
488
489         while ((__u8 *)slotptr < buflimit) {
490                 if (counter == 0)
491                         break;
492                 if (((slotptr->id & ~LAST_LONG_ENTRY_MASK) & 0xff) != counter)
493                         return -1;
494                 slotptr->id = DELETED_FLAG;
495                 slotptr++;
496                 counter--;
497         }
498
499         if ((__u8 *)slotptr >= buflimit) {
500                 if (curclust == 0)
501                         return -1;
502                 curclust = get_fatent_value(mydata, dir_curclust);
503                 if (CHECK_CLUST(curclust, mydata->fatsize)) {
504                         debug("curclust: 0x%x\n", curclust);
505                         printf("Invalid FAT entry\n");
506                         return -1;
507                 }
508
509                 dir_curclust = curclust;
510
511                 if (get_cluster(mydata, curclust, get_vfatname_block,
512                                 mydata->clust_size * mydata->sect_size) != 0) {
513                         debug("Error: reading directory block\n");
514                         return -1;
515                 }
516
517                 slotptr2 = (dir_slot *)get_vfatname_block;
518                 while (counter > 0) {
519                         if (((slotptr2->id & ~LAST_LONG_ENTRY_MASK)
520                             & 0xff) != counter)
521                                 return -1;
522                         slotptr2->id = DELETED_FLAG;
523                         slotptr2++;
524                         counter--;
525                 }
526
527                 /* Save the real directory entry */
528                 realdent = (dir_entry *)slotptr2;
529         } else {
530                 /* Save the real directory entry */
531                 realdent = (dir_entry *)slotptr;
532         }
533         realdent->name[0] = DELETED_FLAG;
534
535         return 0;
536 }
537 #endif
538
539 /*
540  * Set the entry at index 'entry' in a FAT (16/32) table.
541  */
542 static int set_fatent_value(fsdata *mydata, __u32 entry, __u32 entry_value)
543 {
544         __u32 bufnum, offset;
545
546         switch (mydata->fatsize) {
547         case 32:
548                 bufnum = entry / FAT32BUFSIZE;
549                 offset = entry - bufnum * FAT32BUFSIZE;
550                 break;
551         case 16:
552                 bufnum = entry / FAT16BUFSIZE;
553                 offset = entry - bufnum * FAT16BUFSIZE;
554                 break;
555         default:
556                 /* Unsupported FAT size */
557                 return -1;
558         }
559
560         /* Read a new block of FAT entries into the cache. */
561         if (bufnum != mydata->fatbufnum) {
562                 int getsize = FATBUFBLOCKS;
563                 __u8 *bufptr = mydata->fatbuf;
564                 __u32 fatlength = mydata->fatlength;
565                 __u32 startblock = bufnum * FATBUFBLOCKS;
566
567                 fatlength *= mydata->sect_size;
568                 startblock += mydata->fat_sect;
569
570                 if (getsize > fatlength)
571                         getsize = fatlength;
572
573                 if (mydata->fatbufnum != -1) {
574                         if (flush_fat_buffer(mydata) < 0)
575                                 return -1;
576                 }
577
578                 if (disk_read(startblock, getsize, bufptr) < 0) {
579                         debug("Error reading FAT blocks\n");
580                         return -1;
581                 }
582                 mydata->fatbufnum = bufnum;
583         }
584
585         /* Set the actual entry */
586         switch (mydata->fatsize) {
587         case 32:
588                 ((__u32 *) mydata->fatbuf)[offset] = cpu_to_le32(entry_value);
589                 break;
590         case 16:
591                 ((__u16 *) mydata->fatbuf)[offset] = cpu_to_le16(entry_value);
592                 break;
593         default:
594                 return -1;
595         }
596
597         return 0;
598 }
599
600 /*
601  * Determine the entry value at index 'entry' in a FAT (16/32) table
602  */
603 static __u32 determine_fatent(fsdata *mydata, __u32 entry)
604 {
605         __u32 next_fat, next_entry = entry + 1;
606
607         while (1) {
608                 next_fat = get_fatent_value(mydata, next_entry);
609                 if (next_fat == 0) {
610                         set_fatent_value(mydata, entry, next_entry);
611                         break;
612                 }
613                 next_entry++;
614         }
615         debug("FAT%d: entry: %08x, entry_value: %04x\n",
616                mydata->fatsize, entry, next_entry);
617
618         return next_entry;
619 }
620
621 /*
622  * Write at most 'size' bytes from 'buffer' into the specified cluster.
623  * Return 0 on success, -1 otherwise.
624  */
625 static int
626 set_cluster(fsdata *mydata, __u32 clustnum, __u8 *buffer,
627              unsigned long size)
628 {
629         int idx = 0;
630         __u32 startsect;
631
632         if (clustnum > 0)
633                 startsect = mydata->data_begin +
634                                 clustnum * mydata->clust_size;
635         else
636                 startsect = mydata->rootdir_sect;
637
638         debug("clustnum: %d, startsect: %d\n", clustnum, startsect);
639
640         if (disk_write(startsect, size / mydata->sect_size, buffer) < 0) {
641                 debug("Error writing data\n");
642                 return -1;
643         }
644
645         if (size % mydata->sect_size) {
646                 __u8 tmpbuf[mydata->sect_size];
647
648                 idx = size / mydata->sect_size;
649                 buffer += idx * mydata->sect_size;
650                 memcpy(tmpbuf, buffer, size % mydata->sect_size);
651
652                 if (disk_write(startsect + idx, 1, tmpbuf) < 0) {
653                         debug("Error writing data\n");
654                         return -1;
655                 }
656
657                 return 0;
658         }
659
660         return 0;
661 }
662
663 /*
664  * Find the first empty cluster
665  */
666 static int find_empty_cluster(fsdata *mydata)
667 {
668         __u32 fat_val, entry = 3;
669
670         while (1) {
671                 fat_val = get_fatent_value(mydata, entry);
672                 if (fat_val == 0)
673                         break;
674                 entry++;
675         }
676
677         return entry;
678 }
679
680 /*
681  * Write directory entries in 'get_dentfromdir_block' to block device
682  */
683 static void flush_dir_table(fsdata *mydata, dir_entry **dentptr)
684 {
685         int dir_newclust = 0;
686
687         if (set_cluster(mydata, dir_curclust,
688                     get_dentfromdir_block,
689                     mydata->clust_size * mydata->sect_size) != 0) {
690                 printf("error: wrinting directory entry\n");
691                 return;
692         }
693         dir_newclust = find_empty_cluster(mydata);
694         set_fatent_value(mydata, dir_curclust, dir_newclust);
695         if (mydata->fatsize == 32)
696                 set_fatent_value(mydata, dir_newclust, 0xffffff8);
697         else if (mydata->fatsize == 16)
698                 set_fatent_value(mydata, dir_newclust, 0xfff8);
699
700         dir_curclust = dir_newclust;
701
702         if (flush_fat_buffer(mydata) < 0)
703                 return;
704
705         memset(get_dentfromdir_block, 0x00,
706                 mydata->clust_size * mydata->sect_size);
707
708         *dentptr = (dir_entry *) get_dentfromdir_block;
709 }
710
711 /*
712  * Set empty cluster from 'entry' to the end of a file
713  */
714 static int clear_fatent(fsdata *mydata, __u32 entry)
715 {
716         __u32 fat_val;
717
718         while (1) {
719                 fat_val = get_fatent_value(mydata, entry);
720                 if (fat_val != 0)
721                         set_fatent_value(mydata, entry, 0);
722                 else
723                         break;
724
725                 if (fat_val == 0xfffffff || fat_val == 0xffff)
726                         break;
727
728                 entry = fat_val;
729         }
730
731         /* Flush fat buffer */
732         if (flush_fat_buffer(mydata) < 0)
733                 return -1;
734
735         return 0;
736 }
737
738 /*
739  * Write at most 'maxsize' bytes from 'buffer' into
740  * the file associated with 'dentptr'
741  * Return the number of bytes read or -1 on fatal errors.
742  */
743 static int
744 set_contents(fsdata *mydata, dir_entry *dentptr, __u8 *buffer,
745               unsigned long maxsize)
746 {
747         unsigned long filesize = FAT2CPU32(dentptr->size), gotsize = 0;
748         unsigned int bytesperclust = mydata->clust_size * mydata->sect_size;
749         __u32 curclust = START(dentptr);
750         __u32 endclust = 0, newclust = 0;
751         unsigned long actsize;
752
753         debug("Filesize: %ld bytes\n", filesize);
754
755         if (maxsize > 0 && filesize > maxsize)
756                 filesize = maxsize;
757
758         debug("%ld bytes\n", filesize);
759
760         actsize = bytesperclust;
761         endclust = curclust;
762         do {
763                 /* search for consecutive clusters */
764                 while (actsize < filesize) {
765                         newclust = determine_fatent(mydata, endclust);
766
767                         if ((newclust - 1) != endclust)
768                                 goto getit;
769
770                         if (CHECK_CLUST(newclust, mydata->fatsize)) {
771                                 debug("curclust: 0x%x\n", newclust);
772                                 debug("Invalid FAT entry\n");
773                                 return gotsize;
774                         }
775                         endclust = newclust;
776                         actsize += bytesperclust;
777                 }
778                 /* actsize >= file size */
779                 actsize -= bytesperclust;
780                 /* set remaining clusters */
781                 if (set_cluster(mydata, curclust, buffer, (int)actsize) != 0) {
782                         debug("error: writing cluster\n");
783                         return -1;
784                 }
785
786                 /* set remaining bytes */
787                 gotsize += (int)actsize;
788                 filesize -= actsize;
789                 buffer += actsize;
790                 actsize = filesize;
791
792                 if (set_cluster(mydata, endclust, buffer, (int)actsize) != 0) {
793                         debug("error: writing cluster\n");
794                         return -1;
795                 }
796                 gotsize += actsize;
797
798                 /* Mark end of file in FAT */
799                 if (mydata->fatsize == 16)
800                         newclust = 0xffff;
801                 else if (mydata->fatsize == 32)
802                         newclust = 0xfffffff;
803                 set_fatent_value(mydata, endclust, newclust);
804
805                 return gotsize;
806 getit:
807                 if (set_cluster(mydata, curclust, buffer, (int)actsize) != 0) {
808                         debug("error: writing cluster\n");
809                         return -1;
810                 }
811                 gotsize += (int)actsize;
812                 filesize -= actsize;
813                 buffer += actsize;
814
815                 if (CHECK_CLUST(curclust, mydata->fatsize)) {
816                         debug("curclust: 0x%x\n", curclust);
817                         debug("Invalid FAT entry\n");
818                         return gotsize;
819                 }
820                 actsize = bytesperclust;
821                 curclust = endclust = newclust;
822         } while (1);
823 }
824
825 /*
826  * Fill dir_entry
827  */
828 static void fill_dentry(fsdata *mydata, dir_entry *dentptr,
829         const char *filename, __u32 start_cluster, __u32 size, __u8 attr)
830 {
831         if (mydata->fatsize == 32)
832                 dentptr->starthi =
833                         cpu_to_le16((start_cluster & 0xffff0000) >> 16);
834         dentptr->start = cpu_to_le16(start_cluster & 0xffff);
835         dentptr->size = cpu_to_le32(size);
836
837         dentptr->attr = attr;
838
839         set_name(dentptr, filename);
840 }
841
842 /*
843  * Check whether adding a file makes the file system to
844  * exceed the size of the block device
845  * Return -1 when overflow occurs, otherwise return 0
846  */
847 static int check_overflow(fsdata *mydata, __u32 clustnum, unsigned long size)
848 {
849         __u32 startsect, sect_num;
850
851         if (clustnum > 0) {
852                 startsect = mydata->data_begin +
853                                 clustnum * mydata->clust_size;
854         } else {
855                 startsect = mydata->rootdir_sect;
856         }
857
858         sect_num = size / mydata->sect_size;
859         if (size % mydata->sect_size)
860                 sect_num++;
861
862         if (startsect + sect_num > cur_part_info.start + total_sector)
863                 return -1;
864
865         return 0;
866 }
867
868 /*
869  * Check if adding several entries exceed one cluster boundary
870  */
871 static int is_next_clust(fsdata *mydata, dir_entry *dentptr)
872 {
873         int cur_position;
874
875         cur_position = (__u8 *)dentptr - get_dentfromdir_block;
876
877         if (cur_position >= mydata->clust_size * mydata->sect_size)
878                 return 1;
879         else
880                 return 0;
881 }
882
883 static dir_entry *empty_dentptr;
884 /*
885  * Find a directory entry based on filename or start cluster number
886  * If the directory entry is not found,
887  * the new position for writing a directory entry will be returned
888  */
889 static dir_entry *find_directory_entry(fsdata *mydata, int startsect,
890         char *filename, dir_entry *retdent, __u32 start)
891 {
892         __u32 curclust = (startsect - mydata->data_begin) / mydata->clust_size;
893
894         debug("get_dentfromdir: %s\n", filename);
895
896         while (1) {
897                 dir_entry *dentptr;
898
899                 int i;
900
901                 if (get_cluster(mydata, curclust, get_dentfromdir_block,
902                             mydata->clust_size * mydata->sect_size) != 0) {
903                         printf("Error: reading directory block\n");
904                         return NULL;
905                 }
906
907                 dentptr = (dir_entry *)get_dentfromdir_block;
908
909                 dir_curclust = curclust;
910
911                 for (i = 0; i < DIRENTSPERCLUST; i++) {
912                         char s_name[14], l_name[VFAT_MAXLEN_BYTES];
913
914                         l_name[0] = '\0';
915                         if (dentptr->name[0] == DELETED_FLAG) {
916                                 dentptr++;
917                                 if (is_next_clust(mydata, dentptr))
918                                         break;
919                                 continue;
920                         }
921                         if ((dentptr->attr & ATTR_VOLUME)) {
922 #ifdef CONFIG_SUPPORT_VFAT
923                                 if ((dentptr->attr & ATTR_VFAT) &&
924                                     (dentptr->name[0] & LAST_LONG_ENTRY_MASK)) {
925                                         get_long_file_name(mydata, curclust,
926                                                      get_dentfromdir_block,
927                                                      &dentptr, l_name);
928                                         debug("vfatname: |%s|\n", l_name);
929                                 } else
930 #endif
931                                 {
932                                         /* Volume label or VFAT entry */
933                                         dentptr++;
934                                         if (is_next_clust(mydata, dentptr))
935                                                 break;
936                                         continue;
937                                 }
938                         }
939                         if (dentptr->name[0] == 0) {
940                                 debug("Dentname == NULL - %d\n", i);
941                                 empty_dentptr = dentptr;
942                                 return NULL;
943                         }
944
945                         get_name(dentptr, s_name);
946
947                         if (strcmp(filename, s_name)
948                             && strcmp(filename, l_name)) {
949                                 debug("Mismatch: |%s|%s|\n",
950                                         s_name, l_name);
951                                 dentptr++;
952                                 if (is_next_clust(mydata, dentptr))
953                                         break;
954                                 continue;
955                         }
956
957                         memcpy(retdent, dentptr, sizeof(dir_entry));
958
959                         debug("DentName: %s", s_name);
960                         debug(", start: 0x%x", START(dentptr));
961                         debug(", size:  0x%x %s\n",
962                               FAT2CPU32(dentptr->size),
963                               (dentptr->attr & ATTR_DIR) ?
964                               "(DIR)" : "");
965
966                         return dentptr;
967                 }
968
969                 curclust = get_fatent_value(mydata, dir_curclust);
970                 if ((curclust >= 0xffffff8) || (curclust >= 0xfff8)) {
971                         empty_dentptr = dentptr;
972                         return NULL;
973                 }
974                 if (CHECK_CLUST(curclust, mydata->fatsize)) {
975                         debug("curclust: 0x%x\n", curclust);
976                         debug("Invalid FAT entry\n");
977                         return NULL;
978                 }
979         }
980
981         return NULL;
982 }
983
984 static int do_fat_write(const char *filename, void *buffer,
985         unsigned long size)
986 {
987         dir_entry *dentptr, *retdent;
988         __u32 startsect;
989         __u32 start_cluster;
990         boot_sector bs;
991         volume_info volinfo;
992         fsdata datablock;
993         fsdata *mydata = &datablock;
994         int cursect;
995         int ret = -1, name_len;
996         char l_filename[VFAT_MAXLEN_BYTES];
997         int write_size = size;
998
999         dir_curclust = 0;
1000
1001         if (read_bootsectandvi(&bs, &volinfo, &mydata->fatsize)) {
1002                 debug("error: reading boot sector\n");
1003                 return -1;
1004         }
1005
1006         total_sector = bs.total_sect;
1007         if (total_sector == 0)
1008                 total_sector = cur_part_info.size;
1009
1010         if (mydata->fatsize == 32)
1011                 mydata->fatlength = bs.fat32_length;
1012         else
1013                 mydata->fatlength = bs.fat_length;
1014
1015         mydata->fat_sect = bs.reserved;
1016
1017         cursect = mydata->rootdir_sect
1018                 = mydata->fat_sect + mydata->fatlength * bs.fats;
1019         num_of_fats = bs.fats;
1020
1021         mydata->sect_size = (bs.sector_size[1] << 8) + bs.sector_size[0];
1022         mydata->clust_size = bs.cluster_size;
1023
1024         if (mydata->fatsize == 32) {
1025                 mydata->data_begin = mydata->rootdir_sect -
1026                                         (mydata->clust_size * 2);
1027         } else {
1028                 int rootdir_size;
1029
1030                 rootdir_size = ((bs.dir_entries[1]  * (int)256 +
1031                                  bs.dir_entries[0]) *
1032                                  sizeof(dir_entry)) /
1033                                  mydata->sect_size;
1034                 mydata->data_begin = mydata->rootdir_sect +
1035                                         rootdir_size -
1036                                         (mydata->clust_size * 2);
1037         }
1038
1039         mydata->fatbufnum = -1;
1040         mydata->fatbuf = malloc(FATBUFSIZE);
1041         if (mydata->fatbuf == NULL) {
1042                 debug("Error: allocating memory\n");
1043                 return -1;
1044         }
1045
1046         if (disk_read(cursect,
1047                 (mydata->fatsize == 32) ?
1048                 (mydata->clust_size) :
1049                 PREFETCH_BLOCKS, do_fat_read_block) < 0) {
1050                 debug("Error: reading rootdir block\n");
1051                 goto exit;
1052         }
1053         dentptr = (dir_entry *) do_fat_read_block;
1054
1055         name_len = strlen(filename);
1056         if (name_len >= VFAT_MAXLEN_BYTES)
1057                 name_len = VFAT_MAXLEN_BYTES - 1;
1058
1059         memcpy(l_filename, filename, name_len);
1060         l_filename[name_len] = 0; /* terminate the string */
1061         downcase(l_filename);
1062
1063         startsect = mydata->rootdir_sect;
1064         retdent = find_directory_entry(mydata, startsect,
1065                                 l_filename, dentptr, 0);
1066         if (retdent) {
1067                 /* Update file size and start_cluster in a directory entry */
1068                 retdent->size = cpu_to_le32(size);
1069                 start_cluster = FAT2CPU16(retdent->start);
1070                 if (mydata->fatsize == 32)
1071                         start_cluster |=
1072                                 (FAT2CPU16(retdent->starthi) << 16);
1073
1074                 ret = check_overflow(mydata, start_cluster, size);
1075                 if (ret) {
1076                         printf("Error: %ld overflow\n", size);
1077                         goto exit;
1078                 }
1079
1080                 ret = clear_fatent(mydata, start_cluster);
1081                 if (ret) {
1082                         printf("Error: clearing FAT entries\n");
1083                         goto exit;
1084                 }
1085
1086                 ret = set_contents(mydata, retdent, buffer, size);
1087                 if (ret < 0) {
1088                         printf("Error: writing contents\n");
1089                         goto exit;
1090                 }
1091                 write_size = ret;
1092                 debug("attempt to write 0x%x bytes\n", write_size);
1093
1094                 /* Flush fat buffer */
1095                 ret = flush_fat_buffer(mydata);
1096                 if (ret) {
1097                         printf("Error: flush fat buffer\n");
1098                         goto exit;
1099                 }
1100
1101                 /* Write directory table to device */
1102                 ret = set_cluster(mydata, dir_curclust,
1103                             get_dentfromdir_block,
1104                             mydata->clust_size * mydata->sect_size);
1105                 if (ret) {
1106                         printf("Error: writing directory entry\n");
1107                         goto exit;
1108                 }
1109         } else {
1110                 /* Set short name to set alias checksum field in dir_slot */
1111                 set_name(empty_dentptr, filename);
1112                 fill_dir_slot(mydata, &empty_dentptr, filename);
1113
1114                 ret = start_cluster = find_empty_cluster(mydata);
1115                 if (ret < 0) {
1116                         printf("Error: finding empty cluster\n");
1117                         goto exit;
1118                 }
1119
1120                 ret = check_overflow(mydata, start_cluster, size);
1121                 if (ret) {
1122                         printf("Error: %ld overflow\n", size);
1123                         goto exit;
1124                 }
1125
1126                 /* Set attribute as archieve for regular file */
1127                 fill_dentry(mydata, empty_dentptr, filename,
1128                         start_cluster, size, 0x20);
1129
1130                 ret = set_contents(mydata, empty_dentptr, buffer, size);
1131                 if (ret < 0) {
1132                         printf("Error: writing contents\n");
1133                         goto exit;
1134                 }
1135                 write_size = ret;
1136                 debug("attempt to write 0x%x bytes\n", write_size);
1137
1138                 /* Flush fat buffer */
1139                 ret = flush_fat_buffer(mydata);
1140                 if (ret) {
1141                         printf("Error: flush fat buffer\n");
1142                         goto exit;
1143                 }
1144
1145                 /* Write directory table to device */
1146                 ret = set_cluster(mydata, dir_curclust,
1147                             get_dentfromdir_block,
1148                             mydata->clust_size * mydata->sect_size);
1149                 if (ret) {
1150                         printf("Error: writing directory entry\n");
1151                         goto exit;
1152                 }
1153         }
1154
1155 exit:
1156         free(mydata->fatbuf);
1157         return ret < 0 ? ret : write_size;
1158 }
1159
1160 int file_fat_write(const char *filename, void *buffer, unsigned long maxsize)
1161 {
1162         printf("writing %s\n", filename);
1163         return do_fat_write(filename, buffer, maxsize);
1164 }
1165
1166 static int delete_fatent(fsdata *mydata, __u32 entry)
1167 {
1168         __u32 fat_val;
1169
1170         while (1) {
1171                 fat_val = get_fatent_value(mydata, entry);
1172                 if (fat_val != 0) {
1173                         set_fatent_value(mydata, entry, 0);
1174                 }
1175                 else
1176                         break;
1177
1178                 if (fat_val == 0xfffffff || fat_val == 0xffff) {
1179                         break;
1180                 }
1181
1182                 entry = fat_val;
1183         }
1184
1185         /* Flush fat buffer */
1186         if (flush_fat_buffer(mydata) < 0)
1187                 return -1;
1188
1189         return 0;
1190 }
1191
1192 static dir_entry *delete_directory_entry(fsdata *mydata, int startsect,
1193         char *filename, dir_entry *retdent, __u32 start)
1194 {
1195         __u32 curclust = (startsect - mydata->data_begin) / mydata->clust_size;
1196         __u32 find_curclust = curclust;
1197 #ifdef CONFIG_SUPPORT_VFAT
1198         dir_entry *find_dentptr = NULL;
1199 #endif
1200
1201         debug("get_dentfromdir: %s\n", filename);
1202
1203         while (1) {
1204                 dir_entry *dentptr;
1205
1206                 int i;
1207                 int mark_cnt;
1208
1209                 if (get_cluster(mydata, curclust, get_dentfromdir_block,
1210                             mydata->clust_size * mydata->sect_size) != 0) {
1211                         printf("Error: reading directory block\n");
1212                         return NULL;
1213                 }
1214
1215                 dentptr = (dir_entry *)get_dentfromdir_block;
1216
1217                 dir_curclust = curclust;
1218
1219                 for (i = 0; i < DIRENTSPERCLUST; i++) {
1220                         char s_name[14], l_name[VFAT_MAXLEN_BYTES];
1221
1222                         l_name[0] = '\0';
1223                         if (dentptr->name[0] == DELETED_FLAG) {
1224                                 dentptr++;
1225                                 if (is_next_clust(mydata, dentptr))
1226                                         break;
1227                                 continue;
1228                         }
1229                         if ((dentptr->attr & ATTR_VOLUME)) {
1230 #ifdef CONFIG_SUPPORT_VFAT
1231                                 if ((dentptr->attr & ATTR_VFAT) &&
1232                                     (dentptr->name[0] & LAST_LONG_ENTRY_MASK)) {
1233                                         find_dentptr = dentptr;
1234                                         find_curclust = curclust;
1235                                         get_long_file_name(mydata, curclust,
1236                                                      get_dentfromdir_block,
1237                                                      &dentptr, l_name);
1238                                         debug("vfatname: |%s|\n", l_name);
1239                                 } else
1240 #endif
1241                                 {
1242                                         /* Volume label or VFAT entry */
1243                                         dentptr++;
1244                                         if (is_next_clust(mydata, dentptr))
1245                                                 break;
1246                                         continue;
1247                                 }
1248                         }
1249                         if (dentptr->name[0] == 0) {
1250                                 debug("Dentname == NULL - %d\n", i);
1251                                 empty_dentptr = dentptr;
1252                                 return NULL;
1253                         }
1254
1255                         get_name(dentptr, s_name);
1256
1257                         if (strcmp(filename, s_name)
1258                             && strcmp(filename, l_name)) {
1259                                 debug("Mismatch: |%s|%s|\n",
1260                                         s_name, l_name);
1261                                 dentptr++;
1262                                 if (is_next_clust(mydata, dentptr))
1263                                         break;
1264                                 continue;
1265                         }
1266
1267                         memcpy(retdent, dentptr, sizeof(dir_entry));
1268 #ifdef CONFIG_SUPPORT_VFAT
1269                         delete_long_file_name(mydata, find_curclust,
1270                                         get_dentfromdir_block,
1271                                         &find_dentptr);
1272 #else
1273                         dentptr->name[0] = DELETED_FLAG;
1274 #endif
1275                         if (set_cluster(mydata, dir_curclust,
1276                                                 get_dentfromdir_block,
1277                                                 mydata->clust_size * mydata->sect_size) != 0) {
1278                                 printf("error: wrinting directory entry\n");
1279                                 return NULL;
1280                         }
1281                         debug("DentName: %s", s_name);
1282                         debug(", start: 0x%x", START(dentptr));
1283                         debug(", size:  0x%x %s\n",
1284                               FAT2CPU32(dentptr->size),
1285                               (dentptr->attr & ATTR_DIR) ?
1286                               "(DIR)" : "");
1287
1288                         return dentptr;
1289                 }
1290
1291                 curclust = get_fatent_value(mydata, dir_curclust);
1292                 if ((curclust >= 0xffffff8) || (curclust >= 0xfff8)) {
1293                         empty_dentptr = dentptr;
1294                         return NULL;
1295                 }
1296                 if (CHECK_CLUST(curclust, mydata->fatsize)) {
1297                         debug("curclust: 0x%x\n", curclust);
1298                         debug("Invalid FAT entry\n");
1299                         return NULL;
1300                 }
1301         }
1302
1303         return NULL;
1304 }
1305
1306 static int do_fat_rm(const char *filename)
1307 {
1308         dir_entry *dentptr, *retdent;
1309         __u32 startsect;
1310         __u32 start_cluster;
1311         boot_sector bs;
1312         volume_info volinfo;
1313         fsdata datablock;
1314         fsdata *mydata = &datablock;
1315         int cursect;
1316         int ret = -1, name_len;
1317         char l_filename[VFAT_MAXLEN_BYTES];
1318
1319         dir_curclust = 0;
1320
1321         if (read_bootsectandvi(&bs, &volinfo, &mydata->fatsize)) {
1322                 debug("error: reading boot sector\n");
1323                 return -1;
1324         }
1325
1326         total_sector = bs.total_sect;
1327         if (total_sector == 0)
1328                 total_sector = cur_part_info.size;
1329
1330         if (mydata->fatsize == 32)
1331                 mydata->fatlength = bs.fat32_length;
1332         else
1333                 mydata->fatlength = bs.fat_length;
1334
1335         mydata->fat_sect = bs.reserved;
1336
1337         cursect = mydata->rootdir_sect
1338                 = mydata->fat_sect + mydata->fatlength * bs.fats;
1339         num_of_fats = bs.fats;
1340
1341         mydata->sect_size = (bs.sector_size[1] << 8) + bs.sector_size[0];
1342         mydata->clust_size = bs.cluster_size;
1343
1344         if (mydata->fatsize == 32) {
1345                 mydata->data_begin = mydata->rootdir_sect -
1346                                         (mydata->clust_size * 2);
1347         } else {
1348                 int rootdir_size;
1349
1350                 rootdir_size = ((bs.dir_entries[1]  * (int)256 +
1351                                  bs.dir_entries[0]) *
1352                                  sizeof(dir_entry)) /
1353                                  mydata->sect_size;
1354                 mydata->data_begin = mydata->rootdir_sect +
1355                                         rootdir_size -
1356                                         (mydata->clust_size * 2);
1357         }
1358
1359         mydata->fatbufnum = -1;
1360         mydata->fatbuf = malloc(FATBUFSIZE);
1361         if (mydata->fatbuf == NULL) {
1362                 debug("Error: allocating memory\n");
1363                 return -1;
1364         }
1365
1366         if (disk_read(cursect,
1367                 (mydata->fatsize == 32) ?
1368                 (mydata->clust_size) :
1369                 PREFETCH_BLOCKS, do_fat_read_block) < 0) {
1370                 debug("Error: reading rootdir block\n");
1371                 goto exit;
1372         }
1373         dentptr = (dir_entry *) do_fat_read_block;
1374
1375         name_len = strlen(filename);
1376         if (name_len >= VFAT_MAXLEN_BYTES)
1377                 name_len = VFAT_MAXLEN_BYTES - 1;
1378
1379         memcpy(l_filename, filename, name_len);
1380         l_filename[name_len] = 0; /* terminate the string */
1381         downcase(l_filename);
1382
1383         startsect = mydata->rootdir_sect;
1384         retdent = delete_directory_entry(mydata, startsect,
1385                                 l_filename, dentptr, 0);
1386         int i;
1387         if (retdent) {
1388                 /* Update file size and start_cluster in a directory entry */
1389                 start_cluster = FAT2CPU16(retdent->start);
1390                 if (mydata->fatsize == 32)
1391                         start_cluster |=
1392                                 (FAT2CPU16(retdent->starthi) << 16);
1393
1394                 ret = delete_fatent(mydata, start_cluster);
1395                 if (ret) {
1396                         printf("Error: clearing FAT entries\n");
1397                         goto exit;
1398                 }
1399
1400                 /* Write directory table to device */
1401                 ret = set_cluster(mydata, dir_curclust,
1402                             get_dentfromdir_block,
1403                             mydata->clust_size * mydata->sect_size);
1404                 if (ret) {
1405                         printf("Error: writing directory entry\n");
1406                         goto exit;
1407                 }
1408         } else {
1409                 printf("Error: %s not found\n", filename);
1410                 goto exit;
1411         }
1412
1413 exit:
1414         free(mydata->fatbuf);
1415         return ret < 0 ? ret : 0;
1416 }
1417
1418 int file_fat_rm(const char *filename)
1419 {
1420         printf("deleting %s\n", filename);
1421         return do_fat_rm(filename);
1422 }