diff --git a/src/main.cpp b/src/main.cpp index 1e8bb47..71b479a 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -104,6 +104,10 @@ ON_APPLICATION_START(args) { VirtualMountDevice("storage_mlc:/"); VirtualMountDevice("storage_usb:/"); VirtualMountDevice("usb:/"); + + AddVirtualFSPath("vol", nullptr, nullptr); + AddVirtualFSVOLPath("external01", nullptr, nullptr); + AddVirtualFSVOLPath("content", nullptr, nullptr); } thread = BackgroundThread::getInstance(); diff --git a/src/virtualpath.c b/src/virtualpath.c index ea1508e..592c3f3 100644 --- a/src/virtualpath.c +++ b/src/virtualpath.c @@ -32,7 +32,13 @@ #include "utils/logger.h" uint8_t MAX_VIRTUAL_PARTITIONS = 0; -VIRTUAL_PARTITION * VIRTUAL_PARTITIONS = NULL; +VIRTUAL_PARTITION *VIRTUAL_PARTITIONS = NULL; + +uint8_t MAX_VIRTUAL_FS = 0; +VIRTUAL_PARTITION *VIRTUAL_FS = NULL; + +uint8_t MAX_VIRTUAL_FS_VOL = 0; +VIRTUAL_PARTITION *VIRTUAL_FS_VOL = NULL; void VirtualMountDevice(const char * path) { @@ -91,6 +97,45 @@ void AddVirtualPath(const char *name, const char *alias, const char *prefix) MAX_VIRTUAL_PARTITIONS++; } + + void AddVirtualFSPath(const char *name, const char *alias, const char *prefix) { + if (!VIRTUAL_FS) { + VIRTUAL_FS = (VIRTUAL_PARTITION *) malloc(sizeof(VIRTUAL_PARTITION)); + } + + VIRTUAL_PARTITION *tmp = realloc(VIRTUAL_FS, sizeof(VIRTUAL_PARTITION) * (MAX_VIRTUAL_FS + 1)); + if (!tmp) { + free(VIRTUAL_FS); + MAX_VIRTUAL_FS = 0; + return; + } + + VIRTUAL_FS = tmp; + + VIRTUAL_FS[MAX_VIRTUAL_FS].name = strdup(name); + + MAX_VIRTUAL_FS++; + } + + void AddVirtualFSVOLPath(const char *name, const char *alias, const char *prefix) { + if (!VIRTUAL_FS_VOL) { + VIRTUAL_FS_VOL = (VIRTUAL_PARTITION *) malloc(sizeof(VIRTUAL_PARTITION)); + } + + VIRTUAL_PARTITION *tmp = realloc(VIRTUAL_FS_VOL, sizeof(VIRTUAL_PARTITION) * (MAX_VIRTUAL_FS_VOL + 1)); + if (!tmp) { + free(VIRTUAL_FS_VOL); + MAX_VIRTUAL_FS_VOL = 0; + return; + } + + VIRTUAL_FS_VOL = tmp; + + VIRTUAL_FS_VOL[MAX_VIRTUAL_FS_VOL].name = strdup(name); + + MAX_VIRTUAL_FS_VOL++; + } + void MountVirtualDevices() { VirtualMountDevice("fs:/"); diff --git a/src/virtualpath.h b/src/virtualpath.h index e884867..3cae99e 100644 --- a/src/virtualpath.h +++ b/src/virtualpath.h @@ -47,8 +47,16 @@ typedef struct { extern VIRTUAL_PARTITION * VIRTUAL_PARTITIONS; extern uint8_t MAX_VIRTUAL_PARTITIONS; +extern VIRTUAL_PARTITION * VIRTUAL_FS; +extern uint8_t MAX_VIRTUAL_FS; + +extern VIRTUAL_PARTITION * VIRTUAL_FS_VOL; +extern uint8_t MAX_VIRTUAL_FS_VOL; + void VirtualMountDevice(const char * devicepath); void AddVirtualPath(const char *name, const char *alias, const char *prefix); +void AddVirtualFSPath(const char *name, const char *alias, const char *prefix); +void AddVirtualFSVOLPath(const char *name, const char *alias, const char *prefix); void MountVirtualDevices(); void UnmountVirtualPaths(); diff --git a/src/vrt.c b/src/vrt.c index 3d11fb5..e6e2252 100644 --- a/src/vrt.c +++ b/src/vrt.c @@ -155,13 +155,19 @@ char *to_real_path(char *virtual_cwd, char *virtual_path) { } static int checkdir(char *path) { - DIR *dir = opendir(path); - if(dir) - { - closedir(dir); - return 0; - } - return -1; + if (path != NULL && + (strcmp(path, "fs:/vol") == 0 || + strcmp(path, "fs:/vol/") == 0 || + strcmp(path, "fs:/") == 0)) { + return 0; + } + + DIR *dir = opendir(path); + if (dir) { + closedir(dir); + return 0; + } + return -1; } typedef void * (*path_func)(char *path, ...); @@ -264,65 +270,89 @@ int vrt_rename(char *cwd, char *from_path, char *to_path) { /* When in vfs-root this creates a fake DIR_ITER. */ -DIR_P *vrt_opendir(char *cwd, char *path) -{ - char *real_path = to_real_path(cwd, path); - if (!real_path) return NULL; +DIR_P *vrt_opendir(char *cwd, char *path) { + char *real_path = to_real_path(cwd, path); + if (!real_path) { return NULL; } - DIR_P *iter = malloc(sizeof(DIR_P)); - if (!iter) - { - if (*real_path != 0) - free(real_path); - return NULL; - } + DIR_P *iter = malloc(sizeof(DIR_P)); + if (!iter) { + if (*real_path != 0) { + free(real_path); + } + return NULL; + } - iter->virt_root = 0; - iter->path = real_path; + iter->virt_root = 0; + iter->virtual_fs = 0; + iter->virtual_fs_vol = 0; + iter->path = real_path; - if (*iter->path == 0) { - iter->dir = malloc(sizeof(DIR)); - if(!iter->dir) { - // root path is not allocated - free(iter); - return NULL; - } - memset(iter->dir, 0, sizeof(DIR)); - iter->virt_root = 1; // we are at the virtual root - return iter; - } + if (*iter->path == 0 || (strncmp(iter->path, "fs:", 3) == 0 && strlen(iter->path) <= 4) || (strncmp(iter->path, "fs:/vol", 3) == 0 && strlen(iter->path) <= 8)) { + iter->dir = malloc(sizeof(DIR)); + if (!iter->dir) { + // root path is not allocated + free(iter); + return NULL; + } + memset(iter->dir, 0, sizeof(DIR)); - iter->dir = with_virtual_path(cwd, opendir, path, 0, NULL); - if(!iter->dir) - { - free(iter->path); - free(iter); - return NULL; - } + if (strncmp(iter->path, "fs:/vol", 7) == 0) { + iter->virtual_fs_vol = 1; // we are at the virtual fs + } else if (strncmp(iter->path, "fs:", 3) == 0) { + iter->virtual_fs = 1; // we are at the virtual fs + } else { + iter->virt_root = 1; // we are at the virtual root + } - return iter; + return iter; + } + + + iter->dir = with_virtual_path(cwd, opendir, path, 0, NULL); + if (!iter->dir) { + free(iter->path); + free(iter); + return NULL; + } + + + return iter; } /* Yields virtual aliases when pDir->virt_root */ struct dirent *vrt_readdir(DIR_P *pDir) { - if(!pDir || !pDir->dir) return NULL; + if (!pDir || !pDir->dir) { return NULL; } - DIR *iter = pDir->dir; - if (pDir->virt_root) { - for (; (uint32_t)iter->position < MAX_VIRTUAL_PARTITIONS; iter->position++) { - VIRTUAL_PARTITION *partition = VIRTUAL_PARTITIONS + (int)iter->position; - if (partition->inserted) { - iter->fileData.d_type = DT_DIR; - strcpy(iter->fileData.d_name, partition->alias + 1); - iter->position++; - return &iter->fileData; - } - } - return NULL; - } - return readdir(iter); + DIR *iter = pDir->dir; + if (pDir->virt_root || pDir->virtual_fs || pDir->virtual_fs_vol) { + int max = MAX_VIRTUAL_PARTITIONS; + VIRTUAL_PARTITION * PARTITION_PTR = VIRTUAL_PARTITIONS; + if(pDir->virtual_fs){ + max = MAX_VIRTUAL_FS; + PARTITION_PTR = VIRTUAL_FS; + } else if (pDir->virtual_fs_vol){ + max = MAX_VIRTUAL_FS_VOL; + PARTITION_PTR = VIRTUAL_FS_VOL; + } + for (; (uint32_t) iter->position < max; iter->position++) { + VIRTUAL_PARTITION *partition = PARTITION_PTR + (int) iter->position; + if (partition->inserted) { + iter->fileData.d_type = DT_DIR; + if(pDir->virtual_fs || pDir->virtual_fs_vol){ + strcpy(iter->fileData.d_name, partition->name); + }else{ + strcpy(iter->fileData.d_name, partition->alias + 1); + } + iter->position++; + return &iter->fileData; + } + } + return NULL; + } + + return readdir(iter); } int vrt_closedir(DIR_P *iter) { diff --git a/src/vrt.h b/src/vrt.h index 6c4735e..29f18a5 100644 --- a/src/vrt.h +++ b/src/vrt.h @@ -37,6 +37,8 @@ typedef struct DIR *dir; char *path; uint8_t virt_root; + uint8_t virtual_fs; + uint8_t virtual_fs_vol; } DIR_P; char *to_real_path(char *virtual_cwd, char *virtual_path);