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 31af585b92SGleb Natapov static struct kmem_cache *async_pf_cache; 32af585b92SGleb Natapov 33af585b92SGleb Natapov int kvm_async_pf_init(void) 34af585b92SGleb Natapov { 35af585b92SGleb Natapov async_pf_cache = KMEM_CACHE(kvm_async_pf, 0); 36af585b92SGleb Natapov 37af585b92SGleb Natapov if (!async_pf_cache) 38af585b92SGleb Natapov return -ENOMEM; 39af585b92SGleb Natapov 40af585b92SGleb Natapov return 0; 41af585b92SGleb Natapov } 42af585b92SGleb Natapov 43af585b92SGleb Natapov void kvm_async_pf_deinit(void) 44af585b92SGleb Natapov { 45af585b92SGleb Natapov if (async_pf_cache) 46af585b92SGleb Natapov kmem_cache_destroy(async_pf_cache); 47af585b92SGleb Natapov async_pf_cache = NULL; 48af585b92SGleb Natapov } 49af585b92SGleb Natapov 50af585b92SGleb Natapov void kvm_async_pf_vcpu_init(struct kvm_vcpu *vcpu) 51af585b92SGleb Natapov { 52af585b92SGleb Natapov INIT_LIST_HEAD(&vcpu->async_pf.done); 53af585b92SGleb Natapov INIT_LIST_HEAD(&vcpu->async_pf.queue); 54af585b92SGleb Natapov spin_lock_init(&vcpu->async_pf.lock); 55af585b92SGleb Natapov } 56af585b92SGleb Natapov 57af585b92SGleb Natapov static void async_pf_execute(struct work_struct *work) 58af585b92SGleb Natapov { 59af585b92SGleb Natapov struct kvm_async_pf *apf = 60af585b92SGleb Natapov container_of(work, struct kvm_async_pf, work); 61af585b92SGleb Natapov struct mm_struct *mm = apf->mm; 62af585b92SGleb Natapov struct kvm_vcpu *vcpu = apf->vcpu; 63af585b92SGleb Natapov unsigned long addr = apf->addr; 64af585b92SGleb Natapov gva_t gva = apf->gva; 65af585b92SGleb Natapov 66af585b92SGleb Natapov might_sleep(); 67af585b92SGleb Natapov 68af585b92SGleb Natapov use_mm(mm); 69af585b92SGleb Natapov down_read(&mm->mmap_sem); 70f2e10669Schai wen get_user_pages(current, mm, addr, 1, 1, 0, NULL, NULL); 71af585b92SGleb Natapov up_read(&mm->mmap_sem); 72af585b92SGleb Natapov unuse_mm(mm); 73af585b92SGleb Natapov 74af585b92SGleb Natapov spin_lock(&vcpu->async_pf.lock); 75af585b92SGleb Natapov list_add_tail(&apf->link, &vcpu->async_pf.done); 76af585b92SGleb Natapov spin_unlock(&vcpu->async_pf.lock); 77af585b92SGleb Natapov 78af585b92SGleb Natapov /* 79af585b92SGleb Natapov * apf may be freed by kvm_check_async_pf_completion() after 80af585b92SGleb Natapov * this point 81af585b92SGleb Natapov */ 82af585b92SGleb Natapov 83f2e10669Schai wen trace_kvm_async_pf_completed(addr, gva); 84af585b92SGleb Natapov 85af585b92SGleb Natapov if (waitqueue_active(&vcpu->wq)) 86af585b92SGleb Natapov wake_up_interruptible(&vcpu->wq); 87af585b92SGleb Natapov 88af585b92SGleb Natapov mmdrop(mm); 89af585b92SGleb Natapov kvm_put_kvm(vcpu->kvm); 90af585b92SGleb Natapov } 91af585b92SGleb Natapov 92af585b92SGleb Natapov void kvm_clear_async_pf_completion_queue(struct kvm_vcpu *vcpu) 93af585b92SGleb Natapov { 94af585b92SGleb Natapov /* cancel outstanding work queue item */ 95af585b92SGleb Natapov while (!list_empty(&vcpu->async_pf.queue)) { 96af585b92SGleb Natapov struct kvm_async_pf *work = 97af585b92SGleb Natapov list_entry(vcpu->async_pf.queue.next, 98af585b92SGleb Natapov typeof(*work), queue); 99af585b92SGleb Natapov list_del(&work->queue); 10098fda169SRadim Krčmář if (cancel_work_sync(&work->work)) { 10128b441e2SRadim Krčmář mmdrop(work->mm); 10228b441e2SRadim Krčmář kvm_put_kvm(vcpu->kvm); /* == work->vcpu->kvm */ 103af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 104af585b92SGleb Natapov } 10528b441e2SRadim Krčmář } 106af585b92SGleb Natapov 107af585b92SGleb Natapov spin_lock(&vcpu->async_pf.lock); 108af585b92SGleb Natapov while (!list_empty(&vcpu->async_pf.done)) { 109af585b92SGleb Natapov struct kvm_async_pf *work = 110af585b92SGleb Natapov list_entry(vcpu->async_pf.done.next, 111af585b92SGleb Natapov typeof(*work), link); 112af585b92SGleb Natapov list_del(&work->link); 113af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 114af585b92SGleb Natapov } 115af585b92SGleb Natapov spin_unlock(&vcpu->async_pf.lock); 116af585b92SGleb Natapov 117af585b92SGleb Natapov vcpu->async_pf.queued = 0; 118af585b92SGleb Natapov } 119af585b92SGleb Natapov 120af585b92SGleb Natapov void kvm_check_async_pf_completion(struct kvm_vcpu *vcpu) 121af585b92SGleb Natapov { 122af585b92SGleb Natapov struct kvm_async_pf *work; 123af585b92SGleb Natapov 12415096ffcSXiao Guangrong while (!list_empty_careful(&vcpu->async_pf.done) && 12515096ffcSXiao Guangrong kvm_arch_can_inject_async_page_present(vcpu)) { 126af585b92SGleb Natapov spin_lock(&vcpu->async_pf.lock); 12715096ffcSXiao Guangrong work = list_first_entry(&vcpu->async_pf.done, typeof(*work), 12815096ffcSXiao Guangrong link); 129af585b92SGleb Natapov list_del(&work->link); 130af585b92SGleb Natapov spin_unlock(&vcpu->async_pf.lock); 131af585b92SGleb Natapov 13256028d08SGleb Natapov kvm_arch_async_page_ready(vcpu, work); 133af585b92SGleb Natapov kvm_arch_async_page_present(vcpu, work); 134af585b92SGleb Natapov 135af585b92SGleb Natapov list_del(&work->queue); 136af585b92SGleb Natapov vcpu->async_pf.queued--; 137af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 138af585b92SGleb Natapov } 13915096ffcSXiao Guangrong } 140af585b92SGleb Natapov 141af585b92SGleb Natapov int kvm_setup_async_pf(struct kvm_vcpu *vcpu, gva_t gva, gfn_t gfn, 142af585b92SGleb Natapov struct kvm_arch_async_pf *arch) 143af585b92SGleb Natapov { 144af585b92SGleb Natapov struct kvm_async_pf *work; 145af585b92SGleb Natapov 146af585b92SGleb Natapov if (vcpu->async_pf.queued >= ASYNC_PF_PER_VCPU) 147af585b92SGleb Natapov return 0; 148af585b92SGleb Natapov 149af585b92SGleb Natapov /* setup delayed work */ 150af585b92SGleb Natapov 151af585b92SGleb Natapov /* 152af585b92SGleb Natapov * do alloc nowait since if we are going to sleep anyway we 153af585b92SGleb Natapov * may as well sleep faulting in page 154af585b92SGleb Natapov */ 155af585b92SGleb Natapov work = kmem_cache_zalloc(async_pf_cache, GFP_NOWAIT); 156af585b92SGleb Natapov if (!work) 157af585b92SGleb Natapov return 0; 158af585b92SGleb Natapov 159f2e10669Schai wen work->wakeup_all = false; 160af585b92SGleb Natapov work->vcpu = vcpu; 161af585b92SGleb Natapov work->gva = gva; 162af585b92SGleb Natapov work->addr = gfn_to_hva(vcpu->kvm, gfn); 163af585b92SGleb Natapov work->arch = *arch; 164af585b92SGleb Natapov work->mm = current->mm; 165af585b92SGleb Natapov atomic_inc(&work->mm->mm_count); 166af585b92SGleb Natapov kvm_get_kvm(work->vcpu->kvm); 167af585b92SGleb Natapov 168af585b92SGleb Natapov /* this can't really happen otherwise gfn_to_pfn_async 169af585b92SGleb Natapov would succeed */ 170af585b92SGleb Natapov if (unlikely(kvm_is_error_hva(work->addr))) 171af585b92SGleb Natapov goto retry_sync; 172af585b92SGleb Natapov 173af585b92SGleb Natapov INIT_WORK(&work->work, async_pf_execute); 174af585b92SGleb Natapov if (!schedule_work(&work->work)) 175af585b92SGleb Natapov goto retry_sync; 176af585b92SGleb Natapov 177af585b92SGleb Natapov list_add_tail(&work->queue, &vcpu->async_pf.queue); 178af585b92SGleb Natapov vcpu->async_pf.queued++; 179af585b92SGleb Natapov kvm_arch_async_page_not_present(vcpu, work); 180af585b92SGleb Natapov return 1; 181af585b92SGleb Natapov retry_sync: 182af585b92SGleb Natapov kvm_put_kvm(work->vcpu->kvm); 183af585b92SGleb Natapov mmdrop(work->mm); 184af585b92SGleb Natapov kmem_cache_free(async_pf_cache, work); 185af585b92SGleb Natapov return 0; 186af585b92SGleb Natapov } 187344d9588SGleb Natapov 188344d9588SGleb Natapov int kvm_async_pf_wakeup_all(struct kvm_vcpu *vcpu) 189344d9588SGleb Natapov { 190344d9588SGleb Natapov struct kvm_async_pf *work; 191344d9588SGleb Natapov 19264f638c7SXiao Guangrong if (!list_empty_careful(&vcpu->async_pf.done)) 193344d9588SGleb Natapov return 0; 194344d9588SGleb Natapov 195344d9588SGleb Natapov work = kmem_cache_zalloc(async_pf_cache, GFP_ATOMIC); 196344d9588SGleb Natapov if (!work) 197344d9588SGleb Natapov return -ENOMEM; 198344d9588SGleb Natapov 199f2e10669Schai wen work->wakeup_all = true; 200344d9588SGleb Natapov INIT_LIST_HEAD(&work->queue); /* for list_del to work */ 201344d9588SGleb Natapov 20264f638c7SXiao Guangrong spin_lock(&vcpu->async_pf.lock); 203344d9588SGleb Natapov list_add_tail(&work->link, &vcpu->async_pf.done); 20464f638c7SXiao Guangrong spin_unlock(&vcpu->async_pf.lock); 20564f638c7SXiao Guangrong 206344d9588SGleb Natapov vcpu->async_pf.queued++; 207344d9588SGleb Natapov return 0; 208344d9588SGleb Natapov } 209