#include <VP_Com/vp_com_socket.h>
#include <VP_Com/vp_com_error.h>

#include <VP_Os/vp_os_malloc.h>
#include <VP_Os/vp_os_print.h>
#include <VP_Os/vp_os_signal.h>

#include <fcntl.h>
#include <errno.h>

#ifdef __linux__
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <unistd.h>

#define CYGPKG_NET 1
#endif

int32_t vp_com_fill_read_fs(vp_com_socket_t* sockets, int32_t num_sockets, int32_t max, fd_set* read_fs )
{
  while( num_sockets > 0 )
  {
    if( !sockets->is_disable )
    {
      int32_t s = (int32_t) sockets->priv;

      FD_SET( s, read_fs); // add the socket

      if( s > max )
        max = s;
    }

    sockets ++;
    num_sockets--;
  }

  return max;
}

void vp_com_close_client_sockets(vp_com_socket_t* client_sockets, int32_t num_client_sockets)
{
  int32_t s;

  // Select timed out - We close all sockets because it should mean we lost connection with client
  while( num_client_sockets > 0 )
  {
    if( !client_sockets->is_disable )
    {
      s = (int32_t) client_sockets->priv;

      DEBUG_PRINT_SDK("[VP_COM_SERVER] Closing socket %d\n", (int)s);

      client_sockets->select( client_sockets->server,
                              client_sockets,
                              VP_COM_SOCKET_SELECT_DISABLE,
                              (Write) vp_com_write_socket );

      if( client_sockets->protocol == VP_COM_TCP )
      {
        close( s );
      }

      vp_os_memset( client_sockets, 0, sizeof(vp_com_socket_t) );
      client_sockets->is_disable = TRUE;
    }

    client_sockets++;
    num_client_sockets--;
  }
}

C_RESULT vp_com_client_open_socket(vp_com_socket_t* server_socket, vp_com_socket_t* client_socket)
{
  C_RESULT res;
  struct sockaddr_in raddr = { 0 }; // remote address

  socklen_t l = sizeof(raddr);
  int32_t s = (int32_t) server_socket->priv;

  Write write = (Write) (server_socket->protocol == VP_COM_TCP ? vp_com_write_socket : vp_com_write_udp_socket);

  vp_os_memcpy( client_socket, server_socket, sizeof(vp_com_socket_t) );

  res = server_socket->select( server_socket, client_socket, VP_COM_SOCKET_SELECT_ENABLE, write );

  if( SUCCEED(res) )
  {
    if( server_socket->protocol == VP_COM_TCP )
    {
      client_socket->priv = (void*)accept( s, (struct sockaddr*)&raddr, &l );
    }

    DEBUG_PRINT_SDK("[VP_COM_SERVER] Opening socket for server %d\n", (int)s);

    client_socket->server  = server_socket;
  }
  else
  {
    DEBUG_PRINT_SDK("[VP_COM_SERVER] Failed to open socket for server %d\n", (int)s);
    vp_os_memset( client_socket, 0, sizeof(vp_com_socket_t) );
  }

  return res;
}

