Loading...
--- xnu/xnu-517.7.21/iokit/Kernel/IOBufferMemoryDescriptor.cpp
+++ xnu/xnu-124.7/iokit/Kernel/IOBufferMemoryDescriptor.cpp
@@ -27,10 +27,9 @@
__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,
@@ -83,12 +82,8 @@
bool IOBufferMemoryDescriptor::initWithOptions(
IOOptionBits options,
vm_size_t capacity,
- vm_offset_t alignment,
- task_t inTask)
-{
- vm_map_t map = 0;
- IOOptionBits iomdOptions = kIOMemoryAsReference | kIOMemoryTypeVirtual;
-
+ vm_offset_t alignment)
+{
if (!capacity)
return false;
@@ -98,115 +93,46 @@
_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;
- if (inTask == kernel_task)
- {
- /* Allocate some kernel address space. */
- _buffer = IOMallocPageable(capacity, alignment);
- if (_buffer)
- map = IOPageableMapForAddress((vm_address_t) _buffer);
- }
- else
- {
- kern_return_t kr;
-
- if( !reserved) {
- reserved = IONew( ExpansionData, 1 );
- if( !reserved)
- return( false );
- }
- map = get_task_map(inTask);
- vm_map_reference(map);
- reserved->map = map;
- kr = vm_allocate( map, (vm_address_t *) &_buffer, round_page_32(capacity),
- VM_FLAGS_ANYWHERE | VM_MAKE_TAG(VM_MEMORY_IOKIT) );
- if( KERN_SUCCESS != kr)
- return( false );
-
- // we have to make sure that these pages don't get copied on fork.
- kr = vm_inherit( map, (vm_address_t) _buffer, round_page_32(capacity), VM_INHERIT_NONE);
- if( KERN_SUCCESS != kr)
- return( false );
- }
- }
- 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);
- }
+ /* 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;
+ 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) {
+ _flags |= kIOMemoryRequiresWire;
+
kern_return_t kr;
ipc_port_t sharedMem = (ipc_port_t) _memEntry;
- vm_size_t size = round_page_32(_ranges.v[0].length);
+ vm_size_t size = _ranges.v[0].length;
// must create the entry before any pages are allocated
if( 0 == sharedMem) {
-
- // set memory entry cache
- vm_prot_t memEntryCacheMode = VM_PROT_READ | VM_PROT_WRITE;
- 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( map,
+ kr = mach_make_memory_entry( IOPageableMapForAddress( _ranges.v[0].address ),
&size, _ranges.v[0].address,
- memEntryCacheMode, &sharedMem,
+ VM_PROT_READ | VM_PROT_WRITE, &sharedMem,
NULL );
-
- if( (KERN_SUCCESS == kr) && (size != round_page_32(_ranges.v[0].length))) {
+ if( (KERN_SUCCESS == kr) && (size != _ranges.v[0].length)) {
ipc_port_release_send( sharedMem );
kr = kIOReturnVMError;
}
@@ -214,44 +140,37 @@
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;
}
@@ -330,34 +249,21 @@
*/
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 map = 0;
vm_offset_t alignment = _alignment;
- if (reserved)
- {
- map = reserved->map;
- IODelete( reserved, ExpansionData, 1 );
- }
+ if (_physAddrs)
+ IODelete(_physAddrs, IOPhysicalAddress, _physSegCount);
/* super::free may unwire - deallocate buffer afterwards */
super::free();
- if (buffer)
- {
+ if (buffer) {
if (options & kIOMemoryPageable)
- {
- if (map)
- vm_deallocate(map, (vm_address_t) buffer, round_page_32(size));
- else
- IOFreePageable(buffer, size);
- }
- else
- {
+ IOFreePageable(buffer, size);
+ else {
if (options & kIOMemoryPhysicallyContiguous)
IOFreeContiguous(buffer, size);
else if (alignment > 1)
@@ -366,8 +272,6 @@
IOFree(buffer, size);
}
}
- if (map)
- vm_map_deallocate(map);
}
/*
@@ -454,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);