

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <inttypes.h>
#include <errno.h>
#include <sys/stat.h>
#include <sys/time.h>
#include "drm.h"
#include "ioctl_wrappers.h"
#include "drmtest.h"
#include "intel_bufmgr.h"
#include "intel_batchbuffer.h"
#include "intel_io.h"
#include "intel_chipset.h"

static drm_intel_bufmgr *bufmgr;
struct intel_batchbuffer *batch;

/* Testcase: check whether the libdrm vma limiter works
 *
 * We've had reports of the X server exhausting the default rlimit of 64k vma's
 * in the kernel. libdrm has grown facilities to limit the vma caching since,
 * this checks whether they actually work.
 */

IGT_TEST_DESCRIPTION("Check whether the libdrm vma limiter works.");

/* we do both cpu and gtt maps, so only need half of 64k to exhaust */
#define BO_ARRAY_SIZE 35000
drm_intel_bo *bos[BO_ARRAY_SIZE];

igt_simple_main
{
	int fd;
	int i;
	char *ptr;

	igt_skip_on_simulation();

	fd = drm_open_any();

	bufmgr = drm_intel_bufmgr_gem_init(fd, 4096);
	drm_intel_bufmgr_gem_enable_reuse(bufmgr);
	batch = intel_batchbuffer_alloc(bufmgr, intel_get_drm_devid(fd));

	drm_intel_bufmgr_gem_set_vma_cache_size(bufmgr, 500);

	for (i = 0; i < BO_ARRAY_SIZE; i++) {
		bos[i] = drm_intel_bo_alloc(bufmgr, "mmap bo", 4096, 4096);
		igt_assert(bos[i]);

		drm_intel_bo_map(bos[i], 1);
		ptr = bos[i]->virtual;
		igt_assert(ptr);
		*ptr = 'c';
		drm_intel_bo_unmap(bos[i]);

		drm_intel_gem_bo_map_gtt(bos[i]);
		ptr = bos[i]->virtual;
		igt_assert(ptr);
		*ptr = 'c';
		drm_intel_gem_bo_unmap_gtt(bos[i]);
	}

	/* and recheck whether a second map of the same still works */
	for (i = 0; i < BO_ARRAY_SIZE; i++) {
		bos[i] = drm_intel_bo_alloc(bufmgr, "mmap bo", 4096, 4096);
		igt_assert(bos[i]);

		drm_intel_bo_map(bos[i], 1);
		ptr = bos[i]->virtual;
		igt_assert(*ptr = 'c');
		drm_intel_bo_unmap(bos[i]);

		drm_intel_gem_bo_map_gtt(bos[i]);
		ptr = bos[i]->virtual;
		igt_assert(*ptr = 'c');
		drm_intel_gem_bo_unmap_gtt(bos[i]);
	}

	intel_batchbuffer_free(batch);
	drm_intel_bufmgr_destroy(bufmgr);

	close(fd);
}
