Clementine-audio-player-Mac.../3rdparty/gloox/socks5bytestream.cpp

156 lines
3.9 KiB
C++
Raw Normal View History

/*
Copyright (c) 2006-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 "socks5bytestream.h"
#include "bytestreamdatahandler.h"
#include "clientbase.h"
#include "connectionbase.h"
#include "connectionsocks5proxy.h"
#include "sha.h"
#include "logsink.h"
namespace gloox
{
SOCKS5Bytestream::SOCKS5Bytestream( SOCKS5BytestreamManager* manager, ConnectionBase* connection,
LogSink& logInstance, const JID& initiator, const JID& target,
const std::string& sid )
: Bytestream( Bytestream::S5B, logInstance, initiator, target, sid ),
m_manager( manager ), m_connection( 0 ), m_socks5( 0 ), m_connected( false )
{
if( connection && connection->state() == StateConnected )
m_open = true;
setConnectionImpl( connection );
}
SOCKS5Bytestream::~SOCKS5Bytestream()
{
if( m_open )
close();
if( m_socks5 )
delete m_socks5;
}
void SOCKS5Bytestream::setConnectionImpl( ConnectionBase* connection )
{
if( m_socks5 )
delete m_socks5; // deletes m_connection as well
m_connection = connection;
SHA sha;
sha.feed( m_sid );
sha.feed( m_initiator.full() );
sha.feed( m_target.full() );
m_socks5 = new ConnectionSOCKS5Proxy( this, connection, m_logInstance, sha.hex(), 0 );
}
bool SOCKS5Bytestream::connect()
{
if( !m_connection || !m_socks5 || !m_manager )
return false;
if( m_open )
return true;
StreamHostList::const_iterator it = m_hosts.begin();
for( ; it != m_hosts.end(); ++it )
{
if( ++it == m_hosts.end() )
m_connected = true;
--it; // FIXME ++it followed by --it is kinda ugly
m_connection->setServer( (*it).host, (*it).port );
if( m_socks5->connect() == ConnNoError )
{
m_proxy = (*it).jid;
m_connected = true;
return true;
}
}
m_manager->acknowledgeStreamHost( false, JID(), EmptyString );
return false;
}
bool SOCKS5Bytestream::send( const std::string& data )
{
if( !m_open || !m_connection || !m_socks5 || !m_manager )
return false;
return m_socks5->send( data );
}
ConnectionError SOCKS5Bytestream::recv( int timeout )
{
if( !m_connection || !m_socks5 || !m_manager )
return ConnNotConnected;
return m_socks5->recv( timeout );
}
void SOCKS5Bytestream::activate()
{
m_open = true;
if( m_handler )
m_handler->handleBytestreamOpen( this );
}
void SOCKS5Bytestream::close()
{
if( m_open && m_handler )
{
m_open = false;
m_connected = false;
m_socks5->disconnect();
m_handler->handleBytestreamClose( this );
}
}
void SOCKS5Bytestream::handleReceivedData( const ConnectionBase* /*connection*/, const std::string& data )
{
if( !m_handler )
return;
if( !m_open )
{
m_open = true;
m_handler->handleBytestreamOpen( this );
}
// if( !m_open && data.length() == 2 && data[0] == 0x05 && data[1] == 0x00 )
// {
// printf( "received acknowleding zero byte, stream is now open\n" );
// m_open = true;
// m_handler->handleBytestream5Open( this );
// return;
// }
if( m_open )
m_handler->handleBytestreamData( this, data );
}
void SOCKS5Bytestream::handleConnect( const ConnectionBase* /*connection*/ )
{
m_manager->acknowledgeStreamHost( true, m_proxy, m_sid );
}
void SOCKS5Bytestream::handleDisconnect( const ConnectionBase* /*connection*/, ConnectionError /*reason*/ )
{
if( m_handler && m_connected )
m_handler->handleBytestreamClose( this );
}
}