Loading...
--- xnu/xnu-792.1.5/iokit/Kernel/IOBufferMemoryDescriptor.cpp
+++ xnu/xnu-201/iokit/Kernel/IOBufferMemoryDescriptor.cpp
@@ -25,14 +25,11 @@
#include <IOKit/IOLib.h>
#include <IOKit/IOBufferMemoryDescriptor.h>
-#include "IOKitKernelInternal.h"
-
__BEGIN_DECLS
void ipc_port_release_send(ipc_port_t port);
-#include <vm/pmap.h>
-
-vm_map_t IOPageableMapForAddress( vm_address_t address );
__END_DECLS
+
+extern "C" vm_map_t IOPageableMapForAddress( vm_address_t address );
#define super IOGeneralMemoryDescriptor
OSDefineMetaClassAndStructors(IOBufferMemoryDescriptor,
@@ -85,13 +82,8 @@
bool IOBufferMemoryDescriptor::initWithOptions(
IOOptionBits options,
vm_size_t capacity,
- vm_offset_t alignment,
- task_t inTask)
-{
- kern_return_t kr;
- vm_map_t vmmap = 0;
- IOOptionBits iomdOptions = kIOMemoryAsReference | kIOMemoryTypeVirtual;
-
+ vm_offset_t alignment)
+{
if (!capacity)
return false;
@@ -101,170 +93,84 @@
_physSegCount = 0;
_buffer = 0;
- // Grab the direction and the Auto Prepare bits from the Buffer MD options
- iomdOptions |= options & (kIOMemoryDirectionMask | kIOMemoryAutoPrepare);
-
if ((options & kIOMemorySharingTypeMask) && (alignment < page_size))
alignment = page_size;
- if ((inTask != kernel_task) && !(options & kIOMemoryPageable))
- return false;
-
_alignment = alignment;
if (options & kIOMemoryPageable)
- {
- iomdOptions |= kIOMemoryBufferPageable;
-
- ipc_port_t sharedMem;
- vm_size_t size = round_page_32(capacity);
-
- // must create the entry before any pages are allocated
-
- // set flags for entry + object create
- vm_prot_t memEntryCacheMode = VM_PROT_READ | VM_PROT_WRITE
- | MAP_MEM_NAMED_CREATE;
-
- if (options & kIOMemoryPurgeable)
- memEntryCacheMode |= MAP_MEM_PURGABLE;
-
- // 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;
- }
-
- kr = mach_make_memory_entry( vmmap,
- &size, 0,
- 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 IOALLOCDEBUG
- debug_iomallocpageable_size += size;
-#endif
- if ((NULL == inTask) && (options & kIOMemoryPageable))
- inTask = kernel_task;
- else if (inTask == kernel_task)
- {
- vmmap = kernel_map;
- }
- else
- {
-
- if( !reserved) {
- reserved = IONew( ExpansionData, 1 );
- if( !reserved)
- return( false );
- }
- vmmap = get_task_map(inTask);
- vm_map_reference(vmmap);
- reserved->map = vmmap;
- }
- }
- else
- {
- // @@@ gvdl: Need to remove this
- // Buffer should never 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 = IOMallocContiguous(capacity, alignment, 0);
- else if (alignment > 1)
- _buffer = IOMallocAligned(capacity, alignment);
- else
- _buffer = IOMalloc(capacity);
-
- if (!_buffer)
- return false;
- }
+ /* Allocate some kernel address space. */
+ _buffer = IOMallocPageable(capacity, alignment);
+ /* Allocate a wired-down buffer inside kernel space. */
+ else if (options & kIOMemoryPhysicallyContiguous)
+ _buffer = IOMallocContiguous(capacity, alignment, 0);
+ else if (alignment > 1)
+ _buffer = IOMallocAligned(capacity, alignment);
+ else
+ _buffer = IOMalloc(capacity);
+
+ if (!_buffer)
+ return false;
_singleRange.v.address = (vm_address_t) _buffer;
_singleRange.v.length = capacity;
- if (!super::initWithOptions(&_singleRange.v, 1, 0,
- inTask, iomdOptions, /* System mapper */ 0))
+ if (!super::initWithRanges(&_singleRange.v, 1,
+ (IODirection) (options & kIOMemoryDirectionMask),
+ kernel_task, true))
return false;
- if (options & kIOMemoryPageable)
- {
+ if (options & kIOMemoryPageable) {
+ _flags |= kIOMemoryRequiresWire;
+
kern_return_t kr;
-
- if (vmmap)
- {
- kr = doMap(vmmap, (IOVirtualAddress *) &_buffer, kIOMapAnywhere, 0, round_page_32(capacity));
- if (KERN_SUCCESS != kr)
- {
- _buffer = 0;
- return( false );
- }
- _singleRange.v.address = (vm_address_t) _buffer;
- }
+ ipc_port_t sharedMem = (ipc_port_t) _memEntry;
+ vm_size_t size = round_page(_ranges.v[0].length);
+
+ // must create the entry before any pages are allocated
+ if( 0 == sharedMem) {
+ kr = mach_make_memory_entry( IOPageableMapForAddress( _ranges.v[0].address ),
+ &size, _ranges.v[0].address,
+ VM_PROT_READ | VM_PROT_WRITE, &sharedMem,
+ NULL );
+ if( (KERN_SUCCESS == kr) && (size != round_page(_ranges.v[0].length))) {
+ ipc_port_release_send( sharedMem );
+ kr = kIOReturnVMError;
+ }
+ if( KERN_SUCCESS != kr)
+ sharedMem = 0;
+ _memEntry = (void *) sharedMem;
+ }
+
+ } else {
+ /* Precompute virtual-to-physical page mappings. */
+ vm_address_t inBuffer = (vm_address_t) _buffer;
+ _physSegCount = atop(trunc_page(inBuffer + capacity - 1) -
+ trunc_page(inBuffer)) + 1;
+ _physAddrs = IONew(IOPhysicalAddress, _physSegCount);
+ if (!_physAddrs)
+ return false;
+
+ inBuffer = trunc_page(inBuffer);
+ for (unsigned i = 0; i < _physSegCount; i++) {
+ _physAddrs[i] = pmap_extract(get_task_pmap(kernel_task), inBuffer);
+ assert(_physAddrs[i]); /* supposed to be wired */
+ inBuffer += page_size;
+ }
}
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)) {
- 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)
+ vm_offset_t alignment = 1)
{
IOBufferMemoryDescriptor *me = new IOBufferMemoryDescriptor;
- if (me && !me->initWithOptions(options, capacity, alignment, kernel_task)) {
+ if (me && !me->initWithOptions(options, capacity, alignment)) {
me->release();
me = 0;
}
@@ -343,48 +249,29 @@
*/
void IOBufferMemoryDescriptor::free()
{
- // Cache all of the relevant information on the stack for use
- // after we call super::free()!
IOOptionBits options = _options;
vm_size_t size = _capacity;
void * buffer = _buffer;
- vm_map_t vmmap = 0;
vm_offset_t alignment = _alignment;
- if (reserved)
- {
- vmmap = reserved->map;
- IODelete( reserved, ExpansionData, 1 );
- }
+ if (_physAddrs)
+ IODelete(_physAddrs, IOPhysicalAddress, _physSegCount);
/* super::free may unwire - deallocate buffer afterwards */
super::free();
- if (options & kIOMemoryPageable)
- {
-#if IOALLOCDEBUG
- if (!buffer || vmmap)
- debug_iomallocpageable_size -= round_page_32(size);
-#endif
- if (buffer)
- {
- if (vmmap)
- vm_deallocate(vmmap, (vm_address_t) buffer, round_page_32(size));
+ if (buffer) {
+ if (options & kIOMemoryPageable)
+ IOFreePageable(buffer, size);
+ else {
+ if (options & kIOMemoryPhysicallyContiguous)
+ IOFreeContiguous(buffer, size);
+ else if (alignment > 1)
+ IOFreeAligned(buffer, size);
else
- IOFreePageable(buffer, size);
+ IOFree(buffer, size);
}
}
- else if (buffer)
- {
- if (options & kIOMemoryPhysicallyContiguous)
- IOFreeContiguous(buffer, size);
- else if (alignment > 1)
- IOFreeAligned(buffer, size);
- else
- IOFree(buffer, size);
- }
- if (vmmap)
- vm_map_deallocate(vmmap);
}
/*
@@ -471,7 +358,56 @@
return 0;
}
-OSMetaClassDefineReservedUsed(IOBufferMemoryDescriptor, 0);
+/*
+ * getPhysicalSegment:
+ *
+ * Get the physical address of the buffer, relative to the current position.
+ * If the current position is at the end of the buffer, a zero is returned.
+ */
+IOPhysicalAddress
+IOBufferMemoryDescriptor::getPhysicalSegment(IOByteCount offset,
+ IOByteCount * lengthOfSegment)
+{
+ IOPhysicalAddress physAddr;
+
+ if( offset != _position)
+ setPosition( offset );
+
+ assert(_position <= _length);
+
+ /* Fail gracefully if the position is at (or past) the end-of-buffer. */
+ if (_position >= _length) {
+ *lengthOfSegment = 0;
+ return 0;
+ }
+
+ if (_options & kIOMemoryPageable) {
+ physAddr = super::getPhysicalSegment(offset, lengthOfSegment);
+
+ } else {
+ /* Compute the largest contiguous physical length possible. */
+ vm_address_t actualPos = _singleRange.v.address + _position;
+ vm_address_t actualPage = trunc_page(actualPos);
+ unsigned physInd = atop(actualPage-trunc_page(_singleRange.v.address));
+
+ vm_size_t physicalLength = actualPage + page_size - actualPos;
+ for (unsigned index = physInd + 1; index < _physSegCount &&
+ _physAddrs[index] == _physAddrs[index-1] + page_size; index++) {
+ physicalLength += page_size;
+ }
+
+ /* Clip contiguous physical length at the end-of-buffer. */
+ if (physicalLength > _length - _position)
+ physicalLength = _length - _position;
+
+ *lengthOfSegment = physicalLength;
+ physAddr = _physAddrs[physInd] + (actualPos - actualPage);
+ }
+
+ return physAddr;
+}
+
+OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 0);
OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 1);
OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 2);
OSMetaClassDefineReservedUnused(IOBufferMemoryDescriptor, 3);