Loading...
--- xnu/xnu-4570.71.2/libkern/c++/OSData.cpp
+++ xnu/xnu-2782.1.97/libkern/c++/OSData.cpp
@@ -29,15 +29,11 @@
#include <string.h>
-__BEGIN_DECLS
-#include <vm/vm_kern.h>
-__END_DECLS
-
#include <libkern/c++/OSData.h>
#include <libkern/c++/OSSerialize.h>
#include <libkern/c++/OSLib.h>
#include <libkern/c++/OSString.h>
-#include <IOKit/IOLib.h>
+#include <string.h>
#define super OSObject
@@ -53,38 +49,32 @@
#define EXTERNAL ((unsigned int) -1)
+#if OSALLOCDEBUG
+extern int debug_container_malloc_size;
+#define ACCUMSIZE(s) do { debug_container_malloc_size += (s); } while(0)
+#else
+#define ACCUMSIZE(s)
+#endif
+
bool OSData::initWithCapacity(unsigned int inCapacity)
{
- if (data)
- {
- OSCONTAINER_ACCUMSIZE(-((size_t)capacity));
- if (!inCapacity || (capacity < inCapacity))
- {
- // clean out old data's storage if it isn't big enough
- if (capacity < page_size) kfree(data, capacity);
- else kmem_free(kernel_map, (vm_offset_t)data, capacity);
- data = 0;
- capacity = 0;
- }
- }
-
if (!super::init())
return false;
+ if (data && (!inCapacity || capacity < inCapacity) ) {
+ // clean out old data's storage if it isn't big enough
+ kfree(data, capacity);
+ data = 0;
+ ACCUMSIZE(-capacity);
+ }
+
if (inCapacity && !data) {
-
- if (inCapacity < page_size) data = (void *) kalloc_container(inCapacity);
- else {
- kern_return_t kr;
- if (round_page_overflow(inCapacity, &inCapacity)) kr = KERN_RESOURCE_SHORTAGE;
- else kr = kmem_alloc(kernel_map, (vm_offset_t *)&data, inCapacity, IOMemoryTag(kernel_map));
- if (KERN_SUCCESS != kr) data = NULL;
- }
+ data = (void *) kalloc(inCapacity);
if (!data)
return false;
capacity = inCapacity;
- }
- OSCONTAINER_ACCUMSIZE(capacity);
+ ACCUMSIZE(inCapacity);
+ }
length = 0;
if (inCapacity < 16)
@@ -197,10 +187,9 @@
void OSData::free()
{
- if ((capacity != EXTERNAL) && data && capacity) {
- if (capacity < page_size) kfree(data, capacity);
- else kmem_free(kernel_map, (vm_offset_t)data, capacity);
- OSCONTAINER_ACCUMSIZE( -((size_t)capacity) );
+ if (capacity != EXTERNAL && data && capacity) {
+ kfree(data, capacity);
+ ACCUMSIZE( -capacity );
} else if (capacity == EXTERNAL) {
DeallocFunction freemem = reserved ? reserved->deallocFunction : NULL;
if (freemem && data && length) {
@@ -230,8 +219,6 @@
{
unsigned char * newData;
unsigned int finalCapacity;
- void * copydata;
- kern_return_t kr;
if (newCapacity <= capacity)
return capacity;
@@ -240,38 +227,18 @@
* capacityIncrement;
// integer overflow check
- if (finalCapacity < newCapacity) return capacity;
-
- copydata = data;
-
- if (finalCapacity >= page_size) {
- // round up
- finalCapacity = round_page_32(finalCapacity);
- // integer overflow check
- if (finalCapacity < newCapacity) return capacity;
- if (capacity >= page_size) {
- copydata = NULL;
- kr = kmem_realloc(kernel_map,
- (vm_offset_t)data,
- capacity,
- (vm_offset_t *)&newData,
- finalCapacity,
- IOMemoryTag(kernel_map));
- } else {
- kr = kmem_alloc(kernel_map, (vm_offset_t *)&newData, finalCapacity, IOMemoryTag(kernel_map));
- }
- if (KERN_SUCCESS != kr) newData = NULL;
- }
- else newData = (unsigned char *) kalloc_container(finalCapacity);
+ if (finalCapacity < newCapacity)
+ return capacity;
+
+ newData = (unsigned char *) kalloc(finalCapacity);
if ( newData ) {
bzero(newData + capacity, finalCapacity - capacity);
- if (copydata) bcopy(copydata, newData, capacity);
if (data) {
- if (capacity < page_size) kfree(data, capacity);
- else kmem_free(kernel_map, (vm_offset_t)data, capacity);
+ bcopy(data, newData, capacity);
+ kfree(data, capacity);
}
- OSCONTAINER_ACCUMSIZE( ((size_t)finalCapacity) - ((size_t)capacity) );
+ ACCUMSIZE( finalCapacity - capacity );
data = (void *) newData;
capacity = finalCapacity;
}
@@ -343,7 +310,6 @@
if (length
&& start < length
- && (start + inLength) >= inLength // overflow check
&& (start + inLength) <= length)
outData = (const void *) ((char *) data + start);
@@ -479,7 +445,7 @@
{
if (!reserved)
{
- reserved = (typeof(reserved)) kalloc_container(sizeof(ExpansionData));
+ reserved = (typeof(reserved)) kalloc(sizeof(ExpansionData));
if (!reserved) return;
bzero(reserved, sizeof(ExpansionData));
}
@@ -490,14 +456,9 @@
{
if (!reserved)
{
- reserved = (typeof(reserved)) kalloc_container(sizeof(ExpansionData));
+ reserved = (typeof(reserved)) kalloc(sizeof(ExpansionData));
if (!reserved) return;
bzero(reserved, sizeof(ExpansionData));
}
reserved->disableSerialization = (!serializable);
}
-
-bool OSData::isSerializable(void)
-{
- return (!reserved || !reserved->disableSerialization);
-}