nodemodel.h

Go to the documentation of this file.

/*    -*-C++-*- -*-coding: utf-8-unix;-*-
    Classified Ads is Copyright (c) Antti Jarvinen 2013.

    This file is part of Classified Ads.

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

    Classified Ads 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
    Lesser General Public License for more details.

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

#ifndef CLASSIFIED_NODEMODEL_H
#define CLASSIFIED_NODEMODEL_H
#include <QSqlDatabase>
#include <QObject>
#include "../mcontroller.h" // because enum from there is needed
#include "mnodemodelprotocolinterface.h"
#include "../net/connection.h"
#include <QHostAddress>
#include <QPair>
#include "datamodelbase.h"

class QSqlDatabase ;
class QMutex ;
class Hash ;
class QSslCertificate ;
class QSslKey ;
class Model ;

class NodeModel : public ModelBase,
    public MNodeModelProtocolInterface {
    Q_OBJECT

public:
    NodeModel(MController *aController, const Model &aModel) ;
    ~NodeModel() ;
    virtual Hash& nodeFingerPrint(); 
    virtual int listenPortOfThisNode() ; 
    virtual void setListenPortOfThisNode(int port) ;
    virtual const QSslCertificate& nodeCert() const ;
    virtual const QSslKey& nodeKey() const ;
    virtual QByteArray* getNextItemToSend(Connection& aConnection) ;

    virtual Node* nodeByHash(const Hash& aHash) ;
    virtual QList<HostConnectQueueItem> getHotAddresses() ;
    QList<Node *>* getHotNodes(int aMaxNodes) ;
    virtual bool nodeGreetingReceived(Node& aNode ,
                                      bool aWasInitialGreeting = false ) ;
    virtual QList<Node *>* getNodesAfterHash(const Hash& aHash,
            unsigned aMaxNodes,
            int aMaxInactivityMinutes = -1 ) ;
    bool updateNodeLastConnectTimeInDb(Node& aNode) ;
    virtual bool updateNodeLastMutualConnectTimeInDb(const Hash& aNodeFp,
            quint32 aTime ) ;
    virtual QList<Node *>* getNodesBeforeHash(const Hash& aHash,
            unsigned aMaxNodes) ;
    virtual void       closeOldestInactiveConnection() ;
    QString dataDir() ;
    virtual void addNodeFromBroadcast(const Hash& aNodeFingerPrint,
                                      const QHostAddress& aAddr,
                                      int aPort ) ;
    virtual bool addNodeToConnectionWishList(Node* aNode) ;
    virtual bool addNodeToConnectionWishList(const Hash& aNode) ;

    virtual Node* nextConnectionWishListItem() ;
    virtual bool isNodeAlreadyConnected(const Node& aNode) const ;
    virtual bool isNodeAlreadyConnected(const Hash& aHash) const ;
    virtual Hash bucketEndHash(const Hash& aFingerPrintOfNodeAsking) ;
    virtual void setDnsName(QString aName) ;
    virtual QString getDnsName() ;
    virtual void offerNodeToRecentlyFailedList(const Hash& aFailedNodeHash) ;
protected:
    bool insertNodeToDb(Node& aNode) ;
    bool updateNodeInDb(Node& aNode) ;
    void retrieveListOfHotConnections() ;
    void timerEvent(QTimerEvent *event);
private:
    bool openOrCreateSSLCertificate() ; 
    bool saveSslCertToDb(const QByteArray& aCert) ;
    bool saveSslKeyToDb(const QByteArray& aKey) ;
    bool loadSslCertFromDb() ;
    bool loadSslKeyFromDb() ;
    void deleteOldestConnectedNode(); 
    void countNodes() ; 
    Q_IPV6ADDR ipv6AddrFromUints(quint32 aLeastSignificant,
                                 quint32 aLessSignificant,
                                 quint32 aMoreSignificant,
                                 quint32 aMostSignificant ) const ;
    void removeNodeFromRecentlyFailedList(const Hash& aConnectedHostFingerPrint);
signals:
    void error(MController::CAErrorSituation aError,
               const QString& aExplanation) ;
private:
    MController *iController  ;
    Hash* iFingerPrintOfThisNode ; 
    QSslCertificate* iThisNodeCert ; 
    QSslKey* iThisNodeKey ; 
    QList<HostConnectQueueItem> iHotAddresses ;
    const Model& iModel ;
    QString iDataDir ;
    QList<Node *> iConnectionWishList ;
    QList<QPair<Hash,unsigned> > iRecentlyFailedNodes ;
    int iTimerId ; 
} ;
#endif