Loading...
--- xnu/xnu-2422.100.13/iokit/Kernel/IOBufferMemoryDescriptor.cpp
+++ xnu/xnu-1504.7.4/iokit/Kernel/IOBufferMemoryDescriptor.cpp
@@ -35,24 +35,9 @@
#include <IOKit/IOMapper.h>
#include <IOKit/IOBufferMemoryDescriptor.h>
#include <libkern/OSDebug.h>
-#include <mach/mach_vm.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);
@@ -64,33 +49,16 @@
enum
{
- kInternalFlagPhysical = 0x00000001,
- kInternalFlagPageSized = 0x00000002,
- kInternalFlagPageAllocated = 0x00000004
+ kInternalFlagRealloc = 0x00000001,
};
+
+volatile ppnum_t gIOHighestAllocatedPage;
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
#define super IOGeneralMemoryDescriptor
OSDefineMetaClassAndStructors(IOBufferMemoryDescriptor,
IOGeneralMemoryDescriptor);
-
-/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
-
-static uintptr_t IOBMDPageProc(iopa_t * a)
-{
- kern_return_t kr;
- vm_address_t vmaddr = 0;
- int options = 0; // KMA_LOMEM;
-
- kr = kernel_memory_allocate(kernel_map, &vmaddr,
- page_size, 0, options);
-
- if (KERN_SUCCESS != kr) vmaddr = 0;
- else bzero((void *) vmaddr, page_size);
-
- return ((uintptr_t) vmaddr);
-}
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
@@ -113,14 +81,13 @@
mach_vm_address_t alignment,
mach_vm_address_t physicalMask)
{
- kern_return_t kr;
- task_t mapTask = NULL;
- vm_map_t vmmap = NULL;
- mach_vm_address_t highestMask = 0;
- IOOptionBits iomdOptions = kIOMemoryTypeVirtual64 | kIOMemoryAsReference;
- IODMAMapSpecification mapSpec;
- bool mapped = false;
- bool needZero;
+ kern_return_t kr;
+ task_t mapTask = NULL;
+ vm_map_t vmmap = NULL;
+ addr64_t lastIOAddr;
+ mach_vm_address_t highestMask = 0;
+ bool usePhys;
+ IOOptionBits iomdOptions = kIOMemoryTypeVirtual64 | kIOMemoryAsReference;
if (!capacity)
return false;
@@ -136,29 +103,18 @@
return (false);
_ranges.v64->address = 0;
_ranges.v64->length = 0;
- // 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 = (0 != IOMapper::gSystem);
- }
- needZero = mapped;
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))
+ }
+
+ if ((options & (kIOMemorySharingTypeMask | kIOMapCacheMask)) && (alignment < page_size))
alignment = page_size;
if (alignment >= page_size)
@@ -172,17 +128,8 @@
if ((inTask != kernel_task) && !(options & kIOMemoryPageable))
return false;
- bzero(&mapSpec, sizeof(mapSpec));
- mapSpec.alignment = _alignment;
- mapSpec.numAddressBits = 64;
- if (highestMask && mapped)
- {
- if (highestMask <= 0xFFFFFFFF)
- mapSpec.numAddressBits = (32 - __builtin_clz((unsigned int) highestMask));
- else
- mapSpec.numAddressBits = (64 - __builtin_clz((unsigned int) (highestMask >> 32)));
- highestMask = 0;
- }
+ if ((options & kIOMemoryPhysicallyContiguous) && !physicalMask)
+ physicalMask = 0xFFFFFFFF;
// set flags for entry + object create
vm_prot_t memEntryCacheMode = VM_PROT_READ | VM_PROT_WRITE;
@@ -204,10 +151,6 @@
case kIOMapCopybackCache:
SET_MAP_MEM(MAP_MEM_COPYBACK, memEntryCacheMode);
- break;
-
- case kIOMapCopybackInnerCache:
- SET_MAP_MEM(MAP_MEM_INNERWBACK, memEntryCacheMode);
break;
case kIOMapDefaultCache:
@@ -231,67 +174,63 @@
else
{
memEntryCacheMode |= MAP_MEM_NAMED_REUSE;
- 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. */
-
- 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;
-#endif
- }
-
- if (contig || highestMask || (alignment > page_size))
- {
- _internalFlags |= kInternalFlagPhysical;
- if (highestMask)
- {
- _internalFlags |= kInternalFlagPageSized;
- capacity = round_page(capacity);
- }
- _buffer = (void *) IOKernelAllocateWithPhysicalRestrict(
- capacity, highestMask, alignment, contig);
- }
- else if (needZero
- && ((capacity + alignment) <= (page_size - kIOPageAllocChunkBytes)))
- {
- _internalFlags |= kInternalFlagPageAllocated;
- needZero = false;
- _buffer = (void *) iopa_alloc(&gIOBMDPageAllocator, &IOBMDPageProc, capacity, alignment);
- if (_buffer)
- {
- IOStatisticsAlloc(kIOStatisticsMallocAligned, capacity);
-#if IOALLOCDEBUG
- debug_iomalloc_size += capacity;
-#endif
- }
- }
- else if (alignment > 1)
- {
- _buffer = IOMallocAligned(capacity, alignment);
- }
+
+ if (IOMapper::gSystem)
+ // assuming mapped space is 2G
+ lastIOAddr = (1UL << 31) - PAGE_SIZE;
else
- {
- _buffer = IOMalloc(capacity);
- }
- if (!_buffer)
- {
- return false;
- }
- if (needZero) bzero(_buffer, capacity);
- }
-
- if( (options & (kIOMemoryPageable | kIOMapCacheMask))) {
+ lastIOAddr = ptoa_64(gIOHighestAllocatedPage);
+
+ usePhys = (highestMask && (lastIOAddr != (lastIOAddr & highestMask))
+ && (alignment <= page_size));
+
+ if (!usePhys && (options & kIOMemoryPhysicallyContiguous))
+ {
+ _buffer = (void *) IOKernelAllocateContiguous(capacity, highestMask, alignment);
+ usePhys = (NULL == _buffer);
+ }
+ if (usePhys)
+ {
+ mach_vm_address_t address;
+ iomdOptions &= ~kIOMemoryTypeVirtual64;
+ iomdOptions |= kIOMemoryTypePhysical64;
+
+ address = IOMallocPhysical(capacity, highestMask);
+ _buffer = (void *) address;
+ if (!_buffer)
+ return false;
+
+ 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)
+ {
+ // attempted allocate already
+ }
+ else if (alignment > 1)
+ {
+ _buffer = IOMallocAligned(capacity, alignment);
+ }
+ else
+ {
+ _buffer = IOMalloc(capacity);
+ }
+ if (!_buffer)
+ return false;
+ }
+ }
+
+ if( (kIOMemoryTypePhysical64 != (kIOMemoryTypeMask & iomdOptions))
+ && (options & (kIOMemoryPageable | kIOMapCacheMask))) {
ipc_port_t sharedMem;
vm_size_t size = round_page(capacity);
@@ -325,10 +264,9 @@
while (startAddr < endAddr)
{
- UInt8 dummyVar = *startAddr;
- (void) dummyVar;
+ *startAddr;
startAddr += page_size;
- }
+ }
}
}
@@ -339,10 +277,41 @@
inTask, iomdOptions, /* System mapper */ 0))
return false;
- // give any system mapper the allocation params
- if (kIOReturnSuccess != dmaCommandOperation(kIOMDAddDMAMapSpec,
- &mapSpec, sizeof(mapSpec)))
- return false;
+ if (highestMask && !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 & highestMask))
+ {
+ if (kIOMemoryTypePhysical64 != (_flags & kIOMemoryTypeMask))
+ {
+ // flag a retry
+ _internalFlags |= kInternalFlagRealloc;
+ }
+ return false;
+ }
+ }
if (mapTask)
{
@@ -381,8 +350,18 @@
IOBufferMemoryDescriptor *me = new IOBufferMemoryDescriptor;
if (me && !me->initWithPhysicalMask(inTask, options, capacity, alignment, 0)) {
+ bool retry = (0 != (kInternalFlagRealloc & me->_internalFlags));
me->release();
me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithPhysicalMask(inTask, options, capacity, alignment, 0))
+ {
+ me->release();
+ me = 0;
+ }
+ }
}
return me;
}
@@ -397,8 +376,18 @@
if (me && !me->initWithPhysicalMask(inTask, options, capacity, 1, physicalMask))
{
+ bool retry = (0 != (kInternalFlagRealloc & me->_internalFlags));
me->release();
me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithPhysicalMask(inTask, options, capacity, 1, physicalMask))
+ {
+ me->release();
+ me = 0;
+ }
+ }
}
return me;
}
@@ -421,8 +410,18 @@
IOBufferMemoryDescriptor *me = new IOBufferMemoryDescriptor;
if (me && !me->initWithPhysicalMask(kernel_task, options, capacity, alignment, 0)) {
+ bool retry = (0 != (kInternalFlagRealloc & me->_internalFlags));
me->release();
me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithPhysicalMask(kernel_task, options, capacity, alignment, 0))
+ {
+ me->release();
+ me = 0;
+ }
+ }
}
return me;
}
@@ -491,8 +490,22 @@
| (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
inLength, inLength, 0 ))
{
+ bool retry = (0 != (kInternalFlagRealloc & me->_internalFlags));
me->release();
me = 0;
+ if (retry)
+ {
+ me = new IOBufferMemoryDescriptor;
+ if (me && !me->initWithPhysicalMask(
+ kernel_task, inDirection | kIOMemoryUnshared
+ | (inContiguous ? kIOMemoryPhysicallyContiguous : 0),
+ inLength, inLength, 0 ))
+ {
+ me->release();
+ me = 0;
+ }
+ }
+
}
if (me)
@@ -518,13 +531,13 @@
{
// Cache all of the relevant information on the stack for use
// after we call super::free()!
- IOOptionBits flags = _flags;
- IOOptionBits internalFlags = _internalFlags;
+ IOOptionBits flags = _flags;
IOOptionBits options = _options;
vm_size_t size = _capacity;
void * buffer = _buffer;
IOMemoryMap * map = 0;
IOAddressRange * range = _ranges.v64;
+ mach_vm_address_t source = range ? range->address : 0;
vm_offset_t alignment = _alignment;
if (alignment >= page_size)
@@ -549,33 +562,14 @@
}
else if (buffer)
{
- if (kInternalFlagPageSized & internalFlags) size = round_page(size);
-
- if (kInternalFlagPhysical & internalFlags)
- {
- IOKernelFreePhysical((mach_vm_address_t) buffer, size);
- }
- else if (kInternalFlagPageAllocated & internalFlags)
- {
- uintptr_t page;
- page = iopa_free(&gIOBMDPageAllocator, (uintptr_t) buffer, size);
- if (page)
- {
- kmem_free(kernel_map, page, page_size);
- }
-#if IOALLOCDEBUG
- debug_iomalloc_size -= size;
-#endif
- IOStatisticsAlloc(kIOStatisticsFreeAligned, size);
- }
+ 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);
- }
}
if (range && (kIOMemoryAsReference & flags))
IODelete(range, IOAddressRange, 1);