diff --git a/kernel/mem_domain.c b/kernel/mem_domain.c index 8076f61c2c..4cf4fb2535 100644 --- a/kernel/mem_domain.c +++ b/kernel/mem_domain.c @@ -63,6 +63,11 @@ void k_mem_domain_destroy(struct k_mem_domain *domain) key = irq_lock(); + /* Handle architecture specifc destroy only if it is the current thread*/ + if (_current->mem_domain_info.mem_domain == domain) { + _arch_mem_domain_destroy(domain); + } + SYS_DLIST_FOR_EACH_NODE_SAFE(&domain->mem_domain_q, node, next_node) { struct k_thread *thread = CONTAINER_OF(node, struct k_thread, mem_domain_info); @@ -127,6 +132,11 @@ void k_mem_domain_remove_partition(struct k_mem_domain *domain, /* Assert if not found */ __ASSERT(p_idx < max_partitions, ""); + /* Handle architecture specifc remove only if it is the current thread*/ + if (_current->mem_domain_info.mem_domain == domain) { + _arch_mem_domain_partition_remove(domain, p_idx); + } + domain->partitions[p_idx].start = 0; domain->partitions[p_idx].size = 0; domain->partitions[p_idx].attr = 0;