Loading...
--- xnu/xnu-12377.121.6/iokit/Kernel/IOLib.cpp
+++ xnu/xnu-4570.71.2/iokit/Kernel/IOLib.cpp
@@ -1,8 +1,8 @@
-/*
+/*
* Copyright (c) 1998-2006 Apple Computer, 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
@@ -11,10 +11,10 @@
* 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,
@@ -22,7 +22,7 @@
* 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@
*/
/*
@@ -36,18 +36,17 @@
#include <IOKit/system.h>
#include <mach/sync_policy.h>
#include <machine/machine_routines.h>
-#include <vm/vm_kern_xnu.h>
-#include <vm/vm_map_xnu.h>
+#include <vm/vm_kern.h>
#include <libkern/c++/OSCPPDebug.h>
#include <IOKit/assert.h>
#include <IOKit/IOReturn.h>
-#include <IOKit/IOLib.h>
-#include <IOKit/IOLocks.h>
+#include <IOKit/IOLib.h>
+#include <IOKit/IOLocks.h>
#include <IOKit/IOMapper.h>
#include <IOKit/IOBufferMemoryDescriptor.h>
-#include <IOKit/IOKitDebug.h>
+#include <IOKit/IOKitDebug.h>
#include "IOKitKernelInternal.h"
@@ -77,32 +76,34 @@
#endif /* IOKITSTATS */
-#define TRACK_ALLOC (IOTRACKING && (kIOTracking & gIOKitDebug))
+#define TRACK_ALLOC (IOTRACKING && (kIOTracking & gIOKitDebug))
extern "C"
{
+
+
mach_timespec_t IOZeroTvalspec = { 0, 0 };
extern ppnum_t pmap_find_phys(pmap_t pmap, addr64_t va);
extern int
__doprnt(
- const char *fmt,
- va_list argp,
- void (*putc)(int, void *),
+ const char *fmt,
+ va_list argp,
+ void (*putc)(int, void *),
void *arg,
- int radix,
- int is_log);
-
-extern bool bsd_log_lock(bool);
+ int radix,
+ int is_log);
+
+extern void cons_putc_locked(char);
+extern void bsd_log_lock(void);
extern void bsd_log_unlock(void);
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-lck_grp_t io_lck_grp;
-lck_grp_t *IOLockGroup;
+lck_grp_t *IOLockGroup;
/*
* Global variables for use by iLogger
@@ -110,14 +111,12 @@
* Binary compatibility is not guaranteed for kexts that reference these symbols.
*/
-void *_giDebugLogInternal = NULL;
-void *_giDebugLogDataInternal = NULL;
-void *_giDebugReserved1 = NULL;
-void *_giDebugReserved2 = NULL;
-
-#if defined(__x86_64__)
+void *_giDebugLogInternal = NULL;
+void *_giDebugLogDataInternal = NULL;
+void *_giDebugReserved1 = NULL;
+void *_giDebugReserved2 = NULL;
+
iopa_t gIOBMDPageAllocator;
-#endif /* defined(__x86_64__) */
/*
* Static variables for this module.
@@ -127,31 +126,31 @@
static lck_mtx_t * gIOMallocContiguousEntriesLock;
#if __x86_64__
-enum { kIOPageableMaxAllocSize = 512ULL * 1024 * 1024 };
-enum { kIOPageableMapSize = 8ULL * kIOPageableMaxAllocSize };
+enum { kIOMaxPageableMaps = 8 };
+enum { kIOPageableMapSize = 512 * 1024 * 1024 };
+enum { kIOPageableMaxMapSize = 512 * 1024 * 1024 };
#else
-enum { kIOPageableMaxAllocSize = 96ULL * 1024 * 1024 };
-enum { kIOPageableMapSize = 16ULL * kIOPageableMaxAllocSize };
+enum { kIOMaxPageableMaps = 16 };
+enum { kIOPageableMapSize = 96 * 1024 * 1024 };
+enum { kIOPageableMaxMapSize = 96 * 1024 * 1024 };
#endif
typedef struct {
- vm_map_t map;
- vm_offset_t address;
- vm_offset_t end;
+ vm_map_t map;
+ vm_offset_t address;
+ vm_offset_t end;
} IOMapData;
-#ifndef __BUILDING_XNU_LIBRARY__
-/* this makes clang emit a C and C++ symbol which confuses lldb rdar://135688747 */
-static
-#endif /* __BUILDING_XNU_LIBRARY__ */
-SECURITY_READ_ONLY_LATE(struct mach_vm_range) gIOKitPageableFixedRange;
-IOMapData gIOKitPageableMap;
-
-#if defined(__x86_64__)
+static struct {
+ UInt32 count;
+ UInt32 hint;
+ IOMapData maps[ kIOMaxPageableMaps ];
+ lck_mtx_t * lock;
+} gIOKitPageableSpace;
+
static iopa_t gIOPageablePageAllocator;
uint32_t gIOPageAllocChunkBytes;
-#endif /* defined(__x86_64__) */
#if IOTRACKING
IOTrackingQueue * gIOMallocTracking;
@@ -161,1294 +160,953 @@
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-KMEM_RANGE_REGISTER_STATIC(gIOKitPageableFixed,
- &gIOKitPageableFixedRange, kIOPageableMapSize);
+void IOLibInit(void)
+{
+ kern_return_t ret;
+
+ static bool libInitialized;
+
+ if(libInitialized)
+ return;
+
+ IOLockGroup = lck_grp_alloc_init("IOKit", LCK_GRP_ATTR_NULL);
+
+#if IOTRACKING
+ IOTrackingInit();
+ gIOMallocTracking = IOTrackingQueueAlloc(kIOMallocTrackingName, 0, 0, 0,
+ kIOTrackingQueueTypeAlloc,
+ 37);
+ gIOWireTracking = IOTrackingQueueAlloc(kIOWireTrackingName, 0, 0, page_size, 0, 0);
+
+ size_t mapCaptureSize = (kIOTracking & gIOKitDebug) ? page_size : (1024*1024);
+ gIOMapTracking = IOTrackingQueueAlloc(kIOMapTrackingName, 0, 0, mapCaptureSize,
+ kIOTrackingQueueTypeDefaultOn
+ | kIOTrackingQueueTypeMap
+ | kIOTrackingQueueTypeUser,
+ 0);
+#endif
+
+ gIOKitPageableSpace.maps[0].address = 0;
+ ret = kmem_suballoc(kernel_map,
+ &gIOKitPageableSpace.maps[0].address,
+ kIOPageableMapSize,
+ TRUE,
+ VM_FLAGS_ANYWHERE,
+ VM_MAP_KERNEL_FLAGS_NONE,
+ VM_KERN_MEMORY_IOKIT,
+ &gIOKitPageableSpace.maps[0].map);
+ if (ret != KERN_SUCCESS)
+ panic("failed to allocate iokit pageable map\n");
+
+ gIOKitPageableSpace.lock = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
+ gIOKitPageableSpace.maps[0].end = gIOKitPageableSpace.maps[0].address + kIOPageableMapSize;
+ gIOKitPageableSpace.hint = 0;
+ gIOKitPageableSpace.count = 1;
+
+ gIOMallocContiguousEntriesLock = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
+ queue_init( &gIOMallocContiguousEntries );
+
+ gIOPageAllocChunkBytes = PAGE_SIZE/64;
+ assert(sizeof(iopa_page_t) <= gIOPageAllocChunkBytes);
+ iopa_init(&gIOBMDPageAllocator);
+ iopa_init(&gIOPageablePageAllocator);
+
+
+ libInitialized = true;
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+static uint32_t
+log2up(uint32_t size)
+{
+ if (size <= 1) size = 0;
+ else size = 32 - __builtin_clz(size - 1);
+ return (size);
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+IOThread IOCreateThread(IOThreadFunc fcn, void *arg)
+{
+ kern_return_t result;
+ thread_t thread;
+
+ result = kernel_thread_start((thread_continue_t)fcn, arg, &thread);
+ if (result != KERN_SUCCESS)
+ return (NULL);
+
+ thread_deallocate(thread);
+
+ return (thread);
+}
+
+
+void IOExitThread(void)
+{
+ (void) thread_terminate(current_thread());
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#if IOTRACKING
+struct IOLibMallocHeader
+{
+ IOTrackingAddress tracking;
+};
+#endif
+
+#if IOTRACKING
+#define sizeofIOLibMallocHeader (sizeof(IOLibMallocHeader) - (TRACK_ALLOC ? 0 : sizeof(IOTrackingAddress)))
+#else
+#define sizeofIOLibMallocHeader (0)
+#endif
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+void * IOMalloc(vm_size_t size)
+{
+ void * address;
+ vm_size_t allocSize;
+
+ allocSize = size + sizeofIOLibMallocHeader;
+#if IOTRACKING
+ if (sizeofIOLibMallocHeader && (allocSize <= size)) return (NULL); // overflow
+#endif
+ address = kalloc_tag_bt(allocSize, VM_KERN_MEMORY_IOKIT);
+
+ if ( address ) {
+#if IOTRACKING
+ if (TRACK_ALLOC) {
+ IOLibMallocHeader * hdr;
+ hdr = (typeof(hdr)) address;
+ bzero(&hdr->tracking, sizeof(hdr->tracking));
+ hdr->tracking.address = ~(((uintptr_t) address) + sizeofIOLibMallocHeader);
+ hdr->tracking.size = size;
+ IOTrackingAdd(gIOMallocTracking, &hdr->tracking.tracking, size, true, VM_KERN_MEMORY_NONE);
+ }
+#endif
+ address = (typeof(address)) (((uintptr_t) address) + sizeofIOLibMallocHeader);
+
+#if IOALLOCDEBUG
+ OSAddAtomic(size, &debug_iomalloc_size);
+#endif
+ IOStatisticsAlloc(kIOStatisticsMalloc, size);
+ }
+
+ return address;
+}
+
+void IOFree(void * inAddress, vm_size_t size)
+{
+ void * address;
+
+ if ((address = inAddress))
+ {
+ address = (typeof(address)) (((uintptr_t) address) - sizeofIOLibMallocHeader);
+
+#if IOTRACKING
+ if (TRACK_ALLOC)
+ {
+ IOLibMallocHeader * hdr;
+ struct ptr_reference{ void * ptr; };
+ volatile struct ptr_reference ptr;
+
+ // we're about to block in IOTrackingRemove(), make sure the original pointer
+ // exists in memory or a register for leak scanning to find
+ ptr.ptr = inAddress;
+
+ hdr = (typeof(hdr)) address;
+ if (size != hdr->tracking.size)
+ {
+ OSReportWithBacktrace("bad IOFree size 0x%lx should be 0x%lx", size, hdr->tracking.size);
+ size = hdr->tracking.size;
+ }
+ IOTrackingRemove(gIOMallocTracking, &hdr->tracking.tracking, size);
+ ptr.ptr = NULL;
+ }
+#endif
+
+ kfree(address, size + sizeofIOLibMallocHeader);
+#if IOALLOCDEBUG
+ OSAddAtomic(-size, &debug_iomalloc_size);
+#endif
+ IOStatisticsAlloc(kIOStatisticsFree, size);
+ }
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+vm_tag_t
+IOMemoryTag(vm_map_t map)
+{
+ vm_tag_t tag;
+
+ if (!vm_kernel_map_is_kernel(map)) return (VM_MEMORY_IOKIT);
+
+ tag = vm_tag_bt();
+ if (tag == VM_KERN_MEMORY_NONE) tag = VM_KERN_MEMORY_IOKIT;
+
+ return (tag);
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+struct IOLibPageMallocHeader
+{
+ mach_vm_size_t allocationSize;
+ mach_vm_address_t allocationAddress;
+#if IOTRACKING
+ IOTrackingAddress tracking;
+#endif
+};
+
+#if IOTRACKING
+#define sizeofIOLibPageMallocHeader (sizeof(IOLibPageMallocHeader) - (TRACK_ALLOC ? 0 : sizeof(IOTrackingAddress)))
+#else
+#define sizeofIOLibPageMallocHeader (sizeof(IOLibPageMallocHeader))
+#endif
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+void * IOMallocAligned(vm_size_t size, vm_size_t alignment)
+{
+ kern_return_t kr;
+ vm_offset_t address;
+ vm_offset_t allocationAddress;
+ vm_size_t adjustedSize;
+ uintptr_t alignMask;
+ IOLibPageMallocHeader * hdr;
+
+ if (size == 0)
+ return 0;
+
+ alignment = (1UL << log2up(alignment));
+ alignMask = alignment - 1;
+ adjustedSize = size + sizeofIOLibPageMallocHeader;
+
+ if (size > adjustedSize) {
+ address = 0; /* overflow detected */
+ }
+ else if (adjustedSize >= page_size) {
+
+ kr = kernel_memory_allocate(kernel_map, &address,
+ size, alignMask, 0, IOMemoryTag(kernel_map));
+ if (KERN_SUCCESS != kr) address = 0;
+#if IOTRACKING
+ else if (TRACK_ALLOC) IOTrackingAlloc(gIOMallocTracking, address, size);
+#endif
+
+ } else {
+
+ adjustedSize += alignMask;
+
+ if (adjustedSize >= page_size) {
+
+ kr = kernel_memory_allocate(kernel_map, &allocationAddress,
+ adjustedSize, 0, 0, IOMemoryTag(kernel_map));
+ if (KERN_SUCCESS != kr) allocationAddress = 0;
+
+ } else
+ allocationAddress = (vm_address_t) kalloc_tag_bt(adjustedSize, VM_KERN_MEMORY_IOKIT);
+
+ if (allocationAddress) {
+ address = (allocationAddress + alignMask + sizeofIOLibPageMallocHeader)
+ & (~alignMask);
+
+ hdr = (typeof(hdr))(address - sizeofIOLibPageMallocHeader);
+ hdr->allocationSize = adjustedSize;
+ hdr->allocationAddress = allocationAddress;
+#if IOTRACKING
+ if (TRACK_ALLOC) {
+ bzero(&hdr->tracking, sizeof(hdr->tracking));
+ hdr->tracking.address = ~address;
+ hdr->tracking.size = size;
+ IOTrackingAdd(gIOMallocTracking, &hdr->tracking.tracking, size, true, VM_KERN_MEMORY_NONE);
+ }
+#endif
+ } else
+ address = 0;
+ }
+
+ assert(0 == (address & alignMask));
+
+ if( address) {
+#if IOALLOCDEBUG
+ OSAddAtomic(size, &debug_iomalloc_size);
+#endif
+ IOStatisticsAlloc(kIOStatisticsMallocAligned, size);
+ }
+
+ return (void *) address;
+}
+
+void IOFreeAligned(void * address, vm_size_t size)
+{
+ vm_address_t allocationAddress;
+ vm_size_t adjustedSize;
+ IOLibPageMallocHeader * hdr;
+
+ if( !address)
+ return;
+
+ assert(size);
+
+ adjustedSize = size + sizeofIOLibPageMallocHeader;
+ if (adjustedSize >= page_size) {
+#if IOTRACKING
+ if (TRACK_ALLOC) IOTrackingFree(gIOMallocTracking, (uintptr_t) address, size);
+#endif
+ kmem_free( kernel_map, (vm_offset_t) address, size);
+
+ } else {
+ hdr = (typeof(hdr)) (((uintptr_t)address) - sizeofIOLibPageMallocHeader);
+ adjustedSize = hdr->allocationSize;
+ allocationAddress = hdr->allocationAddress;
+
+#if IOTRACKING
+ if (TRACK_ALLOC)
+ {
+ if (size != hdr->tracking.size)
+ {
+ OSReportWithBacktrace("bad IOFreeAligned size 0x%lx should be 0x%lx", size, hdr->tracking.size);
+ size = hdr->tracking.size;
+ }
+ IOTrackingRemove(gIOMallocTracking, &hdr->tracking.tracking, size);
+ }
+#endif
+ if (adjustedSize >= page_size) {
+ kmem_free( kernel_map, allocationAddress, adjustedSize);
+ } else {
+ kfree((void *)allocationAddress, adjustedSize);
+ }
+ }
+
+#if IOALLOCDEBUG
+ OSAddAtomic(-size, &debug_iomalloc_size);
+#endif
+
+ IOStatisticsAlloc(kIOStatisticsFreeAligned, size);
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
void
-IOLibInit(void)
-{
- static bool libInitialized;
-
- if (libInitialized) {
- return;
- }
-
- lck_grp_init(&io_lck_grp, "IOKit", LCK_GRP_ATTR_NULL);
- IOLockGroup = &io_lck_grp;
-
-#if IOTRACKING
- IOTrackingInit();
- gIOMallocTracking = IOTrackingQueueAlloc(kIOMallocTrackingName, 0, 0, 0,
- kIOTrackingQueueTypeAlloc,
- 37);
- gIOWireTracking = IOTrackingQueueAlloc(kIOWireTrackingName, 0, 0, page_size, 0, 0);
-
- size_t mapCaptureSize = (kIOTracking & gIOKitDebug) ? page_size : (1024 * 1024);
- gIOMapTracking = IOTrackingQueueAlloc(kIOMapTrackingName, 0, 0, mapCaptureSize,
- kIOTrackingQueueTypeDefaultOn
- | kIOTrackingQueueTypeMap
- | kIOTrackingQueueTypeUser,
- 0);
-#endif
-
- gIOKitPageableMap.map = kmem_suballoc(kernel_map,
- &gIOKitPageableFixedRange.min_address,
- kIOPageableMapSize,
- VM_MAP_CREATE_DEFAULT,
- VM_FLAGS_FIXED | VM_FLAGS_OVERWRITE,
- (kms_flags_t)(KMS_DATA | KMS_NOFAIL | KMS_NOSOFTLIMIT),
- VM_KERN_MEMORY_IOKIT).kmr_submap;
-
- gIOKitPageableMap.address = gIOKitPageableFixedRange.min_address;
- gIOKitPageableMap.end = gIOKitPageableFixedRange.max_address;
-
- gIOMallocContiguousEntriesLock = lck_mtx_alloc_init(IOLockGroup, LCK_ATTR_NULL);
- queue_init( &gIOMallocContiguousEntries );
-
-#if defined(__x86_64__)
- gIOPageAllocChunkBytes = PAGE_SIZE / 64;
-
- assert(sizeof(iopa_page_t) <= gIOPageAllocChunkBytes);
- iopa_init(&gIOBMDPageAllocator);
- iopa_init(&gIOPageablePageAllocator);
-#endif /* defined(__x86_64__) */
-
-
- libInitialized = true;
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-vm_size_t
-log2up(vm_size_t size)
-{
- if (size <= 1) {
- size = 0;
- } else {
-#if __LP64__
- size = 64 - __builtin_clzl(size - 1);
-#else
- size = 32 - __builtin_clzl(size - 1);
-#endif
- }
- return size;
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-IOThread
-IOCreateThread(IOThreadFunc fcn, void *arg)
-{
- kern_return_t result;
- thread_t thread;
-
- result = kernel_thread_start((thread_continue_t)(void (*)(void))fcn, arg, &thread);
- if (result != KERN_SUCCESS) {
- return NULL;
- }
-
- thread_deallocate(thread);
-
- return thread;
-}
-
-
-void
-IOExitThread(void)
-{
- (void) thread_terminate(current_thread());
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-#if IOTRACKING
-struct IOLibMallocHeader {
- IOTrackingAddress tracking;
-};
-#endif
-
-#if IOTRACKING
-#define sizeofIOLibMallocHeader (sizeof(IOLibMallocHeader) - (TRACK_ALLOC ? 0 : sizeof(IOTrackingAddress)))
-#else
-#define sizeofIOLibMallocHeader (0)
-#endif
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-__typed_allocators_ignore_push // allocator implementation
-
-void *
-(IOMalloc_internal)(struct kalloc_heap *kheap, vm_size_t size,
-zalloc_flags_t flags)
-{
- void * address;
- vm_size_t allocSize;
-
- allocSize = size + sizeofIOLibMallocHeader;
-#if IOTRACKING
- if (sizeofIOLibMallocHeader && (allocSize <= size)) {
- return NULL; // overflow
- }
-#endif
- address = kheap_alloc(kheap, allocSize,
- Z_VM_TAG(Z_WAITOK | flags, VM_KERN_MEMORY_IOKIT));
-
- if (address) {
-#if IOTRACKING
- if (TRACK_ALLOC) {
- IOLibMallocHeader * hdr;
- hdr = (typeof(hdr))address;
- bzero(&hdr->tracking, sizeof(hdr->tracking));
- hdr->tracking.address = ~(((uintptr_t) address) + sizeofIOLibMallocHeader);
- hdr->tracking.size = size;
- IOTrackingAdd(gIOMallocTracking, &hdr->tracking.tracking, size, true, VM_KERN_MEMORY_NONE);
- }
-#endif
- address = (typeof(address))(((uintptr_t) address) + sizeofIOLibMallocHeader);
-
+IOKernelFreePhysical(mach_vm_address_t address, mach_vm_size_t size)
+{
+ mach_vm_address_t allocationAddress;
+ mach_vm_size_t adjustedSize;
+ IOLibPageMallocHeader * hdr;
+
+ if (!address)
+ return;
+
+ assert(size);
+
+ adjustedSize = (2 * size) + sizeofIOLibPageMallocHeader;
+ if (adjustedSize >= page_size) {
+#if IOTRACKING
+ if (TRACK_ALLOC) IOTrackingFree(gIOMallocTracking, address, size);
+#endif
+ kmem_free( kernel_map, (vm_offset_t) address, size);
+
+ } else {
+
+ hdr = (typeof(hdr)) (((uintptr_t)address) - sizeofIOLibPageMallocHeader);
+ adjustedSize = hdr->allocationSize;
+ allocationAddress = hdr->allocationAddress;
+#if IOTRACKING
+ if (TRACK_ALLOC) IOTrackingRemove(gIOMallocTracking, &hdr->tracking.tracking, size);
+#endif
+ kfree((void *)allocationAddress, adjustedSize);
+ }
+
+ IOStatisticsAlloc(kIOStatisticsFreeContiguous, size);
#if IOALLOCDEBUG
- OSAddAtomicLong(size, &debug_iomalloc_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsMalloc, size);
- }
-
- return address;
-}
-
-void
-IOFree_internal(struct kalloc_heap *kheap, void * inAddress, vm_size_t size)
-{
- void * address;
-
- if ((address = inAddress)) {
- address = (typeof(address))(((uintptr_t) address) - sizeofIOLibMallocHeader);
-
-#if IOTRACKING
- if (TRACK_ALLOC) {
- IOLibMallocHeader * hdr;
- struct ptr_reference { void * ptr; };
- volatile struct ptr_reference ptr;
-
- // we're about to block in IOTrackingRemove(), make sure the original pointer
- // exists in memory or a register for leak scanning to find
- ptr.ptr = inAddress;
-
- hdr = (typeof(hdr))address;
- if (size != hdr->tracking.size) {
- OSReportWithBacktrace("bad IOFree size 0x%zx should be 0x%zx",
- (size_t)size, (size_t)hdr->tracking.size);
- size = hdr->tracking.size;
- }
- IOTrackingRemoveAddress(gIOMallocTracking, &hdr->tracking, size);
- ptr.ptr = NULL;
- }
-#endif
-
- kheap_free(kheap, address, size + sizeofIOLibMallocHeader);
+ OSAddAtomic(-size, &debug_iomalloc_size);
+#endif
+}
+
+#if __arm__ || __arm64__
+extern unsigned long gPhysBase, gPhysSize;
+#endif
+
+mach_vm_address_t
+IOKernelAllocateWithPhysicalRestrict(mach_vm_size_t size, mach_vm_address_t maxPhys,
+ mach_vm_size_t alignment, bool contiguous)
+{
+ kern_return_t kr;
+ mach_vm_address_t address;
+ mach_vm_address_t allocationAddress;
+ mach_vm_size_t adjustedSize;
+ mach_vm_address_t alignMask;
+ IOLibPageMallocHeader * hdr;
+
+ if (size == 0)
+ return (0);
+ if (alignment == 0)
+ alignment = 1;
+
+ alignMask = alignment - 1;
+ adjustedSize = (2 * size) + sizeofIOLibPageMallocHeader;
+ if (adjustedSize < size) return (0);
+
+ contiguous = (contiguous && (adjustedSize > page_size))
+ || (alignment > page_size);
+
+ if (contiguous || maxPhys)
+ {
+ int options = 0;
+ vm_offset_t virt;
+
+ adjustedSize = size;
+ contiguous = (contiguous && (adjustedSize > page_size))
+ || (alignment > page_size);
+
+ if (!contiguous)
+ {
+#if __arm__ || __arm64__
+ if (maxPhys >= (mach_vm_address_t)(gPhysBase + gPhysSize))
+ {
+ maxPhys = 0;
+ }
+ else
+#endif
+ if (maxPhys <= 0xFFFFFFFF)
+ {
+ maxPhys = 0;
+ options |= KMA_LOMEM;
+ }
+ else if (gIOLastPage && (atop_64(maxPhys) > gIOLastPage))
+ {
+ maxPhys = 0;
+ }
+ }
+ if (contiguous || maxPhys)
+ {
+ kr = kmem_alloc_contig(kernel_map, &virt, size,
+ alignMask, atop(maxPhys), atop(alignMask), 0, IOMemoryTag(kernel_map));
+ }
+ else
+ {
+ kr = kernel_memory_allocate(kernel_map, &virt,
+ size, alignMask, options, IOMemoryTag(kernel_map));
+ }
+ if (KERN_SUCCESS == kr)
+ {
+ address = virt;
+#if IOTRACKING
+ if (TRACK_ALLOC) IOTrackingAlloc(gIOMallocTracking, address, size);
+#endif
+ }
+ else
+ address = 0;
+ }
+ else
+ {
+ adjustedSize += alignMask;
+ if (adjustedSize < size) return (0);
+ allocationAddress = (mach_vm_address_t) kalloc_tag_bt(adjustedSize, VM_KERN_MEMORY_IOKIT);
+
+ if (allocationAddress) {
+
+
+ address = (allocationAddress + alignMask + sizeofIOLibPageMallocHeader)
+ & (~alignMask);
+
+ if (atop_32(address) != atop_32(address + size - 1))
+ address = round_page(address);
+
+ hdr = (typeof(hdr))(address - sizeofIOLibPageMallocHeader);
+ hdr->allocationSize = adjustedSize;
+ hdr->allocationAddress = allocationAddress;
+#if IOTRACKING
+ if (TRACK_ALLOC) {
+ bzero(&hdr->tracking, sizeof(hdr->tracking));
+ hdr->tracking.address = ~address;
+ hdr->tracking.size = size;
+ IOTrackingAdd(gIOMallocTracking, &hdr->tracking.tracking, size, true, VM_KERN_MEMORY_NONE);
+ }
+#endif
+ } else
+ address = 0;
+ }
+
+ if (address) {
+ IOStatisticsAlloc(kIOStatisticsMallocContiguous, size);
#if IOALLOCDEBUG
- OSAddAtomicLong(-size, &debug_iomalloc_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsFree, size);
- }
-}
-
-void *
-IOMalloc_external(
- vm_size_t size);
-void *
-IOMalloc_external(
- vm_size_t size)
-{
- return IOMalloc_internal(KHEAP_DEFAULT, size, Z_VM_TAG_BT_BIT);
-}
-
-void
-IOFree(void * inAddress, vm_size_t size)
-{
- IOFree_internal(KHEAP_DEFAULT, inAddress, size);
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-void *
-IOMallocZero_external(
- vm_size_t size);
-void *
-IOMallocZero_external(
- vm_size_t size)
-{
- return IOMalloc_internal(KHEAP_DEFAULT, size, Z_ZERO_VM_TAG_BT_BIT);
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-vm_tag_t
-IOMemoryTag(vm_map_t map)
-{
- vm_tag_t tag;
-
- if (!vm_kernel_map_is_kernel(map)) {
- return VM_MEMORY_IOKIT;
- }
-
- tag = vm_tag_bt();
- if (tag == VM_KERN_MEMORY_NONE) {
- tag = VM_KERN_MEMORY_IOKIT;
- }
-
- return tag;
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-struct IOLibPageMallocHeader {
- mach_vm_size_t alignMask;
- mach_vm_offset_t allocationOffset;
-#if IOTRACKING
- IOTrackingAddress tracking;
-#endif
-};
-
-#if IOTRACKING
-#define sizeofIOLibPageMallocHeader (sizeof(IOLibPageMallocHeader) - (TRACK_ALLOC ? 0 : sizeof(IOTrackingAddress)))
-#else
-#define sizeofIOLibPageMallocHeader (sizeof(IOLibPageMallocHeader))
-#endif
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-static __header_always_inline void
-IOMallocAlignedSetHdr(
- IOLibPageMallocHeader *hdr,
- mach_vm_size_t alignMask,
- mach_vm_address_t allocationStart,
- mach_vm_address_t alignedStart)
-{
- mach_vm_offset_t offset = alignedStart - allocationStart;
-#if __has_feature(ptrauth_calls)
- offset = (mach_vm_offset_t) ptrauth_sign_unauthenticated((void *)offset,
- ptrauth_key_process_independent_data,
- ptrauth_blend_discriminator((void *)(alignedStart | alignMask),
- OS_PTRAUTH_DISCRIMINATOR("IOLibPageMallocHeader.allocationOffset")));
-#endif /* __has_feature(ptrauth_calls) */
- hdr->allocationOffset = offset;
- hdr->alignMask = alignMask;
-}
-
-__abortlike
-static void
-IOMallocAlignedHdrCorruptionPanic(
- mach_vm_offset_t offset,
- mach_vm_size_t alignMask,
- mach_vm_address_t alignedStart,
- vm_size_t size)
-{
- mach_vm_address_t address = 0;
- mach_vm_address_t recalAlignedStart = 0;
-
- if (os_sub_overflow(alignedStart, offset, &address)) {
- panic("Invalid offset %p for aligned addr %p", (void *)offset,
- (void *)alignedStart);
- }
- if (os_add3_overflow(address, sizeofIOLibPageMallocHeader, alignMask,
- &recalAlignedStart)) {
- panic("alignMask 0x%llx overflows recalAlignedStart %p for provided addr "
- "%p", alignMask, (void *)recalAlignedStart, (void *)alignedStart);
- }
- if (((recalAlignedStart &= ~alignMask) != alignedStart) &&
- (round_page(recalAlignedStart) != alignedStart)) {
- panic("Recalculated aligned addr %p doesn't match provided addr %p",
- (void *)recalAlignedStart, (void *)alignedStart);
- }
- if (offset < sizeofIOLibPageMallocHeader) {
- panic("Offset %zd doesn't accomodate IOLibPageMallocHeader for aligned "
- "addr %p", (size_t)offset, (void *)alignedStart);
- }
- panic("alignMask 0x%llx overflows adjusted size %zd for aligned addr %p",
- alignMask, (size_t)size, (void *)alignedStart);
-}
-
-static __header_always_inline mach_vm_address_t
-IOMallocAlignedGetAddress(
- IOLibPageMallocHeader *hdr,
- mach_vm_address_t alignedStart,
- vm_size_t *size)
-{
- mach_vm_address_t address = 0;
- mach_vm_address_t recalAlignedStart = 0;
- mach_vm_offset_t offset = hdr->allocationOffset;
- mach_vm_size_t alignMask = hdr->alignMask;
-#if __has_feature(ptrauth_calls)
- offset = (mach_vm_offset_t) ptrauth_auth_data((void *)offset,
- ptrauth_key_process_independent_data,
- ptrauth_blend_discriminator((void *)(alignedStart | alignMask),
- OS_PTRAUTH_DISCRIMINATOR("IOLibPageMallocHeader.allocationOffset")));
-#endif /* __has_feature(ptrauth_calls) */
- if (os_sub_overflow(alignedStart, offset, &address) ||
- os_add3_overflow(address, sizeofIOLibPageMallocHeader, alignMask,
- &recalAlignedStart) ||
- (((recalAlignedStart &= ~alignMask) != alignedStart) &&
- (round_page(recalAlignedStart) != alignedStart)) ||
- (offset < sizeofIOLibPageMallocHeader) ||
- os_add_overflow(*size, alignMask, size)) {
- IOMallocAlignedHdrCorruptionPanic(offset, alignMask, alignedStart, *size);
- }
- return address;
-}
-
-void *
-(IOMallocAligned_internal)(struct kalloc_heap *kheap, vm_size_t size,
-vm_size_t alignment, zalloc_flags_t flags)
-{
- kern_return_t kr;
- vm_offset_t address;
- vm_offset_t allocationAddress;
- vm_size_t adjustedSize;
- uintptr_t alignMask;
- IOLibPageMallocHeader * hdr;
- kma_flags_t kma_flags = KMA_NONE;
-
- if (size == 0) {
- return NULL;
- }
- if (((uint32_t) alignment) != alignment) {
- return NULL;
- }
-
- if (flags & Z_ZERO) {
- kma_flags = KMA_ZERO;
- }
-
- if (kheap == KHEAP_DATA_PRIVATE) {
- kma_flags = (kma_flags_t) (kma_flags | KMA_DATA);
- } else if (kheap == KHEAP_DATA_SHARED) {
- kma_flags = (kma_flags_t) (kma_flags | KMA_DATA_SHARED);
- }
-
- alignment = (1UL << log2up((uint32_t) alignment));
- alignMask = alignment - 1;
- adjustedSize = size + sizeofIOLibPageMallocHeader;
-
- if (size > adjustedSize) {
- address = 0; /* overflow detected */
- } else if (adjustedSize >= page_size) {
- kr = kernel_memory_allocate(kernel_map, &address,
- size, alignMask, kma_flags, IOMemoryTag(kernel_map));
- if (KERN_SUCCESS != kr) {
- address = 0;
- }
-#if IOTRACKING
- else if (TRACK_ALLOC) {
- IOTrackingAlloc(gIOMallocTracking, address, size);
- }
-#endif
- } else {
- adjustedSize += alignMask;
-
- if (adjustedSize >= page_size) {
- kr = kmem_alloc(kernel_map, &allocationAddress,
- adjustedSize, kma_flags, IOMemoryTag(kernel_map));
- if (KERN_SUCCESS != kr) {
- allocationAddress = 0;
- }
- } else {
- allocationAddress = (vm_address_t) kheap_alloc(kheap,
- adjustedSize, Z_VM_TAG(Z_WAITOK | flags, VM_KERN_MEMORY_IOKIT));
- }
-
- if (allocationAddress) {
- address = (allocationAddress + alignMask + sizeofIOLibPageMallocHeader)
- & (~alignMask);
-
- hdr = (typeof(hdr))(address - sizeofIOLibPageMallocHeader);
- IOMallocAlignedSetHdr(hdr, alignMask, allocationAddress, address);
-#if IOTRACKING
- if (TRACK_ALLOC) {
- bzero(&hdr->tracking, sizeof(hdr->tracking));
- hdr->tracking.address = ~address;
- hdr->tracking.size = size;
- IOTrackingAdd(gIOMallocTracking, &hdr->tracking.tracking, size, true, VM_KERN_MEMORY_NONE);
- }
-#endif
- } else {
- address = 0;
- }
- }
-
- assert(0 == (address & alignMask));
-
- if (address) {
-#if IOALLOCDEBUG
- OSAddAtomicLong(size, &debug_iomalloc_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsMallocAligned, size);
- }
-
- return (void *) address;
-}
-
-void
-IOFreeAligned_internal(kalloc_heap_t kheap, void * address, vm_size_t size)
-{
- vm_address_t allocationAddress;
- vm_size_t adjustedSize;
- IOLibPageMallocHeader * hdr;
-
- if (!address) {
- return;
- }
-
- assert(size);
-
- adjustedSize = size + sizeofIOLibPageMallocHeader;
- if (adjustedSize >= page_size) {
-#if IOTRACKING
- if (TRACK_ALLOC) {
- IOTrackingFree(gIOMallocTracking, (uintptr_t) address, size);
- }
-#endif
- kmem_free(kernel_map, (vm_offset_t) address, size);
- } else {
- hdr = (typeof(hdr))(((uintptr_t)address) - sizeofIOLibPageMallocHeader);
- allocationAddress = IOMallocAlignedGetAddress(hdr,
- (mach_vm_address_t)address, &adjustedSize);
-
-#if IOTRACKING
- if (TRACK_ALLOC) {
- if (size != hdr->tracking.size) {
- OSReportWithBacktrace("bad IOFreeAligned size 0x%zx should be 0x%zx",
- (size_t)size, (size_t)hdr->tracking.size);
- size = hdr->tracking.size;
- }
- IOTrackingRemoveAddress(gIOMallocTracking, &hdr->tracking, size);
- }
-#endif
- if (adjustedSize >= page_size) {
- kmem_free(kernel_map, allocationAddress, adjustedSize);
- } else {
- kheap_free(kheap, allocationAddress, adjustedSize);
- }
- }
-
-#if IOALLOCDEBUG
- OSAddAtomicLong(-size, &debug_iomalloc_size);
-#endif
-
- IOStatisticsAlloc(kIOStatisticsFreeAligned, size);
-}
-
-void *
-IOMallocAligned_external(
- vm_size_t size, vm_size_t alignment);
-void *
-IOMallocAligned_external(
- vm_size_t size, vm_size_t alignment)
-{
- return IOMallocAligned_internal(KHEAP_DATA_SHARED, size, alignment,
- Z_VM_TAG_BT_BIT);
-}
-
-void
-IOFreeAligned(
- void * address,
- vm_size_t size)
-{
- IOFreeAligned_internal(KHEAP_DATA_SHARED, address, size);
-}
-
-__typed_allocators_ignore_pop
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-void
-IOKernelFreePhysical(
- kalloc_heap_t kheap,
- mach_vm_address_t address,
- mach_vm_size_t size)
-{
- vm_address_t allocationAddress;
- vm_size_t adjustedSize;
- IOLibPageMallocHeader * hdr;
-
- if (!address) {
- return;
- }
-
- assert(size);
-
- adjustedSize = (2 * size) + sizeofIOLibPageMallocHeader;
- if (adjustedSize >= page_size) {
-#if IOTRACKING
- if (TRACK_ALLOC) {
- IOTrackingFree(gIOMallocTracking, address, size);
- }
-#endif
- kmem_free(kernel_map, (vm_offset_t) address, size);
- } else {
- hdr = (typeof(hdr))(((uintptr_t)address) - sizeofIOLibPageMallocHeader);
- allocationAddress = IOMallocAlignedGetAddress(hdr, address, &adjustedSize);
-#if IOTRACKING
- if (TRACK_ALLOC) {
- IOTrackingRemoveAddress(gIOMallocTracking, &hdr->tracking, size);
- }
-#endif
- __typed_allocators_ignore(kheap_free(kheap, allocationAddress, adjustedSize));
- }
-
- IOStatisticsAlloc(kIOStatisticsFreeContiguous, size);
-#if IOALLOCDEBUG
- OSAddAtomicLong(-size, &debug_iomalloc_size);
-#endif
-}
-
-#if __arm64__
-extern unsigned long gPhysBase, gPhysSize;
-#endif
-
-mach_vm_address_t
-IOKernelAllocateWithPhysicalRestrict(
- kalloc_heap_t kheap,
- mach_vm_size_t size,
- mach_vm_address_t maxPhys,
- mach_vm_size_t alignment,
- bool contiguous,
- bool noSoftLimit)
-{
- kern_return_t kr;
- mach_vm_address_t address;
- mach_vm_address_t allocationAddress;
- mach_vm_size_t adjustedSize;
- mach_vm_address_t alignMask;
- IOLibPageMallocHeader * hdr;
-
- if (size == 0) {
- return 0;
- }
- if (alignment == 0) {
- alignment = 1;
- }
-
- alignMask = alignment - 1;
-
- if (os_mul_and_add_overflow(2, size, sizeofIOLibPageMallocHeader, &adjustedSize)) {
- return 0;
- }
-
- contiguous = (contiguous && (adjustedSize > page_size))
- || (alignment > page_size);
-
- if (contiguous || maxPhys) {
- kma_flags_t options = KMA_ZERO;
- vm_offset_t virt;
-
- if (kheap == KHEAP_DATA_PRIVATE) {
- options = (kma_flags_t) (options | KMA_DATA);
- } else if (kheap == KHEAP_DATA_SHARED) {
- options = (kma_flags_t) (options | KMA_DATA_SHARED);
- }
-
- if (noSoftLimit) {
- options = (kma_flags_t) (options | KMA_NOSOFTLIMIT);
- }
-
- adjustedSize = size;
- contiguous = (contiguous && (adjustedSize > page_size))
- || (alignment > page_size);
-
- if (!contiguous) {
-#if __arm64__
- if (maxPhys >= (mach_vm_address_t)(gPhysBase + gPhysSize)) {
- maxPhys = 0;
- } else
-#endif
- if (maxPhys <= 0xFFFFFFFF) {
- maxPhys = 0;
- options = (kma_flags_t)(options | KMA_LOMEM);
- } else if (gIOLastPage && (atop_64(maxPhys) > gIOLastPage)) {
- maxPhys = 0;
- }
- }
- if (contiguous || maxPhys) {
- kr = kmem_alloc_contig(kernel_map, &virt, size,
- alignMask, (ppnum_t) atop(maxPhys), (ppnum_t) atop(alignMask),
- options, IOMemoryTag(kernel_map));
- } else {
- kr = kernel_memory_allocate(kernel_map, &virt,
- size, alignMask, options, IOMemoryTag(kernel_map));
- }
- if (KERN_SUCCESS == kr) {
- address = virt;
-#if IOTRACKING
- if (TRACK_ALLOC) {
- IOTrackingAlloc(gIOMallocTracking, address, size);
- }
-#endif
- } else {
- address = 0;
- }
- } else {
- zalloc_flags_t zflags = Z_WAITOK;
-
- if (noSoftLimit) {
- zflags = (zalloc_flags_t)(zflags | Z_NOSOFTLIMIT);
- }
-
- adjustedSize += alignMask;
- if (adjustedSize < size) {
- return 0;
- }
-
- /* BEGIN IGNORE CODESTYLE */
- __typed_allocators_ignore_push // allocator implementation
- allocationAddress = (mach_vm_address_t) kheap_alloc(kheap,
- adjustedSize, Z_VM_TAG_BT(zflags, VM_KERN_MEMORY_IOKIT));
- __typed_allocators_ignore_pop
- /* END IGNORE CODESTYLE */
-
- if (allocationAddress) {
- address = (allocationAddress + alignMask + sizeofIOLibPageMallocHeader)
- & (~alignMask);
-
- if (atop_32(address) != atop_32(address + size - 1)) {
- address = round_page(address);
- }
-
- hdr = (typeof(hdr))(address - sizeofIOLibPageMallocHeader);
- IOMallocAlignedSetHdr(hdr, alignMask, allocationAddress, address);
-#if IOTRACKING
- if (TRACK_ALLOC) {
- bzero(&hdr->tracking, sizeof(hdr->tracking));
- hdr->tracking.address = ~address;
- hdr->tracking.size = size;
- IOTrackingAdd(gIOMallocTracking, &hdr->tracking.tracking, size, true, VM_KERN_MEMORY_NONE);
- }
-#endif
- } else {
- address = 0;
- }
- }
-
- if (address) {
- IOStatisticsAlloc(kIOStatisticsMallocContiguous, size);
-#if IOALLOCDEBUG
- OSAddAtomicLong(size, &debug_iomalloc_size);
-#endif
- }
-
- return address;
-}
-
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-struct _IOMallocContiguousEntry {
- mach_vm_address_t virtualAddr;
- IOBufferMemoryDescriptor * md;
- queue_chain_t link;
+ OSAddAtomic(size, &debug_iomalloc_size);
+#endif
+ }
+
+ return (address);
+}
+
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+struct _IOMallocContiguousEntry
+{
+ mach_vm_address_t virtualAddr;
+ IOBufferMemoryDescriptor * md;
+ queue_chain_t link;
};
typedef struct _IOMallocContiguousEntry _IOMallocContiguousEntry;
-void *
-IOMallocContiguous(vm_size_t size, vm_size_t alignment,
- IOPhysicalAddress * physicalAddress)
-{
- mach_vm_address_t address = 0;
-
- if (size == 0) {
- return NULL;
- }
- if (alignment == 0) {
- alignment = 1;
- }
-
- /* Do we want a physical address? */
- if (!physicalAddress) {
- address = IOKernelAllocateWithPhysicalRestrict(KHEAP_DEFAULT,
- size, 0 /*maxPhys*/, alignment, true, false /* noSoftLimit */);
- } else {
- do {
- IOBufferMemoryDescriptor * bmd;
- mach_vm_address_t physicalMask;
- vm_offset_t alignMask;
-
- alignMask = alignment - 1;
- physicalMask = (0xFFFFFFFF ^ alignMask);
-
- bmd = IOBufferMemoryDescriptor::inTaskWithPhysicalMask(
- kernel_task, kIOMemoryPhysicallyContiguous, size, physicalMask);
- if (!bmd) {
- break;
- }
-
- _IOMallocContiguousEntry *
- entry = IOMallocType(_IOMallocContiguousEntry);
- if (!entry) {
- bmd->release();
- break;
- }
- entry->virtualAddr = (mach_vm_address_t) bmd->getBytesNoCopy();
- entry->md = bmd;
- lck_mtx_lock(gIOMallocContiguousEntriesLock);
- queue_enter( &gIOMallocContiguousEntries, entry,
+void * IOMallocContiguous(vm_size_t size, vm_size_t alignment,
+ IOPhysicalAddress * physicalAddress)
+{
+ mach_vm_address_t address = 0;
+
+ if (size == 0)
+ return 0;
+ if (alignment == 0)
+ alignment = 1;
+
+ /* Do we want a physical address? */
+ if (!physicalAddress)
+ {
+ address = IOKernelAllocateWithPhysicalRestrict(size, 0 /*maxPhys*/, alignment, true);
+ }
+ else do
+ {
+ IOBufferMemoryDescriptor * bmd;
+ mach_vm_address_t physicalMask;
+ vm_offset_t alignMask;
+
+ alignMask = alignment - 1;
+ physicalMask = (0xFFFFFFFF ^ alignMask);
+
+ bmd = IOBufferMemoryDescriptor::inTaskWithPhysicalMask(
+ kernel_task, kIOMemoryPhysicallyContiguous, size, physicalMask);
+ if (!bmd)
+ break;
+
+ _IOMallocContiguousEntry *
+ entry = IONew(_IOMallocContiguousEntry, 1);
+ if (!entry)
+ {
+ bmd->release();
+ break;
+ }
+ entry->virtualAddr = (mach_vm_address_t) bmd->getBytesNoCopy();
+ entry->md = bmd;
+ lck_mtx_lock(gIOMallocContiguousEntriesLock);
+ queue_enter( &gIOMallocContiguousEntries, entry,
+ _IOMallocContiguousEntry *, link );
+ lck_mtx_unlock(gIOMallocContiguousEntriesLock);
+
+ address = (mach_vm_address_t) entry->virtualAddr;
+ *physicalAddress = bmd->getPhysicalAddress();
+ }
+ while (false);
+
+ return (void *) address;
+}
+
+void IOFreeContiguous(void * _address, vm_size_t size)
+{
+ _IOMallocContiguousEntry * entry;
+ IOMemoryDescriptor * md = NULL;
+
+ mach_vm_address_t address = (mach_vm_address_t) _address;
+
+ if( !address)
+ return;
+
+ assert(size);
+
+ lck_mtx_lock(gIOMallocContiguousEntriesLock);
+ queue_iterate( &gIOMallocContiguousEntries, entry,
+ _IOMallocContiguousEntry *, link )
+ {
+ if( entry->virtualAddr == address ) {
+ md = entry->md;
+ queue_remove( &gIOMallocContiguousEntries, entry,
_IOMallocContiguousEntry *, link );
- lck_mtx_unlock(gIOMallocContiguousEntriesLock);
-
- address = (mach_vm_address_t) entry->virtualAddr;
- *physicalAddress = bmd->getPhysicalAddress();
- }while (false);
- }
-
- return (void *) address;
-}
-
-void
-IOFreeContiguous(void * _address, vm_size_t size)
-{
- _IOMallocContiguousEntry * entry;
- IOMemoryDescriptor * md = NULL;
-
- mach_vm_address_t address = (mach_vm_address_t) _address;
-
- if (!address) {
- return;
- }
-
- assert(size);
-
- lck_mtx_lock(gIOMallocContiguousEntriesLock);
- queue_iterate( &gIOMallocContiguousEntries, entry,
- _IOMallocContiguousEntry *, link )
- {
- if (entry->virtualAddr == address) {
- md = entry->md;
- queue_remove( &gIOMallocContiguousEntries, entry,
- _IOMallocContiguousEntry *, link );
- break;
- }
- }
- lck_mtx_unlock(gIOMallocContiguousEntriesLock);
-
- if (md) {
- md->release();
- IOFreeType(entry, _IOMallocContiguousEntry);
- } else {
- IOKernelFreePhysical(KHEAP_DEFAULT, (mach_vm_address_t) address, size);
- }
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-kern_return_t
-IOIteratePageableMaps(vm_size_t size,
- IOIteratePageableMapsCallback callback, void * ref)
-{
- if (size > kIOPageableMaxAllocSize) {
- return kIOReturnBadArgument;
- }
- return (*callback)(gIOKitPageableMap.map, ref);
-}
-
-struct IOMallocPageableRef {
- vm_offset_t address;
- vm_size_t size;
- vm_tag_t tag;
+ break;
+ }
+ }
+ lck_mtx_unlock(gIOMallocContiguousEntriesLock);
+
+ if (md)
+ {
+ md->release();
+ IODelete(entry, _IOMallocContiguousEntry, 1);
+ }
+ else
+ {
+ IOKernelFreePhysical((mach_vm_address_t) address, size);
+ }
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+kern_return_t IOIteratePageableMaps(vm_size_t size,
+ IOIteratePageableMapsCallback callback, void * ref)
+{
+ kern_return_t kr = kIOReturnNotReady;
+ vm_size_t segSize;
+ UInt32 attempts;
+ UInt32 index;
+ vm_offset_t min;
+ vm_map_t map;
+
+ if (size > kIOPageableMaxMapSize)
+ return( kIOReturnBadArgument );
+
+ do {
+ index = gIOKitPageableSpace.hint;
+ attempts = gIOKitPageableSpace.count;
+ while( attempts--) {
+ kr = (*callback)(gIOKitPageableSpace.maps[index].map, ref);
+ if( KERN_SUCCESS == kr) {
+ gIOKitPageableSpace.hint = index;
+ break;
+ }
+ if( index)
+ index--;
+ else
+ index = gIOKitPageableSpace.count - 1;
+ }
+ if (KERN_NO_SPACE != kr)
+ break;
+
+ lck_mtx_lock( gIOKitPageableSpace.lock );
+
+ index = gIOKitPageableSpace.count;
+ if( index >= (kIOMaxPageableMaps - 1)) {
+ lck_mtx_unlock( gIOKitPageableSpace.lock );
+ break;
+ }
+
+ if( size < kIOPageableMapSize)
+ segSize = kIOPageableMapSize;
+ else
+ segSize = size;
+
+ min = 0;
+ kr = kmem_suballoc(kernel_map,
+ &min,
+ segSize,
+ TRUE,
+ VM_FLAGS_ANYWHERE,
+ VM_MAP_KERNEL_FLAGS_NONE,
+ VM_KERN_MEMORY_IOKIT,
+ &map);
+ if( KERN_SUCCESS != kr) {
+ lck_mtx_unlock( gIOKitPageableSpace.lock );
+ break;
+ }
+
+ gIOKitPageableSpace.maps[index].map = map;
+ gIOKitPageableSpace.maps[index].address = min;
+ gIOKitPageableSpace.maps[index].end = min + segSize;
+ gIOKitPageableSpace.hint = index;
+ gIOKitPageableSpace.count = index + 1;
+
+ lck_mtx_unlock( gIOKitPageableSpace.lock );
+
+ } while( true );
+
+ return kr;
+}
+
+struct IOMallocPageableRef
+{
+ vm_offset_t address;
+ vm_size_t size;
+ vm_tag_t tag;
};
-static kern_return_t
-IOMallocPageableCallback(vm_map_t map, void * _ref)
-{
- struct IOMallocPageableRef * ref = (struct IOMallocPageableRef *) _ref;
- kma_flags_t flags = (kma_flags_t)(KMA_PAGEABLE | KMA_DATA_SHARED);
-
- return kmem_alloc( map, &ref->address, ref->size, flags, ref->tag );
-}
-
-static void *
-IOMallocPageablePages(vm_size_t size, vm_size_t alignment, vm_tag_t tag)
-{
- kern_return_t kr = kIOReturnNotReady;
- struct IOMallocPageableRef ref;
-
- if (alignment > page_size) {
- return NULL;
- }
- if (size > kIOPageableMaxAllocSize) {
- return NULL;
- }
-
- ref.size = size;
- ref.tag = tag;
- kr = IOIteratePageableMaps( size, &IOMallocPageableCallback, &ref );
- if (kIOReturnSuccess != kr) {
- ref.address = 0;
- }
-
- return (void *) ref.address;
-}
-
-vm_map_t
-IOPageableMapForAddress(uintptr_t address)
-{
- if (address < gIOKitPageableMap.address || address >= gIOKitPageableMap.end) {
- panic("IOPageableMapForAddress: address out of range");
- }
- return gIOKitPageableMap.map;
-}
-
-static void
-IOFreePageablePages(void * address, vm_size_t size)
-{
- vm_map_t map;
-
- map = IOPageableMapForAddress((vm_address_t) address);
- if (map) {
- kmem_free( map, (vm_offset_t) address, size);
- }
-}
-
-#if defined(__x86_64__)
-static uintptr_t
-IOMallocOnePageablePage(kalloc_heap_t kheap __unused, iopa_t * a)
-{
- return (uintptr_t) IOMallocPageablePages(page_size, page_size, VM_KERN_MEMORY_IOKIT);
-}
-#endif /* defined(__x86_64__) */
-
-static void *
-IOMallocPageableInternal(vm_size_t size, vm_size_t alignment, bool zeroed)
-{
- void * addr;
-
- if (((uint32_t) alignment) != alignment) {
- return NULL;
- }
-#if defined(__x86_64__)
- if (size >= (page_size - 4 * gIOPageAllocChunkBytes) ||
- alignment > page_size) {
- addr = IOMallocPageablePages(size, alignment, IOMemoryTag(kernel_map));
- /* Memory allocated this way will already be zeroed. */
- } else {
- addr = ((void *) iopa_alloc(&gIOPageablePageAllocator,
- &IOMallocOnePageablePage, KHEAP_DEFAULT, size, (uint32_t) alignment));
- if (addr && zeroed) {
- bzero(addr, size);
- }
- }
-#else /* !defined(__x86_64__) */
- vm_size_t allocSize = size;
- if (allocSize == 0) {
- allocSize = 1;
- }
- addr = IOMallocPageablePages(allocSize, alignment, IOMemoryTag(kernel_map));
- /* already zeroed */
-#endif /* defined(__x86_64__) */
-
- if (addr) {
+static kern_return_t IOMallocPageableCallback(vm_map_t map, void * _ref)
+{
+ struct IOMallocPageableRef * ref = (struct IOMallocPageableRef *) _ref;
+ kern_return_t kr;
+
+ kr = kmem_alloc_pageable( map, &ref->address, ref->size, ref->tag );
+
+ return( kr );
+}
+
+static void * IOMallocPageablePages(vm_size_t size, vm_size_t alignment, vm_tag_t tag)
+{
+ kern_return_t kr = kIOReturnNotReady;
+ struct IOMallocPageableRef ref;
+
+ if (alignment > page_size)
+ return( 0 );
+ if (size > kIOPageableMaxMapSize)
+ return( 0 );
+
+ ref.size = size;
+ ref.tag = tag;
+ kr = IOIteratePageableMaps( size, &IOMallocPageableCallback, &ref );
+ if( kIOReturnSuccess != kr)
+ ref.address = 0;
+
+ return( (void *) ref.address );
+}
+
+vm_map_t IOPageableMapForAddress( uintptr_t address )
+{
+ vm_map_t map = 0;
+ UInt32 index;
+
+ for( index = 0; index < gIOKitPageableSpace.count; index++) {
+ if( (address >= gIOKitPageableSpace.maps[index].address)
+ && (address < gIOKitPageableSpace.maps[index].end) ) {
+ map = gIOKitPageableSpace.maps[index].map;
+ break;
+ }
+ }
+ if( !map)
+ panic("IOPageableMapForAddress: null");
+
+ return( map );
+}
+
+static void IOFreePageablePages(void * address, vm_size_t size)
+{
+ vm_map_t map;
+
+ map = IOPageableMapForAddress( (vm_address_t) address);
+ if( map)
+ kmem_free( map, (vm_offset_t) address, size);
+}
+
+static uintptr_t IOMallocOnePageablePage(iopa_t * a)
+{
+ return ((uintptr_t) IOMallocPageablePages(page_size, page_size, VM_KERN_MEMORY_IOKIT));
+}
+
+void * IOMallocPageable(vm_size_t size, vm_size_t alignment)
+{
+ void * addr;
+
+ if (size >= (page_size - 4*gIOPageAllocChunkBytes)) addr = IOMallocPageablePages(size, alignment, IOMemoryTag(kernel_map));
+ else addr = ((void * ) iopa_alloc(&gIOPageablePageAllocator, &IOMallocOnePageablePage, size, alignment));
+
+ if (addr) {
#if IOALLOCDEBUG
- OSAddAtomicLong(size, &debug_iomallocpageable_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsMallocPageable, size);
- }
-
- return addr;
-}
-
-void *
-IOMallocPageable(vm_size_t size, vm_size_t alignment)
-{
- return IOMallocPageableInternal(size, alignment, /*zeroed*/ false);
-}
-
-void *
-IOMallocPageableZero(vm_size_t size, vm_size_t alignment)
-{
- return IOMallocPageableInternal(size, alignment, /*zeroed*/ true);
-}
-
-void
-IOFreePageable(void * address, vm_size_t size)
+ OSAddAtomicLong(size, &debug_iomallocpageable_size);
+#endif
+ IOStatisticsAlloc(kIOStatisticsMallocPageable, size);
+ }
+
+ return (addr);
+}
+
+void IOFreePageable(void * address, vm_size_t size)
{
#if IOALLOCDEBUG
OSAddAtomicLong(-size, &debug_iomallocpageable_size);
#endif
- IOStatisticsAlloc(kIOStatisticsFreePageable, size);
-
-#if defined(__x86_64__)
- if (size < (page_size - 4 * gIOPageAllocChunkBytes)) {
- address = (void *) iopa_free(&gIOPageablePageAllocator, (uintptr_t) address, size);
- size = page_size;
- }
- if (address) {
- IOFreePageablePages(address, size);
- }
-#else /* !defined(__x86_64__) */
- if (size == 0) {
- size = 1;
- }
- if (address) {
- IOFreePageablePages(address, size);
- }
-#endif /* defined(__x86_64__) */
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-
-__typed_allocators_ignore_push
-
-void *
-IOMallocData_external(
- vm_size_t size);
-void *
-IOMallocData_external(vm_size_t size)
-{
- return IOMalloc_internal(KHEAP_DATA_PRIVATE, size, Z_VM_TAG_BT_BIT);
-}
-
-void *
-IOMallocZeroData_external(
- vm_size_t size);
-void *
-IOMallocZeroData_external(vm_size_t size)
-{
- return IOMalloc_internal(KHEAP_DATA_PRIVATE, size, Z_ZERO_VM_TAG_BT_BIT);
-}
-
-void *
-IOMallocDataShareable_external(
- vm_size_t size);
-void *
-IOMallocDataShareable_external(vm_size_t size)
-{
- return IOMalloc_internal(KHEAP_DATA_SHARED, size, Z_VM_TAG_BT_BIT);
-}
-
-void *
-IOMallocZeroDataShareable_external(
- vm_size_t size);
-void *
-IOMallocZeroDataShareable_external(vm_size_t size)
-{
- return IOMalloc_internal(KHEAP_DATA_SHARED, size, Z_ZERO_VM_TAG_BT_BIT);
-}
-
-void
-IOFreeData(void * address, vm_size_t size)
-{
- return IOFree_internal(KHEAP_DATA_PRIVATE, address, size);
-}
-
-void
-IOFreeDataShareable(void * address, vm_size_t size)
-{
- return IOFree_internal(KHEAP_DATA_SHARED, address, size);
-}
-
-__typed_allocators_ignore_pop
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-__typed_allocators_ignore_push // allocator implementation
-
-void *
-IOMallocTypeImpl(kalloc_type_view_t kt_view)
-{
-#if IOTRACKING
- /*
- * When leak detection is on default to using IOMalloc as kalloc
- * type infrastructure isn't aware of needing additional space for
- * the header.
- */
- if (TRACK_ALLOC) {
- uint32_t kt_size = kalloc_type_get_size(kt_view->kt_size);
- void *mem = IOMalloc_internal(KHEAP_DEFAULT, kt_size, Z_ZERO);
- if (!IOMallocType_from_vm(kt_view)) {
- assert(mem);
- }
- return mem;
- }
-#endif
- zalloc_flags_t kt_flags = (zalloc_flags_t) (Z_WAITOK | Z_ZERO);
- if (!IOMallocType_from_vm(kt_view)) {
- kt_flags = (zalloc_flags_t) (kt_flags | Z_NOFAIL);
- }
- /*
- * Use external symbol for kalloc_type_impl as
- * kalloc_type_views generated at some external callsites
- * many not have been processed during boot.
- */
- return kalloc_type_impl_external(kt_view, kt_flags);
-}
-
-void
-IOFreeTypeImpl(kalloc_type_view_t kt_view, void * address)
-{
-#if IOTRACKING
- if (TRACK_ALLOC) {
- return IOFree_internal(KHEAP_DEFAULT, address,
- kalloc_type_get_size(kt_view->kt_size));
- }
-#endif
- /*
- * Use external symbol for kalloc_type_impl as
- * kalloc_type_views generated at some external callsites
- * many not have been processed during boot.
- */
- return kfree_type_impl_external(kt_view, address);
-}
-
-void *
-IOMallocTypeVarImpl(kalloc_type_var_view_t kt_view, vm_size_t size)
-{
-#if IOTRACKING
- /*
- * When leak detection is on default to using IOMalloc as kalloc
- * type infrastructure isn't aware of needing additional space for
- * the header.
- */
- if (TRACK_ALLOC) {
- return IOMalloc_internal(KHEAP_DEFAULT, size, Z_ZERO);
- }
-#endif
- zalloc_flags_t kt_flags = (zalloc_flags_t) (Z_WAITOK | Z_ZERO);
-
- kt_flags = Z_VM_TAG_BT(kt_flags, VM_KERN_MEMORY_KALLOC_TYPE);
- return kalloc_type_var_impl(kt_view, size, kt_flags, NULL);
-}
-
-void
-IOFreeTypeVarImpl(kalloc_type_var_view_t kt_view, void * address,
- vm_size_t size)
-{
-#if IOTRACKING
- if (TRACK_ALLOC) {
- return IOFree_internal(KHEAP_DEFAULT, address, size);
- }
-#endif
-
- return kfree_type_var_impl(kt_view, address, size);
-}
-
-__typed_allocators_ignore_pop
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-#if defined(__x86_64__)
-
-
-extern "C" void
+ IOStatisticsAlloc(kIOStatisticsFreePageable, size);
+
+ if (size < (page_size - 4*gIOPageAllocChunkBytes))
+ {
+ address = (void *) iopa_free(&gIOPageablePageAllocator, (uintptr_t) address, size);
+ size = page_size;
+ }
+ if (address) IOFreePageablePages(address, size);
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+extern "C" void
iopa_init(iopa_t * a)
{
- bzero(a, sizeof(*a));
- a->lock = IOLockAlloc();
- queue_init(&a->list);
+ bzero(a, sizeof(*a));
+ a->lock = IOLockAlloc();
+ queue_init(&a->list);
}
static uintptr_t
iopa_allocinpage(iopa_page_t * pa, uint32_t count, uint64_t align)
{
- uint32_t n, s;
- uint64_t avail = pa->avail;
-
- assert(avail);
-
- // find strings of count 1 bits in avail
- for (n = count; n > 1; n -= s) {
- s = n >> 1;
- avail = avail & (avail << s);
- }
- // and aligned
- avail &= align;
-
- if (avail) {
- n = __builtin_clzll(avail);
- pa->avail &= ~((-1ULL << (64 - count)) >> n);
- if (!pa->avail && pa->link.next) {
- remque(&pa->link);
- pa->link.next = NULL;
- }
- return n * gIOPageAllocChunkBytes + trunc_page((uintptr_t) pa);
- }
-
- return 0;
-}
-
-uintptr_t
-iopa_alloc(
- iopa_t * a,
- iopa_proc_t alloc,
- kalloc_heap_t kheap,
- vm_size_t bytes,
- vm_size_t balign)
-{
- static const uint64_t align_masks[] = {
- 0xFFFFFFFFFFFFFFFF,
- 0xAAAAAAAAAAAAAAAA,
- 0x8888888888888888,
- 0x8080808080808080,
- 0x8000800080008000,
- 0x8000000080000000,
- 0x8000000000000000,
- };
- iopa_page_t * pa;
- uintptr_t addr = 0;
- uint32_t count;
- uint64_t align;
- vm_size_t align_masks_idx;
-
- if (((uint32_t) bytes) != bytes) {
- return 0;
- }
- if (!bytes) {
- bytes = 1;
- }
- count = (((uint32_t) bytes) + gIOPageAllocChunkBytes - 1) / gIOPageAllocChunkBytes;
-
- align_masks_idx = log2up((balign + gIOPageAllocChunkBytes - 1) / gIOPageAllocChunkBytes);
- assert(align_masks_idx < sizeof(align_masks) / sizeof(*align_masks));
- align = align_masks[align_masks_idx];
-
- IOLockLock(a->lock);
- __IGNORE_WCASTALIGN(pa = (typeof(pa))queue_first(&a->list));
- while (!queue_end(&a->list, &pa->link)) {
- addr = iopa_allocinpage(pa, count, align);
- if (addr) {
- a->bytecount += bytes;
- break;
- }
- __IGNORE_WCASTALIGN(pa = (typeof(pa))queue_next(&pa->link));
- }
- IOLockUnlock(a->lock);
-
- if (!addr) {
- addr = alloc(kheap, a);
- if (addr) {
- pa = (typeof(pa))(addr + page_size - gIOPageAllocChunkBytes);
- pa->signature = kIOPageAllocSignature;
- pa->avail = -2ULL;
-
- addr = iopa_allocinpage(pa, count, align);
- IOLockLock(a->lock);
- if (pa->avail) {
- enqueue_head(&a->list, &pa->link);
- }
- a->pagecount++;
- if (addr) {
- a->bytecount += bytes;
- }
- IOLockUnlock(a->lock);
- }
- }
-
- assert((addr & ((1 << log2up(balign)) - 1)) == 0);
- return addr;
-}
-
-uintptr_t
+ uint32_t n, s;
+ uint64_t avail = pa->avail;
+
+ assert(avail);
+
+ // find strings of count 1 bits in avail
+ for (n = count; n > 1; n -= s)
+ {
+ s = n >> 1;
+ avail = avail & (avail << s);
+ }
+ // and aligned
+ avail &= align;
+
+ if (avail)
+ {
+ n = __builtin_clzll(avail);
+ pa->avail &= ~((-1ULL << (64 - count)) >> n);
+ if (!pa->avail && pa->link.next)
+ {
+ remque(&pa->link);
+ pa->link.next = 0;
+ }
+ return (n * gIOPageAllocChunkBytes + trunc_page((uintptr_t) pa));
+ }
+
+ return (0);
+}
+
+uintptr_t
+iopa_alloc(iopa_t * a, iopa_proc_t alloc, vm_size_t bytes, uint32_t balign)
+{
+ static const uint64_t align_masks[] = {
+ 0xFFFFFFFFFFFFFFFF,
+ 0xAAAAAAAAAAAAAAAA,
+ 0x8888888888888888,
+ 0x8080808080808080,
+ 0x8000800080008000,
+ 0x8000000080000000,
+ 0x8000000000000000,
+ };
+ iopa_page_t * pa;
+ uintptr_t addr = 0;
+ uint32_t count;
+ uint64_t align;
+
+ if (!bytes) bytes = 1;
+ count = (bytes + gIOPageAllocChunkBytes - 1) / gIOPageAllocChunkBytes;
+ align = align_masks[log2up((balign + gIOPageAllocChunkBytes - 1) / gIOPageAllocChunkBytes)];
+
+ IOLockLock(a->lock);
+ __IGNORE_WCASTALIGN(pa = (typeof(pa)) queue_first(&a->list));
+ while (!queue_end(&a->list, &pa->link))
+ {
+ addr = iopa_allocinpage(pa, count, align);
+ if (addr)
+ {
+ a->bytecount += bytes;
+ break;
+ }
+ __IGNORE_WCASTALIGN(pa = (typeof(pa)) queue_next(&pa->link));
+ }
+ IOLockUnlock(a->lock);
+
+ if (!addr)
+ {
+ addr = alloc(a);
+ if (addr)
+ {
+ pa = (typeof(pa)) (addr + page_size - gIOPageAllocChunkBytes);
+ pa->signature = kIOPageAllocSignature;
+ pa->avail = -2ULL;
+
+ addr = iopa_allocinpage(pa, count, align);
+ IOLockLock(a->lock);
+ if (pa->avail) enqueue_head(&a->list, &pa->link);
+ a->pagecount++;
+ if (addr) a->bytecount += bytes;
+ IOLockUnlock(a->lock);
+ }
+ }
+
+ assert((addr & ((1 << log2up(balign)) - 1)) == 0);
+ return (addr);
+}
+
+uintptr_t
iopa_free(iopa_t * a, uintptr_t addr, vm_size_t bytes)
{
- iopa_page_t * pa;
- uint32_t count;
- uintptr_t chunk;
-
- if (((uint32_t) bytes) != bytes) {
- return 0;
- }
- if (!bytes) {
- bytes = 1;
- }
-
- chunk = (addr & page_mask);
- assert(0 == (chunk & (gIOPageAllocChunkBytes - 1)));
-
- pa = (typeof(pa))(addr | (page_size - gIOPageAllocChunkBytes));
- assert(kIOPageAllocSignature == pa->signature);
-
- count = (((uint32_t) bytes) + gIOPageAllocChunkBytes - 1) / gIOPageAllocChunkBytes;
- chunk /= gIOPageAllocChunkBytes;
-
- IOLockLock(a->lock);
- if (!pa->avail) {
- assert(!pa->link.next);
- enqueue_tail(&a->list, &pa->link);
- }
- pa->avail |= ((-1ULL << (64 - count)) >> chunk);
- if (pa->avail != -2ULL) {
- pa = NULL;
- } else {
- remque(&pa->link);
- pa->link.next = NULL;
- pa->signature = 0;
- a->pagecount--;
- // page to free
- pa = (typeof(pa))trunc_page(pa);
- }
- a->bytecount -= bytes;
- IOLockUnlock(a->lock);
-
- return (uintptr_t) pa;
-}
-
-#endif /* defined(__x86_64__) */
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-IOReturn
-IOSetProcessorCacheMode( task_t task, IOVirtualAddress address,
- IOByteCount length, IOOptionBits cacheMode )
-{
- IOReturn ret = kIOReturnSuccess;
- ppnum_t pagenum;
-
- if (task != kernel_task) {
- return kIOReturnUnsupported;
- }
- if ((address | length) & PAGE_MASK) {
+ iopa_page_t * pa;
+ uint32_t count;
+ uintptr_t chunk;
+
+ if (!bytes) bytes = 1;
+
+ chunk = (addr & page_mask);
+ assert(0 == (chunk & (gIOPageAllocChunkBytes - 1)));
+
+ pa = (typeof(pa)) (addr | (page_size - gIOPageAllocChunkBytes));
+ assert(kIOPageAllocSignature == pa->signature);
+
+ count = (bytes + gIOPageAllocChunkBytes - 1) / gIOPageAllocChunkBytes;
+ chunk /= gIOPageAllocChunkBytes;
+
+ IOLockLock(a->lock);
+ if (!pa->avail)
+ {
+ assert(!pa->link.next);
+ enqueue_tail(&a->list, &pa->link);
+ }
+ pa->avail |= ((-1ULL << (64 - count)) >> chunk);
+ if (pa->avail != -2ULL) pa = 0;
+ else
+ {
+ remque(&pa->link);
+ pa->link.next = 0;
+ pa->signature = 0;
+ a->pagecount--;
+ // page to free
+ pa = (typeof(pa)) trunc_page(pa);
+ }
+ a->bytecount -= bytes;
+ IOLockUnlock(a->lock);
+
+ return ((uintptr_t) pa);
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+IOReturn IOSetProcessorCacheMode( task_t task, IOVirtualAddress address,
+ IOByteCount length, IOOptionBits cacheMode )
+{
+ IOReturn ret = kIOReturnSuccess;
+ ppnum_t pagenum;
+
+ if( task != kernel_task)
+ return( kIOReturnUnsupported );
+ if ((address | length) & PAGE_MASK)
+ {
// OSReportWithBacktrace("IOSetProcessorCacheMode(0x%x, 0x%x, 0x%x) fails\n", address, length, cacheMode);
- return kIOReturnUnsupported;
- }
- length = round_page(address + length) - trunc_page( address );
- address = trunc_page( address );
-
- // make map mode
- cacheMode = (cacheMode << kIOMapCacheShift) & kIOMapCacheMask;
-
- while ((kIOReturnSuccess == ret) && (length > 0)) {
- // Get the physical page number
- pagenum = pmap_find_phys(kernel_pmap, (addr64_t)address);
- if (pagenum) {
- ret = IOUnmapPages( get_task_map(task), address, page_size );
- ret = IOMapPages( get_task_map(task), address, ptoa_64(pagenum), page_size, cacheMode );
- } else {
- ret = kIOReturnVMError;
- }
-
- address += page_size;
- length -= page_size;
- }
-
- return ret;
-}
-
-
-IOReturn
-IOFlushProcessorCache( task_t task, IOVirtualAddress address,
- IOByteCount length )
-{
- if (task != kernel_task) {
- return kIOReturnUnsupported;
- }
-
- flush_dcache64((addr64_t) address, (unsigned) length, false );
-
- return kIOReturnSuccess;
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-vm_offset_t
-OSKernelStackRemaining( void )
-{
- return ml_stack_remaining();
+ return( kIOReturnUnsupported );
+ }
+ length = round_page(address + length) - trunc_page( address );
+ address = trunc_page( address );
+
+ // make map mode
+ cacheMode = (cacheMode << kIOMapCacheShift) & kIOMapCacheMask;
+
+ while( (kIOReturnSuccess == ret) && (length > 0) ) {
+
+ // Get the physical page number
+ pagenum = pmap_find_phys(kernel_pmap, (addr64_t)address);
+ if( pagenum) {
+ ret = IOUnmapPages( get_task_map(task), address, page_size );
+ ret = IOMapPages( get_task_map(task), address, ptoa_64(pagenum), page_size, cacheMode );
+ } else
+ ret = kIOReturnVMError;
+
+ address += page_size;
+ length -= page_size;
+ }
+
+ return( ret );
+}
+
+
+IOReturn IOFlushProcessorCache( task_t task, IOVirtualAddress address,
+ IOByteCount length )
+{
+ if( task != kernel_task)
+ return( kIOReturnUnsupported );
+
+ flush_dcache64( (addr64_t) address, (unsigned) length, false );
+
+ return( kIOReturnSuccess );
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+vm_offset_t OSKernelStackRemaining( void )
+{
+ return (ml_stack_remaining());
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
@@ -1456,93 +1114,77 @@
/*
* Spin for indicated number of milliseconds.
*/
-void
-IOSleep(unsigned milliseconds)
-{
- delay_for_interval(milliseconds, kMillisecondScale);
+void IOSleep(unsigned milliseconds)
+{
+ delay_for_interval(milliseconds, kMillisecondScale);
}
/*
* Spin for indicated number of milliseconds, and potentially an
* additional number of milliseconds up to the leeway values.
*/
-void
-IOSleepWithLeeway(unsigned intervalMilliseconds, unsigned leewayMilliseconds)
-{
- delay_for_interval_with_leeway(intervalMilliseconds, leewayMilliseconds, kMillisecondScale);
+void IOSleepWithLeeway(unsigned intervalMilliseconds, unsigned leewayMilliseconds)
+{
+ delay_for_interval_with_leeway(intervalMilliseconds, leewayMilliseconds, kMillisecondScale);
}
/*
* Spin for indicated number of microseconds.
*/
-void
-IODelay(unsigned microseconds)
-{
- delay_for_interval(microseconds, kMicrosecondScale);
+void IODelay(unsigned microseconds)
+{
+ delay_for_interval(microseconds, kMicrosecondScale);
}
/*
* Spin for indicated number of nanoseconds.
*/
-void
-IOPause(unsigned nanoseconds)
-{
- delay_for_interval(nanoseconds, kNanosecondScale);
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-static void _IOLogv(const char *format, va_list ap, void *caller) __printflike(1, 0);
-
-__attribute__((noinline, not_tail_called))
-void
-IOLog(const char *format, ...)
-{
- void *caller = __builtin_return_address(0);
- va_list ap;
-
- va_start(ap, format);
- _IOLogv(format, ap, caller);
- va_end(ap);
-}
-
-__attribute__((noinline, not_tail_called))
-void
-IOLogv(const char *format, va_list ap)
-{
- void *caller = __builtin_return_address(0);
- _IOLogv(format, ap, caller);
-}
-
-void
-_IOLogv(const char *format, va_list ap, void *caller)
-{
- va_list ap2;
- struct console_printbuf_state info_data;
- console_printbuf_state_init(&info_data, TRUE, TRUE);
-
- va_copy(ap2, ap);
-
-#pragma clang diagnostic push
-#pragma clang diagnostic ignored "-Wformat-nonliteral"
-#pragma clang diagnostic ignored "-Wformat"
- os_log_with_args(OS_LOG_DEFAULT, OS_LOG_TYPE_DEFAULT, format, ap, caller);
-#pragma clang diagnostic pop
-
- if (!disable_iolog_serial_output) {
- __doprnt(format, ap2, console_printbuf_putc, &info_data, 16, TRUE);
- console_printbuf_clear(&info_data);
- }
- va_end(ap2);
-
- assertf(ml_get_interrupts_enabled() || ml_is_quiescing() ||
- debug_mode_active() || !gCPUsRunning,
- "IOLog called with interrupts disabled");
+void IOPause(unsigned nanoseconds)
+{
+ delay_for_interval(nanoseconds, kNanosecondScale);
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+static void _IOLogv(const char *format, va_list ap, void *caller) __printflike(1,0);
+
+__attribute__((noinline,not_tail_called))
+void IOLog(const char *format, ...)
+{
+ void *caller = __builtin_return_address(0);
+ va_list ap;
+
+ va_start(ap, format);
+ _IOLogv(format, ap, caller);
+ va_end(ap);
+}
+
+__attribute__((noinline,not_tail_called))
+void IOLogv(const char *format, va_list ap)
+{
+ void *caller = __builtin_return_address(0);
+ _IOLogv(format, ap, caller);
+}
+
+void _IOLogv(const char *format, va_list ap, void *caller)
+{
+ va_list ap2;
+ struct console_printbuf_state info_data;
+ console_printbuf_state_init(&info_data, TRUE, TRUE);
+
+ va_copy(ap2, ap);
+
+ os_log_with_args(OS_LOG_DEFAULT, OS_LOG_TYPE_DEFAULT, format, ap, caller);
+
+ __doprnt(format, ap2, console_printbuf_putc, &info_data, 16, TRUE);
+ console_printbuf_clear(&info_data);
+ va_end(ap2);
+
+ assertf(ml_get_interrupts_enabled() || ml_is_quiescing() || debug_mode_active() || !gCPUsRunning, "IOLog called with interrupts disabled");
}
#if !__LP64__
-void
-IOPanic(const char *reason)
+void IOPanic(const char *reason)
{
panic("%s", reason);
}
@@ -1550,34 +1192,36 @@
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-void
-IOKitKernelLogBuffer(const char * title, const void * buffer, size_t size,
- void (*output)(const char *format, ...))
-{
- size_t idx, linestart;
- enum { bytelen = (sizeof("0xZZ, ") - 1) };
- char hex[(bytelen * 16) + 1];
- uint8_t c, chars[17];
-
- output("%s(0x%lx):\n", title, size);
- output(" 0 1 2 3 4 5 6 7 8 9 A B C D E F\n");
- if (size > 4096) {
- size = 4096;
- }
- chars[16] = 0;
- for (idx = 0, linestart = 0; idx < size;) {
- c = ((char *)buffer)[idx];
- snprintf(&hex[bytelen * (idx & 15)], bytelen + 1, "0x%02x, ", c);
- chars[idx & 15] = ((c >= 0x20) && (c <= 0x7f)) ? c : ' ';
- idx++;
- if ((idx == size) || !(idx & 15)) {
- if (idx & 15) {
- chars[idx & 15] = 0;
- }
- output("/* %04lx: */ %-96s /* |%-16s| */\n", linestart, hex, chars);
- linestart += 16;
- }
- }
+void IOKitKernelLogBuffer(const char * title, const void * buffer, size_t size,
+ void (*output)(const char *format, ...))
+{
+ uint8_t c, chars[17];
+ size_t idx;
+
+ output("%s(0x%x):\n", title, size);
+ if (size > 4096) size = 4096;
+ chars[16] = idx = 0;
+ while (true) {
+ if (!(idx & 15)) {
+ if (idx) output(" |%s|\n", chars);
+ if (idx >= size) break;
+ output("%04x: ", idx);
+ }
+ else if (!(idx & 7)) output(" ");
+
+ c = ((char *)buffer)[idx];
+ output("%02x ", c);
+ chars[idx & 15] = ((c >= 0x20) && (c <= 0x7f)) ? c : ' ';
+
+ idx++;
+ if ((idx == size) && (idx & 15)) {
+ chars[idx & 15] = 0;
+ while (idx & 15) {
+ idx++;
+ output(" ");
+ }
+ }
+ }
}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
@@ -1585,27 +1229,24 @@
/*
* Convert a integer constant (typically a #define or enum) to a string.
*/
-static char noValue[80]; // that's pretty
-
-const char *
-IOFindNameForValue(int value, const IONamedValue *regValueArray)
-{
- for (; regValueArray->name; regValueArray++) {
- if (regValueArray->value == value) {
- return regValueArray->name;
- }
+static char noValue[80]; // that's pretty
+
+const char *IOFindNameForValue(int value, const IONamedValue *regValueArray)
+{
+ for( ; regValueArray->name; regValueArray++) {
+ if(regValueArray->value == value)
+ return(regValueArray->name);
}
snprintf(noValue, sizeof(noValue), "0x%x (UNDEFINED)", value);
- return (const char *)noValue;
-}
-
-IOReturn
-IOFindValueForName(const char *string,
- const IONamedValue *regValueArray,
- int *value)
-{
- for (; regValueArray->name; regValueArray++) {
- if (!strcmp(regValueArray->name, string)) {
+ return((const char *)noValue);
+}
+
+IOReturn IOFindValueForName(const char *string,
+ const IONamedValue *regValueArray,
+ int *value)
+{
+ for( ; regValueArray->name; regValueArray++) {
+ if(!strcmp(regValueArray->name, string)) {
*value = regValueArray->value;
return kIOReturnSuccess;
}
@@ -1613,42 +1254,42 @@
return kIOReturnBadArgument;
}
-OSString *
-IOCopyLogNameForPID(int pid)
-{
- char buf[128];
- size_t len;
- snprintf(buf, sizeof(buf), "pid %d, ", pid);
- len = strlen(buf);
- proc_name(pid, buf + len, (int) (sizeof(buf) - len));
- return OSString::withCString(buf);
-}
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-IOAlignment
-IOSizeToAlignment(unsigned int size)
-{
- int shift;
- const int intsize = sizeof(unsigned int) * 8;
-
- for (shift = 1; shift < intsize; shift++) {
- if (size & 0x80000000) {
- return (IOAlignment)(intsize - shift);
- }
- size <<= 1;
- }
- return 0;
-}
-
-unsigned int
-IOAlignmentToSize(IOAlignment align)
-{
- unsigned int size;
-
- for (size = 1; align; align--) {
- size <<= 1;
- }
- return size;
-}
+OSString * IOCopyLogNameForPID(int pid)
+{
+ char buf[128];
+ size_t len;
+ snprintf(buf, sizeof(buf), "pid %d, ", pid);
+ len = strlen(buf);
+ proc_name(pid, buf + len, sizeof(buf) - len);
+ return (OSString::withCString(buf));
+}
+
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+IOAlignment IOSizeToAlignment(unsigned int size)
+{
+ int shift;
+ const int intsize = sizeof(unsigned int) * 8;
+
+ for (shift = 1; shift < intsize; shift++) {
+ if (size & 0x80000000)
+ return (IOAlignment)(intsize - shift);
+ size <<= 1;
+ }
+ return 0;
+}
+
+unsigned int IOAlignmentToSize(IOAlignment align)
+{
+ unsigned int size;
+
+ for (size = 1; align; align--) {
+ size <<= 1;
+ }
+ return size;
+}
+
} /* extern "C" */
+
+
+