1 files changed, 12 insertions, 0 deletions

diff --git a/arch/powerpc/kvm/powerpc.c b/arch/powerpc/kvm/powerpc.c
index 646bfd256e5d..9a4dd8146d39 100644
--- a/arch/powerpc/kvm/powerpc.c
+++ b/arch/powerpc/kvm/powerpc.c
@@ -151,6 +151,7 @@ int kvm_dev_ioctl_check_extension(long ext)
         case KVM_CAP_PPC_PAIRED_SINGLES:
         case KVM_CAP_PPC_UNSET_IRQ:
         case KVM_CAP_ENABLE_CAP:
+        case KVM_CAP_PPC_OSI:
                 r = 1;
                 break;
         case KVM_CAP_COALESCED_MMIO:
@@ -433,6 +434,13 @@ int kvm_arch_vcpu_ioctl_run(struct kvm_vcpu *vcpu, struct kvm_run *run)
                 if (!vcpu->arch.dcr_is_write)
                         kvmppc_complete_dcr_load(vcpu, run);
                 vcpu->arch.dcr_needed = 0;
+        } else if (vcpu->arch.osi_needed) {
+                u64 *gprs = run->osi.gprs;
+                int i;
+
+                for (i = 0; i < 32; i++)
+                        kvmppc_set_gpr(vcpu, i, gprs[i]);
+                vcpu->arch.osi_needed = 0;
         }
 
         kvmppc_core_deliver_interrupts(vcpu);
@@ -475,6 +483,10 @@ static int kvm_vcpu_ioctl_enable_cap(struct kvm_vcpu *vcpu,
                 return -EINVAL;
 
         switch (cap->cap) {
+        case KVM_CAP_PPC_OSI:
+                r = 0;
+                vcpu->arch.osi_enabled = true;
+                break;
         default:
                 r = -EINVAL;
                 break;