1 // SPDX-License-Identifier: MIT 2 /* 3 * Copyright © 2025 Intel Corporation 4 */ 5 6 #include <drm/intel/xe_sriov_vfio.h> 7 #include <linux/cleanup.h> 8 9 #include "xe_pci.h" 10 #include "xe_pm.h" 11 #include "xe_sriov_pf_control.h" 12 #include "xe_sriov_pf_helpers.h" 13 #include "xe_sriov_pf_migration.h" 14 15 struct xe_device *xe_sriov_vfio_get_pf(struct pci_dev *pdev) 16 { 17 return xe_pci_to_pf_device(pdev); 18 } 19 EXPORT_SYMBOL_FOR_MODULES(xe_sriov_vfio_get_pf, "xe-vfio-pci"); 20 21 bool xe_sriov_vfio_migration_supported(struct xe_device *xe) 22 { 23 if (!IS_SRIOV_PF(xe)) 24 return false; 25 26 return xe_sriov_pf_migration_supported(xe); 27 } 28 EXPORT_SYMBOL_FOR_MODULES(xe_sriov_vfio_migration_supported, "xe-vfio-pci"); 29 30 #define DEFINE_XE_SRIOV_VFIO_FUNCTION(_type, _func, _impl) \ 31 _type xe_sriov_vfio_##_func(struct xe_device *xe, unsigned int vfid) \ 32 { \ 33 if (!IS_SRIOV_PF(xe)) \ 34 return -EPERM; \ 35 if (vfid == PFID || vfid > xe_sriov_pf_num_vfs(xe)) \ 36 return -EINVAL; \ 37 \ 38 guard(xe_pm_runtime_noresume)(xe); \ 39 \ 40 return xe_sriov_pf_##_impl(xe, vfid); \ 41 } \ 42 EXPORT_SYMBOL_FOR_MODULES(xe_sriov_vfio_##_func, "xe-vfio-pci") 43 44 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, wait_flr_done, control_wait_flr); 45 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, flr_prepare, control_prepare_flr); 46 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, suspend_device, control_pause_vf); 47 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, resume_device, control_resume_vf); 48 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, stop_copy_enter, control_trigger_save_vf); 49 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, stop_copy_exit, control_finish_save_vf); 50 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, resume_data_enter, control_trigger_restore_vf); 51 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, resume_data_exit, control_finish_restore_vf); 52 DEFINE_XE_SRIOV_VFIO_FUNCTION(int, error, control_stop_vf); 53 DEFINE_XE_SRIOV_VFIO_FUNCTION(ssize_t, stop_copy_size, migration_size); 54 55 ssize_t xe_sriov_vfio_data_read(struct xe_device *xe, unsigned int vfid, 56 char __user *buf, size_t len) 57 { 58 if (!IS_SRIOV_PF(xe)) 59 return -EPERM; 60 if (vfid == PFID || vfid > xe_sriov_pf_num_vfs(xe)) 61 return -EINVAL; 62 63 guard(xe_pm_runtime_noresume)(xe); 64 65 return xe_sriov_pf_migration_read(xe, vfid, buf, len); 66 } 67 EXPORT_SYMBOL_FOR_MODULES(xe_sriov_vfio_data_read, "xe-vfio-pci"); 68 69 ssize_t xe_sriov_vfio_data_write(struct xe_device *xe, unsigned int vfid, 70 const char __user *buf, size_t len) 71 { 72 if (!IS_SRIOV_PF(xe)) 73 return -EPERM; 74 if (vfid == PFID || vfid > xe_sriov_pf_num_vfs(xe)) 75 return -EINVAL; 76 77 guard(xe_pm_runtime_noresume)(xe); 78 79 return xe_sriov_pf_migration_write(xe, vfid, buf, len); 80 } 81 EXPORT_SYMBOL_FOR_MODULES(xe_sriov_vfio_data_write, "xe-vfio-pci"); 82