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