static int mali_ump_map(struct mali_session_data *session, mali_mem_allocation *descriptor)
{
    ump_dd_handle ump_mem;
    u32 nr_blocks;
    u32 i;
    ump_dd_physical_block *ump_blocks;
    struct mali_page_directory *pagedir;
    u32 offset = 0;
    u32 prop;
    _mali_osk_errcode_t err;

    MALI_DEBUG_ASSERT_POINTER(session);
    MALI_DEBUG_ASSERT_POINTER(descriptor);
    MALI_DEBUG_ASSERT(MALI_MEM_UMP == descriptor->type);

    ump_mem = descriptor->ump_mem.handle;
    MALI_DEBUG_ASSERT(UMP_DD_HANDLE_INVALID != ump_mem);

    nr_blocks = ump_dd_phys_block_count_get(ump_mem);
    if (nr_blocks == 0) {
        MALI_DEBUG_PRINT(1, ("No block count\n"));
        return -EINVAL;
    }

    ump_blocks = _mali_osk_malloc(sizeof(*ump_blocks)*nr_blocks);
    if (NULL == ump_blocks) {
        return -ENOMEM;
    }

    if (UMP_DD_INVALID == ump_dd_phys_blocks_get(ump_mem, ump_blocks, nr_blocks)) {
        _mali_osk_free(ump_blocks);
        return -EFAULT;
    }

    pagedir = session->page_directory;
    prop = descriptor->mali_mapping.properties;

    err = mali_mem_mali_map_prepare(descriptor);
    if (_MALI_OSK_ERR_OK != err) {
        MALI_DEBUG_PRINT(1, ("Mapping of UMP memory failed\n"));

        _mali_osk_free(ump_blocks);
        return -ENOMEM;
    }

    for(i = 0; i < nr_blocks; ++i) {
        u32 virt = descriptor->mali_mapping.addr + offset;

        MALI_DEBUG_PRINT(7, ("Mapping in 0x%08x size %d\n", ump_blocks[i].addr , ump_blocks[i].size));

        mali_mmu_pagedir_update(pagedir, virt, ump_blocks[i].addr,
                                ump_blocks[i].size, prop);

        offset += ump_blocks[i].size;
    }

    if (descriptor->flags & _MALI_MAP_EXTERNAL_MAP_GUARD_PAGE) {
        u32 virt = descriptor->mali_mapping.addr + offset;

        /* Map in an extra virtual guard page at the end of the VMA */
        MALI_DEBUG_PRINT(6, ("Mapping in extra guard page\n"));

        mali_mmu_pagedir_update(pagedir, virt, ump_blocks[0].addr, _MALI_OSK_MALI_PAGE_SIZE, prop);

        offset += _MALI_OSK_MALI_PAGE_SIZE;
    }

    _mali_osk_free(ump_blocks);

    return 0;
}
static int mali_mem_ump_map(mali_mem_backend *mem_backend)
{
	ump_dd_handle ump_mem;
	mali_mem_allocation *alloc;
	struct mali_session_data *session;
	u32 nr_blocks;
	u32 i;
	ump_dd_physical_block *ump_blocks;
	struct mali_page_directory *pagedir;
	u32 offset = 0;
	_mali_osk_errcode_t err;

	MALI_DEBUG_ASSERT_POINTER(mem_backend);
	MALI_DEBUG_ASSERT(MALI_MEM_UMP == mem_backend->type);

	alloc = mem_backend->mali_allocation;
	MALI_DEBUG_ASSERT_POINTER(alloc);

	session = alloc->session;
	MALI_DEBUG_ASSERT_POINTER(session);

	ump_mem = mem_backend->ump_mem.handle;
	MALI_DEBUG_ASSERT(UMP_DD_HANDLE_INVALID != ump_mem);

	nr_blocks = ump_dd_phys_block_count_get(ump_mem);
	if (nr_blocks == 0) {
		MALI_DEBUG_PRINT(1, ("No block count\n"));
		return -EINVAL;
	}

	ump_blocks = _mali_osk_malloc(sizeof(*ump_blocks) * nr_blocks);
	if (NULL == ump_blocks) {
		return -ENOMEM;
	}

	if (UMP_DD_INVALID == ump_dd_phys_blocks_get(ump_mem, ump_blocks, nr_blocks)) {
		_mali_osk_free(ump_blocks);
		return -EFAULT;
	}

	pagedir = session->page_directory;

	mali_session_memory_lock(session);

	err = mali_mem_mali_map_prepare(alloc);
	if (_MALI_OSK_ERR_OK != err) {
		MALI_DEBUG_PRINT(1, ("Mapping of UMP memory failed\n"));

		_mali_osk_free(ump_blocks);
		mali_session_memory_unlock(session);
		return -ENOMEM;
	}

	for (i = 0; i < nr_blocks; ++i) {
		u32 virt = alloc->mali_vma_node.vm_node.start + offset;

		MALI_DEBUG_PRINT(7, ("Mapping in 0x%08x size %d\n", ump_blocks[i].addr , ump_blocks[i].size));

		mali_mmu_pagedir_update(pagedir, virt, ump_blocks[i].addr,
					ump_blocks[i].size, MALI_MMU_FLAGS_DEFAULT);

		offset += ump_blocks[i].size;
	}

	if (alloc->flags & _MALI_MAP_EXTERNAL_MAP_GUARD_PAGE) {
		u32 virt = alloc->mali_vma_node.vm_node.start + offset;

		/* Map in an extra virtual guard page at the end of the VMA */
		MALI_DEBUG_PRINT(6, ("Mapping in extra guard page\n"));

		mali_mmu_pagedir_update(pagedir, virt, ump_blocks[0].addr, _MALI_OSK_MALI_PAGE_SIZE, MALI_MMU_FLAGS_DEFAULT);

		offset += _MALI_OSK_MALI_PAGE_SIZE;
	}
	session->mali_mem_array[mem_backend->type] += mem_backend->size;
	mali_session_memory_unlock(session);
	_mali_osk_free(ump_blocks);
	return 0;
}