1 /*
2  * SPDX-License-Identifier: MIT
3  *
4  * Copyright © 2018 Intel Corporation
5  */
6 
7 #include <linux/random.h>
8 
9 #include "gem/selftests/igt_gem_utils.h"
10 #include "gem/selftests/mock_context.h"
11 #include "gt/intel_gt.h"
12 
13 #include "i915_selftest.h"
14 
15 #include "igt_flush_test.h"
16 #include "mock_drm.h"
17 
18 static int switch_to_context(struct i915_gem_context *ctx)
19 {
20 	struct i915_gem_engines_iter it;
21 	struct intel_context *ce;
22 	int err = 0;
23 
24 	for_each_gem_engine(ce, i915_gem_context_lock_engines(ctx), it) {
25 		struct i915_request *rq;
26 
27 		rq = intel_context_create_request(ce);
28 		if (IS_ERR(rq)) {
29 			err = PTR_ERR(rq);
30 			break;
31 		}
32 
33 		i915_request_add(rq);
34 	}
35 	i915_gem_context_unlock_engines(ctx);
36 
37 	return err;
38 }
39 
40 static void trash_stolen(struct drm_i915_private *i915)
41 {
42 	struct i915_ggtt *ggtt = &i915->ggtt;
43 	const u64 slot = ggtt->error_capture.start;
44 	const resource_size_t size = resource_size(&i915->dsm);
45 	unsigned long page;
46 	u32 prng = 0x12345678;
47 
48 	/* XXX: fsck. needs some more thought... */
49 	if (!i915_ggtt_has_aperture(ggtt))
50 		return;
51 
52 	for (page = 0; page < size; page += PAGE_SIZE) {
53 		const dma_addr_t dma = i915->dsm.start + page;
54 		u32 __iomem *s;
55 		int x;
56 
57 		ggtt->vm.insert_page(&ggtt->vm, dma, slot, I915_CACHE_NONE, 0);
58 
59 		s = io_mapping_map_atomic_wc(&ggtt->iomap, slot);
60 		for (x = 0; x < PAGE_SIZE / sizeof(u32); x++) {
61 			prng = next_pseudo_random32(prng);
62 			iowrite32(prng, &s[x]);
63 		}
64 		io_mapping_unmap_atomic(s);
65 	}
66 
67 	ggtt->vm.clear_range(&ggtt->vm, slot, PAGE_SIZE);
68 }
69 
70 static void simulate_hibernate(struct drm_i915_private *i915)
71 {
72 	intel_wakeref_t wakeref;
73 
74 	wakeref = intel_runtime_pm_get(&i915->runtime_pm);
75 
76 	/*
77 	 * As a final sting in the tail, invalidate stolen. Under a real S4,
78 	 * stolen is lost and needs to be refilled on resume. However, under
79 	 * CI we merely do S4-device testing (as full S4 is too unreliable
80 	 * for automated testing across a cluster), so to simulate the effect
81 	 * of stolen being trashed across S4, we trash it ourselves.
82 	 */
83 	trash_stolen(i915);
84 
85 	intel_runtime_pm_put(&i915->runtime_pm, wakeref);
86 }
87 
88 static int pm_prepare(struct drm_i915_private *i915)
89 {
90 	i915_gem_suspend(i915);
91 
92 	return 0;
93 }
94 
95 static void pm_suspend(struct drm_i915_private *i915)
96 {
97 	intel_wakeref_t wakeref;
98 
99 	with_intel_runtime_pm(&i915->runtime_pm, wakeref) {
100 		i915_gem_suspend_gtt_mappings(i915);
101 		i915_gem_suspend_late(i915);
102 	}
103 }
104 
105 static void pm_hibernate(struct drm_i915_private *i915)
106 {
107 	intel_wakeref_t wakeref;
108 
109 	with_intel_runtime_pm(&i915->runtime_pm, wakeref) {
110 		i915_gem_suspend_gtt_mappings(i915);
111 
112 		i915_gem_freeze(i915);
113 		i915_gem_freeze_late(i915);
114 	}
115 }
116 
117 static void pm_resume(struct drm_i915_private *i915)
118 {
119 	intel_wakeref_t wakeref;
120 
121 	/*
122 	 * Both suspend and hibernate follow the same wakeup path and assume
123 	 * that runtime-pm just works.
124 	 */
125 	with_intel_runtime_pm(&i915->runtime_pm, wakeref) {
126 		intel_gt_sanitize(&i915->gt, false);
127 		i915_gem_sanitize(i915);
128 
129 		i915_gem_restore_gtt_mappings(i915);
130 		i915_gem_restore_fences(&i915->ggtt);
131 
132 		i915_gem_resume(i915);
133 	}
134 }
135 
136 static int igt_gem_suspend(void *arg)
137 {
138 	struct drm_i915_private *i915 = arg;
139 	struct i915_gem_context *ctx;
140 	struct drm_file *file;
141 	int err;
142 
143 	file = mock_file(i915);
144 	if (IS_ERR(file))
145 		return PTR_ERR(file);
146 
147 	err = -ENOMEM;
148 	ctx = live_context(i915, file);
149 	if (!IS_ERR(ctx))
150 		err = switch_to_context(ctx);
151 	if (err)
152 		goto out;
153 
154 	err = pm_prepare(i915);
155 	if (err)
156 		goto out;
157 
158 	pm_suspend(i915);
159 
160 	/* Here be dragons! Note that with S3RST any S3 may become S4! */
161 	simulate_hibernate(i915);
162 
163 	pm_resume(i915);
164 
165 	err = switch_to_context(ctx);
166 out:
167 	mock_file_free(i915, file);
168 	return err;
169 }
170 
171 static int igt_gem_hibernate(void *arg)
172 {
173 	struct drm_i915_private *i915 = arg;
174 	struct i915_gem_context *ctx;
175 	struct drm_file *file;
176 	int err;
177 
178 	file = mock_file(i915);
179 	if (IS_ERR(file))
180 		return PTR_ERR(file);
181 
182 	err = -ENOMEM;
183 	ctx = live_context(i915, file);
184 	if (!IS_ERR(ctx))
185 		err = switch_to_context(ctx);
186 	if (err)
187 		goto out;
188 
189 	err = pm_prepare(i915);
190 	if (err)
191 		goto out;
192 
193 	pm_hibernate(i915);
194 
195 	/* Here be dragons! */
196 	simulate_hibernate(i915);
197 
198 	pm_resume(i915);
199 
200 	err = switch_to_context(ctx);
201 out:
202 	mock_file_free(i915, file);
203 	return err;
204 }
205 
206 int i915_gem_live_selftests(struct drm_i915_private *i915)
207 {
208 	static const struct i915_subtest tests[] = {
209 		SUBTEST(igt_gem_suspend),
210 		SUBTEST(igt_gem_hibernate),
211 	};
212 
213 	if (intel_gt_is_wedged(&i915->gt))
214 		return 0;
215 
216 	return i915_live_subtests(tests, i915);
217 }
218