1 files changed, 94 insertions, 0 deletions
diff --git a/SoftHSMv2/src/lib/handle_mgr/HandleManager.h b/SoftHSMv2/src/lib/handle_mgr/HandleManager.h
new file mode 100644
index 0000000..e85e628
--- /dev/null
+++ b/SoftHSMv2/src/lib/handle_mgr/HandleManager.h
@@ -0,0 +1,94 @@
+/*
+ * Copyright (c) 2012 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.
+ */
+
+/*****************************************************************************
+ HandleManager.h
+
+ Keeps track of the issued cryptoki handles within SoftHSM
+ *****************************************************************************/
+
+#ifndef _SOFTHSM_V2_HANDLEMANAGER_H
+#define _SOFTHSM_V2_HANDLEMANAGER_H
+
+#include "MutexFactory.h"
+#include "Handle.h"
+#include "cryptoki.h"
+
+#include <map>
+
+#define CK_INTERNAL_SESSION_HANDLE CK_SESSION_HANDLE
+
+class HandleManager
+{
+public:
+    HandleManager();
+
+    virtual ~HandleManager();
+
+    CK_SESSION_HANDLE addSession(CK_SLOT_ID slotID, CK_VOID_PTR session);
+    CK_VOID_PTR getSession(const CK_SESSION_HANDLE hSession);
+
+    // Add the session object and return a handle. For objects that have already been registered, check that the
+    // slotID matches. The hSession may be different as the object may be added as part of a find objects operation.
+    CK_OBJECT_HANDLE addSessionObject(CK_SLOT_ID slotID, CK_SESSION_HANDLE hSession, bool isPrivate, CK_VOID_PTR object);
+
+    // Add the token object and return a handle. For objects that have already been registered, check that the
+    // slotID mathces.
+    CK_OBJECT_HANDLE addTokenObject(CK_SLOT_ID slotID, bool isPrivate, CK_VOID_PTR object);
+
+    // Get the object pointer associated with the given object handle.
+    CK_VOID_PTR getObject(const CK_OBJECT_HANDLE hObject);
+
+    // Get the object handle for the object pointer that has been previously registered.
+    // When the object is not found CK_INVALID_HANDLE is returned.
+    CK_OBJECT_HANDLE getObjectHandle(CK_VOID_PTR object);
+
+    // Remove the given object handle.
+    void destroyObject(const CK_OBJECT_HANDLE hObject);
+
+    // Remove the given session handle and all session object handles for the session.
+    // The token object handles retrieved using the session will remain valid unless
+    // this is the last session of a token being closed. In that case remove all token
+    // object handles for the slot/token associated with the session.
+    void sessionClosed(const CK_SESSION_HANDLE hSession);
+
+    // Remove all session and object handles for the given slotID.
+    // All handles for the given slotID will become invalid.
+    void allSessionsClosed(const CK_SLOT_ID slotID);
+
+    // Remove all handles to private objects for the given slotID.
+    // All handles to public objects for the given slotID remain valid.
+    void tokenLoggedOut(const CK_SLOT_ID slotID);
+
+private:
+    Mutex* handlesMutex;
+    std::map< CK_ULONG, Handle> handles;
+    std::map< CK_VOID_PTR, CK_ULONG> objects;
+    CK_ULONG handleCounter;
+};
+
+#endif // !_SOFTHSM_V2_HANDLEMANAGER_H
+