c-fat16/fat16.c

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdbool.h>

#include "fat16.h"



// ============== INTERNAL PROTOTYPES ==================

/** Read boot sector from given address */
void read_bs(const BLOCKDEV* dev, Fat16BootSector* info, const uint32_t addr);

/** Find absolute address of first BootSector. Returns 0 on failure. */
uint32_t find_bs(const BLOCKDEV* dev);

/** Get cluster's starting address */
uint32_t clu_start(const FAT16* fat, const uint16_t cluster);

/** Find following cluster using FAT for jumps */
uint16_t next_clu(const FAT16* fat, uint16_t cluster);

/** Find relative address in a file, using FAT for cluster lookup */
uint32_t clu_add(const FAT16* fat, uint16_t cluster, uint32_t addr);

/** Read a file entry from directory (dir starting cluster, entry number) */
void fat16_fopen(const FAT16* fat, FAT16_FILE* file, const uint16_t dir_cluster, const uint16_t num);

/** Allocate and chain new cluster to a chain starting at given cluster */
bool append_cluster(const FAT16* fat, const uint16_t clu);

/** Allocate a new cluster, clean it, and mark with 0xFFFF in FAT */
uint16_t alloc_cluster(const FAT16* fat);

/** Zero out entire cluster. */
void wipe_cluster(const FAT16* fat, const uint16_t clu);

/**
 * Check if there is already a file of given RAW name
 * Raw name - name as found on disk, not "display name".
 */
bool dir_contains_file_raw(FAT16_FILE* dir, char* fname);


// =========== INTERNAL FUNCTION IMPLEMENTATIONS =========

/** Find absolute address of first boot sector. Returns 0 on failure. */
uint32_t find_bs(const BLOCKDEV* dev)
{
	//	Reference structure:
	//
	//	typedef struct __attribute__((packed)) {
	//		uint8_t first_byte;
	//		uint8_t start_chs[3];
	//		uint8_t partition_type;
	//		uint8_t end_chs[3];
	//		uint32_t start_sector;
	//		uint32_t length_sectors;
	//	} PartitionTable;

	uint16_t addr = 0x1BE + 4; // fourth byte of structure is the type.
	uint32_t tmp = 0;
	uint16_t tmp2;

	for (uint8_t i = 0; i < 4; i++, addr += 16)
	{
		// Read partition type
		dev->seek(addr);
		tmp = dev->read();

		// Check if type is valid
		if (tmp == 4 || tmp == 6 || tmp == 14)
		{
			// read MBR address
			dev->rseek(3);// skip 3 bytes
			dev->load(&tmp, 4);

			tmp = tmp << 9; // multiply address by 512 (sector size)

			// Verify that the boot sector has a valid signature mark
			dev->seek(tmp + 510);
			dev->load(&tmp2, 2);
			if (tmp2 != 0xAA55) continue; // continue to next entry

			// return absolute MBR address
			return tmp;
		}
	}

	return 0;
}


/** Read the boot sector */
void read_bs(const BLOCKDEV* dev, Fat16BootSector* info, const uint32_t addr)
{
	dev->seek(addr + 13); // skip 13

	dev->load(&(info->sectors_per_cluster), 6); // spc, rs, nf, re

	info->total_sectors = 0;
	dev->load(&(info->total_sectors), 2); // short sectors

	dev->rseek(1); // md

	dev->load(&(info->fat_size_sectors), 2);

	dev->rseek(8); // spt, noh, hs

	// read or skip long sectors field
	if (info->total_sectors == 0)
	{
		dev->load(&(info->total_sectors), 4);
	} else
	{
		dev->rseek(4); // tsl
	}

	dev->rseek(7); // dn, ch, bs, vi

	dev->load(&(info->volume_label), 11);
}


/** Get cluster starting address */
uint32_t clu_start(const FAT16* fat, const uint16_t cluster)
{
	if (cluster < 2) return fat->rd_addr;
	return fat->data_addr + (cluster - 2) * fat->bs.bytes_per_cluster;
}


uint16_t next_clu(const FAT16* fat, uint16_t cluster)
{
	fat->dev->seek(fat->fat_addr + (cluster * 2));
	return fat->dev->read16();
}


/** Find file-relative address in fat table */
uint32_t clu_add(const FAT16* fat, uint16_t cluster, uint32_t addr)
{
	while (addr >= fat->bs.bytes_per_cluster)
	{
		cluster = next_clu(fat, cluster);
		if (cluster == 0xFFFF) return 0xFFFF; // fail
		addr -= fat->bs.bytes_per_cluster;
	}

	return clu_start(fat, cluster) + addr;
}


