mirror of
https://github.com/data61/MP-SPDZ.git
synced 2026-04-20 03:01:31 -04:00
104 lines
2.5 KiB
C++
104 lines
2.5 KiB
C++
/*
|
|
* YaoPlayer.cpp
|
|
*
|
|
*/
|
|
|
|
#include "YaoPlayer.h"
|
|
#include "YaoGarbler.h"
|
|
#include "YaoEvaluator.h"
|
|
#include "Tools/ezOptionParser.h"
|
|
|
|
#include "GC/Machine.hpp"
|
|
|
|
YaoPlayer::YaoPlayer(int argc, const char** argv)
|
|
{
|
|
ez::ezOptionParser opt;
|
|
opt.add(
|
|
"", // Default.
|
|
1, // Required?
|
|
1, // Number of args expected.
|
|
0, // Delimiter if expecting multiple args.
|
|
"This player's number, 0 for garbling, 1 for evaluating.", // Help description.
|
|
"-p", // Flag token.
|
|
"--player" // Flag token.
|
|
);
|
|
opt.add(
|
|
"localhost", // Default.
|
|
0, // Required?
|
|
1, // Number of args expected.
|
|
0, // Delimiter if expecting multiple args.
|
|
"Host where party 0 is running (default: localhost)", // Help description.
|
|
"-h", // Flag token.
|
|
"--hostname" // Flag token.
|
|
);
|
|
opt.add(
|
|
"5000", // Default.
|
|
0, // Required?
|
|
1, // Number of args expected.
|
|
0, // Delimiter if expecting multiple args.
|
|
"Base port number (default: 5000).", // Help description.
|
|
"-pn", // Flag token.
|
|
"--portnum" // Flag token.
|
|
);
|
|
opt.add(
|
|
"", // Default.
|
|
0, // Required?
|
|
0, // Number of args expected.
|
|
0, // Delimiter if expecting multiple args.
|
|
"Evaluate only after garbling (default only with multi-threading).", // Help description.
|
|
"-O", // Flag token.
|
|
"--oneshot" // Flag token.
|
|
);
|
|
opt.add(
|
|
"1024", // Default.
|
|
0, // Required?
|
|
1, // Number of args expected.
|
|
0, // Delimiter if expecting multiple args.
|
|
"Minimum number of gates for multithreading (default: 1024).", // Help description.
|
|
"-t", // Flag token.
|
|
"--threshold" // Flag token.
|
|
);
|
|
OnlineOptions online_opts(opt, argc, argv);
|
|
opt.parse(argc, argv);
|
|
opt.syntax = "./yao-player.x [OPTIONS] <progname>";
|
|
if (opt.lastArgs.size() == 1)
|
|
{
|
|
progname = *opt.lastArgs[0];
|
|
}
|
|
else
|
|
{
|
|
string usage;
|
|
opt.getUsage(usage);
|
|
cerr << usage;
|
|
exit(1);
|
|
}
|
|
|
|
int my_num;
|
|
int pnb;
|
|
string hostname;
|
|
int threshold;
|
|
opt.get("-p")->getInt(my_num);
|
|
opt.get("-pn")->getInt(pnb);
|
|
opt.get("-h")->getString(hostname);
|
|
bool continuous = not opt.get("-O")->isSet;
|
|
opt.get("-t")->getInt(threshold);
|
|
|
|
GC::ThreadMasterBase* master;
|
|
if (my_num == 0)
|
|
master = new YaoGarbleMaster(continuous, online_opts, threshold);
|
|
else
|
|
master = new YaoEvalMaster(continuous, online_opts);
|
|
|
|
server = Server::start_networking(master->N, my_num, 2, hostname, pnb);
|
|
master->run(progname);
|
|
|
|
if (my_num == 1)
|
|
((YaoEvalMaster*)master)->machine.write_memory(0);
|
|
}
|
|
|
|
YaoPlayer::~YaoPlayer()
|
|
{
|
|
if (server)
|
|
delete server;
|
|
}
|