Upstream-Status: Inappropriate This patch fixes populated dosfs image creation with directory structures. Earlier it was causing segfault; and only image population with no subdirectories was working. Issues fixed: 1. (dir->count == dir->entries) check was only needed for root directory entries. And this check is wrong for non-root directories. 2. For each dir entry 2 dir->table entries were needed, one for the file/dir and 2nd for long file name support. Earlier long name support was added for filenames but the 2nd entry allocation, initialization & counting was missed. 3. The memory clearing was missed at the code path after dir->table memroy allocation. 4. Add entries for . & .. directories in all non-root directories. 5. The . directory points to the correct entry in fat now. 6. All directoriy entries' size was not zero as required for dosfsck, Now all directory entries' size is zero. Enhancements: 1. Added support for long names for directory names. This is same as the existing long name support for filenames. 2. Added error messages for previously silent memory allocation and other errors. 3. -d options does not work correctly with fat32, so now throwing an error for that. 4. Use predefined structures from kernel's msdos_fs.h file, rather than defining again here. And accordingly change the names & use of structure variables. Outstanding Issues: 1. The .. directory entry do not point to the parent of current directory. This issue can be fixed by running dosfsck -a after image creation. 2. For files the filesize is correct, but the clusters size is more than it needs to be, this also can be fixed by running dosfsck -a after image creation. Signed-off-by: Nitin A Kamble 2011/12/13 Index: dosfstools-2.11/mkdosfs/mkdosfs.c =================================================================== --- dosfstools-2.11.orig/mkdosfs/mkdosfs.c +++ dosfstools-2.11/mkdosfs/mkdosfs.c @@ -21,7 +21,17 @@ June 2004 - Jordan Crouse (info.linux@amd.com) Added -d support to populate the image Copyright (C) 2004, Advanced Micro Devices, All Rights Reserved - + + 2011-12-13: Nitin A Kamble + Enhanced the -d support for population of image while + creation. Earlier subdirectores support was broken, only files in + the rootdir were supported. Now directory hirarchy is supported. + Also added long filename support to directory names. + The -d option (image population while creation) + is broken with fat32. + Copyright (C) 2011, Intel Corporation, All Rights Reserved + + Fixes/additions May 1998 by Roman Hodek : - Atari format support @@ -86,23 +96,23 @@ # undef __KERNEL__ #endif -#if __BYTE_ORDER == __BIG_ENDIAN - +#ifndef __ASM_STUB_BYTEORDER_H__ #include -#ifdef __le16_to_cpu -/* ++roman: 2.1 kernel headers define these function, they're probably more - * efficient then coding the swaps machine-independently. */ -#define CF_LE_W __le16_to_cpu -#define CF_LE_L __le32_to_cpu -#define CT_LE_W __cpu_to_le16 -#define CT_LE_L __cpu_to_le32 -#else -#define CF_LE_W(v) ((((v) & 0xff) << 8) | (((v) >> 8) & 0xff)) -#define CF_LE_L(v) (((unsigned)(v)>>24) | (((unsigned)(v)>>8)&0xff00) | \ - (((unsigned)(v)<<8)&0xff0000) | ((unsigned)(v)<<24)) +#endif + +#include + +#undef CF_LE_W +#undef CF_LE_L +#undef CT_LE_W +#undef CT_LE_L + +#if __BYTE_ORDER == __BIG_ENDIAN +#include +#define CF_LE_W(v) bswap_16(v) +#define CF_LE_L(v) bswap_32(v) #define CT_LE_W(v) CF_LE_W(v) #define CT_LE_L(v) CF_LE_L(v) -#endif /* defined(__le16_to_cpu) */ #else @@ -253,33 +263,6 @@ struct fat32_fsinfo { __u32 reserved2[4]; }; -/* This stores up to 13 chars of the name */ - -struct msdos_dir_slot { - __u8 id; /* sequence number for slot */ - __u8 name0_4[10]; /* first 5 characters in name */ - __u8 attr; /* attribute byte */ - __u8 reserved; /* always 0 */ - __u8 alias_checksum; /* checksum for 8.3 alias */ - __u8 name5_10[12]; /* 6 more characters in name */ - __u16 start; /* starting cluster number, 0 in long slots */ - __u8 name11_12[4]; /* last 2 characters in name */ -}; - -struct msdos_dir_entry - { - char name[8], ext[3]; /* name and extension */ - __u8 attr; /* attribute bits */ - __u8 lcase; /* Case for base and extension */ - __u8 ctime_ms; /* Creation time, milliseconds */ - __u16 ctime; /* Creation time */ - __u16 cdate; /* Creation date */ - __u16 adate; /* Last access date */ - __u16 starthi; /* high 16 bits of first cl. (FAT32) */ - __u16 time, date, start; /* time, date and first cluster */ - __u32 size; /* file size (in bytes) */ - } __attribute__ ((packed)); - /* The "boot code" we put into the filesystem... it writes a message and tells the user to try again */ @@ -356,7 +339,6 @@ static struct msdos_dir_entry *root_dir; static int size_root_dir; /* Size of the root directory in bytes */ static int sectors_per_cluster = 0; /* Number of sectors per disk cluster */ static int root_dir_entries = 0; /* Number of root directory entries */ -static int root_dir_num_entries = 0; static int last_cluster_written = 0; static char *blank_sector; /* Blank sector - all zeros */ @@ -1315,7 +1297,7 @@ setup_tables (void) de->date = CT_LE_W((unsigned short)(ctime->tm_mday + ((ctime->tm_mon+1) << 5) + ((ctime->tm_year-80) << 9))); - de->ctime_ms = 0; + de->ctime_cs = 0; de->ctime = de->time; de->cdate = de->date; de->adate = de->date; @@ -1451,16 +1433,23 @@ write_tables (void) /* Add a file to the specified directory entry, and also write it into the image */ -static void copy_filename(char *filename, char *base, char *ext) { +static void copy_filename(char *filename, char *dos_name) { char *ch = filename; int i, len; - memset(base, 0x20, 8); - memset(ext, 0x20, 3); + if (!strcmp(filename, ".")) { + strncpy(dos_name, MSDOS_DOT, MSDOS_NAME); + return; + } + if (!strcmp(filename, "..")) { + strncpy(dos_name, MSDOS_DOTDOT, MSDOS_NAME); + return; + } + memset(dos_name, 0x20, MSDOS_NAME); for(len = 0 ; *ch && *ch != '.'; ch++) { - base[len++] = toupper(*ch); + dos_name[len++] = toupper(*ch); if (len == 8) break; } @@ -1468,7 +1457,7 @@ static void copy_filename(char *filename if (*ch) ch++; for(len = 0 ; *ch; ch++) { - ext[len++] = toupper(*ch); + dos_name[8 + len++] = toupper(*ch); if (len == 3) break; } } @@ -1551,7 +1540,7 @@ static int add_file(char *filename, stru int start; int usedsec, totalsec; - char name83[8], ext83[3]; + char dos_name[MSDOS_NAME+1]; struct msdos_dir_slot *slot; int i; @@ -1562,23 +1551,22 @@ static int add_file(char *filename, stru if (dir->root) { if (dir->count == dir->entries) { printf("Error - too many directory entries\n"); + return; } } else { - if (dir->count == dir->entries) { - if (!dir->table) - dir->table = - (struct msdos_dir_entry *) malloc(sizeof(struct msdos_dir_entry)); - else { - dir->table = - (struct msdos_dir_entry *) realloc(dir->table, (dir->entries + 1) * - sizeof(struct msdos_dir_entry)); - - memset(&dir->table[dir->entries], 0, sizeof(struct msdos_dir_entry)); - } - - dir->entries++; - } + /* 2 entries, one extra for long filename */ + if (!dir->table) + dir->table = + (struct msdos_dir_entry *) malloc(2 * sizeof(struct msdos_dir_entry)); + else + dir->table = + (struct msdos_dir_entry *) realloc(dir->table, 2 * (dir->entries + 1) * + sizeof(struct msdos_dir_entry)); + if (!dir->table) + printf("Error - realloc failed\n"); + memset(&dir->table[dir->entries], 0, 2 * sizeof(struct msdos_dir_entry)); + dir->entries += 2; } infile = open(filename, O_RDONLY, 0); @@ -1611,13 +1599,13 @@ static int add_file(char *filename, stru return -1; } - printf("ADD %s\n", filename); + printf("ADD FILE %s\n", filename); /* Grab the basename of the file */ base = basename(filename); - /* Extract out the 8.3 name */ - copy_filename(base, name83, ext83); + /* convert for dos fat structure */ + copy_filename(base, dos_name); /* Make an extended name slot */ @@ -1629,12 +1617,9 @@ static int add_file(char *filename, stru slot->alias_checksum = 0; - for(i = 0; i < 8; i++) - slot->alias_checksum = (((slot->alias_checksum&1)<<7)|((slot->alias_checksum&0xfe)>>1)) + name83[i]; + for(i = 0; i < MSDOS_NAME; i++) + slot->alias_checksum = (((slot->alias_checksum&1)<<7)|((slot->alias_checksum&0xfe)>>1)) + dos_name[i]; - for(i = 0; i < 3; i++) - slot->alias_checksum = (((slot->alias_checksum&1)<<7)|((slot->alias_checksum&0xfe)>>1)) + ext83[i]; - p = base; copy_name(slot->name0_4, 10, &p); @@ -1645,8 +1630,7 @@ static int add_file(char *filename, stru /* Get the entry from the root filesytem */ entry = &dir->table[dir->count++]; - strncpy(entry->name, name83, 8); - strncpy(entry->ext, ext83, 3); + strncpy(entry->name, dos_name, MSDOS_NAME); /* If the user has it read only, then add read only to the incoming @@ -1665,7 +1649,7 @@ static int add_file(char *filename, stru ((ctime->tm_mon+1) << 5) + ((ctime->tm_year-80) << 9))); - entry->ctime_ms = 0; + entry->ctime_cs = 0; entry->ctime = entry->time; entry->cdate = entry->date; entry->adate = entry->date; @@ -1711,6 +1695,7 @@ static int add_file(char *filename, stru exit_add: if (infile) close(infile); + return 0; } /* Add a new directory to the specified directory entry, and in turn populate @@ -1727,10 +1712,18 @@ static void add_directory(char *filename struct dirent *dentry = 0; int remain; char *data; + char *base; + char dos_name[MSDOS_NAME+1]; + struct msdos_dir_slot *slot; + int i; + char *p; /* If the directory doesn't exist */ - if (!rddir) return; - + if (!rddir) { + printf("Error - dir does not exist: %s\n", filename); + return; + } + if (dir->root) { if (dir->count == dir->entries) { printf("Error - too many directory entries\n"); @@ -1738,28 +1731,58 @@ static void add_directory(char *filename } } else { - if (dir->count == dir->entries) { - if (!dir->table) - dir->table = (struct msdos_dir_entry *) malloc(sizeof(struct msdos_dir_entry)); - else { - dir->table = (struct msdos_dir_entry *) realloc(dir->table, (dir->entries + 1) * - sizeof(struct msdos_dir_entry)); - - /* Zero it out to avoid issues */ - memset(&dir->table[dir->entries], 0, sizeof(struct msdos_dir_entry)); - } - dir->entries++; + /* 2 entries, one extra for long name of the directory */ + if (!dir->table) + dir->table = (struct msdos_dir_entry *) malloc(2 * sizeof(struct msdos_dir_entry)); + else + dir->table = (struct msdos_dir_entry *) realloc(dir->table, 2 * (dir->entries + 1) * + sizeof(struct msdos_dir_entry)); + if (!dir->table) { + printf("Error - memory allocation failed\n"); + goto exit_add_dir; } + /* Zero it out to avoid issues */ + memset(&dir->table[dir->entries], 0, 2 * sizeof(struct msdos_dir_entry)); + dir->entries += 2; } + printf("ADD DIR %s\n", filename); /* Now, create a new directory entry for the new directory */ newdir = (struct dir_entry *) calloc(1, sizeof(struct dir_entry)); - if (!newdir) goto exit_add_dir; + if (!newdir) { + printf("Error - calloc failed\n"); + goto exit_add_dir; + } + + /* Grab the basename of the file */ + base = basename(filename); + + /* convert for dos structure */ + copy_filename(base, dos_name); + + /* Make an extended name slot */ + slot = (struct msdos_dir_slot *) &dir->table[dir->count++]; + slot->id = 'A'; + slot->attr = 0x0F; + slot->reserved = 0; + slot->start = 0; + + slot->alias_checksum = 0; + for (i = 0; i < MSDOS_NAME; i++) + slot->alias_checksum = (((slot->alias_checksum&1)<<7)|((slot->alias_checksum&0xfe)>>1)) + dos_name[i]; + + p = base; + + copy_name(slot->name0_4, 10, &p); + copy_name(slot->name5_10, 12, &p); + copy_name(slot->name11_12, 4, &p); + + /* Get the entry from the root filesytem */ entry = &dir->table[dir->count++]; - strncpy(entry->name, basename(filename), sizeof(entry->name)); - + strncpy(entry->name, dos_name, MSDOS_NAME); + entry->attr = ATTR_DIR; ctime = localtime(&create_time); @@ -1770,25 +1793,32 @@ static void add_directory(char *filename ((ctime->tm_mon+1) << 5) + ((ctime->tm_year-80) << 9))); - entry->ctime_ms = 0; + entry->ctime_cs = 0; entry->ctime = entry->time; entry->cdate = entry->date; entry->adate = entry->date; /* Now, read the directory */ - while((dentry = readdir(rddir))) { + + while((base[0] != '.') && (dentry = readdir(rddir))) { struct stat st; char *buffer; - - if (!strcmp(dentry->d_name, ".") || !strcmp(dentry->d_name, "..")) - continue; - /* DOS wouldn't like a typical unix . (dot) file, so we skip those too */ - if (dentry->d_name[0] == '.') continue; + if (dentry->d_name[0] == '.') { + /* dos also has . & .. directory entries */ + if (! ((!strcmp(dentry->d_name, ".")) || (!strcmp(dentry->d_name, "..")))) { + /* ignore other .* files */ + printf("Error - File/Dir name is not dos compatible, ignored: %s\n", dentry->d_name); + continue; + } + } buffer = malloc(strlen(filename) + strlen(dentry->d_name) + 3); - if (!buffer) continue; + if (!buffer) { + printf("Error - malloc failed\n"); + goto exit_add_dir; + } sprintf(buffer, "%s/%s", filename, dentry->d_name); if (!stat(buffer, &st)) { @@ -1806,11 +1836,23 @@ static void add_directory(char *filename /* Now that the entire directory has been written, go ahead and write the directory entry as well */ + entry->size = 0; /* a directory has zero size */ + + if (base[0] == '.') { /* . & .. point to parent's cluster */ + goto exit_add_dir; + } + entry->start = CT_LE_W(last_cluster_written); entry->starthi = CT_LE_W((last_cluster_written & 0xFFFF0000) >> 16); - entry->size = newdir->count * sizeof(struct msdos_dir_entry); + +/* . dir start points to parent */ + newdir->table[1].start = entry->start; +/* .. dir points to parent of parent*/ +/* .. dir start is not set yet, would need more changes to the code, + * but dosfsck can fix these .. entry start pointers correctly */ + + remain = newdir->count * sizeof(struct msdos_dir_entry); - remain = entry->size; data = (char *) newdir->table; while(remain) { @@ -1858,6 +1900,7 @@ static void add_root_directory(char *dir if (!newdir) { closedir(dir); + printf("Error - calloc failed!\n"); return; } @@ -1877,7 +1920,10 @@ static void add_root_directory(char *dir if (entry->d_name[0] == '.') continue; buffer = malloc(strlen(dirname) + strlen(entry->d_name) + 3); - if (!buffer) continue; + if (!buffer) { + printf("Error - malloc failed!\n"); + continue; + } sprintf(buffer, "%s/%s", dirname, entry->d_name); if (!stat(buffer, &st)) { @@ -2245,6 +2291,9 @@ main (int argc, char **argv) if (check && listfile) /* Auto and specified bad block handling are mutually */ die ("-c and -l are incompatible"); /* exclusive of each other! */ + if (dirname && (size_fat == 32)) + die ("-d is incompatible with FAT32"); + if (!create) { check_mount (device_name); /* Is the device already mounted? */ dev = open (device_name, O_RDWR); /* Is it a suitable device to build the FS on? */