/**
 * Zero out entire cluster
 * This is important only for directory clusters, so we can
 * zero only every first byte of each file entry, to indicate
 * that it is unused (FT_NONE).
 */
void wipe_cluster(const FAT16* fat, const uint16_t clu)
{
	uint32_t addr = clu_start(fat, clu);

	const BLOCKDEV* dev = fat->dev;

	dev->seek(addr);

	for (uint32_t b = 0; b < fat->bs.bytes_per_cluster; b += 32)
	{
		dev->write(0);
		dev->rseek(32);
	}
}


/** Allocate a new cluster, clean it, and mark with 0xFFFF in FAT */
uint16_t alloc_cluster(const FAT16* fat)
{
	// find new unclaimed cluster that can be added to the chain.
	uint16_t i, b;
	for (i = 2; i < fat->bs.fat_size_sectors * 256; i++)
	{
		// read value from FAT
		fat->dev->seek(fat->fat_addr + (i * 2));
		b = fat->dev->read16();
		if (b == 0)
		{
			// Write FFFF to "i", to mark end of file
			b = 0xFFFF;
			fat->dev->seek(fat->fat_addr + (i * 2));
			fat->dev->write16(b);

			// Wipe the cluster
			wipe_cluster(fat, i);

			return i;
		}
	}

	return 0xFFFF;//error code
}


/** Allocate and chain new cluster to a chain starting at given cluster */
bool append_cluster(const FAT16* fat, const uint16_t clu)
{
	uint16_t clu2 = alloc_cluster(fat);
	if (clu2 == 0xFFFF) return false;

	// Write "i" to "clu"
	fat->dev->seek(fat->fat_addr + (clu * 2));
	fat->dev->write16(clu2);

	return true;
}


/**
 * Check if there is already a file of given RAW name
 * Raw name - name as found on disk, not "display name".
 */
bool dir_contains_file_raw(FAT16_FILE* dir, char* fname)
{
	do {
		bool diff = false;
		for (uint8_t i = 0; i < 11; i++)
		{
			if (dir->name[i] != fname[i])
			{
				diff = true;
				break;
			}
		}

		if (!diff) return true;

	} while (fat16_next(dir));

	return false;
}


// =============== PUBLIC FUNCTION IMPLEMENTATIONS =================

/** Initialize a FAT16 handle */
void fat16_init(const BLOCKDEV* dev, FAT16* fat)
{
	const uint32_t bs_a = find_bs(dev);
	fat->dev = dev;
	read_bs(dev, &(fat->bs), bs_a);
	fat->fat_addr = bs_a + (fat->bs.reserved_sectors * 512);
	fat->rd_addr = bs_a + (fat->bs.reserved_sectors + fat->bs.fat_size_sectors * fat->bs.num_fats) * 512;
	fat->data_addr = fat->rd_addr + (fat->bs.root_entries * 32); // entry is 32B long

	fat->bs.bytes_per_cluster = (fat->bs.sectors_per_cluster * 512);
}


/** Move file cursor to a position relative to file start */
bool fat16_fseek(FAT16_FILE* file, uint32_t addr)
{
	// Clamp.
	if (addr > file->size)
		return false;

	// Store as rel
	file->cur_rel = addr;

	// Rewind and resolve abs, clu, ofs
	file->cur_clu = file->clu_start;

	while (addr >= file->fat->bs.bytes_per_cluster)
	{
		file->cur_clu = next_clu(file->fat, file->cur_clu);
		addr -= file->fat->bs.bytes_per_cluster;
	}

	file->cur_abs = clu_start(file->fat, file->cur_clu) + addr;
	file->cur_ofs = addr;

	return true;
}


/**
 * Read a file entry
 *
 * dir_cluster ... directory start cluster
 * num ... entry number in the directory
 */
void fat16_fopen(const FAT16* fat, FAT16_FILE* file, const uint16_t dir_cluster, const uint16_t num)
{
	// Resolve starting address
	uint32_t addr;
	if (dir_cluster == 0)
	{
		addr = clu_start(fat, dir_cluster) + num * 32; // root directory, max 512 entries.
	} else
	{
		addr = clu_add(fat, dir_cluster, num * 32); // cluster + N (wrapping to next cluster if needed)
	}

	fat->dev->seek(addr);
	fat->dev->load(file, 12); // name, ext, attribs
	fat->dev->rseek(14); // skip 14 bytes
	fat->dev->load(((void*)file) + 12, 6); // read remaining bytes

	file->clu = dir_cluster;
	file->num = num;

	// Resolve filename & type

	file->type = FT_FILE;

	switch(file->name[0])
	{
		case 0x00:
			file->type = FT_NONE;
			return;

		case 0xE5:
			file->type = FT_DELETED;
			return;

		case 0x05: // Starting with 0xE5
			file->type = FT_FILE;
			file->name[0] = 0xE5; // convert to the real character
			break;

		case 0x2E:
			if (file->name[1] == 0x2E)
			{
				// ".." directory
				file->type = FT_PARENT;
			} else
			{
				// "." directory
				file->type = FT_SELF;
			}
			break;

		default:
			file->type = FT_FILE;
	}

	// handle subdir, label
	if (file->attribs & FA_DIR && file->type == FT_FILE)
	{
		file->type = FT_SUBDIR;
	} else
	if (file->attribs == FA_LABEL)
	{
		file->type = FT_LABEL; // volume label special file
	} else
	if (file->attribs == 0x0F)
	{
		file->type = FT_LFN; // long name special file, can be safely ignored
	}

	// add a FAT pointer
	file->fat = fat;

	// Init cursors
	fat16_fseek(file, 0);
}


