/*
 * SPDX-License-Identifier: MIT
 *
 * Copyright © 2016 Intel Corporation
 */

#include "i915_selftest.h"

#include "huge_gem_object.h"
#include "selftests/igt_flush_test.h"
#include "selftests/mock_gem_device.h"

static int igt_gem_object(void *arg)
{
 struct drm_i915_private *i915 = arg;
 struct drm_i915_gem_object *obj;
 int err;

 /* Basic test to ensure we can create an object */

 obj = i915_gem_object_create_shmem(i915, PAGE_SIZE);
 if (IS_ERR(obj)) {
  err = PTR_ERR(obj);
  pr_err("i915_gem_object_create failed, err=%d\n", err);
  goto out;
 }

 err = 0;
 i915_gem_object_put(obj);
out:
 return err;
}

static int igt_gem_huge(void *arg)
{
 const unsigned long nreal = 509; /* just to be awkward */
 struct drm_i915_private *i915 = arg;
 struct drm_i915_gem_object *obj;
 unsigned long n;
 int err;

 /* Basic sanitycheck of our huge fake object allocation */

 obj = huge_gem_object(i915,
         nreal * PAGE_SIZE,
         to_gt(i915)->ggtt->vm.total + PAGE_SIZE);
 if (IS_ERR(obj))
  return PTR_ERR(obj);

 err = i915_gem_object_pin_pages_unlocked(obj);
 if (err) {
  pr_err("Failed to allocate %lu pages (%lu total), err=%d\n",
         nreal, obj->base.size / PAGE_SIZE, err);
  goto out;
 }

 for (n = 0; n < obj->base.size / PAGE_SIZE; n++) {
  if (i915_gem_object_get_page(obj, n) !=
      i915_gem_object_get_page(obj, n % nreal)) {
   pr_err("Page lookup mismatch at index %lu [%lu]\n",
          n, n % nreal);
   err = -EINVAL;
   goto out_unpin;
  }
 }

out_unpin:
 i915_gem_object_unpin_pages(obj);
out:
 i915_gem_object_put(obj);
 return err;
}

int i915_gem_object_mock_selftests(void)
{
 static const struct i915_subtest tests[] = {
  SUBTEST(igt_gem_object),
 };
 struct drm_i915_private *i915;
 int err;

 i915 = mock_gem_device();
 if (!i915)
  return -ENOMEM;

 err = i915_subtests(tests, i915);

 mock_destroy_device(i915);
 return err;
}

int i915_gem_object_live_selftests(struct drm_i915_private *i915)
{
 static const struct i915_subtest tests[] = {
  SUBTEST(igt_gem_huge),
 };

 return i915_live_subtests(tests, i915);
}