123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184 |
- #include <iostream>
- #include <deque>
- #include "mpcio.hpp"
- #include "preproc.hpp"
- #include "online.hpp"
- #include "options.hpp"
- static void usage(const char *progname)
- {
- std::cerr << "Usage: " << progname << " [-p | -a | -o] [-C num] [-t num] [-e] [-x] player_num player_addrs args ...\n";
- std::cerr << "-p: preprocessing mode\n";
- std::cerr << "-a: append to files in preprocessing mode (implies -p)\n";
- std::cerr << "-o: online-only mode\n";
- std::cerr << "-C num: use num communication threads\n";
- std::cerr << "-t num: use num CPU threads per communication thread\n";
- std::cerr << "-e: store DPFs expanded (default is compressed)\n";
- std::cerr << "-x: use XOR-shared database (default is additive)\n";
- std::cerr << "player_num = 0 or 1 for the computational players\n";
- std::cerr << "player_num = 2 for the server player\n";
- std::cerr << "player_addrs is omitted for player 0\n";
- std::cerr << "player_addrs is p0's hostname for player 1\n";
- std::cerr << "player_addrs is p0's hostname followed by p1's hostname for player 2\n";
- exit(1);
- }
- static void comp_player_main(boost::asio::io_context &io_context,
- unsigned player, const PRACOptions &opts, const char *p0addr,
- char **args)
- {
- std::deque<tcp::socket> peersocks, serversocks;
- mpcio_setup_computational(player, io_context, p0addr,
- opts.num_comm_threads, peersocks, serversocks);
- MPCPeerIO mpcio(player, opts.mode, peersocks, serversocks);
- // Queue up the work to be done
- boost::asio::post(io_context, [&]{
- if (opts.mode == MODE_PREPROCESSING) {
- preprocessing_comp(mpcio, opts, args);
- } else {
- online_main(mpcio, opts, args);
- }
- });
- // Start another thread; one will perform the work and the other
- // will execute the async_write handlers
- boost::thread t([&]{io_context.run();});
- io_context.run();
- t.join();
- mpcio.dump_stats(std::cout);
- }
- static void server_player_main(boost::asio::io_context &io_context,
- const PRACOptions &opts, const char *p0addr,
- const char *p1addr, char **args)
- {
- std::deque<tcp::socket> p0socks, p1socks;
- mpcio_setup_server(io_context, p0addr, p1addr,
- opts.num_comm_threads, p0socks, p1socks);
- MPCServerIO mpcserverio(opts.mode, p0socks, p1socks);
- // Queue up the work to be done
- boost::asio::post(io_context, [&]{
- if (opts.mode == MODE_PREPROCESSING) {
- preprocessing_server(mpcserverio, opts, args);
- } else {
- online_main(mpcserverio, opts, args);
- }
- });
- // Start another thread; one will perform the work and the other
- // will execute the async_write handlers
- boost::thread t([&]{io_context.run();});
- io_context.run();
- t.join();
- mpcserverio.dump_stats(std::cout);
- }
- int main(int argc, char **argv)
- {
- char **args = argv+1; // Skip argv[0] (the program name)
- PRACOptions opts;
- unsigned player = 0;
- const char *p0addr = NULL;
- const char *p1addr = NULL;
- // Get the options
- while (*args && *args[0] == '-') {
- if (!strcmp("-p", *args)) {
- opts.mode = MODE_PREPROCESSING;
- ++args;
- } else if (!strcmp("-a", *args)) {
- opts.mode = MODE_PREPROCESSING;
- opts.append_to_files = true;
- ++args;
- } else if (!strcmp("-o", *args)) {
- opts.mode = MODE_ONLINEONLY;
- ++args;
- } else if (!strcmp("-C", *args)) {
- if (args[1]) {
- opts.num_comm_threads = atoi(args[1]);
- if (opts.num_comm_threads < 1) {
- usage(argv[0]);
- }
- args += 2;
- } else {
- usage(argv[0]);
- }
- } else if (!strcmp("-t", *args)) {
- if (args[1]) {
- opts.num_cpu_threads = atoi(args[1]);
- if (opts.num_cpu_threads < 1) {
- usage(argv[0]);
- }
- args += 2;
- } else {
- usage(argv[0]);
- }
- } else if (!strcmp("-e", *args)) {
- opts.expand_rdpfs = true;
- ++args;
- } else if (!strcmp("-x", *args)) {
- opts.use_xor_db = true;
- ++args;
- } else {
- printf("Unknown option %s\n", *args);
- usage(argv[0]);
- }
- }
- if (*args == NULL) {
- // No arguments?
- usage(argv[0]);
- } else {
- player = atoi(*args);
- ++args;
- }
- if (player > 2) {
- usage(argv[0]);
- }
- if (player > 0) {
- if (*args == NULL) {
- usage(argv[0]);
- } else {
- p0addr = *args;
- ++args;
- }
- }
- if (player > 1) {
- if (*args == NULL) {
- usage(argv[0]);
- } else {
- p1addr = *args;
- ++args;
- }
- }
- /*
- std::cout << "Preprocessing = " <<
- (preprocessing ? "true" : "false") << "\n";
- std::cout << "Thread count = " << num_threads << "\n";
- std::cout << "Player = " << player << "\n";
- if (p0addr) {
- std::cout << "Player 0 addr = " << p0addr << "\n";
- }
- if (p1addr) {
- std::cout << "Player 1 addr = " << p1addr << "\n";
- }
- std::cout << "Args =";
- for (char **a = args; *a; ++a) {
- std::cout << " " << *a;
- }
- std::cout << "\n";
- */
- // Make the network connections
- boost::asio::io_context io_context;
- if (player < 2) {
- comp_player_main(io_context, player, opts, p0addr, args);
- } else {
- server_player_main(io_context, opts, p0addr, p1addr, args);
- }
- return 0;
- }
|