connectiontcpbase.h

/*
  Copyright (c) 2004-2017 by Jakob Schröter <js@camaya.net>
  This file is part of the gloox library. http://camaya.net/gloox

  This software is distributed under a license. The full license
  agreement can be found in the file LICENSE in this distribution.
  This software may not be copied, modified, sold or distributed
  other than expressed in the named license agreement.

  This software is distributed without any warranty.
*/


#ifndef CONNECTIONTCPBASE_H__
#define CONNECTIONTCPBASE_H__

#include "gloox.h"
#include "connectionbase.h"
#include "logsink.h"
#include "mutex.h"

#ifdef __MINGW32__
#include <ws2tcpip.h>
#endif

#include <string>

namespace gloox
{

  namespace util
  {
    class Mutex;
  }

  class GLOOX_API ConnectionTCPBase : public ConnectionBase
  {
    public:
      ConnectionTCPBase( const LogSink& logInstance, const std::string& server, int port = -1 );

      ConnectionTCPBase( ConnectionDataHandler* cdh, const LogSink& logInstance,
                         const std::string& server, int port = -1 );

      virtual ~ConnectionTCPBase();

      // reimplemented from ConnectionBase
      virtual bool send( const std::string& data );

      // reimplemented from ConnectionBase
      virtual ConnectionError receive();

      // reimplemented from ConnectionBase
      virtual void disconnect();

      // reimplemented from ConnectionBase
      virtual void cleanup();

      // reimplemented from ConnectionBase
      virtual void getStatistics( long int &totalIn, long int &totalOut );

      int socket() const { return m_socket; }

      void setSocket( int socket ) { m_cancel = false; m_state = StateConnected; m_socket = socket; }

      virtual int localPort() const;

      virtual const std::string localInterface() const;

    protected:
      ConnectionTCPBase& operator=( const ConnectionTCPBase& );
      void init( const std::string& server, int port );
      bool dataAvailable( int timeout = -1 );
      void cancel();

      const LogSink& m_logInstance;
      util::Mutex m_sendMutex;
      util::Mutex m_recvMutex;

      char* m_buf;
      int m_socket;
      long int m_totalBytesIn;
      long int m_totalBytesOut;
      const int m_bufsize;
      bool m_cancel;

  };

}

#endif // CONNECTIONTCPBASE_H__