1 files changed, 106 insertions, 0 deletions
diff --git a/src/tls/tls_session_manager.cpp b/src/tls/tls_session_manager.cpp
new file mode 100644
index 000000000..72eb83c21
--- /dev/null
+++ b/src/tls/tls_session_manager.cpp
@@ -0,0 +1,106 @@
+/*
+* TLS Session Management
+* (C) 2011 Jack Lloyd
+*
+* Released under the terms of the Botan license
+*/
+
+#include <botan/tls_session_manager.h>
+#include <botan/hex.h>
+#include <chrono>
+
+namespace Botan {
+
+namespace TLS {
+
+bool Session_Manager_In_Memory::load_from_session_str(
+   const std::string& session_str, Session& session)
+   {
+   // assert(lock is held)
+
+   auto i = m_sessions.find(session_str);
+
+   if(i == m_sessions.end())
+      return false;
+
+   // if session has expired, remove it
+   const auto now = std::chrono::system_clock::now();
+
+   if(i->second.start_time() + session_lifetime() < now)
+      {
+      m_sessions.erase(i);
+      return false;
+      }
+
+   session = i->second;
+   return true;
+   }
+
+bool Session_Manager_In_Memory::load_from_session_id(
+   const std::vector<byte>& session_id, Session& session)
+   {
+   std::lock_guard<std::mutex> lock(m_mutex);
+
+   return load_from_session_str(hex_encode(session_id), session);
+   }
+
+bool Session_Manager_In_Memory::load_from_host_info(
+   const std::string& hostname, u16bit port, Session& session)
+   {
+   std::lock_guard<std::mutex> lock(m_mutex);
+
+   std::map<std::string, std::string>::iterator i;
+
+   if(port > 0)
+      i = m_host_sessions.find(hostname + ":" + std::to_string(port));
+   else
+      i = m_host_sessions.find(hostname);
+
+   if(i == m_host_sessions.end())
+      return false;
+
+   if(load_from_session_str(i->second, session))
+      return true;
+
+   // was removed from sessions map, remove m_host_sessions entry
+   m_host_sessions.erase(i);
+
+   return false;
+   }
+
+void Session_Manager_In_Memory::remove_entry(
+   const std::vector<byte>& session_id)
+   {
+   std::lock_guard<std::mutex> lock(m_mutex);
+
+   auto i = m_sessions.find(hex_encode(session_id));
+
+   if(i != m_sessions.end())
+      m_sessions.erase(i);
+   }
+
+void Session_Manager_In_Memory::save(const Session& session)
+   {
+   std::lock_guard<std::mutex> lock(m_mutex);
+
+   if(m_max_sessions != 0)
+      {
+      /*
+      This removes randomly based on ordering of session ids.
+      Instead, remove oldest first?
+      */
+      while(m_sessions.size() >= m_max_sessions)
+         m_sessions.erase(m_sessions.begin());
+      }
+
+   const std::string session_id_str = hex_encode(session.session_id());
+
+   m_sessions[session_id_str] = session;
+
+   if(session.side() == CLIENT && session.sni_hostname() != "")
+      m_host_sessions[session.sni_hostname()] = session_id_str;
+   }
+
+}
+
+}