aboutsummaryrefslogtreecommitdiffstats
path: root/SoftHSMv2/src/lib/object_store/Generation.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/Generation.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/Generation.cpp')
-rw-r--r--SoftHSMv2/src/lib/object_store/Generation.cpp261
1 files changed, 261 insertions, 0 deletions
diff --git a/SoftHSMv2/src/lib/object_store/Generation.cpp b/SoftHSMv2/src/lib/object_store/Generation.cpp
new file mode 100644
index 0000000..196dac2
--- /dev/null
+++ b/SoftHSMv2/src/lib/object_store/Generation.cpp
@@ -0,0 +1,261 @@
+/*
+ * Copyright (c) 2013 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.
+ */
+
+/*****************************************************************************
+ Generation.cpp
+
+ Helper for generation number handling.
+ *****************************************************************************/
+
+#include "config.h"
+#include "log.h"
+#include "Generation.h"
+
+// Factory
+Generation* Generation::create(const std::string path, bool isToken /* = false */)
+{
+ Generation* gen = new Generation(path, isToken);
+ if ((gen != NULL) && isToken && (gen->genMutex == NULL))
+ {
+ delete gen;
+
+ return NULL;
+ }
+ return gen;
+}
+
+// Destructor
+Generation::~Generation()
+{
+ if (isToken)
+ {
+ MutexFactory::i()->recycleMutex(genMutex);
+ }
+}
+
+// Synchronize from locked disk file
+bool Generation::sync(File &objectFile)
+{
+ if (isToken)
+ {
+ ERROR_MSG("Generation sync() called for a token");
+
+ return false;
+ }
+
+ unsigned long onDisk;
+
+ if (!objectFile.readULong(onDisk))
+ {
+ if (objectFile.isEOF())
+ {
+ onDisk = 0;
+ }
+ else
+ {
+ return false;
+ }
+ }
+
+ currentValue = onDisk;
+
+ return objectFile.seek(0L);
+}
+
+// Check if the target was updated
+bool Generation::wasUpdated()
+{
+ if (isToken)
+ {
+ MutexLocker lock(genMutex);
+
+ File genFile(path);
+
+ if (!genFile.isValid())
+ {
+ return true;
+ }
+
+ genFile.lock();
+
+ unsigned long onDisk;
+
+ if (!genFile.readULong(onDisk))
+ {
+ return true;
+ }
+
+ if (onDisk != currentValue)
+ {
+ currentValue = onDisk;
+ return true;
+ }
+
+ return false;
+ }
+ else
+ {
+ File objectFile(path);
+
+ if (!objectFile.isValid())
+ {
+ return true;
+ }
+
+ objectFile.lock();
+
+ unsigned long onDisk;
+
+ if (!objectFile.readULong(onDisk))
+ {
+ return true;
+ }
+
+ return (onDisk != currentValue);
+ }
+}
+
+// Update
+void Generation::update()
+{
+ pendingUpdate = true;
+}
+
+// Commit
+void Generation::commit()
+{
+ if (isToken)
+ {
+ MutexLocker lock(genMutex);
+
+ File genFile(path, true, true, true, false);
+
+ if (!genFile.isValid())
+ {
+ return;
+ }
+
+ genFile.lock();
+
+ if (genFile.isEmpty())
+ {
+ currentValue++;
+
+ if (currentValue == 0)
+ {
+ currentValue++;
+ }
+
+ pendingUpdate = false;
+
+ (void) genFile.writeULong(currentValue);
+
+ genFile.unlock();
+
+ return;
+ }
+
+ unsigned long onDisk;
+
+ bool bOK = true;
+
+ bOK = bOK && genFile.readULong(onDisk);
+ bOK = bOK && genFile.seek(0L);
+
+ if (pendingUpdate)
+ {
+ onDisk++;
+
+ if (onDisk == 0)
+ {
+ onDisk++;
+ }
+ }
+
+ bOK = bOK && genFile.writeULong(onDisk);
+
+ if (bOK)
+ {
+ currentValue = onDisk;
+
+ pendingUpdate = false;
+ }
+
+ genFile.unlock();
+ }
+}
+
+// Set the current value when read from disk
+void Generation::set(unsigned long onDisk)
+{
+ currentValue = onDisk;
+}
+
+// Return new value
+unsigned long Generation::get()
+{
+ pendingUpdate = false;
+
+ currentValue++;
+
+ if (currentValue == 0)
+ {
+ currentValue = 1;
+ }
+
+ return currentValue;
+}
+
+// Rollback (called when the new value failed to be written)
+void Generation::rollback()
+{
+ pendingUpdate = true;
+
+ if (currentValue != 1)
+ {
+ currentValue--;
+ }
+}
+
+// Constructor
+Generation::Generation(const std::string inPath, bool inIsToken)
+{
+ path = inPath;
+ isToken = inIsToken;
+ pendingUpdate = false;
+ currentValue = 0;
+ genMutex = NULL;
+
+ if (isToken)
+ {
+ genMutex = MutexFactory::i()->getMutex();
+
+ if (genMutex != NULL)
+ {
+ commit();
+ }
+ }
+}