1 files changed, 6 insertions, 1 deletions

diff --git a/drivers/media/video/ivtv/ivtv-udma.c b/drivers/media/video/ivtv/ivtv-udma.c
index 1daf1dd65bf7..69cc8166b20b 100644
--- a/drivers/media/video/ivtv/ivtv-udma.c
+++ b/drivers/media/video/ivtv/ivtv-udma.c
@@ -132,7 +132,12 @@ int ivtv_udma_setup(struct ivtv *itv, unsigned long ivtv_dest_addr,
         if (user_dma.page_count != err) {
                 IVTV_DEBUG_WARN("failed to map user pages, returned %d instead of %d\n",
                            err, user_dma.page_count);
-                return -EINVAL;
+                if (err >= 0) {
+                        for (i = 0; i < err; i++)
+                                put_page(dma->map[i]);
+                        return -EINVAL;
+                }
+                return err;
         }
 
         dma->page_count = user_dma.page_count;