mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04:00
e8ef48ffd8
This commit: + Adds support for a protocol which enables inter-op between concrete, tfhe-rs and potentially other contributors to the fhe ecosystem. + Gets rid of hand-made serialization in the compiler, and client/server libs. + Refactors client/server libs to allow more pre/post processing of circuit inputs/outputs. The protocol is supported by a definition in the shape of a capnp file, which defines different types of objects among which: + ProgramInfo object, which is a precise description of a set of fhe circuit coming from the same compilation (understand function type information), and the associated key set. + *Key objects, which represent secret/public keys used to encrypt/execute fhe circuits. + Value object, which represent values that can be transferred between client and server to support calls to fhe circuits. The hand-rolled serialization that was previously used is completely dropped in favor of capnp in the whole codebase. The client/server libs, are refactored to introduce a modular design for pre-post processing. Reading the ProgramInfo file associated with a compilation, the client and server libs assemble a pipeline of transformers (functions) for pre and post processing of values coming in and out of a circuit. This design properly decouples various aspects of the processing, and allows these capabilities to be safely extended. In practice this commit includes the following: + Defines the specification in a concreteprotocol package + Integrate the compilation of this package as a compiler dependency via cmake + Modify the compiler to use the Encodings objects defined in the protocol + Modify the compiler to emit ProgramInfo files as compilation artifact, and gets rid of the bloated ClientParameters. + Introduces a new Common library containing the functionalities shared between the compiler and the client/server libs. + Introduces a functional pre-post processing pipeline to this common library + Modify the client/server libs to support loading ProgramInfo objects, and calling circuits using Value messages. + Drops support of JIT. + Drops support of C-api. + Drops support of Rust bindings. Co-authored-by: Nikita Frolov <nf@mkmks.org>
90 lines
2.7 KiB
C++
90 lines
2.7 KiB
C++
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
|
|
// Exceptions. See
|
|
// https://github.com/zama-ai/concrete-compiler-internal/blob/main/LICENSE.txt
|
|
// for license information.
|
|
|
|
#ifndef CONCRETELANG_CLIENTLIB_REFACTORED_H
|
|
#define CONCRETELANG_CLIENTLIB_REFACTORED_H
|
|
|
|
#include <cassert>
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <optional>
|
|
#include <string>
|
|
#include <variant>
|
|
|
|
#include "boost/outcome.h"
|
|
#include "concrete-protocol.capnp.h"
|
|
#include "concretelang/Common/Csprng.h"
|
|
#include "concretelang/Common/Error.h"
|
|
#include "concretelang/Common/Keysets.h"
|
|
#include "concretelang/Common/Protocol.h"
|
|
#include "concretelang/Common/Transformers.h"
|
|
#include "concretelang/Common/Values.h"
|
|
|
|
using concretelang::error::Result;
|
|
using concretelang::keysets::ClientKeyset;
|
|
using concretelang::transformers::InputTransformer;
|
|
using concretelang::transformers::OutputTransformer;
|
|
using concretelang::transformers::TransformerFactory;
|
|
using concretelang::values::TransportValue;
|
|
using concretelang::values::Value;
|
|
|
|
namespace concretelang {
|
|
namespace clientlib {
|
|
|
|
class ClientCircuit {
|
|
|
|
public:
|
|
static Result<ClientCircuit>
|
|
create(const Message<concreteprotocol::CircuitInfo> &info,
|
|
const ClientKeyset &keyset, std::shared_ptr<CSPRNG> csprng,
|
|
bool useSimulation = false);
|
|
|
|
Result<TransportValue> prepareInput(Value arg, size_t pos);
|
|
|
|
Result<Value> processOutput(TransportValue result, size_t pos);
|
|
|
|
std::string getName();
|
|
|
|
const Message<concreteprotocol::CircuitInfo> &getCircuitInfo();
|
|
|
|
private:
|
|
ClientCircuit() = delete;
|
|
ClientCircuit(const Message<concreteprotocol::CircuitInfo> &circuitInfo,
|
|
std::vector<InputTransformer> inputTransformers,
|
|
std::vector<OutputTransformer> outputTransformers)
|
|
: circuitInfo(circuitInfo), inputTransformers(inputTransformers),
|
|
outputTransformers(outputTransformers){};
|
|
|
|
private:
|
|
Message<concreteprotocol::CircuitInfo> circuitInfo;
|
|
std::vector<InputTransformer> inputTransformers;
|
|
std::vector<OutputTransformer> outputTransformers;
|
|
};
|
|
|
|
/// Contains all the context to generate inputs for a server call by the
|
|
/// server lib.
|
|
class ClientProgram {
|
|
public:
|
|
/// Generates a fresh client program with fresh keyset on the first use.
|
|
static Result<ClientProgram>
|
|
create(const Message<concreteprotocol::ProgramInfo> &info,
|
|
const ClientKeyset &keyset, std::shared_ptr<CSPRNG> csprng,
|
|
bool useSimulation = false);
|
|
|
|
/// Returns a reference to the named client circuit if it exists.
|
|
Result<ClientCircuit> getClientCircuit(std::string circuitName);
|
|
|
|
private:
|
|
ClientProgram() = default;
|
|
|
|
private:
|
|
std::vector<ClientCircuit> circuits;
|
|
};
|
|
|
|
} // namespace clientlib
|
|
} // namespace concretelang
|
|
|
|
#endif
|