| // SPDX-License-Identifier: GPL-2.0 OR MIT | 
 | /* | 
 |  * Copyright 2020-2022 Advanced Micro Devices, Inc. | 
 |  * | 
 |  * Permission is hereby granted, free of charge, to any person obtaining a | 
 |  * copy of this software and associated documentation files (the "Software"), | 
 |  * to deal in the Software without restriction, including without limitation | 
 |  * the rights to use, copy, modify, merge, publish, distribute, sublicense, | 
 |  * and/or sell copies of the Software, and to permit persons to whom the | 
 |  * Software is furnished to do so, subject to the following conditions: | 
 |  * | 
 |  * The above copyright notice and this permission notice shall be included in | 
 |  * all copies or substantial portions of the Software. | 
 |  * | 
 |  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | 
 |  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | 
 |  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL | 
 |  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR | 
 |  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, | 
 |  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR | 
 |  * OTHER DEALINGS IN THE SOFTWARE. | 
 |  */ | 
 |  | 
 | #include <linux/poll.h> | 
 | #include <linux/wait.h> | 
 | #include <linux/anon_inodes.h> | 
 | #include <uapi/linux/kfd_ioctl.h> | 
 | #include "amdgpu.h" | 
 | #include "amdgpu_vm.h" | 
 | #include "kfd_priv.h" | 
 | #include "kfd_smi_events.h" | 
 |  | 
 | struct kfd_smi_client { | 
 | 	struct list_head list; | 
 | 	struct kfifo fifo; | 
 | 	wait_queue_head_t wait_queue; | 
 | 	/* events enabled */ | 
 | 	uint64_t events; | 
 | 	struct kfd_node *dev; | 
 | 	spinlock_t lock; | 
 | 	struct rcu_head rcu; | 
 | 	pid_t pid; | 
 | 	bool suser; | 
 | }; | 
 |  | 
 | #define MAX_KFIFO_SIZE	1024 | 
 |  | 
 | static __poll_t kfd_smi_ev_poll(struct file *, struct poll_table_struct *); | 
 | static ssize_t kfd_smi_ev_read(struct file *, char __user *, size_t, loff_t *); | 
 | static ssize_t kfd_smi_ev_write(struct file *, const char __user *, size_t, | 
 | 				loff_t *); | 
 | static int kfd_smi_ev_release(struct inode *, struct file *); | 
 |  | 
 | static const char kfd_smi_name[] = "kfd_smi_ev"; | 
 |  | 
 | static const struct file_operations kfd_smi_ev_fops = { | 
 | 	.owner = THIS_MODULE, | 
 | 	.poll = kfd_smi_ev_poll, | 
 | 	.read = kfd_smi_ev_read, | 
 | 	.write = kfd_smi_ev_write, | 
 | 	.release = kfd_smi_ev_release | 
 | }; | 
 |  | 
 | static __poll_t kfd_smi_ev_poll(struct file *filep, | 
 | 				struct poll_table_struct *wait) | 
 | { | 
 | 	struct kfd_smi_client *client = filep->private_data; | 
 | 	__poll_t mask = 0; | 
 |  | 
 | 	poll_wait(filep, &client->wait_queue, wait); | 
 |  | 
 | 	spin_lock(&client->lock); | 
 | 	if (!kfifo_is_empty(&client->fifo)) | 
 | 		mask = EPOLLIN | EPOLLRDNORM; | 
 | 	spin_unlock(&client->lock); | 
 |  | 
 | 	return mask; | 
 | } | 
 |  | 
 | static ssize_t kfd_smi_ev_read(struct file *filep, char __user *user, | 
 | 			       size_t size, loff_t *offset) | 
 | { | 
 | 	int ret; | 
 | 	size_t to_copy; | 
 | 	struct kfd_smi_client *client = filep->private_data; | 
 | 	unsigned char *buf; | 
 |  | 
 | 	size = min_t(size_t, size, MAX_KFIFO_SIZE); | 
 | 	buf = kmalloc(size, GFP_KERNEL); | 
 | 	if (!buf) | 
 | 		return -ENOMEM; | 
 |  | 
 | 	/* kfifo_to_user can sleep so we can't use spinlock protection around | 
 | 	 * it. Instead, we kfifo out as spinlocked then copy them to the user. | 
 | 	 */ | 
 | 	spin_lock(&client->lock); | 
 | 	to_copy = kfifo_len(&client->fifo); | 
 | 	if (!to_copy) { | 
 | 		spin_unlock(&client->lock); | 
 | 		ret = -EAGAIN; | 
 | 		goto ret_err; | 
 | 	} | 
 | 	to_copy = min(size, to_copy); | 
 | 	ret = kfifo_out(&client->fifo, buf, to_copy); | 
 | 	spin_unlock(&client->lock); | 
 | 	if (ret <= 0) { | 
 | 		ret = -EAGAIN; | 
 | 		goto ret_err; | 
 | 	} | 
 |  | 
 | 	ret = copy_to_user(user, buf, to_copy); | 
 | 	if (ret) { | 
 | 		ret = -EFAULT; | 
 | 		goto ret_err; | 
 | 	} | 
 |  | 
 | 	kfree(buf); | 
 | 	return to_copy; | 
 |  | 
 | ret_err: | 
 | 	kfree(buf); | 
 | 	return ret; | 
 | } | 
 |  | 
 | static ssize_t kfd_smi_ev_write(struct file *filep, const char __user *user, | 
 | 				size_t size, loff_t *offset) | 
 | { | 
 | 	struct kfd_smi_client *client = filep->private_data; | 
 | 	uint64_t events; | 
 |  | 
 | 	if (!access_ok(user, size) || size < sizeof(events)) | 
 | 		return -EFAULT; | 
 | 	if (copy_from_user(&events, user, sizeof(events))) | 
 | 		return -EFAULT; | 
 |  | 
 | 	WRITE_ONCE(client->events, events); | 
 |  | 
 | 	return sizeof(events); | 
 | } | 
 |  | 
 | static void kfd_smi_ev_client_free(struct rcu_head *p) | 
 | { | 
 | 	struct kfd_smi_client *ev = container_of(p, struct kfd_smi_client, rcu); | 
 |  | 
 | 	kfifo_free(&ev->fifo); | 
 | 	kfree(ev); | 
 | } | 
 |  | 
 | static int kfd_smi_ev_release(struct inode *inode, struct file *filep) | 
 | { | 
 | 	struct kfd_smi_client *client = filep->private_data; | 
 | 	struct kfd_node *dev = client->dev; | 
 |  | 
 | 	spin_lock(&dev->smi_lock); | 
 | 	list_del_rcu(&client->list); | 
 | 	spin_unlock(&dev->smi_lock); | 
 |  | 
 | 	call_rcu(&client->rcu, kfd_smi_ev_client_free); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static bool kfd_smi_ev_enabled(pid_t pid, struct kfd_smi_client *client, | 
 | 			       unsigned int event) | 
 | { | 
 | 	uint64_t all = KFD_SMI_EVENT_MASK_FROM_INDEX(KFD_SMI_EVENT_ALL_PROCESS); | 
 | 	uint64_t events = READ_ONCE(client->events); | 
 |  | 
 | 	if (pid && client->pid != pid && !(client->suser && (events & all))) | 
 | 		return false; | 
 |  | 
 | 	return events & KFD_SMI_EVENT_MASK_FROM_INDEX(event); | 
 | } | 
 |  | 
 | static void add_event_to_kfifo(pid_t pid, struct kfd_node *dev, | 
 | 			       unsigned int smi_event, char *event_msg, int len) | 
 | { | 
 | 	struct kfd_smi_client *client; | 
 |  | 
 | 	rcu_read_lock(); | 
 |  | 
 | 	list_for_each_entry_rcu(client, &dev->smi_clients, list) { | 
 | 		if (!kfd_smi_ev_enabled(pid, client, smi_event)) | 
 | 			continue; | 
 | 		spin_lock(&client->lock); | 
 | 		if (kfifo_avail(&client->fifo) >= len) { | 
 | 			kfifo_in(&client->fifo, event_msg, len); | 
 | 			wake_up_all(&client->wait_queue); | 
 | 		} else { | 
 | 			pr_debug("smi_event(EventID: %u): no space left\n", | 
 | 					smi_event); | 
 | 		} | 
 | 		spin_unlock(&client->lock); | 
 | 	} | 
 |  | 
 | 	rcu_read_unlock(); | 
 | } | 
 |  | 
 | __printf(4, 5) | 
 | static void kfd_smi_event_add(pid_t pid, struct kfd_node *dev, | 
 | 			      unsigned int event, char *fmt, ...) | 
 | { | 
 | 	char fifo_in[KFD_SMI_EVENT_MSG_SIZE]; | 
 | 	int len; | 
 | 	va_list args; | 
 |  | 
 | 	if (list_empty(&dev->smi_clients)) | 
 | 		return; | 
 |  | 
 | 	len = snprintf(fifo_in, sizeof(fifo_in), "%x ", event); | 
 |  | 
 | 	va_start(args, fmt); | 
 | 	len += vsnprintf(fifo_in + len, sizeof(fifo_in) - len, fmt, args); | 
 | 	va_end(args); | 
 |  | 
 | 	add_event_to_kfifo(pid, dev, event, fifo_in, len); | 
 | } | 
 |  | 
 | void kfd_smi_event_update_gpu_reset(struct kfd_node *dev, bool post_reset) | 
 | { | 
 | 	unsigned int event; | 
 |  | 
 | 	if (post_reset) { | 
 | 		event = KFD_SMI_EVENT_GPU_POST_RESET; | 
 | 	} else { | 
 | 		event = KFD_SMI_EVENT_GPU_PRE_RESET; | 
 | 		++(dev->reset_seq_num); | 
 | 	} | 
 | 	kfd_smi_event_add(0, dev, event, "%x\n", dev->reset_seq_num); | 
 | } | 
 |  | 
 | void kfd_smi_event_update_thermal_throttling(struct kfd_node *dev, | 
 | 					     uint64_t throttle_bitmask) | 
 | { | 
 | 	kfd_smi_event_add(0, dev, KFD_SMI_EVENT_THERMAL_THROTTLE, "%llx:%llx\n", | 
 | 			  throttle_bitmask, | 
 | 			  amdgpu_dpm_get_thermal_throttling_counter(dev->adev)); | 
 | } | 
 |  | 
 | void kfd_smi_event_update_vmfault(struct kfd_node *dev, uint16_t pasid) | 
 | { | 
 | 	struct amdgpu_task_info task_info; | 
 |  | 
 | 	memset(&task_info, 0, sizeof(struct amdgpu_task_info)); | 
 | 	amdgpu_vm_get_task_info(dev->adev, pasid, &task_info); | 
 | 	/* Report VM faults from user applications, not retry from kernel */ | 
 | 	if (!task_info.pid) | 
 | 		return; | 
 |  | 
 | 	kfd_smi_event_add(0, dev, KFD_SMI_EVENT_VMFAULT, "%x:%s\n", | 
 | 			  task_info.pid, task_info.task_name); | 
 | } | 
 |  | 
 | void kfd_smi_event_page_fault_start(struct kfd_node *node, pid_t pid, | 
 | 				    unsigned long address, bool write_fault, | 
 | 				    ktime_t ts) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_PAGE_FAULT_START, | 
 | 			  "%lld -%d @%lx(%x) %c\n", ktime_to_ns(ts), pid, | 
 | 			  address, node->id, write_fault ? 'W' : 'R'); | 
 | } | 
 |  | 
 | void kfd_smi_event_page_fault_end(struct kfd_node *node, pid_t pid, | 
 | 				  unsigned long address, bool migration) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_PAGE_FAULT_END, | 
 | 			  "%lld -%d @%lx(%x) %c\n", ktime_get_boottime_ns(), | 
 | 			  pid, address, node->id, migration ? 'M' : 'U'); | 
 | } | 
 |  | 
 | void kfd_smi_event_migration_start(struct kfd_node *node, pid_t pid, | 
 | 				   unsigned long start, unsigned long end, | 
 | 				   uint32_t from, uint32_t to, | 
 | 				   uint32_t prefetch_loc, uint32_t preferred_loc, | 
 | 				   uint32_t trigger) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_MIGRATE_START, | 
 | 			  "%lld -%d @%lx(%lx) %x->%x %x:%x %d\n", | 
 | 			  ktime_get_boottime_ns(), pid, start, end - start, | 
 | 			  from, to, prefetch_loc, preferred_loc, trigger); | 
 | } | 
 |  | 
 | void kfd_smi_event_migration_end(struct kfd_node *node, pid_t pid, | 
 | 				 unsigned long start, unsigned long end, | 
 | 				 uint32_t from, uint32_t to, uint32_t trigger) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_MIGRATE_END, | 
 | 			  "%lld -%d @%lx(%lx) %x->%x %d\n", | 
 | 			  ktime_get_boottime_ns(), pid, start, end - start, | 
 | 			  from, to, trigger); | 
 | } | 
 |  | 
 | void kfd_smi_event_queue_eviction(struct kfd_node *node, pid_t pid, | 
 | 				  uint32_t trigger) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_QUEUE_EVICTION, | 
 | 			  "%lld -%d %x %d\n", ktime_get_boottime_ns(), pid, | 
 | 			  node->id, trigger); | 
 | } | 
 |  | 
 | void kfd_smi_event_queue_restore(struct kfd_node *node, pid_t pid) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_QUEUE_RESTORE, | 
 | 			  "%lld -%d %x\n", ktime_get_boottime_ns(), pid, | 
 | 			  node->id); | 
 | } | 
 |  | 
 | void kfd_smi_event_queue_restore_rescheduled(struct mm_struct *mm) | 
 | { | 
 | 	struct kfd_process *p; | 
 | 	int i; | 
 |  | 
 | 	p = kfd_lookup_process_by_mm(mm); | 
 | 	if (!p) | 
 | 		return; | 
 |  | 
 | 	for (i = 0; i < p->n_pdds; i++) { | 
 | 		struct kfd_process_device *pdd = p->pdds[i]; | 
 |  | 
 | 		kfd_smi_event_add(p->lead_thread->pid, pdd->dev, | 
 | 				  KFD_SMI_EVENT_QUEUE_RESTORE, | 
 | 				  "%lld -%d %x %c\n", ktime_get_boottime_ns(), | 
 | 				  p->lead_thread->pid, pdd->dev->id, 'R'); | 
 | 	} | 
 | 	kfd_unref_process(p); | 
 | } | 
 |  | 
 | void kfd_smi_event_unmap_from_gpu(struct kfd_node *node, pid_t pid, | 
 | 				  unsigned long address, unsigned long last, | 
 | 				  uint32_t trigger) | 
 | { | 
 | 	kfd_smi_event_add(pid, node, KFD_SMI_EVENT_UNMAP_FROM_GPU, | 
 | 			  "%lld -%d @%lx(%lx) %x %d\n", ktime_get_boottime_ns(), | 
 | 			  pid, address, last - address + 1, node->id, trigger); | 
 | } | 
 |  | 
 | int kfd_smi_event_open(struct kfd_node *dev, uint32_t *fd) | 
 | { | 
 | 	struct kfd_smi_client *client; | 
 | 	int ret; | 
 |  | 
 | 	client = kzalloc(sizeof(struct kfd_smi_client), GFP_KERNEL); | 
 | 	if (!client) | 
 | 		return -ENOMEM; | 
 | 	INIT_LIST_HEAD(&client->list); | 
 |  | 
 | 	ret = kfifo_alloc(&client->fifo, MAX_KFIFO_SIZE, GFP_KERNEL); | 
 | 	if (ret) { | 
 | 		kfree(client); | 
 | 		return ret; | 
 | 	} | 
 |  | 
 | 	init_waitqueue_head(&client->wait_queue); | 
 | 	spin_lock_init(&client->lock); | 
 | 	client->events = 0; | 
 | 	client->dev = dev; | 
 | 	client->pid = current->tgid; | 
 | 	client->suser = capable(CAP_SYS_ADMIN); | 
 |  | 
 | 	spin_lock(&dev->smi_lock); | 
 | 	list_add_rcu(&client->list, &dev->smi_clients); | 
 | 	spin_unlock(&dev->smi_lock); | 
 |  | 
 | 	ret = anon_inode_getfd(kfd_smi_name, &kfd_smi_ev_fops, (void *)client, | 
 | 			       O_RDWR); | 
 | 	if (ret < 0) { | 
 | 		spin_lock(&dev->smi_lock); | 
 | 		list_del_rcu(&client->list); | 
 | 		spin_unlock(&dev->smi_lock); | 
 |  | 
 | 		synchronize_rcu(); | 
 |  | 
 | 		kfifo_free(&client->fifo); | 
 | 		kfree(client); | 
 | 		return ret; | 
 | 	} | 
 | 	*fd = ret; | 
 |  | 
 | 	return 0; | 
 | } |