[media] staging: lirc: freeing ERR_PTRs

We call kfree(data_buf) in the error handling and that will oops if this
is an error pointer.

Signed-off-by: Dan Carpenter <dan.carpenter@oracle.com>
Signed-off-by: Mauro Carvalho Chehab <mchehab@osg.samsung.com>

2 files changed, 2 insertions, 0 deletions

diff --git a/drivers/staging/media/lirc/lirc_imon.c b/drivers/staging/media/lirc/lirc_imon.c
index 96c76b33770b..5441f40cf5a1 100644
--- a/drivers/staging/media/lirc/lirc_imon.c
+++ b/drivers/staging/media/lirc/lirc_imon.c
@@ -414,6 +414,7 @@ static ssize_t vfd_write(struct file *file, const char __user *buf,
         data_buf = memdup_user(buf, n_bytes);
         if (IS_ERR(data_buf)) {
                 retval = PTR_ERR(data_buf);
+                data_buf = NULL;
                 goto exit;
         }
 
diff --git a/drivers/staging/media/lirc/lirc_sasem.c b/drivers/staging/media/lirc/lirc_sasem.c
index 81f90e17e1e6..c32e2965da87 100644
--- a/drivers/staging/media/lirc/lirc_sasem.c
+++ b/drivers/staging/media/lirc/lirc_sasem.c
@@ -392,6 +392,7 @@ static ssize_t vfd_write(struct file *file, const char __user *buf,
         data_buf = memdup_user((void const __user *)buf, n_bytes);
         if (IS_ERR(data_buf)) {
                 retval = PTR_ERR(data_buf);
+                data_buf = NULL;
                 goto exit;
         }