filespecutils.cpp

来自「symbian 下的helix player源代码」· C++ 代码 · 共 724 行 · 第 1/2 页

/* ***** BEGIN LICENSE BLOCK *****
 * Source last modified: $Id: filespecutils.cpp,v 1.2.42.5 2004/07/09 01:44:27 hubbe Exp $
 * 
 * Portions Copyright (c) 1995-2004 RealNetworks, Inc. All Rights Reserved.
 * 
 * The contents of this file, and the files included with this file,
 * are subject to the current version of the RealNetworks Public
 * Source License (the "RPSL") available at
 * http://www.helixcommunity.org/content/rpsl unless you have licensed
 * the file under the current version of the RealNetworks Community
 * Source License (the "RCSL") available at
 * http://www.helixcommunity.org/content/rcsl, in which case the RCSL
 * will apply. You may also obtain the license terms directly from
 * RealNetworks.  You may not use this file except in compliance with
 * the RPSL or, if you have a valid RCSL with RealNetworks applicable
 * to this file, the RCSL.  Please see the applicable RPSL or RCSL for
 * the rights, obligations and limitations governing use of the
 * contents of the file.
 * 
 * Alternatively, the contents of this file may be used under the
 * terms of the GNU General Public License Version 2 or later (the
 * "GPL") in which case the provisions of the GPL are applicable
 * instead of those above. If you wish to allow use of your version of
 * this file only under the terms of the GPL, and not to allow others
 * to use your version of this file under the terms of either the RPSL
 * or RCSL, indicate your decision by deleting the provisions above
 * and replace them with the notice and other provisions required by
 * the GPL. If you do not delete the provisions above, a recipient may
 * use your version of this file under the terms of any one of the
 * RPSL, the RCSL or the GPL.
 * 
 * This file is part of the Helix DNA Technology. RealNetworks is the
 * developer of the Original Code and owns the copyrights in the
 * portions it created.
 * 
 * This file, and the files included with this file, is distributed
 * and made available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY
 * KIND, EITHER EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS
 * ALL SUCH WARRANTIES, INCLUDING WITHOUT LIMITATION, ANY WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, QUIET
 * ENJOYMENT OR NON-INFRINGEMENT.
 * 
 * Technology Compatibility Kit Test Suite(s) Location:
 *    http://www.helixcommunity.org/content/tck
 * 
 * Contributor(s):
 * 
 * ***** END LICENSE BLOCK ***** */

#include "hxstring.h"
#include "filespecutils.h"
#include "hxtick.h"
#include "hxrand.h"
#include "dbcs.h"

#include "hlxclib/windows.h"
#include <winbase.h>
#include <shlobj.h>
#include "hlxclib/sys/stat.h"


//////////////////////////////////////////////////////////
//
// Utility base class -- CHXFileSpecUtils
//
//////////////////////////////////////////////////////////

typedef BOOL (HXEXPORT_PTR FPGetFreeSpace) (LPCTSTR lpDir,PULARGE_INTEGER lpFree,PULARGE_INTEGER lpTotBytes,PULARGE_INTEGER lpTotFree);

typedef HRESULT (HXEXPORT_PTR FPSHGetSpecialFolderLocation) (HWND hwndOwner, int nFolder, LPITEMIDLIST *ppidl);  
typedef HRESULT (HXEXPORT_PTR FPSHGetPathFromIDList)(LPCITEMIDLIST pidl, LPSTR pszPath);
typedef	HRESULT (HXEXPORT_PTR FPSHGetMalloc)(LPMALLOC *ppMalloc);

