
3 changed files with 78 additions and 0 deletions
@ -0,0 +1,34 @@ |
|||
#include "passphrase_cache.hh" |
|||
|
|||
namespace pEp { |
|||
void PassphraseCache::add(std::string passphrase) |
|||
{ |
|||
std::lock_guard<std::mutex> lock(_mtx); |
|||
cleanup(); |
|||
_cache.push_back({passphrase, clock::now()}); |
|||
} |
|||
|
|||
bool PassphraseCache::for_each_passphrase(passphrase_callee& callee) |
|||
{ |
|||
std::lock_guard<std::mutex> lock(_mtx); |
|||
cleanup(); |
|||
|
|||
for (auto entry : _cache) { |
|||
if (callee(entry.passphrase)) |
|||
return true; |
|||
} |
|||
|
|||
return false; |
|||
} |
|||
|
|||
void PassphraseCache::cleanup() |
|||
{ |
|||
for (auto entry = _cache.begin(); entry != _cache.end(); ) { |
|||
if (entry->tp < clock::now() - _timeout) |
|||
_cache.erase(entry); |
|||
else |
|||
break; |
|||
} |
|||
} |
|||
}; |
|||
|
@ -0,0 +1,43 @@ |
|||
#pragma once |
|||
|
|||
#include <list> |
|||
#include <string> |
|||
#include <chrono> |
|||
#include <mutex> |
|||
#include <functional> |
|||
|
|||
namespace pEp { |
|||
class PassphraseCache { |
|||
using clock = std::chrono::system_clock; |
|||
using time_point = std::chrono::time_point<clock>; |
|||
using duration = clock::duration; |
|||
|
|||
struct cache_entry { |
|||
std::string passphrase; |
|||
time_point tp; |
|||
}; |
|||
|
|||
std::list<cache_entry> _cache; |
|||
std::mutex _mtx; |
|||
duration _timeout; |
|||
|
|||
public: |
|||
PassphraseCache() : _timeout(std::chrono::minutes(10)) { } |
|||
~PassphraseCache() { } |
|||
PassphraseCache(const PassphraseCache& second) : |
|||
_cache(second._cache), _timeout(second._timeout) { } |
|||
|
|||
// adding a passphrase to the cache, which will timeout
|
|||
void add(std::string passphrase); |
|||
|
|||
// for each passphrase call the callee until it returns true for a
|
|||
// matching passphrase or no passphrases are left
|
|||
// returns true if a passphrase was matching, false otherwise
|
|||
using passphrase_callee = std::function<bool(std::string)>; |
|||
bool for_each_passphrase(passphrase_callee& callee); |
|||
|
|||
protected: |
|||
void cleanup(); |
|||
}; |
|||
}; |
|||
|
Loading…
Reference in new issue