int main ( int argc, char** argv )
{
FILE* fh;
byte_t* buf;
size_t flen;
class_t cls;
flags_t flags;
if ( get_options ( argc, argv, &flags ) )
return 1;
if ( ! ( buf = alloc_buffer ( ) ) )
return 1;
if ( ! ( fh = open_input_file ( argv [ argc - 1 ] ) ) )
return 1;
if ( ! ( flen = load_file ( buf, fh ) ) )
return 1;
if ( classify_image ( &cls, buf, flen ) )
return 1;
if ( flags.action == ACTION_CONVERT )
return action_convert ( &cls, buf, flen, argv [ argc - 1 ] );
else
return action_print ( &cls );
}
static IOReturn
vnodeNotificationHandler(io_connect_t connection)
{
kern_return_t kr;
VnodeWatcherData_t vdata;
UInt32 dataSize;
IODataQueueMemory *queueMappedMemory;
vm_size_t queueMappedMemorySize;
vm_address_t address = nil;
vm_size_t size = 0;
unsigned int msgType = 1; // family-defined port type (arbitrary)
mach_port_t recvPort;
// allocate a Mach port to receive notifications from the IODataQueue
if (!(recvPort = IODataQueueAllocateNotificationPort())) {
fprintf(stderr, "%s: failed to allocate notification port\n", PROGNAME);
return kIOReturnError;
}
// this will call registerNotificationPort() inside our user client class
kr = IOConnectSetNotificationPort(connection, msgType, recvPort, 0);
if (kr != kIOReturnSuccess) {
fprintf(stderr, "%s: failed to register notification port (%d)\n",
PROGNAME, kr);
mach_port_destroy(mach_task_self(), recvPort);
return kr;
}
// this will call clientMemoryForType() inside our user client class
kr = IOConnectMapMemory(connection, kIODefaultMemoryType,
mach_task_self(), &address, &size, kIOMapAnywhere);
if (kr != kIOReturnSuccess) {
fprintf(stderr, "%s: failed to map memory (%d)\n", PROGNAME, kr);
mach_port_destroy(mach_task_self(), recvPort);
return kr;
}
queueMappedMemory = (IODataQueueMemory *)address;
queueMappedMemorySize = size;
while (IODataQueueWaitForAvailableData(queueMappedMemory, recvPort) ==
kIOReturnSuccess) {
while (IODataQueueDataAvailable(queueMappedMemory)) {
dataSize = sizeof(vdata);
kr = IODataQueueDequeue(queueMappedMemory, &vdata, &dataSize);
if (kr == kIOReturnSuccess) {
if (*(UInt8 *)&vdata == kt_kStopListeningToMessages)
goto exit;
printf("\"%s\" %s %s %lu(%s) ",
vdata.path,
vtype_name(vdata.v_type),
vtag_name(vdata.v_tag),
vdata.pid,
vdata.p_comm);
action_print(vdata.action, (vdata.v_type & VDIR));
} else
fprintf(stderr, "*** error in receiving data (%d)\n", kr);
}
}
exit:
kr = IOConnectUnmapMemory(connection, kIODefaultMemoryType,
mach_task_self(), address);
if (kr != kIOReturnSuccess)
fprintf(stderr, "%s: failed to unmap memory (%d)\n", PROGNAME, kr);
mach_port_destroy(mach_task_self(), recvPort);
return kr;
}
