162306a36Sopenharmony_ci/*
262306a36Sopenharmony_ci * SPDX-License-Identifier: MIT
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Copyright © 2018 Intel Corporation
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <linux/random.h>
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include "gem/i915_gem_internal.h"
1062306a36Sopenharmony_ci#include "gem/i915_gem_pm.h"
1162306a36Sopenharmony_ci#include "gem/selftests/igt_gem_utils.h"
1262306a36Sopenharmony_ci#include "gem/selftests/mock_context.h"
1362306a36Sopenharmony_ci#include "gt/intel_gt.h"
1462306a36Sopenharmony_ci#include "gt/intel_gt_pm.h"
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci#include "i915_selftest.h"
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci#include "igt_flush_test.h"
1962306a36Sopenharmony_ci#include "mock_drm.h"
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_cistatic int switch_to_context(struct i915_gem_context *ctx)
2262306a36Sopenharmony_ci{
2362306a36Sopenharmony_ci	struct i915_gem_engines_iter it;
2462306a36Sopenharmony_ci	struct intel_context *ce;
2562306a36Sopenharmony_ci	int err = 0;
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci	for_each_gem_engine(ce, i915_gem_context_lock_engines(ctx), it) {
2862306a36Sopenharmony_ci		struct i915_request *rq;
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci		rq = intel_context_create_request(ce);
3162306a36Sopenharmony_ci		if (IS_ERR(rq)) {
3262306a36Sopenharmony_ci			err = PTR_ERR(rq);
3362306a36Sopenharmony_ci			break;
3462306a36Sopenharmony_ci		}
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci		i915_request_add(rq);
3762306a36Sopenharmony_ci	}
3862306a36Sopenharmony_ci	i915_gem_context_unlock_engines(ctx);
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	return err;
4162306a36Sopenharmony_ci}
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_cistatic void trash_stolen(struct drm_i915_private *i915)
4462306a36Sopenharmony_ci{
4562306a36Sopenharmony_ci	struct i915_ggtt *ggtt = to_gt(i915)->ggtt;
4662306a36Sopenharmony_ci	const u64 slot = ggtt->error_capture.start;
4762306a36Sopenharmony_ci	const resource_size_t size = resource_size(&i915->dsm.stolen);
4862306a36Sopenharmony_ci	unsigned long page;
4962306a36Sopenharmony_ci	u32 prng = 0x12345678;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	/* XXX: fsck. needs some more thought... */
5262306a36Sopenharmony_ci	if (!i915_ggtt_has_aperture(ggtt))
5362306a36Sopenharmony_ci		return;
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	for (page = 0; page < size; page += PAGE_SIZE) {
5662306a36Sopenharmony_ci		const dma_addr_t dma = i915->dsm.stolen.start + page;
5762306a36Sopenharmony_ci		u32 __iomem *s;
5862306a36Sopenharmony_ci		int x;
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci		ggtt->vm.insert_page(&ggtt->vm, dma, slot,
6162306a36Sopenharmony_ci				     i915_gem_get_pat_index(i915,
6262306a36Sopenharmony_ci							    I915_CACHE_NONE),
6362306a36Sopenharmony_ci				     0);
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci		s = io_mapping_map_atomic_wc(&ggtt->iomap, slot);
6662306a36Sopenharmony_ci		for (x = 0; x < PAGE_SIZE / sizeof(u32); x++) {
6762306a36Sopenharmony_ci			prng = next_pseudo_random32(prng);
6862306a36Sopenharmony_ci			iowrite32(prng, &s[x]);
6962306a36Sopenharmony_ci		}
7062306a36Sopenharmony_ci		io_mapping_unmap_atomic(s);
7162306a36Sopenharmony_ci	}
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	ggtt->vm.clear_range(&ggtt->vm, slot, PAGE_SIZE);
7462306a36Sopenharmony_ci}
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_cistatic void simulate_hibernate(struct drm_i915_private *i915)
7762306a36Sopenharmony_ci{
7862306a36Sopenharmony_ci	intel_wakeref_t wakeref;
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	wakeref = intel_runtime_pm_get(&i915->runtime_pm);
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_ci	/*
8362306a36Sopenharmony_ci	 * As a final sting in the tail, invalidate stolen. Under a real S4,
8462306a36Sopenharmony_ci	 * stolen is lost and needs to be refilled on resume. However, under
8562306a36Sopenharmony_ci	 * CI we merely do S4-device testing (as full S4 is too unreliable
8662306a36Sopenharmony_ci	 * for automated testing across a cluster), so to simulate the effect
8762306a36Sopenharmony_ci	 * of stolen being trashed across S4, we trash it ourselves.
8862306a36Sopenharmony_ci	 */
8962306a36Sopenharmony_ci	trash_stolen(i915);
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_ci	intel_runtime_pm_put(&i915->runtime_pm, wakeref);
9262306a36Sopenharmony_ci}
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_cistatic int igt_pm_prepare(struct drm_i915_private *i915)
9562306a36Sopenharmony_ci{
9662306a36Sopenharmony_ci	i915_gem_suspend(i915);
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	return 0;
9962306a36Sopenharmony_ci}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_cistatic void igt_pm_suspend(struct drm_i915_private *i915)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	intel_wakeref_t wakeref;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	with_intel_runtime_pm(&i915->runtime_pm, wakeref) {
10662306a36Sopenharmony_ci		i915_ggtt_suspend(to_gt(i915)->ggtt);
10762306a36Sopenharmony_ci		i915_gem_suspend_late(i915);
10862306a36Sopenharmony_ci	}
10962306a36Sopenharmony_ci}
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_cistatic void igt_pm_hibernate(struct drm_i915_private *i915)
11262306a36Sopenharmony_ci{
11362306a36Sopenharmony_ci	intel_wakeref_t wakeref;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	with_intel_runtime_pm(&i915->runtime_pm, wakeref) {
11662306a36Sopenharmony_ci		i915_ggtt_suspend(to_gt(i915)->ggtt);
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci		i915_gem_freeze(i915);
11962306a36Sopenharmony_ci		i915_gem_freeze_late(i915);
12062306a36Sopenharmony_ci	}
12162306a36Sopenharmony_ci}
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_cistatic void igt_pm_resume(struct drm_i915_private *i915)
12462306a36Sopenharmony_ci{
12562306a36Sopenharmony_ci	intel_wakeref_t wakeref;
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ci	/*
12862306a36Sopenharmony_ci	 * Both suspend and hibernate follow the same wakeup path and assume
12962306a36Sopenharmony_ci	 * that runtime-pm just works.
13062306a36Sopenharmony_ci	 */
13162306a36Sopenharmony_ci	with_intel_runtime_pm(&i915->runtime_pm, wakeref) {
13262306a36Sopenharmony_ci		i915_ggtt_resume(to_gt(i915)->ggtt);
13362306a36Sopenharmony_ci		if (GRAPHICS_VER(i915) >= 8)
13462306a36Sopenharmony_ci			setup_private_pat(to_gt(i915));
13562306a36Sopenharmony_ci		i915_gem_resume(i915);
13662306a36Sopenharmony_ci	}
13762306a36Sopenharmony_ci}
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_cistatic int igt_gem_suspend(void *arg)
14062306a36Sopenharmony_ci{
14162306a36Sopenharmony_ci	struct drm_i915_private *i915 = arg;
14262306a36Sopenharmony_ci	struct i915_gem_context *ctx;
14362306a36Sopenharmony_ci	struct file *file;
14462306a36Sopenharmony_ci	int err;
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	file = mock_file(i915);
14762306a36Sopenharmony_ci	if (IS_ERR(file))
14862306a36Sopenharmony_ci		return PTR_ERR(file);
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	err = -ENOMEM;
15162306a36Sopenharmony_ci	ctx = live_context(i915, file);
15262306a36Sopenharmony_ci	if (!IS_ERR(ctx))
15362306a36Sopenharmony_ci		err = switch_to_context(ctx);
15462306a36Sopenharmony_ci	if (err)
15562306a36Sopenharmony_ci		goto out;
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	err = igt_pm_prepare(i915);
15862306a36Sopenharmony_ci	if (err)
15962306a36Sopenharmony_ci		goto out;
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	igt_pm_suspend(i915);
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci	/* Here be dragons! Note that with S3RST any S3 may become S4! */
16462306a36Sopenharmony_ci	simulate_hibernate(i915);
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci	igt_pm_resume(i915);
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_ci	err = switch_to_context(ctx);
16962306a36Sopenharmony_ciout:
17062306a36Sopenharmony_ci	fput(file);
17162306a36Sopenharmony_ci	return err;
17262306a36Sopenharmony_ci}
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_cistatic int igt_gem_hibernate(void *arg)
17562306a36Sopenharmony_ci{
17662306a36Sopenharmony_ci	struct drm_i915_private *i915 = arg;
17762306a36Sopenharmony_ci	struct i915_gem_context *ctx;
17862306a36Sopenharmony_ci	struct file *file;
17962306a36Sopenharmony_ci	int err;
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_ci	file = mock_file(i915);
18262306a36Sopenharmony_ci	if (IS_ERR(file))
18362306a36Sopenharmony_ci		return PTR_ERR(file);
18462306a36Sopenharmony_ci
18562306a36Sopenharmony_ci	err = -ENOMEM;
18662306a36Sopenharmony_ci	ctx = live_context(i915, file);
18762306a36Sopenharmony_ci	if (!IS_ERR(ctx))
18862306a36Sopenharmony_ci		err = switch_to_context(ctx);
18962306a36Sopenharmony_ci	if (err)
19062306a36Sopenharmony_ci		goto out;
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	err = igt_pm_prepare(i915);
19362306a36Sopenharmony_ci	if (err)
19462306a36Sopenharmony_ci		goto out;
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	igt_pm_hibernate(i915);
19762306a36Sopenharmony_ci
19862306a36Sopenharmony_ci	/* Here be dragons! */
19962306a36Sopenharmony_ci	simulate_hibernate(i915);
20062306a36Sopenharmony_ci
20162306a36Sopenharmony_ci	igt_pm_resume(i915);
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci	err = switch_to_context(ctx);
20462306a36Sopenharmony_ciout:
20562306a36Sopenharmony_ci	fput(file);
20662306a36Sopenharmony_ci	return err;
20762306a36Sopenharmony_ci}
20862306a36Sopenharmony_ci
20962306a36Sopenharmony_cistatic int igt_gem_ww_ctx(void *arg)
21062306a36Sopenharmony_ci{
21162306a36Sopenharmony_ci	struct drm_i915_private *i915 = arg;
21262306a36Sopenharmony_ci	struct drm_i915_gem_object *obj, *obj2;
21362306a36Sopenharmony_ci	struct i915_gem_ww_ctx ww;
21462306a36Sopenharmony_ci	int err = 0;
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_ci	obj = i915_gem_object_create_internal(i915, PAGE_SIZE);
21762306a36Sopenharmony_ci	if (IS_ERR(obj))
21862306a36Sopenharmony_ci		return PTR_ERR(obj);
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_ci	obj2 = i915_gem_object_create_internal(i915, PAGE_SIZE);
22162306a36Sopenharmony_ci	if (IS_ERR(obj2)) {
22262306a36Sopenharmony_ci		err = PTR_ERR(obj2);
22362306a36Sopenharmony_ci		goto put1;
22462306a36Sopenharmony_ci	}
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	i915_gem_ww_ctx_init(&ww, true);
22762306a36Sopenharmony_ciretry:
22862306a36Sopenharmony_ci	/* Lock the objects, twice for good measure (-EALREADY handling) */
22962306a36Sopenharmony_ci	err = i915_gem_object_lock(obj, &ww);
23062306a36Sopenharmony_ci	if (!err)
23162306a36Sopenharmony_ci		err = i915_gem_object_lock_interruptible(obj, &ww);
23262306a36Sopenharmony_ci	if (!err)
23362306a36Sopenharmony_ci		err = i915_gem_object_lock_interruptible(obj2, &ww);
23462306a36Sopenharmony_ci	if (!err)
23562306a36Sopenharmony_ci		err = i915_gem_object_lock(obj2, &ww);
23662306a36Sopenharmony_ci
23762306a36Sopenharmony_ci	if (err == -EDEADLK) {
23862306a36Sopenharmony_ci		err = i915_gem_ww_ctx_backoff(&ww);
23962306a36Sopenharmony_ci		if (!err)
24062306a36Sopenharmony_ci			goto retry;
24162306a36Sopenharmony_ci	}
24262306a36Sopenharmony_ci	i915_gem_ww_ctx_fini(&ww);
24362306a36Sopenharmony_ci	i915_gem_object_put(obj2);
24462306a36Sopenharmony_ciput1:
24562306a36Sopenharmony_ci	i915_gem_object_put(obj);
24662306a36Sopenharmony_ci	return err;
24762306a36Sopenharmony_ci}
24862306a36Sopenharmony_ci
24962306a36Sopenharmony_ciint i915_gem_live_selftests(struct drm_i915_private *i915)
25062306a36Sopenharmony_ci{
25162306a36Sopenharmony_ci	static const struct i915_subtest tests[] = {
25262306a36Sopenharmony_ci		SUBTEST(igt_gem_suspend),
25362306a36Sopenharmony_ci		SUBTEST(igt_gem_hibernate),
25462306a36Sopenharmony_ci		SUBTEST(igt_gem_ww_ctx),
25562306a36Sopenharmony_ci	};
25662306a36Sopenharmony_ci
25762306a36Sopenharmony_ci	if (intel_gt_is_wedged(to_gt(i915)))
25862306a36Sopenharmony_ci		return 0;
25962306a36Sopenharmony_ci
26062306a36Sopenharmony_ci	return i915_live_subtests(tests, i915);
26162306a36Sopenharmony_ci}
262