All skey functions call skey_check_enable at their start, which checks if we are in the PSTATE and injects a privileged operation exception if we are. Unfortunately they continue processing afterwards and perform the operation anyhow as skey_check_enable does not deliver an error if the exception injection was successful. Let's move the PSTATE check into the skey functions and exit them on such an occasion, also we now do not enable skey handling anymore in such a case.
Unified · Split
| @@ -232,8 +232,6 @@ static int try_handle_skey(struct kvm_vcpu *vcpu) · 231 unmodified lines | ||
| 232 | 232 | VCPU_EVENT(vcpu, 4, "%s", "retrying storage key operation"); |
| 233 | 233 | return -EAGAIN; |
| 234 | 234 | } |
| 235 | − if (vcpu->arch.sie_block->gpsw.mask & PSW_MASK_PSTATE) | |
| 236 | − return kvm_s390_inject_program_int(vcpu, PGM_PRIVILEGED_OP); | |
| 237 | 235 | return 0; |
| 238 | 236 | } |
| 239 | 237 | |
| @@ -244,6 +242,9 @@ static int handle_iske(struct kvm_vcpu *vcpu) · 4 unmodified lines | ||
| 244 | 242 | int reg1, reg2; |
| 245 | 243 | int rc; |
| 246 | 244 | |
| 245 | + if (vcpu->arch.sie_block->gpsw.mask & PSW_MASK_PSTATE) | |
| 246 | + return kvm_s390_inject_program_int(vcpu, PGM_PRIVILEGED_OP); | |
| 247 | + | |
| 247 | 248 | rc = try_handle_skey(vcpu); |
| 248 | 249 | if (rc) |
| 249 | 250 | return rc != -EAGAIN ? rc : 0; |
| @@ -273,6 +274,9 @@ static int handle_rrbe(struct kvm_vcpu *vcpu) · 23 unmodified lines | ||
| 273 | 274 | int reg1, reg2; |
| 274 | 275 | int rc; |
| 275 | 276 | |
| 277 | + if (vcpu->arch.sie_block->gpsw.mask & PSW_MASK_PSTATE) | |
| 278 | + return kvm_s390_inject_program_int(vcpu, PGM_PRIVILEGED_OP); | |
| 279 | + | |
| 276 | 280 | rc = try_handle_skey(vcpu); |
| 277 | 281 | if (rc) |
| 278 | 282 | return rc != -EAGAIN ? rc : 0; |
| @@ -308,6 +312,9 @@ static int handle_sske(struct kvm_vcpu *vcpu) · 29 unmodified lines | ||
| 308 | 312 | int reg1, reg2; |
| 309 | 313 | int rc; |
| 310 | 314 | |
| 315 | + if (vcpu->arch.sie_block->gpsw.mask & PSW_MASK_PSTATE) | |
| 316 | + return kvm_s390_inject_program_int(vcpu, PGM_PRIVILEGED_OP); | |
| 317 | + | |
| 311 | 318 | rc = try_handle_skey(vcpu); |
| 312 | 319 | if (rc) |
| 313 | 320 | return rc != -EAGAIN ? rc : 0; |