110 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			110 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* SPDX-License-Identifier: GPL-2.0 */
 | |
| /*
 | |
|  * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 | |
|  */
 | |
| 
 | |
| #ifndef _NE_MISC_DEV_H_
 | |
| #define _NE_MISC_DEV_H_
 | |
| 
 | |
| #include <linux/cpumask.h>
 | |
| #include <linux/list.h>
 | |
| #include <linux/miscdevice.h>
 | |
| #include <linux/mm.h>
 | |
| #include <linux/mutex.h>
 | |
| #include <linux/pci.h>
 | |
| #include <linux/wait.h>
 | |
| 
 | |
| #include "ne_pci_dev.h"
 | |
| 
 | |
| /**
 | |
|  * struct ne_mem_region - Entry in the enclave user space memory regions list.
 | |
|  * @mem_region_list_entry:	Entry in the list of enclave memory regions.
 | |
|  * @memory_size:		Size of the user space memory region.
 | |
|  * @nr_pages:			Number of pages that make up the memory region.
 | |
|  * @pages:			Pages that make up the user space memory region.
 | |
|  * @userspace_addr:		User space address of the memory region.
 | |
|  */
 | |
| struct ne_mem_region {
 | |
| 	struct list_head	mem_region_list_entry;
 | |
| 	u64			memory_size;
 | |
| 	unsigned long		nr_pages;
 | |
| 	struct page		**pages;
 | |
| 	u64			userspace_addr;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * struct ne_enclave - Per-enclave data used for enclave lifetime management.
 | |
|  * @enclave_info_mutex :	Mutex for accessing this internal state.
 | |
|  * @enclave_list_entry :	Entry in the list of created enclaves.
 | |
|  * @eventq:			Wait queue used for out-of-band event notifications
 | |
|  *				triggered from the PCI device event handler to
 | |
|  *				the enclave process via the poll function.
 | |
|  * @has_event:			Variable used to determine if the out-of-band event
 | |
|  *				was triggered.
 | |
|  * @max_mem_regions:		The maximum number of memory regions that can be
 | |
|  *				handled by the hypervisor.
 | |
|  * @mem_regions_list:		Enclave user space memory regions list.
 | |
|  * @mem_size:			Enclave memory size.
 | |
|  * @mm :			Enclave process abstraction mm data struct.
 | |
|  * @nr_mem_regions:		Number of memory regions associated with the enclave.
 | |
|  * @nr_parent_vm_cores :	The size of the threads per core array. The
 | |
|  *				total number of CPU cores available on the
 | |
|  *				parent / primary VM.
 | |
|  * @nr_threads_per_core:	The number of threads that a full CPU core has.
 | |
|  * @nr_vcpus:			Number of vcpus associated with the enclave.
 | |
|  * @numa_node:			NUMA node of the enclave memory and CPUs.
 | |
|  * @slot_uid:			Slot unique id mapped to the enclave.
 | |
|  * @state:			Enclave state, updated during enclave lifetime.
 | |
|  * @threads_per_core:		Enclave full CPU cores array, indexed by core id,
 | |
|  *				consisting of cpumasks with all their threads.
 | |
|  *				Full CPU cores are taken from the NE CPU pool
 | |
|  *				and are available to the enclave.
 | |
|  * @vcpu_ids:			Cpumask of the vCPUs that are set for the enclave.
 | |
|  */
 | |
| struct ne_enclave {
 | |
| 	struct mutex		enclave_info_mutex;
 | |
| 	struct list_head	enclave_list_entry;
 | |
| 	wait_queue_head_t	eventq;
 | |
| 	bool			has_event;
 | |
| 	u64			max_mem_regions;
 | |
| 	struct list_head	mem_regions_list;
 | |
| 	u64			mem_size;
 | |
| 	struct mm_struct	*mm;
 | |
| 	unsigned int		nr_mem_regions;
 | |
| 	unsigned int		nr_parent_vm_cores;
 | |
| 	unsigned int		nr_threads_per_core;
 | |
| 	unsigned int		nr_vcpus;
 | |
| 	int			numa_node;
 | |
| 	u64			slot_uid;
 | |
| 	u16			state;
 | |
| 	cpumask_var_t		*threads_per_core;
 | |
| 	cpumask_var_t		vcpu_ids;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * enum ne_state - States available for an enclave.
 | |
|  * @NE_STATE_INIT:	The enclave has not been started yet.
 | |
|  * @NE_STATE_RUNNING:	The enclave was started and is running as expected.
 | |
|  * @NE_STATE_STOPPED:	The enclave exited without userspace interaction.
 | |
|  */
 | |
| enum ne_state {
 | |
| 	NE_STATE_INIT		= 0,
 | |
| 	NE_STATE_RUNNING	= 2,
 | |
| 	NE_STATE_STOPPED	= U16_MAX,
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * struct ne_devs - Data structure to keep refs to the NE misc and PCI devices.
 | |
|  * @ne_misc_dev:	Nitro Enclaves misc device.
 | |
|  * @ne_pci_dev :	Nitro Enclaves PCI device.
 | |
|  */
 | |
| struct ne_devs {
 | |
| 	struct miscdevice	*ne_misc_dev;
 | |
| 	struct ne_pci_dev	*ne_pci_dev;
 | |
| };
 | |
| 
 | |
| /* Nitro Enclaves (NE) data structure for keeping refs to the NE misc and PCI devices. */
 | |
| extern struct ne_devs ne_devs;
 | |
| 
 | |
| #endif /* _NE_MISC_DEV_H_ */
 |