efi_loader: Check for the native OP-TEE result on mm_communicate calls

Currently we only check for the return value of tee_invoke_func().
Although OP-TEE and StMM will correctly set param[1].u.value.a and we'll
eventually return an error, the correct thing to do is check for the
OP_TEE return code as well.
So let's check for that and move tee_shm_free() and tee_close_session()
before exiting with an error to make sure we always clear the registered
memory.

Fixes: f042e47e8fb43 ("efi_loader: Implement EFI variable handling via OP-TEE")
Signed-off-by: Ilias Apalodimas <ilias.apalodimas@linaro.org>
Use EFI_DEVICE_ERROR for TEE communication problems.
Reviewed-by: Heinrich Schuchardt <xypron.glpk@gmx.de>

diff --git a/lib/efi_loader/efi_variable_tee.c b/lib/efi_loader/efi_variable_tee.c
index 5f4aae60bf23f53b09cb0d9214993774f4cff19e..94c4de87034300f5015482cd68f21907fd3beb91 100644 (file)
--- a/lib/efi_loader/efi_variable_tee.c
+++ b/lib/efi_loader/efi_variable_tee.c
@@ -100,10 +100,10 @@ static efi_status_t optee_mm_communicate(void *comm_buf, ulong dsize)
        param[1].attr = TEE_PARAM_ATTR_TYPE_VALUE_OUTPUT;
 
        rc = tee_invoke_func(conn.tee, &arg, 2, param);
-       if (rc)
-               return EFI_INVALID_PARAMETER;
        tee_shm_free(shm);
        tee_close_session(conn.tee, conn.session);
+       if (rc || arg.ret != TEE_SUCCESS)
+               return EFI_DEVICE_ERROR;
 
        switch (param[1].u.value.a) {
        case ARM_SVC_SPM_RET_SUCCESS: