Loading...
iokit/Kernel/IOMapper.cpp xnu-12377.121.6 xnu-4570.41.2
--- xnu/xnu-12377.121.6/iokit/Kernel/IOMapper.cpp
+++ xnu/xnu-4570.41.2/iokit/Kernel/IOMapper.cpp
@@ -2,7 +2,7 @@
  * Copyright (c) 1998-2016 Apple 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,19 +22,15 @@
  * 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
-
 #include <IOKit/IOLib.h>
 #include <IOKit/IOMapper.h>
 #include <IOKit/IODMACommand.h>
 #include <libkern/c++/OSData.h>
 #include <libkern/OSDebug.h>
 #include <mach_debug/zone_info.h>
-#include <vm/vm_iokit.h>
 #include "IOKitKernelInternal.h"
 
 __BEGIN_DECLS
@@ -64,219 +60,176 @@
 IOMapper * IOMapper::gSystem = (IOMapper *) IOMapper::kUnknown;
 
 class IOMapperLock {
-	IOLock *fWaitLock;
+    IOLock *fWaitLock;
 public:
-	IOMapperLock()
-	{
-		fWaitLock = IOLockAlloc();
-	}
-	~IOMapperLock()
-	{
-		IOLockFree(fWaitLock);
-	}
-
-	void
-	lock()
-	{
-		IOLockLock(fWaitLock);
-	}
-	void
-	unlock()
-	{
-		IOLockUnlock(fWaitLock);
-	}
-	void
-	sleep(void *event)
-	{
-		IOLockSleep(fWaitLock, event, THREAD_UNINT);
-	}
-	void
-	wakeup(void *event)
-	{
-		IOLockWakeup(fWaitLock, event, false);
-	}
+    IOMapperLock() { fWaitLock = IOLockAlloc(); }
+    ~IOMapperLock() { IOLockFree(fWaitLock); }
+
+    void lock()   { IOLockLock(fWaitLock); }
+    void unlock() { IOLockUnlock(fWaitLock); }
+    void sleep(void *event)  { IOLockSleep(fWaitLock, event, THREAD_UNINT); }
+    void wakeup(void *event) { IOLockWakeup(fWaitLock, event, false); }
 };
 
 static IOMapperLock sMapperLock;
 
