From 73b713e7a3d7fdd492adbf7ec7eb6fea231d31da Mon Sep 17 00:00:00 2001
From: Lim Siew Hoon <siew.hoon.lim@intel.com>
Date: Fri, 1 Jul 2016 13:29:09 +0800
Subject: Add assert check memory allocation potential NULL issue

Signed-off-by: Lim Siew Hoon <siew.hoon.lim@intel.com>
(cherry picked from commit c0fa72c00aa661faf51d7c1fb1275ef11c5a9667)
---
 test/putsurface/putsurface_wayland.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/putsurface/putsurface_wayland.c b/test/putsurface/putsurface_wayland.c
index c3b504f..67da475 100644
--- a/test/putsurface/putsurface_wayland.c
+++ b/test/putsurface/putsurface_wayland.c
@@ -25,6 +25,7 @@
 #include <stddef.h>
 #include <errno.h>
 #include <sys/select.h>
+#include <assert.h>
 #ifdef IN_LIBVA
 # include "va/wayland/va_wayland.h"
 #else
@@ -257,6 +258,7 @@ create_window(void *win_display, int x, int y, int width, int height)
     wl_shell_surface_set_toplevel(shell_surface);
 
     drawable1 = malloc(sizeof(*drawable1));
+    assert(drawable1);
     drawable1->display          = display;
     drawable1->surface          = surface1;
     drawable1->redraw_pending   = 0;
@@ -272,6 +274,7 @@ create_window(void *win_display, int x, int y, int width, int height)
     wl_shell_surface_set_toplevel(shell_surface_2);
 
     drawable2 = malloc(sizeof(*drawable2));
+    assert(drawable2);
     drawable2->display          = display;
     drawable1->surface          = surface2;
     drawable2->redraw_pending   = 0;
-- 
cgit v1.2.3