TSystemUtils.cpp   [plain text]

/*
 * Copyright (c) 2003-2008 Apple Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 * 
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 * 
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 * 
 * @APPLE_LICENSE_HEADER_END@
 */


//-----------------------------------------------------------------------------
//	Includes
//-----------------------------------------------------------------------------

#define DEBUG_ASSERT_COMPONENT_NAME_STRING 					"TSystemUtils"
#include <AssertMacros.h>

#include "TSystemUtils.h"

#include <unistd.h>
#include <sys/stat.h>
#include <SystemConfiguration/SystemConfiguration.h>


//-----------------------------------------------------------------------------
//	Macros
//-----------------------------------------------------------------------------

#define DEBUG	0

#define kAppleLanguagesString	"AppleLanguages"
#define kEmptyString			""


//-----------------------------------------------------------------------------
// GetPreferredLanguages - Gets preferred languages of console user.
//															 [STATIC][PUBLIC]
//-----------------------------------------------------------------------------

CFArrayRef
TSystemUtils::GetPreferredLanguages ( void )
{
	
	CFArrayRef			preferredLanguages	= NULL;
	CFPropertyListRef	languages			= NULL;
	CFStringRef			userName			= NULL;
	CFComparisonResult	equal				= kCFCompareEqualTo;
	uid_t				uid					= 0;

	uid = FindUIDToUse ( );
	seteuid ( uid );
	
	userName = ::CFCopyUserName ( );
	require ( ( userName != NULL ), ErrorExit );
	
	equal = ::CFStringCompare ( userName, CFSTR ( kEmptyString ), 0 );
	require ( ( equal != kCFCompareEqualTo ), ReleaseExit );
	
	languages = ::CFPreferencesCopyValue ( CFSTR ( kAppleLanguagesString ),
										   kCFPreferencesAnyApplication,
										   userName,
										   kCFPreferencesAnyHost );
	
	require ( ( languages != NULL ), ReleaseExit );
	require_action ( ( ::CFGetTypeID ( languages ) == ::CFArrayGetTypeID ( ) ),
					 ReleaseExit,
					 ::CFRelease ( languages ) );
	
	preferredLanguages = ( CFArrayRef ) languages;


ReleaseExit:

	
    ::CFRelease ( userName );


ErrorExit:
	
    
	seteuid ( 0 );
	
	return preferredLanguages;
	
}


//-----------------------------------------------------------------------------
// FindUIDToUse - Finds the UID of the console user.		 [STATIC][PUBLIC]
//-----------------------------------------------------------------------------

uid_t
TSystemUtils::FindUIDToUse ( void )
{
	
	uid_t 				uid			= 0;
	gid_t				gid			= 0;
	CFStringRef			userName	= NULL;
	SCDynamicStoreRef	storeRef	= NULL;

	storeRef = ::SCDynamicStoreCreate ( kCFAllocatorDefault,
										CFSTR ( "cddafs.util" ),
										NULL,
										NULL );
	require ( ( storeRef != NULL ), ErrorExit );

	userName = ::SCDynamicStoreCopyConsoleUser ( storeRef,
												 &uid,
												 &gid );
	require ( ( userName != NULL ), ReleaseDynamicStore );
	::CFRelease ( userName );
	
	
ReleaseDynamicStore:
	
	
	require_quiet ( ( storeRef != NULL ), ErrorExit );
	::CFRelease ( storeRef );
	storeRef = NULL;
	
	
ErrorExit:
	
	
	return uid;
	
}


//-----------------------------------------------------------------------------
// ReadDataFromURL - Reads data given a file URL.		 [STATIC][PUBLIC]
//-----------------------------------------------------------------------------

CFDataRef
TSystemUtils::ReadDataFromURL ( CFURLRef url )
{

    CFMutableDataRef    data            = NULL;
    Boolean				result			= false;
	CFNumberRef         fileSizeNumber  = NULL;
    CFIndex             fileSize        = 0;
    UInt8 *             dataPtr         = NULL;
    UInt8 *             endPtr          = NULL;
    CFReadStreamRef     readStream      = NULL;
    CFIndex             bytesRead       = 0;
    
    result = CFURLCopyResourcePropertyForKey ( url, kCFURLFileSizeKey, &fileSizeNumber, NULL );
    require ( result, ErrorExit );
    
    result = CFNumberGetValue ( fileSizeNumber, kCFNumberCFIndexType, &fileSize );
    require ( result, ReleaseNumber );
    
    data = CFDataCreateMutable ( kCFAllocatorDefault, fileSize );
    require ( data, ReleaseNumber );
    
    CFDataSetLength ( data, fileSize );
    
    dataPtr = CFDataGetMutableBytePtr ( data );
    require ( dataPtr, ReleaseNumber );
    
    readStream = CFReadStreamCreateWithFile ( kCFAllocatorDefault, url );
    require ( readStream, ErrorExit );
    
    result = CFReadStreamOpen ( readStream );
    require ( result, ReleaseStream );
    
    endPtr  = ( UInt8 * ) dataPtr + fileSize;
    
    while ( dataPtr < endPtr )
    {

        bytesRead = CFReadStreamRead ( readStream, dataPtr, endPtr - dataPtr );

        if ( bytesRead > 0 )
        {
            
            dataPtr += bytesRead;
            
        }
        
    }
    
    CFReadStreamClose ( readStream );
    
    
ReleaseStream:
    
    
    CFRelease ( readStream );
    readStream = NULL;
    
    
ReleaseNumber:
    
    
    CFRelease ( fileSizeNumber );
    fileSizeNumber = NULL;
    
    
ErrorExit:
    
    
    return data;
    
}

//-----------------------------------------------------------------------------
//					End				Of			File
//-----------------------------------------------------------------------------