#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_host_server.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/ast.h>
#include <kern/counters.h>
#include <kern/queue.h>
#include <kern/zalloc.h>
#include <kern/thread.h>
#include <kern/thread_swap.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
#include <IOKit/IOTypes.h>
#define EXTERN
#define MIGEXTERN
extern void iokit_add_reference( io_object_t obj );
extern void iokit_remove_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 );
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);
}
EXTERN ipc_port_t
iokit_alloc_object_port( io_object_t obj, ipc_kobject_type_t type );
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 );
}
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;
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 %ld\n", msg->msgh_id);
return FALSE;
}
}
unsigned int IODefaultCacheBits(addr64_t pa)
{
unsigned int flags;
#ifndef i386
struct phys_entry * pp;
if ((pp = pmap_find_physentry(pa >> 12))) {
flags = VM_MEM_COHERENT;
if(pp->ppLink & ppG) flags |= VM_MEM_GUARDED;
if(pp->ppLink & ppI) flags |= VM_MEM_NOT_CACHEABLE;
} else
flags = VM_WIMG_IO;
#else
extern vm_offset_t avail_end;
if (pa < avail_end)
flags = VM_WIMG_COPYBACK;
else
flags = VM_WIMG_IO;
#endif
return flags;
}
kern_return_t IOMapPages(vm_map_t map, vm_offset_t va, vm_offset_t pa,
vm_size_t length, unsigned int options)
{
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:
flags = IODefaultCacheBits(pa);
break;
case kIOMapInhibitCache:
flags = VM_WIMG_IO;
break;
case kIOMapWriteThruCache:
flags = VM_WIMG_WTHRU;
break;
case kIOWriteCombineCache:
flags = VM_WIMG_WCOMB;
break;
case kIOMapCopybackCache:
flags = VM_WIMG_COPYBACK;
break;
}
#if __ppc__
pmap_map_block(pmap, (addr64_t)va, (ppnum_t)(pa >> 12), length, prot, flags, 0);
#else
for (off = 0; off < length; off += page_size)
pmap_enter(pmap, va + off, (pa + off) >> 12, prot, flags, TRUE);
#endif
return( KERN_SUCCESS );
}
kern_return_t IOUnmapPages(vm_map_t map, vm_offset_t va, vm_size_t length)
{
pmap_t pmap = map->pmap;
pmap_remove(pmap, trunc_page_64(va), round_page_64(va + length));
return( KERN_SUCCESS );
}
void IOGetTime( mach_timespec_t * clock_time);
void IOGetTime( mach_timespec_t * clock_time)
{
clock_get_system_nanotime(&clock_time->tv_sec, &clock_time->tv_nsec);
}