Check the returned pointer from malloc() before using it

Signed-off-by: Xiang, Haihao <haihao.xiang@intel.com>
(cherry picked from commit 1caf179b1425b13cacaa421c688c6df8369668c6)

diff --git a/src/i965_drv_video.c b/src/i965_drv_video.c
index 606e279..4097750 100755 (executable)
--- a/src/i965_drv_video.c
+++ b/src/i965_drv_video.c
@@ -4291,6 +4291,9 @@ i965_QuerySurfaceAttributes(VADriverContextP ctx,
 
     attribs = malloc(I965_MAX_SURFACE_ATTRIBUTES *sizeof(*attribs));
     
+    if (attribs == NULL)
+        return VA_STATUS_ERROR_ALLOCATION_FAILED;
+
     if (IS_G4X(i965->intel.device_id)) {
         if (obj_config->profile == VAProfileMPEG2Simple ||
             obj_config->profile == VAProfileMPEG2Main) {