X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=src%2Fdrivers%2Fao_fat.c;h=87c4158bf4e3704a23b287613caa2a500b0b4b8a;hb=7e6e2ca60c65a4fe2bee0bd8b9b89d45a7dbcfb3;hp=a19eff701564c04dc30fecbb2b656a920351cf20;hpb=86e1039e14304ac13db540f2ee3afd4ff170b8b4;p=fw%2Faltos

diff --git a/src/drivers/ao_fat.c b/src/drivers/ao_fat.c
index a19eff70..87c4158b 100644
--- a/src/drivers/ao_fat.c
+++ b/src/drivers/ao_fat.c
@@ -22,6 +22,24 @@
 #include "ao_fat.h"
 #include "ao_bufio.h"
 
+/* Include FAT commands */
+#ifndef AO_FAT_TEST
+#define FAT_COMMANDS	1
+#endif
+
+/* Spew FAT tracing */
+#define FAT_TRACE	0
+ 
+#ifdef DBG
+#undef DBG
+#endif
+
+#if FAT_TRACE
+#define DBG(...) printf(__VA_ARGS__)
+#else
+#define DBG(...)
+#endif
+
 /*
  * Basic file system types
  */
@@ -268,101 +286,130 @@ ao_fat_cluster_seek(cluster_t cluster, cluster_t distance)
 static cluster_t
 ao_fat_cluster_set_size(cluster_t first_cluster, cluster_t size)
 {
-	cluster_t	clear_cluster = 0;
+	cluster_t	have;
+	cluster_t	last_cluster;
+	cluster_t	next_cluster;
+
+	/* Walk the cluster chain to the
+	 * spot where it needs to change. That
+	 * will either be the end of the chain (in case it needs to grow),
+	 * or after the desired number of clusters, in which case it needs to shrink
+	 */
+	next_cluster = first_cluster;
+	last_cluster = 0;
+	DBG("\tclusters:");
+	for (have = 0; have < size; have++) {
+		DBG(" %08x", next_cluster);
+		if (!ao_fat_cluster_valid(next_cluster))
+			break;
+		last_cluster = next_cluster;
+		next_cluster = ao_fat_entry_read(next_cluster);
+	}
+	DBG("\n");
+
+	/* At this point, last_cluster points to the last valid
+	 * cluster in the file, if any. That's the spot in the FAT
+	 * that needs to be rewritten, either to truncate the file by
+	 * writing an END marker, or to extend the file by writing
+	 * more clusters. next_cluster will contain the value of the
+	 * FAT at last_cluster.
+	 *
+	 * If this is at the head of the cluster chain, then
+	 * last_cluster will be zero and next_cluster will
+	 * be the first cluster in the chain.
+	 */
+	if (have == size) {
+		/* The file is large enough, truncate as needed */
+		if (ao_fat_cluster_valid(next_cluster)) {
+			DBG("truncate between %08x and %08x\n", last_cluster, next_cluster);
+			if (last_cluster)
+				/*
+				 * Otherwise, rewrite the last cluster
+				 * in the chain with a LAST marker
+				 */
+				(void) ao_fat_entry_replace(last_cluster,
+							    AO_FAT_LAST_CLUSTER);
+			else
+				/*
+				 * If the file is getting erased, then
+				 * rewrite the directory entry cluster
+				 * value
+				 */
+				first_cluster = 0;
 
-	if (size == 0) {
-		clear_cluster = first_cluster;
-		first_cluster = 0;
-	} else {
-		cluster_t	have;
-		cluster_t	last_cluster = 0;
-		cluster_t	next_cluster;
-
-		/* Walk the cluster chain to the
-		 * spot where it needs to change. That
-		 * will either be the end of the chain (in case it needs to grow),
-		 * or after the desired number of clusters, in which case it needs to shrink
-		 */
-		next_cluster = first_cluster;
-		for (have = 0; have < size; have++) {
-			last_cluster = next_cluster;
-			next_cluster = ao_fat_entry_read(last_cluster);
-			if (!ao_fat_cluster_valid(next_cluster))
-				break;
-		}
+			/* Clear the remaining clusters in the chain */
+			ao_fat_free_cluster_chain(next_cluster);
 
-		if (have == size) {
-			/* The file is large enough, truncate as needed */
-			if (ao_fat_cluster_valid(next_cluster)) {
-				/* Rewrite that cluster entry with 0xffff to mark the end of the chain */
-				clear_cluster = ao_fat_entry_replace(last_cluster, AO_FAT_LAST_CLUSTER);
-				filesystem_full = 0;
-			} else {
-				/* The chain is already the right length, don't mess with it */
-				;
-			}
+			/* The file system is no longer full (if it was) */
+			filesystem_full = 0;
 		} else {
-			cluster_t	need;
-			cluster_t	free;
+			DBG("unchanged FAT chain\n");
+			/* The chain is already the right length, don't mess with it */
+			;
+		}
+	} else {
+		cluster_t	need;
+		cluster_t	free;
 
-			if (filesystem_full)
-				return AO_FAT_BAD_CLUSTER;
+		if (filesystem_full)
+			return AO_FAT_BAD_CLUSTER;
 
-			if (next_free < 2 || number_cluster <= next_free) {
-				next_free = 2;
-				fsinfo_dirty = 1;
-			}
+		/* Set next_free if it has wrapped or wasn't set before */
+		if (next_free < 2 || number_cluster <= next_free) {
+			next_free = 2;
+			fsinfo_dirty = 1;
+		}
 
-			/* See if there are enough free clusters in the file system */
-			need = size - have;
+		/* See if there are enough free clusters in the file system */
+		need = size - have;
 
 #define loop_cluster	for (free = next_free; need > 0;)
-#define next_cluster					\
-			if (++free == number_cluster)	\
-				free = 2;		\
-			if (free == next_free) \
-				break;			\
-
-			loop_cluster {
-				if (!ao_fat_entry_read(free))
-					need--;
-				next_cluster;
-			}
-			/* Still need some, tell the user that we've failed */
-			if (need) {
-				filesystem_full = 1;
-				return AO_FAT_BAD_CLUSTER;
-			}
+#define next_cluster				\
+		if (++free == number_cluster)	\
+			free = 2;		\
+		if (free == next_free)		\
+			break;			\
+
+		loop_cluster {
+			if (!ao_fat_entry_read(free))
+				need--;
+			next_cluster;
+		}
 
-			/* Now go allocate those clusters and
-			 * thread them onto the chain
-			 */
-			need = size - have;
-			loop_cluster {
-				if (!ao_fat_entry_read(free)) {
-					next_free = free + 1;
-					if (next_free >= number_cluster)
-						next_free = 2;
-					fsinfo_dirty = 1;
-					if (last_cluster)
-						ao_fat_entry_replace(last_cluster, free);
-					else
-						first_cluster = free;
-					last_cluster = free;
-					need--;
-				}
-				next_cluster;
+		/* Still need some, tell the user that we've failed */
+		if (need) {
+			filesystem_full = 1;
+			return AO_FAT_BAD_CLUSTER;
+		}
+
+		/* Now go allocate those clusters and
+		 * thread them onto the chain
+		 */
+		need = size - have;
+		loop_cluster {
+			if (ao_fat_entry_read(free) == 0) {
+				next_free = free + 1;
+				if (next_free >= number_cluster)
+					next_free = 2;
+				fsinfo_dirty = 1;
+				DBG("\tadd cluster. old %08x new %08x\n", last_cluster, free);
+				if (last_cluster)
+					ao_fat_entry_replace(last_cluster, free);
+				else
+					first_cluster = free;
+				last_cluster = free;
+				need--;
 			}
+			next_cluster;
+		}
 #undef loop_cluster
 #undef next_cluster
-			/* Mark the new end of the chain */
-			ao_fat_entry_replace(last_cluster, AO_FAT_LAST_CLUSTER);
-		}
+		DBG("\tlast cluster %08x\n", last_cluster);
+		/* Mark the new end of the chain */
+		ao_fat_entry_replace(last_cluster, AO_FAT_LAST_CLUSTER);
 	}
 
-	/* Deallocate clusters off the end of the file */
-	if (ao_fat_cluster_valid(clear_cluster))
-		ao_fat_free_cluster_chain(clear_cluster);
+	DBG("\tfirst cluster %08x\n", first_cluster);
 	return first_cluster;
 }
 
@@ -409,7 +456,7 @@ ao_fat_root_put(uint8_t *root, dirent_t e, uint8_t write)
 /*
  * ao_fat_root_extend
  *
- * On FAT32, make the 
+ * On FAT32, make the root directory at least 'ents' entries long
  */
 static int8_t
 ao_fat_root_extend(dirent_t ents)
@@ -419,8 +466,8 @@ ao_fat_root_extend(dirent_t ents)
 	if (!fat32)
 		return 0;
 	
-	byte_size = ents * 0x20;
-	cluster_size = byte_size / bytes_per_cluster;
+	byte_size = (ents + 1) * 0x20;
+	cluster_size = (byte_size + bytes_per_cluster - 1) / bytes_per_cluster;
 	if (ao_fat_cluster_set_size(root_cluster, cluster_size) != AO_FAT_BAD_CLUSTER)
 		return 1;
 	return 0;
@@ -440,14 +487,14 @@ ao_fat_setup_partition(void)
 
 	mbr = ao_bufio_get(0);
 	if (!mbr)
-		return 0;
+		return AO_FAT_FILESYSTEM_MBR_READ_FAILURE;
 
 	/* Check the signature */
 	if (mbr[0x1fe] != 0x55 || mbr[0x1ff] != 0xaa) {
-		printf ("Invalid MBR signature %02x %02x\n",
+		DBG ("Invalid MBR signature %02x %02x\n",
 			mbr[0x1fe], mbr[0x1ff]);
 		ao_bufio_put(mbr, 0);
-		return 0;
+		return AO_FAT_FILESYSTEM_INVALID_MBR_SIGNATURE;
 	}
 
 	/* Check to see if it's actually a boot block, in which
@@ -472,24 +519,22 @@ ao_fat_setup_partition(void)
 		case 0x0c:	/* FAT32 LBA */
 			break;
 		default:
-			printf ("Invalid partition type %02x\n", partition_type);
+			DBG ("Invalid partition type %02x\n", partition_type);
 			ao_bufio_put(mbr, 0);
-			return 0;
+			return AO_FAT_FILESYSTEM_INVALID_PARTITION_TYPE;
 		}
 
 		partition_start = get_u32(partition+8);
 		partition_size = get_u32(partition+12);
 		if (partition_size == 0) {
-			printf ("Zero-sized partition\n");
+			DBG ("Zero-sized partition\n");
 			ao_bufio_put(mbr, 0);
-			return 0;
+			return AO_FAT_FILESYSTEM_ZERO_SIZED_PARTITION;
 		}
 	}
 	partition_end = partition_start + partition_size;
-	printf ("Partition type %02x start %08x end %08x\n",
-		partition_type, partition_start, partition_end);
 	ao_bufio_put(mbr, 0);
-	return 1;
+	return AO_FAT_FILESYSTEM_SUCCESS;
 }
 	
 static uint8_t
