VirgilPFS.h

#ifndef VIRGIL_CRYPTO_PFS_VIRGIL_PFS_H
#define VIRGIL_CRYPTO_PFS_VIRGIL_PFS_H

#include "../VirgilByteArray.h"

#include "VirgilPFSSession.h"
#include "VirgilPFSEncryptedMessage.h"
#include "VirgilPFSInitiatorPublicInfo.h"
#include "VirgilPFSInitiatorPrivateInfo.h"
#include "VirgilPFSResponderPublicInfo.h"
#include "VirgilPFSResponderPrivateInfo.h"

#include "../primitive/VirgilOperationRandom.h"
#include "../primitive/VirgilOperationHash.h"
#include "../primitive/VirgilOperationDH.h"
#include "../primitive/VirgilOperationKDF.h"
#include "../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 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& adSecretKey, const VirgilByteArray& additionalDataMaterial) const;

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

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

}}}

#endif //VIRGIL_CRYPTO_PFS_VIRGIL_PFS_H