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 = &dev_priv->gt.info.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 = struct_size(query_ptr, engines, num_uabi_engines); 113 114 ret = copy_query_item(&query, sizeof(query), len, query_item); 115 if (ret != 0) 116 return ret; 117 118 if (query.num_engines || query.rsvd[0] || query.rsvd[1] || 119 query.rsvd[2]) 120 return -EINVAL; 121 122 info_ptr = &query_ptr->engines[0]; 123 124 for_each_uabi_engine(engine, i915) { 125 info.engine.engine_class = engine->uabi_class; 126 info.engine.engine_instance = engine->uabi_instance; 127 info.capabilities = engine->uabi_capabilities; 128 129 if (copy_to_user(info_ptr, &info, sizeof(info))) 130 return -EFAULT; 131 132 query.num_engines++; 133 info_ptr++; 134 } 135 136 if (copy_to_user(query_ptr, &query, sizeof(query))) 137 return -EFAULT; 138 139 return len; 140 } 141 142 static int can_copy_perf_config_registers_or_number(u32 user_n_regs, 143 u64 user_regs_ptr, 144 u32 kernel_n_regs) 145 { 146 /* 147 * We'll just put the number of registers, and won't copy the 148 * register. 149 */ 150 if (user_n_regs == 0) 151 return 0; 152 153 if (user_n_regs < kernel_n_regs) 154 return -EINVAL; 155 156 return 0; 157 } 158 159 static int copy_perf_config_registers_or_number(const struct i915_oa_reg *kernel_regs, 160 u32 kernel_n_regs, 161 u64 user_regs_ptr, 162 u32 *user_n_regs) 163 { 164 u32 __user *p = u64_to_user_ptr(user_regs_ptr); 165 u32 r; 166 167 if (*user_n_regs == 0) { 168 *user_n_regs = kernel_n_regs; 169 return 0; 170 } 171 172 *user_n_regs = kernel_n_regs; 173 174 if (!user_write_access_begin(p, 2 * sizeof(u32) * kernel_n_regs)) 175 return -EFAULT; 176 177 for (r = 0; r < kernel_n_regs; r++, p += 2) { 178 unsafe_put_user(i915_mmio_reg_offset(kernel_regs[r].addr), 179 p, Efault); 180 unsafe_put_user(kernel_regs[r].value, p + 1, Efault); 181 } 182 user_write_access_end(); 183 return 0; 184 Efault: 185 user_write_access_end(); 186 return -EFAULT; 187 } 188 189 static int query_perf_config_data(struct drm_i915_private *i915, 190 struct drm_i915_query_item *query_item, 191 bool use_uuid) 192 { 193 struct drm_i915_query_perf_config __user *user_query_config_ptr = 194 u64_to_user_ptr(query_item->data_ptr); 195 struct drm_i915_perf_oa_config __user *user_config_ptr = 196 u64_to_user_ptr(query_item->data_ptr + 197 sizeof(struct drm_i915_query_perf_config)); 198 struct drm_i915_perf_oa_config user_config; 199 struct i915_perf *perf = &i915->perf; 200 struct i915_oa_config *oa_config; 201 char uuid[UUID_STRING_LEN + 1]; 202 u64 config_id; 203 u32 flags, total_size; 204 int ret; 205 206 if (!perf->i915) 207 return -ENODEV; 208 209 total_size = 210 sizeof(struct drm_i915_query_perf_config) + 211 sizeof(struct drm_i915_perf_oa_config); 212 213 if (query_item->length == 0) 214 return total_size; 215 216 if (query_item->length < total_size) { 217 DRM_DEBUG("Invalid query config data item size=%u expected=%u\n", 218 query_item->length, total_size); 219 return -EINVAL; 220 } 221 222 if (get_user(flags, &user_query_config_ptr->flags)) 223 return -EFAULT; 224 225 if (flags != 0) 226 return -EINVAL; 227 228 if (use_uuid) { 229 struct i915_oa_config *tmp; 230 int id; 231 232 BUILD_BUG_ON(sizeof(user_query_config_ptr->uuid) >= sizeof(uuid)); 233 234 memset(&uuid, 0, sizeof(uuid)); 235 if (copy_from_user(uuid, user_query_config_ptr->uuid, 236 sizeof(user_query_config_ptr->uuid))) 237 return -EFAULT; 238 239 oa_config = NULL; 240 rcu_read_lock(); 241 idr_for_each_entry(&perf->metrics_idr, tmp, id) { 242 if (!strcmp(tmp->uuid, uuid)) { 243 oa_config = i915_oa_config_get(tmp); 244 break; 245 } 246 } 247 rcu_read_unlock(); 248 } else { 249 if (get_user(config_id, &user_query_config_ptr->config)) 250 return -EFAULT; 251 252 oa_config = i915_perf_get_oa_config(perf, config_id); 253 } 254 if (!oa_config) 255 return -ENOENT; 256 257 if (copy_from_user(&user_config, user_config_ptr, sizeof(user_config))) { 258 ret = -EFAULT; 259 goto out; 260 } 261 262 ret = can_copy_perf_config_registers_or_number(user_config.n_boolean_regs, 263 user_config.boolean_regs_ptr, 264 oa_config->b_counter_regs_len); 265 if (ret) 266 goto out; 267 268 ret = can_copy_perf_config_registers_or_number(user_config.n_flex_regs, 269 user_config.flex_regs_ptr, 270 oa_config->flex_regs_len); 271 if (ret) 272 goto out; 273 274 ret = can_copy_perf_config_registers_or_number(user_config.n_mux_regs, 275 user_config.mux_regs_ptr, 276 oa_config->mux_regs_len); 277 if (ret) 278 goto out; 279 280 ret = copy_perf_config_registers_or_number(oa_config->b_counter_regs, 281 oa_config->b_counter_regs_len, 282 user_config.boolean_regs_ptr, 283 &user_config.n_boolean_regs); 284 if (ret) 285 goto out; 286 287 ret = copy_perf_config_registers_or_number(oa_config->flex_regs, 288 oa_config->flex_regs_len, 289 user_config.flex_regs_ptr, 290 &user_config.n_flex_regs); 291 if (ret) 292 goto out; 293 294 ret = copy_perf_config_registers_or_number(oa_config->mux_regs, 295 oa_config->mux_regs_len, 296 user_config.mux_regs_ptr, 297 &user_config.n_mux_regs); 298 if (ret) 299 goto out; 300 301 memcpy(user_config.uuid, oa_config->uuid, sizeof(user_config.uuid)); 302 303 if (copy_to_user(user_config_ptr, &user_config, sizeof(user_config))) { 304 ret = -EFAULT; 305 goto out; 306 } 307 308 ret = total_size; 309 310 out: 311 i915_oa_config_put(oa_config); 312 return ret; 313 } 314 315 static size_t sizeof_perf_config_list(size_t count) 316 { 317 return sizeof(struct drm_i915_query_perf_config) + sizeof(u64) * count; 318 } 319 320 static size_t sizeof_perf_metrics(struct i915_perf *perf) 321 { 322 struct i915_oa_config *tmp; 323 size_t i; 324 int id; 325 326 i = 1; 327 rcu_read_lock(); 328 idr_for_each_entry(&perf->metrics_idr, tmp, id) 329 i++; 330 rcu_read_unlock(); 331 332 return sizeof_perf_config_list(i); 333 } 334 335 static int query_perf_config_list(struct drm_i915_private *i915, 336 struct drm_i915_query_item *query_item) 337 { 338 struct drm_i915_query_perf_config __user *user_query_config_ptr = 339 u64_to_user_ptr(query_item->data_ptr); 340 struct i915_perf *perf = &i915->perf; 341 u64 *oa_config_ids = NULL; 342 int alloc, n_configs; 343 u32 flags; 344 int ret; 345 346 if (!perf->i915) 347 return -ENODEV; 348 349 if (query_item->length == 0) 350 return sizeof_perf_metrics(perf); 351 352 if (get_user(flags, &user_query_config_ptr->flags)) 353 return -EFAULT; 354 355 if (flags != 0) 356 return -EINVAL; 357 358 n_configs = 1; 359 do { 360 struct i915_oa_config *tmp; 361 u64 *ids; 362 int id; 363 364 ids = krealloc(oa_config_ids, 365 n_configs * sizeof(*oa_config_ids), 366 GFP_KERNEL); 367 if (!ids) 368 return -ENOMEM; 369 370 alloc = fetch_and_zero(&n_configs); 371 372 ids[n_configs++] = 1ull; /* reserved for test_config */ 373 rcu_read_lock(); 374 idr_for_each_entry(&perf->metrics_idr, tmp, id) { 375 if (n_configs < alloc) 376 ids[n_configs] = id; 377 n_configs++; 378 } 379 rcu_read_unlock(); 380 381 oa_config_ids = ids; 382 } while (n_configs > alloc); 383 384 if (query_item->length < sizeof_perf_config_list(n_configs)) { 385 DRM_DEBUG("Invalid query config list item size=%u expected=%zu\n", 386 query_item->length, 387 sizeof_perf_config_list(n_configs)); 388 kfree(oa_config_ids); 389 return -EINVAL; 390 } 391 392 if (put_user(n_configs, &user_query_config_ptr->config)) { 393 kfree(oa_config_ids); 394 return -EFAULT; 395 } 396 397 ret = copy_to_user(user_query_config_ptr + 1, 398 oa_config_ids, 399 n_configs * sizeof(*oa_config_ids)); 400 kfree(oa_config_ids); 401 if (ret) 402 return -EFAULT; 403 404 return sizeof_perf_config_list(n_configs); 405 } 406 407 static int query_perf_config(struct drm_i915_private *i915, 408 struct drm_i915_query_item *query_item) 409 { 410 switch (query_item->flags) { 411 case DRM_I915_QUERY_PERF_CONFIG_LIST: 412 return query_perf_config_list(i915, query_item); 413 case DRM_I915_QUERY_PERF_CONFIG_DATA_FOR_UUID: 414 return query_perf_config_data(i915, query_item, true); 415 case DRM_I915_QUERY_PERF_CONFIG_DATA_FOR_ID: 416 return query_perf_config_data(i915, query_item, false); 417 default: 418 return -EINVAL; 419 } 420 } 421 422 static int (* const i915_query_funcs[])(struct drm_i915_private *dev_priv, 423 struct drm_i915_query_item *query_item) = { 424 query_topology_info, 425 query_engine_info, 426 query_perf_config, 427 }; 428 429 int i915_query_ioctl(struct drm_device *dev, void *data, struct drm_file *file) 430 { 431 struct drm_i915_private *dev_priv = to_i915(dev); 432 struct drm_i915_query *args = data; 433 struct drm_i915_query_item __user *user_item_ptr = 434 u64_to_user_ptr(args->items_ptr); 435 u32 i; 436 437 if (args->flags != 0) 438 return -EINVAL; 439 440 for (i = 0; i < args->num_items; i++, user_item_ptr++) { 441 struct drm_i915_query_item item; 442 unsigned long func_idx; 443 int ret; 444 445 if (copy_from_user(&item, user_item_ptr, sizeof(item))) 446 return -EFAULT; 447 448 if (item.query_id == 0) 449 return -EINVAL; 450 451 if (overflows_type(item.query_id - 1, unsigned long)) 452 return -EINVAL; 453 454 func_idx = item.query_id - 1; 455 456 ret = -EINVAL; 457 if (func_idx < ARRAY_SIZE(i915_query_funcs)) { 458 func_idx = array_index_nospec(func_idx, 459 ARRAY_SIZE(i915_query_funcs)); 460 ret = i915_query_funcs[func_idx](dev_priv, &item); 461 } 462 463 /* Only write the length back to userspace if they differ. */ 464 if (ret != item.length && put_user(ret, &user_item_ptr->length)) 465 return -EFAULT; 466 } 467 468 return 0; 469 } 470