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 General Public License as published by
    the Free Software Foundation, either version 3 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Classified Ads.  If not, see <http://www.gnu.org/licenses/>.
*/

#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<QPair<QHostAddress,int> > 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 void 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() ;
protected:
  bool insertNodeToDb(Node& aNode) ;
  bool updateNodeInDb(Node& aNode) ;
  void retrieveListOfHotConnections() ;
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 ;
signals:
  void error(MController::CAErrorSituation aError,
             const QString& aExplanation) ;
private:
  MController *iController  ;
  Hash* iFingerPrintOfThisNode ; 
  QSslCertificate* iThisNodeCert ; 
  QSslKey* iThisNodeKey ; 
  QList<QPair<QHostAddress,int> > iHotAddresses ;
  const Model& iModel ;
  QString iDataDir ;
  QList<Node *> iConnectionWishList ; 
} ;
#endif