// Windows/FileFind.cpp

#include "StdAfx.h"

#include "FileFind.h"

#ifndef _UNICODE

#include "../Common/StringConvert.h"

#endif

#ifndef _UNICODE

extern bool g_IsNT;

#endif

namespace NWindows {

namespace NFile {

#if defined(WIN_LONG_PATH) && defined(_UNICODE)

#define WIN_LONG_PATH2

#endif

bool GetLongPath(LPCWSTR fileName, UString &res);

namespace NFind {

static const TCHAR kDot = TEXT('.');

bool CFileInfo::IsDots() const

{

if (!IsDirectory() || Name.IsEmpty())

return false;

if (Name[0] != kDot)

return false;

return Name.Length() == 1 || (Name[1] == kDot && Name.Length() == 2);

}

#ifndef _UNICODE

bool CFileInfoW::IsDots() const

{

if (!IsDirectory() || Name.IsEmpty())

return false;

if (Name[0] != kDot)

return false;

return Name.Length() == 1 || (Name[1] == kDot && Name.Length() == 2);

}

#endif

static void ConvertWIN32_FIND_DATA_To_FileInfo(

const WIN32_FIND_DATA &findData,

CFileInfo &fileInfo)

{

fileInfo.Attributes = findData.dwFileAttributes;

fileInfo.CreationTime = findData.ftCreationTime;

fileInfo.LastAccessTime = findData.ftLastAccessTime;

fileInfo.LastWriteTime = findData.ftLastWriteTime;

fileInfo.Size = (((UInt64)findData.nFileSizeHigh) << 32) + findData.nFileSizeLow;

fileInfo.Name = findData.cFileName;

#ifndef _WIN32_WCE

fileInfo.ReparseTag = findData.dwReserved0;

#else

fileInfo.ObjectID = findData.dwOID;

#endif

}

#ifndef _UNICODE

static inline UINT GetCurrentCodePage() { return ::AreFileApisANSI() ? CP_ACP : CP_OEMCP; }

static void ConvertWIN32_FIND_DATA_To_FileInfo(

const WIN32_FIND_DATAW &findData,

CFileInfoW &fileInfo)

{

fileInfo.Attributes = findData.dwFileAttributes;

fileInfo.CreationTime = findData.ftCreationTime;

fileInfo.LastAccessTime = findData.ftLastAccessTime;

fileInfo.LastWriteTime = findData.ftLastWriteTime;

fileInfo.Size = (((UInt64)findData.nFileSizeHigh) << 32) + findData.nFileSizeLow;

fileInfo.Name = findData.cFileName;

#ifndef _WIN32_WCE

fileInfo.ReparseTag = findData.dwReserved0;

#else

fileInfo.ObjectID = findData.dwOID;

#endif

}

static void ConvertWIN32_FIND_DATA_To_FileInfo(

const WIN32_FIND_DATA &findData,

CFileInfoW &fileInfo)

{

fileInfo.Attributes = findData.dwFileAttributes;

fileInfo.CreationTime = findData.ftCreationTime;

fileInfo.LastAccessTime = findData.ftLastAccessTime;

fileInfo.LastWriteTime = findData.ftLastWriteTime;

fileInfo.Size = (((UInt64)findData.nFileSizeHigh) << 32) + findData.nFileSizeLow;

fileInfo.Name = GetUnicodeString(findData.cFileName, GetCurrentCodePage());

#ifndef _WIN32_WCE

fileInfo.ReparseTag = findData.dwReserved0;

#else

fileInfo.ObjectID = findData.dwOID;

#endif

}

#endif

////////////////////////////////

// CFindFile

bool CFindFile::Close()

{

if (_handle == INVALID_HANDLE_VALUE)

return true;

if (!::FindClose(_handle))

return false;

_handle = INVALID_HANDLE_VALUE;

return true;

}

bool CFindFile::FindFirst(LPCTSTR wildcard, CFileInfo &fileInfo)

{

if (!Close())

return false;

WIN32_FIND_DATA findData;

_handle = ::FindFirstFile(wildcard, &findData);

#ifdef WIN_LONG_PATH2

if (_handle == INVALID_HANDLE_VALUE)

{

UString longPath;

if (GetLongPath(wildcard, longPath))

_handle = ::FindFirstFileW(longPath, &findData);

}

#endif

if (_handle == INVALID_HANDLE_VALUE)

return false;

ConvertWIN32_FIND_DATA_To_FileInfo(findData, fileInfo);

return true;

}

#ifndef _UNICODE

bool CFindFile::FindFirst(LPCWSTR wildcard, CFileInfoW &fileInfo)

{

if (!Close())

return false;

if (g_IsNT)

{

WIN32_FIND_DATAW findData;

_handle = ::FindFirstFileW(wildcard, &findData);

#ifdef WIN_LONG_PATH

if (_handle == INVALID_HANDLE_VALUE)

{

UString longPath;

if (GetLongPath(wildcard, longPath))

_handle = ::FindFirstFileW(longPath, &findData);

}

#endif

if (_handle != INVALID_HANDLE_VALUE)

ConvertWIN32_FIND_DATA_To_FileInfo(findData, fileInfo);

}

else

{

WIN32_FIND_DATAA findData;

_handle = ::FindFirstFileA(UnicodeStringToMultiByte(wildcard,

GetCurrentCodePage()), &findData);

if (_handle != INVALID_HANDLE_VALUE)

ConvertWIN32_FIND_DATA_To_FileInfo(findData, fileInfo);

}

return (_handle != INVALID_HANDLE_VALUE);

}

#endif

bool CFindFile::FindNext(CFileInfo &fileInfo)

{

WIN32_FIND_DATA findData;

bool result = BOOLToBool(::FindNextFile(_handle, &findData));

if (result)

ConvertWIN32_FIND_DATA_To_FileInfo(findData, fileInfo);

return result;

}

#ifndef _UNICODE

bool CFindFile::FindNext(CFileInfoW &fileInfo)

{

if (g_IsNT)

{

WIN32_FIND_DATAW findData;

if (!::FindNextFileW(_handle, &findData))

return false;

ConvertWIN32_FIND_DATA_To_FileInfo(findData, fileInfo);

}

else

{

WIN32_FIND_DATAA findData;

if (!::FindNextFileA(_handle, &findData))

return false;

ConvertWIN32_FIND_DATA_To_FileInfo(findData, fileInfo);

}

return true;

}

#endif

bool FindFile(LPCTSTR wildcard, CFileInfo &fileInfo)

{

CFindFile finder;

return finder.FindFirst(wildcard, fileInfo);

}

#ifndef _UNICODE

bool FindFile(LPCWSTR wildcard, CFileInfoW &fileInfo)

{

CFindFile finder;

return finder.FindFirst(wildcard, fileInfo);

}

#endif

bool DoesFileExist(LPCTSTR name)

{

CFileInfo fileInfo;

return FindFile(name, fileInfo);

}

#ifndef _UNICODE

bool DoesFileExist(LPCWSTR name)

{

CFileInfoW fileInfo;

return FindFile(name, fileInfo);

}

#endif

/////////////////////////////////////

// CEnumerator

bool CEnumerator::NextAny(CFileInfo &fileInfo)

{

if (_findFile.IsHandleAllocated())

return _findFile.FindNext(fileInfo);

else

return _findFile.FindFirst(_wildcard, fileInfo);

}

bool CEnumerator::Next(CFileInfo &fileInfo)

{

for (;;)

{

if (!NextAny(fileInfo))

return false;

if (!fileInfo.IsDots())

return true;

}

}

bool CEnumerator::Next(CFileInfo &fileInfo, bool &found)

{

if (Next(fileInfo))

{

found = true;

return true;

}

found = false;

return (::GetLastError() == ERROR_NO_MORE_FILES);

}

#ifndef _UNICODE

bool CEnumeratorW::NextAny(CFileInfoW &fileInfo)

{

if (_findFile.IsHandleAllocated())

return _findFile.FindNext(fileInfo);

else

return _findFile.FindFirst(_wildcard, fileInfo);

}

bool CEnumeratorW::Next(CFileInfoW &fileInfo)

{

for (;;)

{

if (!NextAny(fileInfo))

return false;

if (!fileInfo.IsDots())

return true;

}

}

bool CEnumeratorW::Next(CFileInfoW &fileInfo, bool &found)

{

if (Next(fileInfo))

{

found = true;

return true;

}

found = false;

return (::GetLastError() == ERROR_NO_MORE_FILES);

}

#endif

////////////////////////////////

// CFindChangeNotification

// FindFirstChangeNotification can return 0. MSDN doesn't tell about it.

bool CFindChangeNotification::Close()

{

if (!IsHandleAllocated())

return true;

if (!::FindCloseChangeNotification(_handle))

return false;

_handle = INVALID_HANDLE_VALUE;

return true;

}

HANDLE CFindChangeNotification::FindFirst(LPCTSTR pathName, bool watchSubtree, DWORD notifyFilter)

{

_handle = ::FindFirstChangeNotification(pathName, BoolToBOOL(watchSubtree), notifyFilter);

#ifdef WIN_LONG_PATH2

if (!IsHandleAllocated())

{

UString longPath;

if (GetLongPath(pathName, longPath))

_handle = ::FindFirstChangeNotificationW(longPath, BoolToBOOL(watchSubtree), notifyFilter);

}

#endif

return _handle;

}

#ifndef _UNICODE

HANDLE CFindChangeNotification::FindFirst(LPCWSTR pathName, bool watchSubtree, DWORD notifyFilter)

{

if (!g_IsNT)

return FindFirst(UnicodeStringToMultiByte(pathName, GetCurrentCodePage()), watchSubtree, notifyFilter);

_handle = ::FindFirstChangeNotificationW(pathName, BoolToBOOL(watchSubtree), notifyFilter);

#ifdef WIN_LONG_PATH

if (!IsHandleAllocated())

{

UString longPath;

if (GetLongPath(pathName, longPath))

_handle = ::FindFirstChangeNotificationW(longPath, BoolToBOOL(watchSubtree), notifyFilter);

}

#endif

return _handle;

}

#endif

#ifndef _WIN32_WCE

bool MyGetLogicalDriveStrings(CSysStringVector &driveStrings)

{

driveStrings.Clear();

UINT32 size = GetLogicalDriveStrings(0, NULL);

if (size == 0)

return false;

CSysString buffer;

UINT32 newSize = GetLogicalDriveStrings(size, buffer.GetBuffer(size));

if (newSize == 0)

return false;

if (newSize > size)

return false;

CSysString string;

for(UINT32 i = 0; i < newSize; i++)

{

TCHAR c = buffer[i];

if (c == TEXT('\0'))

{

driveStrings.Add(string);

string.Empty();

}

else

string += c;

}

if (!string.IsEmpty())

return false;

return true;

}

#ifndef _UNICODE

bool MyGetLogicalDriveStrings(UStringVector &driveStrings)

{

driveStrings.Clear();

if (g_IsNT)

{

UINT32 size = GetLogicalDriveStringsW(0, NULL);

if (size == 0)

return false;

UString buffer;

UINT32 newSize = GetLogicalDriveStringsW(size, buffer.GetBuffer(size));

if (newSize == 0)

return false;

if (newSize > size)

return false;

UString string;

for(UINT32 i = 0; i < newSize; i++)

{

WCHAR c = buffer[i];

if (c == L'\0')

{

driveStrings.Add(string);

string.Empty();

}

else

string += c;

}

return string.IsEmpty();

}

CSysStringVector driveStringsA;

bool res = MyGetLogicalDriveStrings(driveStringsA);

for (int i = 0; i < driveStringsA.Size(); i++)

driveStrings.Add(GetUnicodeString(driveStringsA[i]));

return res;

}

#endif

#endif

}}}