// ---------------------------------------------------------------------------80
//                ICQ plugin for Miranda Instant Messenger
//                ________________________________________
//
// Copyright © 2000,2001 Richard Hughes, Roland Rabien, Tristan Van de Vreede
// Copyright © 2001,2002 Jon Keating, Richard Hughes
// Copyright © 2002,2003,2004 Martin Öberg, Sam Kothari, Robert Rainwater
// Copyright © 2004,2005,2006 Joe Kucera
//
// 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.
//
// -----------------------------------------------------------------------------
//
// File name      : $Source: /cvsroot/miranda/miranda/protocols/IcqOscarJ/icq_server.c,v $
// Revision       : $Revision: 3316 $
// Last change on : $Date: 2006-05-16 23:38:00 +0200 (Tue, 16 May 2006) $
// Last change by : $Author: ghazan $
//
// DESCRIPTION:
//
//  Manages main server connection, low-level communication
//
// -----------------------------------------------------------------------------

#include "icqoscar.h"



extern void handleXStatusCaps(HANDLE hContact, char* caps, int capsize);

extern CRITICAL_SECTION connectionHandleMutex;
extern WORD wLocalSequence;
extern CRITICAL_SECTION localSeqMutex;
HANDLE hServerConn;
WORD wListenPort;
WORD wLocalSequence;
DWORD dwLocalDirectConnCookie;
HANDLE hServerPacketRecver;
static pthread_t serverThreadId;
HANDLE hDirectBoundPort;

static int handleServerPackets(unsigned char* buf, int len, serverthread_info* info);


