NodeAddress.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 "Imports.h"
#include "NodeAddress.h"

using namespace Protocols::Transports;

// If we had more special addresses, we could have put them all in an enum.
const NodeAddress NodeAddress::Null = NodeAddress();

NodeAddress::NodeAddress (const QHostAddress &hostAddress, quint16 hostPort)
 :  hostName_(), hostAddress_(), hostPort_ (hostPort)
{
    setHostAddress (hostAddress);
}

NodeAddress::NodeAddress (const QString &hostName, quint16 hostPort)
 :  hostName_(), hostAddress_(), hostPort_ (hostPort)
{
    setHostName (hostName);
}


NodeAddress::NodeAddress (const QString &addressPortString)
 :  hostName_(), hostAddress_(), hostPort_ (0)
{
    // fot the IPv6 addresses as well!
    int pos = addressPortString.indexOf (':');
    if (pos != -1) {
        hostAddress_.setAddress (addressPortString.left (pos));
        if (hostAddress_.isNull())
            hostName_ = addressPortString.left (pos);
        int length = addressPortString.length();
        while (addressPortString.at (pos + 1).isSpace())
            pos++;
        int i = 0;
        for (i = pos + 1; i < length && addressPortString.at(i).isDigit(); i++)
            ;
        hostPort_ = addressPortString.mid (pos + 1, i - pos - 1).toInt();
    }
}

NodeAddress::~NodeAddress()
{
}

bool NodeAddress::operator == (const NodeAddress &nodeAddress) const
{
    return toString() == nodeAddress.toString();
}

bool NodeAddress::operator < (const NodeAddress &nodeAddress) const
{
    // \todo Is that OK?
    return hostAddress_.toString() < nodeAddress.hostAddress_.toString();
}

uint Protocols::Transports::qHash (const NodeAddress &nodeAddress)
{
    return ::qHash (nodeAddress.toString());
}

QString NodeAddress::toString() const
{
    if (hostName_.length() > 0)
        return QString ("%1:%2").arg (hostName_).arg (hostPort_);
    else
        return QString ("%1:%2").arg (hostAddress_.toString()).arg (hostPort_);
}

QDataStream & Protocols::Transports::operator << (QDataStream &stream,
                                                 const NodeAddress &nodeAddress)
{
    if (nodeAddress.hostName().length() > 0) {
        stream << static_cast <unsigned char> (1);
        stream << nodeAddress.hostName();
    } else {
        stream << static_cast <unsigned char> (0);
        stream << nodeAddress.hostAddress().toString();
    }
    stream << nodeAddress.hostPort();
    return stream;
}

QDataStream & Protocols::Transports::operator >> (QDataStream &stream,
                                                 NodeAddress &nodeAddress)
{
    unsigned char isName = 0;
    QString address;
    quint16 port = static_cast <quint16> (-1);

    stream >> isName;
    stream >> address;
    stream >> port;

    if (isName == 1)
        nodeAddress.setHostName (address);
    else
        nodeAddress.setHostAddress (QHostAddress (address));
    nodeAddress.setHostPort (port);
    return stream;
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.