ta: os_test: Unmask cancellation from invoke command handler

The specification seems to indicate (see #731)
that the cancellation mask is reset at the start of every entry point.
As a result, the TEE_UnmakCancellation() should not be called at the end of
TA_OpenSessionEntryPoint(), but at the start of TA_InvokeCommandEntryPoint().

Signed-off-by: Cedric Auger <[email protected]>
Reviewed-by: Jerome Forissier <[email protected]>
Acked-by: Etienne Carriere <[email protected]>

ta/os_test/ta_entry.c

@@ -36,7 +36,6 @@ TEE_Result TA_OpenSessionEntryPoint(uint32_t nParamTypes,
 	(void)pParams;
 	(void)ppSessionContext;
 	DMSG("TA_OpenSessionEntryPoint");
-	TEE_UnmaskCancellation();
 	return TEE_SUCCESS;
 }
 
@@ -63,6 +62,7 @@ TEE_Result TA_InvokeCommandEntryPoint(void *pSessionContext,
 				      TEE_Param pParams[4])
 {
 	(void)pSessionContext;
+	TEE_UnmaskCancellation();
 
 	switch (nCommandID) {
 	case TA_OS_TEST_CMD_INIT: