Diffstat (limited to 'noncore/comm/keypebble/krfbconnection.cpp') (more/less context) (ignore whitespace changes)
| -rw-r--r-- | noncore/comm/keypebble/krfbconnection.cpp | 242 |
1 files changed, 242 insertions, 0 deletions
diff --git a/noncore/comm/keypebble/krfbconnection.cpp b/noncore/comm/keypebble/krfbconnection.cpp new file mode 100644 index 0000000..c413186 --- a/dev/null +++ b/noncore/comm/keypebble/krfbconnection.cpp @@ -0,0 +1,242 @@ +#include <assert.h> +#include <qsocket.h> +#include <qtimer.h> +#include <string.h> + +#include "krfbconnection.h" +#include "krfblogin.h" +#include "krfboptions.h" +#include "krfbdecoder.h" +#include "krfbbuffer.h" + +KRFBConnection::KRFBConnection( QObject *parent ) + : QObject( parent, "KRFBConnection" ) +{ + portBase_ = 5900; + currentState_ = Disconnected; + sock = 0; + minData_ = 0; + options_ = new KRFBOptions(); + updater = 0; + decoder_ = 0; + buffer_ = 0; +} + +KRFBConnection::~KRFBConnection() +{ + if ( ( currentState_ != Disconnected ) && ( currentState_ != Disconnecting ) && sock ) { + disconnectDone(); + } + delete options_; +} + +void KRFBConnection::connectTo( const QCString &host, int display ) +{ + if ( currentState_ != Disconnected ); + disconnect(); + + this->host_= host; + this->display_ = display; + + sock = new QSocket( this, "rfbSocket" ); + CHECK_PTR( sock ); + + // Connect to something to notice connection or error + connect( sock, SIGNAL( error( int ) ), SLOT( gotSocketError( int ) ) ); + connect( sock, SIGNAL( connected() ), SLOT( gotSocketConnection() ) ); + + qWarning( "Connecting..." ); + + currentState_ = Connecting; + sock->connectToHost( host_, portBase_ + display_ ); +} + +void KRFBConnection::disconnect() +{ + qWarning( "Disconnecting from server" ); + + if ( ( currentState_ != Disconnected ) + && ( currentState_ != Disconnecting ) + && sock ) { + currentState_ = Disconnecting; + + connect( sock, SIGNAL( delayedCloseFinished() ), SLOT( disconnectDone() ) ); + sock->close(); + + if ( sock->state() != QSocket::Closing ) + disconnectDone(); + } +} + +void KRFBConnection::disconnectDone() +{ + qWarning( "KRFBConnection disconnected" ); + currentState_ = Disconnected; + delete sock; + sock = 0; + minData_ = 0; + delete updater; + delete decoder_; + delete buffer_; + emit disconnected(); +} + +void KRFBConnection::gotSocketConnection() +{ + currentState_ = LoggingIn; + + qWarning( "Connected, logging in..." ); + + static QString statusMsg = tr( "Connected" ); + emit statusChanged( statusMsg ); + + // Do some login stuff + login = new KRFBLogin( this ); +} + +void KRFBConnection::gotRFBConnection() +{ + qWarning( "Logged into server" ); + + currentState_ = Connected; + emit connected(); + + // Create the decoder and start doing stuff + decoder_ = new KRFBDecoder( this ); + CHECK_PTR( decoder_ ); + + buffer_ = new KRFBBuffer( decoder_, this, "RFB Buffer" ); + CHECK_PTR( buffer_ ); + decoder_->setBuffer( buffer_ ); + + connect( decoder_, SIGNAL( status( const QString & ) ), + this, SIGNAL( statusChanged( const QString & ) ) ); + emit loggedIn(); + + decoder_->start(); + + updater = new QTimer; + connect( updater, SIGNAL( timeout() ), SLOT( updateTimer() ) ); + updater->start( options_->updateRate ); +} + +void KRFBConnection::gotSocketError( int errno ) +{ + currentState_ = Error; + + // Do some error handling stuff + qWarning( "KRFBConnection: Socket error %d", errno ); + + static QString refused = tr( "Connection Refused" ); + static QString host = tr( "Host not found" ); + static QString read = tr( "Read Error: QSocket reported an error reading\n" + "data, the remote host has probably dropped the\n" + "connection." ); + static QString confused = tr( "QSocket reported an invalid error code" ); + + QString msg; + switch ( errno ) { + case QSocket::ErrConnectionRefused: + msg = refused; + break; + case QSocket::ErrHostNotFound: + msg = host; + break; + case QSocket::ErrSocketRead: + msg = read; + break; + default: + msg = confused; + }; + + QObject::disconnect( sock, SIGNAL( readyRead() ), this, SLOT( gotMoreData() ) ); + delete sock; + sock = 0; + currentState_ = Disconnected; + + emit error( msg ); +} + +void KRFBConnection::gotMoreData() +{ + assert( minData_ > 0 ); + + if ( sock->size() >= minData_ ) { + minData_ = 0; + QObject::disconnect( sock, SIGNAL( readyRead() ), this, SLOT( gotMoreData() ) ); + emit gotEnoughData(); + } +} + +void KRFBConnection::waitForData( unsigned int sz ) +{ + assert( minData_ == 0 ); + assert( sz > 0 ); + assert( currentState_ != Error ); + + if ( sock->size() >= sz ) { + // qWarning( "No need to wait for data" ); + emit gotEnoughData(); + } + else { + // qWarning( "Waiting for %u bytes", sz ); + + minData_ = sz; + connect( sock, SIGNAL( readyRead() ), SLOT( gotMoreData() ) ); + } +} + +int KRFBConnection::read( void *buf, int sz ) +{ + return sock->readBlock( (char *) buf, sz ); +} + +int KRFBConnection::write( void *buf, int sz ) +{ + return sock->writeBlock( (const char *) buf, sz ); +} + +KRFBConnection::State KRFBConnection::state() const +{ + return currentState_; +} + +void KRFBConnection::setPortBase( int base ) +{ + portBase_ = base; +} + +int KRFBConnection::portBase() const +{ + return portBase_; +} + +void KRFBConnection::setPassword( const QCString &pass ) +{ + this->pass_ = pass; +} + +void KRFBConnection::updateTimer() +{ + decoder_->sendUpdateRequest( true ); +} + +void KRFBConnection::refresh() +{ + decoder_->sendUpdateRequest( false ); +} + +void KRFBConnection::sendCutText( const QString &text ) +{ + decoder_->sendCutEvent( text ); +} + +const QUrl &KRFBConnection::url() +{ + url_.setProtocol( "vnc" ); + url_.setPort( display() ); + url_.setHost( host() ); + url_.setPath( "/" ); + + return url_; +} |