fix madvise calls

1 files changed, 4 insertions, 2 deletions

diff --git a/drmtest.c b/drmtest.c
index d6d3381..a7d78bd 100644
--- a/drmtest.c
+++ b/drmtest.c
@@ -213,8 +213,10 @@ static void drm_check_content(const char *grp)
 
 static void drm_zap_mappings(void)
 {
-    madvise(fbmem, creq.size, MADV_DONTNEED);
-    madvise(dmabuf_mem, creq.size, MADV_DONTNEED);
+    if (fbmem)
+        madvise(fbmem, creq.size, MADV_DONTNEED);
+    if (dmabuf_mem)
+        madvise(dmabuf_mem, creq.size, MADV_DONTNEED);
 }
 
 /* ------------------------------------------------------------------ */