//

//  NSWorkspace_RBAdditions.m

//  PathProps

//

//  Created by Rainer Brockerhoff on 10/04/2007.

//  Copyright 2007 Rainer Brockerhoff. All rights reserved.

//

#import "NSWorkspace_RBAdditions.h"

#include <IOKit/IOKitLib.h>

#include <sys/mount.h>

#include <mach/mach.h>

NSString* NSWorkspace_RBfstypename = @"NSWorkspace_RBfstypename";

NSString* NSWorkspace_RBmntonname = @"NSWorkspace_RBmntonname";

NSString* NSWorkspace_RBmntfromname = @"NSWorkspace_RBmntfromname";

NSString* NSWorkspace_RBdeviceinfo = @"NSWorkspace_RBdeviceinfo";

NSString* NSWorkspace_RBimagefilepath = @"NSWorkspace_RBimagefilepath";

NSString* NSWorkspace_RBconnectiontype = @"NSWorkspace_RBconnectiontype";

NSString* NSWorkspace_RBpartitionscheme = @"NSWorkspace_RBpartitionscheme";

NSString* NSWorkspace_RBserverURL = @"NSWorkspace_RBserverURL";

// This static funtion concatenates two strings, but first checks several possibilities...

// like one or the other nil, or one containing the other already.

static NSString* AddPart(NSString* first,NSString* second) {

    if (!second) {

        return first;

    }

    second = [second stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];

    if (first) {

        if ([first rangeOfString:second options:NSCaseInsensitiveSearch].location==NSNotFound) {

            if ([second rangeOfString:first options:NSCaseInsensitiveSearch].location==NSNotFound) {

                return [NSString stringWithFormat:@"%@; %@",first,second];

            }

            return second;

        }

        return first;

    }

    return second;

}

// This static functions recurses "upwards" over the IO registry. Returns strings that are concatenated

// and ultimately end up under the NSWorkspace_RBdeviceinfo key.

// This isn't too robust in that it assumes that objects returned by the objectForKey methods are

// either strings or dictionaries. A "standard" implementations would use either only CoreFoundation and

// IOKit calls for this, or do more robust type checking on the returned objects.

//

// Also notice that this works as determined experimentally in 10.4.9, there's no official docs I could find.

// YMMV, and it may stop working in any new version of Mac OS X.

static NSString* CheckParents(io_object_t thing,NSString* part,NSMutableDictionary* dict) {

    NSString* result = part;

    io_iterator_t parentsIterator = 0;

    kern_return_t kernResult = IORegistryEntryGetParentIterator(thing,kIOServicePlane,&parentsIterator);

    if ((kernResult==KERN_SUCCESS)&&parentsIterator) {

        io_object_t nextParent = 0;

        while ((nextParent = IOIteratorNext(parentsIterator))) {

            NSDictionary* props = nil;

            NSString* image = nil;

            NSString* partition = nil;

            NSString* connection = nil;

            kernResult = IORegistryEntryCreateCFProperties(nextParent,(CFMutableDictionaryRef*)&props,kCFAllocatorDefault,0);

            if (IOObjectConformsTo(nextParent,"IOApplePartitionScheme")) {

                partition = [props objectForKey:@"Content Mask"];

            } else if (IOObjectConformsTo(nextParent,"IOMedia")) {

                partition = [props objectForKey:@"Content"];

            } else if (IOObjectConformsTo(nextParent,"IODiskImageBlockStorageDeviceOutKernel")) {

                NSData* data = nil;

                                if ((data = [[props objectForKey:@"Protocol Characteristics"] objectForKey:@"Virtual Interface Location Path"])) {

                    image = [[[NSString alloc] initWithBytes:[data bytes] length:[data length] encoding:NSUTF8StringEncoding] autorelease];

                }

            } else if (IOObjectConformsTo(nextParent,"IOHDIXHDDriveInKernel")) {

                image = [props objectForKey:@"KDIURLPath"];

            }

            NSDictionary* subdict;

                        if ((subdict = [props objectForKey:@"Protocol Characteristics"])) {

                connection = [subdict objectForKey:@"Physical Interconnect"];

            } else {

                connection = [props objectForKey:@"Physical Interconnect"];

            }

            if (connection) {

                [dict setObject:AddPart([dict objectForKey:NSWorkspace_RBconnectiontype],connection) forKey:NSWorkspace_RBconnectiontype];

            }

            if (partition) {

                [dict setObject:partition forKey:NSWorkspace_RBpartitionscheme];

            }

            if (image) {

                [dict setObject:image forKey:NSWorkspace_RBimagefilepath];

            }

            NSString* value;

                        if ((subdict = [props objectForKey:@"Device Characteristics"])) {

                                if ((value = [subdict objectForKey:@"Product Name"])) {

                    result = AddPart(result,value);

                }

                                if ((value = [subdict objectForKey:@"Product Revision Level"])) {

                    result = AddPart(result,value);

                }

                                if ((value = [subdict objectForKey:@"Vendor Name"])) {

                    result = AddPart(result,value);

                }

            }

                        if ((value = [props objectForKey:@"USB Serial Number"])) {

                result = AddPart(result,value);

            }

                        if ((value = [props objectForKey:@"USB Vendor Name"])) {

                result = AddPart(result,value);

            }

            NSString* cls = [(NSString*)IOObjectCopyClass(nextParent) autorelease];

            if (![cls isEqualToString:@"IOPCIDevice"]) {

// Uncomment the following line to have the device tree dumped to the console.

//              NSLog(@"=================================> %@:%@\n",cls,props);

                result = CheckParents(nextParent,result,dict);

            }

            IOObjectRelease(nextParent);

        }

    }

    if (parentsIterator) {

        IOObjectRelease(parentsIterator);

    }

    return result;

}