-bool
-IOMapper::start(IOService *provider)
-{
-	OSSharedPtr<OSObject> obj;
-	if (!super::start(provider)) {
-		return false;
-	}
-
-	if (!initHardware(provider)) {
-		return false;
-	}
-
-	uint64_t pageSize = getPageSize();
-	assert(pageSize <= UINT_MAX);
-	fPageSize = (uint32_t) pageSize;
-
-	if (fIsSystem) {
-		sMapperLock.lock();
-		IOMapper::gSystem = this;
-		sMapperLock.wakeup(&IOMapper::gSystem);
-		sMapperLock.unlock();
-	}
-
-	if (provider) {
-		obj = provider->copyProperty("iommu-id");
-		if (!obj) {
-			obj = provider->copyProperty("AAPL,phandle");
-		}
-		if (obj) {
-			setProperty(gIOMapperIDKey, obj.get());
-		}
-	}
-	return true;
-}
-
-void
-IOMapper::free()
-{
-	super::free();
-}
-
-void
-IOMapper::setMapperRequired(bool hasMapper)
-{
-	if (hasMapper) {
-		IOMapper::gSystem = (IOMapper *) kHasMapper;
-	} else {
-		sMapperLock.lock();
-		IOMapper::gSystem = (IOMapper *) kNoMapper;
-		sMapperLock.unlock();
-		sMapperLock.wakeup(&IOMapper::gSystem);
-	}
-}
-
-void
-IOMapper::waitForSystemMapper()
-{
-	sMapperLock.lock();
-	while ((uintptr_t) IOMapper::gSystem & kWaitMask) {
+bool IOMapper::start(IOService *provider)
+{
+    OSObject * obj;
+    if (!super::start(provider))
+        return false;
+
+    if (!initHardware(provider))
+        return false;
+
+    fPageSize = getPageSize();
+
+    if (fIsSystem) { 
+        sMapperLock.lock();
+        IOMapper::gSystem = this;
+        sMapperLock.wakeup(&IOMapper::gSystem);
+        sMapperLock.unlock();
+    }
+
+    if (provider)
+    {
+    	obj = provider->getProperty("iommu-id");
+	if (!obj)
+	    obj = provider->getProperty("AAPL,phandle");
+	if (obj)
+	    setProperty(gIOMapperIDKey, obj);
+    }
+    return true;
+}
+
+void IOMapper::free()
+{
+    super::free();
+}
+
+void IOMapper::setMapperRequired(bool hasMapper)
+{
+    if (hasMapper)
+        IOMapper::gSystem = (IOMapper *) kHasMapper;
+    else {
+        sMapperLock.lock();
+        IOMapper::gSystem = (IOMapper *) kNoMapper;
+        sMapperLock.unlock();
+        sMapperLock.wakeup(&IOMapper::gSystem);
+    }
+}
+
+void IOMapper::waitForSystemMapper()
+{
+    sMapperLock.lock();
+    while ((uintptr_t) IOMapper::gSystem & kWaitMask)
+    {
 		OSReportWithBacktrace("waitForSystemMapper");
-		sMapperLock.sleep(&IOMapper::gSystem);
-	}
-	sMapperLock.unlock();
-}
-
-OSSharedPtr<IOMapper>
-IOMapper::copyMapperForDevice(IOService * device)
-{
-	return copyMapperForDeviceWithIndex(device, 0);
-}
-
-OSSharedPtr<IOMapper>
-IOMapper::copyMapperForDeviceWithIndex(IOService * device, unsigned int index)
-{
-	OSSharedPtr<OSData> data;
-	OSSharedPtr<OSObject> obj;
-	OSSharedPtr<IOMapper> mapper;
-	OSSharedPtr<OSDictionary> matching;
-
-	obj = device->copyProperty("iommu-parent");
-	if (!obj) {
-		return NULL;
-	}
-
-	if ((mapper = OSDynamicPtrCast<IOMapper>(obj))) {
-		goto found;
-	}
-
-	if ((data = OSDynamicPtrCast<OSData>(obj))) {
-		if (index >= data->getLength() / sizeof(UInt32)) {
-			goto found;
-		}
-
-		data = OSData::withValueNoCopy(*((UInt32 *)data->getBytesNoCopy() + index));
-		if (!data) {
-			goto found;
-		}
-
-		matching = IOService::propertyMatching(gIOMapperIDKey, data.get());
-	} else {
-		matching = IOService::propertyMatching(gIOMapperIDKey, obj.get());
-	}
-
-	if (matching) {
-		mapper = OSDynamicPtrCast<IOMapper>(IOService::waitForMatchingService(matching.get()));
-	}
-
+        sMapperLock.sleep(&IOMapper::gSystem);
+    }
+    sMapperLock.unlock();
+}
+
+IOMapper * IOMapper::copyMapperForDevice(IOService * device)
+{
+    return copyMapperForDeviceWithIndex(device, 0);
+}
+
+IOMapper * IOMapper::copyMapperForDeviceWithIndex(IOService * device, unsigned int index)
+{
+    OSData *data;
+    OSObject * obj;
+    IOMapper * mapper = NULL;
+    OSDictionary * matching;
+    
+    obj = device->copyProperty("iommu-parent");
+    if (!obj) return (NULL);
+
+    if ((mapper = OSDynamicCast(IOMapper, obj))) goto found;
+
+    if ((data = OSDynamicCast(OSData, obj)))
+    {
+        if (index >= data->getLength() / sizeof(UInt32)) goto done;
+        
+        data = OSData::withBytesNoCopy((UInt32 *)data->getBytesNoCopy() + index, sizeof(UInt32));
+        if (!data) goto done;
+
+        matching = IOService::propertyMatching(gIOMapperIDKey, data);
+        data->release();
+    }
+    else
+        matching = IOService::propertyMatching(gIOMapperIDKey, obj);
+
+    if (matching)
+    {
+        mapper = OSDynamicCast(IOMapper, IOService::waitForMatchingService(matching));
+        matching->release();
+    }
+
+done:
+    if (obj) obj->release();
 found:
-	if (mapper) {
-		if (!mapper->fAllocName) {
-			char name[MACH_ZONE_NAME_MAX_LEN];
-			char kmodname[KMOD_MAX_NAME];
-			vm_tag_t tag;
-			uint32_t kmodid;
-
-			tag = IOMemoryTag(kernel_map);
-			if (!(kmodid = vm_tag_get_kext(tag, &kmodname[0], KMOD_MAX_NAME))) {
-				snprintf(kmodname, sizeof(kmodname), "%d", tag);
-			}
-			snprintf(name, sizeof(name), "%s.DMA.%s", kmodname, device->getName());
-			mapper->fAllocName = kern_allocation_name_allocate(name, 16);
-		}
-	}
-
-	return mapper;
+    if (mapper)
+    {
+        if (!mapper->fAllocName)
+        {
+            char name[MACH_ZONE_NAME_MAX_LEN];
+            char kmodname[KMOD_MAX_NAME];
+            vm_tag_t tag;
+            uint32_t kmodid;
+
+            tag = IOMemoryTag(kernel_map);
+            if (!(kmodid = vm_tag_get_kext(tag, &kmodname[0], KMOD_MAX_NAME)))
+            {
+                snprintf(kmodname, sizeof(kmodname), "%d", tag);
+            }
+            snprintf(name, sizeof(name), "%s.DMA.%s", kmodname, device->getName());
+            mapper->fAllocName = kern_allocation_name_allocate(name, 16);
+        }
+    }
+
+    return (mapper);
 }
 
 __BEGIN_DECLS
 
 // These are C accessors to the system mapper for non-IOKit clients
-ppnum_t
-IOMapperIOVMAlloc(unsigned pages)
-{
-	IOReturn ret;
-	uint64_t dmaAddress, dmaLength;
-
-	IOMapper::checkForSystemMapper();
-
-	ret = kIOReturnUnsupported;
-	if (IOMapper::gSystem) {
-		ret = IOMapper::gSystem->iovmMapMemory(
-			NULL, 0, ptoa_64(pages),
-			(kIODMAMapReadAccess | kIODMAMapWriteAccess),
-			NULL, NULL, NULL,
-			&dmaAddress, &dmaLength);
-	}
-
-	if (kIOReturnSuccess == ret) {
-		uint64_t dmaAddressPage64;
-		dmaAddressPage64 = atop_64(dmaAddress);
-		if (dmaAddressPage64 > UINT_MAX) {
-			return 0;
-		}
-		return (ppnum_t) atop_64(dmaAddress);
-	}
-	return 0;
-}
-
-void
-IOMapperIOVMFree(ppnum_t addr, unsigned pages)
-{
-	if (IOMapper::gSystem) {
-		IOMapper::gSystem->iovmUnmapMemory(NULL, NULL, ptoa_64(addr), ptoa_64(pages));
-	}
-}
-
-ppnum_t
-IOMapperInsertPage(ppnum_t addr, unsigned offset, ppnum_t page)
-{
-	if (!IOMapper::gSystem) {
-		return page;
-	}
-	if (!addr) {
-		panic("!addr");
-	}
-	IOMapper::gSystem->iovmInsert((kIODMAMapReadAccess | kIODMAMapWriteAccess),
-	    ptoa_64(addr), ptoa_64(offset), ptoa_64(page), ptoa_64(1));
-	return addr + offset;
+ppnum_t IOMapperIOVMAlloc(unsigned pages)
+{
+    IOReturn ret;
+    uint64_t dmaAddress, dmaLength;
+
+    IOMapper::checkForSystemMapper();
+
+    ret = kIOReturnUnsupported;
+    if (IOMapper::gSystem)
+    {
+        ret = IOMapper::gSystem->iovmMapMemory(
+        		NULL, 0, ptoa_64(pages), 
+        		(kIODMAMapReadAccess | kIODMAMapWriteAccess),
+        		NULL, NULL, NULL,
+        		&dmaAddress, &dmaLength);
+    }
+
+    if (kIOReturnSuccess == ret) return (atop_64(dmaAddress));
+    return (0);
+}
+
+void IOMapperIOVMFree(ppnum_t addr, unsigned pages)
+{
+    if (IOMapper::gSystem)
+    {
+        IOMapper::gSystem->iovmUnmapMemory(NULL, NULL, ptoa_64(addr), ptoa_64(pages));
+    }
+}
+
+ppnum_t IOMapperInsertPage(ppnum_t addr, unsigned offset, ppnum_t page)
+{
+    if (!IOMapper::gSystem) return (page);
+    if (!addr) panic("!addr");
+    IOMapper::gSystem->iovmInsert((kIODMAMapReadAccess | kIODMAMapWriteAccess),
+				  ptoa_64(addr), ptoa_64(offset), ptoa_64(page), ptoa_64(1));
+    return (addr + offset);
 }
 
 /////////////////////////////////////////////////////////////////////////////
@@ -289,108 +242,100 @@
 
 #include <machine/machine_routines.h>
 
-UInt8
-IOMappedRead8(IOPhysicalAddress address)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		return (UInt8) ml_phys_read_byte_64(addr);
-	} else {
-		return (UInt8) ml_phys_read_byte((vm_offset_t) address);
-	}
-}
-
-UInt16
-IOMappedRead16(IOPhysicalAddress address)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		return (UInt16) ml_phys_read_half_64(addr);
-	} else {
-		return (UInt16) ml_phys_read_half((vm_offset_t) address);
-	}
-}
-
-UInt32
-IOMappedRead32(IOPhysicalAddress address)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		return (UInt32) ml_phys_read_word_64(addr);
-	} else {
-		return (UInt32) ml_phys_read_word((vm_offset_t) address);
-	}
-}
-
-UInt64
-IOMappedRead64(IOPhysicalAddress address)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		return (UInt64) ml_phys_read_double_64(addr);
-	} else {
-		return (UInt64) ml_phys_read_double((vm_offset_t) address);
-	}
-}
-
-void
-IOMappedWrite8(IOPhysicalAddress address, UInt8 value)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		ml_phys_write_byte_64(addr, value);
-	} else {
-		ml_phys_write_byte((vm_offset_t) address, value);
-	}
-}
-
-void
-IOMappedWrite16(IOPhysicalAddress address, UInt16 value)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		ml_phys_write_half_64(addr, value);
-	} else {
-		ml_phys_write_half((vm_offset_t) address, value);
-	}
-}
-
-void
-IOMappedWrite32(IOPhysicalAddress address, UInt32 value)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		ml_phys_write_word_64(addr, value);
-	} else {
-		ml_phys_write_word((vm_offset_t) address, value);
-	}
-}
-
-void
-IOMappedWrite64(IOPhysicalAddress address, UInt64 value)
-{
-	IOMapper::checkForSystemMapper();
-
-	if (IOMapper::gSystem) {
-		addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
-		ml_phys_write_double_64(addr, value);
-	} else {
-		ml_phys_write_double((vm_offset_t) address, value);
-	}
+UInt8 IOMappedRead8(IOPhysicalAddress address)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        return (UInt8) ml_phys_read_byte_64(addr);
+    }
+    else
+        return (UInt8) ml_phys_read_byte((vm_offset_t) address);
+}
+
+UInt16 IOMappedRead16(IOPhysicalAddress address)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        return (UInt16) ml_phys_read_half_64(addr);
+    }
+    else
+        return (UInt16) ml_phys_read_half((vm_offset_t) address);
+}
+
+UInt32 IOMappedRead32(IOPhysicalAddress address)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+	return (UInt32) ml_phys_read_word_64(addr);
+    }
+    else
+        return (UInt32) ml_phys_read_word((vm_offset_t) address);
+}
+
+UInt64 IOMappedRead64(IOPhysicalAddress address)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        return (UInt64) ml_phys_read_double_64(addr);
+    }
+    else
+        return (UInt64) ml_phys_read_double((vm_offset_t) address);
+}
+
+void IOMappedWrite8(IOPhysicalAddress address, UInt8 value)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        ml_phys_write_byte_64(addr, value);
+    }
+    else
+        ml_phys_write_byte((vm_offset_t) address, value);
+}
+
+void IOMappedWrite16(IOPhysicalAddress address, UInt16 value)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        ml_phys_write_half_64(addr, value);
+    }
+    else
+        ml_phys_write_half((vm_offset_t) address, value);
+}
+
+void IOMappedWrite32(IOPhysicalAddress address, UInt32 value)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        ml_phys_write_word_64(addr, value);
+    }
+    else
+        ml_phys_write_word((vm_offset_t) address, value);
+}
+
+void IOMappedWrite64(IOPhysicalAddress address, UInt64 value)
+{
+    IOMapper::checkForSystemMapper();
+
+    if (IOMapper::gSystem) {
+        addr64_t addr = IOMapper::gSystem->mapToPhysicalAddress(address);
+        ml_phys_write_double_64(addr, value);
+    }
+    else
+        ml_phys_write_double((vm_offset_t) address, value);
 }
 
 __END_DECLS