core: virt: initialize heap from virt_guest_created()

Replace the preinit_early() guest heap initialization with function call
in virt_guest_created().

Signed-off-by: Jens Wiklander <jens.wiklander@linaro.org>

core/arch/arm/kernel/boot.c

@@ -912,17 +912,6 @@ static void update_external_dt(void)
 }
 #endif /*!CFG_DT*/
 
-#ifdef CFG_NS_VIRTUALIZATION
-static TEE_Result virt_init_heap(void)
-{
-	/* We need to initialize pool for every virtual guest partition */
-	malloc_add_pool(__heap1_start, __heap1_end - __heap1_start);
-
-	return TEE_SUCCESS;
-}
-preinit_early(virt_init_heap);
-#endif
-
 void init_tee_runtime(void)
 {
 #ifndef CFG_WITH_PAGER

core/arch/arm/kernel/virtualization.c

@@ -361,6 +361,7 @@ TEE_Result virt_guest_created(uint16_t guest_id)
 
 	set_current_prtn(prtn);
 
+	malloc_add_pool(__heap1_start, __heap1_end - __heap1_start);
 	/* Initialize threads */
 	thread_init_threads();
 	/* Do the preinitcalls */