1af585b92SGleb Natapov /* 2af585b92SGleb Natapov * kvm asynchronous fault support 3af585b92SGleb Natapov * 4af585b92SGleb Natapov * Copyright 2010 Red Hat, Inc. 5af585b92SGleb Natapov * 6af585b92SGleb Natapov * Author: 7af585b92SGleb Natapov * Gleb Natapov <gleb@redhat.com> 8af585b92SGleb Natapov * 9af585b92SGleb Natapov * This file is free software; you can redistribute it and/or modify 10af585b92SGleb Natapov * it under the terms of version 2 of the GNU General Public License 11af585b92SGleb Natapov * as published by the Free Software Foundation. 12af585b92SGleb Natapov * 13af585b92SGleb Natapov * This program is distributed in the hope that it will be useful, 14af585b92SGleb Natapov * but WITHOUT ANY WARRANTY; without even the implied warranty of 15af585b92SGleb Natapov * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16af585b92SGleb Natapov * GNU General Public License for more details. 17af585b92SGleb Natapov * 18af585b92SGleb Natapov * You should have received a copy of the GNU General Public License 19af585b92SGleb Natapov * along with this program; if not, write to the Free Software Foundation, 20af585b92SGleb Natapov * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. 21af585b92SGleb Natapov */ 22af585b92SGleb Natapov 23af585b92SGleb Natapov #include <linux/kvm_host.h> 24af585b92SGleb Natapov #include <linux/slab.h> 25af585b92SGleb Natapov #include <linux/module.h> 26af585b92SGleb Natapov #include <linux/mmu_context.h> 27af585b92SGleb Natapov 28af585b92SGleb Natapov #include "async_pf.h" 29af585b92SGleb Natapov #include <trace/events/kvm.h> 30af585b92SGleb Natapov 31e0ead41aSDominik Dingel static inline void kvm_async_page_present_sync(struct kvm_vcpu *vcpu, 32e0ead41aSDominik Dingel struct kvm_async_pf *work) 33e0ead41aSDominik Dingel { 34e0ead41aSDominik Dingel #ifdef CONFIG_KVM_ASYNC_PF_SYNC 35e0ead41aSDominik Dingel kvm_arch_async_page_present(vcpu, work); 36e0ead41aSDominik Dingel #endif 37e0ead41aSDominik Dingel } 38e0ead41aSDominik Dingel static inline void kvm_async_page_present_async(struct kvm_vcpu *vcpu, 39e0ead41aSDominik Dingel struct kvm_async_pf *work) 40e0ead41aSDominik Dingel { 41e0ead41aSDominik Dingel #ifndef CONFIG_KVM_ASYNC_PF_SYNC 42e0ead41aSDominik Dingel kvm_arch_async_page_present(vcpu, work); 43e0ead41aSDominik Dingel #endif 44e0ead41aSDominik Dingel } 45e0ead41aSDominik Dingel 46af585b92SGleb Natapov static struct kmem_cache *async_pf_cache; 47af585b92SGleb Natapov 48af585b92SGleb Natapov int kvm_async_pf_init(void) 49af585b92SGleb Natapov { 50af585b92SGleb Natapov async_pf_cache = KMEM_CACHE(kvm_async_pf, 0); 51af585b92SGleb Natapov 52af585b92SGleb Natapov if (!async_pf_cache) 53af585b92SGleb Natapov return -ENOMEM; 54af585b92SGleb Natapov 55af585b92SGleb Natapov return 0; 56af585b92SGleb Natapov } 57af585b92SGleb Natapov 58af585b92SGleb Natapov void kvm_async_pf_deinit(void) 59af585b92SGleb Natapov { 60af585b92SGleb Natapov kmem_cache_destroy(async_pf_cache); 61af585b92SGleb Natapov async_pf_cache = NULL; 62af585b92SGleb Natapov } 63af585b92SGleb Natapov 64af585b92SGleb Natapov void kvm_async_pf_vcpu_init(struct kvm_vcpu *vcpu) 65af585b92SGleb Natapov { 66af585b92SGleb Natapov INIT_LIST_HEAD(&vcpu->async_pf.done); 67af585b92SGleb Natapov INIT_LIST_HEAD(&vcpu->async_pf.queue); 68af585b92SGleb Natapov spin_lock_init(&vcpu->async_pf.lock); 69af585b92SGleb Natapov } 70af585b92SGleb Natapov 71af585b92SGleb Natapov static void async_pf_execute(struct work_struct *work) 72af585b92SGleb Natapov { 73af585b92SGleb Natapov struct kvm_async_pf *apf = 74af585b92SGleb Natapov container_of(work, struct kvm_async_pf, work); 75af585b92SGleb Natapov struct mm_struct *mm = apf->mm; 76af585b92SGleb Natapov struct kvm_vcpu *vcpu = apf->vcpu; 77af585b92SGleb Natapov unsigned long addr = apf->addr; 78af585b92SGleb Natapov gva_t gva = apf->gva; 798b7457efSLorenzo Stoakes int locked = 1; 80af585b92SGleb Natapov 81af585b92SGleb Natapov might_sleep(); 82af585b92SGleb Natapov 831e987790SDave Hansen /* 841e987790SDave Hansen * This work is run asynchromously to the task which owns 851e987790SDave Hansen * mm and might be done in another context, so we must 868b7457efSLorenzo Stoakes * access remotely. 871e987790SDave Hansen */ 888b7457efSLorenzo Stoakes down_read(&mm->mmap_sem); 898b7457efSLorenzo Stoakes get_user_pages_remote(NULL, mm, addr, 1, FOLL_WRITE, NULL, NULL, 908b7457efSLorenzo Stoakes &locked); 918b7457efSLorenzo Stoakes if (locked) 928b7457efSLorenzo Stoakes up_read(&mm->mmap_sem); 931e987790SDave Hansen 94e0ead41aSDominik Dingel kvm_async_page_present_sync(vcpu, apf); 95af585b92SGleb Natapov 96af585b92SGleb Natapov spin_lock(&vcpu->async_pf.lock); 97af585b92SGleb Natapov list_add_tail(&apf->link, &vcpu->async_pf.done); 9822583f0dSPaolo Bonzini apf->vcpu = NULL; 99af585b92SGleb Natapov spin_unlock(&vcpu->async_pf.lock); 100af585b92SGleb Natapov 101af585b92SGleb Natapov /* 102af585b92SGleb Natapov * apf may be freed by kvm_check_async_pf_completion() after 103af585b92SGleb Natapov * this point 104af585b92SGleb Natapov */ 105af585b92SGleb Natapov 106f2e10669Schai wen trace_kvm_async_pf_completed(addr, gva); 107af585b92SGleb Natapov 1086003a420SKosuke Tatsukawa /* 1096003a420SKosuke Tatsukawa * This memory barrier pairs with prepare_to_wait's set_current_state() 1106003a420SKosuke Tatsukawa */ 1116003a420SKosuke Tatsukawa smp_mb(); 1128577370fSMarcelo Tosatti if (swait_active(&vcpu->wq)) 1138577370fSMarcelo Tosatti swake_up(&vcpu->wq); 114af585b92SGleb Natapov 11541c22f62SOleg Nesterov mmput(mm); 116af585b92SGleb Natapov kvm_put_kvm(vcpu->kvm); 117af585b92SGleb Natapov } 118af585b92SGleb Natapov 119af585b92SGleb Natapov void kvm_clear_async_pf_completion_queue(struct kvm_vcpu *vcpu) 120af585b92SGleb Natapov { 12122583f0dSPaolo Bonzini spin_lock(&vcpu->async_pf.lock); 12222583f0dSPaolo Bonzini 123af585b92SGleb Natapov /* cancel outstanding work queue item */ 124af585b92SGleb Natapov while (!list_empty(&vcpu->async_pf.queue)) { 125af585b92SGleb Natapov struct kvm_async_pf *work = 126433da860SGeliang Tang list_first_entry(&vcpu->async_pf.queue, 127af585b92SGleb Natapov typeof(*work), queue); 128af585b92SGleb Natapov list_del(&work->queue); 1299f2ceda4SDominik Dingel 13022583f0dSPaolo Bonzini /* 13122583f0dSPaolo Bonzini * We know it's present in vcpu->async_pf.done, do 13222583f0dSPaolo Bonzini * nothing here. 13322583f0dSPaolo Bonzini */ 13422583f0dSPaolo Bonzini if (!work->vcpu) 13522583f0dSPaolo Bonzini continue; 13622583f0dSPaolo Bonzini 13722583f0dSPaolo Bonzini spin_unlock(&vcpu->async_pf.lock); 1389f2ceda4SDominik Dingel #ifdef CONFIG_KVM_ASYNC_PF_SYNC 1399f2ceda4SDominik Dingel flush_work(&work->work); 1409f2ceda4SDominik Dingel #else 14198fda169SRadim Krčmář if (cancel_work_sync(&work->work)) { 14241c22f62SOleg Nesterov mmput(work->mm); 14328b441e2SRadim Krčmář kvm_put_kvm(vcpu->kvm); /* == work->vcpu->kvm */ 144af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 145af585b92SGleb Natapov } 1469f2ceda4SDominik Dingel #endif 14722583f0dSPaolo Bonzini spin_lock(&vcpu->async_pf.lock); 14828b441e2SRadim Krčmář } 149af585b92SGleb Natapov 150af585b92SGleb Natapov while (!list_empty(&vcpu->async_pf.done)) { 151af585b92SGleb Natapov struct kvm_async_pf *work = 152433da860SGeliang Tang list_first_entry(&vcpu->async_pf.done, 153af585b92SGleb Natapov typeof(*work), link); 154af585b92SGleb Natapov list_del(&work->link); 155af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 156af585b92SGleb Natapov } 157af585b92SGleb Natapov spin_unlock(&vcpu->async_pf.lock); 158af585b92SGleb Natapov 159af585b92SGleb Natapov vcpu->async_pf.queued = 0; 160af585b92SGleb Natapov } 161af585b92SGleb Natapov 162af585b92SGleb Natapov void kvm_check_async_pf_completion(struct kvm_vcpu *vcpu) 163af585b92SGleb Natapov { 164af585b92SGleb Natapov struct kvm_async_pf *work; 165af585b92SGleb Natapov 16615096ffcSXiao Guangrong while (!list_empty_careful(&vcpu->async_pf.done) && 16715096ffcSXiao Guangrong kvm_arch_can_inject_async_page_present(vcpu)) { 168af585b92SGleb Natapov spin_lock(&vcpu->async_pf.lock); 16915096ffcSXiao Guangrong work = list_first_entry(&vcpu->async_pf.done, typeof(*work), 17015096ffcSXiao Guangrong link); 171af585b92SGleb Natapov list_del(&work->link); 172af585b92SGleb Natapov spin_unlock(&vcpu->async_pf.lock); 173af585b92SGleb Natapov 17456028d08SGleb Natapov kvm_arch_async_page_ready(vcpu, work); 1751179ba53SDominik Dingel kvm_async_page_present_async(vcpu, work); 176af585b92SGleb Natapov 177af585b92SGleb Natapov list_del(&work->queue); 178af585b92SGleb Natapov vcpu->async_pf.queued--; 179af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 180af585b92SGleb Natapov } 18115096ffcSXiao Guangrong } 182af585b92SGleb Natapov 183e0ead41aSDominik Dingel int kvm_setup_async_pf(struct kvm_vcpu *vcpu, gva_t gva, unsigned long hva, 184af585b92SGleb Natapov struct kvm_arch_async_pf *arch) 185af585b92SGleb Natapov { 186af585b92SGleb Natapov struct kvm_async_pf *work; 187af585b92SGleb Natapov 188af585b92SGleb Natapov if (vcpu->async_pf.queued >= ASYNC_PF_PER_VCPU) 189af585b92SGleb Natapov return 0; 190af585b92SGleb Natapov 191af585b92SGleb Natapov /* setup delayed work */ 192af585b92SGleb Natapov 193af585b92SGleb Natapov /* 194af585b92SGleb Natapov * do alloc nowait since if we are going to sleep anyway we 195af585b92SGleb Natapov * may as well sleep faulting in page 196af585b92SGleb Natapov */ 197d7444794SChristian Borntraeger work = kmem_cache_zalloc(async_pf_cache, GFP_NOWAIT | __GFP_NOWARN); 198af585b92SGleb Natapov if (!work) 199af585b92SGleb Natapov return 0; 200af585b92SGleb Natapov 201f2e10669Schai wen work->wakeup_all = false; 202af585b92SGleb Natapov work->vcpu = vcpu; 203af585b92SGleb Natapov work->gva = gva; 204e0ead41aSDominik Dingel work->addr = hva; 205af585b92SGleb Natapov work->arch = *arch; 206af585b92SGleb Natapov work->mm = current->mm; 2073fce371bSVegard Nossum mmget(work->mm); 208af585b92SGleb Natapov kvm_get_kvm(work->vcpu->kvm); 209af585b92SGleb Natapov 210af585b92SGleb Natapov /* this can't really happen otherwise gfn_to_pfn_async 211af585b92SGleb Natapov would succeed */ 212af585b92SGleb Natapov if (unlikely(kvm_is_error_hva(work->addr))) 213af585b92SGleb Natapov goto retry_sync; 214af585b92SGleb Natapov 215af585b92SGleb Natapov INIT_WORK(&work->work, async_pf_execute); 216af585b92SGleb Natapov if (!schedule_work(&work->work)) 217af585b92SGleb Natapov goto retry_sync; 218af585b92SGleb Natapov 219af585b92SGleb Natapov list_add_tail(&work->queue, &vcpu->async_pf.queue); 220af585b92SGleb Natapov vcpu->async_pf.queued++; 221af585b92SGleb Natapov kvm_arch_async_page_not_present(vcpu, work); 222af585b92SGleb Natapov return 1; 223af585b92SGleb Natapov retry_sync: 224af585b92SGleb Natapov kvm_put_kvm(work->vcpu->kvm); 22541c22f62SOleg Nesterov mmput(work->mm); 226af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 227af585b92SGleb Natapov return 0; 228af585b92SGleb Natapov } 229344d9588SGleb Natapov 230344d9588SGleb Natapov int kvm_async_pf_wakeup_all(struct kvm_vcpu *vcpu) 231344d9588SGleb Natapov { 232344d9588SGleb Natapov struct kvm_async_pf *work; 233344d9588SGleb Natapov 23464f638c7SXiao Guangrong if (!list_empty_careful(&vcpu->async_pf.done)) 235344d9588SGleb Natapov return 0; 236344d9588SGleb Natapov 237344d9588SGleb Natapov work = kmem_cache_zalloc(async_pf_cache, GFP_ATOMIC); 238344d9588SGleb Natapov if (!work) 239344d9588SGleb Natapov return -ENOMEM; 240344d9588SGleb Natapov 241f2e10669Schai wen work->wakeup_all = true; 242344d9588SGleb Natapov INIT_LIST_HEAD(&work->queue); /* for list_del to work */ 243344d9588SGleb Natapov 24464f638c7SXiao Guangrong spin_lock(&vcpu->async_pf.lock); 245344d9588SGleb Natapov list_add_tail(&work->link, &vcpu->async_pf.done); 24664f638c7SXiao Guangrong spin_unlock(&vcpu->async_pf.lock); 24764f638c7SXiao Guangrong 248344d9588SGleb Natapov vcpu->async_pf.queued++; 249344d9588SGleb Natapov return 0; 250344d9588SGleb Natapov } 251