static DWORD __stdcall icq_serverThread(serverthread_start_info* infoParam)
{
  serverthread_info info = {0};

  info.isLoginServer = 1;
  info.wAuthKeyLen = infoParam->wPassLen;
  strncpy(info.szAuthKey, infoParam->szPass, info.wAuthKeyLen);

  srand(time(NULL));

  dwLocalDirectConnCookie = rand() ^ (rand() << 16);

  ResetSettingsOnConnect();

  // Connect to the login server
  NetLog_Server("Authenticating to server");
  {
    NETLIBOPENCONNECTION nloc = infoParam->nloc;

    hServerConn = NetLib_OpenConnection(ghServerNetlibUser, &nloc);

    SAFE_FREE((void**)&nloc.szHost);
  }
  SAFE_FREE(&infoParam);


  // Login error
  if (hServerConn == NULL)
  {
    DWORD dwError = GetLastError();

    hServerConn = NULL;

    SetCurrentStatus(ID_STATUS_OFFLINE);

    icq_LogUsingErrorCode(LOG_ERROR, dwError, "Unable to connect to ICQ login server");

    return 0;
  }


  // Initialize direct connection ports
  {
    NETLIBBIND nlb = {0};

    nlb.cbSize = sizeof(nlb);
    nlb.pfnNewConnection = icq_newConnectionReceived;
    hDirectBoundPort = (HANDLE)CallService(MS_NETLIB_BINDPORT, (WPARAM)ghDirectNetlibUser, (LPARAM)&nlb);
    if (!hDirectBoundPort && (GetLastError() == 87))
    { // this ensures old Miranda also can bind a port for a dc
      nlb.cbSize = NETLIBBIND_SIZEOF_V1;
      hDirectBoundPort = (HANDLE)CallService(MS_NETLIB_BINDPORT, (WPARAM)ghDirectNetlibUser, (LPARAM)&nlb);
    }
    if (hDirectBoundPort == NULL)
    {
      icq_LogUsingErrorCode(LOG_WARNING, GetLastError(), "Miranda was unable to allocate a port to listen for direct peer-to-peer connections between clients. You will be able to use most of the ICQ network without problems but you may be unable to send or receive files.\n\nIf you have a firewall this may be blocking Miranda, in which case you should configure your firewall to leave some ports open and tell Miranda which ports to use in M->Options->ICQ->Network.");
      wListenPort = 0;
    }
    wListenPort = nlb.wPort;
    if (!ICQGetContactSettingByte(NULL, "ConstRealIP", 0))
      ICQWriteContactSettingDword(NULL, "RealIP", nlb.dwInternalIP);
	
	ICQWriteContactSettingDword(NULL, "OldRealIP", nlb.dwInternalIP);
	ICQWriteContactSettingWord(NULL, "UserPort", wListenPort);
	ICQWriteContactSettingWord(NULL, "OldUserPort", wListenPort);
  }


  // This is the "infinite" loop that receives the packets from the ICQ server
  {
    int recvResult;
    NETLIBPACKETRECVER packetRecv = {0};

    hServerPacketRecver = (HANDLE)CallService(MS_NETLIB_CREATEPACKETRECVER, (WPARAM)hServerConn, 8192);
    packetRecv.cbSize = sizeof(packetRecv);
    packetRecv.dwTimeout = INFINITE;
    while(hServerConn)
    {
      if (info.bReinitRecver)
      { // we reconnected, reinit struct
        info.bReinitRecver = 0;
        ZeroMemory(&packetRecv, sizeof(packetRecv));
        packetRecv.cbSize = sizeof(packetRecv);
        packetRecv.dwTimeout = INFINITE;
      }

      recvResult = CallService(MS_NETLIB_GETMOREPACKETS,(WPARAM)hServerPacketRecver, (LPARAM)&packetRecv);

      if (recvResult == 0)
      {
        NetLog_Server("Clean closure of server socket");
        break;
      }

      if (recvResult == SOCKET_ERROR)
      {
        NetLog_Server("Abortive closure of server socket, error: %d", GetLastError());
        break;
      }

      // Deal with the packet
      packetRecv.bytesUsed = handleServerPackets(packetRecv.buffer, packetRecv.bytesAvailable, &info);
    }

    // Close the packet receiver (connection may still be open)
    Netlib_CloseHandle(hServerPacketRecver);
    hServerPacketRecver = 0;

    // Close DC port
    Netlib_CloseHandle(hDirectBoundPort);
    hDirectBoundPort = 0;
  }

  // Time to shutdown
  icq_serverDisconnect(FALSE);
  if (gnCurrentStatus != ID_STATUS_OFFLINE)
  {    
  	  if (!info.bLoggedIn)
    {
      icq_LogMessage(LOG_FATAL, "Connection failed.\nLogin sequence failed for unknown reason.\nTry again later.");
    }

    SetCurrentStatus(ID_STATUS_OFFLINE);
  }
  
  // Close all open DC connections
  CloseContactDirectConns(NULL);

  // Offline all contacts
  {
    HANDLE hContact;

    hContact= ICQFindFirstContact();

    while (hContact)
    {
      DWORD dwUIN;
      uid_str szUID;

      if (!ICQGetContactSettingUID(hContact, &dwUIN, &szUID))
      {
        if (ICQGetContactStatus(hContact) != ID_STATUS_OFFLINE)
        {
          ICQWriteContactSettingWord(hContact, "Status", ID_STATUS_OFFLINE);

          handleXStatusCaps(hContact, NULL, 0);
        }
      }

      hContact = ICQFindNextContact(hContact);
    }
  }
  ICQWriteContactSettingDword(NULL, "LogonTS", 0); // clear logon time

  FlushServerIDs();         // clear server IDs list
  FlushPendingOperations(); // clear pending operations list
  FlushGroupRenames();      // clear group rename in progress list

  NetLog_Server("%s thread ended.", "Server");

  return 0;
}



void icq_serverDisconnect(BOOL bBlock)
{
  EnterCriticalSection(&connectionHandleMutex);

  if (hServerConn)
  {
    int sck = CallService(MS_NETLIB_GETSOCKET, (WPARAM)hServerConn, (LPARAM)0);
    if (sck!=INVALID_SOCKET) shutdown(sck, 2); // close gracefully
    Netlib_CloseHandle(hServerConn);
    FreeGatewayIndex(hServerConn);
    hServerConn = NULL;
    LeaveCriticalSection(&connectionHandleMutex);
    
    // Not called from network thread?
    if (bBlock && GetCurrentThreadId() != serverThreadId.dwThreadId)
    {
      while (WaitForSingleObjectEx(serverThreadId.hThread, INFINITE, TRUE) != WAIT_OBJECT_0);
      CloseHandle(serverThreadId.hThread);
    }
    else
      CloseHandle(serverThreadId.hThread);
  }
  else
    LeaveCriticalSection(&connectionHandleMutex);

  StopKeepAlive(); // signal keep-alive thread to stop
}



