xref: /linux/drivers/gpu/drm/i915/i915_query.c (revision 3503d56cc7233ced602e38a4c13caa64f00ab2aa)
1 /*
2  * SPDX-License-Identifier: MIT
3  *
4  * Copyright © 2018 Intel Corporation
5  */
6 
7 #include <linux/nospec.h>
8 
9 #include "i915_drv.h"
10 #include "i915_perf.h"
11 #include "i915_query.h"
12 #include <uapi/drm/i915_drm.h>
13 
14 static int copy_query_item(void *query_hdr, size_t query_sz,
15 			   u32 total_length,
16 			   struct drm_i915_query_item *query_item)
17 {
18 	if (query_item->length == 0)
19 		return total_length;
20 
21 	if (query_item->length < total_length)
22 		return -EINVAL;
23 
24 	if (copy_from_user(query_hdr, u64_to_user_ptr(query_item->data_ptr),
25 			   query_sz))
26 		return -EFAULT;
27 
28 	return 0;
29 }
30 
31 static int query_topology_info(struct drm_i915_private *dev_priv,
32 			       struct drm_i915_query_item *query_item)
33 {
34 	const struct sseu_dev_info *sseu = &RUNTIME_INFO(dev_priv)->sseu;
35 	struct drm_i915_query_topology_info topo;
36 	u32 slice_length, subslice_length, eu_length, total_length;
37 	int ret;
38 
39 	if (query_item->flags != 0)
40 		return -EINVAL;
41 
42 	if (sseu->max_slices == 0)
43 		return -ENODEV;
44 
45 	BUILD_BUG_ON(sizeof(u8) != sizeof(sseu->slice_mask));
46 
47 	slice_length = sizeof(sseu->slice_mask);
48 	subslice_length = sseu->max_slices * sseu->ss_stride;
49 	eu_length = sseu->max_slices * sseu->max_subslices * sseu->eu_stride;
50 	total_length = sizeof(topo) + slice_length + subslice_length +
51 		       eu_length;
52 
53 	ret = copy_query_item(&topo, sizeof(topo), total_length,
54 			      query_item);
55 	if (ret != 0)
56 		return ret;
57 
58 	if (topo.flags != 0)
59 		return -EINVAL;
60 
61 	memset(&topo, 0, sizeof(topo));
62 	topo.max_slices = sseu->max_slices;
63 	topo.max_subslices = sseu->max_subslices;
64 	topo.max_eus_per_subslice = sseu->max_eus_per_subslice;
65 
66 	topo.subslice_offset = slice_length;
67 	topo.subslice_stride = sseu->ss_stride;
68 	topo.eu_offset = slice_length + subslice_length;
69 	topo.eu_stride = sseu->eu_stride;
70 
71 	if (copy_to_user(u64_to_user_ptr(query_item->data_ptr),
72 			   &topo, sizeof(topo)))
73 		return -EFAULT;
74 
75 	if (copy_to_user(u64_to_user_ptr(query_item->data_ptr + sizeof(topo)),
76 			   &sseu->slice_mask, slice_length))
77 		return -EFAULT;
78 
79 	if (copy_to_user(u64_to_user_ptr(query_item->data_ptr +
80 					   sizeof(topo) + slice_length),
81 			   sseu->subslice_mask, subslice_length))
82 		return -EFAULT;
83 
84 	if (copy_to_user(u64_to_user_ptr(query_item->data_ptr +
85 					   sizeof(topo) +
86 					   slice_length + subslice_length),
87 			   sseu->eu_mask, eu_length))
88 		return -EFAULT;
89 
90 	return total_length;
91 }
92 
93 static int
94 query_engine_info(struct drm_i915_private *i915,
95 		  struct drm_i915_query_item *query_item)
96 {
97 	struct drm_i915_query_engine_info __user *query_ptr =
98 				u64_to_user_ptr(query_item->data_ptr);
99 	struct drm_i915_engine_info __user *info_ptr;
100 	struct drm_i915_query_engine_info query;
101 	struct drm_i915_engine_info info = { };
102 	unsigned int num_uabi_engines = 0;
103 	struct intel_engine_cs *engine;
104 	int len, ret;
105 
106 	if (query_item->flags)
107 		return -EINVAL;
108 
109 	for_each_uabi_engine(engine, i915)
110 		num_uabi_engines++;
111 
112 	len = sizeof(struct drm_i915_query_engine_info) +
113 	      num_uabi_engines * sizeof(struct drm_i915_engine_info);
114 
115 	ret = copy_query_item(&query, sizeof(query), len, query_item);
116 	if (ret != 0)
117 		return ret;
118 
119 	if (query.num_engines || query.rsvd[0] || query.rsvd[1] ||
120 	    query.rsvd[2])
121 		return -EINVAL;
122 
123 	info_ptr = &query_ptr->engines[0];
124 
125 	for_each_uabi_engine(engine, i915) {
126 		info.engine.engine_class = engine->uabi_class;
127 		info.engine.engine_instance = engine->uabi_instance;
128 		info.capabilities = engine->uabi_capabilities;
129 
130 		if (copy_to_user(info_ptr, &info, sizeof(info)))
131 			return -EFAULT;
132 
133 		query.num_engines++;
134 		info_ptr++;
135 	}
136 
137 	if (copy_to_user(query_ptr, &query, sizeof(query)))
138 		return -EFAULT;
139 
140 	return len;
141 }
142 
143 static int can_copy_perf_config_registers_or_number(u32 user_n_regs,
144 						    u64 user_regs_ptr,
145 						    u32 kernel_n_regs)
146 {
147 	/*
148 	 * We'll just put the number of registers, and won't copy the
149 	 * register.
150 	 */
151 	if (user_n_regs == 0)
152 		return 0;
153 
154 	if (user_n_regs < kernel_n_regs)
155 		return -EINVAL;
156 
157 	return 0;
158 }
159 
160 static int copy_perf_config_registers_or_number(const struct i915_oa_reg *kernel_regs,
161 						u32 kernel_n_regs,
162 						u64 user_regs_ptr,
163 						u32 *user_n_regs)
164 {
165 	u32 __user *p = u64_to_user_ptr(user_regs_ptr);
166 	u32 r;
167 
168 	if (*user_n_regs == 0) {
169 		*user_n_regs = kernel_n_regs;
170 		return 0;
171 	}
172 
173 	*user_n_regs = kernel_n_regs;
174 
175 	if (!user_write_access_begin(p, 2 * sizeof(u32) * kernel_n_regs))
176 		return -EFAULT;
177 
178 	for (r = 0; r < kernel_n_regs; r++, p += 2) {
179 		unsafe_put_user(i915_mmio_reg_offset(kernel_regs[r].addr),
180 				p, Efault);
181 		unsafe_put_user(kernel_regs[r].value, p + 1, Efault);
182 	}
183 	user_write_access_end();
184 	return 0;
185 Efault:
186 	user_write_access_end();
187 	return -EFAULT;
188 }
189 
190 static int query_perf_config_data(struct drm_i915_private *i915,
191 				  struct drm_i915_query_item *query_item,
192 				  bool use_uuid)
193 {
194 	struct drm_i915_query_perf_config __user *user_query_config_ptr =
195 		u64_to_user_ptr(query_item->data_ptr);
196 	struct drm_i915_perf_oa_config __user *user_config_ptr =
197 		u64_to_user_ptr(query_item->data_ptr +
198 				sizeof(struct drm_i915_query_perf_config));
199 	struct drm_i915_perf_oa_config user_config;
200 	struct i915_perf *perf = &i915->perf;
201 	struct i915_oa_config *oa_config;
202 	char uuid[UUID_STRING_LEN + 1];
203 	u64 config_id;
204 	u32 flags, total_size;
205 	int ret;
206 
207 	if (!perf->i915)
208 		return -ENODEV;
209 
210 	total_size =
211 		sizeof(struct drm_i915_query_perf_config) +
212 		sizeof(struct drm_i915_perf_oa_config);
213 
214 	if (query_item->length == 0)
215 		return total_size;
216 
217 	if (query_item->length < total_size) {
218 		DRM_DEBUG("Invalid query config data item size=%u expected=%u\n",
219 			  query_item->length, total_size);
220 		return -EINVAL;
221 	}
222 
223 	if (get_user(flags, &user_query_config_ptr->flags))
224 		return -EFAULT;
225 
226 	if (flags != 0)
227 		return -EINVAL;
228 
229 	if (use_uuid) {
230 		struct i915_oa_config *tmp;
231 		int id;
232 
233 		BUILD_BUG_ON(sizeof(user_query_config_ptr->uuid) >= sizeof(uuid));
234 
235 		memset(&uuid, 0, sizeof(uuid));
236 		if (copy_from_user(uuid, user_query_config_ptr->uuid,
237 				     sizeof(user_query_config_ptr->uuid)))
238 			return -EFAULT;
239 
240 		oa_config = NULL;
241 		rcu_read_lock();
242 		idr_for_each_entry(&perf->metrics_idr, tmp, id) {
243 			if (!strcmp(tmp->uuid, uuid)) {
244 				oa_config = i915_oa_config_get(tmp);
245 				break;
246 			}
247 		}
248 		rcu_read_unlock();
249 	} else {
250 		if (get_user(config_id, &user_query_config_ptr->config))
251 			return -EFAULT;
252 
253 		oa_config = i915_perf_get_oa_config(perf, config_id);
254 	}
255 	if (!oa_config)
256 		return -ENOENT;
257 
258 	if (copy_from_user(&user_config, user_config_ptr, sizeof(user_config))) {
259 		ret = -EFAULT;
260 		goto out;
261 	}
262 
263 	ret = can_copy_perf_config_registers_or_number(user_config.n_boolean_regs,
264 						       user_config.boolean_regs_ptr,
265 						       oa_config->b_counter_regs_len);
266 	if (ret)
267 		goto out;
268 
269 	ret = can_copy_perf_config_registers_or_number(user_config.n_flex_regs,
270 						       user_config.flex_regs_ptr,
271 						       oa_config->flex_regs_len);
272 	if (ret)
273 		goto out;
274 
275 	ret = can_copy_perf_config_registers_or_number(user_config.n_mux_regs,
276 						       user_config.mux_regs_ptr,
277 						       oa_config->mux_regs_len);
278 	if (ret)
279 		goto out;
280 
281 	ret = copy_perf_config_registers_or_number(oa_config->b_counter_regs,
282 						   oa_config->b_counter_regs_len,
283 						   user_config.boolean_regs_ptr,
284 						   &user_config.n_boolean_regs);
285 	if (ret)
286 		goto out;
287 
288 	ret = copy_perf_config_registers_or_number(oa_config->flex_regs,
289 						   oa_config->flex_regs_len,
290 						   user_config.flex_regs_ptr,
291 						   &user_config.n_flex_regs);
292 	if (ret)
293 		goto out;
294 
295 	ret = copy_perf_config_registers_or_number(oa_config->mux_regs,
296 						   oa_config->mux_regs_len,
297 						   user_config.mux_regs_ptr,
298 						   &user_config.n_mux_regs);
299 	if (ret)
300 		goto out;
301 
302 	memcpy(user_config.uuid, oa_config->uuid, sizeof(user_config.uuid));
303 
304 	if (copy_to_user(user_config_ptr, &user_config, sizeof(user_config))) {
305 		ret = -EFAULT;
306 		goto out;
307 	}
308 
309 	ret = total_size;
310 
311 out:
312 	i915_oa_config_put(oa_config);
313 	return ret;
314 }
315 
316 static size_t sizeof_perf_config_list(size_t count)
317 {
318 	return sizeof(struct drm_i915_query_perf_config) + sizeof(u64) * count;
319 }
320 
321 static size_t sizeof_perf_metrics(struct i915_perf *perf)
322 {
323 	struct i915_oa_config *tmp;
324 	size_t i;
325 	int id;
326 
327 	i = 1;
328 	rcu_read_lock();
329 	idr_for_each_entry(&perf->metrics_idr, tmp, id)
330 		i++;
331 	rcu_read_unlock();
332 
333 	return sizeof_perf_config_list(i);
334 }
335 
336 static int query_perf_config_list(struct drm_i915_private *i915,
337 				  struct drm_i915_query_item *query_item)
338 {
339 	struct drm_i915_query_perf_config __user *user_query_config_ptr =
340 		u64_to_user_ptr(query_item->data_ptr);
341 	struct i915_perf *perf = &i915->perf;
342 	u64 *oa_config_ids = NULL;
343 	int alloc, n_configs;
344 	u32 flags;
345 	int ret;
346 
347 	if (!perf->i915)
348 		return -ENODEV;
349 
350 	if (query_item->length == 0)
351 		return sizeof_perf_metrics(perf);
352 
353 	if (get_user(flags, &user_query_config_ptr->flags))
354 		return -EFAULT;
355 
356 	if (flags != 0)
357 		return -EINVAL;
358 
359 	n_configs = 1;
360 	do {
361 		struct i915_oa_config *tmp;
362 		u64 *ids;
363 		int id;
364 
365 		ids = krealloc(oa_config_ids,
366 			       n_configs * sizeof(*oa_config_ids),
367 			       GFP_KERNEL);
368 		if (!ids)
369 			return -ENOMEM;
370 
371 		alloc = fetch_and_zero(&n_configs);
372 
373 		ids[n_configs++] = 1ull; /* reserved for test_config */
374 		rcu_read_lock();
375 		idr_for_each_entry(&perf->metrics_idr, tmp, id) {
376 			if (n_configs < alloc)
377 				ids[n_configs] = id;
378 			n_configs++;
379 		}
380 		rcu_read_unlock();
381 
382 		oa_config_ids = ids;
383 	} while (n_configs > alloc);
384 
385 	if (query_item->length < sizeof_perf_config_list(n_configs)) {
386 		DRM_DEBUG("Invalid query config list item size=%u expected=%zu\n",
387 			  query_item->length,
388 			  sizeof_perf_config_list(n_configs));
389 		kfree(oa_config_ids);
390 		return -EINVAL;
391 	}
392 
393 	if (put_user(n_configs, &user_query_config_ptr->config)) {
394 		kfree(oa_config_ids);
395 		return -EFAULT;
396 	}
397 
398 	ret = copy_to_user(user_query_config_ptr + 1,
399 			   oa_config_ids,
400 			   n_configs * sizeof(*oa_config_ids));
401 	kfree(oa_config_ids);
402 	if (ret)
403 		return -EFAULT;
404 
405 	return sizeof_perf_config_list(n_configs);
406 }
407 
408 static int query_perf_config(struct drm_i915_private *i915,
409 			     struct drm_i915_query_item *query_item)
410 {
411 	switch (query_item->flags) {
412 	case DRM_I915_QUERY_PERF_CONFIG_LIST:
413 		return query_perf_config_list(i915, query_item);
414 	case DRM_I915_QUERY_PERF_CONFIG_DATA_FOR_UUID:
415 		return query_perf_config_data(i915, query_item, true);
416 	case DRM_I915_QUERY_PERF_CONFIG_DATA_FOR_ID:
417 		return query_perf_config_data(i915, query_item, false);
418 	default:
419 		return -EINVAL;
420 	}
421 }
422 
423 static int (* const i915_query_funcs[])(struct drm_i915_private *dev_priv,
424 					struct drm_i915_query_item *query_item) = {
425 	query_topology_info,
426 	query_engine_info,
427 	query_perf_config,
428 };
429 
430 int i915_query_ioctl(struct drm_device *dev, void *data, struct drm_file *file)
431 {
432 	struct drm_i915_private *dev_priv = to_i915(dev);
433 	struct drm_i915_query *args = data;
434 	struct drm_i915_query_item __user *user_item_ptr =
435 		u64_to_user_ptr(args->items_ptr);
436 	u32 i;
437 
438 	if (args->flags != 0)
439 		return -EINVAL;
440 
441 	for (i = 0; i < args->num_items; i++, user_item_ptr++) {
442 		struct drm_i915_query_item item;
443 		unsigned long func_idx;
444 		int ret;
445 
446 		if (copy_from_user(&item, user_item_ptr, sizeof(item)))
447 			return -EFAULT;
448 
449 		if (item.query_id == 0)
450 			return -EINVAL;
451 
452 		if (overflows_type(item.query_id - 1, unsigned long))
453 			return -EINVAL;
454 
455 		func_idx = item.query_id - 1;
456 
457 		ret = -EINVAL;
458 		if (func_idx < ARRAY_SIZE(i915_query_funcs)) {
459 			func_idx = array_index_nospec(func_idx,
460 						      ARRAY_SIZE(i915_query_funcs));
461 			ret = i915_query_funcs[func_idx](dev_priv, &item);
462 		}
463 
464 		/* Only write the length back to userspace if they differ. */
465 		if (ret != item.length && put_user(ret, &user_item_ptr->length))
466 			return -EFAULT;
467 	}
468 
469 	return 0;
470 }
471