fixed possible uninitialised memory read in folderiterator for windows, and changed file folderiterator.cc to unix format

This commit is contained in:
cyril soler 2016-11-14 14:10:49 +01:00
parent fb302002d6
commit d433713bd0
2 changed files with 210 additions and 181 deletions

View File

@ -19,6 +19,13 @@ namespace librs { namespace util {
FolderIterator::FolderIterator(const std::string& folderName)
: mFolderName(folderName)
{
is_open = false ;
validity = false ;
mFileModTime = 0;
mFolderModTime = 0;
mFileSize = 0;
mType = TYPE_UNKNOWN;
// Grab the last modification time for the directory
struct stat64 buf ;
@ -45,9 +52,20 @@ FolderIterator::FolderIterator(const std::string& folderName)
utf16Name += L"/*.*";
// FindFirstFileW does the "next" operation, so after calling it, we need to read the information
// for the current entry.
handle = FindFirstFileW(utf16Name.c_str(), &fileInfo);
is_open = validity = handle != INVALID_HANDLE_VALUE;
bool should_skip ;
validity = validity && updateFileInfo(should_skip);
if(validity && should_skip)
next();
#else
// On linux, we need to call "next()" once the dir is openned. next() will call updateFileInfo() itself.
handle = opendir(folderName.c_str());
is_open = validity = handle != NULL;
next();
@ -61,8 +79,25 @@ FolderIterator::~FolderIterator()
void FolderIterator::next()
{
bool should_skip = false ;
while(readdir())
if(updateFileInfo(should_skip) && !should_skip)
return ;
#ifdef DEBUG_FOLDER_ITERATOR
std::cerr << "End of directory." << std::endl;
#endif
mType = TYPE_UNKNOWN ;
mFileSize = 0 ;
mFileModTime = 0;
validity = false ;
}
bool FolderIterator::updateFileInfo(bool& should_skip)
{
should_skip = false;
#ifdef WINDOWS_SYS
ConvertUtf16ToUtf8(fileInfo.cFileName, mFileName) ;
#else
@ -70,7 +105,10 @@ void FolderIterator::next()
#endif
if(mFileName == "." || mFileName == "..")
continue ;
{
should_skip = true ;
return true ;
}
mFullPath = mFolderName + "/" + mFileName ;
@ -89,7 +127,6 @@ void FolderIterator::next()
#endif
{
mFileModTime = buf.st_mtime ;
mStatInfoOk = true;
if (S_ISDIR(buf.st_mode))
{
@ -101,7 +138,7 @@ void FolderIterator::next()
mFileSize = 0 ;
mFileModTime = buf.st_mtime;
return ;
return true;
}
if (S_ISREG(buf.st_mode))
@ -114,7 +151,7 @@ void FolderIterator::next()
mFileSize = buf.st_size;
mFileModTime = buf.st_mtime;
return ;
return true;
}
}
@ -125,15 +162,8 @@ void FolderIterator::next()
mType = TYPE_UNKNOWN ;
mFileSize = 0 ;
mFileModTime = 0;
}
#ifdef DEBUG_FOLDER_ITERATOR
std::cerr << "End of directory." << std::endl;
#endif
mType = TYPE_UNKNOWN ;
mFileSize = 0 ;
mFileModTime = 0;
validity = false ;
return false;
}
bool FolderIterator::readdir()

View File

@ -59,9 +59,8 @@ private:
DIR* handle;
struct dirent* ent;
#endif
void updateStatsInfo() ;
bool updateFileInfo(bool &should_skip) ;
bool mStatInfoOk ;
time_t mFileModTime ;
time_t mFolderModTime ;
uint64_t mFileSize ;