Browse Source

V4L/DVB: videobuf: add missing checks for kzalloc returning NULL

Signed-off-by: Pawel Osciak <p.osciak@samsung.com>
Signed-off-by: Mauro Carvalho Chehab <mchehab@redhat.com>
Pawel Osciak 15 years ago
parent
commit
bee527f979
2 changed files with 4 additions and 0 deletions
  1. 2 0
      drivers/media/video/videobuf-dma-sg.c
  2. 2 0
      drivers/media/video/videobuf-vmalloc.c

+ 2 - 0
drivers/media/video/videobuf-dma-sg.c

@@ -418,6 +418,8 @@ static void *__videobuf_alloc(size_t size)
 	struct videobuf_buffer *vb;
 
 	vb = kzalloc(size+sizeof(*mem),GFP_KERNEL);
+	if (!vb)
+		return vb;
 
 	mem = vb->priv = ((char *)vb)+size;
 	mem->magic=MAGIC_SG_MEM;

+ 2 - 0
drivers/media/video/videobuf-vmalloc.c

@@ -138,6 +138,8 @@ static void *__videobuf_alloc(size_t size)
 	struct videobuf_buffer *vb;
 
 	vb = kzalloc(size+sizeof(*mem),GFP_KERNEL);
+	if (!vb)
+		return vb;
 
 	mem = vb->priv = ((char *)vb)+size;
 	mem->magic=MAGIC_VMAL_MEM;