//     File: main.m
// Abstract: Main file supporting the startup of the Virtual Scanner.
//  Version: 1.0
// 
// Disclaimer: IMPORTANT:  This Apple software is supplied to you by Apple
// Inc. ("Apple") in consideration of your agreement to the following
// terms, and your use, installation, modification or redistribution of
// this Apple software constitutes acceptance of these terms.  If you do
// not agree with these terms, please do not use, install, modify or
// redistribute this Apple software.
// 
// In consideration of your agreement to abide by the following terms, and
// subject to these terms, Apple grants you a personal, non-exclusive
// license, under Apple's copyrights in this original Apple software (the
// "Apple Software"), to use, reproduce, modify and redistribute the Apple
// Software, with or without modifications, in source and/or binary forms;
// provided that if you redistribute the Apple Software in its entirety and
// without modifications, you must retain this notice and the following
// text and disclaimers in all such redistributions of the Apple Software.
// Neither the name, trademarks, service marks or logos of Apple Inc. may
// be used to endorse or promote products derived from the Apple Software
// without specific prior written permission from Apple.  Except as
// expressly stated in this notice, no other rights or licenses, express or
// implied, are granted by Apple herein, including but not limited to any
// patent rights that may be infringed by your derivative works or by other
// works in which the Apple Software may be incorporated.
// 
// The Apple Software is provided by Apple on an "AS IS" basis.  APPLE
// MAKES NO WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION
// THE IMPLIED WARRANTIES OF NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND
// OPERATION ALONE OR IN COMBINATION WITH YOUR PRODUCTS.
// 
// IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL
// OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION,
// MODIFICATION AND/OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED
// AND WHETHER UNDER THEORY OF CONTRACT, TORT (INCLUDING NEGLIGENCE),
// STRICT LIABILITY OR OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
// 
// Copyright (C) 2012 Apple Inc. All Rights Reserved.
// 

#import <sys/stat.h>
#import <pthread.h>
#import "EntryPoints.h"

//----------------------------------------------------------------------------------------------------------------------

pthread_mutex_t       _gScannersMutex       = PTHREAD_MUTEX_INITIALIZER;
NSMutableDictionary*  _gScannersDictionary  = NULL;
BOOL                  _gIsVirtualScanner    = NO;
DASessionRef          _gDiskArbSession      = NULL;
extern int            gArgc;
extern char*          gArgv[10];

//----------------------------------------------------------------------------------------------------------------------

int main(int argc, char *argv[])
{
    int status          = 0;
    int option          = 0;
    char tempStr[256]   = {0};
    
    while( ( option = getopt( argc, argv, "d:") ) != -1 )
    {
        switch ( option )
        {
            case 'd':
                tempStr[0] = 0;
                strlcpy( tempStr, optarg, 31 );
                gArgc       = 2;
                gArgv[0]    = argv[0];
                gArgv[1]    = strdup( tempStr );
            break;
        }
    }
    
    gICDScannerCallbackFunctions.f_ICD_ScannerOpenUSBDevice                   = _ICD_ScannerOpenUSBDevice;
    gICDScannerCallbackFunctions.f_ICD_ScannerOpenFireWireDeviceWithIORegPath = _ICD_ScannerOpenFWDevice;
    gICDScannerCallbackFunctions.f_ICD_ScannerOpenTCPIPDevice                 = _ICD_ScannerOpenTCPIPDevice;
    gICDScannerCallbackFunctions.f_ICD_ScannerCloseDevice                     = _ICD_ScannerCloseDevice;
    gICDScannerCallbackFunctions.f_ICD_ScannerPeriodicTask                    = _ICD_ScannerPeriodicTask;
    gICDScannerCallbackFunctions.f_ICD_ScannerGetObjectInfo                   = _ICD_ScannerGetObjectInfo;
    gICDScannerCallbackFunctions.f_ICD_ScannerCleanup                         = _ICD_ScannerCleanup;
    gICDScannerCallbackFunctions.f_ICD_ScannerGetPropertyData                 = NULL;   // Unused
    gICDScannerCallbackFunctions.f_ICD_ScannerSetPropertyData                 = NULL;   // Unused
    gICDScannerCallbackFunctions.f_ICD_ScannerReadFileData                    = _ICD_ScannerReadFileData;
    gICDScannerCallbackFunctions.f_ICD_ScannerWriteDataToFile                 = NULL;   // Unused
    gICDScannerCallbackFunctions.f_ICD_ScannerWriteFileData                   = NULL;   // Unused
    gICDScannerCallbackFunctions.f_ICD_ScannerSendMessage                     = _ICD_ScannerSendMessage;
    gICDScannerCallbackFunctions.f_ICD_ScannerAddPropertiesToCFDictionary     = _ICD_ScannerAddPropertiesToCFDictionary;
    
    gICDScannerCallbackFunctions.f_ICD_ScannerOpenSession                     = _ICD_ScannerOpenSession;
    gICDScannerCallbackFunctions.f_ICD_ScannerCloseSession                    = _ICD_ScannerCloseSession;
    gICDScannerCallbackFunctions.f_ICD_ScannerInitialize                      = NULL;   // Unused
    gICDScannerCallbackFunctions.f_ICD_ScannerGetParameters                   = _ICD_ScannerGetParameters;
    gICDScannerCallbackFunctions.f_ICD_ScannerSetParameters                   = _ICD_ScannerSetParameters;
    gICDScannerCallbackFunctions.f_ICD_ScannerStatus                          = _ICD_ScannerStatus;
    gICDScannerCallbackFunctions.f_ICD_ScannerStart                           = _ICD_ScannerStart;
    
    if ( _gScannersDictionary == NULL )
        _gScannersDictionary = [[NSMutableDictionary alloc] initWithCapacity:0];

    CFBundleRef bundle    = CFBundleGetMainBundle();
    NSString*   bundleID  = (NSString*)CFBundleGetIdentifier( bundle );
    
    _gIsVirtualScanner = [bundleID isEqualToString:@"com.apple.VirtualScanner"];
    
    if ( _gIsVirtualScanner )
    {
        _gDiskArbSession = DASessionCreate( kCFAllocatorDefault );
#if MAC_OS_X_VERSION_MAX_ALLOWED < 1070
        DASessionScheduleWithRunLoop( _gDiskArbSession, CFRunLoopGetMain(), kCFRunLoopDefaultMode );
#else
        DASessionSetDispatchQueue( _gDiskArbSession, dispatch_get_global_queue( DISPATCH_QUEUE_PRIORITY_DEFAULT, 0ul ) );
#endif
    }
    
    status = ICD_ScannerMain(argc, (const char **)argv);
    
    if ( _gDiskArbSession )
    {
#if MAC_OS_X_VERSION_MAX_ALLOWED < 1070
        DASessionUnscheduleFromRunLoop( _gDiskArbSession, CFRunLoopGetMain(), kCFRunLoopDefaultMode );
#else
        DASessionSetDispatchQueue( _gDiskArbSession, NULL );
#endif
        CFRelease( _gDiskArbSession );
        _gDiskArbSession = NULL;
    }
    
    pthread_mutex_lock( &_gScannersMutex );
    [_gScannersDictionary release];
    pthread_mutex_unlock( &_gScannersMutex );
    pthread_mutex_destroy( &_gScannersMutex );
    
    return status;
}

//----------------------------------------------------------------------------------------------------------------------