mirror of
https://github.com/emp-toolkit/emp-ag2pc.git
synced 2026-01-08 23:08:04 -05:00
62 lines
1.6 KiB
C++
62 lines
1.6 KiB
C++
#include <emp-tool/emp-tool.h>
|
|
#include "amortized_2pc.h"
|
|
using namespace std;
|
|
using namespace emp;
|
|
double get_io_count(Fpre * fpre, NetIO * io) {
|
|
double res = 0;
|
|
#ifdef COUNT_IO
|
|
for(int i = 0; i < fpre->THDS; ++i) {
|
|
res += fpre->io[i]->counter;
|
|
res += fpre->io2[i]->counter;
|
|
}
|
|
res += io->counter;
|
|
#endif
|
|
return res;
|
|
}
|
|
|
|
const string circuit_file_location = macro_xstr(EMP_CIRCUIT_PATH);
|
|
const static int runs = 128;
|
|
int main(int argc, char** argv) {
|
|
int port, party;
|
|
parse_party_and_port(argv, &party, &port);
|
|
|
|
NetIO* io = new NetIO(party==ALICE ? nullptr:IP, port);
|
|
io->set_nodelay();
|
|
string file = "ands.txt";//circuit_file_location+"/AES-non-expanded.txt";//adder_32bit.txt";
|
|
file = circuit_file_location+"/AES-non-expanded.txt";//adder_32bit.txt";
|
|
// file = circuit_file_location+"/sha-256.txt";
|
|
|
|
CircuitFile cf(file.c_str());
|
|
|
|
auto t1 = clock_start();
|
|
AmortizedC2PC<runs> twopc(io, party, &cf);
|
|
io->flush();
|
|
cout << "one time:\t"<<party<<"\t" <<time_from(t1)/runs<<endl;
|
|
t1 = clock_start();
|
|
twopc.function_independent();
|
|
io->flush();
|
|
cout << "inde:\t"<<party<<"\t"<<time_from(t1)/runs<<endl;
|
|
double c1 = get_io_count(twopc.fpre, io);
|
|
cout <<c1<<endl;
|
|
|
|
t1 = clock_start();
|
|
twopc.function_dependent();
|
|
io->flush();
|
|
cout << "dep:\t"<<party<<"\t"<<(time_from(t1))/runs<<endl;
|
|
double c2 = get_io_count(twopc.fpre, io);
|
|
cout <<c2 - c1<<endl;
|
|
|
|
|
|
bool in[1024], out[512];
|
|
t1 = clock_start();
|
|
for(int i = 0; i < runs; ++i) {
|
|
twopc.online(in, out);
|
|
}
|
|
cout << "online:\t"<<party<<"\t"<<(time_from(t1))/runs<<endl;
|
|
double c3 = get_io_count(twopc.fpre, io);
|
|
cout <<c3 - c2<<endl;
|
|
delete io;
|
|
|
|
return 0;
|
|
}
|