1 files changed, 278 insertions, 0 deletions
diff --git a/SoftHSMv2/src/lib/object_store/Directory.cpp b/SoftHSMv2/src/lib/object_store/Directory.cpp
new file mode 100644
index 0000000..e964d32
--- /dev/null
+++ b/SoftHSMv2/src/lib/object_store/Directory.cpp
@@ -0,0 +1,278 @@
+/*
+ * Copyright (c) 2010 SURFnet bv
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
+ * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
+ * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+ * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
+ * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+/*****************************************************************************
+ Directory.cpp
+
+ Helper functions for accessing directories.
+ *****************************************************************************/
+
+#include "config.h"
+#include "Directory.h"
+#include "OSPathSep.h"
+#include "log.h"
+#include <string>
+#include <vector>
+#ifndef _WIN32
+#include <dirent.h>
+#include <unistd.h>
+#else
+#include <direct.h>
+#include <io.h>
+#endif
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+
+// Constructor
+Directory::Directory(std::string inPath)
+{
+	path = inPath;
+	dirMutex = MutexFactory::i()->getMutex();
+
+	valid = (dirMutex != NULL) && refresh();
+}
+
+// Destructor
+Directory::~Directory()
+{
+	MutexFactory::i()->recycleMutex(dirMutex);
+}
+
+// Check if the directory is valid
+bool Directory::isValid()
+{
+	return valid;
+}
+
+// Return a list of all files in a directory
+std::vector<std::string> Directory::getFiles()
+{
+	// Make sure that no other thread is in the process of changing
+	// the file list when we return it
+	MutexLocker lock(dirMutex);
+
+	return files;
+}
+
+// Return a list of all subdirectories in a directory
+std::vector<std::string> Directory::getSubDirs()
+{
+	// Make sure that no other thread is in the process of changing
+	// the subdirectory list when we return it
+	MutexLocker lock(dirMutex);
+
+	return subDirs;
+}
+
+// Refresh the directory listing
+bool Directory::refresh()
+{
+	// Prevent concurrent call until valid is reset
+	MutexLocker lock(dirMutex);
+
+	// Reset the state
+	valid = false;
+
+	subDirs.clear();
+	files.clear();
+
+#ifndef _WIN32
+	// Enumerate the directory
+	DIR* dir = opendir(path.c_str());
+
+	if (dir == NULL)
+	{
+		DEBUG_MSG("Failed to open directory %s", path.c_str());
+
+		return false;
+	}
+
+	// Enumerate the directory
+	struct dirent* entry = NULL;
+
+	while ((entry = readdir(dir)) != NULL)
+	{
+		bool pushed = false;
+
+		// Check if this is the . or .. entry
+		if (!strcmp(entry->d_name, ".") || !strcmp(entry->d_name, ".."))
+		{
+			continue;
+		}
+
+		// Convert the name of the entry to a C++ string
+		std::string name(entry->d_name);
+
+#if defined(_DIRENT_HAVE_D_TYPE) && defined(_BSD_SOURCE)
+		// Determine the type of the entry
+		switch(entry->d_type)
+		{
+		case DT_DIR:
+			// This is a directory
+			subDirs.push_back(name);
+			pushed = true;
+			break;
+		case DT_REG:
+			// This is a regular file
+			files.push_back(name);
+			pushed = true;
+			break;
+		default:
+			break;
+		}
+#endif
+		if (!pushed) {
+			// The entry type has to be determined using lstat
+			struct stat entryStatus;
+
+			std::string fullPath = path + OS_PATHSEP + name;
+
+			if (!lstat(fullPath.c_str(), &entryStatus))
+			{
+				if (S_ISDIR(entryStatus.st_mode))
+				{
+					subDirs.push_back(name);
+				}
+				else if (S_ISREG(entryStatus.st_mode))
+				{
+					files.push_back(name);
+				}
+				else
+				{
+					DEBUG_MSG("File not used %s", name.c_str());
+				}
+			}
+		}
+	}
+
+	// Close the directory
+	closedir(dir);
+
+#else
+	// Enumerate the directory
+	std::string pattern;
+	intptr_t h;
+	struct _finddata_t fi;
+
+	if ((path.back() == '/') || (path.back() == '\\'))
+		pattern = path + "*";
+	else
+		pattern = path + "/*";
+	memset(&fi, 0, sizeof(fi));
+	h = _findfirst(pattern.c_str(), &fi);
+	if (h == -1)
+	{
+		// empty directory
+		if (errno == ENOENT)
+			goto finished;
+
+		DEBUG_MSG("Failed to open directory %s", path.c_str());
+
+		return false;
+	}
+
+	// scan files & subdirs
+	do {
+		// Check if this is the . or .. entry
+		if (!strcmp(fi.name, ".") || !strcmp(fi.name, ".."))
+			continue;
+
+		if ((fi.attrib & _A_SUBDIR) == 0)
+			files.push_back(fi.name);
+		else
+			subDirs.push_back(fi.name);
+
+		memset(&fi, 0, sizeof(fi));
+	} while (_findnext(h, &fi) == 0);
+
+	(void) _findclose(h);
+
+    finished:
+#endif
+
+	valid = true;
+
+	return true;
+}
+
+// Create a new subdirectory
+bool Directory::mkdir(std::string name)
+{
+	std::string fullPath = path + OS_PATHSEP + name;
+
+#ifndef _WIN32
+	int rv = ::mkdir(fullPath.c_str(), S_IFDIR | S_IRWXU);
+#else
+	int rv = _mkdir(fullPath.c_str());
+#endif
+
+	if (rv != 0)
+	{
+		ERROR_MSG("Failed to create the directory (%s): %s", strerror(errno), fullPath.c_str());
+
+		return false;
+	}
+
+	return refresh();
+}
+
+// Delete a subdirectory in the directory
+bool Directory::rmdir(std::string name, bool doRefresh /* = false */)
+{
+	std::string fullPath;
+
+	if (name.empty())
+		fullPath = path;
+	else
+		fullPath = path + OS_PATHSEP + name;
+
+#ifndef _WIN32
+	if (::rmdir(fullPath.c_str()) != 0)
+		return false;
+#else
+	if (_rmdir(fullPath.c_str()) != 0)
+		return false;
+#endif
+	if (doRefresh)
+		return refresh();
+	return true;
+}
+
+// Delete a file in the directory
+bool Directory::remove(std::string name)
+{
+	std::string fullPath = path + OS_PATHSEP + name;
+
+#ifndef _WIN32
+	return (!::remove(fullPath.c_str()) && refresh());
+#else
+	return (!_unlink(fullPath.c_str()) && refresh());
+#endif
+}
+