* Copyright (c) 1998-2016 Apple Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- *
+ *
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* unlawful or unlicensed copies of an Apple operating system, or to
* circumvent, violate, or enable the circumvention or violation of, any
* terms of an Apple operating system software license agreement.
- *
+ *
* Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this file.
- *
+ *
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
- *
+ *
* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
*/
#define IODEBUG_CTLFLAGS CTLFLAG_RD
#endif
-SYSCTL_QUAD(_debug, OID_AUTO, iokit, IODEBUG_CTLFLAGS | CTLFLAG_LOCKED, &gIOKitDebug, "boot_arg io");
SYSCTL_QUAD(_debug, OID_AUTO, iotrace, CTLFLAG_RW | CTLFLAG_LOCKED, &gIOKitTrace, "trace io");
+static int
+sysctl_debug_iokit
+(__unused struct sysctl_oid *oidp, __unused void *arg1, __unused int arg2, struct sysctl_req *req)
+{
+ SInt64 newValue;
+ int changed, error = sysctl_io_number(req, gIOKitDebug, sizeof(gIOKitDebug), &newValue, &changed);
+ if (changed) {
+ gIOKitDebug = ((gIOKitDebug & ~kIOKitDebugUserOptions) | (newValue & kIOKitDebugUserOptions));
+ }
+ return error;
+}
+
+SYSCTL_PROC(_debug, OID_AUTO, iokit,
+ CTLTYPE_QUAD | IODEBUG_CTLFLAGS | CTLFLAG_NOAUTO | CTLFLAG_KERN | CTLFLAG_LOCKED,
+ &gIOKitDebug, 0, sysctl_debug_iokit, "Q", "boot_arg io");
int debug_malloc_size;
int debug_iomalloc_size;
// int debug_ivars_size; // in OSObject.cpp
extern "C" {
-
#if 0
#define DEBG(fmt, args...) { kprintf(fmt, ## args); }
#else
#define DEBG(fmt, args...) { IOLog(fmt, ## args); }
#endif
-void IOPrintPlane( const IORegistryPlane * plane )
+void
+IOPrintPlane( const IORegistryPlane * plane )
{
- IORegistryEntry * next;
- IORegistryIterator * iter;
- OSOrderedSet * all;
- char format[] = "%xxxs";
- IOService * service;
-
- iter = IORegistryIterator::iterateOver( plane );
- assert( iter );
- all = iter->iterateAll();
- if( all) {
- DEBG("Count %d\n", all->getCount() );
- all->release();
- } else
- DEBG("Empty\n");
-
- iter->reset();
- while( (next = iter->getNextObjectRecursive())) {
- snprintf(format + 1, sizeof(format) - 1, "%ds", 2 * next->getDepth( plane ));
- DEBG( format, "");
- DEBG( "\033[33m%s", next->getName( plane ));
- if( (next->getLocation( plane )))
- DEBG("@%s", next->getLocation( plane ));
- DEBG("\033[0m <class %s", next->getMetaClass()->getClassName());
- if( (service = OSDynamicCast(IOService, next)))
- DEBG(", busy %ld", (long) service->getBusyState());
- DEBG( ">\n");
+ IORegistryEntry * next;
+ IORegistryIterator * iter;
+ OSOrderedSet * all;
+ char format[] = "%xxxs";
+ IOService * service;
+
+ iter = IORegistryIterator::iterateOver( plane );
+ assert( iter );
+ all = iter->iterateAll();
+ if (all) {
+ DEBG("Count %d\n", all->getCount());
+ all->release();
+ } else {
+ DEBG("Empty\n");
+ }
+
+ iter->reset();
+ while ((next = iter->getNextObjectRecursive())) {
+ snprintf(format + 1, sizeof(format) - 1, "%ds", 2 * next->getDepth( plane ));
+ DEBG( format, "");
+ DEBG( "\033[33m%s", next->getName( plane ));
+ if ((next->getLocation( plane ))) {
+ DEBG("@%s", next->getLocation( plane ));
+ }
+ DEBG("\033[0m <class %s", next->getMetaClass()->getClassName());
+ if ((service = OSDynamicCast(IOService, next))) {
+ DEBG(", busy %ld", (long) service->getBusyState());
+ }
+ DEBG( ">\n");
// IOSleep(250);
- }
- iter->release();
+ }
+ iter->release();
}
-void db_piokjunk(void)
+void
+db_piokjunk(void)
{
}
-void db_dumpiojunk( const IORegistryPlane * plane __unused )
+void
+db_dumpiojunk( const IORegistryPlane * plane __unused )
{
}
-void IOPrintMemory( void )
+void
+IOPrintMemory( void )
{
-
// OSMetaClass::printInstanceCounts();
- IOLog("\n"
- "ivar kalloc() 0x%08x\n"
- "malloc() 0x%08x\n"
- "containers kalloc() 0x%08x\n"
- "IOMalloc() 0x%08x\n"
- "----------------------------------------\n",
- debug_ivars_size,
- debug_malloc_size,
- debug_container_malloc_size,
- debug_iomalloc_size
- );
+ IOLog("\n"
+ "ivar kalloc() 0x%08x\n"
+ "malloc() 0x%08x\n"
+ "containers kalloc() 0x%08x\n"
+ "IOMalloc() 0x%08x\n"
+ "----------------------------------------\n",
+ debug_ivars_size,
+ debug_malloc_size,
+ debug_container_malloc_size,
+ debug_iomalloc_size
+ );
}
-
} /* extern "C" */
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
OSObject * IOKitDiagnostics::diagnostics( void )
{
- IOKitDiagnostics * diags;
+ IOKitDiagnostics * diags;
- diags = new IOKitDiagnostics;
- if( diags && !diags->init()) {
- diags->release();
- diags = 0;
- }
+ diags = new IOKitDiagnostics;
+ if (diags && !diags->init()) {
+ diags->release();
+ diags = NULL;
+ }
- return( diags );
+ return diags;
}
-void IOKitDiagnostics::updateOffset( OSDictionary * dict,
- UInt64 value, const char * name )
+void
+IOKitDiagnostics::updateOffset( OSDictionary * dict,
+ UInt64 value, const char * name )
{
- OSNumber * off;
+ OSNumber * off;
- off = OSNumber::withNumber( value, 64 );
- if( !off)
- return;
+ off = OSNumber::withNumber( value, 64 );
+ if (!off) {
+ return;
+ }
- dict->setObject( name, off );
- off->release();
+ dict->setObject( name, off );
+ off->release();
}
-bool IOKitDiagnostics::serialize(OSSerialize *s) const
+bool
+IOKitDiagnostics::serialize(OSSerialize *s) const
{
- OSDictionary * dict;
- bool ok;
+ OSDictionary * dict;
+ bool ok;
- dict = OSDictionary::withCapacity( 5 );
- if( !dict)
- return( false );
+ dict = OSDictionary::withCapacity( 5 );
+ if (!dict) {
+ return false;
+ }
- updateOffset( dict, debug_ivars_size, "Instance allocation" );
- updateOffset( dict, debug_container_malloc_size, "Container allocation" );
- updateOffset( dict, debug_iomalloc_size, "IOMalloc allocation" );
- updateOffset( dict, debug_iomallocpageable_size, "Pageable allocation" );
+ updateOffset( dict, debug_ivars_size, "Instance allocation" );
+ updateOffset( dict, debug_container_malloc_size, "Container allocation" );
+ updateOffset( dict, debug_iomalloc_size, "IOMalloc allocation" );
+ updateOffset( dict, debug_iomallocpageable_size, "Pageable allocation" );
- OSMetaClass::serializeClassDictionary(dict);
+ OSMetaClass::serializeClassDictionary(dict);
- ok = dict->serialize( s );
+ ok = dict->serialize( s );
- dict->release();
+ dict->release();
- return( ok );
+ return ok;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
#include <kern/zalloc.h>
__private_extern__ "C" void qsort(
- void * array,
- size_t nmembers,
- size_t member_size,
- int (*)(const void *, const void *));
+ void * array,
+ size_t nmembers,
+ size_t member_size,
+ int (*)(const void *, const void *));
extern "C" ppnum_t pmap_find_phys(pmap_t pmap, addr64_t va);
extern "C" ppnum_t pmap_valid_page(ppnum_t pn);
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-struct IOTRecursiveLock
-{
- lck_mtx_t * mutex;
- thread_t thread;
- UInt32 count;
+struct IOTRecursiveLock {
+ lck_mtx_t * mutex;
+ thread_t thread;
+ UInt32 count;
};
-struct IOTrackingQueue
-{
- queue_chain_t link;
- IOTRecursiveLock lock;
- const char * name;
- uintptr_t btEntry;
- size_t allocSize;
- size_t minCaptureSize;
- uint32_t siteCount;
- uint32_t type;
- uint32_t numSiteQs;
- uint8_t captureOn;
- queue_head_t sites[];
+struct IOTrackingQueue {
+ queue_chain_t link;
+ IOTRecursiveLock lock;
+ const char * name;
+ uintptr_t btEntry;
+ size_t allocSize;
+ size_t minCaptureSize;
+ uint32_t siteCount;
+ uint32_t type;
+ uint32_t numSiteQs;
+ uint8_t captureOn;
+ queue_head_t sites[];
};
-struct IOTrackingCallSite
-{
- queue_chain_t link;
- IOTrackingQueue * queue;
- uint32_t crc;
+struct IOTrackingCallSite {
+ queue_chain_t link;
+ IOTrackingQueue * queue;
+ uint32_t crc;
- uint32_t count;
- size_t size[2];
- uintptr_t bt[kIOTrackingCallSiteBTs];
+ vm_tag_t tag;
+ uint32_t count;
+ size_t size[2];
+ uintptr_t bt[kIOTrackingCallSiteBTs];
- queue_head_t instances;
- IOTracking * addresses;
+ queue_head_t instances;
+ IOTracking * addresses;
};
-struct IOTrackingLeaksRef
-{
- uintptr_t * instances;
- uint32_t zoneSize;
- uint32_t count;
- uint32_t found;
- size_t bytes;
+struct IOTrackingLeaksRef {
+ uintptr_t * instances;
+ uint32_t zoneSize;
+ uint32_t count;
+ uint32_t found;
+ uint32_t foundzlen;
+ size_t bytes;
};
lck_mtx_t * gIOTrackingLock;
queue_head_t gIOTrackingQ;
-enum
-{
- kTrackingAddressFlagAllocated = 0x00000001
+enum{
+ kTrackingAddressFlagAllocated = 0x00000001
};
#if defined(__LP64__)
-#define IOTrackingAddressFlags(ptr) (ptr->flags)
+#define IOTrackingAddressFlags(ptr) (ptr->flags)
#else
-#define IOTrackingAddressFlags(ptr) (ptr->tracking.flags)
+#define IOTrackingAddressFlags(ptr) (ptr->tracking.flags)
#endif
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-static void
+static void
IOTRecursiveLockLock(IOTRecursiveLock * lock)
{
- if (lock->thread == current_thread()) lock->count++;
- else
- {
- lck_mtx_lock(lock->mutex);
- assert(lock->thread == 0);
- assert(lock->count == 0);
- lock->thread = current_thread();
- lock->count = 1;
- }
+ if (lock->thread == current_thread()) {
+ lock->count++;
+ } else {
+ lck_mtx_lock(lock->mutex);
+ assert(lock->thread == NULL);
+ assert(lock->count == 0);
+ lock->thread = current_thread();
+ lock->count = 1;
+ }
}
-static void
+static void
IOTRecursiveLockUnlock(IOTRecursiveLock * lock)
{
- assert(lock->thread == current_thread());
- if (0 == (--lock->count))
- {
- lock->thread = 0;
- lck_mtx_unlock(lock->mutex);
- }
+ assert(lock->thread == current_thread());
+ if (0 == (--lock->count)) {
+ lock->thread = NULL;
+ lck_mtx_unlock(lock->mutex);
+ }
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingInit(void)
{
- queue_init(&gIOTrackingQ);
- gIOTrackingLock = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
+ queue_init(&gIOTrackingQ);
+ gIOTrackingLock = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
IOTrackingQueue *
IOTrackingQueueAlloc(const char * name, uintptr_t btEntry,
- size_t allocSize, size_t minCaptureSize,
- uint32_t type, uint32_t numSiteQs)
+ size_t allocSize, size_t minCaptureSize,
+ uint32_t type, uint32_t numSiteQs)
{
- IOTrackingQueue * queue;
- uint32_t idx;
-
- if (!numSiteQs) numSiteQs = 1;
- queue = (typeof(queue)) kalloc(sizeof(IOTrackingQueue) + numSiteQs * sizeof(queue->sites[0]));
- bzero(queue, sizeof(IOTrackingQueue));
-
- queue->name = name;
- queue->btEntry = btEntry;
- queue->allocSize = allocSize;
- queue->minCaptureSize = minCaptureSize;
- queue->lock.mutex = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
- queue->numSiteQs = numSiteQs;
- queue->type = type;
- enum { kFlags = (kIOTracking | kIOTrackingBoot) };
- queue->captureOn = (kFlags == (kFlags & gIOKitDebug))
- || (kIOTrackingQueueTypeDefaultOn & type);
-
- for (idx = 0; idx < numSiteQs; idx++) queue_init(&queue->sites[idx]);
-
- lck_mtx_lock(gIOTrackingLock);
- queue_enter(&gIOTrackingQ, queue, IOTrackingQueue *, link);
- lck_mtx_unlock(gIOTrackingLock);
-
- return (queue);
+ IOTrackingQueue * queue;
+ uint32_t idx;
+
+ if (!numSiteQs) {
+ numSiteQs = 1;
+ }
+ queue = (typeof(queue))kalloc(sizeof(IOTrackingQueue) + numSiteQs * sizeof(queue->sites[0]));
+ bzero(queue, sizeof(IOTrackingQueue));
+
+ queue->name = name;
+ queue->btEntry = btEntry;
+ queue->allocSize = allocSize;
+ queue->minCaptureSize = minCaptureSize;
+ queue->lock.mutex = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
+ queue->numSiteQs = numSiteQs;
+ queue->type = type;
+ enum { kFlags = (kIOTracking | kIOTrackingBoot) };
+ queue->captureOn = (kFlags == (kFlags & gIOKitDebug))
+ || (kIOTrackingQueueTypeDefaultOn & type);
+
+ for (idx = 0; idx < numSiteQs; idx++) {
+ queue_init(&queue->sites[idx]);
+ }
+
+ lck_mtx_lock(gIOTrackingLock);
+ queue_enter(&gIOTrackingQ, queue, IOTrackingQueue *, link);
+ lck_mtx_unlock(gIOTrackingLock);
+
+ return queue;
};
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingQueueFree(IOTrackingQueue * queue)
{
- lck_mtx_lock(gIOTrackingLock);
- IOTrackingReset(queue);
- remque(&queue->link);
- lck_mtx_unlock(gIOTrackingLock);
+ lck_mtx_lock(gIOTrackingLock);
+ IOTrackingReset(queue);
+ remque(&queue->link);
+ lck_mtx_unlock(gIOTrackingLock);
- lck_mtx_free(queue->lock.mutex, IOLockGroup);
+ lck_mtx_free(queue->lock.mutex, IOLockGroup);
- kfree(queue, sizeof(IOTrackingQueue) + queue->numSiteQs * sizeof(queue->sites[0]));
+ kfree(queue, sizeof(IOTrackingQueue) + queue->numSiteQs * sizeof(queue->sites[0]));
};
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
/* fasthash
- The MIT License
-
- Copyright (C) 2012 Zilong Tan (eric.zltan@gmail.com)
-
- Permission is hereby granted, free of charge, to any person
- obtaining a copy of this software and associated documentation
- files (the "Software"), to deal in the Software without
- restriction, including without limitation the rights to use, copy,
- modify, merge, publish, distribute, sublicense, and/or sell copies
- of the Software, and to permit persons to whom the Software is
- furnished to do so, subject to the following conditions:
-
- The above copyright notice and this permission notice shall be
- included in all copies or substantial portions of the Software.
-
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
- EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
- MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
- NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
- BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
- ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
- CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- SOFTWARE.
-*/
+ * The MIT License
+ *
+ * Copyright (C) 2012 Zilong Tan (eric.zltan@gmail.com)
+ *
+ * Permission is hereby granted, free of charge, to any person
+ * obtaining a copy of this software and associated documentation
+ * files (the "Software"), to deal in the Software without
+ * restriction, including without limitation the rights to use, copy,
+ * modify, merge, publish, distribute, sublicense, and/or sell copies
+ * of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+ * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+ * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+ * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
// Compression function for Merkle-Damgard construction.
// This function is generated using the framework provided.
#define mix(h) ({ \
- (h) ^= (h) >> 23; \
- (h) *= 0x2127599bf4325c37ULL; \
- (h) ^= (h) >> 47; })
+ (h) ^= (h) >> 23; \
+ (h) *= 0x2127599bf4325c37ULL; \
+ (h) ^= (h) >> 47; })
static uint64_t
fasthash64(const void *buf, size_t len, uint64_t seed)
{
- const uint64_t m = 0x880355f21e6d1965ULL;
- const uint64_t *pos = (const uint64_t *)buf;
- const uint64_t *end = pos + (len / 8);
- const unsigned char *pos2;
- uint64_t h = seed ^ (len * m);
- uint64_t v;
-
- while (pos != end) {
- v = *pos++;
- h ^= mix(v);
- h *= m;
- }
-
- pos2 = (const unsigned char*)pos;
- v = 0;
-
- switch (len & 7) {
- case 7: v ^= (uint64_t)pos2[6] << 48;
- [[clang::fallthrough]];
- case 6: v ^= (uint64_t)pos2[5] << 40;
- [[clang::fallthrough]];
- case 5: v ^= (uint64_t)pos2[4] << 32;
- [[clang::fallthrough]];
- case 4: v ^= (uint64_t)pos2[3] << 24;
- [[clang::fallthrough]];
- case 3: v ^= (uint64_t)pos2[2] << 16;
- [[clang::fallthrough]];
- case 2: v ^= (uint64_t)pos2[1] << 8;
- [[clang::fallthrough]];
- case 1: v ^= (uint64_t)pos2[0];
- h ^= mix(v);
- h *= m;
- }
-
- return mix(h);
-}
+ const uint64_t m = 0x880355f21e6d1965ULL;
+ const uint64_t *pos = (const uint64_t *)buf;
+ const uint64_t *end = pos + (len / 8);
+ const unsigned char *pos2;
+ uint64_t h = seed ^ (len * m);
+ uint64_t v;
+
+ while (pos != end) {
+ v = *pos++;
+ h ^= mix(v);
+ h *= m;
+ }
+
+ pos2 = (const unsigned char*)pos;
+ v = 0;
+
+ switch (len & 7) {
+ case 7: v ^= (uint64_t)pos2[6] << 48;
+ [[clang::fallthrough]];
+ case 6: v ^= (uint64_t)pos2[5] << 40;
+ [[clang::fallthrough]];
+ case 5: v ^= (uint64_t)pos2[4] << 32;
+ [[clang::fallthrough]];
+ case 4: v ^= (uint64_t)pos2[3] << 24;
+ [[clang::fallthrough]];
+ case 3: v ^= (uint64_t)pos2[2] << 16;
+ [[clang::fallthrough]];
+ case 2: v ^= (uint64_t)pos2[1] << 8;
+ [[clang::fallthrough]];
+ case 1: v ^= (uint64_t)pos2[0];
+ h ^= mix(v);
+ h *= m;
+ }
+
+ return mix(h);
+}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static uint32_t
fasthash32(const void *buf, size_t len, uint32_t seed)
{
- // the following trick converts the 64-bit hashcode to Fermat
- // residue, which shall retain information from both the higher
- // and lower parts of hashcode.
- uint64_t h = fasthash64(buf, len, seed);
- return h - (h >> 32);
+ // the following trick converts the 64-bit hashcode to Fermat
+ // residue, which shall retain information from both the higher
+ // and lower parts of hashcode.
+ uint64_t h = fasthash64(buf, len, seed);
+ return h - (h >> 32);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingAddUser(IOTrackingQueue * queue, IOTrackingUser * mem, vm_size_t size)
{
- uint32_t num;
- proc_t self;
-
- if (!queue->captureOn) return;
- if (size < queue->minCaptureSize) return;
-
- assert(!mem->link.next);
-
- num = backtrace(&mem->bt[0], kIOTrackingCallSiteBTs);
- num = 0;
- if ((kernel_task != current_task()) && (self = proc_self()))
- {
- bool user_64;
- mem->btPID = proc_pid(self);
- (void)backtrace_user(&mem->btUser[0], kIOTrackingCallSiteBTs - 1, &num,
- &user_64);
- mem->user32 = !user_64;
- proc_rele(self);
- }
- assert(num <= kIOTrackingCallSiteBTs);
- mem->userCount = num;
-
- IOTRecursiveLockLock(&queue->lock);
- queue_enter/*last*/(&queue->sites[0], mem, IOTrackingUser *, link);
- queue->siteCount++;
- IOTRecursiveLockUnlock(&queue->lock);
+ uint32_t num;
+ proc_t self;
+
+ if (!queue->captureOn) {
+ return;
+ }
+ if (size < queue->minCaptureSize) {
+ return;
+ }
+
+ assert(!mem->link.next);
+
+ num = backtrace(&mem->bt[0], kIOTrackingCallSiteBTs, NULL);
+ num = 0;
+ if ((kernel_task != current_task()) && (self = proc_self())) {
+ bool user_64 = false;
+ mem->btPID = proc_pid(self);
+ (void)backtrace_user(&mem->btUser[0], kIOTrackingCallSiteBTs - 1, &num,
+ &user_64, NULL);
+ mem->user32 = !user_64;
+ proc_rele(self);
+ }
+ assert(num <= kIOTrackingCallSiteBTs);
+ mem->userCount = num;
+
+ IOTRecursiveLockLock(&queue->lock);
+ queue_enter/*last*/ (&queue->sites[0], mem, IOTrackingUser *, link);
+ queue->siteCount++;
+ IOTRecursiveLockUnlock(&queue->lock);
}
void
IOTrackingRemoveUser(IOTrackingQueue * queue, IOTrackingUser * mem)
{
- if (!mem->link.next) return;
-
- IOTRecursiveLockLock(&queue->lock);
- if (mem->link.next)
- {
- remque(&mem->link);
- assert(queue->siteCount);
- queue->siteCount--;
- }
- IOTRecursiveLockUnlock(&queue->lock);
+ if (!mem->link.next) {
+ return;
+ }
+
+ IOTRecursiveLockLock(&queue->lock);
+ if (mem->link.next) {
+ remque(&mem->link);
+ assert(queue->siteCount);
+ queue->siteCount--;
+ }
+ IOTRecursiveLockUnlock(&queue->lock);
}
uint64_t gIOTrackingAddTime;
void
-IOTrackingAdd(IOTrackingQueue * queue, IOTracking * mem, size_t size, bool address)
+IOTrackingAdd(IOTrackingQueue * queue, IOTracking * mem, size_t size, bool address, vm_tag_t tag)
{
- IOTrackingCallSite * site;
- uint32_t crc, num;
- uintptr_t bt[kIOTrackingCallSiteBTs + 1];
- queue_head_t * que;
-
- if (mem->site) return;
- if (!queue->captureOn) return;
- if (size < queue->minCaptureSize) return;
-
- assert(!mem->link.next);
-
- num = backtrace(&bt[0], kIOTrackingCallSiteBTs + 1);
- if (!num) return;
- num--;
- crc = fasthash32(&bt[1], num * sizeof(bt[0]), 0x04C11DB7);
-
- IOTRecursiveLockLock(&queue->lock);
- que = &queue->sites[crc % queue->numSiteQs];
- queue_iterate(que, site, IOTrackingCallSite *, link)
- {
- if (crc == site->crc) break;
- }
-
- if (queue_end(que, (queue_entry_t) site))
- {
- site = (typeof(site)) kalloc(sizeof(IOTrackingCallSite));
-
- queue_init(&site->instances);
- site->addresses = (IOTracking *) &site->instances;
- site->queue = queue;
- site->crc = crc;
- site->count = 0;
- memset(&site->size[0], 0, sizeof(site->size));
- bcopy(&bt[1], &site->bt[0], num * sizeof(site->bt[0]));
- assert(num <= kIOTrackingCallSiteBTs);
- bzero(&site->bt[num], (kIOTrackingCallSiteBTs - num) * sizeof(site->bt[0]));
-
- queue_enter_first(que, site, IOTrackingCallSite *, link);
- queue->siteCount++;
- }
-
- if (address)
- {
- queue_enter/*last*/(&site->instances, mem, IOTracking *, link);
- if (queue_end(&site->instances, (queue_entry_t)site->addresses)) site->addresses = mem;
- }
- else queue_enter_first(&site->instances, mem, IOTracking *, link);
-
- mem->site = site;
- site->size[0] += size;
- site->count++;
-
- IOTRecursiveLockUnlock(&queue->lock);
+ IOTrackingCallSite * site;
+ uint32_t crc, num;
+ uintptr_t bt[kIOTrackingCallSiteBTs + 1];
+ queue_head_t * que;
+
+ if (mem->site) {
+ return;
+ }
+ if (!queue->captureOn) {
+ return;
+ }
+ if (size < queue->minCaptureSize) {
+ return;
+ }
+
+ assert(!mem->link.next);
+
+ num = backtrace(&bt[0], kIOTrackingCallSiteBTs + 1, NULL);
+ if (!num) {
+ return;
+ }
+ num--;
+ crc = fasthash32(&bt[1], num * sizeof(bt[0]), 0x04C11DB7);
+
+ IOTRecursiveLockLock(&queue->lock);
+ que = &queue->sites[crc % queue->numSiteQs];
+ queue_iterate(que, site, IOTrackingCallSite *, link)
+ {
+ if (tag != site->tag) {
+ continue;
+ }
+ if (crc == site->crc) {
+ break;
+ }
+ }
+
+ if (queue_end(que, (queue_entry_t) site)) {
+ site = (typeof(site))kalloc(sizeof(IOTrackingCallSite));
+
+ queue_init(&site->instances);
+ site->addresses = (IOTracking *) &site->instances;
+ site->queue = queue;
+ site->crc = crc;
+ site->count = 0;
+ site->tag = tag;
+ memset(&site->size[0], 0, sizeof(site->size));
+ bcopy(&bt[1], &site->bt[0], num * sizeof(site->bt[0]));
+ assert(num <= kIOTrackingCallSiteBTs);
+ bzero(&site->bt[num], (kIOTrackingCallSiteBTs - num) * sizeof(site->bt[0]));
+
+ queue_enter_first(que, site, IOTrackingCallSite *, link);
+ queue->siteCount++;
+ }
+
+ if (address) {
+ queue_enter/*last*/ (&site->instances, mem, IOTracking *, link);
+ if (queue_end(&site->instances, (queue_entry_t)site->addresses)) {
+ site->addresses = mem;
+ }
+ } else {
+ queue_enter_first(&site->instances, mem, IOTracking *, link);
+ }
+
+ mem->site = site;
+ site->size[0] += size;
+ site->count++;
+
+ IOTRecursiveLockUnlock(&queue->lock);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingRemove(IOTrackingQueue * queue, IOTracking * mem, size_t size)
{
- if (!mem->link.next) return;
-
- IOTRecursiveLockLock(&queue->lock);
- if (mem->link.next)
- {
- assert(mem->site);
-
- if (mem == mem->site->addresses) mem->site->addresses = (IOTracking *) queue_next(&mem->link);
- remque(&mem->link);
-
- assert(mem->site->count);
- mem->site->count--;
- assert(mem->site->size[0] >= size);
- mem->site->size[0] -= size;
- if (!mem->site->count)
- {
- assert(queue_empty(&mem->site->instances));
- assert(!mem->site->size[0]);
- assert(!mem->site->size[1]);
-
- remque(&mem->site->link);
- assert(queue->siteCount);
- queue->siteCount--;
- kfree(mem->site, sizeof(IOTrackingCallSite));
- }
- }
- IOTRecursiveLockUnlock(&queue->lock);
+ if (!mem->link.next) {
+ return;
+ }
+
+ IOTRecursiveLockLock(&queue->lock);
+ if (mem->link.next) {
+ assert(mem->site);
+
+ if (mem == mem->site->addresses) {
+ mem->site->addresses = (IOTracking *) queue_next(&mem->link);
+ }
+ remque(&mem->link);
+
+ assert(mem->site->count);
+ mem->site->count--;
+ assert(mem->site->size[0] >= size);
+ mem->site->size[0] -= size;
+ if (!mem->site->count) {
+ assert(queue_empty(&mem->site->instances));
+ assert(!mem->site->size[0]);
+ assert(!mem->site->size[1]);
+
+ remque(&mem->site->link);
+ assert(queue->siteCount);
+ queue->siteCount--;
+ kfree(mem->site, sizeof(IOTrackingCallSite));
+ }
+ mem->site = NULL;
+ }
+ IOTRecursiveLockUnlock(&queue->lock);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingAlloc(IOTrackingQueue * queue, uintptr_t address, size_t size)
{
- IOTrackingAddress * tracking;
-
- if (!queue->captureOn) return;
- if (size < queue->minCaptureSize) return;
-
- address = ~address;
- tracking = (typeof(tracking)) kalloc(sizeof(IOTrackingAddress));
- bzero(tracking, sizeof(IOTrackingAddress));
- IOTrackingAddressFlags(tracking) |= kTrackingAddressFlagAllocated;
- tracking->address = address;
- tracking->size = size;
-
- IOTrackingAdd(queue, &tracking->tracking, size, true);
+ IOTrackingAddress * tracking;
+
+ if (!queue->captureOn) {
+ return;
+ }
+ if (size < queue->minCaptureSize) {
+ return;
+ }
+
+ address = ~address;
+ tracking = (typeof(tracking))kalloc(sizeof(IOTrackingAddress));
+ bzero(tracking, sizeof(IOTrackingAddress));
+ IOTrackingAddressFlags(tracking) |= kTrackingAddressFlagAllocated;
+ tracking->address = address;
+ tracking->size = size;
+
+ IOTrackingAdd(queue, &tracking->tracking, size, true, VM_KERN_MEMORY_NONE);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingFree(IOTrackingQueue * queue, uintptr_t address, size_t size)
{
- IOTrackingCallSite * site;
- IOTrackingAddress * tracking;
- uint32_t idx;
- bool done;
-
- address = ~address;
- IOTRecursiveLockLock(&queue->lock);
- done = false;
- for (idx = 0; idx < queue->numSiteQs; idx++)
- {
- queue_iterate(&queue->sites[idx], site, IOTrackingCallSite *, link)
- {
- for (tracking = (IOTrackingAddress *) site->addresses;
- !done && !queue_end(&site->instances, &tracking->tracking.link);
- tracking = (IOTrackingAddress *) queue_next(&tracking->tracking.link))
- {
- if ((done = (address == tracking->address)))
- {
- IOTrackingRemove(queue, &tracking->tracking, size);
- kfree(tracking, sizeof(IOTrackingAddress));
- }
- }
- if (done) break;
- }
- if (done) break;
- }
- IOTRecursiveLockUnlock(&queue->lock);
+ IOTrackingCallSite * site;
+ IOTrackingAddress * tracking;
+ uint32_t idx;
+ bool done;
+
+ address = ~address;
+ IOTRecursiveLockLock(&queue->lock);
+ done = false;
+ for (idx = 0; idx < queue->numSiteQs; idx++) {
+ queue_iterate(&queue->sites[idx], site, IOTrackingCallSite *, link)
+ {
+ tracking = (IOTrackingAddress *) site->addresses;
+ while (!queue_end(&site->instances, &tracking->tracking.link)) {
+ if ((done = (address == tracking->address))) {
+ IOTrackingRemove(queue, &tracking->tracking, size);
+ kfree(tracking, sizeof(IOTrackingAddress));
+ break;
+ } else {
+ tracking = (IOTrackingAddress *) queue_next(&tracking->tracking.link);
+ }
+ }
+ if (done) {
+ break;
+ }
+ }
+ if (done) {
+ break;
+ }
+ }
+ IOTRecursiveLockUnlock(&queue->lock);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingAccumSize(IOTrackingQueue * queue, IOTracking * mem, size_t size)
{
- IOTRecursiveLockLock(&queue->lock);
- if (mem->link.next)
- {
- assert(mem->site);
- assert((size > 0) || (mem->site->size[1] >= -size));
- mem->site->size[1] += size;
- };
- IOTRecursiveLockUnlock(&queue->lock);
+ IOTRecursiveLockLock(&queue->lock);
+ if (mem->link.next) {
+ assert(mem->site);
+ assert((size > 0) || (mem->site->size[1] >= -size));
+ mem->site->size[1] += size;
+ }
+ ;
+ IOTRecursiveLockUnlock(&queue->lock);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
void
IOTrackingReset(IOTrackingQueue * queue)
{
- IOTrackingCallSite * site;
- IOTrackingUser * user;
- IOTracking * tracking;
- IOTrackingAddress * trackingAddress;
- uint32_t idx;
- bool addresses;
-
- IOTRecursiveLockLock(&queue->lock);
- for (idx = 0; idx < queue->numSiteQs; idx++)
- {
- while (!queue_empty(&queue->sites[idx]))
- {
- if (kIOTrackingQueueTypeMap & queue->type)
- {
- queue_remove_first(&queue->sites[idx], user, IOTrackingUser *, link);
- user->link.next = user->link.prev = NULL;
- }
- else
- {
- queue_remove_first(&queue->sites[idx], site, IOTrackingCallSite *, link);
- addresses = false;
- while (!queue_empty(&site->instances))
- {
- queue_remove_first(&site->instances, tracking, IOTracking *, link);
- if (tracking == site->addresses) addresses = true;
- if (addresses)
- {
- trackingAddress = (typeof(trackingAddress)) tracking;
- if (kTrackingAddressFlagAllocated & IOTrackingAddressFlags(trackingAddress))
- {
- kfree(tracking, sizeof(IOTrackingAddress));
- }
- }
- }
- kfree(site, sizeof(IOTrackingCallSite));
- }
- }
- }
- queue->siteCount = 0;
- IOTRecursiveLockUnlock(&queue->lock);
+ IOTrackingCallSite * site;
+ IOTrackingUser * user;
+ IOTracking * tracking;
+ IOTrackingAddress * trackingAddress;
+ uint32_t idx;
+ bool addresses;
+
+ IOTRecursiveLockLock(&queue->lock);
+ for (idx = 0; idx < queue->numSiteQs; idx++) {
+ while (!queue_empty(&queue->sites[idx])) {
+ if (kIOTrackingQueueTypeMap & queue->type) {
+ queue_remove_first(&queue->sites[idx], user, IOTrackingUser *, link);
+ user->link.next = user->link.prev = NULL;
+ } else {
+ queue_remove_first(&queue->sites[idx], site, IOTrackingCallSite *, link);
+ addresses = false;
+ while (!queue_empty(&site->instances)) {
+ queue_remove_first(&site->instances, tracking, IOTracking *, link);
+ if (tracking == site->addresses) {
+ addresses = true;
+ }
+ if (addresses) {
+ trackingAddress = (typeof(trackingAddress))tracking;
+ if (kTrackingAddressFlagAllocated & IOTrackingAddressFlags(trackingAddress)) {
+ kfree(tracking, sizeof(IOTrackingAddress));
+ }
+ }
+ }
+ kfree(site, sizeof(IOTrackingCallSite));
+ }
+ }
+ }
+ queue->siteCount = 0;
+ IOTRecursiveLockUnlock(&queue->lock);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static int
IOTrackingCallSiteInfoCompare(const void * left, const void * right)
{
- IOTrackingCallSiteInfo * l = (typeof(l)) left;
- IOTrackingCallSiteInfo * r = (typeof(r)) right;
- size_t lsize, rsize;
+ IOTrackingCallSiteInfo * l = (typeof(l))left;
+ IOTrackingCallSiteInfo * r = (typeof(r))right;
+ size_t lsize, rsize;
- rsize = r->size[0] + r->size[1];
- lsize = l->size[0] + l->size[1];
+ rsize = r->size[0] + r->size[1];
+ lsize = l->size[0] + l->size[1];
- return ((rsize > lsize) ? 1 : ((rsize == lsize) ? 0 : -1));
+ return (rsize > lsize) ? 1 : ((rsize == lsize) ? 0 : -1);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static int
IOTrackingAddressCompare(const void * left, const void * right)
{
- IOTracking * instance;
- uintptr_t inst, laddr, raddr;
-
- inst = ((typeof(inst) *) left)[0];
- instance = (typeof(instance)) INSTANCE_GET(inst);
- if (kInstanceFlagAddress & inst) laddr = ~((IOTrackingAddress *)instance)->address;
- else laddr = (uintptr_t) (instance + 1);
+ IOTracking * instance;
+ uintptr_t inst, laddr, raddr;
+
+ inst = ((typeof(inst) *)left)[0];
+ instance = (typeof(instance))INSTANCE_GET(inst);
+ if (kInstanceFlagAddress & inst) {
+ laddr = ~((IOTrackingAddress *)instance)->address;
+ } else {
+ laddr = (uintptr_t) (instance + 1);
+ }
- inst = ((typeof(inst) *) right)[0];
- instance = (typeof(instance)) (inst & ~kInstanceFlags);
- if (kInstanceFlagAddress & inst) raddr = ~((IOTrackingAddress *)instance)->address;
- else raddr = (uintptr_t) (instance + 1);
+ inst = ((typeof(inst) *)right)[0];
+ instance = (typeof(instance))(inst & ~kInstanceFlags);
+ if (kInstanceFlagAddress & inst) {
+ raddr = ~((IOTrackingAddress *)instance)->address;
+ } else {
+ raddr = (uintptr_t) (instance + 1);
+ }
- return ((laddr > raddr) ? 1 : ((laddr == raddr) ? 0 : -1));
+ return (laddr > raddr) ? 1 : ((laddr == raddr) ? 0 : -1);
}
static int
IOTrackingZoneElementCompare(const void * left, const void * right)
{
- uintptr_t inst, laddr, raddr;
+ uintptr_t inst, laddr, raddr;
- inst = ((typeof(inst) *) left)[0];
- laddr = INSTANCE_PUT(inst);
- inst = ((typeof(inst) *) right)[0];
- raddr = INSTANCE_PUT(inst);
+ inst = ((typeof(inst) *)left)[0];
+ laddr = INSTANCE_PUT(inst);
+ inst = ((typeof(inst) *)right)[0];
+ raddr = INSTANCE_PUT(inst);
- return ((laddr > raddr) ? 1 : ((laddr == raddr) ? 0 : -1));
+ return (laddr > raddr) ? 1 : ((laddr == raddr) ? 0 : -1);
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static void
CopyOutKernelBacktrace(IOTrackingCallSite * site, IOTrackingCallSiteInfo * siteInfo)
{
- uint32_t j;
- mach_vm_address_t bt, btEntry;
-
- btEntry = site->queue->btEntry;
- for (j = 0; j < kIOTrackingCallSiteBTs; j++)
- {
- bt = site->bt[j];
- if (btEntry
- && (!bt || (j == (kIOTrackingCallSiteBTs - 1))))
- {
- bt = btEntry;
- btEntry = 0;
- }
- siteInfo->bt[0][j] = VM_KERNEL_UNSLIDE(bt);
- }
+ uint32_t j;
+ mach_vm_address_t bt, btEntry;
+
+ btEntry = site->queue->btEntry;
+ for (j = 0; j < kIOTrackingCallSiteBTs; j++) {
+ bt = site->bt[j];
+ if (btEntry
+ && (!bt || (j == (kIOTrackingCallSiteBTs - 1)))) {
+ bt = btEntry;
+ btEntry = 0;
+ }
+ siteInfo->bt[0][j] = VM_KERNEL_UNSLIDE(bt);
+ }
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static void
IOTrackingLeakScan(void * refcon)
{
- IOTrackingLeaksRef * ref = (typeof(ref)) refcon;
- uintptr_t * instances;
- IOTracking * instance;
- uint64_t vaddr, vincr;
- ppnum_t ppn;
- uintptr_t ptr, addr, vphysaddr, inst;
- size_t size;
- uint32_t baseIdx, lim, ptrIdx, count;
- boolean_t is;
- AbsoluteTime deadline;
-
- instances = ref->instances;
- count = ref->count;
- size = ref->zoneSize;
-
- for (deadline = 0, vaddr = VM_MIN_KERNEL_AND_KEXT_ADDRESS;
- ;
- vaddr += vincr)
- {
- if ((mach_absolute_time() > deadline) || (vaddr >= VM_MAX_KERNEL_ADDRESS))
- {
- if (deadline)
- {
- ml_set_interrupts_enabled(is);
- IODelay(10);
- }
- if (vaddr >= VM_MAX_KERNEL_ADDRESS) break;
- is = ml_set_interrupts_enabled(false);
- clock_interval_to_deadline(10, kMillisecondScale, &deadline);
- }
-
- ppn = kernel_pmap_present_mapping(vaddr, &vincr, &vphysaddr);
- // check noencrypt to avoid VM structs (map entries) with pointers
- if (ppn && (!pmap_valid_page(ppn) || (!ref->zoneSize && pmap_is_noencrypt(ppn)))) ppn = 0;
- if (!ppn) continue;
-
- for (ptrIdx = 0; ptrIdx < (page_size / sizeof(uintptr_t)); ptrIdx++)
- {
- ptr = ((uintptr_t *)vphysaddr)[ptrIdx];
-
- for (lim = count, baseIdx = 0; lim; lim >>= 1)
- {
- inst = instances[baseIdx + (lim >> 1)];
- instance = (typeof(instance)) INSTANCE_GET(inst);
-
- if (ref->zoneSize)
- {
- addr = INSTANCE_PUT(inst) & ~kInstanceFlags;
- }
- else if (kInstanceFlagAddress & inst)
- {
- addr = ~((IOTrackingAddress *)instance)->address;
- size = ((IOTrackingAddress *)instance)->size;
- }
- else
- {
- addr = (uintptr_t) (instance + 1);
- size = instance->site->queue->allocSize;
- }
- if ((ptr >= addr) && (ptr < (addr + size))
-
- && (((vaddr + ptrIdx * sizeof(uintptr_t)) < addr)
- || ((vaddr + ptrIdx * sizeof(uintptr_t)) >= (addr + size))))
- {
- if (!(kInstanceFlagReferenced & inst))
- {
- inst |= kInstanceFlagReferenced;
- instances[baseIdx + (lim >> 1)] = inst;
- ref->found++;
- }
- break;
- }
- if (ptr > addr)
- {
- // move right
- baseIdx += (lim >> 1) + 1;
- lim--;
- }
- // else move left
- }
- }
- ref->bytes += page_size;
- }
+ IOTrackingLeaksRef * ref = (typeof(ref))refcon;
+ uintptr_t * instances;
+ IOTracking * instance;
+ uint64_t vaddr, vincr;
+ ppnum_t ppn;
+ uintptr_t ptr, addr, vphysaddr, inst;
+ size_t size, origsize;
+ uint32_t baseIdx, lim, ptrIdx, count;
+ boolean_t is;
+ AbsoluteTime deadline;
+
+ instances = ref->instances;
+ count = ref->count;
+ size = origsize = ref->zoneSize;
+
+ for (deadline = 0, vaddr = VM_MIN_KERNEL_AND_KEXT_ADDRESS;
+ ;
+ vaddr += vincr) {
+ if ((mach_absolute_time() > deadline) || (vaddr >= VM_MAX_KERNEL_ADDRESS)) {
+ if (deadline) {
+ ml_set_interrupts_enabled(is);
+ IODelay(10);
+ }
+ if (vaddr >= VM_MAX_KERNEL_ADDRESS) {
+ break;
+ }
+ is = ml_set_interrupts_enabled(false);
+ clock_interval_to_deadline(10, kMillisecondScale, &deadline);
+ }
+
+ ppn = kernel_pmap_present_mapping(vaddr, &vincr, &vphysaddr);
+ // check noencrypt to avoid VM structs (map entries) with pointers
+ if (ppn && (!pmap_valid_page(ppn) || (!ref->zoneSize && pmap_is_noencrypt(ppn)))) {
+ ppn = 0;
+ }
+ if (!ppn) {
+ continue;
+ }
+
+ for (ptrIdx = 0; ptrIdx < (page_size / sizeof(uintptr_t)); ptrIdx++) {
+ ptr = ((uintptr_t *)vphysaddr)[ptrIdx];
+
+ for (lim = count, baseIdx = 0; lim; lim >>= 1) {
+ inst = instances[baseIdx + (lim >> 1)];
+ instance = (typeof(instance))INSTANCE_GET(inst);
+
+ if (ref->zoneSize) {
+ addr = INSTANCE_PUT(inst) & ~kInstanceFlags;
+ } else if (kInstanceFlagAddress & inst) {
+ addr = ~((IOTrackingAddress *)instance)->address;
+ origsize = size = ((IOTrackingAddress *)instance)->size;
+ if (!size) {
+ size = 1;
+ }
+ } else {
+ addr = (uintptr_t) (instance + 1);
+ origsize = size = instance->site->queue->allocSize;
+ }
+ if ((ptr >= addr) && (ptr < (addr + size))
+
+ && (((vaddr + ptrIdx * sizeof(uintptr_t)) < addr)
+ || ((vaddr + ptrIdx * sizeof(uintptr_t)) >= (addr + size)))) {
+ if (!(kInstanceFlagReferenced & inst)) {
+ inst |= kInstanceFlagReferenced;
+ instances[baseIdx + (lim >> 1)] = inst;
+ ref->found++;
+ if (!origsize) {
+ ref->foundzlen++;
+ }
+ }
+ break;
+ }
+ if (ptr > addr) {
+ // move right
+ baseIdx += (lim >> 1) + 1;
+ lim--;
+ }
+ // else move left
+ }
+ }
+ ref->bytes += page_size;
+ }
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
extern "C" void
zone_leaks_scan(uintptr_t * instances, uint32_t count, uint32_t zoneSize, uint32_t * found)
{
- IOTrackingLeaksRef ref;
- IOTrackingCallSiteInfo siteInfo;
- uint32_t idx;
-
- qsort(instances, count, sizeof(*instances), &IOTrackingZoneElementCompare);
-
- bzero(&siteInfo, sizeof(siteInfo));
- bzero(&ref, sizeof(ref));
- ref.instances = instances;
- ref.count = count;
- ref.zoneSize = zoneSize;
-
- for (idx = 0; idx < 2; idx++)
- {
- ref.bytes = 0;
- IOTrackingLeakScan(&ref);
- IOLog("leaks(%d) scanned %ld MB, instance count %d, found %d\n", idx, ref.bytes / 1024 / 1024, count, ref.found);
- if (count <= ref.found) break;
- }
-
- *found = ref.found;
+ IOTrackingLeaksRef ref;
+ IOTrackingCallSiteInfo siteInfo;
+ uint32_t idx;
+
+ qsort(instances, count, sizeof(*instances), &IOTrackingZoneElementCompare);
+
+ bzero(&siteInfo, sizeof(siteInfo));
+ bzero(&ref, sizeof(ref));
+ ref.instances = instances;
+ ref.count = count;
+ ref.zoneSize = zoneSize;
+
+ for (idx = 0; idx < 2; idx++) {
+ ref.bytes = 0;
+ IOTrackingLeakScan(&ref);
+ IOLog("leaks(%d) scanned %ld MB, instance count %d, found %d\n", idx, ref.bytes / 1024 / 1024, count, ref.found);
+ if (count <= ref.found) {
+ break;
+ }
+ }
+
+ *found = ref.found;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static void
ZoneSiteProc(void * refCon, uint32_t siteCount, uint32_t zoneSize,
- uintptr_t * backtrace, uint32_t btCount)
+ uintptr_t * backtrace, uint32_t btCount)
{
- IOTrackingCallSiteInfo siteInfo;
- OSData * leakData;
- uint32_t idx;
+ IOTrackingCallSiteInfo siteInfo;
+ OSData * leakData;
+ uint32_t idx;
- leakData = (typeof(leakData)) refCon;
+ leakData = (typeof(leakData))refCon;
- bzero(&siteInfo, sizeof(siteInfo));
- siteInfo.count = siteCount;
- siteInfo.size[0] = zoneSize * siteCount;
+ bzero(&siteInfo, sizeof(siteInfo));
+ siteInfo.count = siteCount;
+ siteInfo.size[0] = zoneSize * siteCount;
- for (idx = 0; (idx < btCount) && (idx < kIOTrackingCallSiteBTs); idx++)
- {
- siteInfo.bt[0][idx] = VM_KERNEL_UNSLIDE(backtrace[idx]);
- }
+ for (idx = 0; (idx < btCount) && (idx < kIOTrackingCallSiteBTs); idx++) {
+ siteInfo.bt[0][idx] = VM_KERNEL_UNSLIDE(backtrace[idx]);
+ }
- leakData->appendBytes(&siteInfo, sizeof(siteInfo));
+ leakData->appendBytes(&siteInfo, sizeof(siteInfo));
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static OSData *
-IOTrackingLeaks(OSData * data)
+IOTrackingLeaks(LIBKERN_CONSUMED OSData * data)
{
- IOTrackingLeaksRef ref;
- IOTrackingCallSiteInfo siteInfo;
- IOTrackingCallSite * site;
- OSData * leakData;
- uintptr_t * instances;
- IOTracking * instance;
- uintptr_t inst;
- uint32_t count, idx, numSites, dups, siteCount;
-
- instances = (typeof(instances)) data->getBytesNoCopy();
- count = (data->getLength() / sizeof(*instances));
- qsort(instances, count, sizeof(*instances), &IOTrackingAddressCompare);
-
- bzero(&siteInfo, sizeof(siteInfo));
- bzero(&ref, sizeof(ref));
- ref.instances = instances;
- ref.count = count;
- for (idx = 0; idx < 2; idx++)
- {
- ref.bytes = 0;
- IOTrackingLeakScan(&ref);
- IOLog("leaks(%d) scanned %ld MB, instance count %d, found %d\n", idx, ref.bytes / 1024 / 1024, count, ref.found);
- if (count <= ref.found) break;
- }
-
- leakData = OSData::withCapacity(128 * sizeof(IOTrackingCallSiteInfo));
-
- for (numSites = 0, idx = 0; idx < count; idx++)
- {
- inst = instances[idx];
- if (kInstanceFlagReferenced & inst) continue;
- instance = (typeof(instance)) INSTANCE_GET(inst);
- site = instance->site;
- instances[numSites] = (uintptr_t) site;
- numSites++;
- }
-
- for (idx = 0; idx < numSites; idx++)
- {
- inst = instances[idx];
- if (!inst) continue;
- site = (typeof(site)) inst;
- for (siteCount = 1, dups = (idx + 1); dups < numSites; dups++)
- {
- if (instances[dups] == (uintptr_t) site)
- {
- siteCount++;
- instances[dups] = 0;
- }
+ IOTrackingLeaksRef ref;
+ IOTrackingCallSiteInfo siteInfo;
+ IOTrackingCallSite * site;
+ OSData * leakData;
+ uintptr_t * instances;
+ IOTracking * instance;
+ uintptr_t inst;
+ uint32_t count, idx, numSites, dups, siteCount;
+
+ instances = (typeof(instances))data->getBytesNoCopy();
+ count = (data->getLength() / sizeof(*instances));
+ qsort(instances, count, sizeof(*instances), &IOTrackingAddressCompare);
+
+ bzero(&siteInfo, sizeof(siteInfo));
+ bzero(&ref, sizeof(ref));
+ ref.instances = instances;
+ ref.count = count;
+ for (idx = 0; idx < 2; idx++) {
+ ref.bytes = 0;
+ IOTrackingLeakScan(&ref);
+ IOLog("leaks(%d) scanned %ld MB, instance count %d, found %d (zlen %d)\n", idx, ref.bytes / 1024 / 1024, count, ref.found, ref.foundzlen);
+ if (count <= ref.found) {
+ break;
+ }
+ }
+
+ leakData = OSData::withCapacity(128 * sizeof(IOTrackingCallSiteInfo));
+
+ for (numSites = 0, idx = 0; idx < count; idx++) {
+ inst = instances[idx];
+ if (kInstanceFlagReferenced & inst) {
+ continue;
+ }
+ instance = (typeof(instance))INSTANCE_GET(inst);
+ site = instance->site;
+ instances[numSites] = (uintptr_t) site;
+ numSites++;
}
- siteInfo.count = siteCount;
- siteInfo.size[0] = (site->size[0] * site->count) / siteCount;
- siteInfo.size[1] = (site->size[1] * site->count) / siteCount;;
- CopyOutKernelBacktrace(site, &siteInfo);
- leakData->appendBytes(&siteInfo, sizeof(siteInfo));
- }
- data->release();
-
- return (leakData);
+
+ for (idx = 0; idx < numSites; idx++) {
+ inst = instances[idx];
+ if (!inst) {
+ continue;
+ }
+ site = (typeof(site))inst;
+ for (siteCount = 1, dups = (idx + 1); dups < numSites; dups++) {
+ if (instances[dups] == (uintptr_t) site) {
+ siteCount++;
+ instances[dups] = 0;
+ }
+ }
+ siteInfo.count = siteCount;
+ siteInfo.size[0] = (site->size[0] * site->count) / siteCount;
+ siteInfo.size[1] = (site->size[1] * site->count) / siteCount;;
+ CopyOutKernelBacktrace(site, &siteInfo);
+ leakData->appendBytes(&siteInfo, sizeof(siteInfo));
+ }
+ data->release();
+
+ return leakData;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
static bool
SkipName(uint32_t options, const char * name, size_t namesLen, const char * names)
{
- const char * scan;
- const char * next;
- bool exclude, found;
- size_t qLen, sLen;
-
- if (!namesLen || !names) return (false);
- // <len><name>...<len><name><0>
- exclude = (0 != (kIOTrackingExcludeNames & options));
- qLen = strlen(name);
- scan = names;
- found = false;
- do
- {
- sLen = scan[0];
- scan++;
- next = scan + sLen;
- if (next >= (names + namesLen)) break;
- found = ((sLen == qLen) && !strncmp(scan, name, sLen));
- scan = next;
- }
- while (!found && (scan < (names + namesLen)));
-
- return (!(exclude ^ found));
+ const char * scan;
+ const char * next;
+ bool exclude, found;
+ size_t qLen, sLen;
+
+ if (!namesLen || !names) {
+ return false;
+ }
+ // <len><name>...<len><name><0>
+ exclude = (0 != (kIOTrackingExcludeNames & options));
+ qLen = strlen(name);
+ scan = names;
+ found = false;
+ do{
+ sLen = scan[0];
+ scan++;
+ next = scan + sLen;
+ if (next >= (names + namesLen)) {
+ break;
+ }
+ found = ((sLen == qLen) && !strncmp(scan, name, sLen));
+ scan = next;
+ }while (!found && (scan < (names + namesLen)));
+
+ return !(exclude ^ found);
}
#endif /* IOTRACKING */
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-kern_return_t
+static kern_return_t
IOTrackingDebug(uint32_t selector, uint32_t options, uint64_t value,
- const char * names, size_t namesLen,
- size_t size, OSObject ** result)
+ uint32_t intag, uint32_t inzsize,
+ const char * names, size_t namesLen,
+ size_t size, OSObject ** result)
{
- kern_return_t ret;
- OSData * data;
+ kern_return_t ret;
+ OSData * data;
- if (result) *result = 0;
- data = 0;
- ret = kIOReturnNotReady;
+ if (result) {
+ *result = NULL;
+ }
+ data = NULL;
+ ret = kIOReturnNotReady;
#if IOTRACKING
- kern_return_t kr;
- IOTrackingQueue * queue;
- IOTracking * instance;
- IOTrackingCallSite * site;
- IOTrackingCallSiteInfo siteInfo;
- IOTrackingUser * user;
- task_t mapTask;
- mach_vm_address_t mapAddress;
- mach_vm_size_t mapSize;
- uint32_t num, idx, qIdx;
- uintptr_t instFlags;
- proc_t proc;
- bool addresses;
-
- ret = kIOReturnNotFound;
- proc = NULL;
- if (kIOTrackingGetMappings == selector)
- {
- if (value != -1ULL)
- {
- proc = proc_find(value);
- if (!proc) return (kIOReturnNotFound);
+ kern_return_t kr;
+ IOTrackingQueue * queue;
+ IOTracking * instance;
+ IOTrackingCallSite * site;
+ IOTrackingCallSiteInfo siteInfo;
+ IOTrackingUser * user;
+ task_t mapTask;
+ mach_vm_address_t mapAddress;
+ mach_vm_size_t mapSize;
+ uint32_t num, idx, qIdx;
+ uintptr_t instFlags;
+ proc_t proc;
+ bool addresses;
+
+ ret = kIOReturnNotFound;
+ proc = NULL;
+ if (kIOTrackingGetMappings == selector) {
+ if (value != -1ULL) {
+ proc = proc_find(value);
+ if (!proc) {
+ return kIOReturnNotFound;
+ }
+ }
}
- }
-
- bzero(&siteInfo, sizeof(siteInfo));
- lck_mtx_lock(gIOTrackingLock);
- queue_iterate(&gIOTrackingQ, queue, IOTrackingQueue *, link)
- {
- if (SkipName(options, queue->name, namesLen, names)) continue;
-
- if (!(kIOTracking & gIOKitDebug) && (kIOTrackingQueueTypeAlloc & queue->type)) continue;
-
- switch (selector)
- {
- case kIOTrackingResetTracking:
- {
- IOTrackingReset(queue);
- ret = kIOReturnSuccess;
- break;
- }
-
- case kIOTrackingStartCapture:
- case kIOTrackingStopCapture:
- {
- queue->captureOn = (kIOTrackingStartCapture == selector);
- ret = kIOReturnSuccess;
- break;
- }
-
- case kIOTrackingSetMinCaptureSize:
- {
- queue->minCaptureSize = size;
- ret = kIOReturnSuccess;
- break;
- }
-
- case kIOTrackingLeaks:
- {
- if (!(kIOTrackingQueueTypeAlloc & queue->type)) break;
-
- if (!data) data = OSData::withCapacity(1024 * sizeof(uintptr_t));
-
- IOTRecursiveLockLock(&queue->lock);
- for (idx = 0; idx < queue->numSiteQs; idx++)
- {
- queue_iterate(&queue->sites[idx], site, IOTrackingCallSite *, link)
- {
- addresses = false;
- queue_iterate(&site->instances, instance, IOTracking *, link)
- {
- if (instance == site->addresses) addresses = true;
- instFlags = (typeof(instFlags)) instance;
- if (addresses) instFlags |= kInstanceFlagAddress;
- data->appendBytes(&instFlags, sizeof(instFlags));
- }
- }
- }
- // queue is locked
- ret = kIOReturnSuccess;
- break;
- }
-
- case kIOTrackingGetTracking:
- {
- if (kIOTrackingQueueTypeMap & queue->type) break;
-
- if (!data) data = OSData::withCapacity(128 * sizeof(IOTrackingCallSiteInfo));
-
- IOTRecursiveLockLock(&queue->lock);
- num = queue->siteCount;
- idx = 0;
- for (qIdx = 0; qIdx < queue->numSiteQs; qIdx++)
- {
- queue_iterate(&queue->sites[qIdx], site, IOTrackingCallSite *, link)
- {
- assert(idx < num);
- idx++;
-
- if (size && ((site->size[0] + site->size[1]) < size)) continue;
-
- siteInfo.count = site->count;
- siteInfo.size[0] = site->size[0];
- siteInfo.size[1] = site->size[1];
-
- CopyOutKernelBacktrace(site, &siteInfo);
- data->appendBytes(&siteInfo, sizeof(siteInfo));
- }
- }
- assert(idx == num);
- IOTRecursiveLockUnlock(&queue->lock);
- ret = kIOReturnSuccess;
- break;
- }
-
- case kIOTrackingGetMappings:
- {
- if (!(kIOTrackingQueueTypeMap & queue->type)) break;
- if (!data) data = OSData::withCapacity(page_size);
-
- IOTRecursiveLockLock(&queue->lock);
- num = queue->siteCount;
- idx = 0;
- for (qIdx = 0; qIdx < queue->numSiteQs; qIdx++)
- {
- queue_iterate(&queue->sites[qIdx], user, IOTrackingUser *, link)
- {
- assert(idx < num);
- idx++;
-
- kr = IOMemoryMapTracking(user, &mapTask, &mapAddress, &mapSize);
- if (kIOReturnSuccess != kr) continue;
- if (proc && (mapTask != proc_task(proc))) continue;
- if (size && (mapSize < size)) continue;
-
- siteInfo.count = 1;
- siteInfo.size[0] = mapSize;
- siteInfo.address = mapAddress;
- siteInfo.addressPID = task_pid(mapTask);
- siteInfo.btPID = user->btPID;
-
- for (uint32_t j = 0; j < kIOTrackingCallSiteBTs; j++)
- {
- siteInfo.bt[0][j] = VM_KERNEL_UNSLIDE(user->bt[j]);
- }
- uint32_t * bt32 = (typeof(bt32)) &user->btUser[0];
- uint64_t * bt64 = (typeof(bt64)) ((void *) &user->btUser[0]);
- for (uint32_t j = 0; j < kIOTrackingCallSiteBTs; j++)
- {
- if (j >= user->userCount) siteInfo.bt[1][j] = 0;
- else if (user->user32) siteInfo.bt[1][j] = bt32[j];
- else siteInfo.bt[1][j] = bt64[j];
- }
- data->appendBytes(&siteInfo, sizeof(siteInfo));
- }
- }
- assert(idx == num);
- IOTRecursiveLockUnlock(&queue->lock);
- ret = kIOReturnSuccess;
- break;
- }
-
- default:
- ret = kIOReturnUnsupported;
- break;
- }
- }
-
- if ((kIOTrackingLeaks == selector) && data)
- {
- data = IOTrackingLeaks(data);
- queue_iterate(&gIOTrackingQ, queue, IOTrackingQueue *, link)
- {
- if (SkipName(options, queue->name, namesLen, names)) continue;
- if (!(kIOTrackingQueueTypeAlloc & queue->type)) continue;
- IOTRecursiveLockUnlock(&queue->lock);
- }
- }
-
- lck_mtx_unlock(gIOTrackingLock);
-
- if ((kIOTrackingLeaks == selector) && namesLen && names)
- {
- const char * scan;
- const char * next;
- size_t sLen;
-
- if (!data) data = OSData::withCapacity(4096 * sizeof(uintptr_t));
-
- // <len><name>...<len><name><0>
- scan = names;
- do
- {
- sLen = scan[0];
- scan++;
- next = scan + sLen;
- if (next >= (names + namesLen)) break;
- kr = zone_leaks(scan, sLen, &ZoneSiteProc, data);
- if (KERN_SUCCESS == kr) ret = kIOReturnSuccess;
- else if (KERN_INVALID_NAME != kr) ret = kIOReturnVMError;
- scan = next;
- }
- while (scan < (names + namesLen));
- }
-
- if (data) switch (selector)
- {
- case kIOTrackingLeaks:
- case kIOTrackingGetTracking:
- case kIOTrackingGetMappings:
+
+ bzero(&siteInfo, sizeof(siteInfo));
+ lck_mtx_lock(gIOTrackingLock);
+ queue_iterate(&gIOTrackingQ, queue, IOTrackingQueue *, link)
{
- IOTrackingCallSiteInfo * siteInfos;
- siteInfos = (typeof(siteInfos)) data->getBytesNoCopy();
- num = (data->getLength() / sizeof(*siteInfos));
- qsort(siteInfos, num, sizeof(*siteInfos), &IOTrackingCallSiteInfoCompare);
- break;
+ if (SkipName(options, queue->name, namesLen, names)) {
+ continue;
+ }
+
+ if (!(kIOTracking & gIOKitDebug) && (kIOTrackingQueueTypeAlloc & queue->type)) {
+ continue;
+ }
+
+ switch (selector) {
+ case kIOTrackingResetTracking:
+ {
+ IOTrackingReset(queue);
+ ret = kIOReturnSuccess;
+ break;
+ }
+
+ case kIOTrackingStartCapture:
+ case kIOTrackingStopCapture:
+ {
+ queue->captureOn = (kIOTrackingStartCapture == selector);
+ ret = kIOReturnSuccess;
+ break;
+ }
+
+ case kIOTrackingSetMinCaptureSize:
+ {
+ queue->minCaptureSize = size;
+ ret = kIOReturnSuccess;
+ break;
+ }
+
+ case kIOTrackingLeaks:
+ {
+ if (!(kIOTrackingQueueTypeAlloc & queue->type)) {
+ break;
+ }
+
+ if (!data) {
+ data = OSData::withCapacity(1024 * sizeof(uintptr_t));
+ }
+
+ IOTRecursiveLockLock(&queue->lock);
+ for (idx = 0; idx < queue->numSiteQs; idx++) {
+ queue_iterate(&queue->sites[idx], site, IOTrackingCallSite *, link)
+ {
+ addresses = false;
+ queue_iterate(&site->instances, instance, IOTracking *, link)
+ {
+ if (instance == site->addresses) {
+ addresses = true;
+ }
+ instFlags = (typeof(instFlags))instance;
+ if (addresses) {
+ instFlags |= kInstanceFlagAddress;
+ }
+ data->appendBytes(&instFlags, sizeof(instFlags));
+ }
+ }
+ }
+ // queue is locked
+ ret = kIOReturnSuccess;
+ break;
+ }
+
+
+ case kIOTrackingGetTracking:
+ {
+ if (kIOTrackingQueueTypeMap & queue->type) {
+ break;
+ }
+
+ if (!data) {
+ data = OSData::withCapacity(128 * sizeof(IOTrackingCallSiteInfo));
+ }
+
+ IOTRecursiveLockLock(&queue->lock);
+ num = queue->siteCount;
+ idx = 0;
+ for (qIdx = 0; qIdx < queue->numSiteQs; qIdx++) {
+ queue_iterate(&queue->sites[qIdx], site, IOTrackingCallSite *, link)
+ {
+ assert(idx < num);
+ idx++;
+
+ size_t tsize[2];
+ uint32_t count = site->count;
+ tsize[0] = site->size[0];
+ tsize[1] = site->size[1];
+
+ if (intag || inzsize) {
+ uintptr_t addr;
+ vm_size_t size, zoneSize;
+ vm_tag_t tag;
+
+ if (kIOTrackingQueueTypeAlloc & queue->type) {
+ addresses = false;
+ count = 0;
+ tsize[0] = tsize[1] = 0;
+ queue_iterate(&site->instances, instance, IOTracking *, link)
+ {
+ if (instance == site->addresses) {
+ addresses = true;
+ }
+
+ if (addresses) {
+ addr = ~((IOTrackingAddress *)instance)->address;
+ } else {
+ addr = (uintptr_t) (instance + 1);
+ }
+
+ kr = vm_kern_allocation_info(addr, &size, &tag, &zoneSize);
+ if (KERN_SUCCESS != kr) {
+ continue;
+ }
+
+ if ((VM_KERN_MEMORY_NONE != intag) && (intag != tag)) {
+ continue;
+ }
+ if (inzsize && (inzsize != zoneSize)) {
+ continue;
+ }
+
+ count++;
+ tsize[0] += size;
+ }
+ } else {
+ if (!intag || inzsize || (intag != site->tag)) {
+ continue;
+ }
+ }
+ }
+
+ if (!count) {
+ continue;
+ }
+ if (size && ((tsize[0] + tsize[1]) < size)) {
+ continue;
+ }
+
+ siteInfo.count = count;
+ siteInfo.size[0] = tsize[0];
+ siteInfo.size[1] = tsize[1];
+
+ CopyOutKernelBacktrace(site, &siteInfo);
+ data->appendBytes(&siteInfo, sizeof(siteInfo));
+ }
+ }
+ assert(idx == num);
+ IOTRecursiveLockUnlock(&queue->lock);
+ ret = kIOReturnSuccess;
+ break;
+ }
+
+ case kIOTrackingGetMappings:
+ {
+ if (!(kIOTrackingQueueTypeMap & queue->type)) {
+ break;
+ }
+ if (!data) {
+ data = OSData::withCapacity(page_size);
+ }
+
+ IOTRecursiveLockLock(&queue->lock);
+ num = queue->siteCount;
+ idx = 0;
+ for (qIdx = 0; qIdx < queue->numSiteQs; qIdx++) {
+ queue_iterate(&queue->sites[qIdx], user, IOTrackingUser *, link)
+ {
+ assert(idx < num);
+ idx++;
+
+ kr = IOMemoryMapTracking(user, &mapTask, &mapAddress, &mapSize);
+ if (kIOReturnSuccess != kr) {
+ continue;
+ }
+ if (proc && (mapTask != proc_task(proc))) {
+ continue;
+ }
+ if (size && (mapSize < size)) {
+ continue;
+ }
+
+ siteInfo.count = 1;
+ siteInfo.size[0] = mapSize;
+ siteInfo.address = mapAddress;
+ siteInfo.addressPID = task_pid(mapTask);
+ siteInfo.btPID = user->btPID;
+
+ for (uint32_t j = 0; j < kIOTrackingCallSiteBTs; j++) {
+ siteInfo.bt[0][j] = VM_KERNEL_UNSLIDE(user->bt[j]);
+ }
+ uint32_t * bt32 = (typeof(bt32)) & user->btUser[0];
+ uint64_t * bt64 = (typeof(bt64))((void *) &user->btUser[0]);
+ for (uint32_t j = 0; j < kIOTrackingCallSiteBTs; j++) {
+ if (j >= user->userCount) {
+ siteInfo.bt[1][j] = 0;
+ } else if (user->user32) {
+ siteInfo.bt[1][j] = bt32[j];
+ } else {
+ siteInfo.bt[1][j] = bt64[j];
+ }
+ }
+ data->appendBytes(&siteInfo, sizeof(siteInfo));
+ }
+ }
+ assert(idx == num);
+ IOTRecursiveLockUnlock(&queue->lock);
+ ret = kIOReturnSuccess;
+ break;
+ }
+
+ default:
+ ret = kIOReturnUnsupported;
+ break;
+ }
+ }
+
+ if ((kIOTrackingLeaks == selector) && data) {
+ data = IOTrackingLeaks(data);
+ queue_iterate(&gIOTrackingQ, queue, IOTrackingQueue *, link)
+ {
+ if (SkipName(options, queue->name, namesLen, names)) {
+ continue;
+ }
+ if (!(kIOTrackingQueueTypeAlloc & queue->type)) {
+ continue;
+ }
+ IOTRecursiveLockUnlock(&queue->lock);
+ }
+ }
+
+ lck_mtx_unlock(gIOTrackingLock);
+
+ if ((kIOTrackingLeaks == selector) && namesLen && names) {
+ const char * scan;
+ const char * next;
+ size_t sLen;
+
+ if (!data) {
+ data = OSData::withCapacity(4096 * sizeof(uintptr_t));
+ }
+
+ // <len><name>...<len><name><0>
+ scan = names;
+ do{
+ sLen = scan[0];
+ scan++;
+ next = scan + sLen;
+ if (next >= (names + namesLen)) {
+ break;
+ }
+ kr = zone_leaks(scan, sLen, &ZoneSiteProc, data);
+ if (KERN_SUCCESS == kr) {
+ ret = kIOReturnSuccess;
+ } else if (KERN_INVALID_NAME != kr) {
+ ret = kIOReturnVMError;
+ }
+ scan = next;
+ }while (scan < (names + namesLen));
+ }
+
+ if (data) {
+ switch (selector) {
+ case kIOTrackingLeaks:
+ case kIOTrackingGetTracking:
+ case kIOTrackingGetMappings:
+ {
+ IOTrackingCallSiteInfo * siteInfos;
+ siteInfos = (typeof(siteInfos))data->getBytesNoCopy();
+ num = (data->getLength() / sizeof(*siteInfos));
+ qsort(siteInfos, num, sizeof(*siteInfos), &IOTrackingCallSiteInfoCompare);
+ break;
+ }
+ default: assert(false); break;
+ }
}
- default: assert(false); break;
- }
- *result = data;
- if (proc) proc_rele(proc);
+ *result = data;
+ if (proc) {
+ proc_rele(proc);
+ }
#endif /* IOTRACKING */
- return (ret);
+ return ret;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
IOUserClient * IOKitDiagnosticsClient::withTask(task_t owningTask)
{
- IOKitDiagnosticsClient * inst;
+ IOKitDiagnosticsClient * inst;
- inst = new IOKitDiagnosticsClient;
- if (inst && !inst->init())
- {
- inst->release();
- inst = 0;
- }
+ inst = new IOKitDiagnosticsClient;
+ if (inst && !inst->init()) {
+ inst->release();
+ inst = NULL;
+ }
- return (inst);
+ return inst;
}
-IOReturn IOKitDiagnosticsClient::clientClose(void)
+IOReturn
+IOKitDiagnosticsClient::clientClose(void)
{
- terminate();
- return (kIOReturnSuccess);
+ terminate();
+ return kIOReturnSuccess;
}
-IOReturn IOKitDiagnosticsClient::setProperties(OSObject * properties)
+IOReturn
+IOKitDiagnosticsClient::setProperties(OSObject * properties)
{
- IOReturn kr = kIOReturnUnsupported;
- return (kr);
+ IOReturn kr = kIOReturnUnsupported;
+ return kr;
}
-IOReturn IOKitDiagnosticsClient::externalMethod(uint32_t selector, IOExternalMethodArguments * args,
- IOExternalMethodDispatch * dispatch, OSObject * target, void * reference)
+IOReturn
+IOKitDiagnosticsClient::externalMethod(uint32_t selector, IOExternalMethodArguments * args,
+ IOExternalMethodDispatch * dispatch, OSObject * target, void * reference)
{
- IOReturn ret = kIOReturnBadArgument;
- const IOKitDiagnosticsParameters * params;
- const char * names;
- size_t namesLen;
- OSObject * result;
-
- if (args->structureInputSize < sizeof(IOKitDiagnosticsParameters)) return (kIOReturnBadArgument);
- params = (typeof(params)) args->structureInput;
- if (!params) return (kIOReturnBadArgument);
+ IOReturn ret = kIOReturnBadArgument;
+ const IOKitDiagnosticsParameters * params;
+ const char * names;
+ size_t namesLen;
+ OSObject * result;
+
+ if (args->structureInputSize < sizeof(IOKitDiagnosticsParameters)) {
+ return kIOReturnBadArgument;
+ }
+ params = (typeof(params))args->structureInput;
+ if (!params) {
+ return kIOReturnBadArgument;
+ }
- names = 0;
- namesLen = args->structureInputSize - sizeof(IOKitDiagnosticsParameters);
- if (namesLen) names = (typeof(names))(params + 1);
+ names = NULL;
+ namesLen = args->structureInputSize - sizeof(IOKitDiagnosticsParameters);
+ if (namesLen) {
+ names = (typeof(names))(params + 1);
+ }
- ret = IOTrackingDebug(selector, params->options, params->value, names, namesLen, params->size, &result);
+ ret = IOTrackingDebug(selector, params->options, params->value, params->tag, params->zsize, names, namesLen, params->size, &result);
- if ((kIOReturnSuccess == ret) && args->structureVariableOutputData) *args->structureVariableOutputData = result;
- else if (result) result->release();
+ if ((kIOReturnSuccess == ret) && args->structureVariableOutputData) {
+ *args->structureVariableOutputData = result;
+ } else if (result) {
+ result->release();
+ }
- return (ret);
+ return ret;
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
projects / apple / xnu.git / blobdiff