video:uvesafb: check the return value of kzalloc

Michal maybe forgot it merely, we should add code
to check the return value of kzalloc to make the
code more robust.

Signed-off-by: Wang YanQing <udknight@gmail.com>
Signed-off-by: Florian Tobias Schandinat <FlorianSchandinat@gmx.de>

diff --git a/drivers/video/uvesafb.c b/drivers/video/uvesafb.c
index b0e2a42..2f8f82d 100644 (file)
--- a/drivers/video/uvesafb.c
+++ b/drivers/video/uvesafb.c
@@ -659,6 +659,8 @@ static int __devinit uvesafb_vbe_getedid(struct uvesafb_ktask *task,
        task->t.flags = TF_BUF_RET | TF_BUF_ESDI;
        task->t.buf_len = EDID_LENGTH;
        task->buf = kzalloc(EDID_LENGTH, GFP_KERNEL);
+       if (!task->buf)
+               return -ENOMEM;
 
        err = uvesafb_exec(task);