Add GEM mmap() and KMS tests

'gdrm-gem-mmap' creates a GEM object, mmap()s it, and tests access.
'gdrm-kms-addfb' creates a GEM object, switches the active framebuffer to use it,
and writes random colours to it via memory mapped access.

5 files changed, 447 insertions, 2 deletions

diff --git a/Makefile b/Makefile
index bd710f8..3859a82 100644
--- a/Makefile
+++ b/Makefile
@@ -2,7 +2,8 @@ CFLAGS=-g -W -Wall
 LIBS=-lX11 -lXext -ldrm -ldrm_glamo
 CC=gcc
 
-default: gdri-cmdq-submission gdri-mem-manager gdri-buf-cmdq
+default: gdri-cmdq-submission gdri-mem-manager gdri-buf-cmdq gdrm-gem-mmap \
+         gdrm-kms-addfb
 
 .c.o:
 	$(CC) -c ${CFLAGS} ${CROSS_CFLAGS} $*.c
@@ -16,14 +17,22 @@ gdri-mem-manager: gdri-mem-manager.o dri2.o utils.o
 gdri-buf-cmdq: gdri-buf-cmdq.o dri2.o utils.o
 	$(CC) gdri-buf-cmdq.o dri2.o utils.o -o gdri-buf-cmdq ${LIBS}
 
+gdrm-gem-mmap: gdrm-gem-mmap.o drmtest.o
+	$(CC) gdrm-gem-mmap.o drmtest.o -o gdrm-gem-mmap ${LIBS}
+
+gdrm-kms-addfb: gdrm-kms-addfb.o drmtest.o
+	$(CC) gdrm-kms-addfb.o drmtest.o -o gdrm-kms-addfb ${LIBS}
+
 clean:
 	rm -rf dri2.o gdri-cmdq-submission.o utils.o gdri-cmdq-submission \
-		gdri-mem-manager gdri-buf-cmdq
+		gdri-mem-manager gdri-buf-cmdq gdrm-gem-mmap
 
 install:
 	install -d -m 755 ${PREFIX}/bin
 	install -m 755 gdri-cmdq-submission ${PREFIX}/bin
 	install -m 755 gdri-mem-manager ${PREFIX}/bin
 	install -m 755 gdri-buf-cmdq ${PREFIX}/bin
+	install -m 755 gdrm-gem-mmap ${PREFIX}/bin
+	install -m 755 gdrm-kms-addfb ${PREFIX}/bin
 
 .PHONY: clean install default
