#include <mach_kdb.h>
#include <zone_debug.h>
#include <mach_kdb.h>
#include <mach/boolean.h>
#include <mach/kern_return.h>
#include <mach/mig_errors.h>
#include <mach/port.h>
#include <mach/vm_param.h>
#include <mach/notify.h>
#include <mach/mach_types.h>
#include <machine/machparam.h>
#include <ipc/ipc_port.h>
#include <ipc/ipc_space.h>
#include <kern/clock.h>
#include <kern/spl.h>
#include <kern/counters.h>
#include <kern/queue.h>
#include <kern/zalloc.h>
#include <kern/thread.h>
#include <kern/task.h>
#include <kern/sched_prim.h>
#include <kern/misc_protos.h>
#include <vm/pmap.h>
#include <vm/vm_map.h>
#include <vm/vm_kern.h>
#include <device/device_types.h>
#include <device/device_port.h>
#include <device/device_server.h>
#include <machine/machparam.h>
#ifdef __ppc__
#include <ppc/mappings.h>
#endif
#ifdef __i386
#include <i386/pmap.h>
#endif
#include <IOKit/IOTypes.h>
#define EXTERN
#define MIGEXTERN
extern void iokit_add_reference( io_object_t obj );
extern ipc_port_t iokit_port_for_object( io_object_t obj,
ipc_kobject_type_t type );
extern kern_return_t iokit_client_died( io_object_t obj,
ipc_port_t port, ipc_kobject_type_t type, mach_port_mscount_t * mscount );
extern kern_return_t
iokit_client_memory_for_type(
io_object_t connect,
unsigned int type,
unsigned int * flags,
vm_address_t * address,
vm_size_t * size );
extern ppnum_t IOGetLastPageNumber(void);
extern ipc_port_t iokit_alloc_object_port( io_object_t obj,
ipc_kobject_type_t type );
extern kern_return_t iokit_destroy_object_port( ipc_port_t port );
extern mach_port_name_t iokit_make_send_right( task_t task,
io_object_t obj, ipc_kobject_type_t type );
extern kern_return_t iokit_mod_send_right( task_t task, mach_port_name_t name, mach_port_delta_t delta );
extern io_object_t iokit_lookup_connect_ref(io_object_t clientRef, ipc_space_t task);
extern io_object_t iokit_lookup_connect_ref_current_task(io_object_t clientRef);
extern void iokit_retain_port( ipc_port_t port );
extern void iokit_release_port( ipc_port_t port );
extern kern_return_t iokit_switch_object_port( ipc_port_t port, io_object_t obj, ipc_kobject_type_t type );
extern kern_return_t IOMapPages(vm_map_t map, mach_vm_address_t va, mach_vm_address_t pa,
mach_vm_size_t length, unsigned int mapFlags);
extern kern_return_t IOUnmapPages(vm_map_t map, mach_vm_address_t va, mach_vm_size_t length);
extern kern_return_t IOProtectCacheMode(vm_map_t map, mach_vm_address_t va,
mach_vm_size_t length, unsigned int options);
extern unsigned int IODefaultCacheBits(addr64_t pa);
MIGEXTERN io_object_t
iokit_lookup_object_port(
ipc_port_t port)
{
register io_object_t obj;
if (!IP_VALID(port))
return (NULL);
ip_lock(port);
if (ip_active(port) && (ip_kotype(port) == IKOT_IOKIT_OBJECT)) {
obj = (io_object_t) port->ip_kobject;
iokit_add_reference( obj );
}
else
obj = NULL;
ip_unlock(port);
return( obj );
}
MIGEXTERN io_object_t
iokit_lookup_connect_port(
ipc_port_t port)
{
register io_object_t obj;
if (!IP_VALID(port))
return (NULL);
ip_lock(port);
if (ip_active(port) && (ip_kotype(port) == IKOT_IOKIT_CONNECT)) {
obj = (io_object_t) port->ip_kobject;
iokit_add_reference( obj );
}
else
obj = NULL;
ip_unlock(port);
return( obj );
}
EXTERN io_object_t
iokit_lookup_connect_ref(io_object_t connectRef, ipc_space_t space)
{
io_object_t obj = NULL;
if (connectRef && MACH_PORT_VALID((mach_port_name_t)connectRef)) {
ipc_port_t port;
kern_return_t kr;
kr = ipc_object_translate(space, (mach_port_name_t)connectRef, MACH_PORT_RIGHT_SEND, (ipc_object_t *)&port);
if (kr == KERN_SUCCESS) {
assert(IP_VALID(port));
if (ip_active(port) && (ip_kotype(port) == IKOT_IOKIT_CONNECT)) {
obj = (io_object_t) port->ip_kobject;
iokit_add_reference(obj);
}
ip_unlock(port);
}
}
return obj;
}
EXTERN io_object_t
iokit_lookup_connect_ref_current_task(io_object_t connectRef)
{
return iokit_lookup_connect_ref(connectRef, current_space());
}
EXTERN void
iokit_retain_port( ipc_port_t port )
{
ipc_port_reference( port );
}
EXTERN void
iokit_release_port( ipc_port_t port )
{
ipc_port_release( port );
}
MIGEXTERN ipc_port_t
iokit_make_object_port(
io_object_t obj )
{
register ipc_port_t port;
register ipc_port_t sendPort;
if( obj == NULL)
return IP_NULL;
port = iokit_port_for_object( obj, IKOT_IOKIT_OBJECT );
if( port) {
sendPort = ipc_port_make_send( port);
iokit_release_port( port );
} else
sendPort = IP_NULL;
iokit_remove_reference( obj );
return( sendPort);
}
MIGEXTERN ipc_port_t
iokit_make_connect_port(
io_object_t obj )
{
register ipc_port_t port;
register ipc_port_t sendPort;
if( obj == NULL)
return IP_NULL;
port = iokit_port_for_object( obj, IKOT_IOKIT_CONNECT );
if( port) {
sendPort = ipc_port_make_send( port);
iokit_release_port( port );
} else
sendPort = IP_NULL;
iokit_remove_reference( obj );
return( sendPort);
}
int gIOKitPortCount;
EXTERN ipc_port_t
iokit_alloc_object_port( io_object_t obj, ipc_kobject_type_t type )
{
ipc_port_t notify;
ipc_port_t port;
do {
port = ipc_port_alloc_kernel();
if( port == IP_NULL)
continue;
ipc_kobject_set( port, (ipc_kobject_t) obj, type);
notify = ipc_port_make_sonce( port);
ip_lock( port);
ipc_port_nsrequest( port, 1, notify, ¬ify);
assert( notify == IP_NULL);
gIOKitPortCount++;
} while( FALSE);
return( port );
}
EXTERN kern_return_t
iokit_destroy_object_port( ipc_port_t port )
{
ipc_kobject_set( port, IKO_NULL, IKOT_NONE);
ipc_port_dealloc_kernel( port);
gIOKitPortCount--;
return( KERN_SUCCESS);
}
EXTERN kern_return_t
iokit_switch_object_port( ipc_port_t port, io_object_t obj, ipc_kobject_type_t type )
{
ipc_kobject_set( port, (ipc_kobject_t) obj, type);
return( KERN_SUCCESS);
}
EXTERN mach_port_name_t
iokit_make_send_right( task_t task, io_object_t obj, ipc_kobject_type_t type )
{
ipc_port_t port;
ipc_port_t sendPort;
mach_port_name_t name;
if( obj == NULL)
return MACH_PORT_NULL;
port = iokit_port_for_object( obj, type );
if( port) {
sendPort = ipc_port_make_send( port);
iokit_release_port( port );
} else
sendPort = IP_NULL;
if (IP_VALID( sendPort )) {
kern_return_t kr;
kr = ipc_object_copyout( task->itk_space, (ipc_object_t) sendPort,
MACH_MSG_TYPE_PORT_SEND, TRUE, &name);
if ( kr != KERN_SUCCESS)
name = MACH_PORT_NULL;
} else if ( sendPort == IP_NULL)
name = MACH_PORT_NULL;
else if ( sendPort == IP_DEAD)
name = MACH_PORT_DEAD;
iokit_remove_reference( obj );
return( name );
}
EXTERN kern_return_t
iokit_mod_send_right( task_t task, mach_port_name_t name, mach_port_delta_t delta )
{
return (mach_port_mod_refs( task->itk_space, name, MACH_PORT_RIGHT_SEND, delta ));
}
static void
iokit_no_senders( mach_no_senders_notification_t * notification )
{
ipc_port_t port;
io_object_t obj = NULL;
ipc_kobject_type_t type = IKOT_NONE;
ipc_port_t notify;
port = (ipc_port_t) notification->not_header.msgh_remote_port;
if( IP_VALID(port)) {
ip_lock(port);
if( ip_active(port)) {
obj = (io_object_t) port->ip_kobject;
type = ip_kotype( port );
if( (IKOT_IOKIT_OBJECT == type)
|| (IKOT_IOKIT_CONNECT == type))
iokit_add_reference( obj );
else
obj = NULL;
}
ip_unlock(port);
if( obj ) {
mach_port_mscount_t mscount = notification->not_count;
if( KERN_SUCCESS != iokit_client_died( obj, port, type, &mscount ))
{
notify = ipc_port_make_sonce( port);
ip_lock( port);
ipc_port_nsrequest( port, mscount + 1, notify, ¬ify);
assert( notify == IP_NULL);
}
iokit_remove_reference( obj );
}
}
}
EXTERN
boolean_t
iokit_notify( mach_msg_header_t * msg )
{
switch (msg->msgh_id) {
case MACH_NOTIFY_NO_SENDERS:
iokit_no_senders((mach_no_senders_notification_t *) msg);
return TRUE;
case MACH_NOTIFY_PORT_DELETED:
case MACH_NOTIFY_PORT_DESTROYED:
case MACH_NOTIFY_SEND_ONCE:
case MACH_NOTIFY_DEAD_NAME:
default:
printf("iokit_notify: strange notification %d\n", msg->msgh_id);
return FALSE;
}
}
unsigned int IODefaultCacheBits(addr64_t pa)
{
return(pmap_cache_attributes(pa >> PAGE_SHIFT));
}
kern_return_t IOMapPages(vm_map_t map, mach_vm_address_t va, mach_vm_address_t pa,
mach_vm_size_t length, unsigned int options)
{
vm_prot_t prot;
unsigned int flags;
pmap_t pmap = map->pmap;
prot = (options & kIOMapReadOnly)
? VM_PROT_READ : (VM_PROT_READ|VM_PROT_WRITE);
switch(options & kIOMapCacheMask ) {
case kIOMapDefaultCache:
default:
flags = IODefaultCacheBits(pa);
break;
case kIOMapInhibitCache:
flags = VM_WIMG_IO;
break;
case kIOMapWriteThruCache:
flags = VM_WIMG_WTHRU;
break;
case kIOMapWriteCombineCache:
flags = VM_WIMG_WCOMB;
break;
case kIOMapCopybackCache:
flags = VM_WIMG_COPYBACK;
break;
}
pmap_map_block(pmap, va, (ppnum_t)atop_64(pa), (uint32_t) atop_64(round_page_64(length)), prot, flags, 0);
return( KERN_SUCCESS );
}
kern_return_t IOUnmapPages(vm_map_t map, mach_vm_address_t va, mach_vm_size_t length)
{
pmap_t pmap = map->pmap;
pmap_remove(pmap, trunc_page_64(va), round_page_64(va + length));
return( KERN_SUCCESS );
}
kern_return_t IOProtectCacheMode(vm_map_t __unused map, mach_vm_address_t __unused va,
mach_vm_size_t __unused length, unsigned int __unused options)
{
#if __ppc__
#else
mach_vm_size_t off;
vm_prot_t prot;
unsigned int flags;
pmap_t pmap = map->pmap;
prot = (options & kIOMapReadOnly)
? VM_PROT_READ : (VM_PROT_READ|VM_PROT_WRITE);
switch (options & kIOMapCacheMask)
{
case kIOMapDefaultCache:
default:
return (KERN_INVALID_ARGUMENT);
case kIOMapInhibitCache:
flags = VM_WIMG_IO;
break;
case kIOMapWriteThruCache:
flags = VM_WIMG_WTHRU;
break;
case kIOMapWriteCombineCache:
flags = VM_WIMG_WCOMB;
break;
case kIOMapCopybackCache:
flags = VM_WIMG_COPYBACK;
break;
}
for (off = 0; off < length; off += page_size)
{
ppnum_t ppnum = pmap_find_phys(pmap, va + off);
if (ppnum)
pmap_enter(pmap, va + off, ppnum, prot, flags, TRUE);
}
#endif
return (KERN_SUCCESS);
}
ppnum_t IOGetLastPageNumber(void)
{
ppnum_t lastPage, highest = 0;
unsigned int idx;
#if __ppc__
for (idx = 0; idx < pmap_mem_regions_count; idx++)
{
lastPage = pmap_mem_regions[idx].mrEnd;
#elif __i386__
for (idx = 0; idx < pmap_memory_region_count; idx++)
{
lastPage = pmap_memory_regions[idx].end - 1;
#else
#error arch
#endif
if (lastPage > highest)
highest = lastPage;
}
return (highest);
}
void IOGetTime( mach_timespec_t * clock_time);
void IOGetTime( mach_timespec_t * clock_time)
{
clock_get_system_nanotime(&clock_time->tv_sec, (uint32_t *) &clock_time->tv_nsec);
}