X-Git-Url: https://rtime.felk.cvut.cz/gitweb/jailhouse.git/blobdiff_plain/4e0ea74420900c7a66d265117b09fddbc3aaa754..f053c948f5b6fe7b9ec5cf6533e177fd42113ba3:/hypervisor/arch/arm/traps.c

diff --git a/hypervisor/arch/arm/traps.c b/hypervisor/arch/arm/traps.c
index 71246e9..ae0990f 100644
--- a/hypervisor/arch/arm/traps.c
+++ b/hypervisor/arch/arm/traps.c
@@ -209,7 +209,7 @@ static int arch_handle_smc(struct trap_context *ctx)
 {
 	unsigned long *regs = ctx->regs;
 
-	if (IS_PSCI_FN(regs[0]))
+	if (IS_PSCI_32(regs[0]) || IS_PSCI_UBOOT(regs[0]))
 		regs[0] = psci_dispatch(ctx);
 	else
 		regs[0] = smc(regs[0], regs[1], regs[2], regs[3]);
@@ -223,7 +223,7 @@ static int arch_handle_hvc(struct trap_context *ctx)
 {
 	unsigned long *regs = ctx->regs;
 
-	if (IS_PSCI_FN(regs[0]))
+	if (IS_PSCI_32(regs[0]) || IS_PSCI_UBOOT(regs[0]))
 		regs[0] = psci_dispatch(ctx);
 	else
 		regs[0] = hypercall(regs[0], regs[1], regs[2]);