KeyGenerator.hpp

Go to the documentation of this file.

//                                                                            //
// Copyright (C) 2021-2023 Crypto Lab Inc.                                    //
//                                                                            //
// - This file is part of HEaaN homomorphic encryption library.               //
// - HEaaN cannot be copied and/or distributed without the express permission //
//  of Crypto Lab Inc.                                                        //
//                                                                            //
 
#pragma once
 
#include <memory>
#include <optional>
#include <string>
 
#include "HEaaN/Context.hpp"
#include "HEaaN/HEaaNExport.hpp"
#include "HEaaN/KeyPack.hpp"
 
namespace HEaaN {
 
class SecretKey;
class KeyGeneratorImpl;
 
class HEAAN_API KeyGenerator {
public:
    explicit KeyGenerator(const Context &context, const SecretKey &sk);
 
    explicit KeyGenerator(const Context &context, const SecretKey &sk,
                          const KeyPack &pack);
 
    explicit KeyGenerator(const Context &context, const Context &context_sparse,
                          const SecretKey &sk);
 
    explicit KeyGenerator(const Context &context, const Context &context_sparse,
                          const SecretKey &sk, const KeyPack &pack);
 
    // Functions for key generation //
 
    void genEncryptionKey(void) const;
 
    void genMultiplicationKey(void) const;
 
    void genConjugationKey(void) const;
 
    void genLeftRotationKey(u64 rot) const;
 
    void genRightRotationKey(u64 rot) const;
 
    void genRotationKeyBundle(void) const;
 
    void genSparseSecretEncapsulationKey(void) const;
 
    inline void genCommonKeys(void) const {
        genEncryptionKey();
        genMultiplicationKey();
        genConjugationKey();
        genRotationKeyBundle();
    }
 
    void genRotKeysForBootstrap(const u64 log_slots) const;
 
    // Utility functions //
 
    void save(const std::string &dir_path) const;
 
    void flush(void);
 
    KeyPack getKeyPack() const { return pack_; }
 
private:
    const Context context_;
    const std::optional<Context> context_sparse_;
 
    KeyPack pack_;
 
    std::shared_ptr<KeyGeneratorImpl> impl_;
};
} // namespace HEaaN