mirror of
https://github.com/data61/MP-SPDZ.git
synced 2026-04-20 03:01:31 -04:00
125 lines
2.7 KiB
C++
125 lines
2.7 KiB
C++
// (C) 2017 University of Bristol. See License.txt
|
|
|
|
|
|
#include "Networking/sockets.h"
|
|
#include "Networking/ServerSocket.h"
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <pthread.h>
|
|
using namespace std;
|
|
|
|
vector<int> socket_num;
|
|
|
|
|
|
vector<octet*> names;
|
|
|
|
int nmachines;
|
|
|
|
|
|
|
|
/*
|
|
* Get the client ip number on the socket connection for client i.
|
|
*/
|
|
void get_ip(int num)
|
|
{
|
|
struct sockaddr_storage addr;
|
|
socklen_t len = sizeof addr;
|
|
|
|
getpeername(socket_num[num], (struct sockaddr*)&addr, &len);
|
|
|
|
// supports both IPv4 and IPv6:
|
|
char ipstr[INET6_ADDRSTRLEN];
|
|
if (addr.ss_family == AF_INET) {
|
|
struct sockaddr_in *s = (struct sockaddr_in *)&addr;
|
|
inet_ntop(AF_INET, &s->sin_addr, ipstr, sizeof ipstr);
|
|
} else { // AF_INET6
|
|
struct sockaddr_in6 *s = (struct sockaddr_in6 *)&addr;
|
|
inet_ntop(AF_INET6, &s->sin6_addr, ipstr, sizeof ipstr);
|
|
}
|
|
|
|
names[num]=new octet[512];
|
|
strncpy((char*)names[num], ipstr, INET6_ADDRSTRLEN);
|
|
|
|
cerr << "Client IP address: " << names[num] << endl;
|
|
}
|
|
|
|
|
|
void get_name(int num)
|
|
{
|
|
// Now all machines are set up, send GO to start them.
|
|
send(socket_num[num], GO);
|
|
cerr << "Player " << num << " started." << endl;
|
|
|
|
// Receive name sent by client (legacy) - not used here
|
|
octet my_name[512];
|
|
receive(socket_num[num],my_name,512);
|
|
cerr << "Player " << num << " sent name (info only) " << my_name << endl;
|
|
|
|
// Get client IP
|
|
get_ip(num);
|
|
}
|
|
|
|
|
|
void send_names(int num)
|
|
{
|
|
/* Now send the machine names back to each client
|
|
* and the number of machines
|
|
*/
|
|
send(socket_num[num],nmachines);
|
|
for (int i=0; i<nmachines; i++)
|
|
{ send(socket_num[num],names[i],512); }
|
|
}
|
|
|
|
|
|
/* Takes command line arguments of
|
|
- Number of machines connecting
|
|
- Base PORTNUM address
|
|
*/
|
|
|
|
int main(int argc,char **argv)
|
|
{
|
|
if (argc != 3)
|
|
{ cerr << "Call using\n\t";
|
|
cerr << "Server.x n PortnumBase\n";
|
|
cerr << "\t\t n = Number of machines" << endl;
|
|
cerr << "\t\t PortnumBase = Base Portnum\n";
|
|
exit(1);
|
|
}
|
|
nmachines=atoi(argv[1]);
|
|
int PortnumBase=atoi(argv[2]);
|
|
int i;
|
|
|
|
names.resize(nmachines);
|
|
|
|
/* Set up the sockets */
|
|
socket_num.resize(nmachines);
|
|
for (i=0; i<nmachines; i++) { socket_num[i]=-1; }
|
|
|
|
// port number one lower to avoid conflict with players
|
|
ServerSocket server(PortnumBase - 1);
|
|
server.init();
|
|
|
|
// set up connections
|
|
for (i=0; i<nmachines; i++)
|
|
{
|
|
cerr << "Waiting for player " << i << endl;
|
|
socket_num[i] = server.get_connection_socket(i);
|
|
cerr << "Connected to player " << i << endl;
|
|
}
|
|
|
|
// get names
|
|
for (i=0; i<nmachines; i++)
|
|
get_name(i);
|
|
|
|
// send names
|
|
for (i=0; i<nmachines; i++)
|
|
send_names(i);
|
|
|
|
for (i=0; i<nmachines; i++)
|
|
{ delete[] names[i]; }
|
|
|
|
return 0;
|
|
}
|
|
|