2 files changed, 7 insertions, 16 deletions

diff --git a/drivers/media/dvb/dvb-core/dvb_demux.c b/drivers/media/dvb/dvb-core/dvb_demux.c
index 977ddba3e235..4a88a3e4db2b 100644
--- a/drivers/media/dvb/dvb-core/dvb_demux.c
+++ b/drivers/media/dvb/dvb-core/dvb_demux.c
@@ -1130,13 +1130,9 @@ static int dvbdmx_write(struct dmx_demux *demux, const char __user *buf, size_t
         if ((!demux->frontend) || (demux->frontend->source != DMX_MEMORY_FE))
                 return -EINVAL;
 
-        p = kmalloc(count, GFP_USER);
-        if (!p)
-                return -ENOMEM;
-        if (copy_from_user(p, buf, count)) {
-                kfree(p);
-                return -EFAULT;
-        }
+        p = memdup_user(buf, count);
+        if (IS_ERR(p))
+                return PTR_ERR(p);
         if (mutex_lock_interruptible(&dvbdemux->mutex)) {
                 kfree(p);
                 return -ERESTARTSYS;
diff --git a/drivers/media/video/dabusb.c b/drivers/media/video/dabusb.c
index 0f505086774c..5b176bd7afdb 100644
--- a/drivers/media/video/dabusb.c
+++ b/drivers/media/video/dabusb.c
@@ -706,16 +706,11 @@ static long dabusb_ioctl (struct file *file, unsigned int cmd, unsigned long arg
         switch (cmd) {
 
         case IOCTL_DAB_BULK:
-                pbulk = kmalloc(sizeof (bulk_transfer_t), GFP_KERNEL);
+                pbulk = memdup_user((void __user *)arg,
+                                    sizeof(bulk_transfer_t));
 
-                if (!pbulk) {
-                        ret = -ENOMEM;
-                        break;
-                }
-
-                if (copy_from_user (pbulk, (void __user *) arg, sizeof (bulk_transfer_t))) {
-                        ret = -EFAULT;
-                        kfree (pbulk);
+                if (IS_ERR(pbulk)) {
+                        ret = PTR_ERR(pbulk);
                         break;
                 }