change the callback implementation from mmap to get_mem Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com>