1
2
3
4
5
6
7#include "mock_context.h"
8#include "selftests/mock_drm.h"
9#include "selftests/mock_gtt.h"
10
11struct i915_gem_context *
12mock_context(struct drm_i915_private *i915,
13 const char *name)
14{
15 struct i915_gem_context *ctx;
16 struct i915_gem_engines *e;
17
18 ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
19 if (!ctx)
20 return NULL;
21
22 kref_init(&ctx->ref);
23 INIT_LIST_HEAD(&ctx->link);
24 ctx->i915 = i915;
25
26 mutex_init(&ctx->mutex);
27
28 spin_lock_init(&ctx->stale.lock);
29 INIT_LIST_HEAD(&ctx->stale.engines);
30
31 i915_gem_context_set_persistence(ctx);
32
33 mutex_init(&ctx->engines_mutex);
34 e = default_engines(ctx);
35 if (IS_ERR(e))
36 goto err_free;
37 RCU_INIT_POINTER(ctx->engines, e);
38
39 INIT_RADIX_TREE(&ctx->handles_vma, GFP_KERNEL);
40 mutex_init(&ctx->lut_mutex);
41
42 if (name) {
43 struct i915_ppgtt *ppgtt;
44
45 strncpy(ctx->name, name, sizeof(ctx->name) - 1);
46
47 ppgtt = mock_ppgtt(i915, name);
48 if (!ppgtt)
49 goto err_put;
50
51 mutex_lock(&ctx->mutex);
52 __set_ppgtt(ctx, &ppgtt->vm);
53 mutex_unlock(&ctx->mutex);
54
55 i915_vm_put(&ppgtt->vm);
56 }
57
58 return ctx;
59
60err_free:
61 kfree(ctx);
62 return NULL;
63
64err_put:
65 i915_gem_context_set_closed(ctx);
66 i915_gem_context_put(ctx);
67 return NULL;
68}
69
70void mock_context_close(struct i915_gem_context *ctx)
71{
72 context_close(ctx);
73}
74
75void mock_init_contexts(struct drm_i915_private *i915)
76{
77 init_contexts(&i915->gem.contexts);
78}
79
80struct i915_gem_context *
81live_context(struct drm_i915_private *i915, struct file *file)
82{
83 struct i915_gem_context *ctx;
84 int err;
85 u32 id;
86
87 ctx = i915_gem_create_context(i915, 0);
88 if (IS_ERR(ctx))
89 return ctx;
90
91 i915_gem_context_set_no_error_capture(ctx);
92
93 err = gem_context_register(ctx, to_drm_file(file)->driver_priv, &id);
94 if (err < 0)
95 goto err_ctx;
96
97 return ctx;
98
99err_ctx:
100 context_close(ctx);
101 return ERR_PTR(err);
102}
103
104struct i915_gem_context *
105live_context_for_engine(struct intel_engine_cs *engine, struct file *file)
106{
107 struct i915_gem_engines *engines;
108 struct i915_gem_context *ctx;
109 struct intel_context *ce;
110
111 engines = alloc_engines(1);
112 if (!engines)
113 return ERR_PTR(-ENOMEM);
114
115 ctx = live_context(engine->i915, file);
116 if (IS_ERR(ctx)) {
117 __free_engines(engines, 0);
118 return ctx;
119 }
120
121 ce = intel_context_create(engine);
122 if (IS_ERR(ce)) {
123 __free_engines(engines, 0);
124 return ERR_CAST(ce);
125 }
126
127 intel_context_set_gem(ce, ctx);
128 engines->engines[0] = ce;
129 engines->num_engines = 1;
130
131 mutex_lock(&ctx->engines_mutex);
132 i915_gem_context_set_user_engines(ctx);
133 engines = rcu_replace_pointer(ctx->engines, engines, 1);
134 mutex_unlock(&ctx->engines_mutex);
135
136 engines_idle_release(ctx, engines);
137
138 return ctx;
139}
140
141struct i915_gem_context *
142kernel_context(struct drm_i915_private *i915)
143{
144 struct i915_gem_context *ctx;
145
146 ctx = i915_gem_create_context(i915, 0);
147 if (IS_ERR(ctx))
148 return ctx;
149
150 i915_gem_context_clear_bannable(ctx);
151 i915_gem_context_set_persistence(ctx);
152 i915_gem_context_set_no_error_capture(ctx);
153
154 return ctx;
155}
156
157void kernel_context_close(struct i915_gem_context *ctx)
158{
159 context_close(ctx);
160}
161
