codify locking contractspmm-2015-08-16

1 files changed, 4 insertions, 0 deletions

diff --git a/vm/vm_map.c b/vm/vm_map.c
index 1ebccac..57d931b 100644
--- a/vm/vm_map.c
+++ b/vm/vm_map.c
@@ -516,6 +516,8 @@ kern_return_t vm_map_find_entry(
 	vm_offset_t	start;
 	vm_offset_t	end;
 
+	assert(have_vm_map_lock(map));
+
 	/*
 	 *	Look for the first possible address;
 	 *	if there's already something at this
@@ -586,6 +588,7 @@ kern_return_t vm_map_find_entry(
 	 *
 	 *		the map should be locked.
 	 */
+	assert(have_vm_map_lock(map));
 
 	*address = start;
 
@@ -903,6 +906,7 @@ kern_return_t vm_map_enter(
 	 *
 	 *		the map should be locked.
 	 */
+	assert(have_vm_map_lock(map));
 
 	/*
 	 *	See whether we can avoid creating a new entry (and object) by