/*
Copyright (C) 1996-1997 Id Software, Inc.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  

See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/
// net_loop.c

#include "quakedef.h"
#include "net_loop.h"

qboolean  localconnectpending = false;
qsocket_t *loop_client = NULL;
qsocket_t *loop_server = NULL;

int Loop_Init (void)
{
  if (cls.state == ca_dedicated)
    return -1;
  return 0;
}


void Loop_Shutdown (void)
{
}


void Loop_Listen (qboolean state)
{
}


void Loop_SearchForHosts (qboolean xmit)
{
  if (!sv.active)
    return;

  hostCacheCount = 1;
  if (Q_strcmp(hostname.string, "UNNAMED") == 0)
    Q_strcpy(hostcache[0].name, "local");
  else
    Q_strcpy(hostcache[0].name, hostname.string);
  Q_strcpy(hostcache[0].map, sv.name);
  hostcache[0].users = net_activeconnections;
  hostcache[0].maxusers = svs.maxclients;
  hostcache[0].driver = net_driverlevel;
  Q_strcpy(hostcache[0].cname, "local");
}


qsocket_t *Loop_Connect (char *host)
{
  if (Q_strcmp(host,"local") != 0)
    return NULL;
  
  localconnectpending = true;

  if (!loop_client)
  {
    if ((loop_client = NET_NewQSocket ()) == NULL)
    {
      Con_Printf("Loop_Connect: no qsocket available\n");
      return NULL;
    }
    Q_strcpy (loop_client->address, "localhost");
  }
  loop_client->receiveMessageLength = 0;
  loop_client->sendMessageLength = 0;
  loop_client->canSend = true;

  if (!loop_server)
  {
    if ((loop_server = NET_NewQSocket ()) == NULL)
    {
      Con_Printf("Loop_Connect: no qsocket available\n");
      return NULL;
    }
    Q_strcpy (loop_server->address, "LOCAL");
  }
  loop_server->receiveMessageLength = 0;
  loop_server->sendMessageLength = 0;
  loop_server->canSend = true;

  loop_client->driverdata = (void *)loop_server;
  loop_server->driverdata = (void *)loop_client;
  
  return loop_client; 
}


qsocket_t *Loop_CheckNewConnections (void)
{
  if (!localconnectpending)
    return NULL;

  localconnectpending = false;
  loop_server->sendMessageLength = 0;
  loop_server->receiveMessageLength = 0;
  loop_server->canSend = true;
  loop_client->sendMessageLength = 0;
  loop_client->receiveMessageLength = 0;
  loop_client->canSend = true;
  return loop_server;
}


static int IntAlign(int value)
{
  return (value + (sizeof(int) - 1)) & (~(sizeof(int) - 1));
}


int Loop_GetMessage (qsocket_t *sock)
{
  int   ret;
  int   length;

  if (sock->receiveMessageLength == 0)
    return 0;

  ret = sock->receiveMessage[0];
  length = sock->receiveMessage[1] + (sock->receiveMessage[2] << 8);
  // alignment byte skipped here
  SZ_Clear (&net_message);
  SZ_Write (&net_message, &sock->receiveMessage[4], length);

  length = IntAlign(length + 4);
  sock->receiveMessageLength -= length;

  if (sock->receiveMessageLength)
    Q_memcpy(sock->receiveMessage, &sock->receiveMessage[length], sock->receiveMessageLength);

  if (sock->driverdata && ret == 1)
    ((qsocket_t *)sock->driverdata)->canSend = true;

  return ret;
}


int Loop_SendMessage (qsocket_t *sock, sizebuf_t *data)
{
  byte *buffer;
  int  *bufferLength;

  if (!sock->driverdata)
    return -1;

  bufferLength = &((qsocket_t *)sock->driverdata)->receiveMessageLength;

  if ((*bufferLength + data->cursize + 4) > NET_MAXMESSAGE)
    Sys_Error("Loop_SendMessage: overflow\n");

  buffer = ((qsocket_t *)sock->driverdata)->receiveMessage + *bufferLength;

  // message type
  *buffer++ = 1;

  // length
  *buffer++ = data->cursize & 0xff;
  *buffer++ = data->cursize >> 8;

  // align
  buffer++;

  // message
  Q_memcpy(buffer, data->data, data->cursize);
  *bufferLength = IntAlign(*bufferLength + data->cursize + 4);

  sock->canSend = false;
  return 1;
}


int Loop_SendUnreliableMessage (qsocket_t *sock, sizebuf_t *data)
{
  byte *buffer;
  int  *bufferLength;

  if (!sock->driverdata)
    return -1;

  bufferLength = &((qsocket_t *)sock->driverdata)->receiveMessageLength;

  if ((*bufferLength + data->cursize + sizeof(byte) + sizeof(short)) > NET_MAXMESSAGE)
    return 0;

  buffer = ((qsocket_t *)sock->driverdata)->receiveMessage + *bufferLength;

  // message type
  *buffer++ = 2;

  // length
  *buffer++ = data->cursize & 0xff;
  *buffer++ = data->cursize >> 8;

  // align
  buffer++;

  // message
  Q_memcpy(buffer, data->data, data->cursize);
  *bufferLength = IntAlign(*bufferLength + data->cursize + 4);
  return 1;
}


qboolean Loop_CanSendMessage (qsocket_t *sock)
{
  if (!sock->driverdata)
    return false;
  return sock->canSend;
}


qboolean Loop_CanSendUnreliableMessage (qsocket_t *sock)
{
  return true;
}


void Loop_Close (qsocket_t *sock)
{
  if (sock->driverdata)
    ((qsocket_t *)sock->driverdata)->driverdata = NULL;
  sock->receiveMessageLength = 0;
  sock->sendMessageLength = 0;
  sock->canSend = true;
  if (sock == loop_client)
    loop_client = NULL;
  else
    loop_server = NULL;
}