1 /*
2  * Copyright © 2016 Intel Corporation
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice (including the next
12  * paragraph) shall be included in all copies or substantial portions of the
13  * Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21  * IN THE SOFTWARE.
22  *
23  */
24 
25 #include <linux/prime_numbers.h>
26 
27 #include "gem/i915_gem_context.h"
28 #include "gem/selftests/mock_context.h"
29 
30 #include "i915_scatterlist.h"
31 #include "i915_selftest.h"
32 
33 #include "mock_gem_device.h"
34 #include "mock_gtt.h"
35 
36 static bool assert_vma(struct i915_vma *vma,
37 		       struct drm_i915_gem_object *obj,
38 		       struct i915_gem_context *ctx)
39 {
40 	bool ok = true;
41 
42 	if (vma->vm != ctx->vm) {
43 		pr_err("VMA created with wrong VM\n");
44 		ok = false;
45 	}
46 
47 	if (vma->size != obj->base.size) {
48 		pr_err("VMA created with wrong size, found %llu, expected %zu\n",
49 		       vma->size, obj->base.size);
50 		ok = false;
51 	}
52 
53 	if (vma->ggtt_view.type != I915_GGTT_VIEW_NORMAL) {
54 		pr_err("VMA created with wrong type [%d]\n",
55 		       vma->ggtt_view.type);
56 		ok = false;
57 	}
58 
59 	return ok;
60 }
61 
62 static struct i915_vma *
63 checked_vma_instance(struct drm_i915_gem_object *obj,
64 		     struct i915_address_space *vm,
65 		     const struct i915_ggtt_view *view)
66 {
67 	struct i915_vma *vma;
68 	bool ok = true;
69 
70 	vma = i915_vma_instance(obj, vm, view);
71 	if (IS_ERR(vma))
72 		return vma;
73 
74 	/* Manual checks, will be reinforced by i915_vma_compare! */
75 	if (vma->vm != vm) {
76 		pr_err("VMA's vm [%p] does not match request [%p]\n",
77 		       vma->vm, vm);
78 		ok = false;
79 	}
80 
81 	if (i915_is_ggtt(vm) != i915_vma_is_ggtt(vma)) {
82 		pr_err("VMA ggtt status [%d] does not match parent [%d]\n",
83 		       i915_vma_is_ggtt(vma), i915_is_ggtt(vm));
84 		ok = false;
85 	}
86 
87 	if (i915_vma_compare(vma, vm, view)) {
88 		pr_err("i915_vma_compare failed with create parameters!\n");
89 		return ERR_PTR(-EINVAL);
90 	}
91 
92 	if (i915_vma_compare(vma, vma->vm,
93 			     i915_vma_is_ggtt(vma) ? &vma->ggtt_view : NULL)) {
94 		pr_err("i915_vma_compare failed with itself\n");
95 		return ERR_PTR(-EINVAL);
96 	}
97 
98 	if (!ok) {
99 		pr_err("i915_vma_compare failed to detect the difference!\n");
100 		return ERR_PTR(-EINVAL);
101 	}
102 
103 	return vma;
104 }
105 
106 static int create_vmas(struct drm_i915_private *i915,
107 		       struct list_head *objects,
108 		       struct list_head *contexts)
109 {
110 	struct drm_i915_gem_object *obj;
111 	struct i915_gem_context *ctx;
112 	int pinned;
113 
114 	list_for_each_entry(obj, objects, st_link) {
115 		for (pinned = 0; pinned <= 1; pinned++) {
116 			list_for_each_entry(ctx, contexts, link) {
117 				struct i915_address_space *vm;
118 				struct i915_vma *vma;
119 				int err;
120 
121 				vm = i915_gem_context_get_eb_vm(ctx);
122 				vma = checked_vma_instance(obj, vm, NULL);
123 				i915_vm_put(vm);
124 				if (IS_ERR(vma))
125 					return PTR_ERR(vma);
126 
127 				if (!assert_vma(vma, obj, ctx)) {
128 					pr_err("VMA lookup/create failed\n");
129 					return -EINVAL;
130 				}
131 
132 				if (!pinned) {
133 					err = i915_vma_pin(vma, 0, 0, PIN_USER);
134 					if (err) {
135 						pr_err("Failed to pin VMA\n");
136 						return err;
137 					}
138 				} else {
139 					i915_vma_unpin(vma);
140 				}
141 			}
142 		}
143 	}
144 
145 	return 0;
146 }
147 
148 static int igt_vma_create(void *arg)
149 {
150 	struct i915_ggtt *ggtt = arg;
151 	struct drm_i915_private *i915 = ggtt->vm.i915;
152 	struct drm_i915_gem_object *obj, *on;
153 	struct i915_gem_context *ctx, *cn;
154 	unsigned long num_obj, num_ctx;
155 	unsigned long no, nc;
156 	IGT_TIMEOUT(end_time);
157 	LIST_HEAD(contexts);
158 	LIST_HEAD(objects);
159 	int err = -ENOMEM;
160 
161 	/* Exercise creating many vma amonst many objections, checking the
162 	 * vma creation and lookup routines.
163 	 */
164 
165 	no = 0;
166 	for_each_prime_number(num_obj, ULONG_MAX - 1) {
167 		for (; no < num_obj; no++) {
168 			obj = i915_gem_object_create_internal(i915, PAGE_SIZE);
169 			if (IS_ERR(obj))
170 				goto out;
171 
172 			list_add(&obj->st_link, &objects);
173 		}
174 
175 		nc = 0;
176 		for_each_prime_number(num_ctx, 2 * BITS_PER_LONG) {
177 			for (; nc < num_ctx; nc++) {
178 				ctx = mock_context(i915, "mock");
179 				if (!ctx)
180 					goto out;
181 
182 				list_move(&ctx->link, &contexts);
183 			}
184 
185 			err = create_vmas(i915, &objects, &contexts);
186 			if (err)
187 				goto out;
188 
189 			if (igt_timeout(end_time,
190 					"%s timed out: after %lu objects in %lu contexts\n",
191 					__func__, no, nc))
192 				goto end;
193 		}
194 
195 		list_for_each_entry_safe(ctx, cn, &contexts, link) {
196 			list_del_init(&ctx->link);
197 			mock_context_close(ctx);
198 		}
199 
200 		cond_resched();
201 	}
202 
203 end:
204 	/* Final pass to lookup all created contexts */
205 	err = create_vmas(i915, &objects, &contexts);
206 out:
207 	list_for_each_entry_safe(ctx, cn, &contexts, link) {
208 		list_del_init(&ctx->link);
209 		mock_context_close(ctx);
210 	}
211 
212 	list_for_each_entry_safe(obj, on, &objects, st_link)
213 		i915_gem_object_put(obj);
214 	return err;
215 }
216 
217 struct pin_mode {
218 	u64 size;
219 	u64 flags;
220 	bool (*assert)(const struct i915_vma *,
221 		       const struct pin_mode *mode,
222 		       int result);
223 	const char *string;
224 };
225 
226 static bool assert_pin_valid(const struct i915_vma *vma,
227 			     const struct pin_mode *mode,
228 			     int result)
229 {
230 	if (result)
231 		return false;
232 
233 	if (i915_vma_misplaced(vma, mode->size, 0, mode->flags))
234 		return false;
235 
236 	return true;
237 }
238 
239 __maybe_unused
240 static bool assert_pin_enospc(const struct i915_vma *vma,
241 			      const struct pin_mode *mode,
242 			      int result)
243 {
244 	return result == -ENOSPC;
245 }
246 
247 __maybe_unused
248 static bool assert_pin_einval(const struct i915_vma *vma,
249 			      const struct pin_mode *mode,
250 			      int result)
251 {
252 	return result == -EINVAL;
253 }
254 
255 static int igt_vma_pin1(void *arg)
256 {
257 	struct i915_ggtt *ggtt = arg;
258 	const struct pin_mode modes[] = {
259 #define VALID(sz, fl) { .size = (sz), .flags = (fl), .assert = assert_pin_valid, .string = #sz ", " #fl ", (valid) " }
260 #define __INVALID(sz, fl, check, eval) { .size = (sz), .flags = (fl), .assert = (check), .string = #sz ", " #fl ", (invalid " #eval ")" }
261 #define INVALID(sz, fl) __INVALID(sz, fl, assert_pin_einval, EINVAL)
262 #define NOSPACE(sz, fl) __INVALID(sz, fl, assert_pin_enospc, ENOSPC)
263 		VALID(0, PIN_GLOBAL),
264 		VALID(0, PIN_GLOBAL | PIN_MAPPABLE),
265 
266 		VALID(0, PIN_GLOBAL | PIN_OFFSET_BIAS | 4096),
267 		VALID(0, PIN_GLOBAL | PIN_OFFSET_BIAS | 8192),
268 		VALID(0, PIN_GLOBAL | PIN_OFFSET_BIAS | (ggtt->mappable_end - 4096)),
269 		VALID(0, PIN_GLOBAL | PIN_MAPPABLE | PIN_OFFSET_BIAS | (ggtt->mappable_end - 4096)),
270 		VALID(0, PIN_GLOBAL | PIN_OFFSET_BIAS | (ggtt->vm.total - 4096)),
271 
272 		VALID(0, PIN_GLOBAL | PIN_MAPPABLE | PIN_OFFSET_FIXED | (ggtt->mappable_end - 4096)),
273 		INVALID(0, PIN_GLOBAL | PIN_MAPPABLE | PIN_OFFSET_FIXED | ggtt->mappable_end),
274 		VALID(0, PIN_GLOBAL | PIN_OFFSET_FIXED | (ggtt->vm.total - 4096)),
275 		INVALID(0, PIN_GLOBAL | PIN_OFFSET_FIXED | ggtt->vm.total),
276 		INVALID(0, PIN_GLOBAL | PIN_OFFSET_FIXED | round_down(U64_MAX, PAGE_SIZE)),
277 
278 		VALID(4096, PIN_GLOBAL),
279 		VALID(8192, PIN_GLOBAL),
280 		VALID(ggtt->mappable_end - 4096, PIN_GLOBAL | PIN_MAPPABLE),
281 		VALID(ggtt->mappable_end, PIN_GLOBAL | PIN_MAPPABLE),
282 		NOSPACE(ggtt->mappable_end + 4096, PIN_GLOBAL | PIN_MAPPABLE),
283 		VALID(ggtt->vm.total - 4096, PIN_GLOBAL),
284 		VALID(ggtt->vm.total, PIN_GLOBAL),
285 		NOSPACE(ggtt->vm.total + 4096, PIN_GLOBAL),
286 		NOSPACE(round_down(U64_MAX, PAGE_SIZE), PIN_GLOBAL),
287 		INVALID(8192, PIN_GLOBAL | PIN_MAPPABLE | PIN_OFFSET_FIXED | (ggtt->mappable_end - 4096)),
288 		INVALID(8192, PIN_GLOBAL | PIN_OFFSET_FIXED | (ggtt->vm.total - 4096)),
289 		INVALID(8192, PIN_GLOBAL | PIN_OFFSET_FIXED | (round_down(U64_MAX, PAGE_SIZE) - 4096)),
290 
291 		VALID(8192, PIN_GLOBAL | PIN_OFFSET_BIAS | (ggtt->mappable_end - 4096)),
292 
293 #if !IS_ENABLED(CONFIG_DRM_I915_DEBUG_GEM)
294 		/* Misusing BIAS is a programming error (it is not controllable
295 		 * from userspace) so when debugging is enabled, it explodes.
296 		 * However, the tests are still quite interesting for checking
297 		 * variable start, end and size.
298 		 */
299 		NOSPACE(0, PIN_GLOBAL | PIN_MAPPABLE | PIN_OFFSET_BIAS | ggtt->mappable_end),
300 		NOSPACE(0, PIN_GLOBAL | PIN_OFFSET_BIAS | ggtt->vm.total),
301 		NOSPACE(8192, PIN_GLOBAL | PIN_MAPPABLE | PIN_OFFSET_BIAS | (ggtt->mappable_end - 4096)),
302 		NOSPACE(8192, PIN_GLOBAL | PIN_OFFSET_BIAS | (ggtt->vm.total - 4096)),
303 #endif
304 		{ },
305 #undef NOSPACE
306 #undef INVALID
307 #undef __INVALID
308 #undef VALID
309 	}, *m;
310 	struct drm_i915_gem_object *obj;
311 	struct i915_vma *vma;
312 	int err = -EINVAL;
313 
314 	/* Exercise all the weird and wonderful i915_vma_pin requests,
315 	 * focusing on error handling of boundary conditions.
316 	 */
317 
318 	GEM_BUG_ON(!drm_mm_clean(&ggtt->vm.mm));
319 
320 	obj = i915_gem_object_create_internal(ggtt->vm.i915, PAGE_SIZE);
321 	if (IS_ERR(obj))
322 		return PTR_ERR(obj);
323 
324 	vma = checked_vma_instance(obj, &ggtt->vm, NULL);
325 	if (IS_ERR(vma))
326 		goto out;
327 
328 	for (m = modes; m->assert; m++) {
329 		err = i915_vma_pin(vma, m->size, 0, m->flags);
330 		if (!m->assert(vma, m, err)) {
331 			pr_err("%s to pin single page into GGTT with mode[%d:%s]: size=%llx flags=%llx, err=%d\n",
332 			       m->assert == assert_pin_valid ? "Failed" : "Unexpectedly succeeded",
333 			       (int)(m - modes), m->string, m->size, m->flags,
334 			       err);
335 			if (!err)
336 				i915_vma_unpin(vma);
337 			err = -EINVAL;
338 			goto out;
339 		}
340 
341 		if (!err) {
342 			i915_vma_unpin(vma);
343 			err = i915_vma_unbind(vma);
344 			if (err) {
345 				pr_err("Failed to unbind single page from GGTT, err=%d\n", err);
346 				goto out;
347 			}
348 		}
349 
350 		cond_resched();
351 	}
352 
353 	err = 0;
354 out:
355 	i915_gem_object_put(obj);
356 	return err;
357 }
358 
359 static unsigned long rotated_index(const struct intel_rotation_info *r,
360 				   unsigned int n,
361 				   unsigned int x,
362 				   unsigned int y)
363 {
364 	return (r->plane[n].src_stride * (r->plane[n].height - y - 1) +
365 		r->plane[n].offset + x);
366 }
367 
368 static struct scatterlist *
369 assert_rotated(struct drm_i915_gem_object *obj,
370 	       const struct intel_rotation_info *r, unsigned int n,
371 	       struct scatterlist *sg)
372 {
373 	unsigned int x, y;
374 
375 	for (x = 0; x < r->plane[n].width; x++) {
376 		unsigned int left;
377 
378 		for (y = 0; y < r->plane[n].height; y++) {
379 			unsigned long src_idx;
380 			dma_addr_t src;
381 
382 			if (!sg) {
383 				pr_err("Invalid sg table: too short at plane %d, (%d, %d)!\n",
384 				       n, x, y);
385 				return ERR_PTR(-EINVAL);
386 			}
387 
388 			src_idx = rotated_index(r, n, x, y);
389 			src = i915_gem_object_get_dma_address(obj, src_idx);
390 
391 			if (sg_dma_len(sg) != PAGE_SIZE) {
392 				pr_err("Invalid sg.length, found %d, expected %lu for rotated page (%d, %d) [src index %lu]\n",
393 				       sg_dma_len(sg), PAGE_SIZE,
394 				       x, y, src_idx);
395 				return ERR_PTR(-EINVAL);
396 			}
397 
398 			if (sg_dma_address(sg) != src) {
399 				pr_err("Invalid address for rotated page (%d, %d) [src index %lu]\n",
400 				       x, y, src_idx);
401 				return ERR_PTR(-EINVAL);
402 			}
403 
404 			sg = sg_next(sg);
405 		}
406 
407 		left = (r->plane[n].dst_stride - y) * PAGE_SIZE;
408 
409 		if (!left)
410 			continue;
411 
412 		if (!sg) {
413 			pr_err("Invalid sg table: too short at plane %d, (%d, %d)!\n",
414 			       n, x, y);
415 			return ERR_PTR(-EINVAL);
416 		}
417 
418 		if (sg_dma_len(sg) != left) {
419 			pr_err("Invalid sg.length, found %d, expected %u for rotated page (%d, %d)\n",
420 			       sg_dma_len(sg), left, x, y);
421 			return ERR_PTR(-EINVAL);
422 		}
423 
424 		if (sg_dma_address(sg) != 0) {
425 			pr_err("Invalid address, found %pad, expected 0 for remapped page (%d, %d)\n",
426 			       &sg_dma_address(sg), x, y);
427 			return ERR_PTR(-EINVAL);
428 		}
429 
430 		sg = sg_next(sg);
431 	}
432 
433 	return sg;
434 }
435 
436 static unsigned long remapped_index(const struct intel_remapped_info *r,
437 				    unsigned int n,
438 				    unsigned int x,
439 				    unsigned int y)
440 {
441 	return (r->plane[n].src_stride * y +
442 		r->plane[n].offset + x);
443 }
444 
445 static struct scatterlist *
446 assert_remapped(struct drm_i915_gem_object *obj,
447 		const struct intel_remapped_info *r, unsigned int n,
448 		struct scatterlist *sg)
449 {
450 	unsigned int x, y;
451 	unsigned int left = 0;
452 	unsigned int offset;
453 
454 	for (y = 0; y < r->plane[n].height; y++) {
455 		for (x = 0; x < r->plane[n].width; x++) {
456 			unsigned long src_idx;
457 			dma_addr_t src;
458 
459 			if (!sg) {
460 				pr_err("Invalid sg table: too short at plane %d, (%d, %d)!\n",
461 				       n, x, y);
462 				return ERR_PTR(-EINVAL);
463 			}
464 			if (!left) {
465 				offset = 0;
466 				left = sg_dma_len(sg);
467 			}
468 
469 			src_idx = remapped_index(r, n, x, y);
470 			src = i915_gem_object_get_dma_address(obj, src_idx);
471 
472 			if (left < PAGE_SIZE || left & (PAGE_SIZE-1)) {
473 				pr_err("Invalid sg.length, found %d, expected %lu for remapped page (%d, %d) [src index %lu]\n",
474 				       sg_dma_len(sg), PAGE_SIZE,
475 				       x, y, src_idx);
476 				return ERR_PTR(-EINVAL);
477 			}
478 
479 			if (sg_dma_address(sg) + offset != src) {
480 				pr_err("Invalid address for remapped page (%d, %d) [src index %lu]\n",
481 				       x, y, src_idx);
482 				return ERR_PTR(-EINVAL);
483 			}
484 
485 			left -= PAGE_SIZE;
486 			offset += PAGE_SIZE;
487 
488 
489 			if (!left)
490 				sg = sg_next(sg);
491 		}
492 
493 		if (left) {
494 			pr_err("Unexpected sg tail with %d size for remapped page (%d, %d)\n",
495 			       left,
496 			       x, y);
497 			return ERR_PTR(-EINVAL);
498 		}
499 
500 		left = (r->plane[n].dst_stride - r->plane[n].width) * PAGE_SIZE;
501 
502 		if (!left)
503 			continue;
504 
505 		if (!sg) {
506 			pr_err("Invalid sg table: too short at plane %d, (%d, %d)!\n",
507 			       n, x, y);
508 			return ERR_PTR(-EINVAL);
509 		}
510 
511 		if (sg_dma_len(sg) != left) {
512 			pr_err("Invalid sg.length, found %u, expected %u for remapped page (%d, %d)\n",
513 			       sg_dma_len(sg), left,
514 			       x, y);
515 			return ERR_PTR(-EINVAL);
516 		}
517 
518 		if (sg_dma_address(sg) != 0) {
519 			pr_err("Invalid address, found %pad, expected 0 for remapped page (%d, %d)\n",
520 			       &sg_dma_address(sg),
521 			       x, y);
522 			return ERR_PTR(-EINVAL);
523 		}
524 
525 		sg = sg_next(sg);
526 		left = 0;
527 	}
528 
529 	return sg;
530 }
531 
532 static unsigned int remapped_size(enum i915_ggtt_view_type view_type,
533 				  const struct intel_remapped_plane_info *a,
534 				  const struct intel_remapped_plane_info *b)
535 {
536 
537 	if (view_type == I915_GGTT_VIEW_ROTATED)
538 		return a->dst_stride * a->width + b->dst_stride * b->width;
539 	else
540 		return a->dst_stride * a->height + b->dst_stride * b->height;
541 }
542 
543 static int igt_vma_rotate_remap(void *arg)
544 {
545 	struct i915_ggtt *ggtt = arg;
546 	struct i915_address_space *vm = &ggtt->vm;
547 	struct drm_i915_gem_object *obj;
548 	const struct intel_remapped_plane_info planes[] = {
549 		{ .width = 1, .height = 1, .src_stride = 1 },
550 		{ .width = 2, .height = 2, .src_stride = 2 },
551 		{ .width = 4, .height = 4, .src_stride = 4 },
552 		{ .width = 8, .height = 8, .src_stride = 8 },
553 
554 		{ .width = 3, .height = 5, .src_stride = 3 },
555 		{ .width = 3, .height = 5, .src_stride = 4 },
556 		{ .width = 3, .height = 5, .src_stride = 5 },
557 
558 		{ .width = 5, .height = 3, .src_stride = 5 },
559 		{ .width = 5, .height = 3, .src_stride = 7 },
560 		{ .width = 5, .height = 3, .src_stride = 9 },
561 
562 		{ .width = 4, .height = 6, .src_stride = 6 },
563 		{ .width = 6, .height = 4, .src_stride = 6 },
564 
565 		{ .width = 2, .height = 2, .src_stride = 2, .dst_stride = 2 },
566 		{ .width = 3, .height = 3, .src_stride = 3, .dst_stride = 4 },
567 		{ .width = 5, .height = 6, .src_stride = 7, .dst_stride = 8 },
568 
569 		{ }
570 	}, *a, *b;
571 	enum i915_ggtt_view_type types[] = {
572 		I915_GGTT_VIEW_ROTATED,
573 		I915_GGTT_VIEW_REMAPPED,
574 		0,
575 	}, *t;
576 	const unsigned int max_pages = 64;
577 	int err = -ENOMEM;
578 
579 	/* Create VMA for many different combinations of planes and check
580 	 * that the page layout within the rotated VMA match our expectations.
581 	 */
582 
583 	obj = i915_gem_object_create_internal(vm->i915, max_pages * PAGE_SIZE);
584 	if (IS_ERR(obj))
585 		goto out;
586 
587 	for (t = types; *t; t++) {
588 	for (a = planes; a->width; a++) {
589 		for (b = planes + ARRAY_SIZE(planes); b-- != planes; ) {
590 			struct i915_ggtt_view view = {
591 				.type = *t,
592 				.remapped.plane[0] = *a,
593 				.remapped.plane[1] = *b,
594 			};
595 			struct intel_remapped_plane_info *plane_info = view.remapped.plane;
596 			unsigned int n, max_offset;
597 
598 			max_offset = max(plane_info[0].src_stride * plane_info[0].height,
599 					 plane_info[1].src_stride * plane_info[1].height);
600 			GEM_BUG_ON(max_offset > max_pages);
601 			max_offset = max_pages - max_offset;
602 
603 			if (!plane_info[0].dst_stride)
604 				plane_info[0].dst_stride = view.type == I915_GGTT_VIEW_ROTATED ?
605 									plane_info[0].height :
606 									plane_info[0].width;
607 			if (!plane_info[1].dst_stride)
608 				plane_info[1].dst_stride = view.type == I915_GGTT_VIEW_ROTATED ?
609 									plane_info[1].height :
610 									plane_info[1].width;
611 
612 			for_each_prime_number_from(plane_info[0].offset, 0, max_offset) {
613 				for_each_prime_number_from(plane_info[1].offset, 0, max_offset) {
614 					struct scatterlist *sg;
615 					struct i915_vma *vma;
616 					unsigned int expected_pages;
617 
618 					vma = checked_vma_instance(obj, vm, &view);
619 					if (IS_ERR(vma)) {
620 						err = PTR_ERR(vma);
621 						goto out_object;
622 					}
623 
624 					err = i915_vma_pin(vma, 0, 0, PIN_GLOBAL);
625 					if (err) {
626 						pr_err("Failed to pin VMA, err=%d\n", err);
627 						goto out_object;
628 					}
629 
630 					expected_pages = remapped_size(view.type, &plane_info[0], &plane_info[1]);
631 
632 					if (view.type == I915_GGTT_VIEW_ROTATED &&
633 					    vma->size != expected_pages * PAGE_SIZE) {
634 						pr_err("VMA is wrong size, expected %lu, found %llu\n",
635 						       PAGE_SIZE * expected_pages, vma->size);
636 						err = -EINVAL;
637 						goto out_object;
638 					}
639 
640 					if (view.type == I915_GGTT_VIEW_REMAPPED &&
641 					    vma->size > expected_pages * PAGE_SIZE) {
642 						pr_err("VMA is wrong size, expected %lu, found %llu\n",
643 						       PAGE_SIZE * expected_pages, vma->size);
644 						err = -EINVAL;
645 						goto out_object;
646 					}
647 
648 					if (vma->pages->nents > expected_pages) {
649 						pr_err("sg table is wrong sizeo, expected %u, found %u nents\n",
650 						       expected_pages, vma->pages->nents);
651 						err = -EINVAL;
652 						goto out_object;
653 					}
654 
655 					if (vma->node.size < vma->size) {
656 						pr_err("VMA binding too small, expected %llu, found %llu\n",
657 						       vma->size, vma->node.size);
658 						err = -EINVAL;
659 						goto out_object;
660 					}
661 
662 					if (vma->pages == obj->mm.pages) {
663 						pr_err("VMA using unrotated object pages!\n");
664 						err = -EINVAL;
665 						goto out_object;
666 					}
667 
668 					sg = vma->pages->sgl;
669 					for (n = 0; n < ARRAY_SIZE(view.rotated.plane); n++) {
670 						if (view.type == I915_GGTT_VIEW_ROTATED)
671 							sg = assert_rotated(obj, &view.rotated, n, sg);
672 						else
673 							sg = assert_remapped(obj, &view.remapped, n, sg);
674 						if (IS_ERR(sg)) {
675 							pr_err("Inconsistent %s VMA pages for plane %d: [(%d, %d, %d, %d, %d), (%d, %d, %d, %d, %d)]\n",
676 							       view.type == I915_GGTT_VIEW_ROTATED ?
677 							       "rotated" : "remapped", n,
678 							       plane_info[0].width,
679 							       plane_info[0].height,
680 							       plane_info[0].src_stride,
681 							       plane_info[0].dst_stride,
682 							       plane_info[0].offset,
683 							       plane_info[1].width,
684 							       plane_info[1].height,
685 							       plane_info[1].src_stride,
686 							       plane_info[1].dst_stride,
687 							       plane_info[1].offset);
688 							err = -EINVAL;
689 							goto out_object;
690 						}
691 					}
692 
693 					i915_vma_unpin(vma);
694 					err = i915_vma_unbind(vma);
695 					if (err) {
696 						pr_err("Unbinding returned %i\n", err);
697 						goto out_object;
698 					}
699 					cond_resched();
700 				}
701 			}
702 		}
703 	}
704 	}
705 
706 out_object:
707 	i915_gem_object_put(obj);
708 out:
709 	return err;
710 }
711 
712 static bool assert_partial(struct drm_i915_gem_object *obj,
713 			   struct i915_vma *vma,
714 			   unsigned long offset,
715 			   unsigned long size)
716 {
717 	struct sgt_iter sgt;
718 	dma_addr_t dma;
719 
720 	for_each_sgt_daddr(dma, sgt, vma->pages) {
721 		dma_addr_t src;
722 
723 		if (!size) {
724 			pr_err("Partial scattergather list too long\n");
725 			return false;
726 		}
727 
728 		src = i915_gem_object_get_dma_address(obj, offset);
729 		if (src != dma) {
730 			pr_err("DMA mismatch for partial page offset %lu\n",
731 			       offset);
732 			return false;
733 		}
734 
735 		offset++;
736 		size--;
737 	}
738 
739 	return true;
740 }
741 
742 static bool assert_pin(struct i915_vma *vma,
743 		       struct i915_ggtt_view *view,
744 		       u64 size,
745 		       const char *name)
746 {
747 	bool ok = true;
748 
749 	if (vma->size != size) {
750 		pr_err("(%s) VMA is wrong size, expected %llu, found %llu\n",
751 		       name, size, vma->size);
752 		ok = false;
753 	}
754 
755 	if (vma->node.size < vma->size) {
756 		pr_err("(%s) VMA binding too small, expected %llu, found %llu\n",
757 		       name, vma->size, vma->node.size);
758 		ok = false;
759 	}
760 
761 	if (view && view->type != I915_GGTT_VIEW_NORMAL) {
762 		if (memcmp(&vma->ggtt_view, view, sizeof(*view))) {
763 			pr_err("(%s) VMA mismatch upon creation!\n",
764 			       name);
765 			ok = false;
766 		}
767 
768 		if (vma->pages == vma->obj->mm.pages) {
769 			pr_err("(%s) VMA using original object pages!\n",
770 			       name);
771 			ok = false;
772 		}
773 	} else {
774 		if (vma->ggtt_view.type != I915_GGTT_VIEW_NORMAL) {
775 			pr_err("Not the normal ggtt view! Found %d\n",
776 			       vma->ggtt_view.type);
777 			ok = false;
778 		}
779 
780 		if (vma->pages != vma->obj->mm.pages) {
781 			pr_err("VMA not using object pages!\n");
782 			ok = false;
783 		}
784 	}
785 
786 	return ok;
787 }
788 
789 static int igt_vma_partial(void *arg)
790 {
791 	struct i915_ggtt *ggtt = arg;
792 	struct i915_address_space *vm = &ggtt->vm;
793 	const unsigned int npages = 1021; /* prime! */
794 	struct drm_i915_gem_object *obj;
795 	const struct phase {
796 		const char *name;
797 	} phases[] = {
798 		{ "create" },
799 		{ "lookup" },
800 		{ },
801 	}, *p;
802 	unsigned int sz, offset;
803 	struct i915_vma *vma;
804 	int err = -ENOMEM;
805 
806 	/* Create lots of different VMA for the object and check that
807 	 * we are returned the same VMA when we later request the same range.
808 	 */
809 
810 	obj = i915_gem_object_create_internal(vm->i915, npages * PAGE_SIZE);
811 	if (IS_ERR(obj))
812 		goto out;
813 
814 	for (p = phases; p->name; p++) { /* exercise both create/lookup */
815 		unsigned int count, nvma;
816 
817 		nvma = 0;
818 		for_each_prime_number_from(sz, 1, npages) {
819 			for_each_prime_number_from(offset, 0, npages - sz) {
820 				struct i915_ggtt_view view;
821 
822 				view.type = I915_GGTT_VIEW_PARTIAL;
823 				view.partial.offset = offset;
824 				view.partial.size = sz;
825 
826 				if (sz == npages)
827 					view.type = I915_GGTT_VIEW_NORMAL;
828 
829 				vma = checked_vma_instance(obj, vm, &view);
830 				if (IS_ERR(vma)) {
831 					err = PTR_ERR(vma);
832 					goto out_object;
833 				}
834 
835 				err = i915_vma_pin(vma, 0, 0, PIN_GLOBAL);
836 				if (err)
837 					goto out_object;
838 
839 				if (!assert_pin(vma, &view, sz*PAGE_SIZE, p->name)) {
840 					pr_err("(%s) Inconsistent partial pinning for (offset=%d, size=%d)\n",
841 					       p->name, offset, sz);
842 					err = -EINVAL;
843 					goto out_object;
844 				}
845 
846 				if (!assert_partial(obj, vma, offset, sz)) {
847 					pr_err("(%s) Inconsistent partial pages for (offset=%d, size=%d)\n",
848 					       p->name, offset, sz);
849 					err = -EINVAL;
850 					goto out_object;
851 				}
852 
853 				i915_vma_unpin(vma);
854 				nvma++;
855 				err = i915_vma_unbind(vma);
856 				if (err) {
857 					pr_err("Unbinding returned %i\n", err);
858 					goto out_object;
859 				}
860 
861 				cond_resched();
862 			}
863 		}
864 
865 		count = 0;
866 		list_for_each_entry(vma, &obj->vma.list, obj_link)
867 			count++;
868 		if (count != nvma) {
869 			pr_err("(%s) All partial vma were not recorded on the obj->vma_list: found %u, expected %u\n",
870 			       p->name, count, nvma);
871 			err = -EINVAL;
872 			goto out_object;
873 		}
874 
875 		/* Check that we did create the whole object mapping */
876 		vma = checked_vma_instance(obj, vm, NULL);
877 		if (IS_ERR(vma)) {
878 			err = PTR_ERR(vma);
879 			goto out_object;
880 		}
881 
882 		err = i915_vma_pin(vma, 0, 0, PIN_GLOBAL);
883 		if (err)
884 			goto out_object;
885 
886 		if (!assert_pin(vma, NULL, obj->base.size, p->name)) {
887 			pr_err("(%s) inconsistent full pin\n", p->name);
888 			err = -EINVAL;
889 			goto out_object;
890 		}
891 
892 		i915_vma_unpin(vma);
893 
894 		err = i915_vma_unbind(vma);
895 		if (err) {
896 			pr_err("Unbinding returned %i\n", err);
897 			goto out_object;
898 		}
899 
900 		count = 0;
901 		list_for_each_entry(vma, &obj->vma.list, obj_link)
902 			count++;
903 		if (count != nvma) {
904 			pr_err("(%s) allocated an extra full vma!\n", p->name);
905 			err = -EINVAL;
906 			goto out_object;
907 		}
908 	}
909 
910 out_object:
911 	i915_gem_object_put(obj);
912 out:
913 	return err;
914 }
915 
916 int i915_vma_mock_selftests(void)
917 {
918 	static const struct i915_subtest tests[] = {
919 		SUBTEST(igt_vma_create),
920 		SUBTEST(igt_vma_pin1),
921 		SUBTEST(igt_vma_rotate_remap),
922 		SUBTEST(igt_vma_partial),
923 	};
924 	struct drm_i915_private *i915;
925 	struct i915_ggtt *ggtt;
926 	int err;
927 
928 	i915 = mock_gem_device();
929 	if (!i915)
930 		return -ENOMEM;
931 
932 	ggtt = kmalloc(sizeof(*ggtt), GFP_KERNEL);
933 	if (!ggtt) {
934 		err = -ENOMEM;
935 		goto out_put;
936 	}
937 	mock_init_ggtt(i915, ggtt);
938 
939 	err = i915_subtests(tests, ggtt);
940 
941 	mock_device_flush(i915);
942 	i915_gem_drain_freed_objects(i915);
943 	mock_fini_ggtt(ggtt);
944 	kfree(ggtt);
945 out_put:
946 	mock_destroy_device(i915);
947 	return err;
948 }
949 
950 static int igt_vma_remapped_gtt(void *arg)
951 {
952 	struct drm_i915_private *i915 = arg;
953 	const struct intel_remapped_plane_info planes[] = {
954 		{ .width = 1, .height = 1, .src_stride = 1 },
955 		{ .width = 2, .height = 2, .src_stride = 2 },
956 		{ .width = 4, .height = 4, .src_stride = 4 },
957 		{ .width = 8, .height = 8, .src_stride = 8 },
958 
959 		{ .width = 3, .height = 5, .src_stride = 3 },
960 		{ .width = 3, .height = 5, .src_stride = 4 },
961 		{ .width = 3, .height = 5, .src_stride = 5 },
962 
963 		{ .width = 5, .height = 3, .src_stride = 5 },
964 		{ .width = 5, .height = 3, .src_stride = 7 },
965 		{ .width = 5, .height = 3, .src_stride = 9 },
966 
967 		{ .width = 4, .height = 6, .src_stride = 6 },
968 		{ .width = 6, .height = 4, .src_stride = 6 },
969 
970 		{ .width = 2, .height = 2, .src_stride = 2, .dst_stride = 2 },
971 		{ .width = 3, .height = 3, .src_stride = 3, .dst_stride = 4 },
972 		{ .width = 5, .height = 6, .src_stride = 7, .dst_stride = 8 },
973 
974 		{ }
975 	}, *p;
976 	enum i915_ggtt_view_type types[] = {
977 		I915_GGTT_VIEW_ROTATED,
978 		I915_GGTT_VIEW_REMAPPED,
979 		0,
980 	}, *t;
981 	struct drm_i915_gem_object *obj;
982 	intel_wakeref_t wakeref;
983 	int err = 0;
984 
985 	if (!i915_ggtt_has_aperture(&i915->ggtt))
986 		return 0;
987 
988 	obj = i915_gem_object_create_internal(i915, 10 * 10 * PAGE_SIZE);
989 	if (IS_ERR(obj))
990 		return PTR_ERR(obj);
991 
992 	wakeref = intel_runtime_pm_get(&i915->runtime_pm);
993 
994 	for (t = types; *t; t++) {
995 		for (p = planes; p->width; p++) {
996 			struct i915_ggtt_view view = {
997 				.type = *t,
998 				.rotated.plane[0] = *p,
999 			};
1000 			struct intel_remapped_plane_info *plane_info = view.rotated.plane;
1001 			struct i915_vma *vma;
1002 			u32 __iomem *map;
1003 			unsigned int x, y;
1004 
1005 			i915_gem_object_lock(obj, NULL);
1006 			err = i915_gem_object_set_to_gtt_domain(obj, true);
1007 			i915_gem_object_unlock(obj);
1008 			if (err)
1009 				goto out;
1010 
1011 			if (!plane_info[0].dst_stride)
1012 				plane_info[0].dst_stride = *t == I915_GGTT_VIEW_ROTATED ?
1013 								 p->height : p->width;
1014 
1015 			vma = i915_gem_object_ggtt_pin(obj, &view, 0, 0, PIN_MAPPABLE);
1016 			if (IS_ERR(vma)) {
1017 				err = PTR_ERR(vma);
1018 				goto out;
1019 			}
1020 
1021 			GEM_BUG_ON(vma->ggtt_view.type != *t);
1022 
1023 			map = i915_vma_pin_iomap(vma);
1024 			i915_vma_unpin(vma);
1025 			if (IS_ERR(map)) {
1026 				err = PTR_ERR(map);
1027 				goto out;
1028 			}
1029 
1030 			for (y = 0 ; y < plane_info[0].height; y++) {
1031 				for (x = 0 ; x < plane_info[0].width; x++) {
1032 					unsigned int offset;
1033 					u32 val = y << 16 | x;
1034 
1035 					if (*t == I915_GGTT_VIEW_ROTATED)
1036 						offset = (x * plane_info[0].dst_stride + y) * PAGE_SIZE;
1037 					else
1038 						offset = (y * plane_info[0].dst_stride + x) * PAGE_SIZE;
1039 
1040 					iowrite32(val, &map[offset / sizeof(*map)]);
1041 				}
1042 			}
1043 
1044 			i915_vma_unpin_iomap(vma);
1045 
1046 			vma = i915_gem_object_ggtt_pin(obj, NULL, 0, 0, PIN_MAPPABLE);
1047 			if (IS_ERR(vma)) {
1048 				err = PTR_ERR(vma);
1049 				goto out;
1050 			}
1051 
1052 			GEM_BUG_ON(vma->ggtt_view.type != I915_GGTT_VIEW_NORMAL);
1053 
1054 			map = i915_vma_pin_iomap(vma);
1055 			i915_vma_unpin(vma);
1056 			if (IS_ERR(map)) {
1057 				err = PTR_ERR(map);
1058 				goto out;
1059 			}
1060 
1061 			for (y = 0 ; y < plane_info[0].height; y++) {
1062 				for (x = 0 ; x < plane_info[0].width; x++) {
1063 					unsigned int offset, src_idx;
1064 					u32 exp = y << 16 | x;
1065 					u32 val;
1066 
1067 					if (*t == I915_GGTT_VIEW_ROTATED)
1068 						src_idx = rotated_index(&view.rotated, 0, x, y);
1069 					else
1070 						src_idx = remapped_index(&view.remapped, 0, x, y);
1071 					offset = src_idx * PAGE_SIZE;
1072 
1073 					val = ioread32(&map[offset / sizeof(*map)]);
1074 					if (val != exp) {
1075 						pr_err("%s VMA write test failed, expected 0x%x, found 0x%x\n",
1076 						       *t == I915_GGTT_VIEW_ROTATED ? "Rotated" : "Remapped",
1077 						       exp, val);
1078 						i915_vma_unpin_iomap(vma);
1079 						err = -EINVAL;
1080 						goto out;
1081 					}
1082 				}
1083 			}
1084 			i915_vma_unpin_iomap(vma);
1085 
1086 			cond_resched();
1087 		}
1088 	}
1089 
1090 out:
1091 	intel_runtime_pm_put(&i915->runtime_pm, wakeref);
1092 	i915_gem_object_put(obj);
1093 
1094 	return err;
1095 }
1096 
1097 int i915_vma_live_selftests(struct drm_i915_private *i915)
1098 {
1099 	static const struct i915_subtest tests[] = {
1100 		SUBTEST(igt_vma_remapped_gtt),
1101 	};
1102 
1103 	return i915_subtests(tests, i915);
1104 }
1105