CnC_Renegade/Code/Tools/RenegadeGR/tcpmgr.h

/*
**	Command & Conquer Renegade(tm)
**	Copyright 2025 Electronic Arts 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 3 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, see <http://www.gnu.org/licenses/>.
*/

#ifndef TCPMGR_HEADER
#define TCPMGR_HEADER

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <errno.h>
#include <string.h>
#include <assert.h>

#include "wlib/wstypes.h"
#include "wlib/wdebug.h"
#include "wlib/arraylist.h"

#ifdef _WINDOWS
#include <windows.h>
#include <winsock.h>
#include <io.h>

#else  //UNIX
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/time.h>
#include <fcntl.h>
#include <limits.h>

typedef sint32         SOCKET;
#define closesocket    close
#define SOCKET_ERROR   -1
#define INVALID_SOCKET -1
 
#endif

#ifdef AIX
#include <sys/select.h>
#endif

class TCPCon;

#define DEFAULT_PROTOCOL 0
#define INVALID_HANDLE   0


class TCPMgr {
 public:

  enum CONN_STATE
  {
    CLOSED,
    CONNECTING,
    CONNECTED
  };

  // These defines specify a system independent way to
  //	 get error codes for socket services.
  enum STATUS
  {
    OK				=	0,		 // Everything's cool
    UNKNOWN			= -1,		 // There was an error of unknown type
    ISCONN			= -2,		 // The socket is already connected
    INPROGRESS		= -3,		 // The socket is non-blocking and the operation
													 //	 isn't done yet
    ALREADY			= -4,		 // The socket is already attempting a connection
													 //	 but isn't done yet
    AGAIN			= -5,		 // Try again.
    ADDRINUSE		= -6,		 // Address already in use
    ADDRNOTAVAIL	= -7,		 // That address is not available on the remote host
    BADF				= -8,		 // Not a valid FD
    CONNREFUSED	= -9,		 // Connection was refused
    INTR				=-10,		 // Operation was interrupted
    NOTSOCK			=-11,		 // FD wasn't a socket
    PIPE				=-12,		 // That operation just made a SIGPIPE
    WOULDBLOCK		=-13,		 // That operation would block
    INVAL			=-14,	    // Invalid
    TIMEDOUT		=-15		 // Timeout
  };

  enum DIRECTION
  {
    INCOMING = 1,
    OUTGOING = 2,
    EITHER   = 3
  };

            TCPMgr();
            ~TCPMgr();

  bit8      addListener(uint32 ip, uint16 port, bit8 reuseAddr);
  bit8      removeListener(uint32 ip, uint16 port);
  bit8      getListener(uint32 ip, uint16 port, OUT SOCKET &outsock);

  bit8      connect(char *address, uint16 port, uint32 *handle);
  bit8      connect(uint32 ip, uint16 port,OUT uint32 *handle);

  bit8      getOutgoingConnection(TCPCon **conn, uint32 handle, sint32 wait_secs);
  bit8      getIncomingConnection(TCPCon **conn, uint16 port, sint32 wait_secs);

  bit8      setBufferedWrites(TCPCon *con, bit8 enabled);
  void      pumpWriters(void);  // pump the buffered writer connections

  // Static methods
  static int       wait(uint32 sec, uint32 usec, SOCKET *sockets, int count, bit8 readMode=TRUE);
  static STATUS    getStatus(void);

 private:

  SOCKET    createSocket(uint32 ip, uint16 port, bit8 reuseAddr=TRUE);
  bit8      setBlocking(SOCKET fd, bit8 block);
  bit8      getConnection(TCPCon **conn, uint32 handle, uint16 port, sint32 wait_secs, DIRECTION dir);
  void      pumpConnections(void);

  struct ListenSocket
  {
    SOCKET   fd;
    uint32   ip;
    uint32   port;
  };

  ArrayList<ListenSocket>       ListenArray_;

  ArrayList<TCPCon * >          BufferedWriters_;  // need to be pumped

  struct PendingConn
  {
    SOCKET      fd;
    uint32      ip;
    uint16      port;
    time_t      startTime;
    uint32      handle;
    CONN_STATE  state;
    bit8        incoming;
    uint32      remoteIp;
    uint16      remotePort;
  };

  ArrayList<PendingConn>  ConnectArray_;
  uint32                  HandleSequence_;
};

#endif