void vp_com_client_receive( vp_com_socket_t *client_socket )
{

  static int8_t *local_buffer=NULL ;//[VP_COM_THREAD_LOCAL_BUFFER_MAX_SIZE];
  static int8_t *new_buffer=NULL;
  static int32_t local_buffer_length=0;

  struct sockaddr_in from;
  socklen_t fromlen;
  int32_t s, received,available;

  s = (int32_t) client_socket->priv;

  fromlen = sizeof(from);

  switch(client_socket->protocol)
  {
  case VP_COM_TCP:
	  /* TCP : read as much information as we can */

	  /* Resize the buffer */
	  if (local_buffer==NULL || local_buffer_length<VP_COM_THREAD_LOCAL_BUFFER_SIZE)	  {
		  new_buffer=(int8_t *)vp_os_realloc(local_buffer,VP_COM_THREAD_LOCAL_BUFFER_SIZE);
		  if (new_buffer==NULL) { PRINT("Not enough memory to store TCP data.\n"); return;  }
		  local_buffer = new_buffer;
		  local_buffer_length = VP_COM_THREAD_LOCAL_BUFFER_SIZE;
	  }


	  /* Fetch data */
	  received = recvfrom(s, (char*)local_buffer,local_buffer_length, 0, (struct sockaddr*)&from, &fromlen);

	  /* We close the socket if an error occurred */
	  if( received <= 0)
	   {
	 	 PRINT("\nClosing socket on port %i.\n",client_socket->port);
	     client_socket->select( client_socket->server, client_socket, VP_COM_SOCKET_SELECT_DISABLE, (Write) vp_com_write_socket );
	     close( s );
	     vp_os_memset( client_socket, 0, sizeof(vp_com_socket_t) );
	     client_socket->is_disable = TRUE;
	   }
	   /* If some data were received, we pass them to the application.  */
	   else if( received>=0 && client_socket->read != NULL )
	   {
	     client_socket->remotePort = ntohs(from.sin_port);
	     client_socket->read( (void*) client_socket, local_buffer, &received, (&from)->sin_addr.s_addr );
	   }

	  break; // End of TCP processing

  case VP_COM_UDP:
	  /* UDP : read one packet */

	  /* Query packet size */
	  available = recvfrom(s, (char*)local_buffer,0, MSG_PEEK|MSG_TRUNC, (struct sockaddr*)&from, &fromlen);

	  /* Resize the buffer */
		  if(available>0){
			  if (available>VP_COM_THREAD_LOCAL_BUFFER_MAX_SIZE) {
				  PRINT("UDP packet is bigger than maximum authorized size. Dropping packet.\n"); return;
			  }
			  if (local_buffer==NULL || local_buffer_length<available)	  {
				  new_buffer=(int8_t *)vp_os_realloc(local_buffer,available);
				  if (new_buffer==NULL) {
					  PRINT("UDP packet is bigger than available memory. Dropping packet.\n"); return;
				  }
				  local_buffer = new_buffer; local_buffer_length = available;
		  }}
		  else { return ; }

	  /* Fetch data */
		received = recvfrom(s, (char*)local_buffer,local_buffer_length, 0, (struct sockaddr*)&from, &fromlen);


	  /* Closes the socket if an error occurred on an UDP socket   */
		  if( received < 0 )
		  {
			PRINT("\nClosing socket on port %i.\n",client_socket->port);
			client_socket->select( client_socket->server, client_socket, VP_COM_SOCKET_SELECT_DISABLE, (Write) vp_com_write_socket );
			close( s );
			vp_os_memset( client_socket, 0, sizeof(vp_com_socket_t) );
			client_socket->is_disable = TRUE;
		  }

	  /* If some data were received, we pass them to the application.  */
		  else if( received>=0 && client_socket->read != NULL )
		  {
            client_socket->remotePort = ntohs(from.sin_port);
            client_socket->read( (void*) client_socket, local_buffer, &received, (&from)->sin_addr.s_addr );
		  }

	  break;  // End of UDP processing

  default: break;

  }

  //DEBUG_STEPH  - debugging the fact that an empty UDP packet crashes the Mykonos program

  /* PRINT("Received (%i bytes) from (ip %i.%i.%i.%i) on (port %i) - ",
		  received,
		  ((((struct sockaddr_in*)&from)->sin_addr.s_addr))&0x0FF,
		  ((((struct sockaddr_in*)&from)->sin_addr.s_addr)>>8)&0x0FF,
		  ((((struct sockaddr_in*)&from)->sin_addr.s_addr)>>16)&0x0FF,
		  ((((struct sockaddr_in*)&from)->sin_addr.s_addr)>>24)&0x0FF,
		  ntohs(((struct sockaddr_in*)&from)->sin_port)
		  );

	  switch(client_socket->protocol){
	  case VP_COM_UDP: printf(" UDP"); break;
	  case VP_COM_TCP: printf(" TCP"); break;
	  default : printf("unknown\n"); break;
  	  }
*/

}