Files
MP-SPDZ/Processor/ReplicatedPrep.h
2019-01-05 15:10:01 +01:00

77 lines
1.7 KiB
C++

/*
* ReplicatedPrep.h
*
*/
#ifndef PROCESSOR_REPLICATEDPREP_H_
#define PROCESSOR_REPLICATEDPREP_H_
#include "Networking/Player.h"
#include "Data_Files.h"
#include <array>
template<class T>
class BufferPrep : public Preprocessing<T>
{
protected:
vector<array<T, 3>> triples;
vector<array<T, 2>> squares;
vector<array<T, 2>> inverses;
vector<T> bits;
virtual void buffer_triples() = 0;
virtual void buffer_squares() = 0;
virtual void buffer_inverses() = 0;
virtual void buffer_bits() = 0;
virtual void buffer_inverses(MAC_Check_Base<T>& MC, Player& P);
public:
static const int buffer_size = 1000;
virtual ~BufferPrep() {}
void get_three(Dtype dtype, T& a, T& b, T& c);
void get_two(Dtype dtype, T& a, T& b);
void get_one(Dtype dtype, T& a);
void get_input(T& a, typename T::clear& x, int i);
void get(vector<T>& S, DataTag tag, const vector<int>& regs,
int vector_size);
};
template<class T>
class ReplicatedRingPrep : public BufferPrep<T>
{
protected:
template<class U> friend class MaliciousRepPrep;
typename T::Protocol* protocol;
SubProcessor<T>* proc;
void buffer_triples();
void buffer_squares();
void buffer_inverses() { throw runtime_error("not inverses in rings"); }
public:
ReplicatedRingPrep(SubProcessor<T>* proc);
virtual ~ReplicatedRingPrep() {}
void set_protocol(typename T::Protocol& protocol) { this->protocol = &protocol; }
virtual void buffer_bits();
};
template<class T>
class ReplicatedPrep: public ReplicatedRingPrep<T>
{
void buffer_inverses();
public:
ReplicatedPrep(SubProcessor<T>* proc = 0) : ReplicatedRingPrep<T>(proc) {}
void buffer_bits();
};
#endif /* PROCESSOR_REPLICATEDPREP_H_ */