/**
 * Check if file is a valid entry (to be shown)
 */
bool fat16_is_file_valid(const FAT16_FILE* file)
{
	switch (file->type) {
		case FT_FILE:
		case FT_SUBDIR:
		case FT_SELF:
		case FT_PARENT:
			return true;

		default:
			return false;
	}
}


#define MIN(a, b) (((a) < (b)) ? (a) : (b))

bool fat16_fread(FAT16_FILE* file, void* target, uint32_t len)
{
	if (file->cur_abs == 0xFFFF)
		return false; // file at the end already

	if (file->cur_rel + len > file->size)
		return false; // attempt to read outside file size

	const FAT16* fat = file->fat;
	const BLOCKDEV* dev = fat->dev;

	while (len > 0 && file->cur_rel < file->size)
	{
		uint16_t chunk = MIN(file->size - file->cur_rel, MIN(fat->bs.bytes_per_cluster - file->cur_ofs, len));

		dev->seek(file->cur_abs);
		dev->load(target, chunk);

		file->cur_abs += chunk;
		file->cur_rel += chunk;
		file->cur_ofs += chunk;

		target += chunk;

		if (file->cur_ofs >= fat->bs.bytes_per_cluster)
		{
			file->cur_clu = next_clu(fat, file->cur_clu);
			file->cur_abs = clu_start(fat, file->cur_clu);
			file->cur_ofs = 0;
		}

		len -= chunk;
	}

	return true;
}


/** Open next file in the directory */
bool fat16_next(FAT16_FILE* file)
{
	const FAT16* fat = file->fat;
	const BLOCKDEV* dev = fat->dev;

	if (file->clu == 0 && file->num >= fat->bs.root_entries)
		return false; // attempt to read outside root directory.

	uint32_t addr = clu_add(fat, file->clu, (file->num + 1) * 32);
	if (addr == 0xFFFF)
		return false; // next file is out of the directory cluster

	// read first byte of the file entry; if zero, can't read (file is NONE)
	// FIXME this may be a problem when creating a new file...?
	uint8_t x;
	dev->seek(addr);
	x = dev->read();

	if (x == 0)
		return false;

	fat16_fopen(fat, file, file->clu, file->num+1);

	return true;
}


/** Open previous file in the directory */
bool fat16_prev(FAT16_FILE* file)
{
	if (file->num == 0)
		return false; // first file already

	fat16_fopen(file->fat, file, file->clu, file->num-1);

/*	// Skip bad files
	if (!fat16_is_file_valid(file))
		fat16_prev(file);*/

	return true;
}


/** Rewind to first file in directory */
void fat16_first(FAT16_FILE* file)
{
	fat16_fopen(file->fat, file, file->clu, 0);
}


/** Open a directory */
bool fat16_opendir(FAT16_FILE* file)
{
	// Don't open non-dirs and "." directory.
	if (!(file->attribs & FA_DIR) || file->type == FT_SELF)
		return false;

	fat16_fopen(file->fat, file, file->clu_start, 0);
	return true;
}


void fat16_open_root(const FAT16* fat, FAT16_FILE* file)
{
	fat16_fopen(fat, file, 0, 0);
}


/**
 * Find a file with given "display name" in this directory.
 * If file is found, "dir" will contain it's handle.
 * Either way, "dir" gets modified and you may need to rewind it afterwards.
 */
bool fat16_find_file(FAT16_FILE* dir, const char* name)
{
	char fname[11];
	fat16_undisplay_name(name, fname);
	return dir_contains_file_raw(dir, fname);
}


