// (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;
}