diff --git a/drmtest.c b/drmtest.c
new file mode 100644
index 0000000..d053865
--- /dev/null
+++ b/drmtest.c
@@ -0,0 +1,86 @@
+/*
+ * Copyright © 2007 Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+ * IN THE SOFTWARE.
+ *
+ * Authors:
+ *    Eric Anholt <eric@anholt.net>
+ *
+ */
+
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include "drmtest.h"
+
+/** Open the first DRM device we can find, searching up to 16 device nodes */
+int drm_open_any(void)
+{
+	char name[20];
+	int i, fd;
+
+	for (i = 0; i < 16; i++) {
+		sprintf(name, "/dev/dri/card%d", i);
+		fd = open(name, O_RDWR);
+		if (fd != -1)
+			return fd;
+	}
+	abort();
+}
+
+#if 0
+/**
+ * Open the first DRM device we can find where we end up being the master.
+ */
+int drm_open_any_master(void)
+{
+	char name[20];
+	int i, fd;
+
+	for (i = 0; i < 16; i++) {
+		drm_client_t client;
+		int ret;
+
+		sprintf(name, "/dev/dri/card%d", i);
+		fd = open(name, O_RDWR);
+		if (fd == -1)
+			continue;
+
+		/* Check that we're the only opener and authed. */
+		client.idx = 0;
+		ret = ioctl(fd, DRM_IOCTL_GET_CLIENT, &client);
+		assert (ret == 0);
+		if (!client.auth) {
+			close(fd);
+			continue;
+		}
+		client.idx = 1;
+		ret = ioctl(fd, DRM_IOCTL_GET_CLIENT, &client);
+		if (ret != -1 || errno != EINVAL) {
+			close(fd);
+			continue;
+		}
+		return fd;
+	}
+	fprintf(stderr, "Couldn't find an un-controlled DRM device\n");
+	abort();
+}
+#endif
diff --git a/drmtest.h b/drmtest.h
new file mode 100644
index 0000000..0977fe6
--- /dev/null
+++ b/drmtest.h
@@ -0,0 +1,24 @@
+/*
+ * drmtest.h
+ *
+ * A header file for drmtest.c
+ *
+ * (c) 2009 Thomas White <taw@bitwiz.org.uk>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+extern int drm_open_any(void);
+extern int drm_open_any_master(void);
diff --git a/gdrm-gem-mmap.c b/gdrm-gem-mmap.c
new file mode 100644
index 0000000..dc9aef0
--- /dev/null
+++ b/gdrm-gem-mmap.c
@@ -0,0 +1,150 @@
+/*
+ * gdrm-gem-mmap.c
+ *
+ * Test Glamo DRM GEM memory mapping
+ *
+ * (c) 2009 Thomas White <taw@bitwiz.org.uk>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ * Copyright © 2008 Intel Corporation
+ *
+ *
+ * Based on tests/gem_mmap.c from libdrm, to which the following notice applies:
+ *
+ * Copyright © 2008 Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+ * IN THE SOFTWARE.
+ *
+ * Authors:
+ *    Eric Anholt <eric@anholt.net>
+ *
+ */
+
+#define _FILE_OFFSET_BITS 64
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+#include <fcntl.h>
+#include <inttypes.h>
+#include <errno.h>
+#include <sys/stat.h>
+#include <sys/mman.h>
+
+#include "drm.h"
+#include "glamo_drm.h"
+#include "drmtest.h"
+
+#define OBJECT_SIZE 16384
+
+int main(int argc, char **argv)
+{
+	int fd;
+	struct drm_glamo_gem_create create;
+	struct drm_glamo_gem_mmap mmap_arg;
+	struct drm_gem_close unref;
+	uint8_t expected[OBJECT_SIZE];
+	uint8_t buf[OBJECT_SIZE];
+	uint8_t *addr;
+	uint16_t *addr16;
+	int ret;
+	int handle;
+	char tmp[4];
+	int i;
+
+	fd = drm_open_any();
+
+	memset(&mmap_arg, 0, sizeof(mmap_arg));
+	mmap_arg.handle = 0x10101010;
+	printf("Testing mmaping of bad object.\n");
+	ret = ioctl(fd, DRM_IOCTL_GLAMO_GEM_MMAP, &mmap_arg);
+	assert(ret == -1 && errno == EBADF);
+
+	memset(&create, 0, sizeof(create));
+	create.size = OBJECT_SIZE;
+	ret = ioctl(fd, DRM_IOCTL_GLAMO_GEM_CREATE, &create);
+	assert(ret == 0);
+	handle = create.handle;
+
+	printf("Testing mmaping of newly created object.\n");
+	mmap_arg.handle = handle;
+	mmap_arg.offset = 0;
+	ret = ioctl(fd, DRM_IOCTL_GLAMO_GEM_MMAP, &mmap_arg);
+	assert(ret == 0);
+
+	printf("The offset is %8llx\n", (long long int)mmap_arg.offset);
+	addr = mmap(0, OBJECT_SIZE, PROT_READ | PROT_WRITE,
+		    MAP_SHARED, fd, mmap_arg.offset);
+	printf("Virtual address is 0x%p\n", addr);
+
+	printf("Press enter to continue...\n");
+	fgets(tmp, 3, stdin);
+
+	printf("Testing contents of newly created object.\n");
+	addr16 = (uint16_t *)addr;
+	for ( i=0; i<OBJECT_SIZE/2; i++ ) {
+		addr16[i] = 0xABCD;
+	}
+	for ( i=0; i<OBJECT_SIZE/2; i++ ) {
+		if ( addr16[i] != 0xABCD ) {
+			printf("%i = %16x\n", i, addr16[i]);
+		}
+	}
+	for ( i=0; i<OBJECT_SIZE/2; i++ ) {
+		if ( addr16[i] != 0xABCD ) {
+			printf("%i = %16x\n", i, addr16[i]);
+		}
+	}
+
+	printf("Testing that mapping stays after close\n");
+	unref.handle = handle;
+	ret = ioctl(fd, DRM_IOCTL_GEM_CLOSE, &unref);
+	assert(ret == 0);
+	addr16 = (uint16_t *)addr;
+	for ( i=0; i<OBJECT_SIZE/2; i++ ) {
+		if ( addr16[i] != 0xABCD ) {
+			printf("%i = %16x\n", i, addr16[i]);
+		}
+	}
+	for ( i=0; i<OBJECT_SIZE/2; i++ ) {
+		if ( addr16[i] != 0xABCD ) {
+			printf("%i = %16x\n", i, addr16[i]);
+		}
+	}
+
+	printf("Testing unmapping\n");
+	munmap(addr, OBJECT_SIZE);
+
+	close(fd);
+
+	return 0;
+}
diff --git a/gdrm-kms-addfb.c b/gdrm-kms-addfb.c
new file mode 100644
index 0000000..6f9acc2
--- /dev/null
+++ b/gdrm-kms-addfb.c
@@ -0,0 +1,176 @@
+/*
+ * gdrm-gem-mmap_arg.c
+ *
+ * Test Glamo DRM KMS
+ *
+ * (c) 2009 Thomas White <taw@bitwiz.org.uk>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ * Copyright © 2008 Intel Corporation
+ *
+ *
+ * Based on tests/gem_mmap_arg.c from libdrm, to which the following notice applies:
+ *
+ * Copyright © 2008 Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+ * IN THE SOFTWARE.
+ *
+ * Authors:
+ *    Eric Anholt <eric@anholt.net>
+ *
+ */
+
+#define _FILE_OFFSET_BITS 64
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+#include <fcntl.h>
+#include <inttypes.h>
+#include <errno.h>
+#include <sys/stat.h>
+#include <sys/mman.h>
+#include <xf86drmMode.h>
+
+#include "drm.h"
+#include "glamo_drm.h"
+#include "drmtest.h"
+
+#define OBJECT_SIZE (640*480*2)
+
+
+static int addfb(int fd, int handle)
+{
+	int ret = 0;
+	drmModeResPtr r;
+	drmModeCrtcPtr crtc;
+	int crtc_id;
+	int conn_id;
+	int fb_id;
+
+	r = drmModeGetResources(fd);
+	printf("There are %i CRTCs, %i connectors, %i encoders"
+	       " and %i framebuffers\n",
+	       r->count_crtcs, r->count_connectors, r->count_encoders,
+	       r->count_fbs);
+
+	crtc_id = r->crtcs[0];
+	conn_id = r->connectors[0];
+	printf("The first CRTC ID is %i\n", crtc_id);
+	printf("The first connector ID is %i\n", conn_id);
+
+	drmModeAddFB(fd, 480, 640, 16, 16, 480*2, handle, &fb_id);
+	printf("My new FB handle is %i\n", fb_id);
+	if (!fb_id) {
+		printf("Could not add FB\n");
+		return 1;
+	}
+
+	printf("Getting CRTC info\n");
+	crtc = drmModeGetCrtc(fd, crtc_id);
+
+	printf("crtc=%p\n", crtc);
+	ret = drmModeSetCrtc(fd, crtc_id, fb_id, 0, 0, r->connectors, 1,
+	                     &(crtc->mode));
+	if ( ret ) {
+		printf("drmModeSetCrtc returned %i\n", ret);
+	}
+
+	return fb_id;
+}
+
+
+int main(int argc, char **argv)
+{
+	int fd;
+	struct drm_glamo_gem_create create;
+	struct drm_glamo_gem_mmap mmap_arg;
+	struct drm_gem_close unref;
+	uint8_t expected[OBJECT_SIZE];
+	uint8_t buf[OBJECT_SIZE];
+	uint8_t *addr;
+	uint16_t *addr16;
+	int ret;
+	int handle;
+	char tmp[4];
+	int i;
+	int fb_id;
+
+	fd = drm_open_any();
+
+	printf("Creating a GEM object\n");
+	memset(&create, 0, sizeof(create));
+	create.size = OBJECT_SIZE;
+	ret = ioctl(fd, DRM_IOCTL_GLAMO_GEM_CREATE, &create);
+	assert(ret == 0);
+	handle = create.handle;
+
+	printf("Getting an mmap() offset for the new object\n");
+	mmap_arg.handle = handle;
+	mmap_arg.offset = 0;
+	ret = ioctl(fd, DRM_IOCTL_GLAMO_GEM_MMAP, &mmap_arg);
+	assert(ret == 0);
+
+	printf("The offset is %8llx\n", (long long int)mmap_arg.offset);
+	addr = mmap(0, OBJECT_SIZE, PROT_READ | PROT_WRITE,
+		    MAP_SHARED, fd, mmap_arg.offset);
+	addr16 = (uint16_t *)addr;
+	printf("Virtual address is 0x%p\n", addr);
+
+	printf("Adding FB\n");
+	fb_id = addfb(fd, handle);
+
+	do {
+
+		uint16_t val;
+
+		printf("Press <enter> to continue, q<enter> to finish...");
+		fgets(tmp, 3, stdin);
+
+		val = random() & 0xff;
+		val += (random() & 0xff)<<8;
+		for ( i=0; i<OBJECT_SIZE/2; i++ ) {
+			addr16[i] = val;
+		}
+
+	} while ( tmp[0] != 'q' );
+
+	printf("Unmapping\n");
+	munmap(addr, OBJECT_SIZE);
+
+	printf("Removing FB\n");
+	drmModeRmFB(fd, fb_id);
+
+	close(fd);
+
+	return 0;
+}