VirgilPFS.h

#ifndef VIRGIL_CRYPTO_PFS_VIRGIL_PFS_H
#define VIRGIL_CRYPTO_PFS_VIRGIL_PFS_H

#include <virgil/crypto/VirgilByteArray.h>

#include <virgil/crypto/pfs/VirgilPFSSession.h>
#include <virgil/crypto/pfs/VirgilPFSEncryptedMessage.h>
#include <virgil/crypto/pfs/VirgilPFSInitiatorPublicInfo.h>
#include <virgil/crypto/pfs/VirgilPFSInitiatorPrivateInfo.h>
#include <virgil/crypto/pfs/VirgilPFSResponderPublicInfo.h>
#include <virgil/crypto/pfs/VirgilPFSResponderPrivateInfo.h>

#include <virgil/crypto/primitive/VirgilOperationRandom.h>
#include <virgil/crypto/primitive/VirgilOperationHash.h>
#include <virgil/crypto/primitive/VirgilOperationDH.h>
#include <virgil/crypto/primitive/VirgilOperationKDF.h>
#include <virgil/crypto/primitive/VirgilOperationCipher.h>

namespace virgil { namespace crypto { namespace pfs {

class VirgilPFS {
public:
    VirgilPFS();

    VirgilPFSSession startInitiatorSession(
            const VirgilPFSInitiatorPrivateInfo& initiatorPrivateInfo,
            const VirgilPFSResponderPublicInfo& responderPublicInfo,
            const VirgilByteArray& additionalData = VirgilByteArray());

    VirgilPFSSession startResponderSession(
            const VirgilPFSResponderPrivateInfo& responderPrivateInfo,
            const VirgilPFSInitiatorPublicInfo& initiatorPublicInfo,
            const VirgilByteArray& additionalData = VirgilByteArray());

    VirgilPFSEncryptedMessage encrypt(const VirgilByteArray& data);

    VirgilByteArray decrypt(const VirgilPFSEncryptedMessage& encryptedMessage) const;

    void setRandom(VirgilOperationRandom random);

    void setHash(VirgilOperationHash hash);

    void setDH(VirgilOperationDH dh);

    void setKDF(VirgilOperationKDF kdf);

    void setCipher(VirgilOperationCipher cipher);

    VirgilPFSSession getSession() const;

    void setSession(VirgilPFSSession session);

private:
    VirgilByteArray calculateSharedKey(
            const VirgilPFSInitiatorPrivateInfo& initiatorPrivateInfo,
            const VirgilPFSResponderPublicInfo& responderPublicInfo) const;

    VirgilByteArray calculateSharedKey(
            const VirgilPFSResponderPrivateInfo& responderPrivateInfo,
            const VirgilPFSInitiatorPublicInfo& initiatorPublicInfo) const;

    VirgilByteArray calculateSecretKey(const VirgilByteArray& keyMaterial, size_t size);

    VirgilByteArray calculateAdditionalData(const VirgilByteArray& additionalDataMaterial) const;

    VirgilByteArray calculateSessionIdentifier(
            const VirgilByteArray& secretKey, const VirgilByteArray& additionalData) const;

private:
    VirgilOperationRandom random_;
    VirgilOperationHash hash_;
    VirgilOperationDH dh_;
    VirgilOperationKDF kdf_;
    VirgilOperationCipher cipher_;
    VirgilPFSSession session_;
};

}}}

#endif //VIRGIL_CRYPTO_PFS_VIRGIL_PFS_H