//******************************************************************************
HX_RESULT CHXFileSpecUtils::GetFreeSpaceOnDisk(const CHXDirSpecifier& volSpec, 
					       INT64& freeSpace)
{
	CHXDirSpecifier dirRoot = volSpec.GetVolume();
	CHXString strDir = dirRoot.GetPathName();
	OSVERSIONINFO ver;
	HX_RESULT res = HXR_FAIL;
	WORD w;

	// figure out which version of windows the user is running 
	ver.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
	GetVersionEx(&ver);
	w = (WORD)ver.dwBuildNumber;
	if (ver.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS && w <= 1000)
	{
#if !defined(WIN32_PLATFORM_PSPC)
		// the original windows 95
		DWORD dwSecPerClu,dwBytPerSec,dwNumFreeClu,dwNumOfClu;
		if (GetDiskFreeSpace(strDir,&dwSecPerClu,&dwBytPerSec,&dwNumFreeClu,&dwNumOfClu) != 0)
		{
			return HXR_FAIL;
		}

		// do a bit a math to figure out the free space - 
		freeSpace = ((INT64)dwSecPerClu) * ((INT64)dwBytPerSec) * ((INT64)dwNumFreeClu);

#else /* !defined(WIN32_PLATFORM_PSPC) */
		ULARGE_INTEGER   freeBytesAvailableToCaller;
		ULARGE_INTEGER   totalNumberOfBytes;
		ULARGE_INTEGER   totalNumberOfFreeBytes;
		if (!GetDiskFreeSpaceEx(OS_STRING(strDir),
					&freeBytesAvailableToCaller,
					&totalNumberOfBytes,
					&totalNumberOfFreeBytes))
		    return HXR_FAIL;
		freeSpace = freeBytesAvailableToCaller.QuadPart;

#endif /* !defined(WIN32_PLATFORM_PSPC) */
	}
	else
	{
		// windows 95 osr2 or above or winnt
		// the docs for GetDiskFreeSpaceEx say to load kernel32.dll,
		// search for the procaddress and then call the function...

		HINSTANCE hKernel = ::LoadLibrary(OS_STRING("kernel32.dll"));
		if (hKernel)
		{
			FPGetFreeSpace fpGetFreeSpace = (FPGetFreeSpace) GetProcAddress(hKernel, OS_STRING("GetDiskFreeSpaceExA"));
			if (fpGetFreeSpace)
			{
				ULARGE_INTEGER uFreeBytesCaller,uTotNumBytes,uTotNumFreeBytes;

				BOOL b = fpGetFreeSpace(OS_STRING(strDir),&uFreeBytesCaller,&uTotNumBytes,&uTotNumFreeBytes);
				if (b)
				{
					// free space on disk or for the user? That is the question
					freeSpace = uFreeBytesCaller.QuadPart;
					res = HXR_OK;
				}
			}

			::FreeLibrary(hKernel);
		}
	}

	return res;
}

//******************************************************************************
HX_RESULT CHXFileSpecUtils::GetTotalSpaceOnDisk(const CHXDirSpecifier& volSpec, 
						INT64& totalSpace)
{ 
	CHXDirSpecifier dirRoot = volSpec.GetVolume();
	CHXString strDir = dirRoot.GetPathName();
	OSVERSIONINFO ver;
	HX_RESULT res = HXR_FAIL;
	WORD w;

	// figure out which version of windows the user is running 
	ver.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
	GetVersionEx(&ver);
	w = (WORD)ver.dwBuildNumber;
	if (ver.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS && w <= 1000)
	{
#if !defined(WIN32_PLATFORM_PSPC)
		// the original windows 95
		DWORD dwSecPerClu,dwBytPerSec,dwNumFreeClu,dwNumOfClu;
		if (GetDiskFreeSpace(strDir,&dwSecPerClu,&dwBytPerSec,&dwNumFreeClu,&dwNumOfClu) != 0)
		{
			return HXR_FAIL;
		}

		// do a bit a math to figure out the free space - 
		totalSpace = ((INT64)dwSecPerClu) * ((INT64)dwBytPerSec) * ((INT64)dwNumOfClu);
#else /* !defined(WIN32_PLATFORM_PSPC) */
		ULARGE_INTEGER   freeBytesAvailableToCaller;
		ULARGE_INTEGER   totalNumberOfBytes;
		ULARGE_INTEGER   totalNumberOfFreeBytes;
		if (!GetDiskFreeSpaceEx(OS_STRING(strDir),
					&freeBytesAvailableToCaller,
					&totalNumberOfBytes,
					&totalNumberOfFreeBytes))
		    return HXR_FAIL;
		totalSpace = totalNumberOfBytes.QuadPart;
#endif /* !defined(WIN32_PLATFORM_PSPC) */
	}
	else
	{
		// windows 95 osr2 or above or winnt
		// the docs for GetDiskFreeSpaceEx say to load kernel32.dll,
		// search for the procaddress and then call the function...

		HINSTANCE hKernel = ::LoadLibrary(OS_STRING("kernel32.dll"));
		if (hKernel)
		{
			FPGetFreeSpace fpGetFreeSpace = (FPGetFreeSpace)GetProcAddress(hKernel, OS_STRING("GetDiskFreeSpaceExA"));
			if (fpGetFreeSpace)
			{
				ULARGE_INTEGER uFreeBytesCaller,uTotNumBytes,uTotNumFreeBytes;

				BOOL b = fpGetFreeSpace(OS_STRING(strDir),&uFreeBytesCaller,&uTotNumBytes,&uTotNumFreeBytes);
				if (b)
				{
					// free space on disk or for the user? That is the question
					totalSpace = uTotNumBytes.QuadPart;
					res = HXR_OK;
				}
			}

			::FreeLibrary(hKernel);
		}
	}

	return res;
}

