diff options
Diffstat (limited to 'indra/llvfs')
| -rw-r--r-- | indra/llvfs/lldir_win32.cpp | 2 | ||||
| -rw-r--r-- | indra/llvfs/lldiriterator.cpp | 203 | ||||
| -rw-r--r-- | indra/llvfs/lldiriterator.h | 87 | 
3 files changed, 291 insertions, 1 deletions
| diff --git a/indra/llvfs/lldir_win32.cpp b/indra/llvfs/lldir_win32.cpp index 33718e520d..b9a3995e25 100644 --- a/indra/llvfs/lldir_win32.cpp +++ b/indra/llvfs/lldir_win32.cpp @@ -249,7 +249,7 @@ BOOL LLDir_Win32::getNextFileInDir(const std::string &dirname, const std::string  	if (pathname != mCurrentDir)  	{  		// different dir specified, close old search -		if (mCurrentDir[0]) +		if (!mCurrentDir.empty())  		{  			FindClose(mDirSearch_h);  		} diff --git a/indra/llvfs/lldiriterator.cpp b/indra/llvfs/lldiriterator.cpp new file mode 100644 index 0000000000..5536ed8f69 --- /dev/null +++ b/indra/llvfs/lldiriterator.cpp @@ -0,0 +1,203 @@ +/** + * @file lldiriterator.cpp + * @brief Iterator through directory entries matching the search pattern. + * + * $LicenseInfo:firstyear=2010&license=viewerlgpl$ + * Second Life Viewer Source Code + * Copyright (C) 2010, Linden Research, Inc. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; + * version 2.1 of the License only. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + * + * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA + * $/LicenseInfo$ + */ + +#include "lldiriterator.h" + +#include <boost/filesystem.hpp> +#include <boost/regex.hpp> + +namespace fs = boost::filesystem; + +static std::string glob_to_regex(const std::string& glob); + +class LLDirIterator::Impl +{ +public: +	Impl(const std::string &dirname, const std::string &mask); +	~Impl(); + +	bool next(std::string &fname); + +private: +	boost::regex			mFilterExp; +	fs::directory_iterator	mIter; +	bool					mIsValid; +}; + +LLDirIterator::Impl::Impl(const std::string &dirname, const std::string &mask) +	: mIsValid(false) +{ +	fs::path dir_path(dirname); + +	// Check if path exists. +	if (!fs::exists(dir_path)) +	{ +		llerrs << "Invalid path: \"" << dir_path.string() << "\"" << llendl; +		return; +	} + +	// Initialize the directory iterator for the given path. +	try +	{ +		mIter = fs::directory_iterator(dir_path); +	} +	catch (fs::basic_filesystem_error<fs::path>& e) +	{ +		llerrs << e.what() << llendl; +		return; +	} + +	// Convert the glob mask to a regular expression +	std::string exp = glob_to_regex(mask); + +	// Initialize boost::regex with the expression converted from +	// the glob mask. +	// An exception is thrown if the expression is not valid. +	try +	{ +		mFilterExp.assign(exp); +	} +	catch (boost::regex_error& e) +	{ +		llerrs << "\"" << exp << "\" is not a valid regular expression: " +				<< e.what() << llendl; +		return; +	} + +	mIsValid = true; +} + +LLDirIterator::Impl::~Impl() +{ +} + +bool LLDirIterator::Impl::next(std::string &fname) +{ +	fname = ""; + +	if (!mIsValid) +	{ +		llerrs << "The iterator is not correctly initialized." << llendl; +		return false; +	} + +	fs::directory_iterator end_itr; // default construction yields past-the-end +	bool found = false; +	while (mIter != end_itr && !found) +	{ +		boost::smatch match; +		std::string name = mIter->path().filename(); +		if (found = boost::regex_match(name, match, mFilterExp)) +		{ +			fname = name; +		} + +		++mIter; +	} + +	return found; +} + +std::string glob_to_regex(const std::string& glob) +{ +	std::string regex; +	regex.reserve(glob.size()<<1); +	S32 braces = 0; +	bool escaped = false; +	bool square_brace_open = false; + +	for (std::string::const_iterator i = glob.begin(); i != glob.end(); ++i) +	{ +		char c = *i; + +		switch (c) +		{ +			case '.': +				regex+="\\."; +				break; +			case '*': +				if (glob.begin() == i) +				{ +					regex+="[^.].*"; +				} +				else +				{ +					regex+= escaped ? "*" : ".*"; +				} +				break; +			case '?': +				regex+= escaped ? '?' : '.'; +				break; +			case '{': +				braces++; +				regex+='('; +				break; +			case '}': +				if (!braces) +				{ +					llerrs << "glob_to_regex: Closing brace without an equivalent opening brace: " << glob << llendl; +				} + +				regex+=')'; +				braces--; +				break; +			case ',': +				regex+= braces ? '|' : c; +				break; +			case '!': +				regex+= square_brace_open ? '^' : c; +				break; +			default: +				regex+=c; +				break; +		} + +		escaped = ('\\' == c); +		square_brace_open = ('[' == c); +	} + +	if (braces) +	{ +		llerrs << "glob_to_regex: Unterminated brace expression: " << glob << llendl; +	} + +	return regex; +} + +LLDirIterator::LLDirIterator(const std::string &dirname, const std::string &mask) +{ +	mImpl = new Impl(dirname, mask); +} + +LLDirIterator::~LLDirIterator() +{ +	delete mImpl; +} + +bool LLDirIterator::next(std::string &fname) +{ +	return mImpl->next(fname); +} diff --git a/indra/llvfs/lldiriterator.h b/indra/llvfs/lldiriterator.h new file mode 100644 index 0000000000..0b48be41b3 --- /dev/null +++ b/indra/llvfs/lldiriterator.h @@ -0,0 +1,87 @@ +/** + * @file lldiriterator.h + * @brief Iterator through directory entries matching the search pattern. + * + * $LicenseInfo:firstyear=2010&license=viewerlgpl$ + * Second Life Viewer Source Code + * Copyright (C) 2010, Linden Research, Inc. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; + * version 2.1 of the License only. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + * + * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA + * $/LicenseInfo$ + */ + +#ifndef LL_LLDIRITERATOR_H +#define LL_LLDIRITERATOR_H + +#include "linden_common.h" + +/** + * Class LLDirIterator + * + * Iterates through directory entries matching the search pattern. + */ +class LLDirIterator +{ +public: +	/** +	 * Constructs LLDirIterator object to search for glob pattern +	 * matches in a directory. +	 * +	 * @param dirname - name of a directory to search in. +	 * @param mask - search pattern, a glob expression +	 * +	 * Wildcards supported in glob expressions: +	 * -------------------------------------------------------------- +	 * | Wildcard 	| Matches										| +	 * -------------------------------------------------------------- +	 * | 	* 		|zero or more characters						| +	 * | 	?		|exactly one character							| +	 * | [abcde]	|exactly one character listed					| +	 * | [a-e]		|exactly one character in the given range		| +	 * | [!abcde]	|any character that is not listed				| +	 * | [!a-e]		|any character that is not in the given range	| +	 * | {abc,xyz}	|exactly one entire word in the options given	| +	 * -------------------------------------------------------------- +	 */ +	LLDirIterator(const std::string &dirname, const std::string &mask); + +	~LLDirIterator(); + +	/** +	 * Searches for the next directory entry matching the glob mask +	 * specified upon iterator construction. +	 * Returns true if a match is found, sets fname +	 * parameter to the name of the matched directory entry and +	 * increments the iterator position. +	 * +	 * Typical usage: +	 * <code> +	 * LLDirIterator iter(directory, pattern); +	 * if ( iter.next(scanResult) ) +	 * </code> +	 * +	 * @param fname - name of the matched directory entry. +	 * @return true if a match is found, false otherwise. +	 */ +	bool next(std::string &fname); + +protected: +	class Impl; +	Impl* mImpl; +}; + +#endif //LL_LLDIRITERATOR_H | 
