Arbitration.m

/*
 * The MIT License (MIT)
 *
 * Application: NTFS OS X
 * Copyright (c) 2015 Jeevanandam M. (jeeva@myjeeva.com)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

//
//  Arbitration.m
//  NTFS-OSX
//
//  Created by Jeevanandam M. on 6/5/15.
//  Copyright (c) 2015 myjeeva.com. All rights reserved.
//

#import "Arbitration.h"
#import "Disk.h"

DASessionRef session;
DASessionRef approvalSession;
NSMutableSet *ntfsDisks;


void RegisterDA(void) {

	// Disk Arbitration Session
	session = DASessionCreate(kCFAllocatorDefault);
	if (!session) {
		[NSException raise:NSGenericException format:@"Unable to create Disk Arbitration session."];
		return;
	}

	LogDebug(@"Disk Arbitration Session created");

	ntfsDisks = [NSMutableSet new];

	// Matching Conditions
	CFMutableDictionaryRef match = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);

	// Device matching criteria
	// 1. Of-course it shouldn't be internal device since
	CFDictionaryAddValue(match, kDADiskDescriptionDeviceInternalKey, kCFBooleanFalse);

	// Volume matching criteria
	// It should statisfy following
	CFDictionaryAddValue(match, kDADiskDescriptionVolumeKindKey, (__bridge CFStringRef)DADiskDescriptionVolumeKindValue);
	CFDictionaryAddValue(match, kDADiskDescriptionVolumeMountableKey, kCFBooleanTrue);
	CFDictionaryAddValue(match, kDADiskDescriptionVolumeNetworkKey, kCFBooleanFalse);

	//CFDictionaryAddValue(match, kDADiskDescriptionDeviceProtocolKey, CFSTR(kIOPropertyPhysicalInterconnectTypeUSB));

	DASessionScheduleWithRunLoop(session, CFRunLoopGetMain(), kCFRunLoopCommonModes);

	// Registring callbacks
	DARegisterDiskAppearedCallback(session, match, DiskAppearedCallback, (__bridge void *)AppName);
	DARegisterDiskDisappearedCallback(session, match, DiskDisappearedCallback, (__bridge void *)AppName);
	DARegisterDiskDescriptionChangedCallback(session, match, NULL, DiskDescriptionChangedCallback, (__bridge void *)AppName);

	// Disk Arbitration Approval Session
	approvalSession = DAApprovalSessionCreate(kCFAllocatorDefault);
	if (!approvalSession) {
		LogDebug(@"Unable to create Disk Arbitration approval session.");
		return;
	}

	LogDebug(@"Disk Arbitration Approval Session created");
	DAApprovalSessionScheduleWithRunLoop(approvalSession, CFRunLoopGetMain(), kCFRunLoopCommonModes);

	// Same match condition for Approval session too
	DARegisterDiskMountApprovalCallback(approvalSession, match, DiskMountApprovalCallback, (__bridge void *)AppName);

	Release(match);
}

void UnregisterDA(void) {
	// DA Session
	if (session) {
		DAUnregisterCallback(session, DiskAppearedCallback, (__bridge void *)AppName);
		DAUnregisterCallback(session, DiskDisappearedCallback, (__bridge void *)AppName);

		DASessionUnscheduleFromRunLoop(session, CFRunLoopGetMain(), kCFRunLoopCommonModes);
		Release(session);

		LogDebug(@"Disk Arbitration Session destoryed");
	}

	// DA Approval Session
	if (approvalSession) {
		DAUnregisterApprovalCallback(approvalSession, DiskMountApprovalCallback, (__bridge void *)AppName);

		DAApprovalSessionUnscheduleFromRunLoop(approvalSession, CFRunLoopGetMain(), kCFRunLoopCommonModes);
		Release(approvalSession);

		LogDebug(@"Disk Arbitration Approval Session destoryed");
	}

	[ntfsDisks removeAllObjects];
	ntfsDisks = nil;
}

BOOL Validate(DADiskRef diskRef) {

	if (DADiskGetBSDName(diskRef) == NULL) {
		[NSException raise:NSInternalInconsistencyException format:@"NTFS Disk without BSDName"];
	}

	return TRUE;
}

void DiskAppearedCallback(DADiskRef diskRef, void *context) {
	LogDebug(@"DiskAppearedCallback called: %s", DADiskGetBSDName(diskRef));

	if (Validate(diskRef)) {
		Disk *disk = [[Disk alloc] initWithDADiskRef:diskRef];
		LogDebug(@"Name: %@ \tUUID: %@", disk.volumeName, disk.volumeUUID);

		[[NSNotificationCenter defaultCenter] postNotificationName:NTFSDiskAppearedNotification object:disk];
	}
}

void DiskDisappearedCallback(DADiskRef diskRef, void *context) {
	LogDebug(@"DiskDisappearedCallback called: %s", DADiskGetBSDName(diskRef));

	if (Validate(diskRef)) {
		Disk *disk = [Disk getDiskForDARef:diskRef];
		LogDebug(@"Name: %@ \tUUID: %@", disk.volumeName, disk.volumeUUID);

		[[NSNotificationCenter defaultCenter] postNotificationName:NTFSDiskDisappearedNotification object:disk];
	}
}

void DiskDescriptionChangedCallback(DADiskRef diskRef, CFArrayRef keys, void *context) {
	LogDebug(@"DiskDescriptionChangedCallback called: %s", DADiskGetBSDName(diskRef));

	Disk *disk = [Disk getDiskForDARef:diskRef];

	if (disk) {
		CFDictionaryRef newDesc = DADiskCopyDescription(diskRef);
		disk.desc = newDesc;

		LogDebug(@"Updated Disk Description: %@", disk.desc);

		Release(newDesc);
	}
}

DADissenterRef DiskMountApprovalCallback(DADiskRef diskRef, void *context) {
	LogDebug(@"DiskMountApprovalCallback called: %s", DADiskGetBSDName(diskRef));

	if (Validate(diskRef)) {
		Disk *disk = [[Disk alloc] initWithDADiskRef:diskRef];
		LogDebug(@"Name: %@ \tUUID: %@", disk.volumeName, disk.volumeUUID);
	}

	return NULL;
}