Loading...
libkern/c++/OSDictionary.cpp xnu-12377.101.15 xnu-8019.41.5
--- xnu/xnu-12377.101.15/libkern/c++/OSDictionary.cpp
+++ xnu/xnu-8019.41.5/libkern/c++/OSDictionary.cpp
@@ -93,11 +93,13 @@
 
 //fOptions |= kSort;
 
-	dictionary = kallocp_type_container(dictEntry, &inCapacity, Z_WAITOK_ZERO);
+	dictionary = kalloc_type_tag_bt(dictEntry, inCapacity, Z_WAITOK_ZERO,
+	    VM_KERN_MEMORY_LIBKERN);
 	if (!dictionary) {
 		return false;
 	}
 
+	os::uninitialized_value_construct(dictionary, dictionary + inCapacity);
 	OSCONTAINER_ACCUMSIZE(inCapacity * sizeof(dictEntry));
 
 	count = 0;
@@ -316,7 +318,7 @@
 OSDictionary::ensureCapacity(unsigned int newCapacity)
 {
 	dictEntry *newDict;
-	unsigned int finalCapacity;
+	vm_size_t finalCapacity;
 
 	if (newCapacity <= capacity) {
 		return capacity;
@@ -331,12 +333,25 @@
 		return capacity;
 	}
 
-	newDict = kreallocp_type_container(dictEntry, dictionary,
-	    capacity, &finalCapacity, Z_WAITOK_ZERO);
+	newDict = kallocp_type_tag_bt(dictEntry, &finalCapacity, Z_WAITOK,
+	    VM_KERN_MEMORY_LIBKERN);
 	if (newDict) {
+		// use all of the actual allocation size
+		if (finalCapacity > UINT_MAX) {
+			// failure, too large
+			kfree_type(dictEntry, finalCapacity, newDict);
+			return capacity;
+		}
+
+		os::uninitialized_move(dictionary, dictionary + capacity, newDict);
+		os::uninitialized_value_construct(newDict + capacity, newDict + finalCapacity);
+		os::destroy(dictionary, dictionary + capacity);
+
 		OSCONTAINER_ACCUMSIZE(sizeof(dictEntry) * (finalCapacity - capacity));
+
+		kfree_type(dictEntry, capacity, dictionary);
 		dictionary = newDict;
-		capacity = finalCapacity;
+		capacity = (unsigned int) finalCapacity;
 	}
 
 	return capacity;
@@ -348,8 +363,8 @@
 	haveUpdated();
 
 	for (unsigned int i = 0; i < count; i++) {
-		dictionary[i].key.reset();
-		dictionary[i].value.reset();
+		dictionary[i].key->taggedRelease(OSTypeID(OSCollection));
+		dictionary[i].value->taggedRelease(OSTypeID(OSCollection));
 	}
 	count = 0;
 }