aiie/teensy/teensy-filemanager.cpp
2020-07-06 07:04:22 -04:00

281 lines
5.8 KiB
C++

#include <Arduino.h>
#include <wchar.h>
#include "teensy-filemanager.h"
#include <string.h> // strcpy
#include <TeensyThreads.h>
#include "teensy-println.h"
Threads::Mutex fslock;
TeensyFileManager::TeensyFileManager()
{
numCached = 0;
// FIXME: used to have 'enabled = sd.begin()' here, but we weren't
// using the enabled flag, so I've removed it to save the RAM for
// now; but eventually we need better error handling here
SD.begin(BUILTIN_SDCARD);
}
TeensyFileManager::~TeensyFileManager()
{
}
int8_t TeensyFileManager::openFile(const char *name)
{
Threads::Scope locker(fslock);
if (cacheFd != -1) {
cacheFile.close();
cacheFd = -1;
}
// See if there's a hole to re-use...
for (int i=0; i<numCached; i++) {
if (cachedNames[i][0] == '\0') {
strncpy(cachedNames[i], name, MAXPATH-1);
cachedNames[i][MAXPATH-1] = '\0'; // safety: ensure string terminator
fileSeekPositions[i] = 0;
return i;
}
}
// check for too many open files
if (numCached >= MAXFILES)
return -1;
// No, so we'll add it to the end
strncpy(cachedNames[numCached], name, MAXPATH-1);
cachedNames[numCached][MAXPATH-1] = '\0'; // safety: ensure string terminator
fileSeekPositions[numCached] = 0;
numCached++;
return numCached-1;
}
void TeensyFileManager::closeFile(int8_t fd)
{
Threads::Scope locker(fslock);
if (cacheFd != -1) {
cacheFile.close();
cacheFd = -1;
}
// invalid fd provided?
if (fd < 0 || fd >= numCached)
return;
// clear the name
cachedNames[fd][0] = '\0';
}
const char *TeensyFileManager::fileName(int8_t fd)
{
if (fd < 0 || fd >= numCached)
return NULL;
return cachedNames[fd];
}
// suffix may be comma-separated
int8_t TeensyFileManager::readDir(const char *where, const char *suffix, char *outputFN, int8_t startIdx, uint16_t maxlen)
{
Threads::Scope locker(fslock);
// ... open, read, save next name, close, return name. Horribly
// inefficient but hopefully won't break the sd layer. And if it
// works then we can make this more efficient later.
// First entry is always "../"
if (startIdx == 0) {
strcpy(outputFN, "../");
return 0;
}
int8_t idxCount = 1;
File f = SD.open(where, FILE_READ);
while (1) {
File e = f.openNextFile();
if (!e) {
// No more - all done.
f.close();
return -1;
}
// Skip MAC fork files
// FIXME: strncpy
strcpy(outputFN, e.name()); // and we need maxlen-1 for trailing '/' on directories
if (outputFN[0] == '.') {
e.close();
continue;
}
// skip anything that has the wrong suffix and isn't a directory
if (suffix && !e.isDirectory() && strlen(outputFN) >= 3) {
const char *fsuff = &outputFN[strlen(outputFN)-3];
if (strstr(suffix, ",")) {
// multiple suffixes to check - all must be 3 chars long, FIXME
bool matchesAny = false;
const char *p = suffix;
while (*p && strlen(p)) {
if (!strncasecmp(fsuff, p, 3)) {
matchesAny = true;
break;
}
p = strstr(p, ",")+1;
}
if (!matchesAny) {
e.close();
continue;
}
} else {
// one suffix to check
if (strcasecmp(fsuff, suffix)) {
e.close();
continue;
}
}
}
if (idxCount == startIdx) {
if (e.isDirectory()) {
strcat(outputFN, "/");
}
e.close();
f.close();
return idxCount;
}
idxCount++;
}
/* NOTREACHED */
}
bool TeensyFileManager::_prepCache(int8_t fd)
{
if (cacheFd == -1 ||
cacheFd != fd) {
// Not our cached file, or we have no cached file
if (cacheFd != -1) {
// Close the old one if we had one
cacheFile.close();
cacheFd = -1;
}
// Open the new one
cacheFile = SD.open(cachedNames[fd], FILE_WRITE);
if (!cacheFile) {
return false;
}
cacheFd = fd; // cache is live
}
return true; // FIXME error handling
}
void TeensyFileManager::getRootPath(char *toWhere, int8_t maxLen)
{
strcpy(toWhere, "/A2DISKS/");
// strncpy(toWhere, "/A2DISKS/", maxLen);
}
// FIXME: this should be private
bool TeensyFileManager::setSeekPosition(int8_t fd, uint32_t pos)
{
seekToEnd(fd);
uint32_t endPos = getSeekPosition(fd);
if (pos >= endPos) {
return false;
}
fileSeekPositions[fd] = pos;
return true;
}
// FIXME: this should be private
void TeensyFileManager::seekToEnd(int8_t fd)
{
File f = SD.open(cachedNames[fd], FILE_READ);
if (!f) {
return;
}
fileSeekPositions[fd] = f.size();
f.close();
}
int TeensyFileManager::write(int8_t fd, const void *buf, int nbyte)
{
Threads::Scope locker(fslock);
// open, seek, write, close.
if (fd < 0 || fd >= numCached) {
return -1;
}
if (cachedNames[fd][0] == 0) {
return -1;
}
_prepCache(fd);
uint32_t pos = fileSeekPositions[fd];
if (!cacheFile.seek(pos)) {
return -1;
}
if (cacheFile.write((const uint8_t *)buf, (size_t)nbyte) != (size_t)nbyte) {
return -1;
}
fileSeekPositions[fd] += nbyte;
cacheFile.close();
return nbyte;
};
int TeensyFileManager::read(int8_t fd, void *buf, int nbyte)
{
Threads::Scope locker(fslock);
// open, seek, read, close.
if (fd < 0 || fd >= numCached) {
return -1;
}
if (cachedNames[fd][0] == 0) {
return -1;
}
_prepCache(fd);
uint32_t pos = fileSeekPositions[fd];
if (!cacheFile.seek(pos)) {
return -1;
}
fileSeekPositions[fd] += nbyte;
if (cacheFile.read(buf, nbyte) != nbyte) {
return -1;
}
return nbyte;
};
int TeensyFileManager::lseek(int8_t fd, int offset, int whence)
{
Threads::Scope locker(fslock);
if (whence == SEEK_CUR && offset == 0) {
return fileSeekPositions[fd];
}
if (whence == SEEK_SET) {
if (!setSeekPosition(fd, offset))
return -1;
return offset;
}
// Other cases not supported yet
return -1;
};