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>

1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/video/uvesafb.c b/drivers/video/uvesafb.c
index b0e2a4261afe..2f8f82d874a1 100644
--- 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);