7 files changed, 41 insertions, 1 deletions
diff --git a/core/mem_region.c b/core/mem_region.c
index 8adb036..c3da059 100644
--- a/core/mem_region.c
+++ b/core/mem_region.c
@@ -376,8 +376,11 @@ found:
 void *mem_alloc(struct mem_region *region, size_t size, size_t align,
 		const char *location)
 {
-	void *r = __mem_alloc(region, size, align, location);
+	void *r;
 
+	assert(lock_held_by_me(&region->free_list_lock));
+
+	r = __mem_alloc(region, size, align, location);
 	if (r)
 		return r;
 
@@ -394,6 +397,8 @@ void mem_free(struct mem_region *region, void *mem, const char *location)
 	/* This should be a constant. */
 	assert(is_rodata(location));
 
+	assert(lock_held_by_me(&region->free_list_lock));
+
 	/* Freeing NULL is always a noop. */
 	if (!mem)
 		return;
@@ -426,6 +431,8 @@ bool mem_resize(struct mem_region *region, void *mem, size_t len,
 	/* This should be a constant. */
 	assert(is_rodata(location));
 
+	assert(lock_held_by_me(&region->free_list_lock));
+
 	/* Get header. */
 	hdr = mem - sizeof(*hdr);
 	if (hdr->free)
diff --git a/core/test/run-malloc-speed.c b/core/test/run-malloc-speed.c
index 713a74b..279216e 100644
--- a/core/test/run-malloc-speed.c
+++ b/core/test/run-malloc-speed.c
@@ -67,6 +67,11 @@ void unlock(struct lock *l)
 	l->lock_val = 0;
 }
 
+bool lock_held_by_me(struct lock *l)
+{
+	return l->lock_val;
+}
+
 #define TEST_HEAP_ORDER 27
 #define TEST_HEAP_SIZE (1ULL << TEST_HEAP_ORDER)
 
diff --git a/core/test/run-malloc.c b/core/test/run-malloc.c
index 723cb10..4623108 100644
--- a/core/test/run-malloc.c
+++ b/core/test/run-malloc.c
@@ -69,6 +69,11 @@ void unlock(struct lock *l)
 	l->lock_val = 0;
 }
 
+bool lock_held_by_me(struct lock *l)
+{
+	return l->lock_val;
+}
+
 static bool heap_empty(void)
 {
 	const struct alloc_hdr *h = region_start(&skiboot_heap);
diff --git a/core/test/run-mem_region.c b/core/test/run-mem_region.c
index e27459b..b98fe71 100644
--- a/core/test/run-mem_region.c
+++ b/core/test/run-mem_region.c
@@ -72,6 +72,11 @@ void unlock(struct lock *l)
 	l->lock_val--;
 }
 
+bool lock_held_by_me(struct lock *l __attribute__((unused)))
+{
+	return true;
+}
+
 #define TEST_HEAP_ORDER 12
 #define TEST_HEAP_SIZE (1ULL << TEST_HEAP_ORDER)
 
diff --git a/core/test/run-mem_region_init.c b/core/test/run-mem_region_init.c
index 7624057..7ab23d2 100644
--- a/core/test/run-mem_region_init.c
+++ b/core/test/run-mem_region_init.c
@@ -75,6 +75,11 @@ void unlock(struct lock *l)
 	l->lock_val = 0;
 }
 
+bool lock_held_by_me(struct lock *l)
+{
+	return l->lock_val;
+}
+
 /* We actually need a lot of room for the bitmaps! */
 #define TEST_HEAP_ORDER 27
 #define TEST_HEAP_SIZE (1ULL << TEST_HEAP_ORDER)
diff --git a/core/test/run-mem_region_release_unused.c b/core/test/run-mem_region_release_unused.c
index b8acffd..980f4c4 100644
--- a/core/test/run-mem_region_release_unused.c
+++ b/core/test/run-mem_region_release_unused.c
@@ -70,6 +70,11 @@ void unlock(struct lock *l)
 	l->lock_val--;
 }
 
+bool lock_held_by_me(struct lock *l)
+{
+	return l->lock_val;
+}
+
 #define TEST_HEAP_ORDER 12
 #define TEST_HEAP_SIZE (1ULL << TEST_HEAP_ORDER)
 
@@ -130,7 +135,10 @@ int main(void)
 	assert(mem_check(other));
 
 	/* Allocate 1k from other region. */
+	lock(&other->free_list_lock);
 	mem_alloc(other, 1024, 1, "1k");
+	unlock(&other->free_list_lock);
+
 	mem_region_release_unused();
 
 	assert(mem_check(&skiboot_heap));
diff --git a/core/test/run-mem_region_release_unused_noalloc.c b/core/test/run-mem_region_release_unused_noalloc.c
index 8dadddb..82ff89a 100644
--- a/core/test/run-mem_region_release_unused_noalloc.c
+++ b/core/test/run-mem_region_release_unused_noalloc.c
@@ -70,6 +70,11 @@ void unlock(struct lock *l)
 	l->lock_val--;
 }
 
+bool lock_held_by_me(struct lock *l)
+{
+	return l->lock_val;
+}
+
 #define TEST_HEAP_ORDER 12
 #define TEST_HEAP_SIZE (1ULL << TEST_HEAP_ORDER)