2 files changed, 18 insertions, 0 deletions
diff --git a/lib/drmtest.c b/lib/drmtest.c
index ae7e6dd7..f3f05347 100644
--- a/lib/drmtest.c
+++ b/lib/drmtest.c
@@ -30,6 +30,7 @@
 #include <sys/stat.h>
 #include <sys/ioctl.h>
 #include <string.h>
+#include <sys/mman.h>
 #include "drmtest.h"
 #include "i915_drm.h"
 #include "intel_chipset.h"
@@ -202,3 +203,19 @@ uint32_t gem_create(int fd, int size)
 	return create.handle;
 }
 
+void *gem_mmap(int fd, uint32_t handle, int size, int prot)
+{
+	struct drm_i915_gem_mmap_gtt mmap_arg;
+	void *ptr;
+
+	mmap_arg.handle = handle;
+	if (drmIoctl(fd, DRM_IOCTL_I915_GEM_MMAP_GTT, &mmap_arg))
+		return NULL;
+
+	ptr = mmap(0, size, prot, MAP_SHARED, fd, mmap_arg.offset);
+	if (ptr == MAP_FAILED)
+		ptr = NULL;
+
+	return ptr;
+}
+
diff --git a/lib/drmtest.h b/lib/drmtest.h
index a3066c27..38118f17 100644
--- a/lib/drmtest.h
+++ b/lib/drmtest.h
@@ -45,3 +45,4 @@ void gem_set_domain(int fd, uint32_t handle,
 		    uint32_t read_domains, uint32_t write_domain);
 void gem_sync(int fd, uint32_t handle);
 uint32_t gem_create(int fd, int size);
+void *gem_mmap(int fd, uint32_t handle, int size, int prot);