bool fat16_newfile(FAT16_FILE* dir, FAT16_FILE* file, const char* name)
{
	// Convert filename to zero padded raw string
	char fname[11];
	fat16_undisplay_name(name, fname);

	// Abort if file already exists
	bool exists = dir_contains_file_raw(dir, fname);
	fat16_first(dir); // rewind dir
	if (exists) return false;


	// Find free directory entry that can be used
	uint16_t clu = dir->clu;
	const FAT16* fat = dir->fat;

	uint16_t num = 0;
	for (; num < 0xFFFF; num++)
	{
		// root directory has fewer entries, error if trying
		// to add one more.
		if (dir->clu == 0 && num >= fat->bs.root_entries)
			return false;


		// Resolve addres of next file entry
		uint32_t addr;
		do {
			addr = clu_add(fat, dir->clu, num * 32);

			if (addr == 0xFFFF)
			{
				// end of chain of allocated clusters for the directory
				// append new cluster, return false on failure
				if (!append_cluster(fat, dir->clu)) return false;
			}

			// if new cluster was just added, repeat.
		} while (addr == 0xFFFF);


		// Open the file entry
		fat16_fopen(fat, file, clu, num);


		// Check if can be overwritten
		if (file->type == FT_DELETED || file->type == FT_NONE)
		{
			const uint16_t newclu = alloc_cluster(fat);
			const uint32_t entrystart = clu_add(fat, clu, num * 32);

			// store the file name
			fat->dev->seek(entrystart);

			// filename, without dor, zero-padded.
			fat->dev->store(fname, 11);

			fat->dev->write(0); // attributes
			fat->dev->rseek(14);
			fat->dev->write16(newclu); // starting cluster

			// file size (uint32_t)
			fat->dev->write16(0);
			fat->dev->write16(0);

			// reopen file, load the information just written
			fat16_fopen(fat, file, clu, num);
			return true;
		}
	}
	return false;
}


char* fat16_volume_label(const FAT16* fat, char* str)
{
	FAT16_FILE first;
	fat16_open_root(fat, &first);

	if (first.type == FT_LABEL) {
		return fat16_display_name(&first, str);
	}

	// find where spaces end
	uint8_t j = 10;
	for (; j >= 0; j--)
	{
		if (fat->bs.volume_label[j] != ' ') break;
	}

	// copy all until spaces
	uint8_t i;
	for (i = 0; i <= j; i++)
	{
		str[i] = fat->bs.volume_label[i];
	}

	str[i] = 0; // ender

	return str;
}


char* fat16_display_name(const FAT16_FILE* file, char* str)
{
	// Cannot get name for special files
	if (file->type == FT_NONE ||        // not-yet-used directory location
		file->type == FT_DELETED ||     // deleted file entry
		file->attribs == 0x0F)          // long name special entry (system, hidden)
		return NULL;

	// find first non-space
	uint8_t j = 7;
	for (; j >= 0; j--)
	{
		if (file->name[j] != ' ') break;
	}

	// j ... last no-space char

	uint8_t i;
	for (i = 0; i <= j; i++)
	{
		str[i] = file->name[i];
	}


	// directory entry, no extension
	if (file->type == FT_SUBDIR || file->type == FT_SELF || file->type == FT_PARENT)
	{
		str[i] = 0; // end of string
		return str;
	}


	// add a dot
	if (file->type != FT_LABEL) // volume label has no dot!
		str[i++] = '.';

	// Add extension chars
	for (j = 8; j < 11; j++, i++)
	{
		const char c = file->name[j];
		if (c == ' ') break;
		str[i] = c;
	}

	str[i] = 0; // end of string

	return str;
}


char* fat16_undisplay_name(const char* name, char* fixed)
{
	uint8_t name_c = 0, wr_c = 0;
	bool filling = false;
	bool at_ext = false;
	for (; wr_c < 11; wr_c++)
	{
		// start filling with spaces if end of filename reached
		char c = name[name_c];
		// handle special rule for 0xE5
		if (name_c == 0 && c == 0xE5) {
			c = 0x05;
		}

		if (c == '.' || c == 0)
		{
			if (!filling)
			{
				filling = true;

				if (c == '.')
				{
					name_c++; // skip the dot
					c = name[name_c];
					at_ext = true;
				}
			}
		}

		// if at the start of ext
		if (wr_c == 8)
		{
			if (!at_ext)
			{
				// try to advance past dot (if any)
				while(true)
				{
					c = name[name_c++];
					if (c == 0) break;
					if (c == '.') {
						// read char PAST the dot
						c = name[name_c];
						at_ext = true;
						break;
					}
				}
			}

			// if c has valid char for extension
			if (c != 0 && c != '.')
			{
				// start copying again.
				filling = false;
			}
		}

		if (!filling)
		{
			// copy char of filename
			fixed[wr_c] = name[name_c++];
		}
		else
		{
			// add a filler space
			fixed[wr_c] = ' ';
		}
	}

	return fixed;
}