@@ -499,22 +544,22 @@ ao_fat_setup_fs(void)
 	uint32_t	data_sectors;
 
 	if (!boot)
-		return 0;
+		return AO_FAT_FILESYSTEM_BOOT_READ_FAILURE;
 
 	/* Check the signature */
 	if (boot[0x1fe] != 0x55 || boot[0x1ff] != 0xaa) {
-		printf ("Invalid BOOT signature %02x %02x\n",
+		DBG ("Invalid BOOT signature %02x %02x\n",
 			boot[0x1fe], boot[0x1ff]);
 		ao_fat_sector_put(boot, 0);
-		return 0;
+		return AO_FAT_FILESYSTEM_INVALID_BOOT_SIGNATURE;
 	}
 
 	/* Check the sector size */
 	if (get_u16(boot + 0xb) != SECTOR_SIZE) {
-		printf ("Invalid sector size %d\n",
+		DBG ("Invalid sector size %d\n",
 			get_u16(boot + 0xb));
 		ao_fat_sector_put(boot, 0);
-		return 0;
+		return AO_FAT_FILESYSTEM_INVALID_SECTOR_SIZE;
 	}
 
 	sectors_per_cluster = boot[0xd];
@@ -552,47 +597,136 @@ ao_fat_setup_fs(void)
 
 	number_cluster = data_sectors / sectors_per_cluster;
 
-	printf ("fat32: %d\n", fat32);
-	printf ("sectors per cluster %d\n", sectors_per_cluster);
-	printf ("reserved sectors %d\n", reserved_sector_count);
-	printf ("number of FATs %d\n", number_fat);
-	printf ("root entries %d\n", root_entries);
-	printf ("sectors per fat %d\n", sectors_per_fat);
-
-	printf ("fat  start %d\n", fat_start);
-	printf ("root start %d\n", root_start);
-	printf ("data start %d\n", data_start);
-
-	return 1;
+	return AO_FAT_FILESYSTEM_SUCCESS;
 }
 
 /*
- * State for the current opened file
+ * State for an open file
  */
-static struct ao_fat_dirent	ao_file_dirent;
-static uint32_t 		ao_file_offset;
-static uint32_t			ao_file_cluster_offset;
-static cluster_t		ao_file_cluster;
-static uint8_t			ao_file_opened;
+
+struct ao_file {
+	struct ao_fat_dirent	*dirent;
+	offset_t		offset;
+	offset_t		cluster_offset;
+	cluster_t		cluster;
+	uint8_t			busy;
+};
+
+#define AO_FAT_NFILE	8
+
+static struct ao_fat_dirent	ao_file_dirent[AO_FAT_NFILE];
+
+static struct ao_fat_dirent *
+ao_fat_file_dirent_alloc(struct ao_fat_dirent *want)
+{
+	int8_t	d;
+	struct ao_fat_dirent *free = NULL, *dirent;
+
+	for (d = 0; d < AO_FAT_NFILE; d++) {
+
+		dirent = &ao_file_dirent[d];
+		/* See if there's another user of this file already */
+		if (want && dirent->name[0] != 0) {
+			if (dirent->entry == want->entry)
+				return dirent;
+		} else {
+			if (!free) {
+				free = dirent;
+				if (!want)
+					break;
+			}
+		}
+	}
+	if (free && want)
+		*free = *want;
+	return free;
+}
+
+static struct ao_file 		ao_file_table[AO_FAT_NFILE];
+
+static int8_t
+ao_fat_fd_alloc(struct ao_fat_dirent *dirent)
+{
+	int8_t	fd;
+
+	for (fd = 0; fd < AO_FAT_NFILE; fd++)
+		if (!ao_file_table[fd].busy) {
+			ao_file_table[fd].dirent = ao_fat_file_dirent_alloc(dirent);
+			ao_file_table[fd].busy = 1;
+			ao_file_table[fd].offset = 0;
+			ao_file_table[fd].cluster_offset = 0;
+			ao_file_table[fd].cluster = ao_file_table[fd].dirent->cluster;
+				
+			return fd;
+		}
+	return -AO_FAT_EMFILE;
+}
+
+static void
+ao_fat_fd_free(int8_t fd)
+{
+	struct ao_file *file = &ao_file_table[fd];
+	struct ao_fat_dirent *dirent = file->dirent;
+	memset(&ao_file_table[fd], '\0', sizeof (struct ao_file));
+
+	/* Check and see if another ao_file references the same dirent */
+	for (fd = 0; fd < AO_FAT_NFILE; fd++)
+		if (ao_file_table[fd].dirent == dirent)
+			return;
+	memset(dirent, '\0', sizeof (struct ao_fat_dirent));
+}
+
+static struct ao_file *
+ao_fat_fd_to_file(int8_t fd)
+{
+	struct ao_file *file;
+	if (fd < 0 || AO_FAT_NFILE <= fd)
+		return NULL;
+
+	file = &ao_file_table[fd];
+	if (!file->busy)
+		return NULL;
+	return file;
+}
+
+static uint8_t			ao_filesystem_setup;
+static uint8_t			ao_filesystem_status;
 
 static uint8_t
 ao_fat_setup(void)
 {
-	ao_bufio_setup();
+	if (!ao_filesystem_setup) {
+
+		ao_filesystem_setup = 1;
+		ao_bufio_setup();
 	
-	partition_type = partition_start = partition_end = 0;
-	sectors_per_cluster = bytes_per_cluster = reserved_sector_count = 0;
-	number_fat = root_entries = sectors_per_fat = 0;
-	number_cluster = fat_start = root_start = data_start = 0;
-	next_free = filesystem_full = 0;
-	fat32 = fsinfo_dirty = root_cluster = fsinfo_sector = free_count = 0;
-	memset(&ao_file_dirent, '\0', sizeof (ao_file_dirent));
-	ao_file_offset = ao_file_cluster_offset = ao_file_cluster = ao_file_opened = 0;
-	if (!ao_fat_setup_partition())
-		return 0;
-	if (!ao_fat_setup_fs())
-		return 0;
-	return 1;
+		/* Re-initialize all global state; this will help to allow the
+		 * file system to get swapped someday
+		 */
+		partition_type = partition_start = partition_end = 0;
+		sectors_per_cluster = bytes_per_cluster = reserved_sector_count = 0;
+		number_fat = root_entries = sectors_per_fat = 0;
+		number_cluster = fat_start = root_start = data_start = 0;
+		next_free = filesystem_full = 0;
+		fat32 = fsinfo_dirty = root_cluster = fsinfo_sector = free_count = 0;
+
+		/* Reset open file table */
+		memset(&ao_file_table, '\0', sizeof (ao_file_table));
+
+		ao_filesystem_status = ao_fat_setup_partition();
+		if (ao_filesystem_status != AO_FAT_FILESYSTEM_SUCCESS)
+			return ao_filesystem_status;
+		ao_filesystem_status = ao_fat_setup_fs();
+		if (ao_filesystem_status != AO_FAT_FILESYSTEM_SUCCESS)
+			return ao_filesystem_status;
+	}
+	return ao_filesystem_status;
+}
+
+void
+ao_fat_unmount(void)
+{
+	ao_filesystem_setup = 0;
 }
 
 /*
@@ -600,48 +734,82 @@ ao_fat_setup(void)
  */
 
 static uint32_t
-ao_fat_current_sector(void)
+ao_fat_current_sector(struct ao_file *file)
 {
 	cluster_t	cluster_offset;
 	uint32_t	sector_offset;
 	uint16_t	sector_index;
 	cluster_t	cluster;
 
-	if (ao_file_offset > ao_file_dirent.size)
+	DBG("current sector offset %d size %d\n",
+	    file->offset, file->dirent->size);
+
+	if (file->offset > file->dirent->size) {
+		printf ("file offset %d larger than size %d\n",
+			file->offset, file->dirent->size);
 		return 0xffffffff;
+	}
 
-	sector_offset = ao_file_offset >> SECTOR_SHIFT;
+	sector_offset = file->offset >> SECTOR_SHIFT;
 
-	if (!ao_file_cluster || ao_file_offset < ao_file_cluster_offset) {
-		ao_file_cluster = ao_file_dirent.cluster;
-		ao_file_cluster_offset = 0;
+	if (!file->cluster || file->offset < file->cluster_offset) {
+		file->cluster = file->dirent->cluster;
+		file->cluster_offset = 0;
+		DBG("\treset to start of file %08x\n", file->cluster);
 	}
 
-	if (ao_file_cluster_offset + bytes_per_cluster <= ao_file_offset) {
+	if (file->cluster_offset + bytes_per_cluster <= file->offset) {
 		cluster_t	cluster_distance;
 
 		cluster_offset = sector_offset / sectors_per_cluster;
 
-		cluster_distance = cluster_offset - ao_file_cluster_offset / bytes_per_cluster;
+		cluster_distance = cluster_offset - file->cluster_offset / bytes_per_cluster;
 
-		cluster = ao_fat_cluster_seek(ao_file_cluster, cluster_distance);
+		DBG("\tseek forward %d clusters\n", cluster_distance);
+		cluster = ao_fat_cluster_seek(file->cluster, cluster_distance);
 
-		if (!ao_fat_cluster_valid(cluster))
+		if (!ao_fat_cluster_valid(cluster)) {
+			printf ("invalid cluster %08x\n", cluster);
 			return 0xffffffff;
-		ao_file_cluster = cluster;
-		ao_file_cluster_offset = cluster_offset * bytes_per_cluster;
+		}
+		file->cluster = cluster;
+		file->cluster_offset = cluster_offset * bytes_per_cluster;
 	}
 
 	sector_index = sector_offset % sectors_per_cluster;
-	return data_start + (uint32_t) (ao_file_cluster-2) * sectors_per_cluster + sector_index;
+	DBG("current cluster %08x sector_index %d sector %d\n",
+	    file->cluster, sector_index,
+	    data_start + (uint32_t) (file->cluster-2) * sectors_per_cluster + sector_index);
+	return data_start + (uint32_t) (file->cluster-2) * sectors_per_cluster + sector_index;
 }
 
+/*
+ * ao_fat_invaldate_cluster_offset
+ *
+ * When the file size gets shrunk, invalidate
+ * any file structures referencing clusters beyond that point
+ */
+
 static void
-ao_fat_set_offset(uint32_t offset)
+ao_fat_invalidate_cluster_offset(struct ao_fat_dirent *dirent)
 {
-	ao_file_offset = offset;
+	int8_t		fd;
+	struct ao_file	*file;
+
+	for (fd = 0; fd < AO_FAT_NFILE; fd++) {
+		file = &ao_file_table[fd];
+		if (!file->busy)
+			continue;
+		if (file->dirent == dirent) {
+			if (file->cluster_offset >= dirent->size) {
+				file->cluster_offset = 0;
+				file->cluster = dirent->cluster;
+			}
+		}
+	}
 }
 
+ 
 /*
  * ao_fat_set_size
  *
@@ -649,29 +817,38 @@ ao_fat_set_offset(uint32_t offset)
  * the cluster chain as needed
  */
 static int8_t
-ao_fat_set_size(uint32_t size)
+ao_fat_set_size(struct ao_file *file, uint32_t size)
 {
 	uint8_t		*dent;
 	cluster_t	first_cluster;
 	cluster_t	have_clusters, need_clusters;
 
-	if (size == ao_file_dirent.size)
+	DBG ("Set size %d\n", size);
+	if (size == file->dirent->size) {
+		DBG("\tsize match\n");
 		return AO_FAT_SUCCESS;
+	}
 
-	first_cluster = ao_file_dirent.cluster;
-	have_clusters = (ao_file_dirent.size + bytes_per_cluster - 1) / bytes_per_cluster;
+	first_cluster = file->dirent->cluster;
+	have_clusters = (file->dirent->size + bytes_per_cluster - 1) / bytes_per_cluster;
 	need_clusters = (size + bytes_per_cluster - 1) / bytes_per_cluster;
 
+	DBG ("\tfirst cluster %08x have %d need %d\n", first_cluster, have_clusters, need_clusters);
 	if (have_clusters != need_clusters) {
-		if (ao_file_cluster && size >= ao_file_cluster_offset) {
-			cluster_t	offset_clusters = (ao_file_cluster_offset + bytes_per_cluster) / bytes_per_cluster;
+		if (file->cluster && size > file->cluster_offset) {
+			cluster_t	offset_clusters = (file->cluster_offset + bytes_per_cluster) / bytes_per_cluster;
 			cluster_t	extra_clusters = need_clusters - offset_clusters;
 			cluster_t	next_cluster;
 
-			next_cluster = ao_fat_cluster_set_size(ao_file_cluster, extra_clusters);
+			DBG ("\tset size relative offset_clusters %d extra_clusters %d\n",
+			     offset_clusters, extra_clusters);
+
+			/* Need one more to account for file->cluster, which we already have */
+			next_cluster = ao_fat_cluster_set_size(file->cluster, extra_clusters + 1);
 			if (next_cluster == AO_FAT_BAD_CLUSTER)
 				return -AO_FAT_ENOSPC;
 		} else {
+			DBG ("\tset size absolute need_clusters %d\n", need_clusters);
 			first_cluster = ao_fat_cluster_set_size(first_cluster, need_clusters);
 
 			if (first_cluster == AO_FAT_BAD_CLUSTER)
@@ -679,18 +856,24 @@ ao_fat_set_size(uint32_t size)
 		}
 	}
 
+	DBG ("\tupdate directory size\n");
 	/* Update the directory entry */
-	dent = ao_fat_root_get(ao_file_dirent.entry);
-	if (!dent)
+	dent = ao_fat_root_get(file->dirent->entry);
+	if (!dent) {
+		printf ("dent update failed\n");
 		return -AO_FAT_EIO;
+	}
 	put_u32(dent + 0x1c, size);
 	put_u16(dent + 0x1a, first_cluster);
 	if (fat32)
 		put_u16(dent + 0x14, first_cluster >> 16);
-	ao_fat_root_put(dent, ao_file_dirent.entry, 1);
+	ao_fat_root_put(dent, file->dirent->entry, 1);
 
-	ao_file_dirent.size = size;
-	ao_file_dirent.cluster = first_cluster;
+	file->dirent->size = size;
+	file->dirent->cluster = first_cluster;
+	if (have_clusters > need_clusters)
+		ao_fat_invalidate_cluster_offset(file->dirent);
+	DBG ("set size done\n");
 	return AO_FAT_SUCCESS;
 }
 
@@ -699,7 +882,7 @@ ao_fat_set_size(uint32_t size)
  *
  * Initialize a root directory entry
  */
-void
+static void
 ao_fat_root_init(uint8_t *dent, char name[11], uint8_t attr)
 {
 	memset(dent, '\0', 0x20);
@@ -733,7 +916,7 @@ ao_fat_root_init(uint8_t *dent, char name[11], uint8_t attr)
 
 
 static void
-ao_fat_dirent_init(uint8_t *dent, uint16_t entry, struct ao_fat_dirent *dirent)
+ao_fat_dirent_init(struct ao_fat_dirent *dirent, uint8_t *dent, uint16_t entry)
 {
 	memcpy(dirent->name, dent + 0x00, 11);
 	dirent->attr = dent[0x0b];
@@ -750,7 +933,7 @@ ao_fat_dirent_init(uint8_t *dent, uint16_t entry, struct ao_fat_dirent *dirent)
  * Write out any fsinfo changes to disk
  */
 
-void
+static void
 ao_fat_flush_fsinfo(void)
 {
 	uint8_t	*fsinfo;
@@ -776,6 +959,36 @@ ao_fat_flush_fsinfo(void)
  * Public API
  */
 
+/*
+ * ao_fat_sync
+ *
+ * Flush any pending I/O to storage
+ */
+
+void
+ao_fat_sync(void)
+{
+	if (ao_fat_setup() != AO_FAT_FILESYSTEM_SUCCESS)
+		return;
+	ao_fat_flush_fsinfo();
+	ao_bufio_flush();
+}
+
+/*
+ * ao_fat_full
+ *
+ * Returns TRUE if the filesystem cannot take
+ * more data
+ */
+
+int8_t
+ao_fat_full(void)
+{
+	if (ao_fat_setup() != AO_FAT_FILESYSTEM_SUCCESS)
+		return 1;
+	return filesystem_full;
+}
+
 /*
  * ao_fat_open
  *
@@ -786,11 +999,18 @@ ao_fat_open(char name[11], uint8_t mode)
 {
 	uint16_t		entry = 0;
 	struct ao_fat_dirent	dirent;
+	int8_t			status;
 
-	if (ao_file_opened)
-		return -AO_FAT_EMFILE;
-	
-	while (ao_fat_readdir(&entry, &dirent)) {
+	if (ao_fat_setup() != AO_FAT_FILESYSTEM_SUCCESS)
+		return -AO_FAT_EIO;
+
+	for (;;) {
+		status = ao_fat_readdir(&entry, &dirent);
+		if (status < 0) {
+			if (status == -AO_FAT_EDIREOF)
+				return -AO_FAT_ENOENT;
+			return status;
+		}
 		if (!memcmp(name, dirent.name, 11)) {
 			if (AO_FAT_IS_DIR(dirent.attr))
 				return -AO_FAT_EISDIR;
@@ -798,10 +1018,7 @@ ao_fat_open(char name[11], uint8_t mode)
 				return -AO_FAT_EPERM;
 			if (mode > AO_FAT_OPEN_READ && (dirent.attr & AO_FAT_FILE_READ_ONLY))
 				return -AO_FAT_EACCESS;
-			ao_file_dirent = dirent;
-			ao_fat_set_offset(0);
-			ao_file_opened = 1;
-			return AO_FAT_SUCCESS;
+			return ao_fat_fd_alloc(&dirent);
 		}
 	}
 	return -AO_FAT_ENOENT;
@@ -816,46 +1033,62 @@ ao_fat_open(char name[11], uint8_t mode)
 int8_t
 ao_fat_creat(char name[11])
 {
-	uint16_t	entry;
-	int8_t		status;
-	uint8_t		*dent;
+	uint16_t		entry;
+	int8_t			fd;
+	int8_t			status;
+	uint8_t			*dent;
+	struct ao_file		*file;
 
-	if (ao_file_opened)
-		return -AO_FAT_EMFILE;
-
-	status = ao_fat_open(name, AO_FAT_OPEN_WRITE);
+	if (ao_fat_setup() != AO_FAT_FILESYSTEM_SUCCESS)
+		return -AO_FAT_EIO;
 
-	switch (status) {
-	case -AO_FAT_SUCCESS:
-		status = ao_fat_set_size(0);
-		break;
-	case -AO_FAT_ENOENT:
-		entry = 0;
-		for (;;) {
-			dent = ao_fat_root_get(entry);
-			if (!dent) {
-				
-				if (ao_fat_root_extend(entry))
-					continue;
-				status = -AO_FAT_ENOSPC;
-				break;
-			}
+	fd = ao_fat_open(name, AO_FAT_OPEN_WRITE);
+	if (fd >= 0) {
+		file = &ao_file_table[fd];
+		status = ao_fat_set_size(file, 0);
+		if (status < 0) {
+			ao_fat_close(fd);
+			fd = status;
+		}
+	} else {
+		if (fd == -AO_FAT_ENOENT) {
+			entry = 0;
+			for (;;) {
+				dent = ao_fat_root_get(entry);
+				if (!dent) {
 				
-			if (dent[0] == AO_FAT_DENT_EMPTY || dent[0] == AO_FAT_DENT_END) {
-				ao_fat_root_init(dent, name, AO_FAT_FILE_REGULAR);
-				ao_fat_dirent_init(dent, entry,  &ao_file_dirent);
-				ao_fat_root_put(dent, entry, 1);
-				ao_file_opened = 1;
-				ao_fat_set_offset(0);
-				status = -AO_FAT_SUCCESS;
-				break;
-			} else {
-				ao_fat_root_put(dent, entry, 0);
+					if (ao_fat_root_extend(entry))
+						continue;
+					fd = -AO_FAT_ENOSPC;
+					break;
+				}
+				if (dent[0] == AO_FAT_DENT_EMPTY || dent[0] == AO_FAT_DENT_END) {
+					fd = ao_fat_fd_alloc(NULL);
+					if (fd < 0) {
+						ao_fat_root_put(dent, entry, 0);
+						break;
+					}
+
+					file = &ao_file_table[fd];
+					/* Initialize the dent */
+					ao_fat_root_init(dent, name, AO_FAT_FILE_REGULAR);
+
+					/* Now initialize the dirent from the dent */
+					ao_fat_dirent_init(file->dirent, dent, entry);
+
+					/* And write the dent to storage */
+					ao_fat_root_put(dent, entry, 1);
+
+					status = -AO_FAT_SUCCESS;
+					break;
+				} else {
+					ao_fat_root_put(dent, entry, 0);
+				}
+				entry++;
 			}
-			entry++;
 		}
 	}
-	return status;
+	return fd;
 }
 
 /*
@@ -864,57 +1097,74 @@ ao_fat_creat(char name[11])
  * Close the currently open file
  */
 int8_t
-ao_fat_close(void)
+ao_fat_close(int8_t fd)
 {
-	if (!ao_file_opened)
+	struct ao_file *file = ao_fat_fd_to_file(fd);
+	if (!file)
 		return -AO_FAT_EBADF;
 
-	memset(&ao_file_dirent, '\0', sizeof (struct ao_fat_dirent));
-	ao_file_offset = 0;
-	ao_file_cluster = 0;
-	ao_file_opened = 0;
-
-	ao_fat_flush_fsinfo();
-	ao_bufio_flush();
+	ao_fat_fd_free(fd);
+	ao_fat_sync();
 	return AO_FAT_SUCCESS;
 }
 
+/*
+ * ao_fat_map_current
+ *
+ * Map the sector pointed at by the current file offset
+ */
+
+static void *
+ao_fat_map_current(struct ao_file *file, int len, cluster_offset_t *offsetp, cluster_offset_t *this_time)
+{
+	cluster_offset_t 	offset;
+	sector_t		sector;
+	void			*buf;
+
+	offset = file->offset & SECTOR_MASK;
+	sector = ao_fat_current_sector(file);
+	if (sector == 0xffffffff) {
+		printf ("invalid sector at offset %d\n", file->offset);
+		return NULL;
+	}
+	buf = ao_fat_sector_get(sector);
+	if (!buf)
+		printf ("sector get failed. Sector %d. Partition end %d\n", sector, partition_end);
+	if (offset + len < SECTOR_SIZE)
+		*this_time = len;
+	else
+		*this_time = SECTOR_SIZE - offset;
+	*offsetp = offset;
+	return buf;
+}
+
 /*
  * ao_fat_read
  *
  * Read from the file
  */
 int
-ao_fat_read(void *dst, int len)
+ao_fat_read(int8_t fd, void *dst, int len)
 {
-	uint8_t		*dst_b = dst;
-	uint32_t	sector;
-	uint16_t	this_time;
-	uint16_t	offset;
-	uint8_t		*buf;
-	int		ret = 0;
-
-	if (!ao_file_opened)
+	uint8_t			*dst_b = dst;
+	cluster_offset_t	this_time;
+	cluster_offset_t	offset;
+	uint8_t			*buf;
+	int			ret = 0;
+	struct ao_file		*file = ao_fat_fd_to_file(fd);
+	if (!file)
 		return -AO_FAT_EBADF;
 
-	if (ao_file_offset + len > ao_file_dirent.size)
-		len = ao_file_dirent.size - ao_file_offset;
+	if (file->offset + len > file->dirent->size)
+		len = file->dirent->size - file->offset;
 
 	if (len < 0)
 		len = 0;
 
 	while (len) {
-		offset = ao_file_offset & SECTOR_MASK;
-		if (offset + len < SECTOR_SIZE)
-			this_time = len;
-		else
-			this_time = SECTOR_SIZE - offset;
-
-		sector = ao_fat_current_sector();
-		if (sector == 0xffffffff)
-			break;
-		buf = ao_fat_sector_get(sector);
+		buf = ao_fat_map_current(file, len, &offset, &this_time);
 		if (!buf) {
+			printf ("map_current failed\n");
 			ret = -AO_FAT_EIO;
 			break;
 		}
@@ -924,7 +1174,7 @@ ao_fat_read(void *dst, int len)
 		ret += this_time;
 		len -= this_time;
 		dst_b += this_time;
-		ao_fat_set_offset(ao_file_offset + this_time);
+		file->offset = file->offset + this_time;
 	}
 	return ret;
 }
@@ -935,36 +1185,27 @@ ao_fat_read(void *dst, int len)
  * Write to the file, extended as necessary
  */
 int
-ao_fat_write(void *src, int len)
+ao_fat_write(int8_t fd, void *src, int len)
 {
-	uint8_t		*src_b = src;
-	uint32_t	sector;
-	uint16_t	this_time;
-	uint16_t	offset;
-	uint8_t		*buf;
-	int		ret = 0;
-
-	if (!ao_file_opened)
+	uint8_t			*src_b = src;
+	cluster_offset_t	this_time;
+	cluster_offset_t	offset;
+	uint8_t			*buf;
+	int			ret = 0;
+	struct ao_file		*file = ao_fat_fd_to_file(fd);
+	if (!file)
 		return -AO_FAT_EBADF;
 
-	if (ao_file_offset + len > ao_file_dirent.size) {
-		ret = ao_fat_set_size(ao_file_offset + len);
+	if (file->offset + len > file->dirent->size) {
+		ret = ao_fat_set_size(file, file->offset + len);
 		if (ret < 0)
 			return ret;
 	}
 
 	while (len) {
-		offset = ao_file_offset & SECTOR_MASK;
-		if (offset + len < SECTOR_SIZE)
-			this_time = len;
-		else
-			this_time = SECTOR_SIZE - offset;
-
-		sector = ao_fat_current_sector();
-		if (sector == 0xffffffff)
-			break;
-		buf = ao_fat_sector_get(sector);
+		buf = ao_fat_map_current(file, len, &offset, &this_time);
 		if (!buf) {
+			printf ("map_current failed\n");
 			ret = -AO_FAT_EIO;
 			break;
 		}
@@ -974,7 +1215,7 @@ ao_fat_write(void *src, int len)
 		ret += this_time;
 		len -= this_time;
 		src_b += this_time;
-		ao_fat_set_offset(ao_file_offset + this_time);
+		file->offset = file->offset + this_time;
 	}
 	return ret;
 }
@@ -989,13 +1230,14 @@ ao_fat_write(void *src, int len)
  * write
  */
 int32_t
-ao_fat_seek(int32_t pos, uint8_t whence)
+ao_fat_seek(int8_t fd, int32_t pos, uint8_t whence)
 {
-	uint32_t	new_offset = ao_file_offset;
-
-	if (!ao_file_opened)
+	offset_t	new_offset;
+	struct ao_file	*file = ao_fat_fd_to_file(fd);
+	if (!file)
 		return -AO_FAT_EBADF;
 
+	new_offset = file->offset;
 	switch (whence) {
 	case AO_FAT_SEEK_SET:
 		new_offset = pos;
@@ -1004,11 +1246,11 @@ ao_fat_seek(int32_t pos, uint8_t whence)
 		new_offset += pos;
 		break;
 	case AO_FAT_SEEK_END:
-		new_offset = ao_file_dirent.size + pos;
+		new_offset = file->dirent->size + pos;
 		break;
 	}
-	ao_fat_set_offset(new_offset);
-	return ao_file_offset;
+	file->offset = new_offset;
+	return file->offset;
 }
 
 /*
@@ -1023,6 +1265,9 @@ ao_fat_unlink(char name[11])
 	uint16_t		entry = 0;
 	struct ao_fat_dirent	dirent;
 
+	if (ao_fat_setup() != AO_FAT_FILESYSTEM_SUCCESS)
+		return -AO_FAT_EIO;
+
 	while (ao_fat_readdir(&entry, &dirent)) {
 		if (memcmp(name, dirent.name, 11) == 0) {
 			uint8_t	*next;
@@ -1066,54 +1311,243 @@ ao_fat_readdir(uint16_t *entry, struct ao_fat_dirent *dirent)
 {
 	uint8_t	*dent;
 
+	if (ao_fat_setup() != AO_FAT_FILESYSTEM_SUCCESS)
+		return -AO_FAT_EIO;
+
 	for (;;) {
 		dent = ao_fat_root_get(*entry);
 		if (!dent)
-			return 0;
+			return -AO_FAT_EDIREOF;
 
 		if (dent[0] == AO_FAT_DENT_END) {
 			ao_fat_root_put(dent, *entry, 0);
-			return 0;
+			return -AO_FAT_EDIREOF;
 		}
 		if (dent[0] != AO_FAT_DENT_EMPTY && (dent[0xb] & 0xf) != 0xf) {
-			ao_fat_dirent_init(dent, *entry, dirent);
+			ao_fat_dirent_init(dirent, dent, *entry);
 			ao_fat_root_put(dent, *entry, 0);
 			(*entry)++;
-			return 1;
+			return AO_FAT_SUCCESS;
 		}
 		ao_fat_root_put(dent, *entry, 0);
 		(*entry)++;
 	}
 }
 
+#if FAT_COMMANDS
+
+static const char *filesystem_errors[] = {
+	[AO_FAT_FILESYSTEM_SUCCESS] = "FAT file system operating normally",
+	[AO_FAT_FILESYSTEM_MBR_READ_FAILURE] = "MBR media read error",
+	[AO_FAT_FILESYSTEM_INVALID_MBR_SIGNATURE] = "MBR signature invalid",
+	[AO_FAT_FILESYSTEM_INVALID_PARTITION_TYPE] = "Unsupported paritition type",
+	[AO_FAT_FILESYSTEM_ZERO_SIZED_PARTITION] = "Partition has zero sectors",
+	[AO_FAT_FILESYSTEM_BOOT_READ_FAILURE] = "Boot block media read error",
+	[AO_FAT_FILESYSTEM_INVALID_BOOT_SIGNATURE] = "Boot block signature invalid",
+	[AO_FAT_FILESYSTEM_INVALID_SECTOR_SIZE] = "Sector size not 512",
+};
+	
+static void
+ao_fat_mbr_cmd(void)
+{
+	uint8_t	status;
+
+	status = ao_fat_setup();
+	if (status == AO_FAT_FILESYSTEM_SUCCESS) {
+		printf ("partition type: %02x\n", partition_type);
+		printf ("partition start: %08x\n", partition_start);
+
+		printf ("partition end:   %08x\n", partition_end);
+
+		printf ("fat32: %d\n", fat32);
+		printf ("sectors per cluster %d\n", sectors_per_cluster);
+		printf ("reserved sectors %d\n", reserved_sector_count);
+		printf ("number of FATs %d\n", number_fat);
+		printf ("root entries %d\n", root_entries);
+		printf ("sectors per fat %d\n", sectors_per_fat);
+
+		printf ("fat  start %d\n", fat_start);
+		printf ("root start %d\n", root_start);
+		printf ("data start %d\n", data_start);
+	} else {
+		printf ("FAT filesystem not available: %s\n", filesystem_errors[status]);
+	}
+}
+
+struct ao_fat_attr {
+	uint8_t	bit;
+	char	label;
+};
+
+static const struct ao_fat_attr ao_fat_attr[] = {
+	{ .bit = AO_FAT_FILE_READ_ONLY, .label = 'R' },
+	{ .bit = AO_FAT_FILE_HIDDEN, .label = 'H' },
+	{ .bit = AO_FAT_FILE_SYSTEM, .label = 'S' },
+	{ .bit = AO_FAT_FILE_VOLUME_LABEL, .label = 'V' },
+	{ .bit = AO_FAT_FILE_DIRECTORY, .label = 'D' },
+	{ .bit = AO_FAT_FILE_ARCHIVE, .label = 'A' },
+};
+
+#define NUM_FAT_ATTR	(sizeof (ao_fat_attr) / sizeof (ao_fat_attr[0]))
+
 static void
-ao_fat_list(void)
+ao_fat_list_cmd(void)
 {
 	uint16_t		entry = 0;
 	struct ao_fat_dirent	dirent;
+	int			i;
+	int8_t			status;
+
+	while ((status = ao_fat_readdir(&entry, &dirent)) == AO_FAT_SUCCESS) {
+		for (i = 0; i < 8; i++)
+			putchar(dirent.name[i]);
+		putchar('.');
+		for (; i < 11; i++)
+			putchar(dirent.name[i]);
+		for (i = 0; i < NUM_FAT_ATTR; i++)
+			putchar (dirent.attr & ao_fat_attr[i].bit ? ao_fat_attr[i].label : ' ');
+		printf (" @%08x %d\n", dirent.cluster, dirent.size);
+	}
+	if (status != -AO_FAT_EDIREOF)
+		printf ("readdir failed: %d\n", status);
+}
 
-	while (ao_fat_readdir(&entry, &dirent)) {
-		printf ("%-8.8s.%-3.3s %02x %04x %d\n",
-			dirent.name,
-			dirent.name + 8,
-			dirent.attr,
-			dirent.cluster,
-			dirent.size);
+static uint8_t
+ao_fat_parse_name(char name[11])
+{
+	uint8_t	c;
+
+	name[0] = '\0';
+	ao_cmd_white();
+	c = 0;
+	while (ao_cmd_lex_c != '\n') {
+		if (ao_cmd_lex_c == '.') {
+			for (; c < 8; c++)
+				name[c] = ' ';
+		} else {
+			if (c < 11)
+				name[c++] = ao_cmd_lex_c;
+		}
+		ao_cmd_lex();
+	}
+	while (c < 11)
+		name[c++] = ' ';
+}
+
+static void
+ao_fat_dump_cmd(void)
+{
+	char		name[11];
+	int8_t		fd;
+	int		cnt, i;
+	char		buf[32];
+
+	ao_fat_parse_name(name);
+	if (name[0] == '\0') {
+		ao_cmd_status = ao_cmd_syntax_error;
+		return;
 	}
+		
+	fd = ao_fat_open(name, AO_FAT_OPEN_READ);
+	if (fd < 0) {
+		printf ("Open failed: %d\n", fd);
+		return;
+	}
+	while ((cnt = ao_fat_read(fd, buf, sizeof(buf))) > 0) {
+		for (i = 0; i < cnt; i++)
+			putchar(buf[i]);
+	}
+	ao_fat_close(fd);
+}
+
+static void
+ao_fat_write_cmd(void)
+{
+	char		name[11];
+	int8_t		fd;
+	int		cnt, i;
+	char		buf[64];
+	char		c;
+	int		status;
+
+	ao_fat_parse_name(name);
+	if (name[0] == '\0') {
+		ao_cmd_status = ao_cmd_syntax_error;
+		return;
+	}
+		
+	fd = ao_fat_creat(name);
+	if (fd < 0) {
+		printf ("Open failed: %d\n", fd);
+		return;
+	}
+	flush();
+	while ((c = getchar()) != 4) {
+		if (c == '\r') c = '\n';
+		if (ao_echo()) {
+			if (c == '\n') putchar ('\r');
+			putchar(c); flush();
+		}
+		status = ao_fat_write(fd, &c, 1);
+		if (status != 1) {
+			printf ("Write failure %d\n", status);
+			break;
+		}
+	}
+	ao_fat_close(fd);
 }
 
 static void
-ao_fat_test(void)
+put32(uint32_t a)
 {
-	ao_fat_setup();
-	ao_fat_list();
+	ao_cmd_put16(a >> 16);
+	ao_cmd_put16(a);
+}
+
+static void
+ao_fat_hexdump_cmd(void)
+{
+	char		name[11];
+	int8_t		fd;
+	int		cnt, i;
+	char		buf[8];
+	uint32_t	addr;
+
+	ao_fat_parse_name(name);
+	if (name[0] == '\0') {
+		ao_cmd_status = ao_cmd_syntax_error;
+		return;
+	}
+		
+	fd = ao_fat_open(name, AO_FAT_OPEN_READ);
+	if (fd < 0) {
+		printf ("Open failed: %d\n", fd);
+		return;
+	}
+	addr = 0;
+	while ((cnt = ao_fat_read(fd, buf, sizeof(buf))) > 0) {
+		put32(addr);
+		for (i = 0; i < cnt; i++) {
+			putchar(' ');
+			ao_cmd_put8(buf[i]);
+		}
+		putchar('\n');
+		addr += cnt;
+	}
+	ao_fat_close(fd);
 }
 
 static const struct ao_cmds ao_fat_cmds[] = {
-	{ ao_fat_test,	"F\0Test FAT" },
+	{ ao_fat_mbr_cmd,	"M\0Show FAT MBR and other info" },
+	{ ao_fat_list_cmd,	"F\0List FAT directory" },
+	{ ao_fat_dump_cmd,	"D <name>\0Dump FAT file" },
+	{ ao_fat_write_cmd,	"W <name>\0Write FAT file (end with ^D)" },
+	{ ao_fat_hexdump_cmd,	"H <name>\0HEX dump FAT file" },
 	{ 0, NULL },
 };
 
+#endif
+
 void
 ao_fat_init(void)
 {