160 lines
3.7 KiB
C++
160 lines
3.7 KiB
C++
/*
|
|
Copyright (c) 2004-2009 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.
|
|
*/
|
|
|
|
|
|
|
|
#include "gloox.h"
|
|
|
|
#include "connectiontcpclient.h"
|
|
#include "dns.h"
|
|
#include "logsink.h"
|
|
#include "mutexguard.h"
|
|
|
|
#ifdef __MINGW32__
|
|
# include <winsock.h>
|
|
#endif
|
|
|
|
#if ( !defined( _WIN32 ) && !defined( _WIN32_WCE ) ) || defined( __SYMBIAN32__ )
|
|
# include <sys/types.h>
|
|
# include <sys/socket.h>
|
|
# include <sys/select.h>
|
|
# include <unistd.h>
|
|
#elif ( defined( _WIN32 ) || defined( _WIN32_WCE ) ) && !defined( __SYMBIAN32__ )
|
|
# include <winsock.h>
|
|
#endif
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
|
|
namespace gloox
|
|
{
|
|
|
|
ConnectionTCPClient::ConnectionTCPClient( const LogSink& logInstance,
|
|
const std::string& server, int port )
|
|
: ConnectionTCPBase( logInstance, server, port )
|
|
{
|
|
}
|
|
|
|
ConnectionTCPClient::ConnectionTCPClient( ConnectionDataHandler* cdh, const LogSink& logInstance,
|
|
const std::string& server, int port )
|
|
: ConnectionTCPBase( cdh, logInstance, server, port )
|
|
{
|
|
}
|
|
|
|
|
|
ConnectionTCPClient::~ConnectionTCPClient()
|
|
{
|
|
}
|
|
|
|
ConnectionBase* ConnectionTCPClient::newInstance() const
|
|
{
|
|
return new ConnectionTCPClient( m_handler, m_logInstance, m_server, m_port );
|
|
}
|
|
|
|
ConnectionError ConnectionTCPClient::connect()
|
|
{
|
|
m_sendMutex.lock();
|
|
// FIXME CHECKME
|
|
if( !m_handler )
|
|
{
|
|
m_sendMutex.unlock();
|
|
return ConnNotConnected;
|
|
}
|
|
|
|
if( m_socket >= 0 && m_state > StateDisconnected )
|
|
{
|
|
m_sendMutex.unlock();
|
|
return ConnNoError;
|
|
}
|
|
|
|
m_state = StateConnecting;
|
|
|
|
if( m_socket < 0 )
|
|
{
|
|
if( m_port == -1 )
|
|
m_socket = DNS::connect( m_server, m_logInstance );
|
|
else
|
|
m_socket = DNS::connect( m_server, m_port, m_logInstance );
|
|
}
|
|
|
|
m_sendMutex.unlock();
|
|
|
|
if( m_socket < 0 )
|
|
{
|
|
switch( m_socket )
|
|
{
|
|
case -ConnConnectionRefused:
|
|
m_logInstance.err( LogAreaClassConnectionTCPClient,
|
|
m_server + ": connection refused" );
|
|
break;
|
|
case -ConnDnsError:
|
|
m_logInstance.err( LogAreaClassConnectionTCPClient,
|
|
m_server + ": host not found" );
|
|
break;
|
|
default:
|
|
m_logInstance.err( LogAreaClassConnectionTCPClient,
|
|
"Unknown error condition" );
|
|
break;
|
|
}
|
|
m_handler->handleDisconnect( this, (ConnectionError)-m_socket );
|
|
return (ConnectionError)-m_socket;
|
|
}
|
|
else
|
|
{
|
|
m_state = StateConnected;
|
|
}
|
|
|
|
m_cancel = false;
|
|
m_handler->handleConnect( this );
|
|
return ConnNoError;
|
|
}
|
|
|
|
ConnectionError ConnectionTCPClient::recv( int timeout )
|
|
{
|
|
m_recvMutex.lock();
|
|
|
|
if( m_cancel || m_socket < 0 )
|
|
{
|
|
m_recvMutex.unlock();
|
|
return ConnNotConnected;
|
|
}
|
|
|
|
if( !dataAvailable( timeout ) )
|
|
{
|
|
m_recvMutex.unlock();
|
|
return ConnNoError;
|
|
}
|
|
|
|
int size = static_cast<int>( ::recv( m_socket, m_buf, m_bufsize, 0 ) );
|
|
if( size > 0 )
|
|
m_totalBytesIn += size;
|
|
|
|
m_recvMutex.unlock();
|
|
|
|
if( size <= 0 )
|
|
{
|
|
ConnectionError error = ( size ? ConnIoError : ConnStreamClosed );
|
|
if( m_handler )
|
|
m_handler->handleDisconnect( this, error );
|
|
return error;
|
|
}
|
|
|
|
m_buf[size] = '\0';
|
|
|
|
if( m_handler )
|
|
m_handler->handleReceivedData( this, std::string( m_buf, size ) );
|
|
|
|
return ConnNoError;
|
|
}
|
|
|
|
}
|