// ---------------------------------------------------------------------------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,2007 Joe Kucera
// Copyright © 2006,2007 [sss], chaos.persei, [sin], Faith Healer, Theif, nullbie
// 
// 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$
// Revision       : $Revision: 62 $
// Last change on : $Date: 2007-10-16 14:56:05 +0300 (Вт, 16 окт 2007) $
// Last change by : $Author: chaos.persei $
//
// DESCRIPTION:
//
//  Manages main server connection, low-level communication
//
// -----------------------------------------------------------------------------

#include "icqoscar.h"



extern CRITICAL_SECTION connectionHandleMutex;
extern WORD wLocalSequence;
extern CRITICAL_SECTION localSeqMutex;
extern int icqGoingOnlineStatus;
HANDLE hServerConn;
WORD wListenPort;
WORD wLocalSequence;
static DWORD serverThreadId;
static HANDLE serverThreadHandle;
BOOL TryNextWorking;

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

static DWORD __stdcall icq_TryNextServerThread(LPVOID lp)
{
	WORD CurrServ, SrvCount, Port;
	char DBModule[64], buf[1024], buf1[64], buf2[64];
	
	TryNextWorking = 1;

	WaitForSingleObject(serverThreadHandle, INFINITE);

	mir_snprintf(DBModule, 64, "%sSrvs", gpszICQProtoName);	

	SrvCount = DBGetContactSettingWord(0, DBModule, "SrvCount", 0);
	if(SrvCount == 0) { TryNextWorking = 0; return 0; }
	CurrServ = DBGetContactSettingWord(0, DBModule, "CurrServ", 0);
	
	if(CurrServ >= SrvCount){
		DBWriteContactSettingWord(0, DBModule, "CurrServ", 0);
		icq_LogMessage(LOG_FATAL, "Server AutoChange: servers list end, exiting");
		TryNextWorking = 0;
		return 0; 
	}

	
	CurrServ = CurrServ + 1;
	
	DBWriteContactSettingWord(0, DBModule, "CurrServ", CurrServ);
	
	mir_snprintf(buf1, 64, "server%luhost", CurrServ);
	mir_snprintf(buf2, 64, "server%luport", CurrServ);
	mir_snprintf(buf, 64, "server%luuse", CurrServ);

	if(!DBGetContactSettingByte(0, DBModule, buf, 1)){
		TryNextWorking = 0;
		TryNextServer();
		return 0;
	}
	
	Port = DBGetContactSettingWord(0, DBModule, buf2, DEFAULT_SERVER_PORT);	
	

	DBWriteContactSettingWord(0, gpszICQProtoName, "OscarPort", Port);
	ICQWriteContactSettingString(NULL, "OscarServer", UniGetContactSettingUtf(NULL, DBModule, buf1, 0));
	
	icq_LogMessage(LOG_NOTE, "Server AutoChange: connection problems with current server, switching to next");

	IcqSetStatus((WPARAM)iIcqNewStatus, 0);

	TryNextWorking = 0;

	return 0;
}

