// (C) 2017 University of Bristol. See License.txt

#ifndef _sockets
#define _sockets

#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 */


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);

void send(int socket,octet *msg,int len);
void receive(int socket,octet *msg,int len);

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

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,int len)
{
  if (send(socket,msg,len,0)!=len)
    { error("Send error - 1 ");  }

  sent_amount += len;
  sent_counter++;
}

inline void receive(int socket,octet *msg,int len)
{
  int i=0,j;
  while (len-i>0)
    { j=recv(socket,msg+i,len-i,0);
      if (j<0) { error("Receiving error - 1"); }
      i=i+j;
    }
}

#endif