VirgilContentInfo.h

#ifndef VIRGIL_CRYPTO_CONTENT_INFO_H
#define VIRGIL_CRYPTO_CONTENT_INFO_H

#include "VirgilCustomParams.h"
#include "foundation/asn1/VirgilAsn1Compatible.h"

#include <memory>
#include <functional>

namespace virgil { namespace crypto {

class VirgilContentInfo : public foundation::asn1::VirgilAsn1Compatible {
public:
    size_t asn1Write(
            virgil::crypto::foundation::asn1::VirgilAsn1Writer& asn1Writer,
            size_t childWrittenBytes = 0) const override;

    void asn1Read(virgil::crypto::foundation::asn1::VirgilAsn1Reader& asn1Reader) override;
public:
    VirgilContentInfo();

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

    bool hasKeyRecipient(const VirgilByteArray& recipientId) const;

    void removeKeyRecipient(const VirgilByteArray& recipientId);

    void removeKeyRecipients();

    void addPasswordRecipient(const VirgilByteArray& pwd);

    bool hasPasswordRecipient(const VirgilByteArray& password) const;

    void removePasswordRecipient(const VirgilByteArray& pwd);

    void removePasswordRecipients();

    void removeAllRecipients();


    VirgilCustomParams& customParams();

    const VirgilCustomParams& customParams() const;

    static size_t defineSize(const VirgilByteArray& contentInfoData);

private:

    VirgilByteArray decryptKeyRecipient(
            const VirgilByteArray& recipientId,
            std::function<VirgilByteArray(
                    const VirgilByteArray& algorithm, const VirgilByteArray& encryptedKey)> decrypt) const;

    VirgilByteArray decryptPasswordRecipient(
            std::function<VirgilByteArray(
                    const VirgilByteArray& algorithm, const VirgilByteArray& encryptedKey)> decrypt) const;

    struct EncryptionResult {
        VirgilByteArray encryptionAlgorithm;
        VirgilByteArray encryptedContent;
    };

    void encryptKeyRecipients(std::function<EncryptionResult(const VirgilByteArray& publicKey)> encrypt);

    void encryptPasswordRecipients(std::function<EncryptionResult(const VirgilByteArray& pwd)> encrypt);

    void setContentEncryptionAlgorithm(const VirgilByteArray& contentEncryptionAlgorithm);

    VirgilByteArray getContentEncryptionAlgorithm() const;

    bool isReadyForEncryption();

    bool isReadyForDecryption();

    friend class VirgilCipherBase;

public:
    VirgilContentInfo(VirgilContentInfo&& rhs) noexcept;

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

    ~VirgilContentInfo() noexcept;

private:
    class Impl;

    std::unique_ptr<Impl> impl_;
};

}}

#endif //VIRGIL_CRYPTO_CONTENT_INFO_H