connection.h

/*
  Copyright (c) 2004-2006 by Jakob Schroeter <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 CONNECTION_H__
#define CONNECTION_H__

#ifdef WIN32
# include "../config.h.win"
#else
# include "config.h"
#endif

#include "gloox.h"
#include "logsink.h"

#include <string>

#if defined( HAVE_OPENSSL )
# define USE_OPENSSL
# include <openssl/ssl.h>
# define HAVE_TLS
#elif defined( HAVE_GNUTLS )
# define USE_GNUTLS
# include <gnutls/gnutls.h>
# include <gnutls/x509.h>
# define HAVE_TLS
#endif

namespace gloox
{

  class Compression;
  class Packet;
  class Parser;

  class GLOOX_API Connection
  {
    public:
      Connection( Parser *parser, const LogSink& logInstance, const std::string& server,
                  int port = -1 );

      virtual ~Connection();

      ConnectionState connect();

      ConnectionError recv( int timeout = -1 );

      void send( const std::string& data );

      ConnectionError receive();

      void disconnect( ConnectionError e );

      bool isSecure() const { return m_secure; };

      ConnectionState state() const { return m_state; };

      int fileDescriptor();

#ifdef HAVE_ZLIB

      bool initCompression( StreamFeature method );

      void enableCompression();
#endif

#ifdef HAVE_TLS

      bool tlsHandshake();

      void setCACerts( const StringList& cacerts ) { m_cacerts = cacerts; };

      const CertInfo& fetchTLSInfo() const { return m_certInfo; };

      void setClientCert( const std::string& clientKey, const std::string& clientCerts );
#endif

    private:
      void cancel();
      void cleanup();

#if defined( USE_GNUTLS )

      bool verifyAgainstCAs( gnutls_x509_crt_t cert, gnutls_x509_crt_t *CAList, int CAListSize );
      bool verifyAgainst( gnutls_x509_crt_t cert, gnutls_x509_crt_t issuer );

      gnutls_session_t m_session;
      gnutls_certificate_credentials m_credentials;

#elif defined( USE_OPENSSL )
      SSL *m_ssl;
#endif

      StringList m_cacerts;
      std::string m_clientKey;
      std::string m_clientCerts;

      Parser *m_parser;
      ConnectionState m_state;
      CertInfo m_certInfo;
      ConnectionError m_disconnect;
      const LogSink& m_logInstance;
      Compression *m_compression;

      char *m_buf;
      std::string m_server;
      int m_port;
      int m_socket;
      const int m_bufsize;
      bool m_cancel;
      bool m_secure;
      bool m_fdRequested;
      bool m_enableCompression;
  };

}

#endif // CONNECTION_H__

---