add RAMFS implementation

3 files changed, 455 insertions, 0 deletions

diff --git a/components/dfs/filesystems/ramfs/SConscript b/components/dfs/filesystems/ramfs/SConscript
new file mode 100644
index 0000000000..637ee2d9bc
--- /dev/null
+++ b/components/dfs/filesystems/ramfs/SConscript
@@ -0,0 +1,9 @@
+from building import *
+
+cwd     = GetCurrentDir()
+src     = Glob('*.c')
+CPPPATH = [cwd]
+
+group = DefineGroup('Filesystem', src, depend = ['RT_USING_DFS', 'RT_USING_MEMHEAP'], CPPPATH = CPPPATH)
+
+Return('group')
diff --git a/components/dfs/filesystems/ramfs/dfs_ramfs.c b/components/dfs/filesystems/ramfs/dfs_ramfs.c
new file mode 100644
index 0000000000..64f2aec798
--- /dev/null
+++ b/components/dfs/filesystems/ramfs/dfs_ramfs.c
@@ -0,0 +1,400 @@
+/*
+ * File      : dfs_ramfs.c
+ * This file is part of RT-Thread RTOS
+ * COPYRIGHT (C) 2011, Shanghai Real-Thread Technology Co., Ltd
+ *
+ * All rights reserved.
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2013-04-15     Bernard      the first version
+ * 2013-05-05     Bernard      remove CRC for ramfs persistence
+ */
+
+#include <rtthread.h>
+#include <dfs.h>
+#include <dfs_fs.h>
+#include "dfs_ramfs.h"
+
+int dfs_ramfs_mount(struct dfs_filesystem *fs, unsigned long rwflag, const void *data)
+{
+    struct dfs_ramfs* ramfs;
+
+    if (data == RT_NULL) return -DFS_STATUS_EIO;
+
+    ramfs = (struct dfs_ramfs*) data;
+    fs->data = ramfs;
+
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_unmount(struct dfs_filesystem *fs)
+{
+    fs->data = RT_NULL;
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_statfs(struct dfs_filesystem *fs, struct statfs *buf)
+{
+    struct dfs_ramfs *ramfs;
+
+    ramfs = (struct dfs_ramfs *)fs->data;
+    RT_ASSERT(ramfs != RT_NULL);
+    RT_ASSERT(buf != RT_NULL);
+
+    buf->f_bsize = 512;
+    buf->f_blocks = ramfs->memheap.pool_size/512;
+    buf->f_bfree = ramfs->memheap.available_size/512;
+
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_ioctl(struct dfs_fd *file, int cmd, void *args)
+{
+    return -DFS_STATUS_EIO;
+}
+
+struct ramfs_dirent *dfs_ramfs_lookup(struct dfs_ramfs* ramfs, const char *path, rt_size_t *size)
+{
+    const char *subpath;
+    struct ramfs_dirent *dirent;
+
+    subpath = path;
+    while (*subpath == '/' && *subpath) subpath ++;
+    if (! *subpath) /* is root directory */
+    {
+        *size = 0;
+        return &(ramfs->root);
+    }
+
+    for (dirent = rt_list_entry(ramfs->root.list.next, struct ramfs_dirent, list);
+            dirent != &(ramfs->root);
+            dirent = rt_list_entry(dirent->list.next, struct ramfs_dirent, list))
+    {
+        if (rt_strcmp(dirent->name, subpath) == 0)
+        {
+            *size = dirent->size;
+            return dirent;
+        }
+    }
+
+    /* not found */
+    return RT_NULL;
+}
+
+int dfs_ramfs_read(struct dfs_fd *file, void *buf, rt_size_t count)
+{
+    rt_size_t length;
+    struct ramfs_dirent *dirent;
+
+    dirent = (struct ramfs_dirent *)file->data;
+    RT_ASSERT(dirent != RT_NULL);
+
+    if (count < file->size - file->pos)
+        length = count;
+    else
+        length = file->size - file->pos;
+
+    if (length > 0)
+        memcpy(buf, &(dirent->data[file->pos]), length);
+
+    /* update file current position */
+    file->pos += length;
+
+    return length;
+}
+
+int dfs_ramfs_write(struct dfs_fd *fd, const void *buf, rt_size_t count)
+{
+    struct ramfs_dirent *dirent;
+    struct dfs_ramfs *ramfs;
+
+    ramfs = (struct dfs_ramfs*)fd->fs->data;
+    RT_ASSERT(ramfs != RT_NULL);
+    dirent = (struct ramfs_dirent*)fd->data;
+    RT_ASSERT(dirent != RT_NULL);
+
+    if (count + fd->pos > fd->size)
+    {
+        rt_uint8_t *ptr;
+        ptr = rt_memheap_realloc(&(ramfs->memheap), dirent->data, fd->pos + count);
+        if (ptr == RT_NULL)
+        {
+            rt_set_errno(-RT_ENOMEM);
+            return 0;
+        }
+
+        /* update dirent and file size */
+        dirent->data = ptr;
+        dirent->size = fd->pos + count;
+        fd->size = dirent->size;
+    }
+
+    if (count > 0)
+        memcpy(dirent->data + fd->pos, buf, count);
+
+    /* update file current position */
+    fd->pos += count;
+
+    return count;
+}
+
+int dfs_ramfs_lseek(struct dfs_fd *file, rt_off_t offset)
+{
+    if (offset <= (rt_off_t)file->size)
+    {
+        file->pos = offset;
+        return file->pos;
+    }
+
+    return -DFS_STATUS_EIO;
+}
+
+int dfs_ramfs_close(struct dfs_fd *file)
+{
+    file->data = RT_NULL;
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_open(struct dfs_fd *file)
+{
+    rt_size_t size;
+    struct dfs_ramfs *ramfs;
+    struct ramfs_dirent *dirent;
+
+    ramfs = (struct dfs_ramfs *)file->fs->data;
+    RT_ASSERT(ramfs != RT_NULL);
+
+    if (file->flags & DFS_O_DIRECTORY)
+    {
+        if (file->flags & DFS_O_CREAT)
+        {
+            return -DFS_STATUS_ENOSPC;
+        }
+
+        /* open directory */
+        dirent = dfs_ramfs_lookup(ramfs, file->path, &size);
+        if (dirent == RT_NULL)
+            return -DFS_STATUS_ENOENT;
+        if (dirent == &(ramfs->root)) /* it's root directory */
+        {
+            if (!(file->flags & DFS_O_DIRECTORY))
+            {
+                return -DFS_STATUS_ENOENT;
+            }
+        }
+    }
+    else
+    {
+        dirent = dfs_ramfs_lookup(ramfs, file->path, &size);
+        if (dirent == &(ramfs->root)) /* it's root directory */
+        {
+            return -DFS_STATUS_ENOENT;
+        }
+
+        if (dirent == RT_NULL)
+        {
+            if (file->flags & DFS_O_CREAT ||
+                file->flags & DFS_O_WRONLY)
+            {
+                char *name_ptr;
+
+                /* create a file entry */
+                dirent = (struct ramfs_dirent*)rt_memheap_alloc(&(ramfs->memheap), sizeof(struct ramfs_dirent));
+                if (dirent == RT_NULL)
+                {
+                    return -DFS_STATUS_ENOMEM;
+                }
+
+                /* remove '/' separator */
+                name_ptr = file->path;
+                while (*name_ptr == '/' && *name_ptr) name_ptr ++;
+                strncpy(dirent->name, name_ptr, RAMFS_NAME_MAX);
+
+                rt_list_init(&(dirent->list));
+                dirent->data = RT_NULL;
+                dirent->size = 0;
+                /* add to the root directory */
+                rt_list_insert_after(&(ramfs->root.list), &(dirent->list));
+            }
+        }
+
+        /* Creates a new file. If the file is existing, it is truncated and overwritten. */
+        if (file->flags & DFS_O_TRUNC)
+        {
+            dirent->size = 0;
+            if (dirent->data != RT_NULL)
+            {
+                rt_memheap_free(dirent->data);
+                dirent->data = RT_NULL;
+            }
+        }
+    }
+
+    file->data = dirent;
+    file->size = dirent->size;
+    file->pos = 0;
+
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_stat(struct dfs_filesystem *fs, const char *path, struct stat *st)
+{
+    rt_size_t size;
+    struct ramfs_dirent *dirent;
+    struct dfs_ramfs *ramfs;
+
+    ramfs = (struct dfs_ramfs *)fs->data;
+    dirent = dfs_ramfs_lookup(ramfs, path, &size);
+
+    if (dirent == RT_NULL)
+        return -DFS_STATUS_ENOENT;
+
+    st->st_dev = 0;
+    st->st_mode = DFS_S_IFREG | DFS_S_IRUSR | DFS_S_IRGRP | DFS_S_IROTH |
+                  DFS_S_IWUSR | DFS_S_IWGRP | DFS_S_IWOTH;
+
+    st->st_size = dirent->size;
+    st->st_mtime = 0;
+    st->st_blksize = 512;
+
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_getdents(struct dfs_fd *file, struct dirent *dirp, rt_uint32_t count)
+{
+    rt_size_t index, end;
+    struct dirent *d;
+    struct ramfs_dirent *dirent;
+    struct dfs_ramfs* ramfs;
+
+    ramfs = (struct dfs_ramfs*)file->fs->data;
+    dirent = (struct ramfs_dirent *)file->data;
+    if (dirent != &(ramfs->root)) return -DFS_STATUS_EINVAL;
+
+    /* make integer count */
+    count = (count / sizeof(struct dirent));
+    if (count == 0) return -DFS_STATUS_EINVAL;
+
+    end = file->pos + count;
+    index = 0;
+    count = 0;
+    for (dirent = rt_list_entry(dirent->list.next, struct ramfs_dirent, list);
+            dirent != &(ramfs->root) && index < end;
+            dirent = rt_list_entry(dirent->list.next, struct ramfs_dirent, list))
+    {
+        if (index >= (rt_size_t)file->pos)
+        {
+            d = dirp + count;
+            d->d_type = DFS_DT_REG;
+            d->d_namlen = RT_NAME_MAX;
+            d->d_reclen = (rt_uint16_t)sizeof(struct dirent);
+            rt_strncpy(d->d_name, dirent->name, RAMFS_NAME_MAX);
+
+            count += 1;
+            file->pos += 1;
+        }
+        index += 1;
+    }
+
+    return count * sizeof(struct dirent);
+}
+
+int dfs_ramfs_unlink(struct dfs_filesystem *fs, const char *path)
+{
+    rt_size_t size;
+    struct dfs_ramfs *ramfs;
+    struct ramfs_dirent *dirent;
+
+    ramfs = (struct dfs_ramfs *)fs->data;
+    RT_ASSERT(ramfs != RT_NULL);
+
+    dirent = dfs_ramfs_lookup(ramfs, path, &size);
+    if (dirent == RT_NULL) return -DFS_STATUS_ENOENT;
+
+    rt_list_remove(&(dirent->list));
+    if (dirent->data != RT_NULL)
+        rt_memheap_free(dirent->data);
+    rt_memheap_free(dirent);
+
+    return DFS_STATUS_OK;
+}
+
+int dfs_ramfs_rename(struct dfs_filesystem *fs, const char *oldpath, const char *newpath)
+{
+    struct ramfs_dirent *dirent;
+    struct dfs_ramfs *ramfs;
+    rt_size_t size;
+
+    ramfs = (struct dfs_ramfs *)fs->data;
+    RT_ASSERT(ramfs != RT_NULL);
+
+    dirent = dfs_ramfs_lookup(ramfs, newpath, &size);
+    if (dirent != RT_NULL) return -DFS_STATUS_EEXIST;
+
+    dirent = dfs_ramfs_lookup(ramfs, oldpath, &size);
+    if (dirent == RT_NULL) return -DFS_STATUS_ENOENT;
+
+    strncpy(dirent->name, newpath, RAMFS_NAME_MAX);
+
+    return DFS_STATUS_OK;
+}
+
+static const struct dfs_filesystem_operation _ramfs =
+{
+    "ram",
+    DFS_FS_FLAG_DEFAULT,
+    dfs_ramfs_mount,
+    dfs_ramfs_unmount,
+    RT_NULL, /* mkfs */
+    dfs_ramfs_statfs,
+
+    dfs_ramfs_open,
+    dfs_ramfs_close,
+    dfs_ramfs_ioctl,
+    dfs_ramfs_read,
+    dfs_ramfs_write,
+    RT_NULL, /* flush */
+    dfs_ramfs_lseek,
+    dfs_ramfs_getdents,
+    dfs_ramfs_unlink,
+    dfs_ramfs_stat,
+    dfs_ramfs_rename,
+};
+
+int dfs_ramfs_init(void)
+{
+    /* register ram file system */
+    dfs_register(&_ramfs);
+    return 0;
+}
+
+struct dfs_ramfs* dfs_ramfs_create(rt_uint8_t* pool, rt_size_t size)
+{
+    struct dfs_ramfs *ramfs;
+    rt_uint8_t *data_ptr;
+    rt_err_t result;
+
+    size = RT_ALIGN_DOWN(size, RT_ALIGN_SIZE);
+    ramfs = (struct dfs_ramfs*) pool;
+
+    data_ptr = (rt_uint8_t *)(ramfs + 1);
+    size = size - sizeof(struct dfs_ramfs);
+    size = RT_ALIGN_DOWN(size, RT_ALIGN_SIZE);
+
+    result = rt_memheap_init(&ramfs->memheap, "ramfs", data_ptr, size);
+    if (result != RT_EOK) return RT_NULL;
+	/* detach this memheap object from the system */
+	rt_object_detach((rt_object_t)&(ramfs->memheap));
+
+    /* initialize ramfs object */
+    ramfs->magic = RAMFS_MAGIC;
+
+    /* initialize root directory */
+    memset(&(ramfs->root), 0x00, sizeof(ramfs->root));
+    rt_list_init(&(ramfs->root.list));
+    ramfs->root.size = 0;
+    strcpy(ramfs->root.name, ".");
+
+    return ramfs;
+}
diff --git a/components/dfs/filesystems/ramfs/dfs_ramfs.h b/components/dfs/filesystems/ramfs/dfs_ramfs.h
new file mode 100644
index 0000000000..b47422b154
--- /dev/null
+++ b/components/dfs/filesystems/ramfs/dfs_ramfs.h
@@ -0,0 +1,46 @@
+/*
+ * File      : dfs_ramfs.h
+ * This file is part of RT-Thread RTOS
+ * COPYRIGHT (C) 2011, Shanghai Real-Thread Technology Co., Ltd
+ *
+ * All rights reserved.
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2013-04-15     Bernard      the first version
+ * 2013-05-05     Bernard      remove CRC for ramfs persistence
+ */
+
+#ifndef __DFS_ROMFS_H__
+#define __DFS_ROMFS_H__
+
+#include <rtthread.h>
+#include <rtservice.h>
+
+#define RAMFS_NAME_MAX  32
+#define RAMFS_MAGIC		0x0A0A0A0A
+
+struct ramfs_dirent
+{
+    rt_list_t list;
+    char name[RAMFS_NAME_MAX];	/* dirent name */
+    rt_uint8_t* data;
+
+    rt_size_t size;	/* file size */
+};
+
+/**
+ * DFS ramfs object
+ */
+struct dfs_ramfs
+{
+    rt_uint32_t magic;
+
+    struct rt_memheap memheap;
+    struct ramfs_dirent root;
+};
+
+int dfs_ramfs_init(void);
+struct dfs_ramfs* dfs_ramfs_create(rt_uint8_t* pool, rt_size_t size);
+
+#endif