staging: wilc1000: prevent some overflows in debugfs

[cascardo/linux.git] / drivers / staging / wilc1000 / wilc_debugfs.c

diff --git a/drivers/staging/wilc1000/wilc_debugfs.c b/drivers/staging/wilc1000/wilc_debugfs.c
index 74b9fd5..c328208 100644 (file)
--- a/drivers/staging/wilc1000/wilc_debugfs.c
+++ b/drivers/staging/wilc1000/wilc_debugfs.c
@@ -53,6 +53,9 @@ static ssize_t wilc_debug_level_write(struct file *filp, const char *buf, size_t
        char buffer[128] = {};
        int flag = 0;
 
+       if (count > sizeof(buffer))
+               return -EINVAL;
+
        if (copy_from_user(buffer, buf, count)) {
                return -EFAULT;
        }
@@ -99,6 +102,9 @@ static ssize_t wilc_debug_region_write(struct file *filp, const char *buf, size_
        char buffer[128] = {};
        int flag;
 
+       if (count > sizeof(buffer))
+               return -EINVAL;
+
        if (copy_from_user(buffer, buf, count)) {
                return -EFAULT;
        }