//******************************************************************************
BOOL CHXFileSpecUtils::IsDiskEjectable(const CHXDirSpecifier& volSpec)
{
    	BOOL retval = FALSE;

#if !defined(WIN32_PLATFORM_PSPC)
	CHXDirSpecifier dirRoot = volSpec.GetVolume();
	CHXString strDir = dirRoot.GetPathName();
	UINT uiType = ::GetDriveType(strDir);


	switch (uiType)
	{
	case DRIVE_REMOVABLE:
	case DRIVE_CDROM:
		retval = TRUE;
		break;
	default:
		retval = FALSE;
		break;
	}
#endif /* !defined(WIN32_PLATFORM_PSPC) */

	return retval;
}

// IsLocal returns TRUE if the file or directory is on a local volume
// (not on a server)
//******************************************************************************
BOOL CHXFileSpecUtils::IsDiskLocal(const CHXDirSpecifier& volSpec)
{
    BOOL retval = TRUE;

#if !defined(WIN32_PLATFORM_PSPC)
	CHXDirSpecifier dirRoot = volSpec.GetVolume();
	CHXString strDir = dirRoot.GetPathName();
	UINT uiType = ::GetDriveType(strDir);

	switch (uiType)
	{
	case DRIVE_FIXED:
	case DRIVE_CDROM:
	case DRIVE_RAMDISK:
	case DRIVE_REMOVABLE:
		retval = TRUE;
		break;
	default:
		retval = FALSE;
		break;
	}
#endif /* !defined(WIN32_PLATFORM_PSPC) */

	return retval;
}

// file/dir utilities
//******************************************************************** **********
HX_RESULT CHXFileSpecUtils::RemoveDir(const CHXDirSpecifier& dirSpec)
{
    if (::RemoveDirectory(OS_STRING(dirSpec.GetPathName())))
    {
        return HXR_OK;
    }

    return HXR_FAILED;
};


//******************************************************************** **********
HX_RESULT CHXFileSpecUtils::RemoveFile(const CHXFileSpecifier& fileSpec)
{
    if (::DeleteFile(OS_STRING(fileSpec.GetPathName())))
    {
        return HXR_OK;
    }

    return HXR_FAILED;
};


//******************************************************************************
HX_RESULT CHXFileSpecUtils::GetFileSize(const CHXFileSpecifier& fileSpec, INT64& fSize)
{
	CHXString strFile = fileSpec.GetPathName();

	// try to open the file
	HANDLE hFile = ::CreateFile(OS_STRING(strFile),
								GENERIC_READ,
								FILE_SHARE_READ,
								NULL,
								OPEN_EXISTING,
								FILE_ATTRIBUTE_NORMAL,
								NULL);
	if (hFile == INVALID_HANDLE_VALUE)
	{
		return HXR_FAIL;
	}

	DWORD dwLow,dwHigh,dwError;

	dwLow = ::GetFileSize(hFile,&dwHigh);
	if (dwLow == 0xFFFFFFFF && (dwError = GetLastError()) != NO_ERROR)
	{
		return HXR_FAIL;
	}

	fSize = dwHigh;
	fSize = fSize << 32;
	fSize |= dwLow;

	::CloseHandle(hFile);

	return HXR_OK;
}

//******************************************************************************
HX_RESULT CHXFileSpecUtils::GetDirectorySize(const CHXDirSpecifier& dirSpec, BOOL shouldDescend, INT64& fSize)
{
    CHXString currentDir = dirSpec.GetPathName();
    CHXString strFindFilePath = currentDir + "\\*.*";

    WIN32_FIND_DATA findFileData;
    HANDLE hFile = ::FindFirstFile(OS_STRING(strFindFilePath), &findFileData );

    HX_RESULT outResult = HXR_OK;

    while ( hFile != INVALID_HANDLE_VALUE )
    {
	if ((0 != strcmp(".", OS_STRING(findFileData.cFileName))) && 
	    (0 != strcmp("..", OS_STRING(findFileData.cFileName))) )
	{
	    if ( shouldDescend && (findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) )
	    {
		CHXString subDirectoryPath = currentDir + "\\" + OS_STRING(findFileData.cFileName);

		CHXDirSpecifier subDirectorySpec( subDirectoryPath );
		if ( FAILED(GetDirectorySize( subDirectorySpec, shouldDescend, fSize )) )
		{
		    fSize = 0;
		    outResult = HXR_FAIL;
		    break;
		}
	    }
	    else
	    {
		INT64 dFileSize = 0;
		dFileSize = findFileData.nFileSizeHigh;
		dFileSize = dFileSize << 32;
		dFileSize |= findFileData.nFileSizeLow;
		fSize += dFileSize;
	    }