void TryNextServer()
{
	if(TryNextWorking) return;

	ICQCreateThread(icq_TryNextServerThread, NULL);
}


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);
  // store server port
  info.wServerPort = infoParam->nloc.wPort;

  srand(time(NULL));

  ResetSettingsOnConnect();

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

    hServerConn = NetLib_OpenConnection(ghServerNetlibUser, NULL, &nloc);

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


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

    hServerConn = NULL;

    SetCurrentStatus(ID_STATUS_OFFLINE);

	if(bServerAutoChange)
		TryNextServer();
	else
		icq_LogUsingErrorCode(LOG_ERROR, dwError, "Unable to connect to ICQ login server");
	
    return 0;
  }


  // Initialize direct connection ports
  {
    DWORD dwInternalIP;
    BYTE bConstInternalIP = ICQGetContactSettingByte(NULL, "ConstRealIP", 0);

    info.hDirectBoundPort = NetLib_BindPort(icq_newConnectionReceived, NULL, &wListenPort, &dwInternalIP);
    if (!info.hDirectBoundPort)
    {
      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;
      if (!bConstInternalIP) ICQDeleteContactSetting(NULL, "RealIP");
    }
    else if (!bConstInternalIP)
      ICQWriteContactSettingDword(NULL, "RealIP", 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};

    info.hPacketRecver = (HANDLE)CallService(MS_NETLIB_CREATEPACKETRECVER, (WPARAM)hServerConn, 0x2400);
    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)info.hPacketRecver, (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_SafeCloseHandle(&info.hPacketRecver);

    // Close DC port
    NetLib_SafeCloseHandle(&info.hDirectBoundPort);
  }

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

  // disable auto info-update thread
  icq_EnableUserLookup(FALSE);

  // Time to shutdown
  icq_serverDisconnect(FALSE);
  if (gnCurrentStatus != ID_STATUS_OFFLINE && icqGoingOnlineStatus != ID_STATUS_OFFLINE)
  {
    if (!info.bLoggedIn)
    {
      	if(bServerAutoChange)
			TryNextServer();
		else
			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);

  // Close avatar connection if any
  StopAvatarThread();

  // 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, 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
  ratesRelease(&gRates);

  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_CloseConnection(&hServerConn, TRUE);
    LeaveCriticalSection(&connectionHandleMutex);
    
    // Not called from network thread?
    if (bBlock && GetCurrentThreadId() != serverThreadId)
    {
      while (WaitForSingleObjectEx(serverThreadHandle, INFINITE, TRUE) != WAIT_OBJECT_0);
      CloseHandle(serverThreadHandle);
    }
    else
      CloseHandle(serverThreadHandle);
  }
  else
    LeaveCriticalSection(&connectionHandleMutex);
}



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

    // Rates management
    EnterCriticalSection(&ratesMutex);
    ratesPacketSent(gRates, pPacket);
    LeaveCriticalSection(&ratesMutex);

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



typedef struct icq_packet_async_s
{
  icq_packet packet;
  
} icq_packet_async;

static DWORD __stdcall sendPacketAsyncThread(icq_packet_async* pArgs)
{
  sendServPacket(&pArgs->packet);

  SAFE_FREE(&pArgs);
  return 0;
}


void sendServPacketAsync(icq_packet *packet)
{
  icq_packet_async *pArgs;
  
  pArgs = (icq_packet_async*)SAFE_MALLOC(sizeof(icq_packet_async)); // This will be freed in the new thread
  memcpy(&pArgs->packet, packet, sizeof(icq_packet));

  ICQCreateThread(sendPacketAsyncThread, pArgs);
}



int IsServerOverRate(WORD wFamily, WORD wCommand, int nLevel)
{
  WORD wGroup;
  int result = FALSE;

  EnterCriticalSection(&ratesMutex);
  wGroup = ratesGroupFromSNAC(gRates, wFamily, wCommand);

  // check if the rate is not over specified level
  if (ratesNextRateLevel(gRates, wGroup) < ratesGetLimitLevel(gRates, wGroup, nLevel))
    result = TRUE;

  LeaveCriticalSection(&ratesMutex);

  return result;
}



void icq_login(const char* szPassword)
{
  DBVARIANT dbvServer = {DBVT_DELETED};
  char szServer[MAX_PATH];
  serverthread_start_info* stsi;
  DWORD dwUin;
  //MessageBox(0, szPassword, 0, MB_OK);
  if (bImageRequested){
	  icq_serverDisconnect(0);
	  icq_regNewUin = 0;
	  bImageRequested = 0;
	  EnableWindow(GetDlgItem(hwndRegImageDialog, IDC_REGISTER), FALSE);
  }

  dwUin = ICQGetContactSettingUIN(NULL);
  stsi = (serverthread_start_info*)SAFE_MALLOC(sizeof(serverthread_start_info));

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

  serverThreadHandle = ICQCreateThreadEx(icq_serverThread, stsi, &serverThreadId);
}