TcpConnection.cpp

Go to the documentation of this file.

/*

Copyright (C) 2005-2007 by Peter Dimov.

This file is part of Calitko (http://www.calitko.org).

Calitko is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

Calitko is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Calitko; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/

#include "Qt.h"
#include "Imports.h"
#include "TcpConnection.h"

namespace Protocols {
namespace Transports {

enum Constants
{
    ConnectTimeout = 10000
};

} // namespace Transports;
} // namespace Protocols;

using namespace Protocols::Transports;

TcpConnection::TcpConnection (QTcpSocket *s)
 :  socket (s), timer (0)
{
    qDebug() << "TcpConnection::TcpConnection";

    if (socket == 0)
        socket = new QTcpSocket;

    if (socket->state() == QAbstractSocket::UnconnectedState)
        setState (DisconnectedState);
    else if (socket->state() == QAbstractSocket::ConnectingState
                || socket->state() == QAbstractSocket::HostLookupState)
        setState (Connection::ConnectingState);
    else if (socket->state() == QAbstractSocket::ConnectedState) {
        // \todo Incoming connections (from QTcpServer) should also be unbuffered.
        setState (Connection::ConnectedState);
        setOpenMode (QIODevice::ReadWrite | QIODevice::Unbuffered);
    } else
        setState (Connection::DisconnectingState);
    // \todo We should never get a socket that is bound or listening.

    setRemoteNodeAddress (NodeAddress (socket->peerAddress(), socket->peerPort()));

    QObject::connect (socket,   SIGNAL  (connected()),
                      this,     SLOT    (slotConnected()));
    QObject::connect (socket,   SIGNAL  (disconnected()),
                      this,     SLOT    (slotDisconnected()));
    QObject::connect (socket,   SIGNAL  (error(QAbstractSocket::SocketError)),
                      this,     SLOT    (slotError(QAbstractSocket::SocketError)));
    QObject::connect (socket,   SIGNAL  (readyRead()),
                      this,     SIGNAL  (readyRead()));
    QObject::connect (socket,   SIGNAL  (bytesWritten (qint64)),
                      this,     SIGNAL  (bytesWritten (qint64)));

    qDebug() << "TcpConnection::TcpConnection exit";
}

TcpConnection::~TcpConnection()
{
    qDebug() << "TcpConnection::~TcpConnection";
    //Q_ASSERT (state() == DisconnectedState);

    if (socket)
        socket->disconnect (this);
    delete socket;
    delete timer;
    qDebug() << "TcpConnection::~TcpConnection exit";
}

void TcpConnection::startTimer()
{
    timer = new QTimer;
    connect (timer, SIGNAL (timeout()), this, SLOT (connectTimeout()));
    timer->setSingleShot (true);
    timer->start (ConnectTimeout);
}

void TcpConnection::stopTimer()
{
    delete timer;
    timer = 0;
}

void TcpConnection::doConnectToNode (const NodeAddress &nodeAddress)
{
    Q_ASSERT (!nodeAddress.isNull());

    socket->connectToHost (nodeAddress.hostAddress(), nodeAddress.hostPort(),
                           QIODevice::ReadWrite | QIODevice::Unbuffered);
    startTimer();
}

void TcpConnection::doDisconnectFromNode()
{
    socket->disconnectFromHost();
}

void TcpConnection::slotConnected()
{
    stopTimer();

    setState (ConnectedState);
    setOpenMode (QIODevice::ReadWrite | QIODevice::Unbuffered);

    if (remoteNodeAddress() == NodeAddress::Null) {
        NodeAddress address (socket->peerAddress(), socket->peerPort());
        setRemoteNodeAddress (address);
    }
    emit connectionEstablished();
}

void TcpConnection::connectTimeout()
{
    qDebug() << "TcpConnection::connectTimeout";
    Q_ASSERT (state() == ConnectingState);

    setState (TimeoutState);
    socket->abort();
    stopTimer();

    qDebug() << "TcpConnection::connectTimeout exit";
}

void TcpConnection::slotDisconnected()
{
    qDebug() << "TcpConnection::slotDisconnected";
    if (state() == TimeoutState)
        emit connectionFailed();
    else {
        setState (DisconnectedState);
        emit connectionClosed();
    }
    setOpenMode (QIODevice::NotOpen);
    qDebug() << "TcpConnection::slotDisconnectedExit";
}

void TcpConnection::slotError(QTcpSocket::SocketError)
{
    qDebug() << "TcpConnection::slotError";
    socket->close();
    if (state() == ConnectingState) {
        setState (DisconnectedState);
        // Stop the connection timeout timer:
        stopTimer();
        emit connectionFailed();
    } /* \todo That is the Disconnecting... bug!
      else if (socket->error() == QTcpSocket::RemoteHostClosedError) {
        slotDisconnected();
    }*/
    setOpenMode (QIODevice::NotOpen);
    qDebug() << "TcpConnection::slotError exit";
}

qint64 TcpConnection::readData (char * data, qint64 maxSize)
{
    return socket->read (data, maxSize);
}

qint64 TcpConnection::writeData (const char * data, qint64 maxSize)
{
    return socket->write (data, maxSize);
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.