libpEpAdapter/passphrase_cache.cc

#include <cassert>
#include "Adapter.hh"
#include "passphrase_cache.hh"

namespace pEp {
    PassphraseCache::cache_entry::cache_entry(const std::string& p, time_point t) :
            passphrase{p, 0, PassphraseCache::cache_entry::max_len}, tp{t}
    { }

    PassphraseCache::PassphraseCache(size_t max_size, duration timeout) :
            _max_size{max_size}, _timeout{timeout}, _which(_cache.end()),
            first_time(true), synchronous(false)
    { }

    PassphraseCache::PassphraseCache(const PassphraseCache& second) :
            _cache{second._cache}, _max_size{second._max_size},
            _timeout{second._timeout}, _stored{second._stored},
            _which(_cache.end()), first_time(true), synchronous(false)
    {
        cleanup();
    }

    PassphraseCache& PassphraseCache::operator=(const PassphraseCache& second)
    {
        _cache = second._cache;
        _max_size = second._max_size;
        _timeout = second._timeout;
        _which = _cache.end();
        cleanup();
        return *this;
    }

    const char *PassphraseCache::add(const std::string& passphrase)
    {
        if (!passphrase.empty()) {
            const char *result = nullptr;
            {
                std::lock_guard<std::mutex> lock(_mtx);

                while (_cache.size() >= _max_size)
                    _cache.pop_front();
                
                _cache.push_back({passphrase, clock::now()});
                auto back = _cache.end();
                assert(!_cache.empty());
                result = (--back)->passphrase.c_str();
            }
            callback_dispatcher.semaphore.go();
            return result;
        }

        static const char *empty = "";
        return empty;
    }

    const char *PassphraseCache::add_stored(const std::string& passphrase)
    {
        std::lock_guard<std::mutex> lock(_stored_mtx);
        _stored = passphrase;
        return _stored.c_str();
    }

    bool PassphraseCache::for_each_passphrase(const passphrase_callee& callee)
    {
        if (callee(std::string()))
            return true;

        {
            std::lock_guard<std::mutex> lock(_stored_mtx);
            if (!_stored.empty() && callee(_stored))
                return true;
        }

        {
            std::lock_guard<std::mutex> lock(_mtx);
            cleanup();

            for (auto entry=_cache.begin(); entry!=_cache.end(); ++entry) {
                if (callee(entry->passphrase)) {
                    refresh(entry);
                    return true;
                }
            }
        }

        return false;
    }

    void PassphraseCache::cleanup()
    {
        while (!_cache.empty() && _cache.front().tp < clock::now() - _timeout)
            _cache.pop_front();
    }

    void PassphraseCache::refresh(cache::iterator entry)
    {
        entry->tp = clock::now();
        _cache.splice(_cache.end(), _cache, entry);
    }

    const char *PassphraseCache::latest_passphrase(PassphraseCache& c)
    {
        if (c.first_time) {
            c.cleanup();
            c._which = c._cache.end();
            c.first_time = false;
            if (!c._stored.empty())
                return c._stored.c_str();
        }

        if (c._cache.empty()) {
            c.first_time = true;
            throw Empty();
        }

        if (c._which == c._cache.begin()) {
            c.first_time = true;
            throw Exhausted();
        }

        --c._which;
        return c._which->passphrase.c_str();
    }

	PEP_STATUS PassphraseCache::config_next_passphrase(bool reset)
	{
        static pEp::PassphraseCache _copy;
        static bool new_copy = true;

        if (reset) {
            new_copy = true;
            return PEP_STATUS_OK;
        }

        if (new_copy) {
            _copy = passphrase_cache;
            new_copy = false;
        }

        try {
            ::config_passphrase(Adapter::session(), latest_passphrase(_copy));
            return PEP_STATUS_OK;
        }
        catch (pEp::PassphraseCache::Empty&) {
            new_copy = true;
            return PEP_PASSPHRASE_REQUIRED;
        }
        catch (pEp::PassphraseCache::Exhausted&) {
            new_copy = true;
            return PEP_WRONG_PASSPHRASE;
        }
	}
};