MP-SPDZ/Networking/sockets.h

#ifndef _sockets_h
#define _sockets_h

#include "Networking/data.h"

#include <errno.h>      /* Errors */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include <netdb.h>
#include <netinet/in.h>
#include <netinet/tcp.h>

#include <arpa/inet.h>

#include <sys/socket.h>
#include <sys/types.h>
#include <sys/wait.h>   /* Wait for Process Termination */

#include <iostream>
using namespace std;


void error(const char *str1,const char *str2);
void error(const char *str);

void set_up_server_socket(sockaddr_in& dest,int& consocket,int& main_socket,int Portnum);
void close_server_socket(int consocket,int main_socket);

void set_up_client_socket(int& mysocket,const char* hostname,int Portnum);
void close_client_socket(int socket);

/* Send and receive 8 bit integers */
void send(int socket,int a);
void receive(int socket,int& a);

// same for words
void send(int socket, size_t a, size_t len);
void receive(int socket, size_t& a, size_t len);

void send_ack(int socket);
int get_ack(int socket);


extern unsigned long long sent_amount, sent_counter;

inline void send(int socket,octet *msg,size_t len)
{
  size_t i = 0;
  while (i < len)
    {
      int j = send(socket,msg+i,len-i,0);
      i += j;
      if (j < 0 and errno != EINTR)
        { error("Send error - 1 ");  }
    }

  sent_amount += len;
  sent_counter++;
}

inline void receive(int socket,octet *msg,size_t len)
{
  size_t i=0;
  int fail = 0;
  while (len-i>0)
    { int j=recv(socket,msg+i,len-i,0);
      if (j<0)
        {
          if (errno == EAGAIN or errno == EINTR)
            {
              if (++fail > 100)
                error("Unavailable too many times");
              else
                {
                  cout << "Unavailable, trying again in 1 ms" << endl;
                  usleep(1000);
                }
            }
          else
            { error("Receiving error - 1"); }
        }
      else
        i=i+j;
    }
}

inline size_t check_non_blocking_result(int res)
{
  if (res < 0)
    {
      if (errno != EWOULDBLOCK)
        error("Non-blocking receiving error");
      return 0;
    }
  return res;
}

inline size_t receive_non_blocking(int socket,octet *msg,int len)
{
  int res = recv(socket, msg, len, MSG_DONTWAIT);
  return check_non_blocking_result(res);
}

inline size_t receive_all_or_nothing(int socket,octet *msg,int len)
{
  int res = recv(socket, msg, len, MSG_DONTWAIT | MSG_PEEK);
  check_non_blocking_result(res);
  if (res == len)
    {
      if (recv(socket, msg, len, 0) != len)
        error("All or nothing receiving error");
      return len;
    }
  else
    return 0;
}

#endif