/*
 * 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);
}