Signed-off-by: Bryce Harrington <br...@osg.samsung.com>
---
 tests/weston-test-client-helper.c | 11 +++++++++++
 tests/weston-test-client-helper.h | 16 ++++++++++------
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/tests/weston-test-client-helper.c 
b/tests/weston-test-client-helper.c
index b05be83..54b6c95 100644
--- a/tests/weston-test-client-helper.c
+++ b/tests/weston-test-client-helper.c
@@ -32,6 +32,17 @@
 #include "../shared/os-compatibility.h"
 #include "weston-test-client-helper.h"
 
+void *
+fail_on_null(void *p)
+{
+       if (p == NULL) {
+               fprintf(stderr, "out of memory\n");
+               exit(EXIT_FAILURE);
+       }
+       return p;
+}
+
+
 int
 surface_contains(struct surface *surface, int x, int y)
 {
diff --git a/tests/weston-test-client-helper.h 
b/tests/weston-test-client-helper.h
index 429f088..0378804 100644
--- a/tests/weston-test-client-helper.h
+++ b/tests/weston-test-client-helper.h
@@ -132,15 +132,19 @@ struct surface {
        void *data;
 };
 
+void *
+fail_on_null(void *p);
+
 static inline void *
-xzalloc(size_t size)
+xzalloc(size_t s)
 {
-        void *p;
-
-        p = calloc(1, size);
-        assert(p);
+       return fail_on_null(calloc(1, s));
+}
 
-        return p;
+static inline void *
+xmalloc(size_t s)
+{
+       return fail_on_null(malloc(s));
 }
 
 struct client *
-- 
1.9.1

_______________________________________________
wayland-devel mailing list
wayland-devel@lists.freedesktop.org
http://lists.freedesktop.org/mailman/listinfo/wayland-devel

Reply via email to