implement flash protection

2 files changed, 29 insertions, 0 deletions

diff --git a/fs/devfs.c b/fs/devfs.c
index aafcc34439..4fc0dbb91b 100644
--- a/fs/devfs.c
+++ b/fs/devfs.c
@@ -50,6 +50,13 @@ static int devfs_erase(struct device_d *_dev, FILE *f, size_t count, unsigned lo
 	return dev_erase(dev, count, offset);
 }
 
+static int devfs_protect(struct device_d *_dev, FILE *f, size_t count, unsigned long offset, int prot)
+{
+	struct device_d *dev = f->inode;
+
+	return dev_protect(dev, count, offset, prot);
+}
+
 static int devfs_memmap(struct device_d *_dev, FILE *f, void **map, int flags)
 {
 	struct device_d *dev = f->inode;
@@ -153,6 +160,7 @@ static struct fs_driver_d devfs_driver = {
 	.closedir  = devfs_closedir,
 	.stat      = devfs_stat,
 	.erase     = devfs_erase,
+	.protect   = devfs_protect,
 	.memmap    = devfs_memmap,
 	.flags     = FS_DRIVER_NO_DEV,
 	.drv = {
diff --git a/fs/fs.c b/fs/fs.c
index 57d75ddb3b..23a1aa0ad6 100644
--- a/fs/fs.c
+++ b/fs/fs.c
@@ -516,6 +516,27 @@ int erase(int fd, size_t count, unsigned long offset)
 	return errno;
 }
 
+int protect(int fd, size_t count, unsigned long offset, int prot)
+{
+	struct device_d *dev;
+	struct fs_driver_d *fsdrv;
+	FILE *f = &files[fd];
+
+	dev = f->dev;
+
+	fsdrv = (struct fs_driver_d *)dev->driver->type_data;
+
+	if (f->pos + count > f->size)
+		count = f->size - f->pos;
+
+	if (fsdrv->protect)
+		errno = fsdrv->protect(dev, f, count, offset, prot);
+	else
+		errno = -EINVAL;
+
+	return errno;
+}
+
 void *memmap(int fd, int flags)
 {
 	struct device_d *dev;