static int handleServerPackets(unsigned char* buf, int len, serverthread_info* info)
{
  BYTE channel;
  WORD sequence;
  WORD datalen;
  int bytesUsed = 0;

  while (len > 0)
  {
    if (info->bReinitRecver)
      break;

    // All FLAPS begin with 0x2a
    if (*buf++ != FLAP_MARKER)
      break;

    if (len < 6)
      break;

    unpackByte(&buf, &channel);
    unpackWord(&buf, &sequence);
    unpackWord(&buf, &datalen);

    if (len < 6 + datalen)
      break;


#ifdef _DEBUG
    NetLog_Server("Server FLAP: Channel %u, Seq %u, Length %u bytes", channel, sequence, datalen);
#endif

    switch (channel)
    {
    case ICQ_LOGIN_CHAN:
      handleLoginChannel(buf, datalen, info);
      break;

    case ICQ_DATA_CHAN:
      handleDataChannel(buf, datalen, info);
      break;

    case ICQ_ERROR_CHAN:
      handleErrorChannel(buf, datalen);
      break;

    case ICQ_CLOSE_CHAN:
      handleCloseChannel(buf, datalen, info);
      break; // we need this for walking thru proxy

    case ICQ_PING_CHAN:
      handlePingChannel(buf, datalen);
      break;

    default:
      NetLog_Server("Warning: Unhandled %s FLAP Channel: Channel %u, Seq %u, Length %u bytes", "Server", channel, sequence, datalen);
      break;
    }

    /* Increase pointers so we can check for more FLAPs */
    buf += datalen;
    len -= (datalen + 6);
    bytesUsed += (datalen + 6);
  }

  return bytesUsed;
}



void sendServPacket(icq_packet* pPacket)
{
  // This critsec makes sure that the sequence order doesn't get screwed up
  EnterCriticalSection(&localSeqMutex);

  if (hServerConn)
  {
    int nRetries;
    int nSendResult;


    // :IMPORTANT:
    // The FLAP sequence must be a WORD. When it reaches 0xFFFF it should wrap to
    // 0x0000, otherwise we'll get kicked by server.
    wLocalSequence++;

    // Pack sequence number
    pPacket->pData[2] = ((wLocalSequence & 0xff00) >> 8);
    pPacket->pData[3] = (wLocalSequence & 0x00ff);

    for (nRetries = 3; nRetries >= 0; nRetries--)
    {
      nSendResult = Netlib_Send(hServerConn, (const char *)pPacket->pData, pPacket->wLen, 0);

      if (nSendResult != SOCKET_ERROR)
        break;

      Sleep(1000);
    }


    // Send error
    if (nSendResult == SOCKET_ERROR)
    {
      icq_LogUsingErrorCode(LOG_ERROR, GetLastError(), "Your connection with the ICQ server was abortively closed");
      icq_serverDisconnect(FALSE);

      if (gnCurrentStatus != ID_STATUS_OFFLINE)
      {
        SetCurrentStatus(ID_STATUS_OFFLINE);
      }
    }
  }
  else
  {
    NetLog_Server("Error: Failed to send packet (no connection)");
  }

  LeaveCriticalSection(&localSeqMutex);

  SAFE_FREE(&pPacket->pData);
}



void icq_login(const char* szPassword)
{
  DBVARIANT dbvServer = {DBVT_DELETED};
  char szServer[MAX_PATH];
  serverthread_start_info* stsi;
  DWORD dwUin;


  dwUin = ICQGetContactSettingUIN(NULL);
  stsi = (serverthread_start_info*)SAFE_MALLOC(sizeof(serverthread_start_info));
  stsi->nloc.cbSize = sizeof(NETLIBOPENCONNECTION);

  // Server host name
  if (ICQGetContactStaticString(NULL, "OscarServer", szServer, MAX_PATH))
    stsi->nloc.szHost = null_strdup(DEFAULT_SERVER_HOST);
  else
    stsi->nloc.szHost = null_strdup(szServer);

  // Server port
  stsi->nloc.wPort = (WORD)ICQGetContactSettingWord(NULL, "OscarPort", DEFAULT_SERVER_PORT);
  if (stsi->nloc.wPort == 0)
    stsi->nloc.wPort = RandRange(1024, 65535);

  // User password
  stsi->wPassLen = strlennull(szPassword);
  if (stsi->wPassLen > 9) stsi->wPassLen = 9;
  strncpy(stsi->szPass, szPassword, stsi->wPassLen);
  stsi->szPass[stsi->wPassLen] = '\0';

  // Randomize sequence
  wLocalSequence = (WORD)RandRange(0, 0x7fff);

  dwLocalUIN = dwUin;

  serverThreadId.hThread = (HANDLE)forkthreadex(NULL, 0, icq_serverThread, stsi, 0, &serverThreadId.dwThreadId);
}