1 /*
2  * Copyright (c) 2006-2023, RT-Thread Development Team
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  *
6  * Change Logs:
7  * Date           Author       Notes
8  * 2023-07-04     zhkag        first Version
9  */
10 
11 #include <rtthread.h>
12 #include <rthw.h>
13 #include <dfs_file.h>
14 #include "dfs_mqueue.h"
15 
16 static rt_list_t _mqueue_file_list = RT_LIST_OBJECT_INIT(_mqueue_file_list);
17 struct rt_spinlock mqueue_lock;
18 
dfs_mqueue_insert_after(rt_list_t * n)19 void dfs_mqueue_insert_after(rt_list_t *n) {
20     rt_spin_lock(&mqueue_lock);
21     rt_list_insert_after(&(_mqueue_file_list), n);
22     rt_spin_unlock(&mqueue_lock);
23 }
24 
dfs_mqueue_lookup(const char * path,rt_size_t * size)25 struct mqueue_file *dfs_mqueue_lookup(const char *path, rt_size_t *size) {
26     struct mqueue_file *file;
27     rt_list_t *node;
28     rt_spin_lock(&mqueue_lock);
29     rt_list_for_each(node, &_mqueue_file_list) {
30         file = rt_list_entry(node, struct mqueue_file, list);
31 
32         if (rt_strncmp(file->name, path, RT_NAME_MAX) == 0) {
33             *size = file->size;
34             rt_spin_unlock(&mqueue_lock);
35             return file;
36         }
37     }
38     rt_spin_unlock(&mqueue_lock);
39     return RT_NULL;
40 }
41 
dfs_mqueue_mount(struct dfs_filesystem * fs,unsigned long rwflag,const void * data)42 int dfs_mqueue_mount(struct dfs_filesystem *fs, unsigned long rwflag, const void *data) {
43     return RT_EOK;
44 }
45 
dfs_mqueue_unmount(struct dfs_filesystem * fs)46 int dfs_mqueue_unmount(struct dfs_filesystem *fs) { return RT_EOK; }
47 
dfs_mqueue_statfs(struct dfs_filesystem * fs,struct statfs * buf)48 int dfs_mqueue_statfs(struct dfs_filesystem *fs, struct statfs *buf) { return RT_EOK; }
49 
dfs_mqueue_close(struct dfs_file * file)50 int dfs_mqueue_close(struct dfs_file *file) { return RT_EOK; }
51 
dfs_mqueue_open(struct dfs_file * file)52 int dfs_mqueue_open(struct dfs_file *file) {
53     rt_size_t size;
54     if ((file->vnode->path[0] == '/') && (file->vnode->path[1] == '\0'))
55         return 0;
56 
57     if (file->flags & O_DIRECTORY)
58         return -ENOENT;
59 
60     struct mqueue_file *mq_file;
61 
62     mq_file = dfs_mqueue_lookup(file->vnode->path + 1, &size);
63     if (mq_file == RT_NULL && !(file->flags & O_CREAT))
64         return -ENOENT;
65     if (mq_file == RT_NULL) {
66         mq_file = (struct mqueue_file *)rt_malloc(sizeof(struct mqueue_file));
67         if (mq_file == RT_NULL) {
68             return -ENFILE;
69         }
70         mq_file->msg_size = 8192;
71         mq_file->max_msgs = 10;
72         strncpy(mq_file->name, file->vnode->path + 1, RT_NAME_MAX);
73         dfs_mqueue_insert_after(&(mq_file->list));
74     }
75 
76     if (file->flags & O_CREAT) {
77         rt_mq_t mq = rt_mq_create(file->vnode->path + 1, mq_file->msg_size, mq_file->max_msgs,
78                                   RT_IPC_FLAG_FIFO);
79         mq_file->data = (void *)mq;
80         file->vnode->data = mq_file;
81         file->vnode->size = 0;
82     }
83 
84     return 0;
85 }
86 
dfs_mqueue_stat(struct dfs_filesystem * fs,const char * path,struct stat * st)87 int dfs_mqueue_stat(struct dfs_filesystem *fs, const char *path, struct stat *st) {
88     st->st_dev = 0;
89     st->st_mode = S_IFREG | S_IRUSR | S_IRGRP | S_IROTH | S_IWUSR | S_IWGRP | S_IWOTH;
90     st->st_size = 0;
91     st->st_mtime = 0;
92     return RT_EOK;
93 }
94 
dfs_mqueue_getdents(struct dfs_file * file,struct dirent * dirp,uint32_t count)95 int dfs_mqueue_getdents(struct dfs_file *file, struct dirent *dirp, uint32_t count) {
96     rt_size_t index, end;
97     struct dirent *d;
98     count = (count / sizeof(struct dirent));
99     end = file->pos + count;
100     index = 0;
101     count = 0;
102     struct mqueue_file *mq_file;
103     rt_list_t *node;
104     rt_spin_lock(&mqueue_lock);
105     rt_list_for_each(node, &_mqueue_file_list) {
106         if (index >= (rt_size_t)file->pos) {
107             mq_file = rt_list_entry(node, struct mqueue_file, list);
108             d = dirp + count;
109             d->d_namlen = RT_NAME_MAX;
110             d->d_reclen = (rt_uint16_t)sizeof(struct dirent);
111             rt_strncpy(d->d_name, mq_file->name, RT_NAME_MAX);
112             count += 1;
113             file->pos += 1;
114         }
115         index += 1;
116         if (index >= end) {
117             break;
118         }
119     }
120     rt_spin_unlock(&mqueue_lock);
121     return count * sizeof(struct dirent);
122 }
123 
dfs_mqueue_unlink(struct dfs_filesystem * fs,const char * path)124 int dfs_mqueue_unlink(struct dfs_filesystem *fs, const char *path) {
125     rt_size_t size;
126     struct mqueue_file *mq_file;
127     mq_file = dfs_mqueue_lookup(path + 1, &size);
128     if (mq_file == RT_NULL)
129         return -ENOENT;
130     rt_list_remove(&(mq_file->list));
131     if (mq_file->data != RT_NULL)
132         rt_mq_delete((rt_mq_t)mq_file->data);
133     rt_free(mq_file);
134     return RT_EOK;
135 }
136 
137 static const struct dfs_file_ops _mqueue_fops = {
138     .open = dfs_mqueue_open,
139     .close = dfs_mqueue_close,
140     .getdents = dfs_mqueue_getdents,
141 };
142 
143 static const struct dfs_filesystem_ops _mqueue = {
144     .name = "mqueue",
145     .flags = DFS_FS_FLAG_DEFAULT,
146     .fops = &_mqueue_fops,
147 
148     .mount = dfs_mqueue_mount,
149     .unmount = dfs_mqueue_unmount,
150     .statfs = dfs_mqueue_statfs,
151 
152     .unlink = dfs_mqueue_unlink,
153     .stat = dfs_mqueue_stat,
154 };
155 
dfs_mqueue_init(void)156 int dfs_mqueue_init(void) {
157     /* register mqueue file system */
158     dfs_register(&_mqueue);
159     mkdir("/dev/mqueue", 0x777);
160     if (dfs_mount(RT_NULL, "/dev/mqueue", "mqueue", 0, 0) != 0) {
161         rt_kprintf("Dir /dev/mqueue mount failed!\n");
162     }
163     return 0;
164 }
165 INIT_ENV_EXPORT(dfs_mqueue_init);
166