1 files changed, 41 insertions, 5 deletions

diff --git a/kernel/features/fs.c b/kernel/features/fs.c
index 687d7ad..4d19dde 100644
--- a/kernel/features/fs.c
+++ b/kernel/features/fs.c
@@ -131,7 +131,8 @@ s32 vfs_read(const char *path, void *buf, u32 offset, u32 count)
 		path++;
 
 	struct mount_info *m = vfs_find_mount_info(path);
-	assert(m && m->dev && m->dev->vfs && m->dev->vfs->read && m->dev->vfs->perm);
+	if (!(m && m->dev && m->dev->vfs && m->dev->vfs->read && m->dev->vfs->perm))
+		return -1;
 
 	u32 len = strlen(m->path);
 	if (len > 1)
@@ -156,7 +157,8 @@ s32 vfs_write(const char *path, void *buf, u32 offset, u32 count)
 		path++;
 
 	struct mount_info *m = vfs_find_mount_info(path);
-	assert(m && m->dev && m->dev->vfs && m->dev->vfs->write && m->dev->vfs->perm);
+	if (!(m && m->dev && m->dev->vfs && m->dev->vfs->write && m->dev->vfs->perm))
+		return -1;
 
 	u32 len = strlen(m->path);
 	if (len > 1)
@@ -168,6 +170,25 @@ s32 vfs_write(const char *path, void *buf, u32 offset, u32 count)
 	return m->dev->vfs->write(path, buf, offset, count, m->dev);
 }
 
+s32 vfs_ioctl(const char *path, u32 request, void *arg1, void *arg2, void *arg3)
+{
+	while (*path == ' ')
+		path++;
+
+	struct mount_info *m = vfs_find_mount_info(path);
+	if (!(m && m->dev && m->dev->vfs && m->dev->vfs->ioctl && m->dev->vfs->perm))
+		return -1;
+
+	u32 len = strlen(m->path);
+	if (len > 1)
+		path += len;
+
+	if (!m->dev->vfs->perm(path, VFS_WRITE, m->dev) && !proc_super())
+		return -1;
+
+	return m->dev->vfs->ioctl(path, request, arg1, arg2, arg3, m->dev);
+}
+
 s32 vfs_stat(const char *path, struct stat *buf)
 {
 	while (*path == ' ')
@@ -177,12 +198,16 @@ s32 vfs_stat(const char *path, struct stat *buf)
 		return -1;
 
 	struct mount_info *m = vfs_find_mount_info(path);
-	assert(m && m->dev && m->dev->vfs && m->dev->vfs->stat);
+	if (!(m && m->dev && m->dev->vfs && m->dev->vfs->stat && m->dev->vfs->perm))
+		return -1;
 
 	u32 len = strlen(m->path);
 	if (len > 1)
 		path += len;
 
+	if (!m->dev->vfs->perm(path, VFS_WRITE, m->dev) && !proc_super())
+		return -1;
+
 	return m->dev->vfs->stat(path, buf, m->dev);
 }
 
@@ -192,7 +217,8 @@ s32 vfs_wait(const char *path, u32 func_ptr)
 		path++;
 
 	struct mount_info *m = vfs_find_mount_info(path);
-	assert(m && m->dev && m->dev->vfs);
+	if (!(m && m->dev && m->dev->vfs))
+		return -1;
 
 	// Default wait
 	if (!m->dev->vfs->wait) {
@@ -280,10 +306,19 @@ static s32 devfs_read(const char *path, void *buf, u32 offset, u32 count, struct
 {
 	struct device *target = device_get_by_name(path + 1);
 	if (!target || !target->read)
-		return 0;
+		return -1;
 	return target->read(buf, offset, count, dev);
 }
 
+static s32 devfs_ioctl(const char *path, u32 request, void *arg1, void *arg2, void *arg3,
+		       struct device *dev)
+{
+	struct device *target = device_get_by_name(path + 1);
+	if (!target || !target->ioctl)
+		return -1;
+	return target->ioctl(request, arg1, arg2, arg3, dev);
+}
+
 static u8 devfs_perm(const char *path, enum vfs_perm perm, struct device *dev)
 {
 	(void)path;
@@ -309,6 +344,7 @@ void device_install(void)
 	struct vfs *vfs = zalloc(sizeof(*vfs));
 	vfs->type = VFS_DEVFS;
 	vfs->read = devfs_read;
+	vfs->ioctl = devfs_ioctl;
 	vfs->perm = devfs_perm;
 	vfs->ready = devfs_ready;
 	struct device *dev = zalloc(sizeof(*dev));