VirgilCipherBase.h

#ifndef VIRGIL_CIPHER_BASE_H
#define VIRGIL_CIPHER_BASE_H

#include <map>
#include <set>

#include <virgil/crypto/VirgilByteArray.h>
#include <virgil/crypto/VirgilCustomParams.h>

namespace virgil { namespace crypto {
class VirgilCipherBaseImpl;
}}

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

namespace virgil { namespace crypto {

class VirgilCipherBase {
public:
    VirgilCipherBase();

    virtual ~VirgilCipherBase() throw();

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);

    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 tryReadContentInfo(const VirgilByteArray& encryptedData);

    virgil::crypto::foundation::VirgilSymmetricCipher& initEncryption();
    virgil::crypto::foundation::VirgilSymmetricCipher& initDecryption(
            const VirgilByteArray& encryptedDataInfo,
            const VirgilByteArray& recipientId, const VirgilByteArray& privateKey,
            const VirgilByteArray& privateKeyPassword = VirgilByteArray());

    virgil::crypto::foundation::VirgilSymmetricCipher& initDecryptionWithPassword(const VirgilByteArray& pwd);

    virgil::crypto::foundation::VirgilSymmetricCipher& initDecryptionWithKey(
            const VirgilByteArray& recipientId,
            const VirgilByteArray& privateKey, const VirgilByteArray& privateKeyPassword);

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

    void buildContentInfo();

    void clearCipherInfo();

private:
    VirgilCipherBase(const VirgilCipherBase& other);

    VirgilCipherBase& operator=(const VirgilCipherBase& rhs);

private:
    VirgilCipherBaseImpl* impl_;
};

}}

#endif /* VIRGIL_CIPHER_BASE_H */