KVM: s390: vsie: support edat1 / edat2

If guest 2 is allowed to use edat 1 / edat 2, it can also set it up for
guest 3, so let's properly check and forward the edat cpuflags.

Acked-by: Christian Borntraeger <borntraeger@de.ibm.com>
Signed-off-by: David Hildenbrand <dahi@linux.vnet.ibm.com>
Signed-off-by: Christian Borntraeger <borntraeger@de.ibm.com>

1 files changed, 7 insertions, 0 deletions

diff --git a/arch/s390/kvm/vsie.c b/arch/s390/kvm/vsie.c
index 1165baf78535..7c9835b0a33f 100644
--- a/arch/s390/kvm/vsie.c
+++ b/arch/s390/kvm/vsie.c
@@ -97,6 +97,13 @@ static int prepare_cpuflags(struct kvm_vcpu *vcpu, struct vsie_page *vsie_page)
 
         /* intervention requests will be set later */
         newflags = CPUSTAT_ZARCH;
+        if (cpuflags & CPUSTAT_GED && test_kvm_facility(vcpu->kvm, 8))
+                newflags |= CPUSTAT_GED;
+        if (cpuflags & CPUSTAT_GED2 && test_kvm_facility(vcpu->kvm, 78)) {
+                if (cpuflags & CPUSTAT_GED)
+                        return set_validity_icpt(scb_s, 0x0001U);
+                newflags |= CPUSTAT_GED2;
+        }
 
         atomic_set(&scb_s->cpuflags, newflags);
         return 0;