Linux Audio

Check our new training course

Embedded Linux training

Mar 31-Apr 8, 2025
Register
Loading...
v5.4
   1// SPDX-License-Identifier: GPL-2.0 OR MIT
   2/**************************************************************************
   3 *
   4 * Copyright 2009-2015 VMware, Inc., Palo Alto, CA., USA
 
   5 *
   6 * Permission is hereby granted, free of charge, to any person obtaining a
   7 * copy of this software and associated documentation files (the
   8 * "Software"), to deal in the Software without restriction, including
   9 * without limitation the rights to use, copy, modify, merge, publish,
  10 * distribute, sub license, and/or sell copies of the Software, and to
  11 * permit persons to whom the Software is furnished to do so, subject to
  12 * the following conditions:
  13 *
  14 * The above copyright notice and this permission notice (including the
  15 * next paragraph) shall be included in all copies or substantial portions
  16 * of the Software.
  17 *
  18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  21 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
  22 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  23 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  24 * USE OR OTHER DEALINGS IN THE SOFTWARE.
  25 *
  26 **************************************************************************/
  27
  28#include <drm/ttm/ttm_placement.h>
  29
  30#include "vmwgfx_drv.h"
  31#include "vmwgfx_resource_priv.h"
  32#include "vmwgfx_binding.h"
 
  33
  34struct vmw_shader {
  35	struct vmw_resource res;
  36	SVGA3dShaderType type;
  37	uint32_t size;
  38	uint8_t num_input_sig;
  39	uint8_t num_output_sig;
  40};
  41
  42struct vmw_user_shader {
  43	struct ttm_base_object base;
  44	struct vmw_shader shader;
  45};
  46
  47struct vmw_dx_shader {
  48	struct vmw_resource res;
  49	struct vmw_resource *ctx;
  50	struct vmw_resource *cotable;
  51	u32 id;
  52	bool committed;
  53	struct list_head cotable_head;
  54};
  55
  56static uint64_t vmw_user_shader_size;
  57static uint64_t vmw_shader_size;
  58static size_t vmw_shader_dx_size;
  59
  60static void vmw_user_shader_free(struct vmw_resource *res);
  61static struct vmw_resource *
  62vmw_user_shader_base_to_res(struct ttm_base_object *base);
  63
  64static int vmw_gb_shader_create(struct vmw_resource *res);
  65static int vmw_gb_shader_bind(struct vmw_resource *res,
  66			       struct ttm_validate_buffer *val_buf);
  67static int vmw_gb_shader_unbind(struct vmw_resource *res,
  68				 bool readback,
  69				 struct ttm_validate_buffer *val_buf);
  70static int vmw_gb_shader_destroy(struct vmw_resource *res);
  71
  72static int vmw_dx_shader_create(struct vmw_resource *res);
  73static int vmw_dx_shader_bind(struct vmw_resource *res,
  74			       struct ttm_validate_buffer *val_buf);
  75static int vmw_dx_shader_unbind(struct vmw_resource *res,
  76				 bool readback,
  77				 struct ttm_validate_buffer *val_buf);
  78static void vmw_dx_shader_commit_notify(struct vmw_resource *res,
  79					enum vmw_cmdbuf_res_state state);
  80static bool vmw_shader_id_ok(u32 user_key, SVGA3dShaderType shader_type);
  81static u32 vmw_shader_key(u32 user_key, SVGA3dShaderType shader_type);
  82static uint64_t vmw_user_shader_size;
  83
  84static const struct vmw_user_resource_conv user_shader_conv = {
  85	.object_type = VMW_RES_SHADER,
  86	.base_obj_to_res = vmw_user_shader_base_to_res,
  87	.res_free = vmw_user_shader_free
  88};
  89
  90const struct vmw_user_resource_conv *user_shader_converter =
  91	&user_shader_conv;
  92
  93
  94static const struct vmw_res_func vmw_gb_shader_func = {
  95	.res_type = vmw_res_shader,
  96	.needs_backup = true,
  97	.may_evict = true,
  98	.prio = 3,
  99	.dirty_prio = 3,
 100	.type_name = "guest backed shaders",
 101	.backup_placement = &vmw_mob_placement,
 102	.create = vmw_gb_shader_create,
 103	.destroy = vmw_gb_shader_destroy,
 104	.bind = vmw_gb_shader_bind,
 105	.unbind = vmw_gb_shader_unbind
 106};
 107
 108static const struct vmw_res_func vmw_dx_shader_func = {
 109	.res_type = vmw_res_shader,
 110	.needs_backup = true,
 111	.may_evict = true,
 112	.prio = 3,
 113	.dirty_prio = 3,
 114	.type_name = "dx shaders",
 115	.backup_placement = &vmw_mob_placement,
 116	.create = vmw_dx_shader_create,
 117	/*
 118	 * The destroy callback is only called with a committed resource on
 119	 * context destroy, in which case we destroy the cotable anyway,
 120	 * so there's no need to destroy DX shaders separately.
 121	 */
 122	.destroy = NULL,
 123	.bind = vmw_dx_shader_bind,
 124	.unbind = vmw_dx_shader_unbind,
 125	.commit_notify = vmw_dx_shader_commit_notify,
 126};
 127
 128/**
 129 * Shader management:
 130 */
 131
 132static inline struct vmw_shader *
 133vmw_res_to_shader(struct vmw_resource *res)
 134{
 135	return container_of(res, struct vmw_shader, res);
 136}
 137
 138/**
 139 * vmw_res_to_dx_shader - typecast a struct vmw_resource to a
 140 * struct vmw_dx_shader
 141 *
 142 * @res: Pointer to the struct vmw_resource.
 143 */
 144static inline struct vmw_dx_shader *
 145vmw_res_to_dx_shader(struct vmw_resource *res)
 146{
 147	return container_of(res, struct vmw_dx_shader, res);
 148}
 149
 150static void vmw_hw_shader_destroy(struct vmw_resource *res)
 151{
 152	if (likely(res->func->destroy))
 153		(void) res->func->destroy(res);
 154	else
 155		res->id = -1;
 156}
 157
 158
 159static int vmw_gb_shader_init(struct vmw_private *dev_priv,
 160			      struct vmw_resource *res,
 161			      uint32_t size,
 162			      uint64_t offset,
 163			      SVGA3dShaderType type,
 164			      uint8_t num_input_sig,
 165			      uint8_t num_output_sig,
 166			      struct vmw_buffer_object *byte_code,
 167			      void (*res_free) (struct vmw_resource *res))
 168{
 169	struct vmw_shader *shader = vmw_res_to_shader(res);
 170	int ret;
 171
 172	ret = vmw_resource_init(dev_priv, res, true, res_free,
 173				&vmw_gb_shader_func);
 174
 175	if (unlikely(ret != 0)) {
 176		if (res_free)
 177			res_free(res);
 178		else
 179			kfree(res);
 180		return ret;
 181	}
 182
 183	res->backup_size = size;
 184	if (byte_code) {
 185		res->backup = vmw_bo_reference(byte_code);
 186		res->backup_offset = offset;
 187	}
 188	shader->size = size;
 189	shader->type = type;
 190	shader->num_input_sig = num_input_sig;
 191	shader->num_output_sig = num_output_sig;
 192
 193	res->hw_destroy = vmw_hw_shader_destroy;
 194	return 0;
 195}
 196
 197/*
 198 * GB shader code:
 199 */
 200
 201static int vmw_gb_shader_create(struct vmw_resource *res)
 202{
 203	struct vmw_private *dev_priv = res->dev_priv;
 204	struct vmw_shader *shader = vmw_res_to_shader(res);
 205	int ret;
 206	struct {
 207		SVGA3dCmdHeader header;
 208		SVGA3dCmdDefineGBShader body;
 209	} *cmd;
 210
 211	if (likely(res->id != -1))
 212		return 0;
 213
 214	ret = vmw_resource_alloc_id(res);
 215	if (unlikely(ret != 0)) {
 216		DRM_ERROR("Failed to allocate a shader id.\n");
 217		goto out_no_id;
 218	}
 219
 220	if (unlikely(res->id >= VMWGFX_NUM_GB_SHADER)) {
 221		ret = -EBUSY;
 222		goto out_no_fifo;
 223	}
 224
 225	cmd = VMW_FIFO_RESERVE(dev_priv, sizeof(*cmd));
 226	if (unlikely(cmd == NULL)) {
 
 
 227		ret = -ENOMEM;
 228		goto out_no_fifo;
 229	}
 230
 231	cmd->header.id = SVGA_3D_CMD_DEFINE_GB_SHADER;
 232	cmd->header.size = sizeof(cmd->body);
 233	cmd->body.shid = res->id;
 234	cmd->body.type = shader->type;
 235	cmd->body.sizeInBytes = shader->size;
 236	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 237	vmw_fifo_resource_inc(dev_priv);
 238
 239	return 0;
 240
 241out_no_fifo:
 242	vmw_resource_release_id(res);
 243out_no_id:
 244	return ret;
 245}
 246
 247static int vmw_gb_shader_bind(struct vmw_resource *res,
 248			      struct ttm_validate_buffer *val_buf)
 249{
 250	struct vmw_private *dev_priv = res->dev_priv;
 251	struct {
 252		SVGA3dCmdHeader header;
 253		SVGA3dCmdBindGBShader body;
 254	} *cmd;
 255	struct ttm_buffer_object *bo = val_buf->bo;
 256
 257	BUG_ON(bo->mem.mem_type != VMW_PL_MOB);
 258
 259	cmd = VMW_FIFO_RESERVE(dev_priv, sizeof(*cmd));
 260	if (unlikely(cmd == NULL))
 
 
 261		return -ENOMEM;
 
 262
 263	cmd->header.id = SVGA_3D_CMD_BIND_GB_SHADER;
 264	cmd->header.size = sizeof(cmd->body);
 265	cmd->body.shid = res->id;
 266	cmd->body.mobid = bo->mem.start;
 267	cmd->body.offsetInBytes = res->backup_offset;
 268	res->backup_dirty = false;
 269	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 270
 271	return 0;
 272}
 273
 274static int vmw_gb_shader_unbind(struct vmw_resource *res,
 275				bool readback,
 276				struct ttm_validate_buffer *val_buf)
 277{
 278	struct vmw_private *dev_priv = res->dev_priv;
 279	struct {
 280		SVGA3dCmdHeader header;
 281		SVGA3dCmdBindGBShader body;
 282	} *cmd;
 283	struct vmw_fence_obj *fence;
 284
 285	BUG_ON(res->backup->base.mem.mem_type != VMW_PL_MOB);
 286
 287	cmd = VMW_FIFO_RESERVE(dev_priv, sizeof(*cmd));
 288	if (unlikely(cmd == NULL))
 
 
 289		return -ENOMEM;
 
 290
 291	cmd->header.id = SVGA_3D_CMD_BIND_GB_SHADER;
 292	cmd->header.size = sizeof(cmd->body);
 293	cmd->body.shid = res->id;
 294	cmd->body.mobid = SVGA3D_INVALID_ID;
 295	cmd->body.offsetInBytes = 0;
 296	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 297
 298	/*
 299	 * Create a fence object and fence the backup buffer.
 300	 */
 301
 302	(void) vmw_execbuf_fence_commands(NULL, dev_priv,
 303					  &fence, NULL);
 304
 305	vmw_bo_fence_single(val_buf->bo, fence);
 306
 307	if (likely(fence != NULL))
 308		vmw_fence_obj_unreference(&fence);
 309
 310	return 0;
 311}
 312
 313static int vmw_gb_shader_destroy(struct vmw_resource *res)
 314{
 315	struct vmw_private *dev_priv = res->dev_priv;
 316	struct {
 317		SVGA3dCmdHeader header;
 318		SVGA3dCmdDestroyGBShader body;
 319	} *cmd;
 320
 321	if (likely(res->id == -1))
 322		return 0;
 323
 324	mutex_lock(&dev_priv->binding_mutex);
 325	vmw_binding_res_list_scrub(&res->binding_head);
 326
 327	cmd = VMW_FIFO_RESERVE(dev_priv, sizeof(*cmd));
 328	if (unlikely(cmd == NULL)) {
 
 
 329		mutex_unlock(&dev_priv->binding_mutex);
 330		return -ENOMEM;
 331	}
 332
 333	cmd->header.id = SVGA_3D_CMD_DESTROY_GB_SHADER;
 334	cmd->header.size = sizeof(cmd->body);
 335	cmd->body.shid = res->id;
 336	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 337	mutex_unlock(&dev_priv->binding_mutex);
 338	vmw_resource_release_id(res);
 339	vmw_fifo_resource_dec(dev_priv);
 340
 341	return 0;
 342}
 343
 344/*
 345 * DX shader code:
 346 */
 347
 348/**
 349 * vmw_dx_shader_commit_notify - Notify that a shader operation has been
 350 * committed to hardware from a user-supplied command stream.
 351 *
 352 * @res: Pointer to the shader resource.
 353 * @state: Indicating whether a creation or removal has been committed.
 354 *
 355 */
 356static void vmw_dx_shader_commit_notify(struct vmw_resource *res,
 357					enum vmw_cmdbuf_res_state state)
 358{
 359	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 360	struct vmw_private *dev_priv = res->dev_priv;
 361
 362	if (state == VMW_CMDBUF_RES_ADD) {
 363		mutex_lock(&dev_priv->binding_mutex);
 364		vmw_cotable_add_resource(shader->cotable,
 365					 &shader->cotable_head);
 366		shader->committed = true;
 367		res->id = shader->id;
 368		mutex_unlock(&dev_priv->binding_mutex);
 369	} else {
 370		mutex_lock(&dev_priv->binding_mutex);
 371		list_del_init(&shader->cotable_head);
 372		shader->committed = false;
 373		res->id = -1;
 374		mutex_unlock(&dev_priv->binding_mutex);
 375	}
 376}
 377
 378/**
 379 * vmw_dx_shader_unscrub - Have the device reattach a MOB to a DX shader.
 380 *
 381 * @res: The shader resource
 382 *
 383 * This function reverts a scrub operation.
 384 */
 385static int vmw_dx_shader_unscrub(struct vmw_resource *res)
 386{
 387	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 388	struct vmw_private *dev_priv = res->dev_priv;
 389	struct {
 390		SVGA3dCmdHeader header;
 391		SVGA3dCmdDXBindShader body;
 392	} *cmd;
 393
 394	if (!list_empty(&shader->cotable_head) || !shader->committed)
 395		return 0;
 396
 397	cmd = VMW_FIFO_RESERVE_DX(dev_priv, sizeof(*cmd), shader->ctx->id);
 398	if (unlikely(cmd == NULL))
 
 
 
 399		return -ENOMEM;
 
 400
 401	cmd->header.id = SVGA_3D_CMD_DX_BIND_SHADER;
 402	cmd->header.size = sizeof(cmd->body);
 403	cmd->body.cid = shader->ctx->id;
 404	cmd->body.shid = shader->id;
 405	cmd->body.mobid = res->backup->base.mem.start;
 406	cmd->body.offsetInBytes = res->backup_offset;
 407	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 408
 409	vmw_cotable_add_resource(shader->cotable, &shader->cotable_head);
 410
 411	return 0;
 412}
 413
 414/**
 415 * vmw_dx_shader_create - The DX shader create callback
 416 *
 417 * @res: The DX shader resource
 418 *
 419 * The create callback is called as part of resource validation and
 420 * makes sure that we unscrub the shader if it's previously been scrubbed.
 421 */
 422static int vmw_dx_shader_create(struct vmw_resource *res)
 423{
 424	struct vmw_private *dev_priv = res->dev_priv;
 425	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 426	int ret = 0;
 427
 428	WARN_ON_ONCE(!shader->committed);
 429
 430	if (vmw_resource_mob_attached(res)) {
 431		mutex_lock(&dev_priv->binding_mutex);
 432		ret = vmw_dx_shader_unscrub(res);
 433		mutex_unlock(&dev_priv->binding_mutex);
 434	}
 435
 436	res->id = shader->id;
 437	return ret;
 438}
 439
 440/**
 441 * vmw_dx_shader_bind - The DX shader bind callback
 442 *
 443 * @res: The DX shader resource
 444 * @val_buf: Pointer to the validate buffer.
 445 *
 446 */
 447static int vmw_dx_shader_bind(struct vmw_resource *res,
 448			      struct ttm_validate_buffer *val_buf)
 449{
 450	struct vmw_private *dev_priv = res->dev_priv;
 451	struct ttm_buffer_object *bo = val_buf->bo;
 452
 453	BUG_ON(bo->mem.mem_type != VMW_PL_MOB);
 454	mutex_lock(&dev_priv->binding_mutex);
 455	vmw_dx_shader_unscrub(res);
 456	mutex_unlock(&dev_priv->binding_mutex);
 457
 458	return 0;
 459}
 460
 461/**
 462 * vmw_dx_shader_scrub - Have the device unbind a MOB from a DX shader.
 463 *
 464 * @res: The shader resource
 465 *
 466 * This function unbinds a MOB from the DX shader without requiring the
 467 * MOB dma_buffer to be reserved. The driver still considers the MOB bound.
 468 * However, once the driver eventually decides to unbind the MOB, it doesn't
 469 * need to access the context.
 470 */
 471static int vmw_dx_shader_scrub(struct vmw_resource *res)
 472{
 473	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 474	struct vmw_private *dev_priv = res->dev_priv;
 475	struct {
 476		SVGA3dCmdHeader header;
 477		SVGA3dCmdDXBindShader body;
 478	} *cmd;
 479
 480	if (list_empty(&shader->cotable_head))
 481		return 0;
 482
 483	WARN_ON_ONCE(!shader->committed);
 484	cmd = VMW_FIFO_RESERVE(dev_priv, sizeof(*cmd));
 485	if (unlikely(cmd == NULL))
 
 
 486		return -ENOMEM;
 
 487
 488	cmd->header.id = SVGA_3D_CMD_DX_BIND_SHADER;
 489	cmd->header.size = sizeof(cmd->body);
 490	cmd->body.cid = shader->ctx->id;
 491	cmd->body.shid = res->id;
 492	cmd->body.mobid = SVGA3D_INVALID_ID;
 493	cmd->body.offsetInBytes = 0;
 494	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 495	res->id = -1;
 496	list_del_init(&shader->cotable_head);
 497
 498	return 0;
 499}
 500
 501/**
 502 * vmw_dx_shader_unbind - The dx shader unbind callback.
 503 *
 504 * @res: The shader resource
 505 * @readback: Whether this is a readback unbind. Currently unused.
 506 * @val_buf: MOB buffer information.
 507 */
 508static int vmw_dx_shader_unbind(struct vmw_resource *res,
 509				bool readback,
 510				struct ttm_validate_buffer *val_buf)
 511{
 512	struct vmw_private *dev_priv = res->dev_priv;
 513	struct vmw_fence_obj *fence;
 514	int ret;
 515
 516	BUG_ON(res->backup->base.mem.mem_type != VMW_PL_MOB);
 517
 518	mutex_lock(&dev_priv->binding_mutex);
 519	ret = vmw_dx_shader_scrub(res);
 520	mutex_unlock(&dev_priv->binding_mutex);
 521
 522	if (ret)
 523		return ret;
 524
 525	(void) vmw_execbuf_fence_commands(NULL, dev_priv,
 526					  &fence, NULL);
 527	vmw_bo_fence_single(val_buf->bo, fence);
 528
 529	if (likely(fence != NULL))
 530		vmw_fence_obj_unreference(&fence);
 531
 532	return 0;
 533}
 534
 535/**
 536 * vmw_dx_shader_cotable_list_scrub - The cotable unbind_func callback for
 537 * DX shaders.
 538 *
 539 * @dev_priv: Pointer to device private structure.
 540 * @list: The list of cotable resources.
 541 * @readback: Whether the call was part of a readback unbind.
 542 *
 543 * Scrubs all shader MOBs so that any subsequent shader unbind or shader
 544 * destroy operation won't need to swap in the context.
 545 */
 546void vmw_dx_shader_cotable_list_scrub(struct vmw_private *dev_priv,
 547				      struct list_head *list,
 548				      bool readback)
 549{
 550	struct vmw_dx_shader *entry, *next;
 551
 552	lockdep_assert_held_once(&dev_priv->binding_mutex);
 553
 554	list_for_each_entry_safe(entry, next, list, cotable_head) {
 555		WARN_ON(vmw_dx_shader_scrub(&entry->res));
 556		if (!readback)
 557			entry->committed = false;
 558	}
 559}
 560
 561/**
 562 * vmw_dx_shader_res_free - The DX shader free callback
 563 *
 564 * @res: The shader resource
 565 *
 566 * Frees the DX shader resource and updates memory accounting.
 567 */
 568static void vmw_dx_shader_res_free(struct vmw_resource *res)
 569{
 570	struct vmw_private *dev_priv = res->dev_priv;
 571	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 572
 573	vmw_resource_unreference(&shader->cotable);
 574	kfree(shader);
 575	ttm_mem_global_free(vmw_mem_glob(dev_priv), vmw_shader_dx_size);
 576}
 577
 578/**
 579 * vmw_dx_shader_add - Add a shader resource as a command buffer managed
 580 * resource.
 581 *
 582 * @man: The command buffer resource manager.
 583 * @ctx: Pointer to the context resource.
 584 * @user_key: The id used for this shader.
 585 * @shader_type: The shader type.
 586 * @list: The list of staged command buffer managed resources.
 587 */
 588int vmw_dx_shader_add(struct vmw_cmdbuf_res_manager *man,
 589		      struct vmw_resource *ctx,
 590		      u32 user_key,
 591		      SVGA3dShaderType shader_type,
 592		      struct list_head *list)
 593{
 594	struct vmw_dx_shader *shader;
 595	struct vmw_resource *res;
 596	struct vmw_private *dev_priv = ctx->dev_priv;
 597	struct ttm_operation_ctx ttm_opt_ctx = {
 598		.interruptible = true,
 599		.no_wait_gpu = false
 600	};
 601	int ret;
 602
 603	if (!vmw_shader_dx_size)
 604		vmw_shader_dx_size = ttm_round_pot(sizeof(*shader));
 605
 606	if (!vmw_shader_id_ok(user_key, shader_type))
 607		return -EINVAL;
 608
 609	ret = ttm_mem_global_alloc(vmw_mem_glob(dev_priv), vmw_shader_dx_size,
 610				   &ttm_opt_ctx);
 611	if (ret) {
 612		if (ret != -ERESTARTSYS)
 613			DRM_ERROR("Out of graphics memory for shader "
 614				  "creation.\n");
 615		return ret;
 616	}
 617
 618	shader = kmalloc(sizeof(*shader), GFP_KERNEL);
 619	if (!shader) {
 620		ttm_mem_global_free(vmw_mem_glob(dev_priv), vmw_shader_dx_size);
 621		return -ENOMEM;
 622	}
 623
 624	res = &shader->res;
 625	shader->ctx = ctx;
 626	shader->cotable = vmw_resource_reference
 627		(vmw_context_cotable(ctx, SVGA_COTABLE_DXSHADER));
 628	shader->id = user_key;
 629	shader->committed = false;
 630	INIT_LIST_HEAD(&shader->cotable_head);
 631	ret = vmw_resource_init(dev_priv, res, true,
 632				vmw_dx_shader_res_free, &vmw_dx_shader_func);
 633	if (ret)
 634		goto out_resource_init;
 635
 636	/*
 637	 * The user_key name-space is not per shader type for DX shaders,
 638	 * so when hashing, use a single zero shader type.
 639	 */
 640	ret = vmw_cmdbuf_res_add(man, vmw_cmdbuf_res_shader,
 641				 vmw_shader_key(user_key, 0),
 642				 res, list);
 643	if (ret)
 644		goto out_resource_init;
 645
 646	res->id = shader->id;
 647	res->hw_destroy = vmw_hw_shader_destroy;
 648
 649out_resource_init:
 650	vmw_resource_unreference(&res);
 651
 652	return ret;
 653}
 654
 655
 656
 657/**
 658 * User-space shader management:
 659 */
 660
 661static struct vmw_resource *
 662vmw_user_shader_base_to_res(struct ttm_base_object *base)
 663{
 664	return &(container_of(base, struct vmw_user_shader, base)->
 665		 shader.res);
 666}
 667
 668static void vmw_user_shader_free(struct vmw_resource *res)
 669{
 670	struct vmw_user_shader *ushader =
 671		container_of(res, struct vmw_user_shader, shader.res);
 672	struct vmw_private *dev_priv = res->dev_priv;
 673
 674	ttm_base_object_kfree(ushader, base);
 675	ttm_mem_global_free(vmw_mem_glob(dev_priv),
 676			    vmw_user_shader_size);
 677}
 678
 679static void vmw_shader_free(struct vmw_resource *res)
 680{
 681	struct vmw_shader *shader = vmw_res_to_shader(res);
 682	struct vmw_private *dev_priv = res->dev_priv;
 683
 684	kfree(shader);
 685	ttm_mem_global_free(vmw_mem_glob(dev_priv),
 686			    vmw_shader_size);
 687}
 688
 689/**
 690 * This function is called when user space has no more references on the
 691 * base object. It releases the base-object's reference on the resource object.
 692 */
 693
 694static void vmw_user_shader_base_release(struct ttm_base_object **p_base)
 695{
 696	struct ttm_base_object *base = *p_base;
 697	struct vmw_resource *res = vmw_user_shader_base_to_res(base);
 698
 699	*p_base = NULL;
 700	vmw_resource_unreference(&res);
 701}
 702
 703int vmw_shader_destroy_ioctl(struct drm_device *dev, void *data,
 704			      struct drm_file *file_priv)
 705{
 706	struct drm_vmw_shader_arg *arg = (struct drm_vmw_shader_arg *)data;
 707	struct ttm_object_file *tfile = vmw_fpriv(file_priv)->tfile;
 708
 709	return ttm_ref_object_base_unref(tfile, arg->handle,
 710					 TTM_REF_USAGE);
 711}
 712
 713static int vmw_user_shader_alloc(struct vmw_private *dev_priv,
 714				 struct vmw_buffer_object *buffer,
 715				 size_t shader_size,
 716				 size_t offset,
 717				 SVGA3dShaderType shader_type,
 718				 uint8_t num_input_sig,
 719				 uint8_t num_output_sig,
 720				 struct ttm_object_file *tfile,
 721				 u32 *handle)
 722{
 723	struct vmw_user_shader *ushader;
 724	struct vmw_resource *res, *tmp;
 725	struct ttm_operation_ctx ctx = {
 726		.interruptible = true,
 727		.no_wait_gpu = false
 728	};
 729	int ret;
 730
 
 
 
 
 731	if (unlikely(vmw_user_shader_size == 0))
 732		vmw_user_shader_size =
 733			ttm_round_pot(sizeof(struct vmw_user_shader)) +
 734			VMW_IDA_ACC_SIZE + TTM_OBJ_EXTRA_SIZE;
 735
 736	ret = ttm_mem_global_alloc(vmw_mem_glob(dev_priv),
 737				   vmw_user_shader_size,
 738				   &ctx);
 739	if (unlikely(ret != 0)) {
 740		if (ret != -ERESTARTSYS)
 741			DRM_ERROR("Out of graphics memory for shader "
 742				  "creation.\n");
 743		goto out;
 744	}
 745
 746	ushader = kzalloc(sizeof(*ushader), GFP_KERNEL);
 747	if (unlikely(!ushader)) {
 748		ttm_mem_global_free(vmw_mem_glob(dev_priv),
 749				    vmw_user_shader_size);
 750		ret = -ENOMEM;
 751		goto out;
 752	}
 753
 754	res = &ushader->shader.res;
 755	ushader->base.shareable = false;
 756	ushader->base.tfile = NULL;
 757
 758	/*
 759	 * From here on, the destructor takes over resource freeing.
 760	 */
 761
 762	ret = vmw_gb_shader_init(dev_priv, res, shader_size,
 763				 offset, shader_type, num_input_sig,
 764				 num_output_sig, buffer,
 765				 vmw_user_shader_free);
 766	if (unlikely(ret != 0))
 767		goto out;
 768
 769	tmp = vmw_resource_reference(res);
 770	ret = ttm_base_object_init(tfile, &ushader->base, false,
 771				   VMW_RES_SHADER,
 772				   &vmw_user_shader_base_release, NULL);
 773
 774	if (unlikely(ret != 0)) {
 775		vmw_resource_unreference(&tmp);
 776		goto out_err;
 777	}
 778
 779	if (handle)
 780		*handle = ushader->base.handle;
 781out_err:
 782	vmw_resource_unreference(&res);
 783out:
 784	return ret;
 785}
 786
 787
 788static struct vmw_resource *vmw_shader_alloc(struct vmw_private *dev_priv,
 789					     struct vmw_buffer_object *buffer,
 790					     size_t shader_size,
 791					     size_t offset,
 792					     SVGA3dShaderType shader_type)
 793{
 794	struct vmw_shader *shader;
 795	struct vmw_resource *res;
 796	struct ttm_operation_ctx ctx = {
 797		.interruptible = true,
 798		.no_wait_gpu = false
 799	};
 800	int ret;
 801
 
 
 
 
 802	if (unlikely(vmw_shader_size == 0))
 803		vmw_shader_size =
 804			ttm_round_pot(sizeof(struct vmw_shader)) +
 805			VMW_IDA_ACC_SIZE;
 806
 807	ret = ttm_mem_global_alloc(vmw_mem_glob(dev_priv),
 808				   vmw_shader_size,
 809				   &ctx);
 810	if (unlikely(ret != 0)) {
 811		if (ret != -ERESTARTSYS)
 812			DRM_ERROR("Out of graphics memory for shader "
 813				  "creation.\n");
 814		goto out_err;
 815	}
 816
 817	shader = kzalloc(sizeof(*shader), GFP_KERNEL);
 818	if (unlikely(!shader)) {
 819		ttm_mem_global_free(vmw_mem_glob(dev_priv),
 820				    vmw_shader_size);
 821		ret = -ENOMEM;
 822		goto out_err;
 823	}
 824
 825	res = &shader->res;
 826
 827	/*
 828	 * From here on, the destructor takes over resource freeing.
 829	 */
 830	ret = vmw_gb_shader_init(dev_priv, res, shader_size,
 831				 offset, shader_type, 0, 0, buffer,
 832				 vmw_shader_free);
 833
 834out_err:
 835	return ret ? ERR_PTR(ret) : res;
 836}
 837
 838
 839static int vmw_shader_define(struct drm_device *dev, struct drm_file *file_priv,
 840			     enum drm_vmw_shader_type shader_type_drm,
 841			     u32 buffer_handle, size_t size, size_t offset,
 842			     uint8_t num_input_sig, uint8_t num_output_sig,
 843			     uint32_t *shader_handle)
 844{
 845	struct vmw_private *dev_priv = vmw_priv(dev);
 846	struct ttm_object_file *tfile = vmw_fpriv(file_priv)->tfile;
 847	struct vmw_buffer_object *buffer = NULL;
 848	SVGA3dShaderType shader_type;
 849	int ret;
 850
 851	if (buffer_handle != SVGA3D_INVALID_ID) {
 852		ret = vmw_user_bo_lookup(tfile, buffer_handle,
 853					     &buffer, NULL);
 854		if (unlikely(ret != 0)) {
 855			VMW_DEBUG_USER("Couldn't find buffer for shader creation.\n");
 
 856			return ret;
 857		}
 858
 859		if ((u64)buffer->base.num_pages * PAGE_SIZE <
 860		    (u64)size + (u64)offset) {
 861			VMW_DEBUG_USER("Illegal buffer- or shader size.\n");
 862			ret = -EINVAL;
 863			goto out_bad_arg;
 864		}
 865	}
 866
 867	switch (shader_type_drm) {
 868	case drm_vmw_shader_type_vs:
 869		shader_type = SVGA3D_SHADERTYPE_VS;
 870		break;
 871	case drm_vmw_shader_type_ps:
 872		shader_type = SVGA3D_SHADERTYPE_PS;
 873		break;
 874	default:
 875		VMW_DEBUG_USER("Illegal shader type.\n");
 876		ret = -EINVAL;
 877		goto out_bad_arg;
 878	}
 879
 880	ret = ttm_read_lock(&dev_priv->reservation_sem, true);
 881	if (unlikely(ret != 0))
 882		goto out_bad_arg;
 883
 884	ret = vmw_user_shader_alloc(dev_priv, buffer, size, offset,
 885				    shader_type, num_input_sig,
 886				    num_output_sig, tfile, shader_handle);
 887
 888	ttm_read_unlock(&dev_priv->reservation_sem);
 889out_bad_arg:
 890	vmw_bo_unreference(&buffer);
 891	return ret;
 892}
 893
 894/**
 895 * vmw_shader_id_ok - Check whether a compat shader user key and
 896 * shader type are within valid bounds.
 897 *
 898 * @user_key: User space id of the shader.
 899 * @shader_type: Shader type.
 900 *
 901 * Returns true if valid false if not.
 902 */
 903static bool vmw_shader_id_ok(u32 user_key, SVGA3dShaderType shader_type)
 904{
 905	return user_key <= ((1 << 20) - 1) && (unsigned) shader_type < 16;
 906}
 907
 908/**
 909 * vmw_shader_key - Compute a hash key suitable for a compat shader.
 910 *
 911 * @user_key: User space id of the shader.
 912 * @shader_type: Shader type.
 913 *
 914 * Returns a hash key suitable for a command buffer managed resource
 915 * manager hash table.
 916 */
 917static u32 vmw_shader_key(u32 user_key, SVGA3dShaderType shader_type)
 918{
 919	return user_key | (shader_type << 20);
 920}
 921
 922/**
 923 * vmw_shader_remove - Stage a compat shader for removal.
 924 *
 925 * @man: Pointer to the compat shader manager identifying the shader namespace.
 926 * @user_key: The key that is used to identify the shader. The key is
 927 * unique to the shader type.
 928 * @shader_type: Shader type.
 929 * @list: Caller's list of staged command buffer resource actions.
 930 */
 931int vmw_shader_remove(struct vmw_cmdbuf_res_manager *man,
 932		      u32 user_key, SVGA3dShaderType shader_type,
 933		      struct list_head *list)
 934{
 935	struct vmw_resource *dummy;
 936
 937	if (!vmw_shader_id_ok(user_key, shader_type))
 938		return -EINVAL;
 939
 940	return vmw_cmdbuf_res_remove(man, vmw_cmdbuf_res_shader,
 941				     vmw_shader_key(user_key, shader_type),
 942				     list, &dummy);
 943}
 944
 945/**
 946 * vmw_compat_shader_add - Create a compat shader and stage it for addition
 947 * as a command buffer managed resource.
 948 *
 949 * @man: Pointer to the compat shader manager identifying the shader namespace.
 950 * @user_key: The key that is used to identify the shader. The key is
 951 * unique to the shader type.
 952 * @bytecode: Pointer to the bytecode of the shader.
 953 * @shader_type: Shader type.
 954 * @tfile: Pointer to a struct ttm_object_file that the guest-backed shader is
 955 * to be created with.
 956 * @list: Caller's list of staged command buffer resource actions.
 957 *
 958 */
 959int vmw_compat_shader_add(struct vmw_private *dev_priv,
 960			  struct vmw_cmdbuf_res_manager *man,
 961			  u32 user_key, const void *bytecode,
 962			  SVGA3dShaderType shader_type,
 963			  size_t size,
 964			  struct list_head *list)
 965{
 966	struct ttm_operation_ctx ctx = { false, true };
 967	struct vmw_buffer_object *buf;
 968	struct ttm_bo_kmap_obj map;
 969	bool is_iomem;
 970	int ret;
 971	struct vmw_resource *res;
 972
 973	if (!vmw_shader_id_ok(user_key, shader_type))
 974		return -EINVAL;
 975
 976	/* Allocate and pin a DMA buffer */
 977	buf = kzalloc(sizeof(*buf), GFP_KERNEL);
 978	if (unlikely(!buf))
 979		return -ENOMEM;
 980
 981	ret = vmw_bo_init(dev_priv, buf, size, &vmw_sys_ne_placement,
 982			      true, vmw_bo_bo_free);
 983	if (unlikely(ret != 0))
 984		goto out;
 985
 986	ret = ttm_bo_reserve(&buf->base, false, true, NULL);
 987	if (unlikely(ret != 0))
 988		goto no_reserve;
 989
 990	/* Map and copy shader bytecode. */
 991	ret = ttm_bo_kmap(&buf->base, 0, PAGE_ALIGN(size) >> PAGE_SHIFT,
 992			  &map);
 993	if (unlikely(ret != 0)) {
 994		ttm_bo_unreserve(&buf->base);
 995		goto no_reserve;
 996	}
 997
 998	memcpy(ttm_kmap_obj_virtual(&map, &is_iomem), bytecode, size);
 999	WARN_ON(is_iomem);
1000
1001	ttm_bo_kunmap(&map);
1002	ret = ttm_bo_validate(&buf->base, &vmw_sys_placement, &ctx);
1003	WARN_ON(ret != 0);
1004	ttm_bo_unreserve(&buf->base);
1005
1006	res = vmw_shader_alloc(dev_priv, buf, size, 0, shader_type);
1007	if (unlikely(ret != 0))
1008		goto no_reserve;
1009
1010	ret = vmw_cmdbuf_res_add(man, vmw_cmdbuf_res_shader,
1011				 vmw_shader_key(user_key, shader_type),
1012				 res, list);
1013	vmw_resource_unreference(&res);
1014no_reserve:
1015	vmw_bo_unreference(&buf);
1016out:
1017	return ret;
1018}
1019
1020/**
1021 * vmw_shader_lookup - Look up a compat shader
1022 *
1023 * @man: Pointer to the command buffer managed resource manager identifying
1024 * the shader namespace.
1025 * @user_key: The user space id of the shader.
1026 * @shader_type: The shader type.
1027 *
1028 * Returns a refcounted pointer to a struct vmw_resource if the shader was
1029 * found. An error pointer otherwise.
1030 */
1031struct vmw_resource *
1032vmw_shader_lookup(struct vmw_cmdbuf_res_manager *man,
1033		  u32 user_key,
1034		  SVGA3dShaderType shader_type)
1035{
1036	if (!vmw_shader_id_ok(user_key, shader_type))
1037		return ERR_PTR(-EINVAL);
1038
1039	return vmw_cmdbuf_res_lookup(man, vmw_cmdbuf_res_shader,
1040				     vmw_shader_key(user_key, shader_type));
1041}
1042
1043int vmw_shader_define_ioctl(struct drm_device *dev, void *data,
1044			     struct drm_file *file_priv)
1045{
1046	struct drm_vmw_shader_create_arg *arg =
1047		(struct drm_vmw_shader_create_arg *)data;
1048
1049	return vmw_shader_define(dev, file_priv, arg->shader_type,
1050				 arg->buffer_handle,
1051				 arg->size, arg->offset,
1052				 0, 0,
1053				 &arg->shader_handle);
1054}
v4.10.11
 
   1/**************************************************************************
   2 *
   3 * Copyright © 2009-2015 VMware, Inc., Palo Alto, CA., USA
   4 * All Rights Reserved.
   5 *
   6 * Permission is hereby granted, free of charge, to any person obtaining a
   7 * copy of this software and associated documentation files (the
   8 * "Software"), to deal in the Software without restriction, including
   9 * without limitation the rights to use, copy, modify, merge, publish,
  10 * distribute, sub license, and/or sell copies of the Software, and to
  11 * permit persons to whom the Software is furnished to do so, subject to
  12 * the following conditions:
  13 *
  14 * The above copyright notice and this permission notice (including the
  15 * next paragraph) shall be included in all copies or substantial portions
  16 * of the Software.
  17 *
  18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  21 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
  22 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  23 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  24 * USE OR OTHER DEALINGS IN THE SOFTWARE.
  25 *
  26 **************************************************************************/
  27
 
 
  28#include "vmwgfx_drv.h"
  29#include "vmwgfx_resource_priv.h"
  30#include "vmwgfx_binding.h"
  31#include "ttm/ttm_placement.h"
  32
  33struct vmw_shader {
  34	struct vmw_resource res;
  35	SVGA3dShaderType type;
  36	uint32_t size;
  37	uint8_t num_input_sig;
  38	uint8_t num_output_sig;
  39};
  40
  41struct vmw_user_shader {
  42	struct ttm_base_object base;
  43	struct vmw_shader shader;
  44};
  45
  46struct vmw_dx_shader {
  47	struct vmw_resource res;
  48	struct vmw_resource *ctx;
  49	struct vmw_resource *cotable;
  50	u32 id;
  51	bool committed;
  52	struct list_head cotable_head;
  53};
  54
  55static uint64_t vmw_user_shader_size;
  56static uint64_t vmw_shader_size;
  57static size_t vmw_shader_dx_size;
  58
  59static void vmw_user_shader_free(struct vmw_resource *res);
  60static struct vmw_resource *
  61vmw_user_shader_base_to_res(struct ttm_base_object *base);
  62
  63static int vmw_gb_shader_create(struct vmw_resource *res);
  64static int vmw_gb_shader_bind(struct vmw_resource *res,
  65			       struct ttm_validate_buffer *val_buf);
  66static int vmw_gb_shader_unbind(struct vmw_resource *res,
  67				 bool readback,
  68				 struct ttm_validate_buffer *val_buf);
  69static int vmw_gb_shader_destroy(struct vmw_resource *res);
  70
  71static int vmw_dx_shader_create(struct vmw_resource *res);
  72static int vmw_dx_shader_bind(struct vmw_resource *res,
  73			       struct ttm_validate_buffer *val_buf);
  74static int vmw_dx_shader_unbind(struct vmw_resource *res,
  75				 bool readback,
  76				 struct ttm_validate_buffer *val_buf);
  77static void vmw_dx_shader_commit_notify(struct vmw_resource *res,
  78					enum vmw_cmdbuf_res_state state);
  79static bool vmw_shader_id_ok(u32 user_key, SVGA3dShaderType shader_type);
  80static u32 vmw_shader_key(u32 user_key, SVGA3dShaderType shader_type);
  81static uint64_t vmw_user_shader_size;
  82
  83static const struct vmw_user_resource_conv user_shader_conv = {
  84	.object_type = VMW_RES_SHADER,
  85	.base_obj_to_res = vmw_user_shader_base_to_res,
  86	.res_free = vmw_user_shader_free
  87};
  88
  89const struct vmw_user_resource_conv *user_shader_converter =
  90	&user_shader_conv;
  91
  92
  93static const struct vmw_res_func vmw_gb_shader_func = {
  94	.res_type = vmw_res_shader,
  95	.needs_backup = true,
  96	.may_evict = true,
 
 
  97	.type_name = "guest backed shaders",
  98	.backup_placement = &vmw_mob_placement,
  99	.create = vmw_gb_shader_create,
 100	.destroy = vmw_gb_shader_destroy,
 101	.bind = vmw_gb_shader_bind,
 102	.unbind = vmw_gb_shader_unbind
 103};
 104
 105static const struct vmw_res_func vmw_dx_shader_func = {
 106	.res_type = vmw_res_shader,
 107	.needs_backup = true,
 108	.may_evict = false,
 
 
 109	.type_name = "dx shaders",
 110	.backup_placement = &vmw_mob_placement,
 111	.create = vmw_dx_shader_create,
 112	/*
 113	 * The destroy callback is only called with a committed resource on
 114	 * context destroy, in which case we destroy the cotable anyway,
 115	 * so there's no need to destroy DX shaders separately.
 116	 */
 117	.destroy = NULL,
 118	.bind = vmw_dx_shader_bind,
 119	.unbind = vmw_dx_shader_unbind,
 120	.commit_notify = vmw_dx_shader_commit_notify,
 121};
 122
 123/**
 124 * Shader management:
 125 */
 126
 127static inline struct vmw_shader *
 128vmw_res_to_shader(struct vmw_resource *res)
 129{
 130	return container_of(res, struct vmw_shader, res);
 131}
 132
 133/**
 134 * vmw_res_to_dx_shader - typecast a struct vmw_resource to a
 135 * struct vmw_dx_shader
 136 *
 137 * @res: Pointer to the struct vmw_resource.
 138 */
 139static inline struct vmw_dx_shader *
 140vmw_res_to_dx_shader(struct vmw_resource *res)
 141{
 142	return container_of(res, struct vmw_dx_shader, res);
 143}
 144
 145static void vmw_hw_shader_destroy(struct vmw_resource *res)
 146{
 147	if (likely(res->func->destroy))
 148		(void) res->func->destroy(res);
 149	else
 150		res->id = -1;
 151}
 152
 153
 154static int vmw_gb_shader_init(struct vmw_private *dev_priv,
 155			      struct vmw_resource *res,
 156			      uint32_t size,
 157			      uint64_t offset,
 158			      SVGA3dShaderType type,
 159			      uint8_t num_input_sig,
 160			      uint8_t num_output_sig,
 161			      struct vmw_dma_buffer *byte_code,
 162			      void (*res_free) (struct vmw_resource *res))
 163{
 164	struct vmw_shader *shader = vmw_res_to_shader(res);
 165	int ret;
 166
 167	ret = vmw_resource_init(dev_priv, res, true, res_free,
 168				&vmw_gb_shader_func);
 169
 170	if (unlikely(ret != 0)) {
 171		if (res_free)
 172			res_free(res);
 173		else
 174			kfree(res);
 175		return ret;
 176	}
 177
 178	res->backup_size = size;
 179	if (byte_code) {
 180		res->backup = vmw_dmabuf_reference(byte_code);
 181		res->backup_offset = offset;
 182	}
 183	shader->size = size;
 184	shader->type = type;
 185	shader->num_input_sig = num_input_sig;
 186	shader->num_output_sig = num_output_sig;
 187
 188	vmw_resource_activate(res, vmw_hw_shader_destroy);
 189	return 0;
 190}
 191
 192/*
 193 * GB shader code:
 194 */
 195
 196static int vmw_gb_shader_create(struct vmw_resource *res)
 197{
 198	struct vmw_private *dev_priv = res->dev_priv;
 199	struct vmw_shader *shader = vmw_res_to_shader(res);
 200	int ret;
 201	struct {
 202		SVGA3dCmdHeader header;
 203		SVGA3dCmdDefineGBShader body;
 204	} *cmd;
 205
 206	if (likely(res->id != -1))
 207		return 0;
 208
 209	ret = vmw_resource_alloc_id(res);
 210	if (unlikely(ret != 0)) {
 211		DRM_ERROR("Failed to allocate a shader id.\n");
 212		goto out_no_id;
 213	}
 214
 215	if (unlikely(res->id >= VMWGFX_NUM_GB_SHADER)) {
 216		ret = -EBUSY;
 217		goto out_no_fifo;
 218	}
 219
 220	cmd = vmw_fifo_reserve(dev_priv, sizeof(*cmd));
 221	if (unlikely(cmd == NULL)) {
 222		DRM_ERROR("Failed reserving FIFO space for shader "
 223			  "creation.\n");
 224		ret = -ENOMEM;
 225		goto out_no_fifo;
 226	}
 227
 228	cmd->header.id = SVGA_3D_CMD_DEFINE_GB_SHADER;
 229	cmd->header.size = sizeof(cmd->body);
 230	cmd->body.shid = res->id;
 231	cmd->body.type = shader->type;
 232	cmd->body.sizeInBytes = shader->size;
 233	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 234	vmw_fifo_resource_inc(dev_priv);
 235
 236	return 0;
 237
 238out_no_fifo:
 239	vmw_resource_release_id(res);
 240out_no_id:
 241	return ret;
 242}
 243
 244static int vmw_gb_shader_bind(struct vmw_resource *res,
 245			      struct ttm_validate_buffer *val_buf)
 246{
 247	struct vmw_private *dev_priv = res->dev_priv;
 248	struct {
 249		SVGA3dCmdHeader header;
 250		SVGA3dCmdBindGBShader body;
 251	} *cmd;
 252	struct ttm_buffer_object *bo = val_buf->bo;
 253
 254	BUG_ON(bo->mem.mem_type != VMW_PL_MOB);
 255
 256	cmd = vmw_fifo_reserve(dev_priv, sizeof(*cmd));
 257	if (unlikely(cmd == NULL)) {
 258		DRM_ERROR("Failed reserving FIFO space for shader "
 259			  "binding.\n");
 260		return -ENOMEM;
 261	}
 262
 263	cmd->header.id = SVGA_3D_CMD_BIND_GB_SHADER;
 264	cmd->header.size = sizeof(cmd->body);
 265	cmd->body.shid = res->id;
 266	cmd->body.mobid = bo->mem.start;
 267	cmd->body.offsetInBytes = res->backup_offset;
 268	res->backup_dirty = false;
 269	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 270
 271	return 0;
 272}
 273
 274static int vmw_gb_shader_unbind(struct vmw_resource *res,
 275				bool readback,
 276				struct ttm_validate_buffer *val_buf)
 277{
 278	struct vmw_private *dev_priv = res->dev_priv;
 279	struct {
 280		SVGA3dCmdHeader header;
 281		SVGA3dCmdBindGBShader body;
 282	} *cmd;
 283	struct vmw_fence_obj *fence;
 284
 285	BUG_ON(res->backup->base.mem.mem_type != VMW_PL_MOB);
 286
 287	cmd = vmw_fifo_reserve(dev_priv, sizeof(*cmd));
 288	if (unlikely(cmd == NULL)) {
 289		DRM_ERROR("Failed reserving FIFO space for shader "
 290			  "unbinding.\n");
 291		return -ENOMEM;
 292	}
 293
 294	cmd->header.id = SVGA_3D_CMD_BIND_GB_SHADER;
 295	cmd->header.size = sizeof(cmd->body);
 296	cmd->body.shid = res->id;
 297	cmd->body.mobid = SVGA3D_INVALID_ID;
 298	cmd->body.offsetInBytes = 0;
 299	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 300
 301	/*
 302	 * Create a fence object and fence the backup buffer.
 303	 */
 304
 305	(void) vmw_execbuf_fence_commands(NULL, dev_priv,
 306					  &fence, NULL);
 307
 308	vmw_fence_single_bo(val_buf->bo, fence);
 309
 310	if (likely(fence != NULL))
 311		vmw_fence_obj_unreference(&fence);
 312
 313	return 0;
 314}
 315
 316static int vmw_gb_shader_destroy(struct vmw_resource *res)
 317{
 318	struct vmw_private *dev_priv = res->dev_priv;
 319	struct {
 320		SVGA3dCmdHeader header;
 321		SVGA3dCmdDestroyGBShader body;
 322	} *cmd;
 323
 324	if (likely(res->id == -1))
 325		return 0;
 326
 327	mutex_lock(&dev_priv->binding_mutex);
 328	vmw_binding_res_list_scrub(&res->binding_head);
 329
 330	cmd = vmw_fifo_reserve(dev_priv, sizeof(*cmd));
 331	if (unlikely(cmd == NULL)) {
 332		DRM_ERROR("Failed reserving FIFO space for shader "
 333			  "destruction.\n");
 334		mutex_unlock(&dev_priv->binding_mutex);
 335		return -ENOMEM;
 336	}
 337
 338	cmd->header.id = SVGA_3D_CMD_DESTROY_GB_SHADER;
 339	cmd->header.size = sizeof(cmd->body);
 340	cmd->body.shid = res->id;
 341	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 342	mutex_unlock(&dev_priv->binding_mutex);
 343	vmw_resource_release_id(res);
 344	vmw_fifo_resource_dec(dev_priv);
 345
 346	return 0;
 347}
 348
 349/*
 350 * DX shader code:
 351 */
 352
 353/**
 354 * vmw_dx_shader_commit_notify - Notify that a shader operation has been
 355 * committed to hardware from a user-supplied command stream.
 356 *
 357 * @res: Pointer to the shader resource.
 358 * @state: Indicating whether a creation or removal has been committed.
 359 *
 360 */
 361static void vmw_dx_shader_commit_notify(struct vmw_resource *res,
 362					enum vmw_cmdbuf_res_state state)
 363{
 364	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 365	struct vmw_private *dev_priv = res->dev_priv;
 366
 367	if (state == VMW_CMDBUF_RES_ADD) {
 368		mutex_lock(&dev_priv->binding_mutex);
 369		vmw_cotable_add_resource(shader->cotable,
 370					 &shader->cotable_head);
 371		shader->committed = true;
 372		res->id = shader->id;
 373		mutex_unlock(&dev_priv->binding_mutex);
 374	} else {
 375		mutex_lock(&dev_priv->binding_mutex);
 376		list_del_init(&shader->cotable_head);
 377		shader->committed = false;
 378		res->id = -1;
 379		mutex_unlock(&dev_priv->binding_mutex);
 380	}
 381}
 382
 383/**
 384 * vmw_dx_shader_unscrub - Have the device reattach a MOB to a DX shader.
 385 *
 386 * @res: The shader resource
 387 *
 388 * This function reverts a scrub operation.
 389 */
 390static int vmw_dx_shader_unscrub(struct vmw_resource *res)
 391{
 392	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 393	struct vmw_private *dev_priv = res->dev_priv;
 394	struct {
 395		SVGA3dCmdHeader header;
 396		SVGA3dCmdDXBindShader body;
 397	} *cmd;
 398
 399	if (!list_empty(&shader->cotable_head) || !shader->committed)
 400		return 0;
 401
 402	cmd = vmw_fifo_reserve_dx(dev_priv, sizeof(*cmd),
 403				  shader->ctx->id);
 404	if (unlikely(cmd == NULL)) {
 405		DRM_ERROR("Failed reserving FIFO space for shader "
 406			  "scrubbing.\n");
 407		return -ENOMEM;
 408	}
 409
 410	cmd->header.id = SVGA_3D_CMD_DX_BIND_SHADER;
 411	cmd->header.size = sizeof(cmd->body);
 412	cmd->body.cid = shader->ctx->id;
 413	cmd->body.shid = shader->id;
 414	cmd->body.mobid = res->backup->base.mem.start;
 415	cmd->body.offsetInBytes = res->backup_offset;
 416	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 417
 418	vmw_cotable_add_resource(shader->cotable, &shader->cotable_head);
 419
 420	return 0;
 421}
 422
 423/**
 424 * vmw_dx_shader_create - The DX shader create callback
 425 *
 426 * @res: The DX shader resource
 427 *
 428 * The create callback is called as part of resource validation and
 429 * makes sure that we unscrub the shader if it's previously been scrubbed.
 430 */
 431static int vmw_dx_shader_create(struct vmw_resource *res)
 432{
 433	struct vmw_private *dev_priv = res->dev_priv;
 434	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 435	int ret = 0;
 436
 437	WARN_ON_ONCE(!shader->committed);
 438
 439	if (!list_empty(&res->mob_head)) {
 440		mutex_lock(&dev_priv->binding_mutex);
 441		ret = vmw_dx_shader_unscrub(res);
 442		mutex_unlock(&dev_priv->binding_mutex);
 443	}
 444
 445	res->id = shader->id;
 446	return ret;
 447}
 448
 449/**
 450 * vmw_dx_shader_bind - The DX shader bind callback
 451 *
 452 * @res: The DX shader resource
 453 * @val_buf: Pointer to the validate buffer.
 454 *
 455 */
 456static int vmw_dx_shader_bind(struct vmw_resource *res,
 457			      struct ttm_validate_buffer *val_buf)
 458{
 459	struct vmw_private *dev_priv = res->dev_priv;
 460	struct ttm_buffer_object *bo = val_buf->bo;
 461
 462	BUG_ON(bo->mem.mem_type != VMW_PL_MOB);
 463	mutex_lock(&dev_priv->binding_mutex);
 464	vmw_dx_shader_unscrub(res);
 465	mutex_unlock(&dev_priv->binding_mutex);
 466
 467	return 0;
 468}
 469
 470/**
 471 * vmw_dx_shader_scrub - Have the device unbind a MOB from a DX shader.
 472 *
 473 * @res: The shader resource
 474 *
 475 * This function unbinds a MOB from the DX shader without requiring the
 476 * MOB dma_buffer to be reserved. The driver still considers the MOB bound.
 477 * However, once the driver eventually decides to unbind the MOB, it doesn't
 478 * need to access the context.
 479 */
 480static int vmw_dx_shader_scrub(struct vmw_resource *res)
 481{
 482	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 483	struct vmw_private *dev_priv = res->dev_priv;
 484	struct {
 485		SVGA3dCmdHeader header;
 486		SVGA3dCmdDXBindShader body;
 487	} *cmd;
 488
 489	if (list_empty(&shader->cotable_head))
 490		return 0;
 491
 492	WARN_ON_ONCE(!shader->committed);
 493	cmd = vmw_fifo_reserve(dev_priv, sizeof(*cmd));
 494	if (unlikely(cmd == NULL)) {
 495		DRM_ERROR("Failed reserving FIFO space for shader "
 496			  "scrubbing.\n");
 497		return -ENOMEM;
 498	}
 499
 500	cmd->header.id = SVGA_3D_CMD_DX_BIND_SHADER;
 501	cmd->header.size = sizeof(cmd->body);
 502	cmd->body.cid = shader->ctx->id;
 503	cmd->body.shid = res->id;
 504	cmd->body.mobid = SVGA3D_INVALID_ID;
 505	cmd->body.offsetInBytes = 0;
 506	vmw_fifo_commit(dev_priv, sizeof(*cmd));
 507	res->id = -1;
 508	list_del_init(&shader->cotable_head);
 509
 510	return 0;
 511}
 512
 513/**
 514 * vmw_dx_shader_unbind - The dx shader unbind callback.
 515 *
 516 * @res: The shader resource
 517 * @readback: Whether this is a readback unbind. Currently unused.
 518 * @val_buf: MOB buffer information.
 519 */
 520static int vmw_dx_shader_unbind(struct vmw_resource *res,
 521				bool readback,
 522				struct ttm_validate_buffer *val_buf)
 523{
 524	struct vmw_private *dev_priv = res->dev_priv;
 525	struct vmw_fence_obj *fence;
 526	int ret;
 527
 528	BUG_ON(res->backup->base.mem.mem_type != VMW_PL_MOB);
 529
 530	mutex_lock(&dev_priv->binding_mutex);
 531	ret = vmw_dx_shader_scrub(res);
 532	mutex_unlock(&dev_priv->binding_mutex);
 533
 534	if (ret)
 535		return ret;
 536
 537	(void) vmw_execbuf_fence_commands(NULL, dev_priv,
 538					  &fence, NULL);
 539	vmw_fence_single_bo(val_buf->bo, fence);
 540
 541	if (likely(fence != NULL))
 542		vmw_fence_obj_unreference(&fence);
 543
 544	return 0;
 545}
 546
 547/**
 548 * vmw_dx_shader_cotable_list_scrub - The cotable unbind_func callback for
 549 * DX shaders.
 550 *
 551 * @dev_priv: Pointer to device private structure.
 552 * @list: The list of cotable resources.
 553 * @readback: Whether the call was part of a readback unbind.
 554 *
 555 * Scrubs all shader MOBs so that any subsequent shader unbind or shader
 556 * destroy operation won't need to swap in the context.
 557 */
 558void vmw_dx_shader_cotable_list_scrub(struct vmw_private *dev_priv,
 559				      struct list_head *list,
 560				      bool readback)
 561{
 562	struct vmw_dx_shader *entry, *next;
 563
 564	WARN_ON_ONCE(!mutex_is_locked(&dev_priv->binding_mutex));
 565
 566	list_for_each_entry_safe(entry, next, list, cotable_head) {
 567		WARN_ON(vmw_dx_shader_scrub(&entry->res));
 568		if (!readback)
 569			entry->committed = false;
 570	}
 571}
 572
 573/**
 574 * vmw_dx_shader_res_free - The DX shader free callback
 575 *
 576 * @res: The shader resource
 577 *
 578 * Frees the DX shader resource and updates memory accounting.
 579 */
 580static void vmw_dx_shader_res_free(struct vmw_resource *res)
 581{
 582	struct vmw_private *dev_priv = res->dev_priv;
 583	struct vmw_dx_shader *shader = vmw_res_to_dx_shader(res);
 584
 585	vmw_resource_unreference(&shader->cotable);
 586	kfree(shader);
 587	ttm_mem_global_free(vmw_mem_glob(dev_priv), vmw_shader_dx_size);
 588}
 589
 590/**
 591 * vmw_dx_shader_add - Add a shader resource as a command buffer managed
 592 * resource.
 593 *
 594 * @man: The command buffer resource manager.
 595 * @ctx: Pointer to the context resource.
 596 * @user_key: The id used for this shader.
 597 * @shader_type: The shader type.
 598 * @list: The list of staged command buffer managed resources.
 599 */
 600int vmw_dx_shader_add(struct vmw_cmdbuf_res_manager *man,
 601		      struct vmw_resource *ctx,
 602		      u32 user_key,
 603		      SVGA3dShaderType shader_type,
 604		      struct list_head *list)
 605{
 606	struct vmw_dx_shader *shader;
 607	struct vmw_resource *res;
 608	struct vmw_private *dev_priv = ctx->dev_priv;
 
 
 
 
 609	int ret;
 610
 611	if (!vmw_shader_dx_size)
 612		vmw_shader_dx_size = ttm_round_pot(sizeof(*shader));
 613
 614	if (!vmw_shader_id_ok(user_key, shader_type))
 615		return -EINVAL;
 616
 617	ret = ttm_mem_global_alloc(vmw_mem_glob(dev_priv), vmw_shader_dx_size,
 618				   false, true);
 619	if (ret) {
 620		if (ret != -ERESTARTSYS)
 621			DRM_ERROR("Out of graphics memory for shader "
 622				  "creation.\n");
 623		return ret;
 624	}
 625
 626	shader = kmalloc(sizeof(*shader), GFP_KERNEL);
 627	if (!shader) {
 628		ttm_mem_global_free(vmw_mem_glob(dev_priv), vmw_shader_dx_size);
 629		return -ENOMEM;
 630	}
 631
 632	res = &shader->res;
 633	shader->ctx = ctx;
 634	shader->cotable = vmw_context_cotable(ctx, SVGA_COTABLE_DXSHADER);
 
 635	shader->id = user_key;
 636	shader->committed = false;
 637	INIT_LIST_HEAD(&shader->cotable_head);
 638	ret = vmw_resource_init(dev_priv, res, true,
 639				vmw_dx_shader_res_free, &vmw_dx_shader_func);
 640	if (ret)
 641		goto out_resource_init;
 642
 643	/*
 644	 * The user_key name-space is not per shader type for DX shaders,
 645	 * so when hashing, use a single zero shader type.
 646	 */
 647	ret = vmw_cmdbuf_res_add(man, vmw_cmdbuf_res_shader,
 648				 vmw_shader_key(user_key, 0),
 649				 res, list);
 650	if (ret)
 651		goto out_resource_init;
 652
 653	res->id = shader->id;
 654	vmw_resource_activate(res, vmw_hw_shader_destroy);
 655
 656out_resource_init:
 657	vmw_resource_unreference(&res);
 658
 659	return ret;
 660}
 661
 662
 663
 664/**
 665 * User-space shader management:
 666 */
 667
 668static struct vmw_resource *
 669vmw_user_shader_base_to_res(struct ttm_base_object *base)
 670{
 671	return &(container_of(base, struct vmw_user_shader, base)->
 672		 shader.res);
 673}
 674
 675static void vmw_user_shader_free(struct vmw_resource *res)
 676{
 677	struct vmw_user_shader *ushader =
 678		container_of(res, struct vmw_user_shader, shader.res);
 679	struct vmw_private *dev_priv = res->dev_priv;
 680
 681	ttm_base_object_kfree(ushader, base);
 682	ttm_mem_global_free(vmw_mem_glob(dev_priv),
 683			    vmw_user_shader_size);
 684}
 685
 686static void vmw_shader_free(struct vmw_resource *res)
 687{
 688	struct vmw_shader *shader = vmw_res_to_shader(res);
 689	struct vmw_private *dev_priv = res->dev_priv;
 690
 691	kfree(shader);
 692	ttm_mem_global_free(vmw_mem_glob(dev_priv),
 693			    vmw_shader_size);
 694}
 695
 696/**
 697 * This function is called when user space has no more references on the
 698 * base object. It releases the base-object's reference on the resource object.
 699 */
 700
 701static void vmw_user_shader_base_release(struct ttm_base_object **p_base)
 702{
 703	struct ttm_base_object *base = *p_base;
 704	struct vmw_resource *res = vmw_user_shader_base_to_res(base);
 705
 706	*p_base = NULL;
 707	vmw_resource_unreference(&res);
 708}
 709
 710int vmw_shader_destroy_ioctl(struct drm_device *dev, void *data,
 711			      struct drm_file *file_priv)
 712{
 713	struct drm_vmw_shader_arg *arg = (struct drm_vmw_shader_arg *)data;
 714	struct ttm_object_file *tfile = vmw_fpriv(file_priv)->tfile;
 715
 716	return ttm_ref_object_base_unref(tfile, arg->handle,
 717					 TTM_REF_USAGE);
 718}
 719
 720static int vmw_user_shader_alloc(struct vmw_private *dev_priv,
 721				 struct vmw_dma_buffer *buffer,
 722				 size_t shader_size,
 723				 size_t offset,
 724				 SVGA3dShaderType shader_type,
 725				 uint8_t num_input_sig,
 726				 uint8_t num_output_sig,
 727				 struct ttm_object_file *tfile,
 728				 u32 *handle)
 729{
 730	struct vmw_user_shader *ushader;
 731	struct vmw_resource *res, *tmp;
 
 
 
 
 732	int ret;
 733
 734	/*
 735	 * Approximate idr memory usage with 128 bytes. It will be limited
 736	 * by maximum number_of shaders anyway.
 737	 */
 738	if (unlikely(vmw_user_shader_size == 0))
 739		vmw_user_shader_size =
 740			ttm_round_pot(sizeof(struct vmw_user_shader)) + 128;
 
 741
 742	ret = ttm_mem_global_alloc(vmw_mem_glob(dev_priv),
 743				   vmw_user_shader_size,
 744				   false, true);
 745	if (unlikely(ret != 0)) {
 746		if (ret != -ERESTARTSYS)
 747			DRM_ERROR("Out of graphics memory for shader "
 748				  "creation.\n");
 749		goto out;
 750	}
 751
 752	ushader = kzalloc(sizeof(*ushader), GFP_KERNEL);
 753	if (unlikely(ushader == NULL)) {
 754		ttm_mem_global_free(vmw_mem_glob(dev_priv),
 755				    vmw_user_shader_size);
 756		ret = -ENOMEM;
 757		goto out;
 758	}
 759
 760	res = &ushader->shader.res;
 761	ushader->base.shareable = false;
 762	ushader->base.tfile = NULL;
 763
 764	/*
 765	 * From here on, the destructor takes over resource freeing.
 766	 */
 767
 768	ret = vmw_gb_shader_init(dev_priv, res, shader_size,
 769				 offset, shader_type, num_input_sig,
 770				 num_output_sig, buffer,
 771				 vmw_user_shader_free);
 772	if (unlikely(ret != 0))
 773		goto out;
 774
 775	tmp = vmw_resource_reference(res);
 776	ret = ttm_base_object_init(tfile, &ushader->base, false,
 777				   VMW_RES_SHADER,
 778				   &vmw_user_shader_base_release, NULL);
 779
 780	if (unlikely(ret != 0)) {
 781		vmw_resource_unreference(&tmp);
 782		goto out_err;
 783	}
 784
 785	if (handle)
 786		*handle = ushader->base.hash.key;
 787out_err:
 788	vmw_resource_unreference(&res);
 789out:
 790	return ret;
 791}
 792
 793
 794static struct vmw_resource *vmw_shader_alloc(struct vmw_private *dev_priv,
 795					     struct vmw_dma_buffer *buffer,
 796					     size_t shader_size,
 797					     size_t offset,
 798					     SVGA3dShaderType shader_type)
 799{
 800	struct vmw_shader *shader;
 801	struct vmw_resource *res;
 
 
 
 
 802	int ret;
 803
 804	/*
 805	 * Approximate idr memory usage with 128 bytes. It will be limited
 806	 * by maximum number_of shaders anyway.
 807	 */
 808	if (unlikely(vmw_shader_size == 0))
 809		vmw_shader_size =
 810			ttm_round_pot(sizeof(struct vmw_shader)) + 128;
 
 811
 812	ret = ttm_mem_global_alloc(vmw_mem_glob(dev_priv),
 813				   vmw_shader_size,
 814				   false, true);
 815	if (unlikely(ret != 0)) {
 816		if (ret != -ERESTARTSYS)
 817			DRM_ERROR("Out of graphics memory for shader "
 818				  "creation.\n");
 819		goto out_err;
 820	}
 821
 822	shader = kzalloc(sizeof(*shader), GFP_KERNEL);
 823	if (unlikely(shader == NULL)) {
 824		ttm_mem_global_free(vmw_mem_glob(dev_priv),
 825				    vmw_shader_size);
 826		ret = -ENOMEM;
 827		goto out_err;
 828	}
 829
 830	res = &shader->res;
 831
 832	/*
 833	 * From here on, the destructor takes over resource freeing.
 834	 */
 835	ret = vmw_gb_shader_init(dev_priv, res, shader_size,
 836				 offset, shader_type, 0, 0, buffer,
 837				 vmw_shader_free);
 838
 839out_err:
 840	return ret ? ERR_PTR(ret) : res;
 841}
 842
 843
 844static int vmw_shader_define(struct drm_device *dev, struct drm_file *file_priv,
 845			     enum drm_vmw_shader_type shader_type_drm,
 846			     u32 buffer_handle, size_t size, size_t offset,
 847			     uint8_t num_input_sig, uint8_t num_output_sig,
 848			     uint32_t *shader_handle)
 849{
 850	struct vmw_private *dev_priv = vmw_priv(dev);
 851	struct ttm_object_file *tfile = vmw_fpriv(file_priv)->tfile;
 852	struct vmw_dma_buffer *buffer = NULL;
 853	SVGA3dShaderType shader_type;
 854	int ret;
 855
 856	if (buffer_handle != SVGA3D_INVALID_ID) {
 857		ret = vmw_user_dmabuf_lookup(tfile, buffer_handle,
 858					     &buffer, NULL);
 859		if (unlikely(ret != 0)) {
 860			DRM_ERROR("Could not find buffer for shader "
 861				  "creation.\n");
 862			return ret;
 863		}
 864
 865		if ((u64)buffer->base.num_pages * PAGE_SIZE <
 866		    (u64)size + (u64)offset) {
 867			DRM_ERROR("Illegal buffer- or shader size.\n");
 868			ret = -EINVAL;
 869			goto out_bad_arg;
 870		}
 871	}
 872
 873	switch (shader_type_drm) {
 874	case drm_vmw_shader_type_vs:
 875		shader_type = SVGA3D_SHADERTYPE_VS;
 876		break;
 877	case drm_vmw_shader_type_ps:
 878		shader_type = SVGA3D_SHADERTYPE_PS;
 879		break;
 880	default:
 881		DRM_ERROR("Illegal shader type.\n");
 882		ret = -EINVAL;
 883		goto out_bad_arg;
 884	}
 885
 886	ret = ttm_read_lock(&dev_priv->reservation_sem, true);
 887	if (unlikely(ret != 0))
 888		goto out_bad_arg;
 889
 890	ret = vmw_user_shader_alloc(dev_priv, buffer, size, offset,
 891				    shader_type, num_input_sig,
 892				    num_output_sig, tfile, shader_handle);
 893
 894	ttm_read_unlock(&dev_priv->reservation_sem);
 895out_bad_arg:
 896	vmw_dmabuf_unreference(&buffer);
 897	return ret;
 898}
 899
 900/**
 901 * vmw_shader_id_ok - Check whether a compat shader user key and
 902 * shader type are within valid bounds.
 903 *
 904 * @user_key: User space id of the shader.
 905 * @shader_type: Shader type.
 906 *
 907 * Returns true if valid false if not.
 908 */
 909static bool vmw_shader_id_ok(u32 user_key, SVGA3dShaderType shader_type)
 910{
 911	return user_key <= ((1 << 20) - 1) && (unsigned) shader_type < 16;
 912}
 913
 914/**
 915 * vmw_shader_key - Compute a hash key suitable for a compat shader.
 916 *
 917 * @user_key: User space id of the shader.
 918 * @shader_type: Shader type.
 919 *
 920 * Returns a hash key suitable for a command buffer managed resource
 921 * manager hash table.
 922 */
 923static u32 vmw_shader_key(u32 user_key, SVGA3dShaderType shader_type)
 924{
 925	return user_key | (shader_type << 20);
 926}
 927
 928/**
 929 * vmw_shader_remove - Stage a compat shader for removal.
 930 *
 931 * @man: Pointer to the compat shader manager identifying the shader namespace.
 932 * @user_key: The key that is used to identify the shader. The key is
 933 * unique to the shader type.
 934 * @shader_type: Shader type.
 935 * @list: Caller's list of staged command buffer resource actions.
 936 */
 937int vmw_shader_remove(struct vmw_cmdbuf_res_manager *man,
 938		      u32 user_key, SVGA3dShaderType shader_type,
 939		      struct list_head *list)
 940{
 941	struct vmw_resource *dummy;
 942
 943	if (!vmw_shader_id_ok(user_key, shader_type))
 944		return -EINVAL;
 945
 946	return vmw_cmdbuf_res_remove(man, vmw_cmdbuf_res_shader,
 947				     vmw_shader_key(user_key, shader_type),
 948				     list, &dummy);
 949}
 950
 951/**
 952 * vmw_compat_shader_add - Create a compat shader and stage it for addition
 953 * as a command buffer managed resource.
 954 *
 955 * @man: Pointer to the compat shader manager identifying the shader namespace.
 956 * @user_key: The key that is used to identify the shader. The key is
 957 * unique to the shader type.
 958 * @bytecode: Pointer to the bytecode of the shader.
 959 * @shader_type: Shader type.
 960 * @tfile: Pointer to a struct ttm_object_file that the guest-backed shader is
 961 * to be created with.
 962 * @list: Caller's list of staged command buffer resource actions.
 963 *
 964 */
 965int vmw_compat_shader_add(struct vmw_private *dev_priv,
 966			  struct vmw_cmdbuf_res_manager *man,
 967			  u32 user_key, const void *bytecode,
 968			  SVGA3dShaderType shader_type,
 969			  size_t size,
 970			  struct list_head *list)
 971{
 972	struct vmw_dma_buffer *buf;
 
 973	struct ttm_bo_kmap_obj map;
 974	bool is_iomem;
 975	int ret;
 976	struct vmw_resource *res;
 977
 978	if (!vmw_shader_id_ok(user_key, shader_type))
 979		return -EINVAL;
 980
 981	/* Allocate and pin a DMA buffer */
 982	buf = kzalloc(sizeof(*buf), GFP_KERNEL);
 983	if (unlikely(buf == NULL))
 984		return -ENOMEM;
 985
 986	ret = vmw_dmabuf_init(dev_priv, buf, size, &vmw_sys_ne_placement,
 987			      true, vmw_dmabuf_bo_free);
 988	if (unlikely(ret != 0))
 989		goto out;
 990
 991	ret = ttm_bo_reserve(&buf->base, false, true, NULL);
 992	if (unlikely(ret != 0))
 993		goto no_reserve;
 994
 995	/* Map and copy shader bytecode. */
 996	ret = ttm_bo_kmap(&buf->base, 0, PAGE_ALIGN(size) >> PAGE_SHIFT,
 997			  &map);
 998	if (unlikely(ret != 0)) {
 999		ttm_bo_unreserve(&buf->base);
1000		goto no_reserve;
1001	}
1002
1003	memcpy(ttm_kmap_obj_virtual(&map, &is_iomem), bytecode, size);
1004	WARN_ON(is_iomem);
1005
1006	ttm_bo_kunmap(&map);
1007	ret = ttm_bo_validate(&buf->base, &vmw_sys_placement, false, true);
1008	WARN_ON(ret != 0);
1009	ttm_bo_unreserve(&buf->base);
1010
1011	res = vmw_shader_alloc(dev_priv, buf, size, 0, shader_type);
1012	if (unlikely(ret != 0))
1013		goto no_reserve;
1014
1015	ret = vmw_cmdbuf_res_add(man, vmw_cmdbuf_res_shader,
1016				 vmw_shader_key(user_key, shader_type),
1017				 res, list);
1018	vmw_resource_unreference(&res);
1019no_reserve:
1020	vmw_dmabuf_unreference(&buf);
1021out:
1022	return ret;
1023}
1024
1025/**
1026 * vmw_shader_lookup - Look up a compat shader
1027 *
1028 * @man: Pointer to the command buffer managed resource manager identifying
1029 * the shader namespace.
1030 * @user_key: The user space id of the shader.
1031 * @shader_type: The shader type.
1032 *
1033 * Returns a refcounted pointer to a struct vmw_resource if the shader was
1034 * found. An error pointer otherwise.
1035 */
1036struct vmw_resource *
1037vmw_shader_lookup(struct vmw_cmdbuf_res_manager *man,
1038		  u32 user_key,
1039		  SVGA3dShaderType shader_type)
1040{
1041	if (!vmw_shader_id_ok(user_key, shader_type))
1042		return ERR_PTR(-EINVAL);
1043
1044	return vmw_cmdbuf_res_lookup(man, vmw_cmdbuf_res_shader,
1045				     vmw_shader_key(user_key, shader_type));
1046}
1047
1048int vmw_shader_define_ioctl(struct drm_device *dev, void *data,
1049			     struct drm_file *file_priv)
1050{
1051	struct drm_vmw_shader_create_arg *arg =
1052		(struct drm_vmw_shader_create_arg *)data;
1053
1054	return vmw_shader_define(dev, file_priv, arg->shader_type,
1055				 arg->buffer_handle,
1056				 arg->size, arg->offset,
1057				 0, 0,
1058				 &arg->shader_handle);
1059}