DAReturn _DADiskSetEncoding( DADiskRef disk, CFStringEncoding encoding )
{
CFMutableArrayRef keys = NULL;
CFStringRef name1 = NULL;
CFStringRef name2 = NULL;
CFURLRef path1 = NULL;
CFURLRef path2 = NULL;
DAReturn status = kDAReturnSuccess;
path1 = DADiskGetDescription( disk, kDADiskDescriptionVolumePathKey );
if ( path1 == NULL ) { status = kDAReturnBadArgument; goto _DADiskSetEncodingErr; }
status = __DAFileSystemSetEncoding( DADiskGetFileSystem( disk ), path1, encoding );
if ( status ) { status = unix_err( status ); goto _DADiskSetEncodingErr; }
keys = CFArrayCreateMutable( kCFAllocatorDefault, 0, &kCFTypeArrayCallBacks );
if ( keys == NULL ) { status = kDAReturnNoResources; goto _DADiskSetEncodingErr; }
name1 = DADiskGetDescription( disk, kDADiskDescriptionVolumeNameKey );
if ( name1 == NULL ) { status = kDAReturnError; goto _DADiskSetEncodingErr; }
name2 = _DAFileSystemCopyName( DADiskGetFileSystem( disk ), path1 );
if ( name2 == NULL ) { status = kDAReturnError; goto _DADiskSetEncodingErr; }
status = CFEqual( name1, name2 );
///w:start
// if ( status ) { status = kDAReturnSuccess; goto _DADiskSetEncodingErr; }
///w:stop
DADiskSetDescription( disk, kDADiskDescriptionVolumeNameKey, name2 );
CFArrayAppendValue( keys, kDADiskDescriptionVolumeNameKey );
///w:start
if ( status == FALSE )
///w:stop
path2 = DAMountCreateMountPointWithAction( disk, kDAMountPointActionMove );
///w:start
status = kDAReturnSuccess;
///w:stop
if ( path2 )
{
DADiskSetBypath( disk, path2 );
DADiskSetDescription( disk, kDADiskDescriptionVolumePathKey, path2 );
CFArrayAppendValue( keys, kDADiskDescriptionVolumePathKey );
}
DADiskDescriptionChangedCallback( disk, keys );
_DADiskSetEncodingErr:
if ( keys ) CFRelease( keys );
if ( name2 ) CFRelease( name2 );
if ( path2 ) CFRelease( path2 );
return status;
}
static void __DAStageProbeCallback( int status, CFBooleanRef clean, CFStringRef name, CFUUIDRef uuid, void * context )
{
DADiskRef disk = context;
CFMutableArrayRef keys = NULL;
CFStringRef kind = NULL;
DALogDebugHeader( "%s -> %s", gDAProcessNameID, gDAProcessNameID );
if ( status )
{
CFMutableArrayRef candidates;
candidates = ( void * ) DADiskGetContext( disk );
if ( DADiskGetFileSystem( disk ) )
{
kind = DAFileSystemGetKind( DADiskGetFileSystem( disk ) );
DALogDebug( " probed disk, id = %@, with %@, failure.", disk, kind );
if ( status != FSUR_UNRECOGNIZED )
{
DALogError( "unable to probe %@ (status code 0x%08X).", disk, status );
}
}
/*
* Find a probe candidate for this media object.
*/
while ( CFArrayGetCount( candidates ) )
{
CFDictionaryRef candidate;
candidate = CFArrayGetValueAtIndex( candidates, 0 );
if ( candidate )
{
DAFileSystemRef filesystem;
filesystem = ( void * ) CFDictionaryGetValue( candidate, kDAFileSystemKey );
if ( filesystem )
{
CFDictionaryRef properties;
properties = CFDictionaryGetValue( candidate, CFSTR( kFSMediaPropertiesKey ) );
if ( properties )
{
boolean_t match = FALSE;
IOServiceMatchPropertyTable( DADiskGetIOMedia( disk ), properties, &match );
if ( match )
{
/*
* We have found a probe candidate for this media object.
*/
kind = DAFileSystemGetKind( filesystem );
DADiskSetFileSystem( disk, filesystem );
if ( CFDictionaryGetValue( candidate, CFSTR( "autodiskmount" ) ) == kCFBooleanFalse )
{
DADiskSetOption( disk, kDADiskOptionMountAutomatic, FALSE );
DADiskSetOption( disk, kDADiskOptionMountAutomaticNoDefer, FALSE );
}
CFArrayRemoveValueAtIndex( candidates, 0 );
DALogDebug( " probed disk, id = %@, with %@, ongoing.", disk, kind );
DAFileSystemProbe( filesystem, DADiskGetDevice( disk ), __DAStageProbeCallback, context );
return;
}
}
}
}
CFArrayRemoveValueAtIndex( candidates, 0 );
}
}
DADiskSetState( disk, kDADiskStateRequireRepair, FALSE );
DADiskSetState( disk, kDADiskStateRequireRepairQuotas, FALSE );
if ( status )
{
/*
* We have found no probe match for this media object.
*/
kind = NULL;
if ( DADiskGetFileSystem( disk ) )
{
DADiskSetFileSystem( disk, NULL );
DALogDebug( " probed disk, id = %@, no match.", disk );
}
}
else
{
/*
* We have found a probe match for this media object.
*/
kind = DAFileSystemGetKind( DADiskGetFileSystem( disk ) );
///w:start
if ( DADiskGetDescription( disk, kDADiskDescriptionMediaWritableKey ) == kCFBooleanFalse )
{
clean = kCFBooleanTrue;
}
///w:stop
if ( clean == kCFBooleanFalse )
{
DADiskSetState( disk, kDADiskStateRequireRepair, TRUE );
DADiskSetState( disk, kDADiskStateRequireRepairQuotas, TRUE );
}
DALogDebug( " probed disk, id = %@, with %@, success.", disk, kind );
}
keys = CFArrayCreateMutable( kCFAllocatorDefault, 0, &kCFTypeArrayCallBacks );
if ( keys )
{
CFTypeRef object;
object = kind ? kCFBooleanTrue : kCFBooleanFalse;
if ( DADiskCompareDescription( disk, kDADiskDescriptionVolumeMountableKey, object ) )
{
DADiskSetDescription( disk, kDADiskDescriptionVolumeMountableKey, object );
CFArrayAppendValue( keys, kDADiskDescriptionVolumeMountableKey );
}
if ( DADiskCompareDescription( disk, kDADiskDescriptionVolumeKindKey, kind ) )
{
DADiskSetDescription( disk, kDADiskDescriptionVolumeKindKey, kind );
CFArrayAppendValue( keys, kDADiskDescriptionVolumeKindKey );
}
if ( DADiskCompareDescription( disk, kDADiskDescriptionVolumeNameKey, name ) )
{
DADiskSetDescription( disk, kDADiskDescriptionVolumeNameKey, name );
CFArrayAppendValue( keys, kDADiskDescriptionVolumeNameKey );
}
if ( DADiskCompareDescription( disk, kDADiskDescriptionVolumeUUIDKey, uuid ) )
{
DADiskSetDescription( disk, kDADiskDescriptionVolumeUUIDKey, uuid );
CFArrayAppendValue( keys, kDADiskDescriptionVolumeUUIDKey );
}
if ( CFArrayGetCount( keys ) )
{
if ( DADiskGetState( disk, kDADiskStateStagedAppear ) )
{
DADiskDescriptionChangedCallback( disk, keys );
}
}
CFRelease( keys );
}
if ( DADiskGetState( disk, kDADiskStateStagedMount ) == FALSE )
{
struct statfs * mountList;
int mountListCount;
int mountListIndex;
/*
* Determine whether the disk is mounted.
*/
mountListCount = getmntinfo( &mountList, MNT_NOWAIT );
for ( mountListIndex = 0; mountListIndex < mountListCount; mountListIndex++ )
{
if ( mountList[mountListIndex].f_fsid.val[0] == DADiskGetBSDNode( disk ) )
{
/*
* We have determined that the disk is mounted.
*/
CFURLRef path;
path = CFURLCreateFromFileSystemRepresentation( kCFAllocatorDefault,
( void * ) mountList[mountListIndex].f_mntonname,
strlen( mountList[mountListIndex].f_mntonname ),
TRUE );
if ( path )
{
_DAMountCreateTrashFolder( disk, path );
DADiskSetBypath( disk, path );
DADiskSetDescription( disk, kDADiskDescriptionVolumePathKey, path );
CFRelease( path );
}
if ( strcmp( mountList[mountListIndex].f_mntonname, "/" ) == 0 )
{
path = DAMountCreateMountPointWithAction( disk, kDAMountPointActionLink );
if ( path )
{
DADiskSetBypath( disk, path );
CFRelease( path );
}
DADiskSetOption( disk, kDADiskOptionMountAutomatic, TRUE );
DADiskSetOption( disk, kDADiskOptionMountAutomaticNoDefer, TRUE );
}
DADiskSetState( disk, kDADiskStateRequireRepair, FALSE );
DADiskSetState( disk, kDADiskStateRequireRepairQuotas, FALSE );
break;
}
}
}
DAUnitSetState( disk, kDAUnitStateCommandActive, FALSE );
DADiskSetState( disk, kDADiskStateCommandActive, FALSE );
DADiskSetContext( disk, NULL );
DAStageSignal( );
CFRelease( disk );
}
static void __DAMountWithArgumentsCallbackStage1( int status, void * parameter )
{
/*
* Process the repair command's completion.
*/
__DAMountCallbackContext * context = parameter;
DALogDebugHeader( "%s -> %s", gDAProcessNameID, gDAProcessNameID );
if ( context->assertionID != kIOPMNullAssertionID )
{
IOPMAssertionRelease( context->assertionID );
context->assertionID = kIOPMNullAssertionID;
}
if ( status )
{
/*
* We were unable to repair the volume.
*/
if ( status == ECANCELED )
{
status = 0;
}
else
{
DALogDebug( " repaired disk, id = %@, failure.", context->disk );
DALogError( "unable to repair %@ (status code 0x%08X).", context->disk, status );
if ( context->force )
{
status = 0;
}
else
{
__DAMountWithArgumentsCallback( ___EDIRTY, context );
}
}
}
else
{
/*
* We were able to repair the volume.
*/
DADiskSetState( context->disk, kDADiskStateRequireRepair, FALSE );
DALogDebug( " repaired disk, id = %@, success.", context->disk );
}
/*
* Mount the volume.
*/
if ( status == 0 )
{
/*
* Create the mount point, in case one needs to be created.
*/
if ( context->mountpoint == NULL )
{
context->mountpoint = DAMountCreateMountPointWithAction( context->disk, kDAMountPointActionMake );
}
/*
* Execute the mount command.
*/
if ( context->mountpoint )
{
DALogDebug( " mounted disk, id = %@, ongoing.", context->disk );
DAFileSystemMountWithArguments( DADiskGetFileSystem( context->disk ),
DADiskGetDevice( context->disk ),
context->mountpoint,
DADiskGetUserUID( context->disk ),
DADiskGetUserGID( context->disk ),
__DAMountWithArgumentsCallbackStage2,
context,
context->options,
NULL );
}
else
{
__DAMountWithArgumentsCallback( ENOSPC, context );
}
}
}
CFURLRef DAMountCreateMountPoint( DADiskRef disk )
{
return DAMountCreateMountPointWithAction( disk, kDAMountPointActionMake );
}