mm: fix memory corruption when loop create/exit thread in SMP mode

Root casue:
when do thread exit, need add free stack operation to mm_delaylist,
but in SMP mode, CPU0 thread1 exit, at this time, CPU1 call malloc
and free mm_delaylist.

Fix:
Divide mm_delaylist for per CPU in SMP mode.

Change-Id: Ibf7d04614ea2f99fb5b506356b7346a0d94f0590
Signed-off-by: ligd <liguiding1@xiaomi.com>
This commit is contained in:
ligd 2021-06-21 16:20:44 +08:00 committed by Xiang Xiao
parent 554310bd78
commit aa43a0215d
5 changed files with 23 additions and 22 deletions

View File

@ -48,7 +48,11 @@ struct mm_delaynode_s
struct mm_heap_impl_s struct mm_heap_impl_s
{ {
struct mm_delaynode_s *mm_delaylist; #ifdef CONFIG_SMP
struct mm_delaynode_s *mm_delaylist[CONFIG_SMP_NCPUS];
#else
struct mm_delaynode_s *mm_delaylist[1];
#endif
}; };
/**************************************************************************** /****************************************************************************
@ -65,8 +69,8 @@ static void mm_add_delaylist(FAR struct mm_heap_s *heap, FAR void *mem)
flags = enter_critical_section(); flags = enter_critical_section();
tmp->flink = heap->mm_impl->mm_delaylist; tmp->flink = heap->mm_impl->mm_delaylist[up_cpu_index()];
heap->mm_impl->mm_delaylist = tmp; heap->mm_impl->mm_delaylist[up_cpu_index()] = tmp;
leave_critical_section(flags); leave_critical_section(flags);
} }
@ -83,8 +87,8 @@ static void mm_free_delaylist(FAR struct mm_heap_s *heap)
flags = enter_critical_section(); flags = enter_critical_section();
tmp = heap->mm_impl->mm_delaylist; tmp = heap->mm_impl->mm_delaylist[up_cpu_index()];
heap->mm_impl->mm_delaylist = NULL; heap->mm_impl->mm_delaylist[up_cpu_index()] = NULL;
leave_critical_section(flags); leave_critical_section(flags);
@ -135,8 +139,11 @@ void mm_initialize(FAR struct mm_heap_s *heap, FAR void *heap_start,
size_t heap_size) size_t heap_size)
{ {
FAR struct mm_heap_impl_s *impl; FAR struct mm_heap_impl_s *impl;
impl = host_malloc(sizeof(struct mm_heap_impl_s)); impl = host_malloc(sizeof(struct mm_heap_impl_s));
impl->mm_delaylist = NULL; DEBUGASSERT(impl);
memset(impl, 0, sizeof(struct mm_heap_impl_s));
heap->mm_impl = impl; heap->mm_impl = impl;
} }

View File

@ -202,7 +202,11 @@ struct mm_heap_impl_s
/* Free delay list, for some situation can't do free immdiately */ /* Free delay list, for some situation can't do free immdiately */
FAR struct mm_delaynode_s *mm_delaylist; #ifdef CONFIG_SMP
FAR struct mm_delaynode_s *mm_delaylist[CONFIG_SMP_NCPUS];
#else
FAR struct mm_delaynode_s *mm_delaylist[1];
#endif
}; };
/* Functions contained in mm_sem.c ******************************************/ /* Functions contained in mm_sem.c ******************************************/

View File

@ -50,8 +50,8 @@ static void mm_add_delaylist(FAR struct mm_heap_s *heap, FAR void *mem)
flags = enter_critical_section(); flags = enter_critical_section();
tmp->flink = heap_impl->mm_delaylist; tmp->flink = heap_impl->mm_delaylist[up_cpu_index()];
heap_impl->mm_delaylist = tmp; heap_impl->mm_delaylist[up_cpu_index()] = tmp;
leave_critical_section(flags); leave_critical_section(flags);
} }

View File

@ -197,20 +197,10 @@ void mm_initialize(FAR struct mm_heap_s *heap, FAR void *heapstart,
/* Set up global variables */ /* Set up global variables */
heap_impl->mm_heapsize = 0; memset(heap_impl, 0, sizeof(struct mm_heap_impl_s));
#if CONFIG_MM_REGIONS > 1
heap_impl->mm_nregions = 0;
#endif
/* Initialize mm_delaylist */
heap_impl->mm_delaylist = NULL;
/* Initialize the node array */ /* Initialize the node array */
memset(heap_impl->mm_nodelist, 0,
sizeof(struct mm_freenode_s) * MM_NNODES);
for (i = 1; i < MM_NNODES; i++) for (i = 1; i < MM_NNODES; i++)
{ {
heap_impl->mm_nodelist[i - 1].flink = &heap_impl->mm_nodelist[i]; heap_impl->mm_nodelist[i - 1].flink = &heap_impl->mm_nodelist[i];

View File

@ -59,8 +59,8 @@ static void mm_free_delaylist(FAR struct mm_heap_s *heap)
flags = enter_critical_section(); flags = enter_critical_section();
tmp = heap_impl->mm_delaylist; tmp = heap_impl->mm_delaylist[up_cpu_index()];
heap_impl->mm_delaylist = NULL; heap_impl->mm_delaylist[up_cpu_index()] = NULL;
leave_critical_section(flags); leave_critical_section(flags);