aboutsummaryrefslogtreecommitdiffstats
path: root/SoftHSMv2/src/lib/object_store/Directory.cpp
diff options
context:
space:
mode:
authorNingSun <ning.sun@intel.com>2018-02-08 08:34:03 -0800
committerNingSun <ning.sun@intel.com>2018-02-08 09:14:52 -0800
commit0c89b3ccba7c9b7332ab67ae1936aff51ca62367 (patch)
tree70c1b1d160d4c6d0a83395ca9a87c1264d0d3439 /SoftHSMv2/src/lib/object_store/Directory.cpp
parent945613b4db4e07f75d2bc7463db580ddfaa700fd (diff)
Initial sshsm project structure
Issue-ID: AAF-94 Change-Id: I5e82fff418e7567b161acf9b98013a9b85ffc5b4 Signed-off-by: NingSun <ning.sun@intel.com>
Diffstat (limited to 'SoftHSMv2/src/lib/object_store/Directory.cpp')
-rw-r--r--SoftHSMv2/src/lib/object_store/Directory.cpp278
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
+}
+