1/*
2 * SPDX-License-Identifier: MIT
3 *
4 * Copyright © 2016 Intel Corporation
5 */
6
7#include "i915_file_private.h"
8#include "mock_context.h"
9#include "selftests/mock_drm.h"
10#include "selftests/mock_gtt.h"
11
12struct i915_gem_context *
13mock_context(struct drm_i915_private *i915,
14 const char *name)
15{
16 struct i915_gem_context *ctx;
17 struct i915_gem_engines *e;
18 struct intel_sseu null_sseu = {};
19
20 ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
21 if (!ctx)
22 return NULL;
23
24 kref_init(kref: &ctx->ref);
25 INIT_LIST_HEAD(list: &ctx->link);
26 ctx->i915 = i915;
27 INIT_WORK(&ctx->release_work, i915_gem_context_release_work);
28
29 mutex_init(&ctx->mutex);
30
31 spin_lock_init(&ctx->stale.lock);
32 INIT_LIST_HEAD(list: &ctx->stale.engines);
33
34 i915_gem_context_set_persistence(ctx);
35
36 if (name) {
37 struct i915_ppgtt *ppgtt;
38
39 strscpy(ctx->name, name, sizeof(ctx->name));
40
41 ppgtt = mock_ppgtt(i915, name);
42 if (!ppgtt)
43 goto err_free;
44
45 ctx->vm = &ppgtt->vm;
46 }
47
48 mutex_init(&ctx->engines_mutex);
49 e = default_engines(ctx, rcs_sseu: null_sseu);
50 if (IS_ERR(ptr: e))
51 goto err_vm;
52 RCU_INIT_POINTER(ctx->engines, e);
53
54 INIT_RADIX_TREE(&ctx->handles_vma, GFP_KERNEL);
55 mutex_init(&ctx->lut_mutex);
56
57 return ctx;
58
59err_vm:
60 if (ctx->vm)
61 i915_vm_put(vm: ctx->vm);
62err_free:
63 kfree(objp: ctx);
64 return NULL;
65}
66
67void mock_context_close(struct i915_gem_context *ctx)
68{
69 context_close(ctx);
70}
71
72void mock_init_contexts(struct drm_i915_private *i915)
73{
74 init_contexts(gc: &i915->gem.contexts);
75}
76
77struct i915_gem_context *
78live_context(struct drm_i915_private *i915, struct file *file)
79{
80 struct drm_i915_file_private *fpriv = to_drm_file(f: file)->driver_priv;
81 struct i915_gem_proto_context *pc;
82 struct i915_gem_context *ctx;
83 int err;
84 u32 id;
85
86 pc = proto_context_create(fpriv, i915, flags: 0);
87 if (IS_ERR(ptr: pc))
88 return ERR_CAST(ptr: pc);
89
90 ctx = i915_gem_create_context(i915, pc);
91 proto_context_close(i915, pc);
92 if (IS_ERR(ptr: ctx))
93 return ctx;
94
95 i915_gem_context_set_no_error_capture(ctx);
96
97 err = xa_alloc(xa: &fpriv->context_xa, id: &id, NULL, xa_limit_32b, GFP_KERNEL);
98 if (err < 0)
99 goto err_ctx;
100
101 gem_context_register(ctx, fpriv, id);
102
103 return ctx;
104
105err_ctx:
106 context_close(ctx);
107 return ERR_PTR(error: err);
108}
109
110struct i915_gem_context *
111kernel_context(struct drm_i915_private *i915,
112 struct i915_address_space *vm)
113{
114 struct i915_gem_context *ctx;
115 struct i915_gem_proto_context *pc;
116
117 pc = proto_context_create(NULL, i915, flags: 0);
118 if (IS_ERR(ptr: pc))
119 return ERR_CAST(ptr: pc);
120
121 if (vm) {
122 if (pc->vm)
123 i915_vm_put(vm: pc->vm);
124 pc->vm = i915_vm_get(vm);
125 }
126
127 ctx = i915_gem_create_context(i915, pc);
128 proto_context_close(i915, pc);
129 if (IS_ERR(ptr: ctx))
130 return ctx;
131
132 i915_gem_context_clear_bannable(ctx);
133 i915_gem_context_set_persistence(ctx);
134 i915_gem_context_set_no_error_capture(ctx);
135
136 return ctx;
137}
138
139void kernel_context_close(struct i915_gem_context *ctx)
140{
141 context_close(ctx);
142}
143

source code of linux/drivers/gpu/drm/i915/gem/selftests/mock_context.c