// This formats the (partially undocumented) AFPXMountInfo info into a string.

/*

static NSString* FormatAFPURL(AFPXVolMountInfoPtr mountInfo,NSString** devdesc) {

    UInt8* work = ((UInt8*)mountInfo)+mountInfo->serverNameOffset;

    if (devdesc) {

        *devdesc = [[[NSString alloc] initWithBytes:&work[1] length:work[0] encoding:NSUTF8StringEncoding] autorelease];

    }

    work = ((UInt8*)mountInfo)+mountInfo->volNameOffset;

    NSString* volname = [[[NSString alloc] initWithBytes:&work[1] length:work[0] encoding:NSUTF8StringEncoding] autorelease];

    work = ((UInt8*)mountInfo)+mountInfo->alternateAddressOffset;

    AFPAlternateAddress* afpa = (AFPAlternateAddress*)work;

    AFPTagData* afpta = (AFPTagData*)(&afpa->fAddressList);

    NSString* ip = nil;

    NSString* dns = nil;

    int i = afpa->fAddressCount;

    while ((i-->0)) {

        switch (afpta->fType) {

            case kAFPTagTypeIP:

                if (!ip) {

                    ip = [[[NSString alloc] initWithBytes:&afpta->fData[0] length:afpta->fLength-2 encoding:NSUTF8StringEncoding] autorelease];

                }

                break;

            case kAFPTagTypeIPPort:

                ip = [NSString stringWithFormat:@"%u.%u.%u.%u:%u",afpta->fData[0],afpta->fData[1],afpta->fData[2],afpta->fData[3],OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[4])];

                break;

            case kAFPTagTypeDNS:

                dns = [[[NSString alloc] initWithBytes:&afpta->fData[0] length:afpta->fLength-2 encoding:NSUTF8StringEncoding] autorelease];

                break;

            case 0x07:

                ip = [NSString stringWithFormat:@"[%x:%x:%x:%x:%x:%x:%x:%x]",OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[0]),

                    OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[2]),OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[4]),

                    OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[6]),OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[8]),

                    OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[10]),OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[12]),

                    OSSwapBigToHostInt16(*(UInt16*)&afpta->fData[14])];

                break;

        }

        afpta = (AFPTagData*)((char*)afpta+afpta->fLength);

    }

    return [NSString stringWithFormat:@"afp://%@/%@",dns?:(ip?:@""),volname];

}

*/

@implementation NSWorkspace (NSWorkspace_RBAdditions)

// Returns a NSDictionary with properties for the path. See details in the .h file.

// This assumes that the length of path is less than PATH_MAX (currently 1024 characters).

- (NSDictionary*)propertiesForPath:(NSString*)path {

    const char* ccpath = (const char*)[path fileSystemRepresentation];

    NSMutableDictionary* result = nil;

    struct statfs fs;

    if (!statfs(ccpath,&fs)) {

        NSString* from = [NSString stringWithUTF8String:fs.f_mntfromname];

        result = [NSMutableDictionary dictionaryWithObjectsAndKeys:

            [NSString stringWithUTF8String:fs.f_fstypename],NSWorkspace_RBfstypename,

            [NSString stringWithUTF8String:fs.f_mntonname],NSWorkspace_RBmntonname,

            nil];

        if (strncmp(fs.f_mntfromname,"/dev/",5)==0) {

// For a local volume,get the IO registry tree and search it for further info.

            mach_port_t masterPort = 0;

            io_iterator_t mediaIterator = 0;

            kern_return_t kernResult = IOMasterPort(bootstrap_port,&masterPort);

            if (kernResult==KERN_SUCCESS) {

                CFMutableDictionaryRef classesToMatch = IOBSDNameMatching(masterPort,0,&fs.f_mntfromname[5]);

                if (classesToMatch) {

                    kernResult = IOServiceGetMatchingServices(masterPort,classesToMatch,&mediaIterator);

                    if ((kernResult==KERN_SUCCESS)&&mediaIterator) {

                        io_object_t firstMedia = 0;

                        while ((firstMedia = IOIteratorNext(mediaIterator))) {

                            NSString* stuff = CheckParents(firstMedia,nil,result);

                            if (stuff) {

                                [result setObject:stuff forKey:NSWorkspace_RBdeviceinfo];

                            }

                            IOObjectRelease(firstMedia);

                        }

                    }

                }

            }

            if (mediaIterator) {

                IOObjectRelease(mediaIterator);

            }

            if (masterPort) {

                mach_port_deallocate(mach_task_self(),masterPort);

            }

        }

        //Don't need this for disk images, gets around warnings for some deprecated functions

        /* else {

// For a network volume, get the volume reference number and use to get the server URL.

            FSRef ref;

            if (FSPathMakeRef((const UInt8*)ccpath,&ref,NULL)==noErr) {

                FSCatalogInfo info;

                if (FSGetCatalogInfo(&ref,kFSCatInfoVolume,&info,NULL,NULL,NULL)==noErr) {

                    ParamBlockRec pb;

                    UInt16 vmisize = 0;

                    VolumeMountInfoHeaderPtr mountInfo = NULL;

                    pb.ioParam.ioCompletion = NULL;

                    pb.ioParam.ioNamePtr = NULL;

                    pb.ioParam.ioVRefNum = info.volume;

                    pb.ioParam.ioBuffer = (Ptr)&vmisize;

                    pb.ioParam.ioReqCount = sizeof(vmisize);

                    if ((PBGetVolMountInfoSize(&pb)==noErr)&&vmisize) {

                        mountInfo = (VolumeMountInfoHeaderPtr)malloc(vmisize);

                        if (mountInfo) {

                            pb.ioParam.ioBuffer = (Ptr)mountInfo;

                            pb.ioParam.ioReqCount = vmisize;

                            if (PBGetVolMountInfo(&pb)==noErr) {

                                NSString* url = nil;

                                switch (mountInfo->media) {

                                case AppleShareMediaType:

                                    url = FormatAFPURL((AFPXVolMountInfoPtr)mountInfo,&from);

                                    break;

                                case 'http':

                                    url = from;

                                    break;

                                case 'crbm':

                                case 'nfs_':

                                case 'cifs':

                                    url = [NSString stringWithUTF8String:(char*)mountInfo+sizeof(VolumeMountInfoHeader)+sizeof(OSType)];

                                    break;

                                }

                                if (url) {

                                    [result setObject:url forKey:NSWorkspace_RBserverURL];

                                }

                            }

                        }

                        free(mountInfo);

                    }

                }

            }

        }*/

        [result setObject:from forKey:NSWorkspace_RBmntfromname];

    }

    return result;

}

@end