cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

amdgpu_sched.c (3408B)

      1/*
      2 * Copyright 2017 Valve Corporation
      3 *
      4 * Permission is hereby granted, free of charge, to any person obtaining a
      5 * copy of this software and associated documentation files (the "Software"),
      6 * to deal in the Software without restriction, including without limitation
      7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
      8 * and/or sell copies of the Software, and to permit persons to whom the
      9 * Software is furnished to do so, subject to the following conditions:
     10 *
     11 * The above copyright notice and this permission notice shall be included in
     12 * all copies or substantial portions of the Software.
     13 *
     14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
     15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
     16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
     17 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
     18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
     19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
     20 * OTHER DEALINGS IN THE SOFTWARE.
     21 *
     22 * Authors: Andres Rodriguez <andresx7@gmail.com>
     23 */
     24
     25#include <linux/fdtable.h>
     26#include <linux/file.h>
     27#include <linux/pid.h>
     28
     29#include <drm/amdgpu_drm.h>
     30
     31#include "amdgpu.h"
     32#include "amdgpu_sched.h"
     33#include "amdgpu_vm.h"
     34
     35static int amdgpu_sched_process_priority_override(struct amdgpu_device *adev,
     36						  int fd,
     37						  int32_t priority)
     38{
     39	struct fd f = fdget(fd);
     40	struct amdgpu_fpriv *fpriv;
     41	struct amdgpu_ctx *ctx;
     42	uint32_t id;
     43	int r;
     44
     45	if (!f.file)
     46		return -EINVAL;
     47
     48	r = amdgpu_file_to_fpriv(f.file, &fpriv);
     49	if (r) {
     50		fdput(f);
     51		return r;
     52	}
     53
     54	idr_for_each_entry(&fpriv->ctx_mgr.ctx_handles, ctx, id)
     55		amdgpu_ctx_priority_override(ctx, priority);
     56
     57	fdput(f);
     58	return 0;
     59}
     60
     61static int amdgpu_sched_context_priority_override(struct amdgpu_device *adev,
     62						  int fd,
     63						  unsigned ctx_id,
     64						  int32_t priority)
     65{
     66	struct fd f = fdget(fd);
     67	struct amdgpu_fpriv *fpriv;
     68	struct amdgpu_ctx *ctx;
     69	int r;
     70
     71	if (!f.file)
     72		return -EINVAL;
     73
     74	r = amdgpu_file_to_fpriv(f.file, &fpriv);
     75	if (r) {
     76		fdput(f);
     77		return r;
     78	}
     79
     80	ctx = amdgpu_ctx_get(fpriv, ctx_id);
     81
     82	if (!ctx) {
     83		fdput(f);
     84		return -EINVAL;
     85	}
     86
     87	amdgpu_ctx_priority_override(ctx, priority);
     88	amdgpu_ctx_put(ctx);
     89	fdput(f);
     90
     91	return 0;
     92}
     93
     94int amdgpu_sched_ioctl(struct drm_device *dev, void *data,
     95		       struct drm_file *filp)
     96{
     97	union drm_amdgpu_sched *args = data;
     98	struct amdgpu_device *adev = drm_to_adev(dev);
     99	int r;
    100
    101	/* First check the op, then the op's argument.
    102	 */
    103	switch (args->in.op) {
    104	case AMDGPU_SCHED_OP_PROCESS_PRIORITY_OVERRIDE:
    105	case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE:
    106		break;
    107	default:
    108		DRM_ERROR("Invalid sched op specified: %d\n", args->in.op);
    109		return -EINVAL;
    110	}
    111
    112	if (!amdgpu_ctx_priority_is_valid(args->in.priority)) {
    113		WARN(1, "Invalid context priority %d\n", args->in.priority);
    114		return -EINVAL;
    115	}
    116
    117	switch (args->in.op) {
    118	case AMDGPU_SCHED_OP_PROCESS_PRIORITY_OVERRIDE:
    119		r = amdgpu_sched_process_priority_override(adev,
    120							   args->in.fd,
    121							   args->in.priority);
    122		break;
    123	case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE:
    124		r = amdgpu_sched_context_priority_override(adev,
    125							   args->in.fd,
    126							   args->in.ctx_id,
    127							   args->in.priority);
    128		break;
    129	default:
    130		/* Impossible.
    131		 */
    132		r = -EINVAL;
    133		break;
    134	}
    135
    136	return r;
    137}