#include <IOKit/IOLib.h>
#include <IOKit/IOMapper.h>
#include <IOKit/IODMACommand.h>
#include <libkern/c++/OSData.h>
__BEGIN_DECLS
extern ppnum_t pmap_find_phys(pmap_t pmap, addr64_t va);
__END_DECLS
#define super IOService
OSDefineMetaClassAndAbstractStructors(IOMapper, IOService);
OSMetaClassDefineReservedUsed(IOMapper, 0);
OSMetaClassDefineReservedUsed(IOMapper, 1);
OSMetaClassDefineReservedUsed(IOMapper, 2);
OSMetaClassDefineReservedUnused(IOMapper, 3);
OSMetaClassDefineReservedUnused(IOMapper, 4);
OSMetaClassDefineReservedUnused(IOMapper, 5);
OSMetaClassDefineReservedUnused(IOMapper, 6);
OSMetaClassDefineReservedUnused(IOMapper, 7);
OSMetaClassDefineReservedUnused(IOMapper, 8);
OSMetaClassDefineReservedUnused(IOMapper, 9);
OSMetaClassDefineReservedUnused(IOMapper, 10);
OSMetaClassDefineReservedUnused(IOMapper, 11);
OSMetaClassDefineReservedUnused(IOMapper, 12);
OSMetaClassDefineReservedUnused(IOMapper, 13);
OSMetaClassDefineReservedUnused(IOMapper, 14);
OSMetaClassDefineReservedUnused(IOMapper, 15);
IOMapper * IOMapper::gSystem = (IOMapper *) IOMapper::kUnknown;
class IOMapperLock {
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); };
};
static IOMapperLock sMapperLock;
bool IOMapper::start(IOService *provider)
{
OSObject * obj;
if (!super::start(provider))
return false;
if (!initHardware(provider))
return false;
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;
}
bool IOMapper::allocTable(IOByteCount size)
{
assert(!fTable);
fTableSize = size;
fTableHandle = NewARTTable(size, &fTable, &fTablePhys);
return fTableHandle != 0;
}
void IOMapper::free()
{
if (fTableHandle) {
FreeARTTable(fTableHandle, fTableSize);
fTableHandle = 0;
}
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)
sMapperLock.sleep(&IOMapper::gSystem);
sMapperLock.unlock();
}
IOMapper * IOMapper::copyMapperForDevice(IOService * device)
{
OSObject * obj;
IOMapper * mapper;
OSDictionary * matching;
obj = device->copyProperty("iommu-parent");
if (!obj)
return (NULL);
if ((mapper = OSDynamicCast(IOMapper, obj)))
return (mapper);
matching = IOService::propertyMatching(gIOMapperIDKey, obj);
if (matching)
{
mapper = OSDynamicCast(IOMapper, IOService::waitForMatchingService(matching));
matching->release();
}
if (mapper)
device->setProperty("iommu-parent", mapper);
else
obj->release();
return (mapper);
}
ppnum_t IOMapper::iovmAllocDMACommand(IODMACommand * command, IOItemCount pageCount)
{
return (0);
}
void IOMapper::iovmFreeDMACommand(IODMACommand * command,
ppnum_t addr, IOItemCount pageCount)
{
}
void IOMapper::iovmInsert(ppnum_t addr, IOItemCount offset,
ppnum_t *pageList, IOItemCount pageCount)
{
while (pageCount--)
iovmInsert(addr, offset++, *pageList++);
}
void IOMapper::iovmInsert(ppnum_t addr, IOItemCount offset,
upl_page_info_t *pageList, IOItemCount pageCount)
{
for (IOItemCount i = 0; i < pageCount; i++)
iovmInsert(addr, offset + i, pageList[i].phys_addr);
}
OSData * IOMapper::
NewARTTable(IOByteCount size, void ** virtAddrP, ppnum_t *physAddrP)
{
if (!virtAddrP || !physAddrP)
return 0;
kern_return_t kr;
vm_address_t address;
size = round_page(size);
kr = kmem_alloc_contig(kernel_map, &address, size, PAGE_MASK, 0 , 0 , false);
if (kr)
return 0;
ppnum_t pagenum = pmap_find_phys(kernel_pmap, (addr64_t) address);
if (pagenum)
*physAddrP = pagenum;
else {
FreeARTTable((OSData *) address, size);
address = 0;
}
*virtAddrP = (void *) address;
return (OSData *) address;
}
void IOMapper::FreeARTTable(OSData *artHandle, IOByteCount size)
{
vm_address_t address = (vm_address_t) artHandle;
size = round_page(size);
kmem_free(kernel_map, address, size); }
bool IOMapper::getBypassMask(addr64_t *maskP) const
{
return false;
}
__BEGIN_DECLS
ppnum_t IOMapperIOVMAlloc(unsigned pages)
{
IOMapper::checkForSystemMapper();
if (IOMapper::gSystem)
return IOMapper::gSystem->iovmAlloc((IOItemCount) pages);
else
return 0;
}
void IOMapperIOVMFree(ppnum_t addr, unsigned pages)
{
if (IOMapper::gSystem)
IOMapper::gSystem->iovmFree(addr, (IOItemCount) pages);
}
ppnum_t IOMapperInsertPage(ppnum_t addr, unsigned offset, ppnum_t page)
{
if (IOMapper::gSystem) {
IOMapper::gSystem->iovmInsert(addr, (IOItemCount) offset, page);
return addr + offset;
}
else
return page;
}
void IOMapperInsertPPNPages(ppnum_t addr, unsigned offset,
ppnum_t *pageList, unsigned pageCount)
{
if (!IOMapper::gSystem)
panic("IOMapperInsertPPNPages no system mapper");
else
assert(!((vm_address_t) IOMapper::gSystem & 3));
IOMapper::gSystem->
iovmInsert(addr, (IOItemCount) offset, pageList, pageCount);
}
void IOMapperInsertUPLPages(ppnum_t addr, unsigned offset,
upl_page_info_t *pageList, unsigned pageCount)
{
if (!IOMapper::gSystem)
panic("IOMapperInsertUPLPages no system mapper");
else
assert(!((vm_address_t) IOMapper::gSystem & 3));
IOMapper::gSystem->iovmInsert(addr,
(IOItemCount) offset,
pageList,
(IOItemCount) pageCount);
}
#include <machine/machine_routines.h>
UInt8 IOMappedRead8(IOPhysicalAddress address)
{
IOMapper::checkForSystemMapper();
if (IOMapper::gSystem) {
addr64_t addr = IOMapper::gSystem->mapAddr(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->mapAddr(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->mapAddr(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->mapAddr(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->mapAddr(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->mapAddr(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->mapAddr(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->mapAddr(address);
ml_phys_write_double_64(addr, value);
}
else
ml_phys_write_double((vm_offset_t) address, value);
}
__END_DECLS