Loading...
--- xnu/xnu-12377.101.15/iokit/Kernel/IOBufferMemoryDescriptor.cpp
+++ xnu/xnu-1228.7.58/iokit/Kernel/IOBufferMemoryDescriptor.cpp
@@ -2,7 +2,7 @@
* Copyright (c) 1998-2000 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,596 +22,392 @@
* 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 IOKIT_ENABLE_SHARED_PTR
-
-#define _IOMEMORYDESCRIPTOR_INTERNAL_
-
#include <IOKit/assert.h>
#include <IOKit/system.h>
#include <IOKit/IOLib.h>
#include <IOKit/IOMapper.h>
#include <IOKit/IOBufferMemoryDescriptor.h>
-#include <libkern/OSDebug.h>
-#include <mach/mach_vm.h>
-
-#include <vm/vm_kern_xnu.h>
#include "IOKitKernelInternal.h"
-
-#ifdef IOALLOCDEBUG
-#include <libkern/c++/OSCPPDebug.h>
-#endif
-#include <IOKit/IOStatisticsPrivate.h>
-
-#if IOKITSTATS
-#define IOStatisticsAlloc(type, size) \
-do { \
- IOStatistics::countAlloc(type, size); \
-} while (0)
-#else
-#define IOStatisticsAlloc(type, size)
-#endif /* IOKITSTATS */
-
+#include "IOCopyMapper.h"
__BEGIN_DECLS
void ipc_port_release_send(ipc_port_t port);
#include <vm/pmap.h>
-KALLOC_HEAP_DEFINE(KHEAP_IOBMD_CONTROL, "IOBMD_control", KHEAP_ID_KT_VAR);
+vm_map_t IOPageableMapForAddress( vm_address_t address );
__END_DECLS
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-enum{
- kInternalFlagPhysical = 0x00000001,
- kInternalFlagPageSized = 0x00000002,
- kInternalFlagPageAllocated = 0x00000004,
- kInternalFlagInit = 0x00000008,
- kInternalFlagHasPointers = 0x00000010,
- kInternalFlagGuardPages = 0x00000020,
- /**
- * Should the IOBMD behave as if it has no kernel mapping for the
- * underlying buffer? Note that this does not necessarily imply the
- * existence (or non-existence) of a kernel mapping.
- */
- kInternalFlagAsIfUnmapped = 0x00000040,
-};
+volatile ppnum_t gIOHighestAllocatedPage;
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
#define super IOGeneralMemoryDescriptor
-OSDefineMetaClassAndStructorsWithZone(IOBufferMemoryDescriptor,
- IOGeneralMemoryDescriptor, ZC_ZFREE_CLEARMEM);
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-#if defined(__x86_64__)
-static uintptr_t
-IOBMDPageProc(kalloc_heap_t kheap, iopa_t * a)
-{
- kern_return_t kr;
- vm_address_t vmaddr = 0;
- kma_flags_t kma_flags = KMA_ZERO;
-
- if (kheap == KHEAP_DATA_SHARED) {
- kma_flags = (kma_flags_t) (kma_flags | KMA_DATA_SHARED);
- }
- kr = kmem_alloc(kernel_map, &vmaddr, page_size,
- kma_flags, VM_KERN_MEMORY_IOKIT);
-
- if (KERN_SUCCESS != kr) {
- vmaddr = 0;
- }
-
- return (uintptr_t) vmaddr;
-}
-#endif /* defined(__x86_64__) */
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-#ifndef __LP64__
-bool
-IOBufferMemoryDescriptor::initWithOptions(
- IOOptionBits options,
- vm_size_t capacity,
- vm_offset_t alignment,
- task_t inTask)
-{
- mach_vm_address_t physicalMask = 0;
- return initWithPhysicalMask(inTask, options, capacity, alignment, physicalMask);
-}
-#endif /* !__LP64__ */
-
-OSSharedPtr<IOBufferMemoryDescriptor>
-IOBufferMemoryDescriptor::withCopy(
- task_t inTask,
- IOOptionBits options,
- vm_map_t sourceMap,
- mach_vm_address_t source,
- mach_vm_size_t size)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> inst;
- kern_return_t err;
- vm_map_copy_t copy;
- vm_map_address_t address;
-
- copy = NULL;
- do {
- err = kIOReturnNoMemory;
- inst = OSMakeShared<IOBufferMemoryDescriptor>();
- if (!inst) {
- break;
- }
- inst->_ranges.v64 = IOMallocType(IOAddressRange);
-
- err = vm_map_copyin(sourceMap, source, size,
- false /* src_destroy */, ©);
- if (KERN_SUCCESS != err) {
- break;
- }
-
- err = vm_map_copyout(get_task_map(inTask), &address, copy);
- if (KERN_SUCCESS != err) {
- break;
- }
- copy = NULL;
-
- inst->_ranges.v64->address = address;
- inst->_ranges.v64->length = size;
-
- if (!inst->initWithPhysicalMask(inTask, options, size, page_size, 0)) {
- err = kIOReturnError;
- }
- } while (false);
-
- if (KERN_SUCCESS == err) {
- return inst;
- }
-
- if (copy) {
- vm_map_copy_discard(copy);
- }
-
- return nullptr;
-}
-
-
-bool
-IOBufferMemoryDescriptor::initWithPhysicalMask(
- task_t inTask,
- IOOptionBits options,
- mach_vm_size_t capacity,
- mach_vm_address_t alignment,
- mach_vm_address_t physicalMask)
-{
- task_t mapTask = NULL;
- kalloc_heap_t kheap = KHEAP_DATA_SHARED;
- mach_vm_address_t highestMask = 0;
- IOOptionBits iomdOptions = kIOMemoryTypeVirtual64 | kIOMemoryAsReference;
- IODMAMapSpecification mapSpec;
- bool mapped = false;
- bool withCopy = false;
- bool mappedOrShared = false;
- bool noSoftLimit = false;
-
- if (!capacity) {
+OSDefineMetaClassAndStructors(IOBufferMemoryDescriptor,
+ IOGeneralMemoryDescriptor);
+
+bool IOBufferMemoryDescriptor::initWithAddress(
+ void * /* address */ ,
+ IOByteCount /* withLength */ ,
+ IODirection /* withDirection */ )
+{
+ return false;
+}
+
+bool IOBufferMemoryDescriptor::initWithAddress(
+ vm_address_t /* address */ ,
+ IOByteCount /* withLength */ ,
+ IODirection /* withDirection */ ,
+ task_t /* withTask */ )
+{
+ return false;
+}
+
+bool IOBufferMemoryDescriptor::initWithPhysicalAddress(
+ IOPhysicalAddress /* address */ ,
+ IOByteCount /* withLength */ ,
+ IODirection /* withDirection */ )
+{
+ return false;
+}
+
+bool IOBufferMemoryDescriptor::initWithPhysicalRanges(
+ IOPhysicalRange * /* ranges */ ,
+ UInt32 /* withCount */ ,
+ IODirection /* withDirection */ ,
+ bool /* asReference */ )
+{
+ return false;
+}
+
+bool IOBufferMemoryDescriptor::initWithRanges(
+ IOVirtualRange * /* ranges */ ,
+ UInt32 /* withCount */ ,
+ IODirection /* withDirection */ ,
+ task_t /* withTask */ ,
+ bool /* asReference */ )
+{
+ return false;
+}
+
+bool IOBufferMemoryDescriptor::initWithOptions(
+ IOOptionBits options,
+ vm_size_t capacity,
+ vm_offset_t alignment,
+ task_t inTask)
+{
+ mach_vm_address_t physicalMask = 0;
+ return (initWithPhysicalMask(inTask, options, capacity, alignment, physicalMask));
+}
+
+bool IOBufferMemoryDescriptor::initWithPhysicalMask(
+ task_t inTask,
+ IOOptionBits options,
+ mach_vm_size_t capacity,
+ mach_vm_address_t alignment,
+ mach_vm_address_t physicalMask)
+{
+ kern_return_t kr;
+ task_t mapTask = NULL;
+ vm_map_t vmmap = NULL;
+ addr64_t lastIOAddr;
+ IOAddressRange range;
+ IOOptionBits iomdOptions = kIOMemoryTypeVirtual64;
+
+ if (!capacity)
+ return false;
+
+ _options = options;
+ _capacity = capacity;
+ _physAddrs = 0;
+ _physSegCount = 0;
+ _buffer = 0;
+ range.address = 0;
+ range.length = 0;
+ _ranges.v64 = ⦥
+
+ // Grab the direction and the Auto Prepare bits from the Buffer MD options
+ iomdOptions |= options & (kIOMemoryDirectionMask | kIOMemoryAutoPrepare);
+
+ if ((options & (kIOMemorySharingTypeMask | kIOMapCacheMask)) && (alignment < page_size))
+ alignment = page_size;
+
+ if (physicalMask && (alignment <= 1))
+ alignment = ((physicalMask ^ PAGE_MASK) & PAGE_MASK) + 1;
+
+ _alignment = alignment;
+
+ if (((inTask != kernel_task) && !(options & kIOMemoryPageable)) ||
+ (physicalMask && (options & kIOMapCacheMask)))
+ return false;
+
+ if ((options & kIOMemoryPhysicallyContiguous) && !physicalMask)
+ physicalMask = 0xFFFFFFFF;
+
+ // set flags for entry + object create
+ vm_prot_t memEntryCacheMode = VM_PROT_READ | VM_PROT_WRITE;
+
+ // set memory entry cache mode
+ switch (options & kIOMapCacheMask)
+ {
+ case kIOMapInhibitCache:
+ SET_MAP_MEM(MAP_MEM_IO, memEntryCacheMode);
+ break;
+
+ case kIOMapWriteThruCache:
+ SET_MAP_MEM(MAP_MEM_WTHRU, memEntryCacheMode);
+ break;
+
+ case kIOMapWriteCombineCache:
+ SET_MAP_MEM(MAP_MEM_WCOMB, memEntryCacheMode);
+ break;
+
+ case kIOMapCopybackCache:
+ SET_MAP_MEM(MAP_MEM_COPYBACK, memEntryCacheMode);
+ break;
+
+ case kIOMapDefaultCache:
+ default:
+ SET_MAP_MEM(MAP_MEM_NOOP, memEntryCacheMode);
+ break;
+ }
+
+ if (options & kIOMemoryPageable)
+ {
+ iomdOptions |= kIOMemoryBufferPageable;
+
+ // must create the entry before any pages are allocated
+
+ // set flags for entry + object create
+ memEntryCacheMode |= MAP_MEM_NAMED_CREATE;
+
+ if (options & kIOMemoryPurgeable)
+ memEntryCacheMode |= MAP_MEM_PURGABLE;
+ }
+ else
+ {
+ memEntryCacheMode |= MAP_MEM_NAMED_REUSE;
+
+ if (IOMapper::gSystem)
+ // assuming mapped space is 2G
+ lastIOAddr = (1UL << 31) - PAGE_SIZE;
+ else
+ lastIOAddr = ptoa_64(gIOHighestAllocatedPage);
+
+ if (physicalMask && (lastIOAddr != (lastIOAddr & physicalMask)))
+ {
+ mach_vm_address_t address;
+ iomdOptions &= ~kIOMemoryTypeVirtual64;
+ iomdOptions |= kIOMemoryTypePhysical64;
+
+ address = IOMallocPhysical(capacity, physicalMask);
+ _buffer = (void *) address;
+ if (!_buffer)
return false;
- }
-
- /*
- * The IOKit constructor requests the allocator for zeroed memory
- * so the members of the class do not need to be explicitly zeroed.
- */
- _options = options;
- _capacity = capacity;
-
- if (!_ranges.v64) {
- _ranges.v64 = IOMallocType(IOAddressRange);
- _ranges.v64->address = 0;
- _ranges.v64->length = 0;
- } else {
- if (!_ranges.v64->address) {
- return false;
- }
- if (!(kIOMemoryPageable & options)) {
- return false;
- }
- if (!inTask) {
- return false;
- }
- _buffer = (void *) _ranges.v64->address;
- withCopy = true;
- }
-
- /*
- * Set kalloc_heap to KHEAP_IOBMD_CONTROL if allocation contains pointers
- */
- if (kInternalFlagHasPointers & _internalFlags) {
- kheap = KHEAP_IOBMD_CONTROL;
- }
-
- // make sure super::free doesn't dealloc _ranges before super::init
- _flags = kIOMemoryAsReference;
-
- // Grab IOMD bits from the Buffer MD options
- iomdOptions |= (options & kIOBufferDescriptorMemoryFlags);
-
- if (!(kIOMemoryMapperNone & options)) {
- IOMapper::checkForSystemMapper();
- mapped = (NULL != IOMapper::gSystem);
- }
-
- if (physicalMask && (alignment <= 1)) {
- alignment = ((physicalMask ^ (-1ULL)) & (physicalMask - 1));
- highestMask = (physicalMask | alignment);
- alignment++;
- if (alignment < page_size) {
- alignment = page_size;
- }
- }
-
- if ((options & (kIOMemorySharingTypeMask | kIOMapCacheMask | kIOMemoryClearEncrypt)) && (alignment < page_size)) {
- alignment = page_size;
- }
-
- if (alignment >= page_size) {
- if (round_page_overflow(capacity, &capacity)) {
- return false;
- }
- }
-
- if (alignment > page_size) {
- options |= kIOMemoryPhysicallyContiguous;
- }
-
- _alignment = alignment;
-
- if ((capacity + alignment) < _capacity) {
+
+ mapTask = inTask;
+ inTask = 0;
+ }
+ else
+ {
+ vmmap = kernel_map;
+
+ // Buffer shouldn't auto prepare they should be prepared explicitly
+ // But it never was enforced so what are you going to do?
+ iomdOptions |= kIOMemoryAutoPrepare;
+
+ /* Allocate a wired-down buffer inside kernel space. */
+ if (options & kIOMemoryPhysicallyContiguous)
+ _buffer = (void *) IOKernelAllocateContiguous(capacity, alignment);
+ else if (alignment > 1)
+ _buffer = IOMallocAligned(capacity, alignment);
+ else
+ _buffer = IOMalloc(capacity);
+ if (!_buffer)
return false;
}
-
- if (inTask) {
- if ((inTask != kernel_task) && !(options & kIOMemoryPageable)) {
- // Cannot create non-pageable memory in user tasks
- return false;
- }
- } else {
- // Not passing a task implies the memory should not be mapped (or, at
- // least, should behave as if it were not mapped)
- _internalFlags |= kInternalFlagAsIfUnmapped;
-
- // Disable the soft-limit since the mapping, if any, will not escape the
- // IOBMD.
- noSoftLimit = true;
- }
-
- bzero(&mapSpec, sizeof(mapSpec));
- mapSpec.alignment = _alignment;
- mapSpec.numAddressBits = 64;
- if (highestMask && mapped) {
- if (highestMask <= 0xFFFFFFFF) {
- mapSpec.numAddressBits = (uint8_t)(32 - __builtin_clz((unsigned int) highestMask));
- } else {
- mapSpec.numAddressBits = (uint8_t)(64 - __builtin_clz((unsigned int) (highestMask >> 32)));
- }
- highestMask = 0;
- }
-
- // set memory entry cache mode, pageable, purgeable
- iomdOptions |= ((options & kIOMapCacheMask) >> kIOMapCacheShift) << kIOMemoryBufferCacheShift;
- if (options & kIOMemoryPageable) {
- if (_internalFlags & kInternalFlagGuardPages) {
- printf("IOBMD: Unsupported use of guard pages with pageable memory.\n");
- return false;
- }
- iomdOptions |= kIOMemoryBufferPageable;
- if (options & kIOMemoryPurgeable) {
- iomdOptions |= kIOMemoryBufferPurgeable;
- }
- } else {
- // Buffer shouldn't auto prepare they should be prepared explicitly
- // But it never was enforced so what are you going to do?
- iomdOptions |= kIOMemoryAutoPrepare;
-
- /* Allocate a wired-down buffer inside kernel space. */
-
- bool contig = (0 != (options & kIOMemoryHostPhysicallyContiguous));
-
- if (!contig && (0 != (options & kIOMemoryPhysicallyContiguous))) {
- contig |= (!mapped);
- contig |= (0 != (kIOMemoryMapperNone & options));
-#if 0
- // treat kIOMemoryPhysicallyContiguous as kIOMemoryHostPhysicallyContiguous for now
- contig |= true;
+ }
+
+ if( (kIOMemoryTypePhysical64 != (kIOMemoryTypeMask & iomdOptions))
+ && (options & (kIOMemoryPageable | kIOMapCacheMask))) {
+ ipc_port_t sharedMem;
+ vm_size_t size = round_page_32(capacity);
+
+ kr = mach_make_memory_entry(vmmap,
+ &size, (vm_offset_t)_buffer,
+ memEntryCacheMode, &sharedMem,
+ NULL );
+
+ if( (KERN_SUCCESS == kr) && (size != round_page_32(capacity))) {
+ ipc_port_release_send( sharedMem );
+ kr = kIOReturnVMError;
+ }
+ if( KERN_SUCCESS != kr)
+ return( false );
+
+ _memEntry = (void *) sharedMem;
+
+ if( options & kIOMemoryPageable) {
+#if IOALLOCDEBUG
+ debug_iomallocpageable_size += size;
#endif
- }
-
- mappedOrShared = (mapped || (0 != (kIOMemorySharingTypeMask & options)));
- if (contig || highestMask || (alignment > page_size)) {
- if (_internalFlags & kInternalFlagGuardPages) {
- printf("IOBMD: Unsupported use of guard pages with physical mask or contiguous memory.\n");
- return false;
- }
- _internalFlags |= kInternalFlagPhysical;
- if (highestMask) {
- _internalFlags |= kInternalFlagPageSized;
- if (round_page_overflow(capacity, &capacity)) {
- return false;
- }
- }
- _buffer = (void *) IOKernelAllocateWithPhysicalRestrict(kheap,
- capacity, highestMask, alignment, contig, noSoftLimit);
- } else if (_internalFlags & kInternalFlagGuardPages) {
- vm_offset_t address = 0;
- kern_return_t kr;
- uintptr_t alignMask;
- kma_flags_t kma_flags = (kma_flags_t) (KMA_GUARD_FIRST |
- KMA_GUARD_LAST | KMA_ZERO);
-
- if (((uint32_t) alignment) != alignment) {
- return false;
- }
- if (kheap == KHEAP_DATA_SHARED) {
- kma_flags = (kma_flags_t) (kma_flags | KMA_DATA_SHARED);
- }
-
- if (noSoftLimit) {
- kma_flags = (kma_flags_t)(kma_flags | KMA_NOSOFTLIMIT);
- }
-
- alignMask = (1UL << log2up((uint32_t) alignment)) - 1;
- kr = kernel_memory_allocate(kernel_map, &address,
- capacity + page_size * 2, alignMask, kma_flags,
- IOMemoryTag(kernel_map));
- if (kr != KERN_SUCCESS || address == 0) {
- return false;
- }
-#if IOALLOCDEBUG
- OSAddAtomicLong(capacity, &debug_iomalloc_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsMallocAligned, capacity);
- _buffer = (void *)(address + page_size);
-#if defined(__x86_64__)
- } else if (mappedOrShared
- && (capacity + alignment) <= (page_size - gIOPageAllocChunkBytes)) {
- _internalFlags |= kInternalFlagPageAllocated;
- _buffer = (void *) iopa_alloc(&gIOBMDPageAllocator,
- &IOBMDPageProc, kheap, capacity, alignment);
- if (_buffer) {
- bzero(_buffer, capacity);
- IOStatisticsAlloc(kIOStatisticsMallocAligned, capacity);
-#if IOALLOCDEBUG
- OSAddAtomicLong(capacity, &debug_iomalloc_size);
-#endif
- }
-#endif /* defined(__x86_64__) */
- } else {
- zalloc_flags_t zflags = Z_ZERO_VM_TAG_BT_BIT;
- if (noSoftLimit) {
- zflags = (zalloc_flags_t)(zflags | Z_NOSOFTLIMIT);
- }
-
- /* BEGIN IGNORE CODESTYLE */
- __typed_allocators_ignore_push
- if (alignment > 1) {
- _buffer = IOMallocAligned_internal(kheap, capacity, alignment,
- zflags);
- } else {
- _buffer = IOMalloc_internal(kheap, capacity, zflags);
- }
- __typed_allocators_ignore_pop
- /* END IGNORE CODESTYLE */
- }
- if (!_buffer) {
- return false;
- }
- }
-
- if ((options & (kIOMemoryPageable | kIOMapCacheMask))) {
- vm_size_t size = round_page(capacity);
-
- // initWithOptions will create memory entry
- if (!withCopy) {
- iomdOptions |= kIOMemoryPersistent;
- }
-
- if (options & kIOMemoryPageable) {
-#if IOALLOCDEBUG
- OSAddAtomicLong(size, &debug_iomallocpageable_size);
-#endif
- if (!withCopy) {
- mapTask = inTask;
- }
- } else if (options & kIOMapCacheMask) {
- // Prefetch each page to put entries into the pmap
- volatile UInt8 * startAddr = (UInt8 *)_buffer;
- volatile UInt8 * endAddr = (UInt8 *)_buffer + capacity;
-
- while (startAddr < endAddr) {
- UInt8 dummyVar = *startAddr;
- (void) dummyVar;
- startAddr += page_size;
- }
- }
- }
-
- _ranges.v64->address = (mach_vm_address_t) _buffer;
- _ranges.v64->length = _capacity;
-
- if (!super::initWithOptions(
- /* buffers */ _ranges.v64, /* count */ 1, /* offset */ 0,
- // Since we handle all "unmapped" behavior internally and our superclass
- // requires a task, default all unbound IOBMDs to the kernel task.
- /* task */ inTask ?: kernel_task,
- /* options */ iomdOptions,
- /* System mapper */ NULL)) {
- return false;
- }
-
- _internalFlags |= kInternalFlagInit;
-#if IOTRACKING
- if (!(options & kIOMemoryPageable)) {
- trackingAccumSize(capacity);
- }
-#endif /* IOTRACKING */
-
- // give any system mapper the allocation params
- if (kIOReturnSuccess != dmaCommandOperation(kIOMDAddDMAMapSpec,
- &mapSpec, sizeof(mapSpec))) {
- return false;
- }
-
- if (mapTask) {
- if (!reserved) {
- reserved = IOMallocType(ExpansionData);
- if (!reserved) {
- return false;
- }
- }
- reserved->map = createMappingInTask(mapTask, 0,
- kIOMapAnywhere | (options & kIOMapPrefault) | (options & kIOMapCacheMask), 0, 0).detach();
- if (!reserved->map) {
- _buffer = NULL;
- return false;
- }
- release(); // map took a retain on this
- reserved->map->retain();
- removeMapping(reserved->map);
- mach_vm_address_t buffer = reserved->map->getAddress();
- _buffer = (void *) buffer;
- if (kIOMemoryTypeVirtual64 == (kIOMemoryTypeMask & iomdOptions)) {
- _ranges.v64->address = buffer;
- }
- }
-
- setLength(_capacity);
-
- return true;
-}
-
-bool
-IOBufferMemoryDescriptor::initControlWithPhysicalMask(
- task_t inTask,
- IOOptionBits options,
- mach_vm_size_t capacity,
- mach_vm_address_t alignment,
- mach_vm_address_t physicalMask)
-{
- _internalFlags = kInternalFlagHasPointers;
- return initWithPhysicalMask(inTask, options, capacity, alignment,
- physicalMask);
-}
-
-bool
-IOBufferMemoryDescriptor::initWithGuardPages(
- task_t inTask,
- IOOptionBits options,
- mach_vm_size_t capacity)
-{
- mach_vm_size_t roundedCapacity;
-
- _internalFlags = kInternalFlagGuardPages;
-
- if (round_page_overflow(capacity, &roundedCapacity)) {
- return false;
- }
-
- return initWithPhysicalMask(inTask, options, roundedCapacity, page_size,
- (mach_vm_address_t)0);
-}
-
-OSSharedPtr<IOBufferMemoryDescriptor>
-IOBufferMemoryDescriptor::inTaskWithOptions(
- task_t inTask,
- IOOptionBits options,
- vm_size_t capacity,
- vm_offset_t alignment)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> me = OSMakeShared<IOBufferMemoryDescriptor>();
-
- if (me && !me->initWithPhysicalMask(inTask, options, capacity, alignment, 0)) {
- me.reset();
- }
- return me;
-}
-
-OSSharedPtr<IOBufferMemoryDescriptor>
-IOBufferMemoryDescriptor::inTaskWithOptions(
- task_t inTask,
- IOOptionBits options,
- vm_size_t capacity,
- vm_offset_t alignment,
- uint32_t kernTag,
- uint32_t userTag)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> me = OSMakeShared<IOBufferMemoryDescriptor>();
-
- if (me) {
- me->setVMTags(kernTag, userTag);
-
- if (!me->initWithPhysicalMask(inTask, options, capacity, alignment, 0)) {
- me.reset();
- }
- }
- return me;
-}
-
-OSSharedPtr<IOBufferMemoryDescriptor>
-IOBufferMemoryDescriptor::inTaskWithPhysicalMask(
- task_t inTask,
- IOOptionBits options,
- mach_vm_size_t capacity,
- mach_vm_address_t physicalMask)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> me = OSMakeShared<IOBufferMemoryDescriptor>();
-
- if (me && !me->initWithPhysicalMask(inTask, options, capacity, 1, physicalMask)) {
- me.reset();
- }
- return me;
-}
-
-OSSharedPtr<IOBufferMemoryDescriptor>
-IOBufferMemoryDescriptor::inTaskWithGuardPages(
- task_t inTask,
- IOOptionBits options,
- mach_vm_size_t capacity)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> me = OSMakeShared<IOBufferMemoryDescriptor>();
-
- if (me && !me->initWithGuardPages(inTask, options, capacity)) {
- me.reset();
- }
- return me;
-}
-
-#ifndef __LP64__
-bool
-IOBufferMemoryDescriptor::initWithOptions(
- IOOptionBits options,
- vm_size_t capacity,
- vm_offset_t alignment)
-{
- return initWithPhysicalMask(kernel_task, options, capacity, alignment, (mach_vm_address_t)0);
-}
-#endif /* !__LP64__ */
-
-OSSharedPtr<IOBufferMemoryDescriptor>
-IOBufferMemoryDescriptor::withOptions(
- IOOptionBits options,
- vm_size_t capacity,
- vm_offset_t alignment)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> me = OSMakeShared<IOBufferMemoryDescriptor>();
-
- if (me && !me->initWithPhysicalMask(kernel_task, options, capacity, alignment, 0)) {
- me.reset();
- }
- return me;
+ mapTask = inTask;
+ if (NULL == inTask)
+ inTask = kernel_task;
+ }
+ else if (options & kIOMapCacheMask)
+ {
+ // Prefetch each page to put entries into the pmap
+ volatile UInt8 * startAddr = (UInt8 *)_buffer;
+ volatile UInt8 * endAddr = (UInt8 *)_buffer + capacity;
+
+ while (startAddr < endAddr)
+ {
+ *startAddr;
+ startAddr += page_size;
+ }
+ }
+ }
+
+ range.address = (mach_vm_address_t) _buffer;
+ range.length = capacity;
+
+ if (!super::initWithOptions(&range, 1, 0,
+ inTask, iomdOptions, /* System mapper */ 0))
+ return false;
+
+ if (physicalMask && !IOMapper::gSystem)
+ {
+ IOMDDMACharacteristics mdSummary;
+
+ bzero(&mdSummary, sizeof(mdSummary));
+ IOReturn rtn = dmaCommandOperation(
+ kIOMDGetCharacteristics,
+ &mdSummary, sizeof(mdSummary));
+ if (rtn)
+ return false;
+
+ if (mdSummary.fHighestPage)
+ {
+ ppnum_t highest;
+ while (mdSummary.fHighestPage > (highest = gIOHighestAllocatedPage))
+ {
+ if (OSCompareAndSwap(highest, mdSummary.fHighestPage,
+ (UInt32 *) &gIOHighestAllocatedPage))
+ break;
+ }
+ lastIOAddr = ptoa_64(mdSummary.fHighestPage);
+ }
+ else
+ lastIOAddr = ptoa_64(gIOLastPage);
+
+ if (lastIOAddr != (lastIOAddr & physicalMask))
+ {
+ if (kIOMemoryTypePhysical64 != (_flags & kIOMemoryTypeMask))
+ {
+ // flag a retry
+ _physSegCount = 1;
+ }
+ return false;
+ }
+ }
+
+ if (mapTask)
+ {
+ if (!reserved) {
+ reserved = IONew( ExpansionData, 1 );
+ if( !reserved)
+ return( false );
+ }
+ reserved->map = map(mapTask, 0, kIOMapAnywhere, 0, 0);
+ if (!reserved->map)
+ {
+ _buffer = 0;
+ return( false );
+ }
+ release(); // map took a retain on this
+ mach_vm_address_t buffer = reserved->map->getAddress();
+ _buffer = (void *) buffer;
+ if (kIOMemoryTypeVirtual64 == (kIOMemoryTypeMask & iomdOptions))
+ _ranges.v64->address = buffer;
+ }
+
+ setLength(capacity);
+
+ return true;
+}
+
+IOBufferMemoryDescriptor * IOBufferMemoryDescriptor::inTaskWithOptions(
+ task_t inTask,
+ IOOptionBits options,
+ vm_size_t capacity,
+ vm_offset_t alignment)
+{
+ IOBufferMemoryDescriptor *me = new IOBufferMemoryDescriptor;
+
+ if (me && !me->initWithOptions(options, capacity, alignment, inTask)) {
+ bool retry = me->_physSegCount;
+ me->release();
+ me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithOptions(options, capacity, alignment, inTask))
+ {
+ me->release();
+ me = 0;
+ }
+ }
+ }
+ return me;
+}
+
+IOBufferMemoryDescriptor * IOBufferMemoryDescriptor::inTaskWithPhysicalMask(
+ task_t inTask,
+ IOOptionBits options,
+ mach_vm_size_t capacity,
+ mach_vm_address_t physicalMask)
+{
+ IOBufferMemoryDescriptor *me = new IOBufferMemoryDescriptor;
+
+ if (me && !me->initWithPhysicalMask(inTask, options, capacity, 1, physicalMask))
+ {
+ bool retry = me->_physSegCount;
+ me->release();
+ me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithPhysicalMask(inTask, options, capacity, 1, physicalMask))
+ {
+ me->release();
+ me = 0;
+ }
+ }
+ }
+ return me;
+}
+
+bool IOBufferMemoryDescriptor::initWithOptions(
+ IOOptionBits options,
+ vm_size_t capacity,
+ vm_offset_t alignment)
+{
+ return( initWithOptions(options, capacity, alignment, kernel_task) );
+}
+
+IOBufferMemoryDescriptor * IOBufferMemoryDescriptor::withOptions(
+ IOOptionBits options,
+ vm_size_t capacity,
+ vm_offset_t alignment)
+{
+ return(IOBufferMemoryDescriptor::inTaskWithOptions(kernel_task, options, capacity, alignment));
}
@@ -621,46 +417,42 @@
* Returns a new IOBufferMemoryDescriptor with a buffer large enough to
* hold capacity bytes. The descriptor's length is initially set to the capacity.
*/
-OSSharedPtr<IOBufferMemoryDescriptor>
+IOBufferMemoryDescriptor *
IOBufferMemoryDescriptor::withCapacity(vm_size_t inCapacity,
- IODirection inDirection,
- bool inContiguous)
-{
- return IOBufferMemoryDescriptor::withOptions(
- inDirection | kIOMemoryUnshared
- | (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
- inCapacity, inContiguous ? inCapacity : 1 );
-}
-
-#ifndef __LP64__
+ IODirection inDirection,
+ bool inContiguous)
+{
+ return( IOBufferMemoryDescriptor::withOptions(
+ inDirection | kIOMemoryUnshared
+ | (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
+ inCapacity, inContiguous ? inCapacity : 1 ));
+}
+
/*
* initWithBytes:
*
* Initialize a new IOBufferMemoryDescriptor preloaded with bytes (copied).
* The descriptor's length and capacity are set to the input buffer's size.
*/
-bool
-IOBufferMemoryDescriptor::initWithBytes(const void * inBytes,
- vm_size_t inLength,
- IODirection inDirection,
- bool inContiguous)
-{
- if (!initWithPhysicalMask(kernel_task, inDirection | kIOMemoryUnshared
- | (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
- inLength, inLength, (mach_vm_address_t)0)) {
- return false;
- }
-
- // start out with no data
- setLength(0);
-
- if (!appendBytes(inBytes, inLength)) {
- return false;
- }
-
- return true;
-}
-#endif /* !__LP64__ */
+bool IOBufferMemoryDescriptor::initWithBytes(const void * inBytes,
+ vm_size_t inLength,
+ IODirection inDirection,
+ bool inContiguous)
+{
+ if (!initWithOptions(
+ inDirection | kIOMemoryUnshared
+ | (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
+ inLength, inLength ))
+ return false;
+
+ // start out with no data
+ setLength(0);
+
+ if (!appendBytes(inBytes, inLength))
+ return false;
+
+ return true;
+}
/*
* withBytes:
@@ -668,32 +460,31 @@
* Returns a new IOBufferMemoryDescriptor preloaded with bytes (copied).
* The descriptor's length and capacity are set to the input buffer's size.
*/
-OSSharedPtr<IOBufferMemoryDescriptor>
+IOBufferMemoryDescriptor *
IOBufferMemoryDescriptor::withBytes(const void * inBytes,
- vm_size_t inLength,
- IODirection inDirection,
- bool inContiguous)
-{
- OSSharedPtr<IOBufferMemoryDescriptor> me = OSMakeShared<IOBufferMemoryDescriptor>();
- mach_vm_address_t alignment;
-
- alignment = (inLength <= page_size) ? inLength : page_size;
- if (me && !me->initWithPhysicalMask(
- kernel_task, inDirection | kIOMemoryUnshared
- | (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
- inLength, alignment, 0 )) {
- me.reset();
- }
-
- if (me) {
- // start out with no data
- me->setLength(0);
-
- if (!me->appendBytes(inBytes, inLength)) {
- me.reset();
- }
- }
- return me;
+ vm_size_t inLength,
+ IODirection inDirection,
+ bool inContiguous)
+{
+ IOBufferMemoryDescriptor *me = new IOBufferMemoryDescriptor;
+
+ if (me && !me->initWithBytes(inBytes, inLength, inDirection, inContiguous))
+ {
+ bool retry = me->_physSegCount;
+ me->release();
+ me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithBytes(inBytes, inLength, inDirection, inContiguous))
+ {
+ me->release();
+ me = 0;
+ }
+ }
+
+ }
+ return me;
}
/*
@@ -701,101 +492,46 @@
*
* Free resources
*/
-void
-IOBufferMemoryDescriptor::free()
-{
- // Cache all of the relevant information on the stack for use
- // after we call super::free()!
- IOOptionBits flags = _flags;
- IOOptionBits internalFlags = _internalFlags;
- IOOptionBits options = _options;
- vm_size_t size = _capacity;
- void * buffer = _buffer;
- IOMemoryMap * map = NULL;
- IOAddressRange * range = _ranges.v64;
- vm_offset_t alignment = _alignment;
- kalloc_heap_t kheap = KHEAP_DATA_SHARED;
- vm_size_t rsize;
-
- if (alignment >= page_size) {
- if (!round_page_overflow(size, &rsize)) {
- size = rsize;
- }
- }
-
- if (reserved) {
- map = reserved->map;
- IOFreeType(reserved, ExpansionData);
- if (map) {
- map->release();
- }
- }
-
- if ((options & kIOMemoryPageable)
- || (kInternalFlagPageSized & internalFlags)) {
- if (!round_page_overflow(size, &rsize)) {
- size = rsize;
- }
- }
-
- if (internalFlags & kInternalFlagHasPointers) {
- kheap = KHEAP_IOBMD_CONTROL;
- }
-
-#if IOTRACKING
- if (!(options & kIOMemoryPageable)
- && buffer
- && (kInternalFlagInit & _internalFlags)) {
- trackingAccumSize(-size);
- }
-#endif /* IOTRACKING */
-
- /* super::free may unwire - deallocate buffer afterwards */
- super::free();
-
- if (options & kIOMemoryPageable) {
+void IOBufferMemoryDescriptor::free()
+{
+ // Cache all of the relevant information on the stack for use
+ // after we call super::free()!
+ IOOptionBits flags = _flags;
+ IOOptionBits options = _options;
+ vm_size_t size = _capacity;
+ void * buffer = _buffer;
+ mach_vm_address_t source = (_ranges.v) ? _ranges.v64->address : 0;
+ IOMemoryMap * map = 0;
+ vm_offset_t alignment = _alignment;
+
+ if (reserved)
+ {
+ map = reserved->map;
+ IODelete( reserved, ExpansionData, 1 );
+ if (map)
+ map->release();
+ }
+
+ /* super::free may unwire - deallocate buffer afterwards */
+ super::free();
+
+ if (options & kIOMemoryPageable)
+ {
#if IOALLOCDEBUG
- OSAddAtomicLong(-size, &debug_iomallocpageable_size);
+ debug_iomallocpageable_size -= round_page_32(size);
#endif
- } else if (buffer) {
- if (kInternalFlagPhysical & internalFlags) {
- IOKernelFreePhysical(kheap, (mach_vm_address_t) buffer, size);
- } else if (kInternalFlagPageAllocated & internalFlags) {
-#if defined(__x86_64__)
- uintptr_t page;
- page = iopa_free(&gIOBMDPageAllocator, (uintptr_t) buffer, size);
- if (page) {
- kmem_free(kernel_map, page, page_size);
- }
-#if IOALLOCDEBUG
- OSAddAtomicLong(-size, &debug_iomalloc_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsFreeAligned, size);
-#else /* !defined(__x86_64__) */
- /* should be unreachable */
- panic("Attempting to free IOBMD with page allocated flag");
-#endif /* defined(__x86_64__) */
- } else if (kInternalFlagGuardPages & internalFlags) {
- vm_offset_t allocation = (vm_offset_t)buffer - page_size;
- kmem_free(kernel_map, allocation, size + page_size * 2,
- (kmf_flags_t)(KMF_GUARD_FIRST | KMF_GUARD_LAST));
-#if IOALLOCDEBUG
- OSAddAtomicLong(-size, &debug_iomalloc_size);
-#endif
- IOStatisticsAlloc(kIOStatisticsFreeAligned, size);
- } else if (alignment > 1) {
- /* BEGIN IGNORE CODESTYLE */
- __typed_allocators_ignore_push
- IOFreeAligned_internal(kheap, buffer, size);
- } else {
- IOFree_internal(kheap, buffer, size);
- __typed_allocators_ignore_pop
- /* END IGNORE CODESTYLE */
- }
- }
- if (range && (kIOMemoryAsReference & flags)) {
- IOFreeType(range, IOAddressRange);
- }
+ }
+ else if (buffer)
+ {
+ if (kIOMemoryTypePhysical64 == (flags & kIOMemoryTypeMask))
+ IOFreePhysical(source, size);
+ else if (options & kIOMemoryPhysicallyContiguous)
+ IOKernelFreeContiguous((mach_vm_address_t) buffer, size);
+ else if (alignment > 1)
+ IOFreeAligned(buffer, size);
+ else
+ IOFree(buffer, size);
+ }
}
/*
@@ -803,10 +539,9 @@
*
* Get the buffer capacity
*/
-vm_size_t
-IOBufferMemoryDescriptor::getCapacity() const
-{
- return _capacity;
+vm_size_t IOBufferMemoryDescriptor::getCapacity() const
+{
+ return _capacity;
}
/*
@@ -819,16 +554,12 @@
* can reuse an existing one, even for different transfer sizes). Note
* that the specified length must not exceed the capacity of the buffer.
*/
-void
-IOBufferMemoryDescriptor::setLength(vm_size_t length)
-{
- assert(length <= _capacity);
- if (length > _capacity) {
- return;
- }
-
- _length = length;
- _ranges.v64->length = length;
+void IOBufferMemoryDescriptor::setLength(vm_size_t length)
+{
+ assert(length <= _capacity);
+
+ _length = length;
+ _ranges.v64->length = length;
}
/*
@@ -838,13 +569,9 @@
* the descriptor's transfer direction. This eliminates the need to destroy
* and create new buffers when different transfer directions are needed.
*/
-void
-IOBufferMemoryDescriptor::setDirection(IODirection direction)
-{
- _flags = (_flags & ~kIOMemoryDirectionMask) | direction;
-#ifndef __LP64__
- _direction = (IODirection) (_flags & kIOMemoryDirectionMask);
-#endif /* !__LP64__ */
+void IOBufferMemoryDescriptor::setDirection(IODirection direction)
+{
+ _direction = direction;
}
/*
@@ -857,23 +584,22 @@
bool
IOBufferMemoryDescriptor::appendBytes(const void * bytes, vm_size_t withLength)
{
- vm_size_t actualBytesToCopy = IOMin(withLength, _capacity - _length);
- IOByteCount offset;
-
- assert(_length <= _capacity);
-
- offset = _length;
- _length += actualBytesToCopy;
- _ranges.v64->length += actualBytesToCopy;
-
- if (_task == kernel_task) {
- bcopy(/* from */ bytes, (void *)(_ranges.v64->address + offset),
- actualBytesToCopy);
- } else {
- writeBytes(offset, bytes, actualBytesToCopy);
- }
-
- return true;
+ vm_size_t actualBytesToCopy = min(withLength, _capacity - _length);
+ IOByteCount offset;
+
+ assert(_length <= _capacity);
+
+ offset = _length;
+ _length += actualBytesToCopy;
+ _ranges.v64->length += actualBytesToCopy;
+
+ if (_task == kernel_task)
+ bcopy(/* from */ bytes, (void *)(_ranges.v64->address + offset),
+ actualBytesToCopy);
+ else
+ writeBytes(offset, bytes, actualBytesToCopy);
+
+ return true;
}
/*
@@ -881,18 +607,12 @@
*
* Return the virtual address of the beginning of the buffer
*/
-void *
-IOBufferMemoryDescriptor::getBytesNoCopy()
-{
- if (__improbable(_internalFlags & kInternalFlagAsIfUnmapped)) {
- return NULL;
- }
-
- if (kIOMemoryTypePhysical64 == (_flags & kIOMemoryTypeMask)) {
- return _buffer;
- } else {
- return (void *)_ranges.v64->address;
- }
+void * IOBufferMemoryDescriptor::getBytesNoCopy()
+{
+ if (kIOMemoryTypePhysical64 == (_flags & kIOMemoryTypeMask))
+ return _buffer;
+ else
+ return (void *)_ranges.v64->address;
}
@@ -904,50 +624,30 @@
void *
IOBufferMemoryDescriptor::getBytesNoCopy(vm_size_t start, vm_size_t withLength)
{
- IOVirtualAddress address;
-
- if (__improbable(_internalFlags & kInternalFlagAsIfUnmapped)) {
- return NULL;
- }
-
- if ((start + withLength) < start) {
- return NULL;
- }
-
- if (kIOMemoryTypePhysical64 == (_flags & kIOMemoryTypeMask)) {
- address = (IOVirtualAddress) _buffer;
- } else {
- address = _ranges.v64->address;
- }
-
- if (start < _length && (start + withLength) <= _length) {
- return (void *)(address + start);
- }
- return NULL;
-}
-
-#ifndef __LP64__
-void *
-IOBufferMemoryDescriptor::getVirtualSegment(IOByteCount offset,
- IOByteCount * lengthOfSegment)
-{
- void * bytes = getBytesNoCopy(offset, 0);
-
- if (bytes && lengthOfSegment) {
- *lengthOfSegment = _length - offset;
- }
-
- return bytes;
-}
-#endif /* !__LP64__ */
-
-#ifdef __LP64__
-OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 0);
-OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 1);
-#else /* !__LP64__ */
-OSMetaClassDefineReservedUsedX86(IOBufferMemoryDescriptor, 0);
-OSMetaClassDefineReservedUsedX86(IOBufferMemoryDescriptor, 1);
-#endif /* !__LP64__ */
+ IOVirtualAddress address;
+ if (kIOMemoryTypePhysical64 == (_flags & kIOMemoryTypeMask))
+ address = (IOVirtualAddress) _buffer;
+ else
+ address = _ranges.v64->address;
+
+ if (start < _length && (start + withLength) <= _length)
+ return (void *)(address + start);
+ return 0;
+}
+
+/* DEPRECATED */ void * IOBufferMemoryDescriptor::getVirtualSegment(IOByteCount offset,
+/* DEPRECATED */ IOByteCount * lengthOfSegment)
+{
+ void * bytes = getBytesNoCopy(offset, 0);
+
+ if (bytes && lengthOfSegment)
+ *lengthOfSegment = _length - offset;
+
+ return bytes;
+}
+
+OSMetaClassDefineReservedUsed(IOBufferMemoryDescriptor, 0);
+OSMetaClassDefineReservedUsed(IOBufferMemoryDescriptor, 1);
OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 2);
OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 3);
OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 4);