VirgilCipherBase.h

#ifndef VIRGIL_CIPHER_BASE_H
#define VIRGIL_CIPHER_BASE_H

#include <map>
#include <set>
#include <memory>

#include "VirgilByteArray.h"
#include "VirgilCustomParams.h"

namespace virgil { namespace crypto { namespace foundation {
class VirgilSymmetricCipher;
}}}

namespace virgil { namespace crypto {

class VirgilCipherBase {
public:
    VirgilCipherBase();

public:

    void addKeyRecipient(const VirgilByteArray& recipientId, const VirgilByteArray& publicKey);

    void removeKeyRecipient(const VirgilByteArray& recipientId);

    bool keyRecipientExists(const VirgilByteArray& recipientId) const;

    void addPasswordRecipient(const VirgilByteArray& pwd);

    void removePasswordRecipient(const VirgilByteArray& pwd);

    bool passwordRecipientExists(const VirgilByteArray& password) const;

    void removeAllRecipients();


    VirgilByteArray getContentInfo() const;

    void setContentInfo(const VirgilByteArray& contentInfo);

    static size_t defineContentInfoSize(const VirgilByteArray& data);


    VirgilCustomParams& customParams();

    const VirgilCustomParams& customParams() const;


    static VirgilByteArray computeShared(
            const VirgilByteArray& publicKey,
            const VirgilByteArray& privateKey, const VirgilByteArray& privateKeyPassword = VirgilByteArray());

protected:
    VirgilByteArray filterAndSetupContentInfo(const VirgilByteArray& encryptedData, bool isLastChunk);

    void initEncryption();

    void initDecryptionWithPassword(const VirgilByteArray& pwd);

    void initDecryptionWithKey(
            const VirgilByteArray& recipientId,
            const VirgilByteArray& privateKey, const VirgilByteArray& privateKeyPassword);

    bool isInited() const;

    bool isReadyForEncryption() const;

    bool isReadyForDecryption() const;

    virgil::crypto::foundation::VirgilSymmetricCipher& getSymmetricCipher();

    void buildContentInfo();

    void clear();

public:
    VirgilCipherBase(VirgilCipherBase&& rhs) noexcept;

    VirgilCipherBase& operator=(VirgilCipherBase&& rhs) noexcept;

    virtual ~VirgilCipherBase() noexcept;

private:
    virtual VirgilByteArray doDecryptWithKey(
            const VirgilByteArray& algorithm, const VirgilByteArray& encryptedKey,
            const VirgilByteArray& privateKey, const VirgilByteArray& privateKeyPassword) const;

    virtual VirgilByteArray doDecryptWithPassword(
            const VirgilByteArray& encryptedKey, const VirgilByteArray& encryptionAlgorithm,
            const VirgilByteArray& password) const;


    void accomplishInitDecryption();

private:
    class Impl;

    std::unique_ptr<Impl> impl_;
};

}}

#endif /* VIRGIL_